src/vcl/vppcom.c

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