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