Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Simplify the API between Engine and EngineImpl when registering functions
[simgrid.git] / src / kernel / actor / ActorImpl.cpp
1 /* Copyright (c) 2007-2020. 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/Exception.hpp"
8 #include "simgrid/s4u/Actor.hpp"
9 #include "simgrid/s4u/Exec.hpp"
10 #include "src/kernel/activity/CommImpl.hpp"
11 #include "src/kernel/activity/ExecImpl.hpp"
12 #include "src/kernel/activity/IoImpl.hpp"
13 #include "src/kernel/activity/SleepImpl.hpp"
14 #include "src/kernel/activity/SynchroRaw.hpp"
15 #include "src/mc/mc_replay.hpp"
16 #include "src/mc/remote/Client.hpp"
17 #include "src/simix/smx_private.hpp"
18 #include "src/surf/HostImpl.hpp"
19 #include "src/surf/cpu_interface.hpp"
20
21 #include <boost/range/algorithm.hpp>
22 #include <utility>
23
24 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_process, simix, "Logging specific to SIMIX (process)");
25
26 /**
27  * @brief Returns the current agent.
28  *
29  * This functions returns the currently running SIMIX process.
30  *
31  * @return The SIMIX process
32  */
33 smx_actor_t SIMIX_process_self()
34 {
35   return simgrid::kernel::actor::ActorImpl::self();
36 }
37
38 namespace simgrid {
39 namespace kernel {
40 namespace actor {
41
42 static unsigned long maxpid = 0;
43 int get_maxpid()
44 {
45   return maxpid;
46 }
47
48 ActorImpl* ActorImpl::self()
49 {
50   context::Context* self_context = context::Context::self();
51
52   return (self_context != nullptr) ? self_context->get_actor() : nullptr;
53 }
54
55 ActorImpl::ActorImpl(xbt::string name, s4u::Host* host) : host_(host), name_(std::move(name)), piface_(this)
56 {
57   pid_           = maxpid++;
58   simcall.issuer_ = this;
59 }
60
61 ActorImpl::~ActorImpl()
62 {
63   if (simix_global != nullptr && this != simix_global->maestro_)
64     s4u::Actor::on_destruction(*ciface());
65 }
66
67 /* Become an actor in the simulation
68  *
69  * Currently this can only be called by the main thread (once) and only work with some thread factories
70  * (currently ThreadContextFactory).
71  *
72  * In the future, it might be extended in order to attach other threads created by a third party library.
73  */
74
75 ActorImplPtr ActorImpl::attach(const std::string& name, void* data, s4u::Host* host,
76                                const std::unordered_map<std::string, std::string>* properties)
77 {
78   // This is mostly a copy/paste from create(), it'd be nice to share some code between those two functions.
79
80   XBT_DEBUG("Attach actor %s on host '%s'", name.c_str(), host->get_cname());
81
82   if (not host->is_on()) {
83     XBT_WARN("Cannot attach actor '%s' on failed host '%s'", name.c_str(), host->get_cname());
84     throw HostFailureException(XBT_THROW_POINT, "Cannot attach actor on failed host.");
85   }
86
87   ActorImpl* actor = new ActorImpl(xbt::string(name), host);
88   /* Actor data */
89   actor->set_user_data(data);
90   actor->code_ = nullptr;
91
92   XBT_VERB("Create context %s", actor->get_cname());
93   xbt_assert(simix_global != nullptr, "simix is not initialized, please call MSG_init first");
94   actor->context_.reset(simix_global->context_factory->attach(actor));
95
96   /* Add properties */
97   if (properties != nullptr)
98     actor->set_properties(*properties);
99
100   /* Add the actor to it's host actor list */
101   host->pimpl_->add_actor(actor);
102
103   /* Now insert it in the global actor list and in the actors to run list */
104   simix_global->process_list[actor->get_pid()] = actor;
105   XBT_DEBUG("Inserting [%p] %s(%s) in the to_run list", actor, actor->get_cname(), host->get_cname());
106   simix_global->actors_to_run.push_back(actor);
107   intrusive_ptr_add_ref(actor);
108
109   auto* context = dynamic_cast<context::AttachContext*>(actor->context_.get());
110   xbt_assert(nullptr != context, "Not a suitable context");
111   context->attach_start();
112
113   /* The on_creation() signal must be delayed until there, where the pid and everything is set */
114   s4u::Actor::on_creation(*actor->ciface());
115
116   return ActorImplPtr(actor);
117 }
118 /** @brief Detach an actor attached with `attach()`
119  *
120  *  This is called when the current actor has finished its job.
121  *  Used in the main thread, it waits for the simulation to finish before returning. When it returns, the other
122  *  simulated actors and the maestro are destroyed.
123  */
124 void ActorImpl::detach()
125 {
126   auto* context = dynamic_cast<context::AttachContext*>(context::Context::self());
127   if (context == nullptr)
128     xbt_die("Not a suitable context");
129
130   context->get_actor()->cleanup();
131   context->attach_stop();
132 }
133
134 void ActorImpl::cleanup_from_simix()
135 {
136   const std::lock_guard<std::mutex> lock(simix_global->mutex);
137   simix_global->process_list.erase(pid_);
138   if (host_ && host_actor_list_hook.is_linked())
139     host_->pimpl_->remove_actor(this);
140   if (not smx_destroy_list_hook.is_linked()) {
141 #if SIMGRID_HAVE_MC
142     xbt_dynar_push_as(simix_global->dead_actors_vector, ActorImpl*, this);
143 #endif
144     simix_global->actors_to_destroy.push_back(*this);
145   }
146 }
147
148 void ActorImpl::cleanup()
149 {
150   finished_ = true;
151
152   if (has_to_auto_restart() && not get_host()->is_on()) {
153     XBT_DEBUG("Insert host %s to watched_hosts because it's off and %s needs to restart", get_host()->get_cname(),
154               get_cname());
155     watched_hosts.insert(get_host()->get_name());
156   }
157
158   if (on_exit) {
159     // Execute the termination callbacks
160     bool failed = context_->wannadie();
161     for (auto exit_fun = on_exit->crbegin(); exit_fun != on_exit->crend(); ++exit_fun)
162       (*exit_fun)(failed);
163     on_exit.reset();
164   }
165   undaemonize();
166
167   /* cancel non-blocking activities */
168   for (auto activity : comms)
169     boost::static_pointer_cast<activity::CommImpl>(activity)->cancel();
170   comms.clear();
171
172   XBT_DEBUG("%s@%s(%ld) should not run anymore", get_cname(), get_host()->get_cname(), get_pid());
173
174   if (this == simix_global->maestro_) /* Do not cleanup maestro */
175     return;
176
177   XBT_DEBUG("Cleanup actor %s (%p), waiting synchro %p", get_cname(), this, waiting_synchro.get());
178
179   /* Unregister from the kill timer if any */
180   if (kill_timer != nullptr) {
181     kill_timer->remove();
182     kill_timer = nullptr;
183   }
184   cleanup_from_simix();
185
186   context_->set_wannadie(false); // don't let the simcall's yield() do a Context::stop(), to avoid infinite loops
187   actor::simcall([this] { s4u::Actor::on_termination(*ciface()); });
188   context_->set_wannadie();
189 }
190
191 void ActorImpl::exit()
192 {
193   context_->set_wannadie();
194   suspended_          = false;
195   exception_          = nullptr;
196
197   /* destroy the blocking synchro if any */
198   if (waiting_synchro != nullptr) {
199     waiting_synchro->cancel();
200     waiting_synchro->state_ = activity::State::FAILED;
201
202     activity::ExecImplPtr exec   = boost::dynamic_pointer_cast<activity::ExecImpl>(waiting_synchro);
203     activity::CommImplPtr comm   = boost::dynamic_pointer_cast<activity::CommImpl>(waiting_synchro);
204
205     if (exec != nullptr) {
206       exec->clean_action();
207     } else if (comm != nullptr) {
208       comms.remove(waiting_synchro);
209       // Remove first occurrence of &actor->simcall:
210       auto i = boost::range::find(waiting_synchro->simcalls_, &simcall);
211       if (i != waiting_synchro->simcalls_.end())
212         waiting_synchro->simcalls_.remove(&simcall);
213     } else {
214       activity::ActivityImplPtr(waiting_synchro)->finish();
215     }
216
217     waiting_synchro = nullptr;
218   }
219
220   // Forcefully kill the actor if its host is turned off. Not a HostFailureException because you should not survive that
221   this->throw_exception(std::make_exception_ptr(ForcefulKillException(host_->is_on() ? "exited" : "host failed")));
222 }
223
224 void ActorImpl::kill(ActorImpl* actor)
225 {
226   xbt_assert(actor != simix_global->maestro_, "Killing maestro is a rather bad idea");
227   if (actor->finished_) {
228     XBT_DEBUG("Ignoring request to kill actor %s@%s that is already dead", actor->get_cname(),
229               actor->host_->get_cname());
230     return;
231   }
232
233   XBT_DEBUG("Actor '%s'@%s is killing actor '%s'@%s", get_cname(), host_ ? host_->get_cname() : "", actor->get_cname(),
234             actor->host_ ? actor->host_->get_cname() : "");
235
236   actor->exit();
237
238   if (std::find(begin(simix_global->actors_to_run), end(simix_global->actors_to_run), actor) ==
239           end(simix_global->actors_to_run) &&
240       actor != this) {
241     XBT_DEBUG("Inserting %s in the to_run list", actor->get_cname());
242     simix_global->actors_to_run.push_back(actor);
243   }
244 }
245
246 void ActorImpl::kill_all()
247 {
248   for (auto const& kv : simix_global->process_list)
249     if (kv.second != this)
250       this->kill(kv.second);
251 }
252
253 void ActorImpl::set_kill_time(double kill_time)
254 {
255   if (kill_time <= SIMIX_get_clock())
256     return;
257   XBT_DEBUG("Set kill time %f for actor %s@%s", kill_time, get_cname(), host_->get_cname());
258   kill_timer = simix::Timer::set(kill_time, [this] {
259     this->exit();
260     kill_timer = nullptr;
261   });
262 }
263
264 double ActorImpl::get_kill_time()
265 {
266   return kill_timer ? kill_timer->get_date() : 0;
267 }
268
269 void ActorImpl::yield()
270 {
271   XBT_DEBUG("Yield actor '%s'", get_cname());
272
273   /* Go into sleep and return control to maestro */
274   context_->suspend();
275
276   /* Ok, maestro returned control to us */
277   XBT_DEBUG("Control returned to me: '%s'", get_cname());
278
279   if (context_->wannadie()) {
280     XBT_DEBUG("Actor %s@%s is dead", get_cname(), host_->get_cname());
281     context_->stop();
282     THROW_IMPOSSIBLE;
283   }
284
285   if (suspended_) {
286     XBT_DEBUG("Hey! I'm suspended.");
287
288     xbt_assert(exception_ == nullptr, "Gasp! This exception may be lost by subsequent calls.");
289     suspended_ = false;
290     suspend();
291   }
292
293   if (exception_ != nullptr) {
294     XBT_DEBUG("Wait, maestro left me an exception");
295     std::exception_ptr exception = std::move(exception_);
296     exception_                   = nullptr;
297     std::rethrow_exception(std::move(exception));
298   }
299
300   if (SMPI_switch_data_segment && not finished_) {
301     SMPI_switch_data_segment(iface());
302   }
303 }
304
305 /** This actor will be terminated automatically when the last non-daemon actor finishes */
306 void ActorImpl::daemonize()
307 {
308   if (not daemon_) {
309     daemon_ = true;
310     simix_global->daemons.push_back(this);
311   }
312 }
313
314 void ActorImpl::undaemonize()
315 {
316   if (daemon_) {
317     auto& vect = simix_global->daemons;
318     auto it    = std::find(vect.begin(), vect.end(), this);
319     xbt_assert(it != vect.end(), "The dying daemon is not a daemon after all. Please report that bug.");
320     /* Don't move the whole content since we don't really care about the order */
321
322     std::swap(*it, vect.back());
323     vect.pop_back();
324     daemon_ = false;
325   }
326 }
327
328 s4u::Actor* ActorImpl::restart()
329 {
330   xbt_assert(this != simix_global->maestro_, "Restarting maestro is not supported");
331
332   XBT_DEBUG("Restarting actor %s on %s", get_cname(), host_->get_cname());
333
334   // retrieve the arguments of the old actor
335   ProcessArg arg = ProcessArg(host_, this);
336
337   // kill the old actor
338   context::Context::self()->get_actor()->kill(this);
339
340   // start the new actor
341   ActorImplPtr actor =
342       ActorImpl::create(arg.name, std::move(arg.code), arg.data, arg.host, arg.properties.get(), nullptr);
343   *actor->on_exit = std::move(*arg.on_exit);
344   actor->set_kill_time(arg.kill_time);
345   actor->set_auto_restart(arg.auto_restart);
346
347   return actor->ciface();
348 }
349
350 void ActorImpl::suspend()
351 {
352   if (suspended_) {
353     XBT_DEBUG("Actor '%s' is already suspended", get_cname());
354     return;
355   }
356
357   suspended_ = true;
358
359   /* If the suspended actor is waiting on a sync, suspend its synchronization. */
360   if (waiting_synchro == nullptr) {
361     auto exec = new activity::ExecImpl();
362     exec->set_name("suspend").set_host(host_).set_flops_amount(0.0).start();
363     waiting_synchro = activity::ExecImplPtr(exec);
364
365     waiting_synchro->simcalls_.push_back(&simcall);
366   }
367   waiting_synchro->suspend();
368 }
369
370 void ActorImpl::resume()
371 {
372   XBT_IN("actor = %p", this);
373
374   if (context_->wannadie()) {
375     XBT_VERB("Ignoring request to suspend an actor that is currently dying.");
376     return;
377   }
378
379   if (not suspended_)
380     return;
381   suspended_ = false;
382
383   /* resume the synchronization that was blocking the resumed actor. */
384   if (waiting_synchro)
385     waiting_synchro->resume();
386
387   XBT_OUT();
388 }
389
390 activity::ActivityImplPtr ActorImpl::join(const ActorImpl* actor, double timeout)
391 {
392   activity::ActivityImplPtr sleep = this->sleep(timeout);
393   actor->on_exit->emplace_back([sleep](bool) {
394     if (sleep->surf_action_)
395       sleep->surf_action_->finish(resource::Action::State::FINISHED);
396   });
397   return sleep;
398 }
399
400 activity::ActivityImplPtr ActorImpl::sleep(double duration)
401 {
402   if (not host_->is_on())
403     throw_exception(std::make_exception_ptr(HostFailureException(
404         XBT_THROW_POINT, std::string("Host ") + host_->get_cname() + " failed, you cannot sleep there.")));
405
406   auto sleep = new activity::SleepImpl();
407   sleep->set_name("sleep").set_host(host_).set_duration(duration).start();
408   return activity::SleepImplPtr(sleep);
409 }
410
411 void ActorImpl::throw_exception(std::exception_ptr e)
412 {
413   exception_ = e;
414
415   if (suspended_)
416     resume();
417
418   /* cancel the blocking synchro if any */
419   if (waiting_synchro) {
420     waiting_synchro->cancel();
421
422     activity::CommImplPtr comm = boost::dynamic_pointer_cast<activity::CommImpl>(waiting_synchro);
423
424     if (comm != nullptr)
425       comms.remove(comm);
426
427     waiting_synchro = nullptr;
428   }
429 }
430
431 void ActorImpl::simcall_answer()
432 {
433   if (this != simix_global->maestro_) {
434     XBT_DEBUG("Answer simcall %s (%d) issued by %s (%p)", SIMIX_simcall_name(simcall.call_), (int)simcall.call_,
435               get_cname(), this);
436     simcall.call_ = SIMCALL_NONE;
437     xbt_assert(not XBT_LOG_ISENABLED(simix_process, xbt_log_priority_debug) ||
438                    std::find(begin(simix_global->actors_to_run), end(simix_global->actors_to_run), this) ==
439                        end(simix_global->actors_to_run),
440                "Actor %p should not exist in actors_to_run!", this);
441     simix_global->actors_to_run.push_back(this);
442   }
443 }
444
445 void ActorImpl::set_host(s4u::Host* dest)
446 {
447   host_->pimpl_->remove_actor(this);
448   host_ = dest;
449   dest->pimpl_->add_actor(this);
450 }
451
452 ActorImplPtr ActorImpl::init(const std::string& name, s4u::Host* host)
453 {
454   ActorImpl* actor = new ActorImpl(xbt::string(name), host);
455   actor->set_ppid(this->pid_);
456
457   intrusive_ptr_add_ref(actor);
458   /* The on_creation() signal must be delayed until there, where the pid and everything is set */
459   s4u::Actor::on_creation(*actor->ciface());
460
461   return ActorImplPtr(actor);
462 }
463
464 ActorImpl* ActorImpl::start(const ActorCode& code)
465 {
466   xbt_assert(code && host_ != nullptr, "Invalid parameters");
467
468   if (not host_->is_on()) {
469     XBT_WARN("Cannot launch actor '%s' on failed host '%s'", name_.c_str(), host_->get_cname());
470     intrusive_ptr_release(this);
471     throw HostFailureException(XBT_THROW_POINT, "Cannot start actor on failed host.");
472   }
473
474   this->code_ = code;
475   XBT_VERB("Create context %s", get_cname());
476   context_.reset(simix_global->context_factory->create_context(ActorCode(code), this));
477
478   XBT_DEBUG("Start context '%s'", get_cname());
479
480   /* Add the actor to its host's actor list */
481   host_->pimpl_->add_actor(this);
482   simix_global->process_list[pid_] = this;
483
484   /* Now insert it in the global actor list and in the actor to run list */
485   XBT_DEBUG("Inserting [%p] %s(%s) in the to_run list", this, get_cname(), host_->get_cname());
486   simix_global->actors_to_run.push_back(this);
487
488   return this;
489 }
490
491 ActorImplPtr ActorImpl::create(const std::string& name, const ActorCode& code, void* data, s4u::Host* host,
492                                const std::unordered_map<std::string, std::string>* properties, ActorImpl* parent_actor)
493 {
494   XBT_DEBUG("Start actor %s@'%s'", name.c_str(), host->get_cname());
495
496   ActorImplPtr actor;
497   if (parent_actor != nullptr)
498     actor = parent_actor->init(xbt::string(name), host);
499   else
500     actor = self()->init(xbt::string(name), host);
501
502   /* actor data */
503   actor->set_user_data(data);
504
505   /* Add properties */
506   if (properties != nullptr)
507     actor->set_properties(*properties);
508
509   actor->start(code);
510
511   return actor;
512 }
513
514 void create_maestro(const std::function<void()>& code)
515 {
516   /* Create maestro actor and initialize it */
517   ActorImpl* maestro = new ActorImpl(xbt::string(""), /*host*/ nullptr);
518
519   if (not code) {
520     maestro->context_.reset(simix_global->context_factory->create_context(ActorCode(), maestro));
521   } else {
522     maestro->context_.reset(simix_global->context_factory->create_maestro(ActorCode(code), maestro));
523   }
524
525   maestro->simcall.issuer_      = maestro;
526   simix_global->maestro_        = maestro;
527 }
528
529 } // namespace actor
530 } // namespace kernel
531 } // namespace simgrid
532
533 int SIMIX_process_count() // XBT_ATTRIB_DEPRECATED_v329
534 {
535   return simix_global->process_list.size();
536 }
537
538 // XBT_DEPRECATED_v329
539 void* SIMIX_process_self_get_data()
540 {
541   smx_actor_t self = simgrid::kernel::actor::ActorImpl::self();
542
543   if (self == nullptr) {
544     return nullptr;
545   }
546   return self->get_user_data();
547 }
548
549 // XBT_DEPRECATED_v329
550 void SIMIX_process_self_set_data(void* data)
551 {
552   simgrid::kernel::actor::ActorImpl::self()->set_user_data(data);
553 }
554
555 /* needs to be public and without simcall because it is called
556    by exceptions and logging events */
557 const char* SIMIX_process_self_get_name()
558 {
559   return SIMIX_is_maestro() ? "maestro" : simgrid::kernel::actor::ActorImpl::self()->get_cname();
560 }
561
562 /**
563  * @brief Calling this function makes the process to yield.
564  *
565  * Only the current process can call this function, giving back the control to maestro.
566  *
567  * @param self the current process
568  */
569
570 /** @brief Returns the process from PID. */
571 smx_actor_t SIMIX_process_from_PID(aid_t PID)
572 {
573   auto item = simix_global->process_list.find(PID);
574   if (item == simix_global->process_list.end()) {
575     for (auto& a : simix_global->actors_to_destroy)
576       if (a.get_pid() == PID)
577         return &a;
578     return nullptr; // Not found, even in the trash
579   }
580   return item->second;
581 }
582
583 void SIMIX_process_on_exit(smx_actor_t actor,
584                            const std::function<void(bool /*failed*/)>& fun) // XBT_ATTRIB_DEPRECATED_v329
585 {
586   xbt_assert(actor, "current process not found: are you in maestro context ?");
587   actor->on_exit->emplace_back(fun);
588 }
589
590 /** @brief Restart a process, starting it again from the beginning. */
591 /**
592  * @ingroup simix_process_management
593  * @brief Creates and runs a new SIMIX process.
594  *
595  * The structure and the corresponding thread are created and put in the list of ready processes.
596  *
597  * @param name a name for the process. It is for user-level information and can be nullptr.
598  * @param code the main function of the process
599  * @param data a pointer to any data one may want to attach to the new object. It is for user-level information and can
600  * be nullptr.
601  * It can be retrieved with the method ActorImpl::getUserData().
602  * @param host where the new agent is executed.
603  * @param properties the properties of the process
604  */
605 smx_actor_t simcall_process_create(const std::string& name, const simgrid::kernel::actor::ActorCode& code, void* data,
606                                    sg_host_t host, std::unordered_map<std::string, std::string>* properties)
607 {
608   smx_actor_t self = simgrid::kernel::actor::ActorImpl::self();
609   return simgrid::kernel::actor::simcall([&name, &code, data, host, properties, self] {
610     return simgrid::kernel::actor::ActorImpl::create(name, code, data, host, properties, self).get();
611   });
612 }
613
614 void simcall_process_set_data(smx_actor_t process, void* data)
615 {
616   simgrid::kernel::actor::simcall([process, data] { process->set_user_data(data); });
617 }