blob: 71578b076d213bbff77999efa92a7e44971954ff [file] [log] [blame]
Dave Wallace543852a2017-08-03 02:11:34 -04001/*
Florin Coras5e062572019-03-14 19:07:51 -07002 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
Dave Wallace543852a2017-08-03 02:11:34 -04003 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this
5 * You may obtain a copy of the License at:
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#include <stdio.h>
17#include <stdlib.h>
Dave Wallace5c7cf1c2017-10-24 04:12:18 -040018#include <vcl/vppcom.h>
Florin Coras0d427d82018-06-27 03:24:07 -070019#include <vcl/vcl_debug.h>
Florin Coras697faea2018-06-27 17:10:49 -070020#include <vcl/vcl_private.h>
Florin Coras88001c62019-04-24 14:44:46 -070021#include <svm/fifo_segment.h>
Dave Wallace7e607a72018-06-18 18:41:32 -040022
Florin Coras134a9962018-08-28 11:32:04 -070023__thread uword __vcl_worker_index = ~0;
24
Florin Corasd85de682018-11-29 17:02:29 -080025static int
Florin Corasc4c4cf52019-08-24 18:17:34 -070026vcl_segment_is_not_mounted (vcl_worker_t * wrk, u64 segment_handle)
Florin Coras54693d22018-07-17 10:46:29 -070027{
Florin Corasc4c4cf52019-08-24 18:17:34 -070028 u32 segment_index;
Florin Coras54693d22018-07-17 10:46:29 -070029
Florin Corasd85de682018-11-29 17:02:29 -080030 if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
Florin Coras5f45e012019-01-23 09:21:30 -080031 return 0;
Florin Coras54693d22018-07-17 10:46:29 -070032
Florin Corasc4c4cf52019-08-24 18:17:34 -070033 segment_index = vcl_segment_table_lookup (segment_handle);
34 if (segment_index != VCL_INVALID_SEGMENT_INDEX)
35 return 0;
36
Florin Corasd85de682018-11-29 17:02:29 -080037 return 1;
Florin Coras54693d22018-07-17 10:46:29 -070038}
39
Florin Coras30e79c22019-01-02 19:31:22 -080040static inline int
Florin Corase003a1b2019-06-05 10:47:16 -070041vcl_mq_dequeue_batch (vcl_worker_t * wrk, svm_msg_q_t * mq, u32 n_max_msg)
Florin Coras30e79c22019-01-02 19:31:22 -080042{
43 svm_msg_q_msg_t *msg;
44 u32 n_msgs;
45 int i;
46
Florin Corase003a1b2019-06-05 10:47:16 -070047 n_msgs = clib_min (svm_msg_q_size (mq), n_max_msg);
Florin Coras30e79c22019-01-02 19:31:22 -080048 for (i = 0; i < n_msgs; i++)
49 {
50 vec_add2 (wrk->mq_msg_vector, msg, 1);
51 svm_msg_q_sub_w_lock (mq, msg);
52 }
53 return n_msgs;
54}
55
Florin Coras697faea2018-06-27 17:10:49 -070056const char *
Florin Coras288eaab2019-02-03 15:26:14 -080057vppcom_session_state_str (vcl_session_state_t state)
Dave Wallace543852a2017-08-03 02:11:34 -040058{
59 char *st;
60
61 switch (state)
62 {
Florin Coras54140622020-02-04 19:04:34 +000063 case STATE_CLOSED:
64 st = "STATE_CLOSED";
Dave Wallace543852a2017-08-03 02:11:34 -040065 break;
66
67 case STATE_CONNECT:
68 st = "STATE_CONNECT";
69 break;
70
71 case STATE_LISTEN:
72 st = "STATE_LISTEN";
73 break;
74
75 case STATE_ACCEPT:
76 st = "STATE_ACCEPT";
77 break;
78
Florin Coras3c7d4f92018-12-14 11:28:43 -080079 case STATE_VPP_CLOSING:
80 st = "STATE_VPP_CLOSING";
Dave Wallace4878cbe2017-11-21 03:45:09 -050081 break;
82
Dave Wallace543852a2017-08-03 02:11:34 -040083 case STATE_DISCONNECT:
84 st = "STATE_DISCONNECT";
85 break;
86
Florin Corasadcfb152020-02-12 08:50:29 +000087 case STATE_DETACHED:
88 st = "STATE_DETACHED";
Dave Wallace543852a2017-08-03 02:11:34 -040089 break;
90
Florin Coras2d675d72019-01-28 15:54:27 -080091 case STATE_UPDATED:
92 st = "STATE_UPDATED";
93 break;
94
95 case STATE_LISTEN_NO_MQ:
96 st = "STATE_LISTEN_NO_MQ";
97 break;
98
Dave Wallace543852a2017-08-03 02:11:34 -040099 default:
100 st = "UNKNOWN_STATE";
101 break;
102 }
103
104 return st;
105}
106
Dave Wallace543852a2017-08-03 02:11:34 -0400107u8 *
108format_ip4_address (u8 * s, va_list * args)
109{
110 u8 *a = va_arg (*args, u8 *);
111 return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
112}
113
114u8 *
115format_ip6_address (u8 * s, va_list * args)
116{
117 ip6_address_t *a = va_arg (*args, ip6_address_t *);
118 u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
119
120 i_max_n_zero = ARRAY_LEN (a->as_u16);
121 max_n_zeros = 0;
122 i_first_zero = i_max_n_zero;
123 n_zeros = 0;
124 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
125 {
126 u32 is_zero = a->as_u16[i] == 0;
127 if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
128 {
129 i_first_zero = i;
130 n_zeros = 0;
131 }
132 n_zeros += is_zero;
133 if ((!is_zero && n_zeros > max_n_zeros)
134 || (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
135 {
136 i_max_n_zero = i_first_zero;
137 max_n_zeros = n_zeros;
138 i_first_zero = ARRAY_LEN (a->as_u16);
139 n_zeros = 0;
140 }
141 }
142
143 last_double_colon = 0;
144 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
145 {
146 if (i == i_max_n_zero && max_n_zeros > 1)
147 {
148 s = format (s, "::");
149 i += max_n_zeros - 1;
150 last_double_colon = 1;
151 }
152 else
153 {
154 s = format (s, "%s%x",
155 (last_double_colon || i == 0) ? "" : ":",
156 clib_net_to_host_u16 (a->as_u16[i]));
157 last_double_colon = 0;
158 }
159 }
160
161 return s;
162}
163
164/* Format an IP46 address. */
165u8 *
166format_ip46_address (u8 * s, va_list * args)
167{
168 ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
169 ip46_type_t type = va_arg (*args, ip46_type_t);
170 int is_ip4 = 1;
171
172 switch (type)
173 {
174 case IP46_TYPE_ANY:
175 is_ip4 = ip46_address_is_ip4 (ip46);
176 break;
177 case IP46_TYPE_IP4:
178 is_ip4 = 1;
179 break;
180 case IP46_TYPE_IP6:
181 is_ip4 = 0;
182 break;
183 }
184
185 return is_ip4 ?
186 format (s, "%U", format_ip4_address, &ip46->ip4) :
187 format (s, "%U", format_ip6_address, &ip46->ip6);
188}
189
Florin Coras697faea2018-06-27 17:10:49 -0700190/*
191 * VPPCOM Utility Functions
192 */
193
Florin Coras458089b2019-08-21 16:20:44 -0700194static void
195vcl_send_session_listen (vcl_worker_t * wrk, vcl_session_t * s)
196{
197 app_session_evt_t _app_evt, *app_evt = &_app_evt;
198 session_listen_msg_t *mp;
199 svm_msg_q_t *mq;
200
201 mq = vcl_worker_ctrl_mq (wrk);
202 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_LISTEN);
203 mp = (session_listen_msg_t *) app_evt->evt->data;
204 memset (mp, 0, sizeof (*mp));
205 mp->client_index = wrk->my_client_index;
206 mp->context = s->session_index;
207 mp->wrk_index = wrk->vpp_wrk_index;
208 mp->is_ip4 = s->transport.is_ip4;
209 clib_memcpy_fast (&mp->ip, &s->transport.lcl_ip, sizeof (mp->ip));
210 mp->port = s->transport.lcl_port;
211 mp->proto = s->session_type;
Florin Coras1e966172020-05-16 18:18:14 +0000212 if (s->flags & VCL_SESSION_F_CONNECTED)
213 mp->flags = TRANSPORT_CFG_F_CONNECTED;
Florin Coras458089b2019-08-21 16:20:44 -0700214 app_send_ctrl_evt_to_vpp (mq, app_evt);
215}
216
217static void
218vcl_send_session_connect (vcl_worker_t * wrk, vcl_session_t * s)
219{
220 app_session_evt_t _app_evt, *app_evt = &_app_evt;
221 session_connect_msg_t *mp;
222 svm_msg_q_t *mq;
223
224 mq = vcl_worker_ctrl_mq (wrk);
225 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_CONNECT);
226 mp = (session_connect_msg_t *) app_evt->evt->data;
227 memset (mp, 0, sizeof (*mp));
228 mp->client_index = wrk->my_client_index;
229 mp->context = s->session_index;
230 mp->wrk_index = wrk->vpp_wrk_index;
231 mp->is_ip4 = s->transport.is_ip4;
232 mp->parent_handle = s->parent_handle;
233 clib_memcpy_fast (&mp->ip, &s->transport.rmt_ip, sizeof (mp->ip));
Florin Corasef7cbf62019-10-17 09:56:27 -0700234 clib_memcpy_fast (&mp->lcl_ip, &s->transport.lcl_ip, sizeof (mp->lcl_ip));
Florin Coras458089b2019-08-21 16:20:44 -0700235 mp->port = s->transport.rmt_port;
Florin Coras0a1e1832020-03-29 18:54:04 +0000236 mp->lcl_port = s->transport.lcl_port;
Florin Coras458089b2019-08-21 16:20:44 -0700237 mp->proto = s->session_type;
Florin Coras0a1e1832020-03-29 18:54:04 +0000238 if (s->flags & VCL_SESSION_F_CONNECTED)
239 mp->flags |= TRANSPORT_CFG_F_CONNECTED;
Florin Coras458089b2019-08-21 16:20:44 -0700240 app_send_ctrl_evt_to_vpp (mq, app_evt);
241}
242
243void
244vcl_send_session_unlisten (vcl_worker_t * wrk, vcl_session_t * s)
245{
246 app_session_evt_t _app_evt, *app_evt = &_app_evt;
247 session_unlisten_msg_t *mp;
248 svm_msg_q_t *mq;
249
250 mq = vcl_worker_ctrl_mq (wrk);
251 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_UNLISTEN);
252 mp = (session_unlisten_msg_t *) app_evt->evt->data;
253 memset (mp, 0, sizeof (*mp));
254 mp->client_index = wrk->my_client_index;
255 mp->wrk_index = wrk->vpp_wrk_index;
256 mp->handle = s->vpp_handle;
257 mp->context = wrk->wrk_index;
258 app_send_ctrl_evt_to_vpp (mq, app_evt);
259}
260
261static void
262vcl_send_session_disconnect (vcl_worker_t * wrk, vcl_session_t * s)
263{
264 app_session_evt_t _app_evt, *app_evt = &_app_evt;
265 session_disconnect_msg_t *mp;
266 svm_msg_q_t *mq;
267
268 /* Send to thread that owns the session */
269 mq = s->vpp_evt_q;
270 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_DISCONNECT);
271 mp = (session_disconnect_msg_t *) app_evt->evt->data;
272 memset (mp, 0, sizeof (*mp));
273 mp->client_index = wrk->my_client_index;
274 mp->handle = s->vpp_handle;
275 app_send_ctrl_evt_to_vpp (mq, app_evt);
276}
277
278static void
279vcl_send_app_detach (vcl_worker_t * wrk)
280{
281 app_session_evt_t _app_evt, *app_evt = &_app_evt;
282 session_app_detach_msg_t *mp;
283 svm_msg_q_t *mq;
284
285 mq = vcl_worker_ctrl_mq (wrk);
286 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_DETACH);
287 mp = (session_app_detach_msg_t *) app_evt->evt->data;
288 memset (mp, 0, sizeof (*mp));
289 mp->client_index = wrk->my_client_index;
290 app_send_ctrl_evt_to_vpp (mq, app_evt);
291}
Florin Coras697faea2018-06-27 17:10:49 -0700292
Florin Coras54693d22018-07-17 10:46:29 -0700293static void
294vcl_send_session_accepted_reply (svm_msg_q_t * mq, u32 context,
295 session_handle_t handle, int retval)
296{
297 app_session_evt_t _app_evt, *app_evt = &_app_evt;
298 session_accepted_reply_msg_t *rmp;
299 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_ACCEPTED_REPLY);
300 rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
301 rmp->handle = handle;
302 rmp->context = context;
303 rmp->retval = retval;
304 app_send_ctrl_evt_to_vpp (mq, app_evt);
305}
306
Florin Coras99368312018-08-02 10:45:44 -0700307static void
308vcl_send_session_disconnected_reply (svm_msg_q_t * mq, u32 context,
309 session_handle_t handle, int retval)
310{
311 app_session_evt_t _app_evt, *app_evt = &_app_evt;
312 session_disconnected_reply_msg_t *rmp;
313 app_alloc_ctrl_evt_to_vpp (mq, app_evt,
314 SESSION_CTRL_EVT_DISCONNECTED_REPLY);
315 rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
316 rmp->handle = handle;
317 rmp->context = context;
318 rmp->retval = retval;
319 app_send_ctrl_evt_to_vpp (mq, app_evt);
320}
321
Florin Corasc9fbd662018-08-24 12:59:56 -0700322static void
323vcl_send_session_reset_reply (svm_msg_q_t * mq, u32 context,
324 session_handle_t handle, int retval)
325{
326 app_session_evt_t _app_evt, *app_evt = &_app_evt;
327 session_reset_reply_msg_t *rmp;
328 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_RESET_REPLY);
329 rmp = (session_reset_reply_msg_t *) app_evt->evt->data;
330 rmp->handle = handle;
331 rmp->context = context;
332 rmp->retval = retval;
333 app_send_ctrl_evt_to_vpp (mq, app_evt);
334}
335
Florin Coras30e79c22019-01-02 19:31:22 -0800336void
337vcl_send_session_worker_update (vcl_worker_t * wrk, vcl_session_t * s,
338 u32 wrk_index)
339{
340 app_session_evt_t _app_evt, *app_evt = &_app_evt;
341 session_worker_update_msg_t *mp;
342 svm_msg_q_t *mq;
343
344 mq = vcl_session_vpp_evt_q (wrk, s);
345 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_WORKER_UPDATE);
346 mp = (session_worker_update_msg_t *) app_evt->evt->data;
347 mp->client_index = wrk->my_client_index;
348 mp->handle = s->vpp_handle;
349 mp->req_wrk_index = wrk->vpp_wrk_index;
350 mp->wrk_index = wrk_index;
351 app_send_ctrl_evt_to_vpp (mq, app_evt);
352}
353
hanlina3a48962020-07-13 11:09:15 +0800354int
Florin Coras40c07ce2020-07-16 20:46:17 -0700355vcl_send_worker_rpc (u32 dst_wrk_index, void *data, u32 data_len)
356{
357 app_session_evt_t _app_evt, *app_evt = &_app_evt;
358 session_app_wrk_rpc_msg_t *mp;
359 vcl_worker_t *dst_wrk, *wrk;
360 svm_msg_q_t *mq;
hanlina3a48962020-07-13 11:09:15 +0800361 int ret = -1;
Florin Coras40c07ce2020-07-16 20:46:17 -0700362
363 if (data_len > sizeof (mp->data))
364 goto done;
365
366 clib_spinlock_lock (&vcm->workers_lock);
367
368 dst_wrk = vcl_worker_get_if_valid (dst_wrk_index);
369 if (!dst_wrk)
370 goto done;
371
372 wrk = vcl_worker_get_current ();
373 mq = vcl_worker_ctrl_mq (wrk);
374 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_WRK_RPC);
375 mp = (session_app_wrk_rpc_msg_t *) app_evt->evt->data;
376 mp->client_index = wrk->my_client_index;
377 mp->wrk_index = dst_wrk->vpp_wrk_index;
378 clib_memcpy (mp->data, data, data_len);
379 app_send_ctrl_evt_to_vpp (mq, app_evt);
hanlina3a48962020-07-13 11:09:15 +0800380 ret = 0;
Florin Coras40c07ce2020-07-16 20:46:17 -0700381
382done:
383 clib_spinlock_unlock (&vcm->workers_lock);
hanlina3a48962020-07-13 11:09:15 +0800384 return ret;
Florin Coras40c07ce2020-07-16 20:46:17 -0700385}
386
Florin Coras54693d22018-07-17 10:46:29 -0700387static u32
Florin Coras00cca802019-06-06 09:38:44 -0700388vcl_session_accepted_handler (vcl_worker_t * wrk, session_accepted_msg_t * mp,
389 u32 ls_index)
Florin Coras54693d22018-07-17 10:46:29 -0700390{
391 vcl_session_t *session, *listen_session;
392 svm_fifo_t *rx_fifo, *tx_fifo;
Florin Coras134a9962018-08-28 11:32:04 -0700393 u32 vpp_wrk_index;
Florin Coras99368312018-08-02 10:45:44 -0700394 svm_msg_q_t *evt_q;
Florin Coras54693d22018-07-17 10:46:29 -0700395
Florin Coras134a9962018-08-28 11:32:04 -0700396 session = vcl_session_alloc (wrk);
Florin Coras99368312018-08-02 10:45:44 -0700397
Florin Coras00cca802019-06-06 09:38:44 -0700398 listen_session = vcl_session_get (wrk, ls_index);
399 if (listen_session->vpp_handle != mp->listener_handle)
Florin Coras54693d22018-07-17 10:46:29 -0700400 {
Florin Coras00cca802019-06-06 09:38:44 -0700401 VDBG (0, "ERROR: listener handle %lu does not match session %u",
402 mp->listener_handle, ls_index);
403 goto error;
404 }
405
Florin Corasc4c4cf52019-08-24 18:17:34 -0700406 if (vcl_segment_is_not_mounted (wrk, mp->segment_handle))
Florin Coras00cca802019-06-06 09:38:44 -0700407 {
Florin Corasc4c4cf52019-08-24 18:17:34 -0700408 VDBG (0, "ERROR: segment for session %u is not mounted!",
Florin Coras00cca802019-06-06 09:38:44 -0700409 session->session_index);
410 goto error;
Florin Coras54693d22018-07-17 10:46:29 -0700411 }
412
Florin Coras54693d22018-07-17 10:46:29 -0700413 rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
414 tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
Florin Coras653e43f2019-03-04 10:56:23 -0800415 session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
416 svm_msg_q_t *);
417 rx_fifo->client_session_index = session->session_index;
418 tx_fifo->client_session_index = session->session_index;
419 rx_fifo->client_thread_index = vcl_get_worker_index ();
420 tx_fifo->client_thread_index = vcl_get_worker_index ();
421 vpp_wrk_index = tx_fifo->master_thread_index;
422 vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
423 wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
Florin Coras54693d22018-07-17 10:46:29 -0700424
425 session->vpp_handle = mp->handle;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800426 session->vpp_thread_index = rx_fifo->master_thread_index;
Florin Coras54693d22018-07-17 10:46:29 -0700427 session->rx_fifo = rx_fifo;
428 session->tx_fifo = tx_fifo;
429
430 session->session_state = STATE_ACCEPT;
Florin Coras09d18c22019-04-24 11:10:02 -0700431 session->transport.rmt_port = mp->rmt.port;
432 session->transport.is_ip4 = mp->rmt.is_ip4;
433 clib_memcpy_fast (&session->transport.rmt_ip, &mp->rmt.ip,
Dave Barach178cf492018-11-13 16:34:13 -0500434 sizeof (ip46_address_t));
Florin Coras54693d22018-07-17 10:46:29 -0700435
Florin Coras134a9962018-08-28 11:32:04 -0700436 vcl_session_table_add_vpp_handle (wrk, mp->handle, session->session_index);
Florin Coras54693d22018-07-17 10:46:29 -0700437 session->transport.lcl_port = listen_session->transport.lcl_port;
438 session->transport.lcl_ip = listen_session->transport.lcl_ip;
Florin Coras460dce62018-07-27 05:45:06 -0700439 session->session_type = listen_session->session_type;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +0200440 session->is_dgram = vcl_proto_is_dgram (session->session_type);
441 session->listener_index = listen_session->session_index;
442 listen_session->n_accepted_sessions++;
Florin Coras54693d22018-07-17 10:46:29 -0700443
Florin Coras5e062572019-03-14 19:07:51 -0700444 VDBG (1, "session %u [0x%llx]: client accept request from %s address %U"
445 " port %d queue %p!", session->session_index, mp->handle,
Florin Coras09d18c22019-04-24 11:10:02 -0700446 mp->rmt.is_ip4 ? "IPv4" : "IPv6", format_ip46_address, &mp->rmt.ip,
447 mp->rmt.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
448 clib_net_to_host_u16 (mp->rmt.port), session->vpp_evt_q);
Florin Coras54693d22018-07-17 10:46:29 -0700449 vcl_evt (VCL_EVT_ACCEPT, session, listen_session, session_index);
450
Florin Coras00cca802019-06-06 09:38:44 -0700451 vcl_send_session_accepted_reply (session->vpp_evt_q, mp->context,
452 session->vpp_handle, 0);
453
Florin Coras134a9962018-08-28 11:32:04 -0700454 return session->session_index;
Florin Coras00cca802019-06-06 09:38:44 -0700455
456error:
457 evt_q = uword_to_pointer (mp->vpp_event_queue_address, svm_msg_q_t *);
458 vcl_send_session_accepted_reply (evt_q, mp->context, mp->handle,
459 VNET_API_ERROR_INVALID_ARGUMENT);
460 vcl_session_free (wrk, session);
461 return VCL_INVALID_SESSION_INDEX;
Florin Coras54693d22018-07-17 10:46:29 -0700462}
463
464static u32
Florin Coras134a9962018-08-28 11:32:04 -0700465vcl_session_connected_handler (vcl_worker_t * wrk,
466 session_connected_msg_t * mp)
Florin Coras54693d22018-07-17 10:46:29 -0700467{
Florin Coras99368312018-08-02 10:45:44 -0700468 u32 session_index, vpp_wrk_index;
Florin Coras54693d22018-07-17 10:46:29 -0700469 svm_fifo_t *rx_fifo, *tx_fifo;
Florin Coras99368312018-08-02 10:45:44 -0700470 vcl_session_t *session = 0;
Florin Coras54693d22018-07-17 10:46:29 -0700471
472 session_index = mp->context;
Florin Coras134a9962018-08-28 11:32:04 -0700473 session = vcl_session_get (wrk, session_index);
Florin Coras070453d2018-08-24 17:04:27 -0700474 if (!session)
475 {
Florin Coras5e062572019-03-14 19:07:51 -0700476 VDBG (0, "ERROR: vpp handle 0x%llx has no session index (%u)!",
477 mp->handle, session_index);
Florin Coras070453d2018-08-24 17:04:27 -0700478 return VCL_INVALID_SESSION_INDEX;
479 }
Florin Coras54693d22018-07-17 10:46:29 -0700480 if (mp->retval)
481 {
Florin Coras5e062572019-03-14 19:07:51 -0700482 VDBG (0, "ERROR: session index %u: connect failed! %U",
Florin Coras00e01d32019-10-21 16:07:46 -0700483 session_index, format_session_error, mp->retval);
Florin Corasadcfb152020-02-12 08:50:29 +0000484 session->session_state = STATE_DETACHED | STATE_DISCONNECT;
Florin Coras070453d2018-08-24 17:04:27 -0700485 session->vpp_handle = mp->handle;
486 return session_index;
Florin Coras54693d22018-07-17 10:46:29 -0700487 }
488
Florin Corasdbc9c592019-09-25 16:37:43 -0700489 session->vpp_handle = mp->handle;
490 session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
491 svm_msg_q_t *);
Florin Coras460dce62018-07-27 05:45:06 -0700492 rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
493 tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
Florin Corasc4c4cf52019-08-24 18:17:34 -0700494 if (vcl_segment_is_not_mounted (wrk, mp->segment_handle))
Florin Corasd85de682018-11-29 17:02:29 -0800495 {
Florin Corasc4c4cf52019-08-24 18:17:34 -0700496 VDBG (0, "segment for session %u is not mounted!",
Florin Coras653e43f2019-03-04 10:56:23 -0800497 session->session_index);
Florin Corasadcfb152020-02-12 08:50:29 +0000498 session->session_state = STATE_DETACHED | STATE_DISCONNECT;
Florin Corasdbc9c592019-09-25 16:37:43 -0700499 vcl_send_session_disconnect (wrk, session);
500 return session_index;
Florin Corasd85de682018-11-29 17:02:29 -0800501 }
502
Florin Coras460dce62018-07-27 05:45:06 -0700503 rx_fifo->client_session_index = session_index;
504 tx_fifo->client_session_index = session_index;
Florin Coras21795132018-09-09 09:40:51 -0700505 rx_fifo->client_thread_index = vcl_get_worker_index ();
506 tx_fifo->client_thread_index = vcl_get_worker_index ();
Florin Coras460dce62018-07-27 05:45:06 -0700507
Florin Coras653e43f2019-03-04 10:56:23 -0800508 vpp_wrk_index = tx_fifo->master_thread_index;
509 vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
510 wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
Florin Coras99368312018-08-02 10:45:44 -0700511
Florin Coras653e43f2019-03-04 10:56:23 -0800512 if (mp->ct_rx_fifo)
Florin Coras99368312018-08-02 10:45:44 -0700513 {
Florin Coras653e43f2019-03-04 10:56:23 -0800514 session->ct_rx_fifo = uword_to_pointer (mp->ct_rx_fifo, svm_fifo_t *);
515 session->ct_tx_fifo = uword_to_pointer (mp->ct_tx_fifo, svm_fifo_t *);
Florin Corasc4c4cf52019-08-24 18:17:34 -0700516 if (vcl_segment_is_not_mounted (wrk, mp->ct_segment_handle))
Florin Coras653e43f2019-03-04 10:56:23 -0800517 {
Florin Corasc4c4cf52019-08-24 18:17:34 -0700518 VDBG (0, "ct segment for session %u is not mounted!",
Florin Coras653e43f2019-03-04 10:56:23 -0800519 session->session_index);
Florin Corasadcfb152020-02-12 08:50:29 +0000520 session->session_state = STATE_DETACHED | STATE_DISCONNECT;
Florin Corasdbc9c592019-09-25 16:37:43 -0700521 vcl_send_session_disconnect (wrk, session);
522 return session_index;
Florin Coras653e43f2019-03-04 10:56:23 -0800523 }
Florin Coras99368312018-08-02 10:45:44 -0700524 }
Florin Coras54693d22018-07-17 10:46:29 -0700525
Florin Coras54693d22018-07-17 10:46:29 -0700526 session->rx_fifo = rx_fifo;
527 session->tx_fifo = tx_fifo;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800528 session->vpp_thread_index = rx_fifo->master_thread_index;
Florin Coras09d18c22019-04-24 11:10:02 -0700529 session->transport.is_ip4 = mp->lcl.is_ip4;
530 clib_memcpy_fast (&session->transport.lcl_ip, &mp->lcl.ip,
Dave Barach178cf492018-11-13 16:34:13 -0500531 sizeof (session->transport.lcl_ip));
Florin Coras09d18c22019-04-24 11:10:02 -0700532 session->transport.lcl_port = mp->lcl.port;
Florin Corasa5ea8212020-08-24 21:23:51 -0700533
534 /* Application closed session before connect reply */
535 if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK)
536 && session->session_state == STATE_CLOSED)
537 vcl_send_session_disconnect (wrk, session);
538 else
539 session->session_state = STATE_CONNECT;
Florin Coras54693d22018-07-17 10:46:29 -0700540
541 /* Add it to lookup table */
Florin Coras3c7d4f92018-12-14 11:28:43 -0800542 vcl_session_table_add_vpp_handle (wrk, mp->handle, session_index);
Florin Coras54693d22018-07-17 10:46:29 -0700543
Florin Coras5e062572019-03-14 19:07:51 -0700544 VDBG (1, "session %u [0x%llx] connected! rx_fifo %p, refcnt %d, tx_fifo %p,"
545 " refcnt %d", session_index, mp->handle, session->rx_fifo,
Florin Coras54693d22018-07-17 10:46:29 -0700546 session->rx_fifo->refcnt, session->tx_fifo, session->tx_fifo->refcnt);
Florin Coras070453d2018-08-24 17:04:27 -0700547
Florin Coras54693d22018-07-17 10:46:29 -0700548 return session_index;
549}
550
Florin Coras3c7d4f92018-12-14 11:28:43 -0800551static int
552vcl_flag_accepted_session (vcl_session_t * session, u64 handle, u32 flags)
553{
554 vcl_session_msg_t *accepted_msg;
555 int i;
556
557 for (i = 0; i < vec_len (session->accept_evts_fifo); i++)
558 {
559 accepted_msg = &session->accept_evts_fifo[i];
560 if (accepted_msg->accepted_msg.handle == handle)
561 {
Florin Corasb0f662f2018-12-27 14:51:46 -0800562 accepted_msg->flags |= flags;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800563 return 1;
564 }
565 }
566 return 0;
567}
568
Florin Corasc9fbd662018-08-24 12:59:56 -0700569static u32
Florin Coras134a9962018-08-28 11:32:04 -0700570vcl_session_reset_handler (vcl_worker_t * wrk,
571 session_reset_msg_t * reset_msg)
Florin Corasc9fbd662018-08-24 12:59:56 -0700572{
573 vcl_session_t *session;
574 u32 sid;
575
Florin Coras134a9962018-08-28 11:32:04 -0700576 sid = vcl_session_index_from_vpp_handle (wrk, reset_msg->handle);
577 session = vcl_session_get (wrk, sid);
Florin Corasc9fbd662018-08-24 12:59:56 -0700578 if (!session)
579 {
580 VDBG (0, "request to reset unknown handle 0x%llx", reset_msg->handle);
581 return VCL_INVALID_SESSION_INDEX;
582 }
Florin Coras3c7d4f92018-12-14 11:28:43 -0800583
584 /* Caught a reset before actually accepting the session */
585 if (session->session_state == STATE_LISTEN)
586 {
587
588 if (!vcl_flag_accepted_session (session, reset_msg->handle,
589 VCL_ACCEPTED_F_RESET))
590 VDBG (0, "session was not accepted!");
591 return VCL_INVALID_SESSION_INDEX;
592 }
593
Florin Coras080aa502020-05-26 00:47:52 +0000594 if (session->session_state != STATE_CLOSED)
595 session->session_state = STATE_DISCONNECT;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800596 VDBG (0, "reset session %u [0x%llx]", sid, reset_msg->handle);
Florin Corasc9fbd662018-08-24 12:59:56 -0700597 return sid;
598}
599
Florin Coras60116992018-08-27 09:52:18 -0700600static u32
Florin Coras134a9962018-08-28 11:32:04 -0700601vcl_session_bound_handler (vcl_worker_t * wrk, session_bound_msg_t * mp)
Florin Coras60116992018-08-27 09:52:18 -0700602{
603 vcl_session_t *session;
604 u32 sid = mp->context;
605
Florin Coras134a9962018-08-28 11:32:04 -0700606 session = vcl_session_get (wrk, sid);
Florin Coras60116992018-08-27 09:52:18 -0700607 if (mp->retval)
608 {
Florin Coras5e062572019-03-14 19:07:51 -0700609 VERR ("session %u [0x%llx]: bind failed: %U", sid, mp->handle,
Florin Coras00e01d32019-10-21 16:07:46 -0700610 format_session_error, mp->retval);
Florin Coras60116992018-08-27 09:52:18 -0700611 if (session)
612 {
Florin Corasadcfb152020-02-12 08:50:29 +0000613 session->session_state = STATE_DETACHED;
Florin Coras60116992018-08-27 09:52:18 -0700614 session->vpp_handle = mp->handle;
615 return sid;
616 }
617 else
618 {
Florin Coras5e062572019-03-14 19:07:51 -0700619 VDBG (0, "ERROR: session %u [0x%llx]: Invalid session index!",
620 sid, mp->handle);
Florin Coras60116992018-08-27 09:52:18 -0700621 return VCL_INVALID_SESSION_INDEX;
622 }
623 }
624
625 session->vpp_handle = mp->handle;
626 session->transport.is_ip4 = mp->lcl_is_ip4;
Dave Barach178cf492018-11-13 16:34:13 -0500627 clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
628 sizeof (ip46_address_t));
Florin Coras60116992018-08-27 09:52:18 -0700629 session->transport.lcl_port = mp->lcl_port;
Florin Coras134a9962018-08-28 11:32:04 -0700630 vcl_session_table_add_listener (wrk, mp->handle, sid);
Florin Coras60116992018-08-27 09:52:18 -0700631 session->session_state = STATE_LISTEN;
632
Florin Coras5f45e012019-01-23 09:21:30 -0800633 session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
634 vec_validate (wrk->vpp_event_queues, 0);
635 wrk->vpp_event_queues[0] = session->vpp_evt_q;
636
Florin Coras6e3c1f82020-01-15 01:30:46 +0000637 if (vcl_session_is_cl (session))
Florin Coras60116992018-08-27 09:52:18 -0700638 {
639 svm_fifo_t *rx_fifo, *tx_fifo;
640 session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
641 rx_fifo = uword_to_pointer (mp->rx_fifo, svm_fifo_t *);
642 rx_fifo->client_session_index = sid;
643 tx_fifo = uword_to_pointer (mp->tx_fifo, svm_fifo_t *);
644 tx_fifo->client_session_index = sid;
645 session->rx_fifo = rx_fifo;
646 session->tx_fifo = tx_fifo;
647 }
648
Florin Coras05ecfcc2018-12-12 18:19:39 -0800649 VDBG (0, "session %u [0x%llx]: listen succeeded!", sid, mp->handle);
Florin Coras60116992018-08-27 09:52:18 -0700650 return sid;
651}
652
Florin Corasdfae9f92019-02-20 19:48:31 -0800653static void
654vcl_session_unlisten_reply_handler (vcl_worker_t * wrk, void *data)
655{
656 session_unlisten_reply_msg_t *mp = (session_unlisten_reply_msg_t *) data;
657 vcl_session_t *s;
658
659 s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
Florin Coras0a1e1832020-03-29 18:54:04 +0000660 if (!s)
Florin Corasdfae9f92019-02-20 19:48:31 -0800661 {
662 VDBG (0, "Unlisten reply with wrong handle %llx", mp->handle);
663 return;
664 }
Florin Coras0a1e1832020-03-29 18:54:04 +0000665 if (s->session_state != STATE_DISCONNECT)
666 {
667 /* Connected udp listener */
668 if (s->session_type == VPPCOM_PROTO_UDP
669 && s->session_state == STATE_CLOSED)
670 return;
671
672 VDBG (0, "Unlisten session in wrong state %llx", mp->handle);
673 return;
674 }
Florin Corasdfae9f92019-02-20 19:48:31 -0800675
676 if (mp->retval)
677 VDBG (0, "ERROR: session %u [0xllx]: unlisten failed: %U",
Florin Coras00e01d32019-10-21 16:07:46 -0700678 s->session_index, mp->handle, format_session_error, mp->retval);
Florin Corasdfae9f92019-02-20 19:48:31 -0800679
680 if (mp->context != wrk->wrk_index)
681 VDBG (0, "wrong context");
682
683 vcl_session_table_del_vpp_handle (wrk, mp->handle);
684 vcl_session_free (wrk, s);
685}
686
Florin Coras68b7e582020-01-21 18:33:23 -0800687static void
688vcl_session_migrated_handler (vcl_worker_t * wrk, void *data)
689{
690 session_migrated_msg_t *mp = (session_migrated_msg_t *) data;
691 vcl_session_t *s;
692
693 s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
694 if (!s)
695 {
696 VDBG (0, "Migrated notification with wrong handle %llx", mp->handle);
697 return;
698 }
699
700 s->vpp_thread_index = mp->vpp_thread_index;
Florin Coras57660d92020-04-04 22:45:34 +0000701 s->vpp_handle = mp->new_handle;
Florin Coras68b7e582020-01-21 18:33:23 -0800702 s->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
703
704 vec_validate (wrk->vpp_event_queues, s->vpp_thread_index);
705 wrk->vpp_event_queues[s->vpp_thread_index] = s->vpp_evt_q;
706
707 vcl_session_table_del_vpp_handle (wrk, mp->handle);
708 vcl_session_table_add_vpp_handle (wrk, mp->new_handle, s->session_index);
709
710 /* Generate new tx event if we have outstanding data */
711 if (svm_fifo_has_event (s->tx_fifo))
712 app_send_io_evt_to_vpp (s->vpp_evt_q, s->tx_fifo->master_session_index,
713 SESSION_IO_EVT_TX, SVM_Q_WAIT);
714
Florin Coras57660d92020-04-04 22:45:34 +0000715 VDBG (0, "Migrated 0x%lx to thread %u 0x%lx", mp->handle,
716 s->vpp_thread_index, mp->new_handle);
Florin Coras68b7e582020-01-21 18:33:23 -0800717}
718
Florin Coras3c7d4f92018-12-14 11:28:43 -0800719static vcl_session_t *
720vcl_session_accepted (vcl_worker_t * wrk, session_accepted_msg_t * msg)
721{
722 vcl_session_msg_t *vcl_msg;
723 vcl_session_t *session;
724
725 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
726 if (PREDICT_FALSE (session != 0))
Florin Corasb0f662f2018-12-27 14:51:46 -0800727 VWRN ("session overlap handle %lu state %u!", msg->handle,
728 session->session_state);
Florin Coras3c7d4f92018-12-14 11:28:43 -0800729
730 session = vcl_session_table_lookup_listener (wrk, msg->listener_handle);
731 if (!session)
732 {
733 VERR ("couldn't find listen session: listener handle %llx",
734 msg->listener_handle);
735 return 0;
736 }
737
738 clib_fifo_add2 (session->accept_evts_fifo, vcl_msg);
Florin Corase88845e2020-02-13 20:04:28 +0000739 vcl_msg->flags = 0;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800740 vcl_msg->accepted_msg = *msg;
741 /* Session handle points to listener until fully accepted by app */
742 vcl_session_table_add_vpp_handle (wrk, msg->handle, session->session_index);
743
744 return session;
745}
746
747static vcl_session_t *
748vcl_session_disconnected_handler (vcl_worker_t * wrk,
749 session_disconnected_msg_t * msg)
750{
751 vcl_session_t *session;
752
753 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
754 if (!session)
755 {
756 VDBG (0, "request to disconnect unknown handle 0x%llx", msg->handle);
757 return 0;
758 }
759
Florin Corasadcfb152020-02-12 08:50:29 +0000760 /* Late disconnect notification on a session that has been closed */
761 if (session->session_state == STATE_CLOSED)
762 return 0;
763
Florin Coras3c7d4f92018-12-14 11:28:43 -0800764 /* Caught a disconnect before actually accepting the session */
765 if (session->session_state == STATE_LISTEN)
766 {
Florin Coras3c7d4f92018-12-14 11:28:43 -0800767 if (!vcl_flag_accepted_session (session, msg->handle,
768 VCL_ACCEPTED_F_CLOSED))
769 VDBG (0, "session was not accepted!");
770 return 0;
771 }
772
Florin Corasadcfb152020-02-12 08:50:29 +0000773 /* If not already reset change state */
774 if (session->session_state != STATE_DISCONNECT)
775 session->session_state = STATE_VPP_CLOSING;
776
Florin Coras3c7d4f92018-12-14 11:28:43 -0800777 return session;
778}
779
Florin Coras30e79c22019-01-02 19:31:22 -0800780static void
Florin Coras9ace36d2019-10-28 13:14:17 -0700781vcl_session_cleanup_handler (vcl_worker_t * wrk, void *data)
782{
783 session_cleanup_msg_t *msg;
784 vcl_session_t *session;
785
786 msg = (session_cleanup_msg_t *) data;
787 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
788 if (!session)
789 {
790 VDBG (0, "disconnect confirmed for unknown handle 0x%llx", msg->handle);
791 return;
792 }
793
Florin Coras36d49392020-04-24 23:00:11 +0000794 if (msg->type == SESSION_CLEANUP_TRANSPORT)
795 {
796 /* Transport was cleaned up before we confirmed close. Probably the
797 * app is still waiting for some data that cannot be delivered.
798 * Confirm close to make sure everything is cleaned up */
799 if (session->session_state == STATE_VPP_CLOSING)
800 vcl_session_cleanup (wrk, session, vcl_session_handle (session),
801 1 /* do_disconnect */ );
802 return;
803 }
804
Florin Coras9ace36d2019-10-28 13:14:17 -0700805 vcl_session_table_del_vpp_handle (wrk, msg->handle);
Florin Corasadcfb152020-02-12 08:50:29 +0000806 /* Should not happen. App did not close the connection so don't free it. */
807 if (session->session_state != STATE_CLOSED)
808 {
809 VDBG (0, "app did not close session %d", session->session_index);
810 session->session_state = STATE_DETACHED;
811 session->vpp_handle = VCL_INVALID_SESSION_HANDLE;
812 return;
813 }
Florin Coras9ace36d2019-10-28 13:14:17 -0700814 vcl_session_free (wrk, session);
815}
816
817static void
Florin Coras30e79c22019-01-02 19:31:22 -0800818vcl_session_req_worker_update_handler (vcl_worker_t * wrk, void *data)
819{
820 session_req_worker_update_msg_t *msg;
821 vcl_session_t *s;
822
823 msg = (session_req_worker_update_msg_t *) data;
824 s = vcl_session_get_w_vpp_handle (wrk, msg->session_handle);
825 if (!s)
826 return;
827
828 vec_add1 (wrk->pending_session_wrk_updates, s->session_index);
829}
830
831static void
832vcl_session_worker_update_reply_handler (vcl_worker_t * wrk, void *data)
833{
834 session_worker_update_reply_msg_t *msg;
835 vcl_session_t *s;
836
837 msg = (session_worker_update_reply_msg_t *) data;
838 s = vcl_session_get_w_vpp_handle (wrk, msg->handle);
839 if (!s)
840 {
841 VDBG (0, "unknown handle 0x%llx", msg->handle);
842 return;
843 }
Florin Corasc4c4cf52019-08-24 18:17:34 -0700844 if (vcl_segment_is_not_mounted (wrk, msg->segment_handle))
Florin Coras30e79c22019-01-02 19:31:22 -0800845 {
Florin Corasc4c4cf52019-08-24 18:17:34 -0700846 clib_warning ("segment for session %u is not mounted!",
Florin Coras30e79c22019-01-02 19:31:22 -0800847 s->session_index);
848 return;
849 }
Florin Coras30e79c22019-01-02 19:31:22 -0800850
Florin Coras5f45e012019-01-23 09:21:30 -0800851 if (s->rx_fifo)
852 {
853 s->rx_fifo = uword_to_pointer (msg->rx_fifo, svm_fifo_t *);
854 s->tx_fifo = uword_to_pointer (msg->tx_fifo, svm_fifo_t *);
855 s->rx_fifo->client_session_index = s->session_index;
856 s->tx_fifo->client_session_index = s->session_index;
857 s->rx_fifo->client_thread_index = wrk->wrk_index;
858 s->tx_fifo->client_thread_index = wrk->wrk_index;
859 }
Florin Coras30e79c22019-01-02 19:31:22 -0800860 s->session_state = STATE_UPDATED;
861
Florin Coras30e79c22019-01-02 19:31:22 -0800862 VDBG (0, "session %u[0x%llx] moved to worker %u", s->session_index,
863 s->vpp_handle, wrk->wrk_index);
864}
865
Florin Corasc4c4cf52019-08-24 18:17:34 -0700866static void
867vcl_session_app_add_segment_handler (vcl_worker_t * wrk, void *data)
868{
869 ssvm_segment_type_t seg_type = SSVM_SEGMENT_SHM;
870 session_app_add_segment_msg_t *msg;
871 u64 segment_handle;
872 int fd = -1;
873
874 msg = (session_app_add_segment_msg_t *) data;
875
876 if (msg->fd_flags)
877 {
878 vl_socket_client_recv_fd_msg2 (&wrk->bapi_sock_ctx, &fd, 1, 5);
879 seg_type = SSVM_SEGMENT_MEMFD;
880 }
881
882 segment_handle = msg->segment_handle;
883 if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
884 {
885 clib_warning ("invalid segment handle");
886 return;
887 }
888
889 if (vcl_segment_attach (segment_handle, (char *) msg->segment_name,
890 seg_type, fd))
891 {
892 VDBG (0, "vcl_segment_attach ('%s') failed", msg->segment_name);
893 return;
894 }
895
896 VDBG (1, "mapped new segment '%s' size %d", msg->segment_name,
897 msg->segment_size);
898}
899
900static void
901vcl_session_app_del_segment_handler (vcl_worker_t * wrk, void *data)
902{
903 session_app_del_segment_msg_t *msg = (session_app_del_segment_msg_t *) data;
904 vcl_segment_detach (msg->segment_handle);
905 VDBG (1, "Unmapped segment: %d", msg->segment_handle);
906}
907
Florin Coras40c07ce2020-07-16 20:46:17 -0700908static void
909vcl_worker_rpc_handler (vcl_worker_t * wrk, void *data)
910{
911 if (!vcm->wrk_rpc_fn)
912 return;
913
hanlina3a48962020-07-13 11:09:15 +0800914 (vcm->wrk_rpc_fn) (((session_app_wrk_rpc_msg_t *) data)->data);
Florin Coras40c07ce2020-07-16 20:46:17 -0700915}
916
Florin Coras86f04502018-09-12 16:08:01 -0700917static int
918vcl_handle_mq_event (vcl_worker_t * wrk, session_event_t * e)
Florin Coras54693d22018-07-17 10:46:29 -0700919{
Florin Coras54693d22018-07-17 10:46:29 -0700920 session_disconnected_msg_t *disconnected_msg;
Florin Coras54693d22018-07-17 10:46:29 -0700921 vcl_session_t *session;
Florin Coras54693d22018-07-17 10:46:29 -0700922
923 switch (e->event_type)
924 {
Florin Coras653e43f2019-03-04 10:56:23 -0800925 case SESSION_IO_EVT_RX:
926 case SESSION_IO_EVT_TX:
Florin Corasc0737e92019-03-04 14:19:39 -0800927 session = vcl_session_get (wrk, e->session_index);
Florin Coras653e43f2019-03-04 10:56:23 -0800928 if (!session || !(session->session_state & STATE_OPEN))
929 break;
Florin Coras86f04502018-09-12 16:08:01 -0700930 vec_add1 (wrk->unhandled_evts_vector, *e);
Florin Coras54693d22018-07-17 10:46:29 -0700931 break;
932 case SESSION_CTRL_EVT_ACCEPTED:
Florin Coras3c7d4f92018-12-14 11:28:43 -0800933 vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
Florin Coras54693d22018-07-17 10:46:29 -0700934 break;
935 case SESSION_CTRL_EVT_CONNECTED:
Florin Coras134a9962018-08-28 11:32:04 -0700936 vcl_session_connected_handler (wrk,
937 (session_connected_msg_t *) e->data);
Florin Coras54693d22018-07-17 10:46:29 -0700938 break;
939 case SESSION_CTRL_EVT_DISCONNECTED:
940 disconnected_msg = (session_disconnected_msg_t *) e->data;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800941 session = vcl_session_disconnected_handler (wrk, disconnected_msg);
Florin Corasc9fbd662018-08-24 12:59:56 -0700942 if (!session)
Florin Coras3c7d4f92018-12-14 11:28:43 -0800943 break;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800944 VDBG (0, "disconnected session %u [0x%llx]", session->session_index,
945 session->vpp_handle);
Florin Corasc9fbd662018-08-24 12:59:56 -0700946 break;
947 case SESSION_CTRL_EVT_RESET:
Florin Coras134a9962018-08-28 11:32:04 -0700948 vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
Florin Coras60116992018-08-27 09:52:18 -0700949 break;
950 case SESSION_CTRL_EVT_BOUND:
Florin Coras134a9962018-08-28 11:32:04 -0700951 vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
Florin Coras54693d22018-07-17 10:46:29 -0700952 break;
Florin Corasdfae9f92019-02-20 19:48:31 -0800953 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
954 vcl_session_unlisten_reply_handler (wrk, e->data);
955 break;
Florin Coras68b7e582020-01-21 18:33:23 -0800956 case SESSION_CTRL_EVT_MIGRATED:
957 vcl_session_migrated_handler (wrk, e->data);
958 break;
Florin Coras9ace36d2019-10-28 13:14:17 -0700959 case SESSION_CTRL_EVT_CLEANUP:
960 vcl_session_cleanup_handler (wrk, e->data);
961 break;
Florin Coras30e79c22019-01-02 19:31:22 -0800962 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
963 vcl_session_req_worker_update_handler (wrk, e->data);
964 break;
965 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
966 vcl_session_worker_update_reply_handler (wrk, e->data);
967 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -0700968 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
969 vcl_session_app_add_segment_handler (wrk, e->data);
970 break;
971 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
972 vcl_session_app_del_segment_handler (wrk, e->data);
973 break;
hanlina3a48962020-07-13 11:09:15 +0800974 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -0700975 vcl_worker_rpc_handler (wrk, e->data);
976 break;
Florin Coras54693d22018-07-17 10:46:29 -0700977 default:
978 clib_warning ("unhandled %u", e->event_type);
979 }
980 return VPPCOM_OK;
981}
982
Florin Coras30e79c22019-01-02 19:31:22 -0800983static int
Florin Coras697faea2018-06-27 17:10:49 -0700984vppcom_wait_for_session_state_change (u32 session_index,
Florin Coras288eaab2019-02-03 15:26:14 -0800985 vcl_session_state_t state,
Florin Coras697faea2018-06-27 17:10:49 -0700986 f64 wait_for_time)
987{
Florin Coras134a9962018-08-28 11:32:04 -0700988 vcl_worker_t *wrk = vcl_worker_get_current ();
989 f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
Florin Coras697faea2018-06-27 17:10:49 -0700990 vcl_session_t *volatile session;
Florin Coras54693d22018-07-17 10:46:29 -0700991 svm_msg_q_msg_t msg;
992 session_event_t *e;
Dave Wallace543852a2017-08-03 02:11:34 -0400993
Florin Coras697faea2018-06-27 17:10:49 -0700994 do
Dave Wallace543852a2017-08-03 02:11:34 -0400995 {
Florin Coras134a9962018-08-28 11:32:04 -0700996 session = vcl_session_get (wrk, session_index);
Florin Coras070453d2018-08-24 17:04:27 -0700997 if (PREDICT_FALSE (!session))
Florin Coras697faea2018-06-27 17:10:49 -0700998 {
Florin Coras070453d2018-08-24 17:04:27 -0700999 return VPPCOM_EBADFD;
Florin Coras697faea2018-06-27 17:10:49 -07001000 }
1001 if (session->session_state & state)
1002 {
Florin Coras697faea2018-06-27 17:10:49 -07001003 return VPPCOM_OK;
1004 }
Florin Corasadcfb152020-02-12 08:50:29 +00001005 if (session->session_state & STATE_DETACHED)
Florin Coras697faea2018-06-27 17:10:49 -07001006 {
Florin Coras697faea2018-06-27 17:10:49 -07001007 return VPPCOM_ECONNREFUSED;
1008 }
Florin Coras54693d22018-07-17 10:46:29 -07001009
Florin Coras134a9962018-08-28 11:32:04 -07001010 if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0))
Florin Corasdc2e2512018-12-03 17:47:26 -08001011 {
1012 usleep (100);
1013 continue;
1014 }
Florin Coras134a9962018-08-28 11:32:04 -07001015 e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
Florin Coras86f04502018-09-12 16:08:01 -07001016 vcl_handle_mq_event (wrk, e);
Florin Coras134a9962018-08-28 11:32:04 -07001017 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
Florin Corasdcf55ce2017-11-16 15:32:50 -08001018 }
Florin Coras134a9962018-08-28 11:32:04 -07001019 while (clib_time_now (&wrk->clib_time) < timeout);
Florin Corasdcf55ce2017-11-16 15:32:50 -08001020
Florin Coras05ecfcc2018-12-12 18:19:39 -08001021 VDBG (0, "timeout waiting for state 0x%x (%s)", state,
Florin Coras697faea2018-06-27 17:10:49 -07001022 vppcom_session_state_str (state));
1023 vcl_evt (VCL_EVT_SESSION_TIMEOUT, session, session_state);
Dave Wallace543852a2017-08-03 02:11:34 -04001024
Florin Coras697faea2018-06-27 17:10:49 -07001025 return VPPCOM_ETIMEDOUT;
Dave Wallace60caa062017-11-10 17:07:13 -05001026}
1027
Florin Coras30e79c22019-01-02 19:31:22 -08001028static void
1029vcl_handle_pending_wrk_updates (vcl_worker_t * wrk)
1030{
Florin Coras288eaab2019-02-03 15:26:14 -08001031 vcl_session_state_t state;
Florin Coras30e79c22019-01-02 19:31:22 -08001032 vcl_session_t *s;
1033 u32 *sip;
1034
1035 if (PREDICT_TRUE (vec_len (wrk->pending_session_wrk_updates) == 0))
1036 return;
1037
1038 vec_foreach (sip, wrk->pending_session_wrk_updates)
1039 {
1040 s = vcl_session_get (wrk, *sip);
1041 vcl_send_session_worker_update (wrk, s, wrk->wrk_index);
1042 state = s->session_state;
1043 vppcom_wait_for_session_state_change (s->session_index, STATE_UPDATED, 5);
1044 s->session_state = state;
1045 }
1046 vec_reset_length (wrk->pending_session_wrk_updates);
1047}
1048
Florin Corasf9240dc2019-01-15 08:03:17 -08001049void
Florin Coras30e79c22019-01-02 19:31:22 -08001050vcl_flush_mq_events (void)
1051{
1052 vcl_worker_t *wrk = vcl_worker_get_current ();
1053 svm_msg_q_msg_t *msg;
1054 session_event_t *e;
1055 svm_msg_q_t *mq;
1056 int i;
1057
1058 mq = wrk->app_event_queue;
1059 svm_msg_q_lock (mq);
Florin Corase003a1b2019-06-05 10:47:16 -07001060 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras30e79c22019-01-02 19:31:22 -08001061 svm_msg_q_unlock (mq);
1062
1063 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
1064 {
1065 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
1066 e = svm_msg_q_msg_data (mq, msg);
1067 vcl_handle_mq_event (wrk, e);
1068 svm_msg_q_free_msg (mq, msg);
1069 }
1070 vec_reset_length (wrk->mq_msg_vector);
1071 vcl_handle_pending_wrk_updates (wrk);
1072}
1073
Florin Coras697faea2018-06-27 17:10:49 -07001074static int
1075vppcom_app_session_enable (void)
Dave Wallace543852a2017-08-03 02:11:34 -04001076{
Florin Coras697faea2018-06-27 17:10:49 -07001077 int rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001078
Florin Coras697faea2018-06-27 17:10:49 -07001079 if (vcm->app_state != STATE_APP_ENABLED)
1080 {
1081 vppcom_send_session_enable_disable (1 /* is_enabled == TRUE */ );
Florin Coras134a9962018-08-28 11:32:04 -07001082 rv = vcl_wait_for_app_state_change (STATE_APP_ENABLED);
Florin Coras697faea2018-06-27 17:10:49 -07001083 if (PREDICT_FALSE (rv))
1084 {
Florin Coras5e062572019-03-14 19:07:51 -07001085 VDBG (0, "application session enable timed out! returning %d (%s)",
1086 rv, vppcom_retval_str (rv));
Florin Coras697faea2018-06-27 17:10:49 -07001087 return rv;
1088 }
1089 }
1090 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001091}
1092
Florin Coras697faea2018-06-27 17:10:49 -07001093static int
1094vppcom_app_attach (void)
Dave Wallace543852a2017-08-03 02:11:34 -04001095{
Florin Coras697faea2018-06-27 17:10:49 -07001096 int rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001097
Florin Coras697faea2018-06-27 17:10:49 -07001098 vppcom_app_send_attach ();
Florin Coras134a9962018-08-28 11:32:04 -07001099 rv = vcl_wait_for_app_state_change (STATE_APP_ATTACHED);
Florin Coras697faea2018-06-27 17:10:49 -07001100 if (PREDICT_FALSE (rv))
1101 {
Florin Coras5e062572019-03-14 19:07:51 -07001102 VDBG (0, "application attach timed out! returning %d (%s)", rv,
1103 vppcom_retval_str (rv));
Florin Coras697faea2018-06-27 17:10:49 -07001104 return rv;
1105 }
Dave Wallace543852a2017-08-03 02:11:34 -04001106
Florin Coras697faea2018-06-27 17:10:49 -07001107 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001108}
1109
1110static int
Florin Corasab2f6db2018-08-31 14:31:41 -07001111vppcom_session_unbind (u32 session_handle)
Dave Wallace543852a2017-08-03 02:11:34 -04001112{
Florin Coras134a9962018-08-28 11:32:04 -07001113 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Corasaaff5ee2019-07-08 13:00:00 -07001114 session_accepted_msg_t *accepted_msg;
Florin Coras7e12d942018-06-27 14:32:43 -07001115 vcl_session_t *session = 0;
Florin Corasaaff5ee2019-07-08 13:00:00 -07001116 vcl_session_msg_t *evt;
Dave Wallace543852a2017-08-03 02:11:34 -04001117
Florin Corasab2f6db2018-08-31 14:31:41 -07001118 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001119 if (!session)
1120 return VPPCOM_EBADFD;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001121
Florin Corasaaff5ee2019-07-08 13:00:00 -07001122 /* Flush pending accept events, if any */
1123 while (clib_fifo_elts (session->accept_evts_fifo))
1124 {
1125 clib_fifo_sub2 (session->accept_evts_fifo, evt);
1126 accepted_msg = &evt->accepted_msg;
1127 vcl_session_table_del_vpp_handle (wrk, accepted_msg->handle);
1128 vcl_send_session_accepted_reply (session->vpp_evt_q,
1129 accepted_msg->context,
1130 session->vpp_handle, -1);
1131 }
1132 clib_fifo_free (session->accept_evts_fifo);
1133
Florin Coras458089b2019-08-21 16:20:44 -07001134 vcl_send_session_unlisten (wrk, session);
Dave Wallace4878cbe2017-11-21 03:45:09 -05001135
Florin Coras5e062572019-03-14 19:07:51 -07001136 VDBG (1, "session %u [0x%llx]: sending unbind!", session->session_index,
Florin Coras458089b2019-08-21 16:20:44 -07001137 session->vpp_handle);
Florin Coras0d427d82018-06-27 03:24:07 -07001138 vcl_evt (VCL_EVT_UNBIND, session);
Florin Coras458089b2019-08-21 16:20:44 -07001139
1140 session->vpp_handle = ~0;
1141 session->session_state = STATE_DISCONNECT;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001142
Florin Coras070453d2018-08-24 17:04:27 -07001143 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001144}
1145
Florin Coras697faea2018-06-27 17:10:49 -07001146static int
Florin Corasab2f6db2018-08-31 14:31:41 -07001147vppcom_session_disconnect (u32 session_handle)
Dave Wallace543852a2017-08-03 02:11:34 -04001148{
Florin Coras134a9962018-08-28 11:32:04 -07001149 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras99368312018-08-02 10:45:44 -07001150 svm_msg_q_t *vpp_evt_q;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001151 vcl_session_t *session, *listen_session;
Florin Coras288eaab2019-02-03 15:26:14 -08001152 vcl_session_state_t state;
Florin Coras99368312018-08-02 10:45:44 -07001153 u64 vpp_handle;
Dave Wallace543852a2017-08-03 02:11:34 -04001154
Florin Corasab2f6db2018-08-31 14:31:41 -07001155 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras21795132018-09-09 09:40:51 -07001156 if (!session)
1157 return VPPCOM_EBADFD;
1158
Dave Wallace4878cbe2017-11-21 03:45:09 -05001159 vpp_handle = session->vpp_handle;
Florin Coras7e12d942018-06-27 14:32:43 -07001160 state = session->session_state;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001161
Florin Coras5e062572019-03-14 19:07:51 -07001162 VDBG (1, "session %u [0x%llx] state 0x%x (%s)", session->session_index,
1163 vpp_handle, state, vppcom_session_state_str (state));
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001164
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001165 if (PREDICT_FALSE (state & STATE_LISTEN))
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001166 {
Florin Coras5e062572019-03-14 19:07:51 -07001167 VDBG (0, "ERROR: Cannot disconnect a listen socket!");
Florin Coras070453d2018-08-24 17:04:27 -07001168 return VPPCOM_EBADFD;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001169 }
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001170
Florin Coras3c7d4f92018-12-14 11:28:43 -08001171 if (state & STATE_VPP_CLOSING)
Dave Wallace4878cbe2017-11-21 03:45:09 -05001172 {
Florin Coras134a9962018-08-28 11:32:04 -07001173 vpp_evt_q = vcl_session_vpp_evt_q (wrk, session);
Florin Coras47c40e22018-11-26 17:01:36 -08001174 vcl_send_session_disconnected_reply (vpp_evt_q, wrk->my_client_index,
Florin Coras99368312018-08-02 10:45:44 -07001175 vpp_handle, 0);
Florin Coras5e062572019-03-14 19:07:51 -07001176 VDBG (1, "session %u [0x%llx]: sending disconnect REPLY...",
1177 session->session_index, vpp_handle);
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001178 }
1179 else
Dave Wallace227867f2017-11-13 21:21:53 -05001180 {
Florin Corasa5ea8212020-08-24 21:23:51 -07001181 /* Session doesn't have an event queue yet. Probably a non-blocking
1182 * connect. Wait for the reply */
1183 if (PREDICT_FALSE (!session->vpp_evt_q))
1184 return VPPCOM_OK;
1185
Florin Coras5e062572019-03-14 19:07:51 -07001186 VDBG (1, "session %u [0x%llx]: sending disconnect...",
1187 session->session_index, vpp_handle);
Florin Coras458089b2019-08-21 16:20:44 -07001188 vcl_send_session_disconnect (wrk, session);
Dave Wallace227867f2017-11-13 21:21:53 -05001189 }
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001190
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001191 if (session->listener_index != VCL_INVALID_SESSION_INDEX)
1192 {
1193 listen_session = vcl_session_get (wrk, session->listener_index);
1194 listen_session->n_accepted_sessions--;
1195 }
1196
Florin Coras070453d2018-08-24 17:04:27 -07001197 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001198}
1199
Florin Coras940f78f2018-11-30 12:11:20 -08001200/**
1201 * Handle app exit
1202 *
1203 * Notify vpp of the disconnect and mark the worker as free. If we're the
1204 * last worker, do a full cleanup otherwise, since we're probably a forked
1205 * child, avoid syscalls as much as possible. We might've lost privileges.
1206 */
1207void
1208vppcom_app_exit (void)
1209{
1210 if (!pool_elts (vcm->workers))
1211 return;
Florin Coras01f3f892018-12-02 12:45:53 -08001212 vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
1213 vcl_set_worker_index (~0);
Florin Coras940f78f2018-11-30 12:11:20 -08001214 vcl_elog_stop (vcm);
Florin Coras940f78f2018-11-30 12:11:20 -08001215}
1216
Dave Wallace543852a2017-08-03 02:11:34 -04001217/*
1218 * VPPCOM Public API functions
1219 */
1220int
Florin Coras66ec4672020-06-15 07:59:40 -07001221vppcom_app_create (const char *app_name)
Dave Wallace543852a2017-08-03 02:11:34 -04001222{
Dave Wallace543852a2017-08-03 02:11:34 -04001223 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
Dave Wallace543852a2017-08-03 02:11:34 -04001224 int rv;
1225
Florin Coras47c40e22018-11-26 17:01:36 -08001226 if (vcm->is_init)
Dave Wallace543852a2017-08-03 02:11:34 -04001227 {
Florin Coras955bfbb2018-12-04 13:43:45 -08001228 VDBG (1, "already initialized");
1229 return VPPCOM_EEXIST;
Dave Wallace543852a2017-08-03 02:11:34 -04001230 }
1231
Florin Coras47c40e22018-11-26 17:01:36 -08001232 vcm->is_init = 1;
1233 vppcom_cfg (&vcm->cfg);
1234 vcl_cfg = &vcm->cfg;
1235
1236 vcm->main_cpu = pthread_self ();
1237 vcm->main_pid = getpid ();
1238 vcm->app_name = format (0, "%s", app_name);
1239 vppcom_init_error_string_table ();
Florin Coras88001c62019-04-24 14:44:46 -07001240 fifo_segment_main_init (&vcm->segment_main, vcl_cfg->segment_baseva,
1241 20 /* timeout in secs */ );
Florin Coras47c40e22018-11-26 17:01:36 -08001242 pool_alloc (vcm->workers, vcl_cfg->max_workers);
1243 clib_spinlock_init (&vcm->workers_lock);
Florin Corasd85de682018-11-29 17:02:29 -08001244 clib_rwlock_init (&vcm->segment_table_lock);
Florin Coras940f78f2018-11-30 12:11:20 -08001245 atexit (vppcom_app_exit);
Florin Coras47c40e22018-11-26 17:01:36 -08001246
1247 /* Allocate default worker */
1248 vcl_worker_alloc_and_init ();
1249
1250 /* API hookup and connect to VPP */
Florin Coras47c40e22018-11-26 17:01:36 -08001251 vcl_elog_init (vcm);
1252 vcm->app_state = STATE_APP_START;
1253 rv = vppcom_connect_to_vpp (app_name);
1254 if (rv)
Dave Wallace543852a2017-08-03 02:11:34 -04001255 {
Florin Coras47c40e22018-11-26 17:01:36 -08001256 VERR ("couldn't connect to VPP!");
1257 return rv;
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001258 }
Florin Coras47c40e22018-11-26 17:01:36 -08001259 VDBG (0, "sending session enable");
1260 rv = vppcom_app_session_enable ();
1261 if (rv)
1262 {
1263 VERR ("vppcom_app_session_enable() failed!");
1264 return rv;
1265 }
1266
1267 VDBG (0, "sending app attach");
1268 rv = vppcom_app_attach ();
1269 if (rv)
1270 {
1271 VERR ("vppcom_app_attach() failed!");
1272 return rv;
1273 }
1274
1275 VDBG (0, "app_name '%s', my_client_index %d (0x%x)", app_name,
1276 vcm->workers[0].my_client_index, vcm->workers[0].my_client_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001277
1278 return VPPCOM_OK;
1279}
1280
1281void
1282vppcom_app_destroy (void)
1283{
Florin Corasdc0ded72020-04-30 02:59:55 +00001284 vcl_worker_t *wrk, *current_wrk;
Florin Corasce815de2020-04-16 18:47:27 +00001285 struct dlmallinfo mi;
Florin Corasce815de2020-04-16 18:47:27 +00001286 mspace heap;
Dave Wallace543852a2017-08-03 02:11:34 -04001287
Florin Coras940f78f2018-11-30 12:11:20 -08001288 if (!pool_elts (vcm->workers))
1289 return;
1290
Florin Coras0d427d82018-06-27 03:24:07 -07001291 vcl_evt (VCL_EVT_DETACH, vcm);
Keith Burns (alagalah)8aa9aaf2018-01-05 12:16:22 -08001292
Florin Corasdc0ded72020-04-30 02:59:55 +00001293 current_wrk = vcl_worker_get_current ();
Keith Burns (alagalah)8aa9aaf2018-01-05 12:16:22 -08001294
Florin Corasce815de2020-04-16 18:47:27 +00001295 /* *INDENT-OFF* */
1296 pool_foreach (wrk, vcm->workers, ({
Florin Corasdc0ded72020-04-30 02:59:55 +00001297 if (current_wrk != wrk)
1298 vcl_worker_cleanup (wrk, 0 /* notify vpp */ );
Florin Corasce815de2020-04-16 18:47:27 +00001299 }));
1300 /* *INDENT-ON* */
1301
Florin Corasdc0ded72020-04-30 02:59:55 +00001302 vcl_send_app_detach (current_wrk);
1303 vppcom_disconnect_from_vpp ();
1304 vcl_worker_cleanup (current_wrk, 0 /* notify vpp */ );
1305
Florin Coras0d427d82018-06-27 03:24:07 -07001306 vcl_elog_stop (vcm);
Florin Corasce815de2020-04-16 18:47:27 +00001307
1308 /*
1309 * Free the heap and fix vcm
1310 */
1311 heap = clib_mem_get_heap ();
1312 mi = mspace_mallinfo (heap);
1313 munmap (mspace_least_addr (heap), mi.arena);
1314
1315 vcm = &_vppcom_main;
Florin Coras69d97252020-05-11 17:19:31 +00001316 vcm->is_init = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001317}
1318
1319int
Dave Wallacec04cbf12018-02-07 18:14:02 -05001320vppcom_session_create (u8 proto, u8 is_nonblocking)
Dave Wallace543852a2017-08-03 02:11:34 -04001321{
Florin Coras134a9962018-08-28 11:32:04 -07001322 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001323 vcl_session_t *session;
Dave Wallace543852a2017-08-03 02:11:34 -04001324
Florin Coras134a9962018-08-28 11:32:04 -07001325 session = vcl_session_alloc (wrk);
Dave Wallace543852a2017-08-03 02:11:34 -04001326
Florin Coras7e12d942018-06-27 14:32:43 -07001327 session->session_type = proto;
Florin Coras54140622020-02-04 19:04:34 +00001328 session->session_state = STATE_CLOSED;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001329 session->vpp_handle = ~0;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001330 session->is_dgram = vcl_proto_is_dgram (proto);
Dave Wallace543852a2017-08-03 02:11:34 -04001331
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001332 if (is_nonblocking)
1333 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
Dave Wallace543852a2017-08-03 02:11:34 -04001334
Florin Coras7e12d942018-06-27 14:32:43 -07001335 vcl_evt (VCL_EVT_CREATE, session, session_type, session->session_state,
1336 is_nonblocking, session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001337
Florin Coras5e062572019-03-14 19:07:51 -07001338 VDBG (0, "created session %u", session->session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001339
Florin Coras134a9962018-08-28 11:32:04 -07001340 return vcl_session_handle (session);
Dave Wallace543852a2017-08-03 02:11:34 -04001341}
1342
1343int
Florin Corasf9240dc2019-01-15 08:03:17 -08001344vcl_session_cleanup (vcl_worker_t * wrk, vcl_session_t * session,
1345 vcl_session_handle_t sh, u8 do_disconnect)
Dave Wallace543852a2017-08-03 02:11:34 -04001346{
Florin Coras288eaab2019-02-03 15:26:14 -08001347 vcl_session_state_t state;
Florin Coras134a9962018-08-28 11:32:04 -07001348 u32 next_sh, vep_sh;
1349 int rv = VPPCOM_OK;
1350 u64 vpp_handle;
Florin Corasf9240dc2019-01-15 08:03:17 -08001351 u8 is_vep;
Florin Coras47c40e22018-11-26 17:01:36 -08001352
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001353 is_vep = session->is_vep;
Florin Coras134a9962018-08-28 11:32:04 -07001354 next_sh = session->vep.next_sh;
1355 vep_sh = session->vep.vep_sh;
Florin Coras7e12d942018-06-27 14:32:43 -07001356 state = session->session_state;
Dave Wallaceee45d412017-11-24 21:44:06 -05001357 vpp_handle = session->vpp_handle;
Dave Wallace543852a2017-08-03 02:11:34 -04001358
Florin Corasf9240dc2019-01-15 08:03:17 -08001359 VDBG (1, "session %u [0x%llx] closing", session->session_index, vpp_handle);
Dave Wallace543852a2017-08-03 02:11:34 -04001360
Dave Wallace66cf6eb2017-10-26 02:51:07 -04001361 if (is_vep)
Dave Wallace543852a2017-08-03 02:11:34 -04001362 {
Florin Coras134a9962018-08-28 11:32:04 -07001363 while (next_sh != ~0)
Dave Wallace19481612017-09-15 18:47:44 -04001364 {
Florin Corasf9240dc2019-01-15 08:03:17 -08001365 rv = vppcom_epoll_ctl (sh, EPOLL_CTL_DEL, next_sh, 0);
Florin Coras0d427d82018-06-27 03:24:07 -07001366 if (PREDICT_FALSE (rv < 0))
Florin Coras5e062572019-03-14 19:07:51 -07001367 VDBG (0, "vpp handle 0x%llx, sh %u: EPOLL_CTL_DEL vep_idx %u"
Florin Coras47c40e22018-11-26 17:01:36 -08001368 " failed! rv %d (%s)", vpp_handle, next_sh, vep_sh, rv,
1369 vppcom_retval_str (rv));
Dave Wallacef7f809c2017-10-03 01:48:42 -04001370
Florin Coras134a9962018-08-28 11:32:04 -07001371 next_sh = session->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04001372 }
Florin Coras9ace36d2019-10-28 13:14:17 -07001373 goto cleanup;
Dave Wallacef7f809c2017-10-03 01:48:42 -04001374 }
Florin Coras9ace36d2019-10-28 13:14:17 -07001375
1376 if (session->is_vep_session)
Dave Wallacef7f809c2017-10-03 01:48:42 -04001377 {
Florin Coras9ace36d2019-10-28 13:14:17 -07001378 rv = vppcom_epoll_ctl (vep_sh, EPOLL_CTL_DEL, sh, 0);
1379 if (rv < 0)
1380 VDBG (0, "session %u [0x%llx]: EPOLL_CTL_DEL vep_idx %u "
1381 "failed! rv %d (%s)", session->session_index, vpp_handle,
1382 vep_sh, rv, vppcom_retval_str (rv));
Dave Wallace19481612017-09-15 18:47:44 -04001383 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05001384
Florin Coras9ace36d2019-10-28 13:14:17 -07001385 if (!do_disconnect)
1386 {
1387 VDBG (1, "session %u [0x%llx] disconnect skipped",
1388 session->session_index, vpp_handle);
1389 goto cleanup;
1390 }
1391
1392 if (state & STATE_LISTEN)
1393 {
1394 rv = vppcom_session_unbind (sh);
1395 if (PREDICT_FALSE (rv < 0))
1396 VDBG (0, "session %u [0x%llx]: listener unbind failed! "
1397 "rv %d (%s)", session->session_index, vpp_handle, rv,
1398 vppcom_retval_str (rv));
1399 return rv;
1400 }
1401 else if ((state & STATE_OPEN)
1402 || (vcl_session_is_connectable_listener (wrk, session)))
1403 {
1404 rv = vppcom_session_disconnect (sh);
1405 if (PREDICT_FALSE (rv < 0))
1406 VDBG (0, "ERROR: session %u [0x%llx]: disconnect failed!"
1407 " rv %d (%s)", session->session_index, vpp_handle,
1408 rv, vppcom_retval_str (rv));
1409 }
1410 else if (state == STATE_DISCONNECT)
1411 {
1412 svm_msg_q_t *mq = vcl_session_vpp_evt_q (wrk, session);
1413 vcl_send_session_reset_reply (mq, wrk->my_client_index,
1414 session->vpp_handle, 0);
1415 }
Florin Corasadcfb152020-02-12 08:50:29 +00001416 else if (state == STATE_DETACHED)
1417 {
1418 /* Should not happen. VPP cleaned up before app confirmed close */
1419 VDBG (0, "vpp freed session %d before close", session->session_index);
1420 goto free_session;
1421 }
Florin Coras9ace36d2019-10-28 13:14:17 -07001422
Florin Coras54140622020-02-04 19:04:34 +00001423 session->session_state = STATE_CLOSED;
1424
Florin Coras9ace36d2019-10-28 13:14:17 -07001425 /* Session is removed only after vpp confirms the disconnect */
1426 return rv;
Florin Coras0ef8ef22019-01-18 08:37:13 -08001427
Florin Corasf9240dc2019-01-15 08:03:17 -08001428cleanup:
Florin Coras30e79c22019-01-02 19:31:22 -08001429 vcl_session_table_del_vpp_handle (wrk, vpp_handle);
Florin Corasadcfb152020-02-12 08:50:29 +00001430free_session:
Florin Coras134a9962018-08-28 11:32:04 -07001431 vcl_session_free (wrk, session);
Florin Coras0d427d82018-06-27 03:24:07 -07001432 vcl_evt (VCL_EVT_CLOSE, session, rv);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001433
Dave Wallace543852a2017-08-03 02:11:34 -04001434 return rv;
1435}
1436
1437int
Florin Corasf9240dc2019-01-15 08:03:17 -08001438vppcom_session_close (uint32_t session_handle)
1439{
1440 vcl_worker_t *wrk = vcl_worker_get_current ();
1441 vcl_session_t *session;
1442
1443 session = vcl_session_get_w_handle (wrk, session_handle);
1444 if (!session)
1445 return VPPCOM_EBADFD;
1446 return vcl_session_cleanup (wrk, session, session_handle,
1447 1 /* do_disconnect */ );
1448}
1449
1450int
Florin Coras134a9962018-08-28 11:32:04 -07001451vppcom_session_bind (uint32_t session_handle, vppcom_endpt_t * ep)
Dave Wallace543852a2017-08-03 02:11:34 -04001452{
Florin Coras134a9962018-08-28 11:32:04 -07001453 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001454 vcl_session_t *session = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001455
1456 if (!ep || !ep->ip)
1457 return VPPCOM_EINVAL;
1458
Florin Coras134a9962018-08-28 11:32:04 -07001459 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001460 if (!session)
1461 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001462
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001463 if (session->is_vep)
1464 {
Florin Coras5e062572019-03-14 19:07:51 -07001465 VDBG (0, "ERROR: cannot bind to epoll session %u!",
1466 session->session_index);
Florin Coras070453d2018-08-24 17:04:27 -07001467 return VPPCOM_EBADFD;
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001468 }
1469
Florin Coras7e12d942018-06-27 14:32:43 -07001470 session->transport.is_ip4 = ep->is_ip4;
Florin Coras54693d22018-07-17 10:46:29 -07001471 if (ep->is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05001472 clib_memcpy_fast (&session->transport.lcl_ip.ip4, ep->ip,
1473 sizeof (ip4_address_t));
Florin Coras54693d22018-07-17 10:46:29 -07001474 else
Dave Barach178cf492018-11-13 16:34:13 -05001475 clib_memcpy_fast (&session->transport.lcl_ip.ip6, ep->ip,
1476 sizeof (ip6_address_t));
Florin Coras7e12d942018-06-27 14:32:43 -07001477 session->transport.lcl_port = ep->port;
Stevenac1f96d2017-10-24 16:03:58 -07001478
Florin Coras5e062572019-03-14 19:07:51 -07001479 VDBG (0, "session %u handle %u: binding to local %s address %U port %u, "
1480 "proto %s", session->session_index, session_handle,
Florin Coras7e12d942018-06-27 14:32:43 -07001481 session->transport.is_ip4 ? "IPv4" : "IPv6",
1482 format_ip46_address, &session->transport.lcl_ip,
1483 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1484 clib_net_to_host_u16 (session->transport.lcl_port),
Ping Yu34a3a082018-11-30 19:16:17 -05001485 vppcom_proto_str (session->session_type));
Florin Coras0d427d82018-06-27 03:24:07 -07001486 vcl_evt (VCL_EVT_BIND, session);
Florin Coras460dce62018-07-27 05:45:06 -07001487
1488 if (session->session_type == VPPCOM_PROTO_UDP)
Florin Coras134a9962018-08-28 11:32:04 -07001489 vppcom_session_listen (session_handle, 10);
Florin Coras460dce62018-07-27 05:45:06 -07001490
Florin Coras070453d2018-08-24 17:04:27 -07001491 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001492}
1493
1494int
Florin Coras134a9962018-08-28 11:32:04 -07001495vppcom_session_listen (uint32_t listen_sh, uint32_t q_len)
Dave Wallace543852a2017-08-03 02:11:34 -04001496{
Florin Coras134a9962018-08-28 11:32:04 -07001497 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001498 vcl_session_t *listen_session = 0;
Dave Wallaceee45d412017-11-24 21:44:06 -05001499 u64 listen_vpp_handle;
Florin Coras070453d2018-08-24 17:04:27 -07001500 int rv;
1501
Florin Coras134a9962018-08-28 11:32:04 -07001502 listen_session = vcl_session_get_w_handle (wrk, listen_sh);
Florin Coras05ecfcc2018-12-12 18:19:39 -08001503 if (!listen_session || listen_session->is_vep)
Florin Coras070453d2018-08-24 17:04:27 -07001504 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001505
Keith Burns (alagalah)aba98de2018-02-22 03:23:40 -08001506 if (q_len == 0 || q_len == ~0)
1507 q_len = vcm->cfg.listen_queue_size;
1508
Dave Wallaceee45d412017-11-24 21:44:06 -05001509 listen_vpp_handle = listen_session->vpp_handle;
Florin Coras7e12d942018-06-27 14:32:43 -07001510 if (listen_session->session_state & STATE_LISTEN)
Dave Wallacee695cb42017-11-02 22:04:42 -04001511 {
Florin Coras05ecfcc2018-12-12 18:19:39 -08001512 VDBG (0, "session %u [0x%llx]: already in listen state!",
1513 listen_sh, listen_vpp_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001514 return VPPCOM_OK;
Dave Wallacee695cb42017-11-02 22:04:42 -04001515 }
1516
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001517 VDBG (0, "session %u: sending vpp listen request...", listen_sh);
Dave Wallace543852a2017-08-03 02:11:34 -04001518
Florin Coras070453d2018-08-24 17:04:27 -07001519 /*
1520 * Send listen request to vpp and wait for reply
1521 */
Florin Coras458089b2019-08-21 16:20:44 -07001522 vcl_send_session_listen (wrk, listen_session);
Florin Coras134a9962018-08-28 11:32:04 -07001523 rv = vppcom_wait_for_session_state_change (listen_session->session_index,
1524 STATE_LISTEN,
1525 vcm->cfg.session_timeout);
Dave Wallace543852a2017-08-03 02:11:34 -04001526
Florin Coras070453d2018-08-24 17:04:27 -07001527 if (PREDICT_FALSE (rv))
Dave Wallaceee45d412017-11-24 21:44:06 -05001528 {
Florin Coras134a9962018-08-28 11:32:04 -07001529 listen_session = vcl_session_get_w_handle (wrk, listen_sh);
Florin Coras05ecfcc2018-12-12 18:19:39 -08001530 VDBG (0, "session %u [0x%llx]: listen failed! returning %d (%s)",
1531 listen_sh, listen_session->vpp_handle, rv,
1532 vppcom_retval_str (rv));
Florin Coras070453d2018-08-24 17:04:27 -07001533 return rv;
Dave Wallaceee45d412017-11-24 21:44:06 -05001534 }
1535
Florin Coras070453d2018-08-24 17:04:27 -07001536 return VPPCOM_OK;
Keith Burns (alagalah)0d2b0d52018-03-06 15:55:22 -08001537}
1538
Ping Yu34a3a082018-11-30 19:16:17 -05001539int
1540vppcom_session_tls_add_cert (uint32_t session_handle, char *cert,
1541 uint32_t cert_len)
1542{
1543
1544 vcl_worker_t *wrk = vcl_worker_get_current ();
1545 vcl_session_t *session = 0;
1546
1547 session = vcl_session_get_w_handle (wrk, session_handle);
1548 if (!session)
1549 return VPPCOM_EBADFD;
1550
1551 if (cert_len == 0 || cert_len == ~0)
1552 return VPPCOM_EBADFD;
1553
1554 /*
1555 * Send listen request to vpp and wait for reply
1556 */
1557 vppcom_send_application_tls_cert_add (session, cert, cert_len);
Florin Coras458089b2019-08-21 16:20:44 -07001558 vcm->app_state = STATE_APP_ADDING_TLS_DATA;
1559 vcl_wait_for_app_state_change (STATE_APP_READY);
Ping Yu34a3a082018-11-30 19:16:17 -05001560 return VPPCOM_OK;
1561
1562}
1563
1564int
1565vppcom_session_tls_add_key (uint32_t session_handle, char *key,
1566 uint32_t key_len)
1567{
1568
1569 vcl_worker_t *wrk = vcl_worker_get_current ();
1570 vcl_session_t *session = 0;
1571
1572 session = vcl_session_get_w_handle (wrk, session_handle);
1573 if (!session)
1574 return VPPCOM_EBADFD;
1575
1576 if (key_len == 0 || key_len == ~0)
1577 return VPPCOM_EBADFD;
1578
Ping Yu34a3a082018-11-30 19:16:17 -05001579 vppcom_send_application_tls_key_add (session, key, key_len);
Florin Coras458089b2019-08-21 16:20:44 -07001580 vcm->app_state = STATE_APP_ADDING_TLS_DATA;
1581 vcl_wait_for_app_state_change (STATE_APP_READY);
Ping Yu34a3a082018-11-30 19:16:17 -05001582 return VPPCOM_OK;
Ping Yu34a3a082018-11-30 19:16:17 -05001583}
1584
Florin Coras134a9962018-08-28 11:32:04 -07001585static int
Florin Coras5e062572019-03-14 19:07:51 -07001586validate_args_session_accept_ (vcl_worker_t * wrk, vcl_session_t * ls)
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001587{
Florin Coras5e062572019-03-14 19:07:51 -07001588 if (ls->is_vep)
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001589 {
Florin Coras5e062572019-03-14 19:07:51 -07001590 VDBG (0, "ERROR: cannot accept on epoll session %u!",
1591 ls->session_index);
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001592 return VPPCOM_EBADFD;
1593 }
1594
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001595 if ((ls->session_state != STATE_LISTEN)
1596 && (!vcl_session_is_connectable_listener (wrk, ls)))
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001597 {
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001598 VDBG (0,
1599 "ERROR: session [0x%llx]: not in listen state! state 0x%x"
1600 " (%s)", ls->vpp_handle, ls->session_state,
Florin Coras5e062572019-03-14 19:07:51 -07001601 vppcom_session_state_str (ls->session_state));
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001602 return VPPCOM_EBADFD;
1603 }
1604 return VPPCOM_OK;
1605}
1606
1607int
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001608vppcom_unformat_proto (uint8_t * proto, char *proto_str)
1609{
1610 if (!strcmp (proto_str, "TCP"))
1611 *proto = VPPCOM_PROTO_TCP;
1612 else if (!strcmp (proto_str, "tcp"))
1613 *proto = VPPCOM_PROTO_TCP;
1614 else if (!strcmp (proto_str, "UDP"))
1615 *proto = VPPCOM_PROTO_UDP;
1616 else if (!strcmp (proto_str, "udp"))
1617 *proto = VPPCOM_PROTO_UDP;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001618 else if (!strcmp (proto_str, "TLS"))
1619 *proto = VPPCOM_PROTO_TLS;
1620 else if (!strcmp (proto_str, "tls"))
1621 *proto = VPPCOM_PROTO_TLS;
1622 else if (!strcmp (proto_str, "QUIC"))
1623 *proto = VPPCOM_PROTO_QUIC;
1624 else if (!strcmp (proto_str, "quic"))
1625 *proto = VPPCOM_PROTO_QUIC;
1626 else
1627 return 1;
1628 return 0;
1629}
1630
1631int
Florin Coras134a9962018-08-28 11:32:04 -07001632vppcom_session_accept (uint32_t listen_session_handle, vppcom_endpt_t * ep,
Dave Wallace048b1d62018-01-03 22:24:41 -05001633 uint32_t flags)
Dave Wallace543852a2017-08-03 02:11:34 -04001634{
Florin Coras3c7d4f92018-12-14 11:28:43 -08001635 u32 client_session_index = ~0, listen_session_index, accept_flags = 0;
Florin Coras134a9962018-08-28 11:32:04 -07001636 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras54693d22018-07-17 10:46:29 -07001637 session_accepted_msg_t accepted_msg;
Florin Coras7e12d942018-06-27 14:32:43 -07001638 vcl_session_t *listen_session = 0;
1639 vcl_session_t *client_session = 0;
Florin Coras54693d22018-07-17 10:46:29 -07001640 vcl_session_msg_t *evt;
Florin Coras54693d22018-07-17 10:46:29 -07001641 svm_msg_q_msg_t msg;
1642 session_event_t *e;
1643 u8 is_nonblocking;
1644 int rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001645
Florin Coras134a9962018-08-28 11:32:04 -07001646 listen_session = vcl_session_get_w_handle (wrk, listen_session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001647 if (!listen_session)
1648 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001649
Florin Coras134a9962018-08-28 11:32:04 -07001650 listen_session_index = listen_session->session_index;
1651 if ((rv = validate_args_session_accept_ (wrk, listen_session)))
Florin Coras070453d2018-08-24 17:04:27 -07001652 return rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001653
Florin Coras54693d22018-07-17 10:46:29 -07001654 if (clib_fifo_elts (listen_session->accept_evts_fifo))
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001655 {
Florin Coras54693d22018-07-17 10:46:29 -07001656 clib_fifo_sub2 (listen_session->accept_evts_fifo, evt);
Florin Coras3c7d4f92018-12-14 11:28:43 -08001657 accept_flags = evt->flags;
Florin Coras54693d22018-07-17 10:46:29 -07001658 accepted_msg = evt->accepted_msg;
1659 goto handle;
1660 }
1661
1662 is_nonblocking = VCL_SESS_ATTR_TEST (listen_session->attr,
1663 VCL_SESS_ATTR_NONBLOCK);
Florin Coras54693d22018-07-17 10:46:29 -07001664 while (1)
1665 {
Carl Smith592a9092019-11-12 14:57:37 +13001666 if (svm_msg_q_is_empty (wrk->app_event_queue) && is_nonblocking)
1667 return VPPCOM_EAGAIN;
1668
Florin Coras134a9962018-08-28 11:32:04 -07001669 if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_WAIT, 0))
Florin Coras54693d22018-07-17 10:46:29 -07001670 return VPPCOM_EAGAIN;
1671
Florin Coras134a9962018-08-28 11:32:04 -07001672 e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
Florin Coras54693d22018-07-17 10:46:29 -07001673 if (e->event_type != SESSION_CTRL_EVT_ACCEPTED)
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001674 {
wanghanlin96453fd2019-12-16 19:14:39 +08001675 vcl_handle_mq_event (wrk, e);
Florin Coras134a9962018-08-28 11:32:04 -07001676 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
Florin Coras54693d22018-07-17 10:46:29 -07001677 continue;
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001678 }
Dave Barach178cf492018-11-13 16:34:13 -05001679 clib_memcpy_fast (&accepted_msg, e->data, sizeof (accepted_msg));
Florin Coras134a9962018-08-28 11:32:04 -07001680 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
Florin Coras54693d22018-07-17 10:46:29 -07001681 break;
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001682 }
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001683
Florin Coras54693d22018-07-17 10:46:29 -07001684handle:
Keith Burns (alagalah)00f44cc2018-03-07 09:26:38 -08001685
Florin Coras00cca802019-06-06 09:38:44 -07001686 client_session_index = vcl_session_accepted_handler (wrk, &accepted_msg,
1687 listen_session_index);
1688 if (client_session_index == VCL_INVALID_SESSION_INDEX)
1689 return VPPCOM_ECONNABORTED;
1690
Florin Coras134a9962018-08-28 11:32:04 -07001691 listen_session = vcl_session_get (wrk, listen_session_index);
1692 client_session = vcl_session_get (wrk, client_session_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001693
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001694 if (flags & O_NONBLOCK)
1695 VCL_SESS_ATTR_SET (client_session->attr, VCL_SESS_ATTR_NONBLOCK);
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001696
Florin Coras5e062572019-03-14 19:07:51 -07001697 VDBG (1, "listener %u [0x%llx]: Got a connect request! session %u [0x%llx],"
1698 " flags %d, is_nonblocking %u", listen_session->session_index,
1699 listen_session->vpp_handle, client_session_index,
1700 client_session->vpp_handle, flags,
1701 VCL_SESS_ATTR_TEST (client_session->attr, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace543852a2017-08-03 02:11:34 -04001702
Dave Wallace048b1d62018-01-03 22:24:41 -05001703 if (ep)
1704 {
Florin Coras7e12d942018-06-27 14:32:43 -07001705 ep->is_ip4 = client_session->transport.is_ip4;
1706 ep->port = client_session->transport.rmt_port;
1707 if (client_session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05001708 clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip4,
1709 sizeof (ip4_address_t));
Dave Wallace048b1d62018-01-03 22:24:41 -05001710 else
Dave Barach178cf492018-11-13 16:34:13 -05001711 clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip6,
1712 sizeof (ip6_address_t));
Dave Wallace048b1d62018-01-03 22:24:41 -05001713 }
Dave Wallace60caa062017-11-10 17:07:13 -05001714
Florin Coras05ecfcc2018-12-12 18:19:39 -08001715 VDBG (0, "listener %u [0x%llx] accepted %u [0x%llx] peer: %U:%u "
Florin Coras5e062572019-03-14 19:07:51 -07001716 "local: %U:%u", listen_session_handle, listen_session->vpp_handle,
Florin Coras05ecfcc2018-12-12 18:19:39 -08001717 client_session_index, client_session->vpp_handle,
Florin Coras7e12d942018-06-27 14:32:43 -07001718 format_ip46_address, &client_session->transport.rmt_ip,
Florin Coras54693d22018-07-17 10:46:29 -07001719 client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001720 clib_net_to_host_u16 (client_session->transport.rmt_port),
Florin Coras7e12d942018-06-27 14:32:43 -07001721 format_ip46_address, &client_session->transport.lcl_ip,
Florin Coras54693d22018-07-17 10:46:29 -07001722 client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001723 clib_net_to_host_u16 (client_session->transport.lcl_port));
Florin Coras0d427d82018-06-27 03:24:07 -07001724 vcl_evt (VCL_EVT_ACCEPT, client_session, listen_session,
1725 client_session_index);
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001726
Florin Coras3c7d4f92018-12-14 11:28:43 -08001727 /*
1728 * Session might have been closed already
1729 */
1730 if (accept_flags)
1731 {
Florin Coras3c7d4f92018-12-14 11:28:43 -08001732 if (accept_flags & VCL_ACCEPTED_F_CLOSED)
Florin Corasb0f662f2018-12-27 14:51:46 -08001733 client_session->session_state = STATE_VPP_CLOSING;
Florin Coras3c7d4f92018-12-14 11:28:43 -08001734 else if (accept_flags & VCL_ACCEPTED_F_RESET)
Florin Corasb0f662f2018-12-27 14:51:46 -08001735 client_session->session_state = STATE_DISCONNECT;
Florin Coras3c7d4f92018-12-14 11:28:43 -08001736 }
Florin Corasab2f6db2018-08-31 14:31:41 -07001737 return vcl_session_handle (client_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001738}
1739
1740int
Florin Coras134a9962018-08-28 11:32:04 -07001741vppcom_session_connect (uint32_t session_handle, vppcom_endpt_t * server_ep)
Dave Wallace543852a2017-08-03 02:11:34 -04001742{
Florin Coras134a9962018-08-28 11:32:04 -07001743 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001744 vcl_session_t *session = 0;
Florin Coras134a9962018-08-28 11:32:04 -07001745 u32 session_index;
Florin Coras070453d2018-08-24 17:04:27 -07001746 int rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001747
Florin Coras134a9962018-08-28 11:32:04 -07001748 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001749 if (!session)
1750 return VPPCOM_EBADFD;
Florin Coras134a9962018-08-28 11:32:04 -07001751 session_index = session->session_index;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001752
1753 if (PREDICT_FALSE (session->is_vep))
Dave Wallace543852a2017-08-03 02:11:34 -04001754 {
Florin Coras5e062572019-03-14 19:07:51 -07001755 VDBG (0, "ERROR: cannot connect epoll session %u!",
1756 session->session_index);
Florin Coras070453d2018-08-24 17:04:27 -07001757 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001758 }
1759
Florin Coras7e12d942018-06-27 14:32:43 -07001760 if (PREDICT_FALSE (session->session_state & CLIENT_STATE_OPEN))
Dave Wallace543852a2017-08-03 02:11:34 -04001761 {
Florin Coras7baeb712019-01-04 17:05:43 -08001762 VDBG (0, "session handle %u [0x%llx]: session already "
Florin Coras0d427d82018-06-27 03:24:07 -07001763 "connected to %s %U port %d proto %s, state 0x%x (%s)",
Florin Coras7baeb712019-01-04 17:05:43 -08001764 session_handle, session->vpp_handle,
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001765 session->transport.is_ip4 ? "IPv4" : "IPv6", format_ip46_address,
Florin Coras7e12d942018-06-27 14:32:43 -07001766 &session->transport.rmt_ip, session->transport.is_ip4 ?
Florin Coras0d427d82018-06-27 03:24:07 -07001767 IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001768 clib_net_to_host_u16 (session->transport.rmt_port),
Ping Yu34a3a082018-11-30 19:16:17 -05001769 vppcom_proto_str (session->session_type), session->session_state,
Florin Coras7e12d942018-06-27 14:32:43 -07001770 vppcom_session_state_str (session->session_state));
Florin Coras070453d2018-08-24 17:04:27 -07001771 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001772 }
1773
Florin Coras0a1e1832020-03-29 18:54:04 +00001774 /* Attempt to connect a connectionless listener */
1775 if (PREDICT_FALSE (session->session_state & STATE_LISTEN))
1776 {
1777 if (session->session_type != VPPCOM_PROTO_UDP)
1778 return VPPCOM_EINVAL;
1779 vcl_send_session_unlisten (wrk, session);
1780 session->session_state = STATE_CLOSED;
1781 }
1782
Florin Coras7e12d942018-06-27 14:32:43 -07001783 session->transport.is_ip4 = server_ep->is_ip4;
Florin Corasef7cbf62019-10-17 09:56:27 -07001784 vcl_ip_copy_from_ep (&session->transport.rmt_ip, server_ep);
Florin Coras7e12d942018-06-27 14:32:43 -07001785 session->transport.rmt_port = server_ep->port;
Nathan Skrzypczak8ac1d6d2019-07-17 11:02:20 +02001786 session->parent_handle = VCL_INVALID_SESSION_HANDLE;
Florin Coras0a1e1832020-03-29 18:54:04 +00001787 session->flags |= VCL_SESSION_F_CONNECTED;
Dave Wallace543852a2017-08-03 02:11:34 -04001788
Florin Coras0a1e1832020-03-29 18:54:04 +00001789 VDBG (0, "session handle %u (%s): connecting to peer %s %U "
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001790 "port %d proto %s", session_handle,
Florin Coras0a1e1832020-03-29 18:54:04 +00001791 vppcom_session_state_str (session->session_state),
Florin Coras7e12d942018-06-27 14:32:43 -07001792 session->transport.is_ip4 ? "IPv4" : "IPv6",
Florin Coras0d427d82018-06-27 03:24:07 -07001793 format_ip46_address,
Florin Coras7e12d942018-06-27 14:32:43 -07001794 &session->transport.rmt_ip, session->transport.is_ip4 ?
Florin Coras0d427d82018-06-27 03:24:07 -07001795 IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001796 clib_net_to_host_u16 (session->transport.rmt_port),
Ping Yu34a3a082018-11-30 19:16:17 -05001797 vppcom_proto_str (session->session_type));
Dave Wallace543852a2017-08-03 02:11:34 -04001798
Florin Coras458089b2019-08-21 16:20:44 -07001799 vcl_send_session_connect (wrk, session);
Florin Coras57c88932019-08-29 12:03:17 -07001800
1801 if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK))
Florin Corasa5ea8212020-08-24 21:23:51 -07001802 {
1803 session->session_state = STATE_CONNECT;
1804 return VPPCOM_EINPROGRESS;
1805 }
Florin Coras57c88932019-08-29 12:03:17 -07001806
1807 /*
1808 * Wait for reply from vpp if blocking
1809 */
Florin Coras070453d2018-08-24 17:04:27 -07001810 rv = vppcom_wait_for_session_state_change (session_index, STATE_CONNECT,
1811 vcm->cfg.session_timeout);
Dave Wallace4878cbe2017-11-21 03:45:09 -05001812
Florin Coras134a9962018-08-28 11:32:04 -07001813 session = vcl_session_get (wrk, session_index);
Florin Coras5e062572019-03-14 19:07:51 -07001814 VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1815 session->vpp_handle, rv ? "failed" : "succeeded");
Dave Wallaceee45d412017-11-24 21:44:06 -05001816
Dave Wallace4878cbe2017-11-21 03:45:09 -05001817 return rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001818}
1819
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001820int
1821vppcom_session_stream_connect (uint32_t session_handle,
1822 uint32_t parent_session_handle)
1823{
1824 vcl_worker_t *wrk = vcl_worker_get_current ();
1825 vcl_session_t *session, *parent_session;
1826 u32 session_index, parent_session_index;
1827 int rv;
1828
1829 session = vcl_session_get_w_handle (wrk, session_handle);
1830 if (!session)
1831 return VPPCOM_EBADFD;
1832 parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1833 if (!parent_session)
1834 return VPPCOM_EBADFD;
1835
1836 session_index = session->session_index;
1837 parent_session_index = parent_session->session_index;
1838 if (PREDICT_FALSE (session->is_vep))
1839 {
1840 VDBG (0, "ERROR: cannot connect epoll session %u!",
1841 session->session_index);
1842 return VPPCOM_EBADFD;
1843 }
1844
1845 if (PREDICT_FALSE (session->session_state & CLIENT_STATE_OPEN))
1846 {
1847 VDBG (0, "session handle %u [0x%llx]: session already "
1848 "connected to session %u [0x%llx] proto %s, state 0x%x (%s)",
1849 session_handle, session->vpp_handle,
1850 parent_session_handle, parent_session->vpp_handle,
1851 vppcom_proto_str (session->session_type), session->session_state,
1852 vppcom_session_state_str (session->session_state));
1853 return VPPCOM_OK;
1854 }
1855
1856 /* Connect to quic session specifics */
1857 session->transport.is_ip4 = parent_session->transport.is_ip4;
1858 session->transport.rmt_ip.ip4.as_u32 = (uint32_t) 1;
1859 session->transport.rmt_port = 0;
Nathan Skrzypczak8ac1d6d2019-07-17 11:02:20 +02001860 session->parent_handle = parent_session->vpp_handle;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001861
1862 VDBG (0, "session handle %u: connecting to session %u [0x%llx]",
1863 session_handle, parent_session_handle, parent_session->vpp_handle);
1864
1865 /*
1866 * Send connect request and wait for reply from vpp
1867 */
Florin Coras458089b2019-08-21 16:20:44 -07001868 vcl_send_session_connect (wrk, session);
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001869 rv = vppcom_wait_for_session_state_change (session_index, STATE_CONNECT,
1870 vcm->cfg.session_timeout);
1871
1872 session->listener_index = parent_session_index;
1873 parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
Florin Coras028eaf02019-07-19 12:15:52 -07001874 if (parent_session)
1875 parent_session->n_accepted_sessions++;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001876
1877 session = vcl_session_get (wrk, session_index);
1878 VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1879 session->vpp_handle, rv ? "failed" : "succeeded");
1880
1881 return rv;
1882}
1883
Florin Coras54693d22018-07-17 10:46:29 -07001884static u8
1885vcl_is_rx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
1886{
Florin Corasc0737e92019-03-04 14:19:39 -08001887 return (e->event_type == SESSION_IO_EVT_RX && e->session_index == sid);
Florin Coras54693d22018-07-17 10:46:29 -07001888}
1889
Steven58f464e2017-10-25 12:33:12 -07001890static inline int
Florin Coras134a9962018-08-28 11:32:04 -07001891vppcom_session_read_internal (uint32_t session_handle, void *buf, int n,
Steven58f464e2017-10-25 12:33:12 -07001892 u8 peek)
Dave Wallace543852a2017-08-03 02:11:34 -04001893{
Florin Coras134a9962018-08-28 11:32:04 -07001894 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras6d0106e2019-01-29 20:11:58 -08001895 int n_read = 0, is_nonblocking;
Florin Coras460dce62018-07-27 05:45:06 -07001896 vcl_session_t *s = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001897 svm_fifo_t *rx_fifo;
Florin Coras54693d22018-07-17 10:46:29 -07001898 svm_msg_q_msg_t msg;
1899 session_event_t *e;
1900 svm_msg_q_t *mq;
Florin Coras41c9e042018-09-11 00:10:41 -07001901 u8 is_ct;
Dave Wallace543852a2017-08-03 02:11:34 -04001902
Florin Coras070453d2018-08-24 17:04:27 -07001903 if (PREDICT_FALSE (!buf))
1904 return VPPCOM_EINVAL;
Dave Wallace543852a2017-08-03 02:11:34 -04001905
Florin Coras134a9962018-08-28 11:32:04 -07001906 s = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras2cba8532018-09-11 16:33:36 -07001907 if (PREDICT_FALSE (!s || s->is_vep))
Florin Coras070453d2018-08-24 17:04:27 -07001908 return VPPCOM_EBADFD;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001909
Florin Coras6d0106e2019-01-29 20:11:58 -08001910 if (PREDICT_FALSE (!vcl_session_is_open (s)))
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001911 {
Florin Coras5e062572019-03-14 19:07:51 -07001912 VDBG (0, "session %u[0x%llx] is not open! state 0x%x (%s)",
Florin Coras6d0106e2019-01-29 20:11:58 -08001913 s->session_index, s->vpp_handle, s->session_state,
1914 vppcom_session_state_str (s->session_state));
1915 return vcl_session_closed_error (s);
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001916 }
1917
Florin Coras2cba8532018-09-11 16:33:36 -07001918 is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
Florin Coras41c9e042018-09-11 00:10:41 -07001919 is_ct = vcl_session_is_ct (s);
Florin Coras653e43f2019-03-04 10:56:23 -08001920 mq = wrk->app_event_queue;
1921 rx_fifo = is_ct ? s->ct_rx_fifo : s->rx_fifo;
Florin Corasaa27eb92018-10-13 12:20:01 -07001922 s->has_rx_evt = 0;
Dave Wallacef7f809c2017-10-03 01:48:42 -04001923
Sirshak Das28aa5392019-02-05 01:33:33 -06001924 if (svm_fifo_is_empty_cons (rx_fifo))
Dave Wallacef7f809c2017-10-03 01:48:42 -04001925 {
Florin Coras54693d22018-07-17 10:46:29 -07001926 if (is_nonblocking)
Florin Corasc0737e92019-03-04 14:19:39 -08001927 {
Yu Pingb2955352019-12-04 06:49:04 +08001928 if (vcl_session_is_closing (s))
1929 return vcl_session_closing_error (s);
Florin Corasc0737e92019-03-04 14:19:39 -08001930 svm_fifo_unset_event (s->rx_fifo);
1931 return VPPCOM_EWOULDBLOCK;
1932 }
Sirshak Das28aa5392019-02-05 01:33:33 -06001933 while (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras54693d22018-07-17 10:46:29 -07001934 {
Florin Coras6d0106e2019-01-29 20:11:58 -08001935 if (vcl_session_is_closing (s))
1936 return vcl_session_closing_error (s);
1937
Florin Corasc0737e92019-03-04 14:19:39 -08001938 svm_fifo_unset_event (s->rx_fifo);
Florin Coras99368312018-08-02 10:45:44 -07001939 svm_msg_q_lock (mq);
Florin Coras54693d22018-07-17 10:46:29 -07001940 if (svm_msg_q_is_empty (mq))
1941 svm_msg_q_wait (mq);
Florin Coras99368312018-08-02 10:45:44 -07001942
Florin Coras54693d22018-07-17 10:46:29 -07001943 svm_msg_q_sub_w_lock (mq, &msg);
1944 e = svm_msg_q_msg_data (mq, &msg);
Florin Coras99368312018-08-02 10:45:44 -07001945 svm_msg_q_unlock (mq);
Florin Coras41c9e042018-09-11 00:10:41 -07001946 if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
Florin Corasc0737e92019-03-04 14:19:39 -08001947 vcl_handle_mq_event (wrk, e);
Florin Coras54693d22018-07-17 10:46:29 -07001948 svm_msg_q_free_msg (mq, &msg);
Florin Coras54693d22018-07-17 10:46:29 -07001949 }
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001950 }
Florin Coras54693d22018-07-17 10:46:29 -07001951
Florin Coras460dce62018-07-27 05:45:06 -07001952 if (s->is_dgram)
Florin Coras99368312018-08-02 10:45:44 -07001953 n_read = app_recv_dgram_raw (rx_fifo, buf, n, &s->transport, 0, peek);
Dave Wallace4878cbe2017-11-21 03:45:09 -05001954 else
Florin Coras99368312018-08-02 10:45:44 -07001955 n_read = app_recv_stream_raw (rx_fifo, buf, n, 0, peek);
Florin Coras54693d22018-07-17 10:46:29 -07001956
Sirshak Das28aa5392019-02-05 01:33:33 -06001957 if (svm_fifo_is_empty_cons (rx_fifo))
Florin Corasc34118b2020-08-12 18:58:25 -07001958 {
1959 svm_fifo_unset_event (s->rx_fifo);
1960 if (!svm_fifo_is_empty_cons (rx_fifo)
1961 && svm_fifo_set_event (s->rx_fifo) && is_nonblocking)
1962 {
1963 session_event_t *e;
1964 vec_add2 (wrk->unhandled_evts_vector, e, 1);
1965 e->event_type = SESSION_IO_EVT_RX;
1966 e->session_index = s->session_index;
1967 }
1968 }
Florin Coras41c9e042018-09-11 00:10:41 -07001969
Florin Coras17ec5772020-08-12 20:46:29 -07001970 if (PREDICT_FALSE (svm_fifo_needs_deq_ntf (rx_fifo, n_read)))
Florin Coras317b8e02019-04-17 09:57:46 -07001971 {
Florin Coras17ec5772020-08-12 20:46:29 -07001972 svm_fifo_clear_deq_ntf (rx_fifo);
Florin Coras317b8e02019-04-17 09:57:46 -07001973 app_send_io_evt_to_vpp (s->vpp_evt_q, s->rx_fifo->master_session_index,
1974 SESSION_IO_EVT_RX, SVM_Q_WAIT);
Florin Coras317b8e02019-04-17 09:57:46 -07001975 }
1976
Florin Coras5e062572019-03-14 19:07:51 -07001977 VDBG (2, "session %u[0x%llx]: read %d bytes from (%p)", s->session_index,
1978 s->vpp_handle, n_read, rx_fifo);
Florin Coras2cba8532018-09-11 16:33:36 -07001979
Florin Coras54693d22018-07-17 10:46:29 -07001980 return n_read;
Dave Wallace543852a2017-08-03 02:11:34 -04001981}
1982
Steven58f464e2017-10-25 12:33:12 -07001983int
Florin Coras134a9962018-08-28 11:32:04 -07001984vppcom_session_read (uint32_t session_handle, void *buf, size_t n)
Steven58f464e2017-10-25 12:33:12 -07001985{
Florin Coras134a9962018-08-28 11:32:04 -07001986 return (vppcom_session_read_internal (session_handle, buf, n, 0));
Steven58f464e2017-10-25 12:33:12 -07001987}
1988
1989static int
Florin Coras134a9962018-08-28 11:32:04 -07001990vppcom_session_peek (uint32_t session_handle, void *buf, int n)
Steven58f464e2017-10-25 12:33:12 -07001991{
Florin Coras134a9962018-08-28 11:32:04 -07001992 return (vppcom_session_read_internal (session_handle, buf, n, 1));
Steven58f464e2017-10-25 12:33:12 -07001993}
1994
Florin Coras2cba8532018-09-11 16:33:36 -07001995int
1996vppcom_session_read_segments (uint32_t session_handle,
1997 vppcom_data_segments_t ds)
1998{
1999 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras6d0106e2019-01-29 20:11:58 -08002000 int n_read = 0, is_nonblocking;
Florin Coras2cba8532018-09-11 16:33:36 -07002001 vcl_session_t *s = 0;
2002 svm_fifo_t *rx_fifo;
2003 svm_msg_q_msg_t msg;
2004 session_event_t *e;
2005 svm_msg_q_t *mq;
2006 u8 is_ct;
2007
2008 s = vcl_session_get_w_handle (wrk, session_handle);
2009 if (PREDICT_FALSE (!s || s->is_vep))
2010 return VPPCOM_EBADFD;
2011
Florin Coras6d0106e2019-01-29 20:11:58 -08002012 if (PREDICT_FALSE (!vcl_session_is_open (s)))
2013 return vcl_session_closed_error (s);
Florin Coras2cba8532018-09-11 16:33:36 -07002014
2015 is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
2016 is_ct = vcl_session_is_ct (s);
2017 mq = is_ct ? s->our_evt_q : wrk->app_event_queue;
2018 rx_fifo = s->rx_fifo;
Florin Corasaa27eb92018-10-13 12:20:01 -07002019 s->has_rx_evt = 0;
Florin Coras2cba8532018-09-11 16:33:36 -07002020
Florin Coras653e43f2019-03-04 10:56:23 -08002021 if (is_ct)
2022 svm_fifo_unset_event (s->rx_fifo);
2023
Sirshak Das28aa5392019-02-05 01:33:33 -06002024 if (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras2cba8532018-09-11 16:33:36 -07002025 {
2026 if (is_nonblocking)
2027 {
2028 svm_fifo_unset_event (rx_fifo);
Florin Corasaa27eb92018-10-13 12:20:01 -07002029 return VPPCOM_EWOULDBLOCK;
Florin Coras2cba8532018-09-11 16:33:36 -07002030 }
Sirshak Das28aa5392019-02-05 01:33:33 -06002031 while (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras2cba8532018-09-11 16:33:36 -07002032 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002033 if (vcl_session_is_closing (s))
2034 return vcl_session_closing_error (s);
2035
Florin Coras2cba8532018-09-11 16:33:36 -07002036 svm_fifo_unset_event (rx_fifo);
2037 svm_msg_q_lock (mq);
2038 if (svm_msg_q_is_empty (mq))
2039 svm_msg_q_wait (mq);
2040
2041 svm_msg_q_sub_w_lock (mq, &msg);
2042 e = svm_msg_q_msg_data (mq, &msg);
2043 svm_msg_q_unlock (mq);
2044 if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
Florin Coras05ce4b82018-12-15 18:30:43 -08002045 vcl_handle_mq_event (wrk, e);
Florin Coras2cba8532018-09-11 16:33:36 -07002046 svm_msg_q_free_msg (mq, &msg);
Florin Coras2cba8532018-09-11 16:33:36 -07002047 }
2048 }
2049
Florin Coras88001c62019-04-24 14:44:46 -07002050 n_read = svm_fifo_segments (rx_fifo, (svm_fifo_seg_t *) ds);
Florin Coras2cba8532018-09-11 16:33:36 -07002051 svm_fifo_unset_event (rx_fifo);
2052
Florin Coras2cba8532018-09-11 16:33:36 -07002053 return n_read;
2054}
2055
2056void
2057vppcom_session_free_segments (uint32_t session_handle,
2058 vppcom_data_segments_t ds)
2059{
2060 vcl_worker_t *wrk = vcl_worker_get_current ();
2061 vcl_session_t *s;
2062
2063 s = vcl_session_get_w_handle (wrk, session_handle);
2064 if (PREDICT_FALSE (!s || s->is_vep))
2065 return;
2066
Florin Coras88001c62019-04-24 14:44:46 -07002067 svm_fifo_segments_free (s->rx_fifo, (svm_fifo_seg_t *) ds);
Florin Coras2cba8532018-09-11 16:33:36 -07002068}
2069
Florin Coras2cba8532018-09-11 16:33:36 -07002070int
2071vppcom_data_segment_copy (void *buf, vppcom_data_segments_t ds, u32 max_bytes)
2072{
2073 u32 first_copy = clib_min (ds[0].len, max_bytes);
Dave Barach178cf492018-11-13 16:34:13 -05002074 clib_memcpy_fast (buf, ds[0].data, first_copy);
Florin Coras2cba8532018-09-11 16:33:36 -07002075 if (first_copy < max_bytes)
2076 {
Dave Barach178cf492018-11-13 16:34:13 -05002077 clib_memcpy_fast (buf + first_copy, ds[1].data,
2078 clib_min (ds[1].len, max_bytes - first_copy));
Florin Coras2cba8532018-09-11 16:33:36 -07002079 }
2080 return 0;
2081}
2082
Florin Coras54693d22018-07-17 10:46:29 -07002083static u8
2084vcl_is_tx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
2085{
Florin Corasc0737e92019-03-04 14:19:39 -08002086 return (e->event_type == SESSION_IO_EVT_TX && e->session_index == sid);
Dave Wallace543852a2017-08-03 02:11:34 -04002087}
2088
Florin Coras7a2abce2020-04-05 19:25:44 +00002089always_inline u8
2090vcl_fifo_is_writeable (svm_fifo_t * f, u32 len, u8 is_dgram)
Dave Wallace543852a2017-08-03 02:11:34 -04002091{
Florin Coras7a2abce2020-04-05 19:25:44 +00002092 u32 max_enq = svm_fifo_max_enqueue_prod (f);
2093 if (is_dgram)
2094 return max_enq >= (sizeof (session_dgram_hdr_t) + len);
2095 else
2096 return max_enq > 0;
Florin Coras7a2abce2020-04-05 19:25:44 +00002097}
2098
2099always_inline int
2100vppcom_session_write_inline (vcl_worker_t * wrk, vcl_session_t * s, void *buf,
2101 size_t n, u8 is_flush, u8 is_dgram)
2102{
Florin Coras6d0106e2019-01-29 20:11:58 -08002103 int n_write, is_nonblocking;
Florin Coras460dce62018-07-27 05:45:06 -07002104 session_evt_type_t et;
Florin Coras54693d22018-07-17 10:46:29 -07002105 svm_msg_q_msg_t msg;
Florin Coras14ed6df2019-03-06 21:13:42 -08002106 svm_fifo_t *tx_fifo;
Florin Coras54693d22018-07-17 10:46:29 -07002107 session_event_t *e;
Florin Coras3c2fed52018-07-04 04:15:05 -07002108 svm_msg_q_t *mq;
Florin Coras0e88e852018-09-17 22:09:02 -07002109 u8 is_ct;
Dave Wallace543852a2017-08-03 02:11:34 -04002110
jiangxiaomingff31ac62019-11-21 23:34:56 +08002111 if (PREDICT_FALSE (!buf || n == 0))
Florin Coras070453d2018-08-24 17:04:27 -07002112 return VPPCOM_EINVAL;
Dave Wallace543852a2017-08-03 02:11:34 -04002113
Florin Coras460dce62018-07-27 05:45:06 -07002114 if (PREDICT_FALSE (s->is_vep))
Dave Wallace543852a2017-08-03 02:11:34 -04002115 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002116 VDBG (0, "ERROR: session %u [0x%llx]: cannot write to an epoll"
2117 " session!", s->session_index, s->vpp_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002118 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04002119 }
2120
Florin Coras6d0106e2019-01-29 20:11:58 -08002121 if (PREDICT_FALSE (!vcl_session_is_open (s)))
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002122 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002123 VDBG (1, "session %u [0x%llx]: is not open! state 0x%x (%s)",
2124 s->session_index, s->vpp_handle, s->session_state,
2125 vppcom_session_state_str (s->session_state));
2126 return vcl_session_closed_error (s);;
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002127 }
2128
Florin Coras0e88e852018-09-17 22:09:02 -07002129 is_ct = vcl_session_is_ct (s);
Florin Coras653e43f2019-03-04 10:56:23 -08002130 tx_fifo = is_ct ? s->ct_tx_fifo : s->tx_fifo;
Florin Coras0e88e852018-09-17 22:09:02 -07002131 is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
Sirshak Das28aa5392019-02-05 01:33:33 -06002132
Florin Coras653e43f2019-03-04 10:56:23 -08002133 mq = wrk->app_event_queue;
Florin Coras7a2abce2020-04-05 19:25:44 +00002134 if (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
Dave Wallace543852a2017-08-03 02:11:34 -04002135 {
Florin Coras54693d22018-07-17 10:46:29 -07002136 if (is_nonblocking)
2137 {
Florin Coras070453d2018-08-24 17:04:27 -07002138 return VPPCOM_EWOULDBLOCK;
Florin Coras54693d22018-07-17 10:46:29 -07002139 }
Florin Coras7a2abce2020-04-05 19:25:44 +00002140 while (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
Florin Coras54693d22018-07-17 10:46:29 -07002141 {
Florin Coras2d379d82019-06-28 12:45:12 -07002142 svm_fifo_add_want_deq_ntf (tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
Florin Coras6d0106e2019-01-29 20:11:58 -08002143 if (vcl_session_is_closing (s))
2144 return vcl_session_closing_error (s);
Florin Coras99368312018-08-02 10:45:44 -07002145 svm_msg_q_lock (mq);
Florin Corasaa27eb92018-10-13 12:20:01 -07002146 if (svm_msg_q_is_empty (mq))
2147 svm_msg_q_wait (mq);
Florin Coras0e88e852018-09-17 22:09:02 -07002148
Florin Coras54693d22018-07-17 10:46:29 -07002149 svm_msg_q_sub_w_lock (mq, &msg);
2150 e = svm_msg_q_msg_data (mq, &msg);
Florin Coras99368312018-08-02 10:45:44 -07002151 svm_msg_q_unlock (mq);
2152
Florin Coras0e88e852018-09-17 22:09:02 -07002153 if (!vcl_is_tx_evt_for_session (e, s->session_index, is_ct))
Florin Coras86f04502018-09-12 16:08:01 -07002154 vcl_handle_mq_event (wrk, e);
Florin Coras54693d22018-07-17 10:46:29 -07002155 svm_msg_q_free_msg (mq, &msg);
Florin Coras54693d22018-07-17 10:46:29 -07002156 }
Dave Wallace543852a2017-08-03 02:11:34 -04002157 }
Dave Wallace543852a2017-08-03 02:11:34 -04002158
Florin Coras653e43f2019-03-04 10:56:23 -08002159 et = SESSION_IO_EVT_TX;
2160 if (is_flush && !is_ct)
Florin Coras42ceddb2018-12-12 10:56:01 -08002161 et = SESSION_IO_EVT_TX_FLUSH;
2162
Florin Coras7a2abce2020-04-05 19:25:44 +00002163 if (is_dgram)
Florin Coras460dce62018-07-27 05:45:06 -07002164 n_write = app_send_dgram_raw (tx_fifo, &s->transport,
Florin Coras653e43f2019-03-04 10:56:23 -08002165 s->vpp_evt_q, buf, n, et,
Florin Coras14ed6df2019-03-06 21:13:42 -08002166 0 /* do_evt */ , SVM_Q_WAIT);
Florin Coras460dce62018-07-27 05:45:06 -07002167 else
2168 n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
Florin Coras14ed6df2019-03-06 21:13:42 -08002169 0 /* do_evt */ , SVM_Q_WAIT);
Florin Coras653e43f2019-03-04 10:56:23 -08002170
Florin Coras14ed6df2019-03-06 21:13:42 -08002171 if (svm_fifo_set_event (s->tx_fifo))
Florin Corasc0737e92019-03-04 14:19:39 -08002172 app_send_io_evt_to_vpp (s->vpp_evt_q, s->tx_fifo->master_session_index,
2173 et, SVM_Q_WAIT);
Keith Burns (alagalah)410bcca2018-03-23 13:42:49 -07002174
Florin Coras460dce62018-07-27 05:45:06 -07002175 ASSERT (n_write > 0);
Dave Wallace543852a2017-08-03 02:11:34 -04002176
Florin Coras6d0106e2019-01-29 20:11:58 -08002177 VDBG (2, "session %u [0x%llx]: wrote %d bytes", s->session_index,
2178 s->vpp_handle, n_write);
Florin Coras0e88e852018-09-17 22:09:02 -07002179
Florin Coras54693d22018-07-17 10:46:29 -07002180 return n_write;
Dave Wallace543852a2017-08-03 02:11:34 -04002181}
2182
Florin Coras42ceddb2018-12-12 10:56:01 -08002183int
2184vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
2185{
Florin Coras7a2abce2020-04-05 19:25:44 +00002186 vcl_worker_t *wrk = vcl_worker_get_current ();
2187 vcl_session_t *s;
2188
2189 s = vcl_session_get_w_handle (wrk, session_handle);
2190 if (PREDICT_FALSE (!s))
2191 return VPPCOM_EBADFD;
2192
2193 return vppcom_session_write_inline (wrk, s, buf, n,
2194 0 /* is_flush */ , s->is_dgram ? 1 : 0);
Florin Coras42ceddb2018-12-12 10:56:01 -08002195}
2196
Florin Corasb0f662f2018-12-27 14:51:46 -08002197int
2198vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
2199{
Florin Coras7a2abce2020-04-05 19:25:44 +00002200 vcl_worker_t *wrk = vcl_worker_get_current ();
2201 vcl_session_t *s;
2202
2203 s = vcl_session_get_w_handle (wrk, session_handle);
2204 if (PREDICT_FALSE (!s))
2205 return VPPCOM_EBADFD;
2206
2207 return vppcom_session_write_inline (wrk, s, buf, n,
2208 1 /* is_flush */ , s->is_dgram ? 1 : 0);
Florin Corasb0f662f2018-12-27 14:51:46 -08002209}
2210
Florin Corasc0737e92019-03-04 14:19:39 -08002211#define vcl_fifo_rx_evt_valid_or_break(_s) \
Florin Corasbd52e462019-10-28 13:22:37 -07002212if (PREDICT_FALSE (!_s->rx_fifo)) \
2213 break; \
Florin Corasc0737e92019-03-04 14:19:39 -08002214if (PREDICT_FALSE (svm_fifo_is_empty (_s->rx_fifo))) \
2215 { \
2216 if (!vcl_session_is_ct (_s)) \
2217 { \
2218 svm_fifo_unset_event (_s->rx_fifo); \
2219 if (svm_fifo_is_empty (_s->rx_fifo)) \
2220 break; \
2221 } \
2222 else if (svm_fifo_is_empty (_s->ct_rx_fifo)) \
2223 { \
Florin Coras2f630182020-08-13 00:17:51 -07002224 svm_fifo_unset_event (_s->rx_fifo); /* rx evts on actual fifo*/ \
Florin Corasc0737e92019-03-04 14:19:39 -08002225 if (svm_fifo_is_empty (_s->ct_rx_fifo)) \
2226 break; \
2227 } \
2228 } \
Florin Coras6d4bb422018-09-04 22:07:27 -07002229
Florin Coras86f04502018-09-12 16:08:01 -07002230static void
2231vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2232 unsigned long n_bits, unsigned long *read_map,
2233 unsigned long *write_map,
2234 unsigned long *except_map, u32 * bits_set)
Florin Coras54693d22018-07-17 10:46:29 -07002235{
2236 session_disconnected_msg_t *disconnected_msg;
Florin Coras99368312018-08-02 10:45:44 -07002237 session_connected_msg_t *connected_msg;
Florin Coras54693d22018-07-17 10:46:29 -07002238 vcl_session_t *session;
Florin Coras86f04502018-09-12 16:08:01 -07002239 u32 sid;
2240
2241 switch (e->event_type)
2242 {
Florin Corasc0737e92019-03-04 14:19:39 -08002243 case SESSION_IO_EVT_RX:
2244 sid = e->session_index;
Florin Coras86f04502018-09-12 16:08:01 -07002245 session = vcl_session_get (wrk, sid);
Florin Corasf49cf472020-04-19 23:12:08 +00002246 if (!session || !vcl_session_is_open (session))
Florin Coras86f04502018-09-12 16:08:01 -07002247 break;
Florin Corasfff68f72019-03-13 16:01:38 -07002248 vcl_fifo_rx_evt_valid_or_break (session);
Florin Coras86f04502018-09-12 16:08:01 -07002249 if (sid < n_bits && read_map)
2250 {
David Johnsond9818dd2018-12-14 14:53:41 -05002251 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002252 *bits_set += 1;
2253 }
2254 break;
Florin Corasfe97da32019-03-06 10:09:04 -08002255 case SESSION_IO_EVT_TX:
Florin Corasc0737e92019-03-04 14:19:39 -08002256 sid = e->session_index;
Florin Coras86f04502018-09-12 16:08:01 -07002257 session = vcl_session_get (wrk, sid);
Florin Corasf49cf472020-04-19 23:12:08 +00002258 if (!session || !vcl_session_is_open (session))
Florin Coras86f04502018-09-12 16:08:01 -07002259 break;
2260 if (sid < n_bits && write_map)
2261 {
David Johnsond9818dd2018-12-14 14:53:41 -05002262 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002263 *bits_set += 1;
2264 }
2265 break;
Florin Coras86f04502018-09-12 16:08:01 -07002266 case SESSION_CTRL_EVT_ACCEPTED:
Florin Coras3c7d4f92018-12-14 11:28:43 -08002267 session = vcl_session_accepted (wrk,
2268 (session_accepted_msg_t *) e->data);
Florin Coras86f04502018-09-12 16:08:01 -07002269 if (!session)
Florin Coras3c7d4f92018-12-14 11:28:43 -08002270 break;
Florin Coras86f04502018-09-12 16:08:01 -07002271 sid = session->session_index;
2272 if (sid < n_bits && read_map)
2273 {
David Johnsond9818dd2018-12-14 14:53:41 -05002274 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002275 *bits_set += 1;
2276 }
2277 break;
2278 case SESSION_CTRL_EVT_CONNECTED:
2279 connected_msg = (session_connected_msg_t *) e->data;
Florin Coras57c88932019-08-29 12:03:17 -07002280 sid = vcl_session_connected_handler (wrk, connected_msg);
2281 if (sid == VCL_INVALID_SESSION_INDEX)
2282 break;
2283 if (sid < n_bits && write_map)
2284 {
2285 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2286 *bits_set += 1;
2287 }
Florin Coras86f04502018-09-12 16:08:01 -07002288 break;
2289 case SESSION_CTRL_EVT_DISCONNECTED:
2290 disconnected_msg = (session_disconnected_msg_t *) e->data;
Florin Coras3c7d4f92018-12-14 11:28:43 -08002291 session = vcl_session_disconnected_handler (wrk, disconnected_msg);
2292 if (!session)
2293 break;
2294 sid = session->session_index;
Florin Coras86f04502018-09-12 16:08:01 -07002295 if (sid < n_bits && except_map)
2296 {
David Johnsond9818dd2018-12-14 14:53:41 -05002297 clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002298 *bits_set += 1;
2299 }
2300 break;
2301 case SESSION_CTRL_EVT_RESET:
2302 sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2303 if (sid < n_bits && except_map)
2304 {
David Johnsond9818dd2018-12-14 14:53:41 -05002305 clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002306 *bits_set += 1;
2307 }
2308 break;
Florin Corasdfae9f92019-02-20 19:48:31 -08002309 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2310 vcl_session_unlisten_reply_handler (wrk, e->data);
2311 break;
Florin Coras68b7e582020-01-21 18:33:23 -08002312 case SESSION_CTRL_EVT_MIGRATED:
2313 vcl_session_migrated_handler (wrk, e->data);
2314 break;
Florin Coras9ace36d2019-10-28 13:14:17 -07002315 case SESSION_CTRL_EVT_CLEANUP:
2316 vcl_session_cleanup_handler (wrk, e->data);
2317 break;
Florin Coras30e79c22019-01-02 19:31:22 -08002318 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2319 vcl_session_worker_update_reply_handler (wrk, e->data);
2320 break;
2321 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2322 vcl_session_req_worker_update_handler (wrk, e->data);
2323 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -07002324 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2325 vcl_session_app_add_segment_handler (wrk, e->data);
2326 break;
2327 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2328 vcl_session_app_del_segment_handler (wrk, e->data);
2329 break;
hanlina3a48962020-07-13 11:09:15 +08002330 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -07002331 vcl_worker_rpc_handler (wrk, e->data);
2332 break;
Florin Coras86f04502018-09-12 16:08:01 -07002333 default:
2334 clib_warning ("unhandled: %u", e->event_type);
2335 break;
2336 }
2337}
2338
2339static int
2340vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2341 unsigned long n_bits, unsigned long *read_map,
2342 unsigned long *write_map, unsigned long *except_map,
2343 double time_to_wait, u32 * bits_set)
2344{
Florin Coras99368312018-08-02 10:45:44 -07002345 svm_msg_q_msg_t *msg;
Florin Coras54693d22018-07-17 10:46:29 -07002346 session_event_t *e;
Florin Coras86f04502018-09-12 16:08:01 -07002347 u32 i;
Florin Coras54693d22018-07-17 10:46:29 -07002348
2349 svm_msg_q_lock (mq);
2350 if (svm_msg_q_is_empty (mq))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002351 {
Florin Coras54693d22018-07-17 10:46:29 -07002352 if (*bits_set)
Dave Wallace4878cbe2017-11-21 03:45:09 -05002353 {
Florin Coras54693d22018-07-17 10:46:29 -07002354 svm_msg_q_unlock (mq);
2355 return 0;
2356 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05002357
Florin Coras54693d22018-07-17 10:46:29 -07002358 if (!time_to_wait)
2359 {
2360 svm_msg_q_unlock (mq);
2361 return 0;
2362 }
2363 else if (time_to_wait < 0)
2364 {
2365 svm_msg_q_wait (mq);
2366 }
2367 else
2368 {
2369 if (svm_msg_q_timedwait (mq, time_to_wait))
2370 {
2371 svm_msg_q_unlock (mq);
2372 return 0;
2373 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05002374 }
2375 }
Florin Corase003a1b2019-06-05 10:47:16 -07002376 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras54693d22018-07-17 10:46:29 -07002377 svm_msg_q_unlock (mq);
2378
Florin Coras134a9962018-08-28 11:32:04 -07002379 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
Florin Coras54693d22018-07-17 10:46:29 -07002380 {
Florin Coras134a9962018-08-28 11:32:04 -07002381 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
Florin Coras99368312018-08-02 10:45:44 -07002382 e = svm_msg_q_msg_data (mq, msg);
Florin Coras86f04502018-09-12 16:08:01 -07002383 vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
2384 except_map, bits_set);
Florin Coras99368312018-08-02 10:45:44 -07002385 svm_msg_q_free_msg (mq, msg);
Florin Coras54693d22018-07-17 10:46:29 -07002386 }
Florin Coras134a9962018-08-28 11:32:04 -07002387 vec_reset_length (wrk->mq_msg_vector);
Florin Coras30e79c22019-01-02 19:31:22 -08002388 vcl_handle_pending_wrk_updates (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07002389 return *bits_set;
Dave Wallace543852a2017-08-03 02:11:34 -04002390}
2391
Florin Coras99368312018-08-02 10:45:44 -07002392static int
Florin Coras294afe22019-01-07 17:49:17 -08002393vppcom_select_condvar (vcl_worker_t * wrk, int n_bits,
2394 vcl_si_set * read_map, vcl_si_set * write_map,
2395 vcl_si_set * except_map, double time_to_wait,
Florin Coras134a9962018-08-28 11:32:04 -07002396 u32 * bits_set)
Florin Coras99368312018-08-02 10:45:44 -07002397{
Florin Coras14ed6df2019-03-06 21:13:42 -08002398 double wait = 0, start = 0;
2399
2400 if (!*bits_set)
2401 {
2402 wait = time_to_wait;
2403 start = clib_time_now (&wrk->clib_time);
2404 }
2405
2406 do
2407 {
2408 vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
2409 write_map, except_map, wait, bits_set);
2410 if (*bits_set)
2411 return *bits_set;
2412 if (wait == -1)
2413 continue;
2414
2415 wait = wait - (clib_time_now (&wrk->clib_time) - start);
2416 }
2417 while (wait > 0);
2418
2419 return 0;
Florin Coras99368312018-08-02 10:45:44 -07002420}
2421
2422static int
Florin Coras294afe22019-01-07 17:49:17 -08002423vppcom_select_eventfd (vcl_worker_t * wrk, int n_bits,
2424 vcl_si_set * read_map, vcl_si_set * write_map,
2425 vcl_si_set * except_map, double time_to_wait,
Florin Coras134a9962018-08-28 11:32:04 -07002426 u32 * bits_set)
Florin Coras99368312018-08-02 10:45:44 -07002427{
2428 vcl_mq_evt_conn_t *mqc;
2429 int __clib_unused n_read;
2430 int n_mq_evts, i;
2431 u64 buf;
2432
Florin Coras134a9962018-08-28 11:32:04 -07002433 vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
2434 n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
2435 vec_len (wrk->mq_events), time_to_wait);
Florin Coras99368312018-08-02 10:45:44 -07002436 for (i = 0; i < n_mq_evts; i++)
2437 {
Florin Coras134a9962018-08-28 11:32:04 -07002438 mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
Florin Coras99368312018-08-02 10:45:44 -07002439 n_read = read (mqc->mq_fd, &buf, sizeof (buf));
Florin Coras134a9962018-08-28 11:32:04 -07002440 vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
Florin Coras99368312018-08-02 10:45:44 -07002441 except_map, 0, bits_set);
2442 }
2443
2444 return (n_mq_evts > 0 ? (int) *bits_set : 0);
2445}
2446
Dave Wallace543852a2017-08-03 02:11:34 -04002447int
Florin Coras294afe22019-01-07 17:49:17 -08002448vppcom_select (int n_bits, vcl_si_set * read_map, vcl_si_set * write_map,
2449 vcl_si_set * except_map, double time_to_wait)
Dave Wallace543852a2017-08-03 02:11:34 -04002450{
Florin Coras54693d22018-07-17 10:46:29 -07002451 u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
Florin Coras134a9962018-08-28 11:32:04 -07002452 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07002453 vcl_session_t *session = 0;
Florin Corasf49cf472020-04-19 23:12:08 +00002454 int i;
Dave Wallace543852a2017-08-03 02:11:34 -04002455
Dave Wallace7876d392017-10-19 03:53:57 -04002456 if (n_bits && read_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002457 {
Florin Coras134a9962018-08-28 11:32:04 -07002458 clib_bitmap_validate (wrk->rd_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002459 clib_memcpy_fast (wrk->rd_bitmap, read_map,
Florin Coras294afe22019-01-07 17:49:17 -08002460 vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2461 memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002462 }
Dave Wallace7876d392017-10-19 03:53:57 -04002463 if (n_bits && write_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002464 {
Florin Coras134a9962018-08-28 11:32:04 -07002465 clib_bitmap_validate (wrk->wr_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002466 clib_memcpy_fast (wrk->wr_bitmap, write_map,
Florin Coras294afe22019-01-07 17:49:17 -08002467 vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2468 memset (write_map, 0, vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002469 }
Dave Wallace7876d392017-10-19 03:53:57 -04002470 if (n_bits && except_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002471 {
Florin Coras134a9962018-08-28 11:32:04 -07002472 clib_bitmap_validate (wrk->ex_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002473 clib_memcpy_fast (wrk->ex_bitmap, except_map,
Florin Coras294afe22019-01-07 17:49:17 -08002474 vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2475 memset (except_map, 0, vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002476 }
2477
Florin Coras54693d22018-07-17 10:46:29 -07002478 if (!n_bits)
2479 return 0;
2480
2481 if (!write_map)
2482 goto check_rd;
2483
2484 /* *INDENT-OFF* */
Florin Coras134a9962018-08-28 11:32:04 -07002485 clib_bitmap_foreach (sid, wrk->wr_bitmap, ({
2486 if (!(session = vcl_session_get (wrk, sid)))
Florin Coras54693d22018-07-17 10:46:29 -07002487 {
Florin Coras5e6222a2020-04-24 17:09:25 +00002488 clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
2489 bits_set++;
2490 continue;
Florin Coras54693d22018-07-17 10:46:29 -07002491 }
2492
Florin Coras5e6222a2020-04-24 17:09:25 +00002493 if (vcl_session_write_ready (session))
Florin Coras54693d22018-07-17 10:46:29 -07002494 {
David Johnsond9818dd2018-12-14 14:53:41 -05002495 clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
Florin Coras54693d22018-07-17 10:46:29 -07002496 bits_set++;
2497 }
Florin Coras294afe22019-01-07 17:49:17 -08002498 else
Florin Coras2d379d82019-06-28 12:45:12 -07002499 svm_fifo_add_want_deq_ntf (session->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
Florin Coras54693d22018-07-17 10:46:29 -07002500 }));
2501
2502check_rd:
2503 if (!read_map)
2504 goto check_mq;
Florin Coras99368312018-08-02 10:45:44 -07002505
Florin Coras134a9962018-08-28 11:32:04 -07002506 clib_bitmap_foreach (sid, wrk->rd_bitmap, ({
2507 if (!(session = vcl_session_get (wrk, sid)))
Florin Coras54693d22018-07-17 10:46:29 -07002508 {
Florin Coras5e6222a2020-04-24 17:09:25 +00002509 clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
2510 bits_set++;
2511 continue;
Florin Coras54693d22018-07-17 10:46:29 -07002512 }
2513
Florin Coras5e6222a2020-04-24 17:09:25 +00002514 if (vcl_session_read_ready (session))
Florin Coras54693d22018-07-17 10:46:29 -07002515 {
David Johnsond9818dd2018-12-14 14:53:41 -05002516 clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
Florin Coras54693d22018-07-17 10:46:29 -07002517 bits_set++;
2518 }
2519 }));
2520 /* *INDENT-ON* */
2521
2522check_mq:
Dave Wallace543852a2017-08-03 02:11:34 -04002523
Florin Coras86f04502018-09-12 16:08:01 -07002524 for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
2525 {
2526 vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
2527 read_map, write_map, except_map, &bits_set);
2528 }
2529 vec_reset_length (wrk->unhandled_evts_vector);
2530
Florin Coras99368312018-08-02 10:45:44 -07002531 if (vcm->cfg.use_mq_eventfd)
Florin Coras134a9962018-08-28 11:32:04 -07002532 vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
Florin Coras99368312018-08-02 10:45:44 -07002533 time_to_wait, &bits_set);
2534 else
Florin Coras134a9962018-08-28 11:32:04 -07002535 vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
Florin Coras99368312018-08-02 10:45:44 -07002536 time_to_wait, &bits_set);
Florin Coras54693d22018-07-17 10:46:29 -07002537
Dave Wallace543852a2017-08-03 02:11:34 -04002538 return (bits_set);
2539}
2540
Dave Wallacef7f809c2017-10-03 01:48:42 -04002541static inline void
Florin Coras7e5e62b2019-07-30 14:08:23 -07002542vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_handle)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002543{
Florin Coras7e12d942018-06-27 14:32:43 -07002544 vcl_session_t *session;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002545 vppcom_epoll_t *vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002546 u32 sh = vep_handle;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002547
Florin Corasc0737e92019-03-04 14:19:39 -08002548 if (VPPCOM_DEBUG <= 2)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002549 return;
2550
Florin Coras7e5e62b2019-07-30 14:08:23 -07002551 session = vcl_session_get_w_handle (wrk, vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002552 if (PREDICT_FALSE (!session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002553 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002554 VDBG (0, "ERROR: Invalid vep_sh (%u)!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002555 goto done;
2556 }
2557 if (PREDICT_FALSE (!session->is_vep))
2558 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002559 VDBG (0, "ERROR: vep_sh (%u) is not a vep!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002560 goto done;
2561 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05002562 vep = &session->vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002563 VDBG (0, "vep_sh (%u): Dumping epoll chain\n"
Florin Coras5e062572019-03-14 19:07:51 -07002564 "{\n"
2565 " is_vep = %u\n"
2566 " is_vep_session = %u\n"
Florin Coras7e5e62b2019-07-30 14:08:23 -07002567 " next_sh = 0x%x (%u)\n"
2568 "}\n", vep_handle, session->is_vep, session->is_vep_session,
Florin Coras5e062572019-03-14 19:07:51 -07002569 vep->next_sh, vep->next_sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002570
Florin Coras7e5e62b2019-07-30 14:08:23 -07002571 for (sh = vep->next_sh; sh != ~0; sh = vep->next_sh)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002572 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002573 session = vcl_session_get_w_handle (wrk, sh);
Florin Coras070453d2018-08-24 17:04:27 -07002574 if (PREDICT_FALSE (!session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002575 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002576 VDBG (0, "ERROR: Invalid sh (%u)!", sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002577 goto done;
2578 }
2579 if (PREDICT_FALSE (session->is_vep))
Florin Coras5e062572019-03-14 19:07:51 -07002580 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002581 VDBG (0, "ERROR: sh (%u) is a vep!", vep_handle);
Florin Coras5e062572019-03-14 19:07:51 -07002582 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05002583 else if (PREDICT_FALSE (!session->is_vep_session))
2584 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002585 VDBG (0, "ERROR: sh (%u) is not a vep session handle!", sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002586 goto done;
2587 }
2588 vep = &session->vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002589 if (PREDICT_FALSE (vep->vep_sh != vep_handle))
2590 VDBG (0, "ERROR: session (%u) vep_sh (%u) != vep_sh (%u)!",
2591 sh, session->vep.vep_sh, vep_handle);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002592 if (session->is_vep_session)
2593 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002594 VDBG (0, "vep_sh[%u]: sh 0x%x (%u)\n"
Florin Coras5e062572019-03-14 19:07:51 -07002595 "{\n"
Florin Coras7e5e62b2019-07-30 14:08:23 -07002596 " next_sh = 0x%x (%u)\n"
2597 " prev_sh = 0x%x (%u)\n"
2598 " vep_sh = 0x%x (%u)\n"
Florin Coras5e062572019-03-14 19:07:51 -07002599 " ev.events = 0x%x\n"
2600 " ev.data.u64 = 0x%llx\n"
2601 " et_mask = 0x%x\n"
2602 "}\n",
Florin Coras7e5e62b2019-07-30 14:08:23 -07002603 vep_handle, sh, sh, vep->next_sh, vep->next_sh, vep->prev_sh,
Florin Coras5e062572019-03-14 19:07:51 -07002604 vep->prev_sh, vep->vep_sh, vep->vep_sh, vep->ev.events,
2605 vep->ev.data.u64, vep->et_mask);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002606 }
2607 }
Dave Wallacef7f809c2017-10-03 01:48:42 -04002608
2609done:
Florin Coras7e5e62b2019-07-30 14:08:23 -07002610 VDBG (0, "vep_sh (%u): Dump complete!\n", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002611}
2612
2613int
2614vppcom_epoll_create (void)
2615{
Florin Coras134a9962018-08-28 11:32:04 -07002616 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07002617 vcl_session_t *vep_session;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002618
Florin Coras134a9962018-08-28 11:32:04 -07002619 vep_session = vcl_session_alloc (wrk);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002620
2621 vep_session->is_vep = 1;
Florin Coras134a9962018-08-28 11:32:04 -07002622 vep_session->vep.vep_sh = ~0;
2623 vep_session->vep.next_sh = ~0;
2624 vep_session->vep.prev_sh = ~0;
Dave Wallace4878cbe2017-11-21 03:45:09 -05002625 vep_session->vpp_handle = ~0;
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08002626
Florin Corasa7a1a222018-12-30 17:11:31 -08002627 vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
2628 VDBG (0, "Created vep_idx %u", vep_session->session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08002629
Florin Corasab2f6db2018-08-31 14:31:41 -07002630 return vcl_session_handle (vep_session);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002631}
2632
2633int
Florin Coras134a9962018-08-28 11:32:04 -07002634vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
Dave Wallacef7f809c2017-10-03 01:48:42 -04002635 struct epoll_event *event)
2636{
Florin Coras134a9962018-08-28 11:32:04 -07002637 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07002638 vcl_session_t *vep_session;
Florin Coras070453d2018-08-24 17:04:27 -07002639 int rv = VPPCOM_OK;
Florin Coras17ec5772020-08-12 20:46:29 -07002640 vcl_session_t *s;
2641 svm_fifo_t *txf;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002642
Florin Coras134a9962018-08-28 11:32:04 -07002643 if (vep_handle == session_handle)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002644 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002645 VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002646 return VPPCOM_EINVAL;
2647 }
2648
Florin Coras134a9962018-08-28 11:32:04 -07002649 vep_session = vcl_session_get_w_handle (wrk, vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002650 if (PREDICT_FALSE (!vep_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002651 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002652 VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002653 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002654 }
2655 if (PREDICT_FALSE (!vep_session->is_vep))
2656 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002657 VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002658 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002659 }
2660
Florin Coras134a9962018-08-28 11:32:04 -07002661 ASSERT (vep_session->vep.vep_sh == ~0);
2662 ASSERT (vep_session->vep.prev_sh == ~0);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002663
Florin Coras17ec5772020-08-12 20:46:29 -07002664 s = vcl_session_get_w_handle (wrk, session_handle);
2665 if (PREDICT_FALSE (!s))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002666 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002667 VDBG (0, "Invalid session_handle (%u)!", session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002668 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002669 }
Florin Coras17ec5772020-08-12 20:46:29 -07002670 if (PREDICT_FALSE (s->is_vep))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002671 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002672 VDBG (0, "session_handle (%u) is a vep!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002673 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002674 }
2675
2676 switch (op)
2677 {
2678 case EPOLL_CTL_ADD:
2679 if (PREDICT_FALSE (!event))
2680 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002681 VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
Florin Coras070453d2018-08-24 17:04:27 -07002682 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002683 }
Florin Coras134a9962018-08-28 11:32:04 -07002684 if (vep_session->vep.next_sh != ~0)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002685 {
Florin Coras7e12d942018-06-27 14:32:43 -07002686 vcl_session_t *next_session;
Florin Corasab2f6db2018-08-31 14:31:41 -07002687 next_session = vcl_session_get_w_handle (wrk,
2688 vep_session->vep.next_sh);
Florin Coras070453d2018-08-24 17:04:27 -07002689 if (PREDICT_FALSE (!next_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002690 {
Florin Coras5e062572019-03-14 19:07:51 -07002691 VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
Florin Corasa7a1a222018-12-30 17:11:31 -08002692 "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002693 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002694 }
Florin Coras134a9962018-08-28 11:32:04 -07002695 ASSERT (next_session->vep.prev_sh == vep_handle);
2696 next_session->vep.prev_sh = session_handle;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002697 }
Florin Coras17ec5772020-08-12 20:46:29 -07002698 s->vep.next_sh = vep_session->vep.next_sh;
2699 s->vep.prev_sh = vep_handle;
2700 s->vep.vep_sh = vep_handle;
2701 s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2702 s->vep.ev = *event;
2703 s->is_vep = 0;
2704 s->is_vep_session = 1;
Florin Coras134a9962018-08-28 11:32:04 -07002705 vep_session->vep.next_sh = session_handle;
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08002706
Florin Coras17ec5772020-08-12 20:46:29 -07002707 txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2708 if (txf && (event->events & EPOLLOUT))
2709 svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Coras1bcad5c2019-01-09 20:04:38 -08002710
Florin Coras6e3c1f82020-01-15 01:30:46 +00002711 /* Generate EPOLLOUT if tx fifo not full */
Florin Coras17ec5772020-08-12 20:46:29 -07002712 if ((event->events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
hanlin475c9d72019-12-26 11:44:28 +08002713 {
2714 session_event_t e = { 0 };
2715 e.event_type = SESSION_IO_EVT_TX;
Florin Coras17ec5772020-08-12 20:46:29 -07002716 e.session_index = s->session_index;
hanlin475c9d72019-12-26 11:44:28 +08002717 vec_add1 (wrk->unhandled_evts_vector, e);
2718 }
Florin Coras6e3c1f82020-01-15 01:30:46 +00002719 /* Generate EPOLLIN if rx fifo has data */
Florin Coras17ec5772020-08-12 20:46:29 -07002720 if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
Florin Coras6e3c1f82020-01-15 01:30:46 +00002721 {
2722 session_event_t e = { 0 };
2723 e.event_type = SESSION_IO_EVT_RX;
Florin Coras17ec5772020-08-12 20:46:29 -07002724 e.session_index = s->session_index;
Florin Coras6e3c1f82020-01-15 01:30:46 +00002725 vec_add1 (wrk->unhandled_evts_vector, e);
2726 }
Florin Corasa7a1a222018-12-30 17:11:31 -08002727 VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2728 vep_handle, session_handle, event->events, event->data.u64);
Florin Coras17ec5772020-08-12 20:46:29 -07002729 vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002730 break;
2731
2732 case EPOLL_CTL_MOD:
2733 if (PREDICT_FALSE (!event))
2734 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002735 VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
Dave Wallacef7f809c2017-10-03 01:48:42 -04002736 rv = VPPCOM_EINVAL;
2737 goto done;
2738 }
Florin Coras17ec5772020-08-12 20:46:29 -07002739 else if (PREDICT_FALSE (!s->is_vep_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002740 {
Florin Coras5e062572019-03-14 19:07:51 -07002741 VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002742 rv = VPPCOM_EINVAL;
2743 goto done;
2744 }
Florin Coras17ec5772020-08-12 20:46:29 -07002745 else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002746 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002747 VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07002748 session_handle, s->vep.vep_sh, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002749 rv = VPPCOM_EINVAL;
2750 goto done;
2751 }
hanlin475c9d72019-12-26 11:44:28 +08002752
2753 /* Generate EPOLLOUT when tx_fifo/ct_tx_fifo not full */
2754 if ((event->events & EPOLLOUT) &&
Florin Coras17ec5772020-08-12 20:46:29 -07002755 !(s->vep.ev.events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
hanlin475c9d72019-12-26 11:44:28 +08002756 {
2757 session_event_t e = { 0 };
2758 e.event_type = SESSION_IO_EVT_TX;
Florin Coras17ec5772020-08-12 20:46:29 -07002759 e.session_index = s->session_index;
hanlin475c9d72019-12-26 11:44:28 +08002760 vec_add1 (wrk->unhandled_evts_vector, e);
2761 }
Florin Coras17ec5772020-08-12 20:46:29 -07002762 s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2763 s->vep.ev = *event;
2764 txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2765 if (event->events & EPOLLOUT)
2766 svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2767 else
2768 svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Corasa7a1a222018-12-30 17:11:31 -08002769 VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2770 vep_handle, session_handle, event->events, event->data.u64);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002771 break;
2772
2773 case EPOLL_CTL_DEL:
Florin Coras17ec5772020-08-12 20:46:29 -07002774 if (PREDICT_FALSE (!s->is_vep_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002775 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002776 VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002777 rv = VPPCOM_EINVAL;
2778 goto done;
2779 }
Florin Coras17ec5772020-08-12 20:46:29 -07002780 else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002781 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002782 VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07002783 session_handle, s->vep.vep_sh, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002784 rv = VPPCOM_EINVAL;
2785 goto done;
2786 }
2787
Florin Coras17ec5772020-08-12 20:46:29 -07002788 if (s->vep.prev_sh == vep_handle)
2789 vep_session->vep.next_sh = s->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002790 else
2791 {
Florin Coras7e12d942018-06-27 14:32:43 -07002792 vcl_session_t *prev_session;
Florin Coras17ec5772020-08-12 20:46:29 -07002793 prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
Florin Coras070453d2018-08-24 17:04:27 -07002794 if (PREDICT_FALSE (!prev_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002795 {
Florin Coras5e062572019-03-14 19:07:51 -07002796 VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07002797 s->vep.prev_sh, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002798 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002799 }
Florin Coras134a9962018-08-28 11:32:04 -07002800 ASSERT (prev_session->vep.next_sh == session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07002801 prev_session->vep.next_sh = s->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002802 }
Florin Coras17ec5772020-08-12 20:46:29 -07002803 if (s->vep.next_sh != ~0)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002804 {
Florin Coras7e12d942018-06-27 14:32:43 -07002805 vcl_session_t *next_session;
Florin Coras17ec5772020-08-12 20:46:29 -07002806 next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
Florin Coras070453d2018-08-24 17:04:27 -07002807 if (PREDICT_FALSE (!next_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002808 {
Florin Coras5e062572019-03-14 19:07:51 -07002809 VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07002810 s->vep.next_sh, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002811 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002812 }
Florin Coras134a9962018-08-28 11:32:04 -07002813 ASSERT (next_session->vep.prev_sh == session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07002814 next_session->vep.prev_sh = s->vep.prev_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002815 }
2816
Florin Coras17ec5772020-08-12 20:46:29 -07002817 memset (&s->vep, 0, sizeof (s->vep));
2818 s->vep.next_sh = ~0;
2819 s->vep.prev_sh = ~0;
2820 s->vep.vep_sh = ~0;
2821 s->is_vep_session = 0;
Florin Coras1bcad5c2019-01-09 20:04:38 -08002822
Florin Coras17ec5772020-08-12 20:46:29 -07002823 txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2824 if (txf)
2825 svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Coras1bcad5c2019-01-09 20:04:38 -08002826
Florin Coras5e062572019-03-14 19:07:51 -07002827 VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
Florin Corasa7a1a222018-12-30 17:11:31 -08002828 session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07002829 vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002830 break;
2831
2832 default:
Florin Corasa7a1a222018-12-30 17:11:31 -08002833 VDBG (0, "Invalid operation (%d)!", op);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002834 rv = VPPCOM_EINVAL;
2835 }
2836
Florin Coras134a9962018-08-28 11:32:04 -07002837 vep_verify_epoll_chain (wrk, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002838
2839done:
Dave Wallacef7f809c2017-10-03 01:48:42 -04002840 return rv;
2841}
2842
Florin Coras86f04502018-09-12 16:08:01 -07002843static inline void
2844vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2845 struct epoll_event *events, u32 * num_ev)
Florin Coras54693d22018-07-17 10:46:29 -07002846{
2847 session_disconnected_msg_t *disconnected_msg;
2848 session_connected_msg_t *connected_msg;
Florin Coras99368312018-08-02 10:45:44 -07002849 u32 sid = ~0, session_events;
Florin Coras3c7d4f92018-12-14 11:28:43 -08002850 u64 session_evt_data = ~0;
Florin Coras54693d22018-07-17 10:46:29 -07002851 vcl_session_t *session;
Florin Coras86f04502018-09-12 16:08:01 -07002852 u8 add_event = 0;
2853
2854 switch (e->event_type)
2855 {
Florin Coras653e43f2019-03-04 10:56:23 -08002856 case SESSION_IO_EVT_RX:
Florin Corasc0737e92019-03-04 14:19:39 -08002857 sid = e->session_index;
Florin Coras080aa502020-05-26 00:47:52 +00002858 session = vcl_session_get (wrk, sid);
2859 if (vcl_session_is_closed (session))
Florin Corasfa915f82018-12-26 16:29:06 -08002860 break;
Florin Corasc0737e92019-03-04 14:19:39 -08002861 vcl_fifo_rx_evt_valid_or_break (session);
Florin Coras86f04502018-09-12 16:08:01 -07002862 session_events = session->vep.ev.events;
Florin Corasaa27eb92018-10-13 12:20:01 -07002863 if (!(EPOLLIN & session->vep.ev.events) || session->has_rx_evt)
Florin Coras86f04502018-09-12 16:08:01 -07002864 break;
2865 add_event = 1;
2866 events[*num_ev].events |= EPOLLIN;
2867 session_evt_data = session->vep.ev.data.u64;
Florin Corasaa27eb92018-10-13 12:20:01 -07002868 session->has_rx_evt = 1;
Florin Coras86f04502018-09-12 16:08:01 -07002869 break;
Florin Coras653e43f2019-03-04 10:56:23 -08002870 case SESSION_IO_EVT_TX:
Florin Corasc0737e92019-03-04 14:19:39 -08002871 sid = e->session_index;
Florin Coras080aa502020-05-26 00:47:52 +00002872 session = vcl_session_get (wrk, sid);
2873 if (vcl_session_is_closed (session))
Florin Corasfa915f82018-12-26 16:29:06 -08002874 break;
Florin Coras86f04502018-09-12 16:08:01 -07002875 session_events = session->vep.ev.events;
2876 if (!(EPOLLOUT & session_events))
2877 break;
2878 add_event = 1;
2879 events[*num_ev].events |= EPOLLOUT;
2880 session_evt_data = session->vep.ev.data.u64;
Florin Coras17ec5772020-08-12 20:46:29 -07002881 svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (session) ?
2882 session->ct_tx_fifo : session->tx_fifo);
Florin Coras86f04502018-09-12 16:08:01 -07002883 break;
Florin Coras86f04502018-09-12 16:08:01 -07002884 case SESSION_CTRL_EVT_ACCEPTED:
Florin Coras3c7d4f92018-12-14 11:28:43 -08002885 session = vcl_session_accepted (wrk,
2886 (session_accepted_msg_t *) e->data);
Florin Coras86f04502018-09-12 16:08:01 -07002887 if (!session)
Florin Coras3c7d4f92018-12-14 11:28:43 -08002888 break;
Florin Coras86f04502018-09-12 16:08:01 -07002889
Florin Coras86f04502018-09-12 16:08:01 -07002890 session_events = session->vep.ev.events;
2891 if (!(EPOLLIN & session_events))
2892 break;
2893
2894 add_event = 1;
2895 events[*num_ev].events |= EPOLLIN;
2896 session_evt_data = session->vep.ev.data.u64;
2897 break;
2898 case SESSION_CTRL_EVT_CONNECTED:
2899 connected_msg = (session_connected_msg_t *) e->data;
Florin Corasf1653e62019-11-06 15:41:37 -08002900 sid = vcl_session_connected_handler (wrk, connected_msg);
Florin Coras86f04502018-09-12 16:08:01 -07002901 /* Generate EPOLLOUT because there's no connected event */
Florin Coras080aa502020-05-26 00:47:52 +00002902 session = vcl_session_get (wrk, sid);
2903 if (vcl_session_is_closed (session))
Florin Corasfa915f82018-12-26 16:29:06 -08002904 break;
Florin Coras86f04502018-09-12 16:08:01 -07002905 session_events = session->vep.ev.events;
Florin Coras72f77822019-01-22 19:05:52 -08002906 if (!(EPOLLOUT & session_events))
2907 break;
2908 add_event = 1;
2909 events[*num_ev].events |= EPOLLOUT;
2910 session_evt_data = session->vep.ev.data.u64;
Florin Corasadcfb152020-02-12 08:50:29 +00002911 if (session->session_state & STATE_DETACHED)
Florin Corasdbc9c592019-09-25 16:37:43 -07002912 events[*num_ev].events |= EPOLLHUP;
Florin Coras86f04502018-09-12 16:08:01 -07002913 break;
2914 case SESSION_CTRL_EVT_DISCONNECTED:
2915 disconnected_msg = (session_disconnected_msg_t *) e->data;
Florin Coras3c7d4f92018-12-14 11:28:43 -08002916 session = vcl_session_disconnected_handler (wrk, disconnected_msg);
Florin Coras080aa502020-05-26 00:47:52 +00002917 if (vcl_session_is_closed (session))
Florin Coras86f04502018-09-12 16:08:01 -07002918 break;
Florin Coras72f77822019-01-22 19:05:52 -08002919 session_events = session->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07002920 add_event = 1;
2921 events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2922 session_evt_data = session->vep.ev.data.u64;
Florin Coras86f04502018-09-12 16:08:01 -07002923 break;
2924 case SESSION_CTRL_EVT_RESET:
2925 sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
Florin Coras080aa502020-05-26 00:47:52 +00002926 session = vcl_session_get (wrk, sid);
2927 if (vcl_session_is_closed (session))
Florin Coras86f04502018-09-12 16:08:01 -07002928 break;
Florin Coras72f77822019-01-22 19:05:52 -08002929 session_events = session->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07002930 add_event = 1;
2931 events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2932 session_evt_data = session->vep.ev.data.u64;
Florin Coras86f04502018-09-12 16:08:01 -07002933 break;
Florin Corasdfae9f92019-02-20 19:48:31 -08002934 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2935 vcl_session_unlisten_reply_handler (wrk, e->data);
2936 break;
Florin Coras68b7e582020-01-21 18:33:23 -08002937 case SESSION_CTRL_EVT_MIGRATED:
2938 vcl_session_migrated_handler (wrk, e->data);
2939 break;
Florin Coras9ace36d2019-10-28 13:14:17 -07002940 case SESSION_CTRL_EVT_CLEANUP:
2941 vcl_session_cleanup_handler (wrk, e->data);
2942 break;
Florin Coras30e79c22019-01-02 19:31:22 -08002943 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2944 vcl_session_req_worker_update_handler (wrk, e->data);
2945 break;
2946 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2947 vcl_session_worker_update_reply_handler (wrk, e->data);
2948 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -07002949 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2950 vcl_session_app_add_segment_handler (wrk, e->data);
2951 break;
2952 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2953 vcl_session_app_del_segment_handler (wrk, e->data);
2954 break;
hanlina3a48962020-07-13 11:09:15 +08002955 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -07002956 vcl_worker_rpc_handler (wrk, e->data);
2957 break;
Florin Coras86f04502018-09-12 16:08:01 -07002958 default:
2959 VDBG (0, "unhandled: %u", e->event_type);
2960 break;
2961 }
2962
2963 if (add_event)
2964 {
2965 events[*num_ev].data.u64 = session_evt_data;
2966 if (EPOLLONESHOT & session_events)
2967 {
2968 session = vcl_session_get (wrk, sid);
2969 session->vep.ev.events = 0;
2970 }
2971 *num_ev += 1;
2972 }
2973}
2974
2975static int
2976vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2977 struct epoll_event *events, u32 maxevents,
2978 double wait_for_time, u32 * num_ev)
2979{
Florin Coras99368312018-08-02 10:45:44 -07002980 svm_msg_q_msg_t *msg;
Florin Coras54693d22018-07-17 10:46:29 -07002981 session_event_t *e;
Florin Coras54693d22018-07-17 10:46:29 -07002982 int i;
2983
Florin Coras539663c2018-09-28 14:59:37 -07002984 if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
2985 goto handle_dequeued;
2986
Florin Coras54693d22018-07-17 10:46:29 -07002987 svm_msg_q_lock (mq);
2988 if (svm_msg_q_is_empty (mq))
2989 {
2990 if (!wait_for_time)
2991 {
2992 svm_msg_q_unlock (mq);
2993 return 0;
2994 }
2995 else if (wait_for_time < 0)
2996 {
2997 svm_msg_q_wait (mq);
2998 }
2999 else
3000 {
Florin Coras60f1fc12018-08-16 14:57:31 -07003001 if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
Florin Coras54693d22018-07-17 10:46:29 -07003002 {
3003 svm_msg_q_unlock (mq);
3004 return 0;
3005 }
3006 }
3007 }
Florin Corase003a1b2019-06-05 10:47:16 -07003008 ASSERT (maxevents > *num_ev);
hanlind0e646f2020-05-11 22:20:37 +08003009 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras54693d22018-07-17 10:46:29 -07003010 svm_msg_q_unlock (mq);
3011
Florin Coras539663c2018-09-28 14:59:37 -07003012handle_dequeued:
Florin Coras134a9962018-08-28 11:32:04 -07003013 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
Florin Coras54693d22018-07-17 10:46:29 -07003014 {
Florin Coras134a9962018-08-28 11:32:04 -07003015 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
Florin Coras99368312018-08-02 10:45:44 -07003016 e = svm_msg_q_msg_data (mq, msg);
hanlind0e646f2020-05-11 22:20:37 +08003017 if (*num_ev < maxevents)
3018 vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
3019 else
3020 vcl_handle_mq_event (wrk, e);
Florin Coras99368312018-08-02 10:45:44 -07003021 svm_msg_q_free_msg (mq, msg);
Florin Coras54693d22018-07-17 10:46:29 -07003022 }
Florin Corasaa27eb92018-10-13 12:20:01 -07003023 vec_reset_length (wrk->mq_msg_vector);
Florin Coras30e79c22019-01-02 19:31:22 -08003024 vcl_handle_pending_wrk_updates (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07003025 return *num_ev;
3026}
3027
Florin Coras99368312018-08-02 10:45:44 -07003028static int
Florin Coras134a9962018-08-28 11:32:04 -07003029vppcom_epoll_wait_condvar (vcl_worker_t * wrk, struct epoll_event *events,
Florin Coras86f04502018-09-12 16:08:01 -07003030 int maxevents, u32 n_evts, double wait_for_time)
Florin Coras99368312018-08-02 10:45:44 -07003031{
Florin Corase003a1b2019-06-05 10:47:16 -07003032 double wait = 0, start = 0, now;
Florin Coras14ed6df2019-03-06 21:13:42 -08003033
3034 if (!n_evts)
3035 {
3036 wait = wait_for_time;
3037 start = clib_time_now (&wrk->clib_time);
3038 }
3039
3040 do
3041 {
3042 vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
3043 wait, &n_evts);
3044 if (n_evts)
3045 return n_evts;
3046 if (wait == -1)
3047 continue;
3048
Florin Corase003a1b2019-06-05 10:47:16 -07003049 now = clib_time_now (&wrk->clib_time);
Florin Coras0edfb1a2020-08-04 22:45:45 -07003050 wait -= (now - start) * 1e3;
Florin Corase003a1b2019-06-05 10:47:16 -07003051 start = now;
Florin Coras14ed6df2019-03-06 21:13:42 -08003052 }
3053 while (wait > 0);
3054
3055 return 0;
Florin Coras99368312018-08-02 10:45:44 -07003056}
3057
3058static int
Florin Coras134a9962018-08-28 11:32:04 -07003059vppcom_epoll_wait_eventfd (vcl_worker_t * wrk, struct epoll_event *events,
Florin Coras86f04502018-09-12 16:08:01 -07003060 int maxevents, u32 n_evts, double wait_for_time)
Florin Coras99368312018-08-02 10:45:44 -07003061{
3062 vcl_mq_evt_conn_t *mqc;
3063 int __clib_unused n_read;
3064 int n_mq_evts, i;
Florin Coras99368312018-08-02 10:45:44 -07003065 u64 buf;
3066
Florin Coras134a9962018-08-28 11:32:04 -07003067 vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
Florin Corasa4878ed2018-10-12 13:09:36 -07003068again:
Florin Coras134a9962018-08-28 11:32:04 -07003069 n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
3070 vec_len (wrk->mq_events), wait_for_time);
Florin Coras99368312018-08-02 10:45:44 -07003071 for (i = 0; i < n_mq_evts; i++)
3072 {
Florin Coras134a9962018-08-28 11:32:04 -07003073 mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
Florin Coras99368312018-08-02 10:45:44 -07003074 n_read = read (mqc->mq_fd, &buf, sizeof (buf));
Florin Coras134a9962018-08-28 11:32:04 -07003075 vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0, &n_evts);
Florin Coras99368312018-08-02 10:45:44 -07003076 }
Florin Corasa4878ed2018-10-12 13:09:36 -07003077 if (!n_evts && n_mq_evts > 0)
3078 goto again;
Florin Coras99368312018-08-02 10:45:44 -07003079
3080 return (int) n_evts;
3081}
3082
Dave Wallacef7f809c2017-10-03 01:48:42 -04003083int
Florin Coras134a9962018-08-28 11:32:04 -07003084vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
Dave Wallacef7f809c2017-10-03 01:48:42 -04003085 int maxevents, double wait_for_time)
3086{
Florin Coras134a9962018-08-28 11:32:04 -07003087 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07003088 vcl_session_t *vep_session;
Florin Coras86f04502018-09-12 16:08:01 -07003089 u32 n_evts = 0;
3090 int i;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003091
3092 if (PREDICT_FALSE (maxevents <= 0))
3093 {
Florin Coras5e062572019-03-14 19:07:51 -07003094 VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003095 return VPPCOM_EINVAL;
3096 }
Dave Wallacef7f809c2017-10-03 01:48:42 -04003097
Florin Coras134a9962018-08-28 11:32:04 -07003098 vep_session = vcl_session_get_w_handle (wrk, vep_handle);
Florin Coras14598772018-09-04 19:47:52 -07003099 if (!vep_session)
3100 return VPPCOM_EBADFD;
3101
Florin Coras54693d22018-07-17 10:46:29 -07003102 if (PREDICT_FALSE (!vep_session->is_vep))
Dave Wallacef7f809c2017-10-03 01:48:42 -04003103 {
Florin Coras5e062572019-03-14 19:07:51 -07003104 VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
Florin Coras54693d22018-07-17 10:46:29 -07003105 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003106 }
Florin Coras54693d22018-07-17 10:46:29 -07003107
3108 memset (events, 0, sizeof (*events) * maxevents);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003109
Florin Coras86f04502018-09-12 16:08:01 -07003110 if (vec_len (wrk->unhandled_evts_vector))
3111 {
3112 for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3113 {
3114 vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3115 events, &n_evts);
3116 if (n_evts == maxevents)
3117 {
Florin Corase003a1b2019-06-05 10:47:16 -07003118 vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3119 return n_evts;
Florin Coras86f04502018-09-12 16:08:01 -07003120 }
3121 }
Florin Corase003a1b2019-06-05 10:47:16 -07003122 vec_reset_length (wrk->unhandled_evts_vector);
Florin Coras86f04502018-09-12 16:08:01 -07003123 }
3124
3125 if (vcm->cfg.use_mq_eventfd)
3126 return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3127 wait_for_time);
3128
3129 return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3130 wait_for_time);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003131}
3132
Dave Wallace35830af2017-10-09 01:43:42 -04003133int
Florin Coras134a9962018-08-28 11:32:04 -07003134vppcom_session_attr (uint32_t session_handle, uint32_t op,
Dave Wallace35830af2017-10-09 01:43:42 -04003135 void *buffer, uint32_t * buflen)
3136{
Florin Coras134a9962018-08-28 11:32:04 -07003137 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07003138 vcl_session_t *session;
Dave Wallace35830af2017-10-09 01:43:42 -04003139 int rv = VPPCOM_OK;
Florin Coras7baeb712019-01-04 17:05:43 -08003140 u32 *flags = buffer, tmp_flags = 0;
Steven2199aab2017-10-15 20:18:47 -07003141 vppcom_endpt_t *ep = buffer;
Dave Wallace35830af2017-10-09 01:43:42 -04003142
Florin Coras134a9962018-08-28 11:32:04 -07003143 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07003144 if (!session)
3145 return VPPCOM_EBADFD;
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003146
Dave Wallace35830af2017-10-09 01:43:42 -04003147 switch (op)
3148 {
3149 case VPPCOM_ATTR_GET_NREAD:
Florin Coras0ef8ef22019-01-18 08:37:13 -08003150 rv = vcl_session_read_ready (session);
Florin Coras5e062572019-03-14 19:07:51 -07003151 VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3152 rv);
Dave Wallace35830af2017-10-09 01:43:42 -04003153 break;
3154
Dave Wallace227867f2017-11-13 21:21:53 -05003155 case VPPCOM_ATTR_GET_NWRITE:
Florin Coras0ef8ef22019-01-18 08:37:13 -08003156 rv = vcl_session_write_ready (session);
Florin Coras5e062572019-03-14 19:07:51 -07003157 VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3158 rv);
Dave Wallace35830af2017-10-09 01:43:42 -04003159 break;
3160
3161 case VPPCOM_ATTR_GET_FLAGS:
Dave Wallace048b1d62018-01-03 22:24:41 -05003162 if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
Dave Wallace35830af2017-10-09 01:43:42 -04003163 {
Simon Zhang53ec9672020-08-07 05:20:47 +08003164 *flags =
3165 O_RDWR |
3166 (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK) ?
3167 O_NONBLOCK : 0);
Dave Wallace35830af2017-10-09 01:43:42 -04003168 *buflen = sizeof (*flags);
Florin Coras5e062572019-03-14 19:07:51 -07003169 VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3170 "is_nonblocking = %u", session_handle, *flags,
Florin Coras0d427d82018-06-27 03:24:07 -07003171 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace35830af2017-10-09 01:43:42 -04003172 }
3173 else
3174 rv = VPPCOM_EINVAL;
3175 break;
3176
3177 case VPPCOM_ATTR_SET_FLAGS:
Dave Wallace048b1d62018-01-03 22:24:41 -05003178 if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
Dave Wallace35830af2017-10-09 01:43:42 -04003179 {
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003180 if (*flags & O_NONBLOCK)
3181 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
3182 else
3183 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_NONBLOCK);
3184
Florin Coras5e062572019-03-14 19:07:51 -07003185 VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3186 " is_nonblocking = %u", session_handle, *flags,
Florin Coras0d427d82018-06-27 03:24:07 -07003187 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace35830af2017-10-09 01:43:42 -04003188 }
3189 else
3190 rv = VPPCOM_EINVAL;
3191 break;
3192
3193 case VPPCOM_ATTR_GET_PEER_ADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003194 if (PREDICT_TRUE (buffer && buflen &&
3195 (*buflen >= sizeof (*ep)) && ep->ip))
Dave Wallace35830af2017-10-09 01:43:42 -04003196 {
Florin Coras7e12d942018-06-27 14:32:43 -07003197 ep->is_ip4 = session->transport.is_ip4;
3198 ep->port = session->transport.rmt_port;
3199 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05003200 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3201 sizeof (ip4_address_t));
Steven2199aab2017-10-15 20:18:47 -07003202 else
Dave Barach178cf492018-11-13 16:34:13 -05003203 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3204 sizeof (ip6_address_t));
Steven2199aab2017-10-15 20:18:47 -07003205 *buflen = sizeof (*ep);
Florin Coras5e062572019-03-14 19:07:51 -07003206 VDBG (1, "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3207 "addr = %U, port %u", session_handle, ep->is_ip4,
3208 format_ip46_address, &session->transport.rmt_ip,
Florin Coras0d427d82018-06-27 03:24:07 -07003209 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3210 clib_net_to_host_u16 (ep->port));
Dave Wallace35830af2017-10-09 01:43:42 -04003211 }
3212 else
3213 rv = VPPCOM_EINVAL;
3214 break;
3215
3216 case VPPCOM_ATTR_GET_LCL_ADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003217 if (PREDICT_TRUE (buffer && buflen &&
3218 (*buflen >= sizeof (*ep)) && ep->ip))
Dave Wallace35830af2017-10-09 01:43:42 -04003219 {
Florin Coras7e12d942018-06-27 14:32:43 -07003220 ep->is_ip4 = session->transport.is_ip4;
3221 ep->port = session->transport.lcl_port;
3222 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05003223 clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3224 sizeof (ip4_address_t));
Steven2199aab2017-10-15 20:18:47 -07003225 else
Dave Barach178cf492018-11-13 16:34:13 -05003226 clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3227 sizeof (ip6_address_t));
Steven2199aab2017-10-15 20:18:47 -07003228 *buflen = sizeof (*ep);
Florin Coras5e062572019-03-14 19:07:51 -07003229 VDBG (1, "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3230 " port %d", session_handle, ep->is_ip4, format_ip46_address,
Florin Coras7e12d942018-06-27 14:32:43 -07003231 &session->transport.lcl_ip,
Florin Coras0d427d82018-06-27 03:24:07 -07003232 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3233 clib_net_to_host_u16 (ep->port));
Dave Wallace35830af2017-10-09 01:43:42 -04003234 }
3235 else
3236 rv = VPPCOM_EINVAL;
3237 break;
Stevenb5a11602017-10-11 09:59:30 -07003238
Florin Corasef7cbf62019-10-17 09:56:27 -07003239 case VPPCOM_ATTR_SET_LCL_ADDR:
3240 if (PREDICT_TRUE (buffer && buflen &&
3241 (*buflen >= sizeof (*ep)) && ep->ip))
3242 {
3243 session->transport.is_ip4 = ep->is_ip4;
3244 session->transport.lcl_port = ep->port;
3245 vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3246 *buflen = sizeof (*ep);
3247 VDBG (1, "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3248 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3249 &session->transport.lcl_ip,
3250 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3251 clib_net_to_host_u16 (ep->port));
3252 }
3253 else
3254 rv = VPPCOM_EINVAL;
3255 break;
3256
Dave Wallace048b1d62018-01-03 22:24:41 -05003257 case VPPCOM_ATTR_GET_LIBC_EPFD:
3258 rv = session->libc_epfd;
Florin Coras5e062572019-03-14 19:07:51 -07003259 VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
Dave Wallace048b1d62018-01-03 22:24:41 -05003260 break;
3261
3262 case VPPCOM_ATTR_SET_LIBC_EPFD:
3263 if (PREDICT_TRUE (buffer && buflen &&
3264 (*buflen == sizeof (session->libc_epfd))))
3265 {
3266 session->libc_epfd = *(int *) buffer;
3267 *buflen = sizeof (session->libc_epfd);
3268
Florin Coras5e062572019-03-14 19:07:51 -07003269 VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3270 session->libc_epfd, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003271 }
3272 else
3273 rv = VPPCOM_EINVAL;
3274 break;
3275
3276 case VPPCOM_ATTR_GET_PROTOCOL:
3277 if (buffer && buflen && (*buflen >= sizeof (int)))
3278 {
Florin Coras7e12d942018-06-27 14:32:43 -07003279 *(int *) buffer = session->session_type;
Dave Wallace048b1d62018-01-03 22:24:41 -05003280 *buflen = sizeof (int);
3281
Florin Coras5e062572019-03-14 19:07:51 -07003282 VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3283 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003284 }
3285 else
3286 rv = VPPCOM_EINVAL;
3287 break;
3288
3289 case VPPCOM_ATTR_GET_LISTEN:
3290 if (buffer && buflen && (*buflen >= sizeof (int)))
3291 {
3292 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3293 VCL_SESS_ATTR_LISTEN);
3294 *buflen = sizeof (int);
3295
Florin Coras5e062572019-03-14 19:07:51 -07003296 VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3297 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003298 }
3299 else
3300 rv = VPPCOM_EINVAL;
3301 break;
3302
3303 case VPPCOM_ATTR_GET_ERROR:
3304 if (buffer && buflen && (*buflen >= sizeof (int)))
3305 {
3306 *(int *) buffer = 0;
3307 *buflen = sizeof (int);
3308
Florin Coras5e062572019-03-14 19:07:51 -07003309 VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3310 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003311 }
3312 else
3313 rv = VPPCOM_EINVAL;
3314 break;
3315
3316 case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3317 if (buffer && buflen && (*buflen >= sizeof (u32)))
3318 {
Dave Wallace048b1d62018-01-03 22:24:41 -05003319
3320 /* VPP-TBD */
3321 *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
Florin Corasf22f4e52019-12-19 16:10:58 -08003322 session->tx_fifo ?
3323 svm_fifo_size (session->tx_fifo) :
Dave Wallace048b1d62018-01-03 22:24:41 -05003324 vcm->cfg.tx_fifo_size);
3325 *buflen = sizeof (u32);
3326
Florin Coras5e062572019-03-14 19:07:51 -07003327 VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3328 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3329 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003330 }
3331 else
3332 rv = VPPCOM_EINVAL;
3333 break;
3334
3335 case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3336 if (buffer && buflen && (*buflen == sizeof (u32)))
3337 {
3338 /* VPP-TBD */
3339 session->sndbuf_size = *(u32 *) buffer;
Florin Coras5e062572019-03-14 19:07:51 -07003340 VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3341 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3342 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003343 }
3344 else
3345 rv = VPPCOM_EINVAL;
3346 break;
3347
3348 case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3349 if (buffer && buflen && (*buflen >= sizeof (u32)))
3350 {
Dave Wallace048b1d62018-01-03 22:24:41 -05003351
3352 /* VPP-TBD */
3353 *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
Florin Corasf22f4e52019-12-19 16:10:58 -08003354 session->rx_fifo ?
3355 svm_fifo_size (session->rx_fifo) :
Dave Wallace048b1d62018-01-03 22:24:41 -05003356 vcm->cfg.rx_fifo_size);
3357 *buflen = sizeof (u32);
3358
Florin Coras5e062572019-03-14 19:07:51 -07003359 VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3360 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003361 }
3362 else
3363 rv = VPPCOM_EINVAL;
3364 break;
3365
3366 case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3367 if (buffer && buflen && (*buflen == sizeof (u32)))
3368 {
3369 /* VPP-TBD */
3370 session->rcvbuf_size = *(u32 *) buffer;
Florin Coras5e062572019-03-14 19:07:51 -07003371 VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3372 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3373 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003374 }
3375 else
3376 rv = VPPCOM_EINVAL;
3377 break;
3378
3379 case VPPCOM_ATTR_GET_REUSEADDR:
3380 if (buffer && buflen && (*buflen >= sizeof (int)))
3381 {
3382 /* VPP-TBD */
3383 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3384 VCL_SESS_ATTR_REUSEADDR);
3385 *buflen = sizeof (int);
3386
Florin Coras5e062572019-03-14 19:07:51 -07003387 VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3388 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003389 }
3390 else
3391 rv = VPPCOM_EINVAL;
3392 break;
3393
Stevenb5a11602017-10-11 09:59:30 -07003394 case VPPCOM_ATTR_SET_REUSEADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003395 if (buffer && buflen && (*buflen == sizeof (int)) &&
3396 !VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
3397 {
3398 /* VPP-TBD */
3399 if (*(int *) buffer)
3400 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEADDR);
3401 else
3402 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEADDR);
3403
Florin Coras5e062572019-03-14 19:07:51 -07003404 VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3405 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_REUSEADDR),
3406 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003407 }
3408 else
3409 rv = VPPCOM_EINVAL;
3410 break;
3411
3412 case VPPCOM_ATTR_GET_REUSEPORT:
3413 if (buffer && buflen && (*buflen >= sizeof (int)))
3414 {
3415 /* VPP-TBD */
3416 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3417 VCL_SESS_ATTR_REUSEPORT);
3418 *buflen = sizeof (int);
3419
Florin Coras5e062572019-03-14 19:07:51 -07003420 VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3421 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003422 }
3423 else
3424 rv = VPPCOM_EINVAL;
3425 break;
3426
3427 case VPPCOM_ATTR_SET_REUSEPORT:
3428 if (buffer && buflen && (*buflen == sizeof (int)) &&
3429 !VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
3430 {
3431 /* VPP-TBD */
3432 if (*(int *) buffer)
3433 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEPORT);
3434 else
3435 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEPORT);
3436
Florin Coras5e062572019-03-14 19:07:51 -07003437 VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3438 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_REUSEPORT),
3439 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003440 }
3441 else
3442 rv = VPPCOM_EINVAL;
3443 break;
3444
3445 case VPPCOM_ATTR_GET_BROADCAST:
3446 if (buffer && buflen && (*buflen >= sizeof (int)))
3447 {
3448 /* VPP-TBD */
3449 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3450 VCL_SESS_ATTR_BROADCAST);
3451 *buflen = sizeof (int);
3452
Florin Coras5e062572019-03-14 19:07:51 -07003453 VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3454 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003455 }
3456 else
3457 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07003458 break;
3459
3460 case VPPCOM_ATTR_SET_BROADCAST:
Dave Wallace048b1d62018-01-03 22:24:41 -05003461 if (buffer && buflen && (*buflen == sizeof (int)))
3462 {
3463 /* VPP-TBD */
3464 if (*(int *) buffer)
3465 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_BROADCAST);
3466 else
3467 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_BROADCAST);
3468
Florin Coras5e062572019-03-14 19:07:51 -07003469 VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3470 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_BROADCAST),
3471 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003472 }
3473 else
3474 rv = VPPCOM_EINVAL;
3475 break;
3476
3477 case VPPCOM_ATTR_GET_V6ONLY:
3478 if (buffer && buflen && (*buflen >= sizeof (int)))
3479 {
3480 /* VPP-TBD */
3481 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3482 VCL_SESS_ATTR_V6ONLY);
3483 *buflen = sizeof (int);
3484
Florin Coras5e062572019-03-14 19:07:51 -07003485 VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3486 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003487 }
3488 else
3489 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07003490 break;
3491
3492 case VPPCOM_ATTR_SET_V6ONLY:
Dave Wallace048b1d62018-01-03 22:24:41 -05003493 if (buffer && buflen && (*buflen == sizeof (int)))
3494 {
3495 /* VPP-TBD */
3496 if (*(int *) buffer)
3497 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_V6ONLY);
3498 else
3499 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_V6ONLY);
3500
Florin Coras5e062572019-03-14 19:07:51 -07003501 VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3502 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_V6ONLY),
3503 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003504 }
3505 else
3506 rv = VPPCOM_EINVAL;
3507 break;
3508
3509 case VPPCOM_ATTR_GET_KEEPALIVE:
3510 if (buffer && buflen && (*buflen >= sizeof (int)))
3511 {
3512 /* VPP-TBD */
3513 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3514 VCL_SESS_ATTR_KEEPALIVE);
3515 *buflen = sizeof (int);
3516
Florin Coras5e062572019-03-14 19:07:51 -07003517 VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3518 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003519 }
3520 else
3521 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07003522 break;
Stevenbccd3392017-10-12 20:42:21 -07003523
3524 case VPPCOM_ATTR_SET_KEEPALIVE:
Dave Wallace048b1d62018-01-03 22:24:41 -05003525 if (buffer && buflen && (*buflen == sizeof (int)))
3526 {
3527 /* VPP-TBD */
3528 if (*(int *) buffer)
3529 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_KEEPALIVE);
3530 else
3531 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_KEEPALIVE);
3532
Florin Coras5e062572019-03-14 19:07:51 -07003533 VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3534 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_KEEPALIVE),
3535 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003536 }
3537 else
3538 rv = VPPCOM_EINVAL;
3539 break;
3540
3541 case VPPCOM_ATTR_GET_TCP_NODELAY:
3542 if (buffer && buflen && (*buflen >= sizeof (int)))
3543 {
3544 /* VPP-TBD */
3545 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3546 VCL_SESS_ATTR_TCP_NODELAY);
3547 *buflen = sizeof (int);
3548
Florin Coras5e062572019-03-14 19:07:51 -07003549 VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3550 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003551 }
3552 else
3553 rv = VPPCOM_EINVAL;
3554 break;
3555
3556 case VPPCOM_ATTR_SET_TCP_NODELAY:
3557 if (buffer && buflen && (*buflen == sizeof (int)))
3558 {
3559 /* VPP-TBD */
3560 if (*(int *) buffer)
3561 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
3562 else
3563 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
3564
Florin Coras5e062572019-03-14 19:07:51 -07003565 VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3566 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_TCP_NODELAY),
3567 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003568 }
3569 else
3570 rv = VPPCOM_EINVAL;
3571 break;
3572
3573 case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
3574 if (buffer && buflen && (*buflen >= sizeof (int)))
3575 {
3576 /* VPP-TBD */
3577 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3578 VCL_SESS_ATTR_TCP_KEEPIDLE);
3579 *buflen = sizeof (int);
3580
Florin Coras5e062572019-03-14 19:07:51 -07003581 VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3582 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003583 }
3584 else
3585 rv = VPPCOM_EINVAL;
Stevenbccd3392017-10-12 20:42:21 -07003586 break;
3587
3588 case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
Dave Wallace048b1d62018-01-03 22:24:41 -05003589 if (buffer && buflen && (*buflen == sizeof (int)))
3590 {
3591 /* VPP-TBD */
3592 if (*(int *) buffer)
3593 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
3594 else
3595 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
3596
Florin Coras5e062572019-03-14 19:07:51 -07003597 VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
Florin Coras0d427d82018-06-27 03:24:07 -07003598 VCL_SESS_ATTR_TEST (session->attr,
3599 VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003600 }
3601 else
3602 rv = VPPCOM_EINVAL;
3603 break;
3604
3605 case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
3606 if (buffer && buflen && (*buflen >= sizeof (int)))
3607 {
3608 /* VPP-TBD */
3609 *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3610 VCL_SESS_ATTR_TCP_KEEPINTVL);
3611 *buflen = sizeof (int);
3612
Florin Coras5e062572019-03-14 19:07:51 -07003613 VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3614 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003615 }
3616 else
3617 rv = VPPCOM_EINVAL;
Stevenbccd3392017-10-12 20:42:21 -07003618 break;
3619
3620 case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
Dave Wallace048b1d62018-01-03 22:24:41 -05003621 if (buffer && buflen && (*buflen == sizeof (int)))
3622 {
3623 /* VPP-TBD */
3624 if (*(int *) buffer)
3625 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_KEEPINTVL);
3626 else
3627 VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPINTVL);
3628
Florin Coras5e062572019-03-14 19:07:51 -07003629 VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
Florin Coras0d427d82018-06-27 03:24:07 -07003630 VCL_SESS_ATTR_TEST (session->attr,
3631 VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003632 }
3633 else
3634 rv = VPPCOM_EINVAL;
3635 break;
3636
3637 case VPPCOM_ATTR_GET_TCP_USER_MSS:
3638 if (buffer && buflen && (*buflen >= sizeof (u32)))
3639 {
3640 /* VPP-TBD */
3641 *(u32 *) buffer = session->user_mss;
3642 *buflen = sizeof (int);
3643
Florin Coras5e062572019-03-14 19:07:51 -07003644 VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d, #VPP-TBD#",
3645 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003646 }
3647 else
3648 rv = VPPCOM_EINVAL;
3649 break;
3650
3651 case VPPCOM_ATTR_SET_TCP_USER_MSS:
3652 if (buffer && buflen && (*buflen == sizeof (u32)))
3653 {
3654 /* VPP-TBD */
3655 session->user_mss = *(u32 *) buffer;
3656
Florin Coras5e062572019-03-14 19:07:51 -07003657 VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d, #VPP-TBD#",
3658 session->user_mss, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003659 }
3660 else
3661 rv = VPPCOM_EINVAL;
Stevenbccd3392017-10-12 20:42:21 -07003662 break;
Dave Wallacee22aa742017-10-20 12:30:38 -04003663
Florin Coras7baeb712019-01-04 17:05:43 -08003664 case VPPCOM_ATTR_SET_SHUT:
3665 if (*flags == SHUT_RD || *flags == SHUT_RDWR)
3666 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_SHUT_RD);
3667 if (*flags == SHUT_WR || *flags == SHUT_RDWR)
3668 VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_SHUT_WR);
3669 break;
3670
3671 case VPPCOM_ATTR_GET_SHUT:
3672 if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_SHUT_RD))
3673 tmp_flags = 1;
3674 if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_SHUT_WR))
3675 tmp_flags |= 2;
3676 if (tmp_flags == 1)
3677 *(int *) buffer = SHUT_RD;
3678 else if (tmp_flags == 2)
3679 *(int *) buffer = SHUT_WR;
3680 else if (tmp_flags == 3)
3681 *(int *) buffer = SHUT_RDWR;
3682 *buflen = sizeof (int);
3683 break;
Florin Coras1e966172020-05-16 18:18:14 +00003684
3685 case VPPCOM_ATTR_SET_CONNECTED:
3686 session->flags |= VCL_SESSION_F_CONNECTED;
3687 break;
3688
Dave Wallacee22aa742017-10-20 12:30:38 -04003689 default:
3690 rv = VPPCOM_EINVAL;
3691 break;
Dave Wallace35830af2017-10-09 01:43:42 -04003692 }
3693
Dave Wallace35830af2017-10-09 01:43:42 -04003694 return rv;
3695}
3696
Stevenac1f96d2017-10-24 16:03:58 -07003697int
Florin Coras134a9962018-08-28 11:32:04 -07003698vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
Stevenac1f96d2017-10-24 16:03:58 -07003699 uint32_t buflen, int flags, vppcom_endpt_t * ep)
3700{
Florin Coras134a9962018-08-28 11:32:04 -07003701 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras5da10c42020-04-01 04:31:21 +00003702 vcl_session_t *session;
Stevenac1f96d2017-10-24 16:03:58 -07003703 int rv = VPPCOM_OK;
Steven58f464e2017-10-25 12:33:12 -07003704
3705 if (flags == 0)
Florin Coras134a9962018-08-28 11:32:04 -07003706 rv = vppcom_session_read (session_handle, buffer, buflen);
Stevenac1f96d2017-10-24 16:03:58 -07003707 else if (flags & MSG_PEEK)
Florin Coras134a9962018-08-28 11:32:04 -07003708 rv = vppcom_session_peek (session_handle, buffer, buflen);
Stevenac1f96d2017-10-24 16:03:58 -07003709 else
3710 {
Florin Corasa7a1a222018-12-30 17:11:31 -08003711 VDBG (0, "Unsupport flags for recvfrom %d", flags);
Florin Coras460dce62018-07-27 05:45:06 -07003712 return VPPCOM_EAFNOSUPPORT;
Stevenac1f96d2017-10-24 16:03:58 -07003713 }
3714
Florin Coras0a1e1832020-03-29 18:54:04 +00003715 if (ep && rv > 0)
Florin Coras99368312018-08-02 10:45:44 -07003716 {
Florin Coras5da10c42020-04-01 04:31:21 +00003717 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras99368312018-08-02 10:45:44 -07003718 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05003719 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3720 sizeof (ip4_address_t));
Florin Coras99368312018-08-02 10:45:44 -07003721 else
Dave Barach178cf492018-11-13 16:34:13 -05003722 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3723 sizeof (ip6_address_t));
Florin Coras5da10c42020-04-01 04:31:21 +00003724 ep->is_ip4 = session->transport.is_ip4;
3725 ep->port = session->transport.rmt_port;
Florin Coras99368312018-08-02 10:45:44 -07003726 }
Florin Coras460dce62018-07-27 05:45:06 -07003727
Stevenac1f96d2017-10-24 16:03:58 -07003728 return rv;
3729}
3730
3731int
Florin Coras134a9962018-08-28 11:32:04 -07003732vppcom_session_sendto (uint32_t session_handle, void *buffer,
Stevenac1f96d2017-10-24 16:03:58 -07003733 uint32_t buflen, int flags, vppcom_endpt_t * ep)
3734{
Florin Coras7a2abce2020-04-05 19:25:44 +00003735 vcl_worker_t *wrk = vcl_worker_get_current ();
3736 vcl_session_t *s;
3737
3738 s = vcl_session_get_w_handle (wrk, session_handle);
3739 if (!s)
3740 return VPPCOM_EBADFD;
3741
Dave Wallace617dffa2017-10-26 14:47:06 -04003742 if (!buffer)
3743 return VPPCOM_EINVAL;
3744
3745 if (ep)
3746 {
Florin Coras0a1e1832020-03-29 18:54:04 +00003747 if (s->session_type != VPPCOM_PROTO_UDP
3748 || (s->flags & VCL_SESSION_F_CONNECTED))
Florin Coras5da10c42020-04-01 04:31:21 +00003749 return VPPCOM_EINVAL;
3750
3751 /* Session not connected/bound in vpp. Create it by 'connecting' it */
3752 if (PREDICT_FALSE (s->session_state == STATE_CLOSED))
3753 {
3754 vcl_send_session_connect (wrk, s);
3755 }
3756 else
3757 {
3758 s->transport.is_ip4 = ep->is_ip4;
3759 s->transport.rmt_port = ep->port;
3760 vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
3761 }
Dave Wallace617dffa2017-10-26 14:47:06 -04003762 }
3763
3764 if (flags)
3765 {
3766 // TBD check the flags and do the right thing
Florin Coras5e062572019-03-14 19:07:51 -07003767 VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
Dave Wallace617dffa2017-10-26 14:47:06 -04003768 }
3769
Florin Coras7a2abce2020-04-05 19:25:44 +00003770 return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
3771 s->is_dgram ? 1 : 0));
Stevenac1f96d2017-10-24 16:03:58 -07003772}
3773
Dave Wallace048b1d62018-01-03 22:24:41 -05003774int
3775vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
3776{
Florin Coras134a9962018-08-28 11:32:04 -07003777 vcl_worker_t *wrk = vcl_worker_get_current ();
3778 f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
Dave Wallace048b1d62018-01-03 22:24:41 -05003779 u32 i, keep_trying = 1;
Florin Coras6917b942018-11-13 22:44:54 -08003780 svm_msg_q_msg_t msg;
3781 session_event_t *e;
Dave Wallace048b1d62018-01-03 22:24:41 -05003782 int rv, num_ev = 0;
3783
Florin Coras5e062572019-03-14 19:07:51 -07003784 VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
Dave Wallace048b1d62018-01-03 22:24:41 -05003785
3786 if (!vp)
3787 return VPPCOM_EFAULT;
3788
3789 do
3790 {
Florin Coras7e12d942018-06-27 14:32:43 -07003791 vcl_session_t *session;
Dave Wallace048b1d62018-01-03 22:24:41 -05003792
Florin Coras6917b942018-11-13 22:44:54 -08003793 /* Dequeue all events and drop all unhandled io events */
3794 while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
3795 {
3796 e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
3797 vcl_handle_mq_event (wrk, e);
3798 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
3799 }
3800 vec_reset_length (wrk->unhandled_evts_vector);
3801
Dave Wallace048b1d62018-01-03 22:24:41 -05003802 for (i = 0; i < n_sids; i++)
3803 {
Florin Coras7baeb712019-01-04 17:05:43 -08003804 session = vcl_session_get (wrk, vp[i].sh);
Florin Coras070453d2018-08-24 17:04:27 -07003805 if (!session)
Florin Coras6917b942018-11-13 22:44:54 -08003806 {
3807 vp[i].revents = POLLHUP;
3808 num_ev++;
3809 continue;
3810 }
Dave Wallace048b1d62018-01-03 22:24:41 -05003811
Florin Coras6917b942018-11-13 22:44:54 -08003812 vp[i].revents = 0;
Dave Wallace048b1d62018-01-03 22:24:41 -05003813
3814 if (POLLIN & vp[i].events)
3815 {
Florin Coras0ef8ef22019-01-18 08:37:13 -08003816 rv = vcl_session_read_ready (session);
Dave Wallace048b1d62018-01-03 22:24:41 -05003817 if (rv > 0)
3818 {
Florin Coras6917b942018-11-13 22:44:54 -08003819 vp[i].revents |= POLLIN;
Dave Wallace048b1d62018-01-03 22:24:41 -05003820 num_ev++;
3821 }
3822 else if (rv < 0)
3823 {
3824 switch (rv)
3825 {
3826 case VPPCOM_ECONNRESET:
Florin Coras6917b942018-11-13 22:44:54 -08003827 vp[i].revents = POLLHUP;
Dave Wallace048b1d62018-01-03 22:24:41 -05003828 break;
3829
3830 default:
Florin Coras6917b942018-11-13 22:44:54 -08003831 vp[i].revents = POLLERR;
Dave Wallace048b1d62018-01-03 22:24:41 -05003832 break;
3833 }
3834 num_ev++;
3835 }
3836 }
3837
3838 if (POLLOUT & vp[i].events)
3839 {
Florin Coras0ef8ef22019-01-18 08:37:13 -08003840 rv = vcl_session_write_ready (session);
Dave Wallace048b1d62018-01-03 22:24:41 -05003841 if (rv > 0)
3842 {
Florin Coras6917b942018-11-13 22:44:54 -08003843 vp[i].revents |= POLLOUT;
Dave Wallace048b1d62018-01-03 22:24:41 -05003844 num_ev++;
3845 }
3846 else if (rv < 0)
3847 {
3848 switch (rv)
3849 {
3850 case VPPCOM_ECONNRESET:
Florin Coras6917b942018-11-13 22:44:54 -08003851 vp[i].revents = POLLHUP;
Dave Wallace048b1d62018-01-03 22:24:41 -05003852 break;
3853
3854 default:
Florin Coras6917b942018-11-13 22:44:54 -08003855 vp[i].revents = POLLERR;
Dave Wallace048b1d62018-01-03 22:24:41 -05003856 break;
3857 }
3858 num_ev++;
3859 }
3860 }
3861
Dave Wallace7e607a72018-06-18 18:41:32 -04003862 if (0) // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
Dave Wallace048b1d62018-01-03 22:24:41 -05003863 {
Florin Coras6917b942018-11-13 22:44:54 -08003864 vp[i].revents = POLLNVAL;
Dave Wallace048b1d62018-01-03 22:24:41 -05003865 num_ev++;
3866 }
3867 }
3868 if (wait_for_time != -1)
Florin Coras134a9962018-08-28 11:32:04 -07003869 keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
Dave Wallace048b1d62018-01-03 22:24:41 -05003870 }
3871 while ((num_ev == 0) && keep_trying);
3872
Dave Wallace048b1d62018-01-03 22:24:41 -05003873 return num_ev;
3874}
3875
Florin Coras99368312018-08-02 10:45:44 -07003876int
3877vppcom_mq_epoll_fd (void)
3878{
Florin Coras134a9962018-08-28 11:32:04 -07003879 vcl_worker_t *wrk = vcl_worker_get_current ();
3880 return wrk->mqs_epfd;
3881}
3882
3883int
Florin Coras30e79c22019-01-02 19:31:22 -08003884vppcom_session_index (vcl_session_handle_t session_handle)
Florin Coras134a9962018-08-28 11:32:04 -07003885{
3886 return session_handle & 0xFFFFFF;
3887}
3888
3889int
Florin Coras30e79c22019-01-02 19:31:22 -08003890vppcom_session_worker (vcl_session_handle_t session_handle)
3891{
3892 return session_handle >> 24;
3893}
3894
3895int
Florin Coras134a9962018-08-28 11:32:04 -07003896vppcom_worker_register (void)
3897{
Florin Corasd4c70922019-11-28 14:21:21 -08003898 vcl_worker_t *wrk;
3899 u8 *wrk_name = 0;
3900 int rv;
3901
Florin Coras47c40e22018-11-26 17:01:36 -08003902 if (!vcl_worker_alloc_and_init ())
3903 return VPPCOM_EEXIST;
3904
Florin Corasd4c70922019-11-28 14:21:21 -08003905 wrk = vcl_worker_get_current ();
3906 wrk_name = format (0, "%s-wrk-%u", vcm->app_name, wrk->wrk_index);
3907
3908 rv = vppcom_connect_to_vpp ((char *) wrk_name);
3909 vec_free (wrk_name);
3910
3911 if (rv)
3912 return VPPCOM_EFAULT;
Florin Coras47c40e22018-11-26 17:01:36 -08003913
3914 if (vcl_worker_register_with_vpp ())
3915 return VPPCOM_EEXIST;
3916
3917 return VPPCOM_OK;
Florin Coras99368312018-08-02 10:45:44 -07003918}
3919
Florin Coras369db832019-07-08 12:34:45 -07003920void
3921vppcom_worker_unregister (void)
3922{
3923 vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
3924 vcl_set_worker_index (~0);
3925}
3926
Pivo6017ff02020-05-04 17:57:33 +02003927void
3928vppcom_worker_index_set (int index)
3929{
3930 vcl_set_worker_index (index);
3931}
3932
Florin Corasdfe4cf42018-11-28 22:13:45 -08003933int
3934vppcom_worker_index (void)
3935{
3936 return vcl_get_worker_index ();
3937}
3938
Florin Corase0982e52019-01-25 13:19:56 -08003939int
3940vppcom_worker_mqs_epfd (void)
3941{
3942 vcl_worker_t *wrk = vcl_worker_get_current ();
3943 if (!vcm->cfg.use_mq_eventfd)
3944 return -1;
3945 return wrk->mqs_epfd;
3946}
3947
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02003948int
3949vppcom_session_is_connectable_listener (uint32_t session_handle)
3950{
3951 vcl_session_t *session;
3952 vcl_worker_t *wrk = vcl_worker_get_current ();
3953 session = vcl_session_get_w_handle (wrk, session_handle);
3954 if (!session)
3955 return VPPCOM_EBADFD;
3956 return vcl_session_is_connectable_listener (wrk, session);
3957}
3958
3959int
3960vppcom_session_listener (uint32_t session_handle)
3961{
3962 vcl_worker_t *wrk = vcl_worker_get_current ();
3963 vcl_session_t *listen_session, *session;
3964 session = vcl_session_get_w_handle (wrk, session_handle);
3965 if (!session)
3966 return VPPCOM_EBADFD;
3967 if (session->listener_index == VCL_INVALID_SESSION_INDEX)
3968 return VPPCOM_EBADFD;
3969 listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
3970 if (!listen_session)
3971 return VPPCOM_EBADFD;
3972 return vcl_session_handle (listen_session);
3973}
3974
3975int
3976vppcom_session_n_accepted (uint32_t session_handle)
3977{
3978 vcl_worker_t *wrk = vcl_worker_get_current ();
3979 vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
3980 if (!session)
3981 return VPPCOM_EBADFD;
3982 return session->n_accepted_sessions;
3983}
3984
Florin Coras66ec4672020-06-15 07:59:40 -07003985const char *
3986vppcom_proto_str (vppcom_proto_t proto)
3987{
3988 char const *proto_str;
3989
3990 switch (proto)
3991 {
3992 case VPPCOM_PROTO_TCP:
3993 proto_str = "TCP";
3994 break;
3995 case VPPCOM_PROTO_UDP:
3996 proto_str = "UDP";
3997 break;
3998 case VPPCOM_PROTO_TLS:
3999 proto_str = "TLS";
4000 break;
4001 case VPPCOM_PROTO_QUIC:
4002 proto_str = "QUIC";
4003 break;
4004 default:
4005 proto_str = "UNKNOWN";
4006 break;
4007 }
4008 return proto_str;
4009}
4010
4011const char *
4012vppcom_retval_str (int retval)
4013{
4014 char const *st;
4015
4016 switch (retval)
4017 {
4018 case VPPCOM_OK:
4019 st = "VPPCOM_OK";
4020 break;
4021
4022 case VPPCOM_EAGAIN:
4023 st = "VPPCOM_EAGAIN";
4024 break;
4025
4026 case VPPCOM_EFAULT:
4027 st = "VPPCOM_EFAULT";
4028 break;
4029
4030 case VPPCOM_ENOMEM:
4031 st = "VPPCOM_ENOMEM";
4032 break;
4033
4034 case VPPCOM_EINVAL:
4035 st = "VPPCOM_EINVAL";
4036 break;
4037
4038 case VPPCOM_EBADFD:
4039 st = "VPPCOM_EBADFD";
4040 break;
4041
4042 case VPPCOM_EAFNOSUPPORT:
4043 st = "VPPCOM_EAFNOSUPPORT";
4044 break;
4045
4046 case VPPCOM_ECONNABORTED:
4047 st = "VPPCOM_ECONNABORTED";
4048 break;
4049
4050 case VPPCOM_ECONNRESET:
4051 st = "VPPCOM_ECONNRESET";
4052 break;
4053
4054 case VPPCOM_ENOTCONN:
4055 st = "VPPCOM_ENOTCONN";
4056 break;
4057
4058 case VPPCOM_ECONNREFUSED:
4059 st = "VPPCOM_ECONNREFUSED";
4060 break;
4061
4062 case VPPCOM_ETIMEDOUT:
4063 st = "VPPCOM_ETIMEDOUT";
4064 break;
4065
4066 default:
4067 st = "UNKNOWN_STATE";
4068 break;
4069 }
4070
4071 return st;
4072}
4073
Dave Wallacee22aa742017-10-20 12:30:38 -04004074/*
4075 * fd.io coding-style-patch-verification: ON
4076 *
4077 * Local Variables:
4078 * eval: (c-set-style "gnu")
4079 * End:
4080 */