RX features - added timestamp to RX packets

Signed-off-by: imarom <imarom@cisco.com>

1 files changed, 23 insertions, 16 deletions

diff --git a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_port.py b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_port.py
index 2e77b492..aef325ef 100644
--- a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_port.py
+++ b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_port.py
@@ -80,7 +80,7 @@ class Port(object):
 
         self.owner = ''
         self.last_factor_type = None
-
+        
         self.attr_lock = threading.Lock()
         
     # decorator to verify port is up
@@ -430,8 +430,8 @@ class Port(object):
 
         # save this for TUI
         self.last_factor_type = mul['type']
-
-        return self.ok()
+        
+        return rc
 
 
     # stop traffic
@@ -451,8 +451,9 @@ class Port(object):
             return self.err(rc.err())
 
         self.state = self.STATE_STREAMS
+        
         self.last_factor_type = None
-
+        
         # timestamp for last tx
         self.tx_stopped_ts = datetime.now()
         
@@ -580,7 +581,8 @@ class Port(object):
         
         # decode the packets
         for i in range(len(pkts)):
-            pkts[i] = base64.b64decode(pkts[i])
+            pkts[i]['binary'] = base64.b64decode(pkts[i]['binary'])
+            
             
         return pkts
         
@@ -1019,6 +1021,7 @@ class Port(object):
         # until thread is locked - order is important
         self.tx_stopped_ts = datetime.now()
         self.state = self.STATE_STREAMS
+        
         self.last_factor_type = None
 
     def async_event_port_attr_changed (self, new_attr):
@@ -1088,7 +1091,7 @@ class Resolver(object):
         raise NotImplementedError()
         
     # return None for more packets otherwise RC object
-    def on_pkt_rx (self, pkt, dt):
+    def on_pkt_rx (self, pkt):
         raise NotImplementedError()
     
     # return value in case of timeout
@@ -1156,12 +1159,13 @@ class Resolver(object):
         if not rc:
             return rc
 
+        # save the start timestamp
+        self.start_ts = rc.data()['ts']
+        
         # block until traffic finishes
         while self.port.is_active():
             time.sleep(0.01)
 
-        self.tx_time = time.time()
-        
         return self.wait_for_rx_response()
         
              
@@ -1173,10 +1177,10 @@ class Resolver(object):
         while polling > 0:
             # fetch the queue
             rx_pkts = self.port.get_rx_queue_pkts()
-            dt = time.time() - self.tx_time
+            
             # for each packet - examine it
             for pkt in rx_pkts:
-                rc = self.on_pkt_rx(pkt, dt)
+                rc = self.on_pkt_rx(pkt)
                 if rc is not None:
                     return rc
                 
@@ -1222,8 +1226,8 @@ class ARPResolver(Resolver):
 
 
     # return None in case more packets are needed else the status rc
-    def on_pkt_rx (self, pkt, dt):
-        scapy_pkt = Ether(pkt)
+    def on_pkt_rx (self, pkt):
+        scapy_pkt = Ether(pkt['binary'])
         if not 'ARP' in scapy_pkt:
             return None
 
@@ -1286,8 +1290,8 @@ class PingResolver(Resolver):
         return [s1]
         
     # return None for more packets otherwise RC object
-    def on_pkt_rx (self, pkt, dt):
-        scapy_pkt = Ether(pkt)
+    def on_pkt_rx (self, pkt):
+        scapy_pkt = Ether(pkt['binary'])
         if not 'ICMP' in scapy_pkt:
             return None
         
@@ -1295,13 +1299,16 @@ class PingResolver(Resolver):
         ip = scapy_pkt['IP']
         
         icmp = scapy_pkt['ICMP']
+        
+        dt = pkt['ts'] - self.start_ts
+        
         if icmp.type == 0:
             # echo reply
-            return self.port.ok('Reply from {0}: bytes={1}, time={2:.2f}ms, TTL={3}'.format(ip.src, len(pkt), dt * 1000, ip.ttl))
+            return self.port.ok('Reply from {0}: bytes={1}, time={2:.2f}ms, TTL={3}'.format(ip.src, len(pkt['binary']), dt * 1000, ip.ttl))
             
         # unreachable
         elif icmp.type == 3:
-            return self.port.ok('destination {0} is unreachable'.format(icmp.dst))
+            return self.port.ok('Reply from {0}: Destination host unreachable'.format(icmp.src))
         else:
             scapy_pkt.show2()
             return self.port.err('unknown ICMP reply')