What ActivityImpl child should look like IMHO

[simgrid.git] / src / kernel / activity / IoImpl.cpp

diff --git a/src/kernel/activity/IoImpl.cpp b/src/kernel/activity/IoImpl.cpp
index f712154..822ff10 100644 (file)
--- a/src/kernel/activity/IoImpl.cpp
+++ b/src/kernel/activity/IoImpl.cpp
@@ -2,74 +2,148 @@
 
 /* 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 "mc/mc.h"
+#include "simgrid/Exception.hpp"
 #include "simgrid/kernel/resource/Action.hpp"
-#include "src/simix/smx_io_private.hpp"
+#include "simgrid/s4u/Host.hpp"
+#include "src/mc/mc_replay.hpp"
+#include "src/simix/smx_private.hpp"
+#include "src/surf/StorageImpl.hpp"
 
-XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_io);
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_io, simix, "Logging specific to SIMIX (io)");
 
-simgrid::kernel::activity::IoImpl::IoImpl(std::string name, resource::Action* surf_action, s4u::Storage* storage)
-    : ActivityImpl(name), storage_(storage), surf_action_(surf_action)
+void simcall_HANDLER_io_wait(smx_simcall_t simcall, simgrid::kernel::activity::IoImpl* synchro)
 {
-  this->state_ = SIMIX_RUNNING;
+  XBT_DEBUG("Wait for execution of synchro %p, state %d", synchro, (int)synchro->state_);
 
-  surf_action_->set_data(this);
+  /* Associate this simcall to the synchro */
+  synchro->simcalls_.push_back(simcall);
+  simcall->issuer->waiting_synchro = synchro;
+
+  /* set surf's synchro */
+  if (MC_is_active() || MC_record_replay_is_active()) {
+    synchro->state_ = SIMIX_DONE;
+    synchro->finish();
+    return;
+  }
 
-  XBT_DEBUG("Create io %p", this);
+  /* If the synchro is already finished then perform the error handling */
+  if (synchro->state_ != SIMIX_RUNNING)
+    synchro->finish();
 }
 
-simgrid::kernel::activity::IoImpl::~IoImpl()
+namespace simgrid {
+namespace kernel {
+namespace activity {
+
+IoImpl::~IoImpl()
 {
   if (surf_action_ != nullptr)
     surf_action_->unref();
   XBT_DEBUG("Destroy io %p", this);
 }
 
-void simgrid::kernel::activity::IoImpl::cancel()
+IoImplPtr IoImpl::set_name(const std::string& name)
 {
-  XBT_VERB("This exec %p is canceled", this);
-  if (surf_action_ != nullptr)
-    surf_action_->cancel();
+  ActivityImpl::set_name(name);
+  return this;
 }
 
-void simgrid::kernel::activity::IoImpl::suspend()
+IoImplPtr IoImpl::set_type(s4u::Io::OpType type)
 {
-  if (surf_action_ != nullptr)
-    surf_action_->suspend();
+  type_ = type;
+  return this;
+}
+
+IoImplPtr IoImpl::set_size(sg_size_t size)
+{
+  size_ = size;
+  return this;
 }
 
-void simgrid::kernel::activity::IoImpl::resume()
+IoImplPtr IoImpl::set_storage(resource::StorageImpl* storage)
 {
+  storage_ = storage;
+  return this;
+}
+
+IoImpl* IoImpl::start()
+{
+  state_       = SIMIX_RUNNING;
+  surf_action_ = storage_->io_start(size_, type_);
+  surf_action_->set_data(this);
+
+  XBT_DEBUG("Create IO synchro %p %s", this, get_cname());
+  IoImpl::on_start(this);
+
+  return this;
+}
+
+void IoImpl::cancel()
+{
+  XBT_VERB("This exec %p is canceled", this);
   if (surf_action_ != nullptr)
-    surf_action_->resume();
+    surf_action_->cancel();
+  state_ = SIMIX_CANCELED;
 }
 
-double simgrid::kernel::activity::IoImpl::get_remaining()
+double IoImpl::get_remaining()
 {
   return surf_action_ ? surf_action_->get_remains() : 0;
 }
 
-void simgrid::kernel::activity::IoImpl::post()
+void IoImpl::post()
 {
   performed_ioops_ = surf_action_->get_cost();
   switch (surf_action_->get_state()) {
-    case simgrid::kernel::resource::Action::State::FAILED:
+    case resource::Action::State::FAILED:
       state_ = SIMIX_FAILED;
       break;
-    case simgrid::kernel::resource::Action::State::FINISHED:
+    case resource::Action::State::FINISHED:
       state_ = SIMIX_DONE;
       break;
     default:
       THROW_IMPOSSIBLE;
-      break;
   }
   on_completion(this);
 
-  SIMIX_io_finish(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_));
+    }
+
+    simcall->issuer->waiting_synchro = nullptr;
+    if (simcall->issuer->get_host()->is_on())
+      SIMIX_simcall_answer(simcall);
+    else
+      simcall->issuer->context_->iwannadie = true;
+  }
+}
+
 /*************
  * Callbacks *
  *************/
-simgrid::xbt::signal<void(simgrid::kernel::activity::IoImplPtr)> simgrid::kernel::activity::IoImpl::on_creation;
-simgrid::xbt::signal<void(simgrid::kernel::activity::IoImplPtr)> simgrid::kernel::activity::IoImpl::on_completion;
+xbt::signal<void(IoImplPtr)> IoImpl::on_start;
+xbt::signal<void(IoImplPtr)> IoImpl::on_completion;
+
+} // namespace activity
+} // namespace kernel
+} // namespace simgrid