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