From 3931530bce5529e141f0fb10d6c72981aa0d97e6 Mon Sep 17 00:00:00 2001 From: Hanoh Haim Date: Thu, 18 Feb 2016 16:10:30 +0200 Subject: add python profile for new write_mask command --- .../unit_tests/functional_tests/stl_basic_tests.py | 11 +++-- .../trex_control_plane/stl/trex_stl_lib/api.py | 1 + .../trex_stl_lib/trex_stl_packet_builder_scapy.py | 47 +++++++++++++++++++++ scripts/exp/udp_1pkt_mac_mask1.pcap | Bin 0 -> 1544 bytes scripts/exp/udp_1pkt_mac_mask2.pcap | Bin 0 -> 1544 bytes scripts/exp/udp_1pkt_mac_mask3.pcap | Bin 0 -> 1544 bytes scripts/stl/udp_1pkt_mac_mask1.py | 35 +++++++++++++++ scripts/stl/udp_1pkt_mac_mask2.py | 35 +++++++++++++++ scripts/stl/udp_1pkt_mac_mask3.py | 35 +++++++++++++++ 9 files changed, 161 insertions(+), 3 deletions(-) create mode 100644 scripts/exp/udp_1pkt_mac_mask1.pcap create mode 100644 scripts/exp/udp_1pkt_mac_mask2.pcap create mode 100644 scripts/exp/udp_1pkt_mac_mask3.pcap create mode 100644 scripts/stl/udp_1pkt_mac_mask1.py create mode 100644 scripts/stl/udp_1pkt_mac_mask2.py create mode 100644 scripts/stl/udp_1pkt_mac_mask3.py (limited to 'scripts') diff --git a/scripts/automation/regression/unit_tests/functional_tests/stl_basic_tests.py b/scripts/automation/regression/unit_tests/functional_tests/stl_basic_tests.py index a2249d5a..9419ab62 100644 --- a/scripts/automation/regression/unit_tests/functional_tests/stl_basic_tests.py +++ b/scripts/automation/regression/unit_tests/functional_tests/stl_basic_tests.py @@ -139,7 +139,7 @@ class CStlBasic_Test(functional_general_test.CGeneralFunctional_Test): def test_stl_profiles (self): - p = [ + p0 = [ ["udp_1pkt_1mac_override.py","-m 1 -l 50",True], ["syn_attack.py","-m 1 -l 50",False], # can't compare random now ["udp_1pkt_1mac.py","-m 1 -l 50",True], @@ -175,12 +175,17 @@ class CStlBasic_Test(functional_general_test.CGeneralFunctional_Test): ["udp_1pkt_simple_mac_dst.py","-m 1 -l 1 ",True], ["udp_1pkt_simple_mac_src.py","-m 1 -l 1 ",True], ["udp_1pkt_simple_mac_dst_src.py","-m 1 -l 1 ",True], - ["burst_3st_loop_x_times.py","-m 1 -l 20 ",True] + ["burst_3st_loop_x_times.py","-m 1 -l 20 ",True], + ["udp_1pkt_mac_step.py","-m 1 -l 20 ",True], + ["udp_1pkt_mac_mask1.py","-m 1 -l 20 ",True] , + ["udp_1pkt_mac_mask2.py","-m 1 -l 20 ",True], + ["udp_1pkt_mac_mask3.py","-m 1 -l 20 ",True] + ]; - p1 = [ ["udp_1pkt_mac_step.py","-m 1 -l 20 ",True] ] + p = [ ["udp_1pkt_mac_mask3.py","-m 1 -l 20 ",True] ] for obj in p: diff --git a/scripts/automation/trex_control_plane/stl/trex_stl_lib/api.py b/scripts/automation/trex_control_plane/stl/trex_stl_lib/api.py index 9ce9f7f1..d19dcaf0 100644 --- a/scripts/automation/trex_control_plane/stl/trex_stl_lib/api.py +++ b/scripts/automation/trex_control_plane/stl/trex_stl_lib/api.py @@ -16,6 +16,7 @@ STLPktBuilder = CScapyTRexPktBuilder # VM STLVmFlowVar = CTRexVmDescFlowVar STLVmWrFlowVar = CTRexVmDescWrFlowVar +STLVmWrMaskFlowVar = CTRexVmDescWrMaskFlowVar STLVmFixIpv4 = CTRexVmDescFixIpv4 STLVmTrimPktSize = CTRexVmDescTrimPktSize STLVmTupleGen = CTRexVmDescTupleGen diff --git a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_packet_builder_scapy.py b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_packet_builder_scapy.py index c788e277..95c32b2f 100644 --- a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_packet_builder_scapy.py +++ b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_packet_builder_scapy.py @@ -234,6 +234,22 @@ class CTRexVmInsWrFlowVar(CTRexVmInsBase): self.is_big_endian = is_big_endian assert type(is_big_endian)==bool, 'type of is_big_endian is not bool' +class CTRexVmInsWrMaskFlowVar(CTRexVmInsBase): + def __init__(self, fv_name, pkt_offset,pkt_cast_size,mask,shift, is_big_endian=True): + super(CTRexVmInsWrMaskFlowVar, self).__init__("write_mask_flow_var") + self.name = fv_name + assert type(fv_name)==str, 'type of fv_name is not str' + self.pkt_offset = pkt_offset + assert type(pkt_offset)==int, 'type of pkt_offset is not int' + self.pkt_cast_size = pkt_cast_size + assert type(pkt_cast_size)==int, 'type of pkt_cast_size is not int' + self.mask = mask + assert type(mask)==int, 'type of mask is not int' + self.shift = shift + assert type(shift)==int, 'type of shift is not int' + self.is_big_endian = is_big_endian + assert type(is_big_endian)==bool, 'type of is_big_endian is not bool' + class CTRexVmInsTrimPktSize(CTRexVmInsBase): def __init__(self,fv_name): super(CTRexVmInsTrimPktSize, self).__init__("trim_pkt_size") @@ -562,6 +578,37 @@ class CTRexVmDescWrFlowVar(CTRexVmDescBase): t=parent._name_to_offset(self.pkt_offset) self.pkt_offset = t[0] +class CTRexVmDescWrMaskFlowVar(CTRexVmDescBase): + def __init__(self, fv_name, pkt_offset, pkt_cast_size=1, mask=0xff, shift=0, offset_fixup=0, is_big=True): + super(CTRexVmDescWrMaskFlowVar, self).__init__() + self.name =fv_name + assert type(fv_name)==str, 'type of fv_name is not str' + self.offset_fixup =offset_fixup + assert type(offset_fixup)==int, 'type of offset_fixup is not int' + self.pkt_offset =pkt_offset + self.pkt_cast_size =pkt_cast_size + assert type(pkt_cast_size)==int,'type of pkt_cast_size is not int' + if not (pkt_cast_size in [1,2,4]): + raise CTRexPacketBuildException(-10,"not valid cast size"); + + self.mask = mask + assert type(mask)==int,'type of mask is not int' + self.shift = shift + assert type(shift)==int,'type of shift is not int' + self.is_big =is_big; + assert type(is_big)==bool,'type of is_big_endian is not bool' + + def get_var_ref (self): + return self.name + + def get_obj (self): + return CTRexVmInsWrMaskFlowVar(self.name,self.pkt_offset+self.offset_fixup,self.pkt_cast_size,self.mask,self.shift,self.is_big) + + def compile(self,parent): + if type(self.pkt_offset)==str: + t=parent._name_to_offset(self.pkt_offset) + self.pkt_offset = t[0] + class CTRexVmDescTrimPktSize(CTRexVmDescBase): def __init__(self,fv_name): diff --git a/scripts/exp/udp_1pkt_mac_mask1.pcap b/scripts/exp/udp_1pkt_mac_mask1.pcap new file mode 100644 index 00000000..65adb5eb Binary files /dev/null and b/scripts/exp/udp_1pkt_mac_mask1.pcap differ diff --git a/scripts/exp/udp_1pkt_mac_mask2.pcap b/scripts/exp/udp_1pkt_mac_mask2.pcap new file mode 100644 index 00000000..07b00113 Binary files /dev/null and b/scripts/exp/udp_1pkt_mac_mask2.pcap differ diff --git a/scripts/exp/udp_1pkt_mac_mask3.pcap b/scripts/exp/udp_1pkt_mac_mask3.pcap new file mode 100644 index 00000000..d168dee0 Binary files /dev/null and b/scripts/exp/udp_1pkt_mac_mask3.pcap differ diff --git a/scripts/stl/udp_1pkt_mac_mask1.py b/scripts/stl/udp_1pkt_mac_mask1.py new file mode 100644 index 00000000..37f5079d --- /dev/null +++ b/scripts/stl/udp_1pkt_mac_mask1.py @@ -0,0 +1,35 @@ +from trex_stl_lib.api import * + + +# step is not 1. +class STLS1(object): + + def __init__ (self): + self.fsize =64; # the size of the packet + + def create_stream (self): + + # create a base packet and pad it to size + size = self.fsize - 4; # no FCS + base_pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025) + pad = max(0, size - len(base_pkt)) * 'x' + + vm = CTRexScRaw( [ STLVmFlowVar(name="mac_src", min_value=1, max_value=30, size=2, op="dec",step=1), + STLVmWrMaskFlowVar(fv_name="mac_src", pkt_offset= 11,pkt_cast_size=1, mask=0xff) # mask command ->write it as one byte + ] + ) + + return STLStream(packet = STLPktBuilder(pkt = base_pkt/pad,vm = vm), + mode = STLTXCont( pps=10 )) + + def get_streams (self, direction = 0): + # create 1 stream + return [ self.create_stream() ] + + +# dynamic load - used for trex console or simulator +def register(): + return STLS1() + + + diff --git a/scripts/stl/udp_1pkt_mac_mask2.py b/scripts/stl/udp_1pkt_mac_mask2.py new file mode 100644 index 00000000..b72f0c3b --- /dev/null +++ b/scripts/stl/udp_1pkt_mac_mask2.py @@ -0,0 +1,35 @@ +from trex_stl_lib.api import * + + +# step is not 1. +class STLS1(object): + + def __init__ (self): + self.fsize =64; # the size of the packet + + def create_stream (self): + + # create a base packet and pad it to size + size = self.fsize - 4; # no FCS + base_pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025) + pad = max(0, size - len(base_pkt)) * 'x' + + vm = CTRexScRaw( [ STLVmFlowVar(name="mac_src", min_value=1, max_value=30, size=2, op="dec",step=1), + STLVmWrMaskFlowVar(fv_name="mac_src", pkt_offset= 10,pkt_cast_size=2, mask=0xff00,shift=8) # take the var shift it 8 (x256) write only to LSB + ] + ) + + return STLStream(packet = STLPktBuilder(pkt = base_pkt/pad,vm = vm), + mode = STLTXCont( pps=10 )) + + def get_streams (self, direction = 0): + # create 1 stream + return [ self.create_stream() ] + + +# dynamic load - used for trex console or simulator +def register(): + return STLS1() + + + diff --git a/scripts/stl/udp_1pkt_mac_mask3.py b/scripts/stl/udp_1pkt_mac_mask3.py new file mode 100644 index 00000000..ef5bc4f0 --- /dev/null +++ b/scripts/stl/udp_1pkt_mac_mask3.py @@ -0,0 +1,35 @@ +from trex_stl_lib.api import * + + +# step is not 1. +class STLS1(object): + + def __init__ (self): + self.fsize =64; # the size of the packet + + def create_stream (self): + + # create a base packet and pad it to size + size = self.fsize - 4; # no FCS + base_pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025) + pad = max(0, size - len(base_pkt)) * 'x' + + vm = CTRexScRaw( [ STLVmFlowVar(name="mac_src", min_value=1, max_value=30, size=2, op="dec",step=1), + STLVmWrMaskFlowVar(fv_name="mac_src", pkt_offset= 10,pkt_cast_size=1, mask=0x1,shift=-1) # take var mac_src>>1 and write the LSB every two packet there should be a change + ] + ) + + return STLStream(packet = STLPktBuilder(pkt = base_pkt/pad,vm = vm), + mode = STLTXCont( pps=10 )) + + def get_streams (self, direction = 0): + # create 1 stream + return [ self.create_stream() ] + + +# dynamic load - used for trex console or simulator +def register(): + return STLS1() + + + -- cgit 1.2.3-korg