src/vcl/vppcom.c

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