Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Copy models from parent netzone
[simgrid.git] / src / kernel / routing / NetZoneImpl.cpp
1 /* Copyright (c) 2006-2021. 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/NetZoneImpl.hpp"
7 #include "simgrid/kernel/routing/NetPoint.hpp"
8 #include "simgrid/s4u/Engine.hpp"
9 #include "simgrid/s4u/Host.hpp"
10 #include "src/include/simgrid/sg_config.hpp"
11 #include "src/kernel/EngineImpl.hpp"
12 #include "src/kernel/resource/DiskImpl.hpp"
13 #include "src/surf/HostImpl.hpp"
14 #include "src/surf/cpu_interface.hpp"
15 #include "src/surf/network_interface.hpp"
16 #include "src/surf/xml/platf_private.hpp"
17 #include "surf/surf.hpp"
18
19 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(surf_route);
20
21 namespace simgrid {
22 namespace kernel {
23 namespace routing {
24
25 /* Pick the right models for CPU, net and host, and call their model_init_preparse */
26 static void surf_config_models_setup()
27 {
28   std::string host_model_name    = simgrid::config::get_value<std::string>("host/model");
29   std::string network_model_name = simgrid::config::get_value<std::string>("network/model");
30   std::string cpu_model_name     = simgrid::config::get_value<std::string>("cpu/model");
31   std::string disk_model_name    = simgrid::config::get_value<std::string>("disk/model");
32
33   /* The compound host model is needed when using non-default net/cpu models */
34   if ((not simgrid::config::is_default("network/model") || not simgrid::config::is_default("cpu/model")) &&
35       simgrid::config::is_default("host/model")) {
36     host_model_name = "compound";
37     simgrid::config::set_value("host/model", host_model_name);
38   }
39
40   XBT_DEBUG("host model: %s", host_model_name.c_str());
41   if (host_model_name == "compound") {
42     xbt_assert(not cpu_model_name.empty(), "Set a cpu model to use with the 'compound' host model");
43     xbt_assert(not network_model_name.empty(), "Set a network model to use with the 'compound' host model");
44
45     int cpu_id = find_model_description(surf_cpu_model_description, cpu_model_name);
46     surf_cpu_model_description[cpu_id].model_init_preparse();
47
48     int network_id = find_model_description(surf_network_model_description, network_model_name);
49     surf_network_model_description[network_id].model_init_preparse();
50   }
51
52   XBT_DEBUG("Call host_model_init");
53   int host_id = find_model_description(surf_host_model_description, host_model_name);
54   surf_host_model_description[host_id].model_init_preparse();
55
56   XBT_DEBUG("Call vm_model_init");
57   /* ideally we should get back the pointer to CpuModel from model_init_preparse(), but this
58    * requires changing the declaration of surf_cpu_model_description.
59    * To be reviewed in the future */
60   surf_vm_model_init_HL13(
61       simgrid::s4u::Engine::get_instance()->get_netzone_root()->get_impl()->get_cpu_pm_model().get());
62
63   XBT_DEBUG("Call disk_model_init");
64   int disk_id = find_model_description(surf_disk_model_description, disk_model_name);
65   surf_disk_model_description[disk_id].model_init_preparse();
66 }
67
68 NetZoneImpl::NetZoneImpl(const std::string& name) : piface_(this), name_(name)
69 {
70   /* workaroud: first netzoneImpl will be the root netzone.
71    * Without globals and with current surf_*_model_description init functions, we need
72    * the root netzone to exist when creating the models.
73    * This was usually done at sg_platf.cpp, during XML parsing */
74   if (not s4u::Engine::get_instance()->get_netzone_root())
75     s4u::Engine::get_instance()->set_netzone_root(&piface_);
76
77   static bool surf_parse_models_setup_already_called = false;
78   if (not surf_parse_models_setup_already_called) {
79     simgrid::s4u::Engine::on_platform_creation();
80
81     /* Initialize the surf models. That must be done after we got all config, and before we need the models.
82      * That is, after the last <config> tag, if any, and before the first of cluster|peer|zone|trace|trace_connect
83      *
84      * I'm not sure for <trace> and <trace_connect>, there may be a bug here
85      * (FIXME: check it out by creating a file beginning with one of these tags)
86      * but cluster and peer come down to zone creations, so putting this verification here is correct.
87      */
88     surf_parse_models_setup_already_called = true;
89     surf_config_models_setup();
90   }
91
92   xbt_assert(nullptr == s4u::Engine::get_instance()->netpoint_by_name_or_null(get_name()),
93              "Refusing to create a second NetZone called '%s'.", get_cname());
94   netpoint_ = new NetPoint(name_, NetPoint::Type::NetZone);
95   XBT_DEBUG("NetZone '%s' created with the id '%u'", get_cname(), netpoint_->id());
96 }
97
98 NetZoneImpl::~NetZoneImpl()
99 {
100   for (auto const& nz : children_)
101     delete nz;
102
103   for (auto const& kv : bypass_routes_)
104     delete kv.second;
105
106   s4u::Engine::get_instance()->netpoint_unregister(netpoint_);
107 }
108
109 void NetZoneImpl::add_child(NetZoneImpl* new_zone)
110 {
111   xbt_assert(not sealed_, "Cannot add a new child to the sealed zone %s", get_cname());
112   children_.push_back(new_zone);
113 }
114
115 /** @brief Returns the list of the hosts found in this NetZone (not recursively)
116  *
117  * Only the hosts that are directly contained in this NetZone are retrieved,
118  * not the ones contained in sub-netzones.
119  */
120 std::vector<s4u::Host*> NetZoneImpl::get_all_hosts() const
121 {
122   std::vector<s4u::Host*> res;
123   for (auto const& card : get_vertices()) {
124     s4u::Host* host = s4u::Host::by_name_or_null(card->get_name());
125     if (host != nullptr)
126       res.push_back(host);
127   }
128   return res;
129 }
130 int NetZoneImpl::get_host_count() const
131 {
132   int count = 0;
133   for (auto const& card : get_vertices()) {
134     const s4u::Host* host = s4u::Host::by_name_or_null(card->get_name());
135     if (host != nullptr)
136       count++;
137   }
138   return count;
139 }
140
141 s4u::Disk* NetZoneImpl::create_disk(const std::string& name, double read_bandwidth, double write_bandwidth)
142 {
143   auto* l = disk_model_->create_disk(name, read_bandwidth, write_bandwidth);
144
145   return l->get_iface();
146 }
147
148 s4u::Link* NetZoneImpl::create_link(const std::string& name, const std::vector<double>& bandwidths)
149 {
150   return network_model_->create_link(name, bandwidths)->get_iface();
151 }
152
153 s4u::Link* NetZoneImpl::create_wifi_link(const std::string& name, const std::vector<double>& bandwidths)
154 {
155   return network_model_->create_wifi_link(name, bandwidths)->get_iface();
156 }
157
158 s4u::Host* NetZoneImpl::create_host(const std::string& name, const std::vector<double>& speed_per_pstate)
159 {
160   if (hierarchy_ == RoutingMode::unset)
161     hierarchy_ = RoutingMode::base;
162
163   auto* res = (new surf::HostImpl(name))->get_iface();
164   res->set_netpoint((new NetPoint(name, NetPoint::Type::Host))->set_englobing_zone(this));
165
166   cpu_model_pm_->create_cpu(res, speed_per_pstate);
167
168   return res;
169 }
170
171 int NetZoneImpl::add_component(NetPoint* elm)
172 {
173   vertices_.push_back(elm);
174   return vertices_.size() - 1; // The rank of the newly created object
175 }
176
177 void NetZoneImpl::add_route(NetPoint* /*src*/, NetPoint* /*dst*/, NetPoint* /*gw_src*/, NetPoint* /*gw_dst*/,
178                             const std::vector<resource::LinkImpl*>& /*link_list*/, bool /*symmetrical*/)
179 {
180   xbt_die("NetZone '%s' does not accept new routes (wrong class).", get_cname());
181 }
182
183 void NetZoneImpl::add_bypass_route(NetPoint* src, NetPoint* dst, NetPoint* gw_src, NetPoint* gw_dst,
184                                    std::vector<resource::LinkImpl*>& link_list, bool /* symmetrical */)
185 {
186   /* Argument validity checks */
187   if (gw_dst) {
188     XBT_DEBUG("Load bypassNetzoneRoute from %s@%s to %s@%s", src->get_cname(), gw_src->get_cname(), dst->get_cname(),
189               gw_dst->get_cname());
190     xbt_assert(not link_list.empty(), "Bypass route between %s@%s and %s@%s cannot be empty.", src->get_cname(),
191                gw_src->get_cname(), dst->get_cname(), gw_dst->get_cname());
192     xbt_assert(bypass_routes_.find({src, dst}) == bypass_routes_.end(),
193                "The bypass route between %s@%s and %s@%s already exists.", src->get_cname(), gw_src->get_cname(),
194                dst->get_cname(), gw_dst->get_cname());
195   } else {
196     XBT_DEBUG("Load bypassRoute from %s to %s", src->get_cname(), dst->get_cname());
197     xbt_assert(not link_list.empty(), "Bypass route between %s and %s cannot be empty.", src->get_cname(),
198                dst->get_cname());
199     xbt_assert(bypass_routes_.find({src, dst}) == bypass_routes_.end(),
200                "The bypass route between %s and %s already exists.", src->get_cname(), dst->get_cname());
201   }
202
203   /* Build a copy that will be stored in the dict */
204   auto* newRoute = new BypassRoute(gw_src, gw_dst);
205   for (auto const& link : link_list)
206     newRoute->links.push_back(link);
207
208   /* Store it */
209   bypass_routes_.insert({{src, dst}, newRoute});
210 }
211
212 /** @brief Get the common ancestor and its first children in each line leading to src and dst
213  *
214  * In the recursive case, this sets common_ancestor, src_ancestor and dst_ancestor are set as follows.
215  * @verbatim
216  *         platform root
217  *               |
218  *              ...                <- possibly long path
219  *               |
220  *         common_ancestor
221  *           /        \
222  *          /          \
223  *         /            \          <- direct links
224  *        /              \
225  *       /                \
226  *  src_ancestor     dst_ancestor  <- must be different in the recursive case
227  *      |                   |
228  *     ...                 ...     <-- possibly long paths (one hop or more)
229  *      |                   |
230  *     src                 dst
231  *  @endverbatim
232  *
233  *  In the base case (when src and dst are in the same netzone), things are as follows:
234  *  @verbatim
235  *                  platform root
236  *                        |
237  *                       ...                      <-- possibly long path
238  *                        |
239  * common_ancestor==src_ancestor==dst_ancestor    <-- all the same value
240  *                   /        \
241  *                  /          \                  <-- direct links (exactly one hop)
242  *                 /            \
243  *              src              dst
244  *  @endverbatim
245  *
246  * A specific recursive case occurs when src is the ancestor of dst. In this case,
247  * the base case routing should be used so the common_ancestor is specifically set
248  * to src_ancestor==dst_ancestor.
249  * Naturally, things are completely symmetrical if dst is the ancestor of src.
250  * @verbatim
251  *            platform root
252  *                  |
253  *                 ...                <-- possibly long path
254  *                  |
255  *  src == src_ancestor==dst_ancestor==common_ancestor <-- same value
256  *                  |
257  *                 ...                <-- possibly long path (one hop or more)
258  *                  |
259  *                 dst
260  *  @endverbatim
261  */
262 static void find_common_ancestors(NetPoint* src, NetPoint* dst,
263                                   /* OUT */ NetZoneImpl** common_ancestor, NetZoneImpl** src_ancestor,
264                                   NetZoneImpl** dst_ancestor)
265 {
266   /* Deal with the easy base case */
267   if (src->get_englobing_zone() == dst->get_englobing_zone()) {
268     *common_ancestor = src->get_englobing_zone();
269     *src_ancestor    = *common_ancestor;
270     *dst_ancestor    = *common_ancestor;
271     return;
272   }
273
274   /* engage the full recursive search */
275
276   /* (1) find the path to root of src and dst*/
277   const NetZoneImpl* src_as = src->get_englobing_zone();
278   const NetZoneImpl* dst_as = dst->get_englobing_zone();
279
280   xbt_assert(src_as, "Host %s must be in a netzone", src->get_cname());
281   xbt_assert(dst_as, "Host %s must be in a netzone", dst->get_cname());
282
283   /* (2) find the path to the root routing component */
284   std::vector<NetZoneImpl*> path_src;
285   NetZoneImpl* current = src->get_englobing_zone();
286   while (current != nullptr) {
287     path_src.push_back(current);
288     current = current->get_parent();
289   }
290   std::vector<NetZoneImpl*> path_dst;
291   current = dst->get_englobing_zone();
292   while (current != nullptr) {
293     path_dst.push_back(current);
294     current = current->get_parent();
295   }
296
297   /* (3) find the common father.
298    * Before that, index_src and index_dst may be different, they both point to nullptr in path_src/path_dst
299    * So we move them down simultaneously as long as they point to the same content.
300    *
301    * This works because all SimGrid platform have a unique root element (that is the last element of both paths).
302    */
303   NetZoneImpl* father = nullptr; // the netzone we dropped on the previous loop iteration
304   while (path_src.size() > 1 && path_dst.size() > 1 &&
305          path_src.at(path_src.size() - 1) == path_dst.at(path_dst.size() - 1)) {
306     father = path_src.at(path_src.size() - 1);
307     path_src.pop_back();
308     path_dst.pop_back();
309   }
310
311   /* (4) we found the difference at least. Finalize the returned values */
312   *src_ancestor = path_src.at(path_src.size() - 1); /* the first different father of src */
313   *dst_ancestor = path_dst.at(path_dst.size() - 1); /* the first different father of dst */
314   if (*src_ancestor == *dst_ancestor) {             // src is the ancestor of dst, or the contrary
315     *common_ancestor = *src_ancestor;
316   } else {
317     *common_ancestor = father;
318   }
319 }
320
321 /* PRECONDITION: this is the common ancestor of src and dst */
322 bool NetZoneImpl::get_bypass_route(NetPoint* src, NetPoint* dst,
323                                    /* OUT */ std::vector<resource::LinkImpl*>& links, double* latency)
324 {
325   // If never set a bypass route return nullptr without any further computations
326   if (bypass_routes_.empty())
327     return false;
328
329   /* Base case, no recursion is needed */
330   if (dst->get_englobing_zone() == this && src->get_englobing_zone() == this) {
331     if (bypass_routes_.find({src, dst}) != bypass_routes_.end()) {
332       const BypassRoute* bypassedRoute = bypass_routes_.at({src, dst});
333       for (resource::LinkImpl* const& link : bypassedRoute->links) {
334         links.push_back(link);
335         if (latency)
336           *latency += link->get_latency();
337       }
338       XBT_DEBUG("Found a bypass route from '%s' to '%s' with %zu links", src->get_cname(), dst->get_cname(),
339                 bypassedRoute->links.size());
340       return true;
341     }
342     return false;
343   }
344
345   /* Engage recursive search */
346
347   /* (1) find the path to the root routing component */
348   std::vector<NetZoneImpl*> path_src;
349   NetZoneImpl* current = src->get_englobing_zone();
350   while (current != nullptr) {
351     path_src.push_back(current);
352     current = current->parent_;
353   }
354
355   std::vector<NetZoneImpl*> path_dst;
356   current = dst->get_englobing_zone();
357   while (current != nullptr) {
358     path_dst.push_back(current);
359     current = current->parent_;
360   }
361
362   /* (2) find the common father */
363   while (path_src.size() > 1 && path_dst.size() > 1 &&
364          path_src.at(path_src.size() - 1) == path_dst.at(path_dst.size() - 1)) {
365     path_src.pop_back();
366     path_dst.pop_back();
367   }
368
369   int max_index_src = path_src.size() - 1;
370   int max_index_dst = path_dst.size() - 1;
371
372   int max_index = std::max(max_index_src, max_index_dst);
373
374   /* (3) Search for a bypass making the path up to the ancestor useless */
375   const BypassRoute* bypassedRoute = nullptr;
376   std::pair<kernel::routing::NetPoint*, kernel::routing::NetPoint*> key;
377   for (int max = 0; max <= max_index && not bypassedRoute; max++) {
378     for (int i = 0; i < max && not bypassedRoute; i++) {
379       if (i <= max_index_src && max <= max_index_dst) {
380         key      = {path_src.at(i)->netpoint_, path_dst.at(max)->netpoint_};
381         auto bpr = bypass_routes_.find(key);
382         if (bpr != bypass_routes_.end()) {
383           bypassedRoute = bpr->second;
384         }
385       }
386       if (not bypassedRoute && max <= max_index_src && i <= max_index_dst) {
387         key      = {path_src.at(max)->netpoint_, path_dst.at(i)->netpoint_};
388         auto bpr = bypass_routes_.find(key);
389         if (bpr != bypass_routes_.end()) {
390           bypassedRoute = bpr->second;
391         }
392       }
393     }
394
395     if (not bypassedRoute && max <= max_index_src && max <= max_index_dst) {
396       key      = {path_src.at(max)->netpoint_, path_dst.at(max)->netpoint_};
397       auto bpr = bypass_routes_.find(key);
398       if (bpr != bypass_routes_.end()) {
399         bypassedRoute = bpr->second;
400       }
401     }
402   }
403
404   /* (4) If we have the bypass, use it. If not, caller will do the Right Thing. */
405   if (bypassedRoute) {
406     XBT_DEBUG("Found a bypass route from '%s' to '%s' with %zu links. We may have to complete it with recursive "
407               "calls to getRoute",
408               src->get_cname(), dst->get_cname(), bypassedRoute->links.size());
409     if (src != key.first)
410       get_global_route(src, bypassedRoute->gw_src, links, latency);
411     for (resource::LinkImpl* const& link : bypassedRoute->links) {
412       links.push_back(link);
413       if (latency)
414         *latency += link->get_latency();
415     }
416     if (dst != key.second)
417       get_global_route(bypassedRoute->gw_dst, dst, links, latency);
418     return true;
419   }
420   XBT_DEBUG("No bypass route from '%s' to '%s'.", src->get_cname(), dst->get_cname());
421   return false;
422 }
423
424 void NetZoneImpl::get_global_route(NetPoint* src, NetPoint* dst,
425                                    /* OUT */ std::vector<resource::LinkImpl*>& links, double* latency)
426 {
427   RouteCreationArgs route;
428
429   XBT_DEBUG("Resolve route from '%s' to '%s'", src->get_cname(), dst->get_cname());
430
431   /* Find how src and dst are interconnected */
432   NetZoneImpl* common_ancestor;
433   NetZoneImpl* src_ancestor;
434   NetZoneImpl* dst_ancestor;
435   find_common_ancestors(src, dst, &common_ancestor, &src_ancestor, &dst_ancestor);
436   XBT_DEBUG("elements_father: common ancestor '%s' src ancestor '%s' dst ancestor '%s'", common_ancestor->get_cname(),
437             src_ancestor->get_cname(), dst_ancestor->get_cname());
438
439   /* Check whether a direct bypass is defined. If so, use it and bail out */
440   if (common_ancestor->get_bypass_route(src, dst, links, latency))
441     return;
442
443   /* If src and dst are in the same netzone, life is good */
444   if (src_ancestor == dst_ancestor) { /* SURF_ROUTING_BASE */
445     route.link_list = std::move(links);
446     common_ancestor->get_local_route(src, dst, &route, latency);
447     links = std::move(route.link_list);
448     return;
449   }
450
451   /* Not in the same netzone, no bypass. We'll have to find our path between the netzones recursively */
452
453   common_ancestor->get_local_route(src_ancestor->netpoint_, dst_ancestor->netpoint_, &route, latency);
454   xbt_assert((route.gw_src != nullptr) && (route.gw_dst != nullptr), "Bad gateways for route from '%s' to '%s'.",
455              src->get_cname(), dst->get_cname());
456
457   /* If source gateway is not our source, we have to recursively find our way up to this point */
458   if (src != route.gw_src)
459     get_global_route(src, route.gw_src, links, latency);
460   for (auto const& link : route.link_list)
461     links.push_back(link);
462
463   /* If dest gateway is not our destination, we have to recursively find our way from this point */
464   if (route.gw_dst != dst)
465     get_global_route(route.gw_dst, dst, links, latency);
466 }
467
468 void NetZoneImpl::seal()
469 {
470   /* already sealed netzone */
471   if (sealed_)
472     return;
473   do_seal(); // derived class' specific sealing procedure
474
475   /* seals sub-netzones and hosts */
476   for (auto* host : get_all_hosts()) {
477     host->seal();
478   }
479   for (auto* sub_net : *get_children()) {
480     sub_net->seal();
481   }
482   sealed_ = true;
483 }
484
485 void NetZoneImpl::set_parent(NetZoneImpl* parent)
486 {
487   xbt_assert(not sealed_, "Impossible to set parent to an already sealed NetZone(%s)", this->get_cname());
488   parent_ = parent;
489   netpoint_->set_englobing_zone(parent_);
490   /* copying models from parent host, to be reviewed when we allow multi-models */
491   if (parent) {
492     set_network_model(parent->get_network_model());
493     set_cpu_pm_model(parent->get_cpu_pm_model());
494     set_cpu_vm_model(parent->get_cpu_vm_model());
495     set_disk_model(parent->get_disk_model());
496     set_host_model(parent->get_host_model());
497   }
498 }
499
500 void NetZoneImpl::set_network_model(std::shared_ptr<resource::NetworkModel> netmodel)
501 {
502   xbt_assert(not sealed_, "Impossible to set network model to an already sealed NetZone(%s)", this->get_cname());
503   network_model_ = std::move(netmodel);
504 }
505
506 void NetZoneImpl::set_cpu_vm_model(std::shared_ptr<resource::CpuModel> cpu_model)
507 {
508   xbt_assert(not sealed_, "Impossible to set CPU model to an already sealed NetZone(%s)", this->get_cname());
509   cpu_model_vm_ = std::move(cpu_model);
510 }
511
512 void NetZoneImpl::set_cpu_pm_model(std::shared_ptr<resource::CpuModel> cpu_model)
513 {
514   xbt_assert(not sealed_, "Impossible to set CPU model to an already sealed NetZone(%s)", this->get_cname());
515   cpu_model_pm_ = std::move(cpu_model);
516 }
517
518 void NetZoneImpl::set_disk_model(std::shared_ptr<resource::DiskModel> disk_model)
519 {
520   xbt_assert(not sealed_, "Impossible to set disk model to an already sealed NetZone(%s)", this->get_cname());
521   disk_model_ = std::move(disk_model);
522 }
523
524 void NetZoneImpl::set_host_model(std::shared_ptr<surf::HostModel> host_model)
525 {
526   xbt_assert(not sealed_, "Impossible to set host model to an already sealed NetZone(%s)", this->get_cname());
527   host_model_ = std::move(host_model);
528 }
529
530 } // namespace routing
531 } // namespace kernel
532 } // namespace simgrid