-# 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:
+# Copyright (c) 2022 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,
class TrafficProfile(TrafficProfileBaseClass):
"""Traffic profile."""
- def __init__(self):
- """Initialization and setting of profile parameters."""
-
- super(TrafficProfileBaseClass, self).__init__()
-
- # IPs used in packet headers.
- self.p1_src_start_ip = u"172.16.0.0"
- self.p1_src_end_ip = u"172.19.255.255"
- self.p1_dst_start_ip = u"20.16.0.0"
- self.p1_dst_end_ip = u"20.19.255.255"
-
- # UDP messages
- self.udp_req = u"GET"
- self.udp_res = u"ACK"
-
- # Headers length
- self.headers_size = 42 # 14B l2 + 20B ipv4 + 8B udp
-
- # Required UDP keepalive value for T-Rex
- self.udp_keepalive = 2000*1000*100 # 200000s (200,000,000 msec)
-
def define_profile(self):
"""Define profile to be used by advanced stateful traffic generator.
:returns: IP generator and profile templates ASTFProfile().
:rtype: tuple
"""
- self.udp_req += self._gen_padding(self.headers_size + len(self.udp_req))
- self.udp_res += self._gen_padding(self.headers_size + len(self.udp_res))
+ # IPs used in packet headers.
+ p1_src_start_ip = u"172.16.0.0"
+ p1_src_end_ip = u"172.19.255.255"
+ p1_dst_start_ip = u"20.16.0.0"
+ p1_dst_end_ip = u"20.19.255.255"
+
+ # Headers length
+ headers_size = 46 # 18B L2 + 20B IPv4 + 8B UDP.
+
+ # UDP messages, not padded yet.
+ udp_req = u"GET"
+ udp_res = u"ACK"
+
+ # Padd to the required frame size.
+ udp_req += self._gen_padding(headers_size + len(udp_req))
+ udp_res += self._gen_padding(headers_size + len(udp_res))
+
+ # No need to set keepalive, both programs end just after start&send.
# client commands
prog_c = ASTFProgram(stream=False)
- prog_c.set_keepalive_msg(self.udp_keepalive)
# send REQ message
- prog_c.send_msg(self.udp_req)
- # receive RES message
- prog_c.recv_msg(1)
+ prog_c.send_msg(udp_req)
+ # No need to process the response, seeing L2 counter is enough.
+ # Client program can end here.
# server commands
prog_s = ASTFProgram(stream=False)
- prog_c.set_keepalive_msg(self.udp_keepalive)
- # receive REQ message
- prog_s.recv_msg(1)
+ # When server instance is created means REQ is visible in L2 counter.
+ # No need to receive explicitly?
# send RES message
- prog_s.send_msg(self.udp_res)
+ prog_s.send_msg(udp_res)
+ # Server program can end here.
# ip generators
ip_gen_c = ASTFIPGenDist(
- ip_range=[self.p1_src_start_ip, self.p1_src_end_ip],
- distribution=u"seq"
+ ip_range=[p1_src_start_ip, p1_src_end_ip],
+ distribution=u"seq",
)
ip_gen_s = ASTFIPGenDist(
- ip_range=[self.p1_dst_start_ip, self.p1_dst_end_ip],
- distribution=u"seq"
+ ip_range=[p1_dst_start_ip, p1_dst_end_ip],
+ distribution=u"seq",
)
ip_gen = ASTFIPGen(
glob=ASTFIPGenGlobal(ip_offset=u"0.0.0.1"),
dist_client=ip_gen_c,
- dist_server=ip_gen_s
+ dist_server=ip_gen_s,
)
# server association
program=prog_c,
ip_gen=ip_gen,
limit=16515072, # TODO: set via input parameter ?
- port=8080
+ port=8080,
)
temp_s = ASTFTCPServerTemplate(program=prog_s, assoc=s_assoc)
template = ASTFTemplate(client_template=temp_c, server_template=temp_s)
Code Review / csit.git / blobdiff