Added to flow stat profile qinq example

Signed-off-by: Ido Barnea <ibarnea@cisco.com>

1 files changed, 30 insertions, 9 deletions

diff --git a/scripts/stl/flow_stats.py b/scripts/stl/flow_stats.py
index a50ba848..65956ff5 100644
--- a/scripts/stl/flow_stats.py
+++ b/scripts/stl/flow_stats.py
@@ -3,26 +3,47 @@ from trex_stl_lib.api import *
 class STLS1(object):
     """
     Create flow stat stream of UDP packet.
-    Can specify using tunables the packet length (fsize) and packet group id (pg_id)
+    Can specify using tunables following params:
+      Packet length (fsize)
+      Packet group id (pg_id)
+      Packet type(pkt_type)
+      Number of streams (num_streams)
     """
     def __init__ (self):
         self.fsize = 64
         self.pg_id = 0
+        self.pkt_type = "ether"
+        self.num_streams = 1
 
     def _create_stream (self):
         size = self.fsize - 4; # HW will add 4 bytes ethernet CRC
-        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'
-        pkt = STLPktBuilder(pkt = base_pkt/pad)
 
-        return [STLStream(packet = pkt,
-                          mode = STLTXCont(pps=1),
-                          flow_stats = STLFlowStats(pg_id = self.pg_id))
-               ]
+        pkt_types = {"ether" :
+                      Ether() / IP(src = "16.0.0.1", dst = "48.0.0.1") / UDP(dport = 12, sport = 1025)
+                      , "vlan":
+                      Ether() / Dot1Q(vlan=11) / IP(src = "16.0.0.1", dst = "48.0.0.1") / UDP(dport = 12, sport = 1025)
+                      , "qinq":
+                      Ether(type=0x88A8) / Dot1Q(vlan=19) / Dot1Q(vlan=11) / IP(src = "16.0.0.1", dst = "48.0.0.1") / UDP(dport = 12, sport = 1025)
+        }
 
-    def get_streams (self, fsize = 64, pg_id = 7, **kwargs):
+        if self.pkt_type not in pkt_types.keys():
+            print ("Wrong pkt_type given. Allowed values are " + format(pkt_types.keys()))
+            return []
+
+        pad = max(0, size - len(self.pkt_type)) * 'x'
+        pkt = STLPktBuilder(pkt = pkt_types[self.pkt_type]/pad)
+
+        streams = []
+        for pg_id_add in range(0, self.num_streams):
+            streams.append(STLStream(packet = pkt, mode = STLTXCont(pps=1), flow_stats = STLFlowStats(pg_id = self.pg_id + pg_id_add)))
+
+        return streams
+
+    def get_streams (self, fsize = 64, pg_id = 1, pkt_type = "ether", num_streams = 1, **kwargs):
         self.fsize = fsize
         self.pg_id = pg_id
+        self.pkt_type = pkt_type
+        self.num_streams = num_streams
         return self._create_stream()
 
 # dynamic load - used for trex console or simulator