X-Git-Url: http://bilbo.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/f45abd9a448a9196144cb2aa845cb966b741d562..33ab927a5c7716530d8cadec7f1200976713453c:/src/kernel/context/ContextBoost.cpp diff --git a/src/kernel/context/ContextBoost.cpp b/src/kernel/context/ContextBoost.cpp index ac3313465f..8f3a9e4b34 100644 --- a/src/kernel/context/ContextBoost.cpp +++ b/src/kernel/context/ContextBoost.cpp @@ -1,204 +1,67 @@ -/* Copyright (c) 2015-2018. The SimGrid Team. All rights reserved. */ +/* Copyright (c) 2015-2023. 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 "context_private.hpp" #include "simgrid/Exception.hpp" -#include "src/simix/smx_private.hpp" +#include "src/internal_config.h" -XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context); +XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ker_context); -namespace simgrid { -namespace kernel { -namespace context { +namespace simgrid::kernel::context { // BoostContextFactory - -BoostContextFactory::BoostContextFactory() - : ContextFactory("BoostContextFactory"), parallel_(SIMIX_context_is_parallel()) -{ - BoostContext::set_maestro(nullptr); - if (parallel_) - ParallelBoostContext::initialize(); -} - -BoostContextFactory::~BoostContextFactory() -{ - if (parallel_) - ParallelBoostContext::finalize(); -} - -smx_context_t BoostContextFactory::create_context(std::function code, void_pfn_smxprocess_t cleanup_func, - smx_actor_t process) -{ - if (parallel_) - return this->new_context(std::move(code), cleanup_func, process); - return this->new_context(std::move(code), cleanup_func, process); -} - -void BoostContextFactory::run_all() +BoostContext* BoostContextFactory::create_context(std::function&& code, actor::ActorImpl* actor) { - if (parallel_) - ParallelBoostContext::run_all(); - else - SwappedContext::run_all(); + return this->new_context(std::move(code), actor, this); } // BoostContext -BoostContext::BoostContext(std::function code, void_pfn_smxprocess_t cleanup_func, smx_actor_t process) - : SwappedContext(std::move(code), cleanup_func, process) +BoostContext::BoostContext(std::function&& code, actor::ActorImpl* actor, SwappedContextFactory* factory) + : SwappedContext(std::move(code), actor, factory) { - + XBT_VERB("Creating a context of stack %uMb", actor->get_stacksize() / 1024 / 1024); /* if the user provided a function for the process then use it, otherwise it is the context for maestro */ if (has_code()) { - /* 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 - ASAN_ONLY(this->asan_stack_ = stack); #if BOOST_VERSION < 106100 - this->fc_ = boost::context::make_fcontext(stack, smx_context_usable_stack_size, BoostContext::wrapper); + this->fc_ = boost::context::make_fcontext(get_stack_bottom(), actor->get_stacksize(), BoostContext::wrapper); #else - this->fc_ = boost::context::detail::make_fcontext(stack, smx_context_usable_stack_size, BoostContext::wrapper); -#endif - } else { -#if BOOST_VERSION < 105600 - this->fc_ = new boost::context::fcontext_t(); + this->fc_ = + boost::context::detail::make_fcontext(get_stack_bottom(), actor->get_stacksize(), BoostContext::wrapper); #endif - if (get_maestro() == nullptr) - set_maestro(this); } } -BoostContext::~BoostContext() -{ -#if BOOST_VERSION < 105600 - if (not this->stack_) - delete this->fc_; -#endif - if (this == get_maestro()) - set_maestro(nullptr); -} - void BoostContext::wrapper(BoostContext::arg_type arg) { #if BOOST_VERSION < 106100 BoostContext* context = reinterpret_cast(arg); #else 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_); + context->verify_previous_context(static_cast(arg.data)[0]); static_cast(arg.data)[0]->fc_ = arg.fctx; #endif - try { - (*context)(); - } 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; - } - context->Context::stop(); - ASAN_ONLY(context->asan_stop_ = true); - context->suspend(); + smx_ctx_wrapper(context); } -void BoostContext::swap_into(SwappedContext* to_) +void BoostContext::swap_into_for_real(SwappedContext* to_) { - BoostContext* to = static_cast(to_); -#if BOOST_VERSION < 105600 - boost::context::jump_fcontext(this->fc_, to->fc_, reinterpret_cast(to)); -#elif BOOST_VERSION < 106100 + auto* to = static_cast(to_); +#if BOOST_VERSION < 106100 boost::context::jump_fcontext(&this->fc_, to->fc_, reinterpret_cast(to)); #else BoostContext* ctx[2] = {this, to}; - 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_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_); + this->verify_previous_context(static_cast(arg.data)[0]); static_cast(arg.data)[0]->fc_ = arg.fctx; #endif } -void BoostContext::stop() -{ - Context::stop(); - throw StopRequest(); -} - -// ParallelBoostContext - -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() -{ - boost::optional next_work = parmap_->next(); - ParallelBoostContext* next_context; - if (next_work) { - XBT_DEBUG("Run next process"); - next_context = static_cast(next_work.get()->context_); - } else { - XBT_DEBUG("No more processes to run"); - next_context = workers_context_[worker_id_]; - } - - Context::set_current(next_context); - this->swap_into(next_context); -} - -void ParallelBoostContext::resume() -{ - worker_id_ = threads_working_.fetch_add(1, std::memory_order_relaxed); - - ParallelBoostContext* worker_context = static_cast(self()); - workers_context_[worker_id_] = worker_context; - - Context::set_current(this); - worker_context->swap_into(this); -} - - XBT_PRIVATE ContextFactory* boost_factory() { XBT_VERB("Using Boost contexts. Welcome to the 21th century."); return new BoostContextFactory(); } -}}} // namespace +} // namespace simgrid::kernel::context