]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/main.c
fix substring_comp_candidates logic if intersection of candidates
[openldap] / servers / slapd / main.c
index afc42a316ad1a53d18cb1c685bfcb536da8bedda..d3972ca0f1a0c0088d111bb97262a9d5c05c60c4 100644 (file)
 
 #include <stdio.h>
 
+#include <ac/signal.h>
 #include <ac/socket.h>
 #include <ac/string.h>
 #include <ac/time.h>
 #include <ac/unistd.h>
+#include <ac/wait.h>
+#include <ac/signal.h>
+#include <ac/errno.h>
 
-#include "ldapconfig.h"
+#include "ldap_defaults.h"
 #include "slap.h"
-#include "lutil.h"                     /* Get lutil_detach() */
+#include "lutil.h"
 
+#ifdef LDAP_SIGCHLD
+static RETSIGTYPE wait4child( int sig );
+#endif
 
-/*
- * read-only global variables or variables only written by the listener
- * thread (after they are initialized) - no need to protect them with a mutex.
- */
-int            slap_debug = 0;
-
-#ifdef LDAP_DEBUG
-int            ldap_syslog = LDAP_DEBUG_STATS;
+#ifdef HAVE_WINSOCK
+#define MAIN_RETURN(x) return
+struct sockaddr_in     bind_addr;
+
+/* in nt_main.c */
+extern SERVICE_STATUS                  SLAPDServiceStatus;
+extern SERVICE_STATUS_HANDLE   hSLAPDServiceStatus;
+extern ldap_pvt_thread_cond_t  started_event,          stopped_event;
+extern int       is_NT_Service;
+
+void LogSlapdStartedEvent( char *svc, int slap_debug, char *configfile, short port, int udp );
+void LogSlapdStoppedEvent( char *svc );
+
+void CommenceStartupProcessing( LPCTSTR serviceName,
+                                                          void(*stopper)(int));
+void ReportSlapdShutdownComplete( void );
+void *getRegParam( char *svc, char *value );
+
+#define SERVICE_EXIT( e, n ) \
+               if ( is_NT_Service ) \
+{ \
+                       SLAPDServiceStatus.dwWin32ExitCode                              = e; \
+                       SLAPDServiceStatus.dwServiceSpecificExitCode    = n; \
+} 
 #else
-int            ldap_syslog;
+#define SERVICE_EXIT( e, n )
+#define MAIN_RETURN(x) return(x)
 #endif
 
-int            ldap_syslog_level = LOG_DEBUG;
-int            udp;
-char           *default_referral;
-char           *configfile;
-time_t         starttime;
-pthread_t      listener_tid;
-int            g_argc;
-char           **g_argv;
-/*
- * global variables that need mutex protection
- */
-time_t         currenttime;
-pthread_mutex_t        currenttime_mutex;
-pthread_mutex_t        strtok_mutex;
-int            active_threads;
-pthread_mutex_t        active_threads_mutex;
-pthread_mutex_t        new_conn_mutex;
-#ifdef SLAPD_CRYPT
-pthread_mutex_t crypt_mutex;
-#endif
-long           ops_initiated;
-long           ops_completed;
-int            num_conns;
-pthread_mutex_t        ops_mutex;
-long           num_entries_sent;
-long           num_bytes_sent;
-pthread_mutex_t        num_sent_mutex;
+short port = LDAP_PORT;
+#ifdef HAVE_TLS
+short tls_port = LDAP_TLS_PORT;
+#endif
 /*
- * these mutexes must be used when calling the entry2str()
- * routine since it returns a pointer to static data.
+ * when more than one slapd is running on one machine, each one might have
+ * it's own LOCAL for syslogging and must have its own pid/args files
  */
-pthread_mutex_t        entry2str_mutex;
-pthread_mutex_t        replog_mutex;
+
+#ifndef HAVE_MKVERSION
+const char Versionstr[] =
+       OPENLDAP_PACKAGE " " OPENLDAP_VERSION " Standalone LDAP Server (slapd)";
+#endif
+
+#ifdef LOG_LOCAL4
+
+#define DEFAULT_SYSLOG_USER  LOG_LOCAL4
+
+typedef struct _str2intDispatch {
+       char    *stringVal;
+       int      abbr;
+       int      intVal;
+} STRDISP, *STRDISP_P;
+
+
+/* table to compute syslog-options to integer */
+static STRDISP  syslog_types[] = {
+    { "LOCAL0",         6, LOG_LOCAL0 },
+    { "LOCAL1",         6, LOG_LOCAL1 },
+    { "LOCAL2",         6, LOG_LOCAL2 },
+    { "LOCAL3",         6, LOG_LOCAL3 },
+    { "LOCAL4",         6, LOG_LOCAL4 },
+    { "LOCAL5",         6, LOG_LOCAL5 },
+    { "LOCAL6",         6, LOG_LOCAL6 },
+    { "LOCAL7",         6, LOG_LOCAL7 },
+    { NULL }
+};
+
+static int   cnvt_str2int( char *, STRDISP_P, int );
+
+#endif  /* LOG_LOCAL4 */
+
 
 static void
 usage( char *name )
 {
-       fprintf( stderr, "usage: %s [-d ?|debuglevel] [-f configfile] [-p portnumber] [-s sysloglevel]\n", name );
+       fprintf( stderr, "usage: %s [-d ?|debuglevel] [-f configfile] [-p portnumber] [-s sysloglevel]", name );
+    fprintf( stderr, "\n        [-a bind-address] [-i] [-u]" );
+#ifdef HAVE_WINSOCK
+       fprintf( stderr, " [-n NTserviceName]" );
+#endif
+#if LDAP_CONNECTIONLESS
+       fprintf( stderr, " [-c]" );
+#endif
+#ifdef SLAPD_BDB2
+    fprintf( stderr, " [-t]" );
+#endif
+#ifdef LOG_LOCAL4
+    fprintf( stderr, " [-l sysloguser]" );
+#endif
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+    fprintf( stderr, " [-u user] [-g group]" );
+#endif
+    fprintf( stderr, "\n" );
 }
 
-int
-main( int argc, char **argv )
+time_t starttime;
+struct sockaddr_in     bind_addr;
+ber_int_t tcps;
+#ifdef HAVE_TLS
+struct sockaddr_in     tls_bind_addr;
+ber_int_t tls_tcps;
+#endif
+
+#ifdef HAVE_WINSOCK
+void WINAPI ServiceMain( DWORD argc, LPTSTR *argv )
+#else
+int main( int argc, char **argv )
+#endif
 {
        int             i;
        int             inetd = 0;
-       int             port;
-       char            *myname;
-       Backend         *be = NULL;
-       FILE            *fp = NULL;
+       int             rc;
+       struct slapd_args args;
+       int             udp;
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+       char *username = NULL;
+       char *groupname = NULL;
+#endif
+#ifdef LOG_LOCAL4
+    int     syslogUser = DEFAULT_SYSLOG_USER;
+#endif
+#ifdef HAVE_WINSOCK
+       char        *NTservice  = SERVICE_NAME;
+       char            *configfile = ".\\slapd.conf";
+#else
+       char            *configfile = SLAPD_DEFAULT_CONFIGFILE;
+#endif
+       char        *serverName;
+       int         serverMode = SLAP_SERVER_MODE;
+       int             use_tls_port = 0;
+
+       (void) memset( (void*) &bind_addr, '\0', sizeof(bind_addr));
+       bind_addr.sin_family = AF_INET;
+       bind_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+       bind_addr.sin_port = htons(port);
+#ifdef HAVE_TLS
+       tls_bind_addr.sin_family = AF_INET;
+       tls_bind_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+       tls_bind_addr.sin_port = htons(tls_port);
+#endif
 
-       configfile = SLAPD_DEFAULT_CONFIGFILE;
-       port = LDAP_PORT;
        g_argc = argc;
        g_argv = argv;
 
-       while ( (i = getopt( argc, argv, "d:f:ip:s:u" )) != EOF ) {
+#ifdef HAVE_WINSOCK
+       {
+               int *newPort;
+               int *newDebugLevel;
+               char *newConfigFile;
+               ldap_debug = 0xffff;
+               if ( is_NT_Service ) CommenceStartupProcessing( NTservice, slap_set_shutdown );
+               newPort = (int*)getRegParam( NULL, "Port" );
+               if ( newPort != NULL )
+               {
+                       port = *newPort;
+                       bind_addr.sin_port = htons(port);
+                       Debug ( LDAP_DEBUG_ANY, "new port from registry is: %d\n", port, 0, 0 );
+               }
+               newDebugLevel = (int*)getRegParam( NULL, "DebugLevel" );
+               if ( newDebugLevel != NULL ) 
+               {
+                       slap_debug = *newDebugLevel;
+                       Debug( LDAP_DEBUG_ANY, "new debug level from registry is: %d\n", slap_debug, 0, 0 );
+               }
+               newConfigFile = (char*)getRegParam( NULL, "ConfigFile" );
+               if ( newConfigFile != NULL ) 
+               {
+                       configfile = newConfigFile;
+                       Debug ( LDAP_DEBUG_ANY, "new config file from registry is: %s\n", configfile, 0, 0 );
+               }
+       }
+#endif
+
+       while ( (i = getopt( argc, argv,
+                            "d:f:ia:p:s:u"
+#ifdef LOG_LOCAL4
+                            "l:"
+#endif
+#ifdef SLAPD_BDB2
+                            "t"
+#endif
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+                            "u:g:"
+#endif
+#ifdef LDAP_CONNECTIONLESS
+                                "c"
+#endif
+#ifdef HAVE_WINSOCK
+                                "n:"
+#endif
+#ifdef HAVE_TLS
+                            "P:T"
+#endif
+                            )) != EOF ) {
                switch ( i ) {
+               case 'a':       /* bind address */
+#ifdef HAVE_WINSOCK
+                       if(!(bind_addr.sin_addr.S_un.S_addr = inet_addr(optarg)))
+#else
+                       if(!inet_aton(optarg, &bind_addr.sin_addr))
+#endif
+                       {
+                               fprintf(stderr, "invalid address (%s) for -a option", optarg);
+                       }
+#ifdef HAVE_TLS
+#ifdef HAVE_WINSOCK
+                       tls_bind_addr.sin_addr.S_un.S_addr = inet_addr(optarg);
+#else
+                       inet_aton(optarg, &tls_bind_addr.sin_addr);
+#endif
+#endif
+            break;
+
 #ifdef LDAP_DEBUG
                case 'd':       /* turn on debugging */
                        if ( optarg[0] == '?' ) {
@@ -131,152 +282,293 @@ main( int argc, char **argv )
                        inetd = 1;
                        break;
 
-               case 'p':       /* port on which to listen */
-                       port = atoi( optarg );
-                       break;
+               case 'p': {     /* port on which to listen */
+                               port = (short)atoi( optarg );
+                               if(! port ) {
+                                       fprintf(stderr, "-p %s must be numeric\n", optarg);
+                               } else {
+                                       bind_addr.sin_port = htons(port);
+                               }
+                       } break;
+
+#ifdef HAVE_TLS
+               case 'P': {     /* port on which to listen for TLS */
+                               tls_port = (short)atoi( optarg );
+                               if(! tls_port ) {
+                                       fprintf(stderr, "-P %s must be numeric\n", optarg);
+                               } else {
+                                       tls_bind_addr.sin_port = htons(tls_port);
+                               }
+                       } break;
+#endif
 
                case 's':       /* set syslog level */
                        ldap_syslog = atoi( optarg );
                        break;
 
-               case 'u':       /* do udp */
+#ifdef LOG_LOCAL4
+               case 'l':       /* set syslog local user */
+                       syslogUser = cnvt_str2int( optarg, syslog_types,
+                                           DEFAULT_SYSLOG_USER );
+                       break;
+#endif
+
+#ifdef LDAP_CONNECTIONLESS
+               case 'c':       /* do connectionless (udp) */
                        udp = 1;
                        break;
+#endif
+
+#ifdef SLAPD_BDB2
+               case 't':  /* timed server */
+                       serverMode = SLAP_TIMEDSERVER_MODE;
+                       break;
+#endif
 
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+               case 'u':       /* user name */
+                       if( username ) free(username);
+                       username = ch_strdup( optarg );
+                       break;
+
+               case 'g':       /* group name */
+                       if( groupname ) free(groupname);
+                       groupname = ch_strdup( optarg );
+                       break;
+#endif /* SETUID && GETUID */
+#ifdef HAVE_WINSOCK
+               case 'n':  /* NT service name */
+                       NTservice = ch_strdup( optarg );
+                       break;
+#endif
+#ifdef HAVE_TLS
+               case 'T':  /* Bind on TLS port */
+                       use_tls_port = 1;
+                       break;
+#endif
                default:
                        usage( argv[0] );
-                       exit( 1 );
+                       rc = 1;
+                       SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 15 );
+                       goto stop;
                }
        }
 
-       lber_set_option(NULL, LBER_OPT_DEBUG_LEVEL, &slap_debug);
+       ber_set_option(NULL, LBER_OPT_DEBUG_LEVEL, &slap_debug);
        ldap_set_option(NULL, LDAP_OPT_DEBUG_LEVEL, &slap_debug);
        ldif_debug = slap_debug;
 
        Debug( LDAP_DEBUG_TRACE, "%s", Versionstr, 0, 0 );
 
-       if ( (myname = strrchr( argv[0], '/' )) == NULL ) {
-               myname = ch_strdup( argv[0] );
+       if ( (serverName = strrchr( argv[0], *LDAP_DIRSEP )) == NULL ) {
+               serverName = ch_strdup( argv[0] );
        } else {
-               myname = ch_strdup( myname + 1 );
+               serverName = ch_strdup( serverName + 1 );
        }
 
-       if ( ! inetd ) {
-               /* pre-open config file before detach in case it is a relative path */
-               fp = fopen( configfile, "r" );
+#ifdef LOG_LOCAL4
+       openlog( serverName, OPENLOG_OPTIONS, syslogUser );
+#else
+       openlog( serverName, OPENLOG_OPTIONS );
+#endif
+
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+       if ( username != NULL || groupname != NULL )
+               slap_init_user( username, groupname );
+#endif
+
+       if ( slap_init( serverMode, serverName ) != 0 ) {
+               rc = 1;
+               SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 18 );
+               goto destroy;
+       }
+
+       if ( read_config( configfile ) != 0 ) {
+               rc = 1;
+               SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 19 );
+               goto destroy;
+       }
+
+#ifdef HAVE_TLS
+       ldap_pvt_tls_init();
+       ldap_pvt_tls_init_def_ctx();
+#endif
+
+       tcps = set_socket( inetd ? NULL : &bind_addr );
+       if ( tcps == -1 )
+               goto destroy;
+#ifdef HAVE_TLS
+       if ( use_tls_port ) {
+               tls_tcps = set_socket( inetd ? NULL : &tls_bind_addr );
+               if ( tls_tcps == -1 )
+                       goto destroy;
+       } else {
+               tls_tcps = -1;
+       }
+#endif
+
+       (void) SIGNAL( LDAP_SIGUSR1, slap_do_nothing );
+       (void) SIGNAL( LDAP_SIGUSR2, slap_set_shutdown );
+#ifdef SIGPIPE
+       (void) SIGNAL( SIGPIPE, SIG_IGN );
+#endif
+#ifdef SIGHUP
+       (void) SIGNAL( SIGHUP, slap_set_shutdown );
+#endif
+       (void) SIGNAL( SIGINT, slap_set_shutdown );
+       (void) SIGNAL( SIGTERM, slap_set_shutdown );
+#ifdef LDAP_SIGCHLD
+       (void) SIGNAL( LDAP_SIGCHLD, wait4child );
+#endif
+#ifdef SIGBREAK
+       /* SIGBREAK is generated when Ctrl-Break is pressed. */
+       (void) SIGNAL( SIGBREAK, slap_set_shutdown );
+#endif
+
+#ifndef HAVE_WINSOCK
+       if(!inetd) {
 #ifdef LDAP_DEBUG
                lutil_detach( ldap_debug, 0 );
 #else
                lutil_detach( 0, 0 );
 #endif
        }
-#ifdef LOG_LOCAL4
-       openlog( myname, OPENLOG_OPTIONS, LOG_LOCAL4 );
-#else
-       openlog( myname, OPENLOG_OPTIONS );
-#endif
+#endif /* HAVE_WINSOC */
+
+       if ( slap_startup(-1)  != 0 ) {
+               rc = 1;
+               SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
+               goto shutdown;
+       }
+
+       if(!inetd) {
+               FILE *fp;
 
-       init();
-       read_config( configfile, &be, fp );
+               args.addr = &bind_addr;
+#ifdef HAVE_TLS
+               args.tls_addr = &tls_bind_addr;
+#endif
 
-       if ( ! inetd ) {
-               int             status;
+               Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
 
-               time( &starttime );
+               if (( slapd_pid_file != NULL ) &&
+                       (( fp = fopen( slapd_pid_file, "w" )) != NULL ))
+               {
+                       fprintf( fp, "%d\n", (int) getpid() );
+                       fclose( fp );
+               }
 
-               if ( pthread_create( &listener_tid, NULL, slapd_daemon,
-                   (void *) port ) != 0 ) {
-                       Debug( LDAP_DEBUG_ANY,
-                           "listener pthread_create failed\n", 0, 0, 0 );
-                       exit( 1 );
+               if (( slapd_args_file != NULL ) &&
+                       (( fp = fopen( slapd_args_file, "w" )) != NULL ))
+               {
+                       for ( i = 0; i < g_argc; i++ ) {
+                               fprintf( fp, "%s ", g_argv[i] );
+                       }
+                       fprintf( fp, "\n" );
+                       fclose( fp );
                }
 
-#ifdef HAVE_PHREADS_FINAL
-               pthread_join( listener_tid, (void *) NULL );
-#else
-               pthread_join( listener_tid, (void *) &status );
+       } else {
+               args.addr = NULL;
+#ifdef HAVE_TLS
+               args.tls_addr = NULL;
+#endif
+       }
+       args.tcps = tcps;
+#ifdef HAVE_TLS
+       args.tls_tcps = tls_tcps;
 #endif
 
-               return 0;
+       time( &starttime );
+#ifdef HAVE_WINSOCK
+       LogSlapdStartedEvent( NTservice, slap_debug, configfile, port, udp );
+#endif
 
-       } else {
-               Connection              c;
-               Operation               *o;
-               BerElement              ber;
-               unsigned long           len, tag;
-               long                    msgid;
-               int                     flen;
-               struct sockaddr_in      from;
-               struct hostent          *hp;
-
-               c.c_dn = NULL;
-               c.c_ops = NULL;
-               c.c_sb.sb_sd = 0;
-               c.c_sb.sb_options = 0;
-               c.c_sb.sb_naddr = udp ? 1 : 0;
-               c.c_sb.sb_ber.ber_buf = NULL;
-               c.c_sb.sb_ber.ber_ptr = NULL;
-               c.c_sb.sb_ber.ber_end = NULL;
-               pthread_mutex_init( &c.c_dnmutex, pthread_mutexattr_default );
-               pthread_mutex_init( &c.c_opsmutex, pthread_mutexattr_default );
-               pthread_mutex_init( &c.c_pdumutex, pthread_mutexattr_default );
-#ifdef notdefcldap
-               c.c_sb.sb_addrs = (void **) saddrlist;
-               c.c_sb.sb_fromaddr = &faddr;
-               c.c_sb.sb_useaddr = saddrlist[ 0 ] = &saddr;
-#endif
-               flen = sizeof(from);
-               if ( getpeername( 0, (struct sockaddr *) &from, &flen ) == 0 ) {
-#ifdef SLAPD_RLOOKUPS
-                       hp = gethostbyaddr( (char *) &(from.sin_addr.s_addr),
-                           sizeof(from.sin_addr.s_addr), AF_INET );
-#else
-                       hp = NULL;
+       rc = slapd_daemon( &args );
+
+#ifdef HAVE_WINSOCK
+       /* Throw away the event that we used during the startup process. */
+       if ( is_NT_Service )
+               ldap_pvt_thread_cond_destroy( &started_event );
 #endif
 
-                       Debug( LDAP_DEBUG_ARGS, "connection from %s (%s)\n",
-                           hp == NULL ? "unknown" : hp->h_name,
-                           inet_ntoa( from.sin_addr ), 0 );
+shutdown:
+       /* remember an error during shutdown */
+       rc |= slap_shutdown(-1);
+destroy:
+       /* remember an error during destroy */
+       rc |= slap_destroy();
 
-                       c.c_addr = inet_ntoa( from.sin_addr );
-                       c.c_domain = ch_strdup( hp == NULL ? "" : hp->h_name );
-               } else {
-                       Debug( LDAP_DEBUG_ARGS, "connection from unknown\n",
-                           0, 0, 0 );
-               }
+stop:
+#ifdef HAVE_WINSOCK
+       LogSlapdStoppedEvent( NTservice );
+#endif
+       Debug( LDAP_DEBUG_ANY, "slapd stopped.\n", 0, 0, 0 );
+#ifdef HAVE_WINSOCK
+       ReportSlapdShutdownComplete();
+#endif
 
-               ber_init_w_nullc( &ber, 0 );
-
-               while ( (tag = ber_get_next( &c.c_sb, &len, &ber ))
-                   == LDAP_TAG_MESSAGE ) {
-                       pthread_mutex_lock( &currenttime_mutex );
-                       time( &currenttime );
-                       pthread_mutex_unlock( &currenttime_mutex );
-
-                       if ( (tag = ber_get_int( &ber, &msgid ))
-                           != LDAP_TAG_MSGID ) {
-                               /* log and send error */
-                               Debug( LDAP_DEBUG_ANY,
-                                  "ber_get_int returns 0x%lx\n", tag, 0, 0 );
-                               ber_free( &ber, 1 );
-                               return 1;
-                       }
+    closelog();
 
-                       if ( (tag = ber_peek_tag( &ber, &len ))
-                           == LBER_ERROR ) {
-                               /* log, close and send error */
-                               Debug( LDAP_DEBUG_ANY,
-                                  "ber_peek_tag returns 0x%lx\n", tag, 0, 0 );
-                               ber_free( &ber, 1 );
-                               close( c.c_sb.sb_sd );
-                               c.c_sb.sb_sd = -1;
-                               return 1;
-                       }
+       MAIN_RETURN(rc);
+}
 
-                       connection_activity( &c );
 
-                       ber_free( &ber, 1 );
-               }
-       }
-       return 1;
+#ifdef LDAP_SIGCHLD
+
+/*
+ *  Catch and discard terminated child processes, to avoid zombies.
+ */
+
+static RETSIGTYPE
+wait4child( int sig )
+{
+    int save_errno = errno;
+
+#ifdef WNOHANG
+    errno = 0;
+#ifdef HAVE_WAITPID
+    while ( waitpid( (pid_t)-1, NULL, WNOHANG ) >= 0 || errno == EINTR )
+       ;       /* NULL */
+#else
+    while ( wait3( NULL, WNOHANG, NULL ) >= 0 || errno == EINTR )
+       ;       /* NULL */
+#endif
+#else
+    (void) wait( NULL );
+#endif
+    (void) SIGNAL( sig, wait4child );
+    errno = save_errno;
 }
+
+#endif /* SIGCHLD || SIGCLD */
+
+
+#ifdef LOG_LOCAL4
+
+/*
+ *  Convert a string to an integer by means of a dispatcher table
+ *  if the string is not in the table return the default
+ */
+
+static int
+cnvt_str2int( char *stringVal, STRDISP_P dispatcher, int defaultVal )
+{
+    int        retVal = defaultVal;
+    STRDISP_P  disp;
+
+    for (disp = dispatcher; disp->stringVal; disp++) {
+
+        if (!strncasecmp (stringVal, disp->stringVal, disp->abbr)) {
+
+            retVal = disp->intVal;
+            break;
+
+        }
+    }
+
+    return (retVal);
+
+} /* cnvt_str2int */
+
+#endif  /* LOG_LOCAL4 */