src/msg/msg_global.cpp

   1 /* Copyright (c) 2004-2015. 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 MSG_Global_t msg_global = nullptr;
  17 static void MSG_exit();
  18
  19 /********************************* MSG **************************************/
  20
  21 static void _sg_cfg_cb_msg_debug_multiple_use(const char *name)
  22 {
  23   msg_global->debug_multiple_use = xbt_cfg_get_boolean(name);
  24 }
  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(argc, argv);
  34
  35   xbt_getpid = &MSG_process_self_PID;
  36   if (!msg_global) {
  37
  38     msg_global = xbt_new0(s_MSG_Global_t, 1);
  39
  40     xbt_cfg_register_boolean("msg/debug-multiple-use", "no", _sg_cfg_cb_msg_debug_multiple_use,
  41         "Print backtraces of both processes when there is a conflict of multiple use of a task");
  42
  43     SIMIX_global_init(argc, argv);
  44
  45     msg_global->sent_msg = 0;
  46     msg_global->task_copy_callback = nullptr;
  47     msg_global->process_data_cleanup = nullptr;
  48
  49     SIMIX_function_register_process_create(MSG_process_create_from_SIMIX);
  50     SIMIX_function_register_process_cleanup(MSG_process_cleanup_from_SIMIX);
  51
  52     simgrid::s4u::onPlatformCreated.connect(MSG_post_create_environment);
  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   XBT_DEBUG("ADD MSG LEVELS");
  66   MSG_STORAGE_LEVEL = xbt_lib_add_level(storage_lib, (void_f_pvoid_t) __MSG_storage_destroy);
  67   MSG_FILE_LEVEL = xbt_lib_add_level(file_lib, (void_f_pvoid_t) __MSG_file_destroy);
  68   if(xbt_cfg_get_boolean("clean-atexit"))
  69     atexit(MSG_exit);
  70 }
  71
  72 /** \ingroup msg_simulation
  73  * \brief Launch the MSG simulation
  74  */
  75 msg_error_t MSG_main()
  76 {
  77   /* Clean IO before the run */
  78   fflush(stdout);
  79   fflush(stderr);
  80
  81   if (MC_is_active()) {
  82     MC_run();
  83   } else {
  84     SIMIX_run();
  85   }
  86   return MSG_OK;
  87 }
  88
  89 /** \ingroup msg_simulation
  90  * \brief set a configuration variable
  91  *
  92  * Do --help on any simgrid binary to see the list of currently existing configuration variables, and see Section @ref options.
  93  *
  94  * Example:
  95  * MSG_config("host/model","ptask_L07");
  96  */
  97 void MSG_config(const char *key, const char *value){
  98   xbt_assert(msg_global,"ERROR: Please call MSG_init() before using MSG_config()");
  99   xbt_cfg_set_as_string(key, value);
 100 }
 101
 102 /** \ingroup msg_simulation
 103  * \brief Kill all running process
 104
 105  * \param reset_PIDs should we reset the PID numbers. A negative
 106  *   number means no reset and a positive number will be used to set the PID
 107  *   of the next newly created process.
 108  */
 109 int MSG_process_killall(int reset_PIDs)
 110 {
 111   simcall_process_killall(reset_PIDs);
 112
 113   return 0;
 114 }
 115
 116 static void MSG_exit() {
 117   if (msg_global==nullptr)
 118     return;
 119
 120   TRACE_surf_resource_utilization_release();
 121   TRACE_end();
 122   free(msg_global);
 123   msg_global = nullptr;
 124 }
 125
 126 /** \ingroup msg_simulation
 127  * \brief A clock (in second).
 128  */
 129 double MSG_get_clock()
 130 {
 131   return SIMIX_get_clock();
 132 }
 133
 134 unsigned long int MSG_get_sent_msg()
 135 {
 136   return msg_global->sent_msg;
 137 }