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