]> git.sur5r.net Git - openldap/blobdiff - libraries/libldap/os-ip.c
ITS#837: fix server down abandon bug
[openldap] / libraries / libldap / os-ip.c
index c1c75e894ba527b0b879eb2917330abf92fa9a3e..71dcd4be25eed76f07eef7a69ba208f696be03e5 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)
+ldap_int_socket(LDAP *ld, int family, int type )
 {
-       ber_socket_t s = socket(AF_INET, 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;
 }
 
@@ -183,7 +184,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_in *sin, 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 +205,7 @@ ldap_pvt_connect(LDAP *ld, ber_socket_t s, struct sockaddr_in *sin, int async)
        if ( ldap_pvt_ndelay_on(ld, s) == -1 )
                return ( -1 );
 
-       if ( connect(s, (struct sockaddr *) sin, sizeof(struct sockaddr_in)) == 0 )
+       if ( connect(s, sin, addrlen) == 0 )
        {
                if ( ldap_pvt_ndelay_off(ld, s) == -1 )
                        return ( -1 );
@@ -275,27 +278,85 @@ 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];
+               struct addrinfo hints, *res, *sai;
+
+               memset( &hints, '\0', sizeof(hints) );
+               hints.ai_family = AF_UNSPEC;
+               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);
+                       return -1;
+               }
+               sai = res;
+               rc = -1;
+               do {
+                       /* we assume AF_x and PF_x are equal for all x */
+                       s = ldap_int_socket( ld, sai->ai_family, SOCK_STREAM );
+                       if ( s == -1 ) {
+                               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: {
+                               char addr[INET6_ADDRSTRLEN];
+                               inet_ntop( AF_INET6,
+                                       &((struct sockaddr_in6 *)sai->ai_addr)->sin6_addr,
+                                       addr, sizeof addr);
+                               osip_debug(ld, "ldap_connect_to_host: Trying %s %s\n", 
+                                       addr, serv, 0);
+                       } break;
+#endif
+                       case AF_INET: {
+                               char addr[INET_ADDRSTRLEN];
+                               inet_ntop( AF_INET,
+                                       &((struct sockaddr_in *)sai->ai_addr)->sin_addr,
+                                       addr, sizeof addr);
+                               osip_debug(ld, "ldap_connect_to_host: Trying %s:%s\n", 
+                                       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 );
+                               break;
+                       }
+                       ldap_pvt_close_socket(ld, s);
+               } while ((sai = sai->ai_next) != NULL);
+               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);
 
-                       if ( rc < 0 )
-                               ; /*XXX NO MEMORY? */
-
                        if ( (rc < 0) || (hp == NULL) ) {
 #ifdef HAVE_WINSOCK
                                ldap_pvt_set_errno( WSAGetLastError() );
@@ -309,18 +370,20 @@ ldap_connect_to_host(LDAP *ld, Sockbuf *sb, const char *host,
                        use_hp = 1;
                }
                address = in.s_addr;
+#endif
        }
 
        rc = s = -1;
        for ( i = 0; !use_hp || (hp->h_addr_list[i] != 0); ++i, rc = -1 ) {
 
-               if ( (s = ldap_pvt_socket( ld )) == -1 )
+               s = ldap_int_socket( ld, PF_INET, SOCK_STREAM );
+               if ( s == -1 ) {
                        /* 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;
                }
 
@@ -329,12 +392,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, &sin, 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 );
@@ -350,26 +415,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_in      sin;
+       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);
-   
-       (void)memset( (char *)&sin, '\0', sizeof( struct sockaddr_in ));
-       len = sizeof( sin );
+       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 *)&sin, &len ) == -1 ) {
+       if ( getpeername( sd, &sa, &len ) == -1 ) {
                return( NULL );
        }
 
@@ -378,21 +445,63 @@ ldap_host_connected_to( Sockbuf *sb )
         * this is necessary for kerberos to work right, since the official
         * hostname is used as the kerberos instance.
         */
-       if ((ldap_pvt_gethostbyaddr_a( (char *) &sin.sin_addr,
-               sizeof( sin.sin_addr ), 
-               AF_INET, &he_buf, &ha_buf,
-               &hp,&local_h_errno ) ==0 ) && (hp != NULL) )
-       {
-               if ( hp->h_name != NULL ) {
-                       char *host = LDAP_STRDUP( hp->h_name );   
-                       DO_RETURN( host );
+
+       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;
+               len = sizeof( struct in6_addr );
+               break;
+#endif
+       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;
        }
 
-       DO_RETURN( NULL );
+       host = NULL;
+       if ((ldap_pvt_gethostbyaddr_a( addr, len,
+               sa.sa_family, &he_buf, &ha_buf,
+               &hp,&local_h_errno ) == 0 ) &&
+               (hp != NULL) && ( hp->h_name != NULL ) )
+       {
+               host = LDAP_STRDUP( hp->h_name );   
+       }
+
+       LDAP_FREE( ha_buf );
+       return host;
 }
-#undef DO_RETURN   
-   
 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND || HAVE_TLS */