add mode example to hw cs profile

1 files changed, 152 insertions, 2 deletions

diff --git a/scripts/stl/syn_attack_fix_cs_hw.py b/scripts/stl/syn_attack_fix_cs_hw.py
index 694a97a3..4738f2b5 100644
--- a/scripts/stl/syn_attack_fix_cs_hw.py
+++ b/scripts/stl/syn_attack_fix_cs_hw.py
@@ -7,7 +7,7 @@ class STLS1(object):
     def __init__ (self):
         self.max_pkt_size_l3  =9*1024;
 
-    def create_stream (self):
+    def create_stream_tcp_syn (self):
 
         # TCP SYN
         base_pkt  = Ether()/IP(dst="48.0.0.1")/TCP(dport=80,flags="S")
@@ -46,10 +46,160 @@ class STLS1(object):
                          mode = STLTXCont())
 
 
+    def create_stream_udp1 (self):
+
+        # UDP packet
+        base_pkt  = Ether()/IP(dst="48.0.0.1")/UDP(dport=80)/(30*'x')
+
+
+        # vm
+        vm = STLScVmRaw( [ STLVmFlowVar(name="ip_src", 
+                                              min_value="16.0.0.0", 
+                                              max_value="18.0.0.254", 
+                                              size=4, op="random"),
+
+                           STLVmFlowVar(name="src_port", 
+                                              min_value=1025, 
+                                              max_value=65000, 
+                                              size=2, op="random"),
+
+                           STLVmWrFlowVar(fv_name="ip_src", pkt_offset= "IP.src" ),
+
+
+                           STLVmWrFlowVar(fv_name="src_port", 
+                                          pkt_offset= "UDP.sport"), # fix udp len  
+
+                           # must be last 
+                           STLVmFixChecksumHw(l3_offset = "IP",
+                                              l4_offset = "UDP",
+                                              l4_type  = CTRexVmInsFixHwCs.L4_TYPE_UDP )# hint, TRex can know that 
+
+                          ]
+                       )
+
+        pkt = STLPktBuilder(pkt = base_pkt,
+                            vm = vm)
+
+        return STLStream(packet = pkt,
+                         random_seed = 0x1234,# can be remove. will give the same random value any run
+                         mode = STLTXCont())
+
+    def create_stream_ipv6_tcp (self):
+
+        # UDP packet
+        base_pkt  = Ether()/IPv6(dst="2001:0:4137:9350:8000:f12a:b9c8:2815",src="2001:4860:0:2001::68")/TCP(dport=80,flags="S")
+
+        # vm
+        vm = STLScVmRaw( [ STLVmFlowVar(name="ip_src", 
+                                              min_value="16.0.0.0", 
+                                              max_value="18.0.0.254", 
+                                              size=4, op="random"),
+
+                           STLVmFlowVar(name="src_port", 
+                                              min_value=1025, 
+                                              max_value=65000, 
+                                              size=2, op="random"),
+
+                           STLVmWrFlowVar(fv_name="ip_src", pkt_offset ="IPv6.src",offset_fixup=12 ),
+
+
+                           STLVmWrFlowVar(fv_name="src_port", 
+                                          pkt_offset= "TCP.sport"), # fix udp len  
+
+                           # must be last 
+                           STLVmFixChecksumHw(l3_offset = "IPv6",
+                                              l4_offset = "TCP",
+                                              l4_type  = CTRexVmInsFixHwCs.L4_TYPE_TCP )# hint, TRex can know that 
+
+                          ]
+                       )
+
+        pkt = STLPktBuilder(pkt = base_pkt,
+                            vm = vm)
+
+        return STLStream(packet = pkt,
+                         random_seed = 0x1234,# can be remove. will give the same random value any run
+                         mode = STLTXCont())
+
+    def create_stream_ipv6_udp (self):
+
+        # UDP packet
+        base_pkt  = Ether()/IPv6(dst="2001:0:4137:9350:8000:f12a:b9c8:2815",src="2001:4860:0:2001::68")/UDP(dport=80)
+
+        # vm
+        vm = STLScVmRaw( [ STLVmFlowVar(name="ip_src", 
+                                              min_value="16.0.0.0", 
+                                              max_value="18.0.0.254", 
+                                              size=4, op="random"),
+
+                           STLVmFlowVar(name="src_port", 
+                                              min_value=1025, 
+                                              max_value=65000, 
+                                              size=2, op="random"),
+
+                           STLVmWrFlowVar(fv_name="ip_src", pkt_offset ="IPv6.src",offset_fixup=12 ),
+
+
+                           STLVmWrFlowVar(fv_name="src_port", 
+                                          pkt_offset= "UDP.sport"), # fix udp len  
+
+                           # must be last 
+                           STLVmFixChecksumHw(l3_offset = "IPv6",
+                                              l4_offset = "UDP",
+                                              l4_type  = CTRexVmInsFixHwCs.L4_TYPE_UDP )# hint, TRex can know that 
+
+                          ]
+                       )
+
+        pkt = STLPktBuilder(pkt = base_pkt,
+                            vm = vm)
+
+        return STLStream(packet = pkt,
+                         random_seed = 0x1234,# can be remove. will give the same random value any run
+                         mode = STLTXCont())
+
+    def create_stream_not_valid (self):
+
+        # UDP packet
+        base_pkt  = Ether()/IPv6(dst="2001:0:4137:9350:8000:f12a:b9c8:2815",src="2001:4860:0:2001::68")/UDP(dport=80)
+
+        # vm
+        vm = STLScVmRaw( [ STLVmFlowVar(name="ip_src", 
+                                              min_value="16.0.0.0", 
+                                              max_value="18.0.0.254", 
+                                              size=4, op="random"),
+
+                           STLVmFlowVar(name="src_port", 
+                                              min_value=1025, 
+                                              max_value=65000, 
+                                              size=2, op="random"),
+
+                           STLVmWrFlowVar(fv_name="ip_src", pkt_offset ="IPv6.src",offset_fixup=12 ),
+
+
+                           STLVmWrFlowVar(fv_name="src_port", 
+                                          pkt_offset= "UDP.sport"), # fix udp len  
+
+                           # must be last 
+                           STLVmFixChecksumHw(l3_offset = "IPv6",
+                                              l4_offset = "IPv6", # not valid 
+                                              l4_type  = CTRexVmInsFixHwCs.L4_TYPE_UDP )# hint, TRex can know that 
+
+                          ]
+                       )
+
+        pkt = STLPktBuilder(pkt = base_pkt,
+                            vm = vm)
+
+        return STLStream(packet = pkt,
+                         random_seed = 0x1234,# can be remove. will give the same random value any run
+                         mode = STLTXCont())
+
 
     def get_streams (self, direction = 0, **kwargs):
         # create 1 stream 
-        return [ self.create_stream() ]
+        #return [ self.create_stream_ipv6_udp(),self.create_stream_tcp_syn(), self.create_stream_udp1(),self.create_stream_ipv6_tcp()]
+        return [ self.create_stream_tcp_syn()]
 
 
 # dynamic load - used for trex console or simulator