2 //#include "portable_err.h"
10 #include <ac/string.h>
12 #include <ac/unistd.h>
14 #include "ldapconfig.h"
20 int allow_severity = LOG_INFO;
21 int deny_severity = LOG_NOTICE;
22 #endif /* TCP Wrappers */
30 extern ldap_pvt_thread_cond_t started_event;
31 /* forward reference */
34 char *WSAGetLastErrorString();
35 static ldap_pvt_thread_t hit_tid;
37 #define WAKE_LISTENER(w) \
40 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\
45 #define WAKE_LISTENER(w) \
48 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );\
56 volatile sig_atomic_t slapd_shutdown = 0;
58 static int daemon_initialized = 0;
59 static ldap_pvt_thread_t listener_tid;
60 static volatile sig_atomic_t slapd_listener = 0;
64 ldap_pvt_thread_mutex_t sd_mutex;
69 /* In winsock, accept() returns values higher than dtblsize
70 so don't bother with this optimization */
80 * Add a descriptor to daemon control
82 static void slapd_add(int s) {
83 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
85 assert( !FD_ISSET( s, &slap_daemon.sd_actives ));
86 assert( !FD_ISSET( s, &slap_daemon.sd_readers ));
87 assert( !FD_ISSET( s, &slap_daemon.sd_writers ));
90 if (s >= slap_daemon.sd_nfds) {
91 slap_daemon.sd_nfds = s + 1;
95 FD_SET( (unsigned) s, &slap_daemon.sd_actives );
96 FD_SET( (unsigned) s, &slap_daemon.sd_readers );
98 Debug( LDAP_DEBUG_CONNS, "daemon: added %d%s%s\n", s,
99 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
100 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
102 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
106 * Remove the descriptor from daemon control
108 void slapd_remove(int s, int wake) {
109 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
112 Debug( LDAP_DEBUG_CONNS, "daemon: removing %d%s%s\n", s,
113 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
114 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
116 FD_CLR( (unsigned) s, &slap_daemon.sd_actives );
117 FD_CLR( (unsigned) s, &slap_daemon.sd_readers );
118 FD_CLR( (unsigned) s, &slap_daemon.sd_writers );
120 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
123 void slapd_clr_write(int s, int wake) {
124 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
127 assert( FD_ISSET( (unsigned) s, &slap_daemon.sd_actives) );
128 FD_CLR( (unsigned) s, &slap_daemon.sd_writers );
130 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
133 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
137 void slapd_set_write(int s, int wake) {
138 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
141 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
142 FD_SET( (unsigned) s, &slap_daemon.sd_writers );
144 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
147 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
151 void slapd_clr_read(int s, int wake) {
152 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
155 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
156 FD_CLR( (unsigned) s, &slap_daemon.sd_readers );
158 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
161 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
165 void slapd_set_read(int s, int wake) {
166 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
169 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
170 FD_SET( (unsigned) s, &slap_daemon.sd_readers );
172 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
175 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
179 static void slapd_close(int s) {
180 Debug( LDAP_DEBUG_CONNS, "daemon: closing %d\n", s, 0, 0 );
187 set_socket( struct sockaddr_in *addr )
190 if ( !daemon_initialized ) sockinit();
193 dtblsize = sysconf( _SC_OPEN_MAX );
194 #elif HAVE_GETDTABLESIZE
195 dtblsize = getdtablesize();
197 dtblsize = FD_SETSIZE;
201 if(dtblsize > FD_SETSIZE) {
202 dtblsize = FD_SETSIZE;
204 #endif /* !FD_SETSIZE */
209 if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) {
212 Debug( LDAP_DEBUG_ANY,
213 "daemon: socket() failed errno %d (%s)\n", err,
214 err > -1 && err < sys_nerr ? sys_errlist[err] :
217 Debug( LDAP_DEBUG_ANY,
218 "daemon: socket() failed errno %d (%s)\n",
220 WSAGetLastErrorString(), 0 );
226 if ( tcps >= dtblsize ) {
227 Debug( LDAP_DEBUG_ANY,
228 "daemon: listener descriptor %d is too great\n",
236 if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR,
237 (char *) &tmp, sizeof(tmp) ) == -1 )
240 Debug( LDAP_DEBUG_ANY,
241 "slapd(%d): setsockopt() failed errno %d (%s)\n",
243 err > -1 && err < sys_nerr
244 ? sys_errlist[err] : "unknown" );
249 if ( setsockopt( tcps, SOL_SOCKET, SO_KEEPALIVE,
250 (char *) &tmp, sizeof(tmp) ) == -1 )
253 Debug( LDAP_DEBUG_ANY,
254 "slapd(%d): setsockopt(KEEPALIVE) failed errno %d (%s)\n",
256 err > -1 && err < sys_nerr
257 ? sys_errlist[err] : "unknown" );
262 if ( bind( tcps, (struct sockaddr *) addr, sizeof(*addr) ) == -1 ) {
264 Debug( LDAP_DEBUG_ANY, "daemon: bind(%d) failed errno %d (%s)\n",
266 err > -1 && err < sys_nerr
267 ? sys_errlist[err] : "unknown" );
280 struct sockaddr_in *slapd_addr = (struct sockaddr_in*)((int *)ptr)[0];
282 tcps = ((int *)ptr) [1];
285 inetd = ( slapd_addr == NULL);
286 if ( !daemon_initialized ) sockinit();
290 ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex );
291 FD_ZERO( &slap_daemon.sd_readers );
292 FD_ZERO( &slap_daemon.sd_writers );
295 if ( listen( tcps, 5 ) == -1 ) {
297 Debug( LDAP_DEBUG_ANY,
298 "daemon: listen(%d, 5) failed errno %d (%s)\n",
300 err > -1 && err < sys_nerr
301 ? sys_errlist[err] : "unknown" );
308 if( connection_init( 0, NULL, NULL ) ) {
309 Debug( LDAP_DEBUG_ANY,
310 "connection_init(%d) failed.\n",
320 if ( started_event != NULL )
321 ldap_pvt_thread_cond_signal( &started_event );
323 // initialization complete. Here comes the loop.
324 while ( !slapd_shutdown ) {
328 #define SLAPD_EBADF_LIMIT 10
333 struct sockaddr_in from;
334 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
343 FD_ZERO( &writefds );
349 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
351 #ifdef FD_SET_MANUAL_COPY
352 for( s = 0; s < nfds; s++ ) {
353 if(FD_ISSET( &slap_sd_writers, s )) {
354 FD_SET( &writefds, s );
356 if(FD_ISSET( &slap_sd_writers, s )) {
357 FD_SET( &writefds, s );
361 memcpy( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) );
362 memcpy( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) );
365 FD_SET( (unsigned) tcps, &readfds );
368 nfds = slap_daemon.sd_nfds;
373 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
375 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
376 #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
379 tvp = active_threads ? &zero : NULL;
382 Debug( LDAP_DEBUG_CONNS,
383 "daemon: select: tcps=%d active_threads=%d tvp=%s\n",
384 tcps, active_threads,
385 tvp == NULL ? "NULL" : "zero" );
388 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
390 switch(ns = select( nfds, &readfds, &writefds, 0, tvp )) {
391 case -1: { /* failure - try again */
393 int err = WSAGetLastError();
398 if( err == EBADF && ++ebadf < SLAPD_EBADF_LIMIT) {
403 Debug( LDAP_DEBUG_CONNS,
404 "daemon: select failed (%d): %s\n",
406 err >= 0 && err < sys_nerr
407 ? sys_errlist[err] : "unknown",
416 case 0: /* timeout - let threads run */
418 Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n",
420 ldap_pvt_thread_yield();
423 default: /* something happened - deal with it */
425 Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n",
430 if ( FD_ISSET( tcps, &readfds ) ) {
432 int len = sizeof(from);
435 if ( (s = accept( tcps,
436 (struct sockaddr *) &from, &len )) == -1 )
439 Debug( LDAP_DEBUG_ANY,
440 "daemon: accept(%d) failed errno %d (%s)\n", err,
441 tcps, err >= 0 && err < sys_nerr ?
442 sys_errlist[err] : "unknown");
447 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
449 /* newly accepted stream should not be in any of the FD SETS */
451 assert( !FD_ISSET( s, &slap_daemon.sd_actives) );
452 assert( !FD_ISSET( s, &slap_daemon.sd_readers) );
453 assert( !FD_ISSET( s, &slap_daemon.sd_writers) );
455 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
459 /* make sure descriptor number isn't too great */
460 if ( s >= dtblsize ) {
461 Debug( LDAP_DEBUG_ANY,
462 "daemon: %d beyond descriptor table size %d\n",
469 Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %d\n",
473 if ( getpeername( s, (struct sockaddr *) &from, &len ) == 0 ) {
474 client_addr = inet_ntoa( from.sin_addr );
476 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
477 hp = gethostbyaddr( (char *)
478 &(from.sin_addr.s_addr),
479 sizeof(from.sin_addr.s_addr), AF_INET );
483 client_name = hp->h_name;
485 /* normalize the domain */
486 for ( p = client_name; *p; p++ ) {
487 *p = TOLOWER( (unsigned char) *p );
503 if(!hosts_ctl("slapd",
504 client_name != NULL ? client_name : STRING_UNKNOWN,
505 client_addr != NULL ? client_addr : STRING_UNKNOWN,
509 Statslog( LDAP_DEBUG_ANY,
510 "fd=%d connection from %s (%s) denied.\n",
512 client_name == NULL ? "unknown" : client_name,
513 client_addr == NULL ? "unknown" : client_addr,
519 #endif /* HAVE_TCPD */
521 if( (id = connection_init(s, client_name, client_addr)) < 0 ) {
522 Debug( LDAP_DEBUG_ANY,
523 "daemon: connection_init(%d, %s, %s) failed.\n",
525 client_name == NULL ? "unknown" : client_name,
526 client_addr == NULL ? "unknown" : client_addr);
531 Statslog( LDAP_DEBUG_STATS,
532 "daemon: conn=%d fd=%d connection from %s (%s) accepted.\n",
534 client_name == NULL ? "unknown" : client_name,
535 client_addr == NULL ? "unknown" : client_addr,
543 Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 );
545 for ( i = 0; i < readfds.fd_count; i++ ) {
546 Debug( LDAP_DEBUG_CONNS, " %d%s", readfds.fd_array[i], "r" );
548 for ( i = 0; i < writefds.fd_count; i++ ) {
549 Debug( LDAP_DEBUG_CONNS, " %d%s", writefds.fd_array[i], "w" );
552 for ( i = 0; i < nfds; i++ ) {
555 r = FD_ISSET( i, &readfds );
556 w = FD_ISSET( i, &writefds );
557 if ( i != tcps && (r || w) ) {
558 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
559 r ? "r" : "", w ? "w" : "" );
563 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
566 /* loop through the writers */
568 for ( i = 0; i < writefds.fd_count; i++ )
570 for ( i = 0; i < nfds; i++ )
576 wd = writefds.fd_array[i];
578 if( ! FD_ISSET( i, &writefds ) ) {
587 Debug( LDAP_DEBUG_CONNS,
588 "daemon: write active on %d\n",
592 * NOTE: it is possible that the connection was closed
593 * and that the stream is now inactive.
594 * connection_write() must valid the stream is still
598 if ( connection_write( wd ) < 0 ) {
599 FD_CLR( (unsigned) wd, &readfds );
605 for ( i = 0; i < readfds.fd_count; i++ )
607 for ( i = 0; i < nfds; i++ )
613 rd = readfds.fd_array[i];
615 if( ! FD_ISSET( i, &readfds ) ) {
625 Debug ( LDAP_DEBUG_CONNS,
626 "daemon: read activity on %d\n", rd, 0, 0 );
629 * NOTE: it is possible that the connection was closed
630 * and that the stream is now inactive.
631 * connection_read() must valid the stream is still
635 if ( connection_read( rd ) < 0 ) {
639 ldap_pvt_thread_yield();
642 if( slapd_shutdown > 0 ) {
643 Debug( LDAP_DEBUG_TRACE,
644 "daemon: shutdown requested and initiated.\n",
647 } else if ( slapd_shutdown < 0 ) {
648 Debug( LDAP_DEBUG_TRACE,
649 "daemon: abnormal condition, shutdown initiated.\n",
652 Debug( LDAP_DEBUG_TRACE,
653 "daemon: no active streams, shutdown initiated.\n",
661 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
662 Debug( LDAP_DEBUG_ANY,
663 "slapd shutdown: waiting for %d threads to terminate\n",
664 active_threads, 0, 0 );
665 while ( active_threads > 0 ) {
666 ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex);
668 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
674 int slapd_daemon( int inetd, int tcps )
677 int *args = ch_malloc( sizeof( int[2] ) );
681 if ( !daemon_initialized ) sockinit();
685 #define SLAPD_LISTENER_THREAD 1
686 #if defined( SLAPD_LISTENER_THREAD ) || !defined(HAVE_PTHREADS)
688 /* listener as a separate THREAD */
689 rc = ldap_pvt_thread_create( &listener_tid,
690 0, slapd_daemon_task, args );
693 Debug( LDAP_DEBUG_ANY,
694 "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 );
698 /* wait for the listener thread to complete */
699 ldap_pvt_thread_join( listener_tid, (void *) NULL );
701 /* expermimental code */
702 listener_tid = pthread_self();
703 slapd_daemon_task( args );
709 connections_destroy();
721 WORD wVersionRequested;
725 wVersionRequested = MAKEWORD( 2, 0 );
727 err = WSAStartup( wVersionRequested, &wsaData );
729 /* Tell the user that we couldn't find a usable */
734 /* Confirm that the WinSock DLL supports 2.0.*/
735 /* Note that if the DLL supports versions greater */
736 /* than 2.0 in addition to 2.0, it will still return */
737 /* 2.0 in wVersion since that is the version we */
740 if ( LOBYTE( wsaData.wVersion ) != 2 ||
741 HIBYTE( wsaData.wVersion ) != 0 )
743 /* Tell the user that we couldn't find a usable */
748 daemon_initialized = 1;
749 } /* The WinSock DLL is acceptable. Proceed. */
754 extern struct sockaddr_in bind_addr;
756 /* throw something at the socket to terminate the select() in the daemon thread. */
757 if (( s = socket( AF_INET, SOCK_STREAM, 0 )) == INVALID_SOCKET )
758 Debug( LDAP_DEBUG_ANY,
759 "slap_set_shutdown: socket failed\n\tWSAGetLastError=%d (%s)\n",
760 WSAGetLastError(), WSAGetLastErrorString(), 0 );
762 if ( ioctlsocket( s, FIONBIO, &on ) == -1 )
763 Debug( LDAP_DEBUG_ANY,
764 "slap_set_shutdown:FIONBIO ioctl on %d faled\n\tWSAGetLastError=%d (%s)\n",
765 s, WSAGetLastError(), WSAGetLastError() );
767 bind_addr.sin_addr.s_addr = htonl( INADDR_LOOPBACK );
769 if ( connect( s, (struct sockaddr *)&bind_addr, sizeof( struct sockaddr_in )) == SOCKET_ERROR ) {
770 Debug( LDAP_DEBUG_ANY,
771 "hit_socket: error on connect: %d\n", WSAGetLastError(), 0 );
772 /* we can probably expect some error to occur here, mostly WSAEWOULDBLOCK */
781 if ( WSAStartup( 0x0101, &wsaData ) != 0 ) {
784 daemon_initialized = 1;
789 daemon_initialized = 1;
795 slap_set_shutdown( int sig )
797 slapd_shutdown = sig;
800 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
803 Debug( LDAP_DEBUG_TRACE, "Shutdown %d ordered", sig, 0 );
804 // trying to "hit" the socket seems to always get a
805 // EWOULDBLOCK error, so just close the listen socket to
806 // break out of the select since we're shutting down anyway
810 (void) SIGNAL( sig, slap_set_shutdown );
814 slap_do_nothing( int sig )
817 (void) SIGNAL( sig, slap_do_nothing );