src/vnet/session/session_api.c

   1 /*
   2  * Copyright (c) 2015-2019 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   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 <vnet/vnet.h>
  17 #include <vlibmemory/api.h>
  18 #include <vnet/session/application.h>
  19 #include <vnet/session/application_interface.h>
  20 #include <vnet/session/application_local.h>
  21 #include <vnet/session/session_rules_table.h>
  22 #include <vnet/session/session_table.h>
  23 #include <vnet/session/session.h>
  24 #include <vnet/ip/ip_types_api.h>
  25
  26 #include <vnet/format_fns.h>
  27 #include <vnet/session/session.api_enum.h>
  28 #include <vnet/session/session.api_types.h>
  29
  30 #define REPLY_MSG_ID_BASE session_main.msg_id_base
  31 #include <vlibapi/api_helper_macros.h>
  32
  33 static transport_proto_t
  34 api_session_transport_proto_decode (const vl_api_transport_proto_t * api_tp)
  35 {
  36   switch (*api_tp)
  37     {
  38     case TRANSPORT_PROTO_API_TCP:
  39       return TRANSPORT_PROTO_TCP;
  40     case TRANSPORT_PROTO_API_UDP:
  41       return TRANSPORT_PROTO_UDP;
  42     case TRANSPORT_PROTO_API_TLS:
  43       return TRANSPORT_PROTO_TLS;
  44     case TRANSPORT_PROTO_API_QUIC:
  45       return TRANSPORT_PROTO_QUIC;
  46     default:
  47       return TRANSPORT_PROTO_NONE;
  48     }
  49 }
  50
  51 static vl_api_transport_proto_t
  52 api_session_transport_proto_encode (const transport_proto_t tp)
  53 {
  54   switch (tp)
  55     {
  56     case TRANSPORT_PROTO_TCP:
  57       return TRANSPORT_PROTO_API_TCP;
  58     case TRANSPORT_PROTO_UDP:
  59       return TRANSPORT_PROTO_API_UDP;
  60     case TRANSPORT_PROTO_TLS:
  61       return TRANSPORT_PROTO_API_TLS;
  62     case TRANSPORT_PROTO_QUIC:
  63       return TRANSPORT_PROTO_API_QUIC;
  64     default:
  65       return TRANSPORT_PROTO_API_NONE;
  66     }
  67 }
  68
  69 static int
  70 session_send_fds (vl_api_registration_t * reg, int fds[], int n_fds)
  71 {
  72   clib_error_t *error;
  73   if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI)
  74     return SESSION_E_BAPI_NO_FD;
  75   error = vl_api_send_fd_msg (reg, fds, n_fds);
  76   if (error)
  77     {
  78       clib_error_report (error);
  79       return SESSION_E_BAPI_SEND_FD;
  80     }
  81   return 0;
  82 }
  83
  84 static int
  85 mq_try_lock_and_alloc_msg (svm_msg_q_t * app_mq, svm_msg_q_msg_t * msg)
  86 {
  87   int rv;
  88   u8 try = 0;
  89   while (try < 100)
  90     {
  91       rv = svm_msg_q_lock_and_alloc_msg_w_ring (app_mq,
  92                                                 SESSION_MQ_CTRL_EVT_RING,
  93                                                 SVM_Q_NOWAIT, msg);
  94       if (!rv)
  95         return 0;
  96       try++;
  97       usleep (1);
  98     }
  99   clib_warning ("failed to alloc msg");
 100   return -1;
 101 }
 102
 103 static int
 104 mq_send_session_accepted_cb (session_t * s)
 105 {
 106   app_worker_t *app_wrk = app_worker_get (s->app_wrk_index);
 107   svm_msg_q_msg_t _msg, *msg = &_msg;
 108   session_accepted_msg_t m = { 0 };
 109   svm_msg_q_t *app_mq;
 110   fifo_segment_t *eq_seg;
 111   session_t *listener;
 112   session_event_t *evt;
 113   application_t *app;
 114
 115   app = application_get (app_wrk->app_index);
 116
 117   m.context = app->app_index;
 118   m.server_rx_fifo = fifo_segment_fifo_offset (s->rx_fifo);
 119   m.server_tx_fifo = fifo_segment_fifo_offset (s->tx_fifo);
 120   m.segment_handle = session_segment_handle (s);
 121   m.flags = s->flags;
 122
 123   eq_seg = application_get_rx_mqs_segment (app);
 124
 125   if (session_has_transport (s))
 126     {
 127       listener = listen_session_get_from_handle (s->listener_handle);
 128       m.listener_handle = app_listen_session_handle (listener);
 129       if (application_is_proxy (app))
 130         {
 131           listener =
 132             app_worker_first_listener (app_wrk, session_get_fib_proto (s),
 133                                        session_get_transport_proto (s));
 134           if (listener)
 135             m.listener_handle = listen_session_get_handle (listener);
 136         }
 137       m.vpp_event_queue_address =
 138         fifo_segment_msg_q_offset (eq_seg, s->thread_index);
 139       m.mq_index = s->thread_index;
 140       m.handle = session_handle (s);
 141
 142       session_get_endpoint (s, &m.rmt, 0 /* is_lcl */);
 143       session_get_endpoint (s, &m.lcl, 1 /* is_lcl */);
 144     }
 145   else
 146     {
 147       ct_connection_t *ct;
 148
 149       ct = (ct_connection_t *) session_get_transport (s);
 150       listener = listen_session_get_from_handle (s->listener_handle);
 151       m.listener_handle = app_listen_session_handle (listener);
 152       m.rmt.is_ip4 = session_type_is_ip4 (listener->session_type);
 153       m.rmt.port = ct->c_rmt_port;
 154       m.lcl.port = ct->c_lcl_port;
 155       m.handle = session_handle (s);
 156       m.vpp_event_queue_address =
 157         fifo_segment_msg_q_offset (eq_seg, s->thread_index);
 158       m.mq_index = s->thread_index;
 159     }
 160
 161   app_mq = app_wrk->event_queue;
 162   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 163     return SESSION_E_MQ_MSG_ALLOC;
 164
 165   evt = svm_msg_q_msg_data (app_mq, msg);
 166   clib_memset (evt, 0, sizeof (*evt));
 167   evt->event_type = SESSION_CTRL_EVT_ACCEPTED;
 168   clib_memcpy_fast (evt->data, &m, sizeof (m));
 169   svm_msg_q_add_and_unlock (app_mq, msg);
 170
 171   return 0;
 172 }
 173
 174 static inline void
 175 mq_send_session_close_evt (app_worker_t * app_wrk, session_handle_t sh,
 176                            session_evt_type_t evt_type)
 177 {
 178   svm_msg_q_msg_t _msg, *msg = &_msg;
 179   session_disconnected_msg_t *mp;
 180   svm_msg_q_t *app_mq;
 181   session_event_t *evt;
 182
 183   app_mq = app_wrk->event_queue;
 184   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 185     return;
 186   evt = svm_msg_q_msg_data (app_mq, msg);
 187   clib_memset (evt, 0, sizeof (*evt));
 188   evt->event_type = evt_type;
 189   mp = (session_disconnected_msg_t *) evt->data;
 190   mp->handle = sh;
 191   mp->context = app_wrk->api_client_index;
 192   svm_msg_q_add_and_unlock (app_mq, msg);
 193 }
 194
 195 static inline void
 196 mq_notify_close_subscribers (u32 app_index, session_handle_t sh,
 197                              svm_fifo_t * f, session_evt_type_t evt_type)
 198 {
 199   app_worker_t *app_wrk;
 200   application_t *app;
 201   int i;
 202
 203   app = application_get (app_index);
 204   if (!app)
 205     return;
 206
 207   for (i = 0; i < f->shr->n_subscribers; i++)
 208     {
 209       if (!(app_wrk = application_get_worker (app, f->shr->subscribers[i])))
 210         continue;
 211       mq_send_session_close_evt (app_wrk, sh, SESSION_CTRL_EVT_DISCONNECTED);
 212     }
 213 }
 214
 215 static void
 216 mq_send_session_disconnected_cb (session_t * s)
 217 {
 218   app_worker_t *app_wrk = app_worker_get (s->app_wrk_index);
 219   session_handle_t sh = session_handle (s);
 220
 221   mq_send_session_close_evt (app_wrk, session_handle (s),
 222                              SESSION_CTRL_EVT_DISCONNECTED);
 223
 224   if (svm_fifo_n_subscribers (s->rx_fifo))
 225     mq_notify_close_subscribers (app_wrk->app_index, sh, s->rx_fifo,
 226                                  SESSION_CTRL_EVT_DISCONNECTED);
 227 }
 228
 229 static void
 230 mq_send_session_reset_cb (session_t * s)
 231 {
 232   app_worker_t *app_wrk = app_worker_get (s->app_wrk_index);
 233   session_handle_t sh = session_handle (s);
 234
 235   mq_send_session_close_evt (app_wrk, sh, SESSION_CTRL_EVT_RESET);
 236
 237   if (svm_fifo_n_subscribers (s->rx_fifo))
 238     mq_notify_close_subscribers (app_wrk->app_index, sh, s->rx_fifo,
 239                                  SESSION_CTRL_EVT_RESET);
 240 }
 241
 242 int
 243 mq_send_session_connected_cb (u32 app_wrk_index, u32 api_context,
 244                               session_t * s, session_error_t err)
 245 {
 246   svm_msg_q_msg_t _msg, *msg = &_msg;
 247   session_connected_msg_t m = { 0 };
 248   svm_msg_q_t *app_mq;
 249   transport_connection_t *tc;
 250   fifo_segment_t *eq_seg;
 251   app_worker_t *app_wrk;
 252   session_event_t *evt;
 253   application_t *app;
 254
 255   app_wrk = app_worker_get (app_wrk_index);
 256
 257   m.context = api_context;
 258   m.retval = err;
 259
 260   if (err)
 261     goto snd_msg;
 262
 263   app = application_get (app_wrk->app_index);
 264   eq_seg = application_get_rx_mqs_segment (app);
 265
 266   if (session_has_transport (s))
 267     {
 268       tc = session_get_transport (s);
 269       if (!tc)
 270         {
 271           clib_warning ("failed to retrieve transport!");
 272           m.retval = SESSION_E_REFUSED;
 273           goto snd_msg;
 274         }
 275
 276       m.handle = session_handle (s);
 277       m.vpp_event_queue_address =
 278         fifo_segment_msg_q_offset (eq_seg, s->thread_index);
 279
 280       session_get_endpoint (s, &m.lcl, 1 /* is_lcl */);
 281
 282       m.server_rx_fifo = fifo_segment_fifo_offset (s->rx_fifo);
 283       m.server_tx_fifo = fifo_segment_fifo_offset (s->tx_fifo);
 284       m.segment_handle = session_segment_handle (s);
 285       m.mq_index = s->thread_index;
 286     }
 287   else
 288     {
 289       ct_connection_t *cct;
 290       session_t *ss;
 291
 292       cct = (ct_connection_t *) session_get_transport (s);
 293       m.handle = session_handle (s);
 294       m.lcl.port = cct->c_lcl_port;
 295       m.lcl.is_ip4 = cct->c_is_ip4;
 296       m.vpp_event_queue_address =
 297         fifo_segment_msg_q_offset (eq_seg, s->thread_index);
 298       m.server_rx_fifo = fifo_segment_fifo_offset (s->rx_fifo);
 299       m.server_tx_fifo = fifo_segment_fifo_offset (s->tx_fifo);
 300       m.segment_handle = session_segment_handle (s);
 301       ss = ct_session_get_peer (s);
 302       m.ct_rx_fifo = fifo_segment_fifo_offset (ss->tx_fifo);
 303       m.ct_tx_fifo = fifo_segment_fifo_offset (ss->rx_fifo);
 304       m.ct_segment_handle = session_segment_handle (ss);
 305       m.mq_index = s->thread_index;
 306     }
 307
 308   /* Setup client session index in advance, in case data arrives
 309    * before the app processes message and updates it */
 310   s->rx_fifo->shr->client_session_index = api_context;
 311   s->tx_fifo->shr->client_session_index = api_context;
 312
 313 snd_msg:
 314
 315   app_mq = app_wrk->event_queue;
 316
 317   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 318     return SESSION_E_MQ_MSG_ALLOC;
 319
 320   evt = svm_msg_q_msg_data (app_mq, msg);
 321   clib_memset (evt, 0, sizeof (*evt));
 322   evt->event_type = SESSION_CTRL_EVT_CONNECTED;
 323   clib_memcpy_fast (evt->data, &m, sizeof (m));
 324
 325   svm_msg_q_add_and_unlock (app_mq, msg);
 326   return 0;
 327 }
 328
 329 int
 330 mq_send_session_bound_cb (u32 app_wrk_index, u32 api_context,
 331                           session_handle_t handle, int rv)
 332 {
 333   svm_msg_q_msg_t _msg, *msg = &_msg;
 334   session_bound_msg_t m = { 0 };
 335   svm_msg_q_t *app_mq;
 336   transport_endpoint_t tep;
 337   fifo_segment_t *eq_seg;
 338   app_worker_t *app_wrk;
 339   session_event_t *evt;
 340   application_t *app;
 341   app_listener_t *al;
 342   session_t *ls = 0;
 343
 344   app_wrk = app_worker_get (app_wrk_index);
 345
 346   m.context = api_context;
 347   m.retval = rv;
 348
 349   if (rv)
 350     goto snd_msg;
 351
 352   m.handle = handle;
 353   al = app_listener_get_w_handle (handle);
 354   if (al->session_index != SESSION_INVALID_INDEX)
 355     ls = app_listener_get_session (al);
 356   else
 357     ls = app_listener_get_local_session (al);
 358
 359   session_get_endpoint (ls, &tep, 1 /* is_lcl */);
 360   m.lcl_port = tep.port;
 361   m.lcl_is_ip4 = tep.is_ip4;
 362   clib_memcpy_fast (m.lcl_ip, &tep.ip, sizeof (tep.ip));
 363   app = application_get (app_wrk->app_index);
 364   eq_seg = application_get_rx_mqs_segment (app);
 365   m.vpp_evt_q = fifo_segment_msg_q_offset (eq_seg, ls->thread_index);
 366   m.mq_index = ls->thread_index;
 367
 368   if (session_transport_service_type (ls) == TRANSPORT_SERVICE_CL &&
 369       ls->rx_fifo)
 370     {
 371       m.rx_fifo = fifo_segment_fifo_offset (ls->rx_fifo);
 372       m.tx_fifo = fifo_segment_fifo_offset (ls->tx_fifo);
 373       m.segment_handle = session_segment_handle (ls);
 374     }
 375
 376 snd_msg:
 377
 378   app_mq = app_wrk->event_queue;
 379
 380   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 381     return SESSION_E_MQ_MSG_ALLOC;
 382
 383   evt = svm_msg_q_msg_data (app_mq, msg);
 384   clib_memset (evt, 0, sizeof (*evt));
 385   evt->event_type = SESSION_CTRL_EVT_BOUND;
 386   clib_memcpy_fast (evt->data, &m, sizeof (m));
 387
 388   svm_msg_q_add_and_unlock (app_mq, msg);
 389   return 0;
 390 }
 391
 392 void
 393 mq_send_unlisten_reply (app_worker_t * app_wrk, session_handle_t sh,
 394                         u32 context, int rv)
 395 {
 396   svm_msg_q_msg_t _msg, *msg = &_msg;
 397   session_unlisten_reply_msg_t *ump;
 398   svm_msg_q_t *app_mq;
 399   session_event_t *evt;
 400
 401   app_mq = app_wrk->event_queue;
 402   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 403     return;
 404
 405   evt = svm_msg_q_msg_data (app_mq, msg);
 406   clib_memset (evt, 0, sizeof (*evt));
 407   evt->event_type = SESSION_CTRL_EVT_UNLISTEN_REPLY;
 408   ump = (session_unlisten_reply_msg_t *) evt->data;
 409   ump->context = context;
 410   ump->handle = sh;
 411   ump->retval = rv;
 412   svm_msg_q_add_and_unlock (app_mq, msg);
 413 }
 414
 415 static void
 416 mq_send_session_migrate_cb (session_t * s, session_handle_t new_sh)
 417 {
 418   svm_msg_q_msg_t _msg, *msg = &_msg;
 419   session_migrated_msg_t m = { 0 };
 420   fifo_segment_t *eq_seg;
 421   app_worker_t *app_wrk;
 422   session_event_t *evt;
 423   svm_msg_q_t *app_mq;
 424   application_t *app;
 425   u32 thread_index;
 426
 427   thread_index = session_thread_from_handle (new_sh);
 428   app_wrk = app_worker_get (s->app_wrk_index);
 429   app_mq = app_wrk->event_queue;
 430   app = application_get (app_wrk->app_index);
 431   eq_seg = application_get_rx_mqs_segment (app);
 432
 433   m.handle = session_handle (s);
 434   m.new_handle = new_sh;
 435   m.vpp_thread_index = thread_index;
 436   m.vpp_evt_q = fifo_segment_msg_q_offset (eq_seg, thread_index);
 437   m.segment_handle = SESSION_INVALID_HANDLE;
 438
 439   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 440     return;
 441
 442   evt = svm_msg_q_msg_data (app_mq, msg);
 443   clib_memset (evt, 0, sizeof (*evt));
 444   evt->event_type = SESSION_CTRL_EVT_MIGRATED;
 445   clib_memcpy_fast (evt->data, &m, sizeof (m));
 446
 447   svm_msg_q_add_and_unlock (app_mq, msg);
 448 }
 449
 450 static int
 451 mq_send_add_segment_cb (u32 app_wrk_index, u64 segment_handle)
 452 {
 453   int fds[SESSION_N_FD_TYPE], n_fds = 0;
 454   svm_msg_q_msg_t _msg, *msg = &_msg;
 455   session_app_add_segment_msg_t *mp;
 456   vl_api_registration_t *reg;
 457   app_worker_t *app_wrk;
 458   session_event_t *evt;
 459   svm_msg_q_t *app_mq;
 460   fifo_segment_t *fs;
 461   ssvm_private_t *sp;
 462   u8 fd_flags = 0;
 463
 464   app_wrk = app_worker_get (app_wrk_index);
 465
 466   reg = vl_mem_api_client_index_to_registration (app_wrk->api_client_index);
 467   if (!reg)
 468     {
 469       clib_warning ("no api registration for client: %u",
 470                     app_wrk->api_client_index);
 471       return -1;
 472     }
 473
 474   fs = segment_manager_get_segment_w_handle (segment_handle);
 475   sp = &fs->ssvm;
 476   if (ssvm_type (sp) == SSVM_SEGMENT_MEMFD)
 477     {
 478       if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI)
 479         {
 480           clib_warning ("can't send memfd fd");
 481           return -1;
 482         }
 483
 484       fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
 485       fds[n_fds] = sp->fd;
 486       n_fds += 1;
 487     }
 488
 489   app_mq = app_wrk->event_queue;
 490   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 491     return -1;
 492
 493   if (n_fds)
 494     session_send_fds (reg, fds, n_fds);
 495
 496   evt = svm_msg_q_msg_data (app_mq, msg);
 497   clib_memset (evt, 0, sizeof (*evt));
 498   evt->event_type = SESSION_CTRL_EVT_APP_ADD_SEGMENT;
 499   mp = (session_app_add_segment_msg_t *) evt->data;
 500   clib_memset (mp, 0, sizeof (*mp));
 501   mp->segment_size = sp->ssvm_size;
 502   mp->fd_flags = fd_flags;
 503   mp->segment_handle = segment_handle;
 504   strncpy ((char *) mp->segment_name, (char *) sp->name,
 505            sizeof (mp->segment_name) - 1);
 506
 507   svm_msg_q_add_and_unlock (app_mq, msg);
 508
 509   return 0;
 510 }
 511
 512 static int
 513 mq_send_del_segment_cb (u32 app_wrk_index, u64 segment_handle)
 514 {
 515   svm_msg_q_msg_t _msg, *msg = &_msg;
 516   session_app_del_segment_msg_t *mp;
 517   vl_api_registration_t *reg;
 518   app_worker_t *app_wrk;
 519   session_event_t *evt;
 520   svm_msg_q_t *app_mq;
 521
 522   app_wrk = app_worker_get (app_wrk_index);
 523   reg = vl_mem_api_client_index_to_registration (app_wrk->api_client_index);
 524   if (!reg)
 525     {
 526       clib_warning ("no registration: %u", app_wrk->api_client_index);
 527       return -1;
 528     }
 529
 530   app_mq = app_wrk->event_queue;
 531   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 532     return -1;
 533
 534   evt = svm_msg_q_msg_data (app_mq, msg);
 535   clib_memset (evt, 0, sizeof (*evt));
 536   evt->event_type = SESSION_CTRL_EVT_APP_DEL_SEGMENT;
 537   mp = (session_app_del_segment_msg_t *) evt->data;
 538   clib_memset (mp, 0, sizeof (*mp));
 539   mp->segment_handle = segment_handle;
 540   svm_msg_q_add_and_unlock (app_mq, msg);
 541
 542   return 0;
 543 }
 544
 545 static void
 546 mq_send_session_cleanup_cb (session_t * s, session_cleanup_ntf_t ntf)
 547 {
 548   svm_msg_q_msg_t _msg, *msg = &_msg;
 549   session_cleanup_msg_t *mp;
 550   svm_msg_q_t *app_mq;
 551   session_event_t *evt;
 552   app_worker_t *app_wrk;
 553
 554   /* Propagate transport cleanup notifications only if app didn't close */
 555   if (ntf == SESSION_CLEANUP_TRANSPORT
 556       && s->session_state != SESSION_STATE_TRANSPORT_DELETED)
 557     return;
 558
 559   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
 560   if (!app_wrk)
 561     return;
 562
 563   app_mq = app_wrk->event_queue;
 564   if (mq_try_lock_and_alloc_msg (app_mq, msg))
 565     return;
 566
 567   evt = svm_msg_q_msg_data (app_mq, msg);
 568   clib_memset (evt, 0, sizeof (*evt));
 569   evt->event_type = SESSION_CTRL_EVT_CLEANUP;
 570   mp = (session_cleanup_msg_t *) evt->data;
 571   mp->handle = session_handle (s);
 572   mp->type = ntf;
 573   svm_msg_q_add_and_unlock (app_mq, msg);
 574 }
 575
 576 static session_cb_vft_t session_mq_cb_vft = {
 577   .session_accept_callback = mq_send_session_accepted_cb,
 578   .session_disconnect_callback = mq_send_session_disconnected_cb,
 579   .session_connected_callback = mq_send_session_connected_cb,
 580   .session_reset_callback = mq_send_session_reset_cb,
 581   .session_migrate_callback = mq_send_session_migrate_cb,
 582   .session_cleanup_callback = mq_send_session_cleanup_cb,
 583   .add_segment_callback = mq_send_add_segment_cb,
 584   .del_segment_callback = mq_send_del_segment_cb,
 585 };
 586
 587 static void
 588 vl_api_session_enable_disable_t_handler (vl_api_session_enable_disable_t * mp)
 589 {
 590   vl_api_session_enable_disable_reply_t *rmp;
 591   vlib_main_t *vm = vlib_get_main ();
 592   int rv = 0;
 593
 594   vnet_session_enable_disable (vm, mp->is_enable);
 595   REPLY_MACRO (VL_API_SESSION_ENABLE_DISABLE_REPLY);
 596 }
 597
 598 static void
 599 vl_api_app_attach_t_handler (vl_api_app_attach_t * mp)
 600 {
 601   int rv = 0, *fds = 0, n_fds = 0, n_workers, i;
 602   fifo_segment_t *segp, *rx_mqs_seg = 0;
 603   vnet_app_attach_args_t _a, *a = &_a;
 604   vl_api_app_attach_reply_t *rmp;
 605   u8 fd_flags = 0, ctrl_thread;
 606   vl_api_registration_t *reg;
 607   svm_msg_q_t *rx_mq;
 608   application_t *app;
 609
 610   reg = vl_api_client_index_to_registration (mp->client_index);
 611   if (!reg)
 612     return;
 613
 614   n_workers = vlib_num_workers ();
 615   if (!session_main_is_enabled () || appns_sapi_enabled ())
 616     {
 617       rv = VNET_API_ERROR_FEATURE_DISABLED;
 618       goto done;
 619     }
 620   /* Only support binary api with socket transport */
 621   if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI)
 622     {
 623       rv = VNET_API_ERROR_APP_UNSUPPORTED_CFG;
 624       goto done;
 625     }
 626
 627   STATIC_ASSERT (sizeof (u64) * APP_OPTIONS_N_OPTIONS <=
 628                  sizeof (mp->options),
 629                  "Out of options, fix api message definition");
 630
 631   clib_memset (a, 0, sizeof (*a));
 632   a->api_client_index = mp->client_index;
 633   a->options = mp->options;
 634   a->session_cb_vft = &session_mq_cb_vft;
 635   a->namespace_id = vl_api_from_api_to_new_vec (mp, &mp->namespace_id);
 636
 637   if ((rv = vnet_application_attach (a)))
 638     {
 639       clib_warning ("attach returned: %d", rv);
 640       vec_free (a->namespace_id);
 641       goto done;
 642     }
 643   vec_free (a->namespace_id);
 644
 645   vec_validate (fds, 3 /* segs + tx evtfd */ + n_workers);
 646
 647   /* Send rx mqs segment */
 648   app = application_get (a->app_index);
 649   rx_mqs_seg = application_get_rx_mqs_segment (app);
 650
 651   fd_flags |= SESSION_FD_F_VPP_MQ_SEGMENT;
 652   fds[n_fds] = rx_mqs_seg->ssvm.fd;
 653   n_fds += 1;
 654
 655   /* Send fifo segment fd if needed */
 656   if (ssvm_type (a->segment) == SSVM_SEGMENT_MEMFD)
 657     {
 658       fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
 659       fds[n_fds] = a->segment->fd;
 660       n_fds += 1;
 661     }
 662   if (a->options[APP_OPTIONS_FLAGS] & APP_OPTIONS_FLAGS_EVT_MQ_USE_EVENTFD)
 663     {
 664       fd_flags |= SESSION_FD_F_MQ_EVENTFD;
 665       fds[n_fds] = svm_msg_q_get_eventfd (a->app_evt_q);
 666       n_fds += 1;
 667     }
 668
 669   if (application_use_private_rx_mqs ())
 670     {
 671       fd_flags |= SESSION_FD_F_VPP_MQ_EVENTFD;
 672       for (i = 0; i < n_workers + 1; i++)
 673         {
 674           rx_mq = application_rx_mq_get (app, i);
 675           fds[n_fds] = svm_msg_q_get_eventfd (rx_mq);
 676           n_fds += 1;
 677         }
 678     }
 679
 680 done:
 681   /* *INDENT-OFF* */
 682   REPLY_MACRO2 (VL_API_APP_ATTACH_REPLY, ({
 683     if (!rv)
 684       {
 685         ctrl_thread = n_workers ? 1 : 0;
 686         segp = (fifo_segment_t *) a->segment;
 687         rmp->app_index = clib_host_to_net_u32 (a->app_index);
 688         rmp->app_mq = fifo_segment_msg_q_offset (segp, 0);
 689         rmp->vpp_ctrl_mq = fifo_segment_msg_q_offset (rx_mqs_seg, ctrl_thread);
 690         rmp->vpp_ctrl_mq_thread = ctrl_thread;
 691         rmp->n_fds = n_fds;
 692         rmp->fd_flags = fd_flags;
 693         if (vec_len (segp->ssvm.name))
 694           {
 695             vl_api_vec_to_api_string (segp->ssvm.name, &rmp->segment_name);
 696           }
 697         rmp->segment_size = segp->ssvm.ssvm_size;
 698         rmp->segment_handle = clib_host_to_net_u64 (a->segment_handle);
 699       }
 700   }));
 701   /* *INDENT-ON* */
 702
 703   if (n_fds)
 704     session_send_fds (reg, fds, n_fds);
 705   vec_free (fds);
 706 }
 707
 708 static void
 709 vl_api_app_worker_add_del_t_handler (vl_api_app_worker_add_del_t * mp)
 710 {
 711   int rv = 0, fds[SESSION_N_FD_TYPE], n_fds = 0;
 712   vl_api_app_worker_add_del_reply_t *rmp;
 713   vl_api_registration_t *reg;
 714   application_t *app;
 715   u8 fd_flags = 0;
 716
 717   if (!session_main_is_enabled () || appns_sapi_enabled ())
 718     {
 719       rv = VNET_API_ERROR_FEATURE_DISABLED;
 720       goto done;
 721     }
 722
 723   reg = vl_api_client_index_to_registration (mp->client_index);
 724   if (!reg)
 725     return;
 726
 727   app = application_get_if_valid (clib_net_to_host_u32 (mp->app_index));
 728   if (!app)
 729     {
 730       rv = VNET_API_ERROR_INVALID_VALUE;
 731       goto done;
 732     }
 733
 734   vnet_app_worker_add_del_args_t args = {
 735     .app_index = app->app_index,
 736     .wrk_map_index = clib_net_to_host_u32 (mp->wrk_index),
 737     .api_client_index = mp->client_index,
 738     .is_add = mp->is_add
 739   };
 740   rv = vnet_app_worker_add_del (&args);
 741   if (rv)
 742     {
 743       clib_warning ("app worker add/del returned: %d", rv);
 744       goto done;
 745     }
 746
 747   if (!mp->is_add)
 748     goto done;
 749
 750   /* Send fifo segment fd if needed */
 751   if (ssvm_type (args.segment) == SSVM_SEGMENT_MEMFD)
 752     {
 753       fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
 754       fds[n_fds] = args.segment->fd;
 755       n_fds += 1;
 756     }
 757   if (application_segment_manager_properties (app)->use_mq_eventfd)
 758     {
 759       fd_flags |= SESSION_FD_F_MQ_EVENTFD;
 760       fds[n_fds] = svm_msg_q_get_eventfd (args.evt_q);
 761       n_fds += 1;
 762     }
 763
 764   /* *INDENT-OFF* */
 765 done:
 766   REPLY_MACRO2 (VL_API_APP_WORKER_ADD_DEL_REPLY, ({
 767     rmp->is_add = mp->is_add;
 768     rmp->wrk_index = clib_host_to_net_u32 (args.wrk_map_index);
 769     rmp->segment_handle = clib_host_to_net_u64 (args.segment_handle);
 770     if (!rv && mp->is_add)
 771       {
 772         rmp->app_event_queue_address =
 773           fifo_segment_msg_q_offset ((fifo_segment_t *) args.segment, 0);
 774         rmp->n_fds = n_fds;
 775         rmp->fd_flags = fd_flags;
 776         if (vec_len (args.segment->name))
 777           {
 778             vl_api_vec_to_api_string (args.segment->name, &rmp->segment_name);
 779           }
 780       }
 781   }));
 782   /* *INDENT-ON* */
 783
 784   if (n_fds)
 785     session_send_fds (reg, fds, n_fds);
 786 }
 787
 788 static void
 789 vl_api_application_detach_t_handler (vl_api_application_detach_t * mp)
 790 {
 791   vl_api_application_detach_reply_t *rmp;
 792   int rv = VNET_API_ERROR_INVALID_VALUE_2;
 793   vnet_app_detach_args_t _a, *a = &_a;
 794   application_t *app;
 795
 796   if (!session_main_is_enabled () || appns_sapi_enabled ())
 797     {
 798       rv = VNET_API_ERROR_FEATURE_DISABLED;
 799       goto done;
 800     }
 801
 802   app = application_lookup (mp->client_index);
 803   if (app)
 804     {
 805       a->app_index = app->app_index;
 806       a->api_client_index = mp->client_index;
 807       rv = vnet_application_detach (a);
 808     }
 809
 810 done:
 811   REPLY_MACRO (VL_API_APPLICATION_DETACH_REPLY);
 812 }
 813
 814 static void
 815 vl_api_app_namespace_add_del_t_handler (vl_api_app_namespace_add_del_t * mp)
 816 {
 817   vl_api_app_namespace_add_del_reply_t *rmp;
 818   u32 appns_index = 0;
 819   u8 *ns_id = 0;
 820   int rv = 0;
 821   if (session_main_is_enabled () == 0)
 822     {
 823       rv = VNET_API_ERROR_FEATURE_DISABLED;
 824       goto done;
 825     }
 826
 827   ns_id = vl_api_from_api_to_new_vec (mp, &mp->namespace_id);
 828
 829   vnet_app_namespace_add_del_args_t args = {
 830     .ns_id = ns_id,
 831     .secret = clib_net_to_host_u64 (mp->secret),
 832     .sw_if_index = clib_net_to_host_u32 (mp->sw_if_index),
 833     .ip4_fib_id = clib_net_to_host_u32 (mp->ip4_fib_id),
 834     .ip6_fib_id = clib_net_to_host_u32 (mp->ip6_fib_id),
 835     .is_add = 1
 836   };
 837   rv = vnet_app_namespace_add_del (&args);
 838   if (!rv)
 839     {
 840       appns_index = app_namespace_index_from_id (ns_id);
 841       if (appns_index == APP_NAMESPACE_INVALID_INDEX)
 842         {
 843           clib_warning ("app ns lookup failed");
 844           rv = VNET_API_ERROR_UNSPECIFIED;
 845         }
 846     }
 847   vec_free (ns_id);
 848
 849   /* *INDENT-OFF* */
 850 done:
 851   REPLY_MACRO2 (VL_API_APP_NAMESPACE_ADD_DEL_REPLY, ({
 852     if (!rv)
 853       rmp->appns_index = clib_host_to_net_u32 (appns_index);
 854   }));
 855   /* *INDENT-ON* */
 856 }
 857
 858 static void
 859 vl_api_session_rule_add_del_t_handler (vl_api_session_rule_add_del_t * mp)
 860 {
 861   vl_api_session_rule_add_del_reply_t *rmp;
 862   session_rule_add_del_args_t args;
 863   session_rule_table_add_del_args_t *table_args = &args.table_args;
 864   int rv = 0;
 865
 866   clib_memset (&args, 0, sizeof (args));
 867
 868   ip_prefix_decode (&mp->lcl, &table_args->lcl);
 869   ip_prefix_decode (&mp->rmt, &table_args->rmt);
 870
 871   table_args->lcl_port = mp->lcl_port;
 872   table_args->rmt_port = mp->rmt_port;
 873   table_args->action_index = clib_net_to_host_u32 (mp->action_index);
 874   table_args->is_add = mp->is_add;
 875   mp->tag[sizeof (mp->tag) - 1] = 0;
 876   table_args->tag = format (0, "%s", mp->tag);
 877   args.appns_index = clib_net_to_host_u32 (mp->appns_index);
 878   args.scope = mp->scope;
 879   args.transport_proto =
 880     api_session_transport_proto_decode (&mp->transport_proto) ==
 881     TRANSPORT_PROTO_UDP ? 1 : 0;
 882
 883   rv = vnet_session_rule_add_del (&args);
 884   if (rv)
 885     clib_warning ("rule add del returned: %d", rv);
 886   vec_free (table_args->tag);
 887   REPLY_MACRO (VL_API_SESSION_RULE_ADD_DEL_REPLY);
 888 }
 889
 890 static void
 891 send_session_rule_details4 (mma_rule_16_t * rule, u8 is_local,
 892                             u8 transport_proto, u32 appns_index, u8 * tag,
 893                             vl_api_registration_t * reg, u32 context)
 894 {
 895   vl_api_session_rules_details_t *rmp = 0;
 896   session_mask_or_match_4_t *match =
 897     (session_mask_or_match_4_t *) & rule->match;
 898   session_mask_or_match_4_t *mask =
 899     (session_mask_or_match_4_t *) & rule->mask;
 900   fib_prefix_t lcl, rmt;
 901
 902   rmp = vl_msg_api_alloc (sizeof (*rmp));
 903   clib_memset (rmp, 0, sizeof (*rmp));
 904   rmp->_vl_msg_id = ntohs (REPLY_MSG_ID_BASE + VL_API_SESSION_RULES_DETAILS);
 905   rmp->context = context;
 906
 907   clib_memset (&lcl, 0, sizeof (lcl));
 908   clib_memset (&rmt, 0, sizeof (rmt));
 909   ip_set (&lcl.fp_addr, &match->lcl_ip, 1);
 910   ip_set (&rmt.fp_addr, &match->rmt_ip, 1);
 911   lcl.fp_len = ip4_mask_to_preflen (&mask->lcl_ip);
 912   rmt.fp_len = ip4_mask_to_preflen (&mask->rmt_ip);
 913
 914   ip_prefix_encode (&lcl, &rmp->lcl);
 915   ip_prefix_encode (&rmt, &rmp->rmt);
 916   rmp->lcl_port = match->lcl_port;
 917   rmp->rmt_port = match->rmt_port;
 918   rmp->action_index = clib_host_to_net_u32 (rule->action_index);
 919   rmp->scope =
 920     is_local ? SESSION_RULE_SCOPE_API_LOCAL : SESSION_RULE_SCOPE_API_GLOBAL;
 921   rmp->transport_proto = api_session_transport_proto_encode (transport_proto);
 922   rmp->appns_index = clib_host_to_net_u32 (appns_index);
 923   if (tag)
 924     {
 925       clib_memcpy_fast (rmp->tag, tag, vec_len (tag));
 926       rmp->tag[vec_len (tag)] = 0;
 927     }
 928
 929   vl_api_send_msg (reg, (u8 *) rmp);
 930 }
 931
 932 static void
 933 send_session_rule_details6 (mma_rule_40_t * rule, u8 is_local,
 934                             u8 transport_proto, u32 appns_index, u8 * tag,
 935                             vl_api_registration_t * reg, u32 context)
 936 {
 937   vl_api_session_rules_details_t *rmp = 0;
 938   session_mask_or_match_6_t *match =
 939     (session_mask_or_match_6_t *) & rule->match;
 940   session_mask_or_match_6_t *mask =
 941     (session_mask_or_match_6_t *) & rule->mask;
 942   fib_prefix_t lcl, rmt;
 943
 944   rmp = vl_msg_api_alloc (sizeof (*rmp));
 945   clib_memset (rmp, 0, sizeof (*rmp));
 946   rmp->_vl_msg_id = ntohs (REPLY_MSG_ID_BASE + VL_API_SESSION_RULES_DETAILS);
 947   rmp->context = context;
 948
 949   clib_memset (&lcl, 0, sizeof (lcl));
 950   clib_memset (&rmt, 0, sizeof (rmt));
 951   ip_set (&lcl.fp_addr, &match->lcl_ip, 0);
 952   ip_set (&rmt.fp_addr, &match->rmt_ip, 0);
 953   lcl.fp_len = ip6_mask_to_preflen (&mask->lcl_ip);
 954   rmt.fp_len = ip6_mask_to_preflen (&mask->rmt_ip);
 955
 956   ip_prefix_encode (&lcl, &rmp->lcl);
 957   ip_prefix_encode (&rmt, &rmp->rmt);
 958   rmp->lcl_port = match->lcl_port;
 959   rmp->rmt_port = match->rmt_port;
 960   rmp->action_index = clib_host_to_net_u32 (rule->action_index);
 961   rmp->scope =
 962     is_local ? SESSION_RULE_SCOPE_API_LOCAL : SESSION_RULE_SCOPE_API_GLOBAL;
 963   rmp->transport_proto = api_session_transport_proto_encode (transport_proto);
 964   rmp->appns_index = clib_host_to_net_u32 (appns_index);
 965   if (tag)
 966     {
 967       clib_memcpy_fast (rmp->tag, tag, vec_len (tag));
 968       rmp->tag[vec_len (tag)] = 0;
 969     }
 970
 971   vl_api_send_msg (reg, (u8 *) rmp);
 972 }
 973
 974 static void
 975 send_session_rules_table_details (session_rules_table_t * srt, u8 fib_proto,
 976                                   u8 tp, u8 is_local, u32 appns_index,
 977                                   vl_api_registration_t * reg, u32 context)
 978 {
 979   mma_rule_16_t *rule16;
 980   mma_rule_40_t *rule40;
 981   mma_rules_table_16_t *srt16;
 982   mma_rules_table_40_t *srt40;
 983   u32 ri;
 984
 985   if (is_local || fib_proto == FIB_PROTOCOL_IP4)
 986     {
 987       u8 *tag = 0;
 988       /* *INDENT-OFF* */
 989       srt16 = &srt->session_rules_tables_16;
 990       pool_foreach (rule16, srt16->rules)  {
 991         ri = mma_rules_table_rule_index_16 (srt16, rule16);
 992         tag = session_rules_table_rule_tag (srt, ri, 1);
 993         send_session_rule_details4 (rule16, is_local, tp, appns_index, tag,
 994                                     reg, context);
 995       }
 996       /* *INDENT-ON* */
 997     }
 998   if (is_local || fib_proto == FIB_PROTOCOL_IP6)
 999     {
1000       u8 *tag = 0;
1001       /* *INDENT-OFF* */
1002       srt40 = &srt->session_rules_tables_40;
1003       pool_foreach (rule40, srt40->rules)  {
1004         ri = mma_rules_table_rule_index_40 (srt40, rule40);
1005         tag = session_rules_table_rule_tag (srt, ri, 1);
1006         send_session_rule_details6 (rule40, is_local, tp, appns_index, tag,
1007                                     reg, context);
1008       }
1009       /* *INDENT-ON* */
1010     }
1011 }
1012
1013 static void
1014 vl_api_session_rules_dump_t_handler (vl_api_session_rules_dump_t * mp)
1015 {
1016   vl_api_registration_t *reg;
1017   session_table_t *st;
1018   u8 tp;
1019
1020   reg = vl_api_client_index_to_registration (mp->client_index);
1021   if (!reg)
1022     return;
1023
1024   /* *INDENT-OFF* */
1025   session_table_foreach (st, ({
1026     for (tp = 0; tp < TRANSPORT_N_PROTOS; tp++)
1027       {
1028         send_session_rules_table_details (&st->session_rules[tp],
1029                                           st->active_fib_proto, tp,
1030                                           st->is_local, st->appns_index, reg,
1031                                           mp->context);
1032       }
1033   }));
1034   /* *INDENT-ON* */
1035 }
1036
1037 static void
1038 vl_api_app_add_cert_key_pair_t_handler (vl_api_app_add_cert_key_pair_t * mp)
1039 {
1040   vl_api_app_add_cert_key_pair_reply_t *rmp;
1041   vnet_app_add_cert_key_pair_args_t _a, *a = &_a;
1042   u32 certkey_len, key_len, cert_len;
1043   int rv = 0;
1044   if (session_main_is_enabled () == 0)
1045     {
1046       rv = VNET_API_ERROR_FEATURE_DISABLED;
1047       goto done;
1048     }
1049
1050   cert_len = clib_net_to_host_u16 (mp->cert_len);
1051   if (cert_len > 10000)
1052     {
1053       rv = VNET_API_ERROR_INVALID_VALUE;
1054       goto done;
1055     }
1056
1057   certkey_len = clib_net_to_host_u16 (mp->certkey_len);
1058   if (certkey_len < cert_len)
1059     {
1060       rv = VNET_API_ERROR_INVALID_VALUE;
1061       goto done;
1062     }
1063
1064   key_len = certkey_len - cert_len;
1065   if (key_len > 10000)
1066     {
1067       rv = VNET_API_ERROR_INVALID_VALUE;
1068       goto done;
1069     }
1070
1071   clib_memset (a, 0, sizeof (*a));
1072   a->cert = mp->certkey;
1073   a->key = mp->certkey + cert_len;
1074   a->cert_len = cert_len;
1075   a->key_len = key_len;
1076   rv = vnet_app_add_cert_key_pair (a);
1077
1078 done:
1079   /* *INDENT-OFF* */
1080   REPLY_MACRO2 (VL_API_APP_ADD_CERT_KEY_PAIR_REPLY, ({
1081     if (!rv)
1082       rmp->index = clib_host_to_net_u32 (a->index);
1083   }));
1084   /* *INDENT-ON* */
1085 }
1086
1087 static void
1088 vl_api_app_del_cert_key_pair_t_handler (vl_api_app_del_cert_key_pair_t * mp)
1089 {
1090   vl_api_app_del_cert_key_pair_reply_t *rmp;
1091   u32 ckpair_index;
1092   int rv = 0;
1093   if (session_main_is_enabled () == 0)
1094     {
1095       rv = VNET_API_ERROR_FEATURE_DISABLED;
1096       goto done;
1097     }
1098   ckpair_index = clib_net_to_host_u32 (mp->index);
1099   rv = vnet_app_del_cert_key_pair (ckpair_index);
1100
1101 done:
1102   REPLY_MACRO (VL_API_APP_DEL_CERT_KEY_PAIR_REPLY);
1103 }
1104
1105 static clib_error_t *
1106 application_reaper_cb (u32 client_index)
1107 {
1108   application_t *app = application_lookup (client_index);
1109   vnet_app_detach_args_t _a, *a = &_a;
1110   if (app)
1111     {
1112       a->app_index = app->app_index;
1113       a->api_client_index = client_index;
1114       vnet_application_detach (a);
1115     }
1116   return 0;
1117 }
1118
1119 VL_MSG_API_REAPER_FUNCTION (application_reaper_cb);
1120
1121 /*
1122  * Socket api functions
1123  */
1124
1125 static void
1126 sapi_send_fds (app_worker_t * app_wrk, int *fds, int n_fds)
1127 {
1128   app_sapi_msg_t smsg = { 0 };
1129   app_namespace_t *app_ns;
1130   application_t *app;
1131   clib_socket_t *cs;
1132   u32 cs_index;
1133
1134   app = application_get (app_wrk->app_index);
1135   app_ns = app_namespace_get (app->ns_index);
1136   cs_index = appns_sapi_handle_sock_index (app_wrk->api_client_index);
1137   cs = appns_sapi_get_socket (app_ns, cs_index);
1138   if (PREDICT_FALSE (!cs))
1139     return;
1140
1141   /* There's no payload for the message only the type */
1142   smsg.type = APP_SAPI_MSG_TYPE_SEND_FDS;
1143   clib_socket_sendmsg (cs, &smsg, sizeof (smsg), fds, n_fds);
1144 }
1145
1146 static int
1147 mq_send_add_segment_sapi_cb (u32 app_wrk_index, u64 segment_handle)
1148 {
1149   int fds[SESSION_N_FD_TYPE], n_fds = 0;
1150   svm_msg_q_msg_t _msg, *msg = &_msg;
1151   session_app_add_segment_msg_t *mp;
1152   app_worker_t *app_wrk;
1153   session_event_t *evt;
1154   svm_msg_q_t *app_mq;
1155   fifo_segment_t *fs;
1156   ssvm_private_t *sp;
1157   u8 fd_flags = 0;
1158
1159   app_wrk = app_worker_get (app_wrk_index);
1160
1161   fs = segment_manager_get_segment_w_handle (segment_handle);
1162   sp = &fs->ssvm;
1163   ASSERT (ssvm_type (sp) == SSVM_SEGMENT_MEMFD);
1164
1165   fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
1166   fds[n_fds] = sp->fd;
1167   n_fds += 1;
1168
1169   app_mq = app_wrk->event_queue;
1170   if (mq_try_lock_and_alloc_msg (app_mq, msg))
1171     return -1;
1172
1173   /*
1174    * Send the fd over api socket
1175    */
1176   sapi_send_fds (app_wrk, fds, n_fds);
1177
1178   /*
1179    * Send the actual message over mq
1180    */
1181   evt = svm_msg_q_msg_data (app_mq, msg);
1182   clib_memset (evt, 0, sizeof (*evt));
1183   evt->event_type = SESSION_CTRL_EVT_APP_ADD_SEGMENT;
1184   mp = (session_app_add_segment_msg_t *) evt->data;
1185   clib_memset (mp, 0, sizeof (*mp));
1186   mp->segment_size = sp->ssvm_size;
1187   mp->fd_flags = fd_flags;
1188   mp->segment_handle = segment_handle;
1189   strncpy ((char *) mp->segment_name, (char *) sp->name,
1190            sizeof (mp->segment_name) - 1);
1191
1192   svm_msg_q_add_and_unlock (app_mq, msg);
1193
1194   return 0;
1195 }
1196
1197 static int
1198 mq_send_del_segment_sapi_cb (u32 app_wrk_index, u64 segment_handle)
1199 {
1200   svm_msg_q_msg_t _msg, *msg = &_msg;
1201   session_app_del_segment_msg_t *mp;
1202   app_worker_t *app_wrk;
1203   session_event_t *evt;
1204   svm_msg_q_t *app_mq;
1205
1206   app_wrk = app_worker_get (app_wrk_index);
1207
1208   app_mq = app_wrk->event_queue;
1209   if (mq_try_lock_and_alloc_msg (app_mq, msg))
1210     return -1;
1211
1212   evt = svm_msg_q_msg_data (app_mq, msg);
1213   clib_memset (evt, 0, sizeof (*evt));
1214   evt->event_type = SESSION_CTRL_EVT_APP_DEL_SEGMENT;
1215   mp = (session_app_del_segment_msg_t *) evt->data;
1216   clib_memset (mp, 0, sizeof (*mp));
1217   mp->segment_handle = segment_handle;
1218   svm_msg_q_add_and_unlock (app_mq, msg);
1219
1220   return 0;
1221 }
1222
1223 static session_cb_vft_t session_mq_sapi_cb_vft = {
1224   .session_accept_callback = mq_send_session_accepted_cb,
1225   .session_disconnect_callback = mq_send_session_disconnected_cb,
1226   .session_connected_callback = mq_send_session_connected_cb,
1227   .session_reset_callback = mq_send_session_reset_cb,
1228   .session_migrate_callback = mq_send_session_migrate_cb,
1229   .session_cleanup_callback = mq_send_session_cleanup_cb,
1230   .add_segment_callback = mq_send_add_segment_sapi_cb,
1231   .del_segment_callback = mq_send_del_segment_sapi_cb,
1232 };
1233
1234 static void
1235 session_api_attach_handler (app_namespace_t * app_ns, clib_socket_t * cs,
1236                             app_sapi_attach_msg_t * mp)
1237 {
1238   int rv = 0, *fds = 0, n_fds = 0, i, n_workers;
1239   vnet_app_attach_args_t _a, *a = &_a;
1240   app_sapi_attach_reply_msg_t *rmp;
1241   u8 fd_flags = 0, ctrl_thread;
1242   app_ns_api_handle_t *handle;
1243   fifo_segment_t *rx_mqs_seg;
1244   app_sapi_msg_t msg = { 0 };
1245   app_worker_t *app_wrk;
1246   application_t *app;
1247   svm_msg_q_t *rx_mq;
1248
1249   /* Make sure name is null terminated */
1250   mp->name[63] = 0;
1251
1252   clib_memset (a, 0, sizeof (*a));
1253   a->api_client_index = appns_sapi_socket_handle (app_ns, cs);
1254   a->name = format (0, "%s", (char *) mp->name);
1255   a->options = mp->options;
1256   a->session_cb_vft = &session_mq_sapi_cb_vft;
1257   a->use_sock_api = 1;
1258   a->options[APP_OPTIONS_NAMESPACE] = app_namespace_index (app_ns);
1259
1260   if ((rv = vnet_application_attach (a)))
1261     {
1262       clib_warning ("attach returned: %d", rv);
1263       goto done;
1264     }
1265
1266   n_workers = vlib_num_workers ();
1267   vec_validate (fds, 3 /* segs + tx evtfd */ + n_workers);
1268
1269   /* Send event queues segment */
1270   app = application_get (a->app_index);
1271   rx_mqs_seg = application_get_rx_mqs_segment (app);
1272
1273   fd_flags |= SESSION_FD_F_VPP_MQ_SEGMENT;
1274   fds[n_fds] = rx_mqs_seg->ssvm.fd;
1275   n_fds += 1;
1276
1277   /* Send fifo segment fd if needed */
1278   if (ssvm_type (a->segment) == SSVM_SEGMENT_MEMFD)
1279     {
1280       fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
1281       fds[n_fds] = a->segment->fd;
1282       n_fds += 1;
1283     }
1284   if (a->options[APP_OPTIONS_FLAGS] & APP_OPTIONS_FLAGS_EVT_MQ_USE_EVENTFD)
1285     {
1286       fd_flags |= SESSION_FD_F_MQ_EVENTFD;
1287       fds[n_fds] = svm_msg_q_get_eventfd (a->app_evt_q);
1288       n_fds += 1;
1289     }
1290
1291   if (application_use_private_rx_mqs ())
1292     {
1293       fd_flags |= SESSION_FD_F_VPP_MQ_EVENTFD;
1294       for (i = 0; i < n_workers + 1; i++)
1295         {
1296           rx_mq = application_rx_mq_get (app, i);
1297           fds[n_fds] = svm_msg_q_get_eventfd (rx_mq);
1298           n_fds += 1;
1299         }
1300     }
1301
1302 done:
1303
1304   msg.type = APP_SAPI_MSG_TYPE_ATTACH_REPLY;
1305   rmp = &msg.attach_reply;
1306   rmp->retval = rv;
1307   if (!rv)
1308     {
1309       ctrl_thread = n_workers ? 1 : 0;
1310       rmp->app_index = a->app_index;
1311       rmp->app_mq =
1312         fifo_segment_msg_q_offset ((fifo_segment_t *) a->segment, 0);
1313       rmp->vpp_ctrl_mq = fifo_segment_msg_q_offset (rx_mqs_seg, ctrl_thread);
1314       rmp->vpp_ctrl_mq_thread = ctrl_thread;
1315       rmp->n_fds = n_fds;
1316       rmp->fd_flags = fd_flags;
1317       /* No segment name and size since we only support memfds
1318        * in this configuration */
1319       rmp->segment_handle = a->segment_handle;
1320       rmp->api_client_handle = a->api_client_index;
1321
1322       /* Update app index for socket */
1323       handle = (app_ns_api_handle_t *) & cs->private_data;
1324       app_wrk = application_get_worker (app, 0);
1325       handle->aah_app_wrk_index = app_wrk->wrk_index;
1326     }
1327
1328   clib_socket_sendmsg (cs, &msg, sizeof (msg), fds, n_fds);
1329   vec_free (a->name);
1330   vec_free (fds);
1331 }
1332
1333 static void
1334 sapi_socket_close_w_handle (u32 api_handle)
1335 {
1336   app_namespace_t *app_ns = app_namespace_get (api_handle >> 16);
1337   u16 sock_index = api_handle & 0xffff;
1338   app_ns_api_handle_t *handle;
1339   clib_socket_t *cs;
1340   clib_file_t *cf;
1341
1342   cs = appns_sapi_get_socket (app_ns, sock_index);
1343   if (!cs)
1344     return;
1345
1346   handle = (app_ns_api_handle_t *) & cs->private_data;
1347   cf = clib_file_get (&file_main, handle->aah_file_index);
1348   clib_file_del (&file_main, cf);
1349
1350   clib_socket_close (cs);
1351   appns_sapi_free_socket (app_ns, cs);
1352 }
1353
1354 static void
1355 sapi_add_del_worker_handler (app_namespace_t * app_ns,
1356                              clib_socket_t * cs,
1357                              app_sapi_worker_add_del_msg_t * mp)
1358 {
1359   int rv = 0, fds[SESSION_N_FD_TYPE], n_fds = 0;
1360   app_sapi_worker_add_del_reply_msg_t *rmp;
1361   app_ns_api_handle_t *handle;
1362   app_sapi_msg_t msg = { 0 };
1363   app_worker_t *app_wrk;
1364   u32 sapi_handle = -1;
1365   application_t *app;
1366   u8 fd_flags = 0;
1367
1368   app = application_get_if_valid (mp->app_index);
1369   if (!app)
1370     {
1371       rv = VNET_API_ERROR_INVALID_VALUE;
1372       goto done;
1373     }
1374
1375   sapi_handle = appns_sapi_socket_handle (app_ns, cs);
1376
1377   vnet_app_worker_add_del_args_t args = {
1378     .app_index = app->app_index,
1379     .wrk_map_index = mp->wrk_index,
1380     .api_client_index = sapi_handle,
1381     .is_add = mp->is_add
1382   };
1383   rv = vnet_app_worker_add_del (&args);
1384   if (rv)
1385     {
1386       clib_warning ("app worker add/del returned: %d", rv);
1387       goto done;
1388     }
1389
1390   if (!mp->is_add)
1391     {
1392       sapi_socket_close_w_handle (sapi_handle);
1393       goto done;
1394     }
1395
1396   /* Send fifo segment fd if needed */
1397   if (ssvm_type (args.segment) == SSVM_SEGMENT_MEMFD)
1398     {
1399       fd_flags |= SESSION_FD_F_MEMFD_SEGMENT;
1400       fds[n_fds] = args.segment->fd;
1401       n_fds += 1;
1402     }
1403   if (application_segment_manager_properties (app)->use_mq_eventfd)
1404     {
1405       fd_flags |= SESSION_FD_F_MQ_EVENTFD;
1406       fds[n_fds] = svm_msg_q_get_eventfd (args.evt_q);
1407       n_fds += 1;
1408     }
1409
1410 done:
1411
1412   msg.type = APP_SAPI_MSG_TYPE_ADD_DEL_WORKER_REPLY;
1413   rmp = &msg.worker_add_del_reply;
1414   rmp->retval = rv;
1415   rmp->is_add = mp->is_add;
1416   rmp->api_client_handle = sapi_handle;
1417   rmp->wrk_index = args.wrk_map_index;
1418   rmp->segment_handle = args.segment_handle;
1419   if (!rv && mp->is_add)
1420     {
1421       /* No segment name and size. This supports only memfds */
1422       rmp->app_event_queue_address =
1423         fifo_segment_msg_q_offset ((fifo_segment_t *) args.segment, 0);
1424       rmp->n_fds = n_fds;
1425       rmp->fd_flags = fd_flags;
1426
1427       /* Update app index for socket */
1428       handle = (app_ns_api_handle_t *) & cs->private_data;
1429       app_wrk = application_get_worker (app, args.wrk_map_index);
1430       handle->aah_app_wrk_index = app_wrk->wrk_index;
1431     }
1432
1433   clib_socket_sendmsg (cs, &msg, sizeof (msg), fds, n_fds);
1434 }
1435
1436 static void
1437 sapi_socket_detach (app_namespace_t * app_ns, clib_socket_t * cs)
1438 {
1439   app_ns_api_handle_t *handle;
1440   app_worker_t *app_wrk;
1441   u32 api_client_handle;
1442
1443   api_client_handle = appns_sapi_socket_handle (app_ns, cs);
1444   sapi_socket_close_w_handle (api_client_handle);
1445
1446   /* Cleanup everything because app worker closed socket or crashed */
1447   handle = (app_ns_api_handle_t *) & cs->private_data;
1448   app_wrk = app_worker_get (handle->aah_app_wrk_index);
1449
1450   vnet_app_worker_add_del_args_t args = {
1451     .app_index = app_wrk->app_index,
1452     .wrk_map_index = app_wrk->wrk_map_index,
1453     .api_client_index = api_client_handle,
1454     .is_add = 0
1455   };
1456   /* Send rpc to main thread for worker barrier */
1457   vlib_rpc_call_main_thread (vnet_app_worker_add_del, (u8 *) & args,
1458                              sizeof (args));
1459 }
1460
1461 static clib_error_t *
1462 sapi_sock_read_ready (clib_file_t * cf)
1463 {
1464   app_ns_api_handle_t *handle = (app_ns_api_handle_t *) & cf->private_data;
1465   vlib_main_t *vm = vlib_get_main ();
1466   app_sapi_msg_t msg = { 0 };
1467   app_namespace_t *app_ns;
1468   clib_error_t *err = 0;
1469   clib_socket_t *cs;
1470
1471   app_ns = app_namespace_get (handle->aah_app_ns_index);
1472   cs = appns_sapi_get_socket (app_ns, handle->aah_sock_index);
1473   if (!cs)
1474     goto error;
1475
1476   err = clib_socket_recvmsg (cs, &msg, sizeof (msg), 0, 0);
1477   if (err)
1478     {
1479       clib_error_free (err);
1480       sapi_socket_detach (app_ns, cs);
1481       goto error;
1482     }
1483
1484   handle = (app_ns_api_handle_t *) & cs->private_data;
1485
1486   vlib_worker_thread_barrier_sync (vm);
1487
1488   switch (msg.type)
1489     {
1490     case APP_SAPI_MSG_TYPE_ATTACH:
1491       session_api_attach_handler (app_ns, cs, &msg.attach);
1492       break;
1493     case APP_SAPI_MSG_TYPE_ADD_DEL_WORKER:
1494       sapi_add_del_worker_handler (app_ns, cs, &msg.worker_add_del);
1495       break;
1496     default:
1497       clib_warning ("app wrk %u unknown message type: %u",
1498                     handle->aah_app_wrk_index, msg.type);
1499       break;
1500     }
1501
1502   vlib_worker_thread_barrier_release (vm);
1503
1504 error:
1505   return 0;
1506 }
1507
1508 static clib_error_t *
1509 sapi_sock_write_ready (clib_file_t * cf)
1510 {
1511   app_ns_api_handle_t *handle = (app_ns_api_handle_t *) & cf->private_data;
1512   clib_warning ("called for app ns %u", handle->aah_app_ns_index);
1513   return 0;
1514 }
1515
1516 static clib_error_t *
1517 sapi_sock_error (clib_file_t * cf)
1518 {
1519   app_ns_api_handle_t *handle = (app_ns_api_handle_t *) & cf->private_data;
1520   app_namespace_t *app_ns;
1521   clib_socket_t *cs;
1522
1523   app_ns = app_namespace_get (handle->aah_app_ns_index);
1524   cs = appns_sapi_get_socket (app_ns, handle->aah_sock_index);
1525   if (!cs)
1526     return 0;
1527
1528   sapi_socket_detach (app_ns, cs);
1529   return 0;
1530 }
1531
1532 static clib_error_t *
1533 sapi_sock_accept_ready (clib_file_t * scf)
1534 {
1535   app_ns_api_handle_t handle = *(app_ns_api_handle_t *) & scf->private_data;
1536   app_namespace_t *app_ns;
1537   clib_file_t cf = { 0 };
1538   clib_error_t *err = 0;
1539   clib_socket_t *ccs, *scs;
1540
1541   /* Listener files point to namespace */
1542   app_ns = app_namespace_get (handle.aah_app_ns_index);
1543
1544   /*
1545    * Initialize client socket
1546    */
1547   ccs = appns_sapi_alloc_socket (app_ns);
1548
1549   /* Grab server socket after client is initialized  */
1550   scs = appns_sapi_get_socket (app_ns, handle.aah_sock_index);
1551   if (!scs)
1552     goto error;
1553
1554   err = clib_socket_accept (scs, ccs);
1555   if (err)
1556     {
1557       clib_error_report (err);
1558       goto error;
1559     }
1560
1561   cf.read_function = sapi_sock_read_ready;
1562   cf.write_function = sapi_sock_write_ready;
1563   cf.error_function = sapi_sock_error;
1564   cf.file_descriptor = ccs->fd;
1565   /* File points to app namespace and socket */
1566   handle.aah_sock_index = appns_sapi_socket_index (app_ns, ccs);
1567   cf.private_data = handle.as_u64;
1568   cf.description = format (0, "app sock conn fd: %d", ccs->fd);
1569
1570   /* Poll until we get an attach message. Socket points to file and
1571    * application that owns the socket */
1572   handle.aah_app_wrk_index = APP_INVALID_INDEX;
1573   handle.aah_file_index = clib_file_add (&file_main, &cf);
1574   ccs->private_data = handle.as_u64;
1575
1576   return err;
1577
1578 error:
1579   appns_sapi_free_socket (app_ns, ccs);
1580   return err;
1581 }
1582
1583 int
1584 appns_sapi_add_ns_socket (app_namespace_t * app_ns)
1585 {
1586   char *subdir = "/app_ns_sockets/";
1587   app_ns_api_handle_t *handle;
1588   clib_file_t cf = { 0 };
1589   struct stat file_stat;
1590   clib_error_t *err;
1591   clib_socket_t *cs;
1592   u8 *dir = 0;
1593   int rv = 0;
1594
1595   vec_add (dir, vlib_unix_get_runtime_dir (),
1596            strlen (vlib_unix_get_runtime_dir ()));
1597   vec_add (dir, (u8 *) subdir, strlen (subdir));
1598
1599   err = vlib_unix_recursive_mkdir ((char *) dir);
1600   if (err)
1601     {
1602       clib_error_report (err);
1603       rv = -1;
1604       goto error;
1605     }
1606
1607   app_ns->sock_name = format (0, "%v%v%c", dir, app_ns->ns_id, 0);
1608
1609   /*
1610    * Create and initialize socket to listen on
1611    */
1612   cs = appns_sapi_alloc_socket (app_ns);
1613   cs->config = (char *) app_ns->sock_name;
1614   cs->flags = CLIB_SOCKET_F_IS_SERVER |
1615     CLIB_SOCKET_F_ALLOW_GROUP_WRITE |
1616     CLIB_SOCKET_F_SEQPACKET | CLIB_SOCKET_F_PASSCRED;
1617
1618   if ((err = clib_socket_init (cs)))
1619     {
1620       clib_error_report (err);
1621       rv = -1;
1622       goto error;
1623     }
1624
1625   if (stat ((char *) app_ns->sock_name, &file_stat) == -1)
1626     {
1627       rv = -1;
1628       goto error;
1629     }
1630
1631   /*
1632    * Start polling it
1633    */
1634   cf.read_function = sapi_sock_accept_ready;
1635   cf.file_descriptor = cs->fd;
1636   /* File points to namespace */
1637   handle = (app_ns_api_handle_t *) & cf.private_data;
1638   handle->aah_app_ns_index = app_namespace_index (app_ns);
1639   handle->aah_sock_index = appns_sapi_socket_index (app_ns, cs);
1640   cf.description = format (0, "app sock listener: %s", app_ns->sock_name);
1641
1642   /* Socket points to clib file index */
1643   handle = (app_ns_api_handle_t *) & cs->private_data;
1644   handle->aah_file_index = clib_file_add (&file_main, &cf);
1645   handle->aah_app_wrk_index = APP_INVALID_INDEX;
1646
1647 error:
1648   vec_free (dir);
1649   return rv;
1650 }
1651
1652 static void
1653 vl_api_application_tls_cert_add_t_handler (
1654   vl_api_application_tls_cert_add_t *mp)
1655 {
1656   /* deprecated */
1657 }
1658
1659 static void
1660 vl_api_application_tls_key_add_t_handler (vl_api_application_tls_key_add_t *mp)
1661 {
1662   /* deprecated */
1663 }
1664
1665 #include <vnet/session/session.api.c>
1666 static clib_error_t *
1667 session_api_hookup (vlib_main_t *vm)
1668 {
1669   /*
1670    * Set up the (msg_name, crc, message-id) table
1671    */
1672   REPLY_MSG_ID_BASE = setup_message_id_table ();
1673
1674   return 0;
1675 }
1676
1677 VLIB_API_INIT_FUNCTION (session_api_hookup);
1678
1679 /*
1680  * fd.io coding-style-patch-verification: ON
1681  *
1682  * Local Variables:
1683  * eval: (c-set-style "gnu")
1684  * End:
1685  */