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   vcl_session_t *s;
 992
 993   switch (e->event_type)
 994     {
 995     case SESSION_IO_EVT_RX:
 996     case SESSION_IO_EVT_TX:
 997       s = vcl_session_get (wrk, e->session_index);
 998       if (!s || !vcl_session_is_open (s))
 999         break;
1000       vec_add1 (wrk->unhandled_evts_vector, *e);
1001       break;
1002     case SESSION_CTRL_EVT_ACCEPTED:
1003       vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
1004       break;
1005     case SESSION_CTRL_EVT_CONNECTED:
1006       vcl_session_connected_handler (wrk,
1007                                      (session_connected_msg_t *) e->data);
1008       break;
1009     case SESSION_CTRL_EVT_DISCONNECTED:
1010       disconnected_msg = (session_disconnected_msg_t *) e->data;
1011       s = vcl_session_disconnected_handler (wrk, disconnected_msg);
1012       if (!s)
1013         break;
1014       VDBG (0, "disconnected session %u [0x%llx]", s->session_index,
1015             s->vpp_handle);
1016       break;
1017     case SESSION_CTRL_EVT_RESET:
1018       vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
1019       break;
1020     case SESSION_CTRL_EVT_BOUND:
1021       vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
1022       break;
1023     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
1024       vcl_session_unlisten_reply_handler (wrk, e->data);
1025       break;
1026     case SESSION_CTRL_EVT_MIGRATED:
1027       vcl_session_migrated_handler (wrk, e->data);
1028       break;
1029     case SESSION_CTRL_EVT_CLEANUP:
1030       vcl_session_cleanup_handler (wrk, e->data);
1031       break;
1032     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
1033       vcl_session_req_worker_update_handler (wrk, e->data);
1034       break;
1035     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
1036       vcl_session_worker_update_reply_handler (wrk, e->data);
1037       break;
1038     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
1039       vcl_session_app_add_segment_handler (wrk, e->data);
1040       break;
1041     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
1042       vcl_session_app_del_segment_handler (wrk, e->data);
1043       break;
1044     case SESSION_CTRL_EVT_APP_WRK_RPC:
1045       vcl_worker_rpc_handler (wrk, e->data);
1046       break;
1047     default:
1048       clib_warning ("unhandled %u", e->event_type);
1049     }
1050   return VPPCOM_OK;
1051 }
1052
1053 static int
1054 vppcom_wait_for_session_state_change (u32 session_index,
1055                                       vcl_session_state_t state,
1056                                       f64 wait_for_time)
1057 {
1058   vcl_worker_t *wrk = vcl_worker_get_current ();
1059   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
1060   vcl_session_t *volatile session;
1061   svm_msg_q_msg_t msg;
1062   session_event_t *e;
1063
1064   do
1065     {
1066       session = vcl_session_get (wrk, session_index);
1067       if (PREDICT_FALSE (!session))
1068         {
1069           return VPPCOM_EBADFD;
1070         }
1071       if (session->session_state == state)
1072         {
1073           return VPPCOM_OK;
1074         }
1075       if (session->session_state == VCL_STATE_DETACHED)
1076         {
1077           return VPPCOM_ECONNREFUSED;
1078         }
1079
1080       if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0))
1081         {
1082           usleep (100);
1083           continue;
1084         }
1085       e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
1086       vcl_handle_mq_event (wrk, e);
1087       svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1088     }
1089   while (clib_time_now (&wrk->clib_time) < timeout);
1090
1091   VDBG (0, "timeout waiting for state 0x%x (%s)", state,
1092         vppcom_session_state_str (state));
1093   vcl_evt (VCL_EVT_SESSION_TIMEOUT, session, session_state);
1094
1095   return VPPCOM_ETIMEDOUT;
1096 }
1097
1098 static void
1099 vcl_handle_pending_wrk_updates (vcl_worker_t * wrk)
1100 {
1101   vcl_session_state_t state;
1102   vcl_session_t *s;
1103   u32 *sip;
1104
1105   if (PREDICT_TRUE (vec_len (wrk->pending_session_wrk_updates) == 0))
1106     return;
1107
1108   vec_foreach (sip, wrk->pending_session_wrk_updates)
1109   {
1110     s = vcl_session_get (wrk, *sip);
1111     vcl_send_session_worker_update (wrk, s, wrk->wrk_index);
1112     state = s->session_state;
1113     vppcom_wait_for_session_state_change (s->session_index, VCL_STATE_UPDATED,
1114                                           5);
1115     s->session_state = state;
1116   }
1117   vec_reset_length (wrk->pending_session_wrk_updates);
1118 }
1119
1120 void
1121 vcl_flush_mq_events (void)
1122 {
1123   vcl_worker_t *wrk = vcl_worker_get_current ();
1124   svm_msg_q_msg_t *msg;
1125   session_event_t *e;
1126   svm_msg_q_t *mq;
1127   int i;
1128
1129   mq = wrk->app_event_queue;
1130   svm_msg_q_lock (mq);
1131   vcl_mq_dequeue_batch (wrk, mq, ~0);
1132   svm_msg_q_unlock (mq);
1133
1134   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
1135     {
1136       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
1137       e = svm_msg_q_msg_data (mq, msg);
1138       vcl_handle_mq_event (wrk, e);
1139       svm_msg_q_free_msg (mq, msg);
1140     }
1141   vec_reset_length (wrk->mq_msg_vector);
1142   vcl_handle_pending_wrk_updates (wrk);
1143 }
1144
1145 static int
1146 vppcom_session_unbind (u32 session_handle)
1147 {
1148   vcl_worker_t *wrk = vcl_worker_get_current ();
1149   session_accepted_msg_t *accepted_msg;
1150   vcl_session_t *session = 0;
1151   vcl_session_msg_t *evt;
1152
1153   session = vcl_session_get_w_handle (wrk, session_handle);
1154   if (!session)
1155     return VPPCOM_EBADFD;
1156
1157   /* Flush pending accept events, if any */
1158   while (clib_fifo_elts (session->accept_evts_fifo))
1159     {
1160       clib_fifo_sub2 (session->accept_evts_fifo, evt);
1161       accepted_msg = &evt->accepted_msg;
1162       vcl_session_table_del_vpp_handle (wrk, accepted_msg->handle);
1163       vcl_send_session_accepted_reply (session->vpp_evt_q,
1164                                        accepted_msg->context,
1165                                        session->vpp_handle, -1);
1166     }
1167   clib_fifo_free (session->accept_evts_fifo);
1168
1169   vcl_send_session_unlisten (wrk, session);
1170
1171   VDBG (1, "session %u [0x%llx]: sending unbind!", session->session_index,
1172         session->vpp_handle);
1173   vcl_evt (VCL_EVT_UNBIND, session);
1174
1175   session->vpp_handle = ~0;
1176   session->session_state = VCL_STATE_DISCONNECT;
1177
1178   return VPPCOM_OK;
1179 }
1180
1181 /**
1182  * Handle app exit
1183  *
1184  * Notify vpp of the disconnect and mark the worker as free. If we're the
1185  * last worker, do a full cleanup otherwise, since we're probably a forked
1186  * child, avoid syscalls as much as possible. We might've lost privileges.
1187  */
1188 void
1189 vppcom_app_exit (void)
1190 {
1191   if (!pool_elts (vcm->workers))
1192     return;
1193   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
1194   vcl_set_worker_index (~0);
1195   vcl_elog_stop (vcm);
1196 }
1197
1198 static int
1199 vcl_api_attach (void)
1200 {
1201   if (vcm->cfg.vpp_app_socket_api)
1202     return vcl_sapi_attach ();
1203
1204   return vcl_bapi_attach ();
1205 }
1206
1207 static void
1208 vcl_api_detach (vcl_worker_t * wrk)
1209 {
1210   vcl_send_app_detach (wrk);
1211
1212   if (vcm->cfg.vpp_app_socket_api)
1213     return vcl_sapi_detach (wrk);
1214
1215   return vcl_bapi_disconnect_from_vpp ();
1216 }
1217
1218 /*
1219  * VPPCOM Public API functions
1220  */
1221 int
1222 vppcom_app_create (const char *app_name)
1223 {
1224   vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1225   int rv;
1226
1227   if (vcm->is_init)
1228     {
1229       VDBG (1, "already initialized");
1230       return VPPCOM_EEXIST;
1231     }
1232
1233   vcm->is_init = 1;
1234   vppcom_cfg (&vcm->cfg);
1235   vcl_cfg = &vcm->cfg;
1236
1237   vcm->main_cpu = pthread_self ();
1238   vcm->main_pid = getpid ();
1239   vcm->app_name = format (0, "%s", app_name);
1240   fifo_segment_main_init (&vcm->segment_main, vcl_cfg->segment_baseva,
1241                           20 /* timeout in secs */ );
1242   pool_alloc (vcm->workers, vcl_cfg->max_workers);
1243   clib_spinlock_init (&vcm->workers_lock);
1244   clib_rwlock_init (&vcm->segment_table_lock);
1245   atexit (vppcom_app_exit);
1246   vcl_elog_init (vcm);
1247
1248   /* Allocate default worker */
1249   vcl_worker_alloc_and_init ();
1250
1251   if ((rv = vcl_api_attach ()))
1252     return rv;
1253
1254   VDBG (0, "app_name '%s', my_client_index %d (0x%x)", app_name,
1255         vcm->workers[0].api_client_handle, vcm->workers[0].api_client_handle);
1256
1257   return VPPCOM_OK;
1258 }
1259
1260 void
1261 vppcom_app_destroy (void)
1262 {
1263   vcl_worker_t *wrk, *current_wrk;
1264   void *heap;
1265
1266   if (!pool_elts (vcm->workers))
1267     return;
1268
1269   vcl_evt (VCL_EVT_DETACH, vcm);
1270
1271   current_wrk = vcl_worker_get_current ();
1272
1273   /* *INDENT-OFF* */
1274   pool_foreach (wrk, vcm->workers, ({
1275     if (current_wrk != wrk)
1276       vcl_worker_cleanup (wrk, 0 /* notify vpp */ );
1277   }));
1278   /* *INDENT-ON* */
1279
1280   vcl_api_detach (current_wrk);
1281   vcl_worker_cleanup (current_wrk, 0 /* notify vpp */ );
1282
1283   vcl_elog_stop (vcm);
1284
1285   /*
1286    * Free the heap and fix vcm
1287    */
1288   heap = clib_mem_get_heap ();
1289   munmap (clib_mem_get_heap_base (heap), clib_mem_get_heap_size (heap));
1290
1291   vcm = &_vppcom_main;
1292   vcm->is_init = 0;
1293 }
1294
1295 int
1296 vppcom_session_create (u8 proto, u8 is_nonblocking)
1297 {
1298   vcl_worker_t *wrk = vcl_worker_get_current ();
1299   vcl_session_t *session;
1300
1301   session = vcl_session_alloc (wrk);
1302
1303   session->session_type = proto;
1304   session->session_state = VCL_STATE_CLOSED;
1305   session->vpp_handle = ~0;
1306   session->is_dgram = vcl_proto_is_dgram (proto);
1307
1308   if (is_nonblocking)
1309     vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
1310
1311   vcl_evt (VCL_EVT_CREATE, session, session_type, session->session_state,
1312            is_nonblocking, session_index);
1313
1314   VDBG (0, "created session %u", session->session_index);
1315
1316   return vcl_session_handle (session);
1317 }
1318
1319 int
1320 vcl_session_cleanup (vcl_worker_t * wrk, vcl_session_t * s,
1321                      vcl_session_handle_t sh, u8 do_disconnect)
1322 {
1323   int rv = VPPCOM_OK;
1324
1325   VDBG (1, "session %u [0x%llx] closing", s->session_index, s->vpp_handle);
1326
1327   if (s->flags & VCL_SESSION_F_IS_VEP)
1328     {
1329       u32 next_sh = s->vep.next_sh;
1330       while (next_sh != ~0)
1331         {
1332           rv = vppcom_epoll_ctl (sh, EPOLL_CTL_DEL, next_sh, 0);
1333           if (PREDICT_FALSE (rv < 0))
1334             VDBG (0, "vpp handle 0x%llx, sh %u: EPOLL_CTL_DEL vep_idx %u"
1335                   " failed! rv %d (%s)", s->vpp_handle, next_sh,
1336                   s->vep.vep_sh, rv, vppcom_retval_str (rv));
1337           next_sh = s->vep.next_sh;
1338         }
1339       goto cleanup;
1340     }
1341
1342   if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
1343     {
1344       rv = vppcom_epoll_ctl (s->vep.vep_sh, EPOLL_CTL_DEL, sh, 0);
1345       if (rv < 0)
1346         VDBG (0, "session %u [0x%llx]: EPOLL_CTL_DEL vep_idx %u "
1347               "failed! rv %d (%s)", s->session_index, s->vpp_handle,
1348               s->vep.vep_sh, rv, vppcom_retval_str (rv));
1349     }
1350
1351   if (!do_disconnect)
1352     {
1353       VDBG (1, "session %u [0x%llx] disconnect skipped",
1354             s->session_index, s->vpp_handle);
1355       goto cleanup;
1356     }
1357
1358   if (s->session_state == VCL_STATE_LISTEN)
1359     {
1360       rv = vppcom_session_unbind (sh);
1361       if (PREDICT_FALSE (rv < 0))
1362         VDBG (0, "session %u [0x%llx]: listener unbind failed! "
1363               "rv %d (%s)", s->session_index, s->vpp_handle, rv,
1364               vppcom_retval_str (rv));
1365       return rv;
1366     }
1367   else if (vcl_session_is_ready (s)
1368            || (vcl_session_is_connectable_listener (wrk, s)))
1369     {
1370       rv = vppcom_session_disconnect (sh);
1371       if (PREDICT_FALSE (rv < 0))
1372         VDBG (0, "ERROR: session %u [0x%llx]: disconnect failed!"
1373               " rv %d (%s)", s->session_index, s->vpp_handle,
1374               rv, vppcom_retval_str (rv));
1375     }
1376   else if (s->session_state == VCL_STATE_DISCONNECT)
1377     {
1378       svm_msg_q_t *mq = vcl_session_vpp_evt_q (wrk, s);
1379       vcl_send_session_reset_reply (mq, wrk->api_client_handle,
1380                                     s->vpp_handle, 0);
1381     }
1382   else if (s->session_state == VCL_STATE_DETACHED)
1383     {
1384       /* Should not happen. VPP cleaned up before app confirmed close */
1385       VDBG (0, "vpp freed session %d before close", s->session_index);
1386       goto free_session;
1387     }
1388
1389   s->session_state = VCL_STATE_CLOSED;
1390
1391   /* Session is removed only after vpp confirms the disconnect */
1392   return rv;
1393
1394 cleanup:
1395   vcl_session_table_del_vpp_handle (wrk, s->vpp_handle);
1396 free_session:
1397   vcl_session_free (wrk, s);
1398   vcl_evt (VCL_EVT_CLOSE, s, rv);
1399
1400   return rv;
1401 }
1402
1403 int
1404 vppcom_session_close (uint32_t session_handle)
1405 {
1406   vcl_worker_t *wrk = vcl_worker_get_current ();
1407   vcl_session_t *session;
1408
1409   session = vcl_session_get_w_handle (wrk, session_handle);
1410   if (!session)
1411     return VPPCOM_EBADFD;
1412   return vcl_session_cleanup (wrk, session, session_handle,
1413                               1 /* do_disconnect */ );
1414 }
1415
1416 int
1417 vppcom_session_bind (uint32_t session_handle, vppcom_endpt_t * ep)
1418 {
1419   vcl_worker_t *wrk = vcl_worker_get_current ();
1420   vcl_session_t *session = 0;
1421
1422   if (!ep || !ep->ip)
1423     return VPPCOM_EINVAL;
1424
1425   session = vcl_session_get_w_handle (wrk, session_handle);
1426   if (!session)
1427     return VPPCOM_EBADFD;
1428
1429   if (session->flags & VCL_SESSION_F_IS_VEP)
1430     {
1431       VDBG (0, "ERROR: cannot bind to epoll session %u!",
1432             session->session_index);
1433       return VPPCOM_EBADFD;
1434     }
1435
1436   session->transport.is_ip4 = ep->is_ip4;
1437   if (ep->is_ip4)
1438     clib_memcpy_fast (&session->transport.lcl_ip.ip4, ep->ip,
1439                       sizeof (ip4_address_t));
1440   else
1441     clib_memcpy_fast (&session->transport.lcl_ip.ip6, ep->ip,
1442                       sizeof (ip6_address_t));
1443   session->transport.lcl_port = ep->port;
1444
1445   VDBG (0, "session %u handle %u: binding to local %s address %U port %u, "
1446         "proto %s", session->session_index, session_handle,
1447         session->transport.is_ip4 ? "IPv4" : "IPv6",
1448         format_ip46_address, &session->transport.lcl_ip,
1449         session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1450         clib_net_to_host_u16 (session->transport.lcl_port),
1451         vppcom_proto_str (session->session_type));
1452   vcl_evt (VCL_EVT_BIND, session);
1453
1454   if (session->session_type == VPPCOM_PROTO_UDP)
1455     vppcom_session_listen (session_handle, 10);
1456
1457   return VPPCOM_OK;
1458 }
1459
1460 int
1461 vppcom_session_listen (uint32_t listen_sh, uint32_t q_len)
1462 {
1463   vcl_worker_t *wrk = vcl_worker_get_current ();
1464   vcl_session_t *listen_session = 0;
1465   u64 listen_vpp_handle;
1466   int rv;
1467
1468   listen_session = vcl_session_get_w_handle (wrk, listen_sh);
1469   if (!listen_session || (listen_session->flags & VCL_SESSION_F_IS_VEP))
1470     return VPPCOM_EBADFD;
1471
1472   if (q_len == 0 || q_len == ~0)
1473     q_len = vcm->cfg.listen_queue_size;
1474
1475   listen_vpp_handle = listen_session->vpp_handle;
1476   if (listen_session->session_state == VCL_STATE_LISTEN)
1477     {
1478       VDBG (0, "session %u [0x%llx]: already in listen state!",
1479             listen_sh, listen_vpp_handle);
1480       return VPPCOM_OK;
1481     }
1482
1483   VDBG (0, "session %u: sending vpp listen request...", listen_sh);
1484
1485   /*
1486    * Send listen request to vpp and wait for reply
1487    */
1488   vcl_send_session_listen (wrk, listen_session);
1489   rv = vppcom_wait_for_session_state_change (listen_session->session_index,
1490                                              VCL_STATE_LISTEN,
1491                                              vcm->cfg.session_timeout);
1492
1493   if (PREDICT_FALSE (rv))
1494     {
1495       listen_session = vcl_session_get_w_handle (wrk, listen_sh);
1496       VDBG (0, "session %u [0x%llx]: listen failed! returning %d (%s)",
1497             listen_sh, listen_session->vpp_handle, rv,
1498             vppcom_retval_str (rv));
1499       return rv;
1500     }
1501
1502   return VPPCOM_OK;
1503 }
1504
1505 static int
1506 validate_args_session_accept_ (vcl_worker_t * wrk, vcl_session_t * ls)
1507 {
1508   if (ls->flags & VCL_SESSION_F_IS_VEP)
1509     {
1510       VDBG (0, "ERROR: cannot accept on epoll session %u!",
1511             ls->session_index);
1512       return VPPCOM_EBADFD;
1513     }
1514
1515   if ((ls->session_state != VCL_STATE_LISTEN)
1516       && (!vcl_session_is_connectable_listener (wrk, ls)))
1517     {
1518       VDBG (0, "ERROR: session [0x%llx]: not in listen state! state 0x%x"
1519             " (%s)", ls->vpp_handle, ls->session_state,
1520             vppcom_session_state_str (ls->session_state));
1521       return VPPCOM_EBADFD;
1522     }
1523   return VPPCOM_OK;
1524 }
1525
1526 int
1527 vppcom_unformat_proto (uint8_t * proto, char *proto_str)
1528 {
1529   if (!strcmp (proto_str, "TCP"))
1530     *proto = VPPCOM_PROTO_TCP;
1531   else if (!strcmp (proto_str, "tcp"))
1532     *proto = VPPCOM_PROTO_TCP;
1533   else if (!strcmp (proto_str, "UDP"))
1534     *proto = VPPCOM_PROTO_UDP;
1535   else if (!strcmp (proto_str, "udp"))
1536     *proto = VPPCOM_PROTO_UDP;
1537   else if (!strcmp (proto_str, "TLS"))
1538     *proto = VPPCOM_PROTO_TLS;
1539   else if (!strcmp (proto_str, "tls"))
1540     *proto = VPPCOM_PROTO_TLS;
1541   else if (!strcmp (proto_str, "QUIC"))
1542     *proto = VPPCOM_PROTO_QUIC;
1543   else if (!strcmp (proto_str, "quic"))
1544     *proto = VPPCOM_PROTO_QUIC;
1545   else
1546     return 1;
1547   return 0;
1548 }
1549
1550 int
1551 vppcom_session_accept (uint32_t listen_session_handle, vppcom_endpt_t * ep,
1552                        uint32_t flags)
1553 {
1554   u32 client_session_index = ~0, listen_session_index, accept_flags = 0;
1555   vcl_worker_t *wrk = vcl_worker_get_current ();
1556   session_accepted_msg_t accepted_msg;
1557   vcl_session_t *listen_session = 0;
1558   vcl_session_t *client_session = 0;
1559   vcl_session_msg_t *evt;
1560   svm_msg_q_msg_t msg;
1561   session_event_t *e;
1562   u8 is_nonblocking;
1563   int rv;
1564
1565   listen_session = vcl_session_get_w_handle (wrk, listen_session_handle);
1566   if (!listen_session)
1567     return VPPCOM_EBADFD;
1568
1569   listen_session_index = listen_session->session_index;
1570   if ((rv = validate_args_session_accept_ (wrk, listen_session)))
1571     return rv;
1572
1573   if (clib_fifo_elts (listen_session->accept_evts_fifo))
1574     {
1575       clib_fifo_sub2 (listen_session->accept_evts_fifo, evt);
1576       accept_flags = evt->flags;
1577       accepted_msg = evt->accepted_msg;
1578       goto handle;
1579     }
1580
1581   is_nonblocking = vcl_session_has_attr (listen_session,
1582                                          VCL_SESS_ATTR_NONBLOCK);
1583   while (1)
1584     {
1585       if (svm_msg_q_is_empty (wrk->app_event_queue) && is_nonblocking)
1586         return VPPCOM_EAGAIN;
1587
1588       if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_WAIT, 0))
1589         return VPPCOM_EAGAIN;
1590
1591       e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
1592       if (e->event_type != SESSION_CTRL_EVT_ACCEPTED)
1593         {
1594           vcl_handle_mq_event (wrk, e);
1595           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1596           continue;
1597         }
1598       clib_memcpy_fast (&accepted_msg, e->data, sizeof (accepted_msg));
1599       svm_msg_q_free_msg (wrk->app_event_queue, &msg);
1600       break;
1601     }
1602
1603 handle:
1604
1605   client_session_index = vcl_session_accepted_handler (wrk, &accepted_msg,
1606                                                        listen_session_index);
1607   if (client_session_index == VCL_INVALID_SESSION_INDEX)
1608     return VPPCOM_ECONNABORTED;
1609
1610   listen_session = vcl_session_get (wrk, listen_session_index);
1611   client_session = vcl_session_get (wrk, client_session_index);
1612
1613   if (flags & O_NONBLOCK)
1614     vcl_session_set_attr (client_session, VCL_SESS_ATTR_NONBLOCK);
1615
1616   VDBG (1, "listener %u [0x%llx]: Got a connect request! session %u [0x%llx],"
1617         " flags %d, is_nonblocking %u", listen_session->session_index,
1618         listen_session->vpp_handle, client_session_index,
1619         client_session->vpp_handle, flags,
1620         vcl_session_has_attr (client_session, VCL_SESS_ATTR_NONBLOCK));
1621
1622   if (ep)
1623     {
1624       ep->is_ip4 = client_session->transport.is_ip4;
1625       ep->port = client_session->transport.rmt_port;
1626       if (client_session->transport.is_ip4)
1627         clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip4,
1628                           sizeof (ip4_address_t));
1629       else
1630         clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip6,
1631                           sizeof (ip6_address_t));
1632     }
1633
1634   VDBG (0, "listener %u [0x%llx] accepted %u [0x%llx] peer: %U:%u "
1635         "local: %U:%u", listen_session_handle, listen_session->vpp_handle,
1636         client_session_index, client_session->vpp_handle,
1637         format_ip46_address, &client_session->transport.rmt_ip,
1638         client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1639         clib_net_to_host_u16 (client_session->transport.rmt_port),
1640         format_ip46_address, &client_session->transport.lcl_ip,
1641         client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1642         clib_net_to_host_u16 (client_session->transport.lcl_port));
1643   vcl_evt (VCL_EVT_ACCEPT, client_session, listen_session,
1644            client_session_index);
1645
1646   /*
1647    * Session might have been closed already
1648    */
1649   if (accept_flags)
1650     {
1651       if (accept_flags & VCL_ACCEPTED_F_CLOSED)
1652         client_session->session_state = VCL_STATE_VPP_CLOSING;
1653       else if (accept_flags & VCL_ACCEPTED_F_RESET)
1654         client_session->session_state = VCL_STATE_DISCONNECT;
1655     }
1656   return vcl_session_handle (client_session);
1657 }
1658
1659 int
1660 vppcom_session_connect (uint32_t session_handle, vppcom_endpt_t * server_ep)
1661 {
1662   vcl_worker_t *wrk = vcl_worker_get_current ();
1663   vcl_session_t *session = 0;
1664   u32 session_index;
1665   int rv;
1666
1667   session = vcl_session_get_w_handle (wrk, session_handle);
1668   if (!session)
1669     return VPPCOM_EBADFD;
1670   session_index = session->session_index;
1671
1672   if (PREDICT_FALSE (session->flags & VCL_SESSION_F_IS_VEP))
1673     {
1674       VDBG (0, "ERROR: cannot connect epoll session %u!",
1675             session->session_index);
1676       return VPPCOM_EBADFD;
1677     }
1678
1679   if (PREDICT_FALSE (vcl_session_is_ready (session)))
1680     {
1681       VDBG (0, "session handle %u [0x%llx]: session already "
1682             "connected to %s %U port %d proto %s, state 0x%x (%s)",
1683             session_handle, session->vpp_handle,
1684             session->transport.is_ip4 ? "IPv4" : "IPv6", format_ip46_address,
1685             &session->transport.rmt_ip, session->transport.is_ip4 ?
1686             IP46_TYPE_IP4 : IP46_TYPE_IP6,
1687             clib_net_to_host_u16 (session->transport.rmt_port),
1688             vppcom_proto_str (session->session_type), session->session_state,
1689             vppcom_session_state_str (session->session_state));
1690       return VPPCOM_OK;
1691     }
1692
1693   /* Attempt to connect a connectionless listener */
1694   if (PREDICT_FALSE (session->session_state == VCL_STATE_LISTEN))
1695     {
1696       if (session->session_type != VPPCOM_PROTO_UDP)
1697         return VPPCOM_EINVAL;
1698       vcl_send_session_unlisten (wrk, session);
1699       session->session_state = VCL_STATE_CLOSED;
1700     }
1701
1702   session->transport.is_ip4 = server_ep->is_ip4;
1703   vcl_ip_copy_from_ep (&session->transport.rmt_ip, server_ep);
1704   session->transport.rmt_port = server_ep->port;
1705   session->parent_handle = VCL_INVALID_SESSION_HANDLE;
1706   session->flags |= VCL_SESSION_F_CONNECTED;
1707
1708   VDBG (0, "session handle %u (%s): connecting to peer %s %U "
1709         "port %d proto %s", session_handle,
1710         vppcom_session_state_str (session->session_state),
1711         session->transport.is_ip4 ? "IPv4" : "IPv6",
1712         format_ip46_address,
1713         &session->transport.rmt_ip, session->transport.is_ip4 ?
1714         IP46_TYPE_IP4 : IP46_TYPE_IP6,
1715         clib_net_to_host_u16 (session->transport.rmt_port),
1716         vppcom_proto_str (session->session_type));
1717
1718   vcl_send_session_connect (wrk, session);
1719
1720   if (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK))
1721     {
1722       /* State set to STATE_UPDATED to ensure the session is not assumed
1723        * to be ready and to also allow the app to close it prior to vpp's
1724        * connected reply. */
1725       session->session_state = VCL_STATE_UPDATED;
1726       return VPPCOM_EINPROGRESS;
1727     }
1728
1729   /*
1730    * Wait for reply from vpp if blocking
1731    */
1732   rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
1733                                              vcm->cfg.session_timeout);
1734
1735   session = vcl_session_get (wrk, session_index);
1736   VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1737         session->vpp_handle, rv ? "failed" : "succeeded");
1738
1739   return rv;
1740 }
1741
1742 int
1743 vppcom_session_stream_connect (uint32_t session_handle,
1744                                uint32_t parent_session_handle)
1745 {
1746   vcl_worker_t *wrk = vcl_worker_get_current ();
1747   vcl_session_t *session, *parent_session;
1748   u32 session_index, parent_session_index;
1749   int rv;
1750
1751   session = vcl_session_get_w_handle (wrk, session_handle);
1752   if (!session)
1753     return VPPCOM_EBADFD;
1754   parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1755   if (!parent_session)
1756     return VPPCOM_EBADFD;
1757
1758   session_index = session->session_index;
1759   parent_session_index = parent_session->session_index;
1760   if (PREDICT_FALSE (session->flags & VCL_SESSION_F_IS_VEP))
1761     {
1762       VDBG (0, "ERROR: cannot connect epoll session %u!",
1763             session->session_index);
1764       return VPPCOM_EBADFD;
1765     }
1766
1767   if (PREDICT_FALSE (vcl_session_is_ready (session)))
1768     {
1769       VDBG (0, "session handle %u [0x%llx]: session already "
1770             "connected to session %u [0x%llx] proto %s, state 0x%x (%s)",
1771             session_handle, session->vpp_handle,
1772             parent_session_handle, parent_session->vpp_handle,
1773             vppcom_proto_str (session->session_type), session->session_state,
1774             vppcom_session_state_str (session->session_state));
1775       return VPPCOM_OK;
1776     }
1777
1778   /* Connect to quic session specifics */
1779   session->transport.is_ip4 = parent_session->transport.is_ip4;
1780   session->transport.rmt_ip.ip4.as_u32 = (uint32_t) 1;
1781   session->transport.rmt_port = 0;
1782   session->parent_handle = parent_session->vpp_handle;
1783
1784   VDBG (0, "session handle %u: connecting to session %u [0x%llx]",
1785         session_handle, parent_session_handle, parent_session->vpp_handle);
1786
1787   /*
1788    * Send connect request and wait for reply from vpp
1789    */
1790   vcl_send_session_connect (wrk, session);
1791   rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
1792                                              vcm->cfg.session_timeout);
1793
1794   session->listener_index = parent_session_index;
1795   parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1796   if (parent_session)
1797     parent_session->n_accepted_sessions++;
1798
1799   session = vcl_session_get (wrk, session_index);
1800   VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1801         session->vpp_handle, rv ? "failed" : "succeeded");
1802
1803   return rv;
1804 }
1805
1806 static u8
1807 vcl_is_rx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
1808 {
1809   return (e->event_type == SESSION_IO_EVT_RX && e->session_index == sid);
1810 }
1811
1812 static inline int
1813 vppcom_session_read_internal (uint32_t session_handle, void *buf, int n,
1814                               u8 peek)
1815 {
1816   vcl_worker_t *wrk = vcl_worker_get_current ();
1817   int rv, n_read = 0, is_nonblocking;
1818   vcl_session_t *s = 0;
1819   svm_fifo_t *rx_fifo;
1820   svm_msg_q_msg_t msg;
1821   session_event_t *e;
1822   svm_msg_q_t *mq;
1823   u8 is_ct;
1824
1825   if (PREDICT_FALSE (!buf))
1826     return VPPCOM_EINVAL;
1827
1828   s = vcl_session_get_w_handle (wrk, session_handle);
1829   if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
1830     return VPPCOM_EBADFD;
1831
1832   if (PREDICT_FALSE (!vcl_session_is_open (s)))
1833     {
1834       VDBG (0, "session %u[0x%llx] is not open! state 0x%x (%s)",
1835             s->session_index, s->vpp_handle, s->session_state,
1836             vppcom_session_state_str (s->session_state));
1837       return vcl_session_closed_error (s);
1838     }
1839
1840   is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
1841   is_ct = vcl_session_is_ct (s);
1842   mq = wrk->app_event_queue;
1843   rx_fifo = is_ct ? s->ct_rx_fifo : s->rx_fifo;
1844   s->flags &= ~VCL_SESSION_F_HAS_RX_EVT;
1845
1846   if (svm_fifo_is_empty_cons (rx_fifo))
1847     {
1848       if (is_nonblocking)
1849         {
1850           if (vcl_session_is_closing (s))
1851             return vcl_session_closing_error (s);
1852           svm_fifo_unset_event (s->rx_fifo);
1853           return VPPCOM_EWOULDBLOCK;
1854         }
1855       while (svm_fifo_is_empty_cons (rx_fifo))
1856         {
1857           if (vcl_session_is_closing (s))
1858             return vcl_session_closing_error (s);
1859
1860           svm_fifo_unset_event (s->rx_fifo);
1861           svm_msg_q_lock (mq);
1862           if (svm_msg_q_is_empty (mq))
1863             svm_msg_q_wait (mq);
1864
1865           svm_msg_q_sub_w_lock (mq, &msg);
1866           e = svm_msg_q_msg_data (mq, &msg);
1867           svm_msg_q_unlock (mq);
1868           if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
1869             vcl_handle_mq_event (wrk, e);
1870           svm_msg_q_free_msg (mq, &msg);
1871         }
1872     }
1873
1874 read_again:
1875
1876   if (s->is_dgram)
1877     rv = app_recv_dgram_raw (rx_fifo, buf, n, &s->transport, 0, peek);
1878   else
1879     rv = app_recv_stream_raw (rx_fifo, buf, n, 0, peek);
1880
1881   ASSERT (rv >= 0);
1882   n_read += rv;
1883
1884   if (svm_fifo_is_empty_cons (rx_fifo))
1885     {
1886       svm_fifo_unset_event (s->rx_fifo);
1887       if (!svm_fifo_is_empty_cons (rx_fifo)
1888           && svm_fifo_set_event (s->rx_fifo) && is_nonblocking)
1889         {
1890           vec_add2 (wrk->unhandled_evts_vector, e, 1);
1891           e->event_type = SESSION_IO_EVT_RX;
1892           e->session_index = s->session_index;
1893         }
1894     }
1895   else if (PREDICT_FALSE (rv < n))
1896     {
1897       /* More data enqueued while reading. Try to drain it
1898        * or fill the buffer */
1899       buf += rv;
1900       n -= rv;
1901       goto read_again;
1902     }
1903
1904   if (PREDICT_FALSE (svm_fifo_needs_deq_ntf (rx_fifo, n_read)))
1905     {
1906       svm_fifo_clear_deq_ntf (rx_fifo);
1907       app_send_io_evt_to_vpp (s->vpp_evt_q, s->rx_fifo->master_session_index,
1908                               SESSION_IO_EVT_RX, SVM_Q_WAIT);
1909     }
1910
1911   VDBG (2, "session %u[0x%llx]: read %d bytes from (%p)", s->session_index,
1912         s->vpp_handle, n_read, rx_fifo);
1913
1914   return n_read;
1915 }
1916
1917 int
1918 vppcom_session_read (uint32_t session_handle, void *buf, size_t n)
1919 {
1920   return (vppcom_session_read_internal (session_handle, buf, n, 0));
1921 }
1922
1923 static int
1924 vppcom_session_peek (uint32_t session_handle, void *buf, int n)
1925 {
1926   return (vppcom_session_read_internal (session_handle, buf, n, 1));
1927 }
1928
1929 int
1930 vppcom_session_read_segments (uint32_t session_handle,
1931                               vppcom_data_segment_t * ds, uint32_t n_segments,
1932                               uint32_t max_bytes)
1933 {
1934   vcl_worker_t *wrk = vcl_worker_get_current ();
1935   int n_read = 0, is_nonblocking;
1936   vcl_session_t *s = 0;
1937   svm_fifo_t *rx_fifo;
1938   svm_msg_q_msg_t msg;
1939   session_event_t *e;
1940   svm_msg_q_t *mq;
1941   u8 is_ct;
1942
1943   s = vcl_session_get_w_handle (wrk, session_handle);
1944   if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
1945     return VPPCOM_EBADFD;
1946
1947   if (PREDICT_FALSE (!vcl_session_is_open (s)))
1948     return vcl_session_closed_error (s);
1949
1950   is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
1951   is_ct = vcl_session_is_ct (s);
1952   mq = wrk->app_event_queue;
1953   rx_fifo = s->rx_fifo;
1954   s->flags &= ~VCL_SESSION_F_HAS_RX_EVT;
1955
1956   if (is_ct)
1957     svm_fifo_unset_event (s->rx_fifo);
1958
1959   if (svm_fifo_is_empty_cons (rx_fifo))
1960     {
1961       if (is_nonblocking)
1962         {
1963           svm_fifo_unset_event (rx_fifo);
1964           return VPPCOM_EWOULDBLOCK;
1965         }
1966       while (svm_fifo_is_empty_cons (rx_fifo))
1967         {
1968           if (vcl_session_is_closing (s))
1969             return vcl_session_closing_error (s);
1970
1971           svm_fifo_unset_event (rx_fifo);
1972           svm_msg_q_lock (mq);
1973           if (svm_msg_q_is_empty (mq))
1974             svm_msg_q_wait (mq);
1975
1976           svm_msg_q_sub_w_lock (mq, &msg);
1977           e = svm_msg_q_msg_data (mq, &msg);
1978           svm_msg_q_unlock (mq);
1979           if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
1980             vcl_handle_mq_event (wrk, e);
1981           svm_msg_q_free_msg (mq, &msg);
1982         }
1983     }
1984
1985   n_read = svm_fifo_segments (rx_fifo, s->rx_bytes_pending,
1986                               (svm_fifo_seg_t *) ds, n_segments, max_bytes);
1987   if (n_read < 0)
1988     return VPPCOM_EAGAIN;
1989
1990   if (svm_fifo_max_dequeue_cons (rx_fifo) == n_read)
1991     {
1992       svm_fifo_unset_event (s->rx_fifo);
1993       if (svm_fifo_max_dequeue_cons (rx_fifo) != n_read
1994           && svm_fifo_set_event (s->rx_fifo)
1995           && vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK))
1996         {
1997           session_event_t *e;
1998           vec_add2 (wrk->unhandled_evts_vector, e, 1);
1999           e->event_type = SESSION_IO_EVT_RX;
2000           e->session_index = s->session_index;
2001         }
2002     }
2003
2004   s->rx_bytes_pending += n_read;
2005   return n_read;
2006 }
2007
2008 void
2009 vppcom_session_free_segments (uint32_t session_handle, uint32_t n_bytes)
2010 {
2011   vcl_worker_t *wrk = vcl_worker_get_current ();
2012   vcl_session_t *s;
2013
2014   s = vcl_session_get_w_handle (wrk, session_handle);
2015   if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
2016     return;
2017
2018   svm_fifo_dequeue_drop (s->rx_fifo, n_bytes);
2019
2020   ASSERT (s->rx_bytes_pending < n_bytes);
2021   s->rx_bytes_pending -= n_bytes;
2022 }
2023
2024 static u8
2025 vcl_is_tx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
2026 {
2027   return (e->event_type == SESSION_IO_EVT_TX && e->session_index == sid);
2028 }
2029
2030 always_inline u8
2031 vcl_fifo_is_writeable (svm_fifo_t * f, u32 len, u8 is_dgram)
2032 {
2033   u32 max_enq = svm_fifo_max_enqueue_prod (f);
2034   if (is_dgram)
2035     return max_enq >= (sizeof (session_dgram_hdr_t) + len);
2036   else
2037     return max_enq > 0;
2038 }
2039
2040 always_inline int
2041 vppcom_session_write_inline (vcl_worker_t * wrk, vcl_session_t * s, void *buf,
2042                              size_t n, u8 is_flush, u8 is_dgram)
2043 {
2044   int n_write, is_nonblocking;
2045   session_evt_type_t et;
2046   svm_msg_q_msg_t msg;
2047   svm_fifo_t *tx_fifo;
2048   session_event_t *e;
2049   svm_msg_q_t *mq;
2050   u8 is_ct;
2051
2052   if (PREDICT_FALSE (!buf || n == 0))
2053     return VPPCOM_EINVAL;
2054
2055   if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2056     {
2057       VDBG (0, "ERROR: session %u [0x%llx]: cannot write to an epoll"
2058             " session!", s->session_index, s->vpp_handle);
2059       return VPPCOM_EBADFD;
2060     }
2061
2062   if (PREDICT_FALSE (!vcl_session_is_open (s)))
2063     {
2064       VDBG (1, "session %u [0x%llx]: is not open! state 0x%x (%s)",
2065             s->session_index, s->vpp_handle, s->session_state,
2066             vppcom_session_state_str (s->session_state));
2067       return vcl_session_closed_error (s);;
2068     }
2069
2070   is_ct = vcl_session_is_ct (s);
2071   tx_fifo = is_ct ? s->ct_tx_fifo : s->tx_fifo;
2072   is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
2073
2074   mq = wrk->app_event_queue;
2075   if (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2076     {
2077       if (is_nonblocking)
2078         {
2079           return VPPCOM_EWOULDBLOCK;
2080         }
2081       while (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
2082         {
2083           svm_fifo_add_want_deq_ntf (tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
2084           if (vcl_session_is_closing (s))
2085             return vcl_session_closing_error (s);
2086           svm_msg_q_lock (mq);
2087           if (svm_msg_q_is_empty (mq))
2088             svm_msg_q_wait (mq);
2089
2090           svm_msg_q_sub_w_lock (mq, &msg);
2091           e = svm_msg_q_msg_data (mq, &msg);
2092           svm_msg_q_unlock (mq);
2093
2094           if (!vcl_is_tx_evt_for_session (e, s->session_index, is_ct))
2095             vcl_handle_mq_event (wrk, e);
2096           svm_msg_q_free_msg (mq, &msg);
2097         }
2098     }
2099
2100   et = SESSION_IO_EVT_TX;
2101   if (is_flush && !is_ct)
2102     et = SESSION_IO_EVT_TX_FLUSH;
2103
2104   if (is_dgram)
2105     n_write = app_send_dgram_raw (tx_fifo, &s->transport,
2106                                   s->vpp_evt_q, buf, n, et,
2107                                   0 /* do_evt */ , SVM_Q_WAIT);
2108   else
2109     n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
2110                                    0 /* do_evt */ , SVM_Q_WAIT);
2111
2112   if (svm_fifo_set_event (s->tx_fifo))
2113     app_send_io_evt_to_vpp (s->vpp_evt_q, s->tx_fifo->master_session_index,
2114                             et, SVM_Q_WAIT);
2115
2116   /* The underlying fifo segment can run out of memory */
2117   if (PREDICT_FALSE (n_write < 0))
2118     return VPPCOM_EAGAIN;
2119
2120   VDBG (2, "session %u [0x%llx]: wrote %d bytes", s->session_index,
2121         s->vpp_handle, n_write);
2122
2123   return n_write;
2124 }
2125
2126 int
2127 vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
2128 {
2129   vcl_worker_t *wrk = vcl_worker_get_current ();
2130   vcl_session_t *s;
2131
2132   s = vcl_session_get_w_handle (wrk, session_handle);
2133   if (PREDICT_FALSE (!s))
2134     return VPPCOM_EBADFD;
2135
2136   return vppcom_session_write_inline (wrk, s, buf, n,
2137                                       0 /* is_flush */ , s->is_dgram ? 1 : 0);
2138 }
2139
2140 int
2141 vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
2142 {
2143   vcl_worker_t *wrk = vcl_worker_get_current ();
2144   vcl_session_t *s;
2145
2146   s = vcl_session_get_w_handle (wrk, session_handle);
2147   if (PREDICT_FALSE (!s))
2148     return VPPCOM_EBADFD;
2149
2150   return vppcom_session_write_inline (wrk, s, buf, n,
2151                                       1 /* is_flush */ , s->is_dgram ? 1 : 0);
2152 }
2153
2154 #define vcl_fifo_rx_evt_valid_or_break(_s)                              \
2155 if (PREDICT_FALSE (!_s->rx_fifo))                                       \
2156   break;                                                                \
2157 if (PREDICT_FALSE (svm_fifo_is_empty (_s->rx_fifo)))                    \
2158   {                                                                     \
2159     if (!vcl_session_is_ct (_s))                                        \
2160       {                                                                 \
2161         svm_fifo_unset_event (_s->rx_fifo);                             \
2162         if (svm_fifo_is_empty (_s->rx_fifo))                            \
2163           break;                                                        \
2164       }                                                                 \
2165     else if (svm_fifo_is_empty (_s->ct_rx_fifo))                        \
2166       {                                                                 \
2167         svm_fifo_unset_event (_s->rx_fifo); /* rx evts on actual fifo*/ \
2168         if (svm_fifo_is_empty (_s->ct_rx_fifo))                         \
2169           break;                                                        \
2170       }                                                                 \
2171   }                                                                     \
2172
2173 static void
2174 vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2175                             unsigned long n_bits, unsigned long *read_map,
2176                             unsigned long *write_map,
2177                             unsigned long *except_map, u32 * bits_set)
2178 {
2179   session_disconnected_msg_t *disconnected_msg;
2180   session_connected_msg_t *connected_msg;
2181   vcl_session_t *session;
2182   u32 sid;
2183
2184   switch (e->event_type)
2185     {
2186     case SESSION_IO_EVT_RX:
2187       sid = e->session_index;
2188       session = vcl_session_get (wrk, sid);
2189       if (!session || !vcl_session_is_open (session))
2190         break;
2191       vcl_fifo_rx_evt_valid_or_break (session);
2192       if (sid < n_bits && read_map)
2193         {
2194           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2195           *bits_set += 1;
2196         }
2197       break;
2198     case SESSION_IO_EVT_TX:
2199       sid = e->session_index;
2200       session = vcl_session_get (wrk, sid);
2201       if (!session || !vcl_session_is_open (session))
2202         break;
2203       if (sid < n_bits && write_map)
2204         {
2205           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2206           *bits_set += 1;
2207         }
2208       break;
2209     case SESSION_CTRL_EVT_ACCEPTED:
2210       session = vcl_session_accepted (wrk,
2211                                       (session_accepted_msg_t *) e->data);
2212       if (!session)
2213         break;
2214       sid = session->session_index;
2215       if (sid < n_bits && read_map)
2216         {
2217           clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2218           *bits_set += 1;
2219         }
2220       break;
2221     case SESSION_CTRL_EVT_CONNECTED:
2222       connected_msg = (session_connected_msg_t *) e->data;
2223       sid = vcl_session_connected_handler (wrk, connected_msg);
2224       if (sid == VCL_INVALID_SESSION_INDEX)
2225         break;
2226       if (sid < n_bits && write_map)
2227         {
2228           clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2229           *bits_set += 1;
2230         }
2231       break;
2232     case SESSION_CTRL_EVT_DISCONNECTED:
2233       disconnected_msg = (session_disconnected_msg_t *) e->data;
2234       session = vcl_session_disconnected_handler (wrk, disconnected_msg);
2235       if (!session)
2236         break;
2237       sid = session->session_index;
2238       if (sid < n_bits && except_map)
2239         {
2240           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2241           *bits_set += 1;
2242         }
2243       break;
2244     case SESSION_CTRL_EVT_RESET:
2245       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2246       if (sid < n_bits && except_map)
2247         {
2248           clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
2249           *bits_set += 1;
2250         }
2251       break;
2252     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2253       vcl_session_unlisten_reply_handler (wrk, e->data);
2254       break;
2255     case SESSION_CTRL_EVT_MIGRATED:
2256       vcl_session_migrated_handler (wrk, e->data);
2257       break;
2258     case SESSION_CTRL_EVT_CLEANUP:
2259       vcl_session_cleanup_handler (wrk, e->data);
2260       break;
2261     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2262       vcl_session_worker_update_reply_handler (wrk, e->data);
2263       break;
2264     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2265       vcl_session_req_worker_update_handler (wrk, e->data);
2266       break;
2267     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2268       vcl_session_app_add_segment_handler (wrk, e->data);
2269       break;
2270     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2271       vcl_session_app_del_segment_handler (wrk, e->data);
2272       break;
2273     case SESSION_CTRL_EVT_APP_WRK_RPC:
2274       vcl_worker_rpc_handler (wrk, e->data);
2275       break;
2276     default:
2277       clib_warning ("unhandled: %u", e->event_type);
2278       break;
2279     }
2280 }
2281
2282 static int
2283 vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2284                       unsigned long n_bits, unsigned long *read_map,
2285                       unsigned long *write_map, unsigned long *except_map,
2286                       double time_to_wait, u32 * bits_set)
2287 {
2288   svm_msg_q_msg_t *msg;
2289   session_event_t *e;
2290   u32 i;
2291
2292   svm_msg_q_lock (mq);
2293   if (svm_msg_q_is_empty (mq))
2294     {
2295       if (*bits_set)
2296         {
2297           svm_msg_q_unlock (mq);
2298           return 0;
2299         }
2300
2301       if (!time_to_wait)
2302         {
2303           svm_msg_q_unlock (mq);
2304           return 0;
2305         }
2306       else if (time_to_wait < 0)
2307         {
2308           svm_msg_q_wait (mq);
2309         }
2310       else
2311         {
2312           if (svm_msg_q_timedwait (mq, time_to_wait))
2313             {
2314               svm_msg_q_unlock (mq);
2315               return 0;
2316             }
2317         }
2318     }
2319   vcl_mq_dequeue_batch (wrk, mq, ~0);
2320   svm_msg_q_unlock (mq);
2321
2322   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
2323     {
2324       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
2325       e = svm_msg_q_msg_data (mq, msg);
2326       vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
2327                                   except_map, bits_set);
2328       svm_msg_q_free_msg (mq, msg);
2329     }
2330   vec_reset_length (wrk->mq_msg_vector);
2331   vcl_handle_pending_wrk_updates (wrk);
2332   return *bits_set;
2333 }
2334
2335 static int
2336 vppcom_select_condvar (vcl_worker_t * wrk, int n_bits,
2337                        vcl_si_set * read_map, vcl_si_set * write_map,
2338                        vcl_si_set * except_map, double time_to_wait,
2339                        u32 * bits_set)
2340 {
2341   double wait = 0, start = 0;
2342
2343   if (!*bits_set)
2344     {
2345       wait = time_to_wait;
2346       start = clib_time_now (&wrk->clib_time);
2347     }
2348
2349   do
2350     {
2351       vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
2352                             write_map, except_map, wait, bits_set);
2353       if (*bits_set)
2354         return *bits_set;
2355       if (wait == -1)
2356         continue;
2357
2358       wait = wait - (clib_time_now (&wrk->clib_time) - start);
2359     }
2360   while (wait > 0);
2361
2362   return 0;
2363 }
2364
2365 static int
2366 vppcom_select_eventfd (vcl_worker_t * wrk, int n_bits,
2367                        vcl_si_set * read_map, vcl_si_set * write_map,
2368                        vcl_si_set * except_map, double time_to_wait,
2369                        u32 * bits_set)
2370 {
2371   vcl_mq_evt_conn_t *mqc;
2372   int __clib_unused n_read;
2373   int n_mq_evts, i;
2374   u64 buf;
2375
2376   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
2377   n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
2378                           vec_len (wrk->mq_events), time_to_wait);
2379   for (i = 0; i < n_mq_evts; i++)
2380     {
2381       mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
2382       n_read = read (mqc->mq_fd, &buf, sizeof (buf));
2383       vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
2384                             except_map, 0, bits_set);
2385     }
2386
2387   return (n_mq_evts > 0 ? (int) *bits_set : 0);
2388 }
2389
2390 int
2391 vppcom_select (int n_bits, vcl_si_set * read_map, vcl_si_set * write_map,
2392                vcl_si_set * except_map, double time_to_wait)
2393 {
2394   u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
2395   vcl_worker_t *wrk = vcl_worker_get_current ();
2396   vcl_session_t *session = 0;
2397   int i;
2398
2399   if (n_bits && read_map)
2400     {
2401       clib_bitmap_validate (wrk->rd_bitmap, minbits);
2402       clib_memcpy_fast (wrk->rd_bitmap, read_map,
2403                         vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2404       memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2405     }
2406   if (n_bits && write_map)
2407     {
2408       clib_bitmap_validate (wrk->wr_bitmap, minbits);
2409       clib_memcpy_fast (wrk->wr_bitmap, write_map,
2410                         vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2411       memset (write_map, 0, vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2412     }
2413   if (n_bits && except_map)
2414     {
2415       clib_bitmap_validate (wrk->ex_bitmap, minbits);
2416       clib_memcpy_fast (wrk->ex_bitmap, except_map,
2417                         vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2418       memset (except_map, 0, vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2419     }
2420
2421   if (!n_bits)
2422     return 0;
2423
2424   if (!write_map)
2425     goto check_rd;
2426
2427   /* *INDENT-OFF* */
2428   clib_bitmap_foreach (sid, wrk->wr_bitmap, ({
2429     if (!(session = vcl_session_get (wrk, sid)))
2430       {
2431         clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
2432         bits_set++;
2433         continue;
2434       }
2435
2436     if (vcl_session_write_ready (session))
2437       {
2438         clib_bitmap_set_no_check ((uword*)write_map, sid, 1);
2439         bits_set++;
2440       }
2441     else
2442       svm_fifo_add_want_deq_ntf (session->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
2443   }));
2444
2445 check_rd:
2446   if (!read_map)
2447     goto check_mq;
2448
2449   clib_bitmap_foreach (sid, wrk->rd_bitmap, ({
2450     if (!(session = vcl_session_get (wrk, sid)))
2451       {
2452         clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
2453         bits_set++;
2454         continue;
2455       }
2456
2457     if (vcl_session_read_ready (session))
2458       {
2459         clib_bitmap_set_no_check ((uword*)read_map, sid, 1);
2460         bits_set++;
2461       }
2462   }));
2463   /* *INDENT-ON* */
2464
2465 check_mq:
2466
2467   for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
2468     {
2469       vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
2470                                   read_map, write_map, except_map, &bits_set);
2471     }
2472   vec_reset_length (wrk->unhandled_evts_vector);
2473
2474   if (vcm->cfg.use_mq_eventfd)
2475     vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
2476                            time_to_wait, &bits_set);
2477   else
2478     vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
2479                            time_to_wait, &bits_set);
2480
2481   return (bits_set);
2482 }
2483
2484 static inline void
2485 vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_handle)
2486 {
2487   vppcom_epoll_t *vep;
2488   u32 sh = vep_handle;
2489   vcl_session_t *s;
2490
2491   if (VPPCOM_DEBUG <= 2)
2492     return;
2493
2494   s = vcl_session_get_w_handle (wrk, vep_handle);
2495   if (PREDICT_FALSE (!s))
2496     {
2497       VDBG (0, "ERROR: Invalid vep_sh (%u)!", vep_handle);
2498       goto done;
2499     }
2500   if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP)))
2501     {
2502       VDBG (0, "ERROR: vep_sh (%u) is not a vep!", vep_handle);
2503       goto done;
2504     }
2505   vep = &s->vep;
2506   VDBG (0, "vep_sh (%u): Dumping epoll chain\n"
2507         "{\n"
2508         "   is_vep         = %u\n"
2509         "   is_vep_session = %u\n"
2510         "   next_sh        = 0x%x (%u)\n"
2511         "}\n", vep_handle, s->flags & VCL_SESSION_F_IS_VEP,
2512         s->flags & VCL_SESSION_F_IS_VEP_SESSION, vep->next_sh, vep->next_sh);
2513
2514   for (sh = vep->next_sh; sh != ~0; sh = vep->next_sh)
2515     {
2516       s = vcl_session_get_w_handle (wrk, sh);
2517       if (PREDICT_FALSE (!s))
2518         {
2519           VDBG (0, "ERROR: Invalid sh (%u)!", sh);
2520           goto done;
2521         }
2522       if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2523         {
2524           VDBG (0, "ERROR: sh (%u) is a vep!", vep_handle);
2525         }
2526       else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2527         {
2528           VDBG (0, "ERROR: sh (%u) is not a vep session handle!", sh);
2529           goto done;
2530         }
2531       vep = &s->vep;
2532       if (PREDICT_FALSE (vep->vep_sh != vep_handle))
2533         VDBG (0, "ERROR: session (%u) vep_sh (%u) != vep_sh (%u)!",
2534               sh, s->vep.vep_sh, vep_handle);
2535       if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
2536         {
2537           VDBG (0, "vep_sh[%u]: sh 0x%x (%u)\n"
2538                 "{\n"
2539                 "   next_sh        = 0x%x (%u)\n"
2540                 "   prev_sh        = 0x%x (%u)\n"
2541                 "   vep_sh         = 0x%x (%u)\n"
2542                 "   ev.events      = 0x%x\n"
2543                 "   ev.data.u64    = 0x%llx\n"
2544                 "   et_mask        = 0x%x\n"
2545                 "}\n",
2546                 vep_handle, sh, sh, vep->next_sh, vep->next_sh, vep->prev_sh,
2547                 vep->prev_sh, vep->vep_sh, vep->vep_sh, vep->ev.events,
2548                 vep->ev.data.u64, vep->et_mask);
2549         }
2550     }
2551
2552 done:
2553   VDBG (0, "vep_sh (%u): Dump complete!\n", vep_handle);
2554 }
2555
2556 int
2557 vppcom_epoll_create (void)
2558 {
2559   vcl_worker_t *wrk = vcl_worker_get_current ();
2560   vcl_session_t *vep_session;
2561
2562   vep_session = vcl_session_alloc (wrk);
2563
2564   vep_session->flags |= VCL_SESSION_F_IS_VEP;
2565   vep_session->vep.vep_sh = ~0;
2566   vep_session->vep.next_sh = ~0;
2567   vep_session->vep.prev_sh = ~0;
2568   vep_session->vpp_handle = ~0;
2569
2570   vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
2571   VDBG (0, "Created vep_idx %u", vep_session->session_index);
2572
2573   return vcl_session_handle (vep_session);
2574 }
2575
2576 int
2577 vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
2578                   struct epoll_event *event)
2579 {
2580   vcl_worker_t *wrk = vcl_worker_get_current ();
2581   vcl_session_t *vep_session;
2582   int rv = VPPCOM_OK;
2583   vcl_session_t *s;
2584   svm_fifo_t *txf;
2585
2586   if (vep_handle == session_handle)
2587     {
2588       VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
2589       return VPPCOM_EINVAL;
2590     }
2591
2592   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
2593   if (PREDICT_FALSE (!vep_session))
2594     {
2595       VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
2596       return VPPCOM_EBADFD;
2597     }
2598   if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
2599     {
2600       VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
2601       return VPPCOM_EINVAL;
2602     }
2603
2604   ASSERT (vep_session->vep.vep_sh == ~0);
2605   ASSERT (vep_session->vep.prev_sh == ~0);
2606
2607   s = vcl_session_get_w_handle (wrk, session_handle);
2608   if (PREDICT_FALSE (!s))
2609     {
2610       VDBG (0, "Invalid session_handle (%u)!", session_handle);
2611       return VPPCOM_EBADFD;
2612     }
2613   if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
2614     {
2615       VDBG (0, "session_handle (%u) is a vep!", vep_handle);
2616       return VPPCOM_EINVAL;
2617     }
2618
2619   switch (op)
2620     {
2621     case EPOLL_CTL_ADD:
2622       if (PREDICT_FALSE (!event))
2623         {
2624           VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
2625           return VPPCOM_EINVAL;
2626         }
2627       if (vep_session->vep.next_sh != ~0)
2628         {
2629           vcl_session_t *next_session;
2630           next_session = vcl_session_get_w_handle (wrk,
2631                                                    vep_session->vep.next_sh);
2632           if (PREDICT_FALSE (!next_session))
2633             {
2634               VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
2635                     "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
2636               return VPPCOM_EBADFD;
2637             }
2638           ASSERT (next_session->vep.prev_sh == vep_handle);
2639           next_session->vep.prev_sh = session_handle;
2640         }
2641       s->vep.next_sh = vep_session->vep.next_sh;
2642       s->vep.prev_sh = vep_handle;
2643       s->vep.vep_sh = vep_handle;
2644       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2645       s->vep.ev = *event;
2646       s->flags &= ~VCL_SESSION_F_IS_VEP;
2647       s->flags |= VCL_SESSION_F_IS_VEP_SESSION;
2648       vep_session->vep.next_sh = session_handle;
2649
2650       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2651       if (txf && (event->events & EPOLLOUT))
2652         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2653
2654       /* Generate EPOLLOUT if tx fifo not full */
2655       if ((event->events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2656         {
2657           session_event_t e = { 0 };
2658           e.event_type = SESSION_IO_EVT_TX;
2659           e.session_index = s->session_index;
2660           vec_add1 (wrk->unhandled_evts_vector, e);
2661         }
2662       /* Generate EPOLLIN if rx fifo has data */
2663       if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
2664         {
2665           session_event_t e = { 0 };
2666           e.event_type = SESSION_IO_EVT_RX;
2667           e.session_index = s->session_index;
2668           vec_add1 (wrk->unhandled_evts_vector, e);
2669         }
2670       VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2671             vep_handle, session_handle, event->events, event->data.u64);
2672       vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
2673       break;
2674
2675     case EPOLL_CTL_MOD:
2676       if (PREDICT_FALSE (!event))
2677         {
2678           VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
2679           rv = VPPCOM_EINVAL;
2680           goto done;
2681         }
2682       else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2683         {
2684           VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
2685           rv = VPPCOM_EINVAL;
2686           goto done;
2687         }
2688       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2689         {
2690           VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
2691                 session_handle, s->vep.vep_sh, vep_handle);
2692           rv = VPPCOM_EINVAL;
2693           goto done;
2694         }
2695
2696       /* Generate EPOLLOUT when tx_fifo/ct_tx_fifo not full */
2697       if ((event->events & EPOLLOUT) &&
2698           !(s->vep.ev.events & EPOLLOUT) && (vcl_session_write_ready (s) > 0))
2699         {
2700           session_event_t e = { 0 };
2701           e.event_type = SESSION_IO_EVT_TX;
2702           e.session_index = s->session_index;
2703           vec_add1 (wrk->unhandled_evts_vector, e);
2704         }
2705       s->vep.et_mask = VEP_DEFAULT_ET_MASK;
2706       s->vep.ev = *event;
2707       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2708       if (event->events & EPOLLOUT)
2709         svm_fifo_add_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2710       else
2711         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2712       VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2713             vep_handle, session_handle, event->events, event->data.u64);
2714       break;
2715
2716     case EPOLL_CTL_DEL:
2717       if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
2718         {
2719           VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
2720           rv = VPPCOM_EINVAL;
2721           goto done;
2722         }
2723       else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
2724         {
2725           VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
2726                 session_handle, s->vep.vep_sh, vep_handle);
2727           rv = VPPCOM_EINVAL;
2728           goto done;
2729         }
2730
2731       if (s->vep.prev_sh == vep_handle)
2732         vep_session->vep.next_sh = s->vep.next_sh;
2733       else
2734         {
2735           vcl_session_t *prev_session;
2736           prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
2737           if (PREDICT_FALSE (!prev_session))
2738             {
2739               VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
2740                     s->vep.prev_sh, session_handle);
2741               return VPPCOM_EBADFD;
2742             }
2743           ASSERT (prev_session->vep.next_sh == session_handle);
2744           prev_session->vep.next_sh = s->vep.next_sh;
2745         }
2746       if (s->vep.next_sh != ~0)
2747         {
2748           vcl_session_t *next_session;
2749           next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
2750           if (PREDICT_FALSE (!next_session))
2751             {
2752               VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
2753                     s->vep.next_sh, session_handle);
2754               return VPPCOM_EBADFD;
2755             }
2756           ASSERT (next_session->vep.prev_sh == session_handle);
2757           next_session->vep.prev_sh = s->vep.prev_sh;
2758         }
2759
2760       memset (&s->vep, 0, sizeof (s->vep));
2761       s->vep.next_sh = ~0;
2762       s->vep.prev_sh = ~0;
2763       s->vep.vep_sh = ~0;
2764       s->flags &= ~VCL_SESSION_F_IS_VEP_SESSION;
2765
2766       txf = vcl_session_is_ct (s) ? s->ct_tx_fifo : s->tx_fifo;
2767       if (txf)
2768         svm_fifo_del_want_deq_ntf (txf, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2769
2770       VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
2771             session_handle);
2772       vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
2773       break;
2774
2775     default:
2776       VDBG (0, "Invalid operation (%d)!", op);
2777       rv = VPPCOM_EINVAL;
2778     }
2779
2780   vep_verify_epoll_chain (wrk, vep_handle);
2781
2782 done:
2783   return rv;
2784 }
2785
2786 static inline void
2787 vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2788                                 struct epoll_event *events, u32 * num_ev)
2789 {
2790   session_disconnected_msg_t *disconnected_msg;
2791   session_connected_msg_t *connected_msg;
2792   u32 sid = ~0, session_events;
2793   u64 session_evt_data = ~0;
2794   vcl_session_t *session;
2795   u8 add_event = 0;
2796
2797   switch (e->event_type)
2798     {
2799     case SESSION_IO_EVT_RX:
2800       sid = e->session_index;
2801       session = vcl_session_get (wrk, sid);
2802       if (vcl_session_is_closed (session))
2803         break;
2804       vcl_fifo_rx_evt_valid_or_break (session);
2805       session_events = session->vep.ev.events;
2806       if (!(EPOLLIN & session->vep.ev.events)
2807           || (session->flags & VCL_SESSION_F_HAS_RX_EVT))
2808         break;
2809       add_event = 1;
2810       events[*num_ev].events |= EPOLLIN;
2811       session_evt_data = session->vep.ev.data.u64;
2812       session->flags |= VCL_SESSION_F_HAS_RX_EVT;
2813       break;
2814     case SESSION_IO_EVT_TX:
2815       sid = e->session_index;
2816       session = vcl_session_get (wrk, sid);
2817       if (vcl_session_is_closed (session))
2818         break;
2819       session_events = session->vep.ev.events;
2820       if (!(EPOLLOUT & session_events))
2821         break;
2822       add_event = 1;
2823       events[*num_ev].events |= EPOLLOUT;
2824       session_evt_data = session->vep.ev.data.u64;
2825       svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (session) ?
2826                                   session->ct_tx_fifo : session->tx_fifo);
2827       break;
2828     case SESSION_CTRL_EVT_ACCEPTED:
2829       session = vcl_session_accepted (wrk,
2830                                       (session_accepted_msg_t *) e->data);
2831       if (!session)
2832         break;
2833
2834       session_events = session->vep.ev.events;
2835       if (!(EPOLLIN & session_events))
2836         break;
2837
2838       add_event = 1;
2839       events[*num_ev].events |= EPOLLIN;
2840       session_evt_data = session->vep.ev.data.u64;
2841       break;
2842     case SESSION_CTRL_EVT_CONNECTED:
2843       connected_msg = (session_connected_msg_t *) e->data;
2844       sid = vcl_session_connected_handler (wrk, connected_msg);
2845       /* Generate EPOLLOUT because there's no connected event */
2846       session = vcl_session_get (wrk, sid);
2847       if (vcl_session_is_closed (session))
2848         break;
2849       session_events = session->vep.ev.events;
2850       if (!(EPOLLOUT & session_events))
2851         break;
2852       add_event = 1;
2853       events[*num_ev].events |= EPOLLOUT;
2854       session_evt_data = session->vep.ev.data.u64;
2855       if (session->session_state == VCL_STATE_DETACHED)
2856         events[*num_ev].events |= EPOLLHUP;
2857       break;
2858     case SESSION_CTRL_EVT_DISCONNECTED:
2859       disconnected_msg = (session_disconnected_msg_t *) e->data;
2860       session = vcl_session_disconnected_handler (wrk, disconnected_msg);
2861       if (vcl_session_is_closed (session))
2862         break;
2863       session_events = session->vep.ev.events;
2864       add_event = 1;
2865       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2866       session_evt_data = session->vep.ev.data.u64;
2867       break;
2868     case SESSION_CTRL_EVT_RESET:
2869       sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2870       session = vcl_session_get (wrk, sid);
2871       if (vcl_session_is_closed (session))
2872         break;
2873       session_events = session->vep.ev.events;
2874       add_event = 1;
2875       events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
2876       session_evt_data = session->vep.ev.data.u64;
2877       break;
2878     case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2879       vcl_session_unlisten_reply_handler (wrk, e->data);
2880       break;
2881     case SESSION_CTRL_EVT_MIGRATED:
2882       vcl_session_migrated_handler (wrk, e->data);
2883       break;
2884     case SESSION_CTRL_EVT_CLEANUP:
2885       vcl_session_cleanup_handler (wrk, e->data);
2886       break;
2887     case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2888       vcl_session_req_worker_update_handler (wrk, e->data);
2889       break;
2890     case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2891       vcl_session_worker_update_reply_handler (wrk, e->data);
2892       break;
2893     case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2894       vcl_session_app_add_segment_handler (wrk, e->data);
2895       break;
2896     case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2897       vcl_session_app_del_segment_handler (wrk, e->data);
2898       break;
2899     case SESSION_CTRL_EVT_APP_WRK_RPC:
2900       vcl_worker_rpc_handler (wrk, e->data);
2901       break;
2902     default:
2903       VDBG (0, "unhandled: %u", e->event_type);
2904       break;
2905     }
2906
2907   if (add_event)
2908     {
2909       events[*num_ev].data.u64 = session_evt_data;
2910       if (EPOLLONESHOT & session_events)
2911         {
2912           session = vcl_session_get (wrk, sid);
2913           session->vep.ev.events = 0;
2914         }
2915       *num_ev += 1;
2916     }
2917 }
2918
2919 static int
2920 vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2921                           struct epoll_event *events, u32 maxevents,
2922                           double wait_for_time, u32 * num_ev)
2923 {
2924   svm_msg_q_msg_t *msg;
2925   session_event_t *e;
2926   int i;
2927
2928   if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
2929     goto handle_dequeued;
2930
2931   svm_msg_q_lock (mq);
2932   if (svm_msg_q_is_empty (mq))
2933     {
2934       if (!wait_for_time)
2935         {
2936           svm_msg_q_unlock (mq);
2937           return 0;
2938         }
2939       else if (wait_for_time < 0)
2940         {
2941           svm_msg_q_wait (mq);
2942         }
2943       else
2944         {
2945           if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
2946             {
2947               svm_msg_q_unlock (mq);
2948               return 0;
2949             }
2950         }
2951     }
2952   ASSERT (maxevents > *num_ev);
2953   vcl_mq_dequeue_batch (wrk, mq, ~0);
2954   svm_msg_q_unlock (mq);
2955
2956 handle_dequeued:
2957   for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
2958     {
2959       msg = vec_elt_at_index (wrk->mq_msg_vector, i);
2960       e = svm_msg_q_msg_data (mq, msg);
2961       if (*num_ev < maxevents)
2962         vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
2963       else
2964         vcl_handle_mq_event (wrk, e);
2965       svm_msg_q_free_msg (mq, msg);
2966     }
2967   vec_reset_length (wrk->mq_msg_vector);
2968   vcl_handle_pending_wrk_updates (wrk);
2969   return *num_ev;
2970 }
2971
2972 static int
2973 vppcom_epoll_wait_condvar (vcl_worker_t * wrk, struct epoll_event *events,
2974                            int maxevents, u32 n_evts, double wait_for_time)
2975 {
2976   double wait = 0, start = 0, now;
2977
2978   if (!n_evts)
2979     {
2980       wait = wait_for_time;
2981       start = clib_time_now (&wrk->clib_time);
2982     }
2983
2984   do
2985     {
2986       vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
2987                                 wait, &n_evts);
2988       if (n_evts)
2989         return n_evts;
2990       if (wait == -1)
2991         continue;
2992
2993       now = clib_time_now (&wrk->clib_time);
2994       wait -= (now - start) * 1e3;
2995       start = now;
2996     }
2997   while (wait > 0);
2998
2999   return 0;
3000 }
3001
3002 static int
3003 vppcom_epoll_wait_eventfd (vcl_worker_t * wrk, struct epoll_event *events,
3004                            int maxevents, u32 n_evts, double wait_for_time)
3005 {
3006   vcl_mq_evt_conn_t *mqc;
3007   int __clib_unused n_read;
3008   int n_mq_evts, i;
3009   u64 buf;
3010
3011   vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
3012 again:
3013   n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
3014                           vec_len (wrk->mq_events), wait_for_time);
3015   for (i = 0; i < n_mq_evts; i++)
3016     {
3017       mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
3018       n_read = read (mqc->mq_fd, &buf, sizeof (buf));
3019       vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0, &n_evts);
3020     }
3021   if (!n_evts && n_mq_evts > 0)
3022     goto again;
3023
3024   return (int) n_evts;
3025 }
3026
3027 int
3028 vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
3029                    int maxevents, double wait_for_time)
3030 {
3031   vcl_worker_t *wrk = vcl_worker_get_current ();
3032   vcl_session_t *vep_session;
3033   u32 n_evts = 0;
3034   int i;
3035
3036   if (PREDICT_FALSE (maxevents <= 0))
3037     {
3038       VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
3039       return VPPCOM_EINVAL;
3040     }
3041
3042   vep_session = vcl_session_get_w_handle (wrk, vep_handle);
3043   if (!vep_session)
3044     return VPPCOM_EBADFD;
3045
3046   if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
3047     {
3048       VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
3049       return VPPCOM_EINVAL;
3050     }
3051
3052   memset (events, 0, sizeof (*events) * maxevents);
3053
3054   if (vec_len (wrk->unhandled_evts_vector))
3055     {
3056       for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3057         {
3058           vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3059                                           events, &n_evts);
3060           if (n_evts == maxevents)
3061             {
3062               vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3063               return n_evts;
3064             }
3065         }
3066       vec_reset_length (wrk->unhandled_evts_vector);
3067     }
3068
3069   if (vcm->cfg.use_mq_eventfd)
3070     return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3071                                       wait_for_time);
3072
3073   return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3074                                     wait_for_time);
3075 }
3076
3077 int
3078 vppcom_session_attr (uint32_t session_handle, uint32_t op,
3079                      void *buffer, uint32_t * buflen)
3080 {
3081   vcl_worker_t *wrk = vcl_worker_get_current ();
3082   vcl_session_t *session;
3083   int rv = VPPCOM_OK;
3084   u32 *flags = buffer, tmp_flags = 0;
3085   vppcom_endpt_t *ep = buffer;
3086
3087   session = vcl_session_get_w_handle (wrk, session_handle);
3088   if (!session)
3089     return VPPCOM_EBADFD;
3090
3091   switch (op)
3092     {
3093     case VPPCOM_ATTR_GET_NREAD:
3094       rv = vcl_session_read_ready (session);
3095       VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3096             rv);
3097       break;
3098
3099     case VPPCOM_ATTR_GET_NWRITE:
3100       rv = vcl_session_write_ready (session);
3101       VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3102             rv);
3103       break;
3104
3105     case VPPCOM_ATTR_GET_FLAGS:
3106       if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
3107         {
3108           *flags =
3109             O_RDWR |
3110             (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK) ?
3111              O_NONBLOCK : 0);
3112           *buflen = sizeof (*flags);
3113           VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3114                 "is_nonblocking = %u", session_handle, *flags,
3115                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3116         }
3117       else
3118         rv = VPPCOM_EINVAL;
3119       break;
3120
3121     case VPPCOM_ATTR_SET_FLAGS:
3122       if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
3123         {
3124           if (*flags & O_NONBLOCK)
3125             vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
3126           else
3127             vcl_session_clear_attr (session, VCL_SESS_ATTR_NONBLOCK);
3128
3129           VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3130                 " is_nonblocking = %u", session_handle, *flags,
3131                 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
3132         }
3133       else
3134         rv = VPPCOM_EINVAL;
3135       break;
3136
3137     case VPPCOM_ATTR_GET_PEER_ADDR:
3138       if (PREDICT_TRUE (buffer && buflen &&
3139                         (*buflen >= sizeof (*ep)) && ep->ip))
3140         {
3141           ep->is_ip4 = session->transport.is_ip4;
3142           ep->port = session->transport.rmt_port;
3143           if (session->transport.is_ip4)
3144             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3145                               sizeof (ip4_address_t));
3146           else
3147             clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3148                               sizeof (ip6_address_t));
3149           *buflen = sizeof (*ep);
3150           VDBG (1, "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3151                 "addr = %U, port %u", session_handle, ep->is_ip4,
3152                 format_ip46_address, &session->transport.rmt_ip,
3153                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3154                 clib_net_to_host_u16 (ep->port));
3155         }
3156       else
3157         rv = VPPCOM_EINVAL;
3158       break;
3159
3160     case VPPCOM_ATTR_GET_LCL_ADDR:
3161       if (PREDICT_TRUE (buffer && buflen &&
3162                         (*buflen >= sizeof (*ep)) && ep->ip))
3163         {
3164           ep->is_ip4 = session->transport.is_ip4;
3165           ep->port = session->transport.lcl_port;
3166           if (session->transport.is_ip4)
3167             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3168                               sizeof (ip4_address_t));
3169           else
3170             clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3171                               sizeof (ip6_address_t));
3172           *buflen = sizeof (*ep);
3173           VDBG (1, "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3174                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3175                 &session->transport.lcl_ip,
3176                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3177                 clib_net_to_host_u16 (ep->port));
3178         }
3179       else
3180         rv = VPPCOM_EINVAL;
3181       break;
3182
3183     case VPPCOM_ATTR_SET_LCL_ADDR:
3184       if (PREDICT_TRUE (buffer && buflen &&
3185                         (*buflen >= sizeof (*ep)) && ep->ip))
3186         {
3187           session->transport.is_ip4 = ep->is_ip4;
3188           session->transport.lcl_port = ep->port;
3189           vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3190           *buflen = sizeof (*ep);
3191           VDBG (1, "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3192                 " port %d", session_handle, ep->is_ip4, format_ip46_address,
3193                 &session->transport.lcl_ip,
3194                 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3195                 clib_net_to_host_u16 (ep->port));
3196         }
3197       else
3198         rv = VPPCOM_EINVAL;
3199       break;
3200
3201     case VPPCOM_ATTR_GET_LIBC_EPFD:
3202       rv = session->libc_epfd;
3203       VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
3204       break;
3205
3206     case VPPCOM_ATTR_SET_LIBC_EPFD:
3207       if (PREDICT_TRUE (buffer && buflen &&
3208                         (*buflen == sizeof (session->libc_epfd))))
3209         {
3210           session->libc_epfd = *(int *) buffer;
3211           *buflen = sizeof (session->libc_epfd);
3212
3213           VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3214                 session->libc_epfd, *buflen);
3215         }
3216       else
3217         rv = VPPCOM_EINVAL;
3218       break;
3219
3220     case VPPCOM_ATTR_GET_PROTOCOL:
3221       if (buffer && buflen && (*buflen >= sizeof (int)))
3222         {
3223           *(int *) buffer = session->session_type;
3224           *buflen = sizeof (int);
3225
3226           VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3227                 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
3228         }
3229       else
3230         rv = VPPCOM_EINVAL;
3231       break;
3232
3233     case VPPCOM_ATTR_GET_LISTEN:
3234       if (buffer && buflen && (*buflen >= sizeof (int)))
3235         {
3236           *(int *) buffer = vcl_session_has_attr (session,
3237                                                   VCL_SESS_ATTR_LISTEN);
3238           *buflen = sizeof (int);
3239
3240           VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3241                 *buflen);
3242         }
3243       else
3244         rv = VPPCOM_EINVAL;
3245       break;
3246
3247     case VPPCOM_ATTR_GET_ERROR:
3248       if (buffer && buflen && (*buflen >= sizeof (int)))
3249         {
3250           *(int *) buffer = 0;
3251           *buflen = sizeof (int);
3252
3253           VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3254                 *(int *) buffer, *buflen);
3255         }
3256       else
3257         rv = VPPCOM_EINVAL;
3258       break;
3259
3260     case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3261       if (buffer && buflen && (*buflen >= sizeof (u32)))
3262         {
3263
3264           /* VPP-TBD */
3265           *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
3266                                 session->tx_fifo ?
3267                                 svm_fifo_size (session->tx_fifo) :
3268                                 vcm->cfg.tx_fifo_size);
3269           *buflen = sizeof (u32);
3270
3271           VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3272                 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3273                 *buflen);
3274         }
3275       else
3276         rv = VPPCOM_EINVAL;
3277       break;
3278
3279     case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3280       if (buffer && buflen && (*buflen == sizeof (u32)))
3281         {
3282           /* VPP-TBD */
3283           session->sndbuf_size = *(u32 *) buffer;
3284           VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3285                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3286                 *buflen);
3287         }
3288       else
3289         rv = VPPCOM_EINVAL;
3290       break;
3291
3292     case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3293       if (buffer && buflen && (*buflen >= sizeof (u32)))
3294         {
3295
3296           /* VPP-TBD */
3297           *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
3298                                 session->rx_fifo ?
3299                                 svm_fifo_size (session->rx_fifo) :
3300                                 vcm->cfg.rx_fifo_size);
3301           *buflen = sizeof (u32);
3302
3303           VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3304                 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
3305         }
3306       else
3307         rv = VPPCOM_EINVAL;
3308       break;
3309
3310     case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3311       if (buffer && buflen && (*buflen == sizeof (u32)))
3312         {
3313           /* VPP-TBD */
3314           session->rcvbuf_size = *(u32 *) buffer;
3315           VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3316                 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3317                 *buflen);
3318         }
3319       else
3320         rv = VPPCOM_EINVAL;
3321       break;
3322
3323     case VPPCOM_ATTR_GET_REUSEADDR:
3324       if (buffer && buflen && (*buflen >= sizeof (int)))
3325         {
3326           /* VPP-TBD */
3327           *(int *) buffer = vcl_session_has_attr (session,
3328                                                   VCL_SESS_ATTR_REUSEADDR);
3329           *buflen = sizeof (int);
3330
3331           VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3332                 *(int *) buffer, *buflen);
3333         }
3334       else
3335         rv = VPPCOM_EINVAL;
3336       break;
3337
3338     case VPPCOM_ATTR_SET_REUSEADDR:
3339       if (buffer && buflen && (*buflen == sizeof (int)) &&
3340           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3341         {
3342           /* VPP-TBD */
3343           if (*(int *) buffer)
3344             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEADDR);
3345           else
3346             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEADDR);
3347
3348           VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3349                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEADDR),
3350                 *buflen);
3351         }
3352       else
3353         rv = VPPCOM_EINVAL;
3354       break;
3355
3356     case VPPCOM_ATTR_GET_REUSEPORT:
3357       if (buffer && buflen && (*buflen >= sizeof (int)))
3358         {
3359           /* VPP-TBD */
3360           *(int *) buffer = vcl_session_has_attr (session,
3361                                                   VCL_SESS_ATTR_REUSEPORT);
3362           *buflen = sizeof (int);
3363
3364           VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3365                 *(int *) buffer, *buflen);
3366         }
3367       else
3368         rv = VPPCOM_EINVAL;
3369       break;
3370
3371     case VPPCOM_ATTR_SET_REUSEPORT:
3372       if (buffer && buflen && (*buflen == sizeof (int)) &&
3373           !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
3374         {
3375           /* VPP-TBD */
3376           if (*(int *) buffer)
3377             vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEPORT);
3378           else
3379             vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEPORT);
3380
3381           VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3382                 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEPORT),
3383                 *buflen);
3384         }
3385       else
3386         rv = VPPCOM_EINVAL;
3387       break;
3388
3389     case VPPCOM_ATTR_GET_BROADCAST:
3390       if (buffer && buflen && (*buflen >= sizeof (int)))
3391         {
3392           /* VPP-TBD */
3393           *(int *) buffer = vcl_session_has_attr (session,
3394                                                   VCL_SESS_ATTR_BROADCAST);
3395           *buflen = sizeof (int);
3396
3397           VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3398                 *(int *) buffer, *buflen);
3399         }
3400       else
3401         rv = VPPCOM_EINVAL;
3402       break;
3403
3404     case VPPCOM_ATTR_SET_BROADCAST:
3405       if (buffer && buflen && (*buflen == sizeof (int)))
3406         {
3407           /* VPP-TBD */
3408           if (*(int *) buffer)
3409             vcl_session_set_attr (session, VCL_SESS_ATTR_BROADCAST);
3410           else
3411             vcl_session_clear_attr (session, VCL_SESS_ATTR_BROADCAST);
3412
3413           VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3414                 vcl_session_has_attr (session, VCL_SESS_ATTR_BROADCAST),
3415                 *buflen);
3416         }
3417       else
3418         rv = VPPCOM_EINVAL;
3419       break;
3420
3421     case VPPCOM_ATTR_GET_V6ONLY:
3422       if (buffer && buflen && (*buflen >= sizeof (int)))
3423         {
3424           /* VPP-TBD */
3425           *(int *) buffer = vcl_session_has_attr (session,
3426                                                   VCL_SESS_ATTR_V6ONLY);
3427           *buflen = sizeof (int);
3428
3429           VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3430                 *(int *) buffer, *buflen);
3431         }
3432       else
3433         rv = VPPCOM_EINVAL;
3434       break;
3435
3436     case VPPCOM_ATTR_SET_V6ONLY:
3437       if (buffer && buflen && (*buflen == sizeof (int)))
3438         {
3439           /* VPP-TBD */
3440           if (*(int *) buffer)
3441             vcl_session_set_attr (session, VCL_SESS_ATTR_V6ONLY);
3442           else
3443             vcl_session_clear_attr (session, VCL_SESS_ATTR_V6ONLY);
3444
3445           VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3446                 vcl_session_has_attr (session, VCL_SESS_ATTR_V6ONLY),
3447                 *buflen);
3448         }
3449       else
3450         rv = VPPCOM_EINVAL;
3451       break;
3452
3453     case VPPCOM_ATTR_GET_KEEPALIVE:
3454       if (buffer && buflen && (*buflen >= sizeof (int)))
3455         {
3456           /* VPP-TBD */
3457           *(int *) buffer = vcl_session_has_attr (session,
3458                                                   VCL_SESS_ATTR_KEEPALIVE);
3459           *buflen = sizeof (int);
3460
3461           VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3462                 *(int *) buffer, *buflen);
3463         }
3464       else
3465         rv = VPPCOM_EINVAL;
3466       break;
3467
3468     case VPPCOM_ATTR_SET_KEEPALIVE:
3469       if (buffer && buflen && (*buflen == sizeof (int)))
3470         {
3471           /* VPP-TBD */
3472           if (*(int *) buffer)
3473             vcl_session_set_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3474           else
3475             vcl_session_clear_attr (session, VCL_SESS_ATTR_KEEPALIVE);
3476
3477           VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
3478                 vcl_session_has_attr (session, VCL_SESS_ATTR_KEEPALIVE),
3479                 *buflen);
3480         }
3481       else
3482         rv = VPPCOM_EINVAL;
3483       break;
3484
3485     case VPPCOM_ATTR_GET_TCP_NODELAY:
3486       if (buffer && buflen && (*buflen >= sizeof (int)))
3487         {
3488           /* VPP-TBD */
3489           *(int *) buffer = vcl_session_has_attr (session,
3490                                                   VCL_SESS_ATTR_TCP_NODELAY);
3491           *buflen = sizeof (int);
3492
3493           VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3494                 *(int *) buffer, *buflen);
3495         }
3496       else
3497         rv = VPPCOM_EINVAL;
3498       break;
3499
3500     case VPPCOM_ATTR_SET_TCP_NODELAY:
3501       if (buffer && buflen && (*buflen == sizeof (int)))
3502         {
3503           /* VPP-TBD */
3504           if (*(int *) buffer)
3505             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3506           else
3507             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
3508
3509           VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
3510                 vcl_session_has_attr (session, VCL_SESS_ATTR_TCP_NODELAY),
3511                 *buflen);
3512         }
3513       else
3514         rv = VPPCOM_EINVAL;
3515       break;
3516
3517     case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
3518       if (buffer && buflen && (*buflen >= sizeof (int)))
3519         {
3520           /* VPP-TBD */
3521           *(int *) buffer = vcl_session_has_attr (session,
3522                                                   VCL_SESS_ATTR_TCP_KEEPIDLE);
3523           *buflen = sizeof (int);
3524
3525           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3526                 *(int *) buffer, *buflen);
3527         }
3528       else
3529         rv = VPPCOM_EINVAL;
3530       break;
3531
3532     case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
3533       if (buffer && buflen && (*buflen == sizeof (int)))
3534         {
3535           /* VPP-TBD */
3536           if (*(int *) buffer)
3537             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3538           else
3539             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
3540
3541           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
3542                 vcl_session_has_attr (session,
3543                                       VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
3544         }
3545       else
3546         rv = VPPCOM_EINVAL;
3547       break;
3548
3549     case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
3550       if (buffer && buflen && (*buflen >= sizeof (int)))
3551         {
3552           /* VPP-TBD */
3553           *(int *) buffer = vcl_session_has_attr (session,
3554                                                   VCL_SESS_ATTR_TCP_KEEPINTVL);
3555           *buflen = sizeof (int);
3556
3557           VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3558                 *(int *) buffer, *buflen);
3559         }
3560       else
3561         rv = VPPCOM_EINVAL;
3562       break;
3563
3564     case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
3565       if (buffer && buflen && (*buflen == sizeof (int)))
3566         {
3567           /* VPP-TBD */
3568           if (*(int *) buffer)
3569             vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3570           else
3571             vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
3572
3573           VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
3574                 vcl_session_has_attr (session,
3575                                       VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
3576         }
3577       else
3578         rv = VPPCOM_EINVAL;
3579       break;
3580
3581     case VPPCOM_ATTR_GET_TCP_USER_MSS:
3582       if (buffer && buflen && (*buflen >= sizeof (u32)))
3583         {
3584           /* VPP-TBD */
3585           *(u32 *) buffer = session->user_mss;
3586           *buflen = sizeof (int);
3587
3588           VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d, #VPP-TBD#",
3589                 *(int *) buffer, *buflen);
3590         }
3591       else
3592         rv = VPPCOM_EINVAL;
3593       break;
3594
3595     case VPPCOM_ATTR_SET_TCP_USER_MSS:
3596       if (buffer && buflen && (*buflen == sizeof (u32)))
3597         {
3598           /* VPP-TBD */
3599           session->user_mss = *(u32 *) buffer;
3600
3601           VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d, #VPP-TBD#",
3602                 session->user_mss, *buflen);
3603         }
3604       else
3605         rv = VPPCOM_EINVAL;
3606       break;
3607
3608     case VPPCOM_ATTR_SET_SHUT:
3609       if (*flags == SHUT_RD || *flags == SHUT_RDWR)
3610         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_RD);
3611       if (*flags == SHUT_WR || *flags == SHUT_RDWR)
3612         vcl_session_set_attr (session, VCL_SESS_ATTR_SHUT_WR);
3613       break;
3614
3615     case VPPCOM_ATTR_GET_SHUT:
3616       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_RD))
3617         tmp_flags = 1;
3618       if (vcl_session_has_attr (session, VCL_SESS_ATTR_SHUT_WR))
3619         tmp_flags |= 2;
3620       if (tmp_flags == 1)
3621         *(int *) buffer = SHUT_RD;
3622       else if (tmp_flags == 2)
3623         *(int *) buffer = SHUT_WR;
3624       else if (tmp_flags == 3)
3625         *(int *) buffer = SHUT_RDWR;
3626       *buflen = sizeof (int);
3627       break;
3628
3629     case VPPCOM_ATTR_SET_CONNECTED:
3630       session->flags |= VCL_SESSION_F_CONNECTED;
3631       break;
3632
3633     default:
3634       rv = VPPCOM_EINVAL;
3635       break;
3636     }
3637
3638   return rv;
3639 }
3640
3641 int
3642 vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
3643                          uint32_t buflen, int flags, vppcom_endpt_t * ep)
3644 {
3645   vcl_worker_t *wrk = vcl_worker_get_current ();
3646   vcl_session_t *session;
3647   int rv = VPPCOM_OK;
3648
3649   if (flags == 0)
3650     rv = vppcom_session_read (session_handle, buffer, buflen);
3651   else if (flags & MSG_PEEK)
3652     rv = vppcom_session_peek (session_handle, buffer, buflen);
3653   else
3654     {
3655       VDBG (0, "Unsupport flags for recvfrom %d", flags);
3656       return VPPCOM_EAFNOSUPPORT;
3657     }
3658
3659   if (ep && rv > 0)
3660     {
3661       session = vcl_session_get_w_handle (wrk, session_handle);
3662       if (session->transport.is_ip4)
3663         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3664                           sizeof (ip4_address_t));
3665       else
3666         clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3667                           sizeof (ip6_address_t));
3668       ep->is_ip4 = session->transport.is_ip4;
3669       ep->port = session->transport.rmt_port;
3670     }
3671
3672   return rv;
3673 }
3674
3675 int
3676 vppcom_session_sendto (uint32_t session_handle, void *buffer,
3677                        uint32_t buflen, int flags, vppcom_endpt_t * ep)
3678 {
3679   vcl_worker_t *wrk = vcl_worker_get_current ();
3680   vcl_session_t *s;
3681
3682   s = vcl_session_get_w_handle (wrk, session_handle);
3683   if (!s)
3684     return VPPCOM_EBADFD;
3685
3686   if (!buffer)
3687     return VPPCOM_EINVAL;
3688
3689   if (ep)
3690     {
3691       if (!vcl_session_is_cl (s))
3692         return VPPCOM_EINVAL;
3693
3694       /* Session not connected/bound in vpp. Create it by 'connecting' it */
3695       if (PREDICT_FALSE (s->session_state == VCL_STATE_CLOSED))
3696         {
3697           u32 session_index = s->session_index;
3698           f64 timeout = vcm->cfg.session_timeout;
3699           int rv;
3700
3701           vcl_send_session_connect (wrk, s);
3702           rv = vppcom_wait_for_session_state_change (session_index,
3703                                                      VCL_STATE_READY,
3704                                                      timeout);
3705           if (rv < 0)
3706             return rv;
3707           s = vcl_session_get (wrk, session_index);
3708         }
3709
3710       s->transport.is_ip4 = ep->is_ip4;
3711       s->transport.rmt_port = ep->port;
3712       vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
3713     }
3714
3715   if (flags)
3716     {
3717       // TBD check the flags and do the right thing
3718       VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
3719     }
3720
3721   return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
3722                                        s->is_dgram ? 1 : 0));
3723 }
3724
3725 int
3726 vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
3727 {
3728   vcl_worker_t *wrk = vcl_worker_get_current ();
3729   f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
3730   u32 i, keep_trying = 1;
3731   svm_msg_q_msg_t msg;
3732   session_event_t *e;
3733   int rv, num_ev = 0;
3734
3735   VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
3736
3737   if (!vp)
3738     return VPPCOM_EFAULT;
3739
3740   do
3741     {
3742       vcl_session_t *session;
3743
3744       /* Dequeue all events and drop all unhandled io events */
3745       while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
3746         {
3747           e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
3748           vcl_handle_mq_event (wrk, e);
3749           svm_msg_q_free_msg (wrk->app_event_queue, &msg);
3750         }
3751       vec_reset_length (wrk->unhandled_evts_vector);
3752
3753       for (i = 0; i < n_sids; i++)
3754         {
3755           session = vcl_session_get (wrk, vp[i].sh);
3756           if (!session)
3757             {
3758               vp[i].revents = POLLHUP;
3759               num_ev++;
3760               continue;
3761             }
3762
3763           vp[i].revents = 0;
3764
3765           if (POLLIN & vp[i].events)
3766             {
3767               rv = vcl_session_read_ready (session);
3768               if (rv > 0)
3769                 {
3770                   vp[i].revents |= POLLIN;
3771                   num_ev++;
3772                 }
3773               else if (rv < 0)
3774                 {
3775                   switch (rv)
3776                     {
3777                     case VPPCOM_ECONNRESET:
3778                       vp[i].revents = POLLHUP;
3779                       break;
3780
3781                     default:
3782                       vp[i].revents = POLLERR;
3783                       break;
3784                     }
3785                   num_ev++;
3786                 }
3787             }
3788
3789           if (POLLOUT & vp[i].events)
3790             {
3791               rv = vcl_session_write_ready (session);
3792               if (rv > 0)
3793                 {
3794                   vp[i].revents |= POLLOUT;
3795                   num_ev++;
3796                 }
3797               else if (rv < 0)
3798                 {
3799                   switch (rv)
3800                     {
3801                     case VPPCOM_ECONNRESET:
3802                       vp[i].revents = POLLHUP;
3803                       break;
3804
3805                     default:
3806                       vp[i].revents = POLLERR;
3807                       break;
3808                     }
3809                   num_ev++;
3810                 }
3811             }
3812
3813           if (0)                // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
3814             {
3815               vp[i].revents = POLLNVAL;
3816               num_ev++;
3817             }
3818         }
3819       if (wait_for_time != -1)
3820         keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
3821     }
3822   while ((num_ev == 0) && keep_trying);
3823
3824   return num_ev;
3825 }
3826
3827 int
3828 vppcom_mq_epoll_fd (void)
3829 {
3830   vcl_worker_t *wrk = vcl_worker_get_current ();
3831   return wrk->mqs_epfd;
3832 }
3833
3834 int
3835 vppcom_session_index (vcl_session_handle_t session_handle)
3836 {
3837   return session_handle & 0xFFFFFF;
3838 }
3839
3840 int
3841 vppcom_session_worker (vcl_session_handle_t session_handle)
3842 {
3843   return session_handle >> 24;
3844 }
3845
3846 int
3847 vppcom_worker_register (void)
3848 {
3849   if (!vcl_worker_alloc_and_init ())
3850     return VPPCOM_EEXIST;
3851
3852   if (vcl_worker_register_with_vpp ())
3853     return VPPCOM_EEXIST;
3854
3855   return VPPCOM_OK;
3856 }
3857
3858 void
3859 vppcom_worker_unregister (void)
3860 {
3861   vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
3862   vcl_set_worker_index (~0);
3863 }
3864
3865 void
3866 vppcom_worker_index_set (int index)
3867 {
3868   vcl_set_worker_index (index);
3869 }
3870
3871 int
3872 vppcom_worker_index (void)
3873 {
3874   return vcl_get_worker_index ();
3875 }
3876
3877 int
3878 vppcom_worker_mqs_epfd (void)
3879 {
3880   vcl_worker_t *wrk = vcl_worker_get_current ();
3881   if (!vcm->cfg.use_mq_eventfd)
3882     return -1;
3883   return wrk->mqs_epfd;
3884 }
3885
3886 int
3887 vppcom_session_is_connectable_listener (uint32_t session_handle)
3888 {
3889   vcl_session_t *session;
3890   vcl_worker_t *wrk = vcl_worker_get_current ();
3891   session = vcl_session_get_w_handle (wrk, session_handle);
3892   if (!session)
3893     return VPPCOM_EBADFD;
3894   return vcl_session_is_connectable_listener (wrk, session);
3895 }
3896
3897 int
3898 vppcom_session_listener (uint32_t session_handle)
3899 {
3900   vcl_worker_t *wrk = vcl_worker_get_current ();
3901   vcl_session_t *listen_session, *session;
3902   session = vcl_session_get_w_handle (wrk, session_handle);
3903   if (!session)
3904     return VPPCOM_EBADFD;
3905   if (session->listener_index == VCL_INVALID_SESSION_INDEX)
3906     return VPPCOM_EBADFD;
3907   listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
3908   if (!listen_session)
3909     return VPPCOM_EBADFD;
3910   return vcl_session_handle (listen_session);
3911 }
3912
3913 int
3914 vppcom_session_n_accepted (uint32_t session_handle)
3915 {
3916   vcl_worker_t *wrk = vcl_worker_get_current ();
3917   vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
3918   if (!session)
3919     return VPPCOM_EBADFD;
3920   return session->n_accepted_sessions;
3921 }
3922
3923 const char *
3924 vppcom_proto_str (vppcom_proto_t proto)
3925 {
3926   char const *proto_str;
3927
3928   switch (proto)
3929     {
3930     case VPPCOM_PROTO_TCP:
3931       proto_str = "TCP";
3932       break;
3933     case VPPCOM_PROTO_UDP:
3934       proto_str = "UDP";
3935       break;
3936     case VPPCOM_PROTO_TLS:
3937       proto_str = "TLS";
3938       break;
3939     case VPPCOM_PROTO_QUIC:
3940       proto_str = "QUIC";
3941       break;
3942     default:
3943       proto_str = "UNKNOWN";
3944       break;
3945     }
3946   return proto_str;
3947 }
3948
3949 const char *
3950 vppcom_retval_str (int retval)
3951 {
3952   char const *st;
3953
3954   switch (retval)
3955     {
3956     case VPPCOM_OK:
3957       st = "VPPCOM_OK";
3958       break;
3959
3960     case VPPCOM_EAGAIN:
3961       st = "VPPCOM_EAGAIN";
3962       break;
3963
3964     case VPPCOM_EFAULT:
3965       st = "VPPCOM_EFAULT";
3966       break;
3967
3968     case VPPCOM_ENOMEM:
3969       st = "VPPCOM_ENOMEM";
3970       break;
3971
3972     case VPPCOM_EINVAL:
3973       st = "VPPCOM_EINVAL";
3974       break;
3975
3976     case VPPCOM_EBADFD:
3977       st = "VPPCOM_EBADFD";
3978       break;
3979
3980     case VPPCOM_EAFNOSUPPORT:
3981       st = "VPPCOM_EAFNOSUPPORT";
3982       break;
3983
3984     case VPPCOM_ECONNABORTED:
3985       st = "VPPCOM_ECONNABORTED";
3986       break;
3987
3988     case VPPCOM_ECONNRESET:
3989       st = "VPPCOM_ECONNRESET";
3990       break;
3991
3992     case VPPCOM_ENOTCONN:
3993       st = "VPPCOM_ENOTCONN";
3994       break;
3995
3996     case VPPCOM_ECONNREFUSED:
3997       st = "VPPCOM_ECONNREFUSED";
3998       break;
3999
4000     case VPPCOM_ETIMEDOUT:
4001       st = "VPPCOM_ETIMEDOUT";
4002       break;
4003
4004     default:
4005       st = "UNKNOWN_STATE";
4006       break;
4007     }
4008
4009   return st;
4010 }
4011
4012 /*
4013  * fd.io coding-style-patch-verification: ON
4014  *
4015  * Local Variables:
4016  * eval: (c-set-style "gnu")
4017  * End:
4018  */