DTA/Generator/dta_keywrite_basic.py at main · jonlanglet/DTA · GitHub

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from trex_stl_lib.api import *
import argparse


class dta_base(Packet):
	name = "dtaBase"
	fields_desc = [
		XByteField("opcode",		0x01),
		XByteField("seqnum",		0), #DTA sequence number
		BitField("immediate",		0,		1),
		BitField("retransmitable",	0,		1),
		BitField("reserved", 		0,		6)
	]

class dta_keyWrite(Packet):
	name = "dtaKeyWrite"
	fields_desc = [
		ByteField("redundancy",	0),
		IntField("key",		0),
		IntField("data", 	0),
		#IntField("data2", 	0),
		#IntField("data3", 	0),
		#IntField("data4", 	0),
		#IntField("data5", 	0),
		#IntField("data6", 	0),
		#IntField("data7", 	0)
	]

class dta_append(Packet):
	name = "dtaAppend"
	fields_desc = [
		IntField("listID",		0),
		IntField("data", 	0)
	]

class STLS1(object):

	def create_stream (self, redundancy=4, key=1, data=1, seqnum=0):
		#redundancy = 4 #modify this for different redundancy tests
		#key = 1
		#data = 1
		#seqnum = 0

		base_pkt = Ether(dst="b8:ce:f6:d2:12:c7")\
			/IP(src="10.0.0.200",dst="10.0.0.51")\
			/UDP(sport=40041,dport=40040)\
			/dta_base(opcode=0x01,seqnum=seqnum)\
			/dta_keyWrite(redundancy=redundancy, key=key, data=data)#, data2=2, data3=3)

		vm = STLVM()

		#Increment sequence number
		vm.var(name='dta_seqnum', size=1, op='inc', step=1, min_value=0, max_value=255)
		vm.write(fv_name='dta_seqnum', pkt_offset='dta_base.seqnum')

		#Increment key
		vm.var(name='dta_key', size=4, op='inc', step=1, min_value=0, max_value=1000000000)
		vm.write(fv_name='dta_key', pkt_offset='dta_keyWrite.key')

		#Increment data
		vm.var(name='dta_data', size=4, op='inc', step=1, min_value=0, max_value=1000000000)
		vm.write(fv_name='dta_data', pkt_offset='dta_keyWrite.data')

		return STLStream(packet = STLPktBuilder(pkt = base_pkt, vm = vm), mode = STLTXCont())

	def get_streams (self, tunables, **kwargs):
		parser = argparse.ArgumentParser(description='Argparser for {}'.format(os.path.basename(__file__)), formatter_class=argparse.ArgumentDefaultsHelpFormatter)
		parser.add_argument('--redundancy', type=int, default=4, help='The KeyWrite redundancy')
		args = parser.parse_args(tunables)
		print(args)
		# create 1 stream
		return [ self.create_stream(redundancy=int(args.redundancy)) ]


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