src/plugins/nat/extras/nat_ses_open.py

   1 from trex_stl_lib.api import *
   2
   3 class STLS1(object):
   4
   5     def __init__ (self):
   6         self.ip_range = {'local': {'start': "10.0.0.3", 'end': "10.1.255.255"},
   7                          'external': {'start': "172.16.1.3", 'end': "172.16.1.3"},
   8                          'remote': {'start': "2.2.0.1", 'end': "2.2.0.1"}}
   9         self.port_range = {'local': {'start': 1025, 'end': 65535},
  10                            'remote': {'start': 12, 'end': 12}}
  11
  12     def create_stream (self, vm):
  13         base_pkt = Ether()/IP()/UDP()
  14
  15         if len(base_pkt) < 64:
  16             pad_len = 64 - len(base_pkt)
  17             pad = Padding()
  18             pad.load = '\x00' * pad_len
  19             base_pkt = base_pkt/pad
  20
  21         pkt = STLPktBuilder(pkt=base_pkt, vm=vm)
  22         return STLStream(packet=pkt, mode=STLTXCont())
  23
  24     def get_streams (self, direction = 0, **kwargs):
  25         if direction == 0:
  26             ip_src = self.ip_range['remote']
  27             ip_dst = self.ip_range['external']
  28             src_port = self.port_range['remote']
  29             dst_port = self.port_range['local']
  30         else:
  31             ip_src = self.ip_range['local']
  32             ip_dst = self.ip_range['remote']
  33             src_port = self.port_range['local']
  34             dst_port = self.port_range['remote']
  35
  36         vm = STLVM()
  37
  38         vm.var(name="ip_src", min_value=ip_src['start'], max_value=ip_src['end'], size=4, op="random")
  39         vm.var(name="ip_dst", min_value=ip_dst['start'], max_value=ip_dst['end'], size=4, op="random")
  40         vm.var(name="src_port", min_value=src_port['start'], max_value=src_port['end'], size=2, op="random")
  41         vm.var(name="dst_port", min_value=dst_port['start'], max_value=dst_port['end'], size=2, op="random")
  42
  43         vm.write(fv_name="ip_src", pkt_offset="IP.src")
  44         vm.write(fv_name="ip_dst", pkt_offset="IP.dst")
  45         vm.write(fv_name="src_port", pkt_offset="UDP.sport")
  46         vm.write(fv_name="dst_port", pkt_offset="UDP.dport")
  47
  48         vm.fix_chksum()
  49
  50         return [ self.create_stream(vm) ]
  51
  52
  53 # dynamic load - used for trex console or simulator
  54 def register():
  55     return STLS1()
  56
  57
  58