| #!/usr/bin/env python |
| |
| import unittest |
| import socket |
| |
| from framework import VppTestCase, VppTestRunner |
| from vpp_ip_route import VppIpRoute, VppRoutePath, VppMplsRoute, \ |
| VppMplsTable, VppIpMRoute, VppMRoutePath, VppIpTable, \ |
| MRouteEntryFlags, MRouteItfFlags, MPLS_LABEL_INVALID, DpoProto |
| from vpp_bier import * |
| |
| from scapy.packet import Raw |
| from scapy.layers.l2 import Ether |
| from scapy.layers.inet import IP, UDP, ICMP |
| from scapy.layers.inet6 import IPv6 |
| from scapy.contrib.mpls import MPLS |
| from scapy.contrib.bier import * |
| |
| |
| class TestBFIB(VppTestCase): |
| """ BIER FIB Test Case """ |
| |
| def test_bfib(self): |
| """ BFIB Unit Tests """ |
| error = self.vapi.cli("test bier") |
| |
| if error: |
| self.logger.critical(error) |
| self.assertEqual(error.find("Failed"), -1) |
| |
| |
| class TestBier(VppTestCase): |
| """ BIER Test Case """ |
| |
| def setUp(self): |
| super(TestBier, self).setUp() |
| |
| # create 2 pg interfaces |
| self.create_pg_interfaces(range(3)) |
| |
| # create the default MPLS table |
| self.tables = [] |
| tbl = VppMplsTable(self, 0) |
| tbl.add_vpp_config() |
| self.tables.append(tbl) |
| |
| tbl = VppIpTable(self, 10) |
| tbl.add_vpp_config() |
| self.tables.append(tbl) |
| |
| # setup both interfaces |
| for i in self.pg_interfaces: |
| if i == self.pg2: |
| i.set_table_ip4(10) |
| i.admin_up() |
| i.config_ip4() |
| i.resolve_arp() |
| i.enable_mpls() |
| |
| def tearDown(self): |
| for i in self.pg_interfaces: |
| i.disable_mpls() |
| i.unconfig_ip4() |
| i.set_table_ip4(0) |
| i.admin_down() |
| super(TestBier, self).tearDown() |
| |
| def send_and_assert_no_replies(self, intf, pkts, remark): |
| intf.add_stream(pkts) |
| self.pg_enable_capture(self.pg_interfaces) |
| self.pg_start() |
| for i in self.pg_interfaces: |
| i.assert_nothing_captured(remark=remark) |
| |
| def send_and_expect(self, input, pkts, output): |
| self.vapi.cli("trace add bier-mpls-lookup 10") |
| input.add_stream(pkts) |
| self.pg_enable_capture(self.pg_interfaces) |
| self.pg_start() |
| rx = output.get_capture(len(pkts)) |
| |
| def test_bier_midpoint(self): |
| """BIER midpoint""" |
| |
| # |
| # Add a BIER table for sub-domain 0, set 0, and BSL 256 |
| # |
| bti = VppBierTableID(0, 0, BIERLength.BIER_LEN_256) |
| bt = VppBierTable(self, bti, 77) |
| bt.add_vpp_config() |
| |
| # |
| # A packet with no bits set gets dropped |
| # |
| p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / |
| MPLS(label=77, ttl=255) / |
| BIER(length=BIERLength.BIER_LEN_256, |
| BitString=chr(0)*64) / |
| IPv6(src=self.pg0.remote_ip6, dst=self.pg0.remote_ip6) / |
| UDP(sport=1234, dport=1234) / |
| Raw()) |
| pkts = [p] |
| |
| self.send_and_assert_no_replies(self.pg0, pkts, |
| "Empty Bit-String") |
| |
| # |
| # Add a BIER route for each bit-position in the table via a different |
| # next-hop. Testing whether the BIER walk and replicate forwarding |
| # function works for all bit posisitons. |
| # |
| nh_routes = [] |
| bier_routes = [] |
| for i in range(1, 256): |
| nh = "10.0.%d.%d" % (i / 255, i % 255) |
| nh_routes.append(VppIpRoute(self, nh, 32, |
| [VppRoutePath(self.pg1.remote_ip4, |
| self.pg1.sw_if_index, |
| labels=[2000+i])])) |
| nh_routes[-1].add_vpp_config() |
| |
| bier_routes.append(VppBierRoute(self, bti, i, nh, 100+i)) |
| bier_routes[-1].add_vpp_config() |
| |
| # |
| # A packet with all bits set gets spat out to BP:1 |
| # |
| p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / |
| MPLS(label=77, ttl=255) / |
| BIER(length=BIERLength.BIER_LEN_256) / |
| IPv6(src=self.pg0.remote_ip6, dst=self.pg0.remote_ip6) / |
| UDP(sport=1234, dport=1234) / |
| Raw()) |
| pkts = [p] |
| |
| self.pg0.add_stream(pkts) |
| self.pg_enable_capture(self.pg_interfaces) |
| self.pg_start() |
| |
| rx = self.pg1.get_capture(255) |
| |
| for rxp in rx: |
| # |
| # The packets are not required to be sent in bit-position order |
| # when we setup the routes above we used the bit-position to |
| # construct the out-label. so use that here to determine the BP |
| # |
| olabel = rxp[MPLS] |
| bp = olabel.label - 2000 |
| |
| blabel = olabel[MPLS].payload |
| self.assertEqual(blabel.label, 100+bp) |
| |
| bier_hdr = blabel[MPLS].payload |
| |
| self.assertEqual(bier_hdr.id, 5) |
| self.assertEqual(bier_hdr.version, 0) |
| self.assertEqual(bier_hdr.length, BIERLength.BIER_LEN_256) |
| self.assertEqual(bier_hdr.entropy, 0) |
| self.assertEqual(bier_hdr.OAM, 0) |
| self.assertEqual(bier_hdr.RSV, 0) |
| self.assertEqual(bier_hdr.DSCP, 0) |
| self.assertEqual(bier_hdr.Proto, 5) |
| |
| # The bit-string should consist only of the BP given by i. |
| i = 0 |
| bitstring = "" |
| bpi = bp - 1 |
| while (i < bpi/8): |
| bitstring = chr(0) + bitstring |
| i += 1 |
| bitstring = chr(1 << bpi % 8) + bitstring |
| |
| while len(bitstring) < 32: |
| bitstring = chr(0) + bitstring |
| |
| self.assertEqual(len(bitstring), len(bier_hdr.BitString)) |
| self.assertEqual(bitstring, bier_hdr.BitString) |
| |
| def test_bier_head(self): |
| """BIER head""" |
| |
| # |
| # Add a BIER table for sub-domain 0, set 0, and BSL 256 |
| # |
| bti = VppBierTableID(0, 0, BIERLength.BIER_LEN_256) |
| bt = VppBierTable(self, bti, 77) |
| bt.add_vpp_config() |
| |
| # |
| # 2 bit positions via two next hops |
| # |
| nh1 = "10.0.0.1" |
| nh2 = "10.0.0.2" |
| ip_route_1 = VppIpRoute(self, nh1, 32, |
| [VppRoutePath(self.pg1.remote_ip4, |
| self.pg1.sw_if_index, |
| labels=[2001])]) |
| ip_route_2 = VppIpRoute(self, nh2, 32, |
| [VppRoutePath(self.pg1.remote_ip4, |
| self.pg1.sw_if_index, |
| labels=[2002])]) |
| ip_route_1.add_vpp_config() |
| ip_route_2.add_vpp_config() |
| |
| bier_route_1 = VppBierRoute(self, bti, 1, nh1, 101) |
| bier_route_2 = VppBierRoute(self, bti, 2, nh2, 102) |
| bier_route_1.add_vpp_config() |
| bier_route_2.add_vpp_config() |
| |
| # |
| # An imposition object with both bit-positions set |
| # |
| bi = VppBierImp(self, bti, 333, chr(0x3) * 32) |
| bi.add_vpp_config() |
| |
| # |
| # Add a multicast route that will forward into the BIER doamin |
| # |
| route_ing_232_1_1_1 = VppIpMRoute( |
| self, |
| "0.0.0.0", |
| "232.1.1.1", 32, |
| MRouteEntryFlags.MFIB_ENTRY_FLAG_NONE, |
| paths=[VppMRoutePath(self.pg0.sw_if_index, |
| MRouteItfFlags.MFIB_ITF_FLAG_ACCEPT), |
| VppMRoutePath(0xffffffff, |
| MRouteItfFlags.MFIB_ITF_FLAG_FORWARD, |
| proto=DpoProto.DPO_PROTO_BIER, |
| bier_imp=bi.bi_index)]) |
| route_ing_232_1_1_1.add_vpp_config() |
| |
| # |
| # inject a packet an IP. We expect it to be BIER encapped, |
| # replicated. |
| # |
| p = (Ether(dst=self.pg0.local_mac, |
| src=self.pg0.remote_mac) / |
| IP(src="1.1.1.1", dst="232.1.1.1") / |
| UDP(sport=1234, dport=1234)) |
| |
| self.pg0.add_stream([p]) |
| self.pg_enable_capture(self.pg_interfaces) |
| self.pg_start() |
| |
| rx = self.pg1.get_capture(2) |
| |
| def test_bier_tail(self): |
| """BIER Tail""" |
| |
| # |
| # Add a BIER table for sub-domain 0, set 0, and BSL 256 |
| # |
| bti = VppBierTableID(0, 0, BIERLength.BIER_LEN_256) |
| bt = VppBierTable(self, bti, 77) |
| bt.add_vpp_config() |
| |
| # |
| # disposition table |
| # |
| bdt = VppBierDispTable(self, 8) |
| bdt.add_vpp_config() |
| |
| # |
| # BIER route in table that's for-us |
| # |
| bier_route_1 = VppBierRoute(self, bti, 1, "0.0.0.0", 0, |
| disp_table=8) |
| bier_route_1.add_vpp_config() |
| |
| # |
| # An entry in the disposition table |
| # |
| bier_de_1 = VppBierDispEntry(self, bdt.id, 99, |
| BIER_HDR_PAYLOAD.BIER_HDR_PROTO_IPV4, |
| "0.0.0.0", 0, rpf_id=8192) |
| bier_de_1.add_vpp_config() |
| |
| # |
| # A multicast route to forward post BIER disposition |
| # |
| route_eg_232_1_1_1 = VppIpMRoute( |
| self, |
| "0.0.0.0", |
| "232.1.1.1", 32, |
| MRouteEntryFlags.MFIB_ENTRY_FLAG_NONE, |
| paths=[VppMRoutePath(self.pg1.sw_if_index, |
| MRouteItfFlags.MFIB_ITF_FLAG_FORWARD)]) |
| route_eg_232_1_1_1.add_vpp_config() |
| route_eg_232_1_1_1.update_rpf_id(8192) |
| |
| # |
| # A packet with all bits set gets spat out to BP:1 |
| # |
| p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / |
| MPLS(label=77, ttl=255) / |
| BIER(length=BIERLength.BIER_LEN_256, BFRID=99) / |
| IP(src="1.1.1.1", dst="232.1.1.1") / |
| UDP(sport=1234, dport=1234) / |
| Raw()) |
| |
| self.send_and_expect(self.pg0, [p], self.pg1) |
| |
| def test_bier_e2e(self): |
| """ BIER end-to-end """ |
| |
| # |
| # Add a BIER table for sub-domain 0, set 0, and BSL 256 |
| # |
| bti = VppBierTableID(0, 0, BIERLength.BIER_LEN_256) |
| bt = VppBierTable(self, bti, 77) |
| bt.add_vpp_config() |
| |
| # |
| # Impostion Sets bit string 101010101.... |
| # sender 333 |
| # |
| bi = VppBierImp(self, bti, 333, chr(0x5) * 32) |
| bi.add_vpp_config() |
| |
| # |
| # Add a multicast route that will forward into the BIER doamin |
| # |
| route_ing_232_1_1_1 = VppIpMRoute( |
| self, |
| "0.0.0.0", |
| "232.1.1.1", 32, |
| MRouteEntryFlags.MFIB_ENTRY_FLAG_NONE, |
| paths=[VppMRoutePath(self.pg0.sw_if_index, |
| MRouteItfFlags.MFIB_ITF_FLAG_ACCEPT), |
| VppMRoutePath(0xffffffff, |
| MRouteItfFlags.MFIB_ITF_FLAG_FORWARD, |
| proto=DpoProto.DPO_PROTO_BIER, |
| bier_imp=bi.bi_index)]) |
| route_ing_232_1_1_1.add_vpp_config() |
| |
| # |
| # disposition table 8 |
| # |
| bdt = VppBierDispTable(self, 8) |
| bdt.add_vpp_config() |
| |
| # |
| # BIER route in table that's for-us, resolving through |
| # disp table 8. |
| # |
| bier_route_1 = VppBierRoute(self, bti, 1, "0.0.0.0", |
| MPLS_LABEL_INVALID, |
| disp_table=8) |
| bier_route_1.add_vpp_config() |
| |
| # |
| # An entry in the disposition table for sender 333 |
| # lookup in VRF 10 |
| # |
| bier_de_1 = VppBierDispEntry(self, bdt.id, 333, |
| BIER_HDR_PAYLOAD.BIER_HDR_PROTO_IPV4, |
| "0.0.0.0", 10, rpf_id=8192) |
| bier_de_1.add_vpp_config() |
| |
| # |
| # Add a multicast route that will forward the traffic |
| # post-disposition |
| # |
| route_eg_232_1_1_1 = VppIpMRoute( |
| self, |
| "0.0.0.0", |
| "232.1.1.1", 32, |
| MRouteEntryFlags.MFIB_ENTRY_FLAG_NONE, |
| table_id=10, |
| paths=[VppMRoutePath(self.pg1.sw_if_index, |
| MRouteItfFlags.MFIB_ITF_FLAG_FORWARD)]) |
| route_eg_232_1_1_1.add_vpp_config() |
| route_eg_232_1_1_1.update_rpf_id(8192) |
| |
| # |
| # inject a packet in VRF-0. We expect it to be BIER encapped, |
| # replicated, then hit the disposition and be forwarded |
| # out of VRF 10, i.e. on pg1 |
| # |
| p = (Ether(dst=self.pg0.local_mac, |
| src=self.pg0.remote_mac) / |
| IP(src="1.1.1.1", dst="232.1.1.1") / |
| UDP(sport=1234, dport=1234)) |
| |
| self.send_and_expect(self.pg0, p*65, self.pg1) |
| |
| |
| if __name__ == '__main__': |
| unittest.main(testRunner=VppTestRunner) |