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