]> 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 caa19f866f2411570e7d3f765c6b3ba958cf702f..d3972ca0f1a0c0088d111bb97262a9d5c05c60c4 100644 (file)
@@ -2,37 +2,79 @@
 
 #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
+
+#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
+#define SERVICE_EXIT( e, n )
+#define MAIN_RETURN(x) return(x)
+#endif
 
+short port = LDAP_PORT;
+#ifdef HAVE_TLS
+short tls_port = LDAP_TLS_PORT;
+#endif
 /*
  * 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
  */
 
+#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;
-
+       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 },
@@ -41,11 +83,10 @@ static STRDISP  syslog_types[] = {
     { "LOCAL5",         6, LOG_LOCAL5 },
     { "LOCAL6",         6, LOG_LOCAL6 },
     { "LOCAL7",         6, LOG_LOCAL7 },
-    NULL
-
+    { NULL }
 };
 
-static int   cnvt_str2int();
+static int   cnvt_str2int( char *, STRDISP_P, int );
 
 #endif  /* LOG_LOCAL4 */
 
@@ -54,34 +95,143 @@ static void
 usage( char *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;
+       int             rc;
+       struct slapd_args args;
        int             udp;
-       Backend         *be = NULL;
-       FILE            *fp = NULL;
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+       char *username = NULL;
+       char *groupname = NULL;
+#endif
 #ifdef LOG_LOCAL4
     int     syslogUser = DEFAULT_SYSLOG_USER;
 #endif
-       char            *configfile;
+#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] == '?' ) {
@@ -132,165 +282,268 @@ 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;
 
 #ifdef LOG_LOCAL4
-
                case 'l':       /* set syslog local user */
                        syslogUser = cnvt_str2int( optarg, syslog_types,
                                            DEFAULT_SYSLOG_USER );
                        break;
-
 #endif
 
-               case 'u':       /* do udp */
+#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 ( (serverName = strrchr( argv[0], '/' )) == NULL ) {
+       if ( (serverName = strrchr( argv[0], *LDAP_DIRSEP )) == NULL ) {
                serverName = ch_strdup( argv[0] );
        } else {
                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
        }
+#endif /* HAVE_WINSOC */
 
-#ifdef LOG_LOCAL4
-       openlog( serverName, OPENLOG_OPTIONS, syslogUser );
-#else
-       openlog( serverName, OPENLOG_OPTIONS );
-#endif
+       if ( slap_startup(-1)  != 0 ) {
+               rc = 1;
+               SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
+               goto shutdown;
+       }
 
-       init();
-       read_config( configfile, &be, fp );
+       if(!inetd) {
+               FILE *fp;
 
-       if ( ! inetd ) {
-               int             status;
+               args.addr = &bind_addr;
+#ifdef HAVE_TLS
+               args.tls_addr = &tls_bind_addr;
+#endif
 
-               time( &starttime );
+               Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
 
-               if ( status = ldap_pvt_thread_create( &listener_tid, 0,
-                       slapd_daemon, (void *) port ) != 0 )
+               if (( slapd_pid_file != NULL ) &&
+                       (( fp = fopen( slapd_pid_file, "w" )) != NULL ))
                {
-                       Debug( LDAP_DEBUG_ANY,
-                           "listener ldap_pvt_thread_create failed (%d)\n", status, 0, 0 );
-                       exit( 1 );
+                       fprintf( fp, "%d\n", (int) getpid() );
+                       fclose( fp );
                }
 
-               ldap_pvt_thread_join( listener_tid, (void *) NULL );
-
-               return 0;
+               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 );
+               }
 
        } 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_cdn = 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;
-               ldap_pvt_thread_mutex_init( &c.c_dnmutex );
-               ldap_pvt_thread_mutex_init( &c.c_opsmutex );
-               ldap_pvt_thread_mutex_init( &c.c_pdumutex );
-#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;
+               args.addr = NULL;
+#ifdef HAVE_TLS
+               args.tls_addr = NULL;
+#endif
+       }
+       args.tcps = tcps;
+#ifdef HAVE_TLS
+       args.tls_tcps = tls_tcps;
 #endif
 
-                       Debug( LDAP_DEBUG_ARGS, "connection from %s (%s)\n",
-                           hp == NULL ? "unknown" : hp->h_name,
-                           inet_ntoa( from.sin_addr ), 0 );
+       time( &starttime );
+#ifdef HAVE_WINSOCK
+       LogSlapdStartedEvent( NTservice, slap_debug, configfile, port, udp );
+#endif
 
-                       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 );
-               }
+       rc = slapd_daemon( &args );
 
-               ber_init_w_nullc( &ber, 0 );
-
-               while ( (tag = ber_get_next( &c.c_sb, &len, &ber ))
-                   == LDAP_TAG_MESSAGE ) {
-                       ldap_pvt_thread_mutex_lock( &currenttime_mutex );
-                       time( &currenttime );
-                       ldap_pvt_thread_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;
-                       }
+#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
 
-                       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;
-                       }
+shutdown:
+       /* remember an error during shutdown */
+       rc |= slap_shutdown(-1);
+destroy:
+       /* remember an error during destroy */
+       rc |= slap_destroy();
+
+stop:
+#ifdef HAVE_WINSOCK
+       LogSlapdStoppedEvent( NTservice );
+#endif
+       Debug( LDAP_DEBUG_ANY, "slapd stopped.\n", 0, 0, 0 );
+#ifdef HAVE_WINSOCK
+       ReportSlapdShutdownComplete();
+#endif
 
-                       connection_activity( &c );
+    closelog();
 
-                       ber_free( &ber, 1 );
-               }
-       }
-       return 1;
+       MAIN_RETURN(rc);
 }
 
 
+#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
 
 /*
@@ -299,10 +552,7 @@ main( int argc, char **argv )
  */
 
 static int
-cnvt_str2int (stringVal, dispatcher, defaultVal)
-char      *stringVal;
-STRDISP_P  dispatcher;
-int        defaultVal;
+cnvt_str2int( char *stringVal, STRDISP_P dispatcher, int defaultVal )
 {
     int        retVal = defaultVal;
     STRDISP_P  disp;
@@ -322,4 +572,3 @@ int        defaultVal;
 } /* cnvt_str2int */
 
 #endif  /* LOG_LOCAL4 */
-