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
|
import sys
import os
# Should be removed
# TBD fix this
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 scapy.all import *
from trex_stl_api import *
class STLS1(object):
def __init__ (self):
self.mode =0;
self.fsize =64;
def create_pkt_base (self):
t=[
Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025),
Ether()/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025),
Ether()/Dot1Q(vlan=12)/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025),
Ether()/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/TCP(dport=12,sport=1025),
Ether()/Dot1Q(vlan=12)/IPv6(src="::5")/TCP(dport=12,sport=1025),
Ether()/IP()/UDP()/IPv6(src="::5")/TCP(dport=12,sport=1025)
];
return t[self.mode]
def create_stream (self):
# create a base packet and pad it to size
size = self.fsize - 4; # no FCS
base_pkt = self.create_pkt_base ()
pad = max(0, size - len(base_pkt)) * 'x'
pkt = STLPktBuilder(pkt = base_pkt/pad,
vm = [])
return STLStream(packet = pkt,
mode = STLTXCont())
def get_streams (self, direction = 0):
# create 1 stream
return [ self.create_stream() ]
# dynamic load - used for trex console or simulator
def register():
return STLS1()
|