Skip to content

Commit e560b7f

Browse files
committed
Added IPv4 ECN configuration
1 parent 8c04c1e commit e560b7f

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/trafficItem.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def configRawTrafficItems(self, ngpfMacAddresses: dict=None) -> None:
126126
# IPv4
127127
if endpoint['packetHeaders'].get('ipv4', None):
128128
ipv4FieldObj = self.createPacketHeader(rawTrafficItemObj, packetHeaderToAdd='ipv4',
129-
appendToStack=f'^{endpoint["packetHeaders"]["ipv4"]["appendToStack"]}$')
129+
appendToStack=f'^{endpoint["packetHeaders"]["ipv4"]["appendToStack"]}$')
130130
ipv4SrcField = ipv4FieldObj.find(DisplayName='Source Address')
131131
ipv4SrcField.ValueType = endpoint["packetHeaders"]["ipv4"]["src"]["valueType"]
132132
ipv4SrcField.StartValue = endpoint["packetHeaders"]["ipv4"]["src"]["startValue"]
@@ -139,6 +139,12 @@ def configRawTrafficItems(self, ngpfMacAddresses: dict=None) -> None:
139139
ipv4DstField.StepValue = endpoint["packetHeaders"]["ipv4"]["dst"]["stepValue"]
140140
ipv4DstField.CountValue = endpoint["packetHeaders"]["ipv4"]["dst"]["count"]
141141

142+
if endpoint['packetHeaders']['ipv4'].get('ecn', None):
143+
# ECN uses field 9 (Unused field)
144+
ipv4EcnField = ipv4FieldObj.find()[10]
145+
ipv4EcnField.ActiveFieldChoice = True
146+
ipv4EcnField.SingleValue = endpoint["packetHeaders"]["ipv4"]["ecn"]
147+
142148
if endpoint['packetHeaders'].get('udp', None):
143149
udpFieldObj = self.createPacketHeader(rawTrafficItemObj, packetHeaderToAdd='^udp$', appendToStack='ipv4')
144150
udpSrcField = udpFieldObj.find(DisplayName='UDP-Source-Port')
@@ -156,7 +162,7 @@ def configRawTrafficItems(self, ngpfMacAddresses: dict=None) -> None:
156162
# 101 CRITIC/ECP, 110 Internetwork Control, 111 Network Control
157163
ipv4PrecedenceField = ipv4FieldObj.find(DisplayName='Precedence')
158164
ipv4PrecedenceField.ActiveFieldChoice = True
159-
ipv4PrecedenceField.FieldValue = '011 Flash'
165+
ipv4PrecedenceField.FieldValue = endpoint["packetHeaders"]["ipv4"]["precedence"]
160166

161167
# For IPv4 Raw priority: Field/3
162168
#ipv4RawPriorityField = ipv4FieldObj.find(DisplayName='Raw priority')

0 commit comments

Comments
 (0)