nat: add more trex test scripts

Type: improvement
Signed-off-by: Klement Sekera <ksekera@cisco.com>
Change-Id: Ia824d663c86d6a7c9bafe0206e9a93f93884739d

3 files changed, 206 insertions, 0 deletions

diff --git a/src/plugins/nat/extras/nat_test_fast_path.py b/src/plugins/nat/extras/nat_test_fast_path.py
new file mode 100644
index 00000000000..97f8525fdc6
--- /dev/null
+++ b/src/plugins/nat/extras/nat_test_fast_path.py
@@ -0,0 +1,67 @@
+from trex_stl_lib.api import *
+
+
+class STLS1(object):
+
+    def create_stream(self):
+        # base_pkt = Ether()/IP(dst="2.2.0.1")/UDP(dport=12)
+
+        # pad = Padding()
+        # if len(base_pkt) < 64:
+        #     pad_len = 64 - len(base_pkt)
+        #     pad.load = '\x00' * pad_len
+
+        # vm = STLVM()
+
+        # vm.tuple_var(name="tuple", ip_min="10.0.0.3", ip_max="10.0.0.202", port_min=1025, port_max=61124, limit_flows = 100000)
+
+        # vm.write(fv_name="tuple.ip", pkt_offset="IP.src")
+        # vm.fix_chksum()
+
+        # vm.write(fv_name="tuple.port", pkt_offset="UDP.sport")
+
+        # pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        # return STLStream(packet=pkt, mode=STLTXCont())
+
+        vm = STLScVmRaw([STLVmTupleGen(ip_min="10.0.0.1", ip_max="10.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       # name="stuple", limit_flows=10000),
+                                       name="stuple", limit_flows=100),
+                         STLVmTupleGen(ip_min="2.0.0.1", ip_max="2.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       # name="dtuple", limit_flows=100000000),
+                                       name="dtuple", limit_flows=100),
+
+                         # write ip to packet IP.src
+                         STLVmWrFlowVar(fv_name="stuple.ip",
+                                        pkt_offset="IP.src"),
+                         STLVmWrFlowVar(fv_name="dtuple.ip",
+                                        pkt_offset="IP.dst"),
+                         # fix checksum
+                         STLVmFixIpv4(offset="IP"),
+                         # write udp.port
+                         STLVmWrFlowVar(fv_name="stuple.port",
+                                        pkt_offset="UDP.sport"),
+                         STLVmWrFlowVar(fv_name="dtuple.port",
+                                        pkt_offset="UDP.dport"),
+                         ]
+                        )
+
+        base_pkt = Ether()/IP(src="16.0.0.1", dst="2.0.0.1")/UDP(dport=12, sport=1025)
+        pad = Padding()
+        if len(base_pkt) < 64:
+            pad_len = 64 - len(base_pkt)
+            pad.load = '\x00' * pad_len
+
+        pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        return STLStream(packet=pkt, mode=STLTXCont())
+
+    def get_streams(self, direction=0, **kwargs):
+        return [self.create_stream()]
+
+
+# dynamic load - used for trex console or simulator
+def register():
+    return STLS1()
diff --git a/src/plugins/nat/extras/nat_test_slow_path.py b/src/plugins/nat/extras/nat_test_slow_path.py
new file mode 100644
index 00000000000..6da2fe6f6b9
--- /dev/null
+++ b/src/plugins/nat/extras/nat_test_slow_path.py
@@ -0,0 +1,65 @@
+from trex_stl_lib.api import *
+
+
+class STLS1(object):
+
+    def create_stream(self):
+        # base_pkt = Ether()/IP(dst="2.2.0.1")/UDP(dport=12)
+
+        # pad = Padding()
+        # if len(base_pkt) < 64:
+        #     pad_len = 64 - len(base_pkt)
+        #     pad.load = '\x00' * pad_len
+
+        # vm = STLVM()
+
+        # vm.tuple_var(name="tuple", ip_min="10.0.0.3", ip_max="10.0.0.202", port_min=1025, port_max=61124, limit_flows = 100000)
+
+        # vm.write(fv_name="tuple.ip", pkt_offset="IP.src")
+        # vm.fix_chksum()
+
+        # vm.write(fv_name="tuple.port", pkt_offset="UDP.sport")
+
+        # pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        # return STLStream(packet=pkt, mode=STLTXCont())
+
+        vm = STLScVmRaw([STLVmTupleGen(ip_min="10.0.0.1", ip_max="10.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       name="stuple", limit_flows=10000),
+                         STLVmTupleGen(ip_min="2.0.0.1", ip_max="2.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       name="dtuple", limit_flows=100000000),
+
+                         # write ip to packet IP.src
+                         STLVmWrFlowVar(fv_name="stuple.ip",
+                                        pkt_offset="IP.src"),
+                         STLVmWrFlowVar(fv_name="dtuple.ip",
+                                        pkt_offset="IP.dst"),
+                         # fix checksum
+                         STLVmFixIpv4(offset="IP"),
+                         # write udp.port
+                         STLVmWrFlowVar(fv_name="stuple.port",
+                                        pkt_offset="UDP.sport"),
+                         STLVmWrFlowVar(fv_name="dtuple.port",
+                                        pkt_offset="UDP.dport"),
+                         ]
+                        )
+
+        base_pkt = Ether()/IP(src="16.0.0.1", dst="2.0.0.1")/UDP(dport=12, sport=1025)
+        pad = Padding()
+        if len(base_pkt) < 64:
+            pad_len = 64 - len(base_pkt)
+            pad.load = '\x00' * pad_len
+
+        pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        return STLStream(packet=pkt, mode=STLTXCont())
+
+    def get_streams(self, direction=0, **kwargs):
+        return [self.create_stream()]
+
+
+# dynamic load - used for trex console or simulator
+def register():
+    return STLS1()
diff --git a/src/plugins/nat/extras/nat_test_slow_path_with_latency.py b/src/plugins/nat/extras/nat_test_slow_path_with_latency.py
new file mode 100644
index 00000000000..066e491c906
--- /dev/null
+++ b/src/plugins/nat/extras/nat_test_slow_path_with_latency.py
@@ -0,0 +1,74 @@
+from trex_stl_lib.api import *
+
+
+class STLS1(object):
+
+    def create_stream(self, port_id):
+        # base_pkt = Ether()/IP(dst="2.2.0.1")/UDP(dport=12)
+
+        # pad = Padding()
+        # if len(base_pkt) < 64:
+        #     pad_len = 64 - len(base_pkt)
+        #     pad.load = '\x00' * pad_len
+
+        # vm = STLVM()
+
+        # vm.tuple_var(name="tuple", ip_min="10.0.0.3", ip_max="10.0.0.202", port_min=1025, port_max=61124, limit_flows = 100000)
+
+        # vm.write(fv_name="tuple.ip", pkt_offset="IP.src")
+        # vm.fix_chksum()
+
+        # vm.write(fv_name="tuple.port", pkt_offset="UDP.sport")
+
+        # pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        # return STLStream(packet=pkt, mode=STLTXCont())
+
+        vm = STLScVmRaw([STLVmTupleGen(ip_min="10.0.0.1", ip_max="10.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       name="stuple", limit_flows=10000),
+                         STLVmTupleGen(ip_min="2.0.0.1", ip_max="2.255.255.254",
+                                       port_min=1025, port_max=65535,
+                                       name="dtuple", limit_flows=100000000),
+
+                         # write ip to packet IP.src
+                         STLVmWrFlowVar(fv_name="stuple.ip",
+                                        pkt_offset="IP.src"),
+                         STLVmWrFlowVar(fv_name="dtuple.ip",
+                                        pkt_offset="IP.dst"),
+                         # fix checksum
+                         STLVmFixIpv4(offset="IP"),
+                         # write udp.port
+                         STLVmWrFlowVar(fv_name="stuple.port",
+                                        pkt_offset="UDP.sport"),
+                         STLVmWrFlowVar(fv_name="dtuple.port",
+                                        pkt_offset="UDP.dport"),
+                         ]
+                        )
+
+        base_pkt = Ether()/IP(src="16.0.0.1", dst="2.0.0.1")/UDP(dport=12, sport=1025)
+        pad = Padding()
+        if len(base_pkt) < 64:
+            pad_len = 64 - len(base_pkt)
+            pad.load = '\x00' * pad_len
+
+        pad = max(0, 64 - len(base_pkt)) * 'x'
+        pad_latency = max(0, (64-4) - len(base_pkt)) * 'x'
+
+        pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
+
+        return [STLStream(packet=pkt, mode=STLTXCont()),
+                  # latency stream   
+                  STLStream(packet = STLPktBuilder(pkt = base_pkt/pad_latency),
+                            mode = STLTXCont(pps=1000),
+                            flow_stats = STLFlowLatencyStats(pg_id = 12+port_id))
+                  ]
+
+    def get_streams(self, direction=0, **kwargs):
+        # return [self.create_stream()]
+        return self.create_stream(kwargs['port_id'])
+
+
+# dynamic load - used for trex console or simulator
+def register():
+    return STLS1()