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