5 * Added locking of new_conn_mutex when traversing the c[] array.
14 #include <ac/signal.h>
15 #include <ac/socket.h>
16 #include <ac/string.h>
18 #include <ac/unistd.h>
21 #include "ldapconfig.h"
23 #ifdef HAVE_SYS_FILIO_H
24 #include <sys/filio.h>
25 #elif HAVE_SYS_IOCTL_H
26 #include <sys/ioctl.h>
32 int allow_severity = LOG_INFO;
33 int deny_severity = LOG_NOTICE;
34 #endif /* TCP Wrappers */
36 extern Operation *op_add();
38 extern time_t currenttime;
39 extern pthread_mutex_t currenttime_mutex;
40 extern int active_threads;
41 extern pthread_mutex_t active_threads_mutex;
42 extern pthread_mutex_t new_conn_mutex;
43 extern int slapd_shutdown;
44 extern pthread_t listener_tid;
46 extern pthread_mutex_t ops_mutex;
53 static void set_shutdown();
54 static void do_nothing();
63 unsigned long len, tag, msgid;
66 struct sockaddr_in addr;
73 dtblsize = sysconf( _SC_OPEN_MAX );
74 #elif HAVE_GETDTABLESIZE
75 dtblsize = getdtablesize();
81 if(dtblsize > FD_SETSIZE) {
82 dtblsize = FD_SETSIZE;
84 #endif /* !FD_SETSIZE */
86 c = (Connection *) ch_calloc( 1, dtblsize * sizeof(Connection) );
88 for ( i = 0; i < dtblsize; i++ ) {
94 c[i].c_sb.sb_options = LBER_NO_READ_AHEAD;
95 c[i].c_sb.sb_naddr = 0;
96 c[i].c_sb.sb_ber.ber_buf = NULL;
97 c[i].c_sb.sb_ber.ber_ptr = NULL;
98 c[i].c_sb.sb_ber.ber_end = NULL;
99 c[i].c_writewaiter = 0;
101 pthread_mutex_init( &c[i].c_dnmutex,
102 pthread_mutexattr_default );
103 pthread_mutex_init( &c[i].c_opsmutex,
104 pthread_mutexattr_default );
105 pthread_mutex_init( &c[i].c_pdumutex,
106 pthread_mutexattr_default );
107 pthread_cond_init( &c[i].c_wcv, pthread_condattr_default );
110 if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) {
111 Debug( LDAP_DEBUG_ANY, "socket() failed errno %d (%s)", errno,
112 errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
118 if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR, (char *) &i,
119 sizeof(i) ) == -1 ) {
120 Debug( LDAP_DEBUG_ANY, "setsockopt() failed errno %d (%s)",
121 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
125 (void) memset( (void *) &addr, '\0', sizeof(addr) );
126 addr.sin_family = AF_INET;
127 addr.sin_addr.s_addr = INADDR_ANY;
128 addr.sin_port = htons( port );
129 if ( bind( tcps, (struct sockaddr *) &addr, sizeof(addr) ) == -1 ) {
130 Debug( LDAP_DEBUG_ANY, "bind() failed errno %d (%s)\n",
131 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
136 if ( listen( tcps, 5 ) == -1 ) {
137 Debug( LDAP_DEBUG_ANY, "listen() failed errno %d (%s)",
138 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
143 (void) SIGNAL( SIGPIPE, SIG_IGN );
144 #ifdef HAVE_LINUX_THREADS
146 * LinuxThreads are implemented using SIGUSR1/USR2,
147 * so we'll use SIGSTKFLT and SIGUNUSED
149 (void) SIGNAL( SIGSTKFLT, (void *) do_nothing );
150 (void) SIGNAL( SIGUNUSED, (void *) set_shutdown );
152 (void) SIGNAL( SIGUSR1, (void *) do_nothing );
153 (void) SIGNAL( SIGUSR2, (void *) set_shutdown );
155 (void) SIGNAL( SIGTERM, (void *) set_shutdown );
156 (void) SIGNAL( SIGINT, (void *) set_shutdown );
157 (void) SIGNAL( SIGHUP, (void *) set_shutdown );
159 Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
161 if ( (fp = fopen( SLAPD_PIDFILE, "w" )) != NULL ) {
162 fprintf( fp, "%d\n", getpid() );
166 #ifdef SLAPD_ARGSFILE
167 if ( (fp = fopen( SLAPD_ARGSFILE, "w" )) != NULL ) {
168 for ( i = 0; i < g_argc; i++ ) {
169 fprintf( fp, "%s ", g_argv[i] );
176 while ( !slapd_shutdown ) {
177 struct sockaddr_in from;
186 FD_ZERO( &writefds );
188 FD_SET( tcps, &readfds );
193 pthread_mutex_lock( &active_threads_mutex );
194 Debug( LDAP_DEBUG_CONNS,
195 "listening for connections on %d, activity on:",
198 pthread_mutex_lock( &new_conn_mutex );
199 for ( i = 0; i < dtblsize; i++ ) {
200 if ( c[i].c_sb.sb_sd != -1 ) {
201 FD_SET( c[i].c_sb.sb_sd, &readfds );
203 if ( c[i].c_writewaiter ) {
204 FD_SET( c[i].c_sb.sb_sd, &writefds );
206 Debug( LDAP_DEBUG_CONNS, " %dr%s", i,
207 c[i].c_writewaiter ? "w" : "", 0 );
210 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
211 pthread_mutex_unlock( &new_conn_mutex );
213 Debug( LDAP_DEBUG_CONNS, "before select active_threads %d\n",
214 active_threads, 0, 0 );
215 #ifdef PREEMPTIVE_THREADS
218 tvp = active_threads ? &zero : NULL;
220 pthread_mutex_unlock( &active_threads_mutex );
222 switch ( i = select( dtblsize, &readfds, &writefds, 0, tvp ) ) {
223 case -1: /* failure - try again */
224 Debug( LDAP_DEBUG_CONNS,
225 "select failed errno %d (%s)\n",
226 errno, errno > -1 && errno < sys_nerr ?
227 sys_errlist[errno] : "unknown", 0 );
230 case 0: /* timeout - let threads run */
231 Debug( LDAP_DEBUG_CONNS, "select timeout - yielding\n",
236 default: /* something happened - deal with it */
237 Debug( LDAP_DEBUG_CONNS, "select activity on %d descriptors\n", i, 0, 0 );
240 pthread_mutex_lock( ¤ttime_mutex );
241 time( ¤ttime );
242 pthread_mutex_unlock( ¤ttime_mutex );
245 pthread_mutex_lock( &new_conn_mutex );
246 if ( FD_ISSET( tcps, &readfds ) ) {
248 if ( (ns = accept( tcps, (struct sockaddr *) &from,
250 Debug( LDAP_DEBUG_ANY,
251 "accept() failed errno %d (%s)", errno,
252 errno > -1 && errno < sys_nerr ?
253 sys_errlist[errno] : "unknown", 0 );
254 pthread_mutex_unlock( &new_conn_mutex );
257 if ( ioctl( ns, FIONBIO, (caddr_t) &on ) == -1 ) {
258 Debug( LDAP_DEBUG_ANY,
259 "FIONBIO ioctl on %d failed\n", ns, 0, 0 );
262 c[ns].c_sb.sb_sd = ns;
263 Debug( LDAP_DEBUG_CONNS, "new connection on %d\n", ns,
266 pthread_mutex_lock( &ops_mutex );
267 c[ns].c_connid = num_conns++;
268 pthread_mutex_unlock( &ops_mutex );
272 if ( getpeername( ns, (struct sockaddr *) &from, &len )
275 client_addr = inet_ntoa( from.sin_addr );
277 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
278 hp = gethostbyaddr( (char *)
279 &(from.sin_addr.s_addr),
280 sizeof(from.sin_addr.s_addr), AF_INET );
283 client_name = hp->h_name;
285 /* normalize the domain */
286 for ( s = client_name; *s; s++ ) {
303 if(!hosts_ctl("slapd", client_name, client_addr,
307 Statslog( LDAP_DEBUG_STATS,
308 "conn=%d fd=%d connection from %s (%s) denied.\n",
310 client_name == NULL ? "unknown" : client_name,
311 client_addr == NULL ? "unknown" : client_addr,
315 pthread_mutex_unlock( &new_conn_mutex );
318 #endif /* HAVE_TCPD */
320 Statslog( LDAP_DEBUG_STATS,
321 "conn=%d fd=%d connection from %s (%s) accepted.\n",
323 client_name == NULL ? "unknown" : client_name,
324 client_addr == NULL ? "unknown" : client_addr,
327 if ( c[ns].c_addr != NULL ) {
328 free( c[ns].c_addr );
330 c[ns].c_addr = strdup( client_addr );
332 if ( c[ns].c_domain != NULL ) {
333 free( c[ns].c_domain );
336 c[ns].c_domain = strdup( client_name == NULL
337 ? "" : client_name );
339 pthread_mutex_lock( &c[ns].c_dnmutex );
340 if ( c[ns].c_dn != NULL ) {
344 pthread_mutex_unlock( &c[ns].c_dnmutex );
345 c[ns].c_starttime = currenttime;
346 c[ns].c_opsinitiated = 0;
347 c[ns].c_opscompleted = 0;
349 pthread_mutex_unlock( &new_conn_mutex );
351 Debug( LDAP_DEBUG_CONNS, "activity on:", 0, 0, 0 );
352 for ( i = 0; i < dtblsize; i++ ) {
355 r = FD_ISSET( i, &readfds );
356 w = FD_ISSET( i, &writefds );
357 if ( i != tcps && (r || w) ) {
358 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
359 r ? "r" : "", w ? "w" : "" );
362 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
364 for ( i = 0; i < dtblsize; i++ ) {
365 if ( i == tcps || (! FD_ISSET( i, &readfds ) &&
366 ! FD_ISSET( i, &writefds )) ) {
370 if ( FD_ISSET( i, &writefds ) ) {
371 Debug( LDAP_DEBUG_CONNS,
372 "signaling write waiter on %d\n", i, 0, 0 );
374 pthread_mutex_lock( &active_threads_mutex );
375 pthread_cond_signal( &c[i].c_wcv );
376 c[i].c_writewaiter = 0;
378 pthread_mutex_unlock( &active_threads_mutex );
381 if ( FD_ISSET( i, &readfds ) ) {
382 Debug( LDAP_DEBUG_CONNS,
383 "read activity on %d\n", i, 0, 0 );
385 connection_activity( &c[i] );
393 pthread_mutex_lock( &active_threads_mutex );
394 Debug( LDAP_DEBUG_ANY,
395 "slapd shutting down - waiting for %d threads to terminate\n",
396 active_threads, 0, 0 );
397 while ( active_threads > 0 ) {
398 pthread_mutex_unlock( &active_threads_mutex );
400 pthread_mutex_lock( &active_threads_mutex );
402 pthread_mutex_unlock( &active_threads_mutex );
404 /* let backends do whatever cleanup they need to do */
405 Debug( LDAP_DEBUG_TRACE,
406 "slapd shutting down - waiting for backends to close down\n", 0, 0,
409 Debug( LDAP_DEBUG_ANY, "slapd stopping\n", 0, 0, 0 );
415 Debug( LDAP_DEBUG_ANY, "slapd got shutdown signal\n", 0, 0, 0 );
417 #ifdef HAVE_LINUX_THREADS
419 * LinuxThreads are implemented using SIGUSR1/USR2,
420 * so we'll use SIGSTKFLT and SIGUNUSED
422 pthread_kill( listener_tid, SIGSTKFLT );
423 (void) SIGNAL( SIGUNUSED, (void *) set_shutdown );
425 pthread_kill( listener_tid, SIGUSR1 );
426 (void) SIGNAL( SIGUSR2, (void *) set_shutdown );
428 (void) SIGNAL( SIGTERM, (void *) set_shutdown );
429 (void) SIGNAL( SIGINT, (void *) set_shutdown );
430 (void) SIGNAL( SIGHUP, (void *) set_shutdown );
436 Debug( LDAP_DEBUG_TRACE, "slapd got do_nothing signal\n", 0, 0, 0 );
437 #ifdef HAVE_LINUX_THREADS
439 * LinuxThreads are implemented using SIGUSR1/USR2,
440 * so we'll use SIGSTKFLT and SIGUNUSED
442 (void) SIGNAL( SIGSTKFLT, (void *) do_nothing );
444 (void) SIGNAL( SIGUSR1, (void *) do_nothing );