From 9bc72ac8de6dcc74aa82fa6c3223e9f93b2dc3c2 Mon Sep 17 00:00:00 2001 From: Florin Coras Date: Mon, 9 Jan 2023 12:46:07 -0800 Subject: udp: avoid listener cleanups with active opens Type: fix Signed-off-by: Florin Coras Change-Id: Ibff9f32e4fcaf0344207d8e43f3547180cbd4eef --- src/vnet/udp/udp.c | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'src/vnet/udp/udp.c') diff --git a/src/vnet/udp/udp.c b/src/vnet/udp/udp.c index e095a66dcf2..4553701dced 100644 --- a/src/vnet/udp/udp.c +++ b/src/vnet/udp/udp.c @@ -38,6 +38,9 @@ udp_connection_register_port (u16 lcl_port, u8 is_ip4) n = sparse_vec_validate (um->next_by_dst_port6, lcl_port); n[0] = um->local_to_input_edge[is_ip4]; + + __atomic_add_fetch (&um->transport_ports_refcnt[is_ip4][lcl_port], 1, + __ATOMIC_RELAXED); } static void @@ -46,6 +49,11 @@ udp_connection_unregister_port (u16 lcl_port, u8 is_ip4) udp_main_t *um = &udp_main; u16 *n; + /* Needed because listeners are not tracked as local endpoints */ + if (__atomic_sub_fetch (&um->transport_ports_refcnt[is_ip4][lcl_port], 1, + __ATOMIC_RELAXED)) + return; + if (is_ip4) n = sparse_vec_validate (um->next_by_dst_port4, lcl_port); else @@ -548,6 +556,9 @@ udp_enable_disable (vlib_main_t *vm, u8 is_en) udp_realloc_ports_sv (&um->next_by_dst_port4); udp_realloc_ports_sv (&um->next_by_dst_port6); + vec_validate (um->transport_ports_refcnt[0], 65535); + vec_validate (um->transport_ports_refcnt[1], 65535); + return 0; } -- cgit 1.2.3-korg