1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#!/router/bin/python
import outer_packages
from client.trex_hltapi import CTRexHltApi
import traceback
import sys, time
from pprint import pprint
import argparse
def check_res(res):
if res['status'] == 0:
print('Encountered error:\n%s' % res['log'])
sys.exit(1)
return res
def save_streams_id(res, streams_id_arr):
stream_id = res.get('stream_id')
if type(stream_id) in (int, long):
streams_id_arr.append(stream_id)
elif type(stream_id) is list:
streams_id_arr.extend(stream_id)
def print_brief_stats(res):
title_str = ' '*3
tx_str = 'TX:'
rx_str = 'RX:'
for port_id, stat in res.iteritems():
if type(port_id) is not int:
continue
title_str += ' '*10 + 'Port%s' % port_id
tx_str += '%15s' % res[port_id]['aggregate']['tx']['total_pkts']
rx_str += '%15s' % res[port_id]['aggregate']['rx']['total_pkts']
print(title_str)
print(tx_str)
print(rx_str)
def wait_with_progress(seconds):
for i in range(0, seconds):
time.sleep(1)
sys.stdout.write('.')
sys.stdout.flush()
print('')
if __name__ == "__main__":
try:
parser = argparse.ArgumentParser(description='Example of using stateless TRex via HLT API.', formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument('-v', dest = 'verbose', default = 0, help='Stateless API verbosity:\n0: No prints\n1: Commands and their status\n2: Same as 1 + ZMQ in&out')
parser.add_argument('--device', dest = 'device', default = 'localhost', help='Address of TRex server')
args = parser.parse_args()
hlt_client = CTRexHltApi(verbose = int(args.verbose))
streams_id_arr = []
print('Connecting to %s...' % args.device)
res = check_res(hlt_client.connect(device = args.device, port_list = [0, 1], username = 'danklei', break_locks = True, reset = True))
port_handle = res['port_handle']
print('Connected.')
print('Create single_burst 100 packets rate_pps=100 on port 0')
res = check_res(hlt_client.traffic_config('create', port_handle = port_handle[0], transmit_mode = 'single_burst', pkts_per_burst = 100, rate_pps = 100))
save_streams_id(res, streams_id_arr)
print('Create continuous stream for port 1, rate_pps = 1')
res = check_res(hlt_client.traffic_config('create', port_handle = port_handle[1]))
save_streams_id(res, streams_id_arr)
print('Run traffic on both ports for 20s')
check_res(hlt_client.traffic_control(action = 'run', port_handle = port_handle, mul = {'type': 'raw', 'op': 'abs', 'value': 1}, duration = 20))
wait_with_progress(5)
print('All available HLT stats after 5s')
res = check_res(hlt_client.traffic_stats(mode = 'all', port_handle = port_handle))
pprint(res)
# delete the single_burst
print('Port 0 has finished the burst, delete it and put continuous with rate 1000. No stopping of other ports.')
check_res(hlt_client.traffic_config('reset', port_handle = port_handle[0]))
res = check_res(hlt_client.traffic_config('create', port_handle = port_handle[0], rate_pps = 1000))
save_streams_id(res, streams_id_arr)
check_res(hlt_client.traffic_control(action = 'run', port_handle = port_handle[0], mul = {'type': 'raw', 'op': 'abs', 'value': 1}, duration = 15))
wait_with_progress(5)
print('Sample after another %s seconds (only packets count)' % 5)
res = check_res(hlt_client.traffic_stats(mode = 'aggregate', port_handle = port_handle))
print_brief_stats(res)
#print('Stop the traffic on port 1')
#res = check_res(hlt_client.traffic_control('stop', port_handle = port_handle[1]))
#wait_with_progress(sample_duration)
#print('Sample after another %s seconds (only packets count)' % sample_duration)
#res = check_res(hlt_client.traffic_stats(mode = 'all', port_handle = port_handle))
#print_brief_stats(res)
print('Stop traffic at port 1')
res = check_res(hlt_client.traffic_control('stop', port_handle = port_handle[1]))
wait_with_progress(5)
print('Sample after another %s seconds (only packets count)' % 5)
res = check_res(hlt_client.traffic_stats(mode = 'aggregate', port_handle = port_handle))
print_brief_stats(res)
check_res(hlt_client.cleanup_session())
except Exception as e:
print(traceback.print_exc())
print(e)
raise
finally:
print('Done.')
|
