X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/55092bf3f9fe1cccfe72f7ef81fcd51f9a0eb4ca..21f39f6eed6ee5e5346c9f1da16de87702e6eae1:/src/kernel/context/ContextBoost.cpp diff --git a/src/kernel/context/ContextBoost.cpp b/src/kernel/context/ContextBoost.cpp index f861f8d1f2..8e1eabd20d 100644 --- a/src/kernel/context/ContextBoost.cpp +++ b/src/kernel/context/ContextBoost.cpp @@ -1,4 +1,4 @@ -/* Copyright (c) 2015. The SimGrid Team. All rights reserved. */ +/* Copyright (c) 2015-2017. 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. */ @@ -18,6 +18,18 @@ #include "src/internal_config.h" #include "src/kernel/context/ContextBoost.hpp" +#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 + XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context); namespace simgrid { @@ -28,9 +40,8 @@ class BoostSerialContext : public BoostContext { public: BoostSerialContext(std::function code, void_pfn_smxprocess_t cleanup_func, - smx_process_t process) + smx_actor_t process) : BoostContext(std::move(code), cleanup_func, process) {} - void stop() override; void suspend() override; }; @@ -39,9 +50,8 @@ class BoostParallelContext : public BoostContext { public: BoostParallelContext(std::function code, void_pfn_smxprocess_t cleanup_func, - smx_process_t process) + smx_actor_t process) : BoostContext(std::move(code), cleanup_func, process) {} - void stop() override; void suspend() override; void resume() override; }; @@ -49,12 +59,12 @@ public: // BoostContextFactory -bool BoostContext::parallel_ = false; -xbt_parmap_t BoostContext::parmap_ = nullptr; -uintptr_t BoostContext::threads_working_ = 0; +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; +unsigned long BoostContext::process_index_ = 0; +BoostContext* BoostContext::maestro_context_ = nullptr; std::vector BoostContext::workers_context_; BoostContextFactory::BoostContextFactory() @@ -62,15 +72,14 @@ BoostContextFactory::BoostContextFactory() { BoostContext::parallel_ = SIMIX_context_is_parallel(); if (BoostContext::parallel_) { -#if !HAVE_THREAD_CONTEXTS - xbt_die("No thread support for parallel context execution"); -#else - int nthreads = SIMIX_context_get_nthreads(); - BoostContext::parmap_ = xbt_parmap_new(nthreads, SIMIX_context_get_parallel_mode()); +#if HAVE_THREAD_CONTEXTS + BoostContext::parmap_ = nullptr; BoostContext::workers_context_.clear(); - BoostContext::workers_context_.resize(nthreads, nullptr); + 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 } } @@ -78,28 +87,24 @@ BoostContextFactory::BoostContextFactory() BoostContextFactory::~BoostContextFactory() { #if HAVE_THREAD_CONTEXTS - if (BoostContext::parmap_) { - xbt_parmap_destroy(BoostContext::parmap_); - BoostContext::parmap_ = nullptr; - } + delete BoostContext::parmap_; + BoostContext::parmap_ = nullptr; BoostContext::workers_context_.clear(); #endif } -smx_context_t BoostContextFactory::create_context(std::function code, - void_pfn_smxprocess_t cleanup_func, smx_process_t process) +smx_context_t BoostContextFactory::create_context(std::function code, void_pfn_smxprocess_t cleanup_func, + smx_actor_t process) { BoostContext* context = nullptr; if (BoostContext::parallel_) #if HAVE_THREAD_CONTEXTS - context = this->new_context( - std::move(code), cleanup_func, process); + context = this->new_context(std::move(code), cleanup_func, process); #else xbt_die("No support for parallel execution"); #endif else - context = this->new_context( - std::move(code), cleanup_func, process); + context = this->new_context(std::move(code), cleanup_func, process); return context; } @@ -108,18 +113,21 @@ void BoostContextFactory::run_all() #if HAVE_THREAD_CONTEXTS if (BoostContext::parallel_) { BoostContext::threads_working_ = 0; - xbt_parmap_apply(BoostContext::parmap_, - [](void* arg) { - smx_process_t process = static_cast(arg); - BoostContext* context = static_cast(process->context); - return context->resume(); - }, - simix_global->process_to_run); + 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); } else #endif { - smx_process_t first_process = - xbt_dynar_get_as(simix_global->process_to_run, 0, smx_process_t); + 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(); @@ -129,37 +137,68 @@ void BoostContextFactory::run_all() // BoostContext -static void smx_ctx_boost_wrapper(std::intptr_t arg) +void BoostContext::smx_ctx_boost_wrapper(BoostContext::ctx_arg_type arg) { - BoostContext* context = (BoostContext*) arg; - (*context)(); - context->stop(); +#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_); + 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"); + } + ASAN_EVAL(context->asan_stop_ = true); + context->suspend(); +} + +inline void BoostContext::smx_ctx_boost_jump_fcontext(BoostContext* from, BoostContext* to) +{ +#if BOOST_VERSION < 105600 + boost::context::jump_fcontext(from->fc_, to->fc_, reinterpret_cast(to)); +#elif BOOST_VERSION < 106100 + boost::context::jump_fcontext(&from->fc_, to->fc_, reinterpret_cast(to)); +#else + BoostContext* ctx[2] = {from, to}; + void* fake_stack; + 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_); + static_cast(arg.data)[0]->fc_ = arg.fctx; +#endif } BoostContext::BoostContext(std::function code, - void_pfn_smxprocess_t cleanup_func, smx_process_t process) + void_pfn_smxprocess_t cleanup_func, smx_actor_t process) : Context(std::move(code), cleanup_func, process) { - /* if the user provided a function for the process then use it, - otherwise it is the context for maestro */ + /* if the user provided a function for the process then use it, otherwise it is the context for maestro */ if (has_code()) { this->stack_ = SIMIX_context_stack_new(); - // We need to pass the bottom of the stack to make_fcontext, - // depending on the stack direction it may be the lower or higher address: - #if PTH_STACKGROWTH == -1 - void* stack = (char*) this->stack_ + smx_context_usable_stack_size - 1; - #else +// We need to pass the bottom of the stack to make_fcontext, depending on the stack direction it may be the lower +// or higher address: +#if PTH_STACKGROWTH == -1 + void* stack = static_cast(this->stack_) + smx_context_usable_stack_size; +#else void* stack = this->stack_; - #endif - this->fc_ = boost::context::make_fcontext( - stack, - smx_context_usable_stack_size, - smx_ctx_boost_wrapper); +#endif + ASAN_EVAL(this->asan_stack_ = stack); +#if BOOST_VERSION < 106100 + this->fc_ = boost::context::make_fcontext(stack, smx_context_usable_stack_size, smx_ctx_boost_wrapper); +#else + this->fc_ = boost::context::detail::make_fcontext(stack, smx_context_usable_stack_size, smx_ctx_boost_wrapper); +#endif } else { - #if HAVE_BOOST_CONTEXTS == 1 +#if BOOST_VERSION < 105600 this->fc_ = new boost::context::fcontext_t(); - #endif +#endif if (BoostContext::maestro_context_ == nullptr) BoostContext::maestro_context_ = this; } @@ -167,8 +206,8 @@ BoostContext::BoostContext(std::function code, BoostContext::~BoostContext() { -#if HAVE_BOOST_CONTEXTS == 1 - if (!this->stack_) +#if BOOST_VERSION < 105600 + if (not this->stack_) delete this->fc_; #endif if (this == maestro_context_) @@ -178,52 +217,36 @@ BoostContext::~BoostContext() // BoostSerialContext +void BoostContext::stop() +{ + Context::stop(); + throw StopRequest(); +} + void BoostContext::resume() { SIMIX_context_set_current(this); -#if HAVE_BOOST_CONTEXTS == 1 - boost::context::jump_fcontext( - maestro_context_->fc_, this->fc_, - (intptr_t) this); -#else - boost::context::jump_fcontext( - &maestro_context_->fc_, this->fc_, - (intptr_t) this); -#endif + smx_ctx_boost_jump_fcontext(maestro_context_, this); } void BoostSerialContext::suspend() { /* determine the next context */ - BoostSerialContext* next_context = nullptr; - unsigned long int i = process_index_++; + BoostSerialContext* next_context; + unsigned long int i = process_index_; + process_index_++; - if (i < xbt_dynar_length(simix_global->process_to_run)) { + if (i < simix_global->process_to_run.size()) { /* execute the next process */ XBT_DEBUG("Run next process"); - next_context = static_cast(xbt_dynar_get_as( - simix_global->process_to_run, i, smx_process_t)->context); - } - else { + 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(maestro_context_); } - SIMIX_context_set_current((smx_context_t) next_context); - #if HAVE_BOOST_CONTEXTS == 1 - boost::context::jump_fcontext( - this->fc_, next_context->fc_, (intptr_t) next_context); - #else - boost::context::jump_fcontext( - &this->fc_, next_context->fc_, (intptr_t) next_context); - #endif -} - -void BoostSerialContext::stop() -{ - BoostContext::stop(); - this->suspend(); + SIMIX_context_set_current(static_cast(next_context)); + smx_ctx_boost_jump_fcontext(this, next_context); } // BoostParallelContext @@ -232,54 +255,31 @@ void BoostSerialContext::stop() void BoostParallelContext::suspend() { - smx_process_t next_work = (smx_process_t) xbt_parmap_next(parmap_); - BoostParallelContext* next_context = nullptr; - - if (next_work != nullptr) { + boost::optional next_work = parmap_->next(); + BoostParallelContext* next_context; + if (next_work) { XBT_DEBUG("Run next process"); - next_context = static_cast(next_work->context); - } - else { + next_context = static_cast(next_work.get()->context); + } else { XBT_DEBUG("No more processes to run"); - uintptr_t worker_id = - (uintptr_t) xbt_os_thread_get_specific(worker_id_key_); - next_context = static_cast( - workers_context_[worker_id]); + uintptr_t worker_id = reinterpret_cast(xbt_os_thread_get_specific(worker_id_key_)); + next_context = static_cast(workers_context_[worker_id]); } - SIMIX_context_set_current((smx_context_t) next_context); -#if HAVE_BOOST_CONTEXTS == 1 - boost::context::jump_fcontext( - this->fc_, next_context->fc_, (intptr_t)next_context); -#else - boost::context::jump_fcontext( - &this->fc_, next_context->fc_, (intptr_t)next_context); -#endif -} - -void BoostParallelContext::stop() -{ - BoostContext::stop(); - this->suspend(); + SIMIX_context_set_current(static_cast(next_context)); + smx_ctx_boost_jump_fcontext(this, next_context); } void BoostParallelContext::resume() { uintptr_t worker_id = __sync_fetch_and_add(&threads_working_, 1); - xbt_os_thread_set_specific(worker_id_key_, (void*) worker_id); + xbt_os_thread_set_specific(worker_id_key_, reinterpret_cast(worker_id)); - BoostParallelContext* worker_context = - static_cast(SIMIX_context_self()); + BoostParallelContext* worker_context = static_cast(SIMIX_context_self()); workers_context_[worker_id] = worker_context; SIMIX_context_set_current(this); -#if HAVE_BOOST_CONTEXTS == 1 - boost::context::jump_fcontext( - worker_context->fc_, this->fc_, (intptr_t) this); -#else - boost::context::jump_fcontext( - &worker_context->fc_, this->fc_, (intptr_t) this); -#endif + smx_ctx_boost_jump_fcontext(worker_context, this); } #endif