src/vnet/session/application.c

   1 /*
   2  * Copyright (c) 2017 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/session/application.h>
  17 #include <vnet/session/application_interface.h>
  18 #include <vnet/session/application_namespace.h>
  19 #include <vnet/session/session.h>
  20
  21 /**
  22  * Pool from which we allocate all applications
  23  */
  24 static application_t *app_pool;
  25
  26 /**
  27  * Hash table of apps by api client index
  28  */
  29 static uword *app_by_api_client_index;
  30
  31 /**
  32  * Hash table of builtin apps by name
  33  */
  34 static uword *app_by_name;
  35
  36 static u8 *
  37 app_get_name_from_reg_index (application_t * app)
  38 {
  39   u8 *app_name;
  40
  41   vl_api_registration_t *regp;
  42   regp = vl_api_client_index_to_registration (app->api_client_index);
  43   if (!regp)
  44     app_name = format (0, "builtin-%d%c", app->index, 0);
  45   else
  46     app_name = format (0, "%s%c", regp->name, 0);
  47
  48   return app_name;
  49 }
  50
  51 static u8 *
  52 app_get_name (application_t * app)
  53 {
  54   if (!app->name)
  55     return app_get_name_from_reg_index (app);
  56   return app->name;
  57 }
  58
  59 u32
  60 application_session_table (application_t * app, u8 fib_proto)
  61 {
  62   app_namespace_t *app_ns;
  63   app_ns = app_namespace_get (app->ns_index);
  64   if (!application_has_global_scope (app))
  65     return APP_INVALID_INDEX;
  66   if (fib_proto == FIB_PROTOCOL_IP4)
  67     return session_lookup_get_index_for_fib (fib_proto,
  68                                              app_ns->ip4_fib_index);
  69   else
  70     return session_lookup_get_index_for_fib (fib_proto,
  71                                              app_ns->ip6_fib_index);
  72 }
  73
  74 u32
  75 application_local_session_table (application_t * app)
  76 {
  77   app_namespace_t *app_ns;
  78   if (!application_has_local_scope (app))
  79     return APP_INVALID_INDEX;
  80   app_ns = app_namespace_get (app->ns_index);
  81   return app_ns->local_table_index;
  82 }
  83
  84 int
  85 application_api_queue_is_full (application_t * app)
  86 {
  87   svm_queue_t *q;
  88
  89   /* builtin servers are always OK */
  90   if (app->api_client_index == ~0)
  91     return 0;
  92
  93   q = vl_api_client_index_to_input_queue (app->api_client_index);
  94   if (!q)
  95     return 1;
  96
  97   if (q->cursize == q->maxsize)
  98     return 1;
  99   return 0;
 100 }
 101
 102 /**
 103  * Returns app name
 104  *
 105  * Since the name is not stored per app, we generate it on the fly. It is
 106  * the caller's responsibility to free the vector
 107  */
 108 u8 *
 109 application_name_from_index (u32 app_index)
 110 {
 111   application_t *app = application_get (app_index);
 112   if (!app)
 113     return 0;
 114   return app_get_name_from_reg_index (app);
 115 }
 116
 117 static void
 118 application_table_add (application_t * app)
 119 {
 120   if (app->api_client_index != APP_INVALID_INDEX)
 121     hash_set (app_by_api_client_index, app->api_client_index, app->index);
 122   else if (app->name)
 123     hash_set_mem (app_by_name, app->name, app->index);
 124 }
 125
 126 static void
 127 application_table_del (application_t * app)
 128 {
 129   if (app->api_client_index != APP_INVALID_INDEX)
 130     hash_unset (app_by_api_client_index, app->api_client_index);
 131   else if (app->name)
 132     hash_unset_mem (app_by_name, app->name);
 133 }
 134
 135 application_t *
 136 application_lookup (u32 api_client_index)
 137 {
 138   uword *p;
 139   p = hash_get (app_by_api_client_index, api_client_index);
 140   if (p)
 141     return application_get (p[0]);
 142
 143   return 0;
 144 }
 145
 146 application_t *
 147 application_lookup_name (const u8 * name)
 148 {
 149   uword *p;
 150   p = hash_get_mem (app_by_name, name);
 151   if (p)
 152     return application_get (p[0]);
 153
 154   return 0;
 155 }
 156
 157 application_t *
 158 application_new ()
 159 {
 160   application_t *app;
 161   pool_get (app_pool, app);
 162   memset (app, 0, sizeof (*app));
 163   app->index = application_get_index (app);
 164   app->connects_seg_manager = APP_INVALID_SEGMENT_MANAGER_INDEX;
 165   app->first_segment_manager = APP_INVALID_SEGMENT_MANAGER_INDEX;
 166   app->local_segment_manager = APP_INVALID_SEGMENT_MANAGER_INDEX;
 167   if (CLIB_DEBUG > 1)
 168     clib_warning ("[%d] New app (%d)", getpid (), app->index);
 169   return app;
 170 }
 171
 172 void
 173 application_del (application_t * app)
 174 {
 175   vnet_unbind_args_t _a, *a = &_a;
 176   u64 handle, *handles = 0;
 177   segment_manager_t *sm;
 178   u32 index;
 179   int i;
 180
 181   /*
 182    * The app event queue allocated in first segment is cleared with
 183    * the segment manager. No need to explicitly free it.
 184    */
 185   if (CLIB_DEBUG > 1)
 186     clib_warning ("[%d] Delete app (%d)", getpid (), app->index);
 187
 188   if (application_is_proxy (app))
 189     application_remove_proxy (app);
 190
 191   /*
 192    *  Listener cleanup
 193    */
 194
 195   /* *INDENT-OFF* */
 196   hash_foreach (handle, index, app->listeners_table,
 197   ({
 198     vec_add1 (handles, handle);
 199     sm = segment_manager_get (index);
 200     sm->app_index = SEGMENT_MANAGER_INVALID_APP_INDEX;
 201   }));
 202   /* *INDENT-ON* */
 203
 204   for (i = 0; i < vec_len (handles); i++)
 205     {
 206       a->app_index = app->index;
 207       a->handle = handles[i];
 208       /* seg manager is removed when unbind completes */
 209       vnet_unbind (a);
 210     }
 211
 212   /*
 213    * Connects segment manager cleanup
 214    */
 215
 216   if (app->connects_seg_manager != APP_INVALID_SEGMENT_MANAGER_INDEX)
 217     {
 218       sm = segment_manager_get (app->connects_seg_manager);
 219       sm->app_index = SEGMENT_MANAGER_INVALID_APP_INDEX;
 220       segment_manager_init_del (sm);
 221     }
 222
 223   /* If first segment manager is used by a listener */
 224   if (app->first_segment_manager != APP_INVALID_SEGMENT_MANAGER_INDEX
 225       && app->first_segment_manager != app->connects_seg_manager)
 226     {
 227       sm = segment_manager_get (app->first_segment_manager);
 228       /* .. and has no fifos, e.g. it might be used for redirected sessions,
 229        * remove it */
 230       if (!segment_manager_has_fifos (sm))
 231         {
 232           sm->app_index = SEGMENT_MANAGER_INVALID_APP_INDEX;
 233           segment_manager_del (sm);
 234         }
 235     }
 236
 237   /*
 238    * Local connections cleanup
 239    */
 240   application_local_sessions_del (app);
 241
 242   vec_free (app->tls_cert);
 243   vec_free (app->tls_key);
 244
 245   application_table_del (app);
 246   vec_free (app->name);
 247   pool_put (app_pool, app);
 248 }
 249
 250 static void
 251 application_verify_cb_fns (session_cb_vft_t * cb_fns)
 252 {
 253   if (cb_fns->session_accept_callback == 0)
 254     clib_warning ("No accept callback function provided");
 255   if (cb_fns->session_connected_callback == 0)
 256     clib_warning ("No session connected callback function provided");
 257   if (cb_fns->session_disconnect_callback == 0)
 258     clib_warning ("No session disconnect callback function provided");
 259   if (cb_fns->session_reset_callback == 0)
 260     clib_warning ("No session reset callback function provided");
 261 }
 262
 263 /**
 264  * Check app config for given segment type
 265  *
 266  * Returns 1 on success and 0 otherwise
 267  */
 268 static u8
 269 application_verify_cfg (ssvm_segment_type_t st)
 270 {
 271   u8 is_valid;
 272   if (st == SSVM_SEGMENT_MEMFD)
 273     {
 274       is_valid = (session_manager_get_evt_q_segment () != 0);
 275       if (!is_valid)
 276         clib_warning ("memfd seg: vpp's event qs IN binary api svm region");
 277       return is_valid;
 278     }
 279   else if (st == SSVM_SEGMENT_SHM)
 280     {
 281       is_valid = (session_manager_get_evt_q_segment () == 0);
 282       if (!is_valid)
 283         clib_warning ("shm seg: vpp's event qs NOT IN binary api svm region");
 284       return is_valid;
 285     }
 286   else
 287     return 1;
 288 }
 289
 290 int
 291 application_init (application_t * app, u32 api_client_index, u8 * app_name,
 292                   u64 * options, session_cb_vft_t * cb_fns)
 293 {
 294   ssvm_segment_type_t seg_type = SSVM_SEGMENT_MEMFD;
 295   u32 first_seg_size, prealloc_fifo_pairs;
 296   segment_manager_properties_t *props;
 297   vl_api_registration_t *reg;
 298   segment_manager_t *sm;
 299   int rv;
 300
 301   /*
 302    * Make sure we support the requested configuration
 303    */
 304
 305   if (!(options[APP_OPTIONS_FLAGS] & APP_OPTIONS_FLAGS_IS_BUILTIN))
 306     {
 307       reg = vl_api_client_index_to_registration (api_client_index);
 308       if (!reg)
 309         return VNET_API_ERROR_APP_UNSUPPORTED_CFG;
 310       if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI)
 311         seg_type = SSVM_SEGMENT_SHM;
 312     }
 313   else
 314     {
 315       seg_type = SSVM_SEGMENT_PRIVATE;
 316     }
 317
 318   if (!application_verify_cfg (seg_type))
 319     return VNET_API_ERROR_APP_UNSUPPORTED_CFG;
 320
 321   /*
 322    * Setup segment manager
 323    */
 324   sm = segment_manager_new ();
 325   sm->app_index = app->index;
 326   props = application_segment_manager_properties (app);
 327   segment_manager_properties_init (props);
 328   if (options[APP_OPTIONS_ADD_SEGMENT_SIZE])
 329     {
 330       props->add_segment_size = options[APP_OPTIONS_ADD_SEGMENT_SIZE];
 331       props->add_segment = 1;
 332     }
 333   if (options[APP_OPTIONS_RX_FIFO_SIZE])
 334     props->rx_fifo_size = options[APP_OPTIONS_RX_FIFO_SIZE];
 335   if (options[APP_OPTIONS_TX_FIFO_SIZE])
 336     props->tx_fifo_size = options[APP_OPTIONS_TX_FIFO_SIZE];
 337   if (options[APP_OPTIONS_EVT_QUEUE_SIZE])
 338     props->evt_q_size = options[APP_OPTIONS_EVT_QUEUE_SIZE];
 339   if (options[APP_OPTIONS_TLS_ENGINE])
 340     app->tls_engine = options[APP_OPTIONS_TLS_ENGINE];
 341   props->segment_type = seg_type;
 342
 343   first_seg_size = options[APP_OPTIONS_SEGMENT_SIZE];
 344   prealloc_fifo_pairs = options[APP_OPTIONS_PREALLOC_FIFO_PAIRS];
 345
 346   if ((rv = segment_manager_init (sm, first_seg_size, prealloc_fifo_pairs)))
 347     return rv;
 348   sm->first_is_protected = 1;
 349
 350   /*
 351    * Setup application
 352    */
 353   app->first_segment_manager = segment_manager_index (sm);
 354   app->api_client_index = api_client_index;
 355   app->flags = options[APP_OPTIONS_FLAGS];
 356   app->cb_fns = *cb_fns;
 357   app->ns_index = options[APP_OPTIONS_NAMESPACE];
 358   app->listeners_table = hash_create (0, sizeof (u64));
 359   app->local_connects = hash_create (0, sizeof (u64));
 360   app->proxied_transports = options[APP_OPTIONS_PROXY_TRANSPORT];
 361   app->event_queue = segment_manager_event_queue (sm);
 362   app->name = vec_dup (app_name);
 363
 364   /* If no scope enabled, default to global */
 365   if (!application_has_global_scope (app)
 366       && !application_has_local_scope (app))
 367     app->flags |= APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE;
 368
 369   /* Check that the obvious things are properly set up */
 370   application_verify_cb_fns (cb_fns);
 371
 372   /* Add app to lookup by api_client_index table */
 373   application_table_add (app);
 374
 375   /*
 376    * Segment manager for local sessions
 377    */
 378   sm = segment_manager_new ();
 379   sm->app_index = app->index;
 380   app->local_segment_manager = segment_manager_index (sm);
 381
 382   return 0;
 383 }
 384
 385 application_t *
 386 application_get (u32 index)
 387 {
 388   if (index == APP_INVALID_INDEX)
 389     return 0;
 390   return pool_elt_at_index (app_pool, index);
 391 }
 392
 393 application_t *
 394 application_get_if_valid (u32 index)
 395 {
 396   if (pool_is_free_index (app_pool, index))
 397     return 0;
 398
 399   return pool_elt_at_index (app_pool, index);
 400 }
 401
 402 u32
 403 application_get_index (application_t * app)
 404 {
 405   return app - app_pool;
 406 }
 407
 408 static segment_manager_t *
 409 application_alloc_segment_manager (application_t * app)
 410 {
 411   segment_manager_t *sm = 0;
 412
 413   /* If the first segment manager is not in use, don't allocate a new one */
 414   if (app->first_segment_manager != APP_INVALID_SEGMENT_MANAGER_INDEX
 415       && app->first_segment_manager_in_use == 0)
 416     {
 417       sm = segment_manager_get (app->first_segment_manager);
 418       app->first_segment_manager_in_use = 1;
 419       return sm;
 420     }
 421
 422   sm = segment_manager_new ();
 423   sm->app_index = app->index;
 424
 425   return sm;
 426 }
 427
 428 /**
 429  * Start listening local transport endpoint for requested transport.
 430  *
 431  * Creates a 'dummy' stream session with state LISTENING to be used in session
 432  * lookups, prior to establishing connection. Requests transport to build
 433  * it's own specific listening connection.
 434  */
 435 int
 436 application_start_listen (application_t * srv, session_endpoint_t * sep,
 437                           session_handle_t * res)
 438 {
 439   segment_manager_t *sm;
 440   stream_session_t *s;
 441   session_handle_t handle;
 442   session_type_t sst;
 443
 444   sst = session_type_from_proto_and_ip (sep->transport_proto, sep->is_ip4);
 445   s = listen_session_new (sst);
 446   s->app_index = srv->index;
 447
 448   if (stream_session_listen (s, sep))
 449     goto err;
 450
 451   /* Allocate segment manager. All sessions derived out of a listen session
 452    * have fifos allocated by the same segment manager. */
 453   sm = application_alloc_segment_manager (srv);
 454   if (sm == 0)
 455     goto err;
 456
 457   /* Add to app's listener table. Useful to find all child listeners
 458    * when app goes down, although, just for unbinding this is not needed */
 459   handle = listen_session_get_handle (s);
 460   hash_set (srv->listeners_table, handle, segment_manager_index (sm));
 461
 462   *res = handle;
 463   return 0;
 464
 465 err:
 466   listen_session_del (s);
 467   return -1;
 468 }
 469
 470 /**
 471  * Stop listening on session associated to handle
 472  */
 473 int
 474 application_stop_listen (application_t * srv, session_handle_t handle)
 475 {
 476   stream_session_t *listener;
 477   uword *indexp;
 478   segment_manager_t *sm;
 479
 480   if (srv && hash_get (srv->listeners_table, handle) == 0)
 481     {
 482       clib_warning ("app doesn't own handle %llu!", handle);
 483       return -1;
 484     }
 485
 486   listener = listen_session_get_from_handle (handle);
 487   stream_session_stop_listen (listener);
 488
 489   indexp = hash_get (srv->listeners_table, handle);
 490   ASSERT (indexp);
 491
 492   sm = segment_manager_get (*indexp);
 493   if (srv->first_segment_manager == *indexp)
 494     {
 495       /* Delete sessions but don't remove segment manager */
 496       srv->first_segment_manager_in_use = 0;
 497       segment_manager_del_sessions (sm);
 498     }
 499   else
 500     {
 501       segment_manager_init_del (sm);
 502     }
 503   hash_unset (srv->listeners_table, handle);
 504   listen_session_del (listener);
 505
 506   return 0;
 507 }
 508
 509 int
 510 application_open_session (application_t * app, session_endpoint_t * sep,
 511                           u32 api_context)
 512 {
 513   int rv;
 514
 515   /* Make sure we have a segment manager for connects */
 516   application_alloc_connects_segment_manager (app);
 517
 518   if ((rv = session_open (app->index, sep, api_context)))
 519     return rv;
 520
 521   return 0;
 522 }
 523
 524 int
 525 application_alloc_connects_segment_manager (application_t * app)
 526 {
 527   segment_manager_t *sm;
 528
 529   if (app->connects_seg_manager == APP_INVALID_SEGMENT_MANAGER_INDEX)
 530     {
 531       sm = application_alloc_segment_manager (app);
 532       if (sm == 0)
 533         return -1;
 534       app->connects_seg_manager = segment_manager_index (sm);
 535     }
 536   return 0;
 537 }
 538
 539 segment_manager_t *
 540 application_get_connect_segment_manager (application_t * app)
 541 {
 542   ASSERT (app->connects_seg_manager != (u32) ~ 0);
 543   return segment_manager_get (app->connects_seg_manager);
 544 }
 545
 546 segment_manager_t *
 547 application_get_listen_segment_manager (application_t * app,
 548                                         stream_session_t * s)
 549 {
 550   uword *smp;
 551   smp = hash_get (app->listeners_table, listen_session_get_handle (s));
 552   ASSERT (smp != 0);
 553   return segment_manager_get (*smp);
 554 }
 555
 556 segment_manager_t *
 557 application_get_local_segment_manager (application_t * app)
 558 {
 559   return segment_manager_get (app->local_segment_manager);
 560 }
 561
 562 segment_manager_t *
 563 application_get_local_segment_manager_w_session (application_t * app,
 564                                                  local_session_t * ls)
 565 {
 566   stream_session_t *listener;
 567   if (application_local_session_listener_has_transport (ls))
 568     {
 569       listener = listen_session_get (ls->listener_session_type,
 570                                      ls->listener_index);
 571       return application_get_listen_segment_manager (app, listener);
 572     }
 573   return segment_manager_get (app->local_segment_manager);
 574 }
 575
 576 int
 577 application_is_proxy (application_t * app)
 578 {
 579   return (app->flags & APP_OPTIONS_FLAGS_IS_PROXY);
 580 }
 581
 582 int
 583 application_is_builtin (application_t * app)
 584 {
 585   return (app->flags & APP_OPTIONS_FLAGS_IS_BUILTIN);
 586 }
 587
 588 int
 589 application_is_builtin_proxy (application_t * app)
 590 {
 591   return (application_is_proxy (app) && application_is_builtin (app));
 592 }
 593
 594 /**
 595  * Send an API message to the external app, to map new segment
 596  */
 597 int
 598 application_add_segment_notify (u32 app_index, ssvm_private_t * fs)
 599 {
 600   application_t *app = application_get (app_index);
 601   return app->cb_fns.add_segment_callback (app->api_client_index, fs);
 602 }
 603
 604 u8
 605 application_has_local_scope (application_t * app)
 606 {
 607   return app->flags & APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE;
 608 }
 609
 610 u8
 611 application_has_global_scope (application_t * app)
 612 {
 613   return app->flags & APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE;
 614 }
 615
 616 u32
 617 application_n_listeners (application_t * app)
 618 {
 619   return hash_elts (app->listeners_table);
 620 }
 621
 622 stream_session_t *
 623 application_first_listener (application_t * app, u8 fib_proto,
 624                             u8 transport_proto)
 625 {
 626   stream_session_t *listener;
 627   u64 handle;
 628   u32 sm_index;
 629   u8 sst;
 630
 631   sst = session_type_from_proto_and_ip (transport_proto,
 632                                         fib_proto == FIB_PROTOCOL_IP4);
 633
 634   /* *INDENT-OFF* */
 635    hash_foreach (handle, sm_index, app->listeners_table, ({
 636      listener = listen_session_get_from_handle (handle);
 637      if (listener->session_type == sst
 638          && listener->listener_index != SESSION_PROXY_LISTENER_INDEX)
 639        return listener;
 640    }));
 641   /* *INDENT-ON* */
 642
 643   return 0;
 644 }
 645
 646 stream_session_t *
 647 application_proxy_listener (application_t * app, u8 fib_proto,
 648                             u8 transport_proto)
 649 {
 650   stream_session_t *listener;
 651   u64 handle;
 652   u32 sm_index;
 653   u8 sst;
 654
 655   sst = session_type_from_proto_and_ip (transport_proto,
 656                                         fib_proto == FIB_PROTOCOL_IP4);
 657
 658   /* *INDENT-OFF* */
 659    hash_foreach (handle, sm_index, app->listeners_table, ({
 660      listener = listen_session_get_from_handle (handle);
 661      if (listener->session_type == sst
 662          && listener->listener_index == SESSION_PROXY_LISTENER_INDEX)
 663        return listener;
 664    }));
 665   /* *INDENT-ON* */
 666
 667   return 0;
 668 }
 669
 670 static clib_error_t *
 671 application_start_stop_proxy_fib_proto (application_t * app, u8 fib_proto,
 672                                         u8 transport_proto, u8 is_start)
 673 {
 674   app_namespace_t *app_ns = app_namespace_get (app->ns_index);
 675   u8 is_ip4 = (fib_proto == FIB_PROTOCOL_IP4);
 676   session_endpoint_t sep = SESSION_ENDPOINT_NULL;
 677   transport_connection_t *tc;
 678   stream_session_t *s;
 679   u64 handle;
 680
 681   if (is_start)
 682     {
 683       s = application_first_listener (app, fib_proto, transport_proto);
 684       if (!s)
 685         {
 686           sep.is_ip4 = is_ip4;
 687           sep.fib_index = app_namespace_get_fib_index (app_ns, fib_proto);
 688           sep.sw_if_index = app_ns->sw_if_index;
 689           sep.transport_proto = transport_proto;
 690           application_start_listen (app, &sep, &handle);
 691           s = listen_session_get_from_handle (handle);
 692           s->listener_index = SESSION_PROXY_LISTENER_INDEX;
 693         }
 694     }
 695   else
 696     {
 697       s = application_proxy_listener (app, fib_proto, transport_proto);
 698       ASSERT (s);
 699     }
 700
 701   tc = listen_session_get_transport (s);
 702
 703   if (!ip_is_zero (&tc->lcl_ip, 1))
 704     {
 705       u32 sti;
 706       sep.is_ip4 = is_ip4;
 707       sep.fib_index = app_namespace_get_fib_index (app_ns, fib_proto);
 708       sep.transport_proto = transport_proto;
 709       sep.port = 0;
 710       sti = session_lookup_get_index_for_fib (fib_proto, sep.fib_index);
 711       if (is_start)
 712         session_lookup_add_session_endpoint (sti, &sep, s->session_index);
 713       else
 714         session_lookup_del_session_endpoint (sti, &sep);
 715     }
 716
 717   return 0;
 718 }
 719
 720 static void
 721 application_start_stop_proxy_local_scope (application_t * app,
 722                                           u8 transport_proto, u8 is_start)
 723 {
 724   session_endpoint_t sep = SESSION_ENDPOINT_NULL;
 725   app_namespace_t *app_ns;
 726   app_ns = app_namespace_get (app->ns_index);
 727   sep.is_ip4 = 1;
 728   sep.transport_proto = transport_proto;
 729   sep.port = 0;
 730
 731   if (is_start)
 732     {
 733       session_lookup_add_session_endpoint (app_ns->local_table_index, &sep,
 734                                            app->index);
 735       sep.is_ip4 = 0;
 736       session_lookup_add_session_endpoint (app_ns->local_table_index, &sep,
 737                                            app->index);
 738     }
 739   else
 740     {
 741       session_lookup_del_session_endpoint (app_ns->local_table_index, &sep);
 742       sep.is_ip4 = 0;
 743       session_lookup_del_session_endpoint (app_ns->local_table_index, &sep);
 744     }
 745 }
 746
 747 void
 748 application_start_stop_proxy (application_t * app,
 749                               transport_proto_t transport_proto, u8 is_start)
 750 {
 751   if (application_has_local_scope (app))
 752     application_start_stop_proxy_local_scope (app, transport_proto, is_start);
 753
 754   if (application_has_global_scope (app))
 755     {
 756       application_start_stop_proxy_fib_proto (app, FIB_PROTOCOL_IP4,
 757                                               transport_proto, is_start);
 758       application_start_stop_proxy_fib_proto (app, FIB_PROTOCOL_IP6,
 759                                               transport_proto, is_start);
 760     }
 761 }
 762
 763 void
 764 application_setup_proxy (application_t * app)
 765 {
 766   u16 transports = app->proxied_transports;
 767   transport_proto_t tp;
 768
 769   ASSERT (application_is_proxy (app));
 770
 771   /* *INDENT-OFF* */
 772   transport_proto_foreach (tp, ({
 773     if (transports & (1 << tp))
 774       application_start_stop_proxy (app, tp, 1);
 775   }));
 776   /* *INDENT-ON* */
 777 }
 778
 779 void
 780 application_remove_proxy (application_t * app)
 781 {
 782   u16 transports = app->proxied_transports;
 783   transport_proto_t tp;
 784
 785   ASSERT (application_is_proxy (app));
 786
 787   /* *INDENT-OFF* */
 788   transport_proto_foreach (tp, ({
 789     if (transports & (1 << tp))
 790       application_start_stop_proxy (app, tp, 0);
 791   }));
 792   /* *INDENT-ON* */
 793 }
 794
 795 segment_manager_properties_t *
 796 application_segment_manager_properties (application_t * app)
 797 {
 798   return &app->sm_properties;
 799 }
 800
 801 segment_manager_properties_t *
 802 application_get_segment_manager_properties (u32 app_index)
 803 {
 804   application_t *app = application_get (app_index);
 805   return &app->sm_properties;
 806 }
 807
 808 local_session_t *
 809 application_alloc_local_session (application_t * app)
 810 {
 811   local_session_t *s;
 812   pool_get (app->local_sessions, s);
 813   memset (s, 0, sizeof (*s));
 814   s->app_index = app->index;
 815   s->session_index = s - app->local_sessions;
 816   s->session_type = session_type_from_proto_and_ip (TRANSPORT_PROTO_NONE, 0);
 817   return s;
 818 }
 819
 820 void
 821 application_free_local_session (application_t * app, local_session_t * s)
 822 {
 823   pool_put (app->local_sessions, s);
 824   if (CLIB_DEBUG)
 825     memset (s, 0xfc, sizeof (*s));
 826 }
 827
 828 local_session_t *
 829 application_get_local_session (application_t * app, u32 session_index)
 830 {
 831   return pool_elt_at_index (app->local_sessions, session_index);
 832 }
 833
 834 local_session_t *
 835 application_get_local_session_from_handle (session_handle_t handle)
 836 {
 837   application_t *server;
 838   u32 session_index, server_index;
 839   local_session_parse_handle (handle, &server_index, &session_index);
 840   server = application_get (server_index);
 841   return application_get_local_session (server, session_index);
 842 }
 843
 844 always_inline void
 845 application_local_listener_session_endpoint (local_session_t * ll,
 846                                              session_endpoint_t * sep)
 847 {
 848   sep->transport_proto =
 849     session_type_transport_proto (ll->listener_session_type);
 850   sep->port = ll->port;
 851   sep->is_ip4 = ll->listener_session_type & 1;
 852 }
 853
 854 int
 855 application_start_local_listen (application_t * server,
 856                                 session_endpoint_t * sep,
 857                                 session_handle_t * handle)
 858 {
 859   session_handle_t lh;
 860   local_session_t *ll;
 861   u32 table_index;
 862
 863   table_index = application_local_session_table (server);
 864
 865   /* An exact sep match, as opposed to session_lookup_local_listener */
 866   lh = session_lookup_endpoint_listener (table_index, sep, 1);
 867   if (lh != SESSION_INVALID_HANDLE)
 868     return VNET_API_ERROR_ADDRESS_IN_USE;
 869
 870   pool_get (server->local_listen_sessions, ll);
 871   memset (ll, 0, sizeof (*ll));
 872   ll->session_type = session_type_from_proto_and_ip (TRANSPORT_PROTO_NONE, 0);
 873   ll->app_index = server->index;
 874   ll->session_index = ll - server->local_listen_sessions;
 875   ll->port = sep->port;
 876   /* Store the original session type for the unbind */
 877   ll->listener_session_type =
 878     session_type_from_proto_and_ip (sep->transport_proto, sep->is_ip4);
 879   ll->transport_listener_index = ~0;
 880
 881   *handle = application_local_session_handle (ll);
 882   session_lookup_add_session_endpoint (table_index, sep, *handle);
 883
 884   return 0;
 885 }
 886
 887 /**
 888  * Clean up local session table. If we have a listener session use it to
 889  * find the port and proto. If not, the handle must be a local table handle
 890  * so parse it.
 891  */
 892 int
 893 application_stop_local_listen (application_t * server, session_handle_t lh)
 894 {
 895   session_endpoint_t sep = SESSION_ENDPOINT_NULL;
 896   u32 table_index, ll_index, server_index;
 897   stream_session_t *sl = 0;
 898   local_session_t *ll, *ls;
 899
 900   table_index = application_local_session_table (server);
 901
 902   /* We have both local and global table binds. Figure from global what
 903    * the sep we should be cleaning up is.
 904    */
 905   if (!session_handle_is_local (lh))
 906     {
 907       sl = listen_session_get_from_handle (lh);
 908       if (!sl || listen_session_get_local_session_endpoint (sl, &sep))
 909         {
 910           clib_warning ("broken listener");
 911           return -1;
 912         }
 913       lh = session_lookup_endpoint_listener (table_index, &sep, 0);
 914       if (lh == SESSION_INVALID_HANDLE)
 915         return -1;
 916     }
 917
 918   local_session_parse_handle (lh, &server_index, &ll_index);
 919   ASSERT (server->index == server_index);
 920   if (!(ll = application_get_local_listen_session (server, ll_index)))
 921     {
 922       clib_warning ("no local listener");
 923       return -1;
 924     }
 925   application_local_listener_session_endpoint (ll, &sep);
 926   session_lookup_del_session_endpoint (table_index, &sep);
 927
 928   /* *INDENT-OFF* */
 929   pool_foreach (ls, server->local_sessions, ({
 930     if (ls->listener_index == ll->session_index)
 931       application_local_session_disconnect (server->index, ls);
 932   }));
 933   /* *INDENT-ON* */
 934   pool_put_index (server->local_listen_sessions, ll->session_index);
 935
 936   return 0;
 937 }
 938
 939 int
 940 application_local_session_connect (u32 table_index, application_t * client,
 941                                    application_t * server,
 942                                    local_session_t * ll, u32 opaque)
 943 {
 944   u32 seg_size, evt_q_sz, evt_q_elts, margin = 16 << 10;
 945   segment_manager_properties_t *props, *cprops;
 946   int rv, has_transport, seg_index;
 947   svm_fifo_segment_private_t *seg;
 948   segment_manager_t *sm;
 949   local_session_t *ls;
 950   svm_queue_t *sq, *cq;
 951
 952   ls = application_alloc_local_session (server);
 953
 954   props = application_segment_manager_properties (server);
 955   cprops = application_segment_manager_properties (client);
 956   evt_q_elts = props->evt_q_size + cprops->evt_q_size;
 957   evt_q_sz = evt_q_elts * sizeof (session_fifo_event_t);
 958   seg_size = props->rx_fifo_size + props->tx_fifo_size + evt_q_sz + margin;
 959
 960   has_transport = session_has_transport ((stream_session_t *) ll);
 961   if (!has_transport)
 962     {
 963       /* Local sessions don't have backing transport */
 964       ls->port = ll->port;
 965       sm = application_get_local_segment_manager (server);
 966     }
 967   else
 968     {
 969       stream_session_t *sl = (stream_session_t *) ll;
 970       transport_connection_t *tc;
 971       tc = listen_session_get_transport (sl);
 972       ls->port = tc->lcl_port;
 973       sm = application_get_listen_segment_manager (server, sl);
 974     }
 975
 976   seg_index = segment_manager_add_segment (sm, seg_size);
 977   if (seg_index < 0)
 978     {
 979       clib_warning ("failed to add new cut-through segment");
 980       return seg_index;
 981     }
 982   seg = segment_manager_get_segment_w_lock (sm, seg_index);
 983   sq = segment_manager_alloc_queue (seg, props->evt_q_size);
 984   cq = segment_manager_alloc_queue (seg, cprops->evt_q_size);
 985   ls->server_evt_q = pointer_to_uword (sq);
 986   ls->client_evt_q = pointer_to_uword (cq);
 987   rv = segment_manager_try_alloc_fifos (seg, props->rx_fifo_size,
 988                                         props->tx_fifo_size,
 989                                         &ls->server_rx_fifo,
 990                                         &ls->server_tx_fifo);
 991   if (rv)
 992     {
 993       clib_warning ("failed to add fifos in cut-through segment");
 994       segment_manager_segment_reader_unlock (sm);
 995       goto failed;
 996     }
 997   ls->server_rx_fifo->master_session_index = ls->session_index;
 998   ls->server_tx_fifo->master_session_index = ls->session_index;
 999   ls->server_rx_fifo->master_thread_index = ~0;
1000   ls->server_tx_fifo->master_thread_index = ~0;
1001   ls->svm_segment_index = seg_index;
1002   ls->listener_index = ll->session_index;
1003   ls->client_index = client->index;
1004   ls->client_opaque = opaque;
1005   ls->listener_session_type = ll->session_type;
1006
1007   if ((rv = server->cb_fns.add_segment_callback (server->api_client_index,
1008                                                  &seg->ssvm)))
1009     {
1010       clib_warning ("failed to notify server of new segment");
1011       segment_manager_segment_reader_unlock (sm);
1012       goto failed;
1013     }
1014   segment_manager_segment_reader_unlock (sm);
1015   if ((rv = server->cb_fns.session_accept_callback ((stream_session_t *) ls)))
1016     {
1017       clib_warning ("failed to send accept cut-through notify to server");
1018       goto failed;
1019     }
1020   if (server->flags & APP_OPTIONS_FLAGS_IS_BUILTIN)
1021     application_local_session_connect_notify (ls);
1022
1023   return 0;
1024
1025 failed:
1026   if (!has_transport)
1027     segment_manager_del_segment (sm, seg);
1028   return rv;
1029 }
1030
1031 static uword
1032 application_client_local_connect_key (local_session_t * ls)
1033 {
1034   return ((uword) ls->app_index << 32 | (uword) ls->session_index);
1035 }
1036
1037 static void
1038 application_client_local_connect_key_parse (uword key, u32 * app_index,
1039                                             u32 * session_index)
1040 {
1041   *app_index = key >> 32;
1042   *session_index = key & 0xFFFFFFFF;
1043 }
1044
1045 int
1046 application_local_session_connect_notify (local_session_t * ls)
1047 {
1048   svm_fifo_segment_private_t *seg;
1049   application_t *client, *server;
1050   segment_manager_t *sm;
1051   int rv, is_fail = 0;
1052   uword client_key;
1053
1054   client = application_get (ls->client_index);
1055   server = application_get (ls->app_index);
1056   sm = application_get_local_segment_manager_w_session (server, ls);
1057   seg = segment_manager_get_segment_w_lock (sm, ls->svm_segment_index);
1058   if ((rv = client->cb_fns.add_segment_callback (client->api_client_index,
1059                                                  &seg->ssvm)))
1060     {
1061       clib_warning ("failed to notify client %u of new segment",
1062                     ls->client_index);
1063       segment_manager_segment_reader_unlock (sm);
1064       application_local_session_disconnect (ls->client_index, ls);
1065       is_fail = 1;
1066     }
1067   else
1068     {
1069       segment_manager_segment_reader_unlock (sm);
1070     }
1071
1072   client->cb_fns.session_connected_callback (client->index, ls->client_opaque,
1073                                              (stream_session_t *) ls,
1074                                              is_fail);
1075
1076   client_key = application_client_local_connect_key (ls);
1077   hash_set (client->local_connects, client_key, client_key);
1078   return 0;
1079 }
1080
1081 int
1082 application_local_session_disconnect (u32 app_index, local_session_t * ls)
1083 {
1084   svm_fifo_segment_private_t *seg;
1085   application_t *client, *server;
1086   segment_manager_t *sm;
1087   uword client_key;
1088
1089   client = application_get_if_valid (ls->client_index);
1090   server = application_get (ls->app_index);
1091
1092   if (ls->session_state == SESSION_STATE_CLOSED)
1093     {
1094     cleanup:
1095       client_key = application_client_local_connect_key (ls);
1096       sm = application_get_local_segment_manager_w_session (server, ls);
1097       seg = segment_manager_get_segment (sm, ls->svm_segment_index);
1098
1099       if (client)
1100         {
1101           hash_unset (client->local_connects, client_key);
1102           client->cb_fns.del_segment_callback (client->api_client_index,
1103                                                &seg->ssvm);
1104         }
1105
1106       server->cb_fns.del_segment_callback (server->api_client_index,
1107                                            &seg->ssvm);
1108       segment_manager_del_segment (sm, seg);
1109       application_free_local_session (server, ls);
1110       return 0;
1111     }
1112
1113   if (app_index == ls->client_index)
1114     {
1115       send_local_session_disconnect_callback (ls->app_index, ls);
1116     }
1117   else
1118     {
1119       if (!client)
1120         {
1121           goto cleanup;
1122         }
1123       else if (ls->session_state < SESSION_STATE_READY)
1124         {
1125           client->cb_fns.session_connected_callback (client->index,
1126                                                      ls->client_opaque,
1127                                                      (stream_session_t *) ls,
1128                                                      1 /* is_fail */ );
1129           ls->session_state = SESSION_STATE_CLOSED;
1130           goto cleanup;
1131         }
1132       else
1133         {
1134           send_local_session_disconnect_callback (ls->client_index, ls);
1135         }
1136     }
1137
1138   ls->session_state = SESSION_STATE_CLOSED;
1139
1140   return 0;
1141 }
1142
1143 void
1144 application_local_sessions_del (application_t * app)
1145 {
1146   u32 index, server_index, session_index, table_index;
1147   segment_manager_t *sm;
1148   u64 handle, *handles = 0;
1149   local_session_t *ls, *ll;
1150   application_t *server;
1151   session_endpoint_t sep;
1152   int i;
1153
1154   /*
1155    * Local listens. Don't bother with local sessions, we clean them lower
1156    */
1157   table_index = application_local_session_table (app);
1158   /* *INDENT-OFF* */
1159   pool_foreach (ll, app->local_listen_sessions, ({
1160     application_local_listener_session_endpoint (ll, &sep);
1161     session_lookup_del_session_endpoint (table_index, &sep);
1162   }));
1163   /* *INDENT-ON* */
1164
1165   /*
1166    * Local sessions
1167    */
1168   if (app->local_sessions)
1169     {
1170       /* *INDENT-OFF* */
1171       pool_foreach (ls, app->local_sessions, ({
1172         application_local_session_disconnect (app->index, ls);
1173       }));
1174       /* *INDENT-ON* */
1175     }
1176
1177   /*
1178    * Local connects
1179    */
1180   vec_reset_length (handles);
1181   /* *INDENT-OFF* */
1182   hash_foreach (handle, index, app->local_connects, ({
1183     vec_add1 (handles, handle);
1184   }));
1185   /* *INDENT-ON* */
1186
1187   for (i = 0; i < vec_len (handles); i++)
1188     {
1189       application_client_local_connect_key_parse (handles[i], &server_index,
1190                                                   &session_index);
1191       server = application_get_if_valid (server_index);
1192       if (server)
1193         {
1194           ls = application_get_local_session (server, session_index);
1195           application_local_session_disconnect (app->index, ls);
1196         }
1197     }
1198
1199   sm = segment_manager_get (app->local_segment_manager);
1200   sm->app_index = SEGMENT_MANAGER_INVALID_APP_INDEX;
1201   segment_manager_del (sm);
1202 }
1203
1204 clib_error_t *
1205 vnet_app_add_tls_cert (vnet_app_add_tls_cert_args_t * a)
1206 {
1207   application_t *app;
1208   app = application_get (a->app_index);
1209   if (!app)
1210     return clib_error_return_code (0, VNET_API_ERROR_APPLICATION_NOT_ATTACHED,
1211                                    0, "app %u doesn't exist", a->app_index);
1212   app->tls_cert = vec_dup (a->cert);
1213   return 0;
1214 }
1215
1216 clib_error_t *
1217 vnet_app_add_tls_key (vnet_app_add_tls_key_args_t * a)
1218 {
1219   application_t *app;
1220   app = application_get (a->app_index);
1221   if (!app)
1222     return clib_error_return_code (0, VNET_API_ERROR_APPLICATION_NOT_ATTACHED,
1223                                    0, "app %u doesn't exist", a->app_index);
1224   app->tls_key = vec_dup (a->key);
1225   return 0;
1226 }
1227
1228 u8 *
1229 format_application_listener (u8 * s, va_list * args)
1230 {
1231   application_t *app = va_arg (*args, application_t *);
1232   u64 handle = va_arg (*args, u64);
1233   u32 sm_index = va_arg (*args, u32);
1234   int verbose = va_arg (*args, int);
1235   stream_session_t *listener;
1236   u8 *app_name, *str;
1237
1238   if (app == 0)
1239     {
1240       if (verbose)
1241         s = format (s, "%-40s%-20s%-15s%-15s%-10s", "Connection", "App",
1242                     "API Client", "ListenerID", "SegManager");
1243       else
1244         s = format (s, "%-40s%-20s", "Connection", "App");
1245
1246       return s;
1247     }
1248
1249   app_name = app_get_name_from_reg_index (app);
1250   listener = listen_session_get_from_handle (handle);
1251   str = format (0, "%U", format_stream_session, listener, verbose);
1252
1253   if (verbose)
1254     {
1255       s = format (s, "%-40s%-20s%-15u%-15u%-10u", str, app_name,
1256                   app->api_client_index, handle, sm_index);
1257     }
1258   else
1259     s = format (s, "%-40s%-20s", str, app_name);
1260
1261   vec_free (app_name);
1262   return s;
1263 }
1264
1265 void
1266 application_format_connects (application_t * app, int verbose)
1267 {
1268   svm_fifo_segment_private_t *fifo_segment;
1269   vlib_main_t *vm = vlib_get_main ();
1270   segment_manager_t *sm;
1271   u8 *app_name, *s = 0;
1272
1273   /* Header */
1274   if (app == 0)
1275     {
1276       if (verbose)
1277         vlib_cli_output (vm, "%-40s%-20s%-15s%-10s", "Connection", "App",
1278                          "API Client", "SegManager");
1279       else
1280         vlib_cli_output (vm, "%-40s%-20s", "Connection", "App");
1281       return;
1282     }
1283
1284   /* make sure */
1285   if (app->connects_seg_manager == (u32) ~ 0)
1286     return;
1287
1288   app_name = app_get_name_from_reg_index (app);
1289
1290   /* Across all fifo segments */
1291   sm = segment_manager_get (app->connects_seg_manager);
1292
1293   /* *INDENT-OFF* */
1294   segment_manager_foreach_segment_w_lock (fifo_segment, sm, ({
1295     svm_fifo_t *fifo;
1296     u8 *str;
1297
1298     fifo = svm_fifo_segment_get_fifo_list (fifo_segment);
1299     while (fifo)
1300         {
1301           u32 session_index, thread_index;
1302           stream_session_t *session;
1303
1304           session_index = fifo->master_session_index;
1305           thread_index = fifo->master_thread_index;
1306
1307           session = session_get (session_index, thread_index);
1308           str = format (0, "%U", format_stream_session, session, verbose);
1309
1310           if (verbose)
1311             s = format (s, "%-40s%-20s%-15u%-10u", str, app_name,
1312                         app->api_client_index, app->connects_seg_manager);
1313           else
1314             s = format (s, "%-40s%-20s", str, app_name);
1315
1316           vlib_cli_output (vm, "%v", s);
1317           vec_reset_length (s);
1318           vec_free (str);
1319
1320           fifo = fifo->next;
1321         }
1322     vec_free (s);
1323   }));
1324   /* *INDENT-ON* */
1325
1326   vec_free (app_name);
1327 }
1328
1329 void
1330 application_format_local_sessions (application_t * app, int verbose)
1331 {
1332   vlib_main_t *vm = vlib_get_main ();
1333   local_session_t *ls;
1334   transport_proto_t tp;
1335   u8 *conn = 0;
1336
1337   /* Header */
1338   if (app == 0)
1339     {
1340       vlib_cli_output (vm, "%-40s%-15s%-20s", "Connection", "ServerApp",
1341                        "ClientApp");
1342       return;
1343     }
1344
1345   /* *INDENT-OFF* */
1346   pool_foreach (ls, app->local_listen_sessions, ({
1347     tp = session_type_transport_proto(ls->listener_session_type);
1348     conn = format (0, "[L][%U] *:%u", format_transport_proto_short, tp,
1349                    ls->port);
1350     vlib_cli_output (vm, "%-40v%-15u%-20s", conn, ls->app_index, "*");
1351     vec_reset_length (conn);
1352   }));
1353   pool_foreach (ls, app->local_sessions, ({
1354     tp = session_type_transport_proto(ls->listener_session_type);
1355     conn = format (0, "[L][%U] *:%u", format_transport_proto_short, tp,
1356                    ls->port);
1357     vlib_cli_output (vm, "%-40v%-15u%-20u", conn, ls->app_index,
1358                      ls->client_index);
1359     vec_reset_length (conn);
1360   }));
1361   /* *INDENT-ON* */
1362
1363   vec_free (conn);
1364 }
1365
1366 void
1367 application_format_local_connects (application_t * app, int verbose)
1368 {
1369   vlib_main_t *vm = vlib_get_main ();
1370   u32 app_index, session_index;
1371   application_t *server;
1372   local_session_t *ls;
1373   uword client_key;
1374   u64 value;
1375
1376   /* Header */
1377   if (app == 0)
1378     {
1379       if (verbose)
1380         vlib_cli_output (vm, "%-40s%-15s%-20s%-10s", "Connection", "App",
1381                          "Peer App", "SegManager");
1382       else
1383         vlib_cli_output (vm, "%-40s%-15s%-20s", "Connection", "App",
1384                          "Peer App");
1385       return;
1386     }
1387
1388   /* *INDENT-OFF* */
1389   hash_foreach (client_key, value, app->local_connects, ({
1390     application_client_local_connect_key_parse (client_key, &app_index,
1391                                                 &session_index);
1392     server = application_get (app_index);
1393     ls = application_get_local_session (server, session_index);
1394     vlib_cli_output (vm, "%-40s%-15s%-20s", "TODO", ls->app_index, ls->client_index);
1395   }));
1396   /* *INDENT-ON* */
1397 }
1398
1399 u8 *
1400 format_application (u8 * s, va_list * args)
1401 {
1402   application_t *app = va_arg (*args, application_t *);
1403   CLIB_UNUSED (int verbose) = va_arg (*args, int);
1404   segment_manager_properties_t *props;
1405   const u8 *app_ns_name;
1406   u8 *app_name;
1407
1408   if (app == 0)
1409     {
1410       if (verbose)
1411         s = format (s, "%-10s%-20s%-15s%-15s%-15s%-15s%-15s", "Index", "Name",
1412                     "API Client", "Namespace", "Add seg size", "Rx fifo size",
1413                     "Tx fifo size");
1414       else
1415         s = format (s, "%-10s%-20s%-15s%-40s", "Index", "Name", "API Client",
1416                     "Namespace");
1417       return s;
1418     }
1419
1420   app_name = app_get_name (app);
1421   app_ns_name = app_namespace_id_from_index (app->ns_index);
1422   props = application_segment_manager_properties (app);
1423   if (verbose)
1424     s = format (s, "%-10d%-20s%-15d%-15d%-15d%-15d%-15d", app->index,
1425                 app_name, app->api_client_index, app->ns_index,
1426                 props->add_segment_size, props->rx_fifo_size,
1427                 props->tx_fifo_size);
1428   else
1429     s = format (s, "%-10d%-20s%-15d%-40s", app->index, app_name,
1430                 app->api_client_index, app_ns_name);
1431   return s;
1432 }
1433
1434
1435 void
1436 application_format_all_listeners (vlib_main_t * vm, int do_local, int verbose)
1437 {
1438   application_t *app;
1439   u32 sm_index;
1440   u64 handle;
1441
1442   if (!pool_elts (app_pool))
1443     {
1444       vlib_cli_output (vm, "No active server bindings");
1445       return;
1446     }
1447
1448   if (do_local)
1449     {
1450       application_format_local_sessions (0, verbose);
1451       /* *INDENT-OFF* */
1452       pool_foreach (app, app_pool, ({
1453         if (!pool_elts (app->local_sessions)
1454             && !pool_elts(app->local_connects))
1455           continue;
1456         application_format_local_sessions (app, verbose);
1457       }));
1458       /* *INDENT-ON* */
1459     }
1460   else
1461     {
1462       vlib_cli_output (vm, "%U", format_application_listener, 0 /* header */ ,
1463                        0, 0, verbose);
1464
1465       /* *INDENT-OFF* */
1466       pool_foreach (app, app_pool, ({
1467         if (hash_elts (app->listeners_table) == 0)
1468           continue;
1469         hash_foreach (handle, sm_index, app->listeners_table, ({
1470           vlib_cli_output (vm, "%U", format_application_listener, app,
1471                            handle, sm_index, verbose);
1472         }));
1473       }));
1474       /* *INDENT-ON* */
1475     }
1476 }
1477
1478 void
1479 application_format_all_clients (vlib_main_t * vm, int do_local, int verbose)
1480 {
1481   application_t *app;
1482
1483   if (!pool_elts (app_pool))
1484     {
1485       vlib_cli_output (vm, "No active apps");
1486       return;
1487     }
1488
1489   if (do_local)
1490     {
1491       application_format_local_connects (0, verbose);
1492
1493       /* *INDENT-OFF* */
1494       pool_foreach (app, app_pool, ({
1495         if (app->local_connects)
1496           application_format_local_connects (app, verbose);
1497       }));
1498       /* *INDENT-ON* */
1499     }
1500   else
1501     {
1502       application_format_connects (0, verbose);
1503
1504       /* *INDENT-OFF* */
1505       pool_foreach (app, app_pool, ({
1506         if (app->connects_seg_manager == (u32)~0)
1507           continue;
1508         application_format_connects (app, verbose);
1509       }));
1510       /* *INDENT-ON* */
1511     }
1512 }
1513
1514 static clib_error_t *
1515 show_app_command_fn (vlib_main_t * vm, unformat_input_t * input,
1516                      vlib_cli_command_t * cmd)
1517 {
1518   int do_server = 0, do_client = 0, do_local = 0;
1519   application_t *app;
1520   int verbose = 0;
1521
1522   session_cli_return_if_not_enabled ();
1523
1524   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1525     {
1526       if (unformat (input, "server"))
1527         do_server = 1;
1528       else if (unformat (input, "client"))
1529         do_client = 1;
1530       else if (unformat (input, "local"))
1531         do_local = 1;
1532       else if (unformat (input, "verbose"))
1533         verbose = 1;
1534       else
1535         break;
1536     }
1537
1538   if (do_server)
1539     application_format_all_listeners (vm, do_local, verbose);
1540
1541   if (do_client)
1542     application_format_all_clients (vm, do_local, verbose);
1543
1544   /* Print app related info */
1545   if (!do_server && !do_client)
1546     {
1547       vlib_cli_output (vm, "%U", format_application, 0, verbose);
1548       /* *INDENT-OFF* */
1549       pool_foreach (app, app_pool, ({
1550         vlib_cli_output (vm, "%U", format_application, app, verbose);
1551       }));
1552       /* *INDENT-ON* */
1553     }
1554
1555   return 0;
1556 }
1557
1558 /* *INDENT-OFF* */
1559 VLIB_CLI_COMMAND (show_app_command, static) =
1560 {
1561   .path = "show app",
1562   .short_help = "show app [server|client] [verbose]",
1563   .function = show_app_command_fn,
1564 };
1565 /* *INDENT-ON* */
1566
1567 /*
1568  * fd.io coding-style-patch-verification: ON
1569  *
1570  * Local Variables:
1571  * eval: (c-set-style "gnu")
1572  * End:
1573  */