git.sur5r.net Git - openldap/blob - servers/slapd/daemon.c

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