Add load balance example

Author: Mingyumaz

examples/CMakeLists.txt

@@ -72,6 +72,18 @@ build_lib_example(
     ${libcsma}
 )
 
+# 4 hosts 4 switches topology [Link Monitoring]
+build_lib_example(
+  NAME p4-spine-leaf-topo
+  SOURCE_FILES p4-spine-leaf-topo.cc
+  LIBRARIES_TO_LINK
+    ${libp4sim}
+    ${libinternet}
+    ${libapplications}
+    ${libnetwork}
+    ${libcsma}
+)
+
 # [ ================= PSA Arch ================= ]
 
 # simple 2 hosts 1 switch topology [ipv4 forwarding]

examples/p4-basic-tunnel.cc

@@ -44,8 +44,8 @@ double client_stop_time = client_start_time + 5; // sending time
 double sink_stop_time = client_stop_time + 5;
 double global_stop_time = sink_stop_time + 5;
 
-bool first_tx = true;
-bool first_rx = true;
+// bool first_tx = true;
+// bool first_rx = true;
 int counter_sender_10 = 10;
 int counter_receiver_10 = 10;
 double first_packet_send_time_tx = 0.0;
@@ -91,81 +91,29 @@ ConvertMacToHex(Address macAddr)
 void
 TxCallback(Ptr<const Packet> packet)
 {
-    if (first_tx)
-    {
-        // here we just simple jump the first 10 pkts (include some of ARP packets)
-        first_packet_send_time_tx = Simulator::Now().GetSeconds();
-        counter_sender_10--;
-        if (counter_sender_10 == 0)
-        {
-            first_tx = false;
-        }
-    }
-    else
-    {
-        totalTxBytes_1 += packet->GetSize();
-        last_packet_send_time_tx = Simulator::Now().GetSeconds();
-    }
+    totalTxBytes_1 += packet->GetSize();
+    last_packet_send_time_tx = Simulator::Now().GetSeconds();
 }
 
 void
 RxCallback(Ptr<const Packet> packet, const Address& addr)
 {
-    if (first_rx)
-    {
-        // here we just simple jump the first 10 pkts (include some of ARP packets)
-        first_packet_received_time_rx = Simulator::Now().GetSeconds();
-        counter_receiver_10--;
-        if (counter_receiver_10 == 0)
-        {
-            first_rx = false;
-        }
-    }
-    else
-    {
-        totalRxBytes_1 += packet->GetSize();
-        last_packet_received_time_rx = Simulator::Now().GetSeconds();
-    }
+    totalRxBytes_1 += packet->GetSize();
+    last_packet_received_time_rx = Simulator::Now().GetSeconds();
 }
 
 void
 TxCallback_2(Ptr<const Packet> packet)
 {
-    if (first_tx)
-    {
-        // here we just simple jump the first 10 pkts (include some of ARP packets)
-        first_packet_send_time_tx = Simulator::Now().GetSeconds();
-        counter_sender_10--;
-        if (counter_sender_10 == 0)
-        {
-            first_tx = false;
-        }
-    }
-    else
-    {
-        totalTxBytes_2 += packet->GetSize();
-        last_packet_send_time_tx = Simulator::Now().GetSeconds();
-    }
+    totalTxBytes_2 += packet->GetSize();
+    last_packet_send_time_tx = Simulator::Now().GetSeconds();
 }
 
 void
 RxCallback_2(Ptr<const Packet> packet, const Address& addr)
 {
-    if (first_rx)
-    {
-        // here we just simple jump the first 10 pkts (include some of ARP packets)
-        first_packet_received_time_rx = Simulator::Now().GetSeconds();
-        counter_receiver_10--;
-        if (counter_receiver_10 == 0)
-        {
-            first_rx = false;
-        }
-    }
-    else
-    {
-        totalRxBytes_2 += packet->GetSize();
-        last_packet_received_time_rx = Simulator::Now().GetSeconds();
-    }
+    totalRxBytes_2 += packet->GetSize();
+    last_packet_received_time_rx = Simulator::Now().GetSeconds();
 }
 
 void
@@ -231,6 +179,8 @@ main(int argc, char* argv[])
 {
     LogComponentEnable("P4BasicTunnel", LOG_LEVEL_INFO);
 
+    Packet::EnablePrinting();
+
     // ============================ parameters ============================
 
     int running_number = 0;
@@ -397,7 +347,7 @@ main(int argc, char* argv[])
     // Bridge or P4 switch configuration
     P4Helper p4SwitchHelper;
     p4SwitchHelper.SetDeviceAttribute("JsonPath", StringValue(p4JsonPath));
-    p4SwitchHelper.SetDeviceAttribute("ChannelType", UintegerValue(0));
+    // p4SwitchHelper.SetDeviceAttribute("ChannelType", UintegerValue(0));
     p4SwitchHelper.SetDeviceAttribute("P4SwitchArch", UintegerValue(0));
     p4SwitchHelper.SetDeviceAttribute("ChannelType", UintegerValue(1));
 
@@ -460,7 +410,7 @@ main(int argc, char* argv[])
     onOff1.SetAttribute("DataRate", StringValue(appDataRate[0]));
     onOff1.SetAttribute("OnTime", StringValue("ns3::ConstantRandomVariable[Constant=1]"));
     onOff1.SetAttribute("OffTime", StringValue("ns3::ConstantRandomVariable[Constant=0]"));
-    // onOff1.SetAttribute ("MaxBytes", UintegerValue (3000));
+    onOff1.SetAttribute("MaxBytes", UintegerValue(5000));
 
     ApplicationContainer app1 = onOff1.Install(terminals.Get(clientI));
     app1.Start(Seconds(client_start_time));
@@ -491,15 +441,15 @@ main(int argc, char* argv[])
     onOff2.SetAttribute("DataRate", StringValue(appDataRate[1]));
     onOff2.SetAttribute("OnTime", StringValue("ns3::ConstantRandomVariable[Constant=1]"));
     onOff2.SetAttribute("OffTime", StringValue("ns3::ConstantRandomVariable[Constant=0]"));
-    // onOff2.SetAttribute("MaxBytes", UintegerValue(500));
+    onOff2.SetAttribute("MaxBytes", UintegerValue(10000));
 
     ApplicationContainer app2 = onOff2.Install(terminals.Get(clientI));
     app2.Start(Seconds(client_start_time));
     app2.Stop(Seconds(client_stop_time));
 
     // === Setup Tracing ===
     Ptr<OnOffApplication> ptr_app2 =
-        DynamicCast<OnOffApplication>(terminals.Get(clientI)->GetApplication(0));
+        DynamicCast<OnOffApplication>(terminals.Get(clientI)->GetApplication(1));
     ptr_app2->TraceConnectWithoutContext("Tx", MakeCallback(&TxCallback_2));
     sinkApp2.Get(0)->TraceConnectWithoutContext("Rx", MakeCallback(&RxCallback_2));