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