Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
1a51d042ed577a28dae86325178dcaf66da50533
[simgrid.git] / src / kernel / context / Context.cpp
1 /* Copyright (c) 2007-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
8 #include "simgrid/s4u/Host.hpp"
9 #include "src/kernel/activity/CommImpl.hpp"
10 #include "src/kernel/context/Context.hpp"
11 #include "src/simix/smx_private.hpp"
12 #include "src/surf/surf_interface.hpp"
13
14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context);
15
16
17 namespace simgrid {
18 namespace kernel {
19 namespace context {
20
21 ContextFactoryInitializer factory_initializer = nullptr;
22
23 ContextFactory::~ContextFactory() = default;
24
25 static thread_local smx_context_t smx_current_context = nullptr;
26 Context* Context::self()
27 {
28   return smx_current_context;
29 }
30 void Context::set_current(Context* self)
31 {
32   smx_current_context = self;
33 }
34
35 void Context::declare_context(std::size_t size)
36 {
37 #if SIMGRID_HAVE_MC
38   /* Store the address of the stack in heap to compare it apart of heap comparison */
39   if(MC_is_active())
40     MC_ignore_heap(this, size);
41 #endif
42 }
43
44 Context* ContextFactory::attach(smx_actor_t)
45 {
46   xbt_die("Cannot attach with this ContextFactory.\n"
47     "Try using --cfg=contexts/factory:thread instead.\n");
48 }
49
50 Context* ContextFactory::create_maestro(std::function<void()>, smx_actor_t)
51 {
52   xbt_die("Cannot create_maestro with this ContextFactory.\n"
53     "Try using --cfg=contexts/factory:thread instead.\n");
54 }
55
56 Context::Context(std::function<void()> code, smx_actor_t actor) : code_(std::move(code)), actor_(actor)
57 {
58   /* If no function was provided, this is the context for maestro
59    * and we should set it as the current context */
60   if (not has_code())
61     set_current(this);
62 }
63
64 Context::~Context()
65 {
66   if (self() == this)
67     set_current(nullptr);
68 }
69
70 void Context::stop()
71 {
72   actor_->finished_ = true;
73
74   if (actor_->has_to_auto_restart() && not actor_->get_host()->is_on()) {
75     XBT_DEBUG("Insert host %s to watched_hosts because it's off and %s needs to restart",
76               actor_->get_host()->get_cname(), actor_->get_cname());
77     watched_hosts.insert(actor_->get_host()->get_name());
78   }
79
80   // Execute the termination callbacks
81   smx_process_exit_status_t exit_status = (actor_->context_->iwannadie) ? SMX_EXIT_FAILURE : SMX_EXIT_SUCCESS;
82   while (not actor_->on_exit.empty()) {
83     s_smx_process_exit_fun_t exit_fun = actor_->on_exit.back();
84     actor_->on_exit.pop_back();
85     (exit_fun.fun)(exit_status, exit_fun.arg);
86   }
87
88   /* cancel non-blocking activities */
89   for (auto activity : actor_->comms)
90     boost::static_pointer_cast<activity::CommImpl>(activity)->cancel();
91   actor_->comms.clear();
92
93   XBT_DEBUG("%s@%s(%ld) should not run anymore", actor_->get_cname(), actor_->iface()->get_host()->get_cname(),
94             actor_->get_pid());
95
96   if (this->actor_ == simix_global->maestro_process) /* Do not cleanup maestro */
97     this->actor_->cleanup();
98
99   this->iwannadie = false; // don't let the simcall's yield() do a Context::stop(), because that's me
100   simgrid::simix::simcall([this] {
101     simgrid::s4u::Actor::on_destruction(actor_->iface());
102
103     /* Unregister from the kill timer if any */
104     if (actor_->kill_timer != nullptr) {
105       SIMIX_timer_remove(actor_->kill_timer);
106       actor_->kill_timer = nullptr;
107     }
108
109     actor_->cleanup();
110   });
111   this->iwannadie = true;
112 }
113
114 AttachContext::~AttachContext() = default;
115
116 StopRequest::~StopRequest() = default;
117
118 void StopRequest::do_throw()
119 {
120   throw StopRequest();
121 }
122
123 bool StopRequest::try_n_catch(std::function<void(void)> try_block)
124 {
125   bool res;
126   try {
127     try_block();
128     res = true;
129   } catch (StopRequest const&) {
130     XBT_DEBUG("Caught a StopRequest");
131     res = false;
132   }
133   return res;
134 }
135 }}}
136
137 /** @brief Executes all the processes to run (in parallel if possible). */
138 void SIMIX_context_runall()
139 {
140   simix_global->context_factory->run_all();
141 }