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

   1 /* $OpenLDAP$ */
   2 /*
   3  * Copyright 1998-2000 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/signal.h>
  14 #include <ac/socket.h>
  15 #include <ac/string.h>
  16 #include <ac/time.h>
  17 #include <ac/unistd.h>
  18
  19 #include "ldap_pvt.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 #endif /* LDAP_PF_LOCAL */
  33
  34 /* globals */
  35 time_t starttime;
  36 ber_socket_t dtblsize;
  37
  38 typedef union slap_sockaddr {
  39         struct sockaddr sa_addr;
  40         struct sockaddr_in sa_in_addr;
  41 #ifdef LDAP_PF_INET6
  42         struct sockaddr_in6 sa_in6_addr;
  43 #endif
  44 #ifdef LDAP_PF_LOCAL
  45         struct sockaddr_un sa_un_addr;
  46 #endif
  47 } Sockaddr;
  48
  49 typedef struct slap_listener {
  50         char* sl_url;
  51         char* sl_name;
  52 #ifdef HAVE_TLS
  53         int             sl_is_tls;
  54 #endif
  55         ber_socket_t            sl_sd;
  56         Sockaddr sl_sa;
  57 #define sl_addr sl_sa.sa_in_addr
  58 } Listener;
  59
  60 Listener **slap_listeners = NULL;
  61
  62 static ber_socket_t wake_sds[2];
  63
  64 #ifdef NO_THREADS
  65 static int waking;
  66 #define WAKE_LISTENER(w) \
  67 ((w && !waking) ? tcp_write( wake_sds[1], "0", 1 ), waking=1 : 0)
  68 #else
  69 #define WAKE_LISTENER(w) \
  70 do { if (w) tcp_write( wake_sds[1], "0", 1 ); } while(0)
  71 #endif
  72
  73 #ifdef HAVE_NT_SERVICE_MANAGER
  74 /* in nt_main.c */
  75 extern ldap_pvt_thread_cond_t                   started_event;
  76 extern int        is_NT_Service;
  77 #endif
  78
  79 #ifndef HAVE_WINSOCK
  80 static
  81 #endif
  82 volatile sig_atomic_t slapd_shutdown = 0;
  83
  84 static struct slap_daemon {
  85         ldap_pvt_thread_mutex_t sd_mutex;
  86
  87         int sd_nactives;
  88
  89 #ifndef HAVE_WINSOCK
  90         /* In winsock, accept() returns values higher than dtblsize
  91                 so don't bother with this optimization */
  92         int sd_nfds;
  93 #endif
  94
  95         fd_set sd_actives;
  96         fd_set sd_readers;
  97         fd_set sd_writers;
  98 } slap_daemon;
  99
 100 /*
 101  * Add a descriptor to daemon control
 102  */
 103 static void slapd_add(ber_socket_t s) {
 104         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 105
 106         assert( !FD_ISSET( s, &slap_daemon.sd_actives ));
 107         assert( !FD_ISSET( s, &slap_daemon.sd_readers ));
 108         assert( !FD_ISSET( s, &slap_daemon.sd_writers ));
 109
 110 #ifndef HAVE_WINSOCK
 111         if (s >= slap_daemon.sd_nfds) {
 112                 slap_daemon.sd_nfds = s + 1;
 113         }
 114 #endif
 115
 116         FD_SET( s, &slap_daemon.sd_actives );
 117         FD_SET( s, &slap_daemon.sd_readers );
 118
 119         Debug( LDAP_DEBUG_CONNS, "daemon: added %ld%s%s\n",
 120                 (long) s,
 121             FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
 122                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
 123
 124         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 125 }
 126
 127 /*
 128  * Remove the descriptor from daemon control
 129  */
 130 void slapd_remove(ber_socket_t s, int wake) {
 131         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 132
 133         Debug( LDAP_DEBUG_CONNS, "daemon: removing %ld%s%s\n",
 134                 (long) s,
 135             FD_ISSET(s, &slap_daemon.sd_readers) ? "r" : "",
 136                 FD_ISSET(s, &slap_daemon.sd_writers) ? "w" : "" );
 137
 138         FD_CLR( s, &slap_daemon.sd_actives );
 139         FD_CLR( s, &slap_daemon.sd_readers );
 140         FD_CLR( s, &slap_daemon.sd_writers );
 141
 142         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 143         WAKE_LISTENER(wake);
 144 }
 145
 146 void slapd_clr_write(ber_socket_t s, int wake) {
 147         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 148
 149         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
 150         FD_CLR( s, &slap_daemon.sd_writers );
 151
 152         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 153         WAKE_LISTENER(wake);
 154 }
 155
 156 void slapd_set_write(ber_socket_t s, int wake) {
 157         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 158
 159         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
 160         if (!FD_ISSET(s, &slap_daemon.sd_writers))
 161             FD_SET( (unsigned) s, &slap_daemon.sd_writers );
 162
 163         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 164         WAKE_LISTENER(wake);
 165 }
 166
 167 void slapd_clr_read(ber_socket_t s, int wake) {
 168         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 169
 170         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
 171         FD_CLR( s, &slap_daemon.sd_readers );
 172
 173         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 174         WAKE_LISTENER(wake);
 175 }
 176
 177 void slapd_set_read(ber_socket_t s, int wake) {
 178         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 179
 180         assert( FD_ISSET( s, &slap_daemon.sd_actives) );
 181         if (!FD_ISSET(s, &slap_daemon.sd_readers))
 182             FD_SET( s, &slap_daemon.sd_readers );
 183
 184         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 185         WAKE_LISTENER(wake);
 186 }
 187
 188 static void slapd_close(ber_socket_t s) {
 189         Debug( LDAP_DEBUG_CONNS, "daemon: closing %ld\n",
 190                 (long) s, 0, 0 );
 191         tcp_close(s);
 192 }
 193
 194 static slap_free_listener_addresses(struct sockaddr **sal)
 195 {
 196         struct sockaddr **sap;
 197
 198         if (sal == NULL) {
 199                 return;
 200         }
 201
 202         for (sap = sal; *sap != NULL; sap++) {
 203                 ch_free(*sap);
 204         }
 205
 206         ch_free(sal);
 207 }
 208
 209 /* port = 0 indicates AF_LOCAL */
 210 static int slap_get_listener_addresses(
 211         const char *host,
 212         unsigned short port,
 213         struct sockaddr ***sal)
 214 {
 215         struct sockaddr **sap;
 216 #ifdef HAVE_GETADDRINFO
 217         struct addrinfo hints, *res, *sai;
 218         int n, err;
 219
 220         memset( &hints, '\0', sizeof(hints) );
 221         hints.ai_flags = AI_PASSIVE;
 222         hints.ai_socktype = SOCK_STREAM;
 223 #  ifdef LDAP_PF_LOCAL
 224         if ( port == 0 ) {
 225                 hints.ai_family = AF_LOCAL;
 226                 /* host specifies a service in this case */
 227                 if (err = getaddrinfo(NULL, host, &hints, &res)) {
 228                         Debug( LDAP_DEBUG_ANY, "daemon: getaddrinfo failed: %s\n",
 229                                 gai_strerror(err), 0, 0);
 230                         return -1;
 231                 }
 232         } else
 233 #  endif
 234         {
 235                 char serv[7];
 236
 237                 snprintf(serv, sizeof serv, "%d", port);
 238                 hints.ai_family = AF_UNSPEC;
 239                 if (err = getaddrinfo(host, serv, &hints, &res)) {
 240                         Debug( LDAP_DEBUG_ANY, "daemon: getaddrinfo failed: %s\n",
 241                                 gai_strerror(err), 0, 0);
 242                         return -1;
 243                 }
 244         }
 245
 246         sai = res;
 247         for (n=2; (sai = sai->ai_next) != NULL; n++) {
 248                 /* EMPTY */ ;
 249         }
 250         *sal = ch_malloc(n * sizeof(*sal));
 251         if (*sal == NULL) {
 252                 return -1;
 253         }
 254
 255         sai = res;
 256         sap = *sal;
 257         do {
 258                 switch (sai->ai_family) {
 259 #  ifdef LDAP_PF_LOCAL
 260                 case AF_LOCAL: {
 261                         *sap = ch_malloc(sizeof(struct sockaddr_un));
 262                         if (*sap == NULL) {
 263                                 freeaddrinfo(res);
 264                                 goto errexit;
 265                         }
 266                         *(struct sockaddr_un *)*sap =
 267                                 *((struct sockaddr_un *)sai->ai_addr);
 268                 } break;
 269 #  endif
 270 #  ifdef LDAP_PF_INET6
 271                 case AF_INET6: {
 272                         *sap = ch_malloc(sizeof(struct sockaddr_in6));
 273                         if (*sap == NULL) {
 274                                 freeaddrinfo(res);
 275                                 goto errexit;
 276                         }
 277                         *(struct sockaddr_in6 *)*sap =
 278                                 *((struct sockaddr_in6 *)sai->ai_addr);
 279                 } break;
 280 #  endif
 281                 case AF_INET: {
 282                         *sap = ch_malloc(sizeof(struct sockaddr_in));
 283                         if (*sap == NULL) {
 284                                 freeaddrinfo(res);
 285                                 goto errexit;
 286                         }
 287                         *(struct sockaddr_in *)*sap =
 288                                 *((struct sockaddr_in *)sai->ai_addr);
 289                 } break;
 290                 default:
 291                         *sap = NULL;
 292                         break;
 293                 }
 294                 if (*sap != NULL) {
 295                         (*sap)->sa_family = sai->ai_family;
 296                         (*sap)->sa_len = sai->ai_addrlen;
 297                 }
 298                 sap++;
 299         } while ((sai = sai->ai_next) != NULL);
 300
 301         freeaddrinfo(res);
 302
 303 #else
 304 #  ifdef LDAP_PF_LOCAL
 305         if ( port == 0 ) {
 306                 *sal = ch_malloc(2 * sizeof(*sal));
 307                 if (*sal == NULL) {
 308                         return -1;
 309                 }
 310                 sap = *sal;
 311                 *sap = ch_malloc(sizeof(struct sockaddr_un));
 312                 if (*sap == NULL)
 313                         goto errexit;
 314                 (void)memset( (void *)*sap, '\0', sizeof(struct sockaddr_un) );
 315                 (*sap)->sa_family = AF_LOCAL;
 316                 if ( strlen(host) >
 317                      (sizeof(((struct sockaddr_un *)*sal)->sun_path) - 1) ) {
 318                         Debug( LDAP_DEBUG_ANY,
 319                                "daemon: domain socket path (%s) too long in URL",
 320                                host, 0, 0);
 321                         goto errexit;
 322                 }
 323                 strcpy( ((struct sockaddr_un *)*sap)->sun_path, host );
 324         } else
 325 #  endif
 326         {
 327                 struct in_addr in;
 328
 329                 if ( host == NULL ) {
 330                         in.s_addr = htonl(INADDR_ANY);
 331
 332                 } else if ( !inet_aton( host, &in ) ) {
 333                         struct hostent *he = gethostbyname( host );
 334                         if( he == NULL ) {
 335                                 Debug( LDAP_DEBUG_ANY,
 336                                        "daemon: invalid host %s", host, 0, 0);
 337                                 return -1;
 338                         }
 339                         AC_MEMCPY( &in, he->h_addr, sizeof( in ) );
 340                 }
 341
 342                 *sal = ch_malloc(2 * sizeof(*sal));
 343                 if (*sal == NULL) {
 344                         return -1;
 345                 }
 346
 347                 sap = *sal;
 348                 *sap = ch_malloc(sizeof(struct sockaddr_in));
 349                 if (*sap == NULL) {
 350                         goto errexit;
 351                 }
 352
 353                 (void)memset( (void *)*sap, '\0', sizeof(struct sockaddr_in) );
 354                 (*sap)->sa_family = AF_INET;
 355                 ((struct sockaddr_in *)*sap)->sin_port = htons(port);
 356                 ((struct sockaddr_in *)*sap)->sin_addr = in;
 357         }
 358         sap++;
 359 #endif
 360
 361         *sap = NULL;
 362         return 0;
 363
 364 errexit:
 365         slap_free_listener_addresses(*sal);
 366         return -1;
 367 }
 368
 369 static Listener * slap_open_listener(
 370         const char* url )
 371 {
 372         int     tmp, rc;
 373         Listener l;
 374         Listener *li;
 375         LDAPURLDesc *lud;
 376         unsigned short port;
 377         int err, addrlen;
 378         struct sockaddr **sal, **psal;
 379
 380         rc = ldap_url_parse( url, &lud );
 381
 382         if( rc != LDAP_URL_SUCCESS ) {
 383                 Debug( LDAP_DEBUG_ANY,
 384                         "daemon: listen URL \"%s\" parse error=%d\n",
 385                         url, rc, 0 );
 386                 return NULL;
 387         }
 388
 389 #ifndef HAVE_TLS
 390         if( ldap_pvt_url_scheme2tls( lud->lud_scheme ) ) {
 391                 Debug( LDAP_DEBUG_ANY,
 392                         "daemon: TLS not supported (%s)\n",
 393                         url, 0, 0 );
 394                 ldap_free_urldesc( lud );
 395                 return NULL;
 396         }
 397
 398         if(! lud->lud_port ) {
 399                 lud->lud_port = LDAP_PORT;
 400         }
 401
 402 #else
 403         l.sl_is_tls = ldap_pvt_url_scheme2tls( lud->lud_scheme );
 404
 405         if(! lud->lud_port ) {
 406                 lud->lud_port = l.sl_is_tls ? LDAPS_PORT : LDAP_PORT;
 407         }
 408 #endif
 409
 410         port = (unsigned short) lud->lud_port;
 411
 412         if ( ldap_pvt_url_scheme2proto(lud->lud_scheme) == LDAP_PROTO_IPC ) {
 413 #ifdef LDAP_PF_LOCAL
 414                 if ( lud->lud_host == NULL || lud->lud_host[0] == '\0' ) {
 415                         err = slap_get_listener_addresses(LDAPI_SOCK, 0, &sal);
 416                 } else {
 417                         err = slap_get_listener_addresses(lud->lud_host, 0, &sal);
 418                 }
 419 #else
 420                 Debug( LDAP_DEBUG_ANY, "daemon: URL scheme not supported: %s",
 421                         url, 0, 0);
 422                 ldap_free_urldesc( lud );
 423                 return NULL;
 424 #endif
 425         } else {
 426                 if( lud->lud_host == NULL || lud->lud_host[0] == '\0'
 427                         || strcmp(lud->lud_host, "*") == 0 )
 428                 {
 429                         err = slap_get_listener_addresses(NULL, port, &sal);
 430                 } else {
 431                         err = slap_get_listener_addresses(lud->lud_host, port, &sal);
 432                 }
 433         }
 434
 435         ldap_free_urldesc( lud );
 436         if ( err ) {
 437                 return NULL;
 438         }
 439
 440         psal = sal;
 441         while ( *sal != NULL ) {
 442                 switch( (*sal)->sa_family ) {
 443                 case AF_INET:
 444 #ifdef LDAP_PF_INET6
 445                 case AF_INET6:
 446 #endif
 447 #ifdef LDAP_PF_LOCAL
 448                 case AF_LOCAL:
 449 #endif
 450                         break;
 451                 default:
 452                         sal++;
 453                         continue;
 454                 }
 455                 l.sl_sd = socket( (*sal)->sa_family, SOCK_STREAM, 0);
 456                 if ( l.sl_sd == AC_SOCKET_INVALID ) {
 457                         int err = sock_errno();
 458                         Debug( LDAP_DEBUG_ANY,
 459                                 "daemon: socket() failed errno=%d (%s)\n", err,
 460                                 sock_errstr(err), 0 );
 461                         sal++;
 462                         continue;
 463                 }
 464 #ifndef HAVE_WINSOCK
 465                 if ( l.sl_sd >= dtblsize ) {
 466                         Debug( LDAP_DEBUG_ANY,
 467                                "daemon: listener descriptor %ld is too great %ld\n",
 468                                (long) l.sl_sd, (long) dtblsize, 0 );
 469                         tcp_close( l.sl_sd );
 470                         sal++;
 471                         continue;
 472                 }
 473 #endif
 474 #ifdef LDAP_PF_LOCAL
 475                 if ( (*sal)->sa_family == AF_LOCAL ) {
 476                         unlink ( ((struct sockaddr_un *)*sal)->sun_path );
 477                 } else
 478 #endif
 479                 {
 480 #ifdef SO_REUSEADDR
 481                         /* enable address reuse */
 482                         tmp = 1;
 483                         rc = setsockopt( l.sl_sd, SOL_SOCKET, SO_REUSEADDR,
 484                                          (char *) &tmp, sizeof(tmp) );
 485                         if ( rc == AC_SOCKET_ERROR ) {
 486                                 int err = sock_errno();
 487                                 Debug( LDAP_DEBUG_ANY,
 488                                        "slapd(%ld): setsockopt(SO_REUSEADDR) failed errno=%d (%s)\n",
 489                                        (long) l.sl_sd, err, sock_errstr(err) );
 490                         }
 491 #endif
 492                 }
 493
 494                 switch( (*sal)->sa_family ) {
 495                 case AF_INET:
 496                         addrlen = sizeof(struct sockaddr_in);
 497                         break;
 498 #ifdef LDAP_PF_INET6
 499                 case AF_INET6:
 500                         addrlen = sizeof(struct sockaddr_in6);
 501                         break;
 502 #endif
 503 #ifdef LDAP_PF_LOCAL
 504                 case AF_LOCAL:
 505                         addrlen = sizeof(struct sockaddr_un);
 506                         break;
 507 #endif
 508                 }
 509
 510                 if (!bind(l.sl_sd, *sal, addrlen))
 511                         break;
 512                 err = sock_errno();
 513                 Debug( LDAP_DEBUG_ANY, "daemon: bind(%ld) failed errno=%d (%s)\n",
 514                        (long) l.sl_sd, err, sock_errstr(err) );
 515                 tcp_close( l.sl_sd );
 516                 sal++;
 517         } /* while ( *sal != NULL ) */
 518
 519         if ( *sal == NULL ) {
 520                 Debug( LDAP_DEBUG_ANY, "daemon: bind(%ld) failed\n",
 521                         (long) l.sl_sd, 0, 0 );
 522                 slap_free_listener_addresses(psal);
 523                 return NULL;
 524         }
 525
 526         switch ( (*sal)->sa_family ) {
 527 #ifdef LDAP_PF_LOCAL
 528         case AF_LOCAL: {
 529                 char *addr = ((struct sockaddr_un *)*sal)->sun_path;
 530                 if ( chmod( addr, S_IRWXU ) < 0 ) {
 531                         int err = sock_errno();
 532                         Debug( LDAP_DEBUG_ANY, "daemon: fchmod(%ld) failed errno=%d (%s)",
 533                                (long) l.sl_sd, err, sock_errstr(err) );
 534                         tcp_close( l.sl_sd );
 535                         slap_free_listener_addresses(psal);
 536                         return NULL;
 537                 }
 538                 l.sl_name = ch_malloc( strlen(addr) + sizeof("PATH=") );
 539                 sprintf( l.sl_name, "PATH=%s", addr );
 540         } break;
 541 #endif /* LDAP_PF_LOCAL */
 542
 543         case AF_INET: {
 544                 char *s;
 545 #ifdef HAVE_GETADDRINFO
 546                 char addr[INET_ADDRSTRLEN];
 547                 inet_ntop( AF_INET, &((struct sockaddr_in *)*sal)->sin_addr,
 548                            addr, sizeof(addr) );
 549                 s = addr;
 550 #else
 551                 s = inet_ntoa( l.sl_addr.sin_addr );
 552 #endif
 553                 l.sl_name = ch_malloc( sizeof("IP=255.255.255.255:65535") );
 554                 sprintf( l.sl_name, "IP=%s:%d",
 555                          s != NULL ? s : "unknown" , port );
 556         } break;
 557
 558 #ifdef LDAP_PF_INET6
 559         case AF_INET6: {
 560                 char addr[INET6_ADDRSTRLEN];
 561                 inet_ntop( AF_INET6, &((struct sockaddr_in6 *)*sal)->sin6_addr,
 562                            addr, sizeof addr);
 563                 l.sl_name = ch_malloc( strlen(addr) + sizeof("IP= 65535") );
 564                 sprintf( l.sl_name, "IP=%s %d", addr, port );
 565         } break;
 566 #endif /* LDAP_PF_INET6 */
 567
 568         default:
 569                 Debug( LDAP_DEBUG_ANY, "daemon: unsupported address family (%d)\n",
 570                         (int) (*sal)->sa_family, 0, 0 );
 571                 break;
 572         }
 573
 574         slap_free_listener_addresses(psal);
 575
 576         l.sl_url = ch_strdup( url );
 577         li = ch_malloc( sizeof( Listener ) );
 578         *li = l;
 579
 580         Debug( LDAP_DEBUG_TRACE, "daemon: initialized %s\n",
 581                 l.sl_url, 0, 0 );
 582
 583         return li;
 584 }
 585
 586 static int sockinit(void);
 587 static int sockdestroy(void);
 588
 589 int slapd_daemon_init( const char *urls )
 590 {
 591         int i, rc;
 592         char **u;
 593
 594         Debug( LDAP_DEBUG_ARGS, "daemon_init: %s\n",
 595                 urls ? urls : "<null>", 0, 0 );
 596
 597         if( (rc = sockinit()) != 0 ) {
 598                 return rc;
 599         }
 600
 601 #ifdef HAVE_SYSCONF
 602         dtblsize = sysconf( _SC_OPEN_MAX );
 603 #elif HAVE_GETDTABLESIZE
 604         dtblsize = getdtablesize();
 605 #else
 606         dtblsize = FD_SETSIZE;
 607 #endif
 608
 609 #ifdef FD_SETSIZE
 610         if(dtblsize > FD_SETSIZE) {
 611                 dtblsize = FD_SETSIZE;
 612         }
 613 #endif  /* !FD_SETSIZE */
 614
 615         /* open a pipe (or something equivalent connected to itself).
 616          * we write a byte on this fd whenever we catch a signal. The main
 617          * loop will be select'ing on this socket, and will wake up when
 618          * this byte arrives.
 619          */
 620         if( (rc = lutil_pair( wake_sds )) < 0 ) {
 621                 Debug( LDAP_DEBUG_ANY,
 622                         "daemon: lutil_pair() failed rc=%d\n", rc, 0, 0 );
 623                 return rc;
 624         }
 625
 626         FD_ZERO( &slap_daemon.sd_readers );
 627         FD_ZERO( &slap_daemon.sd_writers );
 628
 629         if( urls == NULL ) {
 630                 urls = "ldap:///";
 631         }
 632
 633         u = str2charray( urls, " " );
 634
 635         if( u == NULL || u[0] == NULL ) {
 636                 Debug( LDAP_DEBUG_ANY, "daemon_init: no urls (%s) provided.\n",
 637                         urls, 0, 0 );
 638
 639                 return -1;
 640         }
 641
 642         for( i=0; u[i] != NULL; i++ ) {
 643                 Debug( LDAP_DEBUG_TRACE, "daemon_init: listen on %s\n",
 644                         u[i], 0, 0 );
 645         }
 646
 647         if( i == 0 ) {
 648                 Debug( LDAP_DEBUG_ANY, "daemon_init: no listeners to open (%s)\n",
 649                         urls, 0, 0 );
 650                 charray_free( u );
 651                 return -1;
 652         }
 653
 654         Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners to open...\n",
 655                 i, 0, 0 );
 656
 657         slap_listeners = ch_malloc( (i+1)*sizeof(Listener *) );
 658
 659         for(i = 0; u[i] != NULL; i++ ) {
 660                 slap_listeners[i] = slap_open_listener( u[i] );
 661
 662                 if( slap_listeners[i] == NULL ) {
 663                         charray_free( u );
 664                         return -1;
 665                 }
 666         }
 667         slap_listeners[i] = NULL;
 668
 669         Debug( LDAP_DEBUG_TRACE, "daemon_init: %d listeners opened\n",
 670                 i, 0, 0 );
 671
 672         charray_free( u );
 673         ldap_pvt_thread_mutex_init( &slap_daemon.sd_mutex );
 674         return !i;
 675 }
 676
 677
 678 int
 679 slapd_daemon_destroy(void)
 680 {
 681         connections_destroy();
 682         tcp_close( wake_sds[1] );
 683         tcp_close( wake_sds[0] );
 684         sockdestroy();
 685         return 0;
 686 }
 687
 688
 689 static void *
 690 slapd_daemon_task(
 691         void *ptr
 692 )
 693 {
 694         int l;
 695         time_t  last_idle_check = slap_get_time();
 696         time( &starttime );
 697
 698         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
 699                 if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
 700                         continue;
 701
 702                 if ( listen( slap_listeners[l]->sl_sd, 5 ) == -1 ) {
 703                         int err = sock_errno();
 704                         Debug( LDAP_DEBUG_ANY,
 705                                 "daemon: listen(%s, 5) failed errno=%d (%s)\n",
 706                                         slap_listeners[l]->sl_url, err,
 707                                         sock_errstr(err) );
 708                         return( (void*)-1 );
 709                 }
 710
 711                 slapd_add( slap_listeners[l]->sl_sd );
 712         }
 713
 714 #ifdef HAVE_NT_SERVICE_MANAGER
 715         if ( started_event != NULL ) {
 716                 ldap_pvt_thread_cond_signal( &started_event );
 717         }
 718 #endif
 719         /* initialization complete. Here comes the loop. */
 720
 721         while ( !slapd_shutdown ) {
 722                 ber_socket_t i;
 723                 int ns;
 724                 int at;
 725                 ber_socket_t nfds;
 726 #define SLAPD_EBADF_LIMIT 16
 727                 int ebadf = 0;
 728
 729 #define SLAPD_IDLE_CHECK_LIMIT 4
 730                 time_t  now = slap_get_time();
 731
 732
 733                 fd_set                  readfds;
 734                 fd_set                  writefds;
 735                 Sockaddr                from;
 736
 737 #if defined(SLAPD_RLOOKUPS)
 738         struct hostent          *hp;
 739 #endif
 740                 struct timeval          zero;
 741                 struct timeval          *tvp;
 742
 743                 if( global_idletimeout > 0 && difftime(
 744                         last_idle_check+global_idletimeout/SLAPD_IDLE_CHECK_LIMIT,
 745                         now ) < 0 )
 746                 {
 747                         connections_timeout_idle(now);
 748                 }
 749
 750                 FD_ZERO( &writefds );
 751                 FD_ZERO( &readfds );
 752
 753                 zero.tv_sec = 0;
 754                 zero.tv_usec = 0;
 755
 756                 ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 757
 758 #ifdef FD_SET_MANUAL_COPY
 759                 for( s = 0; s < nfds; s++ ) {
 760                         if(FD_ISSET( &slap_sd_readers, s )) {
 761                                 FD_SET( s, &readfds );
 762                         }
 763                         if(FD_ISSET( &slap_sd_writers, s )) {
 764                                 FD_SET( s, &writefds );
 765                         }
 766                 }
 767 #else
 768                 AC_MEMCPY( &readfds, &slap_daemon.sd_readers, sizeof(fd_set) );
 769                 AC_MEMCPY( &writefds, &slap_daemon.sd_writers, sizeof(fd_set) );
 770 #endif
 771                 assert(!FD_ISSET(wake_sds[0], &readfds));
 772                 FD_SET( wake_sds[0], &readfds );
 773
 774                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
 775                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
 776                                 continue;
 777                         if (!FD_ISSET(slap_listeners[l]->sl_sd, &readfds))
 778                             FD_SET( slap_listeners[l]->sl_sd, &readfds );
 779                 }
 780
 781 #ifndef HAVE_WINSOCK
 782                 nfds = slap_daemon.sd_nfds;
 783 #else
 784                 nfds = dtblsize;
 785 #endif
 786
 787                 ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 788
 789                 at = ldap_pvt_thread_pool_backload(&connection_pool);
 790
 791 #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
 792                 tvp = NULL;
 793 #else
 794                 tvp = at ? &zero : NULL;
 795 #endif
 796
 797                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
 798                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
 799                                 continue;
 800
 801                         Debug( LDAP_DEBUG_CONNS,
 802                                 "daemon: select: listen=%d active_threads=%d tvp=%s\n",
 803                                         slap_listeners[l]->sl_sd, at,
 804                                         tvp == NULL ? "NULL" : "zero" );
 805                 }
 806
 807                 switch(ns = select( nfds, &readfds,
 808 #ifdef HAVE_WINSOCK
 809                         /* don't pass empty fd_set */
 810                         ( writefds.fd_count > 0 ? &writefds : NULL ),
 811 #else
 812                         &writefds,
 813 #endif
 814                         NULL, tvp ))
 815                 {
 816                 case -1: {      /* failure - try again */
 817                                 int err = sock_errno();
 818
 819                                 if( err == EBADF
 820 #ifdef WSAENOTSOCK
 821                                         /* you'd think this would be EBADF */
 822                                         || err == WSAENOTSOCK
 823 #endif
 824                                 ) {
 825                                         if (++ebadf < SLAPD_EBADF_LIMIT)
 826                                                 continue;
 827                                 }
 828
 829                                 if( err != EINTR ) {
 830                                         Debug( LDAP_DEBUG_CONNS,
 831                                                 "daemon: select failed (%d): %s\n",
 832                                                 err, sock_errstr(err), 0 );
 833
 834                                         slapd_shutdown = -1;
 835                                 }
 836                         }
 837                         continue;
 838
 839                 case 0:         /* timeout - let threads run */
 840                         ebadf = 0;
 841                         Debug( LDAP_DEBUG_CONNS, "daemon: select timeout - yielding\n",
 842                             0, 0, 0 );
 843                 ldap_pvt_thread_yield();
 844                         continue;
 845
 846                 default:        /* something happened - deal with it */
 847                         if( slapd_shutdown ) continue;
 848
 849                         ebadf = 0;
 850                         Debug( LDAP_DEBUG_CONNS, "daemon: activity on %d descriptors\n",
 851                                 ns, 0, 0 );
 852                         /* FALL THRU */
 853                 }
 854
 855                 if( FD_ISSET( wake_sds[0], &readfds ) ) {
 856                         char c[BUFSIZ];
 857                         tcp_read( wake_sds[0], c, sizeof(c) );
 858 #ifdef NO_THREADS
 859                         waking = 0;
 860 #endif
 861                         continue;
 862                 }
 863
 864                 for ( l = 0; slap_listeners[l] != NULL; l++ ) {
 865                         ber_int_t s;
 866                         socklen_t len = sizeof(from);
 867                         long id;
 868                         slap_ssf_t ssf = 0;
 869                         char *authid = NULL;
 870
 871                         char    *dnsname;
 872                         char    *peeraddr;
 873 #ifdef LDAP_PF_LOCAL
 874                         char    peername[MAXPATHLEN + sizeof("PATH=")];
 875 #elif defined(LDAP_PF_INET6)
 876                         char    peername[sizeof("IP=ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff 65535")];
 877 #else
 878                         char    peername[sizeof("IP=255.255.255.255:65336")];
 879 #endif /* LDAP_PF_LOCAL */
 880
 881                         peername[0] = '\0';
 882
 883                         if ( slap_listeners[l]->sl_sd == AC_SOCKET_INVALID )
 884                                 continue;
 885
 886                         if ( !FD_ISSET( slap_listeners[l]->sl_sd, &readfds ) )
 887                                 continue;
 888
 889                         s = accept( slap_listeners[l]->sl_sd,
 890                                 (struct sockaddr *) &from, &len );
 891                         if ( s == AC_SOCKET_INVALID ) {
 892                                 int err = sock_errno();
 893                                 Debug( LDAP_DEBUG_ANY,
 894                                     "daemon: accept(%ld) failed errno=%d (%s)\n",
 895                                     (long) slap_listeners[l]->sl_sd, err,
 896                                     sock_errstr(err) );
 897                                 continue;
 898                         }
 899
 900 #ifndef HAVE_WINSOCK
 901                         /* make sure descriptor number isn't too great */
 902                         if ( s >= dtblsize ) {
 903                                 Debug( LDAP_DEBUG_ANY,
 904                                         "daemon: %ld beyond descriptor table size %ld\n",
 905                                         (long) s, (long) dtblsize, 0 );
 906                                 slapd_close(s);
 907                                 continue;
 908                         }
 909 #endif
 910
 911 #ifdef LDAP_DEBUG
 912                         ldap_pvt_thread_mutex_lock( &slap_daemon.sd_mutex );
 913
 914                         /* newly accepted stream should not be in any of the FD SETS */
 915                         assert( !FD_ISSET( s, &slap_daemon.sd_actives) );
 916                         assert( !FD_ISSET( s, &slap_daemon.sd_readers) );
 917                         assert( !FD_ISSET( s, &slap_daemon.sd_writers) );
 918
 919                         ldap_pvt_thread_mutex_unlock( &slap_daemon.sd_mutex );
 920 #endif
 921
 922 #if defined( SO_KEEPALIVE ) || defined( TCP_NODELAY )
 923 #ifdef LDAP_PF_LOCAL
 924                         /* for IPv4 and IPv6 sockets only */
 925                         if ( from.sa_addr.sa_family != AF_LOCAL )
 926 #endif /* LDAP_PF_LOCAL */
 927                         {
 928                                 int rc;
 929                                 int tmp;
 930 #ifdef SO_KEEPALIVE
 931                                 /* enable keep alives */
 932                                 tmp = 1;
 933                                 rc = setsockopt( s, SOL_SOCKET, SO_KEEPALIVE,
 934                                         (char *) &tmp, sizeof(tmp) );
 935                                 if ( rc == AC_SOCKET_ERROR ) {
 936                                         int err = sock_errno();
 937                                         Debug( LDAP_DEBUG_ANY,
 938                                                 "slapd(%ld): setsockopt(SO_KEEPALIVE) failed "
 939                                                 "errno=%d (%s)\n", (long) s, err, sock_errstr(err) );
 940                                 }
 941 #endif
 942 #ifdef TCP_NODELAY
 943                                 /* enable no delay */
 944                                 tmp = 1;
 945                                 rc = setsockopt( s, IPPROTO_TCP, TCP_NODELAY,
 946                                         (char *)&tmp, sizeof(tmp) );
 947                                 if ( rc == AC_SOCKET_ERROR ) {
 948                                         int err = sock_errno();
 949                                         Debug( LDAP_DEBUG_ANY,
 950                                                 "slapd(%ld): setsockopt(TCP_NODELAY) failed "
 951                                                 "errno=%d (%s)\n", (long) s, err, sock_errstr(err) );
 952                                 }
 953 #endif
 954                         }
 955 #endif
 956
 957                         Debug( LDAP_DEBUG_CONNS, "daemon: new connection on %ld\n",
 958                                 (long) s, 0, 0 );
 959
 960                         switch ( from.sa_addr.sa_family ) {
 961 #  ifdef LDAP_PF_LOCAL
 962                         case AF_LOCAL:
 963                                 sprintf( peername, "PATH=%s", from.sa_un_addr.sun_path );
 964                                 ssf = LDAP_PVT_SASL_LOCAL_SSF;
 965                                 break;
 966 #endif /* LDAP_PF_LOCAL */
 967
 968 #  ifdef LDAP_PF_INET6
 969                         case AF_INET6:
 970                         if ( IN6_IS_ADDR_V4MAPPED(&from.sa_in6_addr.sin6_addr) ) {
 971                                 peeraddr = inet_ntoa( *((struct in_addr *)
 972                                                         &from.sa_in6_addr.sin6_addr.s6_addr[12]) );
 973                                 sprintf( peername, "IP=%s:%d",
 974                                          peeraddr != NULL ? peeraddr : "unknown",
 975                                          (unsigned) ntohs( from.sa_in6_addr.sin6_port ) );
 976                         } else {
 977                                 char addr[INET6_ADDRSTRLEN];
 978                                 sprintf( peername, "IP=%s %d",
 979                                          inet_ntop( AF_INET6,
 980                                                     &from.sa_in6_addr.sin6_addr,
 981                                                     addr, sizeof addr) ? addr : "unknown",
 982                                          (unsigned) ntohs( from.sa_in6_addr.sin6_port ) );
 983                         }
 984                         break;
 985 #  endif /* LDAP_PF_INET6 */
 986
 987                         case AF_INET:
 988                         peeraddr = inet_ntoa( from.sa_in_addr.sin_addr );
 989                         sprintf( peername, "IP=%s:%d",
 990                                 peeraddr != NULL ? peeraddr : "unknown",
 991                                 (unsigned) ntohs( from.sa_in_addr.sin_port ) );
 992                                 break;
 993
 994                         default:
 995                                 slapd_close(s);
 996                                 continue;
 997                         }
 998
 999                         if ( ( from.sa_addr.sa_family == AF_INET )
1000 #ifdef LDAP_PF_INET6
1001                                 || ( from.sa_addr.sa_family == AF_INET6 )
1002 #endif
1003                         ) {
1004 #ifdef SLAPD_RLOOKUPS
1005 #  ifdef LDAP_PF_INET6
1006                                 if ( from.sa_addr.sa_family == AF_INET6 )
1007                                         hp = gethostbyaddr(
1008                                                 (char *)&(from.sa_in6_addr.sin6_addr),
1009                                                 sizeof(from.sa_in6_addr.sin6_addr),
1010                                                 AF_INET6 );
1011                                 else
1012 #  endif /* LDAP_PF_INET6 */
1013                                 hp = gethostbyaddr(
1014                                         (char *) &(from.sa_in_addr.sin_addr),
1015                                         sizeof(from.sa_in_addr.sin_addr),
1016                                         AF_INET );
1017                                 dnsname = hp ? ldap_pvt_str2lower( hp->h_name ) : NULL;
1018 #else
1019                                 dnsname = NULL;
1020 #endif /* SLAPD_RLOOKUPS */
1021
1022 #ifdef HAVE_TCPD
1023                                 if ( !hosts_ctl("slapd",
1024                                                 dnsname != NULL ? dnsname : STRING_UNKNOWN,
1025                                                 peeraddr != NULL ? peeraddr : STRING_UNKNOWN,
1026                                                 STRING_UNKNOWN ))
1027                                 {
1028                                         /* DENY ACCESS */
1029                                         Statslog( LDAP_DEBUG_ANY,
1030                                                 "fd=%ld host access from %s (%s) denied.\n",
1031                                                 (long) s,
1032                                                 dnsname != NULL ? dnsname : "unknown",
1033                                                 peeraddr != NULL ? peeraddr : "unknown",
1034                                                 0, 0 );
1035                                         slapd_close(s);
1036                                         continue;
1037                                 }
1038 #endif /* HAVE_TCPD */
1039                         }
1040
1041                         id = connection_init(s,
1042                                 slap_listeners[l]->sl_url,
1043                                 dnsname != NULL ? dnsname : "unknown",
1044                                 peername,
1045                                 slap_listeners[l]->sl_name,
1046 #ifdef HAVE_TLS
1047                                 slap_listeners[l]->sl_is_tls,
1048 #else
1049                                 0,
1050 #endif
1051                                 ssf,
1052                                 authid );
1053
1054                         if( authid ) ch_free(authid);
1055
1056                         if( id < 0 ) {
1057                                 Debug( LDAP_DEBUG_ANY,
1058                                         "daemon: connection_init(%ld, %s, %s) failed.\n",
1059                                         (long) s,
1060                                         peername,
1061                                         slap_listeners[l]->sl_name );
1062                                 slapd_close(s);
1063                                 continue;
1064                         }
1065
1066                         Statslog( LDAP_DEBUG_STATS,
1067                                 "daemon: conn=%ld fd=%ld connection from %s (%s) accepted.\n",
1068                                 id, (long) s,
1069                                 peername,
1070                                 slap_listeners[l]->sl_name,
1071                                 0 );
1072
1073                         slapd_add( s );
1074                         continue;
1075                 }
1076
1077 #ifdef LDAP_DEBUG
1078                 Debug( LDAP_DEBUG_CONNS, "daemon: activity on:", 0, 0, 0 );
1079 #ifdef HAVE_WINSOCK
1080                 for ( i = 0; i < readfds.fd_count; i++ ) {
1081                         Debug( LDAP_DEBUG_CONNS, " %d%s",
1082                                 readfds.fd_array[i], "r", 0 );
1083                 }
1084                 for ( i = 0; i < writefds.fd_count; i++ ) {
1085                         Debug( LDAP_DEBUG_CONNS, " %d%s",
1086                                 writefds.fd_array[i], "w", 0 );
1087                 }
1088 #else
1089                 for ( i = 0; i < nfds; i++ ) {
1090                         int     r, w;
1091                         int     is_listener = 0;
1092
1093                         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1094                                 if ( i == slap_listeners[l]->sl_sd ) {
1095                                         is_listener = 1;
1096                                         break;
1097                                 }
1098                         }
1099                         if ( is_listener ) {
1100                                 continue;
1101                         }
1102                         r = FD_ISSET( i, &readfds );
1103                         w = FD_ISSET( i, &writefds );
1104                         if ( r || w ) {
1105                                 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
1106                                     r ? "r" : "", w ? "w" : "" );
1107                         }
1108                 }
1109 #endif
1110                 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
1111 #endif
1112
1113                 /* loop through the writers */
1114 #ifdef HAVE_WINSOCK
1115                 for ( i = 0; i < writefds.fd_count; i++ )
1116 #else
1117                 for ( i = 0; i < nfds; i++ )
1118 #endif
1119                 {
1120                         ber_socket_t wd;
1121                         int is_listener = 0;
1122 #ifdef HAVE_WINSOCK
1123                         wd = writefds.fd_array[i];
1124 #else
1125                         if( ! FD_ISSET( i, &writefds ) ) {
1126                                 continue;
1127                         }
1128                         wd = i;
1129 #endif
1130
1131                         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1132                                 if ( i == slap_listeners[l]->sl_sd ) {
1133                                         is_listener = 1;
1134                                         break;
1135                                 }
1136                         }
1137                         if ( is_listener ) {
1138                                 continue;
1139                         }
1140                         Debug( LDAP_DEBUG_CONNS,
1141                                 "daemon: write active on %d\n",
1142                                 wd, 0, 0 );
1143
1144                         /*
1145                          * NOTE: it is possible that the connection was closed
1146                          * and that the stream is now inactive.
1147                          * connection_write() must valid the stream is still
1148                          * active.
1149                          */
1150
1151                         if ( connection_write( wd ) < 0 ) {
1152                                 FD_CLR( (unsigned) wd, &readfds );
1153                                 slapd_close( wd );
1154                         }
1155                 }
1156
1157 #ifdef HAVE_WINSOCK
1158                 for ( i = 0; i < readfds.fd_count; i++ )
1159 #else
1160                 for ( i = 0; i < nfds; i++ )
1161 #endif
1162                 {
1163                         ber_socket_t rd;
1164                         int is_listener = 0;
1165
1166 #ifdef HAVE_WINSOCK
1167                         rd = readfds.fd_array[i];
1168 #else
1169                         if( ! FD_ISSET( i, &readfds ) ) {
1170                                 continue;
1171                         }
1172                         rd = i;
1173 #endif
1174
1175                         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1176                                 if ( rd == slap_listeners[l]->sl_sd ) {
1177                                         is_listener = 1;
1178                                         break;
1179                                 }
1180                         }
1181                         if ( is_listener ) {
1182                                 continue;
1183                         }
1184
1185                         Debug ( LDAP_DEBUG_CONNS,
1186                                 "daemon: read activity on %d\n", rd, 0, 0 );
1187
1188                         /*
1189                          * NOTE: it is possible that the connection was closed
1190                          * and that the stream is now inactive.
1191                          * connection_read() must valid the stream is still
1192                          * active.
1193                          */
1194
1195                         if ( connection_read( rd ) < 0 ) {
1196                                 slapd_close( rd );
1197                         }
1198                 }
1199                 ldap_pvt_thread_yield();
1200         }
1201
1202         if( slapd_shutdown > 0 ) {
1203                 Debug( LDAP_DEBUG_TRACE,
1204                         "daemon: shutdown requested and initiated.\n",
1205                         0, 0, 0 );
1206
1207         } else if ( slapd_shutdown < 0 ) {
1208 #ifdef HAVE_NT_SERVICE_MANAGER
1209                 if (slapd_shutdown == -1)
1210                     Debug( LDAP_DEBUG_TRACE,
1211                           "daemon: shutdown initiated by Service Manager.\n",
1212                           0, 0, 0);
1213                 else
1214 #endif
1215                 Debug( LDAP_DEBUG_TRACE,
1216                         "daemon: abnormal condition, shutdown initiated.\n",
1217                         0, 0, 0 );
1218         } else {
1219                 Debug( LDAP_DEBUG_TRACE,
1220                         "daemon: no active streams, shutdown initiated.\n",
1221                         0, 0, 0 );
1222         }
1223
1224         for ( l = 0; slap_listeners[l] != NULL; l++ ) {
1225                 if ( slap_listeners[l]->sl_sd != AC_SOCKET_INVALID ) {
1226 #ifdef LDAP_PF_LOCAL
1227                         if ( slap_listeners[l]->sl_sa.sa_addr.sa_family == AF_LOCAL ) {
1228                                 unlink( slap_listeners[l]->sl_sa.sa_un_addr.sun_path );
1229                         }
1230 #endif /* LDAP_PF_LOCAL */
1231                         slapd_close( slap_listeners[l]->sl_sd );
1232                         break;
1233                 }
1234         }
1235
1236         Debug( LDAP_DEBUG_ANY,
1237             "slapd shutdown: waiting for %d threads to terminate\n",
1238             ldap_pvt_thread_pool_backload(&connection_pool), 0, 0 );
1239
1240         ldap_pvt_thread_pool_destroy(&connection_pool, 1);
1241
1242         return NULL;
1243 }
1244
1245
1246 int slapd_daemon( void )
1247 {
1248         int rc;
1249
1250         connections_init();
1251
1252 #define SLAPD_LISTENER_THREAD 1
1253 #if defined( SLAPD_LISTENER_THREAD )
1254         {
1255                 ldap_pvt_thread_t       listener_tid;
1256
1257                 /* listener as a separate THREAD */
1258                 rc = ldap_pvt_thread_create( &listener_tid,
1259                         0, slapd_daemon_task, NULL );
1260
1261                 if ( rc != 0 ) {
1262                         Debug( LDAP_DEBUG_ANY,
1263                         "listener ldap_pvt_thread_create failed (%d)\n", rc, 0, 0 );
1264                         return rc;
1265                 }
1266
1267                 /* wait for the listener thread to complete */
1268                 ldap_pvt_thread_join( listener_tid, (void *) NULL );
1269         }
1270 #else
1271         /* experimental code */
1272         slapd_daemon_task( NULL );
1273 #endif
1274
1275         return 0;
1276
1277 }
1278
1279 int sockinit(void)
1280 {
1281 #if defined( HAVE_WINSOCK2 )
1282     WORD wVersionRequested;
1283         WSADATA wsaData;
1284         int err;
1285
1286         wVersionRequested = MAKEWORD( 2, 0 );
1287
1288         err = WSAStartup( wVersionRequested, &wsaData );
1289         if ( err != 0 ) {
1290                 /* Tell the user that we couldn't find a usable */
1291                 /* WinSock DLL.                                  */
1292                 return -1;
1293         }
1294
1295         /* Confirm that the WinSock DLL supports 2.0.*/
1296         /* Note that if the DLL supports versions greater    */
1297         /* than 2.0 in addition to 2.0, it will still return */
1298         /* 2.0 in wVersion since that is the version we      */
1299         /* requested.                                        */
1300
1301         if ( LOBYTE( wsaData.wVersion ) != 2 ||
1302                 HIBYTE( wsaData.wVersion ) != 0 )
1303         {
1304             /* Tell the user that we couldn't find a usable */
1305             /* WinSock DLL.                                  */
1306             WSACleanup();
1307             return -1;
1308         }
1309
1310         /* The WinSock DLL is acceptable. Proceed. */
1311 #elif defined( HAVE_WINSOCK )
1312         WSADATA wsaData;
1313         if ( WSAStartup( 0x0101, &wsaData ) != 0 ) {
1314             return -1;
1315         }
1316 #endif
1317         return 0;
1318 }
1319
1320 int sockdestroy(void)
1321 {
1322 #if defined( HAVE_WINSOCK2 ) || defined( HAVE_WINSOCK )
1323         WSACleanup();
1324 #endif
1325         return 0;
1326 }
1327
1328 RETSIGTYPE
1329 slap_sig_shutdown( int sig )
1330 {
1331         Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: signal %d\n", sig, 0, 0);
1332
1333         /*
1334          * If the NT Service Manager is controlling the server, we don't
1335          * want SIGBREAK to kill the server. For some strange reason,
1336          * SIGBREAK is generated when a user logs out.
1337          */
1338
1339 #if HAVE_NT_SERVICE_MANAGER && SIGBREAK
1340         if (is_NT_Service && sig == SIGBREAK)
1341             Debug(LDAP_DEBUG_TRACE, "slap_sig_shutdown: SIGBREAK ignored.\n",
1342                   0, 0, 0);
1343         else
1344 #endif
1345         slapd_shutdown = sig;
1346
1347         WAKE_LISTENER(1);
1348
1349         /* reinstall self */
1350         (void) SIGNAL_REINSTALL( sig, slap_sig_shutdown );
1351 }
1352
1353 RETSIGTYPE
1354 slap_sig_wake( int sig )
1355 {
1356         WAKE_LISTENER(1);
1357
1358         /* reinstall self */
1359         (void) SIGNAL_REINSTALL( sig, slap_sig_wake );
1360 }
1361
1362
1363 void slapd_add_internal(ber_socket_t s) {
1364         slapd_add(s);
1365 }