]> git.sur5r.net Git - openldap/blobdiff - libraries/libldap/util-int.c
allow different solutions to debug sasl shutdown
[openldap] / libraries / libldap / util-int.c
index 31cbaafbe3ca73a925663bd88a851f4c63e1c55f..4bf3f6fa71e6833f56f0060737fed7dfdbbf8b0b 100644 (file)
@@ -1,11 +1,12 @@
+/* $OpenLDAP$ */
 /*
- * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
+ * Copyright 1998-2002 The OpenLDAP Foundation, All Rights Reserved.
  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
  */
 /*
  * util-int.c  Various functions to replace missing threadsafe ones.
- *                               Without the real *_r funcs, things will work, but won't be
- *                               threadsafe. 
+ *                               Without the real *_r funcs, things will
+ *                               work, but might not be threadsafe. 
  * 
  * Written by Bart Hartgers.
  *
  * in file LICENSE in the top-level directory of the distribution.
  */ 
 
+
 #include "portable.h"
 
-#include <stdlib.h>
+#include <ac/stdlib.h>
 
 #include <ac/errno.h>
 #include <ac/socket.h>
 #include <ac/string.h>
 #include <ac/time.h>
+#include <ac/unistd.h>
 
 #include "ldap-int.h"
 
-char *ldap_int_strtok( char *str, const char *delim, char **pos )
-{
-#ifdef HAVE_STRTOK_R
-       return strtok_r(str, delim, pos);
+#if defined( LDAP_R_COMPILE )
+# include <ldap_pvt_thread.h>
 #else
-       return strtok(str, delim);
+# undef HAVE_REENTRANT_FUNCTIONS
 #endif
-}
 
-char *ldap_int_ctime( const time_t *tp, char *buf )
+#if (defined( HAVE_CTIME_R ) || defined( HAVE_REENTRANT_FUNCTIONS)) \
+       && defined( CTIME_R_NARGS )
+#      define USE_CTIME_R
+#endif
+
+#if defined(HAVE_GETHOSTBYNAME_R) && \
+    (GETHOSTBYNAME_R_NARGS > 6 || GETHOSTBYNAME_R_NARGS < 5)
+       /* Don't know how to handle this version, pretend it's not there */
+#      undef HAVE_GETHOSTBYNAME_R
+#endif
+#if defined(HAVE_GETHOSTBYADDR_R) && \
+    (GETHOSTBYADDR_R_NARGS > 8 || GETHOSTBYADDR_R_NARGS < 7)
+       /* Don't know how to handle this version, pretend it's not there */
+#      undef HAVE_GETHOSTBYADDR_R
+#endif
+
+#ifdef LDAP_R_COMPILE
+# ifndef USE_CTIME_R
+       static ldap_pvt_thread_mutex_t ldap_int_ctime_mutex;
+# endif
+# if !defined( HAVE_GETHOSTBYNAME_R ) || !defined( HAVE_GETHOSTBYADDR_R )
+       static ldap_pvt_thread_mutex_t ldap_int_gethostby_mutex;
+# endif
+# ifdef HAVE_RES_QUERY
+       ldap_pvt_thread_mutex_t ldap_int_resolv_mutex;
+# endif
+#endif /* LDAP_R_COMPILE */
+
+char *ldap_pvt_ctime( const time_t *tp, char *buf )
 {
-#if defined( HAVE_CTIME_R ) && defined( CTIME_R_NARGS )
+#ifdef USE_CTIME_R
 # if (CTIME_R_NARGS > 3) || (CTIME_R_NARGS < 2)
-       choke me!  nargs should have 2 or 3
+#      error "CTIME_R_NARGS should be 2 or 3"
+# elif CTIME_R_NARGS > 2 && defined(CTIME_R_RETURNS_INT)
+       return( ctime_r(tp,buf,26) < 0 ? 0 : buf );
 # elif CTIME_R_NARGS > 2
        return ctime_r(tp,buf,26);
 # else
        return ctime_r(tp,buf);
 # endif          
+
 #else
-       return ctime(tp);
+# ifdef LDAP_R_COMPILE
+       ldap_pvt_thread_mutex_lock( &ldap_int_ctime_mutex );
+# endif
+       AC_MEMCPY( buf, ctime(tp), 26 );
+# ifdef LDAP_R_COMPILE
+       ldap_pvt_thread_mutex_unlock( &ldap_int_ctime_mutex );
+# endif
+       return buf;
 #endif 
 }
 
-#define BUFSTART 1024
-#define BUFMAX (32*1024)
+#define BUFSTART (1024-32)
+#define BUFMAX (32*1024-32)
 
-static char *safe_realloc( char **buf, int len )
-{
-       char *tmpbuf;
-       tmpbuf = realloc( *buf, len );
-       if (tmpbuf) {
-               *buf=tmpbuf;
-       } 
-       return tmpbuf;
-}
-int ldap_int_gethostbyname_a(
+static char *safe_realloc( char **buf, int len );
+
+#if !defined(HAVE_GETHOSTBYNAME_R) && defined(LDAP_R_COMPILE)
+static int copy_hostent( struct hostent *res, char **buf, struct hostent * src );
+#endif
+
+int ldap_pvt_gethostbyname_a(
        const char *name, 
        struct hostent *resbuf,
        char **buf,
        struct hostent **result,
        int *herrno_ptr )
 {
-#ifdef HAVE_GETHOSTBYNAME_R
-       int r;
+#if defined( HAVE_GETHOSTBYNAME_R )
+
+# define NEED_SAFE_REALLOC 1   
+       int r=-1;
        int buflen=BUFSTART;
+       *buf = NULL;
+       for(;buflen<BUFMAX;) {
+               if (safe_realloc( buf, buflen )==NULL)
+                       return r;
+
+#if (GETHOSTBYNAME_R_NARGS < 6)
+               r = ((*result=gethostbyname_r( name, resbuf, *buf,
+                                              buflen, herrno_ptr ))== NULL) ?
+                   -1 : 0;
+#else
+               r = gethostbyname_r( name, resbuf, *buf,
+                       buflen, result, herrno_ptr );
+#endif
+
+               Debug( LDAP_DEBUG_TRACE, "ldap_pvt_gethostbyname_a: host=%s, r=%d\n",
+                      name, r, 0 );
 
-       if (safe_realloc( buf, buflen)) {
-               for(;buflen<BUFMAX;) {
-                       r = gethostbyname_r( name, resbuf, *buf,
-                               buflen, result, herrno_ptr );
 #ifdef NETDB_INTERNAL
-                       if ((r<0) &&
-                               (*herrno_ptr==NETDB_INTERNAL) &&
-                               (errno==ERANGE))
-                       {
-                               if (safe_realloc( buf, buflen*=2 )) {
-                                               continue;
-                               }
-                       }
+               if ((r<0) &&
+                       (*herrno_ptr==NETDB_INTERNAL) &&
+                       (errno==ERANGE))
+               {
+                       buflen*=2;
+                       continue;
+               }
 #endif
-                       return r;
-               }
+               return r;
        }
-
-#else /* gethostbyname() */
+       return -1;
+#elif defined( LDAP_R_COMPILE )
+# define NEED_COPY_HOSTENT   
+       struct hostent *he;
+       int     retval;
+       *buf = NULL;
+       
+       ldap_pvt_thread_mutex_lock( &ldap_int_gethostby_mutex );
+       
+       he = gethostbyname( name );
+       
+       if (he==NULL) {
+               *herrno_ptr = h_errno;
+               retval = -1;
+       } else if (copy_hostent( resbuf, buf, he )<0) {
+               *herrno_ptr = -1;
+               retval = -1;
+       } else {
+               *result = resbuf;
+               retval = 0;
+       }
+       
+       ldap_pvt_thread_mutex_unlock( &ldap_int_gethostby_mutex );
+       
+       return retval;
+#else  
+       *buf = NULL;
        *result = gethostbyname( name );
 
        if (*result!=NULL) {
@@ -104,12 +175,12 @@ int ldap_int_gethostbyname_a(
        }
 
        *herrno_ptr = h_errno;
-#endif 
-
+       
        return -1;
+#endif 
 }
         
-int ldap_int_gethostbyaddr_a(
+int ldap_pvt_gethostbyaddr_a(
        const char *addr,
        int len,
        int type,
@@ -118,34 +189,263 @@ int ldap_int_gethostbyaddr_a(
        struct hostent **result,
        int *herrno_ptr )
 {
-#ifdef HAVE_GETHOSTBYADDR_R
-       int r;
+#if defined( HAVE_GETHOSTBYADDR_R )
+
+# undef NEED_SAFE_REALLOC
+# define NEED_SAFE_REALLOC   
+       int r=-1;
        int buflen=BUFSTART;
-       if (safe_realloc( buf, buflen)) {
-               for(;buflen<BUFMAX;) {
-                       r = gethostbyaddr_r( addr, len, type,
-                               resbuf, *buf, buflen, 
-                               result, herrno_ptr );
-#ifdef NETDB_INTERNAL
-                       if ((r<0) &&
-                               (*herrno_ptr==NETDB_INTERNAL) &&
-                               (errno==ERANGE))
-                       {
-                               if (safe_realloc( buf, buflen*=2))
-                                       continue;
-                       }
-#endif
+       *buf = NULL;   
+       for(;buflen<BUFMAX;) {
+               if (safe_realloc( buf, buflen )==NULL)
                        return r;
+#if (GETHOSTBYADDR_R_NARGS < 8)
+               r = ((*result=gethostbyaddr_r( addr, len, type,
+                                              resbuf, *buf, buflen, 
+                                              herrno_ptr )) == NULL) ?
+                   -1 : 0;
+#else
+               r = gethostbyaddr_r( addr, len, type,
+                       resbuf, *buf, buflen, 
+                       result, herrno_ptr );
+#endif
+
+#ifdef NETDB_INTERNAL
+               if ((r<0) &&
+                       (*herrno_ptr==NETDB_INTERNAL) &&
+                       (errno==ERANGE))
+               {
+                       buflen*=2;
+                       continue;
                }
+#endif
+               return r;
        }
-
+       return -1;
+#elif defined( LDAP_R_COMPILE )
+# undef NEED_COPY_HOSTENT
+# define NEED_COPY_HOSTENT   
+       struct hostent *he;
+       int     retval;
+       *buf = NULL;   
+       
+       ldap_pvt_thread_mutex_lock( &ldap_int_gethostby_mutex );
+       
+       he = gethostbyaddr( addr, len, type );
+       
+       if (he==NULL) {
+               *herrno_ptr = h_errno;
+               retval = -1;
+       } else if (copy_hostent( resbuf, buf, he )<0) {
+               *herrno_ptr = -1;
+               retval = -1;
+       } else {
+               *result = resbuf;
+               retval = 0;
+       }
+       
+       ldap_pvt_thread_mutex_unlock( &ldap_int_gethostby_mutex );
+       
+       return retval;   
 #else /* gethostbyaddr() */
+       *buf = NULL;   
        *result = gethostbyaddr( addr, len, type );
 
        if (*result!=NULL) {
                return 0;
        }
+       return -1;
 #endif 
+}
+/* 
+ * ldap_int_utils_init() should be called before any other function.
+ */
 
+void ldap_int_utils_init( void )
+{
+       static int done=0;
+       if (done)
+         return;
+       done=1;
+
+#ifdef LDAP_R_COMPILE
+
+#if !defined( USE_CTIME_R ) && !defined( HAVE_REENTRANT_FUNCTIONS )
+       ldap_pvt_thread_mutex_init( &ldap_int_ctime_mutex );
+#endif
+
+#if !defined( HAVE_GETHOSTBYNAME_R ) || !defined( HAVE_GETHOSTBYADDR_R )
+       ldap_pvt_thread_mutex_init( &ldap_int_gethostby_mutex );
+#endif
+
+#ifdef HAVE_RES_QUERY
+       ldap_pvt_thread_mutex_init( &ldap_int_resolv_mutex );
+#endif
+
+       /* call other module init functions here... */
+#endif
+}
+
+#if defined( NEED_COPY_HOSTENT )
+# undef NEED_SAFE_REALLOC
+#define NEED_SAFE_REALLOC
+
+static char *cpy_aliases( char ***tgtio, char *buf, char **src )
+{
+       int len;
+       char **tgt=*tgtio;
+       for( ; (*src) ; src++ ) {
+               len = strlen( *src ) + 1;
+               AC_MEMCPY( buf, *src, len );
+               *tgt++=buf;
+               buf+=len;
+       }
+       *tgtio=tgt;   
+       return buf;
+}
+
+static char *cpy_addresses( char ***tgtio, char *buf, char **src, int len )
+{
+       char **tgt=*tgtio;
+       for( ; (*src) ; src++ ) {
+               AC_MEMCPY( buf, *src, len );
+               *tgt++=buf;
+               buf+=len;
+       }
+       *tgtio=tgt;      
+       return buf;
+}
+
+static int copy_hostent( struct hostent *res, char **buf, struct hostent * src )
+{
+       char    **p;
+       char    **tp;
+       char    *tbuf;
+       int     name_len;
+       int     n_alias=0;
+       int     total_alias_len=0;
+       int     n_addr=0;
+       int     total_addr_len;
+       int     total_len;
+         
+       /* calculate the size needed for the buffer */
+       name_len = strlen( src->h_name ) + 1;
+       
+       if( src->h_aliases != NULL ) {
+               for( p = src->h_aliases; (*p) != NULL; p++ ) {
+                       total_alias_len += strlen( *p ) + 1;
+                       n_alias++; 
+               }
+       }
+
+       if( src->h_addr_list != NULL ) {
+               for( p = src->h_addr_list; (*p) != NULL; p++ ) {
+                       n_addr++;
+               }
+               total_addr_len = n_addr * src->h_length;
+       }
+       
+       total_len = (n_alias + n_addr + 2) * sizeof( char * ) +
+               total_addr_len + total_alias_len + name_len;
+       
+       if (safe_realloc( buf, total_len )) {                    
+               tp = (char **) *buf;
+               tbuf = *buf + (n_alias + n_addr + 2) * sizeof( char * );
+               AC_MEMCPY( res, src, sizeof( struct hostent ) );
+               /* first the name... */
+               AC_MEMCPY( tbuf, src->h_name, name_len );
+               res->h_name = tbuf; tbuf+=name_len;
+               /* now the aliases */
+               res->h_aliases = tp;
+               if ( src->h_aliases != NULL ) {
+                       tbuf = cpy_aliases( &tp, tbuf, src->h_aliases );
+               }
+               *tp++=NULL;
+               /* finally the addresses */
+               res->h_addr_list = tp;
+               if ( src->h_addr_list != NULL ) {
+                       tbuf = cpy_addresses( &tp, tbuf, src->h_addr_list, src->h_length );
+               }
+               *tp++=NULL;
+               return 0;
+       }
        return -1;
 }
+#endif
+
+#if defined( NEED_SAFE_REALLOC )
+static char *safe_realloc( char **buf, int len )
+{
+       char *tmpbuf;
+       tmpbuf = LDAP_REALLOC( *buf, len );
+       if (tmpbuf) {
+               *buf=tmpbuf;
+       } 
+       return tmpbuf;
+}
+#endif
+
+char * ldap_pvt_get_fqdn( char *name )
+{
+       char *fqdn, *ha_buf;
+       char hostbuf[MAXHOSTNAMELEN+1];
+       struct hostent *hp, he_buf;
+       int rc, local_h_errno;
+
+       if( name == NULL ) {
+               if( gethostname( hostbuf, MAXHOSTNAMELEN ) == 0 ) {
+                       hostbuf[MAXHOSTNAMELEN] = '\0';
+                       name = hostbuf;
+               } else {
+                       name = "localhost";
+               }
+       }
+
+       rc = ldap_pvt_gethostbyname_a( name,
+               &he_buf, &ha_buf, &hp, &local_h_errno );
+
+       if( rc < 0 || hp == NULL || hp->h_name == NULL ) {
+               fqdn = LDAP_STRDUP( name );
+       } else {
+               fqdn = LDAP_STRDUP( hp->h_name );
+       }
+
+       LDAP_FREE( ha_buf );
+       return fqdn;
+}
+
+#if defined( HAVE_GETADDRINFO ) && !defined( HAVE_GAI_STRERROR )
+char *ldap_pvt_gai_strerror (int code) {
+       static struct {
+               int code;
+               const char *msg;
+       } values[] = {
+#ifdef EAI_ADDRFAMILY
+               { EAI_ADDRFAMILY, "Address family for hostname not supported" },
+#endif
+               { EAI_AGAIN, "Temporary failure in name resolution" },
+               { EAI_BADFLAGS, "Bad value for ai_flags" },
+               { EAI_FAIL, "Non-recoverable failure in name resolution" },
+               { EAI_FAMILY, "ai_family not supported" },
+               { EAI_MEMORY, "Memory allocation failure" },
+#ifdef EAI_NODATA
+               { EAI_NODATA, "No address associated with hostname" },
+#endif    
+               { EAI_NONAME, "Name or service not known" },
+               { EAI_SERVICE, "Servname not supported for ai_socktype" },
+               { EAI_SOCKTYPE, "ai_socktype not supported" },
+               { EAI_SYSTEM, "System error" },
+               { 0, NULL }
+       };
+
+       int i;
+
+       for ( i = 0; values[i].msg != NULL; i++ ) {
+               if ( values[i].code == code ) {
+                       return (char *) values[i].msg;
+               }
+       }
+       
+       return "Unknown error";
+}
+#endif