src/vcl/vppcom.c

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