aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/hs_apps/http_tps.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/hs_apps/http_tps.c')
-rw-r--r--src/plugins/hs_apps/http_tps.c179
1 files changed, 154 insertions, 25 deletions
diff --git a/src/plugins/hs_apps/http_tps.c b/src/plugins/hs_apps/http_tps.c
index e149335903a..4407427e652 100644
--- a/src/plugins/hs_apps/http_tps.c
+++ b/src/plugins/hs_apps/http_tps.c
@@ -26,6 +26,7 @@ typedef struct
u64 data_len;
u64 data_offset;
u32 vpp_session_index;
+ u8 *uri;
} hts_session_t;
typedef struct hs_main_
@@ -36,6 +37,9 @@ typedef struct hs_main_
u32 ckpair_index;
u8 *test_data;
+ /** Hash table of listener uris to handles */
+ uword *uri_to_handle;
+
/*
* Configs
*/
@@ -44,6 +48,7 @@ typedef struct hs_main_
u64 segment_size;
u8 debug_level;
u8 no_zc;
+ u8 *default_uri;
} hts_main_t;
static hts_main_t hts_main;
@@ -418,23 +423,53 @@ hts_transport_needs_crypto (transport_proto_t proto)
proto == TRANSPORT_PROTO_QUIC;
}
-static int
-hts_listen (hts_main_t *htm)
+static clib_error_t *
+hts_listen (hts_main_t *htm, u8 *listen_uri, u8 is_del)
{
session_endpoint_cfg_t sep = SESSION_ENDPOINT_CFG_NULL;
vnet_listen_args_t _a, *a = &_a;
- char *uri = "tcp://0.0.0.0/80";
- u8 need_crypto;
+ u8 need_crypto, *uri;
+ hts_session_t *hls;
+ session_t *ls;
+ uword *p;
int rv;
- clib_memset (a, 0, sizeof (*a));
- a->app_index = htm->app_index;
+ uri = listen_uri ? listen_uri : htm->default_uri;
+ p = hash_get_mem (htm->uri_to_handle, uri);
- if (htm->uri)
- uri = (char *) htm->uri;
+ if (is_del)
+ {
+ if (!p)
+ return clib_error_return (0, "not listening on %v", uri);
- if (parse_uri (uri, &sep))
- return -1;
+ hls = hts_session_get (0, *p);
+ ls = listen_session_get (hls->vpp_session_index);
+
+ vnet_unlisten_args_t ua = {
+ .handle = listen_session_get_handle (ls),
+ .app_index = htm->app_index,
+ .wrk_map_index = 0 /* default wrk */
+ };
+
+ hash_unset_mem (htm->uri_to_handle, uri);
+
+ if (vnet_unlisten (&ua))
+ return clib_error_return (0, "failed to unlisten");
+
+ vec_free (hls->uri);
+ hts_session_free (hls);
+
+ return 0;
+ }
+
+ if (p)
+ return clib_error_return (0, "already listening %v", uri);
+
+ if (parse_uri ((char *) uri, &sep))
+ return clib_error_return (0, "failed to parse uri %v", uri);
+
+ clib_memset (a, 0, sizeof (*a));
+ a->app_index = htm->app_index;
need_crypto = hts_transport_needs_crypto (sep.transport_proto);
@@ -453,7 +488,16 @@ hts_listen (hts_main_t *htm)
if (need_crypto)
clib_mem_free (a->sep_ext.ext_cfg);
- return rv;
+ if (rv)
+ return clib_error_return (0, "failed to listen on %v", uri);
+
+ hls = hts_session_alloc (0);
+ hls->uri = vec_dup (uri);
+ ls = listen_session_get_from_handle (a->handle);
+ hls->vpp_session_index = ls->session_index;
+ hash_set_mem (htm->uri_to_handle, hls->uri, hls->session_index);
+
+ return 0;
}
static int
@@ -474,11 +518,9 @@ hts_create (vlib_main_t *vm)
clib_warning ("failed to attach server");
return -1;
}
- if (hts_listen (htm))
- {
- clib_warning ("failed to start listening");
- return -1;
- }
+
+ htm->default_uri = format (0, "tcp://0.0.0.0/80%c", 0);
+ htm->uri_to_handle = hash_create_vec (0, sizeof (u8), sizeof (uword));
return 0;
}
@@ -490,7 +532,9 @@ hts_create_command_fn (vlib_main_t *vm, unformat_input_t *input,
unformat_input_t _line_input, *line_input = &_line_input;
hts_main_t *htm = &hts_main;
clib_error_t *error = 0;
+ u8 is_del = 0;
u64 mem_size;
+ u8 *uri = 0;
/* Get a line of input. */
if (!unformat_user (input, unformat_line_input, line_input))
@@ -504,12 +548,14 @@ hts_create_command_fn (vlib_main_t *vm, unformat_input_t *input,
else if (unformat (line_input, "fifo-size %U", unformat_memory_size,
&mem_size))
htm->fifo_size = mem_size;
- else if (unformat (line_input, "uri %s", &htm->uri))
+ else if (unformat (line_input, "uri %s", &uri))
;
else if (unformat (line_input, "no-zc"))
htm->no_zc = 1;
else if (unformat (line_input, "debug"))
htm->debug_level = 1;
+ else if (unformat (line_input, "del"))
+ is_del = 1;
else
{
error = clib_error_return (0, "unknown input `%U'",
@@ -521,29 +567,112 @@ hts_create_command_fn (vlib_main_t *vm, unformat_input_t *input,
unformat_free (line_input);
if (error)
- return error;
+ goto done;
start_server:
- if (htm->app_index != (u32) ~0)
- return clib_error_return (0, "http tps is already running");
+ if (htm->app_index == (u32) ~0)
+ {
+ vnet_session_enable_disable (vm, 1 /* is_enable */);
- vnet_session_enable_disable (vm, 1 /* is_enable */);
+ if (hts_create (vm))
+ {
+ error = clib_error_return (0, "http tps create failed");
+ goto done;
+ }
+ }
- if (hts_create (vm))
- return clib_error_return (0, "http tps create failed");
+ error = hts_listen (htm, uri, is_del);
- return 0;
+done:
+
+ vec_free (uri);
+ return error;
}
VLIB_CLI_COMMAND (http_tps_command, static) = {
.path = "http tps",
.short_help = "http tps [uri <uri>] [fifo-size <nbytes>] "
- "[segment-size <nMG>] [prealloc-fifos <n>] [debug] [no-zc]",
+ "[segment-size <nMG>] [prealloc-fifos <n>] [debug] [no-zc] "
+ "[del]",
.function = hts_create_command_fn,
};
static clib_error_t *
+hts_show_command_fn (vlib_main_t *vm, unformat_input_t *input,
+ vlib_cli_command_t *cmd)
+{
+ unformat_input_t _line_input, *line_input = &_line_input;
+ hts_main_t *htm = &hts_main;
+ clib_error_t *error = 0;
+ u8 do_listeners = 0;
+ hts_session_t **sessions;
+ u32 n_listeners = 0, n_sessions = 0;
+
+ if (!unformat_user (input, unformat_line_input, line_input))
+ goto no_input;
+
+ while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
+ {
+ if (unformat (line_input, "listeners"))
+ do_listeners = 1;
+ else
+ {
+ error = clib_error_return (0, "unknown input `%U'",
+ format_unformat_error, line_input);
+ break;
+ }
+ }
+
+ if (error)
+ return error;
+
+no_input:
+
+ if (htm->app_index == ~0)
+ {
+ vlib_cli_output (vm, "http tps not enabled");
+ goto done;
+ }
+
+ if (do_listeners)
+ {
+ uword handle;
+ u8 *s = 0, *uri;
+
+ /* clang-format off */
+ hash_foreach (uri, handle, htm->uri_to_handle, ({
+ s = format (s, "%-30v%lx\n", uri, handle);
+ }));
+ /* clang-format on */
+
+ if (s)
+ {
+ vlib_cli_output (vm, "%-29s%s", "URI", "Index");
+ vlib_cli_output (vm, "%v", s);
+ vec_free (s);
+ }
+ goto done;
+ }
+
+ n_listeners = hash_elts (htm->uri_to_handle);
+ vec_foreach (sessions, htm->sessions)
+ n_sessions += pool_elts (*sessions);
+
+ vlib_cli_output (vm, " app index: %u\n listeners: %u\n sesions: %u",
+ htm->app_index, n_listeners, n_sessions - n_listeners);
+
+done:
+ return 0;
+}
+
+VLIB_CLI_COMMAND (show_http_tps_command, static) = {
+ .path = "show http tps",
+ .short_help = "http tps [listeners]",
+ .function = hts_show_command_fn,
+};
+
+static clib_error_t *
hs_main_init (vlib_main_t *vm)
{
hts_main_t *htm = &hts_main;