*lots of small merges form HEAD
[kai/samba.git] / source3 / lib / util_sock.c
index e3673f011a13d83a67507125ae4b9ced3a1a589a..3a7899df3d6982dd1fe2b5e98c0eea141e67afa7 100644 (file)
@@ -1,8 +1,8 @@
 /* 
-   Unix SMB/Netbios implementation.
-   Version 1.9.
+   Unix SMB/CIFS implementation.
    Samba utility functions
    Copyright (C) Andrew Tridgell 1992-1998
+   Copyright (C) Tim Potter      2000-2001
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 
 #include "includes.h"
 
-#ifdef WITH_SSL
-#include <ssl.h>
-#undef Realloc  /* SSLeay defines this and samba has a function of this name */
-extern SSL  *ssl;
-extern int  sslFd;
-#endif  /* WITH_SSL */
-
-extern int DEBUGLEVEL;
-
 /* the last IP received from */
 struct in_addr lastip;
 
@@ -44,23 +35,22 @@ int smb_read_error = 0;
 
 BOOL is_a_socket(int fd)
 {
-  int v,l;
-  l = sizeof(int);
-  return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
+       int v,l;
+       l = sizeof(int);
+       return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
 }
 
 enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
 
-typedef struct smb_socket_option
-{
-  char *name;
-  int level;
-  int option;
-  int value;
-  int opttype;
+typedef struct smb_socket_option {
+       const char *name;
+       int level;
+       int option;
+       int value;
+       int opttype;
 } smb_socket_option;
 
-smb_socket_option socket_options[] = {
+static const smb_socket_option socket_options[] = {
   {"SO_KEEPALIVE",      SOL_SOCKET,    SO_KEEPALIVE,    0,                 OPT_BOOL},
   {"SO_REUSEADDR",      SOL_SOCKET,    SO_REUSEADDR,    0,                 OPT_BOOL},
   {"SO_BROADCAST",      SOL_SOCKET,    SO_BROADCAST,    0,                 OPT_BOOL},
@@ -99,10 +89,11 @@ smb_socket_option socket_options[] = {
 /****************************************************************************
  Print socket options.
 ****************************************************************************/
+
 static void print_socket_options(int s)
 {
        int value, vlen = 4;
-       smb_socket_option *p = &socket_options[0];
+       const smb_socket_option *p = &socket_options[0];
 
        for (; p->name != NULL; p++) {
                if (getsockopt(s, p->level, p->option, (void *)&value, &vlen) == -1) {
@@ -117,7 +108,7 @@ static void print_socket_options(int s)
  Set user socket options.
 ****************************************************************************/
 
-void set_socket_options(int fd, char *options)
+void set_socket_options(int fd, const char *options)
 {
        fstring tok;
 
@@ -127,7 +118,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,235 +165,123 @@ 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);
-  }
-
-  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));
-
-  return(ret);
-}
-
-/****************************************************************************
- Read data from a socket with a timout in msec.
- mincount = if timeout, minimum to read before returning
- maxcount = number to be read.
- time_out = timeout in milliseconds
-****************************************************************************/
-
-static ssize_t read_socket_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
-{
-  fd_set fds;
-  int selrtn;
-  ssize_t readret;
-  size_t nread = 0;
-  struct timeval timeout;
-
-  /* just checking .... */
-  if (maxcnt <= 0)
-    return(0);
-
-  smb_read_error = 0;
-
-  /* Blocking read */
-  if (time_out <= 0) {
-    if (mincnt == 0) mincnt = maxcnt;
-
-    while (nread < mincnt) {
-#ifdef WITH_SSL
-      if(fd == sslFd){
-        readret = SSL_read(ssl, buf + nread, maxcnt - nread);
-      }else{
-        readret = read(fd, buf + nread, maxcnt - nread);
-      }
-#else /* WITH_SSL */
-      readret = read(fd, buf + nread, maxcnt - nread);
-#endif /* WITH_SSL */
-
-      if (readret == 0) {
-        DEBUG(5,("read_socket_with_timeout: blocking read. EOF from client.\n"));
-        smb_read_error = READ_EOF;
-        return -1;
-      }
-
-      if (readret == -1) {
-        DEBUG(0,("read_socket_with_timeout: read error = %s.\n", strerror(errno) ));
-        smb_read_error = READ_ERROR;
-        return -1;
-      }
-      nread += readret;
-    }
-    return((ssize_t)nread);
-  }
-  
-  /* Most difficult - timeout read */
-  /* If this is ever called on a disk file and 
-     mincnt is greater then the filesize then
-     system performance will suffer severely as 
-     select always returns true on disk files */
-
-  /* Set initial timeout */
-  timeout.tv_sec = (time_t)(time_out / 1000);
-  timeout.tv_usec = (long)(1000 * (time_out % 1000));
-
-  for (nread=0; nread < mincnt; ) {      
-    FD_ZERO(&fds);
-    FD_SET(fd,&fds);
-      
-    selrtn = sys_select_intr(fd+1,&fds,&timeout);
+       ssize_t ret;
+       struct sockaddr_in sock;
+       socklen_t socklen = sizeof(sock);
 
-    /* Check if error */
-    if(selrtn == -1) {
-      /* something is wrong. Maybe the socket is dead? */
-      DEBUG(0,("read_socket_with_timeout: timeout read. select error = %s.\n", strerror(errno) ));
-      smb_read_error = READ_ERROR;
-      return -1;
-    }
+       memset((char *)&sock,'\0',socklen);
+       memset((char *)&lastip,'\0',sizeof(lastip));
+       ret = (ssize_t)sys_recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
+       if (ret <= 0) {
+               DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
+               return(0);
+       }
 
-    /* Did we timeout ? */
-    if (selrtn == 0) {
-      DEBUG(10,("read_socket_with_timeout: timeout read. select timed out.\n"));
-      smb_read_error = READ_TIMEOUT;
-      return -1;
-    }
-      
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      readret = SSL_read(ssl, buf + nread, maxcnt - nread);
-    }else{
-      readret = read(fd, buf + nread, maxcnt - nread);
-    }
-#else /* WITH_SSL */
-    readret = read(fd, buf+nread, maxcnt-nread);
-#endif /* WITH_SSL */
-
-    if (readret == 0) {
-      /* we got EOF on the file descriptor */
-      DEBUG(5,("read_socket_with_timeout: timeout read. EOF from client.\n"));
-      smb_read_error = READ_EOF;
-      return -1;
-    }
+       lastip = sock.sin_addr;
+       lastport = ntohs(sock.sin_port);
 
-    if (readret == -1) {
-      /* the descriptor is probably dead */
-      DEBUG(0,("read_socket_with_timeout: timeout read. read error = %s.\n", strerror(errno) ));
-      smb_read_error = READ_ERROR;
-      return -1;
-    }
-      
-    nread += readret;
-  }
+       DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %d\n",
+                       inet_ntoa(lastip), lastport, ret));
 
-  /* Return the number we got */
-  return((ssize_t)nread);
+       return(ret);
 }
 
 /****************************************************************************
- Read data from a fd with a timout in msec.
+ Read data from a socket with a timout in msec.
  mincount = if timeout, minimum to read before returning
  maxcount = number to be read.
  time_out = timeout in milliseconds
 ****************************************************************************/
 
-ssize_t read_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
-{
-  fd_set fds;
-  int selrtn;
-  ssize_t readret;
-  size_t nread = 0;
-  struct timeval timeout;
-
-  /* just checking .... */
-  if (maxcnt <= 0)
-    return(0);
-
-  /* Blocking read */
-  if (time_out <= 0) {
-    if (mincnt == 0) mincnt = maxcnt;
-
-    while (nread < mincnt) {
-#ifdef WITH_SSL
-      if(fd == sslFd){
-        readret = SSL_read(ssl, buf + nread, maxcnt - nread);
-      }else{
-        readret = read(fd, buf + nread, maxcnt - nread);
-      }
-#else /* WITH_SSL */
-      readret = read(fd, buf + nread, maxcnt - nread);
-#endif /* WITH_SSL */
-
-      if (readret <= 0)
-       return readret;
-
-      nread += readret;
-    }
-    return((ssize_t)nread);
-  }
-  
-  /* Most difficult - timeout read */
-  /* If this is ever called on a disk file and 
-     mincnt is greater then the filesize then
-     system performance will suffer severely as 
-     select always returns true on disk files */
-
-  /* Set initial timeout */
-  timeout.tv_sec = (time_t)(time_out / 1000);
-  timeout.tv_usec = (long)(1000 * (time_out % 1000));
-
-  for (nread=0; nread < mincnt; ) {      
-    FD_ZERO(&fds);
-    FD_SET(fd,&fds);
-      
-    selrtn = sys_select_intr(fd+1,&fds,&timeout);
-
-    if(selrtn <= 0)
-      return selrtn;
-      
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      readret = SSL_read(ssl, buf + nread, maxcnt - nread);
-    }else{
-      readret = read(fd, buf + nread, maxcnt - nread);
-    }
-#else /* WITH_SSL */
-    readret = read(fd, buf+nread, maxcnt-nread);
-#endif /* WITH_SSL */
-
-    if (readret <= 0)
-      return readret;
-
-    nread += readret;
-  }
-
-  /* Return the number we got */
-  return((ssize_t)nread);
-}
-
-/****************************************************************************
-send a keepalive packet (rfc1002)
-****************************************************************************/
-
-BOOL send_keepalive(int client)
+ssize_t read_socket_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
 {
-  unsigned char buf[4];
-
-  buf[0] = 0x85;
-  buf[1] = buf[2] = buf[3] = 0;
-
-  return(write_socket_data(client,(char *)buf,4) == 4);
+       fd_set fds;
+       int selrtn;
+       ssize_t readret;
+       size_t nread = 0;
+       struct timeval timeout;
+       
+       /* just checking .... */
+       if (maxcnt <= 0)
+               return(0);
+       
+       smb_read_error = 0;
+       
+       /* Blocking read */
+       if (time_out <= 0) {
+               if (mincnt == 0) mincnt = maxcnt;
+               
+               while (nread < mincnt) {
+                       readret = sys_read(fd, buf + nread, maxcnt - nread);
+                       
+                       if (readret == 0) {
+                               DEBUG(5,("read_socket_with_timeout: blocking read. EOF from client.\n"));
+                               smb_read_error = READ_EOF;
+                               return -1;
+                       }
+                       
+                       if (readret == -1) {
+                               DEBUG(0,("read_socket_with_timeout: read error = %s.\n", strerror(errno) ));
+                               smb_read_error = READ_ERROR;
+                               return -1;
+                       }
+                       nread += readret;
+               }
+               return((ssize_t)nread);
+       }
+       
+       /* Most difficult - timeout read */
+       /* If this is ever called on a disk file and 
+          mincnt is greater then the filesize then
+          system performance will suffer severely as 
+          select always returns true on disk files */
+       
+       /* Set initial timeout */
+       timeout.tv_sec = (time_t)(time_out / 1000);
+       timeout.tv_usec = (long)(1000 * (time_out % 1000));
+       
+       for (nread=0; nread < mincnt; ) {      
+               FD_ZERO(&fds);
+               FD_SET(fd,&fds);
+               
+               selrtn = sys_select_intr(fd+1,&fds,NULL,NULL,&timeout);
+               
+               /* Check if error */
+               if (selrtn == -1) {
+                       /* something is wrong. Maybe the socket is dead? */
+                       DEBUG(0,("read_socket_with_timeout: timeout read. select error = %s.\n", strerror(errno) ));
+                       smb_read_error = READ_ERROR;
+                       return -1;
+               }
+               
+               /* Did we timeout ? */
+               if (selrtn == 0) {
+                       DEBUG(10,("read_socket_with_timeout: timeout read. select timed out.\n"));
+                       smb_read_error = READ_TIMEOUT;
+                       return -1;
+               }
+               
+               readret = sys_read(fd, buf+nread, maxcnt-nread);
+               
+               if (readret == 0) {
+                       /* we got EOF on the file descriptor */
+                       DEBUG(5,("read_socket_with_timeout: timeout read. EOF from client.\n"));
+                       smb_read_error = READ_EOF;
+                       return -1;
+               }
+               
+               if (readret == -1) {
+                       /* the descriptor is probably dead */
+                       DEBUG(0,("read_socket_with_timeout: timeout read. read error = %s.\n", strerror(errno) ));
+                       smb_read_error = READ_ERROR;
+                       return -1;
+               }
+               
+               nread += readret;
+       }
+       
+       /* Return the number we got */
+       return (ssize_t)nread;
 }
 
 /****************************************************************************
@@ -411,38 +290,28 @@ BOOL send_keepalive(int client)
 
 ssize_t read_data(int fd,char *buffer,size_t N)
 {
-  ssize_t  ret;
-  size_t total=0;  
+       ssize_t ret;
+       size_t total=0;  
  
-  smb_read_error = 0;
-
-  while (total < N)
-  {
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      ret = SSL_read(ssl, buffer + total, N - total);
-    }else{
-      ret = read(fd,buffer + total,N - total);
-    }
-#else /* WITH_SSL */
-    ret = read(fd,buffer + total,N - total);
-#endif /* WITH_SSL */
-
-    if (ret == 0)
-    {
-      DEBUG(10,("read_data: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
-      smb_read_error = READ_EOF;
-      return 0;
-    }
-    if (ret == -1)
-    {
-      DEBUG(0,("read_data: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
-      smb_read_error = READ_ERROR;
-      return -1;
-    }
-    total += ret;
-  }
-  return (ssize_t)total;
+       smb_read_error = 0;
+
+       while (total < N) {
+               ret = sys_read(fd,buffer + total,N - total);
+
+               if (ret == 0) {
+                       DEBUG(10,("read_data: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
+                       smb_read_error = READ_EOF;
+                       return 0;
+               }
+
+               if (ret == -1) {
+                       DEBUG(0,("read_data: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
+                       smb_read_error = READ_ERROR;
+                       return -1;
+               }
+               total += ret;
+       }
+       return (ssize_t)total;
 }
 
 /****************************************************************************
@@ -451,38 +320,28 @@ ssize_t read_data(int fd,char *buffer,size_t N)
 
 static ssize_t read_socket_data(int fd,char *buffer,size_t N)
 {
-  ssize_t  ret;
-  size_t total=0;  
+       ssize_t ret;
+       size_t total=0;  
  
-  smb_read_error = 0;
-
-  while (total < N)
-  {
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      ret = SSL_read(ssl, buffer + total, N - total);
-    }else{
-      ret = read(fd,buffer + total,N - total);
-    }
-#else /* WITH_SSL */
-    ret = read(fd,buffer + total,N - total);
-#endif /* WITH_SSL */
-
-    if (ret == 0)
-    {
-      DEBUG(10,("read_socket_data: recv of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
-      smb_read_error = READ_EOF;
-      return 0;
-    }
-    if (ret == -1)
-    {
-      DEBUG(0,("read_socket_data: recv failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
-      smb_read_error = READ_ERROR;
-      return -1;
-    }
-    total += ret;
-  }
-  return (ssize_t)total;
+       smb_read_error = 0;
+
+       while (total < N) {
+               ret = sys_read(fd,buffer + total,N - total);
+
+               if (ret == 0) {
+                       DEBUG(10,("read_socket_data: recv of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
+                       smb_read_error = READ_EOF;
+                       return 0;
+               }
+
+               if (ret == -1) {
+                       DEBUG(0,("read_socket_data: recv failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
+                       smb_read_error = READ_ERROR;
+                       return -1;
+               }
+               total += ret;
+       }
+       return (ssize_t)total;
 }
 
 /****************************************************************************
@@ -491,62 +350,46 @@ static ssize_t read_socket_data(int fd,char *buffer,size_t N)
 
 ssize_t write_data(int fd,char *buffer,size_t N)
 {
-  size_t total=0;
-  ssize_t ret;
-
-  while (total < N)
-  {
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      ret = SSL_write(ssl,buffer + total,N - total);
-    }else{
-      ret = write(fd,buffer + total,N - total);
-    }
-#else /* WITH_SSL */
-    ret = write(fd,buffer + total,N - total);
-#endif /* WITH_SSL */
-
-    if (ret == -1) {
-      DEBUG(0,("write_data: write failure. Error = %s\n", strerror(errno) ));
-      return -1;
-    }
-    if (ret == 0) return total;
+       size_t total=0;
+       ssize_t ret;
 
-    total += ret;
-  }
-  return (ssize_t)total;
+       while (total < N) {
+               ret = sys_write(fd,buffer + total,N - total);
+
+               if (ret == -1) {
+                       DEBUG(0,("write_data: write failure. Error = %s\n", strerror(errno) ));
+                       return -1;
+               }
+               if (ret == 0)
+                       return total;
+
+               total += ret;
+       }
+       return (ssize_t)total;
 }
 
 /****************************************************************************
  Write data to a socket - use send rather than write.
 ****************************************************************************/
 
-ssize_t write_socket_data(int fd,char *buffer,size_t N)
+static ssize_t write_socket_data(int fd,char *buffer,size_t N)
 {
-  size_t total=0;
-  ssize_t ret;
-
-  while (total < N)
-  {
-#ifdef WITH_SSL
-    if(fd == sslFd){
-      ret = SSL_write(ssl,buffer + total,N - total);
-    }else{
-      ret = send(fd,buffer + total,N - total, 0);
-    }
-#else /* WITH_SSL */
-    ret = send(fd,buffer + total,N - total,0);
-#endif /* WITH_SSL */
-
-    if (ret == -1) {
-      DEBUG(0,("write_socket_data: write failure. Error = %s\n", strerror(errno) ));
-      return -1;
-    }
-    if (ret == 0) return total;
+       size_t total=0;
+       ssize_t ret;
 
-    total += ret;
-  }
-  return (ssize_t)total;
+       while (total < N) {
+               ret = sys_send(fd,buffer + total,N - total,0);
+
+               if (ret == -1) {
+                       DEBUG(0,("write_socket_data: write failure. Error = %s\n", strerror(errno) ));
+                       return -1;
+               }
+               if (ret == 0)
+                       return total;
+
+               total += ret;
+       }
+       return (ssize_t)total;
 }
 
 /****************************************************************************
@@ -555,19 +398,34 @@ write to a socket
 
 ssize_t write_socket(int fd,char *buf,size_t len)
 {
-  ssize_t ret=0;
+       ssize_t ret=0;
 
-  DEBUG(6,("write_socket(%d,%d)\n",fd,(int)len));
-  ret = write_socket_data(fd,buf,len);
+       DEBUG(6,("write_socket(%d,%d)\n",fd,(int)len));
+       ret = write_socket_data(fd,buf,len);
       
-  DEBUG(6,("write_socket(%d,%d) wrote %d\n",fd,(int)len,(int)ret));
-  if(ret <= 0)
-    DEBUG(0,("write_socket: Error writing %d bytes to socket %d: ERRNO = %s\n", 
-       (int)len, fd, strerror(errno) ));
+       DEBUG(6,("write_socket(%d,%d) wrote %d\n",fd,(int)len,(int)ret));
+       if(ret <= 0)
+               DEBUG(0,("write_socket: Error writing %d bytes to socket %d: ERRNO = %s\n", 
+                       (int)len, fd, strerror(errno) ));
+
+       return(ret);
+}
+
+/****************************************************************************
+send a keepalive packet (rfc1002)
+****************************************************************************/
+
+BOOL send_keepalive(int client)
+{
+       unsigned char buf[4];
+
+       buf[0] = SMBkeepalive;
+       buf[1] = buf[2] = buf[3] = 0;
 
-  return(ret);
+       return(write_socket_data(client,(char *)buf,4) == 4);
 }
 
+
 /****************************************************************************
 read 4 bytes of a smb packet and return the smb length of the packet
 store the result in the buffer
@@ -578,30 +436,29 @@ timeout is in milliseconds.
 
 static ssize_t read_smb_length_return_keepalive(int fd,char *inbuf,unsigned int timeout)
 {
-  ssize_t len=0;
-  int msg_type;
-  BOOL ok = False;
+       ssize_t len=0;
+       int msg_type;
+       BOOL ok = False;
 
-  while (!ok)
-  {
-    if (timeout > 0)
-      ok = (read_socket_with_timeout(fd,inbuf,4,4,timeout) == 4);
-    else 
-      ok = (read_socket_data(fd,inbuf,4) == 4);
+       while (!ok) {
+               if (timeout > 0)
+                       ok = (read_socket_with_timeout(fd,inbuf,4,4,timeout) == 4);
+               else 
+                       ok = (read_socket_data(fd,inbuf,4) == 4);
 
-    if (!ok)
-      return(-1);
+               if (!ok)
+                       return(-1);
 
-    len = smb_len(inbuf);
-    msg_type = CVAL(inbuf,0);
+               len = smb_len(inbuf);
+               msg_type = CVAL(inbuf,0);
 
-    if (msg_type == 0x85
-      DEBUG(5,("Got keepalive packet\n"));
-  }
+               if (msg_type == SMBkeepalive
+                       DEBUG(5,("Got keepalive packet\n"));
+       }
 
-  DEBUG(10,("got smb length of %d\n",len));
+       DEBUG(10,("got smb length of %d\n",len));
 
-  return(len);
+       return(len);
 }
 
 /****************************************************************************
@@ -613,23 +470,22 @@ timeout is in milliseconds.
 
 ssize_t read_smb_length(int fd,char *inbuf,unsigned int timeout)
 {
-  ssize_t len;
+       ssize_t len;
 
-  for(;;)
-  {
-    len = read_smb_length_return_keepalive(fd, inbuf, timeout);
+       for(;;) {
+               len = read_smb_length_return_keepalive(fd, inbuf, timeout);
 
-    if(len < 0)
-      return len;
+               if(len < 0)
+                       return len;
 
-    /* Ignore session keepalives. */
-    if(CVAL(inbuf,0) != 0x85)
-      break;
-  }
+               /* Ignore session keepalives. */
+               if(CVAL(inbuf,0) != SMBkeepalive)
+                       break;
+       }
 
-  DEBUG(10,("read_smb_length: got smb length of %d\n",len));
+       DEBUG(10,("read_smb_length: got smb length of %d\n",len));
 
-  return len;
+       return len;
 }
 
 /****************************************************************************
@@ -651,18 +507,35 @@ BOOL receive_smb(int fd,char *buffer, unsigned int timeout)
        len = read_smb_length_return_keepalive(fd,buffer,timeout);
        if (len < 0) {
                DEBUG(10,("receive_smb: length < 0!\n"));
-               return(False);
+
+               /*
+                * Correct fix. smb_read_error may have already been
+                * set. Only set it here if not already set. Global
+                * variables still suck :-). JRA.
+                */
+
+               if (smb_read_error == 0)
+                       smb_read_error = READ_ERROR;
+               return False;
        }
 
        /*
         * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
-     * of header. Don't print the error if this fits.... JRA.
+        * of header. Don't print the error if this fits.... JRA.
         */
 
        if (len > (BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
                DEBUG(0,("Invalid packet length! (%d bytes).\n",len));
                if (len > BUFFER_SIZE + (SAFETY_MARGIN/2)) {
-                       smb_read_error = READ_ERROR;
+
+                       /*
+                        * Correct fix. smb_read_error may have already been
+                        * set. Only set it here if not already set. Global
+                        * variables still suck :-). JRA.
+                        */
+
+                       if (smb_read_error == 0)
+                               smb_read_error = READ_ERROR;
                        return False;
                }
        }
@@ -670,7 +543,8 @@ BOOL receive_smb(int fd,char *buffer, unsigned int timeout)
        if(len > 0) {
                ret = read_socket_data(fd,buffer+4,len);
                if (ret != len) {
-                       smb_read_error = READ_ERROR;
+                       if (smb_read_error == 0)
+                               smb_read_error = READ_ERROR;
                        return False;
                }
        }
@@ -678,40 +552,6 @@ BOOL receive_smb(int fd,char *buffer, unsigned int timeout)
        return(True);
 }
 
-/****************************************************************************
-  read an smb from a fd ignoring all keepalive packets. Note that the buffer 
-  *MUST* be of size BUFFER_SIZE+SAFETY_MARGIN.
-  The timeout is in milliseconds
-
-  This is exactly the same as receive_smb except that it never returns
-  a session keepalive packet (just as receive_smb used to do).
-  receive_smb was changed to return keepalives as the oplock processing means this call
-  should never go into a blocking read.
-****************************************************************************/
-
-BOOL client_receive_smb(int fd,char *buffer, unsigned int timeout)
-{
-  BOOL ret;
-
-  for(;;)
-  {
-    ret = receive_smb(fd, buffer, timeout);
-
-    if (!ret)
-    {
-      DEBUG(10,("client_receive_smb failed\n"));
-      show_msg(buffer);
-      return ret;
-    }
-
-    /* Ignore session keepalive packets. */
-    if(CVAL(buffer,0) != 0x85)
-      break;
-  }
-  show_msg(buffer);
-  return ret;
-}
-
 /****************************************************************************
   send an smb to a fd 
 ****************************************************************************/
@@ -737,117 +577,70 @@ BOOL send_smb(int fd,char *buffer)
 }
 
 /****************************************************************************
-send a single packet to a port on another machine
+ Open a socket of the specified type, port, and address for incoming data.
 ****************************************************************************/
 
-BOOL send_one_packet(char *buf,int len,struct in_addr ip,int port,int type)
+int open_socket_in( int type, int port, int dlevel, uint32 socket_addr, BOOL rebind )
 {
-  BOOL ret;
-  int out_fd;
-  struct sockaddr_in sock_out;
-
-  /* create a socket to write to */
-  out_fd = socket(AF_INET, type, 0);
-  if (out_fd == -1) 
-    {
-      DEBUG(0,("socket failed"));
-      return False;
-    }
-
-  /* set the address and port */
-  memset((char *)&sock_out,'\0',sizeof(sock_out));
-  putip((char *)&sock_out.sin_addr,(char *)&ip);
-  sock_out.sin_port = htons( port );
-  sock_out.sin_family = AF_INET;
-  
-  if (DEBUGLEVEL > 0)
-    DEBUG(3,("sending a packet of len %d to (%s) on port %d of type %s\n",
-            len,inet_ntoa(ip),port,type==SOCK_DGRAM?"DGRAM":"STREAM"));
-       
-  /* send it */
-  ret = (sendto(out_fd,buf,len,0,(struct sockaddr *)&sock_out,sizeof(sock_out)) >= 0);
-
-  if (!ret)
-    DEBUG(0,("Packet send to %s(%d) failed ERRNO=%s\n",
-            inet_ntoa(ip),port,strerror(errno)));
-
-  close(out_fd);
-  return(ret);
-}
-
-/****************************************************************************
-open a socket of the specified type, port and address for incoming data
-****************************************************************************/
+       struct sockaddr_in sock;
+       int res;
 
-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;
-
-  /* get my host name */
-  if (gethostname(host_name, MAXHOSTNAMELEN) == -1) 
-    { DEBUG(0,("gethostname failed\n")); return -1; } 
-
-  /* get host info */
-  if ((hp = sys_gethostbyname(host_name)) == 0) 
-    {
-      DEBUG(0,( "sys_gethostbyname: 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; }
-
-  {
-    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) ));
+       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;
+       }
+
+       /* 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 = %s ", 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_PORT1 || port == SMB_PORT2 || 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( 10, ( "bind succeeded on port %d\n", port ) );
 
-  return res;
-}
+       return( res );
+ }
 
 /****************************************************************************
   create an outgoing socket. timeout is in milliseconds.
@@ -904,7 +697,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;
@@ -916,6 +709,37 @@ connect_again:
   return res;
 }
 
+/*
+  open a connected UDP socket to host on port
+*/
+int open_udp_socket(const char *host, int port)
+{
+       int type = SOCK_DGRAM;
+       struct sockaddr_in sock_out;
+       int res;
+       struct in_addr *addr;
+
+       addr = interpret_addr2(host);
+
+       res = socket(PF_INET, type, 0);
+       if (res == -1) {
+               return -1;
+       }
+
+       memset((char *)&sock_out,'\0',sizeof(sock_out));
+       putip((char *)&sock_out.sin_addr,(char *)addr);
+       sock_out.sin_port = htons(port);
+       sock_out.sin_family = PF_INET;
+
+       if (connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out))) {
+               close(res);
+               return -1;
+       }
+
+       return res;
+}
+
+
 /* the following 3 client_*() functions are nasty ways of allowing
    some generic functions to get info that really should be hidden in
    particular modules */
@@ -928,7 +752,7 @@ void client_setfd(int fd)
 
 char *client_name(void)
 {
-       return get_socket_name(client_fd);
+       return get_socket_name(client_fd,False);
 }
 
 char *client_addr(void)
@@ -967,7 +791,7 @@ static BOOL matchname(char *remotehost,struct in_addr  addr)
        
        /* Look up the host address in the address list we just got. */
        for (i = 0; hp->h_addr_list[i]; i++) {
-               if (memcmp(hp->h_addr_list[i], (caddr_t) & addr, sizeof(addr)) == 0)
+               if (memcmp(hp->h_addr_list[i], (char *) & addr, sizeof(addr)) == 0)
                        return True;
        }
        
@@ -986,13 +810,21 @@ static BOOL matchname(char *remotehost,struct in_addr  addr)
 /*******************************************************************
  return the DNS name of the remote end of a socket
  ******************************************************************/
-char *get_socket_name(int fd)
+char *get_socket_name(int fd, BOOL force_lookup)
 {
        static pstring name_buf;
        static fstring addr_buf;
        struct hostent *hp;
        struct in_addr addr;
        char *p;
+
+       /* reverse lookups can be *very* expensive, and in many
+          situations won't work because many networks don't link dhcp
+          with dns. To avoid the delay we avoid the lookup if
+          possible */
+       if (!lp_hostname_lookups() && (force_lookup == False)) {
+               return get_socket_addr(fd);
+       }
        
        p = get_socket_addr(fd);
 
@@ -1017,6 +849,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;
 }
 
@@ -1046,104 +884,111 @@ char *get_socket_addr(int fd)
        return addr_buf;
 }
 
-/*******************************************************************
- opens and connects to a unix pipe socket
- ******************************************************************/
-int open_pipe_sock(char *path)
-{
-       int sock;
-       struct sockaddr_un sa;
-
-       sock = socket(AF_UNIX, SOCK_STREAM, 0);
-
-       if (sock < 0)
-       {
-               DEBUG(0, ("unix socket open failed\n"));
-               return sock;
-       }
-
-       ZERO_STRUCT(sa);
-       sa.sun_family = AF_UNIX;
-       safe_strcpy(sa.sun_path, path, sizeof(sa.sun_path)-1);
-
-       DEBUG(10, ("socket open succeeded.  file name: %s\n", sa.sun_path));
-
-       if (connect(sock, (struct sockaddr*) &sa, sizeof(sa)) < 0)
-       {
-               DEBUG(0,("socket connect to %s failed\n", sa.sun_path));
-               close(sock);
-               return -1;
-       }
 
-       return sock;
-}
+/*******************************************************************
+ Create protected unix domain socket.
 
-int create_pipe_socket(char *dir, int dir_perms,
-                               char *path, int path_perms)
+ some unixen cannot set permissions on a ux-dom-sock, so we
+ have to make sure that the directory contains the protection
+ permissions, instead.
+ ******************************************************************/
+int create_pipe_sock(const char *socket_dir,
+                    const char *socket_name,
+                    mode_t dir_perms)
 {
-       int s;
-       struct sockaddr_un sa;
-
-       DEBUG(0,("create_pipe_socket: %s %d %s %d\n",
-                  dir, dir_perms, path, path_perms));
-
-       DEBUG(0,("*** RACE CONDITION.  PLEASE SOMEONE EXAMINE create_pipe_Socket AND FIX IT ***\n"));
-
-       mkdir(dir, dir_perms);
-
-       if (chmod(dir, dir_perms) < 0)
-       {
-               DEBUG(0, ("chmod on %s failed\n", dir));
-               return -1;
-       }
-
-       if (!remove(path))
-       {
-               DEBUG(0, ("remove on %s failed\n", path));
-       }
-               
-       /* start listening on unix socket */
-       s = socket(AF_UNIX, SOCK_STREAM, 0);
-
-       if (s < 0)
-       {
-               DEBUG(0, ("socket open failed\n"));
-               return -1;
-       }
-
-       ZERO_STRUCT(sa);
-       sa.sun_family = AF_UNIX;
-       safe_strcpy(sa.sun_path, path, sizeof(sa.sun_path)-1);
-
-       if (bind(s, (struct sockaddr*) &sa, sizeof(sa)) < 0)
-       {
-               DEBUG(0, ("socket bind to %s failed\n", sa.sun_path));
-               close(s);
-               remove(path);
-               return -1;
-       }
-
-       if (s == -1)
-       {
-               DEBUG(0,("bind failed\n"));
-               remove(path);
-               return -1;
-       }
-
-       if (path_perms != 0)
-       {
-               chmod(path, path_perms);
-       }
-
-       if (listen(s, 5) == -1)
-       {
-               DEBUG(0,("listen failed\n"));
-               return -1;
-       }
-
-       DEBUG(5,("unix socket opened: %s\n", path));
-
-       return s;
+#ifdef HAVE_UNIXSOCKET
+        struct sockaddr_un sunaddr;
+        struct stat st;
+        int sock;
+        mode_t old_umask;
+        pstring path;
+        
+        /* Create the socket directory or reuse the existing one */
+        
+        if (lstat(socket_dir, &st) == -1) {
+                
+                if (errno == ENOENT) {
+                        
+                        /* Create directory */
+                        
+                        if (mkdir(socket_dir, dir_perms) == -1) {
+                                DEBUG(0, ("error creating socket directory "
+                                          "%s: %s\n", socket_dir, 
+                                          strerror(errno)));
+                                return -1;
+                        }
+                        
+                } else {
+                        
+                        DEBUG(0, ("lstat failed on socket directory %s: %s\n",
+                                  socket_dir, strerror(errno)));
+                        return -1;
+                }
+                
+        } else {
+                
+                /* Check ownership and permission on existing directory */
+                
+                if (!S_ISDIR(st.st_mode)) {
+                        DEBUG(0, ("socket directory %s isn't a directory\n",
+                                  socket_dir));
+                        return -1;
+                }
+                
+                if ((st.st_uid != sec_initial_uid()) || 
+                    ((st.st_mode & 0777) != dir_perms)) {
+                        DEBUG(0, ("invalid permissions on socket directory "
+                                  "%s\n", socket_dir));
+                        return -1;
+                }
+        }
+        
+        /* Create the socket file */
+        
+        old_umask = umask(0);
+        
+        sock = socket(AF_UNIX, SOCK_STREAM, 0);
+        
+        if (sock == -1) {
+                perror("socket");
+               umask(old_umask);
+                return -1;
+        }
+        
+        snprintf(path, sizeof(path), "%s/%s", socket_dir, socket_name);
+        
+        unlink(path);
+        memset(&sunaddr, 0, sizeof(sunaddr));
+        sunaddr.sun_family = AF_UNIX;
+        safe_strcpy(sunaddr.sun_path, path, sizeof(sunaddr.sun_path)-1);
+        
+        if (bind(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) == -1) {
+                DEBUG(0, ("bind failed on pipe socket %s: %s\n",
+                          path,
+                          strerror(errno)));
+                close(sock);
+               umask(old_umask);
+                return -1;
+        }
+        
+        if (listen(sock, 5) == -1) {
+                DEBUG(0, ("listen failed on pipe socket %s: %s\n",
+                          path,
+                          strerror(errno)));
+                close(sock);
+               umask(old_umask);
+                return -1;
+        }
+        
+        umask(old_umask);
+        
+        /* Success! */
+        
+        return sock;
+#else
+        DEBUG(0, ("create_pipe_sock: No Unix sockets on this system\n"));
+        return -1;
+#endif /* HAVE_UNIXSOCKET */
 }
 
 /*******************************************************************
@@ -1157,7 +1002,7 @@ the resulting file descriptors are symmetrical
 static int socketpair_tcp(int fd[2])
 {
        int listener;
-       struct sockaddr sock;
+       struct sockaddr_in sock;
        struct sockaddr_in sock2;
        socklen_t socklen = sizeof(sock);
        int connect_done = 0;
@@ -1178,19 +1023,21 @@ static int socketpair_tcp(int fd[2])
 
        if (listen(listener, 1) != 0) goto failed;
 
-       if (getsockname(listener, &sock, &socklen) != 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, &sock, &socklen)) == -1) goto failed;
+       if ((fd[0] = accept(listener, (struct sockaddr *)&sock, &socklen)) == -1) goto failed;
 
        close(listener);
        if (connect_done == 0) {
@@ -1221,7 +1068,10 @@ attached to the original stderr
 int sock_exec(const char *prog)
 {
        int fd[2];
-       if (socketpair_tcp(fd) != 0) return -1;
+       if (socketpair_tcp(fd) != 0) {
+               DEBUG(0,("socketpair_tcp failed (%s)\n", strerror(errno)));
+               return -1;
+       }
        if (fork() == 0) {
                close(fd[0]);
                close(0);
@@ -1233,4 +1083,3 @@ int sock_exec(const char *prog)
        close(fd[1]);
        return fd[0];
 }
-