diff options
Diffstat (limited to 'resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py')
| -rwxr-xr-x | resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py | 84 |
1 files changed, 59 insertions, 25 deletions
diff --git a/resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py b/resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py index 1a0b0475e5..fc6167d126 100755 --- a/resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py +++ b/resources/traffic_profiles/trex/trex-sl-2n-ethip4-ip4src254.py @@ -1,4 +1,4 @@ -# Copyright (c) 2019 Cisco and/or its affiliates. +# Copyright (c) 2020 Cisco and/or its affiliates. # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at: @@ -37,13 +37,13 @@ class TrafficStreams(TrafficStreamsBaseClass): super(TrafficStreamsBaseClass, self).__init__() # IPs used in packet headers. - self.p1_src_start_ip = '10.10.10.1' - self.p1_src_end_ip = '10.10.10.254' - self.p1_dst_start_ip = '20.20.20.1' + self.p1_src_start_ip = u"10.10.10.1" + self.p1_src_end_ip = u"10.10.10.254" + self.p1_dst_start_ip = u"20.20.20.1" - self.p2_src_start_ip = '20.20.20.1' - self.p2_src_end_ip = '20.20.20.254' - self.p2_dst_start_ip = '10.10.10.1' + self.p2_src_start_ip = u"20.20.20.1" + self.p2_src_end_ip = u"20.20.20.254" + self.p2_dst_start_ip = u"10.10.10.1" def define_packets(self): """Defines the packets to be sent from the traffic generator. @@ -55,28 +55,62 @@ class TrafficStreams(TrafficStreamsBaseClass): """ # Direction 0 --> 1 - base_pkt_a = Ether() / IP(src=self.p1_src_start_ip, - dst=self.p1_dst_start_ip, - proto=61) + base_pkt_a = ( + Ether() / + IP( + src=self.p1_src_start_ip, + dst=self.p1_dst_start_ip, + proto=61 + ) + ) # Direction 1 --> 0 - base_pkt_b = Ether() / IP(src=self.p2_src_start_ip, - dst=self.p2_dst_start_ip, - proto=61) + base_pkt_b = ( + Ether() / + IP( + src=self.p2_src_start_ip, + dst=self.p2_dst_start_ip, + proto=61 + ) + ) # Direction 0 --> 1 - vm1 = STLScVmRaw([STLVmFlowVar(name="src", - min_value=self.p1_src_start_ip, - max_value=self.p1_src_end_ip, - size=4, op="inc"), - STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"), - STLVmFixIpv4(offset="IP")]) + vm1 = STLScVmRaw( + [ + STLVmFlowVar( + name=u"src", + min_value=self.p1_src_start_ip, + max_value=self.p1_src_end_ip, + size=4, + op=u"inc" + ), + STLVmWrFlowVar( + fv_name=u"src", + pkt_offset=u"IP.src" + ), + STLVmFixIpv4( + offset=u"IP" + ) + ] + ) # Direction 1 --> 0 - vm2 = STLScVmRaw([STLVmFlowVar(name="src", - min_value=self.p2_src_start_ip, - max_value=self.p2_src_end_ip, - size=4, op="inc"), - STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"), - STLVmFixIpv4(offset="IP")]) + vm2 = STLScVmRaw( + [ + STLVmFlowVar( + name=u"src", + min_value=self.p2_src_start_ip, + max_value=self.p2_src_end_ip, + size=4, + op=u"inc" + ), + STLVmWrFlowVar( + fv_name=u"src", + pkt_offset=u"IP.src" + ), + STLVmFixIpv4( + offset=u"IP" + ) + ] + ) return base_pkt_a, base_pkt_b, vm1, vm2 |