src/kernel/context/ContextUnix.cpp

   1 /* Copyright (c) 2009-2023. 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 "simgrid/Exception.hpp"
   9 #include "src/kernel/actor/ActorImpl.hpp"
  10 #include "src/mc/mc.h"
  11 #include "src/mc/remote/AppSide.hpp"
  12
  13 #include "ContextUnix.hpp"
  14
  15 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ker_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::kernel::context {
  36
  37 // UContextFactory
  38 UContext* UContextFactory::create_context(std::function<void()>&& code, actor::ActorImpl* actor)
  39 {
  40   return new_context<UContext>(std::move(code), actor, this);
  41 }
  42
  43
  44 // UContext
  45
  46 UContext::UContext(std::function<void()>&& code, actor::ActorImpl* actor, SwappedContextFactory* factory)
  47     : SwappedContext(std::move(code), actor, factory)
  48 {
  49   XBT_VERB("Creating a context of stack %uMb", actor->get_stacksize() / 1024 / 1024);
  50   /* if the user provided a function for the actor then use it. If not, nothing to do for maestro. */
  51   if (has_code()) {
  52     getcontext(&this->uc_);
  53     this->uc_.uc_link = nullptr;
  54     this->uc_.uc_stack.ss_sp   = sg_makecontext_stack_addr(get_stack());
  55     this->uc_.uc_stack.ss_size = sg_makecontext_stack_size(actor->get_stacksize());
  56     // Makecontext expects integer arguments; we want to pass a pointer.
  57     // This context address is decomposed into a series of integers, which are passed as arguments to makecontext.
  58
  59     int ctx_addr[CTX_ADDR_LEN]{};
  60     UContext* arg = this;
  61     memcpy(ctx_addr, &arg, sizeof arg);
  62     makecontext(&this->uc_, (void (*)())sysv_ctx_wrapper, 2, ctx_addr[0], ctx_addr[1]);
  63   }
  64 }
  65
  66 void UContext::swap_into_for_real(SwappedContext* to_)
  67 {
  68   const UContext* to = static_cast<UContext*>(to_);
  69   swapcontext(&this->uc_, &to->uc_);
  70 }
  71
  72
  73 XBT_PRIVATE ContextFactory* sysv_factory()
  74 {
  75   XBT_VERB("Activating SYSV context factory");
  76   return new UContextFactory();
  77 }
  78 } // namespace simgrid::kernel::context