3 * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
4 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
7 * Copyright (c) 1996 Regents of the University of Michigan.
10 * Redistribution and use in source and binary forms are permitted
11 * provided that this notice is preserved and that due credit is given
12 * to the University of Michigan at Ann Arbor. The name of the University
13 * may not be used to endorse or promote products derived from this
14 * software without specific prior written permission. This software
15 * is provided ``as is'' without express or implied warranty.
20 * main.c - main routine for slurpd.
27 #include <ac/stdlib.h>
35 #ifdef HAVE_NT_SERVICE_MANAGER
36 #define MAIN_RETURN(x) return
37 #define SERVICE_EXIT( e, n ) do { \
38 if ( is_NT_Service ) { \
39 lutil_ServiceStatus.dwWin32ExitCode = (e); \
40 lutil_ServiceStatus.dwServiceSpecificExitCode = (n); \
44 #define SERVICE_EXIT( e, n )
45 #define MAIN_RETURN(x) return(x)
48 #ifdef HAVE_NT_SERVICE_MANAGER
49 void WINAPI ServiceMain( DWORD argc, LPTSTR *argv )
51 int main( int argc, char **argv )
55 /* Haven't yet written the non-threaded version */
56 fputs( "slurpd currently requires threads support\n", stderr );
62 /* initialize thread package */
63 ldap_pvt_thread_initialize();
66 * Create and initialize globals. init_globals() also initializes
67 * the main replication queue.
69 if (( sglob = init_globals()) == NULL ) {
70 fprintf( stderr, "Out of memory initializing globals\n" );
71 SERVICE_EXIT( ERROR_NOT_ENOUGH_MEMORY, 0 );
76 #ifdef HAVE_NT_SERVICE_MANAGER
80 char *regService = NULL;
82 if ( is_NT_Service ) {
83 sglob->serverName = argv[0];
84 lutil_CommenceStartupProcessing( sglob->serverName, slurp_set_shutdown );
85 if ( strcmp(sglob->serverName, SERVICE_NAME) )
86 regService = sglob->serverName;
89 i = (int*)lutil_getRegParam( regService, "DebugLevel" );
94 lutil_log_initialize( argc, argv );
95 LDAP_LOG( SLURPD, INFO,
96 "main: new debug level from registry is: %d\n",
99 Debug( LDAP_DEBUG_ANY, "new debug level from registry is: %d\n", ldap_debug, 0, 0 );
103 newConfigFile = (char*)lutil_getRegParam( regService, "ConfigFile" );
104 if ( newConfigFile != NULL )
106 sglob->slapd_configfile = newConfigFile;
108 LDAP_LOG( SLURPD, INFO,
109 "main: new config file from registry is: %s\n", sglob->slapd_configfile, 0, 0 );
111 Debug ( LDAP_DEBUG_ANY, "new config file from registry is: %s\n", sglob->slapd_configfile, 0, 0 );
119 * Process command-line args and fill in globals.
121 if ( doargs( argc, argv, sglob ) < 0 ) {
122 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 15 );
128 * Read slapd config file and initialize Re (per-replica) structs.
130 if ( slurpd_read_config( sglob->slapd_configfile ) < 0 ) {
132 "Errors encountered while processing config file \"%s\"\n",
133 sglob->slapd_configfile );
134 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 19 );
140 if( ldap_pvt_tls_init() || ldap_pvt_tls_init_def_ctx() ) {
141 fprintf( stderr, "TLS Initialization failed.\n" );
142 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
149 * Make sure our directory exists
151 if ( mkdir(sglob->slurpd_rdir, 0755) == -1 && errno != EEXIST) {
152 perror(sglob->slurpd_rdir);
153 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
159 * Get any saved state information off the disk.
161 if ( sglob->st->st_read( sglob->st )) {
162 fprintf( stderr, "Malformed slurpd status file \"%s\"\n",
163 sglob->slurpd_status_file, 0, 0 );
164 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 17 );
170 * All readonly data should now be initialized.
171 * Check for any fatal error conditions before we get started
173 if ( sanity() < 0 ) {
174 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 18 );
180 * Detach from the controlling terminal
181 * unless the -d flag is given or in one-shot mode.
184 if ( ! (sglob->no_detach || sglob->one_shot_mode) )
185 lutil_detach( 0, 0 );
188 if ( (rc = lutil_pair( sglob->wake_sds )) < 0 ) {
189 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
194 #ifdef HAVE_NT_EVENT_LOG
195 if (is_NT_Service) lutil_LogStartedEvent( sglob->serverName, ldap_debug, sglob->slapd_configfile, "n/a" );
199 * Start the main file manager thread (in fm.c).
201 if ( ldap_pvt_thread_create( &(sglob->fm_tid),
202 0, fm, (void *) NULL ) != 0 )
205 LDAP_LOG ( SLURPD, ERR,
206 "main: file manager ldap_pvt_thread_create failed\n" , 0, 0, 0 );
208 Debug( LDAP_DEBUG_ANY, "file manager ldap_pvt_thread_create failed\n",
211 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 21 );
218 * wait for fm to finish if in oneshot mode
220 if ( sglob->one_shot_mode ) {
221 ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
225 * Start threads - one thread for each replica
227 for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
228 start_replica_thread( sglob->replicas[ i ]);
231 #ifdef HAVE_NT_SERVICE_MANAGER
232 if ( started_event ) ldap_pvt_thread_cond_signal( &started_event );
236 * Wait for the fm thread to finish.
238 if ( !sglob->one_shot_mode ) {
239 ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
243 * Wait for the replica threads to finish.
245 for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
246 ldap_pvt_thread_join( sglob->replicas[ i ]->ri_tid, (void *) NULL );
250 #ifdef HAVE_NT_SERVICE_MANAGER
252 ldap_pvt_thread_cond_destroy( &started_event );
253 lutil_LogStoppedEvent( sglob->serverName );
254 lutil_ReportShutdownComplete();
257 /* destroy the thread package */
258 ldap_pvt_thread_destroy();
261 LDAP_LOG ( SLURPD, RESULTS, "main: slurpd terminated\n", 0, 0, 0 );
263 Debug( LDAP_DEBUG_ANY, "slurpd: terminated.\n", 0, 0, 0 );
266 #endif /* !NO_THREADS */