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>
20 #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 #define assert( cond )
38 int allow_severity = LOG_INFO;
39 int deny_severity = LOG_NOTICE;
40 #endif /* TCP Wrappers */
45 static volatile sig_atomic_t slapd_shutdown = 0;
47 /* a link to the slapd.conf configuration parameters */
48 extern char *slapd_pid_file;
49 extern char *slapd_args_file;
58 unsigned long len, tag, msgid;
61 struct sockaddr_in addr;
68 dtblsize = sysconf( _SC_OPEN_MAX );
69 #elif HAVE_GETDTABLESIZE
70 dtblsize = getdtablesize();
76 if(dtblsize > FD_SETSIZE) {
77 dtblsize = FD_SETSIZE;
79 #endif /* !FD_SETSIZE */
81 c = (Connection *) ch_calloc( 1, dtblsize * sizeof(Connection) );
83 for ( i = 0; i < dtblsize; i++ ) {
89 lber_pvt_sb_init( &c[i].c_sb );
90 c[i].c_writewaiter = 0;
92 ldap_pvt_thread_mutex_init( &c[i].c_dnmutex );
93 ldap_pvt_thread_mutex_init( &c[i].c_opsmutex );
94 ldap_pvt_thread_mutex_init( &c[i].c_pdumutex );
95 ldap_pvt_thread_cond_init( &c[i].c_wcv );
98 if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) {
99 Debug( LDAP_DEBUG_ANY, "socket() failed errno %d (%s)", errno,
100 errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
106 if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR, (char *) &i,
107 sizeof(i) ) == -1 ) {
108 Debug( LDAP_DEBUG_ANY, "setsockopt() failed errno %d (%s)",
109 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
113 (void) memset( (void *) &addr, '\0', sizeof(addr) );
114 addr.sin_family = AF_INET;
115 addr.sin_addr.s_addr = INADDR_ANY;
116 addr.sin_port = htons( (int)port );
117 if ( bind( tcps, (struct sockaddr *) &addr, sizeof(addr) ) == -1 ) {
118 Debug( LDAP_DEBUG_ANY, "bind() failed errno %d (%s)\n",
119 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
124 if ( listen( tcps, 5 ) == -1 ) {
125 Debug( LDAP_DEBUG_ANY, "listen() failed errno %d (%s)",
126 errno, errno > -1 && errno < sys_nerr ? sys_errlist[errno] :
131 Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
133 if (( slapd_pid_file != NULL ) &&
134 (( fp = fopen( slapd_pid_file, "w" )) != NULL )) {
135 fprintf( fp, "%d\n", (int) getpid() );
139 if (( slapd_args_file != NULL ) &&
140 (( fp = fopen( slapd_args_file, "w" )) != NULL )) {
141 for ( i = 0; i < g_argc; i++ ) {
142 fprintf( fp, "%s ", g_argv[i] );
148 while ( !slapd_shutdown ) {
149 struct sockaddr_in from;
159 FD_ZERO( &writefds );
161 FD_SET( tcps, &readfds );
166 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
167 Debug( LDAP_DEBUG_CONNS,
168 "listening for connections on %d, activity on:",
173 ldap_pvt_thread_mutex_lock( &new_conn_mutex );
174 for ( i = 0; i < dtblsize; i++ ) {
175 if ( lber_pvt_sb_in_use( &c[i].c_sb )) {
176 FD_SET( lber_pvt_sb_get_desc(&c[i].c_sb),
178 if (lber_pvt_sb_data_ready(&c[i].c_sb))
180 if ( c[i].c_writewaiter ) {
181 FD_SET( lber_pvt_sb_get_desc(&c[i].c_sb),
184 Debug( LDAP_DEBUG_CONNS, " %dr%s", i,
185 c[i].c_writewaiter ? "w" : "", 0 );
188 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
189 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
191 Debug( LDAP_DEBUG_CONNS, "before select active_threads %d\n",
192 active_threads, 0, 0 );
193 #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
194 tvp = (data_ready) ? &zero : NULL;
196 tvp = (active_threads || data_ready) ? &zero : NULL;
198 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
200 switch ( i = select( dtblsize, &readfds, &writefds, 0, tvp ) ) {
201 case -1: /* failure - try again */
202 Debug( LDAP_DEBUG_CONNS,
203 "select failed errno %d (%s)\n",
204 errno, errno > -1 && errno < sys_nerr ?
205 sys_errlist[errno] : "unknown", 0 );
208 case 0: /* timeout - let threads run */
209 Debug( LDAP_DEBUG_CONNS, "select timeout - yielding\n",
212 ldap_pvt_thread_yield();
215 default: /* something happened - deal with it */
216 Debug( LDAP_DEBUG_CONNS, "select activity on %d descriptors\n", i, 0, 0 );
219 ldap_pvt_thread_mutex_lock( ¤ttime_mutex );
220 time( ¤ttime );
221 ldap_pvt_thread_mutex_unlock( ¤ttime_mutex );
224 ldap_pvt_thread_mutex_lock( &new_conn_mutex );
225 if ( FD_ISSET( tcps, &readfds ) ) {
227 if ( (ns = accept( tcps, (struct sockaddr *) &from,
229 Debug( LDAP_DEBUG_ANY,
230 "accept() failed errno %d (%s)", errno,
231 errno > -1 && errno < sys_nerr ?
232 sys_errlist[errno] : "unknown", 0 );
233 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
237 /* make sure descriptor number isn't too great */
238 if ( ns >= dtblsize ) {
239 Debug( LDAP_DEBUG_ANY,
240 "new connection on %d beyond descriptor table size %d\n",
243 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
247 lber_pvt_sb_set_desc( &c[ns].c_sb, ns );
248 lber_pvt_sb_set_io( &c[ns].c_sb, &lber_pvt_sb_io_tcp, NULL );
250 if (lber_pvt_sb_set_nonblock( &c[ns].c_sb, 1)<0) {
251 Debug( LDAP_DEBUG_ANY,
252 "FIONBIO ioctl on %d failed\n", ns, 0, 0 );
255 Debug( LDAP_DEBUG_CONNS, "new connection on %d\n", ns,
260 if ( getpeername( ns, (struct sockaddr *) &from, &len )
263 client_addr = inet_ntoa( from.sin_addr );
265 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
266 hp = gethostbyaddr( (char *)
267 &(from.sin_addr.s_addr),
268 sizeof(from.sin_addr.s_addr), AF_INET );
271 client_name = hp->h_name;
273 /* normalize the domain */
274 for ( s = client_name; *s; s++ ) {
275 *s = TOLOWER( (unsigned char) *s );
291 if(!hosts_ctl("slapd",
292 client_name != NULL ? client_name : STRING_UNKNOWN,
293 client_addr != NULL ? client_addr : STRING_UNKNOWN,
297 Statslog( LDAP_DEBUG_ANY,
298 "fd=%d connection from %s (%s) denied.\n",
300 client_name == NULL ? "unknown" : client_name,
301 client_addr == NULL ? "unknown" : client_addr,
304 lber_pvt_sb_close( &c[ns].c_sb );
305 lber_pvt_sb_destroy( &c[ns].c_sb );
306 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
309 #endif /* HAVE_TCPD */
312 ldap_pvt_thread_mutex_lock( &ops_mutex );
313 c[ns].c_connid = num_conns++;
314 ldap_pvt_thread_mutex_unlock( &ops_mutex );
316 Statslog( LDAP_DEBUG_STATS,
317 "conn=%d fd=%d connection from %s (%s) accepted.\n",
319 client_name == NULL ? "unknown" : client_name,
320 client_addr == NULL ? "unknown" : client_addr,
323 if ( c[ns].c_addr != NULL ) {
324 free( c[ns].c_addr );
326 c[ns].c_addr = ch_strdup( client_addr );
328 if ( c[ns].c_domain != NULL ) {
329 free( c[ns].c_domain );
332 c[ns].c_domain = ch_strdup( client_name == NULL
333 ? "" : client_name );
335 ldap_pvt_thread_mutex_lock( &c[ns].c_dnmutex );
336 if ( c[ns].c_dn != NULL ) {
340 if ( c[ns].c_cdn != NULL ) {
344 ldap_pvt_thread_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 ldap_pvt_thread_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 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
376 c[i].c_writewaiter = 0;
377 ldap_pvt_thread_cond_signal( &c[i].c_wcv );
378 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
381 if ( FD_ISSET( i, &readfds ) ||
382 lber_pvt_sb_data_ready( &c[i].c_sb ) ) {
383 Debug( LDAP_DEBUG_CONNS,
384 "read activity on %d\n", i, 0, 0 );
386 connection_activity( &c[i] );
390 ldap_pvt_thread_yield();
393 Debug( LDAP_DEBUG_TRACE,
394 "slapd shutdown: shutdown initiated.\n",
399 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
400 Debug( LDAP_DEBUG_ANY,
401 "slapd shutdown: waiting for %d threads to terminate\n",
402 active_threads, 0, 0 );
403 while ( active_threads > 0 ) {
404 ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex);
406 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
412 slap_set_shutdown( int sig )
414 Debug( LDAP_DEBUG_ANY, "slapd got shutdown signal %d\n", sig, 0, 0 );
416 ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
419 (void) SIGNAL( sig, slap_set_shutdown );
423 slap_do_nothing( int sig )
425 Debug( LDAP_DEBUG_TRACE, "slapd got do_nothing signal %d\n", sig, 0, 0 );
428 (void) SIGNAL( sig, slap_do_nothing );