Skip to content
Snippets Groups Projects
daemon.c 22.9 KiB
Newer Older
  • Learn to ignore specific revisions
  • #include "portable.h"
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #include <stdio.h>
    
    
    #include <ac/ctype.h>
    #include <ac/errno.h>
    #include <ac/signal.h>
    #include <ac/socket.h>
    #include <ac/string.h>
    #include <ac/time.h>
    #include <ac/unistd.h>
    
    
    #include "ldap_defaults.h"
    
    
    #ifdef HAVE_TCPD
    
    #include <tcpd.h>
    
    int allow_severity = LOG_INFO;
    int deny_severity = LOG_NOTICE;
    
    #endif /* TCP Wrappers */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    time_t starttime;
    
    ber_socket_t dtblsize;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    typedef struct slap_listener {
    	char* sl_url;
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    #ifdef HAVE_TLS
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	int		sl_is_tls;
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	ber_socket_t		sl_sd;
    	struct sockaddr_in	sl_addr;
    } Listener;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    Listener **slap_listeners = NULL;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    extern ldap_pvt_thread_cond_t			started_event;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    static void hit_socket(void);
    
    /* In wsa_err.c */
    char *WSAGetLastErrorString();
    
    static ldap_pvt_thread_t hit_tid;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    do {\
    
            ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
        }\
    } while(0)
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    do {\
    
            ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
        }\
    } while(0)
    
    #ifndef HAVE_WINSOCK
    static 
    #endif
    volatile sig_atomic_t slapd_shutdown = 0;
    
    
    static ldap_pvt_thread_t	listener_tid;
    
    static volatile sig_atomic_t slapd_listener = 0;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    static struct slap_daemon {
    
    	ldap_pvt_thread_mutex_t	sd_mutex;
    
    	int sd_nactives;
    
    #ifndef HAVE_WINSOCK
    	/* In winsock, accept() returns values higher than dtblsize
    		so don't bother with this optimization */
    	int sd_nfds;
    #endif
    
    	fd_set sd_actives;
    	fd_set sd_readers;
    	fd_set sd_writers;
    } slap_daemon; 
    
    /*
     * Add a descriptor to daemon control
     */
    
    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 ));
    	assert( !FD_ISSET( s, &slap_daemon.sd_readers ));
    	assert( !FD_ISSET( s, &slap_daemon.sd_writers ));
    
    
    	if (s >= slap_daemon.sd_nfds) {
    		slap_daemon.sd_nfds = s + 1;
    	}
    
    	FD_SET( s, &slap_daemon.sd_actives );
    	FD_SET( s, &slap_daemon.sd_readers );
    
    	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" : "" );
    
    	ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    }
    
    /*
     * Remove the descriptor from daemon control
     */
    
    void slapd_remove(ber_socket_t s, int wake) {
    
    	ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    	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( 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 );
    }
    
    
    void slapd_clr_write(ber_socket_t s, int wake) {
    
    	ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    	assert( FD_ISSET( s, &slap_daemon.sd_actives) );
    	FD_CLR( s, &slap_daemon.sd_writers );
    
    
    	ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    
    
    	if( wake ) {
    		ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
    	}
    
    void slapd_set_write(ber_socket_t s, int wake) {
    
    	ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    	assert( FD_ISSET( s, &slap_daemon.sd_actives) );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	FD_SET( (unsigned) s, &slap_daemon.sd_writers );
    
    
    	ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    
    
    	if( wake ) {
    		ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
    	}
    
    void slapd_clr_read(ber_socket_t s, int wake) {
    
    	ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    
    	assert( FD_ISSET( s, &slap_daemon.sd_actives) );
    
    	FD_CLR( s, &slap_daemon.sd_readers );
    
    
    	ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    
    
    	if( wake ) {
    		ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
    	}
    
    void slapd_set_read(ber_socket_t s, int wake) {
    
    	ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    
    	assert( FD_ISSET( s, &slap_daemon.sd_actives) );
    
    	FD_SET( s, &slap_daemon.sd_readers );
    
    
    	ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    
    
    	if( wake ) {
    		ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
    	}
    
    static void slapd_close(ber_socket_t s) {
    	Debug( LDAP_DEBUG_CONNS, "daemon: closing %ld\n",
    		(long) s, 0, 0 );
    
    	tcp_close(s);
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    Listener *
    open_listener(
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	const char* url,
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	int port,
    	int tls_port )
    {
    	int	tmp, rc;
    	Listener l;
    	Listener *li;
    	LDAPURLDesc *lud;
    
    	rc = ldap_url_parse( url, &lud );
    
    	if( rc != LDAP_URL_SUCCESS ) {
    		Debug( LDAP_DEBUG_ANY,
    			"daemon: listen URL \"%s\" parse error=%d\n",
    			url, rc, 0 );
    		return NULL;
    	}
    
    #ifndef HAVE_TLS
    	if( lud->lud_ldaps ) {
    		Debug( LDAP_DEBUG_ANY,
    			"daemon: TLS not supported (%s)\n",
    			url, 0, 0 );
    		ldap_free_urldesc( lud );
    		return NULL;
    	}
    
    	if(! lud->lud_port ) {
    		lud->lud_port = port;
    	}
    
    #else
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	l.sl_is_tls = lud->lud_ldaps;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if(! lud->lud_port ) {
    		lud->lud_port = lud->lud_ldaps ? tls_port : port;
    	}
    #endif
    
    	(void) memset( (void*) &l.sl_addr, '\0', sizeof(l.sl_addr) );
    
    	l.sl_addr.sin_family = AF_INET;
    	l.sl_addr.sin_port = htons( (unsigned short) lud->lud_port );
    
    	if( lud->lud_host == NULL || lud->lud_host[0] == '\0'
    		|| strcmp(lud->lud_host, "*") == 0 )
    	{
    		l.sl_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    
    	} else {
    		/* host or address was specified */
    
    		if( isdigit( lud->lud_host[0] ) ) {
    #ifdef HAVE_WINSOCK
    			if(!(l.sl_addr.sin_addr.S_un.S_addr = inet_addr(lud->lud_host)))
    #else
    			if(!inet_aton(lud->lud_host, &l.sl_addr.sin_addr))
    #endif  
    			{
    				Debug( LDAP_DEBUG_ANY, "invalid address (%s) in URL: %s",
    					lud->lud_host, url, 0);
    				ldap_free_urldesc( lud );
    				return NULL;
    			}
    
    		} else {
    			struct hostent *he = gethostbyname( lud->lud_host );
    			if( he == NULL ) {
    				Debug( LDAP_DEBUG_ANY, "invalid host (%s) in URL: %s",
    					lud->lud_host, url, 0);
    				ldap_free_urldesc( lud );
    				return NULL;
    			}
    
    #ifdef HAVE_WINSOCK
    
    			memcpy( &l.sl_addr.sin_addr.S_un.S_addr, he->h_addr,
    			       sizeof( l.sl_addr.sin_addr.S_un.S_addr ) );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #else
    
    			memcpy( &l.sl_addr.sin_addr, he->h_addr,
    			       sizeof( l.sl_addr.sin_addr ) );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #endif  
    		}
    	}
    
    	ldap_free_urldesc( lud );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if ( (l.sl_sd = socket( AF_INET, SOCK_STREAM, 0 )) == AC_SOCKET_INVALID ) {
    #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 );
    #else
    		Debug( LDAP_DEBUG_ANY, 
    			"daemon: socket() failed errno %d (%s)\n",
    			WSAGetLastError(),
    	    	WSAGetLastErrorString(), 0 );
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		return NULL;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    #ifndef HAVE_WINSOCK
    	if ( l.sl_sd >= dtblsize ) {
    		Debug( LDAP_DEBUG_ANY,
    			"daemon: listener descriptor %ld is too great %ld\n",
    			(long) l.sl_sd, (long) dtblsize, 0 );
    		tcp_close( l.sl_sd );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		return NULL;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    #endif
    
    #ifdef SO_REUSEADDR
    	tmp = 1;
    	if ( setsockopt( l.sl_sd, SOL_SOCKET, SO_REUSEADDR,
    		(char *) &tmp, sizeof(tmp) ) == -1 )
    	{
    		int err = errno;
    		Debug( LDAP_DEBUG_ANY,
    	       "slapd(%ld): setsockopt() failed errno %d (%s)\n",
    	    	(long) l.sl_sd, err,
    			err > -1 && err < sys_nerr
    				? sys_errlist[err] : "unknown" );
    	}
    #endif
    #ifdef SO_KEEPALIVE
    	tmp = 1;
    	if ( setsockopt( l.sl_sd, SOL_SOCKET, SO_KEEPALIVE,
    		(char *) &tmp, sizeof(tmp) ) == -1 )
    	{
    		int err = errno;
    		Debug( LDAP_DEBUG_ANY,
    			"slapd(%ld): setsockopt(KEEPALIVE) failed errno %d (%s)\n",
    	    	(long) l.sl_sd, err,
    			err > -1 && err < sys_nerr
    				? sys_errlist[err] : "unknown" );
    	}
    #endif
    
    	if ( bind( l.sl_sd, (struct sockaddr *) &l.sl_addr, sizeof(l.sl_addr) ) == -1 ) {
    		int err = errno;
    		Debug( LDAP_DEBUG_ANY, "daemon: bind(%ld) failed errno %d (%s)\n",
    	    	(long) l.sl_sd, err,
    			err > -1 && err < sys_nerr
    				? sys_errlist[err] : "unknown" );
    		tcp_close( l.sl_sd );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		return NULL;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    	l.sl_url = ch_strdup( url );
    
    	li = ch_malloc( sizeof( Listener ) );
    	*li = l;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	Debug( LDAP_DEBUG_TRACE, "daemon: initialized %s\n",
    		l.sl_url, 0, 0 );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	return li;
    }
    
    static int sockinit(void);
    static int sockdestroy(void);
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    int slapd_daemon_init(char *urls, int port, int tls_port )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	int i, rc;
    	char **u;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #ifndef HAVE_TLS
    	assert( tls_port == 0 );
    #endif
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	Debug( LDAP_DEBUG_ARGS, "daemon_init: %s (%d/%d)\n",
    		urls ? urls : "<null>", port, tls_port );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if( rc = sockinit() ) {
    		return rc;
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    #ifdef HAVE_SYSCONF
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	dtblsize = sysconf( _SC_OPEN_MAX );
    
    #elif HAVE_GETDTABLESIZE
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	dtblsize = getdtablesize();
    
    #else
    
    Gary Williams's avatar
    Gary Williams committed
    	dtblsize = FD_SETSIZE;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #ifdef FD_SETSIZE
    
    	if(dtblsize > FD_SETSIZE) {
    		dtblsize = FD_SETSIZE;
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	FD_ZERO( &slap_daemon.sd_readers );
    	FD_ZERO( &slap_daemon.sd_writers );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if( urls == NULL ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		urls = "ldap:///";
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	u = str2charray( urls, " " );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if( u == NULL || u[0] == NULL ) {
    		Debug( LDAP_DEBUG_ANY, "daemon_init: no urls (%s) provided.\n",
    			urls, 0, 0 );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		return -1;
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	for( i=0; u[i] != NULL; i++ ) {
    		Debug( LDAP_DEBUG_TRACE, "daemon_init: listen on %s\n",
    			u[i], 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if( i == 0 ) {
    		Debug( LDAP_DEBUG_ANY, "daemon_init: no listeners to open (%s)\n",
    			urls, 0, 0 );
    		return -1;
    	}
    
    	Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners to open...\n",
    		i, 0, 0 );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	slap_listeners = ch_malloc( (i+1)*sizeof(Listener *) );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	for(i = 0; u[i] != NULL; i++ ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		slap_listeners[i] = open_listener( u[i], port, tls_port );
    
    		if( slap_listeners[i] == NULL ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	slap_listeners[i] = NULL;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners opened.\n",
    		i, 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    	charray_free( u );
    	ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	return !i;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    slapd_daemon_destroy(void)
    {
    	connections_destroy();
    	sockdestroy();
    	return 0;
    }
    
    
    
    static void *
    slapd_daemon_task(
    	void *ptr
    )
    {
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    	int l;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	time( &starttime );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    		if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
    			continue;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		if ( listen( slap_listeners[l]->sl_sd, 5 ) == -1 ) {
    			int err = errno;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				"daemon: listen(%s, 5) failed errno %d (%s)\n",
    					(long) slap_listeners[l]->sl_url, err,
    					err > -1 && err < sys_nerr
    					? sys_errlist[err] : "unknown" );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		slapd_add( slap_listeners[l]->sl_sd );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if ( started_event != NULL ) {
    
    		ldap_pvt_thread_cond_signal( &started_event );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    	/* initialization complete. Here comes the loop. */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	while ( !slapd_shutdown ) {
    
    		ber_socket_t i;
    		int ns;
    
    		ber_socket_t nfds;
    
    #define SLAPD_EBADF_LIMIT 10
    
    		int ebadf = 0;
    
    #define SLAPD_IDLE_CHECK_LIMIT 4
    		time_t	last_idle_check = slap_get_time();
    		time_t	now;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		struct sockaddr_in	from;
    
    #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
            struct hostent		*hp;
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		struct timeval		*tvp;
    
    
    		if( global_idletimeout > 0 && difftime(
    			last_idle_check+global_idletimeout/SLAPD_IDLE_CHECK_LIMIT,
    			now ) < 0 )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		FD_ZERO( &writefds );
    		FD_ZERO( &readfds );
    
    
    		ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    #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_writers, s )) {
    				FD_SET( &writefds, s );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			}
    		}
    
    #else
    		memcpy( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) );
    		memcpy( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) );
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    			if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    				continue;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			FD_SET( slap_listeners[l]->sl_sd, &readfds );
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    		}
    
    
    #ifndef HAVE_WINSOCK
    		nfds = slap_daemon.sd_nfds;
    #else
    		nfds = dtblsize;
    #endif
    
    		ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    		ldap_pvt_thread_mutex_lock( &active_threads_mutex );
    
    		at = active_threads;
    		ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
    
    
    #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #else
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    			if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    				continue;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    			Debug( LDAP_DEBUG_CONNS,
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				"daemon: select: listen=%d active_threads=%d tvp=%s\n",
    					slap_listeners[l]->sl_sd, at,
    					tvp == NULL ? "NULL" : "zero" );
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    		}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    		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 */
    
    #ifdef HAVE_WINSOCK
    				int err = WSAGetLastError();
    #else
    
    				if( err == EBADF && ++ebadf < SLAPD_EBADF_LIMIT) {
    					continue;
    				}
    
    				if( err != EINTR ) {
    
    					Debug( LDAP_DEBUG_CONNS,
    						"daemon: select failed (%d): %s\n",
    						err,
    						err >= 0 && err < sys_nerr
    							? sys_errlist[err] : "unknown",
    						0 );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			continue;
    
    		case 0:		/* timeout - let threads run */
    
    			ebadf = 0;
    
    			Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n",
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			    0, 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			continue;
    
    		default:	/* something happened - deal with it */
    
    			ebadf = 0;
    
    			Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n",
    				ns, 0, 0 );
    			/* FALL THRU */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		}
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    				continue;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    			if ( !FD_ISSET( slap_listeners[l]->sl_sd, &readfds ) )
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    				continue;
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			if ( (s = accept( slap_listeners[l]->sl_sd,
    
    				(struct sockaddr *) &from, &len )) == AC_SOCKET_INVALID )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				Debug( LDAP_DEBUG_ANY,
    
    				    "daemon: accept(%ld) failed errno %d (%s)\n", err,
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				    (long) slap_listeners[l]->sl_sd,
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    				    err >= 0 && err < sys_nerr ?
    
    				    sys_errlist[err] : "unknown");
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				continue;
    			}
    
    #ifdef LDAP_DEBUG
    			ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
    
    			/* 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) );
    
    			ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
    #endif
    
    			/* make sure descriptor number isn't too great */
    
    					"daemon: %ld beyond descriptor table size %ld\n",
    					(long) s, (long) dtblsize, 0 );
    
    			Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %ld\n",
    				(long) s, 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    			len = sizeof(from);
    
    			if ( getpeername( s, (struct sockaddr *) &from, &len ) == 0 ) {
    
    #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				hp = gethostbyaddr( (char *)
    				    &(from.sin_addr.s_addr),
    				    sizeof(from.sin_addr.s_addr), AF_INET );
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    					for ( p = client_name; *p; p++ ) {
    						*p = TOLOWER( (unsigned char) *p );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			} else {
    
    #ifdef HAVE_TCPD
    
    			if(!hosts_ctl("slapd",
    				client_name != NULL ? client_name : STRING_UNKNOWN,
    				client_addr != NULL ? client_addr : STRING_UNKNOWN,
    
    			   	 "fd=%ld connection from %s (%s) denied.\n",
    			   	 	(long) s,
    
    					client_name == NULL ? "unknown" : client_name,
    					client_addr == NULL ? "unknown" : client_addr,
    
    			   	  0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			}
    
    #endif /* HAVE_TCPD */
    
    			if( (id = connection_init(s, client_name, client_addr,
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #ifdef HAVE_TLS
    				slap_listeners[l]->sl_is_tls
    #else
    				0
    #endif
    				)) < 0 )
    			{
    
    					"daemon: connection_init(%ld, %s, %s) failed.\n",
    					(long) s,
    
    					client_name == NULL ? "unknown" : client_name,
    
    					client_addr == NULL ? "unknown" : client_addr);
    
    				"daemon: conn=%d fd=%ld connection from %s (%s) accepted.\n",
    				id, (long) s,
    
    				client_name == NULL ? "unknown" : client_name,
    				client_addr == NULL ? "unknown" : client_addr,
    				0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		}
    
    
    #ifdef LDAP_DEBUG
    		Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 );
    
    Gary Williams's avatar
    Gary Williams committed
    #ifdef HAVE_WINSOCK
    
    		for ( i = 0; i < readfds.fd_count; i++ ) {
    
    			Debug( LDAP_DEBUG_CONNS, " %d%s",
    				readfds.fd_array[i], "r", 0 );
    
    Gary Williams's avatar
    Gary Williams committed
    		}
    
    		for ( i = 0; i < writefds.fd_count; i++ ) {
    
    			Debug( LDAP_DEBUG_CONNS, " %d%s",
    				writefds.fd_array[i], "w", 0 );
    
    Gary Williams's avatar
    Gary Williams committed
    		}
    #else
    
    		for ( i = 0; i < nfds; i++ ) {
    			int	a, r, w;
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    			int	is_listener = 0;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    				if ( i == slap_listeners[l]->sl_sd ) {
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    					is_listener = 1;
    					break;
    				}
    			}
    			if ( is_listener ) {
    				continue;
    			}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			r = FD_ISSET( i, &readfds );
    			w = FD_ISSET( i, &writefds );
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    			if ( r || w ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
    				    r ? "r" : "", w ? "w" : "" );
    			}
    		}
    
    Gary Williams's avatar
    Gary Williams committed
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    		/* loop through the writers */
    
    Gary Williams's avatar
    Gary Williams committed
    #ifdef HAVE_WINSOCK
    
    		for ( i = 0; i < writefds.fd_count; i++ )
    #else
    		for ( i = 0; i < nfds; i++ )
    #endif
    		{
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    			int is_listener = 0;
    
    #ifdef HAVE_WINSOCK
    			wd = writefds.fd_array[i];
    #else
    			if( ! FD_ISSET( i, &writefds ) ) {
    				continue;
    			}
    			wd = i;
    #endif
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    				if ( i == slap_listeners[l]->sl_sd ) {
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    					is_listener = 1;
    					break;
    				}
    			}
    			if ( is_listener ) {
    
    Gary Williams's avatar
    Gary Williams committed
    				continue;
    			}
    			Debug( LDAP_DEBUG_CONNS,
    
    				"daemon: write active on %d\n",
    
    Gary Williams's avatar
    Gary Williams committed
    
    
    			/*
    			 * 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.
    			 */
    
    			if ( connection_write( wd ) < 0 ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				FD_CLR( (unsigned) wd, &readfds );
    
    Gary Williams's avatar
    Gary Williams committed
    			}
    
    
    #ifdef HAVE_WINSOCK
    		for ( i = 0; i < readfds.fd_count; i++ )
    
    Gary Williams's avatar
    Gary Williams committed
    #else
    
    		for ( i = 0; i < nfds; i++ )
    #endif
    		{
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    			int is_listener = 0;
    
    
    #ifdef HAVE_WINSOCK
    			rd = readfds.fd_array[i];
    #else
    			if( ! FD_ISSET( i, &readfds ) ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    				continue;
    			}
    
    Gary Williams's avatar
    Gary Williams committed
    #endif
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    
    				if ( rd == slap_listeners[l]->sl_sd ) {
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    					is_listener = 1;
    					break;
    				}
    			}
    			if ( is_listener ) {
    
    Gary Williams's avatar
    Gary Williams committed
    				continue;
    			}
    
    Gary Williams's avatar
    Gary Williams committed
    			Debug ( LDAP_DEBUG_CONNS,
    
    				"daemon: read activity on %d\n", rd, 0, 0 );
    
    			/*
    			 * 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.
    			 */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			}
    		}
    
    		ldap_pvt_thread_yield();
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    
    	if( slapd_shutdown > 0 ) {
    		Debug( LDAP_DEBUG_TRACE,
    
    			"daemon: shutdown requested and initiated.\n",
    			0, 0, 0 );
    
    
    	} else if ( slapd_shutdown < 0 ) {
    		Debug( LDAP_DEBUG_TRACE,
    
    			"daemon: abnormal condition, shutdown initiated.\n",
    			0, 0, 0 );
    
    	} else {
    		Debug( LDAP_DEBUG_TRACE,
    			"daemon: no active streams, shutdown initiated.\n",
    			0, 0, 0 );
    	}
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	for ( l = 0; slap_listeners[l] != NULL; l++ ) {
    		if ( slap_listeners[l]->sl_sd != AC_SOCKET_INVALID ) {
    			slapd_close( slap_listeners[l]->sl_sd );
    			break;
    
    Julio Sánchez Fernández's avatar
     
    Julio Sánchez Fernández committed
    		}
    
    	ldap_pvt_thread_mutex_lock( &active_threads_mutex );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	Debug( LDAP_DEBUG_ANY,
    
    	    "slapd shutdown: waiting for %d threads to terminate\n",
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	    active_threads, 0, 0 );
    	while ( active_threads > 0 ) {
    
    		ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex);
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	}
    
    	ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    }
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    int slapd_daemon( void )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	int rc;
    
    #define SLAPD_LISTENER_THREAD 1
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    #if defined( SLAPD_LISTENER_THREAD ) || !defined(HAVE_PTHREADS)
    
    
    	/* listener as a separate THREAD */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	rc = ldap_pvt_thread_create( &listener_tid,
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		0, slapd_daemon_task, NULL );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	if ( rc != 0 ) {
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		    "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		return rc;
    
    	/* wait for the listener thread to complete */
    	ldap_pvt_thread_join( listener_tid, (void *) NULL );
    #else
    	/* expermimental code */
    	listener_tid = pthread_self();
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	slapd_daemon_task( NULL );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	return 0;
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    int sockinit(void)
    
    {
        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.                                  */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    		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.                                        */
     
    	if ( LOBYTE( wsaData.wVersion ) != 2 ||
    		HIBYTE( wsaData.wVersion ) != 0 )
    	{
    	    /* Tell the user that we couldn't find a usable */
    	    /* WinSock DLL.                                  */
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	    WSACleanup();
    	    return -1; 
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    	/* The WinSock DLL is acceptable. Proceed. */
    	return 0;
    }
    
    int sockdestroy(void)
    {
    	WSACleanup();
    	return 0;
    }
    
    void hit_socket(void)
    
    	ber_socket_t s;
    	int 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 )) == AC_SOCKET_INVALID )
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			"slap_set_shutdown: socket failed\n\tWSAGetLastError=%d (%s)\n",
    			WSAGetLastError(), WSAGetLastErrorString(), 0 );
    
    
    	if ( ioctlsocket( s, FIONBIO, &on ) == -1 ) 
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    			"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 );
    
    Kurt Zeilenga's avatar
    Kurt Zeilenga committed
    
    
    	if ( connect( s, (struct sockaddr *)&bind_addr, sizeof( struct sockaddr_in )) == SOCKET_ERROR ) {