src/kernel/activity/MailboxImpl.cpp

   1 /* Copyright (c) 2007-2018. 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 #include "src/kernel/activity/MailboxImpl.hpp"
   7
   8 #include "src/kernel/activity/CommImpl.hpp"
   9
  10 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_mailbox, simix, "Mailbox implementation");
  11
  12 static std::map<std::string, smx_mailbox_t>* mailboxes = new std::map<std::string, smx_mailbox_t>;
  13
  14 void SIMIX_mailbox_exit()
  15 {
  16   for (auto const& elm : *mailboxes)
  17     delete elm.second;
  18   delete mailboxes;
  19 }
  20
  21 /******************************************************************************/
  22 /*                           Rendez-Vous Points                               */
  23 /******************************************************************************/
  24
  25 namespace simgrid {
  26 namespace kernel {
  27 namespace activity {
  28 /** @brief Returns the mailbox of that name, or nullptr */
  29 MailboxImpl* MailboxImpl::byNameOrNull(const char* name)
  30 {
  31   auto mbox = mailboxes->find(name);
  32   if (mbox != mailboxes->end())
  33     return mbox->second;
  34   else
  35     return nullptr;
  36 }
  37 /** @brief Returns the mailbox of that name, newly created on need */
  38 MailboxImpl* MailboxImpl::byNameOrCreate(const char* name)
  39 {
  40   xbt_assert(name, "Mailboxes must have a name");
  41   /* two processes may have pushed the same mbox_create simcall at the same time */
  42   auto m = mailboxes->find(name);
  43   if (m == mailboxes->end()) {
  44     smx_mailbox_t mbox = new MailboxImpl(name);
  45     XBT_DEBUG("Creating a mailbox at %p with name %s", mbox, name);
  46     (*mailboxes)[mbox->name_] = mbox;
  47     return mbox;
  48   } else
  49     return m->second;
  50 }
  51 /** @brief set the receiver of the mailbox to allow eager sends
  52  *  \param actor The receiving dude
  53  */
  54 void MailboxImpl::setReceiver(s4u::ActorPtr actor)
  55 {
  56   if (actor != nullptr)
  57     this->permanent_receiver = actor.get()->get_impl();
  58   else
  59     this->permanent_receiver = nullptr;
  60 }
  61 /** @brief Pushes a communication activity into a mailbox
  62  *  @param comm What to add
  63  */
  64 void MailboxImpl::push(activity::CommImplPtr comm)
  65 {
  66   comm->mbox = this;
  67   this->comm_queue.push_back(std::move(comm));
  68 }
  69
  70 /** @brief Removes a communication activity from a mailbox
  71  *  @param activity What to remove
  72  */
  73 void MailboxImpl::remove(smx_activity_t activity)
  74 {
  75   simgrid::kernel::activity::CommImplPtr comm =
  76       boost::static_pointer_cast<simgrid::kernel::activity::CommImpl>(activity);
  77
  78   xbt_assert(comm->mbox == this, "Comm %p is in mailbox %s, not mailbox %s", comm.get(),
  79              (comm->mbox ? comm->mbox->get_cname() : "(null)"), this->get_cname());
  80   comm->mbox = nullptr;
  81   for (auto it = this->comm_queue.begin(); it != this->comm_queue.end(); it++)
  82     if (*it == comm) {
  83       this->comm_queue.erase(it);
  84       return;
  85     }
  86   xbt_die("Comm %p not found in mailbox %s", comm.get(), this->get_cname());
  87 }
  88 }
  89 }
  90 }