# Copyright (c) 2024 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: - Three parallel bi-directional streams sent as W --> E and E --> W at the same time. - Packet: ETH / IP / """ from trex.stl.api import * from profile_trex_stateless_scale_class import TrafficStreamsScaleClass class TrafficStreams(TrafficStreamsScaleClass): """Stream profile.""" def __init__(self): """Initialization and setting of streams' parameters.""" super(TrafficStreamsScaleClass, self).__init__() self.pkt_data = [ # Direction W --> E: { "src_start_ip": "10.0.0.1", "dst_start_ip": "20.0.0.0", "dst_end_ip": "20.0.39.15", }, # Direction E --> W: { "src_start_ip": "20.0.0.1", "dst_start_ip": "10.0.0.0", "dst_end_ip": "10.0.39.15", }, ] self.pkt_base = [] self.pkt_vm = [] def define_packets(self): """Defines the packets to be sent from the traffic generator. Packet definition: | ETH | IP | :returns: Base packets to be sent and transformation function. :rtype: tuple """ for i in range(len(self.pkt_data)): self.pkt_base.append( Ether() / IP( src=self.pkt_data[i]["src_start_ip"], dst=self.pkt_data[i]["dst_start_ip"], proto=61, ) ) self.pkt_vm.append( STLScVmRaw( [ STLVmFlowVar( name="dst", min_value=self.pkt_data[i]["dst_start_ip"], max_value=self.pkt_data[i]["dst_end_ip"], size=4, op="inc", ), STLVmWrFlowVar( fv_name="dst", pkt_offset="IP.dst", ), STLVmFixIpv4( offset="IP", ), ] ) ) return self.pkt_base, self.pkt_vm def register(): """Register this traffic profile to T-Rex. Do not change this function. :return: Traffic streams. :rtype: Object """ return TrafficStreams()