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

   1
   2 /* Revision history
   3  *
   4  * 5-Jun-96     hodges
   5  *      Added locking of new_conn_mutex when traversing the c[] array.
   6  */
   7
   8 #include "portable.h"
   9
  10 #include <stdio.h>
  11
  12 #include <ac/ctype.h>
  13 #include <ac/errno.h>
  14 #include <ac/signal.h>
  15 #include <ac/socket.h>
  16 #include <ac/string.h>
  17 #include <ac/time.h>
  18 #include <ac/unistd.h>
  19
  20 #include "ldapconfig.h"
  21 #include "slap.h"
  22
  23 #ifdef HAVE_SYS_FILIO_H
  24 #include <sys/filio.h>
  25 #elif HAVE_SYS_IOCTL_H
  26 #include <sys/ioctl.h>
  27 #endif
  28
  29 #ifdef HAVE_TCPD
  30 #include <tcpd.h>
  31
  32 int allow_severity = LOG_INFO;
  33 int deny_severity = LOG_NOTICE;
  34 #endif /* TCP Wrappers */
  35
  36 int             dtblsize;
  37 Connection      *c;
  38
  39 static int slapd_shutdown = 0;
  40
  41 /* a link to the slapd.conf configuration parameters */
  42 extern char *slapd_pid_file;
  43 extern char *slapd_args_file;
  44
  45 void *
  46 slapd_daemon(
  47     void *port
  48 )
  49 {
  50         Operation               *o;
  51         BerElement              ber;
  52         unsigned long           len, tag, msgid;
  53         int                     i;
  54         int                     tcps, ns;
  55         struct sockaddr_in      addr;
  56         fd_set                  readfds;
  57         fd_set                  writefds;
  58         FILE                    *fp;
  59         int                     on = 1;
  60
  61 #ifdef HAVE_SYSCONF
  62         dtblsize = sysconf( _SC_OPEN_MAX );
  63 #elif HAVE_GETDTABLESIZE
  64         dtblsize = getdtablesize();
  65 #else
  66         dtblsize = FD_SETSIZE
  67 #endif
  68
  69 #ifdef FD_SETSIZE
  70         if(dtblsize > FD_SETSIZE) {
  71                 dtblsize = FD_SETSIZE;
  72         }
  73 #endif  /* !FD_SETSIZE */
  74
  75         c = (Connection *) ch_calloc( 1, dtblsize * sizeof(Connection) );
  76
  77         for ( i = 0; i < dtblsize; i++ ) {
  78                 c[i].c_dn = NULL;
  79                 c[i].c_cdn = NULL;
  80                 c[i].c_addr = NULL;
  81                 c[i].c_domain = NULL;
  82                 c[i].c_ops = NULL;
  83                 c[i].c_sb.sb_sd = -1;
  84                 c[i].c_sb.sb_options = LBER_NO_READ_AHEAD;
  85                 c[i].c_sb.sb_naddr = 0;
  86                 c[i].c_sb.sb_ber.ber_buf = NULL;
  87                 c[i].c_sb.sb_ber.ber_ptr = NULL;
  88                 c[i].c_sb.sb_ber.ber_end = NULL;
  89                 c[i].c_writewaiter = 0;
  90                 c[i].c_connid = 0;
  91                 ldap_pvt_thread_mutex_init( &c[i].c_dnmutex );
  92                 ldap_pvt_thread_mutex_init( &c[i].c_opsmutex );
  93                 ldap_pvt_thread_mutex_init( &c[i].c_pdumutex );
  94                 ldap_pvt_thread_cond_init( &c[i].c_wcv );
  95         }
  96
  97         if ( (tcps = socket( AF_INET, SOCK_STREAM, 0 )) == -1 ) {
  98                 int err = errno;
  99                 Debug( LDAP_DEBUG_ANY, "socket() failed errno %d (%s)", err,
 100                     err > -1 && err < sys_nerr ? sys_errlist[err] :
 101                     "unknown", 0 );
 102                 exit( 1 );
 103         }
 104
 105         i = 1;
 106         if ( setsockopt( tcps, SOL_SOCKET, SO_REUSEADDR, (char *) &i,
 107             sizeof(i) ) == -1 ) {
 108                 int err = errno;
 109                 Debug( LDAP_DEBUG_ANY, "setsockopt(reuse) failed errno %d (%s)",
 110                     err, err > -1 && err < sys_nerr ? sys_errlist[err] :
 111                     "unknown", 0 );
 112         }
 113
 114 #ifdef SO_KEEPALIVE
 115         i = 1;
 116         if ( setsockopt( tcps, SOL_SOCKET, SO_KEEPALIVE, (char *) &i,
 117             sizeof(i) ) == -1 ) {
 118                 int err = errno;
 119                 Debug( LDAP_DEBUG_ANY, "setsockopt(keepalive) failed errno %d (%s)",
 120                     err, err > -1 && err < sys_nerr ? sys_errlist[err] :
 121                     "unknown", 0 );
 122         }
 123 #endif
 124
 125 #ifdef TCP_NODELAY
 126         i = 1;
 127         if ( setsockopt( tcps, IPPROTO_TCP, TCP_NODELAY, (char *) &i,
 128             sizeof(i) ) == -1 ) {
 129                 int err = errno;
 130                 Debug( LDAP_DEBUG_ANY, "setsockopt(TCP_NODELAY) failed errno %d (%s)",
 131                     err, err > -1 && err < sys_nerr ? sys_errlist[err] :
 132                     "unknown", 0 );
 133         }
 134 #endif
 135
 136         (void) memset( (void *) &addr, '\0', sizeof(addr) );
 137         addr.sin_family = AF_INET;
 138         addr.sin_addr.s_addr = INADDR_ANY;
 139         addr.sin_port = htons( (int)port );
 140         if ( bind( tcps, (struct sockaddr *) &addr, sizeof(addr) ) == -1 ) {
 141                 int err = errno;
 142                 Debug( LDAP_DEBUG_ANY, "bind() failed errno %d (%s)\n",
 143                     err, err > -1 && err < sys_nerr ? sys_errlist[err] :
 144                     "unknown", 0 );
 145                 exit( 1 );
 146         }
 147
 148         if ( listen( tcps, 5 ) == -1 ) {
 149                 int err = errno;
 150                 Debug( LDAP_DEBUG_ANY, "listen() failed errno %d (%s)",
 151                     err, err > -1 && err < sys_nerr ? sys_errlist[err] :
 152                     "unknown", 0 );
 153                 exit( 1 );
 154         }
 155
 156         Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
 157
 158         if (( slapd_pid_file != NULL ) &&
 159                         (( fp = fopen( slapd_pid_file, "w" )) != NULL )) {
 160                 fprintf( fp, "%d\n", (int) getpid() );
 161                 fclose( fp );
 162         }
 163
 164         if (( slapd_args_file != NULL ) &&
 165                         (( fp = fopen( slapd_args_file, "w" )) != NULL )) {
 166                 for ( i = 0; i < g_argc; i++ ) {
 167                         fprintf( fp, "%s ", g_argv[i] );
 168                 }
 169                 fprintf( fp, "\n" );
 170                 fclose( fp );
 171         }
 172
 173         while ( !slapd_shutdown ) {
 174                 struct sockaddr_in      from;
 175                 struct hostent          *hp;
 176                 struct timeval          zero;
 177                 struct timeval          *tvp;
 178                 int                     len, pid;
 179
 180                 char    *client_name;
 181                 char    *client_addr;
 182
 183                 FD_ZERO( &writefds );
 184                 FD_ZERO( &readfds );
 185                 FD_SET( tcps, &readfds );
 186
 187                 zero.tv_sec = 0;
 188                 zero.tv_usec = 0;
 189
 190                 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
 191                 Debug( LDAP_DEBUG_CONNS,
 192                     "listening for connections on %d, activity on:",
 193                     tcps, 0, 0 );
 194
 195                 ldap_pvt_thread_mutex_lock( &new_conn_mutex );
 196                 for ( i = 0; i < dtblsize; i++ ) {
 197                         if ( c[i].c_sb.sb_sd != -1 ) {
 198                                 FD_SET( c[i].c_sb.sb_sd, &readfds );
 199
 200                                 if ( c[i].c_writewaiter ) {
 201                                         FD_SET( c[i].c_sb.sb_sd, &writefds );
 202                                 }
 203                                 Debug( LDAP_DEBUG_CONNS, " %dr%s", i,
 204                                     c[i].c_writewaiter ? "w" : "", 0 );
 205                         }
 206                 }
 207                 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
 208                 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
 209
 210                 Debug( LDAP_DEBUG_CONNS, "before select active_threads %d\n",
 211                     active_threads, 0, 0 );
 212 #if defined( HAVE_YIELDING_SELECT ) || defined( NO_THREADS )
 213                 tvp = NULL;
 214 #else
 215                 tvp = active_threads ? &zero : NULL;
 216 #endif
 217                 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
 218
 219                 switch ( i = select( dtblsize, &readfds, &writefds, 0, tvp ) ) {
 220                 case -1:        { /* failure - try again */
 221                         int err = errno;
 222                         Debug( LDAP_DEBUG_CONNS,
 223                             "select failed errno %d (%s)\n",
 224                             err, err > -1 && err < sys_nerr ?
 225                             sys_errlist[err] : "unknown", 0 );
 226                         } continue;
 227
 228                 case 0:         /* timeout - let threads run */
 229                         Debug( LDAP_DEBUG_CONNS, "select timeout - yielding\n",
 230                             0, 0, 0 );
 231                         ldap_pvt_thread_yield();
 232                         continue;
 233
 234                 default:        /* something happened - deal with it */
 235                         Debug( LDAP_DEBUG_CONNS, "select activity on %d descriptors\n", i, 0, 0 );
 236                         ;       /* FALL */
 237                 }
 238                 ldap_pvt_thread_mutex_lock( &currenttime_mutex );
 239                 time( &currenttime );
 240                 ldap_pvt_thread_mutex_unlock( &currenttime_mutex );
 241
 242                 /* new connection */
 243                 ldap_pvt_thread_mutex_lock( &new_conn_mutex );
 244                 if ( FD_ISSET( tcps, &readfds ) ) {
 245                         len = sizeof(from);
 246                         if ( (ns = accept( tcps, (struct sockaddr *) &from,
 247                             &len )) == -1 )
 248                         {
 249                                 int err = errno;
 250                                 Debug( LDAP_DEBUG_ANY,
 251                                     "accept() failed errno %d (%s)", err,
 252                                     err > -1 && err < sys_nerr ?
 253                                     sys_errlist[err] : "unknown", 0 );
 254                                 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
 255                                 continue;
 256                         }
 257                         if ( ioctl( ns, FIONBIO, (caddr_t) &on ) == -1 ) {
 258                                 Debug( LDAP_DEBUG_ANY,
 259                                     "FIONBIO ioctl on %d failed\n", ns, 0, 0 );
 260                         }
 261
 262                         Debug( LDAP_DEBUG_CONNS, "new connection on %d\n", ns,
 263                             0, 0 );
 264
 265                         len = sizeof(from);
 266
 267                         if ( getpeername( ns, (struct sockaddr *) &from, &len )
 268                             == 0 ) {
 269                                 char *s;
 270                                 client_addr = inet_ntoa( from.sin_addr );
 271
 272 #if defined(SLAPD_RLOOKUPS) || defined(HAVE_TCPD)
 273                                 hp = gethostbyaddr( (char *)
 274                                     &(from.sin_addr.s_addr),
 275                                     sizeof(from.sin_addr.s_addr), AF_INET );
 276
 277                                 if(hp) {
 278                                         client_name = hp->h_name;
 279
 280                                         /* normalize the domain */
 281                                         for ( s = client_name; *s; s++ ) {
 282                                                 *s = TOLOWER( (unsigned char) *s );
 283                                         }
 284
 285                                 } else {
 286                                         client_name = NULL;
 287                                 }
 288 #else
 289                                 client_name = NULL;
 290 #endif
 291
 292                         } else {
 293                                 client_name = NULL;;
 294                                 client_addr = NULL;
 295                         }
 296
 297 #ifdef HAVE_TCPD
 298                         if(!hosts_ctl("slapd",
 299                                 client_name != NULL ? client_name : STRING_UNKNOWN,
 300                                 client_addr != NULL ? client_addr : STRING_UNKNOWN,
 301                                 STRING_UNKNOWN))
 302                         {
 303                                 /* DENY ACCESS */
 304                                 Statslog( LDAP_DEBUG_ANY,
 305                                  "fd=%d connection from %s (%s) denied.\n",
 306                                         ns,
 307                                                 client_name == NULL ? "unknown" : client_name,
 308                                                 client_addr == NULL ? "unknown" : client_addr,
 309                                   0, 0 );
 310
 311                                 close(ns);
 312                                 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
 313                                 continue;
 314                         }
 315 #endif /* HAVE_TCPD */
 316
 317                         c[ns].c_sb.sb_sd = ns;
 318                         ldap_pvt_thread_mutex_lock( &ops_mutex );
 319                         c[ns].c_connid = num_conns++;
 320                         ldap_pvt_thread_mutex_unlock( &ops_mutex );
 321
 322                         Statslog( LDAP_DEBUG_STATS,
 323                             "conn=%d fd=%d connection from %s (%s) accepted.\n",
 324                                 c[ns].c_connid, ns,
 325                                         client_name == NULL ? "unknown" : client_name,
 326                                         client_addr == NULL ? "unknown" : client_addr,
 327                              0 );
 328
 329                         if ( c[ns].c_addr != NULL ) {
 330                                 free( c[ns].c_addr );
 331                         }
 332                         c[ns].c_addr = ch_strdup( client_addr );
 333
 334                         if ( c[ns].c_domain != NULL ) {
 335                                 free( c[ns].c_domain );
 336                         }
 337
 338                         c[ns].c_domain = ch_strdup( client_name == NULL
 339                                 ? "" : client_name );
 340
 341                         ldap_pvt_thread_mutex_lock( &c[ns].c_dnmutex );
 342                         if ( c[ns].c_dn != NULL ) {
 343                                 free( c[ns].c_dn );
 344                                 c[ns].c_dn = NULL;
 345                         }
 346                         if ( c[ns].c_cdn != NULL ) {
 347                                 free( c[ns].c_cdn );
 348                                 c[ns].c_cdn = NULL;
 349                         }
 350                         ldap_pvt_thread_mutex_unlock( &c[ns].c_dnmutex );
 351                         c[ns].c_starttime = currenttime;
 352                         c[ns].c_opsinitiated = 0;
 353                         c[ns].c_opscompleted = 0;
 354                 }
 355                 ldap_pvt_thread_mutex_unlock( &new_conn_mutex );
 356
 357                 Debug( LDAP_DEBUG_CONNS, "activity on:", 0, 0, 0 );
 358                 for ( i = 0; i < dtblsize; i++ ) {
 359                         int     r, w;
 360
 361                         r = FD_ISSET( i, &readfds );
 362                         w = FD_ISSET( i, &writefds );
 363                         if ( i != tcps && (r || w) ) {
 364                                 Debug( LDAP_DEBUG_CONNS, " %d%s%s", i,
 365                                     r ? "r" : "", w ? "w" : "" );
 366                         }
 367                 }
 368                 Debug( LDAP_DEBUG_CONNS, "\n", 0, 0, 0 );
 369
 370                 for ( i = 0; i < dtblsize; i++ ) {
 371                         if ( i == tcps || (! FD_ISSET( i, &readfds ) &&
 372                             ! FD_ISSET( i, &writefds )) ) {
 373                                 continue;
 374                         }
 375
 376                         if ( FD_ISSET( i, &writefds ) ) {
 377                                 Debug( LDAP_DEBUG_CONNS,
 378                                     "signaling write waiter on %d\n", i, 0, 0 );
 379
 380                                 ldap_pvt_thread_mutex_lock( &active_threads_mutex );
 381                                 active_threads++;
 382                                 c[i].c_writewaiter = 0;
 383                                 ldap_pvt_thread_cond_signal( &c[i].c_wcv );
 384                                 ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
 385                         }
 386
 387                         if ( FD_ISSET( i, &readfds ) ) {
 388                                 Debug( LDAP_DEBUG_CONNS,
 389                                     "read activity on %d\n", i, 0, 0 );
 390
 391                                 connection_activity( &c[i] );
 392                         }
 393                 }
 394
 395                 ldap_pvt_thread_yield();
 396         }
 397
 398         close( tcps );
 399
 400         ldap_pvt_thread_mutex_lock( &active_threads_mutex );
 401         Debug( LDAP_DEBUG_ANY,
 402             "slapd shutting down - waiting for %d threads to terminate\n",
 403             active_threads, 0, 0 );
 404         while ( active_threads > 0 ) {
 405                 ldap_pvt_thread_cond_wait(&active_threads_cond, &active_threads_mutex);
 406         }
 407         ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
 408
 409         /* let backends do whatever cleanup they need to do */
 410         Debug( LDAP_DEBUG_TRACE,
 411             "slapd shutting down - waiting for backends to close down\n", 0, 0,
 412             0 );
 413         be_close();
 414         Debug( LDAP_DEBUG_ANY, "slapd stopping\n", 0, 0, 0 );
 415
 416         return NULL;
 417 }
 418
 419 void
 420 slap_set_shutdown( int sig )
 421 {
 422         slapd_shutdown = 1;
 423         ldap_pvt_thread_kill( listener_tid, LDAP_SIGUSR1 );
 424
 425         /* reinstall self */
 426         (void) SIGNAL( sig, slap_set_shutdown );
 427 }
 428
 429 void
 430 slap_do_nothing( int sig )
 431 {
 432         /* reinstall self */
 433         (void) SIGNAL( sig, slap_do_nothing );
 434 }