From 212bcfefccae082c64b84eddd3c7aa368a5b0f20 Mon Sep 17 00:00:00 2001 From: Klement Sekera Date: Tue, 16 Jun 2020 13:28:09 +0000 Subject: nat: add more trex test scripts Type: improvement Signed-off-by: Klement Sekera Change-Id: Ia824d663c86d6a7c9bafe0206e9a93f93884739d --- src/plugins/nat/extras/nat_test_fast_path.py | 67 ++++++++++++++++++++ src/plugins/nat/extras/nat_test_slow_path.py | 65 +++++++++++++++++++ .../nat/extras/nat_test_slow_path_with_latency.py | 74 ++++++++++++++++++++++ 3 files changed, 206 insertions(+) create mode 100644 src/plugins/nat/extras/nat_test_fast_path.py create mode 100644 src/plugins/nat/extras/nat_test_slow_path.py create mode 100644 src/plugins/nat/extras/nat_test_slow_path_with_latency.py 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() -- cgit 1.2.3-korg