Wifi: Add crosstraffic on WIFI cells

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

diff --git a/src/surf/network_cm02.cpp b/src/surf/network_cm02.cpp
index 94409bb2134d3dcf1639ebb6c78c37d8e42d53df..6049efe98fd2316de0c6316399982546b7f1a0f1 100644 (file)
--- a/src/surf/network_cm02.cpp
+++ b/src/surf/network_cm02.cpp
@@ -244,7 +244,6 @@ Action* NetworkCm02Model::communicate(s4u::Host* src, s4u::Host* dst, double siz
       double src_rate = wifi_link->get_host_rate(src);
       double dst_rate = wifi_link->get_host_rate(dst);
 
-      // TODO: What do to when src and dst are on the same AP ? (for the moment we use src rate)
       if (src_rate != -1 && dst_rate != -1) {
         get_maxmin_system()->expand(link->get_constraint(), action->get_variable(), 1.0 / src_rate);
       } else {
@@ -263,10 +262,26 @@ Action* NetworkCm02Model::communicate(s4u::Host* src, s4u::Host* dst, double siz
 
   if (cfg_crosstraffic) {
     XBT_DEBUG("Crosstraffic active: adding backward flow using 5%% of the available bandwidth");
+    bool wifi_dst_assigned = false; // Used by wifi crosstraffic
     for (auto const& link : back_route) {
-      // Do not add crosstraffic on WIFI links
-      if (link->get_sharing_policy() != s4u::Link::SharingPolicy::WIFI)
+      if (link->get_sharing_policy() == s4u::Link::SharingPolicy::WIFI) {
+        NetworkWifiLink* wifi_link = static_cast<NetworkWifiLink*>(link);
+        /**
+         * For wifi links we should add 0.05/rate.
+         * However since we are using the "back_route" we should encounter in
+         * the first place the dst wifi link.
+         */
+        if (!wifi_dst_assigned && (wifi_link->get_host_rate(dst) != -1)) {
+          get_maxmin_system()->expand(link->get_constraint(), action->get_variable(),
+                                      .05 / wifi_link->get_host_rate(dst));
+          wifi_dst_assigned = true;
+        } else {
+          get_maxmin_system()->expand(link->get_constraint(), action->get_variable(),
+                                      .05 / wifi_link->get_host_rate(src));
+        }
+      } else {
         get_maxmin_system()->expand(link->get_constraint(), action->get_variable(), .05);
+      }
     }
 
     // Change concurrency_share here, if you want that cross-traffic is included in the SURF concurrency