vcl: remove accept state and rename connect to ready
[vpp.git] / src / vcl / vppcom.c
1 /*
2  * Copyright (c) 2017-2019 Cisco and/or its affiliates.
3  * 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>
18 #include <vcl/vppcom.h>
19 #include <vcl/vcl_debug.h>
20 #include <vcl/vcl_private.h>
21 #include <svm/fifo_segment.h>
22
23 __thread uword __vcl_worker_index = ~0;
24
25 static int
26 vcl_segment_is_not_mounted (vcl_worker_t * wrk, u64 segment_handle)
27 {
28   u32 segment_index;
29
30   if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
31     return 0;
32
33   segment_index = vcl_segment_table_lookup (segment_handle);
34   if (segment_index != VCL_INVALID_SEGMENT_INDEX)
35     return 0;
36
37   return 1;
38 }
39
40 static inline int
41 vcl_mq_dequeue_batch (vcl_worker_t * wrk, svm_msg_q_t * mq, u32 n_max_msg)
42 {
43   svm_msg_q_msg_t *msg;
44   u32 n_msgs;
45   int i;
46
47   n_msgs = clib_min (svm_msg_q_size (mq), n_max_msg);
48   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
56 const char *
57 vppcom_session_state_str (vcl_session_state_t state)
58 {
59   char *st;
60
61   switch (state)
62     {
63     case VCL_STATE_CLOSED:
64       st = "STATE_CLOSED";
65       break;
66     case VCL_STATE_LISTEN:
67       st = "STATE_LISTEN";
68       break;
69     case VCL_STATE_READY:
70       st = "STATE_READY";
71       break;
72     case VCL_STATE_VPP_CLOSING:
73       st = "STATE_VPP_CLOSING";
74       break;
75     case VCL_STATE_DISCONNECT:
76       st = "STATE_DISCONNECT";
77       break;
78     case VCL_STATE_DETACHED:
79       st = "STATE_DETACHED";
80       break;
81     case VCL_STATE_UPDATED:
82       st = "STATE_UPDATED";
83       break;
84     case VCL_STATE_LISTEN_NO_MQ:
85       st = "STATE_LISTEN_NO_MQ";
86       break;
87     default:
88       st = "UNKNOWN_STATE";
89       break;
90     }
91
92   return st;
93 }
94
95 u8 *
96 format_ip4_address (u8 * s, va_list * args)
97 {
98   u8 *a = va_arg (*args, u8 *);
99   return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
100 }
101
102 u8 *
103 format_ip6_address (u8 * s, va_list * args)
104 {
105   ip6_address_t *a = va_arg (*args, ip6_address_t *);
106   u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
107
108   i_max_n_zero = ARRAY_LEN (a->as_u16);
109   max_n_zeros = 0;
110   i_first_zero = i_max_n_zero;
111   n_zeros = 0;
112   for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
113     {
114       u32 is_zero = a->as_u16[i] == 0;
115       if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
116         {
117           i_first_zero = i;
118           n_zeros = 0;
119         }
120       n_zeros += is_zero;
121       if ((!is_zero && n_zeros > max_n_zeros)
122           || (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
123         {
124           i_max_n_zero = i_first_zero;
125           max_n_zeros = n_zeros;
126           i_first_zero = ARRAY_LEN (a->as_u16);
127           n_zeros = 0;
128         }
129     }
130
131   last_double_colon = 0;
132   for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
133     {
134       if (i == i_max_n_zero && max_n_zeros > 1)
135         {
136           s = format (s, "::");
137           i += max_n_zeros - 1;
138           last_double_colon = 1;
139         }
140       else
141         {
142           s = format (s, "%s%x",
143                       (last_double_colon || i == 0) ? "" : ":",
144                       clib_net_to_host_u16 (a->as_u16[i]));
145           last_double_colon = 0;
146         }
147     }
148
149   return s;
150 }
151
152 /* Format an IP46 address. */
153 u8 *
154 format_ip46_address (u8 * s, va_list * args)
155 {
156   ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
157   ip46_type_t type = va_arg (*args, ip46_type_t);
158   int is_ip4 = 1;
159
160   switch (type)
161     {
162     case IP46_TYPE_ANY:
163       is_ip4 = ip46_address_is_ip4 (ip46);
164       break;
165     case IP46_TYPE_IP4:
166       is_ip4 = 1;
167       break;
168     case IP46_TYPE_IP6:
169       is_ip4 = 0;
170       break;
171     }
172
173   return is_ip4 ?
174     format (s, "%U", format_ip4_address, &ip46->ip4) :
175     format (s, "%U", format_ip6_address, &ip46->ip6);
176 }
177
178 /*
179  * VPPCOM Utility Functions
180  */
181
182 static void
183 vcl_send_session_listen (vcl_worker_t * wrk, vcl_session_t * s)
184 {
185   app_session_evt_t _app_evt, *app_evt = &_app_evt;
186   session_listen_msg_t *mp;
187   svm_msg_q_t *mq;
188
189   mq = vcl_worker_ctrl_mq (wrk);
190   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_LISTEN);
191   mp = (session_listen_msg_t *) app_evt->evt->data;
192   memset (mp, 0, sizeof (*mp));
193   mp->client_index = wrk->api_client_handle;
194   mp->context = s->session_index;
195   mp->wrk_index = wrk->vpp_wrk_index;
196   mp->is_ip4 = s->transport.is_ip4;
197   clib_memcpy_fast (&mp->ip, &s->transport.lcl_ip, sizeof (mp->ip));
198   mp->port = s->transport.lcl_port;
199   mp->proto = s->session_type;
200   if (s->flags & VCL_SESSION_F_CONNECTED)
201     mp->flags = TRANSPORT_CFG_F_CONNECTED;
202   app_send_ctrl_evt_to_vpp (mq, app_evt);
203 }
204
205 static void
206 vcl_send_session_connect (vcl_worker_t * wrk, vcl_session_t * s)
207 {
208   app_session_evt_t _app_evt, *app_evt = &_app_evt;
209   session_connect_msg_t *mp;
210   svm_msg_q_t *mq;
211
212   mq = vcl_worker_ctrl_mq (wrk);
213   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_CONNECT);
214   mp = (session_connect_msg_t *) app_evt->evt->data;
215   memset (mp, 0, sizeof (*mp));
216   mp->client_index = wrk->api_client_handle;
217   mp->context = s->session_index;
218   mp->wrk_index = wrk->vpp_wrk_index;
219   mp->is_ip4 = s->transport.is_ip4;
220   mp->parent_handle = s->parent_handle;
221   clib_memcpy_fast (&mp->ip, &s->transport.rmt_ip, sizeof (mp->ip));
222   clib_memcpy_fast (&mp->lcl_ip, &s->transport.lcl_ip, sizeof (mp->lcl_ip));
223   mp->port = s->transport.rmt_port;
224   mp->lcl_port = s->transport.lcl_port;
225   mp->proto = s->session_type;
226   if (s->flags & VCL_SESSION_F_CONNECTED)
227     mp->flags |= TRANSPORT_CFG_F_CONNECTED;
228   app_send_ctrl_evt_to_vpp (mq, app_evt);
229 }
230
231 void
232 vcl_send_session_unlisten (vcl_worker_t * wrk, vcl_session_t * s)
233 {
234   app_session_evt_t _app_evt, *app_evt = &_app_evt;
235   session_unlisten_msg_t *mp;
236   svm_msg_q_t *mq;
237
238   mq = vcl_worker_ctrl_mq (wrk);
239   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_UNLISTEN);
240   mp = (session_unlisten_msg_t *) app_evt->evt->data;
241   memset (mp, 0, sizeof (*mp));
242   mp->client_index = wrk->api_client_handle;
243   mp->wrk_index = wrk->vpp_wrk_index;
244   mp->handle = s->vpp_handle;
245   mp->context = wrk->wrk_index;
246   app_send_ctrl_evt_to_vpp (mq, app_evt);
247 }
248
249 static void
250 vcl_send_session_disconnect (vcl_worker_t * wrk, vcl_session_t * s)
251 {
252   app_session_evt_t _app_evt, *app_evt = &_app_evt;
253   session_disconnect_msg_t *mp;
254   svm_msg_q_t *mq;
255
256   /* Send to thread that owns the session */
257   mq = s->vpp_evt_q;
258   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_DISCONNECT);
259   mp = (session_disconnect_msg_t *) app_evt->evt->data;
260   memset (mp, 0, sizeof (*mp));
261   mp->client_index = wrk->api_client_handle;
262   mp->handle = s->vpp_handle;
263   app_send_ctrl_evt_to_vpp (mq, app_evt);
264 }
265
266 static void
267 vcl_send_app_detach (vcl_worker_t * wrk)
268 {
269   app_session_evt_t _app_evt, *app_evt = &_app_evt;
270   session_app_detach_msg_t *mp;
271   svm_msg_q_t *mq;
272
273   mq = vcl_worker_ctrl_mq (wrk);
274   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_DETACH);
275   mp = (session_app_detach_msg_t *) app_evt->evt->data;
276   memset (mp, 0, sizeof (*mp));
277   mp->client_index = wrk->api_client_handle;
278   app_send_ctrl_evt_to_vpp (mq, app_evt);
279 }
280
281 static void
282 vcl_send_session_accepted_reply (svm_msg_q_t * mq, u32 context,
283                                  session_handle_t handle, int retval)
284 {
285   app_session_evt_t _app_evt, *app_evt = &_app_evt;
286   session_accepted_reply_msg_t *rmp;
287   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_ACCEPTED_REPLY);
288   rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
289   rmp->handle = handle;
290   rmp->context = context;
291   rmp->retval = retval;
292   app_send_ctrl_evt_to_vpp (mq, app_evt);
293 }
294
295 static void
296 vcl_send_session_disconnected_reply (svm_msg_q_t * mq, u32 context,
297                                      session_handle_t handle, int retval)
298 {
299   app_session_evt_t _app_evt, *app_evt = &_app_evt;
300   session_disconnected_reply_msg_t *rmp;
301   app_alloc_ctrl_evt_to_vpp (mq, app_evt,
302                              SESSION_CTRL_EVT_DISCONNECTED_REPLY);
303   rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
304   rmp->handle = handle;
305   rmp->context = context;
306   rmp->retval = retval;
307   app_send_ctrl_evt_to_vpp (mq, app_evt);
308 }
309
310 static void
311 vcl_send_session_reset_reply (svm_msg_q_t * mq, u32 context,
312                               session_handle_t handle, int retval)
313 {
314   app_session_evt_t _app_evt, *app_evt = &_app_evt;
315   session_reset_reply_msg_t *rmp;
316   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_RESET_REPLY);
317   rmp = (session_reset_reply_msg_t *) app_evt->evt->data;
318   rmp->handle = handle;
319   rmp->context = context;
320   rmp->retval = retval;
321   app_send_ctrl_evt_to_vpp (mq, app_evt);
322 }
323
324 void
325 vcl_send_session_worker_update (vcl_worker_t * wrk, vcl_session_t * s,
326                                 u32 wrk_index)
327 {
328   app_session_evt_t _app_evt, *app_evt = &_app_evt;
329   session_worker_update_msg_t *mp;
330   svm_msg_q_t *mq;
331
332   mq = vcl_session_vpp_evt_q (wrk, s);
333   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_WORKER_UPDATE);
334   mp = (session_worker_update_msg_t *) app_evt->evt->data;
335   mp->client_index = wrk->api_client_handle;
336   mp->handle = s->vpp_handle;
337   mp->req_wrk_index = wrk->vpp_wrk_index;
338   mp->wrk_index = wrk_index;
339   app_send_ctrl_evt_to_vpp (mq, app_evt);
340 }
341
342 int
343 vcl_send_worker_rpc (u32 dst_wrk_index, void *data, u32 data_len)
344 {
345   app_session_evt_t _app_evt, *app_evt = &_app_evt;
346   session_app_wrk_rpc_msg_t *mp;
347   vcl_worker_t *dst_wrk, *wrk;
348   svm_msg_q_t *mq;
349   int ret = -1;
350
351   if (data_len > sizeof (mp->data))
352     goto done;
353
354   clib_spinlock_lock (&vcm->workers_lock);
355
356   dst_wrk = vcl_worker_get_if_valid (dst_wrk_index);
357   if (!dst_wrk)
358     goto done;
359
360   wrk = vcl_worker_get_current ();
361   mq = vcl_worker_ctrl_mq (wrk);
362   app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_WRK_RPC);
363   mp = (session_app_wrk_rpc_msg_t *) app_evt->evt->data;
364   mp->client_index = wrk->api_client_handle;
365   mp->wrk_index = dst_wrk->vpp_wrk_index;
366   clib_memcpy (mp->data, data, data_len);
367   app_send_ctrl_evt_to_vpp (mq, app_evt);
368   ret = 0;
369
370 done:
371   clib_spinlock_unlock (&vcm->workers_lock);
372   return ret;
373 }
374
375 static u32
376 vcl_session_accepted_handler (vcl_worker_t * wrk, session_accepted_msg_t * mp,
377                               u32 ls_index)
378 {
379   vcl_session_t *session, *listen_session;
380   svm_fifo_t *rx_fifo, *tx_fifo;
381   u32 vpp_wrk_index;
382   svm_msg_q_t *evt_q;
383
384   session = vcl_session_alloc (wrk);
385
386   listen_session = vcl_session_get (wrk, ls_index);
387   if (listen_session->vpp_handle != mp->listener_handle)
388     {
389       VDBG (0, "ERROR: listener handle %lu does not match session %u",
390             mp->listener_handle, ls_index);
391       goto error;
392     }
393
394   if (vcl_segment_is_not_mounted (wrk, mp->segment_handle))
395     {
396       VDBG (0, "ERROR: segment for session %u is not mounted!",
397             session->session_index);
398       goto error;
399     }
400
401   rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
402   tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
403   session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
404                                          svm_msg_q_t *);
405   rx_fifo->client_session_index = session->session_index;
406   tx_fifo->client_session_index = session->session_index;
407   rx_fifo->client_thread_index = vcl_get_worker_index ();
408   tx_fifo->client_thread_index = vcl_get_worker_index ();
409   vpp_wrk_index = tx_fifo->master_thread_index;
410   vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
411   wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
412
413   session->vpp_handle = mp->handle;
414   session->vpp_thread_index = rx_fifo->master_thread_index;
415   session->rx_fifo = rx_fifo;
416   session->tx_fifo = tx_fifo;
417
418   session->session_state = VCL_STATE_READY;
419   session->transport.rmt_port = mp->rmt.port;
420   session->transport.is_ip4 = mp->rmt.is_ip4;
421   clib_memcpy_fast (&session->transport.rmt_ip, &mp->rmt.ip,
422                     sizeof (ip46_address_t));
423
424   vcl_session_table_add_vpp_handle (wrk, mp->handle, session->session_index);
425   session->transport.lcl_port = listen_session->transport.lcl_port;
426   session->transport.lcl_ip = listen_session->transport.lcl_ip;
427   session->session_type = listen_session->session_type;
428   session->is_dgram = vcl_proto_is_dgram (session->session_type);
429   session->listener_index = listen_session->session_index;
430   listen_session->n_accepted_sessions++;
431
432   VDBG (1, "session %u [0x%llx]: client accept request from %s address %U"
433         " port %d queue %p!", session->session_index, mp->handle,
434         mp->rmt.is_ip4 ? "IPv4" : "IPv6", format_ip46_address, &mp->rmt.ip,
435         mp->rmt.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
436         clib_net_to_host_u16 (mp->rmt.port), session->vpp_evt_q);
437   vcl_evt (VCL_EVT_ACCEPT, session, listen_session, session_index);
438
439   vcl_send_session_accepted_reply (session->vpp_evt_q, mp->context,
440                                    session->vpp_handle, 0);
441
442   return session->session_index;
443
444 error:
445   evt_q = uword_to_pointer (mp->vpp_event_queue_address, svm_msg_q_t *);
446   vcl_send_session_accepted_reply (evt_q, mp->context, mp->handle,
447                                    VNET_API_ERROR_INVALID_ARGUMENT);
448   vcl_session_free (wrk, session);
449   return VCL_INVALID_SESSION_INDEX;
450 }
451
452 static u32
453 vcl_session_connected_handler (vcl_worker_t * wrk,
454                                session_connected_msg_t * mp)
455 {
456   u32 session_index, vpp_wrk_index;
457   svm_fifo_t *rx_fifo, *tx_fifo;
458   vcl_session_t *session = 0;
459
460   session_index = mp->context;
461   session = vcl_session_get (wrk, session_index);
462   if (!session)
463     {
464       VDBG (0, "ERROR: vpp handle 0x%llx has no session index (%u)!",
465             mp->handle, session_index);
466       return VCL_INVALID_SESSION_INDEX;
467     }
468   if (mp->retval)
469     {
470       VDBG (0, "ERROR: session index %u: connect failed! %U",
471             session_index, format_session_error, mp->retval);
472       session->session_state = VCL_STATE_DETACHED;
473       session->vpp_handle = mp->handle;
474       return session_index;
475     }
476
477   session->vpp_handle = mp->handle;
478   session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
479                                          svm_msg_q_t *);
480   rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
481   tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
482   if (vcl_segment_is_not_mounted (wrk, mp->segment_handle))
483     {
484       VDBG (0, "segment for session %u is not mounted!",
485             session->session_index);
486       session->session_state = VCL_STATE_DETACHED;
487       vcl_send_session_disconnect (wrk, session);
488       return session_index;
489     }
490
491   rx_fifo->client_session_index = session_index;
492   tx_fifo->client_session_index = session_index;
493   rx_fifo->client_thread_index = vcl_get_worker_index ();
494   tx_fifo->client_thread_index = vcl_get_worker_index ();
495
496   vpp_wrk_index = tx_fifo->master_thread_index;
497   vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
498   wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
499
500   if (mp->ct_rx_fifo)
501     {
502       session->ct_rx_fifo = uword_to_pointer (mp->ct_rx_fifo, svm_fifo_t *);
503       session->ct_tx_fifo = uword_to_pointer (mp->ct_tx_fifo, svm_fifo_t *);
504       if (vcl_segment_is_not_mounted (wrk, mp->ct_segment_handle))
505         {
506           VDBG (0, "ct segment for session %u is not mounted!",
507                 session->session_index);
508           session->session_state = VCL_STATE_DETACHED;
509           vcl_send_session_disconnect (wrk, session);
510           return session_index;
511         }
512     }
513
514   session->rx_fifo = rx_fifo;
515   session->tx_fifo = tx_fifo;
516   session->vpp_thread_index = rx_fifo->master_thread_index;
517   session->transport.is_ip4 = mp->lcl.is_ip4;
518   clib_memcpy_fast (&session->transport.lcl_ip, &mp->lcl.ip,
519                     sizeof (session->transport.lcl_ip));
520   session->transport.lcl_port = mp->lcl.port;
521
522   /* Application closed session before connect reply */
523   if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK)
524       && session->session_state == VCL_STATE_CLOSED)
525     vcl_send_session_disconnect (wrk, session);
526   else
527     session->session_state = VCL_STATE_READY;
528
529   /* Add it to lookup table */
530   vcl_session_table_add_vpp_handle (wrk, mp->handle, session_index);
531
532   VDBG (1, "session %u [0x%llx] connected! rx_fifo %p, refcnt %d, tx_fifo %p,"
533         " refcnt %d", session_index, mp->handle, session->rx_fifo,
534         session->rx_fifo->refcnt, session->tx_fifo, session->tx_fifo->refcnt);
535
536   return session_index;
537 }
538
539 static int
540 vcl_flag_accepted_session (vcl_session_t * session, u64 handle, u32 flags)
541 {
542   vcl_session_msg_t *accepted_msg;
543   int i;
544
545   for (i = 0; i < vec_len (session->accept_evts_fifo); i++)
546     {
547       accepted_msg = &session->accept_evts_fifo[i];
548       if (accepted_msg->accepted_msg.handle == handle)
549         {
550           accepted_msg->flags |= flags;
551           return 1;
552         }
553     }
554   return 0;
555 }
556
557 static u32
558 vcl_session_reset_handler (vcl_worker_t * wrk,
559                            session_reset_msg_t * reset_msg)
560 {
561   vcl_session_t *session;
562   u32 sid;
563
564   sid = vcl_session_index_from_vpp_handle (wrk, reset_msg->handle);
565   session = vcl_session_get (wrk, sid);
566   if (!session)
567     {
568       VDBG (0, "request to reset unknown handle 0x%llx", reset_msg->handle);
569       return VCL_INVALID_SESSION_INDEX;
570     }
571
572   /* Caught a reset before actually accepting the session */
573   if (session->session_state == VCL_STATE_LISTEN)
574     {
575
576       if (!vcl_flag_accepted_session (session, reset_msg->handle,
577                                       VCL_ACCEPTED_F_RESET))
578         VDBG (0, "session was not accepted!");
579       return VCL_INVALID_SESSION_INDEX;
580     }
581
582   if (session->session_state != VCL_STATE_CLOSED)
583     session->session_state = VCL_STATE_DISCONNECT;
584   VDBG (0, "reset session %u [0x%llx]", sid, reset_msg->handle);
585   return sid;
586 }
587
588 static u32
589 vcl_session_bound_handler (vcl_worker_t * wrk, session_bound_msg_t * mp)
590 {
591   vcl_session_t *session;
592   u32 sid = mp->context;
593
594   session = vcl_session_get (wrk, sid);
595   if (mp->retval)
596     {
597       VERR ("session %u [0x%llx]: bind failed: %U", sid, mp->handle,
598             format_session_error, mp->retval);
599       if (session)
600         {
601           session->session_state = VCL_STATE_DETACHED;
602           session->vpp_handle = mp->handle;
603           return sid;
604         }
605       else
606         {
607           VDBG (0, "ERROR: session %u [0x%llx]: Invalid session index!",
608                 sid, mp->handle);
609           return VCL_INVALID_SESSION_INDEX;
610         }
611     }
612
613   session->vpp_handle = mp->handle;
614   session->transport.is_ip4 = mp->lcl_is_ip4;
615   clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
616                     sizeof (ip46_address_t));
617   session->transport.lcl_port = mp->lcl_port;
618   vcl_session_table_add_listener (wrk, mp->handle, sid);
619   session->session_state = VCL_STATE_LISTEN;
620
621   session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
622   vec_validate (wrk->vpp_event_queues, 0);
623   wrk->vpp_event_queues[0] = session->vpp_evt_q;
624
625   if (vcl_session_is_cl (session))
626     {
627       svm_fifo_t *rx_fifo, *tx_fifo;
628       session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
629       rx_fifo = uword_to_pointer (mp->rx_fifo, svm_fifo_t *);
630       rx_fifo->client_session_index = sid;
631       tx_fifo = uword_to_pointer (mp->tx_fifo, svm_fifo_t *);
632       tx_fifo->client_session_index = sid;
633       session->rx_fifo = rx_fifo;
634       session->tx_fifo = tx_fifo;
635     }
636
637   VDBG (0, "session %u [0x%llx]: listen succeeded!", sid, mp->handle);
638   return sid;
639 }
640
641 static void
642 vcl_session_unlisten_reply_handler (vcl_worker_t * wrk, void *data)
643 {
644   session_unlisten_reply_msg_t *mp = (session_unlisten_reply_msg_t *) data;
645   vcl_session_t *s;
646
647   s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
648   if (!s)
649     {
650       VDBG (0, "Unlisten reply with wrong handle %llx", mp->handle);
651       return;
652     }
653   if (s->session_state != VCL_STATE_DISCONNECT)
654     {
655       /* Connected udp listener */
656       if (s->session_type == VPPCOM_PROTO_UDP
657           && s->session_state == VCL_STATE_CLOSED)
658         return;
659
660       VDBG (0, "Unlisten session in wrong state %llx", mp->handle);
661       return;
662     }
663
664   if (mp->retval)
665     VDBG (0, "ERROR: session %u [0xllx]: unlisten failed: %U",
666           s->session_index, mp->handle, format_session_error, mp->retval);
667
668   if (mp->context != wrk->wrk_index)
669     VDBG (0, "wrong context");
670
671   vcl_session_table_del_vpp_handle (wrk, mp->handle);
672   vcl_session_free (wrk, s);
673 }
674
675 static void
676 vcl_session_migrated_handler (vcl_worker_t * wrk, void *data)
677 {
678   session_migrated_msg_t *mp = (session_migrated_msg_t *) data;
679   vcl_session_t *s;
680
681   s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
682   if (!s)
683     {
684       VDBG (0, "Migrated notification with wrong handle %llx", mp->handle);
685       return;
686     }
687
688   s->vpp_thread_index = mp->vpp_thread_index;
689   s->vpp_handle = mp->new_handle;
690   s->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
691
692   vec_validate (wrk->vpp_event_queues, s->vpp_thread_index);
693   wrk->vpp_event_queues[s->vpp_thread_index] = s->vpp_evt_q;
694
695   vcl_session_table_del_vpp_handle (wrk, mp->handle);
696   vcl_session_table_add_vpp_handle (wrk, mp->new_handle, s->session_index);
697
698   /* Generate new tx event if we have outstanding data */
699   if (svm_fifo_has_event (s->tx_fifo))
700     app_send_io_evt_to_vpp (s->vpp_evt_q, s->tx_fifo->master_session_index,
701                             SESSION_IO_EVT_TX, SVM_Q_WAIT);
702
703   VDBG (0, "Migrated 0x%lx to thread %u 0x%lx", mp->handle,
704         s->vpp_thread_index, mp->new_handle);
705 }
706
707 static vcl_session_t *
708 vcl_session_accepted (vcl_worker_t * wrk, session_accepted_msg_t * msg)
709 {
710   vcl_session_msg_t *vcl_msg;
711   vcl_session_t *session;
712
713   session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
714   if (PREDICT_FALSE (session != 0))
715     VWRN ("session overlap handle %lu state %u!", msg->handle,
716           session->session_state);
717
718   session = vcl_session_table_lookup_listener (wrk, msg->listener_handle);
719   if (!session)
720     {
721       VERR ("couldn't find listen session: listener handle %llx",
722             msg->listener_handle);
723       return 0;
724     }
725
726   clib_fifo_add2 (session->accept_evts_fifo, vcl_msg);
727   vcl_msg->flags = 0;
728   vcl_msg->accepted_msg = *msg;
729   /* Session handle points to listener until fully accepted by app */
730   vcl_session_table_add_vpp_handle (wrk, msg->handle, session->session_index);
731
732   return session;
733 }
734
735 static vcl_session_t *
736 vcl_session_disconnected_handler (vcl_worker_t * wrk,
737                                   session_disconnected_msg_t * msg)
738 {
739   vcl_session_t *session;
740
741   session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
742   if (!session)
743     {
744       VDBG (0, "request to disconnect unknown handle 0x%llx", msg->handle);
745       return 0;
746     }
747
748   /* Late disconnect notification on a session that has been closed */
749   if (session->session_state == VCL_STATE_CLOSED)
750     return 0;
751
752   /* Caught a disconnect before actually accepting the session */
753   if (session->session_state == VCL_STATE_LISTEN)
754     {
755       if (!vcl_flag_accepted_session (session, msg->handle,
756                                       VCL_ACCEPTED_F_CLOSED))
757         VDBG (0, "session was not accepted!");
758       return 0;
759     }
760
761   /* If not already reset change state */
762   if (session->session_state != VCL_STATE_DISCONNECT)
763     session->session_state = VCL_STATE_VPP_CLOSING;
764
765   return session;
766 }
767
768 static void
769 vcl_session_cleanup_handler (vcl_worker_t * wrk, void *data)
770 {
771   session_cleanup_msg_t *msg;
772   vcl_session_t *session;
773
774   msg = (session_cleanup_msg_t *) data;
775   session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
776   if (!session)
777     {
778       VDBG (0, "disconnect confirmed for unknown handle 0x%llx", msg->handle);
779       return;
780     }
781
782   if (msg->type == SESSION_CLEANUP_TRANSPORT)
783     {
784       /* Transport was cleaned up before we confirmed close. Probably the
785        * app is still waiting for some data that cannot be delivered.
786        * Confirm close to make sure everything is cleaned up */
787       if (session->session_state == VCL_STATE_VPP_CLOSING)
788         vcl_session_cleanup (wrk, session, vcl_session_handle (session),
789                              1 /* do_disconnect */ );
790       return;
791     }
792
793   vcl_session_table_del_vpp_handle (wrk, msg->handle);
794   /* Should not happen. App did not close the connection so don't free it. */
795   if (session->session_state != VCL_STATE_CLOSED)
796     {
797       VDBG (0, "app did not close session %d", session->session_index);
798       session->session_state = VCL_STATE_DETACHED;
799       session->vpp_handle = VCL_INVALID_SESSION_HANDLE;
800       return;
801     }
802   vcl_session_free (wrk, session);
803 }
804
805 static void
806 vcl_session_req_worker_update_handler (vcl_worker_t * wrk, void *data)
807 {
808   session_req_worker_update_msg_t *msg;
809   vcl_session_t *s;
810
811   msg = (session_req_worker_update_msg_t *) data;
812   s = vcl_session_get_w_vpp_handle (wrk, msg->session_handle);
813   if (!s)
814     return;
815
816   vec_add1 (wrk->pending_session_wrk_updates, s->session_index);
817 }
818
819 static void
820 vcl_session_worker_update_reply_handler (vcl_worker_t * wrk, void *data)
821 {
822   session_worker_update_reply_msg_t *msg;
823   vcl_session_t *s;
824
825   msg = (session_worker_update_reply_msg_t *) data;
826   s = vcl_session_get_w_vpp_handle (wrk, msg->handle);
827   if (!s)
828     {
829       VDBG (0, "unknown handle 0x%llx", msg->handle);
830       return;
831     }
832   if (vcl_segment_is_not_mounted (wrk, msg->segment_handle))
833     {
834       clib_warning ("segment for session %u is not mounted!",
835                     s->session_index);
836       return;
837     }
838
839   if (s->rx_fifo)
840     {
841       s->rx_fifo = uword_to_pointer (msg->rx_fifo, svm_fifo_t *);
842       s->tx_fifo = uword_to_pointer (msg->tx_fifo, svm_fifo_t *);
843       s->rx_fifo->client_session_index = s->session_index;
844       s->tx_fifo->client_session_index = s->session_index;
845       s->rx_fifo->client_thread_index = wrk->wrk_index;
846       s->tx_fifo->client_thread_index = wrk->wrk_index;
847     }
848   s->session_state = VCL_STATE_UPDATED;
849
850   VDBG (0, "session %u[0x%llx] moved to worker %u", s->session_index,
851         s->vpp_handle, wrk->wrk_index);
852 }
853
854 static int
855 vcl_api_recv_fd (vcl_worker_t * wrk, int *fds, int n_fds)
856 {
857
858   if (vcm->cfg.vpp_app_socket_api)
859     return vcl_sapi_recv_fds (wrk, fds, n_fds);
860
861   return vcl_bapi_recv_fds (wrk, fds, n_fds);
862 }
863
864 static void
865 vcl_session_app_add_segment_handler (vcl_worker_t * wrk, void *data)
866 {
867   ssvm_segment_type_t seg_type = SSVM_SEGMENT_SHM;
868   session_app_add_segment_msg_t *msg;
869   u64 segment_handle;
870   int fd = -1;
871
872   msg = (session_app_add_segment_msg_t *) data;
873
874   if (msg->fd_flags)
875     {
876       vcl_api_recv_fd (wrk, &fd, 1);
877       seg_type = SSVM_SEGMENT_MEMFD;
878     }
879
880   segment_handle = msg->segment_handle;
881   if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
882     {
883       clib_warning ("invalid segment handle");
884       return;
885     }
886
887   if (vcl_segment_attach (segment_handle, (char *) msg->segment_name,
888                           seg_type, fd))
889     {
890       VDBG (0, "vcl_segment_attach ('%s') failed", msg->segment_name);
891       return;
892     }
893
894   VDBG (1, "mapped new segment '%s' size %d", msg->segment_name,
895         msg->segment_size);
896 }
897
898 static void
899 vcl_session_app_del_segment_handler (vcl_worker_t * wrk, void *data)
900 {
901   session_app_del_segment_msg_t *msg = (session_app_del_segment_msg_t *) data;
902   vcl_segment_detach (msg->segment_handle);
903   VDBG (1, "Unmapped segment: %d", msg->segment_handle);
904 }
905
906 static void
907 vcl_worker_rpc_handler (vcl_worker_t * wrk, void *data)
908 {
909   if (!vcm->wrk_rpc_fn)
910     return;
911
912   (vcm->wrk_rpc_fn) (((session_app_wrk_rpc_msg_t *) data)->data);
913 }
914
915 static int
916 vcl_handle_mq_event (vcl_worker_t * wrk, session_event_t * e)
917 {
918   session_disconnected_msg_t *disconnected_msg;
919   vcl_session_t *s;
920
921   switch (e->event_type)
922     {
923     case SESSION_IO_EVT_RX:
924     case SESSION_IO_EVT_TX:
925       s = vcl_session_get (wrk, e->session_index);
926       if (!s || !(vcl_session_is_ready (s)))
927         break;
928       vec_add1 (wrk->unhandled_evts_vector, *e);
929       break;
930     case SESSION_CTRL_EVT_ACCEPTED:
931       vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
932       break;
933     case SESSION_CTRL_EVT_CONNECTED:
934       vcl_session_connected_handler (wrk,
935                                      (session_connected_msg_t *) e->data);
936       break;
937     case SESSION_CTRL_EVT_DISCONNECTED:
938       disconnected_msg = (session_disconnected_msg_t *) e->data;
939       s = vcl_session_disconnected_handler (wrk, disconnected_msg);
940       if (!s)
941         break;
942       VDBG (0, "disconnected session %u [0x%llx]", s->session_index,
943             s->vpp_handle);
944       break;
945     case SESSION_CTRL_EVT_RESET:
946       vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
947       break;
948     case SESSION_CTRL_EVT_BOUND:
949       vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
950       break;
951     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
952       vcl_session_unlisten_reply_handler (wrk, e->data);
953       break;
954     case SESSION_CTRL_EVT_MIGRATED:
955       vcl_session_migrated_handler (wrk, e->data);
956       break;
957     case SESSION_CTRL_EVT_CLEANUP:
958       vcl_session_cleanup_handler (wrk, e->data);
959       break;
960     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
961       vcl_session_req_worker_update_handler (wrk, e->data);
962       break;
963     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
964       vcl_session_worker_update_reply_handler (wrk, e->data);
965       break;
966     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
967       vcl_session_app_add_segment_handler (wrk, e->data);
968       break;
969     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
970       vcl_session_app_del_segment_handler (wrk, e->data);
971       break;
972     case SESSION_CTRL_EVT_APP_WRK_RPC:
973       vcl_worker_rpc_handler (wrk, e->data);
974       break;
975     default:
976       clib_warning ("unhandled %u", e->event_type);
977     }
978   return VPPCOM_OK;
979 }
980
981 static int
982 vppcom_wait_for_session_state_change (u32 session_index,
983                                       vcl_session_state_t state,
984                                       f64 wait_for_time)
985 {
986   vcl_worker_t *wrk = vcl_worker_get_current ();
987   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
988   vcl_session_t *volatile session;
989   svm_msg_q_msg_t msg;
990   session_event_t *e;
991
992   do
993     {
994       session = vcl_session_get (wrk, session_index);
995       if (PREDICT_FALSE (!session))
996         {
997           return VPPCOM_EBADFD;
998         }
999       if (session->session_state == state)
1000         {
1001           return VPPCOM_OK;
1002         }
1003       if (session->session_state == VCL_STATE_DETACHED)
1004         {
1005           return VPPCOM_ECONNREFUSED;
1006         }
1007
1008       if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0))
1009         {
1010           usleep (100);
1011           continue;
1012         }
1013       e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
1014       vcl_handle_mq_event (wrk, e);
1015       svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1016     }
1017   while (clib_time_now (&wrk->clib_time) < timeout);
1018
1019   VDBG (0, "timeout waiting for state 0x%x (%s)", state,
1020         vppcom_session_state_str (state));
1021   vcl_evt (VCL_EVT_SESSION_TIMEOUT, session, session_state);
1022
1023   return VPPCOM_ETIMEDOUT;
1024 }
1025
1026 static void
1027 vcl_handle_pending_wrk_updates (vcl_worker_t * wrk)
1028 {
1029   vcl_session_state_t state;
1030   vcl_session_t *s;
1031   u32 *sip;
1032
1033   if (PREDICT_TRUE (vec_len (wrk->pending_session_wrk_updates) == 0))
1034     return;
1035
1036   vec_foreach (sip, wrk->pending_session_wrk_updates)
1037   {
1038     s = vcl_session_get (wrk, *sip);
1039     vcl_send_session_worker_update (wrk, s, wrk->wrk_index);
1040     state = s->session_state;
1041     vppcom_wait_for_session_state_change (s->session_index, VCL_STATE_UPDATED,
1042                                           5);
1043     s->session_state = state;
1044   }
1045   vec_reset_length (wrk->pending_session_wrk_updates);
1046 }
1047
1048 void
1049 vcl_flush_mq_events (void)
1050 {
1051   vcl_worker_t *wrk = vcl_worker_get_current ();
1052   svm_msg_q_msg_t *msg;
1053   session_event_t *e;
1054   svm_msg_q_t *mq;
1055   int i;
1056
1057   mq = wrk->app_event_queue;
1058   svm_msg_q_lock (mq);
1059   vcl_mq_dequeue_batch (wrk, mq, ~0);
1060   svm_msg_q_unlock (mq);
1061
1062   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
1063     {
1064       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
1065       e = svm_msg_q_msg_data (mq, msg);
1066       vcl_handle_mq_event (wrk, e);
1067       svm_msg_q_free_msg (mq, msg);
1068     }
1069   vec_reset_length (wrk->mq_msg_vector);
1070   vcl_handle_pending_wrk_updates (wrk);
1071 }
1072
1073 static int
1074 vppcom_session_unbind (u32 session_handle)
1075 {
1076   vcl_worker_t *wrk = vcl_worker_get_current ();
1077   session_accepted_msg_t *accepted_msg;
1078   vcl_session_t *session = 0;
1079   vcl_session_msg_t *evt;
1080
1081   session = vcl_session_get_w_handle (wrk, session_handle);
1082   if (!session)
1083     return VPPCOM_EBADFD;
1084
1085   /* Flush pending accept events, if any */
1086   while (clib_fifo_elts (session->accept_evts_fifo))
1087     {
1088       clib_fifo_sub2 (session->accept_evts_fifo, evt);
1089       accepted_msg = &evt->accepted_msg;
1090       vcl_session_table_del_vpp_handle (wrk, accepted_msg->handle);
1091       vcl_send_session_accepted_reply (session->vpp_evt_q,
1092                                        accepted_msg->context,
1093                                        session->vpp_handle, -1);
1094     }
1095   clib_fifo_free (session->accept_evts_fifo);
1096
1097   vcl_send_session_unlisten (wrk, session);
1098
1099   VDBG (1, "session %u [0x%llx]: sending unbind!", session->session_index,
1100         session->vpp_handle);
1101   vcl_evt (VCL_EVT_UNBIND, session);
1102
1103   session->vpp_handle = ~0;
1104   session->session_state = VCL_STATE_DISCONNECT;
1105
1106   return VPPCOM_OK;
1107 }
1108
1109 static int
1110 vppcom_session_disconnect (u32 session_handle)
1111 {
1112   vcl_worker_t *wrk = vcl_worker_get_current ();
1113   svm_msg_q_t *vpp_evt_q;
1114   vcl_session_t *session, *listen_session;
1115   vcl_session_state_t state;
1116   u64 vpp_handle;
1117
1118   session = vcl_session_get_w_handle (wrk, session_handle);
1119   if (!session)
1120     return VPPCOM_EBADFD;
1121
1122   vpp_handle = session->vpp_handle;
1123   state = session->session_state;
1124
1125   VDBG (1, "session %u [0x%llx] state 0x%x (%s)", session->session_index,
1126         vpp_handle, state, vppcom_session_state_str (state));
1127
1128   if (PREDICT_FALSE (state == VCL_STATE_LISTEN))
1129     {
1130       VDBG (0, "ERROR: Cannot disconnect a listen socket!");
1131       return VPPCOM_EBADFD;
1132     }
1133
1134   if (state == VCL_STATE_VPP_CLOSING)
1135     {
1136       vpp_evt_q = vcl_session_vpp_evt_q (wrk, session);
1137       vcl_send_session_disconnected_reply (vpp_evt_q, wrk->api_client_handle,
1138                                            vpp_handle, 0);
1139       VDBG (1, "session %u [0x%llx]: sending disconnect REPLY...",
1140             session->session_index, vpp_handle);
1141     }
1142   else
1143     {
1144       /* Session doesn't have an event queue yet. Probably a non-blocking
1145        * connect. Wait for the reply */
1146       if (PREDICT_FALSE (!session->vpp_evt_q))
1147         return VPPCOM_OK;
1148
1149       VDBG (1, "session %u [0x%llx]: sending disconnect...",
1150             session->session_index, vpp_handle);
1151       vcl_send_session_disconnect (wrk, session);
1152     }
1153
1154   if (session->listener_index != VCL_INVALID_SESSION_INDEX)
1155     {
1156       listen_session = vcl_session_get (wrk, session->listener_index);
1157       listen_session->n_accepted_sessions--;
1158     }
1159
1160   return VPPCOM_OK;
1161 }
1162
1163 /**
1164  * Handle app exit
1165  *
1166  * Notify vpp of the disconnect and mark the worker as free. If we're the
1167  * last worker, do a full cleanup otherwise, since we're probably a forked
1168  * child, avoid syscalls as much as possible. We might've lost privileges.
1169  */
1170 void
1171 vppcom_app_exit (void)
1172 {
1173   if (!pool_elts (vcm->workers))
1174     return;
1175   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
1176   vcl_set_worker_index (~0);
1177   vcl_elog_stop (vcm);
1178 }
1179
1180 static int
1181 vcl_api_attach (void)
1182 {
1183   if (vcm->cfg.vpp_app_socket_api)
1184     return vcl_sapi_attach ();
1185
1186   return vcl_bapi_attach ();
1187 }
1188
1189 static void
1190 vcl_api_detach (vcl_worker_t * wrk)
1191 {
1192   vcl_send_app_detach (wrk);
1193
1194   if (vcm->cfg.vpp_app_socket_api)
1195     return vcl_sapi_detach (wrk);
1196
1197   return vcl_bapi_disconnect_from_vpp ();
1198 }
1199
1200 /*
1201  * VPPCOM Public API functions
1202  */
1203 int
1204 vppcom_app_create (const char *app_name)
1205 {
1206   vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1207   int rv;
1208
1209   if (vcm->is_init)
1210     {
1211       VDBG (1, "already initialized");
1212       return VPPCOM_EEXIST;
1213     }
1214
1215   vcm->is_init = 1;
1216   vppcom_cfg (&vcm->cfg);
1217   vcl_cfg = &vcm->cfg;
1218
1219   vcm->main_cpu = pthread_self ();
1220   vcm->main_pid = getpid ();
1221   vcm->app_name = format (0, "%s", app_name);
1222   fifo_segment_main_init (&vcm->segment_main, vcl_cfg->segment_baseva,
1223                           20 /* timeout in secs */ );
1224   pool_alloc (vcm->workers, vcl_cfg->max_workers);
1225   clib_spinlock_init (&vcm->workers_lock);
1226   clib_rwlock_init (&vcm->segment_table_lock);
1227   atexit (vppcom_app_exit);
1228   vcl_elog_init (vcm);
1229
1230   /* Allocate default worker */
1231   vcl_worker_alloc_and_init ();
1232
1233   if ((rv = vcl_api_attach ()))
1234     return rv;
1235
1236   VDBG (0, "app_name '%s', my_client_index %d (0x%x)", app_name,
1237         vcm->workers[0].api_client_handle, vcm->workers[0].api_client_handle);
1238
1239   return VPPCOM_OK;
1240 }
1241
1242 void
1243 vppcom_app_destroy (void)
1244 {
1245   vcl_worker_t *wrk, *current_wrk;
1246   void *heap;
1247
1248   if (!pool_elts (vcm->workers))
1249     return;
1250
1251   vcl_evt (VCL_EVT_DETACH, vcm);
1252
1253   current_wrk = vcl_worker_get_current ();
1254
1255   /* *INDENT-OFF* */
1256   pool_foreach (wrk, vcm->workers, ({
1257     if (current_wrk != wrk)
1258       vcl_worker_cleanup (wrk, 0 /* notify vpp */ );
1259   }));
1260   /* *INDENT-ON* */
1261
1262   vcl_api_detach (current_wrk);
1263   vcl_worker_cleanup (current_wrk, 0 /* notify vpp */ );
1264
1265   vcl_elog_stop (vcm);
1266
1267   /*
1268    * Free the heap and fix vcm
1269    */
1270   heap = clib_mem_get_heap ();
1271   munmap (clib_mem_get_heap_base (heap), clib_mem_get_heap_size (heap));
1272
1273   vcm = &_vppcom_main;
1274   vcm->is_init = 0;
1275 }
1276
1277 int
1278 vppcom_session_create (u8 proto, u8 is_nonblocking)
1279 {
1280   vcl_worker_t *wrk = vcl_worker_get_current ();
1281   vcl_session_t *session;
1282
1283   session = vcl_session_alloc (wrk);
1284
1285   session->session_type = proto;
1286   session->session_state = VCL_STATE_CLOSED;
1287   session->vpp_handle = ~0;
1288   session->is_dgram = vcl_proto_is_dgram (proto);
1289
1290   if (is_nonblocking)
1291     VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
1292
1293   vcl_evt (VCL_EVT_CREATE, session, session_type, session->session_state,
1294            is_nonblocking, session_index);
1295
1296   VDBG (0, "created session %u", session->session_index);
1297
1298   return vcl_session_handle (session);
1299 }
1300
1301 int
1302 vcl_session_cleanup (vcl_worker_t * wrk, vcl_session_t * s,
1303                      vcl_session_handle_t sh, u8 do_disconnect)
1304 {
1305   vcl_session_state_t state;
1306   u32 next_sh, vep_sh;
1307   int rv = VPPCOM_OK;
1308   u64 vpp_handle;
1309   u8 is_vep;
1310
1311   is_vep = s->is_vep;
1312   next_sh = s->vep.next_sh;
1313   vep_sh = s->vep.vep_sh;
1314   state = s->session_state;
1315   vpp_handle = s->vpp_handle;
1316
1317   VDBG (1, "session %u [0x%llx] closing", s->session_index, vpp_handle);
1318
1319   if (is_vep)
1320     {
1321       while (next_sh != ~0)
1322         {
1323           rv = vppcom_epoll_ctl (sh, EPOLL_CTL_DEL, next_sh, 0);
1324           if (PREDICT_FALSE (rv < 0))
1325             VDBG (0, "vpp handle 0x%llx, sh %u: EPOLL_CTL_DEL vep_idx %u"
1326                   " failed! rv %d (%s)", vpp_handle, next_sh, vep_sh, rv,
1327                   vppcom_retval_str (rv));
1328
1329           next_sh = s->vep.next_sh;
1330         }
1331       goto cleanup;
1332     }
1333
1334   if (s->is_vep_session)
1335     {
1336       rv = vppcom_epoll_ctl (vep_sh, EPOLL_CTL_DEL, sh, 0);
1337       if (rv < 0)
1338         VDBG (0, "session %u [0x%llx]: EPOLL_CTL_DEL vep_idx %u "
1339               "failed! rv %d (%s)", s->session_index, vpp_handle,
1340               vep_sh, rv, vppcom_retval_str (rv));
1341     }
1342
1343   if (!do_disconnect)
1344     {
1345       VDBG (1, "session %u [0x%llx] disconnect skipped",
1346             s->session_index, vpp_handle);
1347       goto cleanup;
1348     }
1349
1350   if (state == VCL_STATE_LISTEN)
1351     {
1352       rv = vppcom_session_unbind (sh);
1353       if (PREDICT_FALSE (rv < 0))
1354         VDBG (0, "session %u [0x%llx]: listener unbind failed! "
1355               "rv %d (%s)", s->session_index, vpp_handle, rv,
1356               vppcom_retval_str (rv));
1357       return rv;
1358     }
1359   else if ((vcl_session_is_ready (s))
1360            || (vcl_session_is_connectable_listener (wrk, s)))
1361     {
1362       rv = vppcom_session_disconnect (sh);
1363       if (PREDICT_FALSE (rv < 0))
1364         VDBG (0, "ERROR: session %u [0x%llx]: disconnect failed!"
1365               " rv %d (%s)", s->session_index, vpp_handle,
1366               rv, vppcom_retval_str (rv));
1367     }
1368   else if (state == VCL_STATE_DISCONNECT)
1369     {
1370       svm_msg_q_t *mq = vcl_session_vpp_evt_q (wrk, s);
1371       vcl_send_session_reset_reply (mq, wrk->api_client_handle,
1372                                     s->vpp_handle, 0);
1373     }
1374   else if (state == VCL_STATE_DETACHED)
1375     {
1376       /* Should not happen. VPP cleaned up before app confirmed close */
1377       VDBG (0, "vpp freed session %d before close", s->session_index);
1378       goto free_session;
1379     }
1380
1381   s->session_state = VCL_STATE_CLOSED;
1382
1383   /* Session is removed only after vpp confirms the disconnect */
1384   return rv;
1385
1386 cleanup:
1387   vcl_session_table_del_vpp_handle (wrk, vpp_handle);
1388 free_session:
1389   vcl_session_free (wrk, s);
1390   vcl_evt (VCL_EVT_CLOSE, s, rv);
1391
1392   return rv;
1393 }
1394
1395 int
1396 vppcom_session_close (uint32_t session_handle)
1397 {
1398   vcl_worker_t *wrk = vcl_worker_get_current ();
1399   vcl_session_t *session;
1400
1401   session = vcl_session_get_w_handle (wrk, session_handle);
1402   if (!session)
1403     return VPPCOM_EBADFD;
1404   return vcl_session_cleanup (wrk, session, session_handle,
1405                               1 /* do_disconnect */ );
1406 }
1407
1408 int
1409 vppcom_session_bind (uint32_t session_handle, vppcom_endpt_t * ep)
1410 {
1411   vcl_worker_t *wrk = vcl_worker_get_current ();
1412   vcl_session_t *session = 0;
1413
1414   if (!ep || !ep->ip)
1415     return VPPCOM_EINVAL;
1416
1417   session = vcl_session_get_w_handle (wrk, session_handle);
1418   if (!session)
1419     return VPPCOM_EBADFD;
1420
1421   if (session->is_vep)
1422     {
1423       VDBG (0, "ERROR: cannot bind to epoll session %u!",
1424             session->session_index);
1425       return VPPCOM_EBADFD;
1426     }
1427
1428   session->transport.is_ip4 = ep->is_ip4;
1429   if (ep->is_ip4)
1430     clib_memcpy_fast (&session->transport.lcl_ip.ip4, ep->ip,
1431                       sizeof (ip4_address_t));
1432   else
1433     clib_memcpy_fast (&session->transport.lcl_ip.ip6, ep->ip,
1434                       sizeof (ip6_address_t));
1435   session->transport.lcl_port = ep->port;
1436
1437   VDBG (0, "session %u handle %u: binding to local %s address %U port %u, "
1438         "proto %s", session->session_index, session_handle,
1439         session->transport.is_ip4 ? "IPv4" : "IPv6",
1440         format_ip46_address, &session->transport.lcl_ip,
1441         session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1442         clib_net_to_host_u16 (session->transport.lcl_port),
1443         vppcom_proto_str (session->session_type));
1444   vcl_evt (VCL_EVT_BIND, session);
1445
1446   if (session->session_type == VPPCOM_PROTO_UDP)
1447     vppcom_session_listen (session_handle, 10);
1448
1449   return VPPCOM_OK;
1450 }
1451
1452 int
1453 vppcom_session_listen (uint32_t listen_sh, uint32_t q_len)
1454 {
1455   vcl_worker_t *wrk = vcl_worker_get_current ();
1456   vcl_session_t *listen_session = 0;
1457   u64 listen_vpp_handle;
1458   int rv;
1459
1460   listen_session = vcl_session_get_w_handle (wrk, listen_sh);
1461   if (!listen_session || listen_session->is_vep)
1462     return VPPCOM_EBADFD;
1463
1464   if (q_len == 0 || q_len == ~0)
1465     q_len = vcm->cfg.listen_queue_size;
1466
1467   listen_vpp_handle = listen_session->vpp_handle;
1468   if (listen_session->session_state == VCL_STATE_LISTEN)
1469     {
1470       VDBG (0, "session %u [0x%llx]: already in listen state!",
1471             listen_sh, listen_vpp_handle);
1472       return VPPCOM_OK;
1473     }
1474
1475   VDBG (0, "session %u: sending vpp listen request...", listen_sh);
1476
1477   /*
1478    * Send listen request to vpp and wait for reply
1479    */
1480   vcl_send_session_listen (wrk, listen_session);
1481   rv = vppcom_wait_for_session_state_change (listen_session->session_index,
1482                                              VCL_STATE_LISTEN,
1483                                              vcm->cfg.session_timeout);
1484
1485   if (PREDICT_FALSE (rv))
1486     {
1487       listen_session = vcl_session_get_w_handle (wrk, listen_sh);
1488       VDBG (0, "session %u [0x%llx]: listen failed! returning %d (%s)",
1489             listen_sh, listen_session->vpp_handle, rv,
1490             vppcom_retval_str (rv));
1491       return rv;
1492     }
1493
1494   return VPPCOM_OK;
1495 }
1496
1497 static int
1498 validate_args_session_accept_ (vcl_worker_t * wrk, vcl_session_t * ls)
1499 {
1500   if (ls->is_vep)
1501     {
1502       VDBG (0, "ERROR: cannot accept on epoll session %u!",
1503             ls->session_index);
1504       return VPPCOM_EBADFD;
1505     }
1506
1507   if ((ls->session_state != VCL_STATE_LISTEN)
1508       && (!vcl_session_is_connectable_listener (wrk, ls)))
1509     {
1510       VDBG (0,
1511             "ERROR: session [0x%llx]: not in listen state! state 0x%x"
1512             " (%s)", ls->vpp_handle, ls->session_state,
1513             vppcom_session_state_str (ls->session_state));
1514       return VPPCOM_EBADFD;
1515     }
1516   return VPPCOM_OK;
1517 }
1518
1519 int
1520 vppcom_unformat_proto (uint8_t * proto, char *proto_str)
1521 {
1522   if (!strcmp (proto_str, "TCP"))
1523     *proto = VPPCOM_PROTO_TCP;
1524   else if (!strcmp (proto_str, "tcp"))
1525     *proto = VPPCOM_PROTO_TCP;
1526   else if (!strcmp (proto_str, "UDP"))
1527     *proto = VPPCOM_PROTO_UDP;
1528   else if (!strcmp (proto_str, "udp"))
1529     *proto = VPPCOM_PROTO_UDP;
1530   else if (!strcmp (proto_str, "TLS"))
1531     *proto = VPPCOM_PROTO_TLS;
1532   else if (!strcmp (proto_str, "tls"))
1533     *proto = VPPCOM_PROTO_TLS;
1534   else if (!strcmp (proto_str, "QUIC"))
1535     *proto = VPPCOM_PROTO_QUIC;
1536   else if (!strcmp (proto_str, "quic"))
1537     *proto = VPPCOM_PROTO_QUIC;
1538   else
1539     return 1;
1540   return 0;
1541 }
1542
1543 int
1544 vppcom_session_accept (uint32_t listen_session_handle, vppcom_endpt_t * ep,
1545                        uint32_t flags)
1546 {
1547   u32 client_session_index = ~0, listen_session_index, accept_flags = 0;
1548   vcl_worker_t *wrk = vcl_worker_get_current ();
1549   session_accepted_msg_t accepted_msg;
1550   vcl_session_t *listen_session = 0;
1551   vcl_session_t *client_session = 0;
1552   vcl_session_msg_t *evt;
1553   svm_msg_q_msg_t msg;
1554   session_event_t *e;
1555   u8 is_nonblocking;
1556   int rv;
1557
1558   listen_session = vcl_session_get_w_handle (wrk, listen_session_handle);
1559   if (!listen_session)
1560     return VPPCOM_EBADFD;
1561
1562   listen_session_index = listen_session->session_index;
1563   if ((rv = validate_args_session_accept_ (wrk, listen_session)))
1564     return rv;
1565
1566   if (clib_fifo_elts (listen_session->accept_evts_fifo))
1567     {
1568       clib_fifo_sub2 (listen_session->accept_evts_fifo, evt);
1569       accept_flags = evt->flags;
1570       accepted_msg = evt->accepted_msg;
1571       goto handle;
1572     }
1573
1574   is_nonblocking = VCL_SESS_ATTR_TEST (listen_session->attr,
1575                                        VCL_SESS_ATTR_NONBLOCK);
1576   while (1)
1577     {
1578       if (svm_msg_q_is_empty (wrk->app_event_queue) && is_nonblocking)
1579         return VPPCOM_EAGAIN;
1580
1581       if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_WAIT, 0))
1582         return VPPCOM_EAGAIN;
1583
1584       e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
1585       if (e->event_type != SESSION_CTRL_EVT_ACCEPTED)
1586         {
1587           vcl_handle_mq_event (wrk, e);
1588           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1589           continue;
1590         }
1591       clib_memcpy_fast (&accepted_msg, e->data, sizeof (accepted_msg));
1592       svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1593       break;
1594     }
1595
1596 handle:
1597
1598   client_session_index = vcl_session_accepted_handler (wrk, &accepted_msg,
1599                                                        listen_session_index);
1600   if (client_session_index == VCL_INVALID_SESSION_INDEX)
1601     return VPPCOM_ECONNABORTED;
1602
1603   listen_session = vcl_session_get (wrk, listen_session_index);
1604   client_session = vcl_session_get (wrk, client_session_index);
1605
1606   if (flags & O_NONBLOCK)
1607     VCL_SESS_ATTR_SET (client_session->attr, VCL_SESS_ATTR_NONBLOCK);
1608
1609   VDBG (1, "listener %u [0x%llx]: Got a connect request! session %u [0x%llx],"
1610         " flags %d, is_nonblocking %u", listen_session->session_index,
1611         listen_session->vpp_handle, client_session_index,
1612         client_session->vpp_handle, flags,
1613         VCL_SESS_ATTR_TEST (client_session->attr, VCL_SESS_ATTR_NONBLOCK));
1614
1615   if (ep)
1616     {
1617       ep->is_ip4 = client_session->transport.is_ip4;
1618       ep->port = client_session->transport.rmt_port;
1619       if (client_session->transport.is_ip4)
1620         clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip4,
1621                           sizeof (ip4_address_t));
1622       else
1623         clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip6,
1624                           sizeof (ip6_address_t));
1625     }
1626
1627   VDBG (0, "listener %u [0x%llx] accepted %u [0x%llx] peer: %U:%u "
1628         "local: %U:%u", listen_session_handle, listen_session->vpp_handle,
1629         client_session_index, client_session->vpp_handle,
1630         format_ip46_address, &client_session->transport.rmt_ip,
1631         client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1632         clib_net_to_host_u16 (client_session->transport.rmt_port),
1633         format_ip46_address, &client_session->transport.lcl_ip,
1634         client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1635         clib_net_to_host_u16 (client_session->transport.lcl_port));
1636   vcl_evt (VCL_EVT_ACCEPT, client_session, listen_session,
1637            client_session_index);
1638
1639   /*
1640    * Session might have been closed already
1641    */
1642   if (accept_flags)
1643     {
1644       if (accept_flags & VCL_ACCEPTED_F_CLOSED)
1645         client_session->session_state = VCL_STATE_VPP_CLOSING;
1646       else if (accept_flags & VCL_ACCEPTED_F_RESET)
1647         client_session->session_state = VCL_STATE_DISCONNECT;
1648     }
1649   return vcl_session_handle (client_session);
1650 }
1651
1652 int
1653 vppcom_session_connect (uint32_t session_handle, vppcom_endpt_t * server_ep)
1654 {
1655   vcl_worker_t *wrk = vcl_worker_get_current ();
1656   vcl_session_t *session = 0;
1657   u32 session_index;
1658   int rv;
1659
1660   session = vcl_session_get_w_handle (wrk, session_handle);
1661   if (!session)
1662     return VPPCOM_EBADFD;
1663   session_index = session->session_index;
1664
1665   if (PREDICT_FALSE (session->is_vep))
1666     {
1667       VDBG (0, "ERROR: cannot connect epoll session %u!",
1668             session->session_index);
1669       return VPPCOM_EBADFD;
1670     }
1671
1672   if (PREDICT_FALSE (vcl_session_is_ready (session)))
1673     {
1674       VDBG (0, "session handle %u [0x%llx]: session already "
1675             "connected to %s %U port %d proto %s, state 0x%x (%s)",
1676             session_handle, session->vpp_handle,
1677             session->transport.is_ip4 ? "IPv4" : "IPv6", format_ip46_address,
1678             &session->transport.rmt_ip, session->transport.is_ip4 ?
1679             IP46_TYPE_IP4 : IP46_TYPE_IP6,
1680             clib_net_to_host_u16 (session->transport.rmt_port),
1681             vppcom_proto_str (session->session_type), session->session_state,
1682             vppcom_session_state_str (session->session_state));
1683       return VPPCOM_OK;
1684     }
1685
1686   /* Attempt to connect a connectionless listener */
1687   if (PREDICT_FALSE (session->session_state == VCL_STATE_LISTEN))
1688     {
1689       if (session->session_type != VPPCOM_PROTO_UDP)
1690         return VPPCOM_EINVAL;
1691       vcl_send_session_unlisten (wrk, session);
1692       session->session_state = VCL_STATE_CLOSED;
1693     }
1694
1695   session->transport.is_ip4 = server_ep->is_ip4;
1696   vcl_ip_copy_from_ep (&session->transport.rmt_ip, server_ep);
1697   session->transport.rmt_port = server_ep->port;
1698   session->parent_handle = VCL_INVALID_SESSION_HANDLE;
1699   session->flags |= VCL_SESSION_F_CONNECTED;
1700
1701   VDBG (0, "session handle %u (%s): connecting to peer %s %U "
1702         "port %d proto %s", session_handle,
1703         vppcom_session_state_str (session->session_state),
1704         session->transport.is_ip4 ? "IPv4" : "IPv6",
1705         format_ip46_address,
1706         &session->transport.rmt_ip, session->transport.is_ip4 ?
1707         IP46_TYPE_IP4 : IP46_TYPE_IP6,
1708         clib_net_to_host_u16 (session->transport.rmt_port),
1709         vppcom_proto_str (session->session_type));
1710
1711   vcl_send_session_connect (wrk, session);
1712
1713   if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK))
1714     {
1715       /* State set to STATE_UPDATED to ensure the session is not assumed
1716        * to be ready and to also allow the app to close it prior to vpp's
1717        * connected reply. */
1718       session->session_state = VCL_STATE_UPDATED;
1719       return VPPCOM_EINPROGRESS;
1720     }
1721
1722   /*
1723    * Wait for reply from vpp if blocking
1724    */
1725   rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
1726                                              vcm->cfg.session_timeout);
1727
1728   session = vcl_session_get (wrk, session_index);
1729   VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1730         session->vpp_handle, rv ? "failed" : "succeeded");
1731
1732   return rv;
1733 }
1734
1735 int
1736 vppcom_session_stream_connect (uint32_t session_handle,
1737                                uint32_t parent_session_handle)
1738 {
1739   vcl_worker_t *wrk = vcl_worker_get_current ();
1740   vcl_session_t *session, *parent_session;
1741   u32 session_index, parent_session_index;
1742   int rv;
1743
1744   session = vcl_session_get_w_handle (wrk, session_handle);
1745   if (!session)
1746     return VPPCOM_EBADFD;
1747   parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1748   if (!parent_session)
1749     return VPPCOM_EBADFD;
1750
1751   session_index = session->session_index;
1752   parent_session_index = parent_session->session_index;
1753   if (PREDICT_FALSE (session->is_vep))
1754     {
1755       VDBG (0, "ERROR: cannot connect epoll session %u!",
1756             session->session_index);
1757       return VPPCOM_EBADFD;
1758     }
1759
1760   if (PREDICT_FALSE (vcl_session_is_ready (session)))
1761     {
1762       VDBG (0, "session handle %u [0x%llx]: session already "
1763             "connected to session %u [0x%llx] proto %s, state 0x%x (%s)",
1764             session_handle, session->vpp_handle,
1765             parent_session_handle, parent_session->vpp_handle,
1766             vppcom_proto_str (session->session_type), session->session_state,
1767             vppcom_session_state_str (session->session_state));
1768       return VPPCOM_OK;
1769     }
1770
1771   /* Connect to quic session specifics */
1772   session->transport.is_ip4 = parent_session->transport.is_ip4;
1773   session->transport.rmt_ip.ip4.as_u32 = (uint32_t) 1;
1774   session->transport.rmt_port = 0;
1775   session->parent_handle = parent_session->vpp_handle;
1776
1777   VDBG (0, "session handle %u: connecting to session %u [0x%llx]",
1778         session_handle, parent_session_handle, parent_session->vpp_handle);
1779
1780   /*
1781    * Send connect request and wait for reply from vpp
1782    */
1783   vcl_send_session_connect (wrk, session);
1784   rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
1785                                              vcm->cfg.session_timeout);
1786
1787   session->listener_index = parent_session_index;
1788   parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1789   if (parent_session)
1790     parent_session->n_accepted_sessions++;
1791
1792   session = vcl_session_get (wrk, session_index);
1793   VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1794         session->vpp_handle, rv ? "failed" : "succeeded");
1795
1796   return rv;
1797 }
1798
1799 static u8
1800 vcl_is_rx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
1801 {
1802   return (e->event_type == SESSION_IO_EVT_RX && e->session_index == sid);
1803 }
1804
1805 static inline int
1806 vppcom_session_read_internal (uint32_t session_handle, void *buf, int n,
1807                               u8 peek)
1808 {
1809   vcl_worker_t *wrk = vcl_worker_get_current ();
1810   int rv, n_read = 0, is_nonblocking;
1811   vcl_session_t *s = 0;
1812   svm_fifo_t *rx_fifo;
1813   svm_msg_q_msg_t msg;
1814   session_event_t *e;
1815   svm_msg_q_t *mq;
1816   u8 is_ct;
1817
1818   if (PREDICT_FALSE (!buf))
1819     return VPPCOM_EINVAL;
1820
1821   s = vcl_session_get_w_handle (wrk, session_handle);
1822   if (PREDICT_FALSE (!s || s->is_vep))
1823     return VPPCOM_EBADFD;
1824
1825   if (PREDICT_FALSE (!vcl_session_is_open (s)))
1826     {
1827       VDBG (0, "session %u[0x%llx] is not open! state 0x%x (%s)",
1828             s->session_index, s->vpp_handle, s->session_state,
1829             vppcom_session_state_str (s->session_state));
1830       return vcl_session_closed_error (s);
1831     }
1832
1833   is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
1834   is_ct = vcl_session_is_ct (s);
1835   mq = wrk->app_event_queue;
1836   rx_fifo = is_ct ? s->ct_rx_fifo : s->rx_fifo;
1837   s->has_rx_evt = 0;
1838
1839   if (svm_fifo_is_empty_cons (rx_fifo))
1840     {
1841       if (is_nonblocking)
1842         {
1843           if (vcl_session_is_closing (s))
1844             return vcl_session_closing_error (s);
1845           svm_fifo_unset_event (s->rx_fifo);
1846           return VPPCOM_EWOULDBLOCK;
1847         }
1848       while (svm_fifo_is_empty_cons (rx_fifo))
1849         {
1850           if (vcl_session_is_closing (s))
1851             return vcl_session_closing_error (s);
1852
1853           svm_fifo_unset_event (s->rx_fifo);
1854           svm_msg_q_lock (mq);
1855           if (svm_msg_q_is_empty (mq))
1856             svm_msg_q_wait (mq);
1857
1858           svm_msg_q_sub_w_lock (mq, &msg);
1859           e = svm_msg_q_msg_data (mq, &msg);
1860           svm_msg_q_unlock (mq);
1861           if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
1862             vcl_handle_mq_event (wrk, e);
1863           svm_msg_q_free_msg (mq, &msg);
1864         }
1865     }
1866
1867 read_again:
1868
1869   if (s->is_dgram)
1870     rv = app_recv_dgram_raw (rx_fifo, buf, n, &s->transport, 0, peek);
1871   else
1872     rv = app_recv_stream_raw (rx_fifo, buf, n, 0, peek);
1873
1874   ASSERT (rv >= 0);
1875   n_read += rv;
1876
1877   if (svm_fifo_is_empty_cons (rx_fifo))
1878     {
1879       svm_fifo_unset_event (s->rx_fifo);
1880       if (!svm_fifo_is_empty_cons (rx_fifo)
1881           && svm_fifo_set_event (s->rx_fifo) && is_nonblocking)
1882         {
1883           vec_add2 (wrk->unhandled_evts_vector, e, 1);
1884           e->event_type = SESSION_IO_EVT_RX;
1885           e->session_index = s->session_index;
1886         }
1887     }
1888   else if (PREDICT_FALSE (rv < n))
1889     {
1890       /* More data enqueued while reading. Try to drain it
1891        * or fill the buffer */
1892       buf += rv;
1893       n -= rv;
1894       goto read_again;
1895     }
1896
1897   if (PREDICT_FALSE (svm_fifo_needs_deq_ntf (rx_fifo, n_read)))
1898     {
1899       svm_fifo_clear_deq_ntf (rx_fifo);
1900       app_send_io_evt_to_vpp (s->vpp_evt_q, s->rx_fifo->master_session_index,
1901                               SESSION_IO_EVT_RX, SVM_Q_WAIT);
1902     }
1903
1904   VDBG (2, "session %u[0x%llx]: read %d bytes from (%p)", s->session_index,
1905         s->vpp_handle, n_read, rx_fifo);
1906
1907   return n_read;
1908 }
1909
1910 int
1911 vppcom_session_read (uint32_t session_handle, void *buf, size_t n)
1912 {
1913   return (vppcom_session_read_internal (session_handle, buf, n, 0));
1914 }
1915
1916 static int
1917 vppcom_session_peek (uint32_t session_handle, void *buf, int n)
1918 {
1919   return (vppcom_session_read_internal (session_handle, buf, n, 1));
1920 }
1921
1922 int
1923 vppcom_session_read_segments (uint32_t session_handle,
1924                               vppcom_data_segment_t * ds, uint32_t n_segments,
1925                               uint32_t max_bytes)
1926 {
1927   vcl_worker_t *wrk = vcl_worker_get_current ();
1928   int n_read = 0, is_nonblocking;
1929   vcl_session_t *s = 0;
1930   svm_fifo_t *rx_fifo;
1931   svm_msg_q_msg_t msg;
1932   session_event_t *e;
1933   svm_msg_q_t *mq;
1934   u8 is_ct;
1935
1936   s = vcl_session_get_w_handle (wrk, session_handle);
1937   if (PREDICT_FALSE (!s || s->is_vep))
1938     return VPPCOM_EBADFD;
1939
1940   if (PREDICT_FALSE (!vcl_session_is_open (s)))
1941     return vcl_session_closed_error (s);
1942
1943   is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
1944   is_ct = vcl_session_is_ct (s);
1945   mq = is_ct ? s->our_evt_q : wrk->app_event_queue;
1946   rx_fifo = s->rx_fifo;
1947   s->has_rx_evt = 0;
1948
1949   if (is_ct)
1950     svm_fifo_unset_event (s->rx_fifo);
1951
1952   if (svm_fifo_is_empty_cons (rx_fifo))
1953     {
1954       if (is_nonblocking)
1955         {
1956           svm_fifo_unset_event (rx_fifo);
1957           return VPPCOM_EWOULDBLOCK;
1958         }
1959       while (svm_fifo_is_empty_cons (rx_fifo))
1960         {
1961           if (vcl_session_is_closing (s))
1962             return vcl_session_closing_error (s);
1963
1964           svm_fifo_unset_event (rx_fifo);
1965           svm_msg_q_lock (mq);
1966           if (svm_msg_q_is_empty (mq))
1967             svm_msg_q_wait (mq);
1968
1969           svm_msg_q_sub_w_lock (mq, &msg);
1970           e = svm_msg_q_msg_data (mq, &msg);
1971           svm_msg_q_unlock (mq);
1972           if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
1973             vcl_handle_mq_event (wrk, e);
1974           svm_msg_q_free_msg (mq, &msg);
1975         }
1976     }
1977
1978   n_read = svm_fifo_segments (rx_fifo, s->rx_bytes_pending,
1979                               (svm_fifo_seg_t *) ds, n_segments, max_bytes);
1980   if (n_read < 0)
1981     return VPPCOM_EAGAIN;
1982
1983   if (svm_fifo_max_dequeue_cons (rx_fifo) == n_read)
1984     {
1985       svm_fifo_unset_event (s->rx_fifo);
1986       if (svm_fifo_max_dequeue_cons (rx_fifo) != n_read
1987           && svm_fifo_set_event (s->rx_fifo)
1988           && VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK))
1989         {
1990           session_event_t *e;
1991           vec_add2 (wrk->unhandled_evts_vector, e, 1);
1992           e->event_type = SESSION_IO_EVT_RX;
1993           e->session_index = s->session_index;
1994         }
1995     }
1996
1997   s->rx_bytes_pending += n_read;
1998   return n_read;
1999 }
2000
2001 void
2002 vppcom_session_free_segments (uint32_t session_handle, uint32_t n_bytes)
2003 {
2004   vcl_worker_t *wrk = vcl_worker_get_current ();
2005   vcl_session_t *s;
2006
2007   s = vcl_session_get_w_handle (wrk, session_handle);
2008   if (PREDICT_FALSE (!s || s->is_vep))
2009     return;
2010
2011   svm_fifo_dequeue_drop (s->rx_fifo, n_bytes);
2012
2013   ASSERT (s->rx_bytes_pending < n_bytes);
2014   s->rx_bytes_pending -= n_bytes;
2015 }
2016
2017 static u8
2018 vcl_is_tx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
2019 {
2020   return (e->event_type == SESSION_IO_EVT_TX && e->session_index == sid);
2021 }
2022
2023 always_inline u8
2024 vcl_fifo_is_writeable (svm_fifo_t * f, u32 len, u8 is_dgram)
2025 {
2026   u32 max_enq = svm_fifo_max_enqueue_prod (f);
2027   if (is_dgram)
2028     return max_enq >= (sizeof (session_dgram_hdr_t) + len);
2029   else
2030     return max_enq > 0;
2031 }
2032
2033 always_inline int
2034 vppcom_session_write_inline (vcl_worker_t * wrk, vcl_session_t * s, void *buf,
2035                              size_t n, u8 is_flush, u8 is_dgram)
2036 {
2037   int n_write, is_nonblocking;
2038   session_evt_type_t et;
2039   svm_msg_q_msg_t msg;
2040   svm_fifo_t *tx_fifo;
2041   session_event_t *e;
2042   svm_msg_q_t *mq;
2043   u8 is_ct;
2044
2045   if (PREDICT_FALSE (!buf || n == 0))
2046     return VPPCOM_EINVAL;
2047
2048   if (PREDICT_FALSE (s->is_vep))
2049     {
2050       VDBG (0, "ERROR: session %u [0x%llx]: cannot write to an epoll"
2051             " session!", s->session_index, s->vpp_handle);
2052       return VPPCOM_EBADFD;
2053     }
2054
2055   if (PREDICT_FALSE (!vcl_session_is_open (s)))
2056     {
2057       VDBG (1, "session %u [0x%llx]: is not open! state 0x%x (%s)",
2058             s->session_index, s->vpp_handle, s->session_state,
2059             vppcom_session_state_str (s->session_state));
2060       return vcl_session_closed_error (s);;
2061     }
2062
2063   is_ct = vcl_session_is_ct (s);
2064   tx_fifo = is_ct ? s->ct_tx_fifo : s->tx_fifo;
2065   is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
2066
2067   mq = wrk->app_event_queue;
2068   if (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2069     {
2070       if (is_nonblocking)
2071         {
2072           return VPPCOM_EWOULDBLOCK;
2073         }
2074       while (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2075         {
2076           svm_fifo_add_want_deq_ntf (tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
2077           if (vcl_session_is_closing (s))
2078             return vcl_session_closing_error (s);
2079           svm_msg_q_lock (mq);
2080           if (svm_msg_q_is_empty (mq))
2081             svm_msg_q_wait (mq);
2082
2083           svm_msg_q_sub_w_lock (mq, &msg);
2084           e = svm_msg_q_msg_data (mq, &msg);
2085           svm_msg_q_unlock (mq);
2086
2087           if (!vcl_is_tx_evt_for_session (e, s->session_index, is_ct))
2088             vcl_handle_mq_event (wrk, e);
2089           svm_msg_q_free_msg (mq, &msg);
2090         }
2091     }
2092
2093   et = SESSION_IO_EVT_TX;
2094   if (is_flush && !is_ct)
2095     et = SESSION_IO_EVT_TX_FLUSH;
2096
2097   if (is_dgram)
2098     n_write = app_send_dgram_raw (tx_fifo, &s->transport,
2099                                   s->vpp_evt_q, buf, n, et,
2100                                   0 /* do_evt */ , SVM_Q_WAIT);
2101   else
2102     n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
2103                                    0 /* do_evt */ , SVM_Q_WAIT);
2104
2105   if (svm_fifo_set_event (s->tx_fifo))
2106     app_send_io_evt_to_vpp (s->vpp_evt_q, s->tx_fifo->master_session_index,
2107                             et, SVM_Q_WAIT);
2108
2109   /* The underlying fifo segment can run out of memory */
2110   if (PREDICT_FALSE (n_write < 0))
2111     return VPPCOM_EAGAIN;
2112
2113   VDBG (2, "session %u [0x%llx]: wrote %d bytes", s->session_index,
2114         s->vpp_handle, n_write);
2115
2116   return n_write;
2117 }
2118
2119 int
2120 vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
2121 {
2122   vcl_worker_t *wrk = vcl_worker_get_current ();
2123   vcl_session_t *s;
2124
2125   s = vcl_session_get_w_handle (wrk, session_handle);
2126   if (PREDICT_FALSE (!s))
2127     return VPPCOM_EBADFD;
2128
2129   return vppcom_session_write_inline (wrk, s, buf, n,
2130                                       0 /* is_flush */ , s->is_dgram ? 1 : 0);
2131 }
2132
2133 int
2134 vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
2135 {
2136   vcl_worker_t *wrk = vcl_worker_get_current ();
2137   vcl_session_t *s;
2138
2139   s = vcl_session_get_w_handle (wrk, session_handle);
2140   if (PREDICT_FALSE (!s))
2141     return VPPCOM_EBADFD;
2142
2143   return vppcom_session_write_inline (wrk, s, buf, n,
2144                                       1 /* is_flush */ , s->is_dgram ? 1 : 0);
2145 }
2146
2147 #define vcl_fifo_rx_evt_valid_or_break(_s)                              \
2148 if (PREDICT_FALSE (!_s->rx_fifo))                                       \
2149   break;                                                                \
2150 if (PREDICT_FALSE (svm_fifo_is_empty (_s->rx_fifo)))                    \
2151   {                                                                     \
2152     if (!vcl_session_is_ct (_s))                                        \
2153       {                                                                 \
2154         svm_fifo_unset_event (_s->rx_fifo);                             \
2155         if (svm_fifo_is_empty (_s->rx_fifo))                            \
2156           break;                                                        \
2157       }                                                                 \
2158     else if (svm_fifo_is_empty (_s->ct_rx_fifo))                        \
2159       {                                                                 \
2160         svm_fifo_unset_event (_s->rx_fifo); /* rx evts on actual fifo*/ \
2161         if (svm_fifo_is_empty (_s->ct_rx_fifo))                         \
2162           break;                                                        \
2163       }                                                                 \
2164   }                                                                     \
2165
2166 static void
2167 vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2168                             unsigned long n_bits, unsigned long *read_map,
2169                             unsigned long *write_map,
2170                             unsigned long *except_map, u32 * bits_set)
2171 {
2172   session_disconnected_msg_t *disconnected_msg;
2173   session_connected_msg_t *connected_msg;
2174   vcl_session_t *session;
2175   u32 sid;
2176
2177   switch (e->event_type)
2178     {
2179     case SESSION_IO_EVT_RX:
2180       sid = e->session_index;
2181       session = vcl_session_get (wrk, sid);
2182       if (!session || !vcl_session_is_open (session))
2183         break;
2184       vcl_fifo_rx_evt_valid_or_break (session);
2185       if (sid < n_bits && read_map)
2186         {
2187           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2188           *bits_set += 1;
2189         }
2190       break;
2191     case SESSION_IO_EVT_TX:
2192       sid = e->session_index;
2193       session = vcl_session_get (wrk, sid);
2194       if (!session || !vcl_session_is_open (session))
2195         break;
2196       if (sid < n_bits && write_map)
2197         {
2198           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2199           *bits_set += 1;
2200         }
2201       break;
2202     case SESSION_CTRL_EVT_ACCEPTED:
2203       session = vcl_session_accepted (wrk,
2204                                       (session_accepted_msg_t *) e->data);
2205       if (!session)
2206         break;
2207       sid = session->session_index;
2208       if (sid < n_bits && read_map)
2209         {
2210           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2211           *bits_set += 1;
2212         }
2213       break;
2214     case SESSION_CTRL_EVT_CONNECTED:
2215       connected_msg = (session_connected_msg_t *) e->data;
2216       sid = vcl_session_connected_handler (wrk, connected_msg);
2217       if (sid == VCL_INVALID_SESSION_INDEX)
2218         break;
2219       if (sid < n_bits && write_map)
2220         {
2221           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2222           *bits_set += 1;
2223         }
2224       break;
2225     case SESSION_CTRL_EVT_DISCONNECTED:
2226       disconnected_msg = (session_disconnected_msg_t *) e->data;
2227       session = vcl_session_disconnected_handler (wrk, disconnected_msg);
2228       if (!session)
2229         break;
2230       sid = session->session_index;
2231       if (sid < n_bits && except_map)
2232         {
2233           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2234           *bits_set += 1;
2235         }
2236       break;
2237     case SESSION_CTRL_EVT_RESET:
2238       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2239       if (sid < n_bits && except_map)
2240         {
2241           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2242           *bits_set += 1;
2243         }
2244       break;
2245     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2246       vcl_session_unlisten_reply_handler (wrk, e->data);
2247       break;
2248     case SESSION_CTRL_EVT_MIGRATED:
2249       vcl_session_migrated_handler (wrk, e->data);
2250       break;
2251     case SESSION_CTRL_EVT_CLEANUP:
2252       vcl_session_cleanup_handler (wrk, e->data);
2253       break;
2254     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2255       vcl_session_worker_update_reply_handler (wrk, e->data);
2256       break;
2257     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2258       vcl_session_req_worker_update_handler (wrk, e->data);
2259       break;
2260     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2261       vcl_session_app_add_segment_handler (wrk, e->data);
2262       break;
2263     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2264       vcl_session_app_del_segment_handler (wrk, e->data);
2265       break;
2266     case SESSION_CTRL_EVT_APP_WRK_RPC:
2267       vcl_worker_rpc_handler (wrk, e->data);
2268       break;
2269     default:
2270       clib_warning ("unhandled: %u", e->event_type);
2271       break;
2272     }
2273 }
2274
2275 static int
2276 vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2277                       unsigned long n_bits, unsigned long *read_map,
2278                       unsigned long *write_map, unsigned long *except_map,
2279                       double time_to_wait, u32 * bits_set)
2280 {
2281   svm_msg_q_msg_t *msg;
2282   session_event_t *e;
2283   u32 i;
2284
2285   svm_msg_q_lock (mq);
2286   if (svm_msg_q_is_empty (mq))
2287     {
2288       if (*bits_set)
2289         {
2290           svm_msg_q_unlock (mq);
2291           return 0;
2292         }
2293
2294       if (!time_to_wait)
2295         {
2296           svm_msg_q_unlock (mq);
2297           return 0;
2298         }
2299       else if (time_to_wait < 0)
2300         {
2301           svm_msg_q_wait (mq);
2302         }
2303       else
2304         {
2305           if (svm_msg_q_timedwait (mq, time_to_wait))
2306             {
2307               svm_msg_q_unlock (mq);
2308               return 0;
2309             }
2310         }
2311     }
2312   vcl_mq_dequeue_batch (wrk, mq, ~0);
2313   svm_msg_q_unlock (mq);
2314
2315   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
2316     {
2317       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
2318       e = svm_msg_q_msg_data (mq, msg);
2319       vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
2320                                   except_map, bits_set);
2321       svm_msg_q_free_msg (mq, msg);
2322     }
2323   vec_reset_length (wrk->mq_msg_vector);
2324   vcl_handle_pending_wrk_updates (wrk);
2325   return *bits_set;
2326 }
2327
2328 static int
2329 vppcom_select_condvar (vcl_worker_t * wrk, int n_bits,
2330                        vcl_si_set * read_map, vcl_si_set * write_map,
2331                        vcl_si_set * except_map, double time_to_wait,
2332                        u32 * bits_set)
2333 {
2334   double wait = 0, start = 0;
2335
2336   if (!*bits_set)
2337     {
2338       wait = time_to_wait;
2339       start = clib_time_now (&wrk->clib_time);
2340     }
2341
2342   do
2343     {
2344       vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
2345                             write_map, except_map, wait, bits_set);
2346       if (*bits_set)
2347         return *bits_set;
2348       if (wait == -1)
2349         continue;
2350
2351       wait = wait - (clib_time_now (&wrk->clib_time) - start);
2352     }
2353   while (wait > 0);
2354
2355   return 0;
2356 }
2357
2358 static int
2359 vppcom_select_eventfd (vcl_worker_t * wrk, int n_bits,
2360                        vcl_si_set * read_map, vcl_si_set * write_map,
2361                        vcl_si_set * except_map, double time_to_wait,
2362                        u32 * bits_set)
2363 {
2364   vcl_mq_evt_conn_t *mqc;
2365   int __clib_unused n_read;
2366   int n_mq_evts, i;
2367   u64 buf;
2368
2369   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
2370   n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
2371                           vec_len (wrk->mq_events), time_to_wait);
2372   for (i = 0; i < n_mq_evts; i++)
2373     {
2374       mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
2375       n_read = read (mqc->mq_fd, &buf, sizeof (buf));
2376       vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
2377                             except_map, 0, bits_set);
2378     }
2379
2380   return (n_mq_evts > 0 ? (int) *bits_set : 0);
2381 }
2382
2383 int
2384 vppcom_select (int n_bits, vcl_si_set * read_map, vcl_si_set * write_map,
2385                vcl_si_set * except_map, double time_to_wait)
2386 {
2387   u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
2388   vcl_worker_t *wrk = vcl_worker_get_current ();
2389   vcl_session_t *session = 0;
2390   int i;
2391
2392   if (n_bits && read_map)
2393     {
2394       clib_bitmap_validate (wrk->rd_bitmap, minbits);
2395       clib_memcpy_fast (wrk->rd_bitmap, read_map,
2396                         vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2397       memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2398     }
2399   if (n_bits && write_map)
2400     {
2401       clib_bitmap_validate (wrk->wr_bitmap, minbits);
2402       clib_memcpy_fast (wrk->wr_bitmap, write_map,
2403                         vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2404       memset (write_map, 0, vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2405     }
2406   if (n_bits && except_map)
2407     {
2408       clib_bitmap_validate (wrk->ex_bitmap, minbits);
2409       clib_memcpy_fast (wrk->ex_bitmap, except_map,
2410                         vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2411       memset (except_map, 0, vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2412     }
2413
2414   if (!n_bits)
2415     return 0;
2416
2417   if (!write_map)
2418     goto check_rd;
2419
2420   /* *INDENT-OFF* */
2421   clib_bitmap_foreach (sid, wrk->wr_bitmap, ({
2422     if (!(session = vcl_session_get (wrk, sid)))
2423       {
2424         clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
2425         bits_set++;
2426         continue;
2427       }
2428
2429     if (vcl_session_write_ready (session))
2430       {
2431         clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
2432         bits_set++;
2433       }
2434     else
2435       svm_fifo_add_want_deq_ntf (session->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
2436   }));
2437
2438 check_rd:
2439   if (!read_map)
2440     goto check_mq;
2441
2442   clib_bitmap_foreach (sid, wrk->rd_bitmap, ({
2443     if (!(session = vcl_session_get (wrk, sid)))
2444       {
2445         clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
2446         bits_set++;
2447         continue;
2448       }
2449
2450     if (vcl_session_read_ready (session))
2451       {
2452         clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
2453         bits_set++;
2454       }
2455   }));
2456   /* *INDENT-ON* */
2457
2458 check_mq:
2459
2460   for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
2461     {
2462       vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
2463                                   read_map, write_map, except_map, &bits_set);
2464     }
2465   vec_reset_length (wrk->unhandled_evts_vector);
2466
2467   if (vcm->cfg.use_mq_eventfd)
2468     vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
2469                            time_to_wait, &bits_set);
2470   else
2471     vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
2472                            time_to_wait, &bits_set);
2473
2474   return (bits_set);
2475 }
2476
2477 static inline void
2478 vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_handle)
2479 {
2480   vcl_session_t *session;
2481   vppcom_epoll_t *vep;
2482   u32 sh = vep_handle;
2483
2484   if (VPPCOM_DEBUG <= 2)
2485     return;
2486
2487   session = vcl_session_get_w_handle (wrk, vep_handle);
2488   if (PREDICT_FALSE (!session))
2489     {
2490       VDBG (0, "ERROR: Invalid vep_sh (%u)!", vep_handle);
2491       goto done;
2492     }
2493   if (PREDICT_FALSE (!session->is_vep))
2494     {
2495       VDBG (0, "ERROR: vep_sh (%u) is not a vep!", vep_handle);
2496       goto done;
2497     }
2498   vep = &session->vep;
2499   VDBG (0, "vep_sh (%u): Dumping epoll chain\n"
2500         "{\n"
2501         "   is_vep         = %u\n"
2502         "   is_vep_session = %u\n"
2503         "   next_sh        = 0x%x (%u)\n"
2504         "}\n", vep_handle, session->is_vep, session->is_vep_session,
2505         vep->next_sh, vep->next_sh);
2506
2507   for (sh = vep->next_sh; sh != ~0; sh = vep->next_sh)
2508     {
2509       session = vcl_session_get_w_handle (wrk, sh);
2510       if (PREDICT_FALSE (!session))
2511         {
2512           VDBG (0, "ERROR: Invalid sh (%u)!", sh);
2513           goto done;
2514         }
2515       if (PREDICT_FALSE (session->is_vep))
2516         {
2517           VDBG (0, "ERROR: sh (%u) is a vep!", vep_handle);
2518         }
2519       else if (PREDICT_FALSE (!session->is_vep_session))
2520         {
2521           VDBG (0, "ERROR: sh (%u) is not a vep session handle!", sh);
2522           goto done;
2523         }
2524       vep = &session->vep;
2525       if (PREDICT_FALSE (vep->vep_sh != vep_handle))
2526         VDBG (0, "ERROR: session (%u) vep_sh (%u) != vep_sh (%u)!",
2527               sh, session->vep.vep_sh, vep_handle);
2528       if (session->is_vep_session)
2529         {
2530           VDBG (0, "vep_sh[%u]: sh 0x%x (%u)\n"
2531                 "{\n"
2532                 "   next_sh        = 0x%x (%u)\n"
2533                 "   prev_sh        = 0x%x (%u)\n"
2534                 "   vep_sh         = 0x%x (%u)\n"
2535                 "   ev.events      = 0x%x\n"
2536                 "   ev.data.u64    = 0x%llx\n"
2537                 "   et_mask        = 0x%x\n"
2538                 "}\n",
2539                 vep_handle, sh, sh, vep->next_sh, vep->next_sh, vep->prev_sh,
2540                 vep->prev_sh, vep->vep_sh, vep->vep_sh, vep->ev.events,
2541                 vep->ev.data.u64, vep->et_mask);
2542         }
2543     }
2544
2545 done:
2546   VDBG (0, "vep_sh (%u): Dump complete!\n", vep_handle);
2547 }
2548
2549 int
2550 vppcom_epoll_create (void)
2551 {
2552   vcl_worker_t *wrk = vcl_worker_get_current ();
2553   vcl_session_t *vep_session;
2554
2555   vep_session = vcl_session_alloc (wrk);
2556
2557   vep_session->is_vep = 1;
2558   vep_session->vep.vep_sh = ~0;
2559   vep_session->vep.next_sh = ~0;
2560   vep_session->vep.prev_sh = ~0;
2561   vep_session->vpp_handle = ~0;
2562
2563   vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
2564   VDBG (0, "Created vep_idx %u", vep_session->session_index);
2565
2566   return vcl_session_handle (vep_session);
2567 }
2568
2569 int
2570 vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
2571                   struct epoll_event *event)
2572 {
2573   vcl_worker_t *wrk = vcl_worker_get_current ();
2574   vcl_session_t *vep_session;
2575   int rv = VPPCOM_OK;
2576   vcl_session_t *s;
2577   svm_fifo_t *txf;
2578
2579   if (vep_handle == session_handle)
2580     {
2581       VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
2582       return VPPCOM_EINVAL;
2583     }
2584
2585   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
2586   if (PREDICT_FALSE (!vep_session))
2587     {
2588       VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
2589       return VPPCOM_EBADFD;
2590     }
2591   if (PREDICT_FALSE (!vep_session->is_vep))
2592     {
2593       VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
2594       return VPPCOM_EINVAL;
2595     }
2596
2597   ASSERT (vep_session->vep.vep_sh == ~0);
2598   ASSERT (vep_session->vep.prev_sh == ~0);
2599
2600   s = vcl_session_get_w_handle (wrk, session_handle);
2601   if (PREDICT_FALSE (!s))
2602     {
2603       VDBG (0, "Invalid session_handle (%u)!", session_handle);
2604       return VPPCOM_EBADFD;
2605     }
2606   if (PREDICT_FALSE (s->is_vep))
2607     {
2608       VDBG (0, "session_handle (%u) is a vep!", vep_handle);
2609       return VPPCOM_EINVAL;
2610     }
2611
2612   switch (op)
2613     {
2614     case EPOLL_CTL_ADD:
2615       if (PREDICT_FALSE (!event))
2616         {
2617           VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
2618           return VPPCOM_EINVAL;
2619         }
2620       if (vep_session->vep.next_sh != ~0)
2621         {
2622           vcl_session_t *next_session;
2623           next_session = vcl_session_get_w_handle (wrk,
2624                                                    vep_session->vep.next_sh);
2625           if (PREDICT_FALSE (!next_session))
2626             {
2627               VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
2628                     "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
2629               return VPPCOM_EBADFD;
2630             }
2631           ASSERT (next_session->vep.prev_sh == vep_handle);
2632           next_session->vep.prev_sh = session_handle;
2633         }
2634       s->vep.next_sh = vep_session->vep.next_sh;
2635       s->vep.prev_sh = vep_handle;
2636       s->vep.vep_sh = vep_handle;
2637       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2638       s->vep.ev = *event;
2639       s->is_vep = 0;
2640       s->is_vep_session = 1;
2641       vep_session->vep.next_sh = session_handle;
2642
2643       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2644       if (txf && (event->events & EPOLLOUT))
2645         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2646
2647       /* Generate EPOLLOUT if tx fifo not full */
2648       if ((event->events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2649         {
2650           session_event_t e = { 0 };
2651           e.event_type = SESSION_IO_EVT_TX;
2652           e.session_index = s->session_index;
2653           vec_add1 (wrk->unhandled_evts_vector, e);
2654         }
2655       /* Generate EPOLLIN if rx fifo has data */
2656       if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
2657         {
2658           session_event_t e = { 0 };
2659           e.event_type = SESSION_IO_EVT_RX;
2660           e.session_index = s->session_index;
2661           vec_add1 (wrk->unhandled_evts_vector, e);
2662         }
2663       VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2664             vep_handle, session_handle, event->events, event->data.u64);
2665       vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
2666       break;
2667
2668     case EPOLL_CTL_MOD:
2669       if (PREDICT_FALSE (!event))
2670         {
2671           VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
2672           rv = VPPCOM_EINVAL;
2673           goto done;
2674         }
2675       else if (PREDICT_FALSE (!s->is_vep_session))
2676         {
2677           VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
2678           rv = VPPCOM_EINVAL;
2679           goto done;
2680         }
2681       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2682         {
2683           VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
2684                 session_handle, s->vep.vep_sh, vep_handle);
2685           rv = VPPCOM_EINVAL;
2686           goto done;
2687         }
2688
2689       /* Generate EPOLLOUT when tx_fifo/ct_tx_fifo not full */
2690       if ((event->events & EPOLLOUT) &&
2691           !(s->vep.ev.events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2692         {
2693           session_event_t e = { 0 };
2694           e.event_type = SESSION_IO_EVT_TX;
2695           e.session_index = s->session_index;
2696           vec_add1 (wrk->unhandled_evts_vector, e);
2697         }
2698       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2699       s->vep.ev = *event;
2700       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2701       if (event->events & EPOLLOUT)
2702         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2703       else
2704         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2705       VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2706             vep_handle, session_handle, event->events, event->data.u64);
2707       break;
2708
2709     case EPOLL_CTL_DEL:
2710       if (PREDICT_FALSE (!s->is_vep_session))
2711         {
2712           VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
2713           rv = VPPCOM_EINVAL;
2714           goto done;
2715         }
2716       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2717         {
2718           VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
2719                 session_handle, s->vep.vep_sh, vep_handle);
2720           rv = VPPCOM_EINVAL;
2721           goto done;
2722         }
2723
2724       if (s->vep.prev_sh == vep_handle)
2725         vep_session->vep.next_sh = s->vep.next_sh;
2726       else
2727         {
2728           vcl_session_t *prev_session;
2729           prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
2730           if (PREDICT_FALSE (!prev_session))
2731             {
2732               VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
2733                     s->vep.prev_sh, session_handle);
2734               return VPPCOM_EBADFD;
2735             }
2736           ASSERT (prev_session->vep.next_sh == session_handle);
2737           prev_session->vep.next_sh = s->vep.next_sh;
2738         }
2739       if (s->vep.next_sh != ~0)
2740         {
2741           vcl_session_t *next_session;
2742           next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
2743           if (PREDICT_FALSE (!next_session))
2744             {
2745               VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
2746                     s->vep.next_sh, session_handle);
2747               return VPPCOM_EBADFD;
2748             }
2749           ASSERT (next_session->vep.prev_sh == session_handle);
2750           next_session->vep.prev_sh = s->vep.prev_sh;
2751         }
2752
2753       memset (&s->vep, 0, sizeof (s->vep));
2754       s->vep.next_sh = ~0;
2755       s->vep.prev_sh = ~0;
2756       s->vep.vep_sh = ~0;
2757       s->is_vep_session = 0;
2758
2759       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2760       if (txf)
2761         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2762
2763       VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
2764             session_handle);
2765       vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
2766       break;
2767
2768     default:
2769       VDBG (0, "Invalid operation (%d)!", op);
2770       rv = VPPCOM_EINVAL;
2771     }
2772
2773   vep_verify_epoll_chain (wrk, vep_handle);
2774
2775 done:
2776   return rv;
2777 }
2778
2779 static inline void
2780 vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2781                                 struct epoll_event *events, u32 * num_ev)
2782 {
2783   session_disconnected_msg_t *disconnected_msg;
2784   session_connected_msg_t *connected_msg;
2785   u32 sid = ~0, session_events;
2786   u64 session_evt_data = ~0;
2787   vcl_session_t *session;
2788   u8 add_event = 0;
2789
2790   switch (e->event_type)
2791     {
2792     case SESSION_IO_EVT_RX:
2793       sid = e->session_index;
2794       session = vcl_session_get (wrk, sid);
2795       if (vcl_session_is_closed (session))
2796         break;
2797       vcl_fifo_rx_evt_valid_or_break (session);
2798       session_events = session->vep.ev.events;
2799       if (!(EPOLLIN & session->vep.ev.events) || session->has_rx_evt)
2800         break;
2801       add_event = 1;
2802       events[*num_ev].events |= EPOLLIN;
2803       session_evt_data = session->vep.ev.data.u64;
2804       session->has_rx_evt = 1;
2805       break;
2806     case SESSION_IO_EVT_TX:
2807       sid = e->session_index;
2808       session = vcl_session_get (wrk, sid);
2809       if (vcl_session_is_closed (session))
2810         break;
2811       session_events = session->vep.ev.events;
2812       if (!(EPOLLOUT & session_events))
2813         break;
2814       add_event = 1;
2815       events[*num_ev].events |= EPOLLOUT;
2816       session_evt_data = session->vep.ev.data.u64;
2817       svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (session) ?
2818                                   session->ct_tx_fifo : session->tx_fifo);
2819       break;
2820     case SESSION_CTRL_EVT_ACCEPTED:
2821       session = vcl_session_accepted (wrk,
2822                                       (session_accepted_msg_t *) e->data);
2823       if (!session)
2824         break;
2825
2826       session_events = session->vep.ev.events;
2827       if (!(EPOLLIN & session_events))
2828         break;
2829
2830       add_event = 1;
2831       events[*num_ev].events |= EPOLLIN;
2832       session_evt_data = session->vep.ev.data.u64;
2833       break;
2834     case SESSION_CTRL_EVT_CONNECTED:
2835       connected_msg = (session_connected_msg_t *) e->data;
2836       sid = vcl_session_connected_handler (wrk, connected_msg);
2837       /* Generate EPOLLOUT because there's no connected event */
2838       session = vcl_session_get (wrk, sid);
2839       if (vcl_session_is_closed (session))
2840         break;
2841       session_events = session->vep.ev.events;
2842       if (!(EPOLLOUT & session_events))
2843         break;
2844       add_event = 1;
2845       events[*num_ev].events |= EPOLLOUT;
2846       session_evt_data = session->vep.ev.data.u64;
2847       if (session->session_state == VCL_STATE_DETACHED)
2848         events[*num_ev].events |= EPOLLHUP;
2849       break;
2850     case SESSION_CTRL_EVT_DISCONNECTED:
2851       disconnected_msg = (session_disconnected_msg_t *) e->data;
2852       session = vcl_session_disconnected_handler (wrk, disconnected_msg);
2853       if (vcl_session_is_closed (session))
2854         break;
2855       session_events = session->vep.ev.events;
2856       add_event = 1;
2857       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2858       session_evt_data = session->vep.ev.data.u64;
2859       break;
2860     case SESSION_CTRL_EVT_RESET:
2861       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2862       session = vcl_session_get (wrk, sid);
2863       if (vcl_session_is_closed (session))
2864         break;
2865       session_events = session->vep.ev.events;
2866       add_event = 1;
2867       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2868       session_evt_data = session->vep.ev.data.u64;
2869       break;
2870     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2871       vcl_session_unlisten_reply_handler (wrk, e->data);
2872       break;
2873     case SESSION_CTRL_EVT_MIGRATED:
2874       vcl_session_migrated_handler (wrk, e->data);
2875       break;
2876     case SESSION_CTRL_EVT_CLEANUP:
2877       vcl_session_cleanup_handler (wrk, e->data);
2878       break;
2879     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2880       vcl_session_req_worker_update_handler (wrk, e->data);
2881       break;
2882     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2883       vcl_session_worker_update_reply_handler (wrk, e->data);
2884       break;
2885     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2886       vcl_session_app_add_segment_handler (wrk, e->data);
2887       break;
2888     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2889       vcl_session_app_del_segment_handler (wrk, e->data);
2890       break;
2891     case SESSION_CTRL_EVT_APP_WRK_RPC:
2892       vcl_worker_rpc_handler (wrk, e->data);
2893       break;
2894     default:
2895       VDBG (0, "unhandled: %u", e->event_type);
2896       break;
2897     }
2898
2899   if (add_event)
2900     {
2901       events[*num_ev].data.u64 = session_evt_data;
2902       if (EPOLLONESHOT & session_events)
2903         {
2904           session = vcl_session_get (wrk, sid);
2905           session->vep.ev.events = 0;
2906         }
2907       *num_ev += 1;
2908     }
2909 }
2910
2911 static int
2912 vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2913                           struct epoll_event *events, u32 maxevents,
2914                           double wait_for_time, u32 * num_ev)
2915 {
2916   svm_msg_q_msg_t *msg;
2917   session_event_t *e;
2918   int i;
2919
2920   if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
2921     goto handle_dequeued;
2922
2923   svm_msg_q_lock (mq);
2924   if (svm_msg_q_is_empty (mq))
2925     {
2926       if (!wait_for_time)
2927         {
2928           svm_msg_q_unlock (mq);
2929           return 0;
2930         }
2931       else if (wait_for_time < 0)
2932         {
2933           svm_msg_q_wait (mq);
2934         }
2935       else
2936         {
2937           if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
2938             {
2939               svm_msg_q_unlock (mq);
2940               return 0;
2941             }
2942         }
2943     }
2944   ASSERT (maxevents > *num_ev);
2945   vcl_mq_dequeue_batch (wrk, mq, ~0);
2946   svm_msg_q_unlock (mq);
2947
2948 handle_dequeued:
2949   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
2950     {
2951       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
2952       e = svm_msg_q_msg_data (mq, msg);
2953       if (*num_ev < maxevents)
2954         vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
2955       else
2956         vcl_handle_mq_event (wrk, e);
2957       svm_msg_q_free_msg (mq, msg);
2958     }
2959   vec_reset_length (wrk->mq_msg_vector);
2960   vcl_handle_pending_wrk_updates (wrk);
2961   return *num_ev;
2962 }
2963
2964 static int
2965 vppcom_epoll_wait_condvar (vcl_worker_t * wrk, struct epoll_event *events,
2966                            int maxevents, u32 n_evts, double wait_for_time)
2967 {
2968   double wait = 0, start = 0, now;
2969
2970   if (!n_evts)
2971     {
2972       wait = wait_for_time;
2973       start = clib_time_now (&wrk->clib_time);
2974     }
2975
2976   do
2977     {
2978       vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
2979                                 wait, &n_evts);
2980       if (n_evts)
2981         return n_evts;
2982       if (wait == -1)
2983         continue;
2984
2985       now = clib_time_now (&wrk->clib_time);
2986       wait -= (now - start) * 1e3;
2987       start = now;
2988     }
2989   while (wait > 0);
2990
2991   return 0;
2992 }
2993
2994 static int
2995 vppcom_epoll_wait_eventfd (vcl_worker_t * wrk, struct epoll_event *events,
2996                            int maxevents, u32 n_evts, double wait_for_time)
2997 {
2998   vcl_mq_evt_conn_t *mqc;
2999   int __clib_unused n_read;
3000   int n_mq_evts, i;
3001   u64 buf;
3002
3003   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
3004 again:
3005   n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
3006                           vec_len (wrk->mq_events), wait_for_time);
3007   for (i = 0; i < n_mq_evts; i++)
3008     {
3009       mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
3010       n_read = read (mqc->mq_fd, &buf, sizeof (buf));
3011       vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0, &n_evts);
3012     }
3013   if (!n_evts && n_mq_evts > 0)
3014     goto again;
3015
3016   return (int) n_evts;
3017 }
3018
3019 int
3020 vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
3021                    int maxevents, double wait_for_time)
3022 {
3023   vcl_worker_t *wrk = vcl_worker_get_current ();
3024   vcl_session_t *vep_session;
3025   u32 n_evts = 0;
3026   int i;
3027
3028   if (PREDICT_FALSE (maxevents <= 0))
3029     {
3030       VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
3031       return VPPCOM_EINVAL;
3032     }
3033
3034   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
3035   if (!vep_session)
3036     return VPPCOM_EBADFD;
3037
3038   if (PREDICT_FALSE (!vep_session->is_vep))
3039     {
3040       VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
3041       return VPPCOM_EINVAL;
3042     }
3043
3044   memset (events, 0, sizeof (*events) * maxevents);
3045
3046   if (vec_len (wrk->unhandled_evts_vector))
3047     {
3048       for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3049         {
3050           vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3051                                           events, &n_evts);
3052           if (n_evts == maxevents)
3053             {
3054               vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3055               return n_evts;
3056             }
3057         }
3058       vec_reset_length (wrk->unhandled_evts_vector);
3059     }
3060
3061   if (vcm->cfg.use_mq_eventfd)
3062     return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3063                                       wait_for_time);
3064
3065   return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3066                                     wait_for_time);
3067 }
3068
3069 int
3070 vppcom_session_attr (uint32_t session_handle, uint32_t op,
3071                      void *buffer, uint32_t * buflen)
3072 {
3073   vcl_worker_t *wrk = vcl_worker_get_current ();
3074   vcl_session_t *session;
3075   int rv = VPPCOM_OK;
3076   u32 *flags = buffer, tmp_flags = 0;
3077   vppcom_endpt_t *ep = buffer;
3078
3079   session = vcl_session_get_w_handle (wrk, session_handle);
3080   if (!session)
3081     return VPPCOM_EBADFD;
3082
3083   switch (op)
3084     {
3085     case VPPCOM_ATTR_GET_NREAD:
3086       rv = vcl_session_read_ready (session);
3087       VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3088             rv);
3089       break;
3090
3091     case VPPCOM_ATTR_GET_NWRITE:
3092       rv = vcl_session_write_ready (session);
3093       VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3094             rv);
3095       break;
3096
3097     case VPPCOM_ATTR_GET_FLAGS:
3098       if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
3099         {
3100           *flags =
3101             O_RDWR |
3102             (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK) ?
3103              O_NONBLOCK : 0);
3104           *buflen = sizeof (*flags);
3105           VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3106                 "is_nonblocking = %u", session_handle, *flags,
3107                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
3108         }
3109       else
3110         rv = VPPCOM_EINVAL;
3111       break;
3112
3113     case VPPCOM_ATTR_SET_FLAGS:
3114       if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
3115         {
3116           if (*flags & O_NONBLOCK)
3117             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
3118           else
3119             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_NONBLOCK);
3120
3121           VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3122                 " is_nonblocking = %u", session_handle, *flags,
3123                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
3124         }
3125       else
3126         rv = VPPCOM_EINVAL;
3127       break;
3128
3129     case VPPCOM_ATTR_GET_PEER_ADDR:
3130       if (PREDICT_TRUE (buffer && buflen &&
3131                         (*buflen >= sizeof (*ep)) && ep->ip))
3132         {
3133           ep->is_ip4 = session->transport.is_ip4;
3134           ep->port = session->transport.rmt_port;
3135           if (session->transport.is_ip4)
3136             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3137                               sizeof (ip4_address_t));
3138           else
3139             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3140                               sizeof (ip6_address_t));
3141           *buflen = sizeof (*ep);
3142           VDBG (1, "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3143                 "addr = %U, port %u", session_handle, ep->is_ip4,
3144                 format_ip46_address, &session->transport.rmt_ip,
3145                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3146                 clib_net_to_host_u16 (ep->port));
3147         }
3148       else
3149         rv = VPPCOM_EINVAL;
3150       break;
3151
3152     case VPPCOM_ATTR_GET_LCL_ADDR:
3153       if (PREDICT_TRUE (buffer && buflen &&
3154                         (*buflen >= sizeof (*ep)) && ep->ip))
3155         {
3156           ep->is_ip4 = session->transport.is_ip4;
3157           ep->port = session->transport.lcl_port;
3158           if (session->transport.is_ip4)
3159             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3160                               sizeof (ip4_address_t));
3161           else
3162             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3163                               sizeof (ip6_address_t));
3164           *buflen = sizeof (*ep);
3165           VDBG (1, "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3166                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3167                 &session->transport.lcl_ip,
3168                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3169                 clib_net_to_host_u16 (ep->port));
3170         }
3171       else
3172         rv = VPPCOM_EINVAL;
3173       break;
3174
3175     case VPPCOM_ATTR_SET_LCL_ADDR:
3176       if (PREDICT_TRUE (buffer && buflen &&
3177                         (*buflen >= sizeof (*ep)) && ep->ip))
3178         {
3179           session->transport.is_ip4 = ep->is_ip4;
3180           session->transport.lcl_port = ep->port;
3181           vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3182           *buflen = sizeof (*ep);
3183           VDBG (1, "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3184                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3185                 &session->transport.lcl_ip,
3186                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3187                 clib_net_to_host_u16 (ep->port));
3188         }
3189       else
3190         rv = VPPCOM_EINVAL;
3191       break;
3192
3193     case VPPCOM_ATTR_GET_LIBC_EPFD:
3194       rv = session->libc_epfd;
3195       VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
3196       break;
3197
3198     case VPPCOM_ATTR_SET_LIBC_EPFD:
3199       if (PREDICT_TRUE (buffer && buflen &&
3200                         (*buflen == sizeof (session->libc_epfd))))
3201         {
3202           session->libc_epfd = *(int *) buffer;
3203           *buflen = sizeof (session->libc_epfd);
3204
3205           VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3206                 session->libc_epfd, *buflen);
3207         }
3208       else
3209         rv = VPPCOM_EINVAL;
3210       break;
3211
3212     case VPPCOM_ATTR_GET_PROTOCOL:
3213       if (buffer && buflen && (*buflen >= sizeof (int)))
3214         {
3215           *(int *) buffer = session->session_type;
3216           *buflen = sizeof (int);
3217
3218           VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3219                 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
3220         }
3221       else
3222         rv = VPPCOM_EINVAL;
3223       break;
3224
3225     case VPPCOM_ATTR_GET_LISTEN:
3226       if (buffer && buflen && (*buflen >= sizeof (int)))
3227         {
3228           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3229                                                 VCL_SESS_ATTR_LISTEN);
3230           *buflen = sizeof (int);
3231
3232           VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3233                 *buflen);
3234         }
3235       else
3236         rv = VPPCOM_EINVAL;
3237       break;
3238
3239     case VPPCOM_ATTR_GET_ERROR:
3240       if (buffer && buflen && (*buflen >= sizeof (int)))
3241         {
3242           *(int *) buffer = 0;
3243           *buflen = sizeof (int);
3244
3245           VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3246                 *(int *) buffer, *buflen);
3247         }
3248       else
3249         rv = VPPCOM_EINVAL;
3250       break;
3251
3252     case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3253       if (buffer && buflen && (*buflen >= sizeof (u32)))
3254         {
3255
3256           /* VPP-TBD */
3257           *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
3258                                 session->tx_fifo ?
3259                                 svm_fifo_size (session->tx_fifo) :
3260                                 vcm->cfg.tx_fifo_size);
3261           *buflen = sizeof (u32);
3262
3263           VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3264                 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3265                 *buflen);
3266         }
3267       else
3268         rv = VPPCOM_EINVAL;
3269       break;
3270
3271     case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3272       if (buffer && buflen && (*buflen == sizeof (u32)))
3273         {
3274           /* VPP-TBD */
3275           session->sndbuf_size = *(u32 *) buffer;
3276           VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3277                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3278                 *buflen);
3279         }
3280       else
3281         rv = VPPCOM_EINVAL;
3282       break;
3283
3284     case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3285       if (buffer && buflen && (*buflen >= sizeof (u32)))
3286         {
3287
3288           /* VPP-TBD */
3289           *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
3290                                 session->rx_fifo ?
3291                                 svm_fifo_size (session->rx_fifo) :
3292                                 vcm->cfg.rx_fifo_size);
3293           *buflen = sizeof (u32);
3294
3295           VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3296                 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
3297         }
3298       else
3299         rv = VPPCOM_EINVAL;
3300       break;
3301
3302     case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3303       if (buffer && buflen && (*buflen == sizeof (u32)))
3304         {
3305           /* VPP-TBD */
3306           session->rcvbuf_size = *(u32 *) buffer;
3307           VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3308                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3309                 *buflen);
3310         }
3311       else
3312         rv = VPPCOM_EINVAL;
3313       break;
3314
3315     case VPPCOM_ATTR_GET_REUSEADDR:
3316       if (buffer && buflen && (*buflen >= sizeof (int)))
3317         {
3318           /* VPP-TBD */
3319           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3320                                                 VCL_SESS_ATTR_REUSEADDR);
3321           *buflen = sizeof (int);
3322
3323           VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3324                 *(int *) buffer, *buflen);
3325         }
3326       else
3327         rv = VPPCOM_EINVAL;
3328       break;
3329
3330     case VPPCOM_ATTR_SET_REUSEADDR:
3331       if (buffer && buflen && (*buflen == sizeof (int)) &&
3332           !VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
3333         {
3334           /* VPP-TBD */
3335           if (*(int *) buffer)
3336             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEADDR);
3337           else
3338             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEADDR);
3339
3340           VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3341                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_REUSEADDR),
3342                 *buflen);
3343         }
3344       else
3345         rv = VPPCOM_EINVAL;
3346       break;
3347
3348     case VPPCOM_ATTR_GET_REUSEPORT:
3349       if (buffer && buflen && (*buflen >= sizeof (int)))
3350         {
3351           /* VPP-TBD */
3352           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3353                                                 VCL_SESS_ATTR_REUSEPORT);
3354           *buflen = sizeof (int);
3355
3356           VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3357                 *(int *) buffer, *buflen);
3358         }
3359       else
3360         rv = VPPCOM_EINVAL;
3361       break;
3362
3363     case VPPCOM_ATTR_SET_REUSEPORT:
3364       if (buffer && buflen && (*buflen == sizeof (int)) &&
3365           !VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
3366         {
3367           /* VPP-TBD */
3368           if (*(int *) buffer)
3369             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEPORT);
3370           else
3371             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEPORT);
3372
3373           VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3374                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_REUSEPORT),
3375                 *buflen);
3376         }
3377       else
3378         rv = VPPCOM_EINVAL;
3379       break;
3380
3381     case VPPCOM_ATTR_GET_BROADCAST:
3382       if (buffer && buflen && (*buflen >= sizeof (int)))
3383         {
3384           /* VPP-TBD */
3385           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3386                                                 VCL_SESS_ATTR_BROADCAST);
3387           *buflen = sizeof (int);
3388
3389           VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3390                 *(int *) buffer, *buflen);
3391         }
3392       else
3393         rv = VPPCOM_EINVAL;
3394       break;
3395
3396     case VPPCOM_ATTR_SET_BROADCAST:
3397       if (buffer && buflen && (*buflen == sizeof (int)))
3398         {
3399           /* VPP-TBD */
3400           if (*(int *) buffer)
3401             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_BROADCAST);
3402           else
3403             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_BROADCAST);
3404
3405           VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3406                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_BROADCAST),
3407                 *buflen);
3408         }
3409       else
3410         rv = VPPCOM_EINVAL;
3411       break;
3412
3413     case VPPCOM_ATTR_GET_V6ONLY:
3414       if (buffer && buflen && (*buflen >= sizeof (int)))
3415         {
3416           /* VPP-TBD */
3417           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3418                                                 VCL_SESS_ATTR_V6ONLY);
3419           *buflen = sizeof (int);
3420
3421           VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3422                 *(int *) buffer, *buflen);
3423         }
3424       else
3425         rv = VPPCOM_EINVAL;
3426       break;
3427
3428     case VPPCOM_ATTR_SET_V6ONLY:
3429       if (buffer && buflen && (*buflen == sizeof (int)))
3430         {
3431           /* VPP-TBD */
3432           if (*(int *) buffer)
3433             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_V6ONLY);
3434           else
3435             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_V6ONLY);
3436
3437           VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3438                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_V6ONLY),
3439                 *buflen);
3440         }
3441       else
3442         rv = VPPCOM_EINVAL;
3443       break;
3444
3445     case VPPCOM_ATTR_GET_KEEPALIVE:
3446       if (buffer && buflen && (*buflen >= sizeof (int)))
3447         {
3448           /* VPP-TBD */
3449           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3450                                                 VCL_SESS_ATTR_KEEPALIVE);
3451           *buflen = sizeof (int);
3452
3453           VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3454                 *(int *) buffer, *buflen);
3455         }
3456       else
3457         rv = VPPCOM_EINVAL;
3458       break;
3459
3460     case VPPCOM_ATTR_SET_KEEPALIVE:
3461       if (buffer && buflen && (*buflen == sizeof (int)))
3462         {
3463           /* VPP-TBD */
3464           if (*(int *) buffer)
3465             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_KEEPALIVE);
3466           else
3467             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_KEEPALIVE);
3468
3469           VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3470                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_KEEPALIVE),
3471                 *buflen);
3472         }
3473       else
3474         rv = VPPCOM_EINVAL;
3475       break;
3476
3477     case VPPCOM_ATTR_GET_TCP_NODELAY:
3478       if (buffer && buflen && (*buflen >= sizeof (int)))
3479         {
3480           /* VPP-TBD */
3481           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3482                                                 VCL_SESS_ATTR_TCP_NODELAY);
3483           *buflen = sizeof (int);
3484
3485           VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3486                 *(int *) buffer, *buflen);
3487         }
3488       else
3489         rv = VPPCOM_EINVAL;
3490       break;
3491
3492     case VPPCOM_ATTR_SET_TCP_NODELAY:
3493       if (buffer && buflen && (*buflen == sizeof (int)))
3494         {
3495           /* VPP-TBD */
3496           if (*(int *) buffer)
3497             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
3498           else
3499             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
3500
3501           VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3502                 VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_TCP_NODELAY),
3503                 *buflen);
3504         }
3505       else
3506         rv = VPPCOM_EINVAL;
3507       break;
3508
3509     case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
3510       if (buffer && buflen && (*buflen >= sizeof (int)))
3511         {
3512           /* VPP-TBD */
3513           *(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
3514                                                 VCL_SESS_ATTR_TCP_KEEPIDLE);
3515           *buflen = sizeof (int);
3516
3517           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3518                 *(int *) buffer, *buflen);
3519         }
3520       else
3521         rv = VPPCOM_EINVAL;
3522       break;
3523
3524     case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
3525       if (buffer && buflen && (*buflen == sizeof (int)))
3526         {
3527           /* VPP-TBD */
3528           if (*(int *) buffer)
3529             VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
3530           else
3531             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
3532
3533           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3534                 VCL_SESS_ATTR_TEST (session->attr,
3535                                     VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
3536         }
3537       else
3538         rv = VPPCOM_EINVAL;
3539       break;
3540
3541     case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
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_KEEPINTVL);
3547           *buflen = sizeof (int);
3548
3549           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3550                 *(int *) buffer, *buflen);
3551         }
3552       else
3553         rv = VPPCOM_EINVAL;
3554       break;
3555
3556     case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
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_KEEPINTVL);
3562           else
3563             VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPINTVL);
3564
3565           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3566                 VCL_SESS_ATTR_TEST (session->attr,
3567                                     VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
3568         }
3569       else
3570         rv = VPPCOM_EINVAL;
3571       break;
3572
3573     case VPPCOM_ATTR_GET_TCP_USER_MSS:
3574       if (buffer && buflen && (*buflen >= sizeof (u32)))
3575         {
3576           /* VPP-TBD */
3577           *(u32 *) buffer = session->user_mss;
3578           *buflen = sizeof (int);
3579
3580           VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d, #VPP-TBD#",
3581                 *(int *) buffer, *buflen);
3582         }
3583       else
3584         rv = VPPCOM_EINVAL;
3585       break;
3586
3587     case VPPCOM_ATTR_SET_TCP_USER_MSS:
3588       if (buffer && buflen && (*buflen == sizeof (u32)))
3589         {
3590           /* VPP-TBD */
3591           session->user_mss = *(u32 *) buffer;
3592
3593           VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d, #VPP-TBD#",
3594                 session->user_mss, *buflen);
3595         }
3596       else
3597         rv = VPPCOM_EINVAL;
3598       break;
3599
3600     case VPPCOM_ATTR_SET_SHUT:
3601       if (*flags == SHUT_RD || *flags == SHUT_RDWR)
3602         VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_SHUT_RD);
3603       if (*flags == SHUT_WR || *flags == SHUT_RDWR)
3604         VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_SHUT_WR);
3605       break;
3606
3607     case VPPCOM_ATTR_GET_SHUT:
3608       if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_SHUT_RD))
3609         tmp_flags = 1;
3610       if (VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_SHUT_WR))
3611         tmp_flags |= 2;
3612       if (tmp_flags == 1)
3613         *(int *) buffer = SHUT_RD;
3614       else if (tmp_flags == 2)
3615         *(int *) buffer = SHUT_WR;
3616       else if (tmp_flags == 3)
3617         *(int *) buffer = SHUT_RDWR;
3618       *buflen = sizeof (int);
3619       break;
3620
3621     case VPPCOM_ATTR_SET_CONNECTED:
3622       session->flags |= VCL_SESSION_F_CONNECTED;
3623       break;
3624
3625     default:
3626       rv = VPPCOM_EINVAL;
3627       break;
3628     }
3629
3630   return rv;
3631 }
3632
3633 int
3634 vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
3635                          uint32_t buflen, int flags, vppcom_endpt_t * ep)
3636 {
3637   vcl_worker_t *wrk = vcl_worker_get_current ();
3638   vcl_session_t *session;
3639   int rv = VPPCOM_OK;
3640
3641   if (flags == 0)
3642     rv = vppcom_session_read (session_handle, buffer, buflen);
3643   else if (flags & MSG_PEEK)
3644     rv = vppcom_session_peek (session_handle, buffer, buflen);
3645   else
3646     {
3647       VDBG (0, "Unsupport flags for recvfrom %d", flags);
3648       return VPPCOM_EAFNOSUPPORT;
3649     }
3650
3651   if (ep && rv > 0)
3652     {
3653       session = vcl_session_get_w_handle (wrk, session_handle);
3654       if (session->transport.is_ip4)
3655         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3656                           sizeof (ip4_address_t));
3657       else
3658         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3659                           sizeof (ip6_address_t));
3660       ep->is_ip4 = session->transport.is_ip4;
3661       ep->port = session->transport.rmt_port;
3662     }
3663
3664   return rv;
3665 }
3666
3667 int
3668 vppcom_session_sendto (uint32_t session_handle, void *buffer,
3669                        uint32_t buflen, int flags, vppcom_endpt_t * ep)
3670 {
3671   vcl_worker_t *wrk = vcl_worker_get_current ();
3672   vcl_session_t *s;
3673
3674   s = vcl_session_get_w_handle (wrk, session_handle);
3675   if (!s)
3676     return VPPCOM_EBADFD;
3677
3678   if (!buffer)
3679     return VPPCOM_EINVAL;
3680
3681   if (ep)
3682     {
3683       if (s->session_type != VPPCOM_PROTO_UDP
3684           || (s->flags & VCL_SESSION_F_CONNECTED))
3685         return VPPCOM_EINVAL;
3686
3687       /* Session not connected/bound in vpp. Create it by 'connecting' it */
3688       if (PREDICT_FALSE (s->session_state == VCL_STATE_CLOSED))
3689         {
3690           vcl_send_session_connect (wrk, s);
3691         }
3692       else
3693         {
3694           s->transport.is_ip4 = ep->is_ip4;
3695           s->transport.rmt_port = ep->port;
3696           vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
3697         }
3698     }
3699
3700   if (flags)
3701     {
3702       // TBD check the flags and do the right thing
3703       VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
3704     }
3705
3706   return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
3707                                        s->is_dgram ? 1 : 0));
3708 }
3709
3710 int
3711 vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
3712 {
3713   vcl_worker_t *wrk = vcl_worker_get_current ();
3714   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
3715   u32 i, keep_trying = 1;
3716   svm_msg_q_msg_t msg;
3717   session_event_t *e;
3718   int rv, num_ev = 0;
3719
3720   VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
3721
3722   if (!vp)
3723     return VPPCOM_EFAULT;
3724
3725   do
3726     {
3727       vcl_session_t *session;
3728
3729       /* Dequeue all events and drop all unhandled io events */
3730       while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
3731         {
3732           e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
3733           vcl_handle_mq_event (wrk, e);
3734           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
3735         }
3736       vec_reset_length (wrk->unhandled_evts_vector);
3737
3738       for (i = 0; i < n_sids; i++)
3739         {
3740           session = vcl_session_get (wrk, vp[i].sh);
3741           if (!session)
3742             {
3743               vp[i].revents = POLLHUP;
3744               num_ev++;
3745               continue;
3746             }
3747
3748           vp[i].revents = 0;
3749
3750           if (POLLIN & vp[i].events)
3751             {
3752               rv = vcl_session_read_ready (session);
3753               if (rv > 0)
3754                 {
3755                   vp[i].revents |= POLLIN;
3756                   num_ev++;
3757                 }
3758               else if (rv < 0)
3759                 {
3760                   switch (rv)
3761                     {
3762                     case VPPCOM_ECONNRESET:
3763                       vp[i].revents = POLLHUP;
3764                       break;
3765
3766                     default:
3767                       vp[i].revents = POLLERR;
3768                       break;
3769                     }
3770                   num_ev++;
3771                 }
3772             }
3773
3774           if (POLLOUT & vp[i].events)
3775             {
3776               rv = vcl_session_write_ready (session);
3777               if (rv > 0)
3778                 {
3779                   vp[i].revents |= POLLOUT;
3780                   num_ev++;
3781                 }
3782               else if (rv < 0)
3783                 {
3784                   switch (rv)
3785                     {
3786                     case VPPCOM_ECONNRESET:
3787                       vp[i].revents = POLLHUP;
3788                       break;
3789
3790                     default:
3791                       vp[i].revents = POLLERR;
3792                       break;
3793                     }
3794                   num_ev++;
3795                 }
3796             }
3797
3798           if (0)                // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
3799             {
3800               vp[i].revents = POLLNVAL;
3801               num_ev++;
3802             }
3803         }
3804       if (wait_for_time != -1)
3805         keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
3806     }
3807   while ((num_ev == 0) && keep_trying);
3808
3809   return num_ev;
3810 }
3811
3812 int
3813 vppcom_mq_epoll_fd (void)
3814 {
3815   vcl_worker_t *wrk = vcl_worker_get_current ();
3816   return wrk->mqs_epfd;
3817 }
3818
3819 int
3820 vppcom_session_index (vcl_session_handle_t session_handle)
3821 {
3822   return session_handle & 0xFFFFFF;
3823 }
3824
3825 int
3826 vppcom_session_worker (vcl_session_handle_t session_handle)
3827 {
3828   return session_handle >> 24;
3829 }
3830
3831 int
3832 vppcom_worker_register (void)
3833 {
3834   if (!vcl_worker_alloc_and_init ())
3835     return VPPCOM_EEXIST;
3836
3837   if (vcl_worker_register_with_vpp ())
3838     return VPPCOM_EEXIST;
3839
3840   return VPPCOM_OK;
3841 }
3842
3843 void
3844 vppcom_worker_unregister (void)
3845 {
3846   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
3847   vcl_set_worker_index (~0);
3848 }
3849
3850 void
3851 vppcom_worker_index_set (int index)
3852 {
3853   vcl_set_worker_index (index);
3854 }
3855
3856 int
3857 vppcom_worker_index (void)
3858 {
3859   return vcl_get_worker_index ();
3860 }
3861
3862 int
3863 vppcom_worker_mqs_epfd (void)
3864 {
3865   vcl_worker_t *wrk = vcl_worker_get_current ();
3866   if (!vcm->cfg.use_mq_eventfd)
3867     return -1;
3868   return wrk->mqs_epfd;
3869 }
3870
3871 int
3872 vppcom_session_is_connectable_listener (uint32_t session_handle)
3873 {
3874   vcl_session_t *session;
3875   vcl_worker_t *wrk = vcl_worker_get_current ();
3876   session = vcl_session_get_w_handle (wrk, session_handle);
3877   if (!session)
3878     return VPPCOM_EBADFD;
3879   return vcl_session_is_connectable_listener (wrk, session);
3880 }
3881
3882 int
3883 vppcom_session_listener (uint32_t session_handle)
3884 {
3885   vcl_worker_t *wrk = vcl_worker_get_current ();
3886   vcl_session_t *listen_session, *session;
3887   session = vcl_session_get_w_handle (wrk, session_handle);
3888   if (!session)
3889     return VPPCOM_EBADFD;
3890   if (session->listener_index == VCL_INVALID_SESSION_INDEX)
3891     return VPPCOM_EBADFD;
3892   listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
3893   if (!listen_session)
3894     return VPPCOM_EBADFD;
3895   return vcl_session_handle (listen_session);
3896 }
3897
3898 int
3899 vppcom_session_n_accepted (uint32_t session_handle)
3900 {
3901   vcl_worker_t *wrk = vcl_worker_get_current ();
3902   vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
3903   if (!session)
3904     return VPPCOM_EBADFD;
3905   return session->n_accepted_sessions;
3906 }
3907
3908 const char *
3909 vppcom_proto_str (vppcom_proto_t proto)
3910 {
3911   char const *proto_str;
3912
3913   switch (proto)
3914     {
3915     case VPPCOM_PROTO_TCP:
3916       proto_str = "TCP";
3917       break;
3918     case VPPCOM_PROTO_UDP:
3919       proto_str = "UDP";
3920       break;
3921     case VPPCOM_PROTO_TLS:
3922       proto_str = "TLS";
3923       break;
3924     case VPPCOM_PROTO_QUIC:
3925       proto_str = "QUIC";
3926       break;
3927     default:
3928       proto_str = "UNKNOWN";
3929       break;
3930     }
3931   return proto_str;
3932 }
3933
3934 const char *
3935 vppcom_retval_str (int retval)
3936 {
3937   char const *st;
3938
3939   switch (retval)
3940     {
3941     case VPPCOM_OK:
3942       st = "VPPCOM_OK";
3943       break;
3944
3945     case VPPCOM_EAGAIN:
3946       st = "VPPCOM_EAGAIN";
3947       break;
3948
3949     case VPPCOM_EFAULT:
3950       st = "VPPCOM_EFAULT";
3951       break;
3952
3953     case VPPCOM_ENOMEM:
3954       st = "VPPCOM_ENOMEM";
3955       break;
3956
3957     case VPPCOM_EINVAL:
3958       st = "VPPCOM_EINVAL";
3959       break;
3960
3961     case VPPCOM_EBADFD:
3962       st = "VPPCOM_EBADFD";
3963       break;
3964
3965     case VPPCOM_EAFNOSUPPORT:
3966       st = "VPPCOM_EAFNOSUPPORT";
3967       break;
3968
3969     case VPPCOM_ECONNABORTED:
3970       st = "VPPCOM_ECONNABORTED";
3971       break;
3972
3973     case VPPCOM_ECONNRESET:
3974       st = "VPPCOM_ECONNRESET";
3975       break;
3976
3977     case VPPCOM_ENOTCONN:
3978       st = "VPPCOM_ENOTCONN";
3979       break;
3980
3981     case VPPCOM_ECONNREFUSED:
3982       st = "VPPCOM_ECONNREFUSED";
3983       break;
3984
3985     case VPPCOM_ETIMEDOUT:
3986       st = "VPPCOM_ETIMEDOUT";
3987       break;
3988
3989     default:
3990       st = "UNKNOWN_STATE";
3991       break;
3992     }
3993
3994   return st;
3995 }
3996
3997 /*
3998  * fd.io coding-style-patch-verification: ON
3999  *
4000  * Local Variables:
4001  * eval: (c-set-style "gnu")
4002  * End:
4003  */