1 from scapy.layers.l2 import Dot1Q
4 from vpp_pg_interface import VppPGInterface
5 from vpp_interface import VppInterface
20 @six.add_metaclass(abc.ABCMeta)
21 class VppSubInterface(VppPGInterface):
25 """Parent interface for this sub-interface"""
30 """Sub-interface ID"""
45 def __init__(self, test, parent, sub_id):
46 VppInterface.__init__(self, test)
48 self._parent.add_sub_if(self)
50 self.DOT1AD_TYPE = 0x88A8
51 self.DOT1Q_TYPE = 0x8100
53 def set_sw_if_index(self, sw_if_index):
54 super(VppSubInterface, self).set_sw_if_index(sw_if_index)
55 self.set_vtr(L2_VTR_OP.L2_DISABLED)
58 def create_arp_req(self):
62 def create_ndp_req(self):
65 def resolve_arp(self):
66 super(VppSubInterface, self).resolve_arp(self.parent)
68 def resolve_ndp(self):
69 super(VppSubInterface, self).resolve_ndp(self.parent)
72 def add_dot1_layer(self, pkt):
75 def remove_vpp_config(self):
76 self.test.vapi.delete_subif(self.sw_if_index)
78 def _add_tag(self, packet, vlan, tag_type):
79 payload = packet.payload
80 inner_type = packet.type
81 packet.remove_payload()
82 packet.add_payload(Dot1Q(vlan=vlan) / payload)
83 packet.payload.type = inner_type
84 packet.payload.vlan = vlan
85 packet.type = tag_type
88 def _remove_tag(self, packet, vlan=None, tag_type=None):
90 self.test.instance().assertEqual(packet.type, tag_type)
92 payload = packet.payload
94 self.test.instance().assertEqual(payload.vlan, vlan)
95 inner_type = payload.type
96 payload = payload.payload
97 packet.remove_payload()
98 packet.add_payload(payload)
99 packet.type = inner_type
102 def add_dot1q_layer(self, packet, vlan):
103 return self._add_tag(packet, vlan, self.DOT1Q_TYPE)
105 def add_dot1ad_layer(self, packet, outer, inner):
106 p = self._add_tag(packet, inner, self.DOT1Q_TYPE)
107 return self._add_tag(p, outer, self.DOT1AD_TYPE)
109 def remove_dot1q_layer(self, packet, vlan=None):
110 return self._remove_tag(packet, vlan, self.DOT1Q_TYPE)
112 def remove_dot1ad_layer(self, packet, outer=None, inner=None):
113 p = self._remove_tag(packet, outer, self.DOT1AD_TYPE)
114 return self._remove_tag(p, inner, self.DOT1Q_TYPE)
116 def set_vtr(self, vtr, push1q=0, tag=None, inner=None, outer=None):
121 if (vtr == L2_VTR_OP.L2_PUSH_1 or
122 vtr == L2_VTR_OP.L2_TRANSLATE_1_1 or
123 vtr == L2_VTR_OP.L2_TRANSLATE_2_1):
125 self._push1q = push1q
126 if (vtr == L2_VTR_OP.L2_PUSH_2 or
127 vtr == L2_VTR_OP.L2_TRANSLATE_1_2 or
128 vtr == L2_VTR_OP.L2_TRANSLATE_2_2):
131 self._push1q = push1q
133 self.test.vapi.l2_interface_vlan_tag_rewrite(
134 sw_if_index=self.sw_if_index, vtr_op=vtr, push_dot1q=self._push1q,
135 tag1=self._tag1, tag2=self._tag2)
139 class VppDot1QSubint(VppSubInterface):
146 def __init__(self, test, parent, sub_id, vlan=None):
147 super(VppDot1QSubint, self).__init__(test, parent, sub_id)
151 r = test.vapi.create_vlan_subif(parent.sw_if_index, vlan)
152 self.set_sw_if_index(r.sw_if_index)
154 def create_arp_req(self):
155 packet = VppPGInterface.create_arp_req(self)
156 return self.add_dot1_layer(packet)
158 def create_ndp_req(self):
159 packet = VppPGInterface.create_ndp_req(self)
160 return self.add_dot1_layer(packet)
162 # called before sending packet
163 def add_dot1_layer(self, packet):
164 return self.add_dot1q_layer(packet, self.vlan)
166 # called on received packet to "reverse" the add call
167 def remove_dot1_layer(self, packet):
168 return self.remove_dot1q_layer(packet, self.vlan)
171 class VppDot1ADSubint(VppSubInterface):
174 def outer_vlan(self):
176 return self._outer_vlan
179 def inner_vlan(self):
181 return self._inner_vlan
183 def __init__(self, test, parent, sub_id, outer_vlan, inner_vlan):
184 super(VppDot1ADSubint, self).__init__(test, parent, sub_id)
185 r = test.vapi.create_subif(sw_if_index=parent.sw_if_index,
186 sub_id=sub_id, outer_vlan_id=outer_vlan,
187 inner_vlan_id=inner_vlan, two_tags=1,
188 dot1ad=1, exact_match=1)
189 self.set_sw_if_index(r.sw_if_index)
190 self._outer_vlan = outer_vlan
191 self._inner_vlan = inner_vlan
193 def create_arp_req(self):
194 packet = VppPGInterface.create_arp_req(self)
195 return self.add_dot1_layer(packet)
197 def create_ndp_req(self):
198 packet = VppPGInterface.create_ndp_req(self)
199 return self.add_dot1_layer(packet)
201 def add_dot1_layer(self, packet):
202 return self.add_dot1ad_layer(packet, self.outer_vlan, self.inner_vlan)
204 def remove_dot1_layer(self, packet):
205 return self.remove_dot1ad_layer(packet, self.outer_vlan,
209 class VppP2PSubint(VppSubInterface):
211 def __init__(self, test, parent, sub_id, remote_mac):
212 super(VppP2PSubint, self).__init__(test, parent, sub_id)
213 r = test.vapi.p2p_ethernet_add(parent.sw_if_index,
215 self.set_sw_if_index(r.sw_if_index)
216 self.parent_sw_if_index = parent.sw_if_index
217 self.p2p_remote_mac = remote_mac
219 def add_dot1_layer(self, packet):
222 def remove_dot1_layer(self, packet):
225 def create_arp_req(self):
226 packet = VppPGInterface.create_arp_req(self)
229 def create_ndp_req(self):
230 packet = VppPGInterface.create_ndp_req(self)