Fix a bug in the output from print_canonical_sockaddr() fix from 36f8bafbd3dee66a8....
[ira/wip.git] / source3 / lib / util_sock.c
index b92cd3d624fe2454396da7e3911e0685bbc10772..30a3b83be75894b56e6b2807116b258ca13e4530 100644 (file)
@@ -63,7 +63,7 @@ bool is_ipaddress(const char *str)
                                        sizeof(addr)));
                        sp = addr;
                }
-               ret = inet_pton(AF_INET6, addr, &dest6);
+               ret = inet_pton(AF_INET6, sp, &dest6);
                if (ret > 0) {
                        return true;
                }
@@ -108,11 +108,15 @@ static bool interpret_string_addr_internal(struct addrinfo **ppres,
        hints.ai_socktype = SOCK_STREAM;
        hints.ai_flags = flags;
 
+       /* Linux man page on getaddinfo() says port will be
+          uninitialized when service string in NULL */
+
        ret = getaddrinfo(str, NULL,
                        &hints,
                        ppres);
+
        if (ret) {
-               DEBUG(3,("interpret_string_addr_interal: getaddrinfo failed "
+               DEBUG(3,("interpret_string_addr_internal: getaddrinfo failed "
                        "for name %s [%s]\n",
                        str,
                        gai_strerror(ret) ));
@@ -370,7 +374,7 @@ void in_addr_to_sockaddr_storage(struct sockaddr_storage *ss,
  Convert an IPv6 struct in_addr to a struct sockaddr_storage.
 ********************************************************************/
 
-void in6_addr_to_sockaddr_storage(struct sockaddr_storage *ss,
+ void in6_addr_to_sockaddr_storage(struct sockaddr_storage *ss,
                struct in6_addr ip)
 {
        struct sockaddr_in6 *sa = (struct sockaddr_in6 *)ss;
@@ -475,6 +479,29 @@ bool is_address_any(const struct sockaddr_storage *psa)
        return false;
 }
 
+/****************************************************************************
+ Get a port number in host byte order from a sockaddr_storage.
+****************************************************************************/
+
+uint16_t get_sockaddr_port(const struct sockaddr_storage *pss)
+{
+       uint16_t port = 0;
+
+       if (pss->ss_family != AF_INET) {
+#if defined(HAVE_IPV6)
+               /* IPv6 */
+               const struct sockaddr_in6 *sa6 =
+                       (const struct sockaddr_in6 *)pss;
+               port = ntohs(sa6->sin6_port);
+#endif
+       } else {
+               const struct sockaddr_in *sa =
+                       (const struct sockaddr_in *)pss;
+               port = ntohs(sa->sin_port);
+       }
+       return port;
+}
+
 /****************************************************************************
  Print out an IPv4 or IPv6 address from a struct sockaddr_storage.
 ****************************************************************************/
@@ -518,50 +545,28 @@ char *print_canonical_sockaddr(TALLOC_CTX *ctx,
        char *dest = NULL;
        int ret;
 
-       ret = getnameinfo((const struct sockaddr *)pss,
+       /* Linux getnameinfo() man pages says port is unitialized if
+          service name is NULL. */
+
+       ret = sys_getnameinfo((const struct sockaddr *)pss,
                        sizeof(struct sockaddr_storage),
                        addr, sizeof(addr),
                        NULL, 0,
                        NI_NUMERICHOST);
-       if (ret) {
+       if (ret != 0) {
                return NULL;
        }
+
        if (pss->ss_family != AF_INET) {
 #if defined(HAVE_IPV6)
-               /* IPv6 */
-               const struct sockaddr_in6 *sa6 =
-                       (const struct sockaddr_in6 *)pss;
-               uint16_t port = ntohs(sa6->sin6_port);
-
-               if (port) {
-                       dest = talloc_asprintf(ctx,
-                                       "[%s]:%d",
-                                       addr,
-                                       (unsigned int)port);
-               } else {
-                       dest = talloc_asprintf(ctx,
-                                       "[%s]",
-                                       addr);
-               }
+               dest = talloc_asprintf(ctx, "[%s]", addr);
 #else
                return NULL;
 #endif
        } else {
-               const struct sockaddr_in *sa =
-                       (const struct sockaddr_in *)pss;
-               uint16_t port = ntohs(sa->sin_port);
-
-               if (port) {
-                       dest = talloc_asprintf(ctx,
-                                       "%s:%d",
-                                       addr,
-                                       (unsigned int)port);
-               } else {
-                       dest = talloc_asprintf(ctx,
-                                       "%s",
-                                       addr);
-               }
+               dest = talloc_asprintf(ctx, "%s", addr);
        }
+       
        return dest;
 }
 
@@ -890,12 +895,10 @@ ssize_t read_udp_v4_socket(int fd,
  time_out = timeout in milliseconds
 ****************************************************************************/
 
-ssize_t read_socket_with_timeout(int fd,
-                               char *buf,
-                               size_t mincnt,
-                               size_t maxcnt,
-                               unsigned int time_out,
-                               enum smb_read_errors *pre)
+NTSTATUS read_socket_with_timeout(int fd, char *buf,
+                                 size_t mincnt, size_t maxcnt,
+                                 unsigned int time_out,
+                                 size_t *size_ret)
 {
        fd_set fds;
        int selrtn;
@@ -906,9 +909,7 @@ ssize_t read_socket_with_timeout(int fd,
 
        /* just checking .... */
        if (maxcnt <= 0)
-               return(0);
-
-       set_smb_read_error(pre,SMB_READ_OK);
+               return NT_STATUS_OK;
 
        /* Blocking read */
        if (time_out == 0) {
@@ -922,8 +923,7 @@ ssize_t read_socket_with_timeout(int fd,
                        if (readret == 0) {
                                DEBUG(5,("read_socket_with_timeout: "
                                        "blocking read. EOF from client.\n"));
-                               set_smb_read_error(pre,SMB_READ_EOF);
-                               return -1;
+                               return NT_STATUS_END_OF_FILE;
                        }
 
                        if (readret == -1) {
@@ -939,12 +939,11 @@ ssize_t read_socket_with_timeout(int fd,
                                                "read error = %s.\n",
                                                strerror(errno) ));
                                }
-                               set_smb_read_error(pre,SMB_READ_ERROR);
-                               return -1;
+                               return map_nt_error_from_unix(errno);
                        }
                        nread += readret;
                }
-               return((ssize_t)nread);
+               goto done;
        }
 
        /* Most difficult - timeout read */
@@ -978,16 +977,14 @@ ssize_t read_socket_with_timeout(int fd,
                                "read. select error = %s.\n",
                                strerror(errno) ));
                        }
-                       set_smb_read_error(pre,SMB_READ_ERROR);
-                       return -1;
+                       return map_nt_error_from_unix(errno);
                }
 
                /* Did we timeout ? */
                if (selrtn == 0) {
                        DEBUG(10,("read_socket_with_timeout: timeout read. "
                                "select timed out.\n"));
-                       set_smb_read_error(pre,SMB_READ_TIMEOUT);
-                       return -1;
+                       return NT_STATUS_IO_TIMEOUT;
                }
 
                readret = sys_read(fd, buf+nread, maxcnt-nread);
@@ -996,8 +993,7 @@ ssize_t read_socket_with_timeout(int fd,
                        /* we got EOF on the file descriptor */
                        DEBUG(5,("read_socket_with_timeout: timeout read. "
                                "EOF from client.\n"));
-                       set_smb_read_error(pre,SMB_READ_EOF);
-                       return -1;
+                       return NT_STATUS_END_OF_FILE;
                }
 
                if (readret == -1) {
@@ -1014,61 +1010,27 @@ ssize_t read_socket_with_timeout(int fd,
                                        "read. read error = %s.\n",
                                        strerror(errno) ));
                        }
-                       set_smb_read_error(pre,SMB_READ_ERROR);
-                       return -1;
+                       return map_nt_error_from_unix(errno);
                }
 
                nread += readret;
        }
 
+ done:
        /* Return the number we got */
-       return (ssize_t)nread;
+       if (size_ret) {
+               *size_ret = nread;
+       }
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
  Read data from the client, reading exactly N bytes.
 ****************************************************************************/
 
-ssize_t read_data(int fd,char *buffer,size_t N, enum smb_read_errors *pre)
+NTSTATUS read_data(int fd, char *buffer, size_t N)
 {
-       ssize_t ret;
-       size_t total=0;
-       char addr[INET6_ADDRSTRLEN];
-
-       set_smb_read_error(pre,SMB_READ_OK);
-
-       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) ));
-                       set_smb_read_error(pre,SMB_READ_EOF);
-                       return 0;
-               }
-
-               if (ret == -1) {
-                       if (fd == get_client_fd()) {
-                               /* Try and give an error message saying
-                                * what client failed. */
-                               DEBUG(0,("read_data: read failure for %d "
-                                       "bytes to client %s. Error = %s\n",
-                                       (int)(N - total),
-                                       get_peer_addr(fd,addr,sizeof(addr)),
-                                       strerror(errno) ));
-                       } else {
-                               DEBUG(0,("read_data: read failure for %d. "
-                                       "Error = %s\n",
-                                       (int)(N - total),
-                                       strerror(errno) ));
-                       }
-                       set_smb_read_error(pre,SMB_READ_ERROR);
-                       return -1;
-               }
-               total += ret;
-       }
-       return (ssize_t)total;
+       return read_socket_with_timeout(fd, buffer, N, N, 0, NULL);
 }
 
 /****************************************************************************
@@ -1130,37 +1092,29 @@ bool send_keepalive(int client)
  Timeout is in milliseconds.
 ****************************************************************************/
 
-ssize_t read_smb_length_return_keepalive(int fd,
-                                       char *inbuf,
-                                       unsigned int timeout,
-                                       enum smb_read_errors *pre)
+NTSTATUS read_smb_length_return_keepalive(int fd, char *inbuf,
+                                         unsigned int timeout,
+                                         size_t *len)
 {
-       ssize_t len=0;
        int msg_type;
-       bool ok = false;
+       NTSTATUS status;
 
-       while (!ok) {
-               if (timeout > 0) {
-                       ok = (read_socket_with_timeout(fd,inbuf,4,4,
-                                               timeout,pre) == 4);
-               } else {
-                       ok = (read_data(fd,inbuf,4,pre) == 4);
-               }
-               if (!ok) {
-                       return -1;
-               }
+       status = read_socket_with_timeout(fd, inbuf, 4, 4, timeout, NULL);
 
-               len = smb_len(inbuf);
-               msg_type = CVAL(inbuf,0);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
 
-               if (msg_type == SMBkeepalive) {
-                       DEBUG(5,("Got keepalive packet\n"));
-               }
+       *len = smb_len(inbuf);
+       msg_type = CVAL(inbuf,0);
+
+       if (msg_type == SMBkeepalive) {
+               DEBUG(5,("Got keepalive packet\n"));
        }
 
-       DEBUG(10,("got smb length of %lu\n",(unsigned long)len));
+       DEBUG(10,("got smb length of %lu\n",(unsigned long)(*len)));
 
-       return len;
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
@@ -1170,25 +1124,27 @@ ssize_t read_smb_length_return_keepalive(int fd,
  Timeout is in milliseconds.
 ****************************************************************************/
 
-ssize_t read_smb_length(int fd, char *inbuf, unsigned int timeout, enum smb_read_errors *pre)
+NTSTATUS read_smb_length(int fd, char *inbuf, unsigned int timeout,
+                        size_t *len)
 {
-       ssize_t len;
+       uint8_t msgtype = SMBkeepalive;
 
-       for(;;) {
-               len = read_smb_length_return_keepalive(fd, inbuf, timeout, pre);
+       while (msgtype == SMBkeepalive) {
+               NTSTATUS status;
 
-               if(len < 0)
-                       return len;
+               status = read_smb_length_return_keepalive(fd, inbuf, timeout,
+                                                         len);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
 
-               /* Ignore session keepalives. */
-               if(CVAL(inbuf,0) != SMBkeepalive)
-                       break;
+               msgtype = CVAL(inbuf, 0);
        }
 
        DEBUG(10,("read_smb_length: got smb length of %lu\n",
                  (unsigned long)len));
 
-       return len;
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
@@ -1201,28 +1157,17 @@ ssize_t read_smb_length(int fd, char *inbuf, unsigned int timeout, enum smb_read
  Doesn't check the MAC on signed packets.
 ****************************************************************************/
 
-ssize_t receive_smb_raw(int fd,
-                       char *buffer,
-                       unsigned int timeout,
-                       size_t maxlen,
-                       enum smb_read_errors *pre)
+NTSTATUS receive_smb_raw(int fd, char *buffer, unsigned int timeout,
+                        size_t maxlen, size_t *p_len)
 {
-       ssize_t len,ret;
-
-       set_smb_read_error(pre,SMB_READ_OK);
-
-       len = read_smb_length_return_keepalive(fd,buffer,timeout,pre);
-       if (len < 0) {
-               DEBUG(10,("receive_smb_raw: length < 0!\n"));
+       size_t len;
+       NTSTATUS status;
 
-               /*
-                * Correct fix. smb_read_error may have already been
-                * set. Only set it here if not already set. Global
-                * variables still suck :-). JRA.
-                */
+       status = read_smb_length_return_keepalive(fd,buffer,timeout,&len);
 
-               cond_set_smb_read_error(pre,SMB_READ_ERROR);
-               return -1;
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(10, ("receive_smb_raw: %s!\n", nt_errstr(status)));
+               return status;
        }
 
        /*
@@ -1234,15 +1179,7 @@ ssize_t receive_smb_raw(int fd,
                DEBUG(0,("Invalid packet length! (%lu bytes).\n",
                                        (unsigned long)len));
                if (len > BUFFER_SIZE + (SAFETY_MARGIN/2)) {
-
-                       /*
-                        * Correct fix. smb_read_error may have already been
-                        * set. Only set it here if not already set. Global
-                        * variables still suck :-). JRA.
-                        */
-
-                       cond_set_smb_read_error(pre,SMB_READ_ERROR);
-                       return -1;
+                       return NT_STATUS_INVALID_PARAMETER;
                }
        }
 
@@ -1251,20 +1188,11 @@ ssize_t receive_smb_raw(int fd,
                        len = MIN(len,maxlen);
                }
 
-               if (timeout > 0) {
-                       ret = read_socket_with_timeout(fd,
-                                       buffer+4,
-                                       len,
-                                       len,
-                                       timeout,
-                                       pre);
-               } else {
-                       ret = read_data(fd,buffer+4,len,pre);
-               }
+               status = read_socket_with_timeout(
+                       fd, buffer+4, len, len, timeout, &len);
 
-               if (ret != len) {
-                       cond_set_smb_read_error(pre,SMB_READ_ERROR);
-                       return -1;
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
 
                /* not all of samba3 properly checks for packet-termination
@@ -1273,81 +1201,8 @@ ssize_t receive_smb_raw(int fd,
                SSVAL(buffer+4,len, 0);
        }
 
-       return len;
-}
-
-/****************************************************************************
- Wrapper for receive_smb_raw().
- Checks the MAC on signed packets.
-****************************************************************************/
-
-bool receive_smb(int fd, char *buffer, unsigned int timeout, enum smb_read_errors *pre)
-{
-       if (receive_smb_raw(fd, buffer, timeout, 0, pre) < 0) {
-               return false;
-       }
-
-       if (srv_encryption_on()) {
-               NTSTATUS status = srv_decrypt_buffer(buffer);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(0, ("receive_smb: SMB decryption failed "
-                               "on incoming packet! Error %s\n",
-                               nt_errstr(status) ));
-                       cond_set_smb_read_error(pre, SMB_READ_BAD_DECRYPT);
-                       return false;
-               }
-       }
-
-       /* Check the incoming SMB signature. */
-       if (!srv_check_sign_mac(buffer, true)) {
-               DEBUG(0, ("receive_smb: SMB Signature verification "
-                       "failed on incoming packet!\n"));
-               cond_set_smb_read_error(pre,SMB_READ_BAD_SIG);
-               return false;
-       }
-
-       return true;
-}
-
-/****************************************************************************
- Send an smb to a fd.
-****************************************************************************/
-
-bool send_smb(int fd, char *buffer)
-{
-       size_t len;
-       size_t nwritten=0;
-       ssize_t ret;
-       char *buf_out = buffer;
-
-       /* Sign the outgoing packet if required. */
-       srv_calculate_sign_mac(buf_out);
-
-       if (srv_encryption_on()) {
-               NTSTATUS status = srv_encrypt_buffer(buffer, &buf_out);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(0, ("send_smb: SMB encryption failed "
-                               "on outgoing packet! Error %s\n",
-                               nt_errstr(status) ));
-                       return false;
-               }
-       }
-
-       len = smb_len(buf_out) + 4;
-
-       while (nwritten < len) {
-               ret = write_data(fd,buf_out+nwritten,len - nwritten);
-               if (ret <= 0) {
-                       DEBUG(0,("Error writing %d bytes to client. %d. (%s)\n",
-                               (int)len,(int)ret, strerror(errno) ));
-                       srv_free_enc_buffer(buf_out);
-                       return false;
-               }
-               nwritten += ret;
-       }
-
-       srv_free_enc_buffer(buf_out);
-       return true;
+       *p_len = len;
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
@@ -1921,7 +1776,7 @@ const char *get_peer_name(int fd, bool force_lookup)
        }
 
        /* Look up the remote host name. */
-       ret = getnameinfo((struct sockaddr *)&ss,
+       ret = sys_getnameinfo((struct sockaddr *)&ss,
                        length,
                        name_buf,
                        sizeof(name_buf),
@@ -2031,8 +1886,7 @@ int create_pipe_sock(const char *socket_dir,
                 goto out_close;
        }
 
-       asprintf(&path, "%s/%s", socket_dir, socket_name);
-       if (!path) {
+       if (asprintf(&path, "%s/%s", socket_dir, socket_name) == -1) {
                 goto out_close;
        }
 
@@ -2060,7 +1914,8 @@ int create_pipe_sock(const char *socket_dir,
 
 out_close:
        SAFE_FREE(path);
-       close(sock);
+       if (sock != -1)
+               close(sock);
 
 out_umask:
        umask(old_umask);
@@ -2131,14 +1986,15 @@ const char *get_mydnsfullname(void)
                        data_blob_string_const("get_mydnsfullname"),
                        data_blob_string_const(res->ai_canonname));
 
-       freeaddrinfo(res);
-
        if (!memcache_lookup(NULL, SINGLETON_CACHE,
                        data_blob_string_const("get_mydnsfullname"),
                        &tmp)) {
-               return NULL;
+               tmp = data_blob_talloc(talloc_tos(), res->ai_canonname,
+                               strlen(res->ai_canonname) + 1);
        }
 
+       freeaddrinfo(res);
+
        return (const char *)tmp.data;
 }