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