X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=servers%2Fslapd%2Fmain.c;h=6d310de8c3bd5c1a8e8bf2d9e3c9d3bd969c6742;hb=18393e882cd89e461fc804b8ef58817027e465a1;hp=085e7184fa789282214184d06af4c119bb30d43c;hpb=dc07e765f263ef459dcd2afd1ece01cfc85a0edd;p=openldap

diff --git a/servers/slapd/main.c b/servers/slapd/main.c
index 085e7184fa..6d310de8c3 100644
--- a/servers/slapd/main.c
+++ b/servers/slapd/main.c
@@ -11,21 +11,51 @@
 #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, char *urls );
+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
+
 /*
  * 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 Standalone LDAP Server (slapd)";
+const char Versionstr[] =
+	OPENLDAP_PACKAGE " " OPENLDAP_VERSION " Standalone LDAP Server (slapd)";
 #endif
 
 #ifdef LOG_LOCAL4
@@ -60,58 +90,111 @@ static int   cnvt_str2int( char *, STRDISP_P, int );
 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]" );
+	fprintf( stderr,
+		"usage: %s options\n", name );
+	fprintf( stderr,
 #if LDAP_CONNECTIONLESS
-	fprintf( stderr, " [-c]" );
+		"\t-c\t\tEnable (experimental) Connectionless LDAP\n"
 #endif
-#ifdef SLAPD_BDB2
-    fprintf( stderr, " [-t]" );
+		"\t-d level\tDebug Level" "\n"
+		"\t-f filename\tConfiguration File\n"
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+		"\t-g group\tGroup (id or name) to ran as\n"
 #endif
+		"\t-h URLs\tList of URLs to serve"
 #ifdef LOG_LOCAL4
-    fprintf( stderr, " [-l sysloguser]" );
+		"\t-l sysloguser\tSyslog User (default: LOCAL4)\n"
+#endif
+#ifdef HAVE_WINSOCK
+		"\t-n NTserviceName\tNT service name\n"
+#endif
+
+		"\t-p port\tLDAP Port\n"
+#ifdef HAVE_TLS
+		"\t-P port\tLDAP over TLS Port\n"
+#endif
+		"\t-s level\tSyslog Level\n"
+#ifdef SLAPD_BDB2
+		"\t-t\t\tEnable BDB2 timing\n"
 #endif
 #if defined(HAVE_SETUID) && defined(HAVE_SETGID)
-    fprintf( stderr, " [-u user] [-g group]" );
+		"\t-u user\tUser (id or name) to ran as\n"
 #endif
-    fprintf( stderr, "\n" );
+    );
 }
 
-time_t starttime;
-struct sockaddr_in	bind_addr;
-
-int
-main( int argc, char **argv )
+#ifdef HAVE_WINSOCK
+void WINAPI ServiceMain( DWORD argc, LPTSTR *argv )
+#else
+int main( int argc, char **argv )
+#endif
 {
 	int		i;
-	int		inetd = 0;
 	int		rc;
-	int		tcps;
-#ifdef LDAP_CONNECTIONLESS
-	int		udp;
+	char *urls = NULL;
+#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
+	char *username = NULL;
+	char *groupname = NULL;
 #endif
 #ifdef LOG_LOCAL4
     int     syslogUser = DEFAULT_SYSLOG_USER;
 #endif
-#if defined(HAVE_SETUID) && defined(HAVE_SETGID)
-	char		*username = NULL, *groupname = NULL;
+#ifdef HAVE_WINSOCK
+	char        *NTservice  = SERVICE_NAME;
+	char		*configfile = ".\\slapd.conf";
+#else
+	char		*configfile = SLAPD_DEFAULT_CONFIGFILE;
 #endif
-	char		*configfile;
 	char        *serverName;
 	int         serverMode = SLAP_SERVER_MODE;
 
-	configfile = SLAPD_DEFAULT_CONFIGFILE;
-
-	(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(LDAP_PORT);
+	int port = LDAP_PORT;
+#ifdef HAVE_TLS
+	int tls_port = LDAPS_PORT;
+#else
+	int tls_port = 0;
+#endif
 
 	g_argc = argc;
 	g_argv = argv;
 
+#ifdef HAVE_WINSOCK
+	{
+		int *i;
+		char *newConfigFile;
+		ldap_debug = 0xffff;
+		if ( is_NT_Service ) CommenceStartupProcessing( NTservice, slap_set_shutdown );
+		i = (int*)getRegParam( NULL, "Port" );
+		if ( i != NULL )
+		{
+			port = *i;
+			Debug ( LDAP_DEBUG_ANY, "new port from registry is: %d\n", port, 0, 0 );
+		}
+#ifdef HAVE_TLS
+		i = (int*)getRegParam( NULL, "TLSPort" );
+		if ( i != NULL )
+		{
+			tls_port = *i;
+			Debug ( LDAP_DEBUG_ANY, "new TLS port from registry is: %d\n", tls_port, 0, 0 );
+		}
+#endif
+		i = (int*)getRegParam( NULL, "DebugLevel" );
+		if ( i != NULL ) 
+		{
+			slap_debug = *i;
+			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:"
+			     "d:f:h:p:s:u"
 #ifdef LOG_LOCAL4
 			     "l:"
 #endif
@@ -124,53 +207,21 @@ main( int argc, char **argv )
 #ifdef LDAP_CONNECTIONLESS
 				 "c"
 #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))
+				 "n:"
 #endif
-			{
-				fprintf(stderr, "invalid address (%s) for -a option", optarg);
-			}
+#ifdef HAVE_TLS
+			     "P:"
+#endif
+			     )) != EOF ) {
+		switch ( i ) {
+		case 'h':	/* listen URLs */
+			urls = ch_strdup( optarg );
             break;
 
 #ifdef LDAP_DEBUG
 		case 'd':	/* turn on debugging */
-			if ( optarg[0] == '?' ) {
-				printf( "Debug levels:\n" );
-				printf( "\tLDAP_DEBUG_TRACE\t%d\n",
-				    LDAP_DEBUG_TRACE );
-				printf( "\tLDAP_DEBUG_PACKETS\t%d\n",
-				    LDAP_DEBUG_PACKETS );
-				printf( "\tLDAP_DEBUG_ARGS\t\t%d\n",
-				    LDAP_DEBUG_ARGS );
-				printf( "\tLDAP_DEBUG_CONNS\t%d\n",
-				    LDAP_DEBUG_CONNS );
-				printf( "\tLDAP_DEBUG_BER\t\t%d\n",
-				    LDAP_DEBUG_BER );
-				printf( "\tLDAP_DEBUG_FILTER\t%d\n",
-				    LDAP_DEBUG_FILTER );
-				printf( "\tLDAP_DEBUG_CONFIG\t%d\n",
-				    LDAP_DEBUG_CONFIG );
-				printf( "\tLDAP_DEBUG_ACL\t\t%d\n",
-				    LDAP_DEBUG_ACL );
-				printf( "\tLDAP_DEBUG_STATS\t%d\n",
-				    LDAP_DEBUG_STATS );
-				printf( "\tLDAP_DEBUG_STATS2\t%d\n",
-				    LDAP_DEBUG_STATS2 );
-				printf( "\tLDAP_DEBUG_SHELL\t%d\n",
-				    LDAP_DEBUG_SHELL );
-				printf( "\tLDAP_DEBUG_PARSE\t%d\n",
-				    LDAP_DEBUG_PARSE );
-				printf( "\tLDAP_DEBUG_ANY\t\t%d\n",
-				    LDAP_DEBUG_ANY );
-				exit( 0 );
-			} else {
-				slap_debug |= atoi( optarg );
-			}
+			slap_debug |= atoi( optarg );
 			break;
 #else
 		case 'd':	/* turn on debugging */
@@ -183,18 +234,29 @@ main( int argc, char **argv )
 			configfile = ch_strdup( optarg );
 			break;
 
-		case 'i':	/* run from inetd */
-			inetd = 1;
-			break;
-
 		case 'p': {	/* port on which to listen */
-				short port = (short)atoi( optarg );
-				if(! port ) {
+				int p = atoi( optarg );
+				if(! p ) {
 					fprintf(stderr, "-p %s must be numeric\n", optarg);
+				} else if( p < 0 || p >= 1<<16) {
+					fprintf(stderr, "-p %s invalid\n", optarg);
+				} else {
+					port = p;
+				}
+			} break;
+
+#ifdef HAVE_TLS
+		case 'P': {	/* port on which to listen for TLS */
+				int p = atoi( optarg );
+				if(! p ) {
+					fprintf(stderr, "-P %s must be numeric\n", optarg);
+				} else if( p < 0 || p >= 1<<16) {
+					fprintf(stderr, "-P %s invalid\n", optarg);
 				} else {
-					bind_addr.sin_port = htons(port);
+					tls_port = p;
 				}
 			} break;
+#endif
 
 		case 's':	/* set syslog level */
 			ldap_syslog = atoi( optarg );
@@ -202,8 +264,8 @@ main( int argc, char **argv )
 
 #ifdef LOG_LOCAL4
 		case 'l':	/* set syslog local user */
-			syslogUser = cnvt_str2int( optarg, syslog_types,
-                                           DEFAULT_SYSLOG_USER );
+			syslogUser = cnvt_str2int( optarg,
+				syslog_types, DEFAULT_SYSLOG_USER );
 			break;
 #endif
 
@@ -231,9 +293,16 @@ main( int argc, char **argv )
 			break;
 #endif /* SETUID && GETUID */
 
+#ifdef HAVE_WINSOCK
+		case 'n':  /* NT service name */
+			NTservice = ch_strdup( optarg );
+			break;
+#endif
 		default:
 			usage( argv[0] );
-			exit( 1 );
+			rc = 1;
+			SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 15 );
+			goto stop;
 		}
 	}
 
@@ -243,7 +312,7 @@ main( int argc, char **argv )
 
 	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 );
@@ -255,23 +324,35 @@ main( int argc, char **argv )
 	openlog( serverName, OPENLOG_OPTIONS );
 #endif
 
-	tcps = set_socket( inetd ? NULL : &bind_addr );
+	if( slapd_daemon_init( urls, port, tls_port ) != 0 ) {
+		rc = 1;
+		SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
+		goto stop;
+	}
 
 #if defined(HAVE_SETUID) && defined(HAVE_SETGID)
-	if ( username != NULL || groupname != NULL )
+	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
+
 	(void) SIGNAL( LDAP_SIGUSR1, slap_do_nothing );
 	(void) SIGNAL( LDAP_SIGUSR2, slap_set_shutdown );
 #ifdef SIGPIPE
@@ -285,23 +366,26 @@ main( int argc, char **argv )
 #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 WIN32
-	if(!inetd) {
+#ifndef HAVE_WINSOCK
 #ifdef LDAP_DEBUG
 		lutil_detach( ldap_debug, 0 );
 #else
 		lutil_detach( 0, 0 );
 #endif
-	}
-#endif /* WIN32 */
+#endif /* HAVE_WINSOCK */
 
 	if ( slap_startup(-1)  != 0 ) {
 		rc = 1;
+		SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
 		goto shutdown;
 	}
 
-	if(!inetd) {
+	{
 		FILE *fp;
 
 		Debug( LDAP_DEBUG_ANY, "slapd starting\n", 0, 0, 0 );
@@ -324,9 +408,17 @@ main( int argc, char **argv )
 		}
 	}
 
-	time( &starttime );
+#ifdef HAVE_WINSOCK
+	LogSlapdStartedEvent( NTservice, slap_debug, configfile, urls );
+#endif
 
-	rc = slapd_daemon( inetd, tcps );
+	rc = slapd_daemon();
+
+#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
 
 shutdown:
 	/* remember an error during shutdown */
@@ -335,11 +427,21 @@ 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 );
 
-	closelog();
+#ifdef HAVE_WINSOCK
+	ReportSlapdShutdownComplete();
+#endif
+
+    closelog();
+	slapd_daemon_destroy();
 
-	return rc;
+	MAIN_RETURN(rc);
 }
 
 
@@ -397,7 +499,6 @@ cnvt_str2int( char *stringVal, STRDISP_P dispatcher, int defaultVal )
     }
 
     return (retVal);
-
-} /* cnvt_str2int */
+}
 
 #endif  /* LOG_LOCAL4 */