Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
90d6cc768cc2a2db6874125898c46ee69c2aa575
[simgrid.git] / src / kernel / context / ContextBoost.hpp
1 /* Copyright (c) 2015-2018. 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 SIMGRID_SIMIX_BOOST_CONTEXT_HPP
7 #define SIMGRID_SIMIX_BOOST_CONTEXT_HPP
8
9 #include <boost/version.hpp>
10 #if BOOST_VERSION < 106100
11 #include <boost/context/fcontext.hpp>
12 #else
13 #include <boost/context/detail/fcontext.hpp>
14 #endif
15
16 #include <atomic>
17 #include <cstdint>
18 #include <functional>
19 #include <vector>
20
21 #include <simgrid/simix.hpp>
22 #include <xbt/parmap.hpp>
23 #include <xbt/xbt_os_thread.h>
24
25 #include "Context.hpp"
26 #include "src/internal_config.h"
27
28 namespace simgrid {
29 namespace kernel {
30 namespace context {
31
32 /** @brief Userspace context switching implementation based on Boost.Context */
33 class BoostContext : public Context {
34 public:
35   BoostContext(std::function<void()> code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process);
36   ~BoostContext() override;
37   void stop() override;
38   virtual void resume() = 0;
39
40   static void swap(BoostContext* from, BoostContext* to);
41   static BoostContext* getMaestro() { return maestro_context_; }
42   static void setMaestro(BoostContext* maestro) { maestro_context_ = maestro; }
43
44 private:
45   static BoostContext* maestro_context_;
46   void* stack_ = nullptr;
47
48 #if BOOST_VERSION < 105600
49   boost::context::fcontext_t* fc_ = nullptr;
50   typedef intptr_t arg_type;
51 #elif BOOST_VERSION < 106100
52   boost::context::fcontext_t fc_;
53   typedef intptr_t arg_type;
54 #else
55   boost::context::detail::fcontext_t fc_;
56   typedef boost::context::detail::transfer_t arg_type;
57 #endif
58 #if HAVE_SANITIZE_ADDRESS_FIBER_SUPPORT
59   const void* asan_stack_ = nullptr;
60   size_t asan_stack_size_ = 0;
61   bool asan_stop_         = false;
62 #endif
63
64   static void wrapper(arg_type arg);
65 };
66
67 class SerialBoostContext : public BoostContext {
68 public:
69   SerialBoostContext(std::function<void()> code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process)
70       : BoostContext(std::move(code), cleanup_func, process)
71   {
72   }
73   void suspend() override;
74   void resume() override;
75
76   static void run_all();
77
78 private:
79   static unsigned long process_index_;
80 };
81
82 #if HAVE_THREAD_CONTEXTS
83 class ParallelBoostContext : public BoostContext {
84 public:
85   ParallelBoostContext(std::function<void()> code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process)
86       : BoostContext(std::move(code), cleanup_func, process)
87   {
88   }
89   void suspend() override;
90   void resume() override;
91
92   static void initialize();
93   static void finalize();
94   static void run_all();
95
96 private:
97   static simgrid::xbt::Parmap<smx_actor_t>* parmap_;
98   static std::vector<ParallelBoostContext*> workers_context_;
99   static std::atomic<uintptr_t> threads_working_;
100   static xbt_os_thread_key_t worker_id_key_;
101 };
102 #endif
103
104 class BoostContextFactory : public ContextFactory {
105 public:
106   BoostContextFactory();
107   ~BoostContextFactory() override;
108   Context* create_context(std::function<void()> code, void_pfn_smxprocess_t cleanup, smx_actor_t process) override;
109   void run_all() override;
110
111 private:
112   bool parallel_;
113 };
114 }}} // namespace
115
116 #endif