]> git.sur5r.net Git - openldap/blob - servers/slapd/daemon.c
Fix select/yield
[openldap] / servers / slapd / daemon.c
1 /* $OpenLDAP$ */
2 /*
3  * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
4  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5  */
6
7 #include "portable.h"
8
9 #include <stdio.h>
10
11 #include <ac/ctype.h>
12 #include <ac/errno.h>
13 #include <ac/socket.h>
14 #include <ac/string.h>
15 #include <ac/time.h>
16 #include <ac/unistd.h>
17
18 #include "ldap_pvt.h"
19 #include "ldap_pvt_thread.h"
20 #include "lutil.h"
21 #include "slap.h"
22
23 #ifdef LDAP_SYNCREPL
24 #include "ldap_rq.h"
25 #endif
26
27 #ifdef HAVE_TCPD
28 #include <tcpd.h>
29 #define SLAP_STRING_UNKNOWN     STRING_UNKNOWN
30
31 int allow_severity = LOG_INFO;
32 int deny_severity = LOG_NOTICE;
33 #else /* ! TCP Wrappers */
34 #define SLAP_STRING_UNKNOWN     "unknown"
35 #endif /* ! TCP Wrappers */
36
37 #ifdef LDAP_PF_LOCAL
38 #include <sys/stat.h>
39 /* this should go in <ldap.h> as soon as it is accepted */
40 #define LDAPI_MOD_URLEXT                "x-mod"
41 #endif /* LDAP_PF_LOCAL */
42
43 #ifdef LDAP_PF_INET6
44 int slap_inet4or6 = AF_UNSPEC;
45 #else
46 int slap_inet4or6 = AF_INET;
47 #endif
48
49 /* globals */
50 time_t starttime;
51 ber_socket_t dtblsize;
52
53 Listener **slap_listeners = NULL;
54
55 #define SLAPD_LISTEN 10
56
57 static ber_socket_t wake_sds[2];
58 static int emfile;
59
60 #if defined(NO_THREADS) || defined(HAVE_GNU_PTH)
61 static int waking;
62 #define WAKE_LISTENER(w) \
63 ((w && !waking) ? tcp_write( wake_sds[1], "0", 1 ), waking=1 : 0)
64 #else
65 #define WAKE_LISTENER(w) \
66 do { if (w) tcp_write( wake_sds[1], "0", 1 ); } while(0)
67 #endif
68
69 #ifndef HAVE_WINSOCK
70 static
71 #endif
72 volatile sig_atomic_t slapd_shutdown = 0, slapd_gentle_shutdown = 0;
73
74 static struct slap_daemon {
75         ldap_pvt_thread_mutex_t sd_mutex;
76
77         ber_socket_t sd_nactives;
78
79 #ifndef HAVE_WINSOCK
80         /* In winsock, accept() returns values higher than dtblsize
81                 so don't bother with this optimization */
82         int sd_nfds;
83 #endif
84
85         fd_set sd_actives;
86         fd_set sd_readers;
87         fd_set sd_writers;
88 } slap_daemon;
89
90
91
92 #ifdef HAVE_SLP
93 /*
94  * SLP related functions
95  */
96 #include <slp.h>
97
98 #define LDAP_SRVTYPE_PREFIX "service:ldap://"
99 #define LDAPS_SRVTYPE_PREFIX "service:ldaps://"
100 static char** slapd_srvurls = NULL;
101 static SLPHandle slapd_hslp = 0;
102
103 void slapd_slp_init( const char* urls ) {
104         int i;
105
106         slapd_srvurls = ldap_str2charray( urls, " " );
107
108         if( slapd_srvurls == NULL ) return;
109
110         /* find and expand INADDR_ANY URLs */
111         for( i=0; slapd_srvurls[i] != NULL; i++ ) {
112                 if( strcmp( slapd_srvurls[i], "ldap:///" ) == 0) {
113                         char *host = ldap_pvt_get_fqdn( NULL );
114                         if ( host != NULL ) {
115                                 slapd_srvurls[i] = (char *) ch_realloc( slapd_srvurls[i],
116                                         strlen( host ) +
117                                         sizeof( LDAP_SRVTYPE_PREFIX ) );
118                                 strcpy( lutil_strcopy(slapd_srvurls[i],
119                                         LDAP_SRVTYPE_PREFIX ), host );
120
121                                 ch_free( host );
122                         }
123
124                 } else if ( strcmp( slapd_srvurls[i], "ldaps:///" ) == 0) {
125                         char *host = ldap_pvt_get_fqdn( NULL );
126                         if ( host != NULL ) {
127                                 slapd_srvurls[i] = (char *) ch_realloc( slapd_srvurls[i],
128                                         strlen( host ) +
129                                         sizeof( LDAPS_SRVTYPE_PREFIX ) );
130                                 strcpy( lutil_strcopy(slapd_srvurls[i],
131                                         LDAPS_SRVTYPE_PREFIX ), host );
132
133                                 ch_free( host );
134                         }
135                 }
136         }
137
138         /* open the SLP handle */
139         SLPOpen( "en", 0, &slapd_hslp );
140 }
141
142 void slapd_slp_deinit() {
143         if( slapd_srvurls == NULL ) return;
144
145         ldap_charray_free( slapd_srvurls );
146         slapd_srvurls = NULL;
147
148         /* close the SLP handle */
149         SLPClose( slapd_hslp );
150 }
151
152 void slapd_slp_regreport(
153         SLPHandle hslp,
154         SLPError errcode,
155         void* cookie )
156 {
157         /* empty report */
158 }
159
160 void slapd_slp_reg() {
161         int i;
162
163         if( slapd_srvurls == NULL ) return;
164
165         for( i=0; slapd_srvurls[i] != NULL; i++ ) {
166                 if( strncmp( slapd_srvurls[i], LDAP_SRVTYPE_PREFIX,
167                                 sizeof( LDAP_SRVTYPE_PREFIX ) - 1 ) == 0 ||
168                     strncmp( slapd_srvurls[i], LDAPS_SRVTYPE_PREFIX,
169                                 sizeof( LDAPS_SRVTYPE_PREFIX ) - 1 ) == 0 )
170                 {
171                         SLPReg( slapd_hslp,
172                                 slapd_srvurls[i],
173                                 SLP_LIFETIME_MAXIMUM,
174                                 "ldap",
175                                 "",
176                                 1,
177                                 slapd_slp_regreport,
178                                 NULL );
179                 }
180         }
181 }
182
183 void slapd_slp_dereg() {
184         int i;
185
186         if( slapd_srvurls == NULL ) return;
187
188         for( i=0; slapd_srvurls[i] != NULL; i++ ) {
189                 SLPDereg( slapd_hslp,
190                         slapd_srvurls[i],
191                         slapd_slp_regreport,
192                         NULL );
193         }
194 }
195 #endif /* HAVE_SLP */
196
197 /*
198  * Add a descriptor to daemon control
199  */
200 static void slapd_add(ber_socket_t s) {
201         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
202
203         assert( !FD_ISSET( s, &slap_daemon.sd_actives ));
204         assert( !FD_ISSET( s, &slap_daemon.sd_readers ));
205         assert( !FD_ISSET( s, &slap_daemon.sd_writers ));
206
207 #ifndef HAVE_WINSOCK
208         if (s >= slap_daemon.sd_nfds) {
209                 slap_daemon.sd_nfds = s + 1;
210         }
211 #endif
212
213         slap_daemon.sd_nactives++;
214
215         FD_SET( s, &slap_daemon.sd_actives );
216         FD_SET( s, &slap_daemon.sd_readers );
217
218 #ifdef NEW_LOGGING
219         LDAP_LOG( CONNECTION, DETAIL1, 
220                 "slapd_add: added %ld%s%s\n", (long)s,
221                 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
222                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
223 #else
224         Debug( LDAP_DEBUG_CONNS, "daemon: added %ld%s%s\n",
225                 (long) s,
226             FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
227                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
228 #endif
229         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
230 }
231
232 /*
233  * Remove the descriptor from daemon control
234  */
235 void slapd_remove(ber_socket_t s, int wake) {
236         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
237
238         slap_daemon.sd_nactives--;
239
240 #ifdef NEW_LOGGING
241         LDAP_LOG( CONNECTION, DETAIL1, 
242                 "slapd_remove: removing %ld%s%s\n", (long) s,
243                 FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
244                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : ""  );
245 #else
246         Debug( LDAP_DEBUG_CONNS, "daemon: removing %ld%s%s\n",
247                 (long) s,
248             FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
249                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
250 #endif
251         FD_CLR( s, &slap_daemon.sd_actives );
252         FD_CLR( s, &slap_daemon.sd_readers );
253         FD_CLR( s, &slap_daemon.sd_writers );
254
255         /* If we ran out of file descriptors, we dropped a listener from
256          * the select() loop. Now that we're removing a session from our
257          * control, we can try to resume a dropped listener to use.
258          */
259         if ( emfile ) {
260                 int i;
261                 for ( i = 0; slap_listeners[i] != NULL; i++ ) {
262                         if ( slap_listeners[i]->sl_sd != AC_SOCKET_INVALID ) {
263                                 if ( slap_listeners[i]->sl_sd == s ) continue;
264                                 if ( slap_listeners[i]->sl_is_mute ) {
265                                         slap_listeners[i]->sl_is_mute = 0;
266                                         emfile--;
267                                         break;
268                                 }
269                         }
270                 }
271                 /* Walked the entire list without enabling anything; emfile
272                  * counter is stale. Reset it.
273                  */
274                 if ( slap_listeners[i] == NULL )
275                         emfile = 0;
276         }
277         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
278         WAKE_LISTENER(wake || slapd_gentle_shutdown == 2);
279 }
280
281 void slapd_clr_write(ber_socket_t s, int wake) {
282         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
283
284         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
285         FD_CLR( s, &slap_daemon.sd_writers );
286
287         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
288         WAKE_LISTENER(wake);
289 }
290
291 void slapd_set_write(ber_socket_t s, int wake) {
292         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
293
294         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
295         if (!FD_ISSET(s, &slap_daemon.sd_writers))
296             FD_SET( (unsigned) s, &slap_daemon.sd_writers );
297
298         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
299         WAKE_LISTENER(wake);
300 }
301
302 void slapd_clr_read(ber_socket_t s, int wake) {
303         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
304
305         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
306         FD_CLR( s, &slap_daemon.sd_readers );
307
308         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
309         WAKE_LISTENER(wake);
310 }
311
312 void slapd_set_read(ber_socket_t s, int wake) {
313         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
314
315         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
316         if (!FD_ISSET(s, &slap_daemon.sd_readers))
317             FD_SET( s, &slap_daemon.sd_readers );
318
319         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
320         WAKE_LISTENER(wake);
321 }
322
323 static void slapd_close(ber_socket_t s) {
324 #ifdef NEW_LOGGING
325         LDAP_LOG( CONNECTION, DETAIL1, "slapd_close: closing %ld\n", (long)s, 0, 0);
326 #else
327         Debug( LDAP_DEBUG_CONNS, "daemon: closing %ld\n",
328                 (long) s, 0, 0 );
329 #endif
330         tcp_close(s);
331 }
332
333 static void slap_free_listener_addresses(struct sockaddr **sal)
334 {
335         struct sockaddr **sap;
336
337         if (sal == NULL) {
338                 return;
339         }
340
341         for (sap = sal; *sap != NULL; sap++) {
342                 ch_free(*sap);
343         }
344
345         ch_free(sal);
346 }
347
348 #if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD)
349 static int get_url_perms(
350         char    **exts,
351         mode_t  *perms,
352         int     *crit )
353 {
354         int     i;
355
356         assert( exts );
357         assert( perms );
358         assert( crit );
359
360         *crit = 0;
361         for ( i = 0; exts[ i ]; i++ ) {
362                 char    *type = exts[ i ];
363                 int     c = 0;
364
365                 if ( type[ 0 ] == '!' ) {
366                         c = 1;
367                         type++;
368                 }
369
370                 if ( strncasecmp( type, LDAPI_MOD_URLEXT "=", sizeof(LDAPI_MOD_URLEXT "=") - 1 ) == 0 ) {
371                         char    *value = type
372                                 + ( sizeof(LDAPI_MOD_URLEXT "=") - 1 );
373                         mode_t  p = 0;
374                         int     j;
375
376                         switch (strlen(value)) {
377                         case 4:
378                                 /* skip leading '0' */
379                                 if ( value[ 0 ] != '0' ) {
380                                         return LDAP_OTHER;
381                                 }
382                                 value++;
383
384                         case 3:
385                                 for ( j = 0; j < 3; j++) {
386                                         int     v;
387
388                                         v = value[ j ] - '0';
389
390                                         if ( v < 0 || v > 7 ) {
391                                                 return LDAP_OTHER;
392                                         }
393
394                                         p |= v << 3*(2-j);
395                                 }
396                                 break;
397
398                         case 10:
399                                 for ( j = 1; j < 10; j++ ) {
400                                         static mode_t   m[] = { 0, 
401                                                 S_IRUSR, S_IWUSR, S_IXUSR,
402                                                 S_IRGRP, S_IWGRP, S_IXGRP,
403                                                 S_IROTH, S_IWOTH, S_IXOTH
404                                         };
405                                         static char     c[] = "-rwxrwxrwx"; 
406
407                                         if ( value[ j ] == c[ j ] ) {
408                                                 p |= m[ j ];
409         
410                                         } else if ( value[ j ] != '-' ) {
411                                                 return LDAP_OTHER;
412                                         }
413                                 }
414                                 break;
415
416                         default:
417                                 return LDAP_OTHER;
418                         } 
419
420                         *crit = c;
421                         *perms = p;
422
423                         return LDAP_SUCCESS;
424                 }
425         }
426
427         return LDAP_OTHER;
428 }
429 #endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */
430
431 /* port = 0 indicates AF_LOCAL */
432 static int slap_get_listener_addresses(
433         const char *host,
434         unsigned short port,
435         struct sockaddr ***sal)
436 {
437         struct sockaddr **sap;
438
439 #ifdef LDAP_PF_LOCAL
440         if ( port == 0 ) {
441                 *sal = ch_malloc(2 * sizeof(void *));
442                 if (*sal == NULL) {
443                         return -1;
444                 }
445
446                 sap = *sal;
447                 *sap = ch_malloc(sizeof(struct sockaddr_un));
448                 if (*sap == NULL)
449                         goto errexit;
450                 sap[1] = NULL;
451
452                 if ( strlen(host) >
453                      (sizeof(((struct sockaddr_un *)*sap)->sun_path) - 1) ) {
454 #ifdef NEW_LOGGING
455                         LDAP_LOG( CONNECTION, INFO, 
456                                 "slap_get_listener_addresses: domain socket path (%s) "
457                                 "too long in URL\n", host, 0, 0 );
458 #else
459                         Debug( LDAP_DEBUG_ANY,
460                                "daemon: domain socket path (%s) too long in URL",
461                                host, 0, 0);
462 #endif
463                         goto errexit;
464                 }
465
466                 (void)memset( (void *)*sap, '\0', sizeof(struct sockaddr_un) );
467                 (*sap)->sa_family = AF_LOCAL;
468                 strcpy( ((struct sockaddr_un *)*sap)->sun_path, host );
469         } else
470 #endif
471         {
472 #ifdef HAVE_GETADDRINFO
473                 struct addrinfo hints, *res, *sai;
474                 int n, err;
475                 char serv[7];
476
477                 memset( &hints, '\0', sizeof(hints) );
478                 hints.ai_flags = AI_PASSIVE;
479                 hints.ai_socktype = SOCK_STREAM;
480                 hints.ai_family = slap_inet4or6;
481                 snprintf(serv, sizeof serv, "%d", port);
482
483                 if ( (err = getaddrinfo(host, serv, &hints, &res)) ) {
484 #ifdef NEW_LOGGING
485                         LDAP_LOG( CONNECTION, INFO, 
486                                    "slap_get_listener_addresses: getaddrinfo failed: %s\n",
487                                    AC_GAI_STRERROR(err), 0, 0 );
488 #else
489                         Debug( LDAP_DEBUG_ANY, "daemon: getaddrinfo failed: %s\n",
490                                 AC_GAI_STRERROR(err), 0, 0);
491 #endif
492                         return -1;
493                 }
494
495                 sai = res;
496                 for (n=2; (sai = sai->ai_next) != NULL; n++) {
497                         /* EMPTY */ ;
498                 }
499                 *sal = ch_calloc(n, sizeof(void *));
500                 if (*sal == NULL) {
501                         return -1;
502                 }
503
504                 sap = *sal;
505                 *sap = NULL;
506
507                 for ( sai=res; sai; sai=sai->ai_next ) {
508                         if( sai->ai_addr == NULL ) {
509 #ifdef NEW_LOGGING
510                                 LDAP_LOG( CONNECTION, INFO,
511                                         "slap_get_listener_addresses: "
512                                         "getaddrinfo ai_addr is NULL?\n", 0, 0, 0 );
513 #else
514                                 Debug( LDAP_DEBUG_ANY, "slap_get_listener_addresses: "
515                                         "getaddrinfo ai_addr is NULL?\n", 0, 0, 0 );
516 #endif
517                                 freeaddrinfo(res);
518                                 goto errexit;
519                         }
520
521                         switch (sai->ai_family) {
522 #  ifdef LDAP_PF_INET6
523                         case AF_INET6:
524                                 *sap = ch_malloc(sizeof(struct sockaddr_in6));
525                                 if (*sap == NULL) {
526                                         freeaddrinfo(res);
527                                         goto errexit;
528                                 }
529                                 *(struct sockaddr_in6 *)*sap =
530                                         *((struct sockaddr_in6 *)sai->ai_addr);
531                                 break;
532 #  endif
533                         case AF_INET:
534                                 *sap = ch_malloc(sizeof(struct sockaddr_in));
535                                 if (*sap == NULL) {
536                                         freeaddrinfo(res);
537                                         goto errexit;
538                                 }
539                                 *(struct sockaddr_in *)*sap =
540                                         *((struct sockaddr_in *)sai->ai_addr);
541                                 break;
542                         default:
543                                 *sap = NULL;
544                                 break;
545                         }
546
547                         if (*sap != NULL) {
548                                 (*sap)->sa_family = sai->ai_family;
549                                 sap++;
550                                 *sap = NULL;
551                         }
552                 }
553
554                 freeaddrinfo(res);
555 #else
556                 int i, n = 1;
557                 struct in_addr in;
558                 struct hostent *he = NULL;
559
560                 if ( host == NULL ) {
561                         in.s_addr = htonl(INADDR_ANY);
562
563                 } else if ( !inet_aton( host, &in ) ) {
564                         he = gethostbyname( host );
565                         if( he == NULL ) {
566 #ifdef NEW_LOGGING
567                                 LDAP_LOG( CONNECTION, INFO, 
568                                         "slap_get_listener_addresses: invalid host %s\n", host, 0, 0 );
569 #else
570                                 Debug( LDAP_DEBUG_ANY,
571                                        "daemon: invalid host %s", host, 0, 0);
572 #endif
573                                 return -1;
574                         }
575                         for (n = 0; he->h_addr_list[n]; n++) ;
576                 }
577
578                 *sal = ch_malloc((n+1) * sizeof(void *));
579                 if (*sal == NULL) {
580                         return -1;
581                 }
582
583                 sap = *sal;
584                 for ( i = 0; i<n; i++ ) {
585                         sap[i] = ch_malloc(sizeof(struct sockaddr_in));
586                         if (*sap == NULL) {
587                                 goto errexit;
588                         }
589                         (void)memset( (void *)sap[i], '\0', sizeof(struct sockaddr_in) );
590                         sap[i]->sa_family = AF_INET;
591                         ((struct sockaddr_in *)sap[i])->sin_port = htons(port);
592                         if (he) {
593                                 AC_MEMCPY( &((struct sockaddr_in *)sap[i])->sin_addr, he->h_addr_list[i], sizeof(struct in_addr) );
594                         } else {
595                                 AC_MEMCPY( &((struct sockaddr_in *)sap[i])->sin_addr, &in, sizeof(struct in_addr) );
596                         }
597                 }
598                 sap[i] = NULL;
599 #endif
600         }
601
602         return 0;
603
604 errexit:
605         slap_free_listener_addresses(*sal);
606         return -1;
607 }
608
609 static int slap_open_listener(
610         const char* url,
611         int *listeners,
612         int *cur
613         )
614 {
615         int     num, tmp, rc;
616         Listener l;
617         Listener *li;
618         LDAPURLDesc *lud;
619         unsigned short port;
620         int err, addrlen = 0;
621         struct sockaddr **sal, **psal;
622         int socktype = SOCK_STREAM;     /* default to COTS */
623
624 #if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD)
625         /*
626          * use safe defaults
627          */
628         int     crit = 1;
629 #endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */
630
631         rc = ldap_url_parse( url, &lud );
632
633         if( rc != LDAP_URL_SUCCESS ) {
634 #ifdef NEW_LOGGING
635                 LDAP_LOG( CONNECTION, ERR, 
636                         "slap_open_listener: listen URL \"%s\" parse error %d\n",
637                         url, rc , 0 );
638 #else
639                 Debug( LDAP_DEBUG_ANY,
640                         "daemon: listen URL \"%s\" parse error=%d\n",
641                         url, rc, 0 );
642 #endif
643                 return rc;
644         }
645
646         l.sl_url.bv_val = NULL;
647         l.sl_is_mute = 0;
648
649 #ifndef HAVE_TLS
650         if( ldap_pvt_url_scheme2tls( lud->lud_scheme ) ) {
651 #ifdef NEW_LOGGING
652                 LDAP_LOG( CONNECTION, INFO, 
653                            "slap_open_listener: TLS is not supported (%s)\n", url, 0, 0 );
654 #else
655                 Debug( LDAP_DEBUG_ANY,
656                         "daemon: TLS not supported (%s)\n",
657                         url, 0, 0 );
658 #endif
659                 ldap_free_urldesc( lud );
660                 return -1;
661         }
662
663         if(! lud->lud_port ) {
664                 lud->lud_port = LDAP_PORT;
665         }
666
667 #else
668         l.sl_is_tls = ldap_pvt_url_scheme2tls( lud->lud_scheme );
669
670         if(! lud->lud_port ) {
671                 lud->lud_port = l.sl_is_tls ? LDAPS_PORT : LDAP_PORT;
672         }
673 #endif
674
675         port = (unsigned short) lud->lud_port;
676
677         tmp = ldap_pvt_url_scheme2proto(lud->lud_scheme);
678         if ( tmp == LDAP_PROTO_IPC ) {
679 #ifdef LDAP_PF_LOCAL
680                 if ( lud->lud_host == NULL || lud->lud_host[0] == '\0' ) {
681                         err = slap_get_listener_addresses(LDAPI_SOCK, 0, &sal);
682                 } else {
683                         err = slap_get_listener_addresses(lud->lud_host, 0, &sal);
684                 }
685 #else
686
687 #ifdef NEW_LOGGING
688                 LDAP_LOG( CONNECTION, INFO, 
689                         "slap_open_listener: URL scheme is not supported: %s\n", url, 0, 0 );
690 #else
691                 Debug( LDAP_DEBUG_ANY, "daemon: URL scheme not supported: %s",
692                         url, 0, 0);
693 #endif
694                 ldap_free_urldesc( lud );
695                 return -1;
696 #endif
697         } else {
698                 if( lud->lud_host == NULL || lud->lud_host[0] == '\0'
699                         || strcmp(lud->lud_host, "*") == 0 )
700                 {
701                         err = slap_get_listener_addresses(NULL, port, &sal);
702                 } else {
703                         err = slap_get_listener_addresses(lud->lud_host, port, &sal);
704                 }
705         }
706 #ifdef LDAP_CONNECTIONLESS
707         l.sl_is_udp = ( tmp == LDAP_PROTO_UDP );
708 #endif
709
710 #if defined(LDAP_PF_LOCAL) || defined(SLAP_X_LISTENER_MOD)
711         if ( lud->lud_exts ) {
712                 err = get_url_perms( lud->lud_exts, &l.sl_perms, &crit );
713         } else {
714                 l.sl_perms = S_IRWXU | S_IRWXO;
715         }
716 #endif /* LDAP_PF_LOCAL || SLAP_X_LISTENER_MOD */
717
718         ldap_free_urldesc( lud );
719         if ( err ) {
720                 return -1;
721         }
722
723         /* If we got more than one address returned, we need to make space
724          * for it in the slap_listeners array.
725          */
726         for ( num=0; sal[num]; num++ );
727         if ( num > 1 ) {
728                 *listeners += num-1;
729                 slap_listeners = ch_realloc( slap_listeners, (*listeners + 1) * sizeof(Listener *) );
730         }
731
732         psal = sal;
733         while ( *sal != NULL ) {
734                 switch( (*sal)->sa_family ) {
735                 case AF_INET:
736 #ifdef LDAP_PF_INET6
737                 case AF_INET6:
738 #endif
739 #ifdef LDAP_PF_LOCAL
740                 case AF_LOCAL:
741 #endif
742                         break;
743                 default:
744                         sal++;
745                         continue;
746                 }
747 #ifdef LDAP_CONNECTIONLESS
748                 if( l.sl_is_udp ) socktype = SOCK_DGRAM;
749 #endif
750                 l.sl_sd = socket( (*sal)->sa_family, socktype, 0);
751                 if ( l.sl_sd == AC_SOCKET_INVALID ) {
752                         int err = sock_errno();
753 #ifdef NEW_LOGGING
754                         LDAP_LOG( CONNECTION, ERR, 
755                                 "slap_open_listener: socket() failed errno=%d (%s)\n",
756                                 err, sock_errstr(err), 0 );
757 #else
758                         Debug( LDAP_DEBUG_ANY,
759                                 "daemon: socket() failed errno=%d (%s)\n", err,
760                                 sock_errstr(err), 0 );
761 #endif
762                         sal++;
763                         continue;
764                 }
765 #ifndef HAVE_WINSOCK
766                 if ( l.sl_sd >= dtblsize ) {
767 #ifdef NEW_LOGGING
768                         LDAP_LOG( CONNECTION, ERR, 
769                                 "slap_open_listener: listener descriptor %ld is too "
770                                 "great %ld\n", (long)l.sl_sd, (long)dtblsize, 0 );
771 #else
772                         Debug( LDAP_DEBUG_ANY,
773                                "daemon: listener descriptor %ld is too great %ld\n",
774                                (long) l.sl_sd, (long) dtblsize, 0 );
775 #endif
776                         tcp_close( l.sl_sd );
777                         sal++;
778                         continue;
779                 }
780 #endif
781 #ifdef LDAP_PF_LOCAL
782                 if ( (*sal)->sa_family == AF_LOCAL ) {
783                         unlink ( ((struct sockaddr_un *)*sal)->sun_path );
784                 } else
785 #endif
786                 {
787 #ifdef SO_REUSEADDR
788                         /* enable address reuse */
789                         tmp = 1;
790                         rc = setsockopt( l.sl_sd, SOL_SOCKET, SO_REUSEADDR,
791                                          (char *) &tmp, sizeof(tmp) );
792                         if ( rc == AC_SOCKET_ERROR ) {
793                                 int err = sock_errno();
794 #ifdef NEW_LOGGING
795                                 LDAP_LOG( CONNECTION, INFO, 
796                                         "slap_open_listener: setsockopt( %ld, SO_REUSEADDR ) "
797                                         "failed errno %d (%s)\n", (long)l.sl_sd, err, 
798                                         sock_errstr(err) );
799 #else
800                                 Debug( LDAP_DEBUG_ANY,
801                                        "slapd(%ld): setsockopt(SO_REUSEADDR) failed errno=%d (%s)\n",
802                                        (long) l.sl_sd, err, sock_errstr(err) );
803 #endif
804                         }
805 #endif
806                 }
807
808                 switch( (*sal)->sa_family ) {
809                 case AF_INET:
810                         addrlen = sizeof(struct sockaddr_in);
811                         break;
812 #ifdef LDAP_PF_INET6
813                 case AF_INET6:
814 #ifdef IPV6_V6ONLY
815                         /* Try to use IPv6 sockets for IPv6 only */
816                         tmp = 1;
817                         rc = setsockopt( l.sl_sd, IPPROTO_IPV6, IPV6_V6ONLY,
818                                          (char *) &tmp, sizeof(tmp) );
819                         if ( rc == AC_SOCKET_ERROR ) {
820                                 int err = sock_errno();
821 #ifdef NEW_LOGGING
822                                 LDAP_LOG( CONNECTION, INFO,
823                                            "slap_open_listener: setsockopt( %ld, IPV6_V6ONLY ) failed errno %d (%s)\n",
824                                            (long)l.sl_sd, err, sock_errstr(err) );
825 #else
826                                 Debug( LDAP_DEBUG_ANY,
827                                        "slapd(%ld): setsockopt(IPV6_V6ONLY) failed errno=%d (%s)\n",
828                                        (long) l.sl_sd, err, sock_errstr(err) );
829 #endif
830                         }
831 #endif
832                         addrlen = sizeof(struct sockaddr_in6);
833                         break;
834 #endif
835 #ifdef LDAP_PF_LOCAL
836                 case AF_LOCAL:
837                         addrlen = sizeof(struct sockaddr_un);
838                         break;
839 #endif
840                 }
841
842                 if (bind(l.sl_sd, *sal, addrlen)) {
843                         err = sock_errno();
844 #ifdef NEW_LOGGING
845                 LDAP_LOG( CONNECTION, INFO, 
846                         "slap_open_listener: bind(%ld) failed errno=%d (%s)\n",
847                         (long)l.sl_sd, err, sock_errstr(err) );
848 #else
849                 Debug( LDAP_DEBUG_ANY, "daemon: bind(%ld) failed errno=%d (%s)\n",
850                        (long) l.sl_sd, err, sock_errstr(err) );
851 #endif
852                         tcp_close( l.sl_sd );
853                         sal++;
854                         continue;
855                 }
856
857         switch ( (*sal)->sa_family ) {
858 #ifdef LDAP_PF_LOCAL
859         case AF_LOCAL: {
860                 char *addr = ((struct sockaddr_un *)*sal)->sun_path;
861 #if 0 /* don't muck with socket perms */
862                 if ( chmod( addr, l.sl_perms ) < 0 && crit ) {
863                         int err = sock_errno();
864 #ifdef NEW_LOGGING
865                         LDAP_LOG( CONNECTION, INFO, 
866                                 "slap_open_listener: fchmod(%ld) failed errno=%d (%s)\n",
867                                 (long)l.sl_sd, err, sock_errstr(err) );
868 #else
869                         Debug( LDAP_DEBUG_ANY, "daemon: fchmod(%ld) failed errno=%d (%s)",
870                                (long) l.sl_sd, err, sock_errstr(err) );
871 #endif
872                         tcp_close( l.sl_sd );
873                         slap_free_listener_addresses(psal);
874                         return -1;
875                 }
876 #endif
877                 l.sl_name.bv_len = strlen(addr) + sizeof("PATH=") - 1;
878                 l.sl_name.bv_val = ber_memalloc( l.sl_name.bv_len + 1 );
879                 snprintf( l.sl_name.bv_val, l.sl_name.bv_len + 1, 
880                                 "PATH=%s", addr );
881         } break;
882 #endif /* LDAP_PF_LOCAL */
883
884         case AF_INET: {
885                 char *s;
886 #if defined( HAVE_GETADDRINFO ) && defined( HAVE_INET_NTOP )
887                 char addr[INET_ADDRSTRLEN];
888                 inet_ntop( AF_INET, &((struct sockaddr_in *)*sal)->sin_addr,
889                            addr, sizeof(addr) );
890                 s = addr;
891 #else
892                 s = inet_ntoa( ((struct sockaddr_in *) *sal)->sin_addr );
893 #endif
894                 port = ntohs( ((struct sockaddr_in *)*sal) ->sin_port );
895                 l.sl_name.bv_val = ber_memalloc( sizeof("IP=255.255.255.255:65535") );
896                 snprintf( l.sl_name.bv_val, sizeof("IP=255.255.255.255:65535"),
897                         "IP=%s:%d",
898                          s != NULL ? s : SLAP_STRING_UNKNOWN, port );
899                 l.sl_name.bv_len = strlen( l.sl_name.bv_val );
900         } break;
901
902 #ifdef LDAP_PF_INET6
903         case AF_INET6: {
904                 char addr[INET6_ADDRSTRLEN];
905                 inet_ntop( AF_INET6, &((struct sockaddr_in6 *)*sal)->sin6_addr,
906                            addr, sizeof addr);
907                 port = ntohs( ((struct sockaddr_in6 *)*sal)->sin6_port );
908                 l.sl_name.bv_len = strlen(addr) + sizeof("IP= 65535");
909                 l.sl_name.bv_val = ber_memalloc( l.sl_name.bv_len );
910                 snprintf( l.sl_name.bv_val, l.sl_name.bv_len, "IP=%s %d", 
911                                 addr, port );
912                 l.sl_name.bv_len = strlen( l.sl_name.bv_val );
913         } break;
914 #endif /* LDAP_PF_INET6 */
915
916         default:
917 #ifdef NEW_LOGGING
918                 LDAP_LOG( CONNECTION, INFO, 
919                         "slap_open_listener: unsupported address family (%d)\n",
920                         (int)(*sal)->sa_family, 0, 0 );
921 #else
922                 Debug( LDAP_DEBUG_ANY, "daemon: unsupported address family (%d)\n",
923                         (int) (*sal)->sa_family, 0, 0 );
924 #endif
925                 break;
926         }
927
928         AC_MEMCPY(&l.sl_sa, *sal, addrlen);
929         ber_str2bv( url, 0, 1, &l.sl_url);
930         li = ch_malloc( sizeof( Listener ) );
931         *li = l;
932         slap_listeners[*cur] = li;
933         (*cur)++;
934         sal++;
935
936         } /* while ( *sal != NULL ) */
937
938         slap_free_listener_addresses(psal);
939
940         if ( l.sl_url.bv_val == NULL )
941         {
942 #ifdef NEW_LOGGING
943                 LDAP_LOG( CONNECTION, RESULTS, 
944                         "slap_open_listener: failed on %s\n", url, 0, 0 );
945 #else
946                 Debug( LDAP_DEBUG_TRACE,
947                         "slap_open_listener: failed on %s\n", url, 0, 0 );
948 #endif
949                 return -1;
950         }
951
952 #ifdef NEW_LOGGING
953         LDAP_LOG( CONNECTION, RESULTS, 
954                 "slap_open_listener: daemon initialized %s\n",
955                 l.sl_url.bv_val, 0, 0 );
956 #else
957         Debug( LDAP_DEBUG_TRACE, "daemon: initialized %s\n",
958                 l.sl_url.bv_val, 0, 0 );
959 #endif
960         return 0;
961 }
962
963 static int sockinit(void);
964 static int sockdestroy(void);
965
966 int slapd_daemon_init( const char *urls )
967 {
968         int i, j, n, rc;
969         char **u;
970
971 #ifdef NEW_LOGGING
972         LDAP_LOG( CONNECTION, ARGS, 
973                 "slapd_daemon_init: %s\n", urls ? urls : "<null>", 0, 0 );
974 #else
975         Debug( LDAP_DEBUG_ARGS, "daemon_init: %s\n",
976                 urls ? urls : "<null>", 0, 0 );
977 #endif
978         if( (rc = sockinit()) != 0 ) {
979                 return rc;
980         }
981
982 #ifdef HAVE_SYSCONF
983         dtblsize = sysconf( _SC_OPEN_MAX );
984 #elif HAVE_GETDTABLESIZE
985         dtblsize = getdtablesize();
986 #else
987         dtblsize = FD_SETSIZE;
988 #endif
989
990 #ifdef FD_SETSIZE
991         if(dtblsize > FD_SETSIZE) {
992                 dtblsize = FD_SETSIZE;
993         }
994 #endif  /* !FD_SETSIZE */
995
996         /* open a pipe (or something equivalent connected to itself).
997          * we write a byte on this fd whenever we catch a signal. The main
998          * loop will be select'ing on this socket, and will wake up when
999          * this byte arrives.
1000          */
1001         if( (rc = lutil_pair( wake_sds )) < 0 ) {
1002 #ifdef NEW_LOGGING
1003                 LDAP_LOG( CONNECTION, ERR, 
1004                         "slap_daemon_init: lutil_pair() failed rc=%d\n", rc, 0, 0);
1005 #else
1006                 Debug( LDAP_DEBUG_ANY,
1007                         "daemon: lutil_pair() failed rc=%d\n", rc, 0, 0 );
1008 #endif
1009                 return rc;
1010         }
1011
1012         FD_ZERO( &slap_daemon.sd_readers );
1013         FD_ZERO( &slap_daemon.sd_writers );
1014
1015         if( urls == NULL ) {
1016                 urls = "ldap:///";
1017         }
1018
1019         u = ldap_str2charray( urls, " " );
1020
1021         if( u == NULL || u[0] == NULL ) {
1022 #ifdef NEW_LOGGING
1023                 LDAP_LOG( CONNECTION, ERR, 
1024                         "slap_daemon_init: no urls (%s) provided.\n", urls, 0, 0 );
1025 #else
1026                 Debug( LDAP_DEBUG_ANY, "daemon_init: no urls (%s) provided.\n",
1027                         urls, 0, 0 );
1028 #endif
1029                 return -1;
1030         }
1031
1032         for( i=0; u[i] != NULL; i++ ) {
1033 #ifdef NEW_LOGGING
1034                 LDAP_LOG( CONNECTION, DETAIL1, 
1035                         "slap_daemon_init: listen on %s\n", u[i], 0, 0 );
1036 #else
1037                 Debug( LDAP_DEBUG_TRACE, "daemon_init: listen on %s\n",
1038                         u[i], 0, 0 );
1039 #endif
1040         }
1041
1042         if( i == 0 ) {
1043 #ifdef NEW_LOGGING
1044                 LDAP_LOG( CONNECTION, INFO, 
1045                          "slap_daemon_init: no listeners to open (%s)\n", urls, 0, 0 );
1046 #else
1047                 Debug( LDAP_DEBUG_ANY, "daemon_init: no listeners to open (%s)\n",
1048                         urls, 0, 0 );
1049 #endif
1050                 ldap_charray_free( u );
1051                 return -1;
1052         }
1053
1054 #ifdef NEW_LOGGING
1055         LDAP_LOG( CONNECTION, INFO, 
1056                 "slap_daemon_init: %d listeners to open...\n", i, 0, 0 );
1057 #else
1058         Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners to open...\n",
1059                 i, 0, 0 );
1060 #endif
1061         slap_listeners = ch_malloc( (i+1)*sizeof(Listener *) );
1062
1063         for(n = 0, j = 0; u[n]; n++ ) {
1064                 if ( slap_open_listener( u[n], &i, &j ) ) {
1065                         ldap_charray_free( u );
1066                         return -1;
1067                 }
1068         }
1069         slap_listeners[j] = NULL;
1070
1071 #ifdef NEW_LOGGING
1072         LDAP_LOG( CONNECTION, DETAIL1, 
1073                 "slap_daemon_init: %d listeners opened\n", i, 0, 0 );
1074 #else
1075         Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners opened\n",
1076                 i, 0, 0 );
1077 #endif
1078
1079 #ifdef HAVE_SLP
1080         slapd_slp_init( urls );
1081         slapd_slp_reg();
1082 #endif
1083
1084         ldap_charray_free( u );
1085         ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex );
1086         return !i;
1087 }
1088
1089
1090 int
1091 slapd_daemon_destroy(void)
1092 {
1093         connections_destroy();
1094         tcp_close( wake_sds[1] );
1095         tcp_close( wake_sds[0] );
1096         sockdestroy();
1097
1098 #ifdef HAVE_SLP
1099         slapd_slp_dereg();
1100         slapd_slp_deinit();
1101 #endif
1102
1103         return 0;
1104 }
1105
1106
1107 static void
1108 close_listeners(
1109         int remove
1110 )
1111 {
1112         int l;
1113
1114         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1115                 if ( slap_listeners[l]->sl_sd != AC_SOCKET_INVALID ) {
1116                         if ( remove )
1117                                 slapd_remove( slap_listeners[l]->sl_sd, 0 );
1118 #ifdef LDAP_PF_LOCAL
1119                         if ( slap_listeners[l]->sl_sa.sa_addr.sa_family == AF_LOCAL ) {
1120                                 unlink( slap_listeners[l]->sl_sa.sa_un_addr.sun_path );
1121                         }
1122 #endif /* LDAP_PF_LOCAL */
1123                         slapd_close( slap_listeners[l]->sl_sd );
1124                 }
1125                 if ( slap_listeners[l]->sl_url.bv_val )
1126                         ber_memfree( slap_listeners[l]->sl_url.bv_val );
1127                 if ( slap_listeners[l]->sl_name.bv_val )
1128                         ber_memfree( slap_listeners[l]->sl_name.bv_val );
1129                 free ( slap_listeners[l] );
1130                 slap_listeners[l] = NULL;
1131         }
1132 }
1133
1134
1135 static void *
1136 slapd_daemon_task(
1137         void *ptr
1138 )
1139 {
1140         int l;
1141         time_t  last_idle_check = 0;
1142         struct timeval idle;
1143
1144 #define SLAPD_IDLE_CHECK_LIMIT 4
1145
1146         if ( global_idletimeout > 0 ) {
1147                 last_idle_check = slap_get_time();
1148                 /* Set the select timeout.
1149                  * Don't just truncate, preserve the fractions of
1150                  * seconds to prevent sleeping for zero time.
1151                  */
1152                 idle.tv_sec = global_idletimeout/SLAPD_IDLE_CHECK_LIMIT;
1153                 idle.tv_usec = global_idletimeout - idle.tv_sec * SLAPD_IDLE_CHECK_LIMIT;
1154                 idle.tv_usec *= 1000000 / SLAPD_IDLE_CHECK_LIMIT;
1155         } else {
1156                 idle.tv_sec = 0;
1157                 idle.tv_usec = 0;
1158         }
1159
1160         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1161                 if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
1162                         continue;
1163 #ifdef LDAP_CONNECTIONLESS
1164                 /* Since this is connectionless, the data port is the
1165                  * listening port. The listen() and accept() calls
1166                  * are unnecessary.
1167                  */
1168                 if ( slap_listeners[l]->sl_is_udp ) {
1169                         slapd_add( slap_listeners[l]->sl_sd );
1170                         continue;
1171                 }
1172 #endif
1173
1174                 if ( listen( slap_listeners[l]->sl_sd, SLAPD_LISTEN ) == -1 ) {
1175                         int err = sock_errno();
1176
1177 #ifdef LDAP_PF_INET6
1178                         /* If error is EADDRINUSE, we are trying to listen to INADDR_ANY and
1179                          * we are already listening to in6addr_any, then we want to ignore
1180                          * this and continue.
1181                          */
1182                         if ( err == EADDRINUSE ) {
1183                                 int i;
1184                                 struct sockaddr_in sa = slap_listeners[l]->sl_sa.sa_in_addr;
1185                                 struct sockaddr_in6 sa6;
1186                                 
1187                                 if ( sa.sin_family == AF_INET &&
1188                                      sa.sin_addr.s_addr == htonl(INADDR_ANY) ) {
1189                                         for ( i = 0 ; i < l; i++ ) {
1190                                                 sa6 = slap_listeners[i]->sl_sa.sa_in6_addr;
1191                                                 if ( sa6.sin6_family == AF_INET6 &&
1192                                                      !memcmp( &sa6.sin6_addr, &in6addr_any, sizeof(struct in6_addr) ) )
1193                                                         break;
1194                                         }
1195
1196                                         if ( i < l ) {
1197                                                 /* We are already listening to in6addr_any */
1198 #ifdef NEW_LOGGING
1199                                                 LDAP_LOG(CONNECTION, WARNING,
1200                                                            "slapd_daemon_task: Attempt to listen to 0.0.0.0 failed, already listening on ::, assuming IPv4 included\n", 0, 0, 0 );
1201 #else
1202                                                 Debug( LDAP_DEBUG_CONNS,
1203                                                        "daemon: Attempt to listen to 0.0.0.0 failed, already listening on ::, assuming IPv4 included\n",
1204                                                        0, 0, 0 );
1205 #endif
1206                                                 slapd_close( slap_listeners[l]->sl_sd );
1207                                                 slap_listeners[l]->sl_sd = AC_SOCKET_INVALID;
1208                                                 continue;
1209                                         }
1210                                 }
1211                         }
1212 #endif                          
1213 #ifdef NEW_LOGGING
1214                         LDAP_LOG( CONNECTION, ERR, 
1215                                 "slapd_daemon_task: listen( %s, 5 ) failed errno=%d (%s)\n",
1216                                 slap_listeners[l]->sl_url.bv_val, err, sock_errstr(err) );
1217 #else
1218                         Debug( LDAP_DEBUG_ANY,
1219                                 "daemon: listen(%s, 5) failed errno=%d (%s)\n",
1220                                         slap_listeners[l]->sl_url.bv_val, err,
1221                                         sock_errstr(err) );
1222 #endif
1223                         return( (void*)-1 );
1224                 }
1225
1226                 slapd_add( slap_listeners[l]->sl_sd );
1227         }
1228
1229 #ifdef HAVE_NT_SERVICE_MANAGER
1230         if ( started_event != NULL ) {
1231                 ldap_pvt_thread_cond_signal( &started_event );
1232         }
1233 #endif
1234         /* initialization complete. Here comes the loop. */
1235
1236         while ( !slapd_shutdown ) {
1237                 ber_socket_t i;
1238                 int ns;
1239                 int at;
1240                 ber_socket_t nfds;
1241 #define SLAPD_EBADF_LIMIT 16
1242                 int ebadf = 0;
1243
1244                 time_t  now;
1245
1246                 fd_set                  readfds;
1247                 fd_set                  writefds;
1248                 Sockaddr                from;
1249
1250                 struct timeval          tv;
1251                 struct timeval          *tvp;
1252
1253 #ifdef LDAP_SYNCREPL
1254                 struct timeval          *cat;
1255                 time_t                          tdelta = 1;
1256                 struct re_s*            rtask;
1257 #endif
1258                 now = slap_get_time();
1259
1260                 if( ( global_idletimeout > 0 ) &&
1261                         difftime( last_idle_check +
1262                         global_idletimeout/SLAPD_IDLE_CHECK_LIMIT, now ) < 0 ) {
1263                         connections_timeout_idle( now );
1264                         last_idle_check = now;
1265                 }
1266                 tv = idle;
1267
1268 #ifdef SIGHUP
1269                 if( slapd_gentle_shutdown ) {
1270                         ber_socket_t active;
1271
1272                         if( slapd_gentle_shutdown == 1 ) {
1273                                 Debug( LDAP_DEBUG_ANY, "slapd gentle shutdown\n", 0, 0, 0 );
1274                                 close_listeners( 1 );
1275                                 global_restrictops |= SLAP_RESTRICT_OP_WRITES;
1276                                 slapd_gentle_shutdown = 2;
1277                         }
1278
1279                         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
1280                         active = slap_daemon.sd_nactives;
1281                         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
1282                         if( active == 0 ) {
1283                                 slapd_shutdown = 2;
1284                                 break;
1285                         }
1286                 }
1287 #endif
1288
1289                 FD_ZERO( &writefds );
1290                 FD_ZERO( &readfds );
1291
1292                 at = 0;
1293
1294                 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
1295
1296 #ifdef FD_SET_MANUAL_COPY
1297                 for( s = 0; s < nfds; s++ ) {
1298                         if(FD_ISSET( &slap_sd_readers, s )) {
1299                                 FD_SET( s, &readfds );
1300                         }
1301                         if(FD_ISSET( &slap_sd_writers, s )) {
1302                                 FD_SET( s, &writefds );
1303                         }
1304                 }
1305 #else
1306                 AC_MEMCPY( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) );
1307                 AC_MEMCPY( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) );
1308 #endif
1309                 assert(!FD_ISSET(wake_sds[0], &readfds));
1310                 FD_SET( wake_sds[0], &readfds );
1311
1312                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1313                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
1314                                 continue;
1315                         if ( slap_listeners[l]->sl_is_mute )
1316                                 FD_CLR( slap_listeners[l]->sl_sd, &readfds );
1317                         else
1318                         if (!FD_ISSET(slap_listeners[l]->sl_sd, &readfds))
1319                             FD_SET( slap_listeners[l]->sl_sd, &readfds );
1320                 }
1321
1322 #ifndef HAVE_WINSOCK
1323                 nfds = slap_daemon.sd_nfds;
1324 #else
1325                 nfds = dtblsize;
1326 #endif
1327                 if ( global_idletimeout && slap_daemon.sd_nactives )
1328                         at = 1;
1329
1330                 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
1331
1332                 if ( !at ) {
1333                         at = ldap_pvt_thread_pool_backload(&connection_pool) -
1334                                  ldap_pvt_runqueue_persistent_backload( &syncrepl_rq );
1335                 }
1336
1337                 if ( at 
1338 #if defined(HAVE_YIELDING_SELECT) || defined(NO_THREADS)
1339                         &&  ( tv.tv_sec || tv.tv_usec )
1340 #endif
1341                         )
1342                         tvp = &tv;
1343                 else
1344                         tvp = NULL;
1345
1346 #ifdef LDAP_SYNCREPL
1347                 ldap_pvt_thread_mutex_lock( &syncrepl_rq.rq_mutex );
1348                 rtask = ldap_pvt_runqueue_next_sched( &syncrepl_rq, &cat );
1349                 while ( cat && cat->tv_sec && cat->tv_sec <= now ) {
1350                         if ( ldap_pvt_runqueue_isrunning( &syncrepl_rq, rtask )) {
1351                                 ldap_pvt_runqueue_resched( &syncrepl_rq, rtask );
1352                         } else {
1353                                 ldap_pvt_runqueue_runtask( &syncrepl_rq, rtask );
1354                                 ldap_pvt_runqueue_resched( &syncrepl_rq, rtask );
1355                                 ldap_pvt_thread_mutex_unlock( &syncrepl_rq.rq_mutex );
1356                                 ldap_pvt_thread_pool_submit( &connection_pool,
1357                                                                                         rtask->routine, (void *) rtask );
1358                         }
1359                         rtask = ldap_pvt_runqueue_next_sched( &syncrepl_rq, &cat );
1360                 }
1361                 rtask = ldap_pvt_runqueue_next_sched( &syncrepl_rq, &cat );
1362                 ldap_pvt_thread_mutex_unlock( &syncrepl_rq.rq_mutex );
1363
1364                 if ( cat != NULL ) {
1365                         time_t diff = difftime( cat->tv_sec, now );
1366                         if ( diff == 0 )
1367                                 diff = tdelta;
1368                         if ( tvp == NULL || diff < tv.tv_sec ) {
1369                                 tv.tv_sec = diff;
1370                                 tv.tv_usec = 0;
1371                                 tvp = &tv;
1372                         }
1373                 }
1374 #endif
1375
1376                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1377                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID ||
1378                             slap_listeners[l]->sl_is_mute )
1379                                 continue;
1380
1381 #ifdef NEW_LOGGING
1382                         LDAP_LOG( CONNECTION, DETAIL1, 
1383                                 "slapd_daemon_task: select: listen=%d "
1384                                 "active_threads=%d tvp=%s\n",
1385                                 slap_listeners[l]->sl_sd, at, tvp == NULL ? "NULL" : "zero" );
1386 #else
1387                         Debug( LDAP_DEBUG_CONNS,
1388                                 "daemon: select: listen=%d active_threads=%d tvp=%s\n",
1389                                         slap_listeners[l]->sl_sd, at,
1390                                         tvp == NULL ? "NULL" : "zero" );
1391 #endif
1392                 }
1393
1394                 switch(ns = select( nfds, &readfds,
1395 #ifdef HAVE_WINSOCK
1396                         /* don't pass empty fd_set */
1397                         ( writefds.fd_count > 0 ? &writefds : NULL ),
1398 #else
1399                         &writefds,
1400 #endif
1401                         NULL, tvp ))
1402                 {
1403                 case -1: {      /* failure - try again */
1404                                 int err = sock_errno();
1405
1406                                 if( err == EBADF
1407 #ifdef WSAENOTSOCK
1408                                         /* you'd think this would be EBADF */
1409                                         || err == WSAENOTSOCK
1410 #endif
1411                                 ) {
1412                                         if (++ebadf < SLAPD_EBADF_LIMIT)
1413                                                 continue;
1414                                 }
1415
1416                                 if( err != EINTR ) {
1417 #ifdef NEW_LOGGING
1418                                         LDAP_LOG( CONNECTION, INFO, 
1419                                                 "slapd_daemon_task: select failed (%d): %s\n",
1420                                                 err, sock_errstr(err), 0 );
1421 #else
1422                                         Debug( LDAP_DEBUG_CONNS,
1423                                                 "daemon: select failed (%d): %s\n",
1424                                                 err, sock_errstr(err), 0 );
1425 #endif
1426                                         slapd_shutdown = 2;
1427                                 }
1428                         }
1429                         continue;
1430
1431                 case 0:         /* timeout - let threads run */
1432                         ebadf = 0;
1433 #ifdef NEW_LOGGING
1434                         LDAP_LOG( CONNECTION, DETAIL2,
1435                                    "slapd_daemon_task: select timeout - yielding\n", 0, 0, 0 );
1436 #else
1437                         Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n",
1438                             0, 0, 0 );
1439 #endif
1440
1441                         ldap_pvt_thread_yield();
1442                         continue;
1443
1444                 default:        /* something happened - deal with it */
1445                         if( slapd_shutdown ) continue;
1446
1447                         ebadf = 0;
1448 #ifdef NEW_LOGGING
1449                         LDAP_LOG( CONNECTION, DETAIL2, 
1450                                    "slapd_daemon_task: activity on %d descriptors\n", ns, 0, 0 );
1451 #else
1452                         Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n",
1453                                 ns, 0, 0 );
1454 #endif
1455                         /* FALL THRU */
1456                 }
1457
1458                 if( FD_ISSET( wake_sds[0], &readfds ) ) {
1459                         char c[BUFSIZ];
1460                         tcp_read( wake_sds[0], c, sizeof(c) );
1461 #if defined(NO_THREADS) || defined(HAVE_GNU_PTH)
1462                         waking = 0;
1463 #endif
1464                         continue;
1465                 }
1466
1467                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1468                         ber_socket_t s;
1469                         socklen_t len = sizeof(from);
1470                         long id;
1471                         slap_ssf_t ssf = 0;
1472                         char *authid = NULL;
1473 #ifdef SLAPD_RLOOKUPS
1474                         char hbuf[NI_MAXHOST];
1475 #endif
1476
1477                         char    *dnsname = NULL;
1478                         char    *peeraddr = NULL;
1479 #ifdef LDAP_PF_LOCAL
1480                         char    peername[MAXPATHLEN + sizeof("PATH=")];
1481 #elif defined(LDAP_PF_INET6)
1482                         char    peername[sizeof("IP=ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff 65535")];
1483 #else
1484                         char    peername[sizeof("IP=255.255.255.255:65336")];
1485 #endif /* LDAP_PF_LOCAL */
1486
1487                         peername[0] = '\0';
1488
1489                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
1490                                 continue;
1491
1492                         if ( !FD_ISSET( slap_listeners[l]->sl_sd, &readfds ) )
1493                                 continue;
1494                         
1495 #ifdef LDAP_CONNECTIONLESS
1496                         if ( slap_listeners[l]->sl_is_udp ) {
1497                                 /* The first time we receive a query, we set this
1498                                  * up as a "connection". It remains open for the life
1499                                  * of the slapd.
1500                                  */
1501                                 if ( slap_listeners[l]->sl_is_udp < 2 ) {
1502                                     id = connection_init(
1503                                         slap_listeners[l]->sl_sd,
1504                                         slap_listeners[l], "", "",
1505                                         2, ssf, authid );
1506                                     slap_listeners[l]->sl_is_udp++;
1507                                 }
1508                                 continue;
1509                         }
1510 #endif
1511
1512                         /* Don't need to look at this in the data loops */
1513                         FD_CLR( slap_listeners[l]->sl_sd, &readfds );
1514                         FD_CLR( slap_listeners[l]->sl_sd, &writefds );
1515
1516                         s = accept( slap_listeners[l]->sl_sd,
1517                                 (struct sockaddr *) &from, &len );
1518                         if ( s == AC_SOCKET_INVALID ) {
1519                                 int err = sock_errno();
1520
1521                                 if(
1522 #ifdef EMFILE
1523                                     err == EMFILE ||
1524 #endif
1525 #ifdef ENFILE
1526                                     err == ENFILE ||
1527 #endif
1528                                     0 )
1529                                 {
1530                                         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
1531                                         emfile++;
1532                                         /* Stop listening until an existing session closes */
1533                                         slap_listeners[l]->sl_is_mute = 1;
1534                                         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
1535                                 }
1536
1537 #ifdef NEW_LOGGING
1538                                 LDAP_LOG( CONNECTION, ERR, 
1539                                         "slapd_daemon_task: accept(%ld) failed errno=%d (%s)\n",
1540                                         (long)slap_listeners[l]->sl_sd, 
1541                                         err, sock_errstr(err) );
1542 #else
1543                                 Debug( LDAP_DEBUG_ANY,
1544                                         "daemon: accept(%ld) failed errno=%d (%s)\n",
1545                                         (long) slap_listeners[l]->sl_sd, err,
1546                                         sock_errstr(err) );
1547 #endif
1548                                 ldap_pvt_thread_yield();
1549                                 continue;
1550                         }
1551
1552 #ifndef HAVE_WINSOCK
1553                         /* make sure descriptor number isn't too great */
1554                         if ( s >= dtblsize ) {
1555 #ifdef NEW_LOGGING
1556                                 LDAP_LOG( CONNECTION, ERR, 
1557                                    "slapd_daemon_task: %ld beyond descriptor table size %ld\n",
1558                                    (long)s, (long)dtblsize, 0 );
1559 #else
1560                                 Debug( LDAP_DEBUG_ANY,
1561                                         "daemon: %ld beyond descriptor table size %ld\n",
1562                                         (long) s, (long) dtblsize, 0 );
1563 #endif
1564
1565                                 slapd_close(s);
1566                                 ldap_pvt_thread_yield();
1567                                 continue;
1568                         }
1569 #endif
1570
1571 #ifdef LDAP_DEBUG
1572                         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
1573
1574                         /* newly accepted stream should not be in any of the FD SETS */
1575                         assert( !FD_ISSET( s, &slap_daemon.sd_actives) );
1576                         assert( !FD_ISSET( s, &slap_daemon.sd_readers) );
1577                         assert( !FD_ISSET( s, &slap_daemon.sd_writers) );
1578
1579                         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
1580 #endif
1581
1582 #if defined( SO_KEEPALIVE ) || defined( TCP_NODELAY )
1583 #ifdef LDAP_PF_LOCAL
1584                         /* for IPv4 and IPv6 sockets only */
1585                         if ( from.sa_addr.sa_family != AF_LOCAL )
1586 #endif /* LDAP_PF_LOCAL */
1587                         {
1588                                 int rc;
1589                                 int tmp;
1590 #ifdef SO_KEEPALIVE
1591                                 /* enable keep alives */
1592                                 tmp = 1;
1593                                 rc = setsockopt( s, SOL_SOCKET, SO_KEEPALIVE,
1594                                         (char *) &tmp, sizeof(tmp) );
1595                                 if ( rc == AC_SOCKET_ERROR ) {
1596                                         int err = sock_errno();
1597 #ifdef NEW_LOGGING
1598                                         LDAP_LOG( CONNECTION, ERR, 
1599                                                 "slapd_daemon_task: setsockopt( %ld, SO_KEEPALIVE)"
1600                                            " failed errno=%d (%s)\n",
1601                                                 (long)s, err, sock_errstr(err) );
1602 #else
1603                                         Debug( LDAP_DEBUG_ANY,
1604                                                 "slapd(%ld): setsockopt(SO_KEEPALIVE) failed "
1605                                                 "errno=%d (%s)\n", (long) s, err, sock_errstr(err) );
1606 #endif
1607                                 }
1608 #endif
1609 #ifdef TCP_NODELAY
1610                                 /* enable no delay */
1611                                 tmp = 1;
1612                                 rc = setsockopt( s, IPPROTO_TCP, TCP_NODELAY,
1613                                         (char *)&tmp, sizeof(tmp) );
1614                                 if ( rc == AC_SOCKET_ERROR ) {
1615                                         int err = sock_errno();
1616 #ifdef NEW_LOGGING
1617                                         LDAP_LOG( CONNECTION, ERR, 
1618                                                 "slapd_daemon_task: setsockopt( %ld, "
1619                                                 "TCP_NODELAY) failed errno=%d (%s)\n",
1620                                                 (long)s, err, sock_errstr(err) );
1621 #else
1622                                         Debug( LDAP_DEBUG_ANY,
1623                                                 "slapd(%ld): setsockopt(TCP_NODELAY) failed "
1624                                                 "errno=%d (%s)\n", (long) s, err, sock_errstr(err) );
1625 #endif
1626                                 }
1627 #endif
1628                         }
1629 #endif
1630
1631 #ifdef NEW_LOGGING
1632                         LDAP_LOG( CONNECTION, DETAIL1, 
1633                                 "slapd_daemon_task: new connection on %ld\n", (long)s, 0, 0 );
1634 #else
1635                         Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %ld\n",
1636                                 (long) s, 0, 0 );
1637 #endif
1638                         switch ( from.sa_addr.sa_family ) {
1639 #  ifdef LDAP_PF_LOCAL
1640                         case AF_LOCAL:
1641                                 sprintf( peername, "PATH=%s", from.sa_un_addr.sun_path );
1642                                 ssf = LDAP_PVT_SASL_LOCAL_SSF;
1643                                 {
1644                                         uid_t uid;
1645                                         gid_t gid;
1646
1647                                         if( getpeereid( s, &uid, &gid ) == 0 ) {
1648                                                 authid = ch_malloc(
1649                                                         sizeof("uidnumber=4294967295+gidnumber=4294967295,"
1650                                                                 "cn=peercred,cn=external,cn=auth"));
1651                                                 sprintf(authid, "uidnumber=%d+gidnumber=%d,"
1652                                                         "cn=peercred,cn=external,cn=auth",
1653                                                         (int) uid, (int) gid);
1654                                         }
1655                                 }
1656                                 dnsname = "local";
1657                                 break;
1658 #endif /* LDAP_PF_LOCAL */
1659
1660 #  ifdef LDAP_PF_INET6
1661                         case AF_INET6:
1662                         if ( IN6_IS_ADDR_V4MAPPED(&from.sa_in6_addr.sin6_addr) ) {
1663                                 peeraddr = inet_ntoa( *((struct in_addr *)
1664                                                         &from.sa_in6_addr.sin6_addr.s6_addr[12]) );
1665                                 sprintf( peername, "IP=%s:%d",
1666                                          peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN,
1667                                          (unsigned) ntohs( from.sa_in6_addr.sin6_port ) );
1668                         } else {
1669                                 char addr[INET6_ADDRSTRLEN];
1670
1671                                 peeraddr = (char *) inet_ntop( AF_INET6,
1672                                                       &from.sa_in6_addr.sin6_addr,
1673                                                       addr, sizeof addr );
1674                                 sprintf( peername, "IP=%s %d",
1675                                          peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN,
1676                                          (unsigned) ntohs( from.sa_in6_addr.sin6_port ) );
1677                         }
1678                         break;
1679 #  endif /* LDAP_PF_INET6 */
1680
1681                         case AF_INET:
1682                         peeraddr = inet_ntoa( from.sa_in_addr.sin_addr );
1683                         sprintf( peername, "IP=%s:%d",
1684                                 peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN,
1685                                 (unsigned) ntohs( from.sa_in_addr.sin_port ) );
1686                                 break;
1687
1688                         default:
1689                                 slapd_close(s);
1690                                 continue;
1691                         }
1692
1693                         if ( ( from.sa_addr.sa_family == AF_INET )
1694 #ifdef LDAP_PF_INET6
1695                                 || ( from.sa_addr.sa_family == AF_INET6 )
1696 #endif
1697                         ) {
1698 #ifdef SLAPD_RLOOKUPS
1699                                 if ( use_reverse_lookup ) {
1700                                         char *herr;
1701                                         if (ldap_pvt_get_hname( (const struct sockaddr *)&from, len, hbuf,
1702                                                 sizeof(hbuf), &herr ) == 0) {
1703                                                 ldap_pvt_str2lower( hbuf );
1704                                                 dnsname = hbuf;
1705                                         }
1706                                 }
1707 #else
1708                                 dnsname = NULL;
1709 #endif /* SLAPD_RLOOKUPS */
1710
1711 #ifdef HAVE_TCPD
1712                                 if ( !hosts_ctl("slapd",
1713                                                 dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN,
1714                                                 peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN,
1715                                                 SLAP_STRING_UNKNOWN ))
1716                                 {
1717                                         /* DENY ACCESS */
1718                                         Statslog( LDAP_DEBUG_STATS,
1719                                                 "fd=%ld DENIED from %s (%s)\n",
1720                                                 (long) s,
1721                                                 dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN,
1722                                                 peeraddr != NULL ? peeraddr : SLAP_STRING_UNKNOWN,
1723                                                 0, 0 );
1724                                         slapd_close(s);
1725                                         continue;
1726                                 }
1727 #endif /* HAVE_TCPD */
1728                         }
1729
1730                         id = connection_init(s,
1731                                 slap_listeners[l],
1732                                 dnsname != NULL ? dnsname : SLAP_STRING_UNKNOWN,
1733                                 peername,
1734 #ifdef HAVE_TLS
1735                                 slap_listeners[l]->sl_is_tls,
1736 #else
1737                                 0,
1738 #endif
1739                                 ssf,
1740                                 authid );
1741
1742                         if( authid ) ch_free(authid);
1743
1744                         if( id < 0 ) {
1745 #ifdef NEW_LOGGING
1746                                 LDAP_LOG( CONNECTION, INFO, 
1747                                         "slapd_daemon_task: "
1748                                         "connection_init(%ld, %s, %s) "
1749                                         "failed.\n",
1750                                         (long)s, peername, 
1751                                         slap_listeners[l]->sl_name.bv_val );
1752 #else
1753                                 Debug( LDAP_DEBUG_ANY,
1754                                         "daemon: connection_init(%ld, %s, %s) "
1755                                         "failed.\n",
1756                                         (long) s,
1757                                         peername,
1758                                         slap_listeners[l]->sl_name.bv_val );
1759 #endif
1760                                 slapd_close(s);
1761                                 continue;
1762                         }
1763
1764                         Statslog( LDAP_DEBUG_STATS,
1765                                 "conn=%ld fd=%ld ACCEPT from %s (%s)\n",
1766                                 id, (long) s,
1767                                 peername,
1768                                 slap_listeners[l]->sl_name.bv_val,
1769                                 0 );
1770
1771                         slapd_add( s );
1772                         continue;
1773                 }
1774
1775 #ifdef LDAP_DEBUG
1776 #ifdef NEW_LOGGING
1777                 LDAP_LOG( CONNECTION, DETAIL2,
1778                            "slapd_daemon_task: activity on ", 0, 0, 0 );
1779 #else
1780                 Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 );
1781 #endif
1782 #ifdef HAVE_WINSOCK
1783                 for ( i = 0; i < readfds.fd_count; i++ ) {
1784 #ifdef NEW_LOGGING
1785                         LDAP_LOG( CONNECTION, DETAIL2, 
1786                                 " %d%s", readfds.fd_array[i], "r", 0, 0 );
1787 #else
1788                         Debug( LDAP_DEBUG_CONNS, " %d%s",
1789                                 readfds.fd_array[i], "r", 0 );
1790 #endif
1791                 }
1792                 for ( i = 0; i < writefds.fd_count; i++ ) {
1793 #ifdef NEW_LOGGING
1794                         LDAP_LOG( CONNECTION, DETAIL2, 
1795                                 " %d%s", writefds.fd_array[i], "w" , 0 );
1796 #else
1797                         Debug( LDAP_DEBUG_CONNS, " %d%s",
1798                                 writefds.fd_array[i], "w", 0 );
1799 #endif
1800                 }
1801
1802 #else
1803                 for ( i = 0; i < nfds; i++ ) {
1804                         int     r, w;
1805
1806                         r = FD_ISSET( i, &readfds );
1807                         w = FD_ISSET( i, &writefds );
1808                         if ( r || w ) {
1809 #ifdef NEW_LOGGING
1810                                 LDAP_LOG( CONNECTION, DETAIL2, 
1811                                         " %d%s%s", i, r ? "r" : "", w ? "w" : "" );
1812 #else
1813                                 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
1814                                     r ? "r" : "", w ? "w" : "" );
1815 #endif
1816                         }
1817                 }
1818 #endif
1819 #ifdef NEW_LOGGING
1820                 LDAP_LOG( CONNECTION, DETAIL2, "\n", 0, 0, 0 );
1821 #else
1822                 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
1823 #endif
1824
1825 #endif
1826
1827                 /* loop through the writers */
1828 #ifdef HAVE_WINSOCK
1829                 for ( i = 0; i < writefds.fd_count; i++ )
1830 #else
1831                 for ( i = 0; i < nfds; i++ )
1832 #endif
1833                 {
1834                         ber_socket_t wd;
1835 #ifdef HAVE_WINSOCK
1836                         wd = writefds.fd_array[i];
1837 #else
1838                         if( ! FD_ISSET( i, &writefds ) ) {
1839                                 continue;
1840                         }
1841                         wd = i;
1842 #endif
1843
1844 #ifdef NEW_LOGGING
1845                         LDAP_LOG( CONNECTION, DETAIL2, 
1846                                 "slapd_daemon_task: write active on %d\n", wd, 0, 0 );
1847 #else
1848                         Debug( LDAP_DEBUG_CONNS,
1849                                 "daemon: write active on %d\n",
1850                                 wd, 0, 0 );
1851 #endif
1852                         /*
1853                          * NOTE: it is possible that the connection was closed
1854                          * and that the stream is now inactive.
1855                          * connection_write() must valid the stream is still
1856                          * active.
1857                          */
1858
1859                         if ( connection_write( wd ) < 0 ) {
1860                                 FD_CLR( (unsigned) wd, &readfds );
1861                                 slapd_close( wd );
1862                         }
1863                 }
1864
1865 #ifdef HAVE_WINSOCK
1866                 for ( i = 0; i < readfds.fd_count; i++ )
1867 #else
1868                 for ( i = 0; i < nfds; i++ )
1869 #endif
1870                 {
1871                         ber_socket_t rd;
1872 #ifdef HAVE_WINSOCK
1873                         rd = readfds.fd_array[i];
1874 #else
1875                         if( ! FD_ISSET( i, &readfds ) ) {
1876                                 continue;
1877                         }
1878                         rd = i;
1879 #endif
1880
1881 #ifdef NEW_LOGGING
1882                         LDAP_LOG( CONNECTION, DETAIL2, 
1883                                 "slapd_daemon_task: read activity on %d\n", rd, 0, 0 );
1884 #else
1885                         Debug ( LDAP_DEBUG_CONNS,
1886                                 "daemon: read activity on %d\n", rd, 0, 0 );
1887 #endif
1888                         /*
1889                          * NOTE: it is possible that the connection was closed
1890                          * and that the stream is now inactive.
1891                          * connection_read() must valid the stream is still
1892                          * active.
1893                          */
1894
1895                         if ( connection_read( rd ) < 0 ) {
1896                                 slapd_close( rd );
1897                         }
1898                 }
1899                 ldap_pvt_thread_yield();
1900         }
1901
1902         if( slapd_shutdown == 1 ) {
1903 #ifdef NEW_LOGGING
1904                 LDAP_LOG( CONNECTION, CRIT,
1905                    "slapd_daemon_task: shutdown requested and initiated.\n", 0, 0, 0 );
1906 #else
1907                 Debug( LDAP_DEBUG_TRACE,
1908                         "daemon: shutdown requested and initiated.\n",
1909                         0, 0, 0 );
1910 #endif
1911
1912         } else if ( slapd_shutdown == 2 ) {
1913 #ifdef HAVE_NT_SERVICE_MANAGER
1914 #ifdef NEW_LOGGING
1915                         LDAP_LOG( CONNECTION, CRIT,
1916                            "slapd_daemon_task: shutdown initiated by Service Manager.\n",
1917                            0, 0, 0);
1918 #else
1919                         Debug( LDAP_DEBUG_TRACE,
1920                                "daemon: shutdown initiated by Service Manager.\n",
1921                                0, 0, 0);
1922 #endif
1923 #else /* !HAVE_NT_SERVICE_MANAGER */
1924 #ifdef NEW_LOGGING
1925                         LDAP_LOG( CONNECTION, CRIT,
1926                            "slapd_daemon_task: abnormal condition, "
1927                            "shutdown initiated.\n", 0, 0, 0 );
1928 #else
1929                         Debug( LDAP_DEBUG_TRACE,
1930                                "daemon: abnormal condition, shutdown initiated.\n",
1931                                0, 0, 0 );
1932 #endif
1933 #endif /* !HAVE_NT_SERVICE_MANAGER */
1934         } else {
1935 #ifdef NEW_LOGGING
1936                 LDAP_LOG( CONNECTION, CRIT,
1937                    "slapd_daemon_task: no active streams, shutdown initiated.\n", 
1938                    0, 0, 0 );
1939 #else
1940                 Debug( LDAP_DEBUG_TRACE,
1941                        "daemon: no active streams, shutdown initiated.\n",
1942                        0, 0, 0 );
1943 #endif
1944         }
1945
1946         if( slapd_gentle_shutdown != 2 ) {
1947                 close_listeners ( 0 );
1948         }
1949
1950         free ( slap_listeners );
1951         slap_listeners = NULL;
1952
1953         if( !slapd_gentle_shutdown ) {
1954                 connections_shutdown();
1955         }
1956
1957 #ifdef NEW_LOGGING
1958         LDAP_LOG( CONNECTION, CRIT, 
1959                 "slapd_daemon_task: shutdown waiting for %d threads to terminate.\n",
1960                 ldap_pvt_thread_pool_backload(&connection_pool), 0, 0 );
1961 #else
1962         Debug( LDAP_DEBUG_ANY,
1963             "slapd shutdown: waiting for %d threads to terminate\n",
1964             ldap_pvt_thread_pool_backload(&connection_pool), 0, 0 );
1965 #endif
1966         ldap_pvt_thread_pool_destroy(&connection_pool, 1);
1967
1968         return NULL;
1969 }
1970
1971
1972 int slapd_daemon( void )
1973 {
1974         int rc;
1975
1976         connections_init();
1977
1978 #define SLAPD_LISTENER_THREAD 1
1979 #if defined( SLAPD_LISTENER_THREAD )
1980         {
1981                 ldap_pvt_thread_t       listener_tid;
1982
1983                 /* listener as a separate THREAD */
1984                 rc = ldap_pvt_thread_create( &listener_tid,
1985                         0, slapd_daemon_task, NULL );
1986
1987                 if ( rc != 0 ) {
1988 #ifdef NEW_LOGGING
1989                         LDAP_LOG( CONNECTION, ERR, 
1990                                 "slapd_daemon: listener ldap_pvt_thread_create failed (%d).\n",
1991                                 rc, 0, 0 );
1992 #else
1993                         Debug( LDAP_DEBUG_ANY,
1994                         "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 );
1995 #endif
1996                         return rc;
1997                 }
1998  
1999                 /* wait for the listener thread to complete */
2000                 ldap_pvt_thread_join( listener_tid, (void *) NULL );
2001         }
2002 #else
2003         /* experimental code */
2004         slapd_daemon_task( NULL );
2005 #endif
2006
2007         return 0;
2008
2009 }
2010
2011 int sockinit(void)
2012 {
2013 #if defined( HAVE_WINSOCK2 )
2014     WORD wVersionRequested;
2015         WSADATA wsaData;
2016         int err;
2017
2018         wVersionRequested = MAKEWORD( 2, 0 );
2019
2020         err = WSAStartup( wVersionRequested, &wsaData );
2021         if ( err != 0 ) {
2022                 /* Tell the user that we couldn't find a usable */
2023                 /* WinSock DLL.                                  */
2024                 return -1;
2025         }
2026
2027         /* Confirm that the WinSock DLL supports 2.0.*/
2028         /* Note that if the DLL supports versions greater    */
2029         /* than 2.0 in addition to 2.0, it will still return */
2030         /* 2.0 in wVersion since that is the version we      */
2031         /* requested.                                        */
2032
2033         if ( LOBYTE( wsaData.wVersion ) != 2 ||
2034                 HIBYTE( wsaData.wVersion ) != 0 )
2035         {
2036             /* Tell the user that we couldn't find a usable */
2037             /* WinSock DLL.                                  */
2038             WSACleanup();
2039             return -1;
2040         }
2041
2042         /* The WinSock DLL is acceptable. Proceed. */
2043 #elif defined( HAVE_WINSOCK )
2044         WSADATA wsaData;
2045         if ( WSAStartup( 0x0101, &wsaData ) != 0 ) {
2046             return -1;
2047         }
2048 #endif
2049         return 0;
2050 }
2051
2052 int sockdestroy(void)
2053 {
2054 #if defined( HAVE_WINSOCK2 ) || defined( HAVE_WINSOCK )
2055         WSACleanup();
2056 #endif
2057         return 0;
2058 }
2059
2060 RETSIGTYPE
2061 slap_sig_shutdown( int sig )
2062 {
2063 #if 0
2064 #ifdef NEW_LOGGING
2065         LDAP_LOG( CONNECTION, CRIT, 
2066                 "slap_sig_shutdown: signal %d\n", sig, 0, 0 );
2067 #else
2068         Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: signal %d\n", sig, 0, 0);
2069 #endif
2070 #endif
2071
2072         /*
2073          * If the NT Service Manager is controlling the server, we don't
2074          * want SIGBREAK to kill the server. For some strange reason,
2075          * SIGBREAK is generated when a user logs out.
2076          */
2077
2078 #if HAVE_NT_SERVICE_MANAGER && SIGBREAK
2079         if (is_NT_Service && sig == SIGBREAK)
2080 #if 0
2081 #ifdef NEW_LOGGING
2082             LDAP_LOG( CONNECTION, CRIT,
2083                     "slap_sig_shutdown: SIGBREAK ignored.\n", 0, 0, 0 );
2084 #else
2085             Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: SIGBREAK ignored.\n",
2086                   0, 0, 0);
2087 #endif
2088 #endif
2089         else
2090 #endif
2091 #ifdef SIGHUP
2092         if (sig == SIGHUP && global_gentlehup && slapd_gentle_shutdown == 0)
2093                 slapd_gentle_shutdown = 1;
2094         else
2095 #endif
2096         slapd_shutdown = 1;
2097
2098         WAKE_LISTENER(1);
2099
2100         /* reinstall self */
2101         (void) SIGNAL_REINSTALL( sig, slap_sig_shutdown );
2102 }
2103
2104 RETSIGTYPE
2105 slap_sig_wake( int sig )
2106 {
2107         WAKE_LISTENER(1);
2108
2109         /* reinstall self */
2110         (void) SIGNAL_REINSTALL( sig, slap_sig_wake );
2111 }
2112
2113
2114 void slapd_add_internal(ber_socket_t s) {
2115         slapd_add(s);
2116 }
2117
2118 Listener ** slapd_get_listeners(void) {
2119         return slap_listeners;
2120 }