src/vcl/vppcom.c

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