X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Ftools%2Ft-rex%2Ft-rex-stateless.py;h=46c67cc82909765171708806044e90f57e8a354e;hp=eacf4b8d1875d6a4ca1efeb8fe8a8a4a91b21bf5;hb=d637af34421fffe61417bb16e45a7105aae118b9;hpb=bf06c80efd215b28b2e23cb9db7c464e9e29c321

diff --git a/resources/tools/t-rex/t-rex-stateless.py b/resources/tools/t-rex/t-rex-stateless.py
index eacf4b8d18..46c67cc829 100755
--- a/resources/tools/t-rex/t-rex-stateless.py
+++ b/resources/tools/t-rex/t-rex-stateless.py
@@ -79,8 +79,8 @@ def create_packets(traffic_options, frame_size=64):
     p2_dst_start_ip = traffic_options['p2_dst_start_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)
-    base_pkt_b = Ether()/IP(src=p2_src_start_ip, dst=p2_dst_start_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)
 
     vm1 = CTRexScRaw([STLVmTupleGen(ip_min=p1_src_start_ip, ip_max=p1_src_end_ip,
                                     name="tuple"), # define tuple gen
@@ -89,7 +89,7 @@ def create_packets(traffic_options, frame_size=64):
                       STLVmFixIpv4(offset="IP"),                               # fix checksum
                      ]
                      , split_by_field="tuple") # split to cores base on the tuple generator
- 
+
     vm2 = CTRexScRaw([STLVmTupleGen(ip_min=p2_src_start_ip, ip_max=p2_src_end_ip,
                                     name="tuple"), # define tuple gen
 
@@ -158,7 +158,7 @@ def simple_burst(pkt_a, pkt_b, duration=10, rate="1mpps",
         c.start(ports=[0, 1], mult=rate, duration=duration)
 
         # block until done
-        c.wait_on_traffic(ports=[0, 1])
+        c.wait_on_traffic(ports=[0, 1], timeout=(duration+30))
 
         # read the stats after the test
         stats = c.get_stats()