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