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
|
import sys
import os
# we need the API path
CURRENT_PATH = os.path.dirname(os.path.realpath(__file__))
API_PATH = os.path.join(CURRENT_PATH, "../../api/stl")
sys.path.insert(0, API_PATH)
from trex_stl_api import *
from scapy.all import *
# IMIX profile - involves 3 streams of UDP packets
# 1 - 60 bytes
# 2 - 590 bytes
# 3 - 1514 bytes
class STLImix(object):
def __init__ (self):
# default IP range
self.ip_range = {'src': {'start': "10.0.0.1", 'end': "10.0.0.254"},
'dst': {'start': "8.0.0.1", 'end': "8.0.0.254"}}
# default IMIX properties
self.imix_table = [ {'size': 60, 'pps': 28},
{'size': 590, 'pps': 20},
{'size': 1514, 'pps': 4}]
def create_stream (self, size, pps, vm):
# create a base packet and pad it to size
base_pkt = Ether()/IP()/UDP()
pad = max(0, size - len(base_pkt)) * 'x'
pkt = STLPktBuilder(pkt = base_pkt/pad,
vm = vm)
return STLStream(packet = pkt,
mode = STLTXCont())
def get_streams (self, flip = False):
if not flip:
src = self.ip_range['src']
dst = self.ip_range['dst']
else:
src = self.ip_range['dst']
dst = self.ip_range['src']
# construct the base packet for the profile
vm =[
# src
STLVmFlowVar(name="src",min_value=src['start'],max_value=src['end'],size=4,op="inc"),
STLVmWriteFlowVar(fv_name="src",pkt_offset= "IP.src"),
# dst
STLVmFlowVar(name="dst",min_value=dst['start'],max_value=dst['end'],size=4,op="inc"),
STLVmWriteFlowVar(fv_name="dst",pkt_offset= "IP.dst"),
# checksum
STLVmFixIpv4(offset = "IP")
]
# create imix streams
return [self.create_stream(x['size'], x['pps'], vm) for x in self.imix_table]
# dynamic load - used for trex console or simulator
def register():
return STLImix()
|
