summaryrefslogtreecommitdiffstats
path: root/src/plugins/nat/extras/nat_10ks.py
blob: c210d5e81d8aa667e66053fecfe5891abca30d7d (plain)
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
from trex_stl_lib.api import *


class STLS1:
    def create_stream(self):
        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.102",
            port_min=1025,
            port_max=1124,
            limit_flows=10000,
        )

        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())

    def get_streams(self, direction=0, **kwargs):
        return [self.create_stream()]


# dynamic load - used for trex console or simulator
def register():
    return STLS1()