Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Boost contexts: add an assert checking that we don't mix up originating context.
[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   BoostContext* asan_ctx_ = nullptr;
62   bool asan_stop_         = false;
63 #endif
64
65   static void wrapper(arg_type arg);
66 };
67
68 class SerialBoostContext : public BoostContext {
69 public:
70   SerialBoostContext(std::function<void()> code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process)
71       : BoostContext(std::move(code), cleanup_func, process)
72   {
73   }
74   void suspend() override;
75   void resume() override;
76
77   static void run_all();
78
79 private:
80   static unsigned long process_index_;
81 };
82
83 #if HAVE_THREAD_CONTEXTS
84 class ParallelBoostContext : public BoostContext {
85 public:
86   ParallelBoostContext(std::function<void()> code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process)
87       : BoostContext(std::move(code), cleanup_func, process)
88   {
89   }
90   void suspend() override;
91   void resume() override;
92
93   static void initialize();
94   static void finalize();
95   static void run_all();
96
97 private:
98   static simgrid::xbt::Parmap<smx_actor_t>* parmap_;
99   static std::vector<ParallelBoostContext*> workers_context_;
100   static std::atomic<uintptr_t> threads_working_;
101   static xbt_os_thread_key_t worker_id_key_;
102 };
103 #endif
104
105 class BoostContextFactory : public ContextFactory {
106 public:
107   BoostContextFactory();
108   ~BoostContextFactory() override;
109   Context* create_context(std::function<void()> code, void_pfn_smxprocess_t cleanup, smx_actor_t process) override;
110   void run_all() override;
111
112 private:
113   bool parallel_;
114 };
115 }}} // namespace
116
117 #endif