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-2007 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( 0 ) ) {
 159                 rc = 0;
 160                 /* See if we actually need TLS */
 161                 for ( i=0; i < sglob->num_replicas; i++ ) {
 162                         if ( sglob->replicas[i]->ri_tls || ( sglob->replicas[i]->ri_uri &&
 163                                 !strncmp( sglob->replicas[i]->ri_uri, "ldaps:", 6 ))) {
 164                                 rc = 1;
 165                                 break;
 166                         }
 167                 }
 168                 if ( rc ) {
 169                         fprintf( stderr, "TLS Initialization failed.\n" );
 170                         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 20 );
 171                         goto stop;
 172                 }
 173         }
 174 #endif
 175
 176     /*
 177      * Make sure our directory exists
 178      */
 179     if ( mkdir(sglob->slurpd_rdir, 0755) == -1 && errno != EEXIST) {
 180         perror(sglob->slurpd_rdir);
 181         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
 182         rc = 1;
 183         goto stop;
 184     }
 185
 186     /*
 187      * Get any saved state information off the disk.
 188      */
 189     if ( sglob->st->st_read( sglob->st )) {
 190         fprintf( stderr, "Malformed slurpd status file \"%s\"\n",
 191                 sglob->slurpd_status_file );
 192         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 17 );
 193         rc = 1;
 194         goto stop;
 195     }
 196
 197     /*
 198      * All readonly data should now be initialized.
 199      * Check for any fatal error conditions before we get started
 200      */
 201      if ( sanity() < 0 ) {
 202         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 18 );
 203         rc = 1;
 204         goto stop;
 205     }
 206
 207
 208     /*
 209      * Detach from the controlling terminal
 210      * unless the -d flag is given or in one-shot mode.
 211      */
 212 #ifndef HAVE_WINSOCK
 213         if ( ! (sglob->no_detach || sglob->one_shot_mode) ) {
 214                 lutil_detach( 0, 0 );
 215         }
 216 #endif
 217
 218         /*
 219          * don't open pid/args file in one-shot mode (ITS#4152)
 220          *
 221          * bail out if files were specified but cannot be opened (ITS#4074)
 222          */
 223         if ( !sglob->one_shot_mode) {
 224                 if ( slurpd_pid_file != NULL ) {
 225                         FILE *fp = fopen( slurpd_pid_file, "w" );
 226
 227                         if ( fp == NULL ) {
 228                                 int save_errno = errno;
 229
 230                                 fprintf( stderr, "unable to open pid file "
 231                                         "\"%s\": %d (%s)\n",
 232                                         slurpd_pid_file,
 233                                         save_errno, strerror( save_errno ) );
 234
 235                                 free( slurpd_pid_file );
 236                                 slurpd_pid_file = NULL;
 237
 238                                 rc = 1;
 239                                 goto stop;
 240                         }
 241
 242                         fprintf( fp, "%d\n", (int) getpid() );
 243                         fclose( fp );
 244                 }
 245
 246                 if ( slurpd_args_file != NULL ) {
 247                         FILE *fp = fopen( slurpd_args_file, "w" );
 248
 249                         if ( fp == NULL ) {
 250                                 int save_errno = errno;
 251
 252                                 fprintf( stderr, "unable to open args file "
 253                                         "\"%s\": %d (%s)\n",
 254                                         slurpd_args_file,
 255                                         save_errno, strerror( save_errno ) );
 256
 257                                 free( slurpd_args_file );
 258                                 slurpd_pid_file = NULL;
 259
 260                                 rc = 1;
 261                                 goto stop;
 262                         }
 263
 264                         for ( i = 0; i < argc; i++ ) {
 265                                 fprintf( fp, "%s ", argv[i] );
 266                         }
 267                         fprintf( fp, "\n" );
 268                         fclose( fp );
 269                 }
 270         }
 271
 272     if ( (rc = lutil_pair( sglob->wake_sds )) < 0 ) {
 273         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 16 );
 274         rc = 1;
 275         goto stop;
 276     }
 277
 278 #ifdef HAVE_NT_EVENT_LOG
 279         if (is_NT_Service) lutil_LogStartedEvent( sglob->serverName, ldap_debug, sglob->slapd_configfile, "n/a" );
 280 #endif
 281
 282     /*
 283      * Start the main file manager thread (in fm.c).
 284      */
 285     if ( ldap_pvt_thread_create( &(sglob->fm_tid),
 286                 0, fm, (void *) NULL ) != 0 )
 287         {
 288         Debug( LDAP_DEBUG_ANY, "file manager ldap_pvt_thread_create failed\n",
 289                 0, 0, 0 );
 290         SERVICE_EXIT( ERROR_SERVICE_SPECIFIC_ERROR, 21 );
 291         rc = 1;
 292         goto stop;
 293
 294     }
 295
 296     /*
 297      * wait for fm to finish if in oneshot mode
 298      */
 299     if ( sglob->one_shot_mode ) {
 300         ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
 301     }
 302
 303     /*
 304      * Start threads - one thread for each replica
 305      */
 306     for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
 307         start_replica_thread( sglob->replicas[ i ]);
 308     }
 309
 310 #ifdef HAVE_NT_SERVICE_MANAGER
 311     if ( started_event ) ldap_pvt_thread_cond_signal( &started_event );
 312 #endif
 313
 314     /*
 315      * Wait for the fm thread to finish.
 316      */
 317     if ( !sglob->one_shot_mode ) {
 318         ldap_pvt_thread_join( sglob->fm_tid, (void *) NULL );
 319     }
 320
 321     /*
 322      * Wait for the replica threads to finish.
 323      */
 324     for ( i = 0; sglob->replicas[ i ] != NULL; i++ ) {
 325         ldap_pvt_thread_join( sglob->replicas[ i ]->ri_tid, (void *) NULL );
 326     }
 327
 328 stop:
 329 #ifdef HAVE_NT_SERVICE_MANAGER
 330         if (is_NT_Service) {
 331                 ldap_pvt_thread_cond_destroy( &started_event );
 332                 lutil_LogStoppedEvent( sglob->serverName );
 333                 lutil_ReportShutdownComplete();
 334         }
 335 #endif
 336     /* destroy the thread package */
 337     ldap_pvt_thread_destroy();
 338
 339 #ifdef HAVE_TLS
 340     ldap_pvt_tls_destroy();
 341 #endif
 342
 343     Debug( LDAP_DEBUG_ANY, "slurpd: terminated.\n", 0, 0, 0 );
 344
 345     if ( slurpd_pid_file != NULL ) {
 346         unlink( slurpd_pid_file );
 347     }
 348     if ( slurpd_args_file != NULL ) {
 349         unlink( slurpd_args_file );
 350     }
 351
 352
 353         MAIN_RETURN(rc);
 354 #endif /* !NO_THREADS */
 355 }