diff options
Diffstat (limited to 'scripts/stl')
-rw-r--r-- | scripts/stl/profiles/udp_1pkt_tuple_gen.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/scripts/stl/profiles/udp_1pkt_tuple_gen.py b/scripts/stl/profiles/udp_1pkt_tuple_gen.py new file mode 100644 index 00000000..f556cc31 --- /dev/null +++ b/scripts/stl/profiles/udp_1pkt_tuple_gen.py @@ -0,0 +1,56 @@ +import sys +import os + +#./stl-sim -f stl/profiles/udp_1pkt_tuple_gen.py -l 10 -o a.pcap + +# 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.fsize =64; + + def create_stream (self): + # create a base packet and pad it to size + size = self.fsize - 4; # no FCS + + base_pkt = Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025) + + pad = max(0, size - len(base_pkt)) * 'x' + + vm = CTRexScRaw( [ CTRexVmDescTupleGen ( ip_min="16.0.0.1", ip_max="16.0.0.2", + port_min=1025, port_max=65535, + name="tuple"), # define tuple gen + + CTRexVmDescWrFlowVar (fv_name="tuple.ip", pkt_offset= "IP.src" ), # write ip to packet IP.src + CTRexVmDescFixIpv4(offset = "IP"), # fix checksum + CTRexVmDescWrFlowVar (fv_name="tuple.port", pkt_offset= "UDP.sport" ) #write udp.port + ] + ); + + pkt = STLPktBuilder(pkt = base_pkt/pad, + vm = 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() + + + |