src/kernel/context/ContextUnix.cpp

   1 /* Copyright (c) 2009-2021. 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 /* \file UContext.cpp Context switching with ucontexts from System V        */
   7
   8 #include "mc/mc.h"
   9 #include "simgrid/Exception.hpp"
  10 #include "src/kernel/actor/ActorImpl.hpp"
  11 #include "src/mc/mc_ignore.hpp"
  12
  13 #include "ContextUnix.hpp"
  14
  15 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_context);
  16
  17 /** Up to two integers may be needed to store a pointer on the system we target */
  18 constexpr int CTX_ADDR_LEN = 2;
  19
  20 static_assert(sizeof(simgrid::kernel::context::UContext*) <= CTX_ADDR_LEN * sizeof(int),
  21               "Ucontexts are not supported on this arch yet. Please increase CTX_ADDR_LEN.");
  22
  23 // This function is called by makecontext(3): use extern "C" to have C language linkage for its type
  24 extern "C" {
  25 XBT_ATTRIB_NORETURN static void sysv_ctx_wrapper(int i1, int i2)
  26 {
  27   // Rebuild the Context* pointer from the integers:
  28   int ctx_addr[CTX_ADDR_LEN] = {i1, i2};
  29   simgrid::kernel::context::UContext* context;
  30   memcpy(&context, ctx_addr, sizeof context);
  31   smx_ctx_wrapper(context);
  32 }
  33 }
  34
  35 namespace simgrid {
  36 namespace kernel {
  37 namespace context {
  38
  39 // UContextFactory
  40 UContext* UContextFactory::create_context(std::function<void()>&& code, actor::ActorImpl* actor)
  41 {
  42   return new_context<UContext>(std::move(code), actor, this);
  43 }
  44
  45
  46 // UContext
  47
  48 UContext::UContext(std::function<void()>&& code, actor::ActorImpl* actor, SwappedContextFactory* factory)
  49     : SwappedContext(std::move(code), actor, factory)
  50 {
  51   XBT_VERB("Creating a context of stack %uMb", actor->get_stacksize() / 1024 / 1024);
  52   /* if the user provided a function for the actor then use it. If not, nothing to do for maestro. */
  53   if (has_code()) {
  54     getcontext(&this->uc_);
  55     this->uc_.uc_link = nullptr;
  56     this->uc_.uc_stack.ss_sp   = sg_makecontext_stack_addr(get_stack());
  57     this->uc_.uc_stack.ss_size = sg_makecontext_stack_size(actor->get_stacksize());
  58     // Makecontext expects integer arguments; we want to pass a pointer.
  59     // This context address is decomposed into a series of integers, which are passed as arguments to makecontext.
  60
  61     int ctx_addr[CTX_ADDR_LEN]{};
  62     UContext* arg = this;
  63     memcpy(ctx_addr, &arg, sizeof arg);
  64     makecontext(&this->uc_, (void (*)())sysv_ctx_wrapper, 2, ctx_addr[0], ctx_addr[1]);
  65
  66 #if SIMGRID_HAVE_MC
  67     if (MC_is_active()) {
  68       MC_register_stack_area(get_stack(), &(this->uc_), smx_context_stack_size);
  69     }
  70 #endif
  71   }
  72 }
  73
  74 void UContext::swap_into_for_real(SwappedContext* to_)
  75 {
  76   const UContext* to = static_cast<UContext*>(to_);
  77   swapcontext(&this->uc_, &to->uc_);
  78 }
  79
  80
  81 XBT_PRIVATE ContextFactory* sysv_factory()
  82 {
  83   XBT_VERB("Activating SYSV context factory");
  84   return new UContextFactory();
  85 }
  86 } // namespace context
  87 } // namespace kernel
  88 } // namespace simgrid