X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftest_ip4_vrf_multi_instance.py;h=cbda790637bf11036710c2635bbfed577e1b2aac;hb=4941afb4f96a20df7dc8b6688f7921a3d713b77d;hp=38604a534741479b930070202856e3c9ce60c506;hpb=90cf21b5d8fd2d3e531e841dcd752311df5f8a50;p=vpp.git diff --git a/test/test_ip4_vrf_multi_instance.py b/test/test_ip4_vrf_multi_instance.py index 38604a53474..cbda790637b 100644 --- a/test/test_ip4_vrf_multi_instance.py +++ b/test/test_ip4_vrf_multi_instance.py @@ -1,11 +1,11 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """IP4 VRF Multi-instance Test Case HLD: **NOTES:** - higher number of pg-ip4 interfaces causes problems => only 15 pg-ip4 \ - interfaces in 5 VRFs are tested + interfaces in 5 VRFs are tested - jumbo packets in configuration with 15 pg-ip4 interfaces leads to \ - problems too + problems too **config 1** - add 15 pg-ip4 interfaces @@ -18,8 +18,8 @@ **verify 1** - check VRF data by parsing output of ip_fib_dump API command - - all packets received correctly in case of pg-ip4 interfaces in the same - VRF + - all packets received correctly in case of pg-ip4 interfaces in the + same VRF - no packet received in case of pg-ip4 interfaces not in VRF - no packet received in case of pg-ip4 interfaces in different VRFs @@ -30,8 +30,8 @@ - send IP4 packets between all pg-ip4 interfaces in all VRF groups **verify 2** - - all packets received correctly in case of pg-ip4 interfaces in the same - VRF + - all packets received correctly in case of pg-ip4 interfaces in the + same VRF - no packet received in case of pg-ip4 interfaces not in VRF - no packet received in case of pg-ip4 interfaces in different VRFs @@ -43,8 +43,8 @@ **verify 3** - check VRF data by parsing output of ip_fib_dump API command - - all packets received correctly in case of pg-ip4 interfaces in the same - VRF + - all packets received correctly in case of pg-ip4 interfaces in the + same VRF - no packet received in case of pg-ip4 interfaces not in VRF - no packet received in case of pg-ip4 interfaces in different VRFs @@ -56,35 +56,34 @@ **verify 4** - check VRF data by parsing output of ip_fib_dump API command - - all packets received correctly in case of pg-ip4 interfaces in the same - VRF + - all packets received correctly in case of pg-ip4 interfaces in the + same VRF - no packet received in case of pg-ip4 interfaces not in VRF - no packet received in case of pg-ip4 interfaces in different VRFs """ import unittest import random -import socket -import scapy.compat from scapy.packet import Raw -from scapy.layers.l2 import Ether -from scapy.layers.inet import IP, UDP, ARP +from scapy.layers.l2 import Ether, ARP +from scapy.layers.inet import IP, UDP -from framework import VppTestCase, VppTestRunner +from framework import VppTestCase +from asfframework import VppTestRunner from util import ppp from vrf import VRFState def is_ipv4_misc(p): - """ Is packet one of uninteresting IPv4 broadcasts? """ + """Is packet one of uninteresting IPv4 broadcasts?""" if p.haslayer(ARP): return True return False class TestIp4VrfMultiInst(VppTestCase): - """ IP4 VRF Multi-instance Test Case """ + """IP4 VRF Multi-instance Test Case""" @classmethod def setUpClass(cls): @@ -102,17 +101,17 @@ class TestIp4VrfMultiInst(VppTestCase): try: # Create pg interfaces - cls.create_pg_interfaces( - range(cls.nr_of_vrfs * cls.pg_ifs_per_vrf)) + cls.create_pg_interfaces(range(cls.nr_of_vrfs * cls.pg_ifs_per_vrf)) # Packet flows mapping pg0 -> pg1, pg2 etc. cls.flows = dict() for i in range(len(cls.pg_interfaces)): - multiplicand = i / cls.pg_ifs_per_vrf + multiplicand = i // cls.pg_ifs_per_vrf pg_list = [ cls.pg_interfaces[multiplicand * cls.pg_ifs_per_vrf + j] for j in range(cls.pg_ifs_per_vrf) - if (multiplicand * cls.pg_ifs_per_vrf + j) != i] + if (multiplicand * cls.pg_ifs_per_vrf + j) != i + ] cls.flows[cls.pg_interfaces[i]] = pg_list # Packet sizes - jumbo packet (9018 bytes) skipped @@ -136,13 +135,14 @@ class TestIp4VrfMultiInst(VppTestCase): cls.pg_not_in_vrf = [pg_if for pg_if in cls.pg_interfaces] # Create mapping of pg_interfaces to VRF IDs - cls.pg_if_by_vrf_id = dict() + cls.pg_if_sets = dict() for i in range(cls.nr_of_vrfs): - vrf_id = i + 1 + set_id = i + 1 pg_list = [ cls.pg_interfaces[i * cls.pg_ifs_per_vrf + j] - for j in range(cls.pg_ifs_per_vrf)] - cls.pg_if_by_vrf_id[vrf_id] = pg_list + for j in range(cls.pg_ifs_per_vrf) + ] + cls.pg_if_sets[set_id] = pg_list except Exception: super(TestIp4VrfMultiInst, cls).tearDownClass() @@ -167,7 +167,21 @@ class TestIp4VrfMultiInst(VppTestCase): def show_commands_at_teardown(self): self.logger.info(self.vapi.ppcli("show ip fib")) - self.logger.info(self.vapi.ppcli("show ip arp")) + self.logger.info(self.vapi.ppcli("show ip4 neighbors")) + + def _assign_interfaces(self, vrf_id, if_set_id): + for i in range(self.pg_ifs_per_vrf): + pg_if = self.pg_if_sets[if_set_id][i] + pg_if.set_table_ip4(vrf_id) + self.logger.info( + "pg-interface %s added to IPv4 VRF ID %d" % (pg_if.name, vrf_id) + ) + if pg_if not in self.pg_in_vrf: + self.pg_in_vrf.append(pg_if) + if pg_if in self.pg_not_in_vrf: + self.pg_not_in_vrf.remove(pg_if) + pg_if.config_ip4() + pg_if.configure_ipv4_neighbors() def create_vrf_and_assign_interfaces(self, count, start=1): """ @@ -181,43 +195,52 @@ class TestIp4VrfMultiInst(VppTestCase): for i in range(count): vrf_id = i + start - pg_if = self.pg_if_by_vrf_id[vrf_id][0] - dest_addr = pg_if.local_ip4n - dest_addr_len = 24 - self.vapi.ip_table_add_del(is_add=1, table_id=vrf_id) + self.vapi.ip_table_add_del(is_add=1, table={"table_id": vrf_id}) self.logger.info("IPv4 VRF ID %d created" % vrf_id) if vrf_id not in self.vrf_list: self.vrf_list.append(vrf_id) if vrf_id in self.vrf_reset_list: self.vrf_reset_list.remove(vrf_id) - for j in range(self.pg_ifs_per_vrf): - pg_if = self.pg_if_by_vrf_id[vrf_id][j] - pg_if.set_table_ip4(vrf_id) - self.logger.info("pg-interface %s added to IPv4 VRF ID %d" - % (pg_if.name, vrf_id)) - if pg_if not in self.pg_in_vrf: - self.pg_in_vrf.append(pg_if) - if pg_if in self.pg_not_in_vrf: - self.pg_not_in_vrf.remove(pg_if) - pg_if.config_ip4() - pg_if.configure_ipv4_neighbors() + self._assign_interfaces(vrf_id, vrf_id) + self.logger.debug(self.vapi.ppcli("show ip fib")) + self.logger.debug(self.vapi.ppcli("show ip4 neighbors")) + + def create_vrf_by_id_and_assign_interfaces(self, set_id, vrf_id=0xFFFFFFFF): + """ + Create a FIB table / VRF by vrf_id, put 3 pg-ip4 interfaces + to FIB table / VRF. + + :param int vrf_id: Required table ID / VRF ID. \ + (Default value = 0xffffffff, ID will be selected automatically) + """ + ret = self.vapi.ip_table_allocate(table={"table_id": vrf_id}) + vrf_id = ret.table.table_id + self.logger.info("IPv4 VRF ID %d created" % vrf_id) + if vrf_id not in self.vrf_list: + self.vrf_list.append(vrf_id) + if vrf_id in self.vrf_reset_list: + self.vrf_reset_list.remove(vrf_id) + self._assign_interfaces(vrf_id, set_id) self.logger.debug(self.vapi.ppcli("show ip fib")) - self.logger.debug(self.vapi.ppcli("show ip arp")) + self.logger.debug(self.vapi.ppcli("show ip4 neighbors")) + + return vrf_id - def reset_vrf_and_remove_from_vrf_list(self, vrf_id): + def reset_vrf_and_remove_from_vrf_list(self, vrf_id, if_set_id=None): """ Reset required FIB table / VRF and remove it from VRF list. :param int vrf_id: The FIB table / VRF ID to be reset. """ - # self.vapi.reset_vrf(vrf_id, is_ipv6=0) - self.vapi.reset_fib(vrf_id, is_ipv6=0) + if if_set_id is None: + if_set_id = vrf_id + self.vapi.ip_table_flush(table={"table_id": vrf_id}) if vrf_id in self.vrf_list: self.vrf_list.remove(vrf_id) if vrf_id not in self.vrf_reset_list: self.vrf_reset_list.append(vrf_id) for j in range(self.pg_ifs_per_vrf): - pg_if = self.pg_if_by_vrf_id[vrf_id][j] + pg_if = self.pg_if_sets[if_set_id][j] pg_if.unconfig_ip4() if pg_if in self.pg_in_vrf: self.pg_in_vrf.remove(pg_if) @@ -225,8 +248,8 @@ class TestIp4VrfMultiInst(VppTestCase): self.pg_not_in_vrf.append(pg_if) self.logger.info("IPv4 VRF ID %d reset finished" % vrf_id) self.logger.debug(self.vapi.ppcli("show ip fib")) - self.logger.debug(self.vapi.ppcli("show ip arp")) - self.vapi.ip_table_add_del(is_add=0, table_id=vrf_id) + self.logger.debug(self.vapi.ppcli("show ip neighbors")) + self.vapi.ip_table_add_del(is_add=0, table={"table_id": vrf_id}) def create_stream(self, src_if, packet_sizes): """ @@ -243,16 +266,20 @@ class TestIp4VrfMultiInst(VppTestCase): src_host = random.choice(src_hosts) pkt_info = self.create_packet_info(src_if, dst_if) payload = self.info_to_payload(pkt_info) - p = (Ether(dst=src_if.local_mac, src=src_host.mac) / - IP(src=src_host.ip4, dst=dst_host.ip4) / - UDP(sport=1234, dport=1234) / - Raw(payload)) + p = ( + Ether(dst=src_if.local_mac, src=src_host.mac) + / IP(src=src_host.ip4, dst=dst_host.ip4) + / UDP(sport=1234, dport=1234) + / Raw(payload) + ) pkt_info.data = p.copy() size = random.choice(packet_sizes) self.extend_packet(p, size) pkts.append(p) - self.logger.debug("Input stream created for port %s. Length: %u pkt(s)" - % (src_if.name, len(pkts))) + self.logger.debug( + "Input stream created for port %s. Length: %u pkt(s)" + % (src_if.name, len(pkts)) + ) return pkts def create_stream_crosswise_vrf(self, src_if, vrf_id, packet_sizes): @@ -270,21 +297,25 @@ class TestIp4VrfMultiInst(VppTestCase): vrf_lst = list(self.vrf_list) vrf_lst.remove(vrf_id) for vrf in vrf_lst: - for dst_if in self.pg_if_by_vrf_id[vrf]: + for dst_if in self.pg_if_sets[vrf]: for dst_host in dst_if.remote_hosts: src_host = random.choice(src_hosts) pkt_info = self.create_packet_info(src_if, dst_if) payload = self.info_to_payload(pkt_info) - p = (Ether(dst=src_if.local_mac, src=src_host.mac) / - IP(src=src_host.ip4, dst=dst_host.ip4) / - UDP(sport=1234, dport=1234) / - Raw(payload)) + p = ( + Ether(dst=src_if.local_mac, src=src_host.mac) + / IP(src=src_host.ip4, dst=dst_host.ip4) + / UDP(sport=1234, dport=1234) + / Raw(payload) + ) pkt_info.data = p.copy() size = random.choice(packet_sizes) self.extend_packet(p, size) pkts.append(p) - self.logger.debug("Input stream created for port %s. Length: %u pkt(s)" - % (src_if.name, len(pkts))) + self.logger.debug( + "Input stream created for port %s. Length: %u pkt(s)" + % (src_if.name, len(pkts)) + ) return pkts def verify_capture(self, pg_if, capture): @@ -305,11 +336,13 @@ class TestIp4VrfMultiInst(VppTestCase): payload_info = self.payload_to_info(packet[Raw]) packet_index = payload_info.index self.assertEqual(payload_info.dst, dst_sw_if_index) - self.logger.debug("Got packet on port %s: src=%u (id=%u)" % - (pg_if.name, payload_info.src, packet_index)) + self.logger.debug( + "Got packet on port %s: src=%u (id=%u)" + % (pg_if.name, payload_info.src, packet_index) + ) next_info = self.get_next_packet_info_for_interface2( - payload_info.src, dst_sw_if_index, - last_info[payload_info.src]) + payload_info.src, dst_sw_if_index, last_info[payload_info.src] + ) last_info[payload_info.src] = next_info self.assertIsNotNone(next_info) self.assertEqual(packet_index, next_info.index) @@ -324,37 +357,37 @@ class TestIp4VrfMultiInst(VppTestCase): raise for i in self.pg_interfaces: remaining_packet = self.get_next_packet_info_for_interface2( - i, dst_sw_if_index, last_info[i.sw_if_index]) + i, dst_sw_if_index, last_info[i.sw_if_index] + ) self.assertIsNone( remaining_packet, - "Port %u: Packet expected from source %u didn't arrive" % - (dst_sw_if_index, i.sw_if_index)) + "Port %u: Packet expected from source %u didn't arrive" + % (dst_sw_if_index, i.sw_if_index), + ) - def verify_vrf(self, vrf_id): + def verify_vrf(self, vrf_id, if_set_id=None): """ Check if the FIB table / VRF ID is configured. :param int vrf_id: The FIB table / VRF ID to be verified. :return: 1 if the FIB table / VRF ID is configured, otherwise return 0. """ - ip_fib_dump = self.vapi.ip_fib_dump() - vrf_exist = False + if if_set_id is None: + if_set_id = vrf_id + ip_fib_dump = self.vapi.ip_route_dump(vrf_id) + vrf_exist = len(ip_fib_dump) vrf_count = 0 for ip_fib_details in ip_fib_dump: - if ip_fib_details.table_id == vrf_id: - if not vrf_exist: - vrf_exist = True - addr = socket.inet_ntoa(ip_fib_details.address) - found = False - for pg_if in self.pg_if_by_vrf_id[vrf_id]: - if found: + addr = ip_fib_details.route.prefix.network_address + found = False + for pg_if in self.pg_if_sets[if_set_id]: + if found: + break + for host in pg_if.remote_hosts: + if str(addr) == host.ip4: + vrf_count += 1 + found = True break - for host in pg_if.remote_hosts: - if scapy.compat.raw(addr) == \ - scapy.compat.raw(host.ip4): - vrf_count += 1 - found = True - break if not vrf_exist and vrf_count == 0: self.logger.info("IPv4 VRF ID %d is not configured" % vrf_id) return VRFState.not_configured @@ -394,8 +427,9 @@ class TestIp4VrfMultiInst(VppTestCase): capture = pg_if.get_capture(remark="interface is in VRF") self.verify_capture(pg_if, capture) elif pg_if in self.pg_not_in_vrf: - pg_if.assert_nothing_captured(remark="interface is not in VRF", - filter_out_fn=is_ipv4_misc) + pg_if.assert_nothing_captured( + remark="interface is not in VRF", filter_out_fn=is_ipv4_misc + ) self.logger.debug("No capture for interface %s" % pg_if.name) else: raise Exception("Unknown interface: %s" % pg_if.name) @@ -403,18 +437,20 @@ class TestIp4VrfMultiInst(VppTestCase): def run_crosswise_vrf_test(self): """ Create packet streams for every pg-ip4 interface in VRF towards all - pg-ip4 interfaces in other VRFs, send all prepared packet streams and \ + pg-ip4 interfaces in other VRFs, send all prepared packet streams and verify that: - - no packet received on all configured pg-ip4 interfaces + + - no packet received on all configured pg-ip4 interfaces :raise RuntimeError: If any packet is captured on any pg-ip4 interface. """ # Test # Create incoming packet streams for packet-generator interfaces for vrf_id in self.vrf_list: - for pg_if in self.pg_if_by_vrf_id[vrf_id]: + for pg_if in self.pg_if_sets[vrf_id]: pkts = self.create_stream_crosswise_vrf( - pg_if, vrf_id, self.pg_if_packet_sizes) + pg_if, vrf_id, self.pg_if_packet_sizes + ) pg_if.add_stream(pkts) # Enable packet capture and start packet sending @@ -424,29 +460,27 @@ class TestIp4VrfMultiInst(VppTestCase): # Verify # Verify outgoing packet streams per packet-generator interface for pg_if in self.pg_interfaces: - pg_if.assert_nothing_captured(remark="interface is in other VRF", - filter_out_fn=is_ipv4_misc) + pg_if.assert_nothing_captured( + remark="interface is in other VRF", filter_out_fn=is_ipv4_misc + ) self.logger.debug("No capture for interface %s" % pg_if.name) def test_ip4_vrf_01(self): - """ IP4 VRF Multi-instance test 1 - create 4 VRFs - """ + """IP4 VRF Multi-instance test 1 - create 4 VRFs""" # Config 1 # Create 4 VRFs self.create_vrf_and_assign_interfaces(4) # Verify 1 for vrf_id in self.vrf_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.configured, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.configured, VRFState) # Test 1 self.run_verify_test() self.run_crosswise_vrf_test() def test_ip4_vrf_02(self): - """ IP4 VRF Multi-instance test 2 - reset 2 VRFs - """ + """IP4 VRF Multi-instance test 2 - reset 2 VRFs""" # Config 2 # Reset 2 VRFs self.reset_vrf_and_remove_from_vrf_list(1) @@ -454,19 +488,16 @@ class TestIp4VrfMultiInst(VppTestCase): # Verify 2 for vrf_id in self.vrf_reset_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.reset, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.reset, VRFState) for vrf_id in self.vrf_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.configured, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.configured, VRFState) # Test 2 self.run_verify_test() self.run_crosswise_vrf_test() def test_ip4_vrf_03(self): - """ IP4 VRF Multi-instance 3 - add 2 VRFs - """ + """IP4 VRF Multi-instance 3 - add 2 VRFs""" # Config 3 # Add 1 of reset VRFs and 1 new VRF self.create_vrf_and_assign_interfaces(1) @@ -474,19 +505,16 @@ class TestIp4VrfMultiInst(VppTestCase): # Verify 3 for vrf_id in self.vrf_reset_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.reset, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.reset, VRFState) for vrf_id in self.vrf_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.configured, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.configured, VRFState) # Test 3 self.run_verify_test() self.run_crosswise_vrf_test() def test_ip4_vrf_04(self): - """ IP4 VRF Multi-instance test 4 - reset 4 VRFs - """ + """IP4 VRF Multi-instance test 4 - reset 4 VRFs""" # Config 4 # Reset all VRFs (i.e. no VRF except VRF=0 configured) for i in range(len(self.vrf_list)): @@ -494,17 +522,57 @@ class TestIp4VrfMultiInst(VppTestCase): # Verify 4 for vrf_id in self.vrf_reset_list: - self.assert_equal(self.verify_vrf(vrf_id), - VRFState.reset, VRFState) + self.assert_equal(self.verify_vrf(vrf_id), VRFState.reset, VRFState) vrf_list_length = len(self.vrf_list) self.assertEqual( - vrf_list_length, 0, - "List of configured VRFs is not empty: %s != 0" % vrf_list_length) + vrf_list_length, + 0, + "List of configured VRFs is not empty: %s != 0" % vrf_list_length, + ) # Test 4 self.run_verify_test() self.run_crosswise_vrf_test() + def test_ip4_vrf_05(self): + """IP4 VRF Multi-instance test 5 - id allocation""" + # Config 5 + # Create several VRFs + # Set vrf_id manually first + self.create_vrf_by_id_and_assign_interfaces(1, 1) + # Set vrf_id automatically a few times + auto_vrf_id = [ + self.create_vrf_by_id_and_assign_interfaces(i) for i in range(2, 5) + ] + + # Verify 5 + self.assert_equal(self.verify_vrf(1, 1), VRFState.configured, VRFState) + for i, vrf in enumerate(auto_vrf_id): + self.assert_equal( + self.verify_vrf(vrf, i + 2), VRFState.configured, VRFState + ) + + # Test 5 + self.run_verify_test() + + # Config 5.1 + # Reset VRFs + self.reset_vrf_and_remove_from_vrf_list(1) + for i, vrf in enumerate(auto_vrf_id): + self.reset_vrf_and_remove_from_vrf_list(vrf, i + 2) + + # Verify 5.1 + self.assert_equal(self.verify_vrf(1, 1), VRFState.reset, VRFState) + for i, vrf in enumerate(auto_vrf_id): + self.assert_equal(self.verify_vrf(vrf, i + 2), VRFState.reset, VRFState) + + vrf_list_length = len(self.vrf_list) + self.assertEqual( + vrf_list_length, + 0, + "List of configured VRFs is not empty: %s != 0" % vrf_list_length, + ) + -if __name__ == '__main__': +if __name__ == "__main__": unittest.main(testRunner=VppTestRunner)