-/*
- * Copyright (c) 1996 Regents of the University of Michigan.
+/* $OpenLDAP$ */
+/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
+ *
+ * Copyright 1998-2003 The OpenLDAP Foundation.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted only as authorized by the OpenLDAP
+ * Public License.
+ *
+ * A copy of this license is available in file LICENSE in the
+ * top-level directory of the distribution or, alternatively, at
+ * <http://www.OpenLDAP.org/license.html>.
+ */
+/* Portions Copyright (c) 1996 Regents of the University of Michigan.
* All rights reserved.
*
* Redistribution and use in source and binary forms are permitted
* software without specific prior written permission. This software
* is provided ``as is'' without express or implied warranty.
*/
+/* ACKNOWLEDGEMENTS:
+ * This work was originally developed by the University of Michigan
+ * (as part of U-MICH LDAP). Additional significant contributors
+ * include:
+ * Howard Chu
+ */
/*
#include "portable.h"
#include <stdio.h>
+#include <sys/stat.h>
+#include <ac/stdlib.h>
+#include <ac/unistd.h>
#include "slurp.h"
#include "globals.h"
+#include "lutil.h"
+#include <ldap_pvt.h>
-extern int doargs( int, char **, Globals * );
-extern void fm();
-extern int start_replica_thread( Ri * );
-extern Globals *init_globals();
-extern int sanity();
-
-#if defined( HAVE_LWP )
-extern void start_lwp_scheduler();
-#endif /* HAVE_LWP */
+#ifdef HAVE_NT_SERVICE_MANAGER
+#define MAIN_RETURN(x) return
+#define SERVICE_EXIT( e, n ) do { \
+ if ( is_NT_Service ) { \
+ lutil_ServiceStatus.dwWin32ExitCode = (e); \
+ lutil_ServiceStatus.dwServiceSpecificExitCode = (n); \
+ } \
+} while ( 0 )
+#else
+#define SERVICE_EXIT( e, n )
+#define MAIN_RETURN(x) return(x)
+#endif
-main(
- int argc,
- char **argv
-)
+#ifdef HAVE_NT_SERVICE_MANAGER
+void WINAPI ServiceMain( DWORD argc, LPTSTR *argv )
+#else
+int main( int argc, char **argv )
+#endif
{
- pthread_attr_t attr;
- int status;
- int i;
-
#ifdef NO_THREADS
/* Haven't yet written the non-threaded version */
- fprintf( stderr, "slurpd currently requires threads support\n" );
- exit( 1 );
+ fputs( "slurpd currently requires threads support\n", stderr );
+ return( 1 );
#else
+ int i, rc = 0;
+
+ /* initialize thread package */
+ ldap_pvt_thread_initialize();
+
/*
* Create and initialize globals. init_globals() also initializes
* the main replication queue.
*/
if (( sglob = init_globals()) == NULL ) {
fprintf( stderr, "Out of memory initializing globals\n" );
- exit( 1 );
+ SERVICE_EXIT( ERROR_NOT_ENOUGH_MEMORY, 0 );
+ rc = 1;
+ goto stop;
}
+#ifdef HAVE_NT_SERVICE_MANAGER
+ {
+ int *i;
+ char *newConfigFile;
+ char *regService = NULL;
+
+ if ( is_NT_Service ) {
+ sglob->serverName = argv[0];
+ lutil_CommenceStartupProcessing( sglob->serverName, slurp_set_shutdown );
+ if ( strcmp(sglob->serverName, SERVICE_NAME) )
+ regService = sglob->serverName;
+ }
+
+ i = (int*)lutil_getRegParam( regService, "DebugLevel" );
+ if ( i != NULL )
+ {
+ ldap_debug = *i;
+#ifdef NEW_LOGGING
+ lutil_log_initialize( argc, argv );
+ LDAP_LOG( SLURPD, INFO,
+ "main: new debug level from registry is: %d\n",
+ ldap_debug, 0, 0 );
+#else
+ Debug( LDAP_DEBUG_ANY, "new debug level from registry is: %d\n", ldap_debug, 0, 0 );
+#endif
+ }
+
+ newConfigFile = (char*)lutil_getRegParam( regService, "ConfigFile" );
+ if ( newConfigFile != NULL )
+ {
+ sglob->slapd_configfile = newConfigFile;
+#ifdef NEW_LOGGING
+ LDAP_LOG( SLURPD, INFO,
+ "main: new config file from registry is: %s\n", sglob->slapd_configfile, 0, 0 );
+#else
+ Debug ( LDAP_DEBUG_ANY, "new config file from registry is: %s\n", sglob->slapd_configfile, 0, 0 );
+#endif
+
+ }
+ }
+#endif
+
/*
* Process command-line args and fill in globals.
*/
if ( doargs( argc, argv, sglob ) < 0 ) {
- exit( 1 );
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 15 );
+ rc = 1;
+ goto stop;
}
/*
fprintf( stderr,
"Errors encountered while processing config file \"%s\"\n",
sglob->slapd_configfile );
- exit( 1 );
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 19 );
+ rc = 1;
+ goto stop;
+ }
+
+#ifdef HAVE_TLS
+ if( ldap_pvt_tls_init() || ldap_pvt_tls_init_def_ctx() ) {
+ fprintf( stderr, "TLS Initialization failed.\n" );
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
+ rc = 1;
+ goto stop;
+ }
+#endif
+
+ /*
+ * Make sure our directory exists
+ */
+ if ( mkdir(sglob->slurpd_rdir, 0755) == -1 && errno != EEXIST) {
+ perror(sglob->slurpd_rdir);
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
+ rc = 1;
+ goto stop;
}
/*
*/
if ( sglob->st->st_read( sglob->st )) {
fprintf( stderr, "Malformed slurpd status file \"%s\"\n",
- sglob->slurpd_status_file, 0, 0 );
- exit( 1 );
+ sglob->slurpd_status_file );
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 17 );
+ rc = 1;
+ goto stop;
}
/*
* Check for any fatal error conditions before we get started
*/
if ( sanity() < 0 ) {
- exit( 1 );
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 18 );
+ rc = 1;
+ goto stop;
}
- /*
- * Detach from the controlling terminal, if debug level = 0,
- * and if not in one-shot mode.
- */
-#ifdef LDAP_DEBUG
- if (( ldap_debug == 0 ) && !sglob->one_shot_mode )
-#else /* LDAP_DEBUG */
- if ( !sglob->one_shot_mode )
-#endif /* LDAP_DEBUG */
- {
- detach();
- }
-#if defined( HAVE_LWP )
/*
- * Need to start a scheduler thread under SunOS 4
+ * Detach from the controlling terminal
+ * unless the -d flag is given or in one-shot mode.
*/
- start_lwp_scheduler();
-#endif /* HAVE_LWP */
+#ifndef HAVE_WINSOCK
+ if ( ! (sglob->no_detach || sglob->one_shot_mode) ) {
+ lutil_detach( 0, 0 );
+ }
+#endif
+ if ( slurpd_pid_file != NULL ) {
+ FILE *fp = fopen( slurpd_pid_file, "w" );
- /*
- * Start threads - one thread for each replica
- */
- for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
- start_replica_thread( sglob->replicas[ i ]);
+ if( fp != NULL ) {
+ fprintf( fp, "%d\n", (int) getpid() );
+ fclose( fp );
+
+ } else {
+ free(slurpd_pid_file);
+ slurpd_pid_file = NULL;
+ }
+ }
+
+ if ( slurpd_args_file != NULL ) {
+ FILE *fp = fopen( slurpd_args_file, "w" );
+
+ if( fp != NULL ) {
+ for ( i = 0; i < argc; i++ ) {
+ fprintf( fp, "%s ", argv[i] );
+ }
+ fprintf( fp, "\n" );
+ fclose( fp );
+ } else {
+ free(slurpd_args_file);
+ slurpd_args_file = NULL;
+ }
+ }
+
+ if ( (rc = lutil_pair( sglob->wake_sds )) < 0 ) {
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
+ rc = 1;
+ goto stop;
}
+
+#ifdef HAVE_NT_EVENT_LOG
+ if (is_NT_Service) lutil_LogStartedEvent( sglob->serverName, ldap_debug, sglob->slapd_configfile, "n/a" );
+#endif
/*
* Start the main file manager thread (in fm.c).
*/
- pthread_attr_init( &attr );
-
-#if !defined(HAVE_PTHREADS_D4) && !defined(HAVE_DCE)
- /* POSIX_THREADS or compatible
- * This is a draft 10 or standard pthreads implementation
- */
- if ( pthread_create( &(sglob->fm_tid), &attr, fm, (void *) NULL )
- != 0 ) {
- Debug( LDAP_DEBUG_ANY, "file manager pthread_create failed\n",
+ if ( ldap_pvt_thread_create( &(sglob->fm_tid),
+ 0, fm, (void *) NULL ) != 0 )
+ {
+#ifdef NEW_LOGGING
+ LDAP_LOG ( SLURPD, ERR,
+ "main: file manager ldap_pvt_thread_create failed\n" , 0, 0, 0 );
+#else
+ Debug( LDAP_DEBUG_ANY, "file manager ldap_pvt_thread_create failed\n",
0, 0, 0 );
- exit( 1 );
+#endif
+ SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 21 );
+ rc = 1;
+ goto stop;
}
-#else /* !PTHREADS_FINAL */
+
/*
- * This is a draft 4 or earlier pthreads implementation
+ * wait for fm to finish if in oneshot mode
*/
- if ( pthread_create( &(sglob->fm_tid), attr, fm, (void *) NULL )
- != 0 ) {
- Debug( LDAP_DEBUG_ANY, "file manager pthread_create failed\n",
- 0, 0, 0 );
- exit( 1 );
+ if ( sglob->one_shot_mode ) {
+ ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
+ }
+ /*
+ * Start threads - one thread for each replica
+ */
+ for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
+ start_replica_thread( sglob->replicas[ i ]);
}
-#endif /* !PTHREADS_FINAL */
- pthread_attr_destroy( &attr );
+#ifdef HAVE_NT_SERVICE_MANAGER
+ if ( started_event ) ldap_pvt_thread_cond_signal( &started_event );
+#endif
/*
* Wait for the fm thread to finish.
*/
-#ifdef HAVE_PTHREADS_FINAL
- pthread_join( sglob->fm_tid, (void *) NULL );
-#else
- pthread_join( sglob->fm_tid, (void *) &status );
-#endif
+ if ( !sglob->one_shot_mode ) {
+ ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
+ }
+
/*
* Wait for the replica threads to finish.
*/
for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
-#ifdef HAVE_PTHREADS_FINAL
- pthread_join( sglob->replicas[ i ]->ri_tid, (void *) NULL );
+ ldap_pvt_thread_join( sglob->replicas[ i ]->ri_tid, (void *) NULL );
+ }
+
+stop:
+#ifdef HAVE_NT_SERVICE_MANAGER
+ if (is_NT_Service) {
+ ldap_pvt_thread_cond_destroy( &started_event );
+ lutil_LogStoppedEvent( sglob->serverName );
+ lutil_ReportShutdownComplete();
+ }
+#endif
+ /* destroy the thread package */
+ ldap_pvt_thread_destroy();
+
+#ifdef HAVE_TLS
+ ldap_pvt_tls_destroy();
+#endif
+
+#ifdef NEW_LOGGING
+ LDAP_LOG ( SLURPD, RESULTS, "main: slurpd terminated\n", 0, 0, 0 );
#else
- pthread_join( sglob->replicas[ i ]->ri_tid, (void *) &status );
+ Debug( LDAP_DEBUG_ANY, "slurpd: terminated.\n", 0, 0, 0 );
#endif
+
+ if ( slurpd_pid_file != NULL ) {
+ unlink( slurpd_pid_file );
}
- Debug( LDAP_DEBUG_ANY, "slurpd: terminating normally\n", 0, 0, 0 );
- sglob->slurpd_shutdown = 1;
- pthread_exit( 0 );
+ if ( slurpd_args_file != NULL ) {
+ unlink( slurpd_args_file );
+ }
+
+ MAIN_RETURN(rc);
#endif /* !NO_THREADS */
}