X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/c93f29cd7a1453e10d8635f11e48cf0d537a83a2..4d5c6aed22b8ae1c8a73cd04d8ab800322d177bf:/src/kernel/context/ContextBoost.cpp diff --git a/src/kernel/context/ContextBoost.cpp b/src/kernel/context/ContextBoost.cpp index 6a8beb7c37..e9ea749708 100644 --- a/src/kernel/context/ContextBoost.cpp +++ b/src/kernel/context/ContextBoost.cpp @@ -1,24 +1,12 @@ -/* Copyright (c) 2015-2017. The SimGrid Team. All rights reserved. */ +/* Copyright (c) 2015-2018. The SimGrid Team. All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ #include "ContextBoost.hpp" - -#include -#include - -#if HAVE_SANITIZE_ADDRESS_FIBER_SUPPORT -#include -#define ASAN_EVAL(expr) (expr) -#define ASAN_START_SWITCH(fake_stack_save, bottom, size) __sanitizer_start_switch_fiber(fake_stack_save, bottom, size) -#define ASAN_FINISH_SWITCH(fake_stack_save, bottom_old, size_old) \ - __sanitizer_finish_switch_fiber(fake_stack_save, bottom_old, size_old) -#else -#define ASAN_EVAL(expr) (void)0 -#define ASAN_START_SWITCH(fake_stack_save, bottom, size) (void)0 -#define ASAN_FINISH_SWITCH(fake_stack_save, bottom_old, size_old) (void)(fake_stack_save) -#endif +#include "context_private.hpp" +#include "simgrid/Exception.hpp" +#include "src/simix/smx_private.hpp" XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context); @@ -29,78 +17,38 @@ namespace context { // BoostContextFactory BoostContextFactory::BoostContextFactory() - : ContextFactory("BoostContextFactory") + : ContextFactory("BoostContextFactory"), parallel_(SIMIX_context_is_parallel()) { - BoostContext::parallel_ = SIMIX_context_is_parallel(); - if (BoostContext::parallel_) { -#if HAVE_THREAD_CONTEXTS - BoostContext::parmap_ = nullptr; - BoostContext::workers_context_.clear(); - BoostContext::workers_context_.resize(SIMIX_context_get_nthreads(), nullptr); - BoostContext::maestro_context_ = nullptr; - xbt_os_thread_key_create(&BoostContext::worker_id_key_); -#else - xbt_die("No thread support for parallel context execution"); -#endif - } + BoostContext::set_maestro(nullptr); + if (parallel_) + ParallelBoostContext::initialize(); } BoostContextFactory::~BoostContextFactory() { -#if HAVE_THREAD_CONTEXTS - delete BoostContext::parmap_; - BoostContext::parmap_ = nullptr; - BoostContext::workers_context_.clear(); -#endif + if (parallel_) + ParallelBoostContext::finalize(); } smx_context_t BoostContextFactory::create_context(std::function code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process) { -#if HAVE_THREAD_CONTEXTS - if (BoostContext::parallel_) + if (parallel_) return this->new_context(std::move(code), cleanup_func, process); -#endif - return this->new_context(std::move(code), cleanup_func, process); } void BoostContextFactory::run_all() { -#if HAVE_THREAD_CONTEXTS - if (BoostContext::parallel_) { - BoostContext::threads_working_ = 0; - if (not BoostContext::parmap_) - BoostContext::parmap_ = - new simgrid::xbt::Parmap(SIMIX_context_get_nthreads(), SIMIX_context_get_parallel_mode()); - BoostContext::parmap_->apply( - [](smx_actor_t process) { - BoostContext* context = static_cast(process->context); - return context->resume(); - }, - simix_global->process_to_run); - return; - } -#endif - - if (simix_global->process_to_run.empty()) - return; - smx_actor_t first_process = simix_global->process_to_run.front(); - BoostContext::process_index_ = 1; - /* execute the first process */ - static_cast(first_process->context)->resume(); + if (parallel_) + ParallelBoostContext::run_all(); + else + SerialBoostContext::run_all(); } - // BoostContext -bool BoostContext::parallel_ = false; -simgrid::xbt::Parmap* BoostContext::parmap_ = nullptr; -uintptr_t BoostContext::threads_working_ = 0; -xbt_os_thread_key_t BoostContext::worker_id_key_; -unsigned long BoostContext::process_index_ = 0; BoostContext* BoostContext::maestro_context_ = nullptr; -std::vector BoostContext::workers_context_; BoostContext::BoostContext(std::function code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process) : Context(std::move(code), cleanup_func, process) @@ -116,7 +64,7 @@ BoostContext::BoostContext(std::function code, void_pfn_smxprocess_t cle #else void* stack = this->stack_; #endif - ASAN_EVAL(this->asan_stack_ = stack); + ASAN_ONLY(this->asan_stack_ = stack); #if BOOST_VERSION < 106100 this->fc_ = boost::context::make_fcontext(stack, smx_context_usable_stack_size, BoostContext::wrapper); #else @@ -147,18 +95,21 @@ void BoostContext::wrapper(BoostContext::arg_type arg) #if BOOST_VERSION < 106100 BoostContext* context = reinterpret_cast(arg); #else - ASAN_FINISH_SWITCH(nullptr, &static_cast(arg.data)[0]->asan_stack_, - &static_cast(arg.data)[0]->asan_stack_size_); + BoostContext* context = static_cast(arg.data)[1]; + ASAN_ONLY(xbt_assert(context->asan_ctx_ == static_cast(arg.data)[0])); + ASAN_FINISH_SWITCH(nullptr, &context->asan_ctx_->asan_stack_, &context->asan_ctx_->asan_stack_size_); static_cast(arg.data)[0]->fc_ = arg.fctx; - BoostContext* context = static_cast(arg.data)[1]; #endif try { (*context)(); - context->Context::stop(); } catch (StopRequest const&) { XBT_DEBUG("Caught a StopRequest"); + } catch (simgrid::Exception const& e) { + XBT_INFO("Actor killed by an uncatched exception %s", simgrid::xbt::demangle(typeid(e).name()).get()); + throw; } - ASAN_EVAL(context->asan_stop_ = true); + context->Context::stop(); + ASAN_ONLY(context->asan_stop_ = true); context->suspend(); } @@ -170,11 +121,12 @@ inline void BoostContext::swap(BoostContext* from, BoostContext* to) boost::context::jump_fcontext(&from->fc_, to->fc_, reinterpret_cast(to)); #else BoostContext* ctx[2] = {from, to}; - void* fake_stack; + ASAN_ONLY(void* fake_stack = nullptr); + ASAN_ONLY(to->asan_ctx_ = from); ASAN_START_SWITCH(from->asan_stop_ ? nullptr : &fake_stack, to->asan_stack_, to->asan_stack_size_); boost::context::detail::transfer_t arg = boost::context::detail::jump_fcontext(to->fc_, ctx); - ASAN_FINISH_SWITCH(fake_stack, &static_cast(arg.data)[0]->asan_stack_, - &static_cast(arg.data)[0]->asan_stack_size_); + ASAN_ONLY(xbt_assert(from->asan_ctx_ == static_cast(arg.data)[0])); + ASAN_FINISH_SWITCH(fake_stack, &from->asan_ctx_->asan_stack_, &from->asan_ctx_->asan_stack_size_); static_cast(arg.data)[0]->fc_ = arg.fctx; #endif } @@ -187,6 +139,8 @@ void BoostContext::stop() // SerialBoostContext +unsigned long SerialBoostContext::process_index_; + void SerialBoostContext::suspend() { /* determine the next context */ @@ -197,25 +151,66 @@ void SerialBoostContext::suspend() if (i < simix_global->process_to_run.size()) { /* execute the next process */ XBT_DEBUG("Run next process"); - next_context = static_cast(simix_global->process_to_run[i]->context); + next_context = static_cast(simix_global->process_to_run[i]->context_); } else { /* all processes were run, return to maestro */ XBT_DEBUG("No more process to run"); - next_context = static_cast(maestro_context_); + next_context = static_cast(BoostContext::get_maestro()); } - SIMIX_context_set_current(static_cast(next_context)); + Context::set_current(next_context); BoostContext::swap(this, next_context); } void SerialBoostContext::resume() { - SIMIX_context_set_current(this); - BoostContext::swap(maestro_context_, this); + BoostContext* old = static_cast(self()); + Context::set_current(this); + BoostContext::swap(old, this); +} + +void SerialBoostContext::run_all() +{ + if (simix_global->process_to_run.empty()) + return; + smx_actor_t first_process = simix_global->process_to_run.front(); + process_index_ = 1; + /* execute the first process */ + static_cast(first_process->context_)->resume(); } // ParallelBoostContext -#if HAVE_THREAD_CONTEXTS +simgrid::xbt::Parmap* ParallelBoostContext::parmap_; +std::atomic ParallelBoostContext::threads_working_; +thread_local uintptr_t ParallelBoostContext::worker_id_; +std::vector ParallelBoostContext::workers_context_; + +void ParallelBoostContext::initialize() +{ + parmap_ = nullptr; + workers_context_.clear(); + workers_context_.resize(SIMIX_context_get_nthreads(), nullptr); +} + +void ParallelBoostContext::finalize() +{ + delete parmap_; + parmap_ = nullptr; + workers_context_.clear(); +} + +void ParallelBoostContext::run_all() +{ + threads_working_ = 0; + if (parmap_ == nullptr) + parmap_ = new simgrid::xbt::Parmap(SIMIX_context_get_nthreads(), SIMIX_context_get_parallel_mode()); + parmap_->apply( + [](smx_actor_t process) { + ParallelBoostContext* context = static_cast(process->context_); + context->resume(); + }, + simix_global->process_to_run); +} void ParallelBoostContext::suspend() { @@ -223,30 +218,27 @@ void ParallelBoostContext::suspend() ParallelBoostContext* next_context; if (next_work) { XBT_DEBUG("Run next process"); - next_context = static_cast(next_work.get()->context); + next_context = static_cast(next_work.get()->context_); } else { XBT_DEBUG("No more processes to run"); - uintptr_t worker_id = reinterpret_cast(xbt_os_thread_get_specific(worker_id_key_)); - next_context = static_cast(workers_context_[worker_id]); + next_context = workers_context_[worker_id_]; } - SIMIX_context_set_current(static_cast(next_context)); + Context::set_current(next_context); BoostContext::swap(this, next_context); } void ParallelBoostContext::resume() { - uintptr_t worker_id = __sync_fetch_and_add(&threads_working_, 1); - xbt_os_thread_set_specific(worker_id_key_, reinterpret_cast(worker_id)); + worker_id_ = threads_working_.fetch_add(1, std::memory_order_relaxed); - ParallelBoostContext* worker_context = static_cast(SIMIX_context_self()); - workers_context_[worker_id] = worker_context; + ParallelBoostContext* worker_context = static_cast(self()); + workers_context_[worker_id_] = worker_context; - SIMIX_context_set_current(this); + Context::set_current(this); BoostContext::swap(worker_context, this); } -#endif XBT_PRIVATE ContextFactory* boost_factory() {