src/kernel/actor/ActorImpl.hpp

   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 #ifndef SIMIX_ACTORIMPL_H
   7 #define SIMIX_ACTORIMPL_H
   8
   9 #include "simgrid/s4u/Actor.hpp"
  10 #include "src/simix/popping_private.hpp"
  11 #include "src/surf/PropertyHolder.hpp"
  12 #include <boost/intrusive/list.hpp>
  13 #include <functional>
  14 #include <list>
  15 #include <map>
  16 #include <memory>
  17
  18 namespace simgrid {
  19 namespace kernel {
  20 namespace actor {
  21
  22 class XBT_PUBLIC ActorImpl : public surf::PropertyHolder {
  23   s4u::Host* host_   = nullptr; /* the host on which the actor is running */
  24   void* userdata_    = nullptr; /* kept for compatibility, it should be replaced with moddata */
  25   aid_t pid_         = 0;
  26   aid_t ppid_        = -1;
  27   bool daemon_       = false; /* Daemon actors are automatically killed when the last non-daemon leaves */
  28   bool auto_restart_ = false;
  29
  30 public:
  31   xbt::string name_;
  32   ActorImpl(const xbt::string& name, s4u::Host* host);
  33   ActorImpl(const ActorImpl&) = delete;
  34   ActorImpl& operator=(const ActorImpl&) = delete;
  35   ~ActorImpl();
  36
  37   double get_kill_time();
  38   void set_kill_time(double kill_time);
  39   boost::intrusive::list_member_hook<> host_process_list_hook; /* simgrid::simix::Host::process_list */
  40   boost::intrusive::list_member_hook<> smx_destroy_list_hook;  /* simix_global->actors_to_destroy */
  41   boost::intrusive::list_member_hook<> smx_synchro_hook;       /* {mutex,cond,sem}->sleeping */
  42
  43   const xbt::string& get_name() const { return name_; }
  44   const char* get_cname() const { return name_.c_str(); }
  45
  46   // Accessors to private fields
  47   s4u::Host* get_host() { return host_; }
  48   void set_host(s4u::Host* dest);
  49   void* get_user_data() { return userdata_; }
  50   void set_user_data(void* data) { userdata_ = data; }
  51   aid_t get_pid() const { return pid_; }
  52   aid_t get_ppid() const { return ppid_; }
  53   void set_ppid(aid_t ppid) { ppid_ = ppid; }
  54   bool is_daemon() { return daemon_; } /** Whether this actor has been daemonized */
  55   bool has_to_auto_restart() { return auto_restart_; }
  56   void set_auto_restart(bool autorestart) { auto_restart_ = autorestart; }
  57
  58   std::unique_ptr<context::Context> context_; /* the context (uctx/raw/thread) that executes the user function */
  59
  60   std::exception_ptr exception_;
  61   bool finished_  = false;
  62   bool suspended_ = false;
  63
  64   activity::ActivityImplPtr waiting_synchro = nullptr; /* the current blocking synchro if any */
  65   std::list<activity::ActivityImplPtr> comms;          /* the current non-blocking communication synchros */
  66   s_smx_simcall simcall;
  67   /* list of functions executed when the process dies */
  68   std::shared_ptr<std::vector<std::function<void(bool)>>> on_exit =
  69       std::make_shared<std::vector<std::function<void(bool)>>>();
  70
  71   std::function<void()> code_;
  72   simix::Timer* kill_timer = nullptr;
  73
  74 private:
  75   /* Refcounting */
  76   std::atomic_int_fast32_t refcount_{0};
  77
  78 public:
  79   int get_refcount() { return refcount_; }
  80   friend void intrusive_ptr_add_ref(ActorImpl* actor)
  81   {
  82     // std::memory_order_relaxed ought to be enough here instead of std::memory_order_seq_cst
  83     // But then, we have a threading issue when an actor commits a suicide:
  84     //  it seems that in this case, the worker thread kills the last occurrence of the actor
  85     //  while usually, the maestro does so. FIXME: we should change how actors suicide
  86     actor->refcount_.fetch_add(1, std::memory_order_seq_cst);
  87   }
  88   friend void intrusive_ptr_release(ActorImpl* actor)
  89   {
  90     // inspired from http://www.boost.org/doc/libs/1_55_0/doc/html/atomic/usage_examples.html
  91     if (actor->refcount_.fetch_sub(1, std::memory_order_release) == 1) {
  92       // Make sure that any changes done on other threads before their acquire are committed before our delete
  93       // http://stackoverflow.com/questions/27751025/why-is-an-acquire-barrier-needed-before-deleting-the-data-in-an-atomically-refer
  94       std::atomic_thread_fence(std::memory_order_acquire);
  95       delete actor;
  96     }
  97   }
  98
  99   /* S4U/implem interfaces */
 100 private:
 101   s4u::Actor piface_; // Our interface is part of ourselves
 102
 103   void undaemonize();
 104
 105 public:
 106   s4u::ActorPtr iface() { return s4u::ActorPtr(&piface_); }
 107   s4u::Actor* ciface() { return &piface_; }
 108
 109   ActorImplPtr init(const std::string& name, s4u::Host* host);
 110   ActorImpl* start(const simix::ActorCode& code);
 111
 112   static ActorImplPtr create(const std::string& name, const simix::ActorCode& code, void* data, s4u::Host* host,
 113                              const std::unordered_map<std::string, std::string>* properties, ActorImpl* parent_actor);
 114   static ActorImplPtr attach(const std::string& name, void* data, s4u::Host* host,
 115                              const std::unordered_map<std::string, std::string>* properties);
 116   static void detach();
 117   void cleanup();
 118   void exit();
 119   void kill(ActorImpl* actor);
 120   void kill_all();
 121
 122   void yield();
 123   void daemonize();
 124   bool is_suspended() { return suspended_; }
 125   s4u::Actor* restart();
 126   activity::ActivityImplPtr suspend(ActorImpl* issuer);
 127   void resume();
 128   activity::ActivityImplPtr join(ActorImpl* actor, double timeout);
 129   activity::ActivityImplPtr sleep(double duration);
 130   /** Ask the actor to throw an exception right away */
 131   void throw_exception(std::exception_ptr e);
 132 };
 133
 134 class ProcessArg {
 135 public:
 136   std::string name;
 137   std::function<void()> code;
 138   void* data                                                               = nullptr;
 139   s4u::Host* host                                                          = nullptr;
 140   double kill_time                                                         = 0.0;
 141   std::shared_ptr<const std::unordered_map<std::string, std::string>> properties = nullptr;
 142   bool auto_restart                                                        = false;
 143   bool daemon_                                                             = false;
 144   /* list of functions executed when the process dies */
 145   const std::shared_ptr<std::vector<std::function<void(bool)>>> on_exit;
 146
 147   ProcessArg()                                                             = default;
 148
 149   explicit ProcessArg(const std::string& name, const std::function<void()>& code, void* data, s4u::Host* host,
 150                       double kill_time, std::shared_ptr<std::unordered_map<std::string, std::string>> properties,
 151                       bool auto_restart)
 152       : name(name)
 153       , code(code)
 154       , data(data)
 155       , host(host)
 156       , kill_time(kill_time)
 157       , properties(properties)
 158       , auto_restart(auto_restart)
 159   {
 160   }
 161
 162   explicit ProcessArg(s4u::Host* host, ActorImpl* actor)
 163       : name(actor->get_name())
 164       , code(actor->code_)
 165       , data(actor->get_user_data())
 166       , host(host)
 167       , kill_time(actor->get_kill_time())
 168       , auto_restart(actor->has_to_auto_restart())
 169       , daemon_(actor->is_daemon())
 170       , on_exit(actor->on_exit)
 171   {
 172     properties.reset(actor->get_properties(), [](decltype(actor->get_properties())) {});
 173   }
 174 };
 175
 176 /* Used to keep the list of actors blocked on a synchro  */
 177 typedef boost::intrusive::list<ActorImpl, boost::intrusive::member_hook<ActorImpl, boost::intrusive::list_member_hook<>,
 178                                                                         &ActorImpl::smx_synchro_hook>>
 179     SynchroList;
 180
 181 XBT_PUBLIC void create_maestro(const std::function<void()>& code);
 182 XBT_PUBLIC int get_maxpid();
 183 } // namespace actor
 184 } // namespace kernel
 185 } // namespace simgrid
 186
 187 extern void (*SMPI_switch_data_segment)(simgrid::s4u::ActorPtr actor);
 188
 189 #endif