5 from framework import VppTestCase, VppTestRunner
6 from vpp_ip_route import VppIpTable
8 from scapy.packet import Raw
9 from scapy.layers.l2 import Ether
10 from scapy.layers.inet import IP, UDP, ICMP
11 from scapy.layers.inet6 import IPv6
13 from vpp_papi import VppEnum
18 class TestSVS(VppTestCase):
23 super(TestSVS, cls).setUpClass()
26 def tearDownClass(cls):
27 super(TestSVS, cls).tearDownClass()
30 super(TestSVS, self).setUp()
32 # create 2 pg interfaces
33 self.create_pg_interfaces(range(4))
37 for i in self.pg_interfaces:
41 tbl = VppIpTable(self, table_id)
43 tbl = VppIpTable(self, table_id, is_ip6=1)
46 i.set_table_ip4(table_id)
47 i.set_table_ip6(table_id)
55 for i in self.pg_interfaces:
61 super(TestSVS, self).tearDown()
64 """ Source VRF Select IP4 """
67 # packets destined out of the 3 non-default table interfaces
69 pkts_0 = [(Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
70 IP(src="1.1.1.1", dst=self.pg1.remote_ip4) /
71 UDP(sport=1234, dport=1234) /
73 (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
74 IP(src="2.2.2.2", dst=self.pg2.remote_ip4) /
75 UDP(sport=1234, dport=1234) /
77 (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
78 IP(src="3.3.3.3", dst=self.pg3.remote_ip4) /
79 UDP(sport=1234, dport=1234) /
81 pkts_1 = [(Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
82 IP(src="1.1.1.1", dst=self.pg1.remote_ip4) /
83 UDP(sport=1234, dport=1234) /
85 (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
86 IP(src="2.2.2.2", dst=self.pg2.remote_ip4) /
87 UDP(sport=1234, dport=1234) /
89 (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
90 IP(src="3.3.3.3", dst=self.pg3.remote_ip4) /
91 UDP(sport=1234, dport=1234) /
95 # before adding the SVS config all these packets are dropped when
96 # ingressing on pg0 since pg0 is in the default table
99 self.send_and_assert_no_replies(self.pg0, p * 1)
102 # Add table 1001 & 1002 into which we'll add the routes
103 # determining the source VRF selection
105 table_ids = [101, 102]
107 for table_id in table_ids:
108 self.vapi.svs_table_add_del(
110 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
114 # map X.0.0.0/8 to each SVS table for lookup in table X
116 for i in range(1, 4):
117 self.vapi.svs_route_add_del(
119 prefix="%d.0.0.0/8" % i,
124 # Enable SVS on pg0/pg1 using table 1001/1002
126 self.vapi.svs_enable_disable(
128 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
129 table_id=table_ids[0],
130 sw_if_index=self.pg0.sw_if_index)
131 self.vapi.svs_enable_disable(
133 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
134 table_id=table_ids[1],
135 sw_if_index=self.pg1.sw_if_index)
138 # now all the packets should be delivered out the respective interface
140 self.send_and_expect(self.pg0, pkts_0[0] * NUM_PKTS, self.pg1)
141 self.send_and_expect(self.pg0, pkts_0[1] * NUM_PKTS, self.pg2)
142 self.send_and_expect(self.pg0, pkts_0[2] * NUM_PKTS, self.pg3)
143 self.send_and_expect(self.pg1, pkts_1[0] * NUM_PKTS, self.pg1)
144 self.send_and_expect(self.pg1, pkts_1[1] * NUM_PKTS, self.pg2)
145 self.send_and_expect(self.pg1, pkts_1[2] * NUM_PKTS, self.pg3)
148 # check that if the SVS lookup does not match a route the packet
149 # is forwarded using the interface's routing table
151 p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
152 IP(src=self.pg0.remote_ip4, dst=self.pg0.remote_ip4) /
153 UDP(sport=1234, dport=1234) /
155 self.send_and_expect(self.pg0, p * NUM_PKTS, self.pg0)
157 p = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
158 IP(src=self.pg1.remote_ip4, dst=self.pg1.remote_ip4) /
159 UDP(sport=1234, dport=1234) /
161 self.send_and_expect(self.pg1, p * NUM_PKTS, self.pg1)
164 # dump the SVS configs
166 ss = self.vapi.svs_dump()
168 self.assertEqual(ss[0].table_id, table_ids[0])
169 self.assertEqual(ss[0].sw_if_index, self.pg0.sw_if_index)
170 self.assertEqual(ss[0].af, VppEnum.vl_api_address_family_t.ADDRESS_IP4)
171 self.assertEqual(ss[1].table_id, table_ids[1])
172 self.assertEqual(ss[1].sw_if_index, self.pg1.sw_if_index)
173 self.assertEqual(ss[1].af, VppEnum.vl_api_address_family_t.ADDRESS_IP4)
178 self.vapi.svs_enable_disable(
180 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
181 table_id=table_ids[0],
182 sw_if_index=self.pg0.sw_if_index)
183 self.vapi.svs_enable_disable(
185 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
186 table_id=table_ids[1],
187 sw_if_index=self.pg1.sw_if_index)
189 for table_id in table_ids:
190 for i in range(1, 4):
191 self.vapi.svs_route_add_del(
193 prefix="%d.0.0.0/8" % i,
197 self.vapi.svs_table_add_del(
199 af=VppEnum.vl_api_address_family_t.ADDRESS_IP4,
203 """ Source VRF Select IP6 """
206 # packets destined out of the 3 non-default table interfaces
208 pkts_0 = [(Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
209 IPv6(src="2001:1::1", dst=self.pg1.remote_ip6) /
210 UDP(sport=1234, dport=1234) /
212 (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
213 IPv6(src="2001:2::1", dst=self.pg2.remote_ip6) /
214 UDP(sport=1234, dport=1234) /
216 (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
217 IPv6(src="2001:3::1", dst=self.pg3.remote_ip6) /
218 UDP(sport=1234, dport=1234) /
220 pkts_1 = [(Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
221 IPv6(src="2001:1::1", dst=self.pg1.remote_ip6) /
222 UDP(sport=1234, dport=1234) /
224 (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
225 IPv6(src="2001:2::1", dst=self.pg2.remote_ip6) /
226 UDP(sport=1234, dport=1234) /
228 (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
229 IPv6(src="2001:3::1", dst=self.pg3.remote_ip6) /
230 UDP(sport=1234, dport=1234) /
234 # before adding the SVS config all these packets are dropped when
235 # ingressing on pg0 since pg0 is in the default table
238 self.send_and_assert_no_replies(self.pg0, p * 1)
241 # Add table 1001 & 1002 into which we'll add the routes
242 # determining the source VRF selection
244 table_ids = [101, 102]
246 for table_id in table_ids:
247 self.vapi.svs_table_add_del(
249 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
253 # map X.0.0.0/8 to each SVS table for lookup in table X
255 for i in range(1, 4):
256 self.vapi.svs_route_add_del(
258 prefix="2001:%d::/32" % i,
263 # Enable SVS on pg0/pg1 using table 1001/1002
265 self.vapi.svs_enable_disable(
267 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
268 table_id=table_ids[0],
269 sw_if_index=self.pg0.sw_if_index)
270 self.vapi.svs_enable_disable(
272 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
273 table_id=table_ids[1],
274 sw_if_index=self.pg1.sw_if_index)
277 # now all the packets should be delivered out the respective interface
279 self.send_and_expect(self.pg0, pkts_0[0] * NUM_PKTS, self.pg1)
280 self.send_and_expect(self.pg0, pkts_0[1] * NUM_PKTS, self.pg2)
281 self.send_and_expect(self.pg0, pkts_0[2] * NUM_PKTS, self.pg3)
282 self.send_and_expect(self.pg1, pkts_1[0] * NUM_PKTS, self.pg1)
283 self.send_and_expect(self.pg1, pkts_1[1] * NUM_PKTS, self.pg2)
284 self.send_and_expect(self.pg1, pkts_1[2] * NUM_PKTS, self.pg3)
287 # check that if the SVS lookup does not match a route the packet
288 # is forwarded using the interface's routing table
290 p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
291 IPv6(src=self.pg0.remote_ip6, dst=self.pg0.remote_ip6) /
292 UDP(sport=1234, dport=1234) /
294 self.send_and_expect(self.pg0, p * NUM_PKTS, self.pg0)
296 p = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
297 IPv6(src=self.pg1.remote_ip6, dst=self.pg1.remote_ip6) /
298 UDP(sport=1234, dport=1234) /
300 self.send_and_expect(self.pg1, p * NUM_PKTS, self.pg1)
303 # dump the SVS configs
305 ss = self.vapi.svs_dump()
307 self.assertEqual(ss[0].table_id, table_ids[0])
308 self.assertEqual(ss[0].sw_if_index, self.pg0.sw_if_index)
309 self.assertEqual(ss[0].af, VppEnum.vl_api_address_family_t.ADDRESS_IP6)
310 self.assertEqual(ss[1].table_id, table_ids[1])
311 self.assertEqual(ss[1].sw_if_index, self.pg1.sw_if_index)
312 self.assertEqual(ss[1].af, VppEnum.vl_api_address_family_t.ADDRESS_IP6)
317 self.vapi.svs_enable_disable(
319 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
320 table_id=table_ids[0],
321 sw_if_index=self.pg0.sw_if_index)
322 self.vapi.svs_enable_disable(
324 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
325 table_id=table_ids[1],
326 sw_if_index=self.pg1.sw_if_index)
328 for table_id in table_ids:
329 for i in range(1, 4):
330 self.vapi.svs_route_add_del(
332 prefix="2001:%d::/32" % i,
336 self.vapi.svs_table_add_del(
338 af=VppEnum.vl_api_address_family_t.ADDRESS_IP6,
341 if __name__ == '__main__':
342 unittest.main(testRunner=VppTestRunner)