# Copyright (c) 2021 Cisco and/or its affiliates.## SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later## Licensed under the Apache License 2.0 or# GNU General Public License v2.0 or later; you may not use this file# except in compliance with one of these Licenses. You# may obtain a copy of the Licenses at:## http://www.apache.org/licenses/LICENSE-2.0# https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html## Note: If this file is linked with Scapy, which is GPLv2+, your use of it# must be under GPLv2+. If at any point in the future it is no longer linked# with Scapy (or other GPLv2+ licensed software), you are free to choose# Apache 2.## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License."""Stream profile for T-rex traffic generator.Stream profile: - Two streams sent in directions 0 --> 1 and 1 --> 0 at the same time. - Packet: ETH / IP / - Direction 0 --> 1: - Source IP address range: 20.20.20.2 - Destination IP address range: 2.1.1.2 - 2.1.1.254 - Direction 1 --> 0: - Source IP address range: 10.10.10.2 - Destination IP address range: 1.1.1.2 - 1.1.1.254"""fromtrex.stl.apiimport*fromprofile_trex_stateless_base_classimportTrafficStreamsBaseClassclassTrafficStreams(TrafficStreamsBaseClass):"""Stream profile."""def__init__(self):"""Initialization and setting of streams' parameters."""super(TrafficStreamsBaseClass,self).__init__()# IPs used in packet headers.self.p1_src_start_ip=u"20.20.20.2"self.p1_dst_start_ip=u"2.1.1.2"self.p1_dst_end_ip=u"2.1.1.254"self.p2_src_start_ip=u"10.10.10.2"self.p2_dst_start_ip=u"1.1.1.2"self.p2_dst_end_ip=u"1.1.1.254"defdefine_packets(self):"""Defines the packets to be sent from the traffic generator. Packet definition: | ETH | IP | :returns: Packets to be sent from the traffic generator. :rtype: tuple """# Direction 0 --> 1base_pkt_a=(Ether()/IP(src=self.p1_src_start_ip,dst=self.p1_dst_start_ip,proto=61))# Direction 1 --> 0base_pkt_b=(Ether()/IP(src=self.p2_src_start_ip,dst=self.p2_dst_start_ip,proto=61))# Direction 0 --> 1vm1=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 --> 0vm2=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")])returnbase_pkt_a,base_pkt_b,vm1,vm2defregister():"""Register this traffic profile to T-rex. Do not change this function. :return: Traffic streams. :rtype: Object """returnTrafficStreams()