OpenSRF/src/libstack/osrf_system.c

   1 #include "osrf_system.h"
   2 #include <signal.h>
   3 #include "osrf_application.h"
   4 #include "osrf_prefork.h"
   5
   6 void __osrfSystemSignalHandler( int sig );
   7
   8 transport_client* __osrfGlobalTransportClient;
   9
  10 transport_client* osrfSystemGetTransportClient() {
  11         return __osrfGlobalTransportClient;
  12 }
  13
  14 transport_client* osrf_system_get_transport_client() {
  15         return __osrfGlobalTransportClient;
  16 }
  17
  18 int osrf_system_bootstrap_client( char* config_file, char* contextnode ) {
  19         return osrf_system_bootstrap_client_resc(config_file, contextnode, NULL);
  20 }
  21
  22 int osrfSystemBootstrapClientResc( char* config_file, char* contextnode, char* resource ) {
  23         return osrf_system_bootstrap_client_resc( config_file, contextnode, resource );
  24 }
  25
  26
  27 int _osrfSystemInitCache() {
  28
  29         jsonObject* cacheServers = osrf_settings_host_value_object("/cache/global/servers/server");
  30         char* maxCache = osrf_settings_host_value("/cache/global/max_cache_time");
  31
  32         if( cacheServers && maxCache) {
  33
  34                 if( cacheServers->type == JSON_ARRAY ) {
  35                         int i;
  36                         char* servers[cacheServers->size];
  37                         for( i = 0; i != cacheServers->size; i++ ) {
  38                                 servers[i] = jsonObjectGetString( jsonObjectGetIndex(cacheServers, i) );
  39                                 osrfLogInfo("Adding cache server %s", servers[i]);
  40                         }
  41                         osrfCacheInit( servers, cacheServers->size, atoi(maxCache) );
  42
  43                 } else {
  44                         char* servers[] = { jsonObjectGetString(cacheServers) };
  45                         osrfLogInfo("Adding cache server %s", servers[0]);
  46                         osrfCacheInit( servers, 1, atoi(maxCache) );
  47                 }
  48
  49         } else {
  50                 osrfLogError( "Missing config value for /cache/global/servers/server _or_ "
  51                         "/cache/global/max_cache_time");
  52         }
  53
  54         return 0;
  55 }
  56
  57
  58 int osrfSystemBootstrap( char* hostname, char* configfile, char* contextNode ) {
  59         if( !(hostname && configfile && contextNode) ) return -1;
  60
  61         /* first we grab the settings */
  62         if(!osrfSystemBootstrapClientResc(configfile, contextNode, "settings_grabber" )) {
  63                 osrfLogError("Unable to bootstrap");
  64                 return -1;
  65         }
  66
  67         osrf_settings_retrieve(hostname);
  68         osrf_system_disconnect_client();
  69
  70         jsonObject* apps = osrf_settings_host_value_object("/activeapps/appname");
  71         osrfStringArray* arr = osrfNewStringArray(8);
  72
  73         _osrfSystemInitCache();
  74
  75         if(apps) {
  76                 int i = 0;
  77
  78                 if(apps->type == JSON_STRING) {
  79                         osrfStringArrayAdd(arr, jsonObjectGetString(apps));
  80
  81                 } else {
  82                         jsonObject* app;
  83                         while( (app = jsonObjectGetIndex(apps, i++)) )
  84                                 osrfStringArrayAdd(arr, jsonObjectGetString(app));
  85                 }
  86
  87                 char* appname = NULL;
  88                 i = 0;
  89                 while( (appname = osrfStringArrayGetString(arr, i++)) ) {
  90
  91                         char* lang = osrf_settings_host_value("/apps/%s/language", appname);
  92
  93                         if(lang && !strcasecmp(lang,"c"))  {
  94
  95                                 char* libfile = osrf_settings_host_value("/apps/%s/implementation", appname);
  96
  97                                 if(! (appname && libfile) ) {
  98                                         osrfLogWarning("Missing appname / libfile in settings config");
  99                                         continue;
 100                                 }
 101
 102                                 osrfLogInfo("Launching application %s with implementation %s", appname, libfile);
 103
 104                                 int pid;
 105
 106                                 if( (pid = fork()) ) {
 107                                         // storage pid in local table for re-launching dead children...
 108                                         osrfLogInfo("Launched application child %d", pid);
 109
 110                                 } else {
 111
 112                                         fprintf(stderr, " * Running application %s\n", appname);
 113                                         if( osrfAppRegisterApplication( appname, libfile ) == 0 )
 114                                                 osrf_prefork_run(appname);
 115
 116                                         osrfLogDebug("Server exiting for app %s and library %s", appname, libfile );
 117                                         exit(0);
 118                                 }
 119                         } // language == c
 120                 }
 121         }
 122
 123         /** daemonize me **/
 124
 125         /* background and let our children do their thing */
 126         daemonize();
 127         while(1) {
 128                 signal(SIGCHLD, __osrfSystemSignalHandler);
 129                 sleep(10000);
 130         }
 131
 132         return 0;
 133 }
 134
 135 int osrf_system_bootstrap_client_resc( char* config_file, char* contextnode, char* resource ) {
 136
 137         if( !( config_file && contextnode ) && ! osrfConfigHasDefaultConfig() ) {
 138                 osrfLogError("No Config File Specified\n" );
 139                 return -1;
 140         }
 141
 142         if( config_file ) {
 143                 osrfConfigCleanup();
 144                 osrfConfig* cfg = osrfConfigInit( config_file, contextnode );
 145                 osrfConfigSetDefaultConfig(cfg);
 146         }
 147
 148
 149         char* log_file          = osrfConfigGetValue( NULL, "/logfile");
 150         char* log_level = osrfConfigGetValue( NULL, "/loglevel" );
 151         osrfStringArray* arr = osrfNewStringArray(8);
 152         osrfConfigGetValueList(NULL, arr, "/domains/domain");
 153         char* username          = osrfConfigGetValue( NULL, "/username" );
 154         char* password          = osrfConfigGetValue( NULL, "/passwd" );
 155         char* port                      = osrfConfigGetValue( NULL, "/port" );
 156         char* unixpath          = osrfConfigGetValue( NULL, "/unixpath" );
 157         char* facility          = osrfConfigGetValue( NULL, "/syslog" );
 158
 159         char* domain = strdup(osrfStringArrayGetString( arr, 0 )); /* just the first for now */
 160         osrfStringArrayFree(arr);
 161
 162
 163         int llevel = 0;
 164         int iport = 0;
 165         if(port) iport = atoi(port);
 166         if(log_level) llevel = atoi(log_level);
 167
 168         if(!log_file) { fprintf(stderr, "Log file needed\n"); return -1; }
 169
 170         if(!strcmp(log_file, "syslog")) {
 171                 osrfLogInit( OSRF_LOG_TYPE_SYSLOG, contextnode, llevel );
 172                 osrfLogSetSyslogFacility(osrfLogFacilityToInt(facility));
 173
 174         } else {
 175                 osrfLogInit( OSRF_LOG_TYPE_FILE, contextnode, llevel );
 176                 osrfLogSetFile( log_file );
 177         }
 178
 179         osrfLogInfo("Bootstrapping system with domain %s, port %d, and unixpath %s", domain, iport, unixpath );
 180
 181         transport_client* client = client_init( domain, iport, unixpath, 0 );
 182
 183         char* host;
 184         host = getenv("HOSTNAME");
 185
 186         if(!host) host = "";
 187         if(!resource) resource = "";
 188
 189         int len = strlen(resource) + 256;
 190         char buf[len];
 191         memset(buf,0,len);
 192         snprintf(buf, len - 1, "%s_%s_%d", resource, host, getpid() );
 193
 194         if(client_connect( client, username, password, buf, 10, AUTH_DIGEST )) {
 195                 /* child nodes will leak the parents client... but we can't free
 196                         it without disconnecting the parents client :( */
 197                 __osrfGlobalTransportClient = client;
 198         }
 199
 200         free(facility);
 201         free(log_level);
 202         free(log_file);
 203         free(username);
 204         free(password);
 205         free(port);
 206         free(unixpath);
 207
 208         if(__osrfGlobalTransportClient)
 209                 return 1;
 210
 211         return 0;
 212 }
 213
 214 int osrf_system_disconnect_client() {
 215         client_disconnect( __osrfGlobalTransportClient );
 216         client_free( __osrfGlobalTransportClient );
 217         __osrfGlobalTransportClient = NULL;
 218         return 0;
 219 }
 220
 221 int osrf_system_shutdown() {
 222         osrfConfigCleanup();
 223         osrf_system_disconnect_client();
 224         osrf_settings_free_host_config(NULL);
 225         return 1;
 226 }
 227
 228
 229
 230
 231 void __osrfSystemSignalHandler( int sig ) {
 232
 233         pid_t pid;
 234         int status;
 235
 236         while( (pid = waitpid(-1, &status, WNOHANG)) > 0) {
 237                 osrfLogWarning("We lost child %d", pid);
 238         }
 239
 240         /** relaunch the server **/
 241 }
 242
 243