ip: fix punt for ipv6

Type: fix

Change-Id: I583c30e9b63c0b0b6cd5fef0b2cb9ed7ec9856e2
Signed-off-by: Benoît Ganne <bganne@cisco.com>
This commit is contained in:
Benoît Ganne
2021-10-04 12:03:20 +02:00
parent bf37bf6f79
commit 7c7b505461
2 changed files with 153 additions and 4 deletions
+9 -4
View File
@@ -369,6 +369,8 @@ punt_l4_add_del (vlib_main_t * vm,
ip_address_family_t af,
ip_protocol_t protocol, u16 port, bool is_add)
{
int is_ip4 = af == AF_IP4;
/* For now we only support TCP and UDP punt */
if (protocol != IP_PROTOCOL_UDP && protocol != IP_PROTOCOL_TCP)
return clib_error_return (0,
@@ -378,19 +380,22 @@ punt_l4_add_del (vlib_main_t * vm,
if (port == (u16) ~ 0)
{
if (protocol == IP_PROTOCOL_UDP)
udp_punt_unknown (vm, af == AF_IP4, is_add);
udp_punt_unknown (vm, is_ip4, is_add);
else if (protocol == IP_PROTOCOL_TCP)
tcp_punt_unknown (vm, af == AF_IP4, is_add);
tcp_punt_unknown (vm, is_ip4, is_add);
return 0;
}
else if (is_add)
{
const vlib_node_registration_t *punt_node =
is_ip4 ? &udp4_punt_node : &udp6_punt_node;
if (protocol == IP_PROTOCOL_TCP)
return clib_error_return (0, "punt TCP ports is not supported yet");
udp_register_dst_port (vm, port, udp4_punt_node.index, af == AF_IP4);
udp_register_dst_port (vm, port, punt_node->index, is_ip4);
return 0;
}
@@ -399,7 +404,7 @@ punt_l4_add_del (vlib_main_t * vm,
if (protocol == IP_PROTOCOL_TCP)
return clib_error_return (0, "punt TCP ports is not supported yet");
udp_unregister_dst_port (vm, port, af == AF_IP4);
udp_unregister_dst_port (vm, port, is_ip4);
return 0;
}
+144
View File
@@ -3597,5 +3597,149 @@ class TestIPxAF(VppTestCase):
self.assertEqual(rx[IPv6].dst, "3001::2")
class TestIPv6Punt(VppTestCase):
""" IPv6 Punt Police/Redirect """
def setUp(self):
super(TestIPv6Punt, self).setUp()
self.create_pg_interfaces(range(4))
for i in self.pg_interfaces:
i.admin_up()
i.config_ip6()
i.resolve_ndp()
def tearDown(self):
super(TestIPv6Punt, self).tearDown()
for i in self.pg_interfaces:
i.unconfig_ip6()
i.admin_down()
def test_ip6_punt(self):
""" IPv6 punt police and redirect """
# use UDP packet that have a port we need to explicitly
# register to get punted.
pt_l4 = VppEnum.vl_api_punt_type_t.PUNT_API_TYPE_L4
af_ip6 = VppEnum.vl_api_address_family_t.ADDRESS_IP6
udp_proto = VppEnum.vl_api_ip_proto_t.IP_API_PROTO_UDP
punt_udp = {
'type': pt_l4,
'punt': {
'l4': {
'af': af_ip6,
'protocol': udp_proto,
'port': 7654,
}
}
}
self.vapi.set_punt(is_add=1, punt=punt_udp)
pkts = (Ether(src=self.pg0.remote_mac,
dst=self.pg0.local_mac) /
IPv6(src=self.pg0.remote_ip6, dst=self.pg0.local_ip6) /
UDP(sport=1234, dport=7654) /
Raw(b'\xa5' * 100)) * 1025
#
# Configure a punt redirect via pg1.
#
nh_addr = self.pg1.remote_ip6
ip_punt_redirect = VppIpPuntRedirect(self, self.pg0.sw_if_index,
self.pg1.sw_if_index, nh_addr)
ip_punt_redirect.add_vpp_config()
self.send_and_expect(self.pg0, pkts, self.pg1)
#
# add a policer
#
policer = VppPolicer(self, "ip6-punt", 400, 0, 10, 0, rate_type=1)
policer.add_vpp_config()
ip_punt_policer = VppIpPuntPolicer(self, policer.policer_index,
is_ip6=True)
ip_punt_policer.add_vpp_config()
self.vapi.cli("clear trace")
self.pg0.add_stream(pkts)
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
#
# the number of packet received should be greater than 0,
# but not equal to the number sent, since some were policed
#
rx = self.pg1._get_capture(1)
stats = policer.get_stats()
# Single rate policer - expect conform, violate but no exceed
self.assertGreater(stats['conform_packets'], 0)
self.assertEqual(stats['exceed_packets'], 0)
self.assertGreater(stats['violate_packets'], 0)
self.assertGreater(len(rx), 0)
self.assertLess(len(rx), len(pkts))
#
# remove the policer. back to full rx
#
ip_punt_policer.remove_vpp_config()
policer.remove_vpp_config()
self.send_and_expect(self.pg0, pkts, self.pg1)
#
# remove the redirect. expect full drop.
#
ip_punt_redirect.remove_vpp_config()
self.send_and_assert_no_replies(self.pg0, pkts,
"IP no punt config")
#
# Add a redirect that is not input port selective
#
ip_punt_redirect = VppIpPuntRedirect(self, 0xffffffff,
self.pg1.sw_if_index, nh_addr)
ip_punt_redirect.add_vpp_config()
self.send_and_expect(self.pg0, pkts, self.pg1)
ip_punt_redirect.remove_vpp_config()
def test_ip6_punt_dump(self):
""" IPv6 punt redirect dump"""
#
# Configure a punt redirects
#
nh_address = self.pg3.remote_ip6
ipr_03 = VppIpPuntRedirect(self, self.pg0.sw_if_index,
self.pg3.sw_if_index, nh_address)
ipr_13 = VppIpPuntRedirect(self, self.pg1.sw_if_index,
self.pg3.sw_if_index, nh_address)
ipr_23 = VppIpPuntRedirect(self, self.pg2.sw_if_index,
self.pg3.sw_if_index, "::")
ipr_03.add_vpp_config()
ipr_13.add_vpp_config()
ipr_23.add_vpp_config()
#
# Dump pg0 punt redirects
#
self.assertTrue(ipr_03.query_vpp_config())
self.assertTrue(ipr_13.query_vpp_config())
self.assertTrue(ipr_23.query_vpp_config())
#
# Dump punt redirects for all interfaces
#
punts = self.vapi.ip_punt_redirect_dump(sw_if_index=0xffffffff,
is_ipv6=True)
self.assertEqual(len(punts), 3)
for p in punts:
self.assertEqual(p.punt.tx_sw_if_index, self.pg3.sw_if_index)
self.assertNotEqual(punts[1].punt.nh, self.pg3.remote_ip6)
self.assertEqual(str(punts[2].punt.nh), '::')
if __name__ == '__main__':
unittest.main(testRunner=VppTestRunner)