srtp: basic implementation based on libsrtp2
[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 = listen_session->transport.lcl_port;
475   session->transport.lcl_ip = listen_session->transport.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 (vep_session->vep.next_sh != ~0)
2717         {
2718           vcl_session_t *next_session;
2719           next_session = vcl_session_get_w_handle (wrk,
2720                                                    vep_session->vep.next_sh);
2721           if (PREDICT_FALSE (!next_session))
2722             {
2723               VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
2724                     "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
2725               return VPPCOM_EBADFD;
2726             }
2727           ASSERT (next_session->vep.prev_sh == vep_handle);
2728           next_session->vep.prev_sh = session_handle;
2729         }
2730       s->vep.next_sh = vep_session->vep.next_sh;
2731       s->vep.prev_sh = vep_handle;
2732       s->vep.vep_sh = vep_handle;
2733       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2734       s->vep.ev = *event;
2735       s->flags &= ~VCL_SESSION_F_IS_VEP;
2736       s->flags |= VCL_SESSION_F_IS_VEP_SESSION;
2737       vep_session->vep.next_sh = session_handle;
2738
2739       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2740       if (txf && (event->events & EPOLLOUT))
2741         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2742
2743       /* Generate EPOLLOUT if tx fifo not full */
2744       if ((event->events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2745         {
2746           session_event_t e = { 0 };
2747           e.event_type = SESSION_IO_EVT_TX;
2748           e.session_index = s->session_index;
2749           vec_add1 (wrk->unhandled_evts_vector, e);
2750         }
2751       /* Generate EPOLLIN if rx fifo has data */
2752       if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
2753         {
2754           session_event_t e = { 0 };
2755           e.event_type = SESSION_IO_EVT_RX;
2756           e.session_index = s->session_index;
2757           vec_add1 (wrk->unhandled_evts_vector, e);
2758         }
2759       VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2760             vep_handle, session_handle, event->events, event->data.u64);
2761       vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
2762       break;
2763
2764     case EPOLL_CTL_MOD:
2765       if (PREDICT_FALSE (!event))
2766         {
2767           VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
2768           rv = VPPCOM_EINVAL;
2769           goto done;
2770         }
2771       else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2772         {
2773           VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
2774           rv = VPPCOM_EINVAL;
2775           goto done;
2776         }
2777       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2778         {
2779           VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
2780                 session_handle, s->vep.vep_sh, vep_handle);
2781           rv = VPPCOM_EINVAL;
2782           goto done;
2783         }
2784
2785       /* Generate EPOLLOUT when tx_fifo/ct_tx_fifo not full */
2786       if ((event->events & EPOLLOUT) &&
2787           !(s->vep.ev.events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2788         {
2789           session_event_t e = { 0 };
2790           e.event_type = SESSION_IO_EVT_TX;
2791           e.session_index = s->session_index;
2792           vec_add1 (wrk->unhandled_evts_vector, e);
2793         }
2794       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2795       s->vep.ev = *event;
2796       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2797       if (txf)
2798         {
2799           if (event->events & EPOLLOUT)
2800             svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2801           else
2802             svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2803         }
2804       VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2805             vep_handle, session_handle, event->events, event->data.u64);
2806       break;
2807
2808     case EPOLL_CTL_DEL:
2809       if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2810         {
2811           VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
2812           rv = VPPCOM_EINVAL;
2813           goto done;
2814         }
2815       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2816         {
2817           VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
2818                 session_handle, s->vep.vep_sh, vep_handle);
2819           rv = VPPCOM_EINVAL;
2820           goto done;
2821         }
2822
2823       if (s->vep.prev_sh == vep_handle)
2824         vep_session->vep.next_sh = s->vep.next_sh;
2825       else
2826         {
2827           vcl_session_t *prev_session;
2828           prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
2829           if (PREDICT_FALSE (!prev_session))
2830             {
2831               VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
2832                     s->vep.prev_sh, session_handle);
2833               return VPPCOM_EBADFD;
2834             }
2835           ASSERT (prev_session->vep.next_sh == session_handle);
2836           prev_session->vep.next_sh = s->vep.next_sh;
2837         }
2838       if (s->vep.next_sh != ~0)
2839         {
2840           vcl_session_t *next_session;
2841           next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
2842           if (PREDICT_FALSE (!next_session))
2843             {
2844               VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
2845                     s->vep.next_sh, session_handle);
2846               return VPPCOM_EBADFD;
2847             }
2848           ASSERT (next_session->vep.prev_sh == session_handle);
2849           next_session->vep.prev_sh = s->vep.prev_sh;
2850         }
2851
2852       memset (&s->vep, 0, sizeof (s->vep));
2853       s->vep.next_sh = ~0;
2854       s->vep.prev_sh = ~0;
2855       s->vep.vep_sh = ~0;
2856       s->flags &= ~VCL_SESSION_F_IS_VEP_SESSION;
2857
2858       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2859       if (txf)
2860         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2861
2862       VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
2863             session_handle);
2864       vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
2865       break;
2866
2867     default:
2868       VDBG (0, "Invalid operation (%d)!", op);
2869       rv = VPPCOM_EINVAL;
2870     }
2871
2872   vep_verify_epoll_chain (wrk, vep_handle);
2873
2874 done:
2875   return rv;
2876 }
2877
2878 static inline void
2879 vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2880                                 struct epoll_event *events, u32 * num_ev)
2881 {
2882   session_disconnected_msg_t *disconnected_msg;
2883   session_connected_msg_t *connected_msg;
2884   u32 sid = ~0, session_events;
2885   u64 session_evt_data = ~0;
2886   vcl_session_t *s;
2887   u8 add_event = 0;
2888
2889   switch (e->event_type)
2890     {
2891     case SESSION_IO_EVT_RX:
2892       sid = e->session_index;
2893       s = vcl_session_get (wrk, sid);
2894       if (vcl_session_is_closed (s))
2895         break;
2896       vcl_fifo_rx_evt_valid_or_break (s);
2897       session_events = s->vep.ev.events;
2898       if (!(EPOLLIN & s->vep.ev.events)
2899           || (s->flags & VCL_SESSION_F_HAS_RX_EVT))
2900         break;
2901       add_event = 1;
2902       events[*num_ev].events |= EPOLLIN;
2903       session_evt_data = s->vep.ev.data.u64;
2904       s->flags |= VCL_SESSION_F_HAS_RX_EVT;
2905       break;
2906     case SESSION_IO_EVT_TX:
2907       sid = e->session_index;
2908       s = vcl_session_get (wrk, sid);
2909       if (vcl_session_is_closed (s))
2910         break;
2911       session_events = s->vep.ev.events;
2912       if (!(EPOLLOUT & session_events))
2913         break;
2914       add_event = 1;
2915       events[*num_ev].events |= EPOLLOUT;
2916       session_evt_data = s->vep.ev.data.u64;
2917       svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (s) ?
2918                                   s->ct_tx_fifo : s->tx_fifo);
2919       break;
2920     case SESSION_CTRL_EVT_ACCEPTED:
2921       if (!e->postponed)
2922         s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
2923       else
2924         s = vcl_session_get (wrk, e->session_index);
2925       if (!s)
2926         break;
2927       session_events = s->vep.ev.events;
2928       sid = s->session_index;
2929       if (!(EPOLLIN & session_events))
2930         break;
2931       add_event = 1;
2932       events[*num_ev].events |= EPOLLIN;
2933       session_evt_data = s->vep.ev.data.u64;
2934       break;
2935     case SESSION_CTRL_EVT_CONNECTED:
2936       if (!e->postponed)
2937         {
2938           connected_msg = (session_connected_msg_t *) e->data;
2939           sid = vcl_session_connected_handler (wrk, connected_msg);
2940         }
2941       else
2942         sid = e->session_index;
2943       s = vcl_session_get (wrk, sid);
2944       if (vcl_session_is_closed (s))
2945         break;
2946       session_events = s->vep.ev.events;
2947       /* Generate EPOLLOUT because there's no connected event */
2948       if (!(EPOLLOUT & session_events))
2949         break;
2950       add_event = 1;
2951       events[*num_ev].events |= EPOLLOUT;
2952       session_evt_data = s->vep.ev.data.u64;
2953       if (s->session_state == VCL_STATE_DETACHED)
2954         events[*num_ev].events |= EPOLLHUP;
2955       break;
2956     case SESSION_CTRL_EVT_DISCONNECTED:
2957       disconnected_msg = (session_disconnected_msg_t *) e->data;
2958       s = vcl_session_disconnected_handler (wrk, disconnected_msg);
2959       if (vcl_session_is_closed (s))
2960         break;
2961       sid = s->session_index;
2962       session_events = s->vep.ev.events;
2963       add_event = 1;
2964       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2965       session_evt_data = s->vep.ev.data.u64;
2966       break;
2967     case SESSION_CTRL_EVT_RESET:
2968       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2969       s = vcl_session_get (wrk, sid);
2970       if (vcl_session_is_closed (s))
2971         break;
2972       session_events = s->vep.ev.events;
2973       add_event = 1;
2974       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2975       session_evt_data = s->vep.ev.data.u64;
2976       break;
2977     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2978       vcl_session_unlisten_reply_handler (wrk, e->data);
2979       break;
2980     case SESSION_CTRL_EVT_MIGRATED:
2981       vcl_session_migrated_handler (wrk, e->data);
2982       break;
2983     case SESSION_CTRL_EVT_CLEANUP:
2984       vcl_session_cleanup_handler (wrk, e->data);
2985       break;
2986     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2987       vcl_session_req_worker_update_handler (wrk, e->data);
2988       break;
2989     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2990       vcl_session_worker_update_reply_handler (wrk, e->data);
2991       break;
2992     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2993       vcl_session_app_add_segment_handler (wrk, e->data);
2994       break;
2995     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2996       vcl_session_app_del_segment_handler (wrk, e->data);
2997       break;
2998     case SESSION_CTRL_EVT_APP_WRK_RPC:
2999       vcl_worker_rpc_handler (wrk, e->data);
3000       break;
3001     default:
3002       VDBG (0, "unhandled: %u", e->event_type);
3003       break;
3004     }
3005
3006   if (add_event)
3007     {
3008       events[*num_ev].data.u64 = session_evt_data;
3009       if (EPOLLONESHOT & session_events)
3010         {
3011           s = vcl_session_get (wrk, sid);
3012           s->vep.ev.events = 0;
3013         }
3014       *num_ev += 1;
3015     }
3016 }
3017
3018 static int
3019 vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
3020                           struct epoll_event *events, u32 maxevents,
3021                           double wait_for_time, u32 * num_ev)
3022 {
3023   svm_msg_q_msg_t *msg;
3024   session_event_t *e;
3025   int i;
3026
3027   if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
3028     goto handle_dequeued;
3029
3030   if (svm_msg_q_is_empty (mq))
3031     {
3032       if (!wait_for_time)
3033         return 0;
3034       else if (wait_for_time < 0)
3035         svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
3036       else
3037         {
3038           if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
3039             return 0;
3040         }
3041     }
3042   ASSERT (maxevents > *num_ev);
3043   vcl_mq_dequeue_batch (wrk, mq, ~0);
3044
3045 handle_dequeued:
3046   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
3047     {
3048       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
3049       e = svm_msg_q_msg_data (mq, msg);
3050       if (*num_ev < maxevents)
3051         vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
3052       else
3053         vcl_handle_mq_event (wrk, e);
3054       svm_msg_q_free_msg (mq, msg);
3055     }
3056   vec_reset_length (wrk->mq_msg_vector);
3057   vcl_handle_pending_wrk_updates (wrk);
3058   return *num_ev;
3059 }
3060
3061 static int
3062 vppcom_epoll_wait_condvar (vcl_worker_t *wrk, struct epoll_event *events,
3063                            int maxevents, u32 n_evts, double timeout_ms)
3064 {
3065   double end = -1;
3066
3067   if (!n_evts)
3068     {
3069       if (timeout_ms > 0)
3070         end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
3071     }
3072
3073   do
3074     {
3075       vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
3076                                 timeout_ms, &n_evts);
3077       if (n_evts || !timeout_ms)
3078         return n_evts;
3079     }
3080   while (end == -1 || clib_time_now (&wrk->clib_time) < end);
3081
3082   return 0;
3083 }
3084
3085 static int
3086 vppcom_epoll_wait_eventfd (vcl_worker_t *wrk, struct epoll_event *events,
3087                            int maxevents, u32 n_evts, double timeout_ms)
3088 {
3089   int __clib_unused n_read;
3090   vcl_mq_evt_conn_t *mqc;
3091   int n_mq_evts, i;
3092   double end = -1;
3093   u64 buf;
3094
3095   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
3096   if (!n_evts)
3097     {
3098       if (timeout_ms > 0)
3099         end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
3100     }
3101
3102   do
3103     {
3104       n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
3105                               vec_len (wrk->mq_events), timeout_ms);
3106       if (n_mq_evts < 0)
3107         {
3108           VDBG (0, "epoll_wait error %u", errno);
3109           return n_evts;
3110         }
3111
3112       for (i = 0; i < n_mq_evts; i++)
3113         {
3114           mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
3115           n_read = read (mqc->mq_fd, &buf, sizeof (buf));
3116           vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0,
3117                                     &n_evts);
3118         }
3119
3120       if (n_evts || !timeout_ms)
3121         return n_evts;
3122     }
3123   while (end == -1 || clib_time_now (&wrk->clib_time) < end);
3124
3125   return 0;
3126 }
3127
3128 int
3129 vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
3130                    int maxevents, double wait_for_time)
3131 {
3132   vcl_worker_t *wrk = vcl_worker_get_current ();
3133   vcl_session_t *vep_session;
3134   u32 n_evts = 0;
3135   int i;
3136
3137   if (PREDICT_FALSE (maxevents <= 0))
3138     {
3139       VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
3140       return VPPCOM_EINVAL;
3141     }
3142
3143   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
3144   if (!vep_session)
3145     return VPPCOM_EBADFD;
3146
3147   if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
3148     {
3149       VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
3150       return VPPCOM_EINVAL;
3151     }
3152
3153   memset (events, 0, sizeof (*events) * maxevents);
3154
3155   if (vec_len (wrk->unhandled_evts_vector))
3156     {
3157       for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3158         {
3159           vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3160                                           events, &n_evts);
3161           if (n_evts == maxevents)
3162             {
3163               vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3164               return n_evts;
3165             }
3166         }
3167       vec_reset_length (wrk->unhandled_evts_vector);
3168     }
3169   /* Request to only drain unhandled */
3170   if ((int) wait_for_time == -2)
3171     return n_evts;
3172
3173   if (vcm->cfg.use_mq_eventfd)
3174     return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3175                                       wait_for_time);
3176
3177   return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3178                                     wait_for_time);
3179 }
3180
3181 int
3182 vppcom_session_attr (uint32_t session_handle, uint32_t op,
3183                      void *buffer, uint32_t * buflen)
3184 {
3185   vcl_worker_t *wrk = vcl_worker_get_current ();
3186   u32 *flags = buffer, tmp_flags = 0;
3187   vppcom_endpt_t *ep = buffer;
3188   transport_endpt_attr_t tea;
3189   vcl_session_t *session;
3190   int rv = VPPCOM_OK;
3191
3192   session = vcl_session_get_w_handle (wrk, session_handle);
3193   if (!session)
3194     return VPPCOM_EBADFD;
3195
3196   switch (op)
3197     {
3198     case VPPCOM_ATTR_GET_NREAD:
3199       rv = vcl_session_read_ready (session);
3200       VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3201             rv);
3202       break;
3203
3204     case VPPCOM_ATTR_GET_NWRITE:
3205       rv = vcl_session_write_ready (session);
3206       VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3207             rv);
3208       break;
3209
3210     case VPPCOM_ATTR_GET_FLAGS:
3211       if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
3212         {
3213           *flags =
3214             O_RDWR |
3215             (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK) ?
3216              O_NONBLOCK : 0);
3217           *buflen = sizeof (*flags);
3218           VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3219                 "is_nonblocking = %u", session_handle, *flags,
3220                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3221         }
3222       else
3223         rv = VPPCOM_EINVAL;
3224       break;
3225
3226     case VPPCOM_ATTR_SET_FLAGS:
3227       if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
3228         {
3229           if (*flags & O_NONBLOCK)
3230             vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
3231           else
3232             vcl_session_clear_attr (session, VCL_SESS_ATTR_NONBLOCK);
3233
3234           VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3235                 " is_nonblocking = %u", session_handle, *flags,
3236                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3237         }
3238       else
3239         rv = VPPCOM_EINVAL;
3240       break;
3241
3242     case VPPCOM_ATTR_GET_PEER_ADDR:
3243       if (PREDICT_TRUE (buffer && buflen &&
3244                         (*buflen >= sizeof (*ep)) && ep->ip))
3245         {
3246           ep->is_ip4 = session->transport.is_ip4;
3247           ep->port = session->transport.rmt_port;
3248           if (session->transport.is_ip4)
3249             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3250                               sizeof (ip4_address_t));
3251           else
3252             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3253                               sizeof (ip6_address_t));
3254           *buflen = sizeof (*ep);
3255           VDBG (1, "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3256                 "addr = %U, port %u", session_handle, ep->is_ip4,
3257                 format_ip46_address, &session->transport.rmt_ip,
3258                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3259                 clib_net_to_host_u16 (ep->port));
3260         }
3261       else
3262         rv = VPPCOM_EINVAL;
3263       break;
3264
3265     case VPPCOM_ATTR_GET_LCL_ADDR:
3266       if (PREDICT_TRUE (buffer && buflen &&
3267                         (*buflen >= sizeof (*ep)) && ep->ip))
3268         {
3269           ep->is_ip4 = session->transport.is_ip4;
3270           ep->port = session->transport.lcl_port;
3271           if (session->transport.is_ip4)
3272             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3273                               sizeof (ip4_address_t));
3274           else
3275             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3276                               sizeof (ip6_address_t));
3277           *buflen = sizeof (*ep);
3278           VDBG (1, "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3279                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3280                 &session->transport.lcl_ip,
3281                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3282                 clib_net_to_host_u16 (ep->port));
3283         }
3284       else
3285         rv = VPPCOM_EINVAL;
3286       break;
3287
3288     case VPPCOM_ATTR_SET_LCL_ADDR:
3289       if (PREDICT_TRUE (buffer && buflen &&
3290                         (*buflen >= sizeof (*ep)) && ep->ip))
3291         {
3292           session->transport.is_ip4 = ep->is_ip4;
3293           session->transport.lcl_port = ep->port;
3294           vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3295           *buflen = sizeof (*ep);
3296           VDBG (1, "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3297                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3298                 &session->transport.lcl_ip,
3299                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3300                 clib_net_to_host_u16 (ep->port));
3301         }
3302       else
3303         rv = VPPCOM_EINVAL;
3304       break;
3305
3306     case VPPCOM_ATTR_GET_LIBC_EPFD:
3307       rv = session->libc_epfd;
3308       VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
3309       break;
3310
3311     case VPPCOM_ATTR_SET_LIBC_EPFD:
3312       if (PREDICT_TRUE (buffer && buflen &&
3313                         (*buflen == sizeof (session->libc_epfd))))
3314         {
3315           session->libc_epfd = *(int *) buffer;
3316           *buflen = sizeof (session->libc_epfd);
3317
3318           VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3319                 session->libc_epfd, *buflen);
3320         }
3321       else
3322         rv = VPPCOM_EINVAL;
3323       break;
3324
3325     case VPPCOM_ATTR_GET_PROTOCOL:
3326       if (buffer && buflen && (*buflen >= sizeof (int)))
3327         {
3328           *(int *) buffer = session->session_type;
3329           *buflen = sizeof (int);
3330
3331           VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3332                 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
3333         }
3334       else
3335         rv = VPPCOM_EINVAL;
3336       break;
3337
3338     case VPPCOM_ATTR_GET_LISTEN:
3339       if (buffer && buflen && (*buflen >= sizeof (int)))
3340         {
3341           *(int *) buffer = vcl_session_has_attr (session,
3342                                                   VCL_SESS_ATTR_LISTEN);
3343           *buflen = sizeof (int);
3344
3345           VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3346                 *buflen);
3347         }
3348       else
3349         rv = VPPCOM_EINVAL;
3350       break;
3351
3352     case VPPCOM_ATTR_GET_ERROR:
3353       if (buffer && buflen && (*buflen >= sizeof (int)))
3354         {
3355           *(int *) buffer = 0;
3356           *buflen = sizeof (int);
3357
3358           VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3359                 *(int *) buffer, *buflen);
3360         }
3361       else
3362         rv = VPPCOM_EINVAL;
3363       break;
3364
3365     case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3366       if (buffer && buflen && (*buflen >= sizeof (u32)))
3367         {
3368
3369           /* VPP-TBD */
3370           *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
3371                                 session->tx_fifo ?
3372                                 svm_fifo_size (session->tx_fifo) :
3373                                 vcm->cfg.tx_fifo_size);
3374           *buflen = sizeof (u32);
3375
3376           VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3377                 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3378                 *buflen);
3379         }
3380       else
3381         rv = VPPCOM_EINVAL;
3382       break;
3383
3384     case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3385       if (buffer && buflen && (*buflen == sizeof (u32)))
3386         {
3387           /* VPP-TBD */
3388           session->sndbuf_size = *(u32 *) buffer;
3389           VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3390                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3391                 *buflen);
3392         }
3393       else
3394         rv = VPPCOM_EINVAL;
3395       break;
3396
3397     case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3398       if (buffer && buflen && (*buflen >= sizeof (u32)))
3399         {
3400
3401           /* VPP-TBD */
3402           *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
3403                                 session->rx_fifo ?
3404                                 svm_fifo_size (session->rx_fifo) :
3405                                 vcm->cfg.rx_fifo_size);
3406           *buflen = sizeof (u32);
3407
3408           VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3409                 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
3410         }
3411       else
3412         rv = VPPCOM_EINVAL;
3413       break;
3414
3415     case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3416       if (buffer && buflen && (*buflen == sizeof (u32)))
3417         {
3418           /* VPP-TBD */
3419           session->rcvbuf_size = *(u32 *) buffer;
3420           VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3421                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3422                 *buflen);
3423         }
3424       else
3425         rv = VPPCOM_EINVAL;
3426       break;
3427
3428     case VPPCOM_ATTR_GET_REUSEADDR:
3429       if (buffer && buflen && (*buflen >= sizeof (int)))
3430         {
3431           /* VPP-TBD */
3432           *(int *) buffer = vcl_session_has_attr (session,
3433                                                   VCL_SESS_ATTR_REUSEADDR);
3434           *buflen = sizeof (int);
3435
3436           VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3437                 *(int *) buffer, *buflen);
3438         }
3439       else
3440         rv = VPPCOM_EINVAL;
3441       break;
3442
3443     case VPPCOM_ATTR_SET_REUSEADDR:
3444       if (buffer && buflen && (*buflen == sizeof (int)) &&
3445           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3446         {
3447           /* VPP-TBD */
3448           if (*(int *) buffer)
3449             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEADDR);
3450           else
3451             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEADDR);
3452
3453           VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3454                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEADDR),
3455                 *buflen);
3456         }
3457       else
3458         rv = VPPCOM_EINVAL;
3459       break;
3460
3461     case VPPCOM_ATTR_GET_REUSEPORT:
3462       if (buffer && buflen && (*buflen >= sizeof (int)))
3463         {
3464           /* VPP-TBD */
3465           *(int *) buffer = vcl_session_has_attr (session,
3466                                                   VCL_SESS_ATTR_REUSEPORT);
3467           *buflen = sizeof (int);
3468
3469           VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3470                 *(int *) buffer, *buflen);
3471         }
3472       else
3473         rv = VPPCOM_EINVAL;
3474       break;
3475
3476     case VPPCOM_ATTR_SET_REUSEPORT:
3477       if (buffer && buflen && (*buflen == sizeof (int)) &&
3478           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3479         {
3480           /* VPP-TBD */
3481           if (*(int *) buffer)
3482             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEPORT);
3483           else
3484             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEPORT);
3485
3486           VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3487                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEPORT),
3488                 *buflen);
3489         }
3490       else
3491         rv = VPPCOM_EINVAL;
3492       break;
3493
3494     case VPPCOM_ATTR_GET_BROADCAST:
3495       if (buffer && buflen && (*buflen >= sizeof (int)))
3496         {
3497           /* VPP-TBD */
3498           *(int *) buffer = vcl_session_has_attr (session,
3499                                                   VCL_SESS_ATTR_BROADCAST);
3500           *buflen = sizeof (int);
3501
3502           VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3503                 *(int *) buffer, *buflen);
3504         }
3505       else
3506         rv = VPPCOM_EINVAL;
3507       break;
3508
3509     case VPPCOM_ATTR_SET_BROADCAST:
3510       if (buffer && buflen && (*buflen == sizeof (int)))
3511         {
3512           /* VPP-TBD */
3513           if (*(int *) buffer)
3514             vcl_session_set_attr (session, VCL_SESS_ATTR_BROADCAST);
3515           else
3516             vcl_session_clear_attr (session, VCL_SESS_ATTR_BROADCAST);
3517
3518           VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3519                 vcl_session_has_attr (session, VCL_SESS_ATTR_BROADCAST),
3520                 *buflen);
3521         }
3522       else
3523         rv = VPPCOM_EINVAL;
3524       break;
3525
3526     case VPPCOM_ATTR_GET_V6ONLY:
3527       if (buffer && buflen && (*buflen >= sizeof (int)))
3528         {
3529           /* VPP-TBD */
3530           *(int *) buffer = vcl_session_has_attr (session,
3531                                                   VCL_SESS_ATTR_V6ONLY);
3532           *buflen = sizeof (int);
3533
3534           VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3535                 *(int *) buffer, *buflen);
3536         }
3537       else
3538         rv = VPPCOM_EINVAL;
3539       break;
3540
3541     case VPPCOM_ATTR_SET_V6ONLY:
3542       if (buffer && buflen && (*buflen == sizeof (int)))
3543         {
3544           /* VPP-TBD */
3545           if (*(int *) buffer)
3546             vcl_session_set_attr (session, VCL_SESS_ATTR_V6ONLY);
3547           else
3548             vcl_session_clear_attr (session, VCL_SESS_ATTR_V6ONLY);
3549
3550           VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3551                 vcl_session_has_attr (session, VCL_SESS_ATTR_V6ONLY),
3552                 *buflen);
3553         }
3554       else
3555         rv = VPPCOM_EINVAL;
3556       break;
3557
3558     case VPPCOM_ATTR_GET_KEEPALIVE:
3559       if (buffer && buflen && (*buflen >= sizeof (int)))
3560         {
3561           /* VPP-TBD */
3562           *(int *) buffer = vcl_session_has_attr (session,
3563                                                   VCL_SESS_ATTR_KEEPALIVE);
3564           *buflen = sizeof (int);
3565
3566           VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3567                 *(int *) buffer, *buflen);
3568         }
3569       else
3570         rv = VPPCOM_EINVAL;
3571       break;
3572
3573     case VPPCOM_ATTR_SET_KEEPALIVE:
3574       if (buffer && buflen && (*buflen == sizeof (int)))
3575         {
3576           /* VPP-TBD */
3577           if (*(int *) buffer)
3578             vcl_session_set_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3579           else
3580             vcl_session_clear_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3581
3582           VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3583                 vcl_session_has_attr (session, VCL_SESS_ATTR_KEEPALIVE),
3584                 *buflen);
3585         }
3586       else
3587         rv = VPPCOM_EINVAL;
3588       break;
3589
3590     case VPPCOM_ATTR_GET_TCP_NODELAY:
3591       if (buffer && buflen && (*buflen >= sizeof (int)))
3592         {
3593           /* VPP-TBD */
3594           *(int *) buffer = vcl_session_has_attr (session,
3595                                                   VCL_SESS_ATTR_TCP_NODELAY);
3596           *buflen = sizeof (int);
3597
3598           VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3599                 *(int *) buffer, *buflen);
3600         }
3601       else
3602         rv = VPPCOM_EINVAL;
3603       break;
3604
3605     case VPPCOM_ATTR_SET_TCP_NODELAY:
3606       if (buffer && buflen && (*buflen == sizeof (int)))
3607         {
3608           /* VPP-TBD */
3609           if (*(int *) buffer)
3610             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3611           else
3612             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3613
3614           VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3615                 vcl_session_has_attr (session, VCL_SESS_ATTR_TCP_NODELAY),
3616                 *buflen);
3617         }
3618       else
3619         rv = VPPCOM_EINVAL;
3620       break;
3621
3622     case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
3623       if (buffer && buflen && (*buflen >= sizeof (int)))
3624         {
3625           /* VPP-TBD */
3626           *(int *) buffer = vcl_session_has_attr (session,
3627                                                   VCL_SESS_ATTR_TCP_KEEPIDLE);
3628           *buflen = sizeof (int);
3629
3630           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3631                 *(int *) buffer, *buflen);
3632         }
3633       else
3634         rv = VPPCOM_EINVAL;
3635       break;
3636
3637     case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
3638       if (buffer && buflen && (*buflen == sizeof (int)))
3639         {
3640           /* VPP-TBD */
3641           if (*(int *) buffer)
3642             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3643           else
3644             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3645
3646           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3647                 vcl_session_has_attr (session,
3648                                       VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
3649         }
3650       else
3651         rv = VPPCOM_EINVAL;
3652       break;
3653
3654     case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
3655       if (buffer && buflen && (*buflen >= sizeof (int)))
3656         {
3657           /* VPP-TBD */
3658           *(int *) buffer = vcl_session_has_attr (session,
3659                                                   VCL_SESS_ATTR_TCP_KEEPINTVL);
3660           *buflen = sizeof (int);
3661
3662           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3663                 *(int *) buffer, *buflen);
3664         }
3665       else
3666         rv = VPPCOM_EINVAL;
3667       break;
3668
3669     case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
3670       if (buffer && buflen && (*buflen == sizeof (int)))
3671         {
3672           /* VPP-TBD */
3673           if (*(int *) buffer)
3674             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3675           else
3676             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3677
3678           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3679                 vcl_session_has_attr (session,
3680                                       VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
3681         }
3682       else
3683         rv = VPPCOM_EINVAL;
3684       break;
3685
3686     case VPPCOM_ATTR_GET_TCP_USER_MSS:
3687       if (!(buffer && buflen && (*buflen >= sizeof (u32))))
3688         {
3689           rv = VPPCOM_EINVAL;
3690           break;
3691         }
3692
3693       tea.type = TRANSPORT_ENDPT_ATTR_MSS;
3694       tea.mss = *(u32 *) buffer;
3695       if (vcl_session_transport_attr (wrk, session, 1 /* is_get */, &tea))
3696         rv = VPPCOM_ENOPROTOOPT;
3697
3698       if (!rv)
3699         {
3700           *(u32 *) buffer = tea.mss;
3701           *buflen = sizeof (int);
3702         }
3703
3704       VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d", *(int *) buffer,
3705             *buflen);
3706       break;
3707
3708     case VPPCOM_ATTR_SET_TCP_USER_MSS:
3709       if (!(buffer && buflen && (*buflen == sizeof (u32))))
3710         {
3711           rv = VPPCOM_EINVAL;
3712           break;
3713         }
3714
3715       tea.type = TRANSPORT_ENDPT_ATTR_MSS;
3716       tea.mss = *(u32 *) buffer;
3717       if (vcl_session_transport_attr (wrk, session, 0 /* is_get */, &tea))
3718         rv = VPPCOM_ENOPROTOOPT;
3719
3720       VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d", tea.mss,
3721             *buflen);
3722       break;
3723
3724     case VPPCOM_ATTR_SET_SHUT:
3725       if (*flags == SHUT_RD || *flags == SHUT_RDWR)
3726         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_RD);
3727       if (*flags == SHUT_WR || *flags == SHUT_RDWR)
3728         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_WR);
3729       break;
3730
3731     case VPPCOM_ATTR_GET_SHUT:
3732       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_RD))
3733         tmp_flags = 1;
3734       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_WR))
3735         tmp_flags |= 2;
3736       if (tmp_flags == 1)
3737         *(int *) buffer = SHUT_RD;
3738       else if (tmp_flags == 2)
3739         *(int *) buffer = SHUT_WR;
3740       else if (tmp_flags == 3)
3741         *(int *) buffer = SHUT_RDWR;
3742       *buflen = sizeof (int);
3743       break;
3744
3745     case VPPCOM_ATTR_SET_CONNECTED:
3746       session->flags |= VCL_SESSION_F_CONNECTED;
3747       break;
3748
3749     case VPPCOM_ATTR_SET_CKPAIR:
3750       if (!(buffer && buflen && (*buflen == sizeof (int))) ||
3751           !vcl_session_has_crypto (session))
3752         {
3753           rv = VPPCOM_EINVAL;
3754           break;
3755         }
3756       if (!session->ext_config)
3757         {
3758           vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_CRYPTO,
3759                                      sizeof (transport_endpt_ext_cfg_t));
3760         }
3761       else if (session->ext_config->type != TRANSPORT_ENDPT_EXT_CFG_CRYPTO)
3762         {
3763           rv = VPPCOM_EINVAL;
3764           break;
3765         }
3766
3767       session->ext_config->crypto.ckpair_index = *(uint32_t *) buffer;
3768       break;
3769
3770     case VPPCOM_ATTR_SET_VRF:
3771       if (!(buffer && buflen && (*buflen == sizeof (u32))))
3772         {
3773           rv = VPPCOM_EINVAL;
3774           break;
3775         }
3776       session->vrf = *(u32 *) buffer;
3777       break;
3778
3779     case VPPCOM_ATTR_GET_VRF:
3780       if (!(buffer && buflen && (*buflen >= sizeof (u32))))
3781         {
3782           rv = VPPCOM_EINVAL;
3783           break;
3784         }
3785       *(u32 *) buffer = session->vrf;
3786       *buflen = sizeof (u32);
3787       break;
3788
3789     case VPPCOM_ATTR_GET_DOMAIN:
3790       if (!(buffer && buflen && (*buflen >= sizeof (int))))
3791         {
3792           rv = VPPCOM_EINVAL;
3793           break;
3794         }
3795
3796       if (session->transport.is_ip4)
3797         *(int *) buffer = AF_INET;
3798       else
3799         *(int *) buffer = AF_INET6;
3800       *buflen = sizeof (int);
3801
3802       VDBG (2, "VPPCOM_ATTR_GET_DOMAIN: %d, buflen %u", *(int *) buffer,
3803             *buflen);
3804       break;
3805
3806     case VPPCOM_ATTR_SET_ENDPT_EXT_CFG:
3807       if (!(buffer && buflen && (*buflen > 0)))
3808         {
3809           rv = VPPCOM_EINVAL;
3810           break;
3811         }
3812       if (session->ext_config)
3813         {
3814           rv = VPPCOM_EINVAL;
3815           break;
3816         }
3817       vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_NONE,
3818                                  *buflen + sizeof (u32));
3819       clib_memcpy (session->ext_config->data, buffer, *buflen);
3820       session->ext_config->len = *buflen;
3821       break;
3822
3823     default:
3824       rv = VPPCOM_EINVAL;
3825       break;
3826     }
3827
3828   return rv;
3829 }
3830
3831 int
3832 vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
3833                          uint32_t buflen, int flags, vppcom_endpt_t * ep)
3834 {
3835   vcl_worker_t *wrk = vcl_worker_get_current ();
3836   vcl_session_t *session;
3837   int rv = VPPCOM_OK;
3838
3839   if (flags == 0)
3840     rv = vppcom_session_read (session_handle, buffer, buflen);
3841   else if (flags & MSG_PEEK)
3842     rv = vppcom_session_peek (session_handle, buffer, buflen);
3843   else
3844     {
3845       VDBG (0, "Unsupport flags for recvfrom %d", flags);
3846       return VPPCOM_EAFNOSUPPORT;
3847     }
3848
3849   if (ep && rv > 0)
3850     {
3851       session = vcl_session_get_w_handle (wrk, session_handle);
3852       if (session->transport.is_ip4)
3853         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3854                           sizeof (ip4_address_t));
3855       else
3856         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3857                           sizeof (ip6_address_t));
3858       ep->is_ip4 = session->transport.is_ip4;
3859       ep->port = session->transport.rmt_port;
3860     }
3861
3862   return rv;
3863 }
3864
3865 int
3866 vppcom_session_sendto (uint32_t session_handle, void *buffer,
3867                        uint32_t buflen, int flags, vppcom_endpt_t * ep)
3868 {
3869   vcl_worker_t *wrk = vcl_worker_get_current ();
3870   vcl_session_t *s;
3871
3872   s = vcl_session_get_w_handle (wrk, session_handle);
3873   if (!s)
3874     return VPPCOM_EBADFD;
3875
3876   if (!buffer)
3877     return VPPCOM_EINVAL;
3878
3879   if (ep)
3880     {
3881       if (!vcl_session_is_cl (s))
3882         return VPPCOM_EINVAL;
3883
3884       /* Session not connected/bound in vpp. Create it by 'connecting' it */
3885       if (PREDICT_FALSE (s->session_state == VCL_STATE_CLOSED))
3886         {
3887           u32 session_index = s->session_index;
3888           f64 timeout = vcm->cfg.session_timeout;
3889           int rv;
3890
3891           vcl_send_session_connect (wrk, s);
3892           rv = vppcom_wait_for_session_state_change (session_index,
3893                                                      VCL_STATE_READY,
3894                                                      timeout);
3895           if (rv < 0)
3896             return rv;
3897           s = vcl_session_get (wrk, session_index);
3898         }
3899
3900       s->transport.is_ip4 = ep->is_ip4;
3901       s->transport.rmt_port = ep->port;
3902       vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
3903     }
3904
3905   if (flags)
3906     {
3907       // TBD check the flags and do the right thing
3908       VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
3909     }
3910
3911   return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
3912                                        s->is_dgram ? 1 : 0));
3913 }
3914
3915 int
3916 vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
3917 {
3918   vcl_worker_t *wrk = vcl_worker_get_current ();
3919   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
3920   u32 i, keep_trying = 1;
3921   svm_msg_q_msg_t msg;
3922   session_event_t *e;
3923   int rv, num_ev = 0;
3924
3925   VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
3926
3927   if (!vp)
3928     return VPPCOM_EFAULT;
3929
3930   do
3931     {
3932       vcl_session_t *session;
3933
3934       /* Dequeue all events and drop all unhandled io events */
3935       while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
3936         {
3937           e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
3938           vcl_handle_mq_event (wrk, e);
3939           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
3940         }
3941       vec_reset_length (wrk->unhandled_evts_vector);
3942
3943       for (i = 0; i < n_sids; i++)
3944         {
3945           session = vcl_session_get (wrk, vp[i].sh);
3946           if (!session)
3947             {
3948               vp[i].revents = POLLHUP;
3949               num_ev++;
3950               continue;
3951             }
3952
3953           vp[i].revents = 0;
3954
3955           if (POLLIN & vp[i].events)
3956             {
3957               rv = vcl_session_read_ready (session);
3958               if (rv > 0)
3959                 {
3960                   vp[i].revents |= POLLIN;
3961                   num_ev++;
3962                 }
3963               else if (rv < 0)
3964                 {
3965                   switch (rv)
3966                     {
3967                     case VPPCOM_ECONNRESET:
3968                       vp[i].revents = POLLHUP;
3969                       break;
3970
3971                     default:
3972                       vp[i].revents = POLLERR;
3973                       break;
3974                     }
3975                   num_ev++;
3976                 }
3977             }
3978
3979           if (POLLOUT & vp[i].events)
3980             {
3981               rv = vcl_session_write_ready (session);
3982               if (rv > 0)
3983                 {
3984                   vp[i].revents |= POLLOUT;
3985                   num_ev++;
3986                 }
3987               else if (rv < 0)
3988                 {
3989                   switch (rv)
3990                     {
3991                     case VPPCOM_ECONNRESET:
3992                       vp[i].revents = POLLHUP;
3993                       break;
3994
3995                     default:
3996                       vp[i].revents = POLLERR;
3997                       break;
3998                     }
3999                   num_ev++;
4000                 }
4001             }
4002
4003           if (0)                // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
4004             {
4005               vp[i].revents = POLLNVAL;
4006               num_ev++;
4007             }
4008         }
4009       if (wait_for_time != -1)
4010         keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
4011     }
4012   while ((num_ev == 0) && keep_trying);
4013
4014   return num_ev;
4015 }
4016
4017 int
4018 vppcom_mq_epoll_fd (void)
4019 {
4020   vcl_worker_t *wrk = vcl_worker_get_current ();
4021   return wrk->mqs_epfd;
4022 }
4023
4024 int
4025 vppcom_session_index (vcl_session_handle_t session_handle)
4026 {
4027   return session_handle & 0xFFFFFF;
4028 }
4029
4030 int
4031 vppcom_session_worker (vcl_session_handle_t session_handle)
4032 {
4033   return session_handle >> 24;
4034 }
4035
4036 int
4037 vppcom_worker_register (void)
4038 {
4039   if (!vcl_worker_alloc_and_init ())
4040     return VPPCOM_EEXIST;
4041
4042   if (vcl_worker_register_with_vpp ())
4043     return VPPCOM_EEXIST;
4044
4045   return VPPCOM_OK;
4046 }
4047
4048 void
4049 vppcom_worker_unregister (void)
4050 {
4051   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
4052   vcl_set_worker_index (~0);
4053 }
4054
4055 void
4056 vppcom_worker_index_set (int index)
4057 {
4058   vcl_set_worker_index (index);
4059 }
4060
4061 int
4062 vppcom_worker_index (void)
4063 {
4064   return vcl_get_worker_index ();
4065 }
4066
4067 int
4068 vppcom_worker_mqs_epfd (void)
4069 {
4070   vcl_worker_t *wrk = vcl_worker_get_current ();
4071   if (!vcm->cfg.use_mq_eventfd)
4072     return -1;
4073   return wrk->mqs_epfd;
4074 }
4075
4076 int
4077 vppcom_session_is_connectable_listener (uint32_t session_handle)
4078 {
4079   vcl_session_t *session;
4080   vcl_worker_t *wrk = vcl_worker_get_current ();
4081   session = vcl_session_get_w_handle (wrk, session_handle);
4082   if (!session)
4083     return VPPCOM_EBADFD;
4084   return vcl_session_is_connectable_listener (wrk, session);
4085 }
4086
4087 int
4088 vppcom_session_listener (uint32_t session_handle)
4089 {
4090   vcl_worker_t *wrk = vcl_worker_get_current ();
4091   vcl_session_t *listen_session, *session;
4092   session = vcl_session_get_w_handle (wrk, session_handle);
4093   if (!session)
4094     return VPPCOM_EBADFD;
4095   if (session->listener_index == VCL_INVALID_SESSION_INDEX)
4096     return VPPCOM_EBADFD;
4097   listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
4098   if (!listen_session)
4099     return VPPCOM_EBADFD;
4100   return vcl_session_handle (listen_session);
4101 }
4102
4103 int
4104 vppcom_session_n_accepted (uint32_t session_handle)
4105 {
4106   vcl_worker_t *wrk = vcl_worker_get_current ();
4107   vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
4108   if (!session)
4109     return VPPCOM_EBADFD;
4110   return session->n_accepted_sessions;
4111 }
4112
4113 const char *
4114 vppcom_proto_str (vppcom_proto_t proto)
4115 {
4116   char const *proto_str;
4117
4118   switch (proto)
4119     {
4120     case VPPCOM_PROTO_TCP:
4121       proto_str = "TCP";
4122       break;
4123     case VPPCOM_PROTO_UDP:
4124       proto_str = "UDP";
4125       break;
4126     case VPPCOM_PROTO_TLS:
4127       proto_str = "TLS";
4128       break;
4129     case VPPCOM_PROTO_QUIC:
4130       proto_str = "QUIC";
4131       break;
4132     case VPPCOM_PROTO_DTLS:
4133       proto_str = "DTLS";
4134       break;
4135     case VPPCOM_PROTO_SRTP:
4136       proto_str = "SRTP";
4137       break;
4138     default:
4139       proto_str = "UNKNOWN";
4140       break;
4141     }
4142   return proto_str;
4143 }
4144
4145 const char *
4146 vppcom_retval_str (int retval)
4147 {
4148   char const *st;
4149
4150   switch (retval)
4151     {
4152     case VPPCOM_OK:
4153       st = "VPPCOM_OK";
4154       break;
4155
4156     case VPPCOM_EAGAIN:
4157       st = "VPPCOM_EAGAIN";
4158       break;
4159
4160     case VPPCOM_EFAULT:
4161       st = "VPPCOM_EFAULT";
4162       break;
4163
4164     case VPPCOM_ENOMEM:
4165       st = "VPPCOM_ENOMEM";
4166       break;
4167
4168     case VPPCOM_EINVAL:
4169       st = "VPPCOM_EINVAL";
4170       break;
4171
4172     case VPPCOM_EBADFD:
4173       st = "VPPCOM_EBADFD";
4174       break;
4175
4176     case VPPCOM_EAFNOSUPPORT:
4177       st = "VPPCOM_EAFNOSUPPORT";
4178       break;
4179
4180     case VPPCOM_ECONNABORTED:
4181       st = "VPPCOM_ECONNABORTED";
4182       break;
4183
4184     case VPPCOM_ECONNRESET:
4185       st = "VPPCOM_ECONNRESET";
4186       break;
4187
4188     case VPPCOM_ENOTCONN:
4189       st = "VPPCOM_ENOTCONN";
4190       break;
4191
4192     case VPPCOM_ECONNREFUSED:
4193       st = "VPPCOM_ECONNREFUSED";
4194       break;
4195
4196     case VPPCOM_ETIMEDOUT:
4197       st = "VPPCOM_ETIMEDOUT";
4198       break;
4199
4200     default:
4201       st = "UNKNOWN_STATE";
4202       break;
4203     }
4204
4205   return st;
4206 }
4207
4208 int
4209 vppcom_add_cert_key_pair (vppcom_cert_key_pair_t *ckpair)
4210 {
4211   if (vcm->cfg.vpp_app_socket_api)
4212     {
4213       clib_warning ("not supported");
4214       return VPPCOM_EINVAL;
4215     }
4216   return vcl_bapi_add_cert_key_pair (ckpair);
4217 }
4218
4219 int
4220 vppcom_del_cert_key_pair (uint32_t ckpair_index)
4221 {
4222   if (vcm->cfg.vpp_app_socket_api)
4223     {
4224       clib_warning ("not supported");
4225       return VPPCOM_EINVAL;
4226     }
4227   return vcl_bapi_del_cert_key_pair (ckpair_index);
4228 }
4229
4230 /*
4231  * fd.io coding-style-patch-verification: ON
4232  *
4233  * Local Variables:
4234  * eval: (c-set-style "gnu")
4235  * End:
4236  */