"""This script uses T-REX stateless API to drive t-rex instance.
Requirements:
-- T-REX: https://github.com/cisco-system-traffic-generator/trex-core
+- T-REX: https://github.com/cisco-system-traffic-generator/trex-core
- compiled and running T-REX process (eg. ./t-rex-64 -i -c 4)
- trex_stl_lib.api library
- Script must be executed on a node with T-REX instance
- port_limit : 2 # numbers of ports to use
version : 2
interfaces : ["84:00.0","84:00.1"] # PCI address of interfaces
- port_info : # set eth mac addr
+ port_info : # set eth mac addr
- dest_mac : [0x90,0xe2,0xba,0x1f,0x97,0xd5] # port 0
src_mac : [0x90,0xe2,0xba,0x1f,0x97,0xd4]
- dest_mac : [0x90,0xe2,0xba,0x1f,0x97,0xd4] # port 1
"""
-
+import json
+import socket
+import string
+import struct
import sys, getopt
-sys.path.insert(0, "/opt/trex-core-1.91/scripts/automation/trex_control_plane/stl/")
+sys.path.insert(0, "/opt/trex-core-2.00/scripts/automation/"+\
+ "trex_control_plane/stl/")
from trex_stl_lib.api import *
-import dpkt
-import json
-import string
def generate_payload(length):
+ """Generate payload.
+
+ :param length: Length of payload.
+ :type length: int
+ :return: Payload filled with chars.
+ :rtype string
+ """
+
word = ''
alphabet_size = len(string.letters)
for i in range(length):
word += string.letters[(i % alphabet_size)]
+
return word
+
+def get_start_end_ipv6(start_ip, end_ip):
+ """Get start and end host from IPv6 as integer.
+
+ :param start_ip: Start IPv6.
+ :param end_ip: End IPv6.
+ :type start_ip: string
+ :type end_ip: string
+ :return: Start host, end host.
+ :rtype int
+ """
+
+ try:
+ ip1 = socket.inet_pton(socket.AF_INET6, start_ip)
+ ip2 = socket.inet_pton(socket.AF_INET6, end_ip)
+
+ hi1, lo1 = struct.unpack('!QQ', ip1)
+ hi2, lo2 = struct.unpack('!QQ', ip2)
+
+ if ((hi1 << 64) | lo1) > ((hi2 << 64) | lo2):
+ print "IPv6: start_ip is greater then end_ip"
+ sys.exit(2)
+
+ max_p1 = abs(int(lo1) - int(lo2)) + 1
+ base_p1 = lo1
+ except AddressValueError as ex_error:
+ print ex_error
+ sys.exit(2)
+
+ return base_p1, max_p1
+
+
def create_packets(traffic_options, frame_size=64):
+ """Create two IP packets to be used in stream.
+
+ :param traffic_options: Parameters for packets.
+ :param frame_size: Size of L2 frame.
+ :type traffic_options: list
+ :type frame_size: int
+ :return: Packet instances.
+ :rtype STLPktBuilder
+ """
if frame_size < 64:
- print "Packet min. size is 64B"
+ print_error("Packet min. size is 64B")
+ sys.exit(1)
+
+ fsize_no_fcs = frame_size - 4 # no FCS
+
+ p1_src_start_ip = traffic_options['p1_src_start_ip']
+ p1_src_end_ip = traffic_options['p1_src_end_ip']
+ p1_dst_start_ip = traffic_options['p1_dst_start_ip']
+ p2_src_start_ip = traffic_options['p2_src_start_ip']
+ p2_src_end_ip = traffic_options['p2_src_end_ip']
+ p2_dst_start_ip = traffic_options['p2_dst_start_ip']
+
+ base_pkt_a = Ether()/IP(src=p1_src_start_ip, dst=p1_dst_start_ip, proto=61)
+ base_pkt_b = Ether()/IP(src=p2_src_start_ip, dst=p2_dst_start_ip, proto=61)
+
+ # The following code applies raw instructions to packet (IP src increment).
+ # It splits the generated traffic by "ip_src" variable to cores and fix
+ # IPv4 header checksum.
+ vm1 = STLScVmRaw([STLVmFlowVar(name="src",
+ min_value=p1_src_start_ip,
+ max_value=p1_src_end_ip,
+ size=4, op="inc"),
+ STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"),
+ STLVmFixIpv4(offset="IP"),
+ ], split_by_field="src")
+ # The following code applies raw instructions to packet (IP src increment).
+ # It splits the generated traffic by "ip_src" variable to cores and fix
+ # IPv4 header checksum.
+ vm2 = STLScVmRaw([STLVmFlowVar(name="src",
+ min_value=p2_src_start_ip,
+ max_value=p2_src_end_ip,
+ size=4, op="inc"),
+ STLVmWrFlowVar(fv_name="src", pkt_offset="IP.src"),
+ STLVmFixIpv4(offset="IP"),
+ ], split_by_field="src")
+
+ pkt_a = STLPktBuilder(pkt=base_pkt_a/generate_payload(
+ fsize_no_fcs-len(base_pkt_a)), vm=vm1)
+ pkt_b = STLPktBuilder(pkt=base_pkt_b/generate_payload(
+ fsize_no_fcs-len(base_pkt_b)), vm=vm2)
+
+ return(pkt_a, pkt_b)
+
+
+def create_packets_v6(traffic_options, frame_size=78):
+ """Create two IPv6 packets to be used in stream.
+
+ :param traffic_options: Parameters for packets.
+ :param frame_size: Size of L2 frame.
+ :type traffic_options: List
+ :type frame_size: int
+ :return: Packet instances.
+ :rtype STLPktBuilder
+ """
+
+ if frame_size < 78:
+ print "Packet min. size is 78B"
sys.exit(2)
fsize_no_fcs = frame_size - 4 # no FCS
- #p1_src_mac = traffic_options['p1_src_mac']
- #p1_dst_mac = traffic_options['p1_dst_mac']
p1_src_start_ip = traffic_options['p1_src_start_ip']
p1_src_end_ip = traffic_options['p1_src_end_ip']
p1_dst_start_ip = traffic_options['p1_dst_start_ip']
- #p1_dst_end_ip = traffic_options['p1_dst_end_ip']
- #p2_src_mac = traffic_options['p2_src_mac']
- #p2_dst_mac = traffic_options['p2_dst_mac']
p2_src_start_ip = traffic_options['p2_src_start_ip']
p2_src_end_ip = traffic_options['p2_src_end_ip']
p2_dst_start_ip = traffic_options['p2_dst_start_ip']
- #p2_dst_end_ip = traffic_options['p2_dst_end_ip']
- base_pkt_a = Ether()/IP(src=p1_src_start_ip, dst=p1_dst_start_ip)
- base_pkt_b = Ether()/IP(src=p2_src_start_ip, dst=p2_dst_start_ip)
+ base_p1, max_p1 = get_start_end_ipv6(p1_src_start_ip, p1_src_end_ip)
+ base_p2, max_p2 = get_start_end_ipv6(p2_src_start_ip, p2_src_end_ip)
- vm1 = CTRexScRaw([STLVmTupleGen(ip_min=p1_src_start_ip, ip_max=p1_src_end_ip,
- name="tuple"), # define tuple gen
+ base_pkt_a = Ether()/IPv6(src=p1_src_start_ip, dst=p1_dst_start_ip)
+ base_pkt_b = Ether()/IPv6(src=p2_src_start_ip, dst=p2_dst_start_ip)
- STLVmWrFlowVar(fv_name="tuple.ip", pkt_offset="IP.src"), # write ip to packet IP.src
- STLVmFixIpv4(offset="IP"), # fix checksum
+ # The following code applies raw instructions to packet (IP src increment).
+ # It splits the generated traffic by "ip_src" variable to cores
+ vm1 = STLScVmRaw([STLVmFlowVar(name="ipv6_src",
+ min_value=base_p1,
+ max_value=max_p1+base_p1,
+ size=8, op="inc"),
+ STLVmWrFlowVar(fv_name="ipv6_src", pkt_offset="IPv6.src",
+ offset_fixup=8)
]
- , split_by_field="tuple") # split to cores base on the tuple generator
-
- vm2 = CTRexScRaw([STLVmTupleGen(ip_min=p2_src_start_ip, ip_max=p2_src_end_ip,
- name="tuple"), # define tuple gen
-
- STLVmWrFlowVar(fv_name="tuple.ip", pkt_offset="IP.src"), # write ip to packet IP.src
- STLVmFixIpv4(offset="IP"), # fix checksum
+ , split_by_field="ipv6_src")
+
+ # The following code applies raw instructions to packet (IP src increment).
+ # It splits the generated traffic by "ip_src" variable to cores
+ vm2 = STLScVmRaw([STLVmFlowVar(name="ipv6_src",
+ min_value=base_p2,
+ max_value=max_p2+base_p2,
+ size=8, op="inc"),
+ STLVmWrFlowVar(fv_name="ipv6_src", pkt_offset="IPv6.src",
+ offset_fixup=8)
]
- , split_by_field="tuple") # split to cores base on the tuple generator
+ , split_by_field="ipv6_src")
- pkt_a = STLPktBuilder(pkt=base_pkt_a/generate_payload(fsize_no_fcs-len(base_pkt_a)), vm=vm1)
- pkt_b = STLPktBuilder(pkt=base_pkt_b/generate_payload(fsize_no_fcs-len(base_pkt_b)), vm=vm2)
+ pkt_a = STLPktBuilder(pkt=base_pkt_a/generate_payload(
+ max(0, fsize_no_fcs-len(base_pkt_a))), vm=vm1)
+ pkt_b = STLPktBuilder(pkt=base_pkt_b/generate_payload(
+ max(0, fsize_no_fcs-len(base_pkt_b))), vm=vm2)
return(pkt_a, pkt_b)
-def simple_burst(pkt_a, pkt_b, duration=10, rate="1mpps",
- warmup=True, warmup_time=5):
+
+def simple_burst(pkt_a, pkt_b, duration=10, rate="1mpps", warmup_time=5):
+ """Run the traffic with specific parameters.
+
+ :param pkt_a: Base packet for stream 1.
+ :param pkt_b: Base packet for stream 2.
+ :param duration: Duration of traffic run in seconds.
+ :param rate: Rate of traffic run [percentage, pps, bps].
+ :param warmup_time: Warm up duration.
+ :type pkt_a: STLPktBuilder
+ :type pkt_b: STLPktBuilder
+ :type duration: int
+ :type rate: string
+ :type warmup_time: int
+ :return: nothing
+ """
# create client
- c = STLClient()
- passed = True
+ client = STLClient()
+
+ total_rcvd = 0
+ total_sent = 0
+ lost_a = 0
+ lost_b = 0
try:
- # turn this on for some information
- #c.set_verbose("high")
+ # turn this off if too many logs
+ #client.set_verbose("high")
# create two streams
- s1 = STLStream(packet=pkt_a,
- mode=STLTXCont(pps=100))
+ stream1 = STLStream(packet=pkt_a,
+ mode=STLTXCont(pps=100))
# second stream with a phase of 10ns (inter stream gap)
- s2 = STLStream(packet=pkt_b,
- isg=10.0,
- mode=STLTXCont(pps=100))
-
+ stream2 = STLStream(packet=pkt_b,
+ isg=10.0,
+ mode=STLTXCont(pps=100))
# connect to server
- c.connect()
+ client.connect()
# prepare our ports (my machine has 0 <--> 1 with static route)
- c.reset(ports=[0, 1])
+ client.reset(ports=[0, 1])
# add both streams to ports
- c.add_streams(s1, ports=[0])
- c.add_streams(s2, ports=[1])
+ client.add_streams(stream1, ports=[0])
+ client.add_streams(stream2, ports=[1])
#warmup phase
- if warmup == True:
- c.clear_stats()
- c.start(ports=[0, 1], mult=rate, duration=warmup_time)
- c.wait_on_traffic(ports=[0, 1])
- stats = c.get_stats()
+ if warmup_time is not None:
+ client.clear_stats()
+ client.start(ports=[0, 1], mult=rate, duration=warmup_time)
+ client.wait_on_traffic(ports=[0, 1], timeout=(warmup_time+30))
+ stats = client.get_stats()
print stats
print "#####warmup statistics#####"
print json.dumps(stats, indent=4,
# clear the stats before injecting
- c.clear_stats()
+ client.clear_stats()
+ total_rcvd = 0
+ total_sent = 0
+ lost_a = 0
+ lost_b = 0
# choose rate and start traffic
- c.start(ports=[0, 1], mult=rate, duration=duration)
+ client.start(ports=[0, 1], mult=rate, duration=duration)
# block until done
- c.wait_on_traffic(ports=[0, 1])
+ client.wait_on_traffic(ports=[0, 1], timeout=(duration+30))
# read the stats after the test
- stats = c.get_stats()
+ stats = client.get_stats()
print "#####statistics#####"
print json.dumps(stats, indent=4,
print "\npackets lost from 0 --> 1: {0} pkts".format(lost_a)
print "packets lost from 1 --> 0: {0} pkts".format(lost_b)
- print "rate={0}, totalReceived={1}, totalSent={2}, frameLoss={3}"\
- .format(rate, total_rcvd, total_sent, lost_a+lost_b)
- if (lost_a == 0) and (lost_b == 0):
- passed = True
- else:
- passed = False
-
- except STLError as e:
- passed = False
- print e
+ except STLError as ex_error:
+ print_error(str(ex_error))
+ sys.exit(1)
finally:
- c.disconnect()
+ client.disconnect()
+ print "rate={0}, totalReceived={1}, totalSent={2}, frameLoss={3}"\
+ .format(rate, total_rcvd, total_sent, lost_a+lost_b)
+
def print_help():
+ """Print help on stdout."""
print "args: [-h] -d <duration> -s <size of frame in bytes>"+\
" [-r] <traffic rate with unit: %, mpps> "+\
+ " [-6] Use of ipv6 "+\
"--p1_src_mac <port1_src_mac> "+\
"--p1_dst_mac <port1_dst_mac> "+\
"--p1_src_start_ip <port1_src_start_ip> "+\
"--p2_dst_end_ip <port2_dst_end_ip>"
+def print_error(msg):
+ """Print error message on stderr.
+
+ :param msg: Error message to print.
+ :type msg: string
+ :return: nothing
+ """
+
+ sys.stderr.write(msg+'\n')
+
+
def main(argv):
+ """Main function."""
_duration = 10
_frame_size = 64
_rate = '1mpps'
_traffic_options = {}
+ _use_ipv6 = False
try:
- opts, args = getopt.getopt(argv, "hd:s:r:o:",
- ["help",
- "p1_src_mac=",
- "p1_dst_mac=",
- "p1_src_start_ip=",
- "p1_src_end_ip=",
- "p1_dst_start_ip=",
- "p1_dst_end_ip=",
- "p2_src_mac=",
- "p2_dst_mac=",
- "p2_src_start_ip=",
- "p2_src_end_ip=",
- "p2_dst_start_ip=",
- "p2_dst_end_ip="])
+ opts, _ = getopt.getopt(argv, "hd:s:r:6o:",
+ ["help",
+ "p1_src_mac=",
+ "p1_dst_mac=",
+ "p1_src_start_ip=",
+ "p1_src_end_ip=",
+ "p1_dst_start_ip=",
+ "p1_dst_end_ip=",
+ "p2_src_mac=",
+ "p2_dst_mac=",
+ "p2_src_start_ip=",
+ "p2_src_end_ip=",
+ "p2_dst_start_ip=",
+ "p2_dst_end_ip="])
except getopt.GetoptError:
print_help()
- sys.exit(2)
+ sys.exit(1)
for opt, arg in opts:
if opt in ('-h', "--help"):
print_help()
_frame_size = int(arg)
elif opt == '-r':
_rate = arg
+ elif opt == '-6':
+ _use_ipv6 = True
elif opt.startswith("--p"):
_traffic_options[opt[2:]] = arg
print _traffic_options
if len(_traffic_options) != 6:
- print "Supported only: src_start_ip, src_end_ip, dst_start_ip"
+ print_error("Supported only: src_start_ip, src_end_ip, dst_start_ip")
print_help()
- sys.exit(2)
+ sys.exit(1)
- pkt_a, pkt_b = create_packets(_traffic_options,
- frame_size=_frame_size)
+ if _use_ipv6:
+ pkt_a, pkt_b = create_packets_v6(_traffic_options,
+ frame_size=_frame_size)
+ else:
+ pkt_a, pkt_b = create_packets(_traffic_options,
+ frame_size=_frame_size)
simple_burst(pkt_a, pkt_b, duration=_duration, rate=_rate)
if __name__ == "__main__":
main(sys.argv[1:])
-