vpp/test/test_sixrd.py
Klement Sekera d9b0c6fbf7 tests: replace pycodestyle with black
Drop pycodestyle for code style checking in favor of black. Black is
much faster, stable PEP8 compliant code style checker offering also
automatic formatting. It aims to be very stable and produce smallest
diffs. It's used by many small and big projects.

Running checkstyle with black takes a few seconds with a terse output.
Thus, test-checkstyle-diff is no longer necessary.

Expand scope of checkstyle to all python files in the repo, replacing
test-checkstyle with checkstyle-python.

Also, fixstyle-python is now available for automatic style formatting.

Note: python virtualenv has been consolidated in test/Makefile,
test/requirements*.txt which will eventually be moved to a central
location.  This is required to simply the automated generation of
docker executor images in the CI.

Type: improvement
Change-Id: I022a326603485f58585e879ac0f697fceefbc9c8
Signed-off-by: Klement Sekera <klement.sekera@gmail.com>
Signed-off-by: Dave Wallace <dwallacelf@gmail.com>
2022-05-10 18:52:08 +00:00

424 lines
13 KiB
Python

#!/usr/bin/env python3
""" 6RD RFC5969 functional tests """
import unittest
from scapy.layers.inet import IP, UDP, Ether
from scapy.layers.inet6 import IPv6
from scapy.packet import Raw
from framework import VppTestCase, VppTestRunner
from vpp_ip import DpoProto
from vpp_ip_route import VppIpRoute, VppRoutePath, VppIpTable, FibPathProto
from socket import AF_INET, AF_INET6, inet_pton
""" Test6rd is a subclass of VPPTestCase classes.
6RD tests.
"""
class Test6RD(VppTestCase):
"""6RD Test Case"""
@classmethod
def setUpClass(cls):
super(Test6RD, cls).setUpClass()
cls.create_pg_interfaces(range(4))
cls.interfaces = list(cls.pg_interfaces)
@classmethod
def tearDownClass(cls):
super(Test6RD, cls).tearDownClass()
def setUp(self):
super(Test6RD, self).setUp()
t4 = VppIpTable(self, 10)
t6 = VppIpTable(self, 20, True)
t4.add_vpp_config()
t6.add_vpp_config()
for n in range(4):
i = self.pg_interfaces[n]
i.admin_up()
if n > 1:
i.set_table_ip4(10)
i.set_table_ip6(20)
i.config_ip4()
i.config_ip6()
i.disable_ipv6_ra()
i.resolve_arp()
i.resolve_ndp()
def tearDown(self):
for i in self.pg_interfaces:
i.unconfig_ip4()
i.unconfig_ip6()
i.set_table_ip4(0)
i.set_table_ip6(0)
i.admin_down()
super(Test6RD, self).tearDown()
def validate_6in4(self, rx, expected):
if IP not in rx:
self.fail()
if IPv6 not in rx:
self.fail()
self.assertEqual(rx[IP].src, expected[IP].src)
self.assertEqual(rx[IP].dst, expected[IP].dst)
self.assertEqual(rx[IP].proto, expected[IP].proto)
self.assertEqual(rx[IPv6].src, expected[IPv6].src)
self.assertEqual(rx[IPv6].dst, expected[IPv6].dst)
def validate_4in6(self, rx, expected):
if IPv6 not in rx:
self.fail()
if IP in rx:
self.fail()
self.assertEqual(rx[IPv6].src, expected[IPv6].src)
self.assertEqual(rx[IPv6].dst, expected[IPv6].dst)
self.assertEqual(rx[IPv6].nh, expected[IPv6].nh)
def payload(self, len):
return "x" * len
def test_6rd_ip6_to_ip4(self):
"""ip6 -> ip4 (encap) 6rd test"""
p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
p_ip6 = IPv6(src="1::1", dst="2002:AC10:0202::1", nh="UDP")
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
self.vapi.cli("show ip6 fib")
p_payload = UDP(sport=1234, dport=1234)
p = p_ether / p_ip6 / p_payload
p_reply = (
IP(src=self.pg0.local_ip4, dst=self.pg1.remote_ip4, proto="ipv6") / p_ip6
)
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_6in4(p, p_reply)
# MTU tests (default is 1480)
plen = 1481 - 40 - 8
p_ip6 = IPv6(src="1::1", dst="2002:AC10:0202::1")
p_payload = UDP(sport=1234, dport=1234) / Raw(self.payload(plen))
p = p_ether / p_ip6 / p_payload
p_reply = (
IP(src=self.pg0.local_ip4, dst=self.pg1.remote_ip4, proto="ipv6") / p_ip6
)
rx = self.send_and_assert_no_replies(self.pg0, p * 10)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_ip6_to_ip4_vrf(self):
"""ip6 -> ip4 (encap) 6rd VRF test"""
p_ether = Ether(src=self.pg2.remote_mac, dst=self.pg2.local_mac)
p_ip6 = IPv6(src="1::1", dst="2002:AC10:0402::1", nh="UDP")
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=20,
ip4_table_id=10,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg2.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
self.vapi.cli("show ip6 fib")
p_payload = UDP(sport=1234, dport=1234)
p = p_ether / p_ip6 / p_payload
p_reply = (
IP(src=self.pg2.local_ip4, dst=self.pg3.remote_ip4, proto="ipv6") / p_ip6
)
rx = self.send_and_expect(self.pg2, p * 10, self.pg3)
for p in rx:
self.validate_6in4(p, p_reply)
# MTU tests (default is 1480)
plen = 1481 - 40 - 8
p_ip6 = IPv6(src="1::1", dst="2002:AC10:0402::1")
p_payload = UDP(sport=1234, dport=1234) / Raw(self.payload(plen))
p = p_ether / p_ip6 / p_payload
p_reply = (
IP(src=self.pg2.local_ip4, dst=self.pg3.remote_ip4, proto="ipv6") / p_ip6
)
rx = self.send_and_assert_no_replies(self.pg0, p * 10)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_ip4_to_ip6(self):
"""ip4 -> ip6 (decap) 6rd test"""
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
rv = self.vapi.ipip_6rd_del_tunnel(rv.sw_if_index)
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
p_ip6 = IPv6(src="2002:AC10:0202::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = (
Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
/ IP(src=self.pg1.remote_ip4, dst=self.pg0.local_ip4)
/ p_ip6
)
p_reply = p_ip6
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_reply)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_ip4_to_ip6_vrf(self):
"""ip4 -> ip6 (decap) 6rd VRF test"""
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=20,
ip4_table_id=10,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg2.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
rv = self.vapi.ipip_6rd_del_tunnel(rv.sw_if_index)
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=20,
ip4_table_id=10,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg2.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
self.vapi.sw_interface_set_table(self.tunnel_index, 1, 20)
p_ip6 = IPv6(src="2002:AC10:0402::1", dst=self.pg3.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = (
Ether(src=self.pg2.remote_mac, dst=self.pg2.local_mac)
/ IP(src=self.pg3.remote_ip4, dst=self.pg2.local_ip4)
/ p_ip6
)
p_reply = p_ip6
rx = self.send_and_expect(self.pg2, p * 10, self.pg3)
for p in rx:
self.validate_4in6(p, p_reply)
self.vapi.sw_interface_set_table(self.tunnel_index, 1, 0)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_ip4_to_ip6_multiple(self):
"""ip4 -> ip6 (decap) 6rd test"""
self.tunnel_index = []
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index.append(rv.sw_if_index)
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2003::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg1.local_ip4,
security_check=True,
)
self.tunnel_index.append(rv.sw_if_index)
self.vapi.cli("show ip6 fib")
p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
p_ip4 = IP(src=self.pg1.remote_ip4, dst=self.pg0.local_ip4)
p_ip6_1 = IPv6(src="2002:AC10:0202::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p_ip6_2 = IPv6(src="2003:AC10:0202::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = p_ether / p_ip4 / p_ip6_1
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_ip6_1)
p = p_ether / p_ip4 / p_ip6_2
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_ip6_2)
for i in self.tunnel_index:
self.vapi.ipip_6rd_del_tunnel(i)
def test_6rd_ip4_to_ip6_suffix(self):
"""ip4 -> ip6 (decap) 6rd test"""
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="172.0.0.0/8",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
self.vapi.cli("show ip6 fib")
p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
p_ip4 = IP(src=self.pg1.remote_ip4, dst=self.pg0.local_ip4)
p_ip6 = IPv6(src="2002:1002:0200::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = p_ether / p_ip4 / p_ip6
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_ip6)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_ip4_to_ip6_sec_check(self):
"""ip4 -> ip6 (decap) security check 6rd test"""
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=True,
)
self.tunnel_index = rv.sw_if_index
self.vapi.cli("show ip6 fib")
p_ip6 = IPv6(src="2002:AC10:0202::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p_ip6_fail = IPv6(src="2002:DEAD:0202::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = (
Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
/ IP(src=self.pg1.remote_ip4, dst=self.pg0.local_ip4)
/ p_ip6
)
p_reply = p_ip6
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_reply)
p = (
Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
/ IP(src=self.pg1.remote_ip4, dst=self.pg0.local_ip4)
/ p_ip6_fail
)
rx = self.send_and_assert_no_replies(self.pg0, p * 10)
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
def test_6rd_bgp_tunnel(self):
"""6rd BGP tunnel"""
rv = self.vapi.ipip_6rd_add_tunnel(
ip6_table_id=0,
ip4_table_id=0,
ip6_prefix="2002::/16",
ip4_prefix="0.0.0.0/0",
ip4_src=self.pg0.local_ip4,
security_check=False,
)
self.tunnel_index = rv.sw_if_index
default_route = VppIpRoute(
self, "DEAD::", 16, [VppRoutePath("2002:0808:0808::", self.tunnel_index)]
)
default_route.add_vpp_config()
ip4_route = VppIpRoute(
self, "8.0.0.0", 8, [VppRoutePath(self.pg1.remote_ip4, 0xFFFFFFFF)]
)
ip4_route.add_vpp_config()
# Via recursive route 6 -> 4
p = (
Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
/ IPv6(src="1::1", dst="DEAD:BEEF::1")
/ UDP(sport=1234, dport=1234)
)
p_reply = IP(src=self.pg0.local_ip4, dst="8.8.8.8", proto="ipv6") / IPv6(
src="1::1", dst="DEAD:BEEF::1", nh="UDP"
)
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_6in4(p, p_reply)
# Via recursive route 4 -> 6 (Security check must be disabled)
p_ip6 = IPv6(src="DEAD:BEEF::1", dst=self.pg1.remote_ip6) / UDP(
sport=1234, dport=1234
)
p = (
Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
/ IP(src="8.8.8.8", dst=self.pg0.local_ip4)
/ p_ip6
)
p_reply = p_ip6
rx = self.send_and_expect(self.pg0, p * 10, self.pg1)
for p in rx:
self.validate_4in6(p, p_reply)
ip4_route.remove_vpp_config()
default_route.remove_vpp_config()
self.vapi.ipip_6rd_del_tunnel(self.tunnel_index)
if __name__ == "__main__":
unittest.main(testRunner=VppTestRunner)