converted another bunch of stuff to NTSTATUS
[jra/samba/.git] / source3 / lib / util_sock.c
index d741f038ceaffb26d5e4adefc7dc288b4913c426..16b457c413086ac461e16d90a95d386e18110b62 100644 (file)
@@ -127,7 +127,7 @@ void set_socket_options(int fd, char *options)
                char *p;
                BOOL got_value = False;
 
-               if ((p = strchr(tok,'='))) {
+               if ((p = strchr_m(tok,'='))) {
                        *p = 0;
                        value = atoi(p+1);
                        got_value = True;
@@ -174,26 +174,25 @@ void set_socket_options(int fd, char *options)
 
 ssize_t read_udp_socket(int fd,char *buf,size_t len)
 {
-  ssize_t ret;
-  struct sockaddr_in sock;
-  int socklen;
-  
-  socklen = sizeof(sock);
-  memset((char *)&sock,'\0',socklen);
-  memset((char *)&lastip,'\0',sizeof(lastip));
-  ret = (ssize_t)recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
-  if (ret <= 0) {
-    DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
-    return(0);
-  }
+       ssize_t ret;
+       struct sockaddr_in sock;
+       socklen_t socklen = sizeof(sock);
+
+       memset((char *)&sock,'\0',socklen);
+       memset((char *)&lastip,'\0',sizeof(lastip));
+       ret = (ssize_t)recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
+       if (ret <= 0) {
+               DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
+               return(0);
+       }
 
-  lastip = sock.sin_addr;
-  lastport = ntohs(sock.sin_port);
+       lastip = sock.sin_addr;
+       lastport = ntohs(sock.sin_port);
 
-  DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %d\n",
-             inet_ntoa(lastip), lastport, ret));
+       DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %d\n",
+                       inet_ntoa(lastip), lastport, ret));
 
-  return(ret);
+       return(ret);
 }
 
 /****************************************************************************
@@ -776,78 +775,70 @@ BOOL send_one_packet(char *buf,int len,struct in_addr ip,int port,int type)
 }
 
 /****************************************************************************
-open a socket of the specified type, port and address for incoming data
+ Open a socket of the specified type, port, and address for incoming data.
 ****************************************************************************/
 
-int open_socket_in(int type, int port, int dlevel,uint32 socket_addr, BOOL rebind)
+int open_socket_in( int type, int port, int dlevel, uint32 socket_addr, BOOL rebind )
 {
-  struct hostent *hp;
-  struct sockaddr_in sock;
-  pstring host_name;
-  int res;
+       struct sockaddr_in sock;
+       int res;
 
-  /* get my host name */
-  if (gethostname(host_name, MAXHOSTNAMELEN) == -1) 
-    { DEBUG(0,("gethostname failed\n")); return -1; } 
-
-  /* get host info */
-  if ((hp = Get_Hostbyname(host_name)) == 0) 
-    {
-      DEBUG(0,( "Get_Hostbyname: Unknown host %s\n",host_name));
-      return -1;
-    }
-  
-  memset((char *)&sock,'\0',sizeof(sock));
-  memcpy((char *)&sock.sin_addr,(char *)hp->h_addr, hp->h_length);
+       memset( (char *)&sock, '\0', sizeof(sock) );
 
 #ifdef HAVE_SOCK_SIN_LEN
-  sock.sin_len = sizeof(sock);
+       sock.sin_len         = sizeof(sock);
 #endif
-  sock.sin_port = htons( port );
-  sock.sin_family = hp->h_addrtype;
-  sock.sin_addr.s_addr = socket_addr;
-  res = socket(hp->h_addrtype, type, 0);
-  if (res == -1) 
-    { DEBUG(0,("socket failed\n")); return -1; }
+       sock.sin_port        = htons( port );
+       sock.sin_family      = AF_INET;
+       sock.sin_addr.s_addr = socket_addr;
+
+       res = socket( AF_INET, type, 0 );
+       if( res == -1 ) {
+               if( DEBUGLVL(0) ) {
+                       dbgtext( "open_socket_in(): socket() call failed: " );
+                       dbgtext( "%s\n", strerror( errno ) );
+               }
+               return -1;
+       }
 
-  {
-    int val=1;
-       if(rebind)
-               val=1;
-       else
-               val=0;
-    if(setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val)) == -1)
-               DEBUG(dlevel,("setsockopt: SO_REUSEADDR=%d on port %d failed with error = %s\n",
-                       val, port, strerror(errno) ));
+       /* This block sets/clears the SO_REUSEADDR and possibly SO_REUSEPORT. */
+       {
+               int val = rebind ? 1 : 0;
+               if( setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val)) == -1 ) {
+                       if( DEBUGLVL( dlevel ) ) {
+                               dbgtext( "open_socket_in(): setsockopt: " );
+                               dbgtext( "SO_REUSEADDR = %s ", val?"True":"False" );
+                               dbgtext( "on port %d failed ", port );
+                               dbgtext( "with error = %s\n", strerror(errno) );
+                       }
+               }
 #ifdef SO_REUSEPORT
-    if(setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val)) == -1)
-               DEBUG(dlevel,("setsockopt: SO_REUSEPORT=%d on port %d failed with error = %s\n",
-                       val, port, strerror(errno) ));
+               if( setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val)) == -1 ) {
+                       if( DEBUGLVL( dlevel ) ) {
+                               dbgtext( "open_socket_in(): setsockopt: "
+                               dbgtext( "SO_REUSEPORT = %d ", val?"True":"False" );
+                               dbgtext( "on port %d failed ", port );
+                               dbgtext( "with error = %s\n", strerror(errno) );
+                       }
+               }
 #endif /* SO_REUSEPORT */
-  }
-
-  /* now we've got a socket - we need to bind it */
-  if (bind(res, (struct sockaddr * ) &sock,sizeof(sock)) < 0) 
-    { 
-      if (port) {
-       if (port == SMB_PORT || port == NMB_PORT)
-         DEBUG(dlevel,("bind failed on port %d socket_addr=%s (%s)\n",
-                       port,inet_ntoa(sock.sin_addr),strerror(errno))); 
-       close(res); 
-
-       if (dlevel > 0 && port < 1000)
-         port = 7999;
+       }
 
-       if (port >= 1000 && port < 9000)
-         return(open_socket_in(type,port+1,dlevel,socket_addr,rebind));
-      }
+       /* now we've got a socket - we need to bind it */
+       if( bind( res, (struct sockaddr *)&sock, sizeof(sock) ) == -1 ) {
+               if( DEBUGLVL(dlevel) && (port == SMB_PORT || port == NMB_PORT) ) {
+                       dbgtext( "bind failed on port %d ", port );
+                       dbgtext( "socket_addr = %s.\n", inet_ntoa( sock.sin_addr ) );
+                       dbgtext( "Error = %s\n", strerror(errno) );
+               }
+               close( res ); 
+               return( -1 ); 
+       }
 
-      return(-1); 
-    }
-  DEBUG(3,("bind succeeded on port %d\n",port));
+       DEBUG( 3, ( "bind succeeded on port %d\n", port ) );
 
-  return res;
-}
+       return( res );
+ }
 
 /****************************************************************************
   create an outgoing socket. timeout is in milliseconds.
@@ -904,7 +895,7 @@ connect_again:
 #endif
 
   if (ret < 0) {
-    DEBUG(1,("error connecting to %s:%d (%s)\n",
+    DEBUG(2,("error connecting to %s:%d (%s)\n",
             inet_ntoa(*addr),port,strerror(errno)));
     close(res);
     return -1;
@@ -945,8 +936,8 @@ static BOOL matchname(char *remotehost,struct in_addr  addr)
        struct hostent *hp;
        int     i;
        
-       if ((hp = Get_Hostbyname(remotehost)) == 0) {
-               DEBUG(0,("Get_Hostbyname(%s): lookup failure.\n", remotehost));
+       if ((hp = sys_gethostbyname(remotehost)) == 0) {
+               DEBUG(0,("sys_gethostbyname(%s): lookup failure.\n", remotehost));
                return False;
        } 
 
@@ -1017,6 +1008,12 @@ char *get_socket_name(int fd)
                        pstrcpy(name_buf,"UNKNOWN");
                }
        }
+
+       alpha_strcpy(name_buf, name_buf, "_-.", sizeof(name_buf));
+       if (strstr(name_buf,"..")) {
+               pstrcpy(name_buf, "UNKNOWN");
+       }
+
        return name_buf;
 }
 
@@ -1145,3 +1142,97 @@ int create_pipe_socket(char *dir, int dir_perms,
 
        return s;
 }
+
+/*******************************************************************
+this is like socketpair but uses tcp. It is used by the Samba
+regression test code
+The function guarantees that nobody else can attach to the socket,
+or if they do that this function fails and the socket gets closed
+returns 0 on success, -1 on failure
+the resulting file descriptors are symmetrical
+ ******************************************************************/
+static int socketpair_tcp(int fd[2])
+{
+       int listener;
+       struct sockaddr_in sock;
+       struct sockaddr_in sock2;
+       socklen_t socklen = sizeof(sock);
+       int connect_done = 0;
+       
+       fd[0] = fd[1] = listener = -1;
+
+       memset(&sock, 0, sizeof(sock));
+       
+       if ((listener = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
+
+        memset(&sock2, 0, sizeof(sock2));
+#ifdef HAVE_SOCK_SIN_LEN
+        sock2.sin_len = sizeof(sock2);
+#endif
+        sock2.sin_family = PF_INET;
+
+        bind(listener, (struct sockaddr *)&sock2, sizeof(sock2));
+
+       if (listen(listener, 1) != 0) goto failed;
+
+       if (getsockname(listener, (struct sockaddr *)&sock, &socklen) != 0) goto failed;
+
+       if ((fd[1] = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
+
+       set_blocking(fd[1], 0);
+
+       sock.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+
+       if (connect(fd[1],(struct sockaddr *)&sock,sizeof(sock)) == -1) {
+               if (errno != EINPROGRESS) goto failed;
+       } else {
+               connect_done = 1;
+       }
+
+       if ((fd[0] = accept(listener, (struct sockaddr *)&sock, &socklen)) == -1) goto failed;
+
+       close(listener);
+       if (connect_done == 0) {
+               if (connect(fd[1],(struct sockaddr *)&sock,sizeof(sock)) != 0
+                   && errno != EISCONN) goto failed;
+       }
+
+       set_blocking(fd[1], 1);
+
+       /* all OK! */
+       return 0;
+
+ failed:
+       if (fd[0] != -1) close(fd[0]);
+       if (fd[1] != -1) close(fd[1]);
+       if (listener != -1) close(listener);
+       return -1;
+}
+
+
+/*******************************************************************
+run a program on a local tcp socket, this is used to launch smbd
+when regression testing
+the return value is a socket which is attached to a subprocess
+running "prog". stdin and stdout are attached. stderr is left
+attached to the original stderr
+ ******************************************************************/
+int sock_exec(const char *prog)
+{
+       int fd[2];
+       if (socketpair_tcp(fd) != 0) {
+               DEBUG(0,("socketpair_tcp failed (%s)\n", strerror(errno)));
+               return -1;
+       }
+       if (fork() == 0) {
+               close(fd[0]);
+               close(0);
+               close(1);
+               dup(fd[1]);
+               dup(fd[1]);
+               exit(system(prog));
+       }
+       close(fd[1]);
+       return fd[0];
+}
+