src/msg/instr_msg_task.cpp

   1 /* Copyright (c) 2010-2019. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "mc/mc.h"
   8 #include "src/instr/instr_private.hpp"
   9 #include "src/msg/msg_private.hpp"
  10
  11 #include <atomic>
  12
  13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(instr_msg, instr, "MSG instrumentation");
  14
  15 void TRACE_msg_set_task_category(msg_task_t task, const char *category)
  16 {
  17   xbt_assert(task->category == nullptr, "Task %p(%s) already has a category (%s).",
  18       task, task->name, task->category);
  19
  20   //if user provides a nullptr category, task is no longer traced
  21   if (category == nullptr) {
  22     xbt_free (task->category);
  23     task->category = nullptr;
  24     XBT_DEBUG("MSG task %p(%s), category removed", task, task->name);
  25   } else {
  26     // set task category
  27     task->category = xbt_strdup(category);
  28     XBT_DEBUG("MSG task %p(%s), category %s", task, task->name, task->category);
  29   }
  30 }
  31
  32 void TRACE_msg_task_get_start()
  33 {
  34   XBT_DEBUG("GET,in");
  35
  36   if (TRACE_actor_is_enabled())
  37     simgrid::instr::Container::by_name(instr_pid(MSG_process_self()))->get_state("ACTOR_STATE")->push_event("receive");
  38 }
  39
  40 void TRACE_msg_task_get_end(msg_task_t task)
  41 {
  42   XBT_DEBUG("GET,out %p, %lld, %s", task, task->counter, task->category);
  43
  44   if (TRACE_actor_is_enabled()) {
  45     container_t process_container = simgrid::instr::Container::by_name(instr_pid(MSG_process_self()));
  46     process_container->get_state("ACTOR_STATE")->pop_event();
  47
  48     std::string key = std::string("p") + std::to_string(task->counter);
  49     simgrid::instr::Container::get_root()->get_link("ACTOR_TASK_LINK")->end_event(process_container, "SR", key);
  50   }
  51 }
  52
  53 /* MSG_task_put related functions */
  54 void TRACE_msg_task_put_start(msg_task_t task)
  55 {
  56   XBT_DEBUG("PUT,in %p, %lld, %s", task, task->counter, task->category);
  57
  58   if (TRACE_actor_is_enabled()) {
  59     container_t process_container = simgrid::instr::Container::by_name(instr_pid(MSG_process_self()));
  60     std::string key = std::string("p") + std::to_string(task->counter);
  61     simgrid::instr::Container::get_root()->get_link("ACTOR_TASK_LINK")->start_event(process_container, "SR", key);
  62   }
  63 }
  64
  65 void TRACE_msg_task_put_end()
  66 {
  67   XBT_DEBUG("PUT,out");
  68
  69   if (TRACE_actor_is_enabled())
  70     simgrid::instr::Container::by_name(instr_pid(MSG_process_self()))->get_state("ACTOR_STATE")->pop_event();
  71 }