blob: c4cb5d01da69aa99b76c8058ec895f4c62201bee (
plain)
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
|
from trex_stl_streams import *
from trex_stl_packet_builder_scapy import *
# map ports
# will destroy all streams/data on the ports
def stl_map_ports (client, ports = None):
# by default use all ports
if ports == None:
ports = client.get_all_ports()
# reset the ports
client.reset(ports)
# generate streams
base_pkt = CScapyTRexPktBuilder(pkt = Ether()/IP())
tx_pkts = {}
pkts = 1
for port in ports:
tx_pkts[pkts] = port
stream = STLStream(packet = base_pkt,
mode = STLTXSingleBurst(pps = 100000, total_pkts = pkts))
client.add_streams(stream, [port])
pkts = pkts * 2
# inject
client.clear_stats()
client.start(ports, mult = "1mpps")
client.wait_on_traffic(ports)
stats = client.get_stats()
# cleanup
client.reset(ports = ports)
table = {'map': {}, 'bi' : [], 'unknown': []}
# actual mapping
for port in ports:
ipackets = stats[port]["ipackets"]
table['map'][port] = None
for pkts in tx_pkts.keys():
if ( (pkts & ipackets) == pkts ):
tx_port = tx_pkts[pkts]
table['map'][port] = tx_port
unmapped = list(ports)
while len(unmapped) > 0:
port_a = unmapped.pop(0)
port_b = table['map'][port_a]
# if unknown - add to the unknown list
if port_b == None:
table['unknown'].append(port_a)
# self-loop, due to bug?
elif port_a == port_b:
continue
# bi-directional ports
elif (table['map'][port_b] == port_a):
unmapped.remove(port_b)
table['bi'].append( (port_a, port_b) )
return table
|
