src/msg/msg_global.cpp

   1 /* Copyright (c) 2004-2017. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "simgrid/s4u/Engine.hpp"
   7 #include "simgrid/s4u/Host.hpp"
   8
   9 #include "instr/instr_interface.h"
  10 #include "mc/mc.h"
  11 #include "src/msg/msg_private.h"
  12
  13 XBT_LOG_NEW_CATEGORY(msg, "All MSG categories");
  14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(msg_kernel, msg, "Logging specific to MSG (kernel)");
  15
  16 SG_BEGIN_DECL()
  17
  18 MSG_Global_t msg_global = nullptr;
  19 static void MSG_exit();
  20
  21 /********************************* MSG **************************************/
  22
  23 static void _sg_cfg_cb_msg_debug_multiple_use(const char *name)
  24 {
  25   msg_global->debug_multiple_use = xbt_cfg_get_boolean(name);
  26 }
  27
  28 /**
  29  * \ingroup msg_simulation
  30  * \brief Initialize MSG with less verifications
  31  * You should use the MSG_init() function instead. Failing to do so may turn into PEBKAC some day. You've been warned.
  32  */
  33 void MSG_init_nocheck(int *argc, char **argv) {
  34
  35   TRACE_global_init(argc, argv);
  36
  37   xbt_getpid = &MSG_process_self_PID;
  38   if (not msg_global) {
  39
  40     msg_global = new s_MSG_Global_t();
  41
  42     xbt_cfg_register_boolean("msg/debug-multiple-use", "no", _sg_cfg_cb_msg_debug_multiple_use,
  43         "Print backtraces of both processes when there is a conflict of multiple use of a task");
  44
  45     SIMIX_global_init(argc, argv);
  46
  47     msg_global->sent_msg = 0;
  48     msg_global->task_copy_callback = nullptr;
  49     msg_global->process_data_cleanup = nullptr;
  50
  51     SIMIX_function_register_process_create(MSG_process_create_from_SIMIX);
  52     SIMIX_function_register_process_cleanup(MSG_process_cleanup_from_SIMIX);
  53
  54     simgrid::MsgHostExt::EXTENSION_ID = simgrid::s4u::Host::extension_create<simgrid::MsgHostExt>();
  55     simgrid::s4u::Host::onCreation.connect([](simgrid::s4u::Host& host) {
  56       host.extension_set<simgrid::MsgHostExt>(new simgrid::MsgHostExt());
  57     });
  58   }
  59
  60   if(MC_is_active()){
  61     /* Ignore total amount of messages sent during the simulation for heap comparison */
  62     MC_ignore_heap(&(msg_global->sent_msg), sizeof(msg_global->sent_msg));
  63   }
  64
  65   if (xbt_cfg_get_boolean("clean-atexit"))
  66     atexit(MSG_exit);
  67 }
  68
  69 /** \ingroup msg_simulation
  70  * \brief Launch the MSG simulation
  71  */
  72 msg_error_t MSG_main()
  73 {
  74   /* Clean IO before the run */
  75   fflush(stdout);
  76   fflush(stderr);
  77
  78   if (MC_is_active()) {
  79     MC_run();
  80   } else {
  81     SIMIX_run();
  82   }
  83   return MSG_OK;
  84 }
  85
  86 /** \ingroup msg_simulation
  87  * \brief set a configuration variable
  88  *
  89  * Do --help on any simgrid binary to see the list of currently existing configuration variables, and see Section @ref options.
  90  *
  91  * Example:
  92  * MSG_config("host/model","ptask_L07");
  93  */
  94 void MSG_config(const char *key, const char *value){
  95   xbt_assert(msg_global,"ERROR: Please call MSG_init() before using MSG_config()");
  96   xbt_cfg_set_as_string(key, value);
  97 }
  98
  99 /** \ingroup msg_simulation
 100  * \brief Kill all running process
 101
 102  * \param reset_PIDs should we reset the PID numbers. A negative
 103  *   number means no reset and a positive number will be used to set the PID
 104  *   of the next newly created process.
 105  */
 106 int MSG_process_killall(int reset_PIDs)
 107 {
 108   simcall_process_killall(reset_PIDs);
 109
 110   return 0;
 111 }
 112
 113 static void MSG_exit() {
 114   if (msg_global==nullptr)
 115     return;
 116
 117   TRACE_surf_resource_utilization_release();
 118   TRACE_end();
 119   delete msg_global;
 120   msg_global = nullptr;
 121 }
 122
 123 /** \ingroup msg_simulation
 124  * \brief A clock (in second).
 125  */
 126 double MSG_get_clock()
 127 {
 128   return SIMIX_get_clock();
 129 }
 130
 131 unsigned long int MSG_get_sent_msg()
 132 {
 133   return msg_global->sent_msg;
 134 }
 135
 136 SG_END_DECL()