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 osrfSystemBootstrap( char* hostname, char* configfile, char* contextNode ) {
  28         if( !(configfile && contextNode) ) return -1;
  29
  30         /* first we grab the settings */
  31         if(!osrfSystemBootstrapClientResc(configfile, contextNode, "settings_grabber" )) {
  32                 return fatal_handler("Unable to bootstrap");
  33         }
  34
  35         osrf_settings_retrieve(hostname);
  36         osrf_system_disconnect_client();
  37
  38         jsonObject* apps = osrf_settings_host_value_object("/activeapps/appname");
  39         osrfStringArray* arr = osrfNewStringArray(8);
  40
  41         if(apps) {
  42                 int i = 0;
  43
  44                 if(apps->type == JSON_STRING) {
  45                         osrfStringArrayAdd(arr, jsonObjectGetString(apps));
  46
  47                 } else {
  48                         jsonObject* app;
  49                         while( (app = jsonObjectGetIndex(apps, i++)) )
  50                                 osrfStringArrayAdd(arr, jsonObjectGetString(app));
  51                 }
  52
  53                 char* appname = NULL;
  54                 i = 0;
  55                 while( (appname = osrfStringArrayGetString(arr, i++)) ) {
  56
  57                         char* libfile = osrf_settings_host_value("/apps/%s/implementation", appname);
  58                         info_handler("Launching application %s with implementation %s", appname, libfile);
  59
  60                         if(! (appname && libfile) ) {
  61                                 warning_handler("Missing appname / libfile in settings config");
  62                                 continue;
  63                         }
  64
  65                         int pid;
  66
  67                         if( (pid = fork()) ) {
  68                                 // storage pid in local table for re-launching dead children...
  69                                 info_handler("Launched application child %d", pid);
  70
  71                         } else {
  72
  73                                 osrfAppRegisterApplication( appname, libfile );
  74                                 osrf_prefork_run(appname);
  75                                 exit(0);
  76                         }
  77                 }
  78         }
  79
  80         /** daemonize me **/
  81
  82         /* let our children do their thing */
  83         while(1) {
  84                 signal(SIGCHLD, __osrfSystemSignalHandler);
  85                 sleep(10000);
  86         }
  87
  88         return 0;
  89 }
  90
  91 int osrf_system_bootstrap_client_resc( char* config_file, char* contextnode, char* resource ) {
  92
  93         if( !( config_file && contextnode ) && ! osrfConfigHasDefaultConfig() )
  94                 return fatal_handler("No Config File Specified\n" );
  95
  96         if( config_file ) {
  97                 osrfConfigCleanup();
  98                 osrfConfig* cfg = osrfConfigInit( config_file, contextnode );
  99                 osrfConfigSetDefaultConfig(cfg);
 100         }
 101
 102
 103         char* log_file          = osrfConfigGetValue( NULL, "/logfile");
 104         char* log_level = osrfConfigGetValue( NULL, "/loglevel" );
 105         osrfStringArray* arr = osrfNewStringArray(8);
 106         osrfConfigGetValueList(NULL, arr, "/domains/domain");
 107         char* username          = osrfConfigGetValue( NULL, "/username" );
 108         char* password          = osrfConfigGetValue( NULL, "/passwd" );
 109         char* port                      = osrfConfigGetValue( NULL, "/port" );
 110         char* unixpath          = osrfConfigGetValue( NULL, "/unixpath" );
 111
 112         char* domain = osrfStringArrayGetString( arr, 0 ); /* just the first for now */
 113         osrfStringArrayFree(arr);
 114
 115
 116         int llevel = 0;
 117         int iport = 0;
 118         if(port) iport = atoi(port);
 119         if(log_level) llevel = atoi(log_level);
 120
 121         log_init( llevel, log_file );
 122
 123         info_handler("Bootstrapping system with domain %s, port %d, and unixpath %s", domain, iport, unixpath );
 124
 125         transport_client* client = client_init( domain, iport, unixpath, 0 );
 126
 127         char* host = getenv("HOSTNAME");
 128
 129         if(!resource) resource = "";
 130         int len = strlen(resource) + 256;
 131         char buf[len];
 132         memset(buf,0,len);
 133         snprintf(buf, len - 1, "opensrf_%s_%s_%d", resource, host, getpid() );
 134
 135         if(client_connect( client, username, password, buf, 10, AUTH_DIGEST )) {
 136                 /* child nodes will leak the parents client... but we can't free
 137                         it without disconnecting the parents client :( */
 138                 __osrfGlobalTransportClient = client;
 139         }
 140
 141         free(log_level);
 142         free(log_file);
 143         free(username);
 144         free(password);
 145         free(port);
 146         free(unixpath);
 147
 148         if(__osrfGlobalTransportClient)
 149                 return 1;
 150
 151         return 0;
 152 }
 153
 154 int osrf_system_disconnect_client() {
 155         client_disconnect( __osrfGlobalTransportClient );
 156         client_free( __osrfGlobalTransportClient );
 157         __osrfGlobalTransportClient = NULL;
 158         return 0;
 159 }
 160
 161 int osrf_system_shutdown() {
 162         osrfConfigCleanup();
 163         osrf_system_disconnect_client();
 164         osrf_settings_free_host_config(NULL);
 165         log_free();
 166         return 1;
 167 }
 168
 169
 170
 171
 172 void __osrfSystemSignalHandler( int sig ) {
 173
 174         pid_t pid;
 175         int status;
 176
 177         while( (pid = waitpid(-1, &status, WNOHANG)) > 0) {
 178                 warning_handler("We lost child %d", pid);
 179         }
 180
 181         /** relaunch the server **/
 182 }