Code Review / csit.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | review | tree
raw | inline | side by side
CSIT-545: Performance tests for SNAT
[csit.git] / resources / tools / t-rex / t-rex-stateless.py
diff --git a/resources/tools/t-rex/t-rex-stateless.py b/resources/tools/t-rex/t-rex-stateless.py
index 504ef50..ae8d187 100755 (executable)
--- a/resources/tools/t-rex/t-rex-stateless.py
+++ b/resources/tools/t-rex/t-rex-stateless.py
@@ -172,12 +172,13 @@ def create_streams_v46(base_pkt_a, base_pkt_b, vm1, vm2, frame_size):
 
     return (stream1, stream2, lat_stream1, lat_stream2)
 
+
 def create_streams(traffic_options, frame_size=64):
     """Create two IP packets to be used in stream.
 
     :param traffic_options: Parameters for packets.
     :param frame_size: Size of L2 frame.
-    :type traffic_options: list
+    :type traffic_options: dict
     :type frame_size: int
     :return: Packet instances.
     :rtype: Tuple of STLPktBuilder
@@ -197,13 +198,91 @@ def create_streams(traffic_options, frame_size=64):
     p1_dst_end_ip = traffic_options['p1_dst_end_ip']
     p2_dst_end_ip = traffic_options['p2_dst_end_ip']
 
-    base_pkt_a = Ether()/IP(src=p1_src_start_ip, dst=p1_dst_start_ip, proto=61)
-    base_pkt_b = Ether()/IP(src=p2_src_start_ip, dst=p2_dst_start_ip, proto=61)
+    try:
+        p1_src_start_udp_port = traffic_options['p1_src_start_udp_port']
+        p1_src_end_udp_port = traffic_options['p1_src_end_udp_port']
+        p1_dst_start_udp_port = traffic_options['p1_dst_start_udp_port']
+        p2_src_start_udp_port = traffic_options['p2_src_start_udp_port']
+        p2_dst_start_udp_port = traffic_options['p2_dst_start_udp_port']
+        p2_dst_end_udp_port = traffic_options['p2_dst_end_udp_port']
+        ports_defined = True
+    except KeyError:
+        ports_defined = False
+
+    if ports_defined:
+        base_pkt_a = (Ether() /
+                      IP(src=p1_src_start_ip, dst=p1_dst_start_ip, proto=17) /
+                      UDP(sport=int(p1_src_start_udp_port),
+                          dport=int(p1_dst_start_udp_port)))
+        base_pkt_b = (Ether() /
+                      IP(src=p2_src_start_ip, dst=p2_dst_start_ip, proto=17) /
+                      UDP(sport=int(p2_src_start_udp_port),
+                          dport=int(p2_dst_start_udp_port)))
+    else:
+        base_pkt_a = Ether() / IP(src=p1_src_start_ip, dst=p1_dst_start_ip,
+                                  proto=61)
+        base_pkt_b = Ether() / IP(src=p2_src_start_ip, dst=p2_dst_start_ip,
+                                  proto=61)
 
     # The following code applies raw instructions to packet (IP src/dst
     # increment). It splits the generated traffic by "ip_src"/"ip_dst" variable
     # to cores and fix IPv4 header checksum.
-    if p1_dst_end_ip and p2_dst_end_ip:
+    if ports_defined:
+        if p1_src_start_udp_port != p1_src_end_udp_port and \
+                        p1_src_start_ip != p1_src_end_ip:
+            vm1 = STLScVmRaw([
+                STLVmTupleGen(ip_min=p1_src_start_ip,
+                              ip_max=p1_src_end_ip,
+                              port_min=int(p1_src_start_udp_port),
+                              port_max=int(p1_src_end_udp_port),
+                              name="tuple"),
+                STLVmWrFlowVar(fv_name="tuple.ip", pkt_offset="IP.src"),
+                STLVmFixIpv4(offset="IP"),
+                STLVmWrFlowVar(fv_name="tuple.port", pkt_offset="UDP.sport")
+            ])
+            vm2 = STLScVmRaw([
+                STLVmTupleGen(ip_min=p2_dst_start_ip,
+                              ip_max=p2_dst_start_ip,
+                              port_min=int(p2_dst_start_udp_port),
+                              port_max=int(p2_dst_end_udp_port),
+                              name="tuple"),
+                STLVmWrFlowVar(fv_name="tuple.ip", pkt_offset="IP.dst"),
+                STLVmFixIpv4(offset="IP"),
+                STLVmWrFlowVar(fv_name="tuple.port", pkt_offset="UDP.dport")
+            ])
+        elif p1_src_start_udp_port != p1_src_end_udp_port:
+            vm1 = STLScVmRaw([
+                STLVmFlowVar(name="sport",
+                             min_value=int(p1_src_start_udp_port),
+                             max_value=int(p1_src_end_udp_port),
+                             size=2, op="inc"),
+                STLVmWrFlowVar(fv_name="sport", pkt_offset="UDP.sport")
+                ])
+            vm2 = STLScVmRaw([
+                STLVmFlowVar(name="dport",
+                             min_value=int(p2_dst_start_udp_port),
+                             max_value=int(p2_dst_end_udp_port),
+                             size=2, op="inc"),
+                STLVmWrFlowVar(fv_name="dport", pkt_offset="UDP.dport")
+            ])
+        elif p1_src_start_udp_port == p1_src_end_udp_port:
+            vm1 = STLScVmRaw([STLVmFlowVar(name="src",
+                                           min_value=p1_src_start_ip,
+                                           max_value=p1_src_end_ip,
+                                           size=4, op="inc"),
+                              STLVmWrFlowVar(fv_name="src",
+                                             pkt_offset="IP.src"),
+                              STLVmFixIpv4(offset="IP"),
+                              ], split_by_field="src")
+            vm2 = STLScVmRaw([STLVmFlowVar(name="src",
+                                           min_value=p2_src_start_ip,
+                                           max_value=p2_src_end_ip,
+                                           size=4, op="inc"),
+                              STLVmWrFlowVar(fv_name="src",
+                                             pkt_offset="IP.src"),
+                              STLVmFixIpv4(offset="IP"),
+                              ], split_by_field="src")
+    elif p1_dst_end_ip and p2_dst_end_ip:
         vm1 = STLScVmRaw([STLVmFlowVar(name="dst",
                                        min_value=p1_dst_start_ip,
                                        max_value=p1_dst_end_ip,
@@ -524,6 +603,7 @@ def parse_args():
 #                        help="Port 1 source MAC address")
 #    parser.add_argument("--p1_dst_mac",
 #                        help="Port 1 destination MAC address")
+
     parser.add_argument("--p1_src_start_ip", required=True,
                         help="Port 1 source start IP address")
     parser.add_argument("--p1_src_end_ip",
@@ -534,10 +614,26 @@ def parse_args():
     parser.add_argument("--p1_dst_end_ip",
                         default=False,
                         help="Port 1 destination end IP address")
+
+    parser.add_argument("--p1_src_start_udp_port",
+                        default=None,
+                        help="Port 1 source start UDP port.")
+    parser.add_argument("--p1_src_end_udp_port",
+                        default=None,
+                        help="Port 1 source end UDP port.")
+
+    parser.add_argument("--p1_dst_start_udp_port",
+                        default=None,
+                        help="Port 1 destination start UDP port.")
+    parser.add_argument("--p1_dst_end_udp_port",
+                        default=None,
+                        help="Port 1 destination end UDP port.")
+
 #    parser.add_argument("--p2_src_mac",
 #                        help="Port 2 source MAC address")
 #    parser.add_argument("--p2_dst_mac",
 #                        help="Port 2 destination MAC address")
+
     parser.add_argument("--p2_src_start_ip", required=True,
                         help="Port 2 source start IP address")
     parser.add_argument("--p2_src_end_ip",
@@ -549,6 +645,20 @@ def parse_args():
                         default=False,
                         help="Port 2 destination end IP address")
 
+    parser.add_argument("--p2_src_start_udp_port",
+                        default=None,
+                        help="Port 2 source start UDP port.")
+    parser.add_argument("--p2_src_end_udp_port",
+                        default=None,
+                        help="Port 2 source end UDP port.")
+
+    parser.add_argument("--p2_dst_start_udp_port",
+                        default=None,
+                        help="Port 2 destination start UDP port.")
+    parser.add_argument("--p2_dst_end_udp_port",
+                        default=None,
+                        help="Port 2 destination end UDP port.")
+
     return parser.parse_args()
 
 
Integration tests