Horizontal (nSessions) scaling draft
[vpp.git] / src / vnet / session / session_cli.c
old mode 100644 (file)
new mode 100755 (executable)
index b029ee6..e06bc58
 #include <vnet/session/application.h>
 #include <vnet/session/session.h>
 
+u8 *
+format_stream_session_fifos (u8 * s, va_list * args)
+{
+  stream_session_t *ss = va_arg (*args, stream_session_t *);
+  s = format (s, " Rx fifo: %U", format_svm_fifo, ss->server_rx_fifo, 1);
+  s = format (s, " Tx fifo: %U", format_svm_fifo, ss->server_tx_fifo, 1);
+  return s;
+}
+
 /**
  * Format stream session as per the following format
  *
@@ -30,33 +39,43 @@ format_stream_session (u8 * s, va_list * args)
   int verbose = va_arg (*args, int);
   transport_proto_vft_t *tp_vft;
   u8 *str = 0;
-
   tp_vft = session_get_transport_vft (ss->session_type);
 
-  if (verbose)
-    str = format (0, "%-20llp%-20llp%-15lld", ss->server_rx_fifo,
-                 ss->server_tx_fifo, stream_session_get_index (ss));
+  if (verbose == 1)
+    str = format (0, "%-10u%-10u%-10lld",
+                 svm_fifo_max_dequeue (ss->server_rx_fifo),
+                 svm_fifo_max_enqueue (ss->server_tx_fifo),
+                 stream_session_get_index (ss));
 
-  if (ss->session_state == SESSION_STATE_READY)
+  if (ss->session_state == SESSION_STATE_READY
+      || ss->session_state == SESSION_STATE_ACCEPTING)
     {
-      s = format (s, "%-40U%v", tp_vft->format_connection,
-                 ss->connection_index, ss->thread_index, str);
+      s = format (s, "%U", tp_vft->format_connection, ss->connection_index,
+                 ss->thread_index, verbose);
+      if (verbose == 1)
+       s = format (s, "%v", str);
+      if (verbose > 1)
+       s = format (s, "%U", format_stream_session_fifos, ss);
     }
   else if (ss->session_state == SESSION_STATE_LISTENING)
     {
       s = format (s, "%-40U%v", tp_vft->format_listener, ss->connection_index,
                  str);
     }
-  else if (ss->session_state == SESSION_STATE_READY)
+  else if (ss->session_state == SESSION_STATE_CONNECTING)
     {
-      s =
-       format (s, "%-40U%v", tp_vft->format_half_open, ss->connection_index,
-               str);
+      s = format (s, "%-40U%v", tp_vft->format_half_open,
+                 ss->connection_index, str);
     }
   else if (ss->session_state == SESSION_STATE_CLOSED)
     {
-      s = format (s, "[CL] %-40U%v", tp_vft->format_connection,
-                 ss->connection_index, ss->thread_index, str);
+      s =
+       format (s, "[CL] %U", tp_vft->format_connection, ss->connection_index,
+               ss->thread_index, verbose);
+      if (verbose == 1)
+       s = format (s, "%v", str);
+      if (verbose > 1)
+       s = format (s, "%U", format_stream_session_fifos, ss);
     }
   else
     {
@@ -76,21 +95,62 @@ show_session_command_fn (vlib_main_t * vm, unformat_input_t * input,
   int verbose = 0, i;
   stream_session_t *pool;
   stream_session_t *s;
-  u8 *str = 0;
+  u8 *str = 0, one_session = 0, proto_set = 0, proto = 0;
+  u8 is_ip4 = 0, s_type = 0;
+  ip4_address_t lcl_ip4, rmt_ip4;
+  u32 lcl_port = 0, rmt_port = 0;
+
+  memset (&lcl_ip4, 0, sizeof (lcl_ip4));
+  memset (&rmt_ip4, 0, sizeof (rmt_ip4));
 
   if (!smm->is_enabled)
     {
-      clib_error_return (0, "session layer is not enabled");
+      return clib_error_return (0, "session layer is not enabled");
     }
 
   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
     {
-      if (unformat (input, "verbose"))
+      if (unformat (input, "verbose %d", &verbose))
+       ;
+      else if (unformat (input, "verbose"))
        verbose = 1;
+      else if (unformat (input, "tcp"))
+       {
+         proto_set = 1;
+         proto = TRANSPORT_PROTO_TCP;
+       }
+      else if (unformat (input, "%U:%d->%U:%d",
+                        unformat_ip4_address, &lcl_ip4, &lcl_port,
+                        unformat_ip4_address, &rmt_ip4, &rmt_port))
+       {
+         one_session = 1;
+         is_ip4 = 1;
+       }
+
       else
        break;
     }
 
+  if (one_session)
+    {
+      if (!proto_set)
+       {
+         vlib_cli_output (vm, "proto not set");
+         return clib_error_return (0, "proto not set");
+       }
+
+      s_type = session_type_from_proto_and_ip (proto, is_ip4);
+      s = stream_session_lookup4 (&lcl_ip4, &rmt_ip4,
+                                 clib_host_to_net_u16 (lcl_port),
+                                 clib_host_to_net_u16 (rmt_port), s_type);
+      if (s)
+       vlib_cli_output (vm, "%U", format_stream_session, s, 2);
+      else
+       vlib_cli_output (vm, "session does not exist");
+
+      return 0;
+    }
+
   for (i = 0; i < vec_len (smm->sessions); i++)
     {
       u32 once_per_pool;
@@ -105,11 +165,11 @@ show_session_command_fn (vlib_main_t * vm, unformat_input_t * input,
                           i, pool_elts (pool));
          if (verbose)
            {
-             if (once_per_pool)
+             if (once_per_pool && verbose == 1)
                {
-                 str = format (str, "%-40s%-20s%-20s%-15s",
-                               "Connection", "Rx fifo", "Tx fifo",
-                               "Session Index");
+                 str =
+                   format (str, "%-50s%-15s%-10s%-10s%-10s", "Connection",
+                           "State", "Rx-f", "Tx-f", "S-idx");
                  vlib_cli_output (vm, "%v", str);
                  vec_reset_length (str);
                  once_per_pool = 0;
@@ -118,13 +178,16 @@ show_session_command_fn (vlib_main_t * vm, unformat_input_t * input,
               /* *INDENT-OFF* */
               pool_foreach (s, pool,
               ({
-                vlib_cli_output (vm, "%U", format_stream_session, s, verbose);
+               vec_reset_length (str);
+                str = format (str, "%U", format_stream_session, s, verbose);
+                vlib_cli_output (vm, "%v", str);
               }));
               /* *INDENT-ON* */
            }
        }
       else
        vlib_cli_output (vm, "Thread %d: no active sessions", i);
+      vec_reset_length (str);
     }
   vec_free (str);
 
@@ -140,19 +203,26 @@ VLIB_CLI_COMMAND (show_session_command, static) =
 };
 /* *INDENT-ON* */
 
+static int
+clear_session (stream_session_t * s)
+{
+  application_t *server = application_get (s->app_index);
+  server->cb_fns.session_disconnect_callback (s);
+  return 0;
+}
+
 static clib_error_t *
 clear_session_command_fn (vlib_main_t * vm, unformat_input_t * input,
                          vlib_cli_command_t * cmd)
 {
   session_manager_main_t *smm = &session_manager_main;
-  u32 thread_index = 0;
+  u32 thread_index = 0, clear_all = 0;
   u32 session_index = ~0;
-  stream_session_t *pool, *session;
-  application_t *server;
+  stream_session_t **pool, *session;
 
   if (!smm->is_enabled)
     {
-      clib_error_return (0, "session layer is not enabled");
+      return clib_error_return (0, "session layer is not enabled");
     }
 
   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
@@ -161,28 +231,36 @@ clear_session_command_fn (vlib_main_t * vm, unformat_input_t * input,
        ;
       else if (unformat (input, "session %d", &session_index))
        ;
+      else if (unformat (input, "all"))
+       clear_all = 1;
       else
        return clib_error_return (0, "unknown input `%U'",
                                  format_unformat_error, input);
     }
 
-  if (session_index == ~0)
+  if (!clear_all && session_index == ~0)
     return clib_error_return (0, "session <nn> required, but not set.");
 
-  if (thread_index > vec_len (smm->sessions))
-    return clib_error_return (0, "thread %d out of range [0-%d]",
-                             thread_index, vec_len (smm->sessions));
-
-  pool = smm->sessions[thread_index];
-
-  if (pool_is_free_index (pool, session_index))
-    return clib_error_return (0, "session %d not active", session_index);
-
-  session = pool_elt_at_index (pool, session_index);
-  server = application_get (session->app_index);
+  if (session_index != ~0)
+    {
+      session = stream_session_get_if_valid (session_index, thread_index);
+      if (!session)
+       return clib_error_return (0, "no session %d on thread %d",
+                                 session_index, thread_index);
+      clear_session (session);
+    }
 
-  /* Disconnect both app and transport */
-  server->cb_fns.session_disconnect_callback (session);
+  if (clear_all)
+    {
+      /* *INDENT-OFF* */
+      vec_foreach (pool, smm->sessions)
+       {
+         pool_foreach(session, *pool, ({
+           clear_session (session);
+         }));
+       };
+      /* *INDENT-ON* */
+    }
 
   return 0;
 }