From 40903ac34f89d9e2ad775e98b7bcec5b7feb0207 Mon Sep 17 00:00:00 2001 From: Florin Coras Date: Sun, 10 Jun 2018 14:41:23 -0700 Subject: udp: fix for multiple workers and add test Since the main thread is not used for session polling anymore, when vpp is started with multiple wokers, allocate connections on the first. Also add a simple udp make test. Change-Id: Id869f5d89e0fced51048f0384fa86a5022258b7c Signed-off-by: Florin Coras --- src/vnet/session-apps/echo_client.c | 5 +++-- src/vnet/session/session.c | 3 ++- src/vnet/session/transport.c | 6 ++++++ src/vnet/session/transport.h | 1 + src/vnet/udp/udp.c | 16 ++++++++++++---- 5 files changed, 24 insertions(+), 7 deletions(-) (limited to 'src/vnet') diff --git a/src/vnet/session-apps/echo_client.c b/src/vnet/session-apps/echo_client.c index 1d010f23d4d..6ee91f9c54b 100644 --- a/src/vnet/session-apps/echo_client.c +++ b/src/vnet/session-apps/echo_client.c @@ -369,7 +369,7 @@ echo_clients_session_connected_callback (u32 app_index, u32 api_context, echo_client_main_t *ecm = &echo_client_main; eclient_session_t *session; u32 session_index; - u8 thread_index = vlib_get_thread_index (); + u8 thread_index = s->thread_index; if (is_fail) { @@ -378,7 +378,8 @@ echo_clients_session_connected_callback (u32 app_index, u32 api_context, return 0; } - ASSERT (s->thread_index == thread_index); + ASSERT (thread_index == vlib_get_thread_index () + || session_transport_service_type (s) == TRANSPORT_SERVICE_CL); if (!ecm->vpp_event_queue[thread_index]) ecm->vpp_event_queue[thread_index] = diff --git a/src/vnet/session/session.c b/src/vnet/session/session.c index ee02526c190..cfba31ec406 100644 --- a/src/vnet/session/session.c +++ b/src/vnet/session/session.c @@ -142,7 +142,8 @@ session_alloc_for_connection (transport_connection_t * tc) stream_session_t *s; u32 thread_index = tc->thread_index; - ASSERT (thread_index == vlib_get_thread_index ()); + ASSERT (thread_index == vlib_get_thread_index () + || transport_protocol_is_cl (tc->proto)); s = session_alloc (thread_index); s->session_type = session_type_from_proto_and_ip (tc->proto, tc->is_ip4); diff --git a/src/vnet/session/transport.c b/src/vnet/session/transport.c index b3d42d0f7da..a401190e426 100644 --- a/src/vnet/session/transport.c +++ b/src/vnet/session/transport.c @@ -207,6 +207,12 @@ transport_protocol_tx_fn_type (transport_proto_t tp) return tp_vfts[tp].tx_type; } +u8 +transport_protocol_is_cl (transport_proto_t tp) +{ + return (tp_vfts[tp].service_type == TRANSPORT_SERVICE_CL); +} + #define PORT_MASK ((1 << 16)- 1) void diff --git a/src/vnet/session/transport.h b/src/vnet/session/transport.h index 8340fd859ac..e29f3ca9557 100644 --- a/src/vnet/session/transport.h +++ b/src/vnet/session/transport.h @@ -131,6 +131,7 @@ int transport_alloc_local_endpoint (u8 proto, transport_endpoint_t * rmt, ip46_address_t * lcl_addr, u16 * lcl_port); void transport_endpoint_cleanup (u8 proto, ip46_address_t * lcl_ip, u16 port); +u8 transport_protocol_is_cl (transport_proto_t tp); void transport_init (void); #endif /* VNET_VNET_URI_TRANSPORT_H_ */ diff --git a/src/vnet/udp/udp.c b/src/vnet/udp/udp.c index 947cc1e38e1..888a6cb844b 100644 --- a/src/vnet/udp/udp.c +++ b/src/vnet/udp/udp.c @@ -288,6 +288,10 @@ udp_open_connection (transport_endpoint_t * rmt) node_index = rmt->is_ip4 ? udp4_input_node.index : udp6_input_node.index; udp_register_dst_port (vm, lcl_port, node_index, 1 /* is_ipv4 */ ); + /* We don't poll main thread if we have workers */ + if (vlib_num_workers ()) + thread_index = 1; + uc = udp_connection_alloc (thread_index); ip_copy (&uc->c_rmt_ip, &rmt->ip, rmt->is_ip4); ip_copy (&uc->c_lcl_ip, &lcl_addr, rmt->is_ip4); @@ -301,10 +305,14 @@ udp_open_connection (transport_endpoint_t * rmt) } transport_connection_t * -udp_half_open_session_get_transport (u32 conn_index) +udp_session_get_half_open (u32 conn_index) { udp_connection_t *uc; - uc = udp_connection_get (conn_index, vlib_get_thread_index ()); + u32 thread_index; + + /* We don't poll main thread if we have workers */ + thread_index = vlib_num_workers ()? 1 : 0; + uc = udp_connection_get (conn_index, thread_index); return &uc->connection; } @@ -316,7 +324,7 @@ const static transport_proto_vft_t udp_proto = { .push_header = udp_push_header, .get_connection = udp_session_get, .get_listener = udp_session_get_listener, - .get_half_open = udp_half_open_session_get_transport, + .get_half_open = udp_session_get_half_open, .close = udp_session_close, .cleanup = udp_session_cleanup, .send_mss = udp_send_mss, @@ -360,7 +368,7 @@ const static transport_proto_vft_t udpc_proto = { .push_header = udp_push_header, .get_connection = udp_session_get, .get_listener = udp_session_get_listener, - .get_half_open = udp_half_open_session_get_transport, + .get_half_open = udp_session_get_half_open, .close = udp_session_close, .cleanup = udp_session_cleanup, .send_mss = udp_send_mss, -- cgit 1.2.3-korg