src/vcl/vppcom.c

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