=================================================================== RCS file: /home/cvs/OpenXM_contrib2/asir2000/io/sio.c,v retrieving revision 1.1 retrieving revision 1.9.2.4 diff -u -p -r1.1 -r1.9.2.4 --- OpenXM_contrib2/asir2000/io/sio.c 1999/12/03 07:39:11 1.1 +++ OpenXM_contrib2/asir2000/io/sio.c 2000/11/11 06:37:08 1.9.2.4 @@ -1,6 +1,54 @@ -/* $OpenXM: OpenXM/src/asir99/io/sio.c,v 1.3 1999/11/18 09:00:38 noro Exp $ */ +/* + * Copyright (c) 1994-2000 FUJITSU LABORATORIES LIMITED + * All rights reserved. + * + * FUJITSU LABORATORIES LIMITED ("FLL") hereby grants you a limited, + * non-exclusive and royalty-free license to use, copy, modify and + * redistribute, solely for non-commercial and non-profit purposes, the + * computer program, "Risa/Asir" ("SOFTWARE"), subject to the terms and + * conditions of this Agreement. For the avoidance of doubt, you acquire + * only a limited right to use the SOFTWARE hereunder, and FLL or any + * third party developer retains all rights, including but not limited to + * copyrights, in and to the SOFTWARE. + * + * (1) FLL does not grant you a license in any way for commercial + * purposes. You may use the SOFTWARE only for non-commercial and + * non-profit purposes only, such as academic, research and internal + * business use. + * (2) The SOFTWARE is protected by the Copyright Law of Japan and + * international copyright treaties. If you make copies of the SOFTWARE, + * with or without modification, as permitted hereunder, you shall affix + * to all such copies of the SOFTWARE the above copyright notice. + * (3) An explicit reference to this SOFTWARE and its copyright owner + * shall be made on your publication or presentation in any form of the + * results obtained by use of the SOFTWARE. + * (4) In the event that you modify the SOFTWARE, you shall notify FLL by + * e-mail at risa-admin@sec.flab.fujitsu.co.jp of the detailed specification + * for such modification or the source code of the modified part of the + * SOFTWARE. + * + * THE SOFTWARE IS PROVIDED AS IS WITHOUT ANY WARRANTY OF ANY KIND. FLL + * MAKES ABSOLUTELY NO WARRANTIES, EXPRESSED, IMPLIED OR STATUTORY, AND + * EXPRESSLY DISCLAIMS ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS + * FOR A PARTICULAR PURPOSE OR NONINFRINGEMENT OF THIRD PARTIES' + * RIGHTS. NO FLL DEALER, AGENT, EMPLOYEES IS AUTHORIZED TO MAKE ANY + * MODIFICATIONS, EXTENSIONS, OR ADDITIONS TO THIS WARRANTY. + * UNDER NO CIRCUMSTANCES AND UNDER NO LEGAL THEORY, TORT, CONTRACT, + * OR OTHERWISE, SHALL FLL BE LIABLE TO YOU OR ANY OTHER PERSON FOR ANY + * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, PUNITIVE OR CONSEQUENTIAL + * DAMAGES OF ANY CHARACTER, INCLUDING, WITHOUT LIMITATION, DAMAGES + * ARISING OUT OF OR RELATING TO THE SOFTWARE OR THIS AGREEMENT, DAMAGES + * FOR LOSS OF GOODWILL, WORK STOPPAGE, OR LOSS OF DATA, OR FOR ANY + * DAMAGES, EVEN IF FLL SHALL HAVE BEEN INFORMED OF THE POSSIBILITY OF + * SUCH DAMAGES, OR FOR ANY CLAIM BY ANY OTHER PARTY. EVEN IF A PART + * OF THE SOFTWARE HAS BEEN DEVELOPED BY A THIRD PARTY, THE THIRD PARTY + * DEVELOPER SHALL HAVE NO LIABILITY IN CONNECTION WITH THE USE, + * PERFORMANCE OR NON-PERFORMANCE OF THE SOFTWARE. + * $OpenXM: OpenXM_contrib2/asir2000/io/sio.c,v 1.9.2.3 2000/11/10 14:57:24 maekawa Exp $ +*/ #if INET #include "ca.h" +#include "setjmp.h" #include "ox.h" #if defined(VISUAL) #include @@ -17,6 +65,7 @@ extern int little_endian; +int I_am_server; struct IOFP iofp[MAXIOFP]; #if !defined(_PA_RISC1_1) @@ -37,17 +86,18 @@ void getremotename(s,name) int s; char *name; { - struct sockaddr_in peer; - struct hostent *hp; - int peerlen; + struct sockaddr_storage ss; + struct sockaddr *sa; + socklen_t len; + char host[NI_MAXHOST]; + int rs; - peerlen = sizeof(peer); - getpeername(getremotesocket(s),(struct sockaddr *)&peer,&peerlen); - hp = gethostbyaddr((char *)&peer.sin_addr,sizeof(struct in_addr),AF_INET); - if ( hp ) - strcpy(name,hp->h_name); - else - strcpy(name,(char *)inet_ntoa(peer.sin_addr)); + rs = getremotesocket(s); + len = sizeof(ss); + getpeername(rs, (struct sockaddr *)&ss, &len); + sa = (struct sockaddr *)&ss; + getnameinfo(sa, sa->sa_len, host, sizeof(host), NULL, 0, 0); + strcpy(name, host); } int generate_port(use_unix,port_str) @@ -59,11 +109,14 @@ char *port_str; unsigned int port; static int count=0; +#if !defined(VISUAL) if ( use_unix ) { sprintf(port_str,"/tmp/ox%02x.XXXXXX",count); count++; mktemp(port_str); - } else { + } else +#endif + { port = ((unsigned int)mt_genrand()+(unsigned int)get_current_time()) %(65536-1024)+1024; sprintf(port_str,"%d",port); @@ -74,61 +127,63 @@ int try_bind_listen(use_unix,port_str) int use_unix; char *port_str; { - struct sockaddr_in sin; - struct sockaddr_un s_un; - struct sockaddr *saddr; - int len; - int service; + struct addrinfo hints, *res, *ai; + int s, error; + char *errstr; - if ( use_unix ) { - service = socket(AF_UNIX, SOCK_STREAM, 0); - if (service < 0) { - perror("in socket"); - return -1; - } - s_un.sun_family = AF_UNIX; - strcpy(s_un.sun_path,port_str); -#if defined(__FreeBSD__) - len = SUN_LEN(&s_un); - s_un.sun_len = len+1; /* XXX */ + memset(&hints, 0, sizeof(hints)); +#if defined(VISUAL) + hints.ai_family = PF_UNSPEC; #else - len = strlen(s_un.sun_path)+sizeof(s_un.sun_family); -#endif - saddr = (struct sockaddr *)&s_un; - } else { - service = socket(AF_INET, SOCK_STREAM, 0); - if ( service < 0 ) { - perror("in socket"); - return -1; + if (use_unix) + hints.ai_family = PF_UNIX; + else + hints.ai_family = PF_UNSPEC; +#endif /* VISUAL */ + hints.ai_socktype = SOCK_STREAM; + + error = getaddrinfo(NULL, port_str, &hints, &res); + if (error) { + warnx("try_bind_listen(): %s", gai_strerror(error)); + return (-1); + } + + s = -1; + for (ai = res ; ai != NULL ; ai = ai->ai_next) { + if ((s = socket(ai->ai_family, ai->ai_socktype, + ai->ai_protocol)) < 0 ) { + errstr = "in socket"; + continue; } - sin.sin_family = AF_INET; sin.sin_addr.s_addr = INADDR_ANY; - sin.sin_port = htons(atoi(port_str)); - len = sizeof(sin); - saddr = (struct sockaddr *)&sin; + + if (bind(s, ai->ai_addr, ai->ai_addrlen) < 0) { + errstr = "in bind"; + close(s); + s = -1; + continue; + } + + if (listen(s, SOCKQUEUELENGTH) < 0) { + errstr = "in listen"; + close(s); + s = -1; + continue; + } + + /* established connection */ + break; } - if (bind(service, saddr, len) < 0) { - perror("in bind"); - close(service); - return -1; - } - if (getsockname(service,saddr, &len) < 0) { - perror("in getsockname"); - close(service); - return -1; - } - if (listen(service, SOCKQUEUELENGTH) < 0) { - perror("in listen"); - close(service); - return -1; - } - return service; + freeaddrinfo(res); + + if (s < 0) + perror(errstr); + return (s); } /* try to accept a connection request Input - af_unix: s is UNIX domain socket if af_unix is nonzero s: socket Output @@ -138,87 +193,92 @@ char *port_str; the original socket is always closed. */ -int try_accept(af_unix,s) -int af_unix,s; +int try_accept(s) +int s; { - int len,c,i; - struct sockaddr_un s_un; - struct sockaddr_in sin; + struct sockaddr_storage ss; + socklen_t len; + int c, i; - if ( af_unix ) { - len = sizeof(s_un); - for ( c = -1, i = 0; (c < 0)&&(i = 10) ; i++ ) - c = accept(s, (struct sockaddr *) &s_un, &len); - } else { + len = sizeof(ss); + if (getsockname(s, (struct sockaddr *)&ss, &len) < 0) { + close(s); + return (-1); + } - len = sizeof(sin); - for ( c = -1, i = 0; (c < 0)&&(i = 10) ; i++ ) - c = accept(s, (struct sockaddr *) &sin, &len); + for (i = 0 ; i < 10 ; i++) { + c = accept(s, (struct sockaddr *)&ss, &len); + if (c >= 0) { + close(s); + return (c); + } } - if ( i == 10 ) - c = -1; + close(s); - return c; + return (-1); } int try_connect(use_unix,host,port_str) int use_unix; char *host,*port_str; { - struct sockaddr_in sin; - struct sockaddr_un s_un; - struct sockaddr *saddr; - struct hostent *hp; - int len,s,i; + struct addrinfo hints, *res, *ai; + int s, error, i; + char *errstr; - for ( i = 0; i < 10; i++ ) { - if ( use_unix ) { - if ( (s = socket(AF_UNIX,SOCK_STREAM,0)) < 0 ) { - perror("socket"); - return -1; - } - bzero(&s_un,sizeof(s_un)); - s_un.sun_family = AF_UNIX; - strcpy(s_un.sun_path,port_str); -#if defined(__FreeBSD__) - len = SUN_LEN(&s_un); - s_un.sun_len = len+1; /* XXX */ + memset(&hints, 0, sizeof(hints)); +#if defined(VISUAL) + hints.ai_family = PF_UNSPEC; #else - len = strlen(s_un.sun_path)+sizeof(s_un.sun_family); -#endif - saddr = (struct sockaddr *)&s_un; - } else { - if ( (s = socket(AF_INET,SOCK_STREAM,0)) < 0 ) { - perror("socket"); - return -1; + if (use_unix) + hints.ai_family = PF_UNIX; + else + hints.ai_family = PF_UNSPEC; +#endif /* VISUAL */ + hints.ai_socktype = SOCK_STREAM; + + error = getaddrinfo(host, port_str, &hints, &res); + if (error) { + warnx("try_connect: %s", gai_strerror(error)); + return (-1); + } + for (i = 0 ; i < 10 ; i++) { + s = -1; + for (ai = res ; ai != NULL ; ai = ai->ai_next) { + if ((s = socket(ai->ai_family, ai->ai_socktype, + ai->ai_protocol)) < 0 ) { + errstr = "socket"; + continue; } - bzero(&sin,sizeof(sin)); - sin.sin_port = htons(atoi(port_str)); - sin.sin_addr.s_addr = inet_addr(host); - if ( sin.sin_addr.s_addr != -1 ) { - sin.sin_family = AF_INET; - } else { - hp = gethostbyname(host); - bcopy(hp->h_addr,&sin.sin_addr,hp->h_length); - sin.sin_family = hp->h_addrtype; + if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0) { + errstr = "connect"; + close(s); + s = -1; + continue; } - len = sizeof(sin); - saddr = (struct sockaddr *)&sin; - } - if ( connect(s,saddr,len) >= 0 ) + + /* established a connection */ break; - else { - close(s); - usleep(100000); } + + if (s >= 0) { + freeaddrinfo(res); + return (s); + } + +#if defined(VISUAL) + Sleep(100); +#else + usleep(100000); +#endif } - if ( i == 10 ) { - perror("connect"); - return -1; - } else - return s; + freeaddrinfo(res); + + perror(errstr); + return (-1); } +#if 0 close_allconnections() { int s; @@ -242,6 +302,12 @@ int s; free_iofp(s); } } +#else +close_allconnections() +{ + shutdown_all(); +} +#endif free_iofp(s) int s; @@ -265,12 +331,27 @@ int is_server; { int i; unsigned char c,rc; + extern int mpi_myid; +#if MPI + iofp[s1].s = s1; + if ( mpi_myid == s1 ) { + iofp[s1].in = 0; + iofp[s1].out = 0; + } else { + iofp[s1].in = WSIO_open(s1,"r"); + iofp[s1].out = WSIO_open(s1,"w"); + } + iofp[s1].conv = 0; + iofp[s1].socket = 0; + + return s1; +#else for ( i = 0; i < MAXIOFP; i++ ) if ( !iofp[i].in ) break; iofp[i].s = s1; -#if defined(VISUAL) || MPI +#if defined(VISUAL) iofp[i].in = WSIO_open(s1,"r"); iofp[i].out = WSIO_open(s1,"w"); #else @@ -279,10 +360,6 @@ int is_server; setbuffer(iofp[i].in,(char *)malloc(LBUFSIZ),LBUFSIZ); setbuffer(iofp[i].out,(char *)malloc(LBUFSIZ),LBUFSIZ); #endif -#if MPI - iofp[i].conv = 0; - iofp[i].socket = 0; -#else if ( little_endian ) c = 1; else @@ -294,6 +371,9 @@ int is_server; } else { /* client : read -> write */ read_char(iofp[i].in,&rc); + /* special care for a failure of spawing a server */ + if ( rc !=0 && rc != 1 && rc != 0xff ) + return -1; write_char(iofp[i].out,&c); ox_flush_stream_force(i); } iofp[i].conv = c == rc ? 0 : 1; @@ -302,8 +382,8 @@ int is_server; strcpy(iofp[i].socket,af_sock); } else iofp[i].socket = 0; -#endif return i; +#endif } #if defined(VISUAL) @@ -339,4 +419,6 @@ int fd; return i; return -1; } + #endif /* INET */ +