X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fvpp_sub_interface.py;h=255cfffd827a46a481830562aae37055a9b9e5a1;hb=13a08cc;hp=b387d27b49db23c4d3771689c056c2b29d05b7ce;hpb=0178d52384e0428368f1acf3163e664ecda7b64c;p=vpp.git diff --git a/test/vpp_sub_interface.py b/test/vpp_sub_interface.py index b387d27b49d..255cfffd827 100644 --- a/test/vpp_sub_interface.py +++ b/test/vpp_sub_interface.py @@ -1,7 +1,8 @@ -from scapy.layers.l2 import Ether, Dot1Q +from scapy.layers.l2 import Dot1Q from abc import abstractmethod, ABCMeta -from vpp_interface import VppInterface from vpp_pg_interface import VppPGInterface +from vpp_papi_provider import L2_VTR_OP +from vpp_interface import VppInterface class VppSubInterface(VppPGInterface): @@ -17,11 +18,29 @@ class VppSubInterface(VppPGInterface): """Sub-interface ID""" return self._sub_id + @property + def tag1(self): + return self._tag1 + + @property + def tag2(self): + return self._tag2 + + @property + def vtr(self): + return self._vtr + def __init__(self, test, parent, sub_id): - self._test = test + VppInterface.__init__(self, test) self._parent = parent self._parent.add_sub_if(self) self._sub_id = sub_id + self.DOT1AD_TYPE = 0x88A8 + self.DOT1Q_TYPE = 0x8100 + + def set_sw_if_index(self, sw_if_index): + super(VppSubInterface, self).set_sw_if_index(sw_if_index) + self.set_vtr(L2_VTR_OP.L2_DISABLED) @abstractmethod def create_arp_req(self): @@ -41,6 +60,69 @@ class VppSubInterface(VppPGInterface): def add_dot1_layer(self, pkt): pass + def remove_vpp_config(self): + self.test.vapi.delete_subif(self.sw_if_index) + + def _add_tag(self, packet, vlan, tag_type): + payload = packet.payload + inner_type = packet.type + packet.remove_payload() + packet.add_payload(Dot1Q(vlan=vlan) / payload) + packet.payload.type = inner_type + packet.payload.vlan = vlan + packet.type = tag_type + return packet + + def _remove_tag(self, packet, vlan=None, tag_type=None): + if tag_type: + self.test.instance().assertEqual(packet.type, tag_type) + + payload = packet.payload + if vlan: + self.test.instance().assertEqual(payload.vlan, vlan) + inner_type = payload.type + payload = payload.payload + packet.remove_payload() + packet.add_payload(payload) + packet.type = inner_type + return packet + + def add_dot1q_layer(self, packet, vlan): + return self._add_tag(packet, vlan, self.DOT1Q_TYPE) + + def add_dot1ad_layer(self, packet, outer, inner): + p = self._add_tag(packet, inner, self.DOT1Q_TYPE) + return self._add_tag(p, outer, self.DOT1AD_TYPE) + + def remove_dot1q_layer(self, packet, vlan=None): + return self._remove_tag(packet, vlan, self.DOT1Q_TYPE) + + def remove_dot1ad_layer(self, packet, outer=None, inner=None): + p = self._remove_tag(packet, outer, self.DOT1AD_TYPE) + return self._remove_tag(p, inner, self.DOT1Q_TYPE) + + def set_vtr(self, vtr, push1q=0, tag=None, inner=None, outer=None): + self._tag1 = 0 + self._tag2 = 0 + self._push1q = 0 + + if (vtr == L2_VTR_OP.L2_PUSH_1 or + vtr == L2_VTR_OP.L2_TRANSLATE_1_1 or + vtr == L2_VTR_OP.L2_TRANSLATE_2_1): + self._tag1 = tag + self._push1q = push1q + if (vtr == L2_VTR_OP.L2_PUSH_2 or + vtr == L2_VTR_OP.L2_TRANSLATE_1_2 or + vtr == L2_VTR_OP.L2_TRANSLATE_2_2): + self._tag1 = outer + self._tag2 = inner + self._push1q = push1q + + self.test.vapi.sw_interface_set_l2_tag_rewrite( + self.sw_if_index, vtr, push=self._push1q, + tag1=self._tag1, tag2=self._tag2) + self._vtr = vtr + class VppDot1QSubint(VppSubInterface): @@ -50,13 +132,12 @@ class VppDot1QSubint(VppSubInterface): return self._vlan def __init__(self, test, parent, sub_id, vlan=None): + super(VppDot1QSubint, self).__init__(test, parent, sub_id) if vlan is None: vlan = sub_id - super(VppDot1QSubint, self).__init__(test, parent, sub_id) self._vlan = vlan - r = self.test.vapi.create_vlan_subif(parent.sw_if_index, self.vlan) - self._sw_if_index = r.sw_if_index - VppInterface.post_init_setup(self) + r = test.vapi.create_vlan_subif(parent.sw_if_index, vlan) + self.set_sw_if_index(r.sw_if_index) def create_arp_req(self): packet = VppPGInterface.create_arp_req(self) @@ -66,20 +147,13 @@ class VppDot1QSubint(VppSubInterface): packet = VppPGInterface.create_ndp_req(self) return self.add_dot1_layer(packet) + # called before sending packet def add_dot1_layer(self, packet): - payload = packet.payload - packet.remove_payload() - packet.add_payload(Dot1Q(vlan=self.sub_id) / payload) - return packet + return self.add_dot1q_layer(packet, self.vlan) + # called on received packet to "reverse" the add call def remove_dot1_layer(self, packet): - payload = packet.payload - self.test.instance().assertEqual(type(payload), Dot1Q) - self.test.instance().assertEqual(payload.vlan, self.vlan) - payload = payload.payload - packet.remove_payload() - packet.add_payload(payload) - return packet + return self.remove_dot1q_layer(packet, self.vlan) class VppDot1ADSubint(VppSubInterface): @@ -96,20 +170,12 @@ class VppDot1ADSubint(VppSubInterface): def __init__(self, test, parent, sub_id, outer_vlan, inner_vlan): super(VppDot1ADSubint, self).__init__(test, parent, sub_id) - self.DOT1AD_TYPE = 0x88A8 - self.DOT1Q_TYPE = 0x8100 + r = test.vapi.create_subif(parent.sw_if_index, sub_id, outer_vlan, + inner_vlan, dot1ad=1, two_tags=1, + exact_match=1) + self.set_sw_if_index(r.sw_if_index) self._outer_vlan = outer_vlan self._inner_vlan = inner_vlan - r = self.test.vapi.create_subif( - parent.sw_if_index, - self.sub_id, - self.outer_vlan, - self.inner_vlan, - dot1ad=1, - two_tags=1, - exact_match=1) - self._sw_if_index = r.sw_if_index - VppInterface.post_init_setup(self) def create_arp_req(self): packet = VppPGInterface.create_arp_req(self) @@ -120,25 +186,33 @@ class VppDot1ADSubint(VppSubInterface): return self.add_dot1_layer(packet) def add_dot1_layer(self, packet): - payload = packet.payload - packet.remove_payload() - packet.add_payload(Dot1Q(vlan=self.outer_vlan) / - Dot1Q(vlan=self.inner_vlan) / payload) - packet.type = self.DOT1AD_TYPE + return self.add_dot1ad_layer(packet, self.outer_vlan, self.inner_vlan) + + def remove_dot1_layer(self, packet): + return self.remove_dot1ad_layer(packet, self.outer_vlan, + self.inner_vlan) + + +class VppP2PSubint(VppSubInterface): + + def __init__(self, test, parent, sub_id, remote_mac): + super(VppP2PSubint, self).__init__(test, parent, sub_id) + r = test.vapi.create_p2pethernet_subif(parent.sw_if_index, + remote_mac, sub_id) + self.set_sw_if_index(r.sw_if_index) + self.parent_sw_if_index = parent.sw_if_index + self.p2p_remote_mac = remote_mac + + def add_dot1_layer(self, packet): return packet def remove_dot1_layer(self, packet): - self.test.instance().assertEqual(type(packet), Ether) - self.test.instance().assertEqual(packet.type, self.DOT1AD_TYPE) - packet.type = self.DOT1Q_TYPE - packet = Ether(str(packet)) - payload = packet.payload - self.test.instance().assertEqual(type(payload), Dot1Q) - self.test.instance().assertEqual(payload.vlan, self.outer_vlan) - payload = payload.payload - self.test.instance().assertEqual(type(payload), Dot1Q) - self.test.instance().assertEqual(payload.vlan, self.inner_vlan) - payload = payload.payload - packet.remove_payload() - packet.add_payload(payload) + return packet + + def create_arp_req(self): + packet = VppPGInterface.create_arp_req(self) + return packet + + def create_ndp_req(self): + packet = VppPGInterface.create_ndp_req(self) return packet