[dts] [PATCH 2/3] framework packet: support ip layer elements strip
Marvin Liu
yong.liu at intel.com
Wed Oct 26 09:00:40 CEST 2016
Signed-off-by: Marvin Liu <yong.liu at intel.com>
diff --git a/framework/packet.py b/framework/packet.py
index 93d6879..05a3d1f 100755
--- a/framework/packet.py
+++ b/framework/packet.py
@@ -209,6 +209,21 @@ class scapy(object):
value = layer.src
elif element == 'dst':
value = layer.dst
+ elif element == 'type':
+ value = layer.type
+
+ return value
+
+ def strip_layer3(self, element):
+ value = None
+ layer = self.pkt.getlayer(1)
+ if layer is None:
+ return None
+
+ if element == 'src':
+ value = layer.src
+ elif element == 'dst':
+ value = layer.dst
return value
@@ -631,6 +646,9 @@ class Packet(object):
def strip_element_layer2(self, element):
return self.pktgen.strip_layer2(element)
+ def strip_element_layer3(self, element):
+ return self.pktgen.strip_layer3(element)
+
def strip_element_vlan(self, element):
return self.pktgen.strip_vlan(element)
--
1.9.3
More information about the dts
mailing list