src/vcl/vppcom.c

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