src/msg/msg_global.cpp

   1 /* Copyright (c) 2004-2019. 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 "mc/mc.h"
   7 #include "simgrid/s4u/Engine.hpp"
   8 #include "simgrid/s4u/Host.hpp"
   9 #include "src/instr/instr_private.hpp"
  10 #include "src/msg/msg_private.hpp"
  11 #include "src/simix/smx_private.hpp"
  12 #include <xbt/config.hpp>
  13
  14 XBT_LOG_NEW_CATEGORY(msg, "All MSG categories");
  15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(msg_kernel, msg, "Logging specific to MSG (kernel)");
  16
  17 bool MSG_Global_t::debug_multiple_use = false;
  18
  19 MSG_Global_t* msg_global = nullptr;
  20 simgrid::xbt::Extension<simgrid::s4u::Actor, simgrid::msg::ActorUserData> simgrid::msg::ActorUserData::EXTENSION_ID;
  21
  22 static void MSG_exit();
  23
  24 /********************************* MSG **************************************/
  25
  26 /**
  27  * @ingroup msg_simulation
  28  * @brief Initialize MSG with less verifications
  29  * You should use the MSG_init() function instead. Failing to do so may turn into PEBKAC some day. You've been warned.
  30  */
  31 void MSG_init_nocheck(int *argc, char **argv) {
  32
  33   TRACE_global_init();
  34
  35   if (not msg_global) {
  36     simgrid::config::bind_flag(MSG_Global_t::debug_multiple_use, "msg/debug-multiple-use",
  37                                "Print backtraces of both processes when there is a conflict of multiple use of a task");
  38
  39     SIMIX_global_init(argc, argv);
  40
  41     msg_global = new MSG_Global_t();
  42     if (not simgrid::msg::ActorUserData::EXTENSION_ID.valid())
  43       simgrid::msg::ActorUserData::EXTENSION_ID = simgrid::s4u::Actor::extension_create<simgrid::msg::ActorUserData>();
  44
  45     msg_global->sent_msg = 0;
  46     msg_global->task_copy_callback = nullptr;
  47     msg_global->process_data_cleanup = nullptr;
  48
  49     simgrid::s4u::Actor::on_creation.connect([](simgrid::s4u::Actor& actor) {
  50       XBT_DEBUG("creating the extension to store user data");
  51       actor.extension_set(new simgrid::msg::ActorUserData());
  52     });
  53
  54     simgrid::s4u::Actor::on_destruction.connect([](simgrid::s4u::Actor const& actor) {
  55       // free the data if a function was provided
  56       void* userdata = actor.extension<simgrid::msg::ActorUserData>()->get_user_data();
  57       if (userdata && msg_global->process_data_cleanup) {
  58         msg_global->process_data_cleanup(userdata);
  59       }
  60     });
  61   }
  62
  63   if(MC_is_active()){
  64     /* Ignore total amount of messages sent during the simulation for heap comparison */
  65     MC_ignore_heap(&(msg_global->sent_msg), sizeof(msg_global->sent_msg));
  66   }
  67
  68   if (simgrid::config::get_value<bool>("clean-atexit"))
  69     atexit(MSG_exit);
  70 }
  71
  72 void MSG_config(const char *key, const char *value){
  73   xbt_assert(msg_global,"ERROR: Please call MSG_init() before using MSG_config()");
  74   simgrid::config::set_as_string(key, value);
  75 }
  76
  77 static void MSG_exit() {
  78   delete msg_global;
  79   msg_global = nullptr;
  80 }
  81
  82 unsigned long int MSG_get_sent_msg()
  83 {
  84   return msg_global->sent_msg;
  85 }
  86
  87 /** @brief register functions bypassing the parser */
  88 void MSG_set_function(const char* host_id, const char* function_name, xbt_dynar_t arguments)
  89 {
  90   SIMIX_process_set_function(host_id, function_name, arguments, -1, -1);
  91 }