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