Python version of the masterworkers monkey

[simgrid.git] / src / surf / HostImpl.cpp

diff --git a/src/surf/HostImpl.cpp b/src/surf/HostImpl.cpp
index 73347dfe057be2ccdd1b78393bb997adf470d7e3..9dabe15b660df6fd8261e6c79942a922170dbc1e 100644 (file)
--- a/src/surf/HostImpl.cpp
+++ b/src/surf/HostImpl.cpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2013-2021. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2013-2022. 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. */
@@ -99,11 +99,14 @@ void HostImpl::turn_off(const actor::ActorImpl* issuer)
               actor.get_host()->get_cname(), issuer->get_cname());
     issuer->kill(&actor);
   }
-  for (auto& activity : EngineImpl::get_instance()->get_maestro()->activities_) {
+  for (const auto& activity : EngineImpl::get_instance()->get_maestro()->activities_) {
     auto* exec = dynamic_cast<activity::ExecImpl*>(activity.get());
-    if (exec != nullptr && exec->get_host() == &piface_) {
-      exec->cancel();
-      exec->state_ = activity::State::FAILED;
+    if (exec != nullptr) {
+      auto hosts = exec->get_hosts();
+      if (std::find(hosts.begin(), hosts.end(), &piface_) != hosts.end()) {
+        exec->cancel();
+        exec->set_state(activity::State::FAILED);
+      }
     }
   }
   // When a host is turned off, we want to keep only the actors that should restart for when it will boot again.