]> git.sur5r.net Git - openldap/blobdiff - libraries/libldap/os-ip.c
Fix off by one bug
[openldap] / libraries / libldap / os-ip.c
index 5c2f922021f406bf64649df4f117770611f3f7ee..8cef102b3ec4fc27b57d37b99f05f7467a9e73a4 100644 (file)
@@ -74,7 +74,7 @@ ldap_int_timeval_dup( struct timeval **dest, const struct timeval *src )
                return 1;
        }
 
-       SAFEMEMCPY( (char *) new, (const char *) src, sizeof(struct timeval));
+       AC_MEMCPY( (char *) new, (const char *) src, sizeof(struct timeval));
 
        *dest = new;
        return 0;
@@ -95,9 +95,9 @@ ldap_pvt_ndelay_off(LDAP *ld, int fd)
 }
 
 static ber_socket_t
-ldap_pvt_socket(LDAP *ld, int family)
+ldap_int_socket(LDAP *ld, int family, int type )
 {
-       ber_socket_t s = socket(family, SOCK_STREAM, 0);
+       ber_socket_t s = socket(family, type, 0);
        osip_debug(ld, "ldap_new_socket: %d\n",s,0,0);
        return ( s );
 }
@@ -110,22 +110,23 @@ ldap_pvt_close_socket(LDAP *ld, int s)
 }
 
 static int
-ldap_pvt_prepare_socket(LDAP *ld, int fd)
+ldap_int_prepare_socket(LDAP *ld, int s, int proto )
 {
-       osip_debug(ld, "ldap_prepare_socket: %d\n",fd,0,0);
+       osip_debug(ld, "ldap_prepare_socket: %d\n", s,0,0);
 
 #ifdef TCP_NODELAY
-{
-       int dummy = 1;
-       if ( setsockopt( fd, IPPROTO_TCP, TCP_NODELAY,
-               (char*) &dummy, sizeof(dummy) ) == AC_SOCKET_ERROR )
-       {
-               osip_debug(ld, "ldap_prepare_socket: "
-                       "setsockopt(%d, TCP_NODELAY) failed (ignored).\n",
-                       fd, 0, 0);
+       if( proto == LDAP_PROTO_TCP ) {
+               int dummy = 1;
+               if ( setsockopt( s, IPPROTO_TCP, TCP_NODELAY,
+                       (char*) &dummy, sizeof(dummy) ) == AC_SOCKET_ERROR )
+               {
+                       osip_debug(ld, "ldap_prepare_socket: "
+                               "setsockopt(%d, TCP_NODELAY) failed (ignored).\n",
+                               s, 0, 0);
+               }
        }
-}
 #endif
+
        return 0;
 }
 
@@ -150,7 +151,9 @@ ldap_pvt_is_socket_ready(LDAP *ld, int s)
 {
        int so_errno;
        int dummy = sizeof(so_errno);
-       if ( getsockopt( s, SOL_SOCKET, SO_ERROR, &so_errno, &dummy ) == -1 ) {
+       if ( getsockopt( s, SOL_SOCKET, SO_ERROR, &so_errno, &dummy )
+               == AC_SOCKET_ERROR )
+       {
                return -1;
        }
        if ( so_errno ) {
@@ -166,7 +169,9 @@ ldap_pvt_is_socket_ready(LDAP *ld, int s)
        struct sockaddr_in sin;
        char ch;
        int dummy = sizeof(sin);
-       if ( getpeername( s, (struct sockaddr *) &sin, &dummy ) == -1 ) {
+       if ( getpeername( s, (struct sockaddr *) &sin, &dummy )
+               == AC_SOCKET_ERROR )
+       {
                /* XXX: needs to be replace with ber_stream_read() */
                read(s, &ch, 1);
 #ifdef HAVE_WINSOCK
@@ -183,7 +188,9 @@ ldap_pvt_is_socket_ready(LDAP *ld, int s)
 #undef TRACE
 
 static int
-ldap_pvt_connect(LDAP *ld, ber_socket_t s, struct sockaddr *sin, socklen_t addrlen, int async)
+ldap_pvt_connect(LDAP *ld, ber_socket_t s,
+       struct sockaddr *sin, socklen_t addrlen,
+       int async)
 {
        struct timeval  tv, *opt_tv=NULL;
        fd_set          wfds, *z=NULL;
@@ -202,7 +209,7 @@ ldap_pvt_connect(LDAP *ld, ber_socket_t s, struct sockaddr *sin, socklen_t addrl
        if ( ldap_pvt_ndelay_on(ld, s) == -1 )
                return ( -1 );
 
-       if ( connect(s, sin, addrlen) == 0 )
+       if ( connect(s, sin, addrlen) != AC_SOCKET_ERROR )
        {
                if ( ldap_pvt_ndelay_off(ld, s) == -1 )
                        return ( -1 );
@@ -235,13 +242,14 @@ ldap_pvt_connect(LDAP *ld, ber_socket_t s, struct sockaddr *sin, socklen_t addrl
 #else
                    z,
 #endif
-                   opt_tv ? &tv : NULL) == -1)
+                   opt_tv ? &tv : NULL) == AC_SOCKET_ERROR )
+       {
                return ( -1 );
+       }
 
 #ifdef HAVE_WINSOCK
        /* This means the connection failed */
-       if (FD_ISSET(s, &efds))
-       {
+       if ( FD_ISSET(s, &efds) ) {
            ldap_pvt_set_errno(WSAECONNREFUSED);
            osip_debug(ld, "ldap_pvt_connect: error on socket %d: "
                       "errno: %d (%s)\n", s, errno, sock_errstr(errno));
@@ -275,22 +283,23 @@ ldap_pvt_inet_aton( const char *host, struct in_addr *in)
 
 
 int
-ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
-               unsigned long address, int port, int async)
+ldap_connect_to_host(LDAP *ld, Sockbuf *sb,
+       int proto,
+       const char *host,
+       unsigned long address, int port, int async)
 {
        struct sockaddr_in      sin;
-       struct in_addr          in;
        ber_socket_t            s = AC_SOCKET_INVALID;
        int                     rc, i, use_hp = 0;
-       struct hostent          *hp, he_buf;
-       int                     local_h_errno;
+       struct hostent          *hp = NULL;
        char                    *ha_buf=NULL, *p, *q;
 
        osip_debug(ld, "ldap_connect_to_host\n",0,0,0);
        
        if (host != NULL) {
 #ifdef HAVE_GETADDRINFO
-               char serv[7];
+               char serv[7];
+               int err;
                struct addrinfo hints, *res, *sai;
 
                memset( &hints, '\0', sizeof(hints) );
@@ -298,18 +307,24 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
                hints.ai_socktype = SOCK_STREAM;
 
                snprintf(serv, sizeof serv, "%d", ntohs(port));
-               if ( getaddrinfo(host, serv, &hints, &res) ) {
-                       osip_debug(ld, "ldap_connect_to_host:getaddrinfo failed\n",0,0,0);
+               if ( err = getaddrinfo(host, serv, &hints, &res) ) {
+                       osip_debug(ld, "ldap_connect_to_host: getaddrinfo failed: %s\n", AC_GAI_STRERROR(err), 0, 0);
                        return -1;
                }
                sai = res;
                rc = -1;
                do {
-                       s = ldap_pvt_socket( ld, sai->ai_family, ld );
-                       if ( s == -1 ) {
+                       /* we assume AF_x and PF_x are equal for all x */
+                       s = ldap_int_socket( ld, sai->ai_family, SOCK_STREAM );
+                       if ( s == AC_SOCKET_INVALID ) {
                                continue;
                        }
 
+                       if ( ldap_int_prepare_socket(ld, s, proto ) == -1 ) {
+                               ldap_pvt_close_socket(ld, s);
+                               break;
+                       }
+
                        switch (sai->ai_family) {
 #ifdef LDAP_PF_INET6
                        case AF_INET6: {
@@ -330,6 +345,7 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
                                        addr, serv, 0);
                        } break;
                        }
+
                        rc = ldap_pvt_connect(ld, s, sai->ai_addr, sai->ai_addrlen, async);
                        if ( (rc == 0) || (rc == -2) ) {
                                ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_FD, &s );
@@ -340,7 +356,10 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
                freeaddrinfo(res);
                return rc;
 #else
+               struct in_addr in;
                if (! inet_aton( host, &in) ) {
+                       int local_h_errno;
+                       struct hostent he_buf;
                        rc = ldap_pvt_gethostbyname_a(host, &he_buf, &ha_buf,
                                        &hp, &local_h_errno);
 
@@ -363,15 +382,14 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
        rc = s = -1;
        for ( i = 0; !use_hp || (hp->h_addr_list[i] != 0); ++i, rc = -1 ) {
 
-               s = ldap_pvt_socket( ld, AF_INET );
-               if ( s == -1 ) {
+               s = ldap_int_socket( ld, PF_INET, SOCK_STREAM );
+               if ( s == AC_SOCKET_INVALID ) {
                        /* use_hp ? continue : break; */
                        break;
                }
           
-               if ( ldap_pvt_prepare_socket(ld, s) == -1 ) {
+               if ( ldap_int_prepare_socket( ld, s, proto ) == -1 ) {
                        ldap_pvt_close_socket(ld, s);
-                       /* use_hp ? continue : break; */
                        break;
                }
 
@@ -380,12 +398,14 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
                sin.sin_port = port;
                p = (char *)&sin.sin_addr;
                q = use_hp ? (char *)hp->h_addr_list[i] : (char *)&address;
-               SAFEMEMCPY(p, q, sizeof(sin.sin_addr) );
+               AC_MEMCPY(p, q, sizeof(sin.sin_addr) );
 
                osip_debug(ld, "ldap_connect_to_host: Trying %s:%d\n", 
                                inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),0);
 
-               rc = ldap_pvt_connect(ld, s, (struct sockaddr *)&sin, sizeof(struct sockaddr_in), async);
+               rc = ldap_pvt_connect(ld, s,
+                       (struct sockaddr *)&sin, sizeof(struct sockaddr_in),
+                       async);
    
                if ( (rc == 0) || (rc == -2) ) {
                        ber_sockbuf_ctrl( sb, LBER_SB_OPT_SET_FD, &s );
@@ -401,27 +421,28 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
        return rc;
 }
 
-#if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND ) || defined( HAVE_TLS ) || defined( HAVE_CYRUS_SASL )
+#if defined( LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND ) \
+       || defined( HAVE_TLS ) || defined( HAVE_CYRUS_SASL )
 char *
 ldap_host_connected_to( Sockbuf *sb )
 {
-       struct hostent          *hp;
-       socklen_t               len;
-       struct sockaddr         sa;
-       char                    *addr;
+       struct hostent  *hp;
+       socklen_t               len;
+       struct sockaddr sa;
+       char                    *addr;
+       char                    *host;
 
        /* buffers for gethostbyaddr_r */
-       struct hostent          he_buf;
-        int                    local_h_errno;
+       struct hostent  he_buf;
+       int                             local_h_errno;
        char                    *ha_buf=NULL;
-       ber_socket_t            sd;
-#define DO_RETURN(x) if (ha_buf) LDAP_FREE(ha_buf); return (x);
-   
+       ber_socket_t    sd;
+
        (void)memset( (char *)&sa, '\0', sizeof( struct sockaddr ));
        len = sizeof( sa );
 
        ber_sockbuf_ctrl( sb, LBER_SB_OPT_GET_FD, &sd );
-       if ( getpeername( sd, (struct sockaddr *)&sa, &len ) == -1 ) {
+       if ( getpeername( sd, &sa, &len ) == -1 ) {
                return( NULL );
        }
 
@@ -432,6 +453,10 @@ ldap_host_connected_to( Sockbuf *sb )
         */
 
        switch (sa.sa_family) {
+#ifdef LDAP_PF_LOCAL
+       case AF_LOCAL:
+               return LDAP_STRDUP( ldap_int_hostname );
+#endif
 #ifdef LDAP_PF_INET6
        case AF_INET6:
                addr = (char *) &((struct sockaddr_in6 *)&sa)->sin6_addr;
@@ -441,25 +466,48 @@ ldap_host_connected_to( Sockbuf *sb )
        case AF_INET:
                addr = (char *) &((struct sockaddr_in *)&sa)->sin_addr;
                len = sizeof( struct in_addr );
+
+               {
+                       struct sockaddr_in localhost;
+                       localhost.sin_addr.s_addr = htonl( INADDR_ANY );
+
+                       if( memcmp ( &localhost.sin_addr,
+                               &((struct sockaddr_in *)&sa)->sin_addr,
+                               sizeof(localhost.sin_addr) ) == 0 )
+                       {
+                               return LDAP_STRDUP( ldap_int_hostname );
+                       }
+
+#ifdef INADDR_LOOPBACK
+                       localhost.sin_addr.s_addr = htonl( INADDR_LOOPBACK );
+
+                       if( memcmp ( &localhost.sin_addr,
+                               &((struct sockaddr_in *)&sa)->sin_addr,
+                               sizeof(localhost.sin_addr) ) == 0 )
+                       {
+                               return LDAP_STRDUP( ldap_int_hostname );
+                       }
+#endif
+               }
                break;
+
        default:
                return( NULL );
                break;
        }
+
+       host = NULL;
        if ((ldap_pvt_gethostbyaddr_a( addr, len,
                sa.sa_family, &he_buf, &ha_buf,
-               &hp,&local_h_errno ) ==0 ) && (hp != NULL) )
+               &hp,&local_h_errno ) == 0 ) &&
+               (hp != NULL) && ( hp->h_name != NULL ) )
        {
-               if ( hp->h_name != NULL ) {
-                       char *host = LDAP_STRDUP( hp->h_name );   
-                       DO_RETURN( host );
-               }
+               host = LDAP_STRDUP( hp->h_name );   
        }
 
-       DO_RETURN( NULL );
+       LDAP_FREE( ha_buf );
+       return host;
 }
-#undef DO_RETURN   
-   
 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND || HAVE_TLS */