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