git.sur5r.net Git - openldap/blob - servers/slurpd/main.c

   1 /* $OpenLDAP$ */
   2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   3  *
   4  * Copyright 1998-2005 The OpenLDAP Foundation.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted only as authorized by the OpenLDAP
   9  * Public License.
  10  *
  11  * A copy of this license is available in file LICENSE in the
  12  * top-level directory of the distribution or, alternatively, at
  13  * <http://www.OpenLDAP.org/license.html>.
  14  */
  15 /* Portions Copyright (c) 1996 Regents of the University of Michigan.
  16  * All rights reserved.
  17  *
  18  * Redistribution and use in source and binary forms are permitted
  19  * provided that this notice is preserved and that due credit is given
  20  * to the University of Michigan at Ann Arbor. The name of the University
  21  * may not be used to endorse or promote products derived from this
  22  * software without specific prior written permission. This software
  23  * is provided ``as is'' without express or implied warranty.
  24  */
  25 /* ACKNOWLEDGEMENTS:
  26  * This work was originally developed by the University of Michigan
  27  * (as part of U-MICH LDAP).  Additional significant contributors
  28  * include:
  29  *     Howard Chu
  30  */
  31
  32
  33 /*
  34  * main.c - main routine for slurpd.
  35  */
  36
  37 #include "portable.h"
  38
  39 #include <stdio.h>
  40 #include <sys/stat.h>
  41 #include <ac/stdlib.h>
  42 #include <ac/unistd.h>
  43
  44 #include "slurp.h"
  45 #include "globals.h"
  46 #include "lutil.h"
  47
  48 #include <ldap_pvt.h>
  49
  50 #ifdef HAVE_NT_SERVICE_MANAGER
  51 #define MAIN_RETURN(x)  return
  52 #define SERVICE_EXIT( e, n )    do { \
  53         if ( is_NT_Service ) { \
  54                 lutil_ServiceStatus.dwWin32ExitCode = (e); \
  55                 lutil_ServiceStatus.dwServiceSpecificExitCode = (n); \
  56         } \
  57 } while ( 0 )
  58 #else
  59 #define SERVICE_EXIT( e, n )
  60 #define MAIN_RETURN(x)  return(x)
  61 #endif
  62
  63 #ifndef HAVE_MKVERSION
  64 const char Versionstr[] =
  65         OPENLDAP_PACKAGE " " OPENLDAP_VERSION " Standalone LDAP Replicator (slurpd)";
  66 #endif
  67
  68 #ifdef HAVE_NT_SERVICE_MANAGER
  69 void WINAPI ServiceMain( DWORD argc, LPTSTR *argv )
  70 #else
  71 int main( int argc, char **argv )
  72 #endif
  73 {
  74 #ifdef NO_THREADS
  75     /* Haven't yet written the non-threaded version */
  76     fputs( "slurpd currently requires threads support\n", stderr );
  77     return( 1 );
  78 #else
  79
  80     int                 i, rc = 0;
  81
  82     /* initialize thread package */
  83     ldap_pvt_thread_initialize();
  84
  85     /*
  86      * Create and initialize globals.  init_globals() also initializes
  87      * the main replication queue.
  88      */
  89     if (( sglob = init_globals()) == NULL ) {
  90         fprintf( stderr, "Out of memory initializing globals\n" );
  91         SERVICE_EXIT( ERROR_NOT_ENOUGH_MEMORY, 0 );
  92         rc = 1;
  93         goto stop;
  94     }
  95
  96 #ifdef HAVE_NT_SERVICE_MANAGER
  97         {
  98                 int *i;
  99                 char *newConfigFile;
 100                 char *regService = NULL;
 101
 102                 if ( is_NT_Service ) {
 103                         sglob->serverName = argv[0];
 104                         lutil_CommenceStartupProcessing( sglob->serverName, slurp_set_shutdown );
 105                         if ( strcmp(sglob->serverName, SERVICE_NAME) )
 106                             regService = sglob->serverName;
 107                 }
 108
 109                 i = (int*)lutil_getRegParam( regService, "DebugLevel" );
 110                 if ( i != NULL )
 111                 {
 112                         ldap_debug = *i;
 113                         Debug( LDAP_DEBUG_ANY, "new debug level from registry is: %d\n", ldap_debug, 0, 0 );
 114                 }
 115
 116                 newConfigFile = (char*)lutil_getRegParam( regService, "ConfigFile" );
 117                 if ( newConfigFile != NULL )
 118                 {
 119                         sglob->slapd_configfile = newConfigFile;
 120                         Debug ( LDAP_DEBUG_ANY, "new config file from registry is: %s\n", sglob->slapd_configfile, 0, 0 );
 121
 122                 }
 123         }
 124 #endif
 125
 126     /*
 127      * Process command-line args and fill in globals.
 128      */
 129     if ( doargs( argc, argv, sglob ) < 0 ) {
 130         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 15 );
 131         rc = 1;
 132         goto stop;
 133     }
 134
 135     if ( sglob->version ) {
 136                 fprintf(stderr, "%s\n", Versionstr);
 137                 if (sglob->version > 1 ) {
 138                         rc = 1;
 139                         goto stop;
 140                 }
 141     }
 142
 143         Debug ( LDAP_DEBUG_ANY, "%s\n", Versionstr, 0, 0 );
 144
 145     /*
 146      * Read slapd config file and initialize Re (per-replica) structs.
 147      */
 148     if ( slurpd_read_config( sglob->slapd_configfile ) < 0 ) {
 149         fprintf( stderr,
 150                 "Errors encountered while processing config file \"%s\"\n",
 151                 sglob->slapd_configfile );
 152         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 19 );
 153         rc = 1;
 154         goto stop;
 155     }
 156
 157 #ifdef HAVE_TLS
 158         if( ldap_pvt_tls_init() || ldap_pvt_tls_init_def_ctx() ) {
 159                 fprintf( stderr, "TLS Initialization failed.\n" );
 160                 SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
 161                 rc = 1;
 162                 goto stop;
 163         }
 164 #endif
 165
 166     /*
 167      * Make sure our directory exists
 168      */
 169     if ( mkdir(sglob->slurpd_rdir, 0755) == -1 && errno != EEXIST) {
 170         perror(sglob->slurpd_rdir);
 171         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
 172         rc = 1;
 173         goto stop;
 174     }
 175
 176     /*
 177      * Get any saved state information off the disk.
 178      */
 179     if ( sglob->st->st_read( sglob->st )) {
 180         fprintf( stderr, "Malformed slurpd status file \"%s\"\n",
 181                 sglob->slurpd_status_file );
 182         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 17 );
 183         rc = 1;
 184         goto stop;
 185     }
 186
 187     /*
 188      * All readonly data should now be initialized.
 189      * Check for any fatal error conditions before we get started
 190      */
 191      if ( sanity() < 0 ) {
 192         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 18 );
 193         rc = 1;
 194         goto stop;
 195     }
 196
 197
 198     /*
 199      * Detach from the controlling terminal
 200      * unless the -d flag is given or in one-shot mode.
 201      */
 202 #ifndef HAVE_WINSOCK
 203         if ( ! (sglob->no_detach || sglob->one_shot_mode) ) {
 204                 lutil_detach( 0, 0 );
 205         }
 206 #endif
 207
 208         if ( slurpd_pid_file != NULL ) {
 209                 FILE *fp = fopen( slurpd_pid_file, "w" );
 210
 211                 if( fp != NULL ) {
 212                         fprintf( fp, "%d\n", (int) getpid() );
 213                         fclose( fp );
 214
 215                 } else {
 216                 free(slurpd_pid_file);
 217                 slurpd_pid_file = NULL;
 218                 }
 219         }
 220
 221         if ( slurpd_args_file != NULL ) {
 222                 FILE *fp = fopen( slurpd_args_file, "w" );
 223
 224                 if( fp != NULL ) {
 225                         for ( i = 0; i < argc; i++ ) {
 226                                 fprintf( fp, "%s ", argv[i] );
 227                         }
 228                         fprintf( fp, "\n" );
 229                         fclose( fp );
 230                 } else {
 231                         free(slurpd_args_file);
 232                         slurpd_args_file = NULL;
 233                 }
 234         }
 235
 236     if ( (rc = lutil_pair( sglob->wake_sds )) < 0 ) {
 237         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
 238         rc = 1;
 239         goto stop;
 240     }
 241
 242 #ifdef HAVE_NT_EVENT_LOG
 243         if (is_NT_Service) lutil_LogStartedEvent( sglob->serverName, ldap_debug, sglob->slapd_configfile, "n/a" );
 244 #endif
 245
 246     /*
 247      * Start the main file manager thread (in fm.c).
 248      */
 249     if ( ldap_pvt_thread_create( &(sglob->fm_tid),
 250                 0, fm, (void *) NULL ) != 0 )
 251         {
 252         Debug( LDAP_DEBUG_ANY, "file manager ldap_pvt_thread_create failed\n",
 253                 0, 0, 0 );
 254         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 21 );
 255         rc = 1;
 256         goto stop;
 257
 258     }
 259
 260     /*
 261      * wait for fm to finish if in oneshot mode
 262      */
 263     if ( sglob->one_shot_mode ) {
 264         ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
 265     }
 266
 267     /*
 268      * Start threads - one thread for each replica
 269      */
 270     for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
 271         start_replica_thread( sglob->replicas[ i ]);
 272     }
 273
 274 #ifdef HAVE_NT_SERVICE_MANAGER
 275     if ( started_event ) ldap_pvt_thread_cond_signal( &started_event );
 276 #endif
 277
 278     /*
 279      * Wait for the fm thread to finish.
 280      */
 281     if ( !sglob->one_shot_mode ) {
 282         ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
 283     }
 284
 285     /*
 286      * Wait for the replica threads to finish.
 287      */
 288     for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
 289         ldap_pvt_thread_join( sglob->replicas[ i ]->ri_tid, (void *) NULL );
 290     }
 291
 292 stop:
 293 #ifdef HAVE_NT_SERVICE_MANAGER
 294         if (is_NT_Service) {
 295                 ldap_pvt_thread_cond_destroy( &started_event );
 296                 lutil_LogStoppedEvent( sglob->serverName );
 297                 lutil_ReportShutdownComplete();
 298         }
 299 #endif
 300     /* destroy the thread package */
 301     ldap_pvt_thread_destroy();
 302
 303 #ifdef HAVE_TLS
 304     ldap_pvt_tls_destroy();
 305 #endif
 306
 307     Debug( LDAP_DEBUG_ANY, "slurpd: terminated.\n", 0, 0, 0 );
 308
 309     if ( slurpd_pid_file != NULL ) {
 310         unlink( slurpd_pid_file );
 311     }
 312     if ( slurpd_args_file != NULL ) {
 313         unlink( slurpd_args_file );
 314     }
 315
 316
 317         MAIN_RETURN(rc);
 318 #endif /* !NO_THREADS */
 319 }