src/vcl/vppcom.c

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