src/gras/Virtu/process.c

   1 /* process - GRAS process handling (common code for RL and SG)              */
   2
   3 /* Copyright (c) 2004, 2005, 2006, 2007, 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 "xbt/ex.h"
  10 #include "xbt/sysdep.h"
  11 #include "xbt/log.h"
  12 #include "gras/transport.h"
  13 #include "gras/datadesc.h"
  14 #include "gras/messages.h"
  15 #include "gras_modinter.h"
  16
  17 #include "gras/Virtu/virtu_private.h"
  18
  19 XBT_LOG_NEW_SUBCATEGORY(gras_virtu, gras, "Virtualization code");
  20 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_virtu_process, gras_virtu,
  21                                 "Process manipulation code");
  22
  23 /* Functions to handle gras_procdata_t->libdata cells*/
  24 typedef struct {
  25   char *name;
  26   pvoid_f_void_t constructor;
  27   void_f_pvoid_t destructor;
  28 } s_gras_procdata_fabric_t, *gras_procdata_fabric_t;
  29
  30 static xbt_dynar_t _gras_procdata_fabrics = NULL;       /* content: s_gras_procdata_fabric_t */
  31
  32 static void gras_procdata_fabric_free(void *fab)
  33 {
  34   free(((gras_procdata_fabric_t) fab)->name);
  35 }
  36
  37 /** @brief declare the functions in charge of creating/destructing the procdata of a module
  38  *
  39  *  This is intended to be called from the gras_<module>_register function.
  40  *  This returns the module ID you can use for gras_libdata_by_id()
  41  */
  42 int gras_procdata_add(const char *name, pvoid_f_void_t constructor,
  43                       void_f_pvoid_t destructor)
  44 {
  45
  46   gras_procdata_fabric_t fab;
  47
  48   if (!_gras_procdata_fabrics) {
  49     /* create the dynar if needed */
  50     _gras_procdata_fabrics =
  51         xbt_dynar_new(sizeof(s_gras_procdata_fabric_t),
  52                       gras_procdata_fabric_free);
  53   }
  54
  55   fab = xbt_dynar_push_ptr(_gras_procdata_fabrics);
  56
  57   fab->name = xbt_strdup(name);
  58   fab->constructor = constructor;
  59   fab->destructor = destructor;
  60
  61   return xbt_dynar_length(_gras_procdata_fabrics) - 1;
  62 }
  63
  64 /* **************************************************************************
  65  * Process data
  66  * **************************************************************************/
  67 void *gras_userdata_get(void)
  68 {
  69   gras_procdata_t *pd = gras_procdata_get();
  70   return pd->userdata;
  71 }
  72
  73 void *gras_userdata_set(void *ud)
  74 {
  75   gras_procdata_t *pd = gras_procdata_get();
  76
  77   pd->userdata = ud;
  78   return ud;
  79 }
  80
  81 void *gras_libdata_by_name(const char *name)
  82 {
  83   gras_procdata_t *pd = gras_procdata_get();
  84   return gras_libdata_by_name_from_procdata(name, pd);
  85 }
  86
  87 /* this function is splitted from previous to allow SG to get libdata from remote hosts */
  88 void *gras_libdata_by_name_from_procdata(const char *name,
  89                                          gras_procdata_t * pd)
  90 {
  91   void *res = NULL;
  92   if (xbt_set_length(pd->libdata) <
  93       xbt_dynar_length(_gras_procdata_fabrics)) {
  94     /* Damn, some new modules were added since procdata_init(). Amok? */
  95     /* Get 'em all */
  96     gras_procdata_init();
  97   }
  98   TRY {
  99     res = xbt_set_get_by_name(pd->libdata, name);
 100   }
 101   CATCH_ANONYMOUS {
 102     RETHROWF("Cannot retrieve the libdata associated to %s: %s", name);
 103   }
 104   return res;
 105 }
 106
 107 void *gras_libdata_by_id(int id)
 108 {
 109   gras_procdata_t *pd = gras_procdata_get();
 110   if (xbt_set_length(pd->libdata) <
 111       xbt_dynar_length(_gras_procdata_fabrics)) {
 112     /* Damn, some new modules were added since procdata_init(). Amok? */
 113     /* Get 'em all */
 114     gras_procdata_init();
 115   }
 116   return xbt_set_get_by_id(pd->libdata, id);
 117 }
 118
 119
 120 void gras_procdata_init()
 121 {
 122   gras_procdata_t *pd = gras_procdata_get();
 123   s_gras_procdata_fabric_t fab;
 124
 125   unsigned int cursor;
 126
 127   xbt_ex_t e;
 128   xbt_set_elm_t elem;
 129
 130   if (!pd->libdata) {
 131     pd->userdata = NULL;
 132     pd->libdata = xbt_set_new();
 133   }
 134
 135   xbt_dynar_foreach(_gras_procdata_fabrics, cursor, fab) {
 136     volatile int found = 0;
 137
 138     if (cursor + 1 <= xbt_set_length(pd->libdata)) {
 139       XBT_DEBUG("Skip fabric %d: there is already %ld libdata",
 140              cursor, xbt_set_length(pd->libdata));
 141       continue;                 /* allow to recall this function to get recently added fabrics */
 142     }
 143     XBT_DEBUG("Go ahead for cursor %d, there is %ld libdata",
 144            cursor, xbt_set_length(pd->libdata));
 145
 146     xbt_assert(fab.name, "Name of fabric #%d is NULL!", cursor);
 147     XBT_DEBUG("Create the procdata for %s", fab.name);
 148     /* Check for our own errors */
 149     TRY {
 150       xbt_set_get_by_name(pd->libdata, fab.name);
 151       found = 1;
 152     }
 153     CATCH(e) {
 154       xbt_ex_free(e);
 155       found = 0;
 156     }
 157     if (found)
 158       THROWF(unknown_error, 0,
 159              "MayDay: two modules use '%s' as libdata name", fab.name);
 160
 161     /* Add the data in place, after some more sanity checking */
 162     elem = (*(fab.constructor)) ();
 163     if (elem->name_len && elem->name_len != strlen(elem->name)) {
 164       elem->name_len = strlen(elem->name);
 165       XBT_WARN
 166           ("Module '%s' constructor is borken: it does not set elem->name_len",
 167            fab.name);
 168     }
 169     xbt_set_add(pd->libdata, elem, fab.destructor);
 170   }
 171 }
 172
 173 void gras_procdata_exit()
 174 {
 175   int len;
 176   gras_procdata_t *pd = gras_procdata_get();
 177
 178   xbt_set_free(&(pd->libdata));
 179
 180   /* Remove procdata in reverse order wrt creation */
 181   while ((len = xbt_dynar_length(_gras_procdata_fabrics))) {
 182     xbt_dynar_remove_at(_gras_procdata_fabrics, len - 1, NULL);
 183   }
 184   xbt_dynar_free(&_gras_procdata_fabrics);
 185 }
 186
 187
 188 const char *gras_os_hostport()
 189 {
 190   static char *res = NULL;
 191   if (res)
 192     free(res);                  /* my port may have changed */
 193   res = bprintf("%s:%d", gras_os_myname(), gras_os_myport());
 194   return (const char *) res;
 195 }