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
|
from trex_stl_lib.api import *
# split the range of IP to cores
#
class STLS1(object):
def __init__ (self):
self.fsize =64;
def create_stream (self, direction, cache_size):
# Create base packet and pad it to size
size = self.fsize - 4; # HW will add 4 bytes ethernet FCS
src_ip = '16.0.0.1'
dst_ip = '48.0.0.1'
if direction:
src_ip, dst_ip = dst_ip, src_ip
base_pkt = Ether()/IP(src=src_ip,dst=dst_ip)/UDP(dport=12,sport=1025)
pad = max(0, size - len(base_pkt)) * 'x'
vm = STLScVmRaw( [ STLVmFlowVar ( "ip_src", min_value="10.0.0.1",
max_value="10.0.0.255", size=4, step=1,op="inc"),
STLVmWrFlowVar (fv_name="ip_src", pkt_offset= "IP.src" ), # write ip to packet IP.src
STLVmFixIpv4(offset = "IP") # fix checksum
]
,split_by_field = "ip_src" # split to cores base on the tuple generator
,cache_size = cache_size # the cache size
);
pkt = STLPktBuilder(pkt = base_pkt/pad,
vm = vm)
stream = STLStream(packet = pkt,
mode = STLTXCont())
#print(stream.to_code())
return stream
def get_streams (self, direction = 0, cache_size = 255, **kwargs):
# create 1 stream
return [ self.create_stream(direction, cache_size) ]
# dynamic load - used for trex console or simulator
def register():
return STLS1()
|
