11 #include <ac/unistd.h>
13 #include "ldapconfig.h"
19 int allow_severity = LOG_INFO;
20 int deny_severity = LOG_NOTICE;
21 #endif /* TCP Wrappers */
26 static ldap_pvt_thread_t listener_tid;
27 static volatile sig_atomic_t slapd_shutdown = 0;
28 static volatile sig_atomic_t slapd_listener = 0;
31 ldap_pvt_thread_mutex_t sd_mutex;
36 /* In winsock, accept() returns values higher than dtblsize
37 so don't bother with this optimization */
47 * Add a descriptor to daemon control
49 static void slapd_add(int s) {
50 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
52 assert( !FD_ISSET( s, &slap_daemon.sd_actives ));
53 assert( !FD_ISSET( s, &slap_daemon.sd_readers ));
54 assert( !FD_ISSET( s, &slap_daemon.sd_writers ));
57 if (s >= slap_daemon.sd_nfds) {
58 slap_daemon.sd_nfds = s + 1;
62 FD_SET( s, &slap_daemon.sd_actives );
63 FD_SET( s, &slap_daemon.sd_readers );
65 Debug( LDAP_DEBUG_CONNS, "daemon: added %d%s%s\n", s,
66 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
67 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
69 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
73 * Remove the descriptor from daemon control
75 void slapd_remove(int s) {
76 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
78 assert( FD_ISSET( s, &slap_daemon.sd_actives ));
80 Debug( LDAP_DEBUG_CONNS, "daemon: removing %d%s%s\n", s,
81 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
82 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
84 FD_CLR( s, &slap_daemon.sd_actives );
85 FD_CLR( s, &slap_daemon.sd_readers );
86 FD_CLR( s, &slap_daemon.sd_writers );
88 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
91 void slapd_clr_write(int s, int wake) {
92 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
94 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
95 FD_CLR( s, &slap_daemon.sd_writers );
97 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
100 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
104 void slapd_set_write(int s, int wake) {
105 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
107 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
108 FD_SET( s, &slap_daemon.sd_writers );
110 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
113 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
117 void slapd_clr_read(int s, int wake) {
118 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
120 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
121 FD_CLR( s, &slap_daemon.sd_readers );
123 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
126 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
130 void slapd_set_read(int s, int wake) {
131 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
133 assert( FD_ISSET( s, &slap_daemon.sd_actives) );
134 FD_SET( s, &slap_daemon.sd_readers );
136 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
139 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
143 static void slapd_close(int s) {
146 Debug( LDAP_DEBUG_CONNS, "daemon: closing %d\n", s, 0, 0 );
151 set_socket( struct sockaddr_in *addr )
156 dtblsize = sysconf( _SC_OPEN_MAX );
157 #elif HAVE_GETDTABLESIZE
158 dtblsize = getdtablesize();
160 dtblsize = FD_SETSIZE;
164 if(dtblsize > FD_SETSIZE) {
165 dtblsize = FD_SETSIZE;
167 #endif /* !FD_SETSIZE */
172 if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) {
174 Debug( LDAP_DEBUG_ANY,
175 "daemon: socket() failed errno %d (%s)\n", err,
176 err > -1 && err < sys_nerr ? sys_errlist[err] :
182 if ( tcps >= dtblsize ) {
183 Debug( LDAP_DEBUG_ANY,
184 "daemon: listener descriptor %d is too great\n",
191 if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR,
192 (char *) &tmp, sizeof(tmp) ) == -1 )
195 Debug( LDAP_DEBUG_ANY,
196 "slapd(%d): setsockopt() failed errno %d (%s)\n",
198 err > -1 && err < sys_nerr
199 ? sys_errlist[err] : "unknown" );
202 if ( bind( tcps, (struct sockaddr *) addr, sizeof(*addr) ) == -1 ) {
204 Debug( LDAP_DEBUG_ANY, "daemon: bind(%d) failed errno %d (%s)\n",
206 err > -1 && err < sys_nerr
207 ? sys_errlist[err] : "unknown" );
220 int inetd = ((int *)ptr) [0];
221 int tcps = ((int *)ptr) [1];
226 ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex );
227 FD_ZERO( &slap_daemon.sd_readers );
228 FD_ZERO( &slap_daemon.sd_writers );
231 if ( listen( tcps, 5 ) == -1 ) {
233 Debug( LDAP_DEBUG_ANY,
234 "daemon: listen(%d, 5) failed errno %d (%s)\n",
236 err > -1 && err < sys_nerr
237 ? sys_errlist[err] : "unknown" );
244 if( connection_init( 0, NULL, NULL ) ) {
245 Debug( LDAP_DEBUG_ANY,
246 "connection_init(%d) failed.\n",
255 while ( !slapd_shutdown ) {
262 struct sockaddr_in from;
263 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
272 FD_ZERO( &writefds );
278 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
280 #ifdef FD_SET_MANUAL_COPY
281 for( s = 0; s < nfds; s++ ) {
282 if(FD_ISSET( &slap_sd_writers, s )) {
283 FD_SET( &writefds, s );
285 if(FD_ISSET( &slap_sd_writers, s )) {
286 FD_SET( &writefds, s );
290 memcpy( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) );
291 memcpy( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) );
294 FD_SET( tcps, &readfds );
297 nfds = slap_daemon.sd_nfds;
302 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
304 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
305 #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
308 tvp = active_threads ? &zero : NULL;
311 Debug( LDAP_DEBUG_CONNS,
312 "daemon: select: tcps=%d active_threads=%d tvp=%s\n",
313 tcps, active_threads,
314 tvp == NULL ? "NULL" : "zero" );
317 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
319 switch(ns = select( nfds, &readfds, &writefds, 0, tvp )) {
320 case -1: { /* failure - try again */
323 Debug( LDAP_DEBUG_CONNS,
324 "daemon: select failed (%d): %s\n",
326 err >= 0 && err < sys_nerr
327 ? sys_errlist[err] : "unknown",
335 case 0: /* timeout - let threads run */
336 Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n",
338 ldap_pvt_thread_yield();
341 default: /* something happened - deal with it */
342 Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n",
347 if ( FD_ISSET( tcps, &readfds ) ) {
349 int len = sizeof(from);
352 if ( (s = accept( tcps,
353 (struct sockaddr *) &from, &len )) == -1 )
356 Debug( LDAP_DEBUG_ANY,
357 "daemon: accept(%d) failed errno %d (%s)\n", err,
358 tcps, err >= 0 && err < sys_nerr ?
359 sys_errlist[err] : "unknown");
363 assert( !FD_ISSET( 0, &slap_daemon.sd_actives) );
364 assert( !FD_ISSET( 0, &slap_daemon.sd_readers) );
365 assert( !FD_ISSET( 0, &slap_daemon.sd_writers) );
368 /* make sure descriptor number isn't too great */
369 if ( s >= dtblsize ) {
370 Debug( LDAP_DEBUG_ANY,
371 "daemon: %d beyond descriptor table size %d\n",
378 Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %d\n",
382 if ( getpeername( s, (struct sockaddr *) &from, &len ) == 0 ) {
383 client_addr = inet_ntoa( from.sin_addr );
385 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
386 hp = gethostbyaddr( (char *)
387 &(from.sin_addr.s_addr),
388 sizeof(from.sin_addr.s_addr), AF_INET );
392 client_name = hp->h_name;
394 /* normalize the domain */
395 for ( p = client_name; *p; p++ ) {
396 *p = TOLOWER( (unsigned char) *p );
412 if(!hosts_ctl("slapd",
413 client_name != NULL ? client_name : STRING_UNKNOWN,
414 client_addr != NULL ? client_addr : STRING_UNKNOWN,
418 Statslog( LDAP_DEBUG_ANY,
419 "fd=%d connection from %s (%s) denied.\n",
421 client_name == NULL ? "unknown" : client_name,
422 client_addr == NULL ? "unknown" : client_addr,
428 #endif /* HAVE_TCPD */
430 if( (id = connection_init(s, client_name, client_addr)) < 0 ) {
431 Debug( LDAP_DEBUG_ANY,
432 "daemon: connection_init(%d, %s, %s) failed.\n",
434 client_name == NULL ? "unknown" : client_name,
435 client_addr == NULL ? "unknown" : client_addr);
440 Statslog( LDAP_DEBUG_STATS,
441 "daemon: conn=%d fd=%d connection from %s (%s) accepted.\n",
443 client_name == NULL ? "unknown" : client_name,
444 client_addr == NULL ? "unknown" : client_addr,
452 Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 );
454 for ( i = 0; i < readfds.fd_count; i++ )
456 Debug( LDAP_DEBUG_CONNS, " %d%s", readfds.fd_array[i], "r" );
458 for ( i = 0; i < writefds.fd_count; i++ )
460 Debug( LDAP_DEBUG_CONNS, " %d%s", writefds.fd_array[i], "w" );
463 for ( i = 0; i < nfds; i++ ) {
466 r = FD_ISSET( i, &readfds );
467 w = FD_ISSET( i, &writefds );
468 if ( i != tcps && (r || w) ) {
469 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
470 r ? "r" : "", w ? "w" : "" );
474 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
477 /* loop through the writers */
479 for ( i = 0; i < writefds.fd_count; i++ ) {
480 int wd = writefds.fd_array[i];
486 Debug( LDAP_DEBUG_CONNS,
487 "daemon: signalling write waiter on %d\n",
490 assert( FD_ISSET( wd, &slap_daemon.sd_actives) );
492 slapd_clr_write( wd, 0 );
493 if ( connection_write( wd ) < 0 ) {
494 FD_CLR( wd, &readfds );
499 for ( i = 0; i < nfds; i++ ) {
503 if ( FD_ISSET( i, &writefds ) ) {
504 Debug( LDAP_DEBUG_CONNS,
505 "daemon: signaling write waiter on %d\n", i, 0, 0 );
507 assert( FD_ISSET( i, &slap_daemon.sd_actives) );
509 /* clear the write flag */
510 slapd_clr_write( i, 0 );
512 if( connection_write( i ) < 0 ) {
513 FD_CLR( i, &readfds );
521 for ( i = 0; i < readfds.fd_count; i++ ) {
522 int rd = readfds.fd_array[i];
526 Debug ( LDAP_DEBUG_CONNS,
527 "daemon: read activity on %d\n", rd, 0, 0 );
528 assert( FD_ISSET( rd, &slap_daemon.sd_actives) );
530 if ( connection_read( rd ) < 0 ) {
535 for ( i = 0; i < nfds; i++ ) {
540 if ( FD_ISSET( i, &readfds ) ) {
541 Debug( LDAP_DEBUG_CONNS,
542 "daemon: read activity on %d\n", i, 0, 0 );
544 assert( FD_ISSET( i, &slap_daemon.sd_actives) );
546 if( connection_read( i ) < 0) {
552 ldap_pvt_thread_yield();
555 if( slapd_shutdown > 0 ) {
556 Debug( LDAP_DEBUG_TRACE,
557 "daemon: shutdown requested (%d) and initiated.\n",
558 (int) slapd_shutdown, 0, 0 );
560 } else if ( slapd_shutdown < 0 ) {
561 Debug( LDAP_DEBUG_TRACE,
562 "daemon: abnormal condition (%d), shutdown initiated.\n",
563 (int) slapd_shutdown, 0, 0 );
565 Debug( LDAP_DEBUG_TRACE,
566 "daemon: no active streams, shutdown initiated.\n",
574 /* we only implement "quick" shutdown */
575 connections_shutdown();
577 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
578 Debug( LDAP_DEBUG_ANY,
579 "slapd shutdown: waiting for %d threads to terminate\n",
580 active_threads, 0, 0 );
581 while ( active_threads > 0 ) {
582 ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex);
584 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
590 int slapd_daemon( int inetd, int tcps )
593 int *args = ch_malloc( sizeof( int[2] ) );
599 WORD vers = MAKEWORD( 2, 0);
602 err = WSAStartup( vers, &wsaData );
608 #define SLAPD_LISTENER_THREAD 1
609 #if SLAPD_LISTENER_THREAD
610 /* listener as a separate THREAD */
611 rc = ldap_pvt_thread_create( &listener_tid,
612 0, slapd_daemon_task, args );
615 Debug( LDAP_DEBUG_ANY,
616 "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 );
620 /* wait for the listener thread to complete */
621 ldap_pvt_thread_join( listener_tid, (void *) NULL );
623 /* expermimental code */
624 listener_tid = pthread_self();
625 slapd_daemon_task( args );
631 connections_destroy();
641 slap_set_shutdown( int sig )
643 slapd_shutdown = sig;
646 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
650 (void) SIGNAL( sig, slap_set_shutdown );
654 slap_do_nothing( int sig )
657 (void) SIGNAL( sig, slap_do_nothing );