Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
(crudly) deal with I/O launched by maestro
[simgrid.git] / src / kernel / activity / IoImpl.cpp
index 3bbef38..a6dcf3a 100644 (file)
-/* Copyright (c) 2007-2019. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2007-2021. 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 "src/kernel/activity/IoImpl.hpp"
+
+#include <simgrid/Exception.hpp>
+#include <simgrid/kernel/routing/NetPoint.hpp>
+#include <simgrid/s4u/Engine.hpp>
+#include <simgrid/s4u/Host.hpp>
+
 #include "mc/mc.h"
-#include "simgrid/Exception.hpp"
-#include "simgrid/kernel/resource/Action.hpp"
-#include "simgrid/s4u/Host.hpp"
+#include "src/kernel/activity/IoImpl.hpp"
+#include "src/kernel/actor/ActorImpl.hpp"
+#include "src/kernel/actor/SimcallObserver.hpp"
+#include "src/kernel/context/Context.hpp"
+#include "src/kernel/resource/CpuImpl.hpp"
+#include "src/kernel/resource/DiskImpl.hpp"
 #include "src/mc/mc_replay.hpp"
-#include "src/simix/smx_private.hpp"
-#include "src/surf/StorageImpl.hpp"
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_io, simix, "Logging specific to SIMIX (io)");
 
-void simcall_HANDLER_io_wait(smx_simcall_t simcall, simgrid::kernel::activity::IoImpl* synchro)
-{
-  XBT_DEBUG("Wait for execution of synchro %p, state %d", synchro, (int)synchro->state_);
-
-  /* Associate this simcall to the synchro */
-  synchro->simcalls_.push_back(simcall);
-  simcall->issuer->waiting_synchro = synchro;
+namespace simgrid {
+namespace kernel {
+namespace activity {
 
-  /* set surf's synchro */
-  if (MC_is_active() || MC_record_replay_is_active()) {
-    synchro->state_ = SIMIX_DONE;
-    synchro->finish();
-    return;
+IoImpl::IoImpl()
+{
+  piface_ = new s4u::Io(this);
+  actor::ActorImpl* self = actor::ActorImpl::self();
+  if (self) {
+    actor_ = self;
+    self->activities_.emplace_back(this);
   }
+}
 
-  /* If the synchro is already finished then perform the error handling */
-  if (synchro->state_ != SIMIX_RUNNING)
-    synchro->finish();
+IoImpl& IoImpl::set_sharing_penalty(double sharing_penalty)
+{
+  sharing_penalty_ = sharing_penalty;
+  return *this;
 }
 
-namespace simgrid {
-namespace kernel {
-namespace activity {
+IoImpl& IoImpl::update_sharing_penalty(double sharing_penalty)
+{
+  sharing_penalty_ = sharing_penalty;
+  surf_action_->set_sharing_penalty(sharing_penalty);
+  return *this;
+}
 
-IoImplPtr IoImpl::set_name(const std::string& name)
+IoImpl& IoImpl::set_timeout(double timeout)
 {
-  ActivityImpl::set_name(name);
-  return this;
+  const s4u::Host* host = get_disk()->get_host();
+  timeout_detector_     = host->get_cpu()->sleep(timeout);
+  timeout_detector_->set_activity(this);
+  return *this;
 }
 
-IoImplPtr IoImpl::set_type(s4u::Io::OpType type)
+IoImpl& IoImpl::set_type(s4u::Io::OpType type)
 {
   type_ = type;
-  return this;
+  return *this;
 }
 
-IoImplPtr IoImpl::set_size(sg_size_t size)
+IoImpl& IoImpl::set_size(sg_size_t size)
 {
   size_ = size;
-  return this;
+  return *this;
 }
 
-IoImplPtr IoImpl::set_storage(resource::StorageImpl* storage)
+IoImpl& IoImpl::set_disk(resource::DiskImpl* disk)
 {
-  storage_ = storage;
-  return this;
+  disk_ = disk;
+  return *this;
 }
 
 IoImpl* IoImpl::start()
 {
-  state_       = SIMIX_RUNNING;
-  surf_action_ = storage_->io_start(size_, type_);
-  surf_action_->set_data(this);
+  state_ = State::RUNNING;
+  surf_action_ =
+      disk_->get_host()->get_netpoint()->get_englobing_zone()->get_disk_model()->io_start(disk_, size_, type_);
+  surf_action_->set_sharing_penalty(sharing_penalty_);
+  surf_action_->set_activity(this);
 
   XBT_DEBUG("Create IO synchro %p %s", this, get_cname());
-  IoImpl::on_start(*this);
 
   return this;
 }
 
-void IoImpl::cancel()
+void IoImpl::post()
 {
-  XBT_VERB("This exec %p is canceled", this);
-  if (surf_action_ != nullptr)
-    surf_action_->cancel();
-  state_ = SIMIX_CANCELED;
-}
+  performed_ioops_ = surf_action_->get_cost();
+  if (surf_action_->get_state() == resource::Action::State::FAILED) {
+    if (disk_ && not disk_->is_on())
+      state_ = State::FAILED;
+    else
+      state_ = State::CANCELED;
+  } else if (timeout_detector_ && timeout_detector_->get_state() == resource::Action::State::FINISHED) {
+    if (surf_action_->get_remains() > 0.0) {
+      surf_action_->set_state(resource::Action::State::FAILED);
+      state_ = State::TIMEOUT;
+    } else {
+      state_ = State::DONE;
+    }
+  } else {
+    state_ = State::DONE;
+  }
 
-double IoImpl::get_remaining()
-{
-  return surf_action_ ? surf_action_->get_remains() : 0;
-}
+  clean_action();
+  if (timeout_detector_) {
+    timeout_detector_->unref();
+    timeout_detector_ = nullptr;
+  }
 
-void IoImpl::post()
+  /* Answer all simcalls associated with the synchro */
+  finish();
+}
+void IoImpl::set_exception(actor::ActorImpl* issuer)
 {
-  performed_ioops_ = surf_action_->get_cost();
-  switch (surf_action_->get_state()) {
-    case resource::Action::State::FAILED:
-      state_ = SIMIX_FAILED;
+  switch (state_) {
+    case State::FAILED:
+      issuer->context_->set_wannadie();
+      piface_->complete(s4u::Activity::State::FAILED);
+      issuer->exception_ = std::make_exception_ptr(StorageFailureException(XBT_THROW_POINT, "Storage failed"));
+      break;
+    case State::CANCELED:
+      issuer->exception_ = std::make_exception_ptr(CancelException(XBT_THROW_POINT, "I/O Canceled"));
       break;
-    case resource::Action::State::FINISHED:
-      state_ = SIMIX_DONE;
+    case State::TIMEOUT:
+      issuer->exception_ = std::make_exception_ptr(TimeoutException(XBT_THROW_POINT, "Timeouted"));
       break;
     default:
-      THROW_IMPOSSIBLE;
+      xbt_assert(state_ == State::DONE, "Internal error in IoImpl::finish(): unexpected synchro state %s",
+                 to_c_str(state_));
   }
-  on_completion(*this);
-
-  finish();
 }
 
 void IoImpl::finish()
 {
-  for (smx_simcall_t const& simcall : simcalls_) {
-    switch (state_) {
-      case SIMIX_DONE:
-        /* do nothing, synchro done */
-        break;
-      case SIMIX_FAILED:
-        simcall->issuer->exception_ =
-            std::make_exception_ptr(simgrid::StorageFailureException(XBT_THROW_POINT, "Storage failed"));
-        break;
-      case SIMIX_CANCELED:
-        simcall->issuer->exception_ =
-            std::make_exception_ptr(simgrid::CancelException(XBT_THROW_POINT, "I/O Canceled"));
-        break;
-      default:
-        xbt_die("Internal error in IoImpl::finish(): unexpected synchro state %d", static_cast<int>(state_));
+  XBT_DEBUG("IoImpl::finish() in state %s", to_c_str(state_));
+  while (not simcalls_.empty()) {
+    smx_simcall_t simcall = simcalls_.front();
+    simcalls_.pop_front();
+
+    /* If a waitany simcall is waiting for this synchro to finish, then remove it from the other synchros in the waitany
+     * list. Afterwards, get the position of the actual synchro in the waitany list and return it as the result of the
+     * simcall */
+
+    if (simcall->call_ == simix::Simcall::NONE) // FIXME: maybe a better way to handle this case
+      continue;                                 // if process handling comm is killed
+    if (auto* observer = dynamic_cast<kernel::actor::IoWaitanySimcall*>(simcall->observer_)) { // simcall is a wait_any?
+      const auto& ios = observer->get_ios();
+
+      for (auto* io : ios) {
+        io->unregister_simcall(simcall);
+
+        if (simcall->timeout_cb_) {
+          simcall->timeout_cb_->remove();
+          simcall->timeout_cb_ = nullptr;
+        }
+      }
+
+      if (not MC_is_active() && not MC_record_replay_is_active()) {
+        auto element = std::find(ios.begin(), ios.end(), this);
+        int rank     = element != ios.end() ? static_cast<int>(std::distance(ios.begin(), element)) : -1;
+        observer->set_result(rank);
+      }
     }
 
-    simcall->issuer->waiting_synchro = nullptr;
-    if (simcall->issuer->get_host()->is_on())
-      SIMIX_simcall_answer(simcall);
-    else
-      simcall->issuer->context_->iwannadie = true;
+    set_exception(simcall->issuer_);
+
+    simcall->issuer_->waiting_synchro_ = nullptr;
+    simcall->issuer_->simcall_answer();
   }
 }
 
-/*************
- * Callbacks *
- *************/
-xbt::signal<void(IoImpl const&)> IoImpl::on_start;
-xbt::signal<void(IoImpl const&)> IoImpl::on_completion;
+void IoImpl::wait_any_for(actor::ActorImpl* issuer, const std::vector<IoImpl*>& ios, double timeout)
+{
+  if (timeout < 0.0) {
+    issuer->simcall_.timeout_cb_ = nullptr;
+  } else {
+    issuer->simcall_.timeout_cb_ = timer::Timer::set(s4u::Engine::get_clock() + timeout, [issuer, &ios]() {
+      issuer->simcall_.timeout_cb_ = nullptr;
+      for (auto* io : ios)
+        io->unregister_simcall(&issuer->simcall_);
+      // default result (-1) is set in actor::IoWaitanySimcall
+      issuer->simcall_answer();
+    });
+  }
+
+  for (auto* io : ios) {
+    /* associate this simcall to the the synchro */
+    io->simcalls_.push_back(&issuer->simcall_);
+
+    /* see if the synchro is already finished */
+    if (io->state_ != State::WAITING && io->state_ != State::RUNNING) {
+      io->finish();
+      break;
+    }
+  }
+}
 
 } // namespace activity
 } // namespace kernel