Fix T-rex profile base class bug

[csit.git] / resources / tools / t-rex / stream_profiles / profile_trex_stateless_base_class.py

diff --git a/resources/tools/t-rex/stream_profiles/profile_trex_stateless_base_class.py b/resources/tools/t-rex/stream_profiles/profile_trex_stateless_base_class.py
index 2d07e7f..a6cb974 100755 (executable)
--- a/resources/tools/t-rex/stream_profiles/profile_trex_stateless_base_class.py
+++ b/resources/tools/t-rex/stream_profiles/profile_trex_stateless_base_class.py
@@ -134,7 +134,7 @@ class TrafficStreamsBaseClass(object):
                 vm=vm1)
             # Direction 1 --> 0
             pkt_b = STLPktBuilder(
-                pkt=base_pkt_a / self._gen_payload(payload_len),
+                pkt=base_pkt_b / self._gen_payload(payload_len),
                 vm=vm2)
 
             # Packets for latency measurement:
@@ -143,7 +143,7 @@ class TrafficStreamsBaseClass(object):
                 pkt=base_pkt_a / self._gen_payload(payload_len))
             # Direction 1 --> 0
             pkt_lat_b = STLPktBuilder(
-                pkt=base_pkt_a / self._gen_payload(payload_len))
+                pkt=base_pkt_b / self._gen_payload(payload_len))
 
             # Create the streams:
             # Direction 0 --> 1
@@ -173,13 +173,14 @@ class TrafficStreamsBaseClass(object):
             stream2 = []
 
             for stream in self.STREAM_TABLE[self.framesize]:
-                payload_len = max(0, stream['size'] - len(base_pkt_a) - 4)
+                payload_len_a = max(0, stream['size'] - len(base_pkt_a) - 4)
+                payload_len_b = max(0, stream['size'] - len(base_pkt_b) - 4)
                 # Create a base packet and pad it to size
                 pkt_a = STLPktBuilder(
-                    pkt=base_pkt_a / self._gen_payload(payload_len),
+                    pkt=base_pkt_a / self._gen_payload(payload_len_a),
                     vm=vm1)
                 pkt_b = STLPktBuilder(
-                    pkt=base_pkt_b / self._gen_payload(payload_len),
+                    pkt=base_pkt_b / self._gen_payload(payload_len_b),
                     vm=vm2)
 
                 # Create the streams: