| #!/usr/bin/env python3 |
| """ ACL plugin extended stateful tests """ |
| |
| import unittest |
| from config import config |
| from framework import VppTestCase |
| from scapy.layers.inet import IP, UDP, TCP |
| from scapy.packet import Packet |
| from socket import AF_INET, AF_INET6 |
| from scapy.layers.inet6 import IPv6 |
| from util import L4_Conn |
| from ipaddress import ip_network |
| |
| from vpp_acl import AclRule, VppAcl, VppAclInterface |
| |
| |
| def to_acl_rule(self, is_permit, wildcard_sport=False): |
| p = self |
| rule_family = AF_INET6 if p.haslayer(IPv6) else AF_INET |
| rule_prefix_len = 128 if p.haslayer(IPv6) else 32 |
| rule_l3_layer = IPv6 if p.haslayer(IPv6) else IP |
| rule_l4_sport = p.sport |
| rule_l4_dport = p.dport |
| if p.haslayer(IPv6): |
| rule_l4_proto = p[IPv6].nh |
| else: |
| rule_l4_proto = p[IP].proto |
| |
| if wildcard_sport: |
| rule_l4_sport_first = 0 |
| rule_l4_sport_last = 65535 |
| else: |
| rule_l4_sport_first = rule_l4_sport |
| rule_l4_sport_last = rule_l4_sport |
| |
| new_rule = AclRule( |
| is_permit=is_permit, |
| proto=rule_l4_proto, |
| src_prefix=ip_network((p[rule_l3_layer].src, rule_prefix_len)), |
| dst_prefix=ip_network((p[rule_l3_layer].dst, rule_prefix_len)), |
| sport_from=rule_l4_sport_first, |
| sport_to=rule_l4_sport_last, |
| dport_from=rule_l4_dport, |
| dport_to=rule_l4_dport, |
| ) |
| |
| return new_rule |
| |
| |
| Packet.to_acl_rule = to_acl_rule |
| |
| |
| class IterateWithSleep: |
| def __init__(self, testcase, n_iters, description, sleep_sec): |
| self.curr = 0 |
| self.testcase = testcase |
| self.n_iters = n_iters |
| self.sleep_sec = sleep_sec |
| self.description = description |
| |
| def __iter__(self): |
| for x in range(0, self.n_iters): |
| yield x |
| self.testcase.sleep(self.sleep_sec) |
| |
| |
| class Conn(L4_Conn): |
| def apply_acls(self, reflect_side, acl_side): |
| pkts = [] |
| pkts.append(self.pkt(0)) |
| pkts.append(self.pkt(1)) |
| pkt = pkts[reflect_side] |
| |
| r = [] |
| r.append(pkt.to_acl_rule(2, wildcard_sport=True)) |
| r.append(self.wildcard_rule(0)) |
| reflect_acl = VppAcl(self.testcase, r) |
| reflect_acl.add_vpp_config() |
| |
| r = [] |
| r.append(self.wildcard_rule(0)) |
| deny_acl = VppAcl(self.testcase, r) |
| deny_acl.add_vpp_config() |
| |
| if reflect_side == acl_side: |
| acl_if0 = VppAclInterface( |
| self.testcase, |
| self.ifs[acl_side].sw_if_index, |
| [reflect_acl, deny_acl], |
| n_input=1, |
| ) |
| acl_if1 = VppAclInterface( |
| self.testcase, self.ifs[1 - acl_side].sw_if_index, [], n_input=0 |
| ) |
| acl_if0.add_vpp_config() |
| acl_if1.add_vpp_config() |
| else: |
| acl_if0 = VppAclInterface( |
| self.testcase, |
| self.ifs[acl_side].sw_if_index, |
| [deny_acl, reflect_acl], |
| n_input=1, |
| ) |
| acl_if1 = VppAclInterface( |
| self.testcase, self.ifs[1 - acl_side].sw_if_index, [], n_input=0 |
| ) |
| acl_if0.add_vpp_config() |
| acl_if1.add_vpp_config() |
| |
| def wildcard_rule(self, is_permit): |
| any_addr = ["0.0.0.0", "::"] |
| rule_family = self.address_family |
| is_ip6 = 1 if rule_family == AF_INET6 else 0 |
| new_rule = AclRule( |
| is_permit=is_permit, |
| proto=0, |
| src_prefix=ip_network((any_addr[is_ip6], 0)), |
| dst_prefix=ip_network((any_addr[is_ip6], 0)), |
| sport_from=0, |
| sport_to=65535, |
| dport_from=0, |
| dport_to=65535, |
| ) |
| return new_rule |
| |
| |
| @unittest.skipUnless(config.extended, "part of extended tests") |
| class ACLPluginConnTestCase(VppTestCase): |
| """ACL plugin connection-oriented extended testcases""" |
| |
| @classmethod |
| def setUpClass(cls): |
| super(ACLPluginConnTestCase, cls).setUpClass() |
| # create pg0 and pg1 |
| cls.create_pg_interfaces(range(2)) |
| cmd = "set acl-plugin session table event-trace 1" |
| cls.logger.info(cls.vapi.cli(cmd)) |
| for i in cls.pg_interfaces: |
| i.admin_up() |
| i.config_ip4() |
| i.config_ip6() |
| i.resolve_arp() |
| i.resolve_ndp() |
| |
| @classmethod |
| def tearDownClass(cls): |
| super(ACLPluginConnTestCase, cls).tearDownClass() |
| |
| def tearDown(self): |
| """Run standard test teardown and log various show commands""" |
| super(ACLPluginConnTestCase, self).tearDown() |
| |
| def show_commands_at_teardown(self): |
| self.logger.info(self.vapi.cli("show ip neighbors")) |
| self.logger.info(self.vapi.cli("show ip6 neighbors")) |
| self.logger.info(self.vapi.cli("show acl-plugin sessions")) |
| self.logger.info(self.vapi.cli("show acl-plugin acl")) |
| self.logger.info(self.vapi.cli("show acl-plugin interface")) |
| self.logger.info(self.vapi.cli("show acl-plugin tables")) |
| self.logger.info(self.vapi.cli("show event-logger all")) |
| |
| def run_basic_conn_test(self, af, acl_side): |
| """Basic conn timeout test""" |
| conn1 = Conn(self, self.pg0, self.pg1, af, UDP, 42001, 4242) |
| conn1.apply_acls(0, acl_side) |
| conn1.send_through(0) |
| # the return packets should pass |
| conn1.send_through(1) |
| # send some packets on conn1, ensure it doesn't go away |
| for i in IterateWithSleep(self, 20, "Keep conn active", 0.3): |
| conn1.send_through(1) |
| # allow the conn to time out |
| for i in IterateWithSleep(self, 30, "Wait for timeout", 0.1): |
| pass |
| # now try to send a packet on the reflected side |
| try: |
| p2 = conn1.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| self.assert_equal(p2, None, "packet on long-idle conn") |
| |
| def run_active_conn_test(self, af, acl_side): |
| """Idle connection behind active connection test""" |
| base = 10000 + 1000 * acl_side |
| conn1 = Conn(self, self.pg0, self.pg1, af, UDP, base + 1, 2323) |
| conn2 = Conn(self, self.pg0, self.pg1, af, UDP, base + 2, 2323) |
| conn3 = Conn(self, self.pg0, self.pg1, af, UDP, base + 3, 2323) |
| conn1.apply_acls(0, acl_side) |
| conn1.send(0) |
| conn1.recv(1) |
| # create and check that the conn2/3 work |
| self.sleep(0.1) |
| conn2.send_pingpong(0) |
| self.sleep(0.1) |
| conn3.send_pingpong(0) |
| # send some packets on conn1, keep conn2/3 idle |
| for i in IterateWithSleep(self, 20, "Keep conn active", 0.2): |
| conn1.send_through(1) |
| try: |
| p2 = conn2.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| # We should have not received the packet on a long-idle |
| # connection, because it should have timed out |
| # If it didn't - it is a problem |
| self.assert_equal(p2, None, "packet on long-idle conn") |
| |
| def run_clear_conn_test(self, af, acl_side): |
| """Clear the connections via CLI""" |
| conn1 = Conn(self, self.pg0, self.pg1, af, UDP, 42001, 4242) |
| conn1.apply_acls(0, acl_side) |
| conn1.send_through(0) |
| # the return packets should pass |
| conn1.send_through(1) |
| # send some packets on conn1, ensure it doesn't go away |
| for i in IterateWithSleep(self, 20, "Keep conn active", 0.3): |
| conn1.send_through(1) |
| # clear all connections |
| self.vapi.ppcli("clear acl-plugin sessions") |
| # now try to send a packet on the reflected side |
| try: |
| p2 = conn1.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| self.assert_equal(p2, None, "packet on supposedly deleted conn") |
| |
| def run_tcp_transient_setup_conn_test(self, af, acl_side): |
| conn1 = Conn(self, self.pg0, self.pg1, af, TCP, 53001, 5151) |
| conn1.apply_acls(0, acl_side) |
| conn1.send_through(0, "S") |
| # the return packets should pass |
| conn1.send_through(1, "SA") |
| # allow the conn to time out |
| for i in IterateWithSleep(self, 30, "Wait for timeout", 0.1): |
| pass |
| # ensure conn times out |
| try: |
| p2 = conn1.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| self.assert_equal(p2, None, "packet on supposedly deleted conn") |
| |
| def run_tcp_established_conn_test(self, af, acl_side): |
| conn1 = Conn(self, self.pg0, self.pg1, af, TCP, 53002, 5052) |
| conn1.apply_acls(0, acl_side) |
| conn1.send_through(0, "S") |
| # the return packets should pass |
| conn1.send_through(1, "SA") |
| # complete the threeway handshake |
| # (NB: sequence numbers not tracked, so not set!) |
| conn1.send_through(0, "A") |
| # allow the conn to time out if it's in embryonic timer |
| for i in IterateWithSleep(self, 30, "Wait for transient timeout", 0.1): |
| pass |
| # Try to send the packet from the "forbidden" side - it must pass |
| conn1.send_through(1, "A") |
| # ensure conn times out for real |
| for i in IterateWithSleep(self, 130, "Wait for timeout", 0.1): |
| pass |
| try: |
| p2 = conn1.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| self.assert_equal(p2, None, "packet on supposedly deleted conn") |
| |
| def run_tcp_transient_teardown_conn_test(self, af, acl_side): |
| conn1 = Conn(self, self.pg0, self.pg1, af, TCP, 53002, 5052) |
| conn1.apply_acls(0, acl_side) |
| conn1.send_through(0, "S") |
| # the return packets should pass |
| conn1.send_through(1, "SA") |
| # complete the threeway handshake |
| # (NB: sequence numbers not tracked, so not set!) |
| conn1.send_through(0, "A") |
| # allow the conn to time out if it's in embryonic timer |
| for i in IterateWithSleep(self, 30, "Wait for transient timeout", 0.1): |
| pass |
| # Try to send the packet from the "forbidden" side - it must pass |
| conn1.send_through(1, "A") |
| # Send the FIN to bounce the session out of established |
| conn1.send_through(1, "FA") |
| # If conn landed on transient timer it will time out here |
| for i in IterateWithSleep(self, 30, "Wait for transient timeout", 0.1): |
| pass |
| # Now it should have timed out already |
| try: |
| p2 = conn1.send_through(1).command() |
| except: |
| # If we asserted while waiting, it's good. |
| # the conn should have timed out. |
| p2 = None |
| self.assert_equal(p2, None, "packet on supposedly deleted conn") |
| |
| def test_0000_conn_prepare_test(self): |
| """Prepare the settings""" |
| self.vapi.ppcli("set acl-plugin session timeout udp idle 1") |
| |
| def test_0001_basic_conn_test(self): |
| """IPv4: Basic conn timeout test reflect on ingress""" |
| self.run_basic_conn_test(AF_INET, 0) |
| |
| def test_0002_basic_conn_test(self): |
| """IPv4: Basic conn timeout test reflect on egress""" |
| self.run_basic_conn_test(AF_INET, 1) |
| |
| def test_0005_clear_conn_test(self): |
| """IPv4: reflect egress, clear conn""" |
| self.run_clear_conn_test(AF_INET, 1) |
| |
| def test_0006_clear_conn_test(self): |
| """IPv4: reflect ingress, clear conn""" |
| self.run_clear_conn_test(AF_INET, 0) |
| |
| def test_0011_active_conn_test(self): |
| """IPv4: Idle conn behind active conn, reflect on ingress""" |
| self.run_active_conn_test(AF_INET, 0) |
| |
| def test_0012_active_conn_test(self): |
| """IPv4: Idle conn behind active conn, reflect on egress""" |
| self.run_active_conn_test(AF_INET, 1) |
| |
| def test_1001_basic_conn_test(self): |
| """IPv6: Basic conn timeout test reflect on ingress""" |
| self.run_basic_conn_test(AF_INET6, 0) |
| |
| def test_1002_basic_conn_test(self): |
| """IPv6: Basic conn timeout test reflect on egress""" |
| self.run_basic_conn_test(AF_INET6, 1) |
| |
| def test_1005_clear_conn_test(self): |
| """IPv6: reflect egress, clear conn""" |
| self.run_clear_conn_test(AF_INET6, 1) |
| |
| def test_1006_clear_conn_test(self): |
| """IPv6: reflect ingress, clear conn""" |
| self.run_clear_conn_test(AF_INET6, 0) |
| |
| def test_1011_active_conn_test(self): |
| """IPv6: Idle conn behind active conn, reflect on ingress""" |
| self.run_active_conn_test(AF_INET6, 0) |
| |
| def test_1012_active_conn_test(self): |
| """IPv6: Idle conn behind active conn, reflect on egress""" |
| self.run_active_conn_test(AF_INET6, 1) |
| |
| def test_2000_prepare_for_tcp_test(self): |
| """Prepare for TCP session tests""" |
| # ensure the session hangs on if it gets treated as UDP |
| self.vapi.ppcli("set acl-plugin session timeout udp idle 200") |
| # let the TCP connection time out at 5 seconds |
| self.vapi.ppcli("set acl-plugin session timeout tcp idle 10") |
| self.vapi.ppcli("set acl-plugin session timeout tcp transient 1") |
| |
| def test_2001_tcp_transient_conn_test(self): |
| """IPv4: transient TCP session (incomplete 3WHS), ref. on ingress""" |
| self.run_tcp_transient_setup_conn_test(AF_INET, 0) |
| |
| def test_2002_tcp_transient_conn_test(self): |
| """IPv4: transient TCP session (incomplete 3WHS), ref. on egress""" |
| self.run_tcp_transient_setup_conn_test(AF_INET, 1) |
| |
| def test_2003_tcp_transient_conn_test(self): |
| """IPv4: established TCP session (complete 3WHS), ref. on ingress""" |
| self.run_tcp_established_conn_test(AF_INET, 0) |
| |
| def test_2004_tcp_transient_conn_test(self): |
| """IPv4: established TCP session (complete 3WHS), ref. on egress""" |
| self.run_tcp_established_conn_test(AF_INET, 1) |
| |
| def test_2005_tcp_transient_teardown_conn_test(self): |
| """IPv4: transient TCP session (3WHS,ACK,FINACK), ref. on ingress""" |
| self.run_tcp_transient_teardown_conn_test(AF_INET, 0) |
| |
| def test_2006_tcp_transient_teardown_conn_test(self): |
| """IPv4: transient TCP session (3WHS,ACK,FINACK), ref. on egress""" |
| self.run_tcp_transient_teardown_conn_test(AF_INET, 1) |
| |
| def test_3001_tcp_transient_conn_test(self): |
| """IPv6: transient TCP session (incomplete 3WHS), ref. on ingress""" |
| self.run_tcp_transient_setup_conn_test(AF_INET6, 0) |
| |
| def test_3002_tcp_transient_conn_test(self): |
| """IPv6: transient TCP session (incomplete 3WHS), ref. on egress""" |
| self.run_tcp_transient_setup_conn_test(AF_INET6, 1) |
| |
| def test_3003_tcp_transient_conn_test(self): |
| """IPv6: established TCP session (complete 3WHS), ref. on ingress""" |
| self.run_tcp_established_conn_test(AF_INET6, 0) |
| |
| def test_3004_tcp_transient_conn_test(self): |
| """IPv6: established TCP session (complete 3WHS), ref. on egress""" |
| self.run_tcp_established_conn_test(AF_INET6, 1) |
| |
| def test_3005_tcp_transient_teardown_conn_test(self): |
| """IPv6: transient TCP session (3WHS,ACK,FINACK), ref. on ingress""" |
| self.run_tcp_transient_teardown_conn_test(AF_INET6, 0) |
| |
| def test_3006_tcp_transient_teardown_conn_test(self): |
| """IPv6: transient TCP session (3WHS,ACK,FINACK), ref. on egress""" |
| self.run_tcp_transient_teardown_conn_test(AF_INET6, 1) |