src/gras/Virtu/rl_process.c

   1 /* process_rl - GRAS process handling on real life                          */
   2
   3 /* Copyright (c) 2004, 2005, 2006, 2007, 2008, 2009, 2010. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 #include "gras_modinter.h"      /* module initialization interface */
  10 #include "gras/Virtu/virtu_rl.h"
  11 #include "portable.h"
  12 #include "xbt/xbt_os_thread.h"
  13
  14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_virtu_process);
  15
  16 /* globals */
  17 static gras_procdata_t *_gras_procdata = NULL;
  18 XBT_EXPORT_NO_IMPORT(char const *) _gras_procname = NULL;
  19 static xbt_dict_t _process_properties = NULL;
  20 static xbt_dict_t _host_properties = NULL;
  21
  22 # ifdef __APPLE__
  23         /* under darwin, the environment gets added to the process at startup time. So, it's not defined at library link time, forcing us to extra tricks */
  24 # include <crt_externs.h>
  25 # define environ (*_NSGetEnviron())
  26 # else
  27 #ifdef _XBT_WIN32
  28                  /* the environment, as specified by the opengroup, used to initialize the process properties */
  29 extern char **wenviron;
  30 #else
  31 extern char **environ;
  32 #endif
  33 # endif
  34
  35 void gras_process_init()
  36 {
  37   char **env_iter;
  38   _gras_procdata = xbt_new0(gras_procdata_t, 1);
  39   gras_procdata_init();
  40
  41   /* initialize the host & process properties */
  42   _host_properties = xbt_dict_new();
  43   _process_properties = xbt_dict_new();
  44   env_iter = environ;
  45   while (*env_iter) {
  46     char *equal, *buf = xbt_strdup(*env_iter);
  47     equal = strchr(buf, '=');
  48     if (!equal) {
  49       WARN1
  50           ("The environment contains an entry without '=' char: %s (ignore it)",
  51            *env_iter);
  52       continue;
  53     }
  54     *equal = '\0';
  55     xbt_dict_set(_process_properties, buf, xbt_strdup(equal + 1),
  56                  xbt_free_f);
  57     free(buf);
  58     env_iter++;
  59   }
  60 }
  61
  62 void gras_process_exit()
  63 {
  64   gras_procdata_exit();
  65   free(_gras_procdata);
  66   xbt_dict_free(&_process_properties);
  67 }
  68
  69 const char *xbt_procname(void)
  70 {
  71   if (_gras_procname)
  72     return _gras_procname;
  73   else
  74     return "";
  75 }
  76
  77 int gras_os_getpid(void)
  78 {
  79 #ifdef _XBT_WIN32
  80   return (long int) GetCurrentProcessId();
  81 #else
  82   return (long int) getpid();
  83 #endif
  84 }
  85
  86 /* **************************************************************************
  87  * Process data
  88  * **************************************************************************/
  89
  90 gras_procdata_t *gras_procdata_get(void)
  91 {
  92   xbt_assert0(_gras_procdata, "Run gras_process_init (ie, gras_init)!");
  93
  94   return _gras_procdata;
  95 }
  96
  97 typedef struct {
  98   xbt_main_func_t code;
  99   int argc;
 100   char **argv;
 101 } spawner_wrapper_args;
 102
 103 static void *spawner_wrapper(void *data) {
 104   spawner_wrapper_args *a = data;
 105   (*(a->code))(a->argc,a->argv);
 106   free(a);
 107   return NULL;
 108 }
 109
 110 void gras_agent_spawn(const char *name,
 111                       xbt_main_func_t code, int argc, char *argv[],
 112                       xbt_dict_t properties)
 113 {
 114   spawner_wrapper_args *args =malloc(sizeof(spawner_wrapper_args));
 115   args->argc=argc;
 116   args->argv=argv;
 117   args->code=code;
 118   xbt_os_thread_create(name,spawner_wrapper,args);
 119 }
 120
 121 /* **************************************************************************
 122  * Properties
 123  * **************************************************************************/
 124
 125 const char *gras_process_property_value(const char *name)
 126 {
 127   return xbt_dict_get_or_null(_process_properties, name);
 128 }
 129
 130 xbt_dict_t gras_process_properties(void)
 131 {
 132   return _process_properties;
 133 }
 134
 135 const char *gras_os_host_property_value(const char *name)
 136 {
 137   return xbt_dict_get_or_null(_host_properties, name);
 138 }
 139
 140 xbt_dict_t gras_os_host_properties(void)
 141 {
 142   return _host_properties;
 143 }