src/vcl/vppcom.c

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