Merge trex-204

1 files changed, 14 insertions, 11 deletions

diff --git a/scripts/stl/udp_1pkt_simple.py b/scripts/stl/udp_1pkt_simple.py
index 0407f9c9..b84626a1 100644
--- a/scripts/stl/udp_1pkt_simple.py
+++ b/scripts/stl/udp_1pkt_simple.py
@@ -2,18 +2,21 @@ from trex_stl_lib.api import *
 
 class STLS1(object):
 
-    def create_stream (self):
-        return STLStream( 
-            packet = 
-                    STLPktBuilder(
-                        pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/
-                                UDP(dport=12,sport=1025)/(10*'x')
-                    ),
-             mode = STLTXCont())
-
-    def get_streams (self, direction = 0, **kwargs):
+    def create_stream (self, packet_len, packet_count):
+        base_pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)
+        base_pkt_len = len(base_pkt)
+        base_pkt /= 'x' * max(0, packet_len - base_pkt_len)
+        packets = []
+        for i in range(packet_count):
+            packets.append(STLStream(
+                packet = STLPktBuilder(pkt = base_pkt),
+                mode = STLTXCont()
+                ))
+        return packets
+
+    def get_streams (self, direction = 0, packet_len = 64, packet_count = 1, **kwargs):
         # create 1 stream 
-        return [ self.create_stream() ]
+        return self.create_stream(packet_len - 4, packet_count)
 
 
 # dynamic load - used for trex console or simulator