X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=servers%2Fslapd%2Fdaemon.c;h=e42565af46a7fc541a696d07c6ec4d69ffcafb7d;hb=331e6275418bda0735d4952f64d51518881d10a8;hp=a0f6a0b11bb1ed0ea56d67f48b264fb2274e17a0;hpb=5b218b76718546e5ef0172b4e811b933b66ac434;p=openldap diff --git a/servers/slapd/daemon.c b/servers/slapd/daemon.c index a0f6a0b11b..e42565af46 100644 --- a/servers/slapd/daemon.c +++ b/servers/slapd/daemon.c @@ -1,60 +1,75 @@ +/* $OpenLDAP$ */ +/* + * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved. + * COPYING RESTRICTIONS APPLY, see COPYRIGHT file + */ + #include "portable.h" #include #include #include -#include #include #include #include #include -#include "ldapconfig.h" +#include "ldap_pvt.h" +#include "ldap_pvt_thread.h" +#include "lutil.h" #include "slap.h" #ifdef HAVE_TCPD #include +#define SLAP_STRING_UNKNOWN STRING_UNKNOWN int allow_severity = LOG_INFO; int deny_severity = LOG_NOTICE; -#endif /* TCP Wrappers */ +#else /* ! TCP Wrappers */ +#define SLAP_STRING_UNKNOWN "unknown" +#endif /* ! TCP Wrappers */ + +#ifdef LDAP_PF_LOCAL +#include +/* this should go in as soon as it is accepted */ +#define LDAPI_MOD_URLEXT "x-mod" +#endif /* LDAP_PF_LOCAL */ + +#ifdef LDAP_PF_INET6 +int slap_inet4or6 = AF_UNSPEC; +#else +int slap_inet4or6 = AF_INET; +#endif /* globals */ -int dtblsize; - -#ifdef HAVE_WINSOCK2 -/* forward reference */ -void hit_socket(); -/* In wsa_err.c */ -char *WSAGetLastErrorString(); - -#define WAKE_LISTENER \ -do {\ - if( wake ) {\ - ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\ - hit_socket();\ - }\ -} while(0) -#else -#define WAKE_LISTENER \ -do {\ - if( wake ) {\ - ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\ - }\ -} while(0) -#endif - -static int daemon_initialized = 0; -static ldap_pvt_thread_t listener_tid; -static volatile sig_atomic_t slapd_shutdown = 0; -static volatile sig_atomic_t slapd_listener = 0; -void sockinit(); - -struct slap_daemon { +time_t starttime; +ber_socket_t dtblsize; + +Listener **slap_listeners = NULL; + +#define SLAPD_LISTEN 10 + +static ber_socket_t wake_sds[2]; + +#if defined(NO_THREADS) || defined(HAVE_GNU_PTH) +static int waking; +#define WAKE_LISTENER(w) \ +((w && !waking) ? tcp_write( wake_sds[1], "0", 1 ), waking=1 : 0) +#else +#define WAKE_LISTENER(w) \ +do { if (w) tcp_write( wake_sds[1], "0", 1 ); } while(0) +#endif + +#ifndef HAVE_WINSOCK +static +#endif +volatile sig_atomic_t slapd_shutdown = 0, slapd_gentle_shutdown = 0; + +static struct slap_daemon { ldap_pvt_thread_mutex_t sd_mutex; - int sd_nactives; + ber_socket_t sd_nactives; #ifndef HAVE_WINSOCK /* In winsock, accept() returns values higher than dtblsize @@ -65,12 +80,119 @@ struct slap_daemon { fd_set sd_actives; fd_set sd_readers; fd_set sd_writers; -} slap_daemon; +} slap_daemon; + + + +#ifdef HAVE_SLP +/* + * SLP related functions + */ +#include + +#define LDAP_SRVTYPE_PREFIX "service:ldap://" +#define LDAPS_SRVTYPE_PREFIX "service:ldaps://" +static char** slapd_srvurls = NULL; +static SLPHandle slapd_hslp = 0; + +void slapd_slp_init( const char* urls ) { + int i; + + slapd_srvurls = ldap_str2charray( urls, " " ); + + if( slapd_srvurls == NULL ) return; + + /* find and expand INADDR_ANY URLs */ + for( i=0; slapd_srvurls[i] != NULL; i++ ) { + if( strcmp( slapd_srvurls[i], "ldap:///" ) == 0) { + char *host = ldap_pvt_get_fqdn( NULL ); + if ( host != NULL ) { + slapd_srvurls[i] = (char *) ch_realloc( slapd_srvurls[i], + strlen( host ) + + sizeof( LDAP_SRVTYPE_PREFIX ) ); + strcpy( lutil_strcopy(slapd_srvurls[i], + LDAP_SRVTYPE_PREFIX ), host ); + + ch_free( host ); + } + + } else if ( strcmp( slapd_srvurls[i], "ldaps:///" ) == 0) { + char *host = ldap_pvt_get_fqdn( NULL ); + if ( host != NULL ) { + slapd_srvurls[i] = (char *) ch_realloc( slapd_srvurls[i], + strlen( host ) + + sizeof( LDAPS_SRVTYPE_PREFIX ) ); + strcpy( lutil_strcopy(slapd_srvurls[i], + LDAPS_SRVTYPE_PREFIX ), host ); + + ch_free( host ); + } + } + } + + /* open the SLP handle */ + SLPOpen( "en", 0, &slapd_hslp ); +} + +void slapd_slp_deinit() { + if( slapd_srvurls == NULL ) return; + + ldap_charray_free( slapd_srvurls ); + slapd_srvurls = NULL; + + /* close the SLP handle */ + SLPClose( slapd_hslp ); +} + +void slapd_slp_regreport( + SLPHandle hslp, + SLPError errcode, + void* cookie ) +{ + /* empty report */ +} + +void slapd_slp_reg() { + int i; + + if( slapd_srvurls == NULL ) return; + + for( i=0; slapd_srvurls[i] != NULL; i++ ) { + if( strncmp( slapd_srvurls[i], LDAP_SRVTYPE_PREFIX, + sizeof( LDAP_SRVTYPE_PREFIX ) - 1 ) == 0 || + strncmp( slapd_srvurls[i], LDAPS_SRVTYPE_PREFIX, + sizeof( LDAPS_SRVTYPE_PREFIX ) - 1 ) == 0 ) + { + SLPReg( slapd_hslp, + slapd_srvurls[i], + SLP_LIFETIME_MAXIMUM, + "ldap", + "", + 1, + slapd_slp_regreport, + NULL ); + } + } +} + +void slapd_slp_dereg() { + int i; + + if( slapd_srvurls == NULL ) return; + + for( i=0; slapd_srvurls[i] != NULL; i++ ) { + SLPDereg( slapd_hslp, + slapd_srvurls[i], + slapd_slp_regreport, + NULL ); + } +} +#endif /* HAVE_SLP */ /* * Add a descriptor to daemon control */ -static void slapd_add(int s) { +static void slapd_add(ber_socket_t s) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); assert( !FD_ISSET( s, &slap_daemon.sd_actives )); @@ -83,89 +205,751 @@ static void slapd_add(int s) { } #endif - FD_SET( (unsigned) s, &slap_daemon.sd_actives ); - FD_SET( (unsigned) s, &slap_daemon.sd_readers ); + slap_daemon.sd_nactives++; + + FD_SET( s, &slap_daemon.sd_actives ); + FD_SET( s, &slap_daemon.sd_readers ); - Debug( LDAP_DEBUG_CONNS, "daemon: added %d%s%s\n", s, +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slapd_add: added %ld%s%s\n", (long)s, + FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "", + FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" ); +#else + Debug( LDAP_DEBUG_CONNS, "daemon: added %ld%s%s\n", + (long) s, FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "", FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" ); - +#endif ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); } /* * Remove the descriptor from daemon control */ -void slapd_remove(int s) { +void slapd_remove(ber_socket_t s, int wake) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - assert( FD_ISSET( s, &slap_daemon.sd_actives )); + slap_daemon.sd_nactives--; - Debug( LDAP_DEBUG_CONNS, "daemon: removing %d%s%s\n", s, +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slapd_remove: removing %ld%s%s\n", (long) s, + FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "", + FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" ); +#else + Debug( LDAP_DEBUG_CONNS, "daemon: removing %ld%s%s\n", + (long) s, FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "", FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" ); - - FD_CLR( (unsigned) s, &slap_daemon.sd_actives ); - FD_CLR( (unsigned) s, &slap_daemon.sd_readers ); - FD_CLR( (unsigned) s, &slap_daemon.sd_writers ); +#endif + FD_CLR( s, &slap_daemon.sd_actives ); + FD_CLR( s, &slap_daemon.sd_readers ); + FD_CLR( s, &slap_daemon.sd_writers ); ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); + WAKE_LISTENER(wake || slapd_gentle_shutdown == 2); } -void slapd_clr_write(int s, int wake) { +void slapd_clr_write(ber_socket_t s, int wake) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - WAKE_LISTENER; - assert( FD_ISSET( (unsigned) s, &slap_daemon.sd_actives) ); - FD_CLR( (unsigned) s, &slap_daemon.sd_writers ); + assert( FD_ISSET( s, &slap_daemon.sd_actives) ); + FD_CLR( s, &slap_daemon.sd_writers ); ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); + WAKE_LISTENER(wake); } -void slapd_set_write(int s, int wake) { +void slapd_set_write(ber_socket_t s, int wake) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - WAKE_LISTENER; - assert( FD_ISSET( s, &slap_daemon.sd_actives) ); - FD_SET( (unsigned) s, &slap_daemon.sd_writers ); + if (!FD_ISSET(s, &slap_daemon.sd_writers)) + FD_SET( (unsigned) s, &slap_daemon.sd_writers ); ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); + WAKE_LISTENER(wake); } -void slapd_clr_read(int s, int wake) { +void slapd_clr_read(ber_socket_t s, int wake) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - WAKE_LISTENER; assert( FD_ISSET( s, &slap_daemon.sd_actives) ); - FD_CLR( (unsigned) s, &slap_daemon.sd_readers ); + FD_CLR( s, &slap_daemon.sd_readers ); ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); - + WAKE_LISTENER(wake); } -void slapd_set_read(int s, int wake) { +void slapd_set_read(ber_socket_t s, int wake) { ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - WAKE_LISTENER; assert( FD_ISSET( s, &slap_daemon.sd_actives) ); - FD_SET( (unsigned) s, &slap_daemon.sd_readers ); + if (!FD_ISSET(s, &slap_daemon.sd_readers)) + FD_SET( s, &slap_daemon.sd_readers ); ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); + WAKE_LISTENER(wake); } -static void slapd_close(int s) { - slapd_remove(s); - - Debug( LDAP_DEBUG_CONNS, "daemon: closing %d\n", s, 0, 0 ); +static void slapd_close(ber_socket_t s) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, "slapd_close: closing %ld\n", (long)s, 0, 0); +#else + Debug( LDAP_DEBUG_CONNS, "daemon: closing %ld\n", + (long) s, 0, 0 ); +#endif tcp_close(s); } -int -set_socket( struct sockaddr_in *addr ) +static void slap_free_listener_addresses(struct sockaddr **sal) +{ + struct sockaddr **sap; + + if (sal == NULL) { + return; + } + + for (sap = sal; *sap != NULL; sap++) { + ch_free(*sap); + } + + ch_free(sal); +} + +#if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD) +static int get_url_perms( + char **exts, + mode_t *perms, + int *crit ) +{ + int i; + + assert( exts ); + assert( perms ); + assert( crit ); + + *crit = 0; + for ( i = 0; exts[ i ]; i++ ) { + char *type = exts[ i ]; + int c = 0; + + if ( type[ 0 ] == '!' ) { + c = 1; + type++; + } + + if ( strncasecmp( type, LDAPI_MOD_URLEXT "=", sizeof(LDAPI_MOD_URLEXT "=") - 1 ) == 0 ) { + char *value = type + + ( sizeof(LDAPI_MOD_URLEXT "=") - 1 ); + mode_t p = 0; + int j; + + switch (strlen(value)) { + case 4: + /* skip leading '0' */ + if ( value[ 0 ] != '0' ) { + return LDAP_OTHER; + } + value++; + + case 3: + for ( j = 0; j < 3; j++) { + int v; + + v = value[ j ] - '0'; + + if ( v < 0 || v > 7 ) { + return LDAP_OTHER; + } + + p |= v << 3*(2-j); + } + break; + + case 10: + for ( j = 1; j < 10; j++ ) { + static mode_t m[] = { 0, + S_IRUSR, S_IWUSR, S_IXUSR, + S_IRGRP, S_IWGRP, S_IXGRP, + S_IROTH, S_IWOTH, S_IXOTH + }; + static char c[] = "-rwxrwxrwx"; + + if ( value[ j ] == c[ j ] ) { + p |= m[ j ]; + + } else if ( value[ j ] != '-' ) { + return LDAP_OTHER; + } + } + break; + + default: + return LDAP_OTHER; + } + + *crit = c; + *perms = p; + + return LDAP_SUCCESS; + } + } + + return LDAP_OTHER; +} +#endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */ + +/* port = 0 indicates AF_LOCAL */ +static int slap_get_listener_addresses( + const char *host, + unsigned short port, + struct sockaddr ***sal) { - int tcps = -1; - if ( !daemon_initialized ) sockinit(); + struct sockaddr **sap; + +#ifdef LDAP_PF_LOCAL + if ( port == 0 ) { + *sal = ch_malloc(2 * sizeof(void *)); + if (*sal == NULL) { + return -1; + } + + sap = *sal; + *sap = ch_malloc(sizeof(struct sockaddr_un)); + if (*sap == NULL) + goto errexit; + sap[1] = NULL; + + if ( strlen(host) > + (sizeof(((struct sockaddr_un *)*sap)->sun_path) - 1) ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_get_listener_addresses: domain socket path (%s) " + "too long in URL\n", host, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: domain socket path (%s) too long in URL", + host, 0, 0); +#endif + goto errexit; + } + + (void)memset( (void *)*sap, '\0', sizeof(struct sockaddr_un) ); + (*sap)->sa_family = AF_LOCAL; + strcpy( ((struct sockaddr_un *)*sap)->sun_path, host ); + } else +#endif + { +#ifdef HAVE_GETADDRINFO + struct addrinfo hints, *res, *sai; + int n, err; + char serv[7]; + + memset( &hints, '\0', sizeof(hints) ); + hints.ai_flags = AI_PASSIVE; + hints.ai_socktype = SOCK_STREAM; + hints.ai_family = slap_inet4or6; + snprintf(serv, sizeof serv, "%d", port); + + if ( (err = getaddrinfo(host, serv, &hints, &res)) ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_get_listener_addresses: getaddrinfo failed: %s\n", + AC_GAI_STRERROR(err), 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, "daemon: getaddrinfo failed: %s\n", + AC_GAI_STRERROR(err), 0, 0); +#endif + return -1; + } + + sai = res; + for (n=2; (sai = sai->ai_next) != NULL; n++) { + /* EMPTY */ ; + } + *sal = ch_calloc(n, sizeof(void *)); + if (*sal == NULL) { + return -1; + } + + sap = *sal; + *sap = NULL; + + for ( sai=res; sai; sai=sai->ai_next ) { + if( sai->ai_addr == NULL ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_get_listener_addresses: " + "getaddrinfo ai_addr is NULL?\n", 0, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, "slap_get_listener_addresses: " + "getaddrinfo ai_addr is NULL?\n", 0, 0, 0 ); +#endif + freeaddrinfo(res); + goto errexit; + } + + switch (sai->ai_family) { +# ifdef LDAP_PF_INET6 + case AF_INET6: + *sap = ch_malloc(sizeof(struct sockaddr_in6)); + if (*sap == NULL) { + freeaddrinfo(res); + goto errexit; + } + *(struct sockaddr_in6 *)*sap = + *((struct sockaddr_in6 *)sai->ai_addr); + break; +# endif + case AF_INET: + *sap = ch_malloc(sizeof(struct sockaddr_in)); + if (*sap == NULL) { + freeaddrinfo(res); + goto errexit; + } + *(struct sockaddr_in *)*sap = + *((struct sockaddr_in *)sai->ai_addr); + break; + default: + *sap = NULL; + break; + } + + if (*sap != NULL) { + (*sap)->sa_family = sai->ai_family; + sap++; + *sap = NULL; + } + } + + freeaddrinfo(res); +#else + int i, n = 1; + struct in_addr in; + struct hostent *he = NULL; + + if ( host == NULL ) { + in.s_addr = htonl(INADDR_ANY); + + } else if ( !inet_aton( host, &in ) ) { + he = gethostbyname( host ); + if( he == NULL ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_get_listener_addresses: invalid host %s\n", host, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: invalid host %s", host, 0, 0); +#endif + return -1; + } + for (n = 0; he->h_addr_list[n]; n++) ; + } + + *sal = ch_malloc((n+1) * sizeof(void *)); + if (*sal == NULL) { + return -1; + } + + sap = *sal; + for ( i = 0; isa_family = AF_INET; + ((struct sockaddr_in *)sap[i])->sin_port = htons(port); + if (he) { + AC_MEMCPY( &((struct sockaddr_in *)sap[i])->sin_addr, he->h_addr_list[i], sizeof(struct in_addr) ); + } else { + AC_MEMCPY( &((struct sockaddr_in *)sap[i])->sin_addr, &in, sizeof(struct in_addr) ); + } + } + sap[i] = NULL; +#endif + } + + return 0; + +errexit: + slap_free_listener_addresses(*sal); + return -1; +} + +static int slap_open_listener( + const char* url, + int *listeners, + int *cur + ) +{ + int num, tmp, rc; + Listener l; + Listener *li; + LDAPURLDesc *lud; + unsigned short port; + int err, addrlen = 0; + struct sockaddr **sal, **psal; + int socktype = SOCK_STREAM; /* default to COTS */ + +#if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD) + /* + * use safe defaults + */ + int crit = 1; +#endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */ + + rc = ldap_url_parse( url, &lud ); + + if( rc != LDAP_URL_SUCCESS ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slap_open_listener: listen URL \"%s\" parse error %d\n", + url, rc , 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: listen URL \"%s\" parse error=%d\n", + url, rc, 0 ); +#endif + return rc; + } + + l.sl_url.bv_val = NULL; + +#ifndef HAVE_TLS + if( ldap_pvt_url_scheme2tls( lud->lud_scheme ) ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: TLS is not supported (%s)\n", url, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: TLS not supported (%s)\n", + url, 0, 0 ); +#endif + ldap_free_urldesc( lud ); + return -1; + } + + if(! lud->lud_port ) { + lud->lud_port = LDAP_PORT; + } + +#else + l.sl_is_tls = ldap_pvt_url_scheme2tls( lud->lud_scheme ); + + if(! lud->lud_port ) { + lud->lud_port = l.sl_is_tls ? LDAPS_PORT : LDAP_PORT; + } +#endif + + port = (unsigned short) lud->lud_port; + + tmp = ldap_pvt_url_scheme2proto(lud->lud_scheme); + if ( tmp == LDAP_PROTO_IPC ) { +#ifdef LDAP_PF_LOCAL + if ( lud->lud_host == NULL || lud->lud_host[0] == '\0' ) { + err = slap_get_listener_addresses(LDAPI_SOCK, 0, &sal); + } else { + err = slap_get_listener_addresses(lud->lud_host, 0, &sal); + } +#else + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: URL scheme is not supported: %s\n", url, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, "daemon: URL scheme not supported: %s", + url, 0, 0); +#endif + ldap_free_urldesc( lud ); + return -1; +#endif + } else { + if( lud->lud_host == NULL || lud->lud_host[0] == '\0' + || strcmp(lud->lud_host, "*") == 0 ) + { + err = slap_get_listener_addresses(NULL, port, &sal); + } else { + err = slap_get_listener_addresses(lud->lud_host, port, &sal); + } + } +#ifdef LDAP_CONNECTIONLESS + l.sl_is_udp = ( tmp == LDAP_PROTO_UDP ); +#endif + +#if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD) + if ( lud->lud_exts ) { + err = get_url_perms( lud->lud_exts, &l.sl_perms, &crit ); + } else { + l.sl_perms = S_IRWXU | S_IRWXO; + } +#endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */ + + ldap_free_urldesc( lud ); + if ( err ) { + return -1; + } + + /* If we got more than one address returned, we need to make space + * for it in the slap_listeners array. + */ + for ( num=0; sal[num]; num++ ); + if ( num > 1 ) { + *listeners += num-1; + slap_listeners = ch_realloc( slap_listeners, (*listeners + 1) * sizeof(Listener *) ); + } + + psal = sal; + while ( *sal != NULL ) { + switch( (*sal)->sa_family ) { + case AF_INET: +#ifdef LDAP_PF_INET6 + case AF_INET6: +#endif +#ifdef LDAP_PF_LOCAL + case AF_LOCAL: +#endif + break; + default: + sal++; + continue; + } +#ifdef LDAP_CONNECTIONLESS + if( l.sl_is_udp ) socktype = SOCK_DGRAM; +#endif + l.sl_sd = socket( (*sal)->sa_family, socktype, 0); + if ( l.sl_sd == AC_SOCKET_INVALID ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slap_open_listener: socket() failed errno=%d (%s)\n", + err, sock_errstr(err), 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: socket() failed errno=%d (%s)\n", err, + sock_errstr(err), 0 ); +#endif + sal++; + continue; + } +#ifndef HAVE_WINSOCK + if ( l.sl_sd >= dtblsize ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slap_open_listener: listener descriptor %ld is too " + "great %ld\n", (long)l.sl_sd, (long)dtblsize, 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: listener descriptor %ld is too great %ld\n", + (long) l.sl_sd, (long) dtblsize, 0 ); +#endif + tcp_close( l.sl_sd ); + sal++; + continue; + } +#endif +#ifdef LDAP_PF_LOCAL + if ( (*sal)->sa_family == AF_LOCAL ) { + unlink ( ((struct sockaddr_un *)*sal)->sun_path ); + } else +#endif + { +#ifdef SO_REUSEADDR + /* enable address reuse */ + tmp = 1; + rc = setsockopt( l.sl_sd, SOL_SOCKET, SO_REUSEADDR, + (char *) &tmp, sizeof(tmp) ); + if ( rc == AC_SOCKET_ERROR ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: setsockopt( %ld, SO_REUSEADDR ) " + "failed errno %d (%s)\n", (long)l.sl_sd, err, + sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, + "slapd(%ld): setsockopt(SO_REUSEADDR) failed errno=%d (%s)\n", + (long) l.sl_sd, err, sock_errstr(err) ); +#endif + } +#endif + } + + switch( (*sal)->sa_family ) { + case AF_INET: + addrlen = sizeof(struct sockaddr_in); + break; +#ifdef LDAP_PF_INET6 + case AF_INET6: +#ifdef IPV6_V6ONLY + /* Try to use IPv6 sockets for IPv6 only */ + tmp = 1; + rc = setsockopt( l.sl_sd, IPPROTO_IPV6, IPV6_V6ONLY, + (char *) &tmp, sizeof(tmp) ); + if ( rc == AC_SOCKET_ERROR ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: setsockopt( %ld, IPV6_V6ONLY ) failed errno %d (%s)\n", + (long)l.sl_sd, err, sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, + "slapd(%ld): setsockopt(IPV6_V6ONLY) failed errno=%d (%s)\n", + (long) l.sl_sd, err, sock_errstr(err) ); +#endif + } +#endif + addrlen = sizeof(struct sockaddr_in6); + break; +#endif +#ifdef LDAP_PF_LOCAL + case AF_LOCAL: + addrlen = sizeof(struct sockaddr_un); + break; +#endif + } + + if (bind(l.sl_sd, *sal, addrlen)) { + err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: bind(%ld) failed errno=%d (%s)\n", + (long)l.sl_sd, err, sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, "daemon: bind(%ld) failed errno=%d (%s)\n", + (long) l.sl_sd, err, sock_errstr(err) ); +#endif + tcp_close( l.sl_sd ); + sal++; + continue; + } + + switch ( (*sal)->sa_family ) { +#ifdef LDAP_PF_LOCAL + case AF_LOCAL: { + char *addr = ((struct sockaddr_un *)*sal)->sun_path; +#if 0 /* don't muck with socket perms */ + if ( chmod( addr, l.sl_perms ) < 0 && crit ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: fchmod(%ld) failed errno=%d (%s)\n", + (long)l.sl_sd, err, sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, "daemon: fchmod(%ld) failed errno=%d (%s)", + (long) l.sl_sd, err, sock_errstr(err) ); +#endif + tcp_close( l.sl_sd ); + slap_free_listener_addresses(psal); + return -1; + } +#endif + l.sl_name.bv_len = strlen(addr) + sizeof("PATH=") - 1; + l.sl_name.bv_val = ber_memalloc( l.sl_name.bv_len + 1 ); + snprintf( l.sl_name.bv_val, l.sl_name.bv_len + 1, + "PATH=%s", addr ); + } break; +#endif /* LDAP_PF_LOCAL */ + + case AF_INET: { + char *s; +#if defined( HAVE_GETADDRINFO ) && defined( HAVE_INET_NTOP ) + char addr[INET_ADDRSTRLEN]; + inet_ntop( AF_INET, &((struct sockaddr_in *)*sal)->sin_addr, + addr, sizeof(addr) ); + s = addr; +#else + s = inet_ntoa( ((struct sockaddr_in *) *sal)->sin_addr ); +#endif + port = ntohs( ((struct sockaddr_in *)*sal) ->sin_port ); + l.sl_name.bv_val = ber_memalloc( sizeof("IP=255.255.255.255:65535") ); + snprintf( l.sl_name.bv_val, sizeof("IP=255.255.255.255:65535"), + "IP=%s:%d", + s != NULL ? s : SLAP_STRING_UNKNOWN, port ); + l.sl_name.bv_len = strlen( l.sl_name.bv_val ); + } break; + +#ifdef LDAP_PF_INET6 + case AF_INET6: { + char addr[INET6_ADDRSTRLEN]; + inet_ntop( AF_INET6, &((struct sockaddr_in6 *)*sal)->sin6_addr, + addr, sizeof addr); + port = ntohs( ((struct sockaddr_in6 *)*sal)->sin6_port ); + l.sl_name.bv_len = strlen(addr) + sizeof("IP= 65535"); + l.sl_name.bv_val = ber_memalloc( l.sl_name.bv_len ); + snprintf( l.sl_name.bv_val, l.sl_name.bv_len, "IP=%s %d", + addr, port ); + l.sl_name.bv_len = strlen( l.sl_name.bv_val ); + } break; +#endif /* LDAP_PF_INET6 */ + + default: +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_open_listener: unsupported address family (%d)\n", + (int)(*sal)->sa_family, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, "daemon: unsupported address family (%d)\n", + (int) (*sal)->sa_family, 0, 0 ); +#endif + break; + } + + AC_MEMCPY(&l.sl_sa, *sal, addrlen); + ber_str2bv( url, 0, 1, &l.sl_url); + li = ch_malloc( sizeof( Listener ) ); + *li = l; + slap_listeners[*cur] = li; + (*cur)++; + sal++; + + } /* while ( *sal != NULL ) */ + + slap_free_listener_addresses(psal); + + if ( l.sl_url.bv_val == NULL ) + { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, RESULTS, + "slap_open_listener: failed on %s\n", url, 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, + "slap_open_listener: failed on %s\n", url, 0, 0 ); +#endif + return -1; + } + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, RESULTS, + "slap_open_listener: daemon initialized %s\n", + l.sl_url.bv_val, 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, "daemon: initialized %s\n", + l.sl_url.bv_val, 0, 0 ); +#endif + return 0; +} + +static int sockinit(void); +static int sockdestroy(void); + +int slapd_daemon_init( const char *urls ) +{ + int i, j, n, rc; + char **u; + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ARGS, + "slapd_daemon_init: %s\n", urls ? urls : "", 0, 0 ); +#else + Debug( LDAP_DEBUG_ARGS, "daemon_init: %s\n", + urls ? urls : "", 0, 0 ); +#endif + if( (rc = sockinit()) != 0 ) { + return rc; + } #ifdef HAVE_SYSCONF dtblsize = sysconf( _SC_OPEN_MAX ); @@ -181,133 +965,284 @@ set_socket( struct sockaddr_in *addr ) } #endif /* !FD_SETSIZE */ - if( addr != NULL ) { - int tmp; + /* open a pipe (or something equivalent connected to itself). + * we write a byte on this fd whenever we catch a signal. The main + * loop will be select'ing on this socket, and will wake up when + * this byte arrives. + */ + if( (rc = lutil_pair( wake_sds )) < 0 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slap_daemon_init: lutil_pair() failed rc=%d\n", rc, 0, 0); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: lutil_pair() failed rc=%d\n", rc, 0, 0 ); +#endif + return rc; + } - if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) { -#ifndef HAVE_WINSOCK - int err = errno; - Debug( LDAP_DEBUG_ANY, - "daemon: socket() failed errno %d (%s)\n", err, - err > -1 && err < sys_nerr ? sys_errlist[err] : - "unknown", 0 ); + FD_ZERO( &slap_daemon.sd_readers ); + FD_ZERO( &slap_daemon.sd_writers ); + + if( urls == NULL ) { + urls = "ldap:///"; + } + + u = ldap_str2charray( urls, " " ); + + if( u == NULL || u[0] == NULL ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slap_daemon_init: no urls (%s) provided.\n", urls, 0, 0 ); #else - Debug( LDAP_DEBUG_ANY, - "daemon: socket() failed errno %d (%s)\n", - WSAGetLastError(), - WSAGetLastErrorString(), 0 ); + Debug( LDAP_DEBUG_ANY, "daemon_init: no urls (%s) provided.\n", + urls, 0, 0 ); #endif - exit( 1 ); - } + return -1; + } -#ifndef HAVE_WINSOCK - if ( tcps >= dtblsize ) { - Debug( LDAP_DEBUG_ANY, - "daemon: listener descriptor %d is too great\n", - tcps, dtblsize, 0 ); - exit( 1 ); - } + for( i=0; u[i] != NULL; i++ ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slap_daemon_init: listen on %s\n.", u[i], 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, "daemon_init: listen on %s\n", + u[i], 0, 0 ); #endif + } -#ifdef SO_REUSEADDR - tmp = 1; - if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR, - (char *) &tmp, sizeof(tmp) ) == -1 ) - { - int err = errno; - Debug( LDAP_DEBUG_ANY, - "slapd(%d): setsockopt() failed errno %d (%s)\n", - tcps, err, - err > -1 && err < sys_nerr - ? sys_errlist[err] : "unknown" ); - } + if( i == 0 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_daemon_init: no listeners to open (%s)\n", urls, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, "daemon_init: no listeners to open (%s)\n", + urls, 0, 0 ); #endif -#ifdef SO_KEEPALIVE - tmp = 1; - if ( setsockopt( tcps, SOL_SOCKET, SO_KEEPALIVE, - (char *) &tmp, sizeof(tmp) ) == -1 ) - { - int err = errno; - Debug( LDAP_DEBUG_ANY, - "slapd(%d): setsockopt(KEEPALIVE) failed errno %d (%s)\n", - tcps, err, - err > -1 && err < sys_nerr - ? sys_errlist[err] : "unknown" ); + ldap_charray_free( u ); + return -1; + } + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slap_daemon_init: %d listeners to open...\n", i, 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners to open...\n", + i, 0, 0 ); +#endif + slap_listeners = ch_malloc( (i+1)*sizeof(Listener *) ); + + for(n = 0, j = 0; u[n]; n++ ) { + if ( slap_open_listener( u[n], &i, &j ) ) { + ldap_charray_free( u ); + return -1; } + } + slap_listeners[j] = NULL; + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slap_daemon_init: %d listeners opened\n", i, 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners opened\n", + i, 0, 0 ); +#endif + +#ifdef HAVE_SLP + slapd_slp_init( urls ); + slapd_slp_reg(); +#endif + + ldap_charray_free( u ); + ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex ); + return !i; +} + + +int +slapd_daemon_destroy(void) +{ + connections_destroy(); + tcp_close( wake_sds[1] ); + tcp_close( wake_sds[0] ); + sockdestroy(); + +#ifdef HAVE_SLP + slapd_slp_dereg(); + slapd_slp_deinit(); #endif + return 0; +} + - if ( bind( tcps, (struct sockaddr *) addr, sizeof(*addr) ) == -1 ) { - int err = errno; - Debug( LDAP_DEBUG_ANY, "daemon: bind(%d) failed errno %d (%s)\n", - tcps, err, - err > -1 && err < sys_nerr - ? sys_errlist[err] : "unknown" ); - exit( 1 ); +static void +close_listeners( + int remove +) +{ + int l; + + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( slap_listeners[l]->sl_sd != AC_SOCKET_INVALID ) { + if ( remove ) + slapd_remove( slap_listeners[l]->sl_sd, 0 ); +#ifdef LDAP_PF_LOCAL + if ( slap_listeners[l]->sl_sa.sa_addr.sa_family == AF_LOCAL ) { + unlink( slap_listeners[l]->sl_sa.sa_un_addr.sun_path ); + } +#endif /* LDAP_PF_LOCAL */ + slapd_close( slap_listeners[l]->sl_sd ); } + if ( slap_listeners[l]->sl_url.bv_val ) + ber_memfree( slap_listeners[l]->sl_url.bv_val ); + if ( slap_listeners[l]->sl_name.bv_val ) + ber_memfree( slap_listeners[l]->sl_name.bv_val ); + free ( slap_listeners[l] ); + slap_listeners[l] = NULL; } - - return tcps; } + static void * slapd_daemon_task( void *ptr ) { - int inetd = ((int *)ptr) [0]; - int tcps = ((int *)ptr) [1]; - free( ptr ); - - if ( !daemon_initialized ) sockinit(); - - slapd_listener=1; + int l; + time_t last_idle_check = 0; + time( &starttime ); - ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex ); - FD_ZERO( &slap_daemon.sd_readers ); - FD_ZERO( &slap_daemon.sd_writers ); - - if( !inetd ) { - if ( listen( tcps, 5 ) == -1 ) { - int err = errno; - Debug( LDAP_DEBUG_ANY, - "daemon: listen(%d, 5) failed errno %d (%s)\n", - tcps, err, - err > -1 && err < sys_nerr - ? sys_errlist[err] : "unknown" ); - exit( 1 ); + if ( global_idletimeout > 0 ) { + last_idle_check = slap_get_time(); + } + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID ) + continue; +#ifdef LDAP_CONNECTIONLESS + /* Since this is connectionless, the data port is the + * listening port. The listen() and accept() calls + * are unnecessary. + */ + if ( slap_listeners[l]->sl_is_udp ) { + slapd_add( slap_listeners[l]->sl_sd ); + continue; } +#endif - slapd_add( tcps ); + if ( listen( slap_listeners[l]->sl_sd, SLAPD_LISTEN ) == -1 ) { + int err = sock_errno(); + +#ifdef LDAP_PF_INET6 + /* If error is EADDRINUSE, we are trying to listen to INADDR_ANY and + * we are already listening to in6addr_any, then we want to ignore + * this and continue. + */ + if ( err == EADDRINUSE ) { + int i; + struct sockaddr_in sa = slap_listeners[l]->sl_sa.sa_in_addr; + struct sockaddr_in6 sa6; + + if ( sa.sin_family == AF_INET && + sa.sin_addr.s_addr == htonl(INADDR_ANY) ) { + for ( i = 0 ; i < l; i++ ) { + sa6 = slap_listeners[i]->sl_sa.sa_in6_addr; + if ( sa6.sin6_family == AF_INET6 && + !memcmp( &sa6.sin6_addr, &in6addr_any, sizeof(struct in6_addr) ) ) + break; + } - } else { - if( connection_init( 0, NULL, NULL ) ) { + if ( i < l ) { + /* We are already listening to in6addr_any */ +#ifdef NEW_LOGGING + LDAP_LOG(CONNECTION, WARNING, + "slapd_daemon_task: Attempt to listen to 0.0.0.0 failed, already listening on ::, assuming IPv4 included\n", 0, 0, 0 ); +#else + Debug( LDAP_DEBUG_CONNS, + "daemon: Attempt to listen to 0.0.0.0 failed, already listening on ::, assuming IPv4 included\n", + 0, 0, 0 ); +#endif + slapd_close( slap_listeners[l]->sl_sd ); + slap_listeners[l]->sl_sd = AC_SOCKET_INVALID; + continue; + } + } + } +#endif +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon_task: listen( %s, 5 ) failed errno=%d (%s)\n", + slap_listeners[l]->sl_url.bv_val, err, sock_errstr(err) ); +#else Debug( LDAP_DEBUG_ANY, - "connection_init(%d) failed.\n", - 0, 0, 0 ); - - exit( 1 ); + "daemon: listen(%s, 5) failed errno=%d (%s)\n", + slap_listeners[l]->sl_url.bv_val, err, + sock_errstr(err) ); +#endif + return( (void*)-1 ); } - slapd_add( 0 ); + slapd_add( slap_listeners[l]->sl_sd ); + } + +#ifdef HAVE_NT_SERVICE_MANAGER + if ( started_event != NULL ) { + ldap_pvt_thread_cond_signal( &started_event ); } +#endif + /* initialization complete. Here comes the loop. */ while ( !slapd_shutdown ) { - unsigned int i; - int ns, nfds; + ber_socket_t i; + int ns; + int at; + ber_socket_t nfds; +#define SLAPD_EBADF_LIMIT 16 + int ebadf = 0; + int emfile = 0; + +#define SLAPD_IDLE_CHECK_LIMIT 4 + time_t now; + fd_set readfds; fd_set writefds; + Sockaddr from; - struct sockaddr_in from; -#if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD) - struct hostent *hp; -#endif - struct timeval zero; + struct timeval zero; struct timeval *tvp; - char *client_name; - char *client_addr; + if( emfile ) { + now = slap_get_time(); + connections_timeout_idle( now ); + } + else if ( global_idletimeout > 0 ) { + now = slap_get_time(); + if ( difftime( last_idle_check+global_idletimeout/SLAPD_IDLE_CHECK_LIMIT, now ) < 0 ) { + connections_timeout_idle( now ); + } + } + +#ifdef SIGHUP + if( slapd_gentle_shutdown ) { + ber_socket_t active; + + if( slapd_gentle_shutdown == 1 ) { + Debug( LDAP_DEBUG_ANY, "slapd gentle shutdown\n", 0, 0, 0 ); + close_listeners( 1 ); + global_restrictops |= SLAP_RESTRICT_OP_WRITES; + slapd_gentle_shutdown = 2; + } + + ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); + active = slap_daemon.sd_nactives; + ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); + if( active == 0 ) { + slapd_shutdown = 2; + break; + } + } +#endif FD_ZERO( &writefds ); FD_ZERO( &readfds ); @@ -319,19 +1254,26 @@ slapd_daemon_task( #ifdef FD_SET_MANUAL_COPY for( s = 0; s < nfds; s++ ) { - if(FD_ISSET( &slap_sd_writers, s )) { - FD_SET( &writefds, s ); + if(FD_ISSET( &slap_sd_readers, s )) { + FD_SET( s, &readfds ); } if(FD_ISSET( &slap_sd_writers, s )) { - FD_SET( &writefds, s ); + FD_SET( s, &writefds ); } } #else - memcpy( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) ); - memcpy( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) ); + AC_MEMCPY( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) ); + AC_MEMCPY( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) ); #endif + assert(!FD_ISSET(wake_sds[0], &readfds)); + FD_SET( wake_sds[0], &readfds ); - FD_SET( (unsigned) tcps, &readfds ); + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID ) + continue; + if (!FD_ISSET(slap_listeners[l]->sl_sd, &readfds)) + FD_SET( slap_listeners[l]->sl_sd, &readfds ); + } #ifndef HAVE_WINSOCK nfds = slap_daemon.sd_nfds; @@ -341,148 +1283,410 @@ slapd_daemon_task( ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); - ldap_pvt_thread_mutex_lock( &active_threads_mutex ); + at = ldap_pvt_thread_pool_backload(&connection_pool); + #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS ) tvp = NULL; #else - tvp = active_threads ? &zero : NULL; + tvp = at ? &zero : NULL; #endif - Debug( LDAP_DEBUG_CONNS, - "daemon: select: tcps=%d active_threads=%d tvp=%s\n", - tcps, active_threads, - tvp == NULL ? "NULL" : "zero" ); - + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID ) + continue; - ldap_pvt_thread_mutex_unlock( &active_threads_mutex ); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slapd_daemon_task: select: listen=%d " + "active_threads=%d tvp=%s\n", + slap_listeners[l]->sl_sd, at, tvp == NULL ? "NULL" : "zero" ); +#else + Debug( LDAP_DEBUG_CONNS, + "daemon: select: listen=%d active_threads=%d tvp=%s\n", + slap_listeners[l]->sl_sd, at, + tvp == NULL ? "NULL" : "zero" ); +#endif + } - switch(ns = select( nfds, &readfds, &writefds, 0, tvp )) { + switch(ns = select( nfds, &readfds, +#ifdef HAVE_WINSOCK + /* don't pass empty fd_set */ + ( writefds.fd_count > 0 ? &writefds : NULL ), +#else + &writefds, +#endif + NULL, tvp )) + { case -1: { /* failure - try again */ - int err = errno; + int err = sock_errno(); + + if( err == EBADF +#ifdef WSAENOTSOCK + /* you'd think this would be EBADF */ + || err == WSAENOTSOCK +#endif + ) { + if (++ebadf < SLAPD_EBADF_LIMIT) + continue; + } + if( err != EINTR ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slapd_daemon_task: select failed (%d): %s\n", + err, sock_errstr(err), 0 ); +#else Debug( LDAP_DEBUG_CONNS, "daemon: select failed (%d): %s\n", - err, - err >= 0 && err < sys_nerr - ? sys_errlist[err] : "unknown", - 0 ); - - slapd_shutdown = -1; + err, sock_errstr(err), 0 ); +#endif + slapd_shutdown = 2; } } continue; case 0: /* timeout - let threads run */ + ebadf = 0; +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + "slapd_daemon_task: select timeout - yielding\n", 0, 0, 0 ); +#else Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n", 0, 0, 0 ); - ldap_pvt_thread_yield(); +#endif + ldap_pvt_thread_yield(); continue; default: /* something happened - deal with it */ + if( slapd_shutdown ) continue; + + ebadf = 0; +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + "slapd_daemon_task: activity on %d descriptors\n", ns, 0, 0 ); +#else Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n", ns, 0, 0 ); +#endif /* FALL THRU */ } - if ( FD_ISSET( tcps, &readfds ) ) { - int s; - int len = sizeof(from); + if( FD_ISSET( wake_sds[0], &readfds ) ) { + char c[BUFSIZ]; + tcp_read( wake_sds[0], c, sizeof(c) ); +#if defined(NO_THREADS) || defined(HAVE_GNU_PTH) + waking = 0; +#endif + continue; + } + + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + ber_socket_t s; + socklen_t len = sizeof(from); long id; + slap_ssf_t ssf = 0; + char *authid = NULL; +#ifdef SLAPD_RLOOKUPS + char hbuf[NI_MAXHOST]; +#endif - if ( (s = accept( tcps, - (struct sockaddr *) &from, &len )) == -1 ) - { - int err = errno; - Debug( LDAP_DEBUG_ANY, - "daemon: accept(%d) failed errno %d (%s)\n", err, - tcps, err >= 0 && err < sys_nerr ? - sys_errlist[err] : "unknown"); + char *dnsname = NULL; + char *peeraddr = NULL; +#ifdef LDAP_PF_LOCAL + char peername[MAXPATHLEN + sizeof("PATH=")]; +#elif defined(LDAP_PF_INET6) + char peername[sizeof("IP=ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff 65535")]; +#else + char peername[sizeof("IP=255.255.255.255:65336")]; +#endif /* LDAP_PF_LOCAL */ + + peername[0] = '\0'; + + if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID ) + continue; + + if ( !FD_ISSET( slap_listeners[l]->sl_sd, &readfds ) ) + continue; + +#ifdef LDAP_CONNECTIONLESS + if ( slap_listeners[l]->sl_is_udp ) { + /* The first time we receive a query, we set this + * up as a "connection". It remains open for the life + * of the slapd. + */ + if ( slap_listeners[l]->sl_is_udp < 2 ) { + id = connection_init( + slap_listeners[l]->sl_sd, + slap_listeners[l], "", "", + 2, ssf, authid ); + slap_listeners[l]->sl_is_udp++; + } continue; } +#endif + + s = accept( slap_listeners[l]->sl_sd, + (struct sockaddr *) &from, &len ); + if ( s == AC_SOCKET_INVALID ) { + int err = sock_errno(); + +#ifdef EMFILE + if( err == EMFILE ) { + emfile++; + } else +#endif +#ifdef ENFILE + if( err == ENFILE ) { + emfile++; + } else +#endif + { + emfile=0; + } - assert( !FD_ISSET( 0, &slap_daemon.sd_actives) ); - assert( !FD_ISSET( 0, &slap_daemon.sd_readers) ); - assert( !FD_ISSET( 0, &slap_daemon.sd_writers) ); + if( emfile < 3 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon_task: accept(%ld) failed errno=%d (%s)\n", + (long)slap_listeners[l]->sl_sd, + err, sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, + "daemon: accept(%ld) failed errno=%d (%s)\n", + (long) slap_listeners[l]->sl_sd, err, + sock_errstr(err) ); +#endif + } else { + /* prevent busy loop */ +# ifdef HAVE_USLEEP + if( emfile % 4 == 3 ) usleep( 250 ); +# else + if( emfile % 8 == 7 ) sleep( 1 ); +# endif + } + + ldap_pvt_thread_yield(); + continue; + } + emfile = 0; #ifndef HAVE_WINSOCK /* make sure descriptor number isn't too great */ if ( s >= dtblsize ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon_task: %ld beyond descriptor table size %ld\n", + (long)s, (long)dtblsize, 0 ); +#else Debug( LDAP_DEBUG_ANY, - "daemon: %d beyond descriptor table size %d\n", - s, dtblsize, 0 ); - tcp_close(s); + "daemon: %ld beyond descriptor table size %ld\n", + (long) s, (long) dtblsize, 0 ); +#endif + + slapd_close(s); + ldap_pvt_thread_yield(); continue; } #endif - - Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %d\n", - s, 0, 0 ); - len = sizeof(from); - if ( getpeername( s, (struct sockaddr *) &from, &len ) == 0 ) { - client_addr = inet_ntoa( from.sin_addr ); - -#if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD) - hp = gethostbyaddr( (char *) - &(from.sin_addr.s_addr), - sizeof(from.sin_addr.s_addr), AF_INET ); +#ifdef LDAP_DEBUG + ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex ); - if(hp) { - char *p; - client_name = hp->h_name; + /* newly accepted stream should not be in any of the FD SETS */ + assert( !FD_ISSET( s, &slap_daemon.sd_actives) ); + assert( !FD_ISSET( s, &slap_daemon.sd_readers) ); + assert( !FD_ISSET( s, &slap_daemon.sd_writers) ); - /* normalize the domain */ - for ( p = client_name; *p; p++ ) { - *p = TOLOWER( (unsigned char) *p ); - } + ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex ); +#endif - } else { - client_name = NULL; +#if defined( SO_KEEPALIVE ) || defined( TCP_NODELAY ) +#ifdef LDAP_PF_LOCAL + /* for IPv4 and IPv6 sockets only */ + if ( from.sa_addr.sa_family != AF_LOCAL ) +#endif /* LDAP_PF_LOCAL */ + { + int rc; + int tmp; +#ifdef SO_KEEPALIVE + /* enable keep alives */ + tmp = 1; + rc = setsockopt( s, SOL_SOCKET, SO_KEEPALIVE, + (char *) &tmp, sizeof(tmp) ); + if ( rc == AC_SOCKET_ERROR ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon_task: setsockopt( %ld, SO_KEEPALIVE)" + " failed errno=%d (%s)\n", + (long)s, err, sock_errstr(err) ); +#else + Debug( LDAP_DEBUG_ANY, + "slapd(%ld): setsockopt(SO_KEEPALIVE) failed " + "errno=%d (%s)\n", (long) s, err, sock_errstr(err) ); +#endif } +#endif +#ifdef TCP_NODELAY + /* enable no delay */ + tmp = 1; + rc = setsockopt( s, IPPROTO_TCP, TCP_NODELAY, + (char *)&tmp, sizeof(tmp) ); + if ( rc == AC_SOCKET_ERROR ) { + int err = sock_errno(); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon_task: setsockopt( %ld, " + "TCP_NODELAY) failed errno=%d (%s)\n", + (long)s, err, sock_errstr(err) ); #else - client_name = NULL; + Debug( LDAP_DEBUG_ANY, + "slapd(%ld): setsockopt(TCP_NODELAY) failed " + "errno=%d (%s)\n", (long) s, err, sock_errstr(err) ); +#endif + } +#endif + } #endif +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL1, + "slapd_daemon_task: new connection on %ld\n", (long)s, 0, 0 ); +#else + Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %ld\n", + (long) s, 0, 0 ); +#endif + switch ( from.sa_addr.sa_family ) { +# ifdef LDAP_PF_LOCAL + case AF_LOCAL: + sprintf( peername, "PATH=%s", from.sa_un_addr.sun_path ); + ssf = LDAP_PVT_SASL_LOCAL_SSF; + { + uid_t uid; + gid_t gid; + + if( getpeereid( s, &uid, &gid ) == 0 ) { + authid = ch_malloc( + sizeof("uidnumber=4294967295+gidnumber=4294967295," + "cn=peercred,cn=external,cn=auth")); + sprintf(authid, "uidnumber=%d+gidnumber=%d," + "cn=peercred,cn=external,cn=auth", + (int) uid, (int) gid); + } + } + dnsname = "local"; + break; +#endif /* LDAP_PF_LOCAL */ + +# ifdef LDAP_PF_INET6 + case AF_INET6: + if ( IN6_IS_ADDR_V4MAPPED(&from.sa_in6_addr.sin6_addr) ) { + peeraddr = inet_ntoa( *((struct in_addr *) + &from.sa_in6_addr.sin6_addr.s6_addr[12]) ); + sprintf( peername, "IP=%s:%d", + peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN, + (unsigned) ntohs( from.sa_in6_addr.sin6_port ) ); } else { - client_name = NULL;; - client_addr = NULL; + char addr[INET6_ADDRSTRLEN]; + + peeraddr = (char *) inet_ntop( AF_INET6, + &from.sa_in6_addr.sin6_addr, + addr, sizeof addr ); + sprintf( peername, "IP=%s %d", + peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN, + (unsigned) ntohs( from.sa_in6_addr.sin6_port ) ); } - -#ifdef HAVE_TCPD - if(!hosts_ctl("slapd", - client_name != NULL ? client_name : STRING_UNKNOWN, - client_addr != NULL ? client_addr : STRING_UNKNOWN, - STRING_UNKNOWN)) - { - /* DENY ACCESS */ - Statslog( LDAP_DEBUG_ANY, - "fd=%d connection from %s (%s) denied.\n", - s, - client_name == NULL ? "unknown" : client_name, - client_addr == NULL ? "unknown" : client_addr, - 0, 0 ); - - tcp_close(s); + break; +# endif /* LDAP_PF_INET6 */ + + case AF_INET: + peeraddr = inet_ntoa( from.sa_in_addr.sin_addr ); + sprintf( peername, "IP=%s:%d", + peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN, + (unsigned) ntohs( from.sa_in_addr.sin_port ) ); + break; + + default: + slapd_close(s); continue; } -#endif /* HAVE_TCPD */ + if ( ( from.sa_addr.sa_family == AF_INET ) +#ifdef LDAP_PF_INET6 + || ( from.sa_addr.sa_family == AF_INET6 ) +#endif + ) { +#ifdef SLAPD_RLOOKUPS + if ( use_reverse_lookup ) { + char *herr; + if (ldap_pvt_get_hname( (const struct sockaddr *)&from, len, hbuf, + sizeof(hbuf), &herr ) == 0) { + ldap_pvt_str2lower( hbuf ); + dnsname = hbuf; + } + } +#else + dnsname = NULL; +#endif /* SLAPD_RLOOKUPS */ + +#ifdef HAVE_TCPD + if ( !hosts_ctl("slapd", + dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN, + peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN, + SLAP_STRING_UNKNOWN )) + { + /* DENY ACCESS */ + Statslog( LDAP_DEBUG_STATS, + "fd=%ld DENIED from %s (%s)", + (long) s, + dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN, + peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN, + 0, 0 ); + slapd_close(s); + continue; + } +#endif /* HAVE_TCPD */ + } - if( (id = connection_init(s, client_name, client_addr)) < 0 ) { + id = connection_init(s, + slap_listeners[l], + dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN, + peername, +#ifdef HAVE_TLS + slap_listeners[l]->sl_is_tls, +#else + 0, +#endif + ssf, + authid ); + + if( authid ) ch_free(authid); + + if( id < 0 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, INFO, + "slapd_daemon_task: " + "connection_init(%ld, %s, %s) " + "failed.\n", + (long)s, peername, + slap_listeners[l]->sl_name.bv_val ); +#else Debug( LDAP_DEBUG_ANY, - "daemon: connection_init(%d, %s, %s) failed.\n", - s, - client_name == NULL ? "unknown" : client_name, - client_addr == NULL ? "unknown" : client_addr); - tcp_close(s); + "daemon: connection_init(%ld, %s, %s) " + "failed.\n", + (long) s, + peername, + slap_listeners[l]->sl_name.bv_val ); +#endif + slapd_close(s); continue; } Statslog( LDAP_DEBUG_STATS, - "daemon: conn=%d fd=%d connection from %s (%s) accepted.\n", - id, s, - client_name == NULL ? "unknown" : client_name, - client_addr == NULL ? "unknown" : client_addr, + "conn=%ld fd=%ld ACCEPT from %s (%s)\n", + id, (long) s, + peername, + slap_listeners[l]->sl_name.bv_val, 0 ); slapd_add( s ); @@ -490,271 +1694,388 @@ slapd_daemon_task( } #ifdef LDAP_DEBUG +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + "slapd_daemon_task: activity on ", 0, 0, 0 ); +#else Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 ); +#endif #ifdef HAVE_WINSOCK - for ( i = 0; i < readfds.fd_count; i++ ) - { - Debug( LDAP_DEBUG_CONNS, " %d%s", readfds.fd_array[i], "r" ); + for ( i = 0; i < readfds.fd_count; i++ ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + " %d%s", readfds.fd_array[i], "r", 0, 0 ); +#else + Debug( LDAP_DEBUG_CONNS, " %d%s", + readfds.fd_array[i], "r", 0 ); +#endif } - for ( i = 0; i < writefds.fd_count; i++ ) - { - Debug( LDAP_DEBUG_CONNS, " %d%s", writefds.fd_array[i], "w" ); + for ( i = 0; i < writefds.fd_count; i++ ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + " %d%s", writefds.fd_array[i], "w" , 0 ); +#else + Debug( LDAP_DEBUG_CONNS, " %d%s", + writefds.fd_array[i], "w", 0 ); +#endif } + #else for ( i = 0; i < nfds; i++ ) { - int a, r, w; - + int r, w; + int is_listener = 0; + + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( i == slap_listeners[l]->sl_sd ) { +#ifdef LDAP_CONNECTIONLESS + /* The listener is the data port. Don't + * skip it. + */ + if (slap_listeners[l]->sl_is_udp) continue; +#endif + is_listener = 1; + break; + } + } + if ( is_listener ) { + continue; + } r = FD_ISSET( i, &readfds ); w = FD_ISSET( i, &writefds ); - if ( i != tcps && (r || w) ) { + if ( r || w ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + " %d%s%s", i, r ? "r" : "", w ? "w" : "" ); +#else Debug( LDAP_DEBUG_CONNS, " %d%s%s", i, r ? "r" : "", w ? "w" : "" ); +#endif } } #endif +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, "\n", 0, 0, 0 ); +#else Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 ); +#endif + #endif /* loop through the writers */ #ifdef HAVE_WINSOCK - for ( i = 0; i < writefds.fd_count; i++ ) { - int wd = writefds.fd_array[i]; - - if ( wd == tcps ) { + for ( i = 0; i < writefds.fd_count; i++ ) +#else + for ( i = 0; i < nfds; i++ ) +#endif + { + ber_socket_t wd; + int is_listener = 0; +#ifdef HAVE_WINSOCK + wd = writefds.fd_array[i]; +#else + if( ! FD_ISSET( i, &writefds ) ) { continue; } + wd = i; +#endif + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( i == slap_listeners[l]->sl_sd ) { +#ifdef LDAP_CONNECTIONLESS + if (slap_listeners[l]->sl_is_udp) continue; +#endif + is_listener = 1; + break; + } + } + if ( is_listener ) { + continue; + } +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + "slapd_daemon_task: write active on %d\n", wd, 0, 0 ); +#else Debug( LDAP_DEBUG_CONNS, - "daemon: signalling write waiter on %d\n", + "daemon: write active on %d\n", wd, 0, 0 ); +#endif + /* + * NOTE: it is possible that the connection was closed + * and that the stream is now inactive. + * connection_write() must valid the stream is still + * active. + */ - assert( FD_ISSET( wd, &slap_daemon.sd_actives) ); - - slapd_clr_write( wd, 0 ); if ( connection_write( wd ) < 0 ) { FD_CLR( (unsigned) wd, &readfds ); slapd_close( wd ); } } + +#ifdef HAVE_WINSOCK + for ( i = 0; i < readfds.fd_count; i++ ) #else - for ( i = 0; i < nfds; i++ ) { - if ( i == tcps ) { + for ( i = 0; i < nfds; i++ ) +#endif + { + ber_socket_t rd; + int is_listener = 0; + +#ifdef HAVE_WINSOCK + rd = readfds.fd_array[i]; +#else + if( ! FD_ISSET( i, &readfds ) ) { continue; } - if ( FD_ISSET( i, &writefds ) ) { - Debug( LDAP_DEBUG_CONNS, - "daemon: signaling write waiter on %d\n", i, 0, 0 ); - - assert( FD_ISSET( i, &slap_daemon.sd_actives) ); + rd = i; +#endif - /* clear the write flag */ - slapd_clr_write( i, 0 ); - - if( connection_write( i ) < 0 ) { - FD_CLR( i, &readfds ); - slapd_close( i ); + for ( l = 0; slap_listeners[l] != NULL; l++ ) { + if ( rd == slap_listeners[l]->sl_sd ) { +#ifdef LDAP_CONNECTIONLESS + if (slap_listeners[l]->sl_is_udp) continue; +#endif + is_listener = 1; + break; } } - } -#endif - -#ifdef HAVE_WINSOCK - for ( i = 0; i < readfds.fd_count; i++ ) { - int rd = readfds.fd_array[i]; - if ( rd == tcps ) { + if ( is_listener ) { continue; } + +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, DETAIL2, + "slapd_daemon_task: read activity on %d\n", rd, 0, 0 ); +#else Debug ( LDAP_DEBUG_CONNS, "daemon: read activity on %d\n", rd, 0, 0 ); - assert( FD_ISSET( rd, &slap_daemon.sd_actives) ); +#endif + /* + * NOTE: it is possible that the connection was closed + * and that the stream is now inactive. + * connection_read() must valid the stream is still + * active. + */ if ( connection_read( rd ) < 0 ) { slapd_close( rd ); } } -#else - for ( i = 0; i < nfds; i++ ) { - if ( i == tcps ) { - continue; - } - - if ( FD_ISSET( i, &readfds ) ) { - Debug( LDAP_DEBUG_CONNS, - "daemon: read activity on %d\n", i, 0, 0 ); - - assert( FD_ISSET( i, &slap_daemon.sd_actives) ); - - if( connection_read( i ) < 0) { - slapd_close( i ); - } - } - } -#endif ldap_pvt_thread_yield(); } - if( slapd_shutdown > 0 ) { + if( slapd_shutdown == 1 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slapd_daemon_task: shutdown requested and initiated.\n", 0, 0, 0 ); +#else Debug( LDAP_DEBUG_TRACE, - "daemon: shutdown requested (%d) and initiated.\n", - (int) slapd_shutdown, 0, 0 ); + "daemon: shutdown requested and initiated.\n", + 0, 0, 0 ); +#endif - } else if ( slapd_shutdown < 0 ) { - Debug( LDAP_DEBUG_TRACE, - "daemon: abnormal condition (%d), shutdown initiated.\n", - (int) slapd_shutdown, 0, 0 ); + } else if ( slapd_shutdown == 2 ) { +#ifdef HAVE_NT_SERVICE_MANAGER +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slapd_daemon_task: shutdown initiated by Service Manager.\n", + 0, 0, 0); +#else + Debug( LDAP_DEBUG_TRACE, + "daemon: shutdown initiated by Service Manager.\n", + 0, 0, 0); +#endif +#else /* !HAVE_NT_SERVICE_MANAGER */ +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slapd_daemon_task: abnormal condition, " + "shutdown initiated.\n", 0, 0, 0 ); +#else + Debug( LDAP_DEBUG_TRACE, + "daemon: abnormal condition, shutdown initiated.\n", + 0, 0, 0 ); +#endif +#endif /* !HAVE_NT_SERVICE_MANAGER */ } else { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slapd_daemon_task: no active streams, shutdown initiated.\n", + 0, 0, 0 ); +#else Debug( LDAP_DEBUG_TRACE, - "daemon: no active streams, shutdown initiated.\n", - 0, 0, 0 ); + "daemon: no active streams, shutdown initiated.\n", + 0, 0, 0 ); +#endif } - if( tcps >= 0 ) { - tcp_close( tcps ); + if( slapd_gentle_shutdown != 2 ) { + close_listeners ( 0 ); } - /* we only implement "quick" shutdown */ - connections_shutdown(); + free ( slap_listeners ); + slap_listeners = NULL; + + if( !slapd_gentle_shutdown ) { + connections_shutdown(); + } - ldap_pvt_thread_mutex_lock( &active_threads_mutex ); +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slapd_daemon_task: shutdown waiting for %d threads to terminate.\n", + ldap_pvt_thread_pool_backload(&connection_pool), 0, 0 ); +#else Debug( LDAP_DEBUG_ANY, "slapd shutdown: waiting for %d threads to terminate\n", - active_threads, 0, 0 ); - while ( active_threads > 0 ) { - ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex); - } - ldap_pvt_thread_mutex_unlock( &active_threads_mutex ); + ldap_pvt_thread_pool_backload(&connection_pool), 0, 0 ); +#endif + ldap_pvt_thread_pool_destroy(&connection_pool, 1); - slapd_listener = 0; return NULL; } -int slapd_daemon( int inetd, int tcps ) + +int slapd_daemon( void ) { int rc; - int *args = ch_malloc( sizeof( int[2] ) ); - args[0] = inetd; - args[1] = tcps; - - if ( !daemon_initialized ) sockinit(); connections_init(); #define SLAPD_LISTENER_THREAD 1 -#if defined( SLAPD_LISTENER_THREAD ) || !defined(HAVE_PTHREADS) +#if defined( SLAPD_LISTENER_THREAD ) + { + ldap_pvt_thread_t listener_tid; - /* listener as a separate THREAD */ - rc = ldap_pvt_thread_create( &listener_tid, - 0, slapd_daemon_task, args ); + /* listener as a separate THREAD */ + rc = ldap_pvt_thread_create( &listener_tid, + 0, slapd_daemon_task, NULL ); - if ( rc != 0 ) { - Debug( LDAP_DEBUG_ANY, - "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 ); - goto destory; + if ( rc != 0 ) { +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, ERR, + "slapd_daemon: listener ldap_pvt_thread_create failed (%d).\n", + rc, 0, 0 ); +#else + Debug( LDAP_DEBUG_ANY, + "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 ); +#endif + return rc; + } + + /* wait for the listener thread to complete */ + ldap_pvt_thread_join( listener_tid, (void *) NULL ); } - - /* wait for the listener thread to complete */ - ldap_pvt_thread_join( listener_tid, (void *) NULL ); #else - /* expermimental code */ - listener_tid = pthread_self(); - slapd_daemon_task( args ); + /* experimental code */ + slapd_daemon_task( NULL ); #endif - rc = 0; + return 0; -destory: - connections_destroy(); - -#ifdef HAVE_WINSOCK - WSACleanup( ); -#endif - - return rc; } -#ifdef HAVE_WINSOCK2 -void sockinit() +int sockinit(void) { +#if defined( HAVE_WINSOCK2 ) WORD wVersionRequested; WSADATA wsaData; int err; - + wVersionRequested = MAKEWORD( 2, 0 ); - + err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { /* Tell the user that we couldn't find a usable */ - /* WinSock DLL. */ - return; + /* WinSock DLL. */ + return -1; } - + /* Confirm that the WinSock DLL supports 2.0.*/ /* Note that if the DLL supports versions greater */ /* than 2.0 in addition to 2.0, it will still return */ - /* 2.0 in wVersion since that is the version we */ - /* requested. */ - + /* 2.0 in wVersion since that is the version we */ + /* requested. */ + if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0 ) { /* Tell the user that we couldn't find a usable */ - /* WinSock DLL. */ - WSACleanup( ); - return; + /* WinSock DLL. */ + WSACleanup(); + return -1; } - daemon_initialized = 1; -} /* The WinSock DLL is acceptable. Proceed. */ - -void hit_socket( void ) -{ - int s, on = 1; - extern struct sockaddr_in bind_addr; - /* throw something at the socket to terminate the select() in the daemon thread. */ - if (( s = socket( AF_INET, SOCK_STREAM, 0 )) == INVALID_SOCKET ) - Debug( LDAP_DEBUG_TRACE, "slap_set_shutdown: socket failed\n\tWSAGetLastError=%d (%s)\n", WSAGetLastError(), WSAGetLastErrorString(), 0 ); - if ( ioctlsocket( s, FIONBIO, &on ) == -1 ) - Debug( LDAP_DEBUG_TRACE, "slap_set_shutdown:FIONBIO ioctl on %d faled\n\tWSAGetLastError=%d (%s)\n", s, WSAGetLastError(), WSAGetLastError() ); - - bind_addr.sin_addr.s_addr = htonl( INADDR_LOOPBACK ); - if ( connect( s, (struct sockaddr *)&bind_addr, sizeof( struct sockaddr_in )) == SOCKET_ERROR ) { - /* we can probably expect some error to occur here, mostly WSAEWOULDBLOCK */ - } -} -#elif HAVE_WINSOCK -void sockinit() -{ WSADATA wsaData; + /* The WinSock DLL is acceptable. Proceed. */ +#elif defined( HAVE_WINSOCK ) + WSADATA wsaData; if ( WSAStartup( 0x0101, &wsaData ) != 0 ) { - return( NULL ); + return -1; } - daemon_initialized = 1; +#endif + return 0; } -#else -void sockinit() + +int sockdestroy(void) { - daemon_initialized = 1; - return; -} +#if defined( HAVE_WINSOCK2 ) || defined( HAVE_WINSOCK ) + WSACleanup(); #endif + return 0; +} -void -slap_set_shutdown( int sig ) +RETSIGTYPE +slap_sig_shutdown( int sig ) { - slapd_shutdown = sig; -#ifndef HAVE_WINSOCK - if(slapd_listener) { - ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 ); - } +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slap_sig_shutdown: signal %d\n", sig, 0, 0 ); +#else + Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: signal %d\n", sig, 0, 0); +#endif + + /* + * If the NT Service Manager is controlling the server, we don't + * want SIGBREAK to kill the server. For some strange reason, + * SIGBREAK is generated when a user logs out. + */ + +#if HAVE_NT_SERVICE_MANAGER && SIGBREAK + if (is_NT_Service && sig == SIGBREAK) +#ifdef NEW_LOGGING + LDAP_LOG( CONNECTION, CRIT, + "slap_sig_shutdown: SIGBREAK ignored.\n", 0, 0, 0 ); #else - hit_socket(); + Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: SIGBREAK ignored.\n", + 0, 0, 0); +#endif + else +#endif +#ifdef SIGHUP + if (sig == SIGHUP && global_gentlehup && slapd_gentle_shutdown == 0) + slapd_gentle_shutdown = 1; + else #endif + slapd_shutdown = 1; + + WAKE_LISTENER(1); + /* reinstall self */ - (void) SIGNAL( sig, slap_set_shutdown ); + (void) SIGNAL_REINSTALL( sig, slap_sig_shutdown ); } -void -slap_do_nothing( int sig ) +RETSIGTYPE +slap_sig_wake( int sig ) { + WAKE_LISTENER(1); + /* reinstall self */ - (void) SIGNAL( sig, slap_do_nothing ); + (void) SIGNAL_REINSTALL( sig, slap_sig_wake ); +} + + +void slapd_add_internal(ber_socket_t s) { + slapd_add(s); +} + +Listener ** slapd_get_listeners(void) { + return slap_listeners; }