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