summaryrefslogtreecommitdiffstats
path: root/scripts/stl/profiles/udp_1pkt_mpls_vm.py
blob: b324a88d9b35ef4f3f90de946c70c6fc3fcb246a (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
40
41
42
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 scapy.contrib.mpls import * # import from contrib folder of scapy 
from trex_stl_api import *


class STLS1(object):

    def __init__ (self):
        pass;

    def create_stream (self):
        # 2 MPLS label the internal with  s=1 (last one)
        pkt =  Ether()/MPLS(label=17,cos=1,s=0,ttl=255)/MPLS(label=0,cos=1,s=1,ttl=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)/('x'*20)

        vm = CTRexScRaw( [ CTRexVmDescFlowVar(name="mlabel", min_value=1, max_value=2000, size=2, op="inc"), # 2 bytes var
                           CTRexVmDescWrFlowVar(fv_name="mlabel", pkt_offset= "MPLS:1.label")                # LABEL is 20 bits expected is val*8 as 3 LSB are off, 16,32,64 .. using new instruction it will be possible to write to any bits
                          ]
                       )

        # burst of 100 packets
        return STLStream(packet = STLPktBuilder(pkt = pkt ,vm = vm),
                         mode = STLTXSingleBurst( pps = 1, total_pkts = 100) )


    def get_streams (self, direction = 0):
        # create 1 stream 
        return [ self.create_stream() ]

def register():
    return STLS1()