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
|
from trex_stl_lib.api import *
class STLS1:
def create_stream(self, port_id):
# base_pkt = Ether()/IP(dst="2.2.0.1")/UDP(dport=12)
# pad = Padding()
# if len(base_pkt) < 64:
# pad_len = 64 - len(base_pkt)
# pad.load = '\x00' * pad_len
# vm = STLVM()
# vm.tuple_var(name="tuple", ip_min="10.0.0.3", ip_max="10.0.0.202", port_min=1025, port_max=61124, limit_flows = 100000)
# vm.write(fv_name="tuple.ip", pkt_offset="IP.src")
# vm.fix_chksum()
# vm.write(fv_name="tuple.port", pkt_offset="UDP.sport")
# pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
# return STLStream(packet=pkt, mode=STLTXCont())
vm = STLScVmRaw(
[
STLVmTupleGen(
ip_min="10.0.0.1",
ip_max="10.255.255.254",
port_min=1025,
port_max=65535,
name="stuple",
limit_flows=10000,
),
STLVmTupleGen(
ip_min="2.0.0.1",
ip_max="2.255.255.254",
port_min=1025,
port_max=65535,
name="dtuple",
limit_flows=100000000,
),
# write ip to packet IP.src
STLVmWrFlowVar(fv_name="stuple.ip", pkt_offset="IP.src"),
STLVmWrFlowVar(fv_name="dtuple.ip", pkt_offset="IP.dst"),
# fix checksum
STLVmFixIpv4(offset="IP"),
# write udp.port
STLVmWrFlowVar(fv_name="stuple.port", pkt_offset="UDP.sport"),
STLVmWrFlowVar(fv_name="dtuple.port", pkt_offset="UDP.dport"),
]
)
base_pkt = (
Ether() / IP(src="16.0.0.1", dst="2.0.0.1") / UDP(dport=12, sport=1025)
)
pad = Padding()
if len(base_pkt) < 64:
pad_len = 64 - len(base_pkt)
pad.load = "\x00" * pad_len
pad = max(0, 64 - len(base_pkt)) * "x"
pad_latency = max(0, (64 - 4) - len(base_pkt)) * "x"
pkt = STLPktBuilder(pkt=base_pkt / pad, vm=vm)
return [
STLStream(packet=pkt, mode=STLTXCont()),
# latency stream
STLStream(
packet=STLPktBuilder(pkt=base_pkt / pad_latency),
mode=STLTXCont(pps=1000),
flow_stats=STLFlowLatencyStats(pg_id=12 + port_id),
),
]
def get_streams(self, direction=0, **kwargs):
# return [self.create_stream()]
return self.create_stream(kwargs["port_id"])
# dynamic load - used for trex console or simulator
def register():
return STLS1()
|
