vcl: accept zero length writes
[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   /* Accept zero length writes but just return */
2156   if (PREDICT_FALSE (!n))
2157     return VPPCOM_OK;
2158
2159   if (PREDICT_FALSE (!buf))
2160     return VPPCOM_EFAULT;
2161
2162   if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2163     {
2164       VDBG (0, "ERROR: session %u [0x%llx]: cannot write to an epoll"
2165             " session!", s->session_index, s->vpp_handle);
2166       return VPPCOM_EBADFD;
2167     }
2168
2169   if (PREDICT_FALSE (!vcl_session_is_open (s) ||
2170                      (s->flags & VCL_SESSION_F_SHUTDOWN)))
2171     {
2172       VDBG (1, "session %u [0x%llx]: is not open! state 0x%x (%s)",
2173             s->session_index, s->vpp_handle, s->session_state,
2174             vppcom_session_state_str (s->session_state));
2175       return vcl_session_closed_error (s);;
2176     }
2177
2178   is_ct = vcl_session_is_ct (s);
2179   tx_fifo = is_ct ? s->ct_tx_fifo : s->tx_fifo;
2180   is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
2181
2182   mq = wrk->app_event_queue;
2183   if (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2184     {
2185       if (is_nonblocking)
2186         {
2187           return VPPCOM_EWOULDBLOCK;
2188         }
2189       while (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2190         {
2191           svm_fifo_add_want_deq_ntf (tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
2192           if (vcl_session_is_closing (s))
2193             return vcl_session_closing_error (s);
2194
2195           svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
2196           vcl_worker_flush_mq_events (wrk);
2197         }
2198     }
2199
2200   et = SESSION_IO_EVT_TX;
2201   if (is_flush && !is_ct)
2202     et = SESSION_IO_EVT_TX_FLUSH;
2203
2204   if (is_dgram)
2205     n_write = app_send_dgram_raw (tx_fifo, &s->transport,
2206                                   s->vpp_evt_q, buf, n, et,
2207                                   0 /* do_evt */ , SVM_Q_WAIT);
2208   else
2209     n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
2210                                    0 /* do_evt */ , SVM_Q_WAIT);
2211
2212   if (svm_fifo_set_event (s->tx_fifo))
2213     app_send_io_evt_to_vpp (
2214       s->vpp_evt_q, s->tx_fifo->shr->master_session_index, et, SVM_Q_WAIT);
2215
2216   /* The underlying fifo segment can run out of memory */
2217   if (PREDICT_FALSE (n_write < 0))
2218     return VPPCOM_EAGAIN;
2219
2220   VDBG (2, "session %u [0x%llx]: wrote %d bytes", s->session_index,
2221         s->vpp_handle, n_write);
2222
2223   return n_write;
2224 }
2225
2226 int
2227 vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
2228 {
2229   vcl_worker_t *wrk = vcl_worker_get_current ();
2230   vcl_session_t *s;
2231
2232   s = vcl_session_get_w_handle (wrk, session_handle);
2233   if (PREDICT_FALSE (!s))
2234     return VPPCOM_EBADFD;
2235
2236   return vppcom_session_write_inline (wrk, s, buf, n,
2237                                       0 /* is_flush */ , s->is_dgram ? 1 : 0);
2238 }
2239
2240 int
2241 vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
2242 {
2243   vcl_worker_t *wrk = vcl_worker_get_current ();
2244   vcl_session_t *s;
2245
2246   s = vcl_session_get_w_handle (wrk, session_handle);
2247   if (PREDICT_FALSE (!s))
2248     return VPPCOM_EBADFD;
2249
2250   return vppcom_session_write_inline (wrk, s, buf, n,
2251                                       1 /* is_flush */ , s->is_dgram ? 1 : 0);
2252 }
2253
2254 #define vcl_fifo_rx_evt_valid_or_break(_s)                              \
2255 if (PREDICT_FALSE (!_s->rx_fifo))                                       \
2256   break;                                                                \
2257 if (PREDICT_FALSE (svm_fifo_is_empty (_s->rx_fifo)))                    \
2258   {                                                                     \
2259     if (!vcl_session_is_ct (_s))                                        \
2260       {                                                                 \
2261         svm_fifo_unset_event (_s->rx_fifo);                             \
2262         if (svm_fifo_is_empty (_s->rx_fifo))                            \
2263           break;                                                        \
2264       }                                                                 \
2265     else if (svm_fifo_is_empty (_s->ct_rx_fifo))                        \
2266       {                                                                 \
2267         svm_fifo_unset_event (_s->rx_fifo); /* rx evts on actual fifo*/ \
2268         if (svm_fifo_is_empty (_s->ct_rx_fifo))                         \
2269           break;                                                        \
2270       }                                                                 \
2271   }                                                                     \
2272
2273 static void
2274 vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2275                             unsigned long n_bits, unsigned long *read_map,
2276                             unsigned long *write_map,
2277                             unsigned long *except_map, u32 * bits_set)
2278 {
2279   session_disconnected_msg_t *disconnected_msg;
2280   session_connected_msg_t *connected_msg;
2281   vcl_session_t *s;
2282   u32 sid;
2283
2284   switch (e->event_type)
2285     {
2286     case SESSION_IO_EVT_RX:
2287       sid = e->session_index;
2288       s = vcl_session_get (wrk, sid);
2289       if (!s || !vcl_session_is_open (s))
2290         break;
2291       vcl_fifo_rx_evt_valid_or_break (s);
2292       if (sid < n_bits && read_map)
2293         {
2294           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2295           *bits_set += 1;
2296         }
2297       break;
2298     case SESSION_IO_EVT_TX:
2299       sid = e->session_index;
2300       s = vcl_session_get (wrk, sid);
2301       if (!s || !vcl_session_is_open (s))
2302         break;
2303       if (sid < n_bits && write_map)
2304         {
2305           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2306           *bits_set += 1;
2307         }
2308       break;
2309     case SESSION_CTRL_EVT_ACCEPTED:
2310       if (!e->postponed)
2311         s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
2312       else
2313         s = vcl_session_get (wrk, e->session_index);
2314       if (!s)
2315         break;
2316       sid = s->session_index;
2317       if (sid < n_bits && read_map)
2318         {
2319           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2320           *bits_set += 1;
2321         }
2322       break;
2323     case SESSION_CTRL_EVT_CONNECTED:
2324       if (!e->postponed)
2325         {
2326           connected_msg = (session_connected_msg_t *) e->data;
2327           sid = vcl_session_connected_handler (wrk, connected_msg);
2328         }
2329       else
2330         sid = e->session_index;
2331       if (sid == VCL_INVALID_SESSION_INDEX)
2332         break;
2333       if (sid < n_bits && write_map)
2334         {
2335           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2336           *bits_set += 1;
2337         }
2338       break;
2339     case SESSION_CTRL_EVT_DISCONNECTED:
2340       disconnected_msg = (session_disconnected_msg_t *) e->data;
2341       s = vcl_session_disconnected_handler (wrk, disconnected_msg);
2342       if (!s)
2343         break;
2344       sid = s->session_index;
2345       if (sid < n_bits && except_map)
2346         {
2347           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2348           *bits_set += 1;
2349         }
2350       break;
2351     case SESSION_CTRL_EVT_RESET:
2352       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2353       if (sid < n_bits && except_map)
2354         {
2355           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2356           *bits_set += 1;
2357         }
2358       break;
2359     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2360       vcl_session_unlisten_reply_handler (wrk, e->data);
2361       break;
2362     case SESSION_CTRL_EVT_MIGRATED:
2363       vcl_session_migrated_handler (wrk, e->data);
2364       break;
2365     case SESSION_CTRL_EVT_CLEANUP:
2366       vcl_session_cleanup_handler (wrk, e->data);
2367       break;
2368     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2369       vcl_session_worker_update_reply_handler (wrk, e->data);
2370       break;
2371     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2372       vcl_session_req_worker_update_handler (wrk, e->data);
2373       break;
2374     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2375       vcl_session_app_add_segment_handler (wrk, e->data);
2376       break;
2377     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2378       vcl_session_app_del_segment_handler (wrk, e->data);
2379       break;
2380     case SESSION_CTRL_EVT_APP_WRK_RPC:
2381       vcl_worker_rpc_handler (wrk, e->data);
2382       break;
2383     default:
2384       clib_warning ("unhandled: %u", e->event_type);
2385       break;
2386     }
2387 }
2388
2389 static int
2390 vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2391                       unsigned long n_bits, unsigned long *read_map,
2392                       unsigned long *write_map, unsigned long *except_map,
2393                       double time_to_wait, u32 * bits_set)
2394 {
2395   svm_msg_q_msg_t *msg;
2396   session_event_t *e;
2397   u32 i;
2398
2399   if (svm_msg_q_is_empty (mq))
2400     {
2401       if (*bits_set)
2402         return 0;
2403
2404       if (!time_to_wait)
2405         return 0;
2406       else if (time_to_wait < 0)
2407         svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
2408       else
2409         {
2410           if (svm_msg_q_timedwait (mq, time_to_wait))
2411             return 0;
2412         }
2413     }
2414   vcl_mq_dequeue_batch (wrk, mq, ~0);
2415
2416   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
2417     {
2418       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
2419       e = svm_msg_q_msg_data (mq, msg);
2420       vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
2421                                   except_map, bits_set);
2422       svm_msg_q_free_msg (mq, msg);
2423     }
2424   vec_reset_length (wrk->mq_msg_vector);
2425   vcl_handle_pending_wrk_updates (wrk);
2426   return *bits_set;
2427 }
2428
2429 static int
2430 vppcom_select_condvar (vcl_worker_t * wrk, int n_bits,
2431                        vcl_si_set * read_map, vcl_si_set * write_map,
2432                        vcl_si_set * except_map, double time_to_wait,
2433                        u32 * bits_set)
2434 {
2435   double wait = 0, start = 0;
2436
2437   if (!*bits_set)
2438     {
2439       wait = time_to_wait;
2440       start = clib_time_now (&wrk->clib_time);
2441     }
2442
2443   do
2444     {
2445       vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
2446                             write_map, except_map, wait, bits_set);
2447       if (*bits_set)
2448         return *bits_set;
2449       if (wait == -1)
2450         continue;
2451
2452       wait = wait - (clib_time_now (&wrk->clib_time) - start);
2453     }
2454   while (wait > 0);
2455
2456   return 0;
2457 }
2458
2459 static int
2460 vppcom_select_eventfd (vcl_worker_t * wrk, int n_bits,
2461                        vcl_si_set * read_map, vcl_si_set * write_map,
2462                        vcl_si_set * except_map, double time_to_wait,
2463                        u32 * bits_set)
2464 {
2465   vcl_mq_evt_conn_t *mqc;
2466   int __clib_unused n_read;
2467   int n_mq_evts, i;
2468   u64 buf;
2469
2470   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
2471   n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
2472                           vec_len (wrk->mq_events), time_to_wait);
2473   for (i = 0; i < n_mq_evts; i++)
2474     {
2475       mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
2476       n_read = read (mqc->mq_fd, &buf, sizeof (buf));
2477       vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
2478                             except_map, 0, bits_set);
2479     }
2480
2481   return (n_mq_evts > 0 ? (int) *bits_set : 0);
2482 }
2483
2484 int
2485 vppcom_select (int n_bits, vcl_si_set * read_map, vcl_si_set * write_map,
2486                vcl_si_set * except_map, double time_to_wait)
2487 {
2488   u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
2489   vcl_worker_t *wrk = vcl_worker_get_current ();
2490   vcl_session_t *s = 0;
2491   int i;
2492
2493   if (n_bits && read_map)
2494     {
2495       clib_bitmap_validate (wrk->rd_bitmap, minbits);
2496       clib_memcpy_fast (wrk->rd_bitmap, read_map,
2497                         vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2498       memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2499     }
2500   if (n_bits && write_map)
2501     {
2502       clib_bitmap_validate (wrk->wr_bitmap, minbits);
2503       clib_memcpy_fast (wrk->wr_bitmap, write_map,
2504                         vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2505       memset (write_map, 0, vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2506     }
2507   if (n_bits && except_map)
2508     {
2509       clib_bitmap_validate (wrk->ex_bitmap, minbits);
2510       clib_memcpy_fast (wrk->ex_bitmap, except_map,
2511                         vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2512       memset (except_map, 0, vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2513     }
2514
2515   if (!n_bits)
2516     return 0;
2517
2518   if (!write_map)
2519     goto check_rd;
2520
2521   clib_bitmap_foreach (sid, wrk->wr_bitmap)
2522     {
2523       if (!(s = vcl_session_get (wrk, sid)))
2524         {
2525           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2526           bits_set++;
2527           continue;
2528         }
2529
2530       if (vcl_session_write_ready (s))
2531         {
2532           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2533           bits_set++;
2534         }
2535       else
2536         {
2537           svm_fifo_t *txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2538           svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF);
2539         }
2540     }
2541
2542 check_rd:
2543   if (!read_map)
2544     goto check_mq;
2545
2546   clib_bitmap_foreach (sid, wrk->rd_bitmap)
2547     {
2548       if (!(s = vcl_session_get (wrk, sid)))
2549         {
2550           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2551           bits_set++;
2552           continue;
2553         }
2554
2555       if (vcl_session_read_ready (s))
2556         {
2557           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2558           bits_set++;
2559         }
2560     }
2561
2562 check_mq:
2563
2564   for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
2565     {
2566       vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
2567                                   read_map, write_map, except_map, &bits_set);
2568     }
2569   vec_reset_length (wrk->unhandled_evts_vector);
2570
2571   if (vcm->cfg.use_mq_eventfd)
2572     vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
2573                            time_to_wait, &bits_set);
2574   else
2575     vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
2576                            time_to_wait, &bits_set);
2577
2578   return (bits_set);
2579 }
2580
2581 static inline void
2582 vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_handle)
2583 {
2584   vppcom_epoll_t *vep;
2585   u32 sh = vep_handle;
2586   vcl_session_t *s;
2587
2588   if (VPPCOM_DEBUG <= 2)
2589     return;
2590
2591   s = vcl_session_get_w_handle (wrk, vep_handle);
2592   if (PREDICT_FALSE (!s))
2593     {
2594       VDBG (0, "ERROR: Invalid vep_sh (%u)!", vep_handle);
2595       goto done;
2596     }
2597   if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP)))
2598     {
2599       VDBG (0, "ERROR: vep_sh (%u) is not a vep!", vep_handle);
2600       goto done;
2601     }
2602   vep = &s->vep;
2603   VDBG (0, "vep_sh (%u): Dumping epoll chain\n"
2604         "{\n"
2605         "   is_vep         = %u\n"
2606         "   is_vep_session = %u\n"
2607         "   next_sh        = 0x%x (%u)\n"
2608         "}\n", vep_handle, s->flags & VCL_SESSION_F_IS_VEP,
2609         s->flags & VCL_SESSION_F_IS_VEP_SESSION, vep->next_sh, vep->next_sh);
2610
2611   for (sh = vep->next_sh; sh != ~0; sh = vep->next_sh)
2612     {
2613       s = vcl_session_get_w_handle (wrk, sh);
2614       if (PREDICT_FALSE (!s))
2615         {
2616           VDBG (0, "ERROR: Invalid sh (%u)!", sh);
2617           goto done;
2618         }
2619       if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2620         {
2621           VDBG (0, "ERROR: sh (%u) is a vep!", vep_handle);
2622         }
2623       else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2624         {
2625           VDBG (0, "ERROR: sh (%u) is not a vep session handle!", sh);
2626           goto done;
2627         }
2628       vep = &s->vep;
2629       if (PREDICT_FALSE (vep->vep_sh != vep_handle))
2630         VDBG (0, "ERROR: session (%u) vep_sh (%u) != vep_sh (%u)!",
2631               sh, s->vep.vep_sh, vep_handle);
2632       if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
2633         {
2634           VDBG (0, "vep_sh[%u]: sh 0x%x (%u)\n"
2635                 "{\n"
2636                 "   next_sh        = 0x%x (%u)\n"
2637                 "   prev_sh        = 0x%x (%u)\n"
2638                 "   vep_sh         = 0x%x (%u)\n"
2639                 "   ev.events      = 0x%x\n"
2640                 "   ev.data.u64    = 0x%llx\n"
2641                 "   et_mask        = 0x%x\n"
2642                 "}\n",
2643                 vep_handle, sh, sh, vep->next_sh, vep->next_sh, vep->prev_sh,
2644                 vep->prev_sh, vep->vep_sh, vep->vep_sh, vep->ev.events,
2645                 vep->ev.data.u64, vep->et_mask);
2646         }
2647     }
2648
2649 done:
2650   VDBG (0, "vep_sh (%u): Dump complete!\n", vep_handle);
2651 }
2652
2653 int
2654 vppcom_epoll_create (void)
2655 {
2656   vcl_worker_t *wrk = vcl_worker_get_current ();
2657   vcl_session_t *vep_session;
2658
2659   vep_session = vcl_session_alloc (wrk);
2660
2661   vep_session->flags |= VCL_SESSION_F_IS_VEP;
2662   vep_session->vep.vep_sh = ~0;
2663   vep_session->vep.next_sh = ~0;
2664   vep_session->vep.prev_sh = ~0;
2665   vep_session->vpp_handle = ~0;
2666
2667   vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
2668   VDBG (0, "Created vep_idx %u", vep_session->session_index);
2669
2670   return vcl_session_handle (vep_session);
2671 }
2672
2673 int
2674 vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
2675                   struct epoll_event *event)
2676 {
2677   vcl_worker_t *wrk = vcl_worker_get_current ();
2678   vcl_session_t *vep_session;
2679   int rv = VPPCOM_OK;
2680   vcl_session_t *s;
2681   svm_fifo_t *txf;
2682
2683   if (vep_handle == session_handle)
2684     {
2685       VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
2686       return VPPCOM_EINVAL;
2687     }
2688
2689   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
2690   if (PREDICT_FALSE (!vep_session))
2691     {
2692       VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
2693       return VPPCOM_EBADFD;
2694     }
2695   if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
2696     {
2697       VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
2698       return VPPCOM_EINVAL;
2699     }
2700
2701   ASSERT (vep_session->vep.vep_sh == ~0);
2702   ASSERT (vep_session->vep.prev_sh == ~0);
2703
2704   s = vcl_session_get_w_handle (wrk, session_handle);
2705   if (PREDICT_FALSE (!s))
2706     {
2707       VDBG (0, "Invalid session_handle (%u)!", session_handle);
2708       return VPPCOM_EBADFD;
2709     }
2710   if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2711     {
2712       VDBG (0, "session_handle (%u) is a vep!", vep_handle);
2713       return VPPCOM_EINVAL;
2714     }
2715
2716   switch (op)
2717     {
2718     case EPOLL_CTL_ADD:
2719       if (PREDICT_FALSE (!event))
2720         {
2721           VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
2722           return VPPCOM_EINVAL;
2723         }
2724       if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
2725         {
2726           VDBG (0, "EPOLL_CTL_ADD: %u already epolled!", s->session_index);
2727           rv = VPPCOM_EEXIST;
2728           goto done;
2729         }
2730       if (vep_session->vep.next_sh != ~0)
2731         {
2732           vcl_session_t *next_session;
2733           next_session = vcl_session_get_w_handle (wrk,
2734                                                    vep_session->vep.next_sh);
2735           if (PREDICT_FALSE (!next_session))
2736             {
2737               VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
2738                     "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
2739               return VPPCOM_EBADFD;
2740             }
2741           ASSERT (next_session->vep.prev_sh == vep_handle);
2742           next_session->vep.prev_sh = session_handle;
2743         }
2744       s->vep.next_sh = vep_session->vep.next_sh;
2745       s->vep.prev_sh = vep_handle;
2746       s->vep.vep_sh = vep_handle;
2747       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2748       s->vep.ev = *event;
2749       s->flags &= ~VCL_SESSION_F_IS_VEP;
2750       s->flags |= VCL_SESSION_F_IS_VEP_SESSION;
2751       vep_session->vep.next_sh = session_handle;
2752
2753       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2754       if (txf && (event->events & EPOLLOUT))
2755         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2756
2757       /* Generate EPOLLOUT if tx fifo not full */
2758       if ((event->events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2759         {
2760           session_event_t e = { 0 };
2761           e.event_type = SESSION_IO_EVT_TX;
2762           e.session_index = s->session_index;
2763           vec_add1 (wrk->unhandled_evts_vector, e);
2764         }
2765       /* Generate EPOLLIN if rx fifo has data */
2766       if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
2767         {
2768           session_event_t e = { 0 };
2769           e.event_type = SESSION_IO_EVT_RX;
2770           e.session_index = s->session_index;
2771           vec_add1 (wrk->unhandled_evts_vector, e);
2772         }
2773       VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2774             vep_handle, session_handle, event->events, event->data.u64);
2775       vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
2776       break;
2777
2778     case EPOLL_CTL_MOD:
2779       if (PREDICT_FALSE (!event))
2780         {
2781           VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
2782           rv = VPPCOM_EINVAL;
2783           goto done;
2784         }
2785       else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2786         {
2787           VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
2788           rv = VPPCOM_ENOENT;
2789           goto done;
2790         }
2791       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2792         {
2793           VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
2794                 session_handle, s->vep.vep_sh, vep_handle);
2795           rv = VPPCOM_EINVAL;
2796           goto done;
2797         }
2798
2799       /* Generate EPOLLOUT if session write ready nd event was not on */
2800       if ((event->events & EPOLLOUT) && !(s->vep.ev.events & EPOLLOUT) &&
2801           (vcl_session_write_ready (s) > 0))
2802         {
2803           session_event_t e = { 0 };
2804           e.event_type = SESSION_IO_EVT_TX;
2805           e.session_index = s->session_index;
2806           vec_add1 (wrk->unhandled_evts_vector, e);
2807         }
2808       /* Generate EPOLLIN if session read ready and event was not on */
2809       if ((event->events & EPOLLIN) && !(s->vep.ev.events & EPOLLIN) &&
2810           (vcl_session_read_ready (s) > 0))
2811         {
2812           session_event_t e = { 0 };
2813           e.event_type = SESSION_IO_EVT_RX;
2814           e.session_index = s->session_index;
2815           vec_add1 (wrk->unhandled_evts_vector, e);
2816         }
2817       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2818       s->vep.ev = *event;
2819       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2820       if (txf)
2821         {
2822           if (event->events & EPOLLOUT)
2823             svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2824           else
2825             svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2826         }
2827       VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2828             vep_handle, session_handle, event->events, event->data.u64);
2829       break;
2830
2831     case EPOLL_CTL_DEL:
2832       if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2833         {
2834           VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
2835           rv = VPPCOM_ENOENT;
2836           goto done;
2837         }
2838       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2839         {
2840           VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
2841                 session_handle, s->vep.vep_sh, vep_handle);
2842           rv = VPPCOM_EINVAL;
2843           goto done;
2844         }
2845
2846       if (s->vep.prev_sh == vep_handle)
2847         vep_session->vep.next_sh = s->vep.next_sh;
2848       else
2849         {
2850           vcl_session_t *prev_session;
2851           prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
2852           if (PREDICT_FALSE (!prev_session))
2853             {
2854               VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
2855                     s->vep.prev_sh, session_handle);
2856               return VPPCOM_EBADFD;
2857             }
2858           ASSERT (prev_session->vep.next_sh == session_handle);
2859           prev_session->vep.next_sh = s->vep.next_sh;
2860         }
2861       if (s->vep.next_sh != ~0)
2862         {
2863           vcl_session_t *next_session;
2864           next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
2865           if (PREDICT_FALSE (!next_session))
2866             {
2867               VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
2868                     s->vep.next_sh, session_handle);
2869               return VPPCOM_EBADFD;
2870             }
2871           ASSERT (next_session->vep.prev_sh == session_handle);
2872           next_session->vep.prev_sh = s->vep.prev_sh;
2873         }
2874
2875       memset (&s->vep, 0, sizeof (s->vep));
2876       s->vep.next_sh = ~0;
2877       s->vep.prev_sh = ~0;
2878       s->vep.vep_sh = ~0;
2879       s->flags &= ~VCL_SESSION_F_IS_VEP_SESSION;
2880
2881       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2882       if (txf)
2883         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2884
2885       VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
2886             session_handle);
2887       vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
2888       break;
2889
2890     default:
2891       VDBG (0, "Invalid operation (%d)!", op);
2892       rv = VPPCOM_EINVAL;
2893     }
2894
2895   vep_verify_epoll_chain (wrk, vep_handle);
2896
2897 done:
2898   return rv;
2899 }
2900
2901 static inline void
2902 vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2903                                 struct epoll_event *events, u32 * num_ev)
2904 {
2905   session_disconnected_msg_t *disconnected_msg;
2906   session_connected_msg_t *connected_msg;
2907   u32 sid = ~0, session_events;
2908   u64 session_evt_data = ~0;
2909   vcl_session_t *s;
2910   u8 add_event = 0;
2911
2912   switch (e->event_type)
2913     {
2914     case SESSION_IO_EVT_RX:
2915       sid = e->session_index;
2916       s = vcl_session_get (wrk, sid);
2917       if (vcl_session_is_closed (s))
2918         break;
2919       vcl_fifo_rx_evt_valid_or_break (s);
2920       session_events = s->vep.ev.events;
2921       if (!(EPOLLIN & s->vep.ev.events)
2922           || (s->flags & VCL_SESSION_F_HAS_RX_EVT))
2923         break;
2924       add_event = 1;
2925       events[*num_ev].events |= EPOLLIN;
2926       session_evt_data = s->vep.ev.data.u64;
2927       s->flags |= VCL_SESSION_F_HAS_RX_EVT;
2928       break;
2929     case SESSION_IO_EVT_TX:
2930       sid = e->session_index;
2931       s = vcl_session_get (wrk, sid);
2932       if (vcl_session_is_closed (s))
2933         break;
2934       session_events = s->vep.ev.events;
2935       if (!(EPOLLOUT & session_events))
2936         break;
2937       add_event = 1;
2938       events[*num_ev].events |= EPOLLOUT;
2939       session_evt_data = s->vep.ev.data.u64;
2940       svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (s) ?
2941                                   s->ct_tx_fifo : s->tx_fifo);
2942       break;
2943     case SESSION_CTRL_EVT_ACCEPTED:
2944       if (!e->postponed)
2945         s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
2946       else
2947         s = vcl_session_get (wrk, e->session_index);
2948       if (!s)
2949         break;
2950       session_events = s->vep.ev.events;
2951       sid = s->session_index;
2952       if (!(EPOLLIN & session_events))
2953         break;
2954       add_event = 1;
2955       events[*num_ev].events |= EPOLLIN;
2956       session_evt_data = s->vep.ev.data.u64;
2957       break;
2958     case SESSION_CTRL_EVT_CONNECTED:
2959       if (!e->postponed)
2960         {
2961           connected_msg = (session_connected_msg_t *) e->data;
2962           sid = vcl_session_connected_handler (wrk, connected_msg);
2963         }
2964       else
2965         sid = e->session_index;
2966       s = vcl_session_get (wrk, sid);
2967       if (vcl_session_is_closed (s))
2968         break;
2969       session_events = s->vep.ev.events;
2970       /* Generate EPOLLOUT because there's no connected event */
2971       if (!(EPOLLOUT & session_events))
2972         break;
2973       add_event = 1;
2974       events[*num_ev].events |= EPOLLOUT;
2975       session_evt_data = s->vep.ev.data.u64;
2976       if (s->session_state == VCL_STATE_DETACHED)
2977         events[*num_ev].events |= EPOLLHUP;
2978       break;
2979     case SESSION_CTRL_EVT_DISCONNECTED:
2980       disconnected_msg = (session_disconnected_msg_t *) e->data;
2981       s = vcl_session_disconnected_handler (wrk, disconnected_msg);
2982       if (vcl_session_is_closed (s))
2983         break;
2984       sid = s->session_index;
2985       session_events = s->vep.ev.events;
2986       add_event = 1;
2987       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2988       session_evt_data = s->vep.ev.data.u64;
2989       break;
2990     case SESSION_CTRL_EVT_RESET:
2991       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2992       s = vcl_session_get (wrk, sid);
2993       if (vcl_session_is_closed (s))
2994         break;
2995       session_events = s->vep.ev.events;
2996       add_event = 1;
2997       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2998       session_evt_data = s->vep.ev.data.u64;
2999       break;
3000     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
3001       vcl_session_unlisten_reply_handler (wrk, e->data);
3002       break;
3003     case SESSION_CTRL_EVT_MIGRATED:
3004       vcl_session_migrated_handler (wrk, e->data);
3005       break;
3006     case SESSION_CTRL_EVT_CLEANUP:
3007       vcl_session_cleanup_handler (wrk, e->data);
3008       break;
3009     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
3010       vcl_session_req_worker_update_handler (wrk, e->data);
3011       break;
3012     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
3013       vcl_session_worker_update_reply_handler (wrk, e->data);
3014       break;
3015     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
3016       vcl_session_app_add_segment_handler (wrk, e->data);
3017       break;
3018     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
3019       vcl_session_app_del_segment_handler (wrk, e->data);
3020       break;
3021     case SESSION_CTRL_EVT_APP_WRK_RPC:
3022       vcl_worker_rpc_handler (wrk, e->data);
3023       break;
3024     default:
3025       VDBG (0, "unhandled: %u", e->event_type);
3026       break;
3027     }
3028
3029   if (add_event)
3030     {
3031       events[*num_ev].data.u64 = session_evt_data;
3032       if (EPOLLONESHOT & session_events)
3033         {
3034           s = vcl_session_get (wrk, sid);
3035           s->vep.ev.events = 0;
3036         }
3037       *num_ev += 1;
3038     }
3039 }
3040
3041 static int
3042 vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
3043                           struct epoll_event *events, u32 maxevents,
3044                           double wait_for_time, u32 * num_ev)
3045 {
3046   svm_msg_q_msg_t *msg;
3047   session_event_t *e;
3048   int i;
3049
3050   if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
3051     goto handle_dequeued;
3052
3053   if (svm_msg_q_is_empty (mq))
3054     {
3055       if (!wait_for_time)
3056         return 0;
3057       else if (wait_for_time < 0)
3058         svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
3059       else
3060         {
3061           if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
3062             return 0;
3063         }
3064     }
3065   ASSERT (maxevents > *num_ev);
3066   vcl_mq_dequeue_batch (wrk, mq, ~0);
3067
3068 handle_dequeued:
3069   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
3070     {
3071       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
3072       e = svm_msg_q_msg_data (mq, msg);
3073       if (*num_ev < maxevents)
3074         vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
3075       else
3076         vcl_handle_mq_event (wrk, e);
3077       svm_msg_q_free_msg (mq, msg);
3078     }
3079   vec_reset_length (wrk->mq_msg_vector);
3080   vcl_handle_pending_wrk_updates (wrk);
3081   return *num_ev;
3082 }
3083
3084 static int
3085 vppcom_epoll_wait_condvar (vcl_worker_t *wrk, struct epoll_event *events,
3086                            int maxevents, u32 n_evts, double timeout_ms)
3087 {
3088   double end = -1;
3089
3090   if (!n_evts)
3091     {
3092       if (timeout_ms > 0)
3093         end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
3094     }
3095
3096   do
3097     {
3098       vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
3099                                 timeout_ms, &n_evts);
3100       if (n_evts || !timeout_ms)
3101         return n_evts;
3102     }
3103   while (end == -1 || clib_time_now (&wrk->clib_time) < end);
3104
3105   return 0;
3106 }
3107
3108 static int
3109 vppcom_epoll_wait_eventfd (vcl_worker_t *wrk, struct epoll_event *events,
3110                            int maxevents, u32 n_evts, double timeout_ms)
3111 {
3112   int __clib_unused n_read;
3113   vcl_mq_evt_conn_t *mqc;
3114   int n_mq_evts, i;
3115   double end = -1;
3116   u64 buf;
3117
3118   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
3119   if (!n_evts)
3120     {
3121       if (timeout_ms > 0)
3122         end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
3123     }
3124
3125   do
3126     {
3127       n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
3128                               vec_len (wrk->mq_events), timeout_ms);
3129       if (n_mq_evts < 0)
3130         {
3131           VDBG (0, "epoll_wait error %u", errno);
3132           return n_evts;
3133         }
3134
3135       for (i = 0; i < n_mq_evts; i++)
3136         {
3137           mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
3138           n_read = read (mqc->mq_fd, &buf, sizeof (buf));
3139           vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0,
3140                                     &n_evts);
3141         }
3142
3143       if (n_evts || !timeout_ms)
3144         return n_evts;
3145     }
3146   while (end == -1 || clib_time_now (&wrk->clib_time) < end);
3147
3148   return 0;
3149 }
3150
3151 int
3152 vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
3153                    int maxevents, double wait_for_time)
3154 {
3155   vcl_worker_t *wrk = vcl_worker_get_current ();
3156   vcl_session_t *vep_session;
3157   u32 n_evts = 0;
3158   int i;
3159
3160   if (PREDICT_FALSE (maxevents <= 0))
3161     {
3162       VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
3163       return VPPCOM_EINVAL;
3164     }
3165
3166   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
3167   if (!vep_session)
3168     return VPPCOM_EBADFD;
3169
3170   if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
3171     {
3172       VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
3173       return VPPCOM_EINVAL;
3174     }
3175
3176   memset (events, 0, sizeof (*events) * maxevents);
3177
3178   if (vec_len (wrk->unhandled_evts_vector))
3179     {
3180       for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3181         {
3182           vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3183                                           events, &n_evts);
3184           if (n_evts == maxevents)
3185             {
3186               vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3187               return n_evts;
3188             }
3189         }
3190       vec_reset_length (wrk->unhandled_evts_vector);
3191     }
3192   /* Request to only drain unhandled */
3193   if ((int) wait_for_time == -2)
3194     return n_evts;
3195
3196   if (vcm->cfg.use_mq_eventfd)
3197     return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3198                                       wait_for_time);
3199
3200   return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3201                                     wait_for_time);
3202 }
3203
3204 int
3205 vppcom_session_attr (uint32_t session_handle, uint32_t op,
3206                      void *buffer, uint32_t * buflen)
3207 {
3208   vcl_worker_t *wrk = vcl_worker_get_current ();
3209   u32 *flags = buffer, tmp_flags = 0;
3210   vppcom_endpt_t *ep = buffer;
3211   transport_endpt_attr_t tea;
3212   vcl_session_t *session;
3213   int rv = VPPCOM_OK;
3214
3215   session = vcl_session_get_w_handle (wrk, session_handle);
3216   if (!session)
3217     return VPPCOM_EBADFD;
3218
3219   switch (op)
3220     {
3221     case VPPCOM_ATTR_GET_NREAD:
3222       rv = vcl_session_read_ready (session);
3223       VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3224             rv);
3225       break;
3226
3227     case VPPCOM_ATTR_GET_NWRITE:
3228       rv = vcl_session_write_ready (session);
3229       VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3230             rv);
3231       break;
3232
3233     case VPPCOM_ATTR_GET_FLAGS:
3234       if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
3235         {
3236           *flags =
3237             O_RDWR |
3238             (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK) ?
3239              O_NONBLOCK : 0);
3240           *buflen = sizeof (*flags);
3241           VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3242                 "is_nonblocking = %u", session_handle, *flags,
3243                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3244         }
3245       else
3246         rv = VPPCOM_EINVAL;
3247       break;
3248
3249     case VPPCOM_ATTR_SET_FLAGS:
3250       if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
3251         {
3252           if (*flags & O_NONBLOCK)
3253             vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
3254           else
3255             vcl_session_clear_attr (session, VCL_SESS_ATTR_NONBLOCK);
3256
3257           VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3258                 " is_nonblocking = %u", session_handle, *flags,
3259                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3260         }
3261       else
3262         rv = VPPCOM_EINVAL;
3263       break;
3264
3265     case VPPCOM_ATTR_GET_PEER_ADDR:
3266       if (PREDICT_TRUE (buffer && buflen &&
3267                         (*buflen >= sizeof (*ep)) && ep->ip))
3268         {
3269           ep->is_ip4 = session->transport.is_ip4;
3270           ep->port = session->transport.rmt_port;
3271           if (session->transport.is_ip4)
3272             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3273                               sizeof (ip4_address_t));
3274           else
3275             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3276                               sizeof (ip6_address_t));
3277           *buflen = sizeof (*ep);
3278           VDBG (1, "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3279                 "addr = %U, port %u", session_handle, ep->is_ip4,
3280                 format_ip46_address, &session->transport.rmt_ip,
3281                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3282                 clib_net_to_host_u16 (ep->port));
3283         }
3284       else
3285         rv = VPPCOM_EINVAL;
3286       break;
3287
3288     case VPPCOM_ATTR_GET_LCL_ADDR:
3289       if (PREDICT_TRUE (buffer && buflen &&
3290                         (*buflen >= sizeof (*ep)) && ep->ip))
3291         {
3292           ep->is_ip4 = session->transport.is_ip4;
3293           ep->port = session->transport.lcl_port;
3294           if (session->transport.is_ip4)
3295             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3296                               sizeof (ip4_address_t));
3297           else
3298             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3299                               sizeof (ip6_address_t));
3300           *buflen = sizeof (*ep);
3301           VDBG (1, "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3302                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3303                 &session->transport.lcl_ip,
3304                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3305                 clib_net_to_host_u16 (ep->port));
3306         }
3307       else
3308         rv = VPPCOM_EINVAL;
3309       break;
3310
3311     case VPPCOM_ATTR_SET_LCL_ADDR:
3312       if (PREDICT_TRUE (buffer && buflen &&
3313                         (*buflen >= sizeof (*ep)) && ep->ip))
3314         {
3315           session->transport.is_ip4 = ep->is_ip4;
3316           session->transport.lcl_port = ep->port;
3317           vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3318           *buflen = sizeof (*ep);
3319           VDBG (1, "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3320                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3321                 &session->transport.lcl_ip,
3322                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3323                 clib_net_to_host_u16 (ep->port));
3324         }
3325       else
3326         rv = VPPCOM_EINVAL;
3327       break;
3328
3329     case VPPCOM_ATTR_GET_LIBC_EPFD:
3330       rv = session->libc_epfd;
3331       VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
3332       break;
3333
3334     case VPPCOM_ATTR_SET_LIBC_EPFD:
3335       if (PREDICT_TRUE (buffer && buflen &&
3336                         (*buflen == sizeof (session->libc_epfd))))
3337         {
3338           session->libc_epfd = *(int *) buffer;
3339           *buflen = sizeof (session->libc_epfd);
3340
3341           VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3342                 session->libc_epfd, *buflen);
3343         }
3344       else
3345         rv = VPPCOM_EINVAL;
3346       break;
3347
3348     case VPPCOM_ATTR_GET_PROTOCOL:
3349       if (buffer && buflen && (*buflen >= sizeof (int)))
3350         {
3351           *(int *) buffer = session->session_type;
3352           *buflen = sizeof (int);
3353
3354           VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3355                 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
3356         }
3357       else
3358         rv = VPPCOM_EINVAL;
3359       break;
3360
3361     case VPPCOM_ATTR_GET_LISTEN:
3362       if (buffer && buflen && (*buflen >= sizeof (int)))
3363         {
3364           *(int *) buffer = vcl_session_has_attr (session,
3365                                                   VCL_SESS_ATTR_LISTEN);
3366           *buflen = sizeof (int);
3367
3368           VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3369                 *buflen);
3370         }
3371       else
3372         rv = VPPCOM_EINVAL;
3373       break;
3374
3375     case VPPCOM_ATTR_GET_ERROR:
3376       if (buffer && buflen && (*buflen >= sizeof (int)))
3377         {
3378           *(int *) buffer = 0;
3379           *buflen = sizeof (int);
3380
3381           VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3382                 *(int *) buffer, *buflen);
3383         }
3384       else
3385         rv = VPPCOM_EINVAL;
3386       break;
3387
3388     case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3389       if (buffer && buflen && (*buflen >= sizeof (u32)))
3390         {
3391
3392           /* VPP-TBD */
3393           *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
3394                                 session->tx_fifo ?
3395                                 svm_fifo_size (session->tx_fifo) :
3396                                 vcm->cfg.tx_fifo_size);
3397           *buflen = sizeof (u32);
3398
3399           VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3400                 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3401                 *buflen);
3402         }
3403       else
3404         rv = VPPCOM_EINVAL;
3405       break;
3406
3407     case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3408       if (buffer && buflen && (*buflen == sizeof (u32)))
3409         {
3410           /* VPP-TBD */
3411           session->sndbuf_size = *(u32 *) buffer;
3412           VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3413                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3414                 *buflen);
3415         }
3416       else
3417         rv = VPPCOM_EINVAL;
3418       break;
3419
3420     case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3421       if (buffer && buflen && (*buflen >= sizeof (u32)))
3422         {
3423
3424           /* VPP-TBD */
3425           *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
3426                                 session->rx_fifo ?
3427                                 svm_fifo_size (session->rx_fifo) :
3428                                 vcm->cfg.rx_fifo_size);
3429           *buflen = sizeof (u32);
3430
3431           VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3432                 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
3433         }
3434       else
3435         rv = VPPCOM_EINVAL;
3436       break;
3437
3438     case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3439       if (buffer && buflen && (*buflen == sizeof (u32)))
3440         {
3441           /* VPP-TBD */
3442           session->rcvbuf_size = *(u32 *) buffer;
3443           VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3444                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3445                 *buflen);
3446         }
3447       else
3448         rv = VPPCOM_EINVAL;
3449       break;
3450
3451     case VPPCOM_ATTR_GET_REUSEADDR:
3452       if (buffer && buflen && (*buflen >= sizeof (int)))
3453         {
3454           /* VPP-TBD */
3455           *(int *) buffer = vcl_session_has_attr (session,
3456                                                   VCL_SESS_ATTR_REUSEADDR);
3457           *buflen = sizeof (int);
3458
3459           VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3460                 *(int *) buffer, *buflen);
3461         }
3462       else
3463         rv = VPPCOM_EINVAL;
3464       break;
3465
3466     case VPPCOM_ATTR_SET_REUSEADDR:
3467       if (buffer && buflen && (*buflen == sizeof (int)) &&
3468           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3469         {
3470           /* VPP-TBD */
3471           if (*(int *) buffer)
3472             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEADDR);
3473           else
3474             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEADDR);
3475
3476           VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3477                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEADDR),
3478                 *buflen);
3479         }
3480       else
3481         rv = VPPCOM_EINVAL;
3482       break;
3483
3484     case VPPCOM_ATTR_GET_REUSEPORT:
3485       if (buffer && buflen && (*buflen >= sizeof (int)))
3486         {
3487           /* VPP-TBD */
3488           *(int *) buffer = vcl_session_has_attr (session,
3489                                                   VCL_SESS_ATTR_REUSEPORT);
3490           *buflen = sizeof (int);
3491
3492           VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3493                 *(int *) buffer, *buflen);
3494         }
3495       else
3496         rv = VPPCOM_EINVAL;
3497       break;
3498
3499     case VPPCOM_ATTR_SET_REUSEPORT:
3500       if (buffer && buflen && (*buflen == sizeof (int)) &&
3501           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3502         {
3503           /* VPP-TBD */
3504           if (*(int *) buffer)
3505             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEPORT);
3506           else
3507             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEPORT);
3508
3509           VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3510                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEPORT),
3511                 *buflen);
3512         }
3513       else
3514         rv = VPPCOM_EINVAL;
3515       break;
3516
3517     case VPPCOM_ATTR_GET_BROADCAST:
3518       if (buffer && buflen && (*buflen >= sizeof (int)))
3519         {
3520           /* VPP-TBD */
3521           *(int *) buffer = vcl_session_has_attr (session,
3522                                                   VCL_SESS_ATTR_BROADCAST);
3523           *buflen = sizeof (int);
3524
3525           VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3526                 *(int *) buffer, *buflen);
3527         }
3528       else
3529         rv = VPPCOM_EINVAL;
3530       break;
3531
3532     case VPPCOM_ATTR_SET_BROADCAST:
3533       if (buffer && buflen && (*buflen == sizeof (int)))
3534         {
3535           /* VPP-TBD */
3536           if (*(int *) buffer)
3537             vcl_session_set_attr (session, VCL_SESS_ATTR_BROADCAST);
3538           else
3539             vcl_session_clear_attr (session, VCL_SESS_ATTR_BROADCAST);
3540
3541           VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3542                 vcl_session_has_attr (session, VCL_SESS_ATTR_BROADCAST),
3543                 *buflen);
3544         }
3545       else
3546         rv = VPPCOM_EINVAL;
3547       break;
3548
3549     case VPPCOM_ATTR_GET_V6ONLY:
3550       if (buffer && buflen && (*buflen >= sizeof (int)))
3551         {
3552           /* VPP-TBD */
3553           *(int *) buffer = vcl_session_has_attr (session,
3554                                                   VCL_SESS_ATTR_V6ONLY);
3555           *buflen = sizeof (int);
3556
3557           VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3558                 *(int *) buffer, *buflen);
3559         }
3560       else
3561         rv = VPPCOM_EINVAL;
3562       break;
3563
3564     case VPPCOM_ATTR_SET_V6ONLY:
3565       if (buffer && buflen && (*buflen == sizeof (int)))
3566         {
3567           /* VPP-TBD */
3568           if (*(int *) buffer)
3569             vcl_session_set_attr (session, VCL_SESS_ATTR_V6ONLY);
3570           else
3571             vcl_session_clear_attr (session, VCL_SESS_ATTR_V6ONLY);
3572
3573           VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3574                 vcl_session_has_attr (session, VCL_SESS_ATTR_V6ONLY),
3575                 *buflen);
3576         }
3577       else
3578         rv = VPPCOM_EINVAL;
3579       break;
3580
3581     case VPPCOM_ATTR_GET_KEEPALIVE:
3582       if (buffer && buflen && (*buflen >= sizeof (int)))
3583         {
3584           /* VPP-TBD */
3585           *(int *) buffer = vcl_session_has_attr (session,
3586                                                   VCL_SESS_ATTR_KEEPALIVE);
3587           *buflen = sizeof (int);
3588
3589           VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3590                 *(int *) buffer, *buflen);
3591         }
3592       else
3593         rv = VPPCOM_EINVAL;
3594       break;
3595
3596     case VPPCOM_ATTR_SET_KEEPALIVE:
3597       if (buffer && buflen && (*buflen == sizeof (int)))
3598         {
3599           /* VPP-TBD */
3600           if (*(int *) buffer)
3601             vcl_session_set_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3602           else
3603             vcl_session_clear_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3604
3605           VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3606                 vcl_session_has_attr (session, VCL_SESS_ATTR_KEEPALIVE),
3607                 *buflen);
3608         }
3609       else
3610         rv = VPPCOM_EINVAL;
3611       break;
3612
3613     case VPPCOM_ATTR_GET_TCP_NODELAY:
3614       if (buffer && buflen && (*buflen >= sizeof (int)))
3615         {
3616           /* VPP-TBD */
3617           *(int *) buffer = vcl_session_has_attr (session,
3618                                                   VCL_SESS_ATTR_TCP_NODELAY);
3619           *buflen = sizeof (int);
3620
3621           VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3622                 *(int *) buffer, *buflen);
3623         }
3624       else
3625         rv = VPPCOM_EINVAL;
3626       break;
3627
3628     case VPPCOM_ATTR_SET_TCP_NODELAY:
3629       if (buffer && buflen && (*buflen == sizeof (int)))
3630         {
3631           /* VPP-TBD */
3632           if (*(int *) buffer)
3633             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3634           else
3635             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3636
3637           VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3638                 vcl_session_has_attr (session, VCL_SESS_ATTR_TCP_NODELAY),
3639                 *buflen);
3640         }
3641       else
3642         rv = VPPCOM_EINVAL;
3643       break;
3644
3645     case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
3646       if (buffer && buflen && (*buflen >= sizeof (int)))
3647         {
3648           /* VPP-TBD */
3649           *(int *) buffer = vcl_session_has_attr (session,
3650                                                   VCL_SESS_ATTR_TCP_KEEPIDLE);
3651           *buflen = sizeof (int);
3652
3653           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3654                 *(int *) buffer, *buflen);
3655         }
3656       else
3657         rv = VPPCOM_EINVAL;
3658       break;
3659
3660     case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
3661       if (buffer && buflen && (*buflen == sizeof (int)))
3662         {
3663           /* VPP-TBD */
3664           if (*(int *) buffer)
3665             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3666           else
3667             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3668
3669           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3670                 vcl_session_has_attr (session,
3671                                       VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
3672         }
3673       else
3674         rv = VPPCOM_EINVAL;
3675       break;
3676
3677     case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
3678       if (buffer && buflen && (*buflen >= sizeof (int)))
3679         {
3680           /* VPP-TBD */
3681           *(int *) buffer = vcl_session_has_attr (session,
3682                                                   VCL_SESS_ATTR_TCP_KEEPINTVL);
3683           *buflen = sizeof (int);
3684
3685           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3686                 *(int *) buffer, *buflen);
3687         }
3688       else
3689         rv = VPPCOM_EINVAL;
3690       break;
3691
3692     case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
3693       if (buffer && buflen && (*buflen == sizeof (int)))
3694         {
3695           /* VPP-TBD */
3696           if (*(int *) buffer)
3697             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3698           else
3699             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3700
3701           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3702                 vcl_session_has_attr (session,
3703                                       VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
3704         }
3705       else
3706         rv = VPPCOM_EINVAL;
3707       break;
3708
3709     case VPPCOM_ATTR_GET_TCP_USER_MSS:
3710       if (!(buffer && buflen && (*buflen >= sizeof (u32))))
3711         {
3712           rv = VPPCOM_EINVAL;
3713           break;
3714         }
3715
3716       tea.type = TRANSPORT_ENDPT_ATTR_MSS;
3717       tea.mss = *(u32 *) buffer;
3718       if (vcl_session_transport_attr (wrk, session, 1 /* is_get */, &tea))
3719         rv = VPPCOM_ENOPROTOOPT;
3720
3721       if (!rv)
3722         {
3723           *(u32 *) buffer = tea.mss;
3724           *buflen = sizeof (int);
3725         }
3726
3727       VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d", *(int *) buffer,
3728             *buflen);
3729       break;
3730
3731     case VPPCOM_ATTR_SET_TCP_USER_MSS:
3732       if (!(buffer && buflen && (*buflen == sizeof (u32))))
3733         {
3734           rv = VPPCOM_EINVAL;
3735           break;
3736         }
3737
3738       tea.type = TRANSPORT_ENDPT_ATTR_MSS;
3739       tea.mss = *(u32 *) buffer;
3740       if (vcl_session_transport_attr (wrk, session, 0 /* is_get */, &tea))
3741         rv = VPPCOM_ENOPROTOOPT;
3742
3743       VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d", tea.mss,
3744             *buflen);
3745       break;
3746
3747     case VPPCOM_ATTR_SET_SHUT:
3748       if (*flags == SHUT_RD || *flags == SHUT_RDWR)
3749         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_RD);
3750       if (*flags == SHUT_WR || *flags == SHUT_RDWR)
3751         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_WR);
3752       break;
3753
3754     case VPPCOM_ATTR_GET_SHUT:
3755       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_RD))
3756         tmp_flags = 1;
3757       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_WR))
3758         tmp_flags |= 2;
3759       if (tmp_flags == 1)
3760         *(int *) buffer = SHUT_RD;
3761       else if (tmp_flags == 2)
3762         *(int *) buffer = SHUT_WR;
3763       else if (tmp_flags == 3)
3764         *(int *) buffer = SHUT_RDWR;
3765       *buflen = sizeof (int);
3766       break;
3767
3768     case VPPCOM_ATTR_SET_CONNECTED:
3769       session->flags |= VCL_SESSION_F_CONNECTED;
3770       break;
3771
3772     case VPPCOM_ATTR_SET_CKPAIR:
3773       if (!(buffer && buflen && (*buflen == sizeof (int))) ||
3774           !vcl_session_has_crypto (session))
3775         {
3776           rv = VPPCOM_EINVAL;
3777           break;
3778         }
3779       if (!session->ext_config)
3780         {
3781           vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_CRYPTO,
3782                                      sizeof (transport_endpt_ext_cfg_t));
3783         }
3784       else if (session->ext_config->type != TRANSPORT_ENDPT_EXT_CFG_CRYPTO)
3785         {
3786           rv = VPPCOM_EINVAL;
3787           break;
3788         }
3789
3790       session->ext_config->crypto.ckpair_index = *(uint32_t *) buffer;
3791       break;
3792
3793     case VPPCOM_ATTR_SET_VRF:
3794       if (!(buffer && buflen && (*buflen == sizeof (u32))))
3795         {
3796           rv = VPPCOM_EINVAL;
3797           break;
3798         }
3799       session->vrf = *(u32 *) buffer;
3800       break;
3801
3802     case VPPCOM_ATTR_GET_VRF:
3803       if (!(buffer && buflen && (*buflen >= sizeof (u32))))
3804         {
3805           rv = VPPCOM_EINVAL;
3806           break;
3807         }
3808       *(u32 *) buffer = session->vrf;
3809       *buflen = sizeof (u32);
3810       break;
3811
3812     case VPPCOM_ATTR_GET_DOMAIN:
3813       if (!(buffer && buflen && (*buflen >= sizeof (int))))
3814         {
3815           rv = VPPCOM_EINVAL;
3816           break;
3817         }
3818
3819       if (session->transport.is_ip4)
3820         *(int *) buffer = AF_INET;
3821       else
3822         *(int *) buffer = AF_INET6;
3823       *buflen = sizeof (int);
3824
3825       VDBG (2, "VPPCOM_ATTR_GET_DOMAIN: %d, buflen %u", *(int *) buffer,
3826             *buflen);
3827       break;
3828
3829     case VPPCOM_ATTR_SET_ENDPT_EXT_CFG:
3830       if (!(buffer && buflen && (*buflen > 0)))
3831         {
3832           rv = VPPCOM_EINVAL;
3833           break;
3834         }
3835       if (session->ext_config)
3836         {
3837           rv = VPPCOM_EINVAL;
3838           break;
3839         }
3840       vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_NONE,
3841                                  *buflen + sizeof (u32));
3842       clib_memcpy (session->ext_config->data, buffer, *buflen);
3843       session->ext_config->len = *buflen;
3844       break;
3845
3846     default:
3847       rv = VPPCOM_EINVAL;
3848       break;
3849     }
3850
3851   return rv;
3852 }
3853
3854 int
3855 vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
3856                          uint32_t buflen, int flags, vppcom_endpt_t * ep)
3857 {
3858   vcl_worker_t *wrk = vcl_worker_get_current ();
3859   vcl_session_t *session;
3860   int rv = VPPCOM_OK;
3861
3862   if (flags == 0)
3863     rv = vppcom_session_read (session_handle, buffer, buflen);
3864   else if (flags & MSG_PEEK)
3865     rv = vppcom_session_peek (session_handle, buffer, buflen);
3866   else
3867     {
3868       VDBG (0, "Unsupport flags for recvfrom %d", flags);
3869       return VPPCOM_EAFNOSUPPORT;
3870     }
3871
3872   if (ep && rv > 0)
3873     {
3874       session = vcl_session_get_w_handle (wrk, session_handle);
3875       if (session->transport.is_ip4)
3876         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3877                           sizeof (ip4_address_t));
3878       else
3879         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3880                           sizeof (ip6_address_t));
3881       ep->is_ip4 = session->transport.is_ip4;
3882       ep->port = session->transport.rmt_port;
3883     }
3884
3885   return rv;
3886 }
3887
3888 int
3889 vppcom_session_sendto (uint32_t session_handle, void *buffer,
3890                        uint32_t buflen, int flags, vppcom_endpt_t * ep)
3891 {
3892   vcl_worker_t *wrk = vcl_worker_get_current ();
3893   vcl_session_t *s;
3894
3895   s = vcl_session_get_w_handle (wrk, session_handle);
3896   if (PREDICT_FALSE (!s))
3897     return VPPCOM_EBADFD;
3898
3899   if (ep)
3900     {
3901       if (!vcl_session_is_cl (s))
3902         return VPPCOM_EINVAL;
3903
3904       /* Session not connected/bound in vpp. Create it by 'connecting' it */
3905       if (PREDICT_FALSE (s->session_state == VCL_STATE_CLOSED))
3906         {
3907           u32 session_index = s->session_index;
3908           f64 timeout = vcm->cfg.session_timeout;
3909           int rv;
3910
3911           vcl_send_session_connect (wrk, s);
3912           rv = vppcom_wait_for_session_state_change (session_index,
3913                                                      VCL_STATE_READY,
3914                                                      timeout);
3915           if (rv < 0)
3916             return rv;
3917           s = vcl_session_get (wrk, session_index);
3918         }
3919
3920       s->transport.is_ip4 = ep->is_ip4;
3921       s->transport.rmt_port = ep->port;
3922       vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
3923     }
3924
3925   if (flags)
3926     {
3927       // TBD check the flags and do the right thing
3928       VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
3929     }
3930
3931   return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
3932                                        s->is_dgram ? 1 : 0));
3933 }
3934
3935 int
3936 vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
3937 {
3938   vcl_worker_t *wrk = vcl_worker_get_current ();
3939   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
3940   u32 i, keep_trying = 1;
3941   svm_msg_q_msg_t msg;
3942   session_event_t *e;
3943   int rv, num_ev = 0;
3944
3945   VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
3946
3947   if (!vp)
3948     return VPPCOM_EFAULT;
3949
3950   do
3951     {
3952       vcl_session_t *session;
3953
3954       /* Dequeue all events and drop all unhandled io events */
3955       while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
3956         {
3957           e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
3958           vcl_handle_mq_event (wrk, e);
3959           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
3960         }
3961       vec_reset_length (wrk->unhandled_evts_vector);
3962
3963       for (i = 0; i < n_sids; i++)
3964         {
3965           session = vcl_session_get (wrk, vp[i].sh);
3966           if (!session)
3967             {
3968               vp[i].revents = POLLHUP;
3969               num_ev++;
3970               continue;
3971             }
3972
3973           vp[i].revents = 0;
3974
3975           if (POLLIN & vp[i].events)
3976             {
3977               rv = vcl_session_read_ready (session);
3978               if (rv > 0)
3979                 {
3980                   vp[i].revents |= POLLIN;
3981                   num_ev++;
3982                 }
3983               else if (rv < 0)
3984                 {
3985                   switch (rv)
3986                     {
3987                     case VPPCOM_ECONNRESET:
3988                       vp[i].revents = POLLHUP;
3989                       break;
3990
3991                     default:
3992                       vp[i].revents = POLLERR;
3993                       break;
3994                     }
3995                   num_ev++;
3996                 }
3997             }
3998
3999           if (POLLOUT & vp[i].events)
4000             {
4001               rv = vcl_session_write_ready (session);
4002               if (rv > 0)
4003                 {
4004                   vp[i].revents |= POLLOUT;
4005                   num_ev++;
4006                 }
4007               else if (rv < 0)
4008                 {
4009                   switch (rv)
4010                     {
4011                     case VPPCOM_ECONNRESET:
4012                       vp[i].revents = POLLHUP;
4013                       break;
4014
4015                     default:
4016                       vp[i].revents = POLLERR;
4017                       break;
4018                     }
4019                   num_ev++;
4020                 }
4021             }
4022
4023           if (0)                // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
4024             {
4025               vp[i].revents = POLLNVAL;
4026               num_ev++;
4027             }
4028         }
4029       if (wait_for_time != -1)
4030         keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
4031     }
4032   while ((num_ev == 0) && keep_trying);
4033
4034   return num_ev;
4035 }
4036
4037 int
4038 vppcom_mq_epoll_fd (void)
4039 {
4040   vcl_worker_t *wrk = vcl_worker_get_current ();
4041   return wrk->mqs_epfd;
4042 }
4043
4044 int
4045 vppcom_session_index (vcl_session_handle_t session_handle)
4046 {
4047   return session_handle & 0xFFFFFF;
4048 }
4049
4050 int
4051 vppcom_session_worker (vcl_session_handle_t session_handle)
4052 {
4053   return session_handle >> 24;
4054 }
4055
4056 int
4057 vppcom_worker_register (void)
4058 {
4059   if (!vcl_worker_alloc_and_init ())
4060     return VPPCOM_EEXIST;
4061
4062   if (vcl_worker_register_with_vpp ())
4063     return VPPCOM_EEXIST;
4064
4065   return VPPCOM_OK;
4066 }
4067
4068 void
4069 vppcom_worker_unregister (void)
4070 {
4071   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
4072   vcl_set_worker_index (~0);
4073 }
4074
4075 void
4076 vppcom_worker_index_set (int index)
4077 {
4078   vcl_set_worker_index (index);
4079 }
4080
4081 int
4082 vppcom_worker_index (void)
4083 {
4084   return vcl_get_worker_index ();
4085 }
4086
4087 int
4088 vppcom_worker_mqs_epfd (void)
4089 {
4090   vcl_worker_t *wrk = vcl_worker_get_current ();
4091   if (!vcm->cfg.use_mq_eventfd)
4092     return -1;
4093   return wrk->mqs_epfd;
4094 }
4095
4096 int
4097 vppcom_session_is_connectable_listener (uint32_t session_handle)
4098 {
4099   vcl_session_t *session;
4100   vcl_worker_t *wrk = vcl_worker_get_current ();
4101   session = vcl_session_get_w_handle (wrk, session_handle);
4102   if (!session)
4103     return VPPCOM_EBADFD;
4104   return vcl_session_is_connectable_listener (wrk, session);
4105 }
4106
4107 int
4108 vppcom_session_listener (uint32_t session_handle)
4109 {
4110   vcl_worker_t *wrk = vcl_worker_get_current ();
4111   vcl_session_t *listen_session, *session;
4112   session = vcl_session_get_w_handle (wrk, session_handle);
4113   if (!session)
4114     return VPPCOM_EBADFD;
4115   if (session->listener_index == VCL_INVALID_SESSION_INDEX)
4116     return VPPCOM_EBADFD;
4117   listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
4118   if (!listen_session)
4119     return VPPCOM_EBADFD;
4120   return vcl_session_handle (listen_session);
4121 }
4122
4123 int
4124 vppcom_session_n_accepted (uint32_t session_handle)
4125 {
4126   vcl_worker_t *wrk = vcl_worker_get_current ();
4127   vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
4128   if (!session)
4129     return VPPCOM_EBADFD;
4130   return session->n_accepted_sessions;
4131 }
4132
4133 const char *
4134 vppcom_proto_str (vppcom_proto_t proto)
4135 {
4136   char const *proto_str;
4137
4138   switch (proto)
4139     {
4140     case VPPCOM_PROTO_TCP:
4141       proto_str = "TCP";
4142       break;
4143     case VPPCOM_PROTO_UDP:
4144       proto_str = "UDP";
4145       break;
4146     case VPPCOM_PROTO_TLS:
4147       proto_str = "TLS";
4148       break;
4149     case VPPCOM_PROTO_QUIC:
4150       proto_str = "QUIC";
4151       break;
4152     case VPPCOM_PROTO_DTLS:
4153       proto_str = "DTLS";
4154       break;
4155     case VPPCOM_PROTO_SRTP:
4156       proto_str = "SRTP";
4157       break;
4158     default:
4159       proto_str = "UNKNOWN";
4160       break;
4161     }
4162   return proto_str;
4163 }
4164
4165 const char *
4166 vppcom_retval_str (int retval)
4167 {
4168   char const *st;
4169
4170   switch (retval)
4171     {
4172     case VPPCOM_OK:
4173       st = "VPPCOM_OK";
4174       break;
4175
4176     case VPPCOM_EAGAIN:
4177       st = "VPPCOM_EAGAIN";
4178       break;
4179
4180     case VPPCOM_EFAULT:
4181       st = "VPPCOM_EFAULT";
4182       break;
4183
4184     case VPPCOM_ENOMEM:
4185       st = "VPPCOM_ENOMEM";
4186       break;
4187
4188     case VPPCOM_EINVAL:
4189       st = "VPPCOM_EINVAL";
4190       break;
4191
4192     case VPPCOM_EBADFD:
4193       st = "VPPCOM_EBADFD";
4194       break;
4195
4196     case VPPCOM_EAFNOSUPPORT:
4197       st = "VPPCOM_EAFNOSUPPORT";
4198       break;
4199
4200     case VPPCOM_ECONNABORTED:
4201       st = "VPPCOM_ECONNABORTED";
4202       break;
4203
4204     case VPPCOM_ECONNRESET:
4205       st = "VPPCOM_ECONNRESET";
4206       break;
4207
4208     case VPPCOM_ENOTCONN:
4209       st = "VPPCOM_ENOTCONN";
4210       break;
4211
4212     case VPPCOM_ECONNREFUSED:
4213       st = "VPPCOM_ECONNREFUSED";
4214       break;
4215
4216     case VPPCOM_ETIMEDOUT:
4217       st = "VPPCOM_ETIMEDOUT";
4218       break;
4219
4220     default:
4221       st = "UNKNOWN_STATE";
4222       break;
4223     }
4224
4225   return st;
4226 }
4227
4228 int
4229 vppcom_add_cert_key_pair (vppcom_cert_key_pair_t *ckpair)
4230 {
4231   if (vcm->cfg.vpp_app_socket_api)
4232     {
4233       clib_warning ("not supported");
4234       return VPPCOM_EINVAL;
4235     }
4236   return vcl_bapi_add_cert_key_pair (ckpair);
4237 }
4238
4239 int
4240 vppcom_del_cert_key_pair (uint32_t ckpair_index)
4241 {
4242   if (vcm->cfg.vpp_app_socket_api)
4243     {
4244       clib_warning ("not supported");
4245       return VPPCOM_EINVAL;
4246     }
4247   return vcl_bapi_del_cert_key_pair (ckpair_index);
4248 }
4249
4250 /*
4251  * fd.io coding-style-patch-verification: ON
4252  *
4253  * Local Variables:
4254  * eval: (c-set-style "gnu")
4255  * End:
4256  */