-#!/usr/bin/python
-
-# Copyright (c) 2016 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:
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# 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.
-
-"""This script uses T-REX stateless API to drive t-rex process.
-
-!!! OUTDATED 1.88 T-REX STATELESS API, USE WITH CAUTION !!!
-
-Requirements:
-- 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
-- 2 interfaces must be configured in configuretion file /etc/trex_cfg.yaml
-
-##################### Example of /etc/trex_cfg.yaml ##########################
-- 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
- - 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
- src_mac : [0x90,0xe2,0xba,0x1f,0x97,0xd5]
-##############################################################################
-
-Functionality:
-1. Configure traffic on running T-REX instance
-2. Clear statistics on all ports
-3. Ctart traffic with specified duration
-4. Print statistics
-
-"""
-
-
-import sys, getopt
-sys.path.insert(0, "/opt/trex-core-1.88/scripts/api/stl/")
-
-from trex_stl_api import *
-
-import dpkt
-import json
-import string
-
-def generate_payload(length):
- word = ''
- alphabet_size = len(string.letters)
- for i in range(length):
- word += string.letters[(i % alphabet_size)]
- return word
-
-def create_packets(traffic_options, frame_size=64):
-
- if frame_size < 64:
- print "Packet min. size is 64B"
- sys.exit(2)
-
- # build A side packet
- pkt_a = STLPktBuilder()
-
- pkt_a.add_pkt_layer("l2", dpkt.ethernet.Ethernet())
- pkt_a.add_pkt_layer("l3_ip", dpkt.ip.IP())
- pkt_a.add_pkt_layer("l4_udp", dpkt.udp.UDP())
- pkt_a.set_pkt_payload(generate_payload(frame_size -
- pkt_a.get_packet_length()))
- pkt_a.set_layer_attr("l3_ip", "len", len(pkt_a.get_layer('l3_ip')))
-
- # build B side packet
- pkt_b = pkt_a.clone()
-
- 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']
-
- pkt_a.set_eth_layer_addr(layer_name="l2",
- attr="src",
- mac_addr=p1_src_mac)
- pkt_a.set_eth_layer_addr(layer_name="l2",
- attr="dst",
- mac_addr=p1_dst_mac)
- pkt_b.set_eth_layer_addr(layer_name="l2",
- attr="src",
- mac_addr=p2_src_mac)
- pkt_b.set_eth_layer_addr(layer_name="l2",
- attr="dst",
- mac_addr=p2_dst_mac)
-
- # set IP range for pkt and split it by multiple cores
- pkt_a.set_vm_ip_range(ip_layer_name="l3_ip",
- ip_field="src",
- ip_start=p1_src_start_ip, ip_end=p1_src_end_ip,
- operation="inc",
- split=True)
-
- pkt_a.set_vm_ip_range(ip_layer_name="l3_ip",
- ip_field="dst",
- ip_start=p1_dst_start_ip, ip_end=p1_dst_end_ip,
- operation="inc")
-
-
- # build B side packet
- pkt_b.set_vm_ip_range(ip_layer_name="l3_ip",
- ip_field="src",
- ip_start=p2_src_start_ip, ip_end=p2_src_end_ip,
- operation="inc",
- split=True)
-
- pkt_b.set_vm_ip_range(ip_layer_name="l3_ip",
- ip_field="dst",
- ip_start=p2_dst_start_ip, ip_end=p2_dst_end_ip,
- operation="inc")
-
- return(pkt_a, pkt_b)
-
-def simple_burst(pkt_a, pkt_b, duration=10, rate="1mpps",
- warmup=True, warmup_time=5):
-
- # create client
- c = STLClient()
- passed = True
-
- try:
- # turn this on for some information
- #c.set_verbose("high")
-
- # create two streams
- s1 = STLStream(packet=pkt_a,
- mode=STLTXCont(pps=100))
-
- # second stream with a phase of 1ms (inter stream gap)
- s2 = STLStream(packet=pkt_b,
- isg=1000,
- mode=STLTXCont(pps=100))
-
-
- # connect to server
- c.connect()
-
- # prepare our ports (my machine has 0 <--> 1 with static route)
- c.reset(ports=[0, 1])
-
- # add both streams to ports
- c.add_streams(s1, ports=[0])
- c.add_streams(s2, 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()
- print "#####warmup statistics#####"
- print json.dumps(stats["port 0"], indent=4,
- separators=(',', ': '), sort_keys=True)
- print json.dumps(stats["port 1"], indent=4,
- separators=(',', ': '), sort_keys=True)
- lost_a = stats["port 0"]["opackets"] - stats["port 1"]["ipackets"]
- lost_b = stats["port 1"]["opackets"] - stats["port 0"]["ipackets"]
-
- print "\npackets lost from 0 --> 1: {0} pkts".format(lost_a)
- print "packets lost from 1 --> 0: {0} pkts".format(lost_b)
-
-
- # clear the stats before injecting
- c.clear_stats()
-
- # choose rate and start traffic
- c.start(ports=[0, 1], mult=rate, duration=duration)
-
- # block until done
- c.wait_on_traffic(ports=[0, 1])
-
- # read the stats after the test
- stats = c.get_stats()
-
- print "#####statistics#####"
- print json.dumps(stats["port 0"], indent=4,
- separators=(',', ': '), sort_keys=True)
- print json.dumps(stats["port 1"], indent=4,
- separators=(',', ': '), sort_keys=True)
-
- lost_a = stats["port 0"]["opackets"] - stats["port 1"]["ipackets"]
- lost_b = stats["port 1"]["opackets"] - stats["port 0"]["ipackets"]
-
- total_sent = stats["port 0"]["opackets"] + stats["port 1"]["opackets"]
- total_rcvd = stats["port 0"]["ipackets"] + stats["port 1"]["ipackets"]
-
- 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
-
- finally:
- c.disconnect()
-
-def print_help():
-
- print "args: [-h] -d <duration> -s <size of frame in bytes>"+\
- " [-r] <traffic rate with unit: %, mpps> "+\
- "--p1_src_mac <port1_src_mac> "+\
- "--p1_dst_mac <port1_dst_mac> "+\
- "--p1_src_start_ip <port1_src_start_ip> "+\
- "--p1_src_end_ip <port1_src_end_ip> "+\
- "--p1_dst_start_ip <port1_dst_start_ip> "+\
- "--p1_dst_end_ip <port1_dst_end_ip> "+\
- "--p2_src_mac <port2_src_mac> "+\
- "--p2_dst_mac <port2_dst_mac> "+\
- "--p2_src_start_ip <port2_src_start_ip> "+\
- "--p2_src_end_ip <port2_src_end_ip> "+\
- "--p2_dst_start_ip <port2_dst_start_ip> "+\
- "--p2_dst_end_ip <port2_dst_end_ip>"
-
-
-def main(argv):
-
- _duration = 10
- _frame_size = 64
- _rate = '1mpps'
- _traffic_options = {}
-
- 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="])
- except getopt.GetoptError:
- print_help()
- sys.exit(2)
- for opt, arg in opts:
- if opt in ('-h', "--help"):
- print_help()
- sys.exit()
- elif opt == '-d':
- _duration = int(arg)
- elif opt == '-s':
- _frame_size = int(arg)
- elif opt == '-r':
- _rate = arg
- elif opt.startswith( "--p" ):
- _traffic_options[opt[2:]] = arg
-
- print _traffic_options
- if len(_traffic_options) != 12:
- print "Expected all 12 traffic options"
- print_help()
- sys.exit(2)
-
- 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:])
-
Code Review / csit.git / commitdiff