src/gras/gras.c

   1 /* gras.c -- generic functions not fitting anywhere else                    */
   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 "xbt/log.h"
  10 #include "xbt/virtu.h"          /* set the XBT virtualization to use GRAS */
  11 #include "xbt/module.h"         /* xbt_init/exit */
  12 #include "xbt/xbt_os_time.h"    /* xbt_os_time */
  13 #include "xbt/synchro.h"
  14 #include "xbt/socket.h"
  15
  16 #include "Virtu/virtu_interface.h"      /* Module mechanism FIXME: deplace&rename */
  17 #include "Virtu/virtu_private.h"
  18 #include "gras_modinter.h"      /* module init/exit */
  19 #include "amok/amok_modinter.h" /* module init/exit */
  20 #include "xbt_modinter.h"       /* module init/exit */
  21
  22 #include "gras.h"
  23 #include "gras/process.h"       /* FIXME: killme and put process_init in modinter */
  24 #include "gras/transport.h"
  25 #include "gras/Msg/msg_private.h"
  26 #include "portable.h"           /* hexa_*(); signalling stuff */
  27
  28 XBT_LOG_NEW_DEFAULT_CATEGORY(gras,
  29                              "All GRAS categories (cf. section \ref GRAS_API)");
  30 static int gras_running_process = 0;
  31 #if defined(HAVE_SIGNAL) && defined(HAVE_SIGNAL_H)
  32 static void gras_sigusr_handler(int sig)
  33 {
  34   XBT_INFO("SIGUSR1 received. Display the backtrace");
  35   xbt_backtrace_display_current();
  36 }
  37
  38 static void gras_sigint_handler(int sig)
  39 {
  40   static double lastone = 0;
  41   if (lastone == 0 || xbt_os_time() - lastone > 5) {
  42     if (gras_if_RL())
  43       xbt_backtrace_display_current();
  44     else
  45       SIMIX_display_process_status();
  46     fprintf(stderr,
  47             "\nBacktrace displayed because Ctrl-C was pressed. Press again (within 5 sec) to abort the process.\n");
  48     lastone = xbt_os_time();
  49   } else {
  50     exit(1);
  51   }
  52 }
  53 #endif
  54
  55 /**
  56  * @ingroup GRAS_API
  57  * \brief Initialize the gras mechanisms.
  58  */
  59 void gras_init(int *argc, char **argv)
  60 {
  61   int first = 0;
  62   gras_procdata_t *pd;
  63   gras_msg_procdata_t msg_pd;
  64
  65   xbt_getpid = gras_os_getpid;
  66   /* First initialize the XBT */
  67   xbt_init(argc, argv);
  68
  69   XBT_VERB("Initialize GRAS");
  70
  71   /* module registrations:
  72    *    - declare process specific data we need (without creating them)
  73    */
  74   if (gras_running_process == 0) {
  75     first = 1;
  76     /* Connect our log channels: that must be done manually under windows */
  77
  78     XBT_LOG_CONNECT(gras_modules, gras);
  79
  80     XBT_LOG_CONNECT(gras_msg, gras);
  81     XBT_LOG_CONNECT(gras_msg_read, gras_msg);
  82     XBT_LOG_CONNECT(gras_msg_rpc, gras_msg);
  83
  84     XBT_LOG_CONNECT(gras_timer, gras);
  85
  86     XBT_LOG_CONNECT(gras_virtu, gras);
  87     XBT_LOG_CONNECT(gras_virtu_emul, gras_virtu);
  88     XBT_LOG_CONNECT(gras_virtu_process, gras_virtu);
  89
  90     if (!getenv("GRAS_NO_WARN_EXPERIMENTAL"))
  91       XBT_WARN("GRAS is not well maintained anymore. We consider it to be experimental (and not stable anymore) since SimGrid 3.6. Sorry about it, please consider contributing to improve this situation");
  92
  93     gras_trp_register();
  94     gras_msg_register();
  95
  96     xbt_trp_plugin_new("file", gras_trp_file_setup);
  97     if (gras_if_SG()) {
  98       xbt_trp_plugin_new("sg", gras_trp_sg_setup);
  99     }
 100     /* the TCP plugin (used in RL mode) is automatically loaded by XBT */
 101   }
 102   gras_running_process++;
 103
 104   /*
 105    * Initialize the process specific stuff
 106    */
 107   gras_process_init();          /* calls procdata_init, which creates process specific data for each module */
 108
 109   /*
 110    * Initialize the global stuff if it's not the first process created
 111    */
 112   if (first) {
 113     gras_emul_init();
 114     gras_msg_init();
 115 #if defined(HAVE_SIGNAL) && defined(HAVE_SIGNAL_H)
 116 # ifdef SIGUSR1
 117     signal(SIGUSR1, gras_sigusr_handler);
 118 # endif
 119     signal(SIGINT, gras_sigint_handler);
 120 #endif
 121   }
 122
 123   /* and then init amok */
 124   amok_init();
 125
 126   /* And finally, launch the listener thread */
 127   pd = gras_procdata_get();
 128   msg_pd = gras_libdata_by_name("gras_msg");
 129   pd->listener = gras_msg_listener_launch(msg_pd->msg_received);
 130 }
 131
 132 /**
 133  * @ingroup GRAS_API
 134  * @brief Finalize the gras mechanisms.
 135  * */
 136 void gras_exit(void)
 137 {
 138   XBT_INFO("Exiting GRAS");
 139   amok_exit();
 140   gras_moddata_leave();
 141   gras_msg_listener_shutdown();
 142   gras_process_exit();
 143   if (--gras_running_process == 0) {
 144     gras_msg_exit();
 145     gras_emul_exit();
 146     gras_moddata_exit();
 147   }
 148 }
 149
 150 const char *hexa_str(unsigned char *data, int size, int downside)
 151 {
 152   static char *buff = NULL;
 153   static int buffsize = 0;
 154   int i, pos = 0;
 155   int begin, increment;
 156
 157   if (buffsize < 5 * (size + 1)) {
 158     free(buff);
 159     buffsize = 5 * (size + 1);
 160     buff = xbt_malloc(buffsize);
 161   }
 162
 163
 164   if (downside) {
 165     begin = size - 1;
 166     increment = -1;
 167   } else {
 168     begin = 0;
 169     increment = 1;
 170   }
 171
 172   for (i = begin; 0 <= i && i < size; i += increment) {
 173     if (data[i] < 32 || data[i] > 126)
 174       sprintf(buff + pos, ".");
 175     else
 176       sprintf(buff + pos, "%c", data[i]);
 177     while (buff[++pos]);
 178   }
 179   sprintf(buff + pos, "(");
 180   while (buff[++pos]);
 181   for (i = begin; 0 <= i && i < size; i += increment) {
 182     sprintf(buff + pos, "%02x", data[i]);
 183     while (buff[++pos]);
 184   }
 185   sprintf(buff + pos, ")");
 186   while (buff[++pos]);
 187   buff[pos] = '\0';
 188   return buff;
 189 }
 190
 191 void hexa_print(const char *name, unsigned char *data, int size)
 192 {
 193   printf("%s: %s\n", name, hexa_str(data, size, 0));
 194 }