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