Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Merge branch 'task-dispatch' into 'master'
[simgrid.git] / src / kernel / resource / HostImpl.cpp
1 /* Copyright (c) 2013-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 #include <simgrid/kernel/routing/NetPoint.hpp>
7 #include <simgrid/s4u/Engine.hpp>
8 #include <simgrid/s4u/Host.hpp>
9
10 #include "src/kernel/EngineImpl.hpp"
11 #include "src/kernel/resource/VirtualMachineImpl.hpp"
12 #include "xbt/asserts.hpp"
13
14 #include <string>
15
16 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(res_host, ker_resource, "Host resources agregate CPU, networking and I/O features");
17
18 /*************
19  * Callbacks *t
20  *************/
21
22 namespace simgrid::kernel::resource {
23
24 /*********
25  * Model *
26  *********/
27 /************
28  * Resource *
29  ************/
30 HostImpl::HostImpl(const std::string& name) : piface_(this), name_(name)
31 {
32   xbt_enforce(s4u::Host::by_name_or_null(name_) == nullptr, "Refusing to create a second host named '%s'.",
33               get_cname());
34 }
35
36 HostImpl::~HostImpl()
37 {
38   /* All actors should be gone when the host is turned off (by the end of the simulation). */
39   if (not actor_list_.empty()) {
40     const char* msg = "Shutting down host, but it's not empty";
41     try {
42       std::string actors;
43       for (auto const& actor : actor_list_)
44         actors += "\n\t" + actor.get_name();
45
46       EngineImpl::get_instance()->display_all_actor_status();
47       xbt_die("%s:%s", msg, actors.c_str());
48     } catch (const std::bad_alloc& ba) {
49       xbt_die("%s (cannot print actor list: %s)", msg, ba.what());
50     }
51   }
52   for (auto const& arg : actors_at_boot_)
53     delete arg;
54   actors_at_boot_.clear();
55
56   for (auto const& [_, vm] : vms_)
57     vm->vm_destroy();
58 }
59
60 /** @brief Fire the required callbacks and destroy the object
61  *
62  * Don't delete directly a Host, call h->destroy() instead.
63  */
64 void HostImpl::destroy()
65 {
66   s4u::Host::on_destruction(*this->get_iface());
67   this->get_iface()->on_this_destruction(*this->get_iface());
68   delete this;
69 }
70
71 /** Re-starts all the actors that are marked as restartable.
72  *
73  * Weird things will happen if you turn on a host that is already on. S4U is fool-proof, not this.
74  */
75 void HostImpl::turn_on() const
76 {
77   for (auto const& arg : actors_at_boot_) {
78     XBT_DEBUG("Booting Actor %s(%s) right now", arg->name.c_str(), arg->host->get_cname());
79     actor::ActorImplPtr actor = actor::ActorImpl::create(arg);
80   }
81 }
82
83 /** Kill all actors hosted here */
84 void HostImpl::turn_off(const actor::ActorImpl* issuer)
85 {
86   /* turn_off VMs running on host */
87   for (const auto& [_, vm] : vms_) {
88     // call s4u functions to generate the good on_state_change signal, maybe one day this wont be necessary
89     vm->get_iface()->shutdown();
90     vm->get_iface()->turn_off();
91   }
92   for (auto& actor : actor_list_) {
93     XBT_DEBUG("Killing Actor %s@%s on behalf of %s which turned off that host.", actor.get_cname(),
94               actor.get_host()->get_cname(), issuer->get_cname());
95     issuer->kill(&actor);
96   }
97   for (const auto& activity : EngineImpl::get_instance()->get_maestro()->activities_) {
98     auto const& hosts = activity->get_hosts();
99     if (std::find(hosts.begin(), hosts.end(), &piface_) != hosts.end()) {
100       activity->cancel();
101       activity->set_state(activity::State::FAILED);
102     }
103   }
104   // When a host is turned off, we want to keep only the actors that should restart for when it will boot again.
105   // Then get rid of the others.
106   auto elm = remove_if(begin(actors_at_boot_), end(actors_at_boot_), [](const actor::ProcessArg* arg) {
107     if (arg->auto_restart)
108       return false;
109     delete arg;
110     return true;
111   });
112   actors_at_boot_.erase(elm, end(actors_at_boot_));
113 }
114
115 HostImpl* HostImpl::set_englobing_zone(routing::NetZoneImpl* englobing_zone)
116 {
117   englobing_zone_ = englobing_zone;
118   return this;
119 }
120
121 std::vector<s4u::ActorPtr> HostImpl::get_all_actors()
122 {
123   std::vector<s4u::ActorPtr> res;
124   for (auto& actor : actor_list_)
125     res.emplace_back(actor.get_ciface());
126   return res;
127 }
128 size_t HostImpl::get_actor_count() const
129 {
130   return actor_list_.size();
131 }
132
133 std::vector<s4u::Disk*> HostImpl::get_disks() const
134 {
135   std::vector<s4u::Disk*> disks;
136   for (auto const& [_, d] : disks_)
137     disks.push_back(d->get_iface());
138   return disks;
139 }
140
141 s4u::VirtualMachine* HostImpl::create_vm(const std::string& name, int core_amount, size_t ramsize)
142 {
143   auto* host_vm = new kernel::resource::VirtualMachineImpl(name, get_iface(), core_amount, ramsize);
144   auto* vm      = new s4u::VirtualMachine(host_vm);
145   host_vm->set_piface(vm);
146   return create_vm(name, vm);
147 }
148
149 s4u::VirtualMachine* HostImpl::create_vm(const std::string& name, s4u::VirtualMachine* vm)
150 {
151   vms_[name] = vm->get_vm_impl();
152
153   // Create a VCPU for this VM
154   std::vector<double> speeds;
155   for (unsigned long i = 0; i < get_iface()->get_pstate_count(); i++)
156     speeds.push_back(get_iface()->get_pstate_speed(i));
157
158   auto* cpu =
159       englobing_zone_->get_cpu_vm_model()->create_cpu(vm, speeds)->set_core_count(vm->get_vm_impl()->get_core_amount());
160
161   cpu->seal();
162
163   if (get_iface()->get_pstate() != 0) {
164     cpu->set_pstate(get_iface()->get_pstate());
165   }
166
167   /* Currently, a VM uses the network resource of its physical host */
168   vm->set_netpoint(get_iface()->get_netpoint());
169
170   vm->seal();
171
172   return vm;
173 }
174
175 void HostImpl::move_vm(VirtualMachineImpl* vm, HostImpl* destination)
176 {
177   xbt_assert(vm && destination);
178
179   vms_.erase(vm->get_name());
180   destination->vms_[vm->get_name()] = vm;
181 }
182
183 void HostImpl::destroy_vm(const std::string& name)
184 {
185   auto* vm = vms_[name];
186   vms_.erase(name);
187   vm->vm_destroy();
188 }
189
190 VirtualMachineImpl* HostImpl::get_vm_by_name_or_null(const std::string& name) const
191 {
192   auto vm_it = vms_.find(name);
193   return vm_it == vms_.end() ? nullptr : vm_it->second;
194 }
195
196 std::vector<s4u::VirtualMachine*> HostImpl::get_vms() const
197 {
198   std::vector<s4u::VirtualMachine*> vms;
199   for (const auto& [_, vm] : vms_) {
200     vms.push_back(vm->get_iface());
201   }
202   return vms;
203 }
204
205 s4u::Disk* HostImpl::create_disk(const std::string& name, double read_bandwidth, double write_bandwidth)
206 {
207   auto* disk = piface_.get_netpoint()->get_englobing_zone()->get_disk_model()->create_disk(name, read_bandwidth,
208                                                                                            write_bandwidth);
209   if (sealed_)
210     disk->seal();
211   return disk->set_host(&piface_)->get_iface();
212 }
213
214 void HostImpl::add_disk(const s4u::Disk* disk)
215 {
216   disks_[disk->get_name()] = kernel::resource::DiskImplPtr(disk->get_impl());
217 }
218
219 void HostImpl::remove_disk(const std::string& name)
220 {
221   disks_.erase(name);
222 }
223
224 void HostImpl::seal()
225 {
226   if (sealed_) {
227     return;
228   }
229   // seals host's CPU
230   get_iface()->get_cpu()->seal();
231   sealed_ = true;
232
233   /* seal its disks */
234   for (auto const& [_, disk] : disks_)
235     disk->seal();
236
237   /* seal its VMs */
238   for (auto const& [_, vm] : vms_)
239     vm->seal();
240 }
241 } // namespace simgrid::kernel::resource