src/vcl/vppcom.c

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