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