-# Copyright (c) 2017 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:
- Destination IP address range: 10.0.0.0 - 10.1.134.159
"""
-from trex_stl_lib.api import *
+from trex.stl.api import *
from profile_trex_stateless_base_class import TrafficStreamsBaseClass
super(TrafficStreamsBaseClass, self).__init__()
# IPs used in packet headers.
- self.p1_src_start_ip = '10.0.0.1'
- self.p1_dst_start_ip = '20.0.0.0'
- self.p1_dst_end_ip = '20.1.134.159'
+ self.p1_src_start_ip = u"10.0.0.1"
+ self.p1_dst_start_ip = u"20.0.0.0"
+ self.p1_dst_end_ip = u"20.1.134.159"
- self.p2_src_start_ip = '20.0.0.1'
- self.p2_dst_start_ip = '10.0.0.0'
- self.p2_dst_end_ip = '10.1.134.159'
+ self.p2_src_start_ip = u"20.0.0.1"
+ self.p2_dst_start_ip = u"10.0.0.0"
+ self.p2_dst_end_ip = u"10.1.134.159"
def define_packets(self):
"""Defines the packets to be sent from the traffic generator.
"""
# 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="dst",
- min_value=self.p1_dst_start_ip,
- max_value=self.p1_dst_end_ip,
- size=4, op="inc"),
- STLVmWrFlowVar(fv_name="dst", pkt_offset="IP.dst"),
- STLVmFixIpv4(offset="IP")])
+ vm1 = STLScVmRaw(
+ [
+ STLVmFlowVar(
+ name=u"dst",
+ min_value=self.p1_dst_start_ip,
+ max_value=self.p1_dst_end_ip,
+ size=4,
+ op=u"inc"
+ ),
+ STLVmWrFlowVar(
+ fv_name=u"dst",
+ pkt_offset=u"IP.dst"
+ ),
+ STLVmFixIpv4(
+ offset=u"IP"
+ )
+ ]
+ )
# Direction 1 --> 0
- vm2 = STLScVmRaw([STLVmFlowVar(name="dst",
- min_value=self.p2_dst_start_ip,
- max_value=self.p2_dst_end_ip,
- size=4, op="inc"),
- STLVmWrFlowVar(fv_name="dst", pkt_offset="IP.dst"),
- STLVmFixIpv4(offset="IP")])
+ vm2 = STLScVmRaw(
+ [
+ STLVmFlowVar(
+ name=u"dst",
+ min_value=self.p2_dst_start_ip,
+ max_value=self.p2_dst_end_ip,
+ size=4,
+ op=u"inc"
+ ),
+ STLVmWrFlowVar(
+ fv_name=u"dst",
+ pkt_offset=u"IP.dst"
+ ),
+ STLVmFixIpv4(
+ offset=u"IP"
+ )
+ ]
+ )
return base_pkt_a, base_pkt_b, vm1, vm2