From a1175b726d428ab6e67e2fe35514099bffa43bc2 Mon Sep 17 00:00:00 2001 From: Florin Coras Date: Wed, 26 Jan 2022 00:15:03 -0800 Subject: hsa: cleanup and rename http test server - cleanup data structures, functions and cli - remove option to return static html. For similar results, use http static server - rename to http_cli.c as it better describes what the app does Type: refactor Signed-off-by: Florin Coras Change-Id: I502e7566fba4376c68fbe41de9e45079a159e864 --- src/plugins/hs_apps/CMakeLists.txt | 2 +- src/plugins/hs_apps/http_cli.c | 668 ++++++++++++++++++++++++++++++++++ src/plugins/hs_apps/http_server.c | 718 ------------------------------------- 3 files changed, 669 insertions(+), 719 deletions(-) create mode 100644 src/plugins/hs_apps/http_cli.c delete mode 100644 src/plugins/hs_apps/http_server.c (limited to 'src/plugins/hs_apps') diff --git a/src/plugins/hs_apps/CMakeLists.txt b/src/plugins/hs_apps/CMakeLists.txt index 1f474828b15..701d1374c92 100644 --- a/src/plugins/hs_apps/CMakeLists.txt +++ b/src/plugins/hs_apps/CMakeLists.txt @@ -19,7 +19,7 @@ add_vpp_plugin(hs_apps echo_client.c echo_server.c hs_apps.c - http_server.c + http_cli.c proxy.c ) diff --git a/src/plugins/hs_apps/http_cli.c b/src/plugins/hs_apps/http_cli.c new file mode 100644 index 00000000000..50cedce3524 --- /dev/null +++ b/src/plugins/hs_apps/http_cli.c @@ -0,0 +1,668 @@ +/* +* Copyright (c) 2017-2019 Cisco and/or its affiliates. +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at: +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +#include +#include +#include +#include + +typedef struct +{ + u32 hs_index; + u32 thread_index; + u64 node_index; + u8 *buf; +} hcs_cli_args_t; + +typedef struct +{ + CLIB_CACHE_LINE_ALIGN_MARK (cacheline0); + u32 session_index; + u32 thread_index; + u8 *tx_buf; + u32 tx_offset; + u32 vpp_session_index; +} hcs_session_t; + +typedef struct +{ + hcs_session_t **sessions; + u32 *free_http_cli_process_node_indices; + u32 app_index; + + /* Cert key pair for tls */ + u32 ckpair_index; + + u32 prealloc_fifos; + u32 private_segment_size; + u32 fifo_size; + u8 *uri; + vlib_main_t *vlib_main; +} hcs_main_t; + +static hcs_main_t hcs_main; + +static hcs_session_t * +hcs_session_alloc (u32 thread_index) +{ + hcs_main_t *hcm = &hcs_main; + hcs_session_t *hs; + pool_get (hcm->sessions[thread_index], hs); + memset (hs, 0, sizeof (*hs)); + hs->session_index = hs - hcm->sessions[thread_index]; + hs->thread_index = thread_index; + return hs; +} + +static hcs_session_t * +hcs_session_get (u32 thread_index, u32 hs_index) +{ + hcs_main_t *hcm = &hcs_main; + if (pool_is_free_index (hcm->sessions[thread_index], hs_index)) + return 0; + return pool_elt_at_index (hcm->sessions[thread_index], hs_index); +} + +static void +hcs_session_free (hcs_session_t *hs) +{ + hcs_main_t *hcm = &hcs_main; + u32 thread = hs->thread_index; + if (CLIB_DEBUG) + memset (hs, 0xfa, sizeof (*hs)); + pool_put (hcm->sessions[thread], hs); +} + +static void +hcs_cli_process_free (hcs_cli_args_t *args) +{ + vlib_main_t *vm = vlib_get_first_main (); + hcs_main_t *hcm = &hcs_main; + hcs_cli_args_t **save_args; + vlib_node_runtime_t *rt; + vlib_node_t *n; + u32 node_index; + + node_index = args->node_index; + ASSERT (node_index != 0); + + n = vlib_get_node (vm, node_index); + rt = vlib_node_get_runtime (vm, n->index); + save_args = vlib_node_get_runtime_data (vm, n->index); + + /* Reset process session pointer */ + clib_mem_free (*save_args); + *save_args = 0; + + /* Turn off the process node */ + vlib_node_set_state (vm, rt->node_index, VLIB_NODE_STATE_DISABLED); + + /* add node index to the freelist */ + vec_add1 (hcm->free_http_cli_process_node_indices, node_index); +} + +/* Header, including incantation to suppress favicon.ico requests */ +static const char *html_header_template = + "%v" + "" + "
";
+
+static const char *html_footer =
+    "
\r\n"; + +static void +hcs_cli_output (uword arg, u8 *buffer, uword buffer_bytes) +{ + u8 **output_vecp = (u8 **) arg; + u8 *output_vec; + u32 offset; + + output_vec = *output_vecp; + + offset = vec_len (output_vec); + vec_validate (output_vec, offset + buffer_bytes - 1); + clib_memcpy_fast (output_vec + offset, buffer, buffer_bytes); + + *output_vecp = output_vec; +} + +static void +start_send_data (hcs_session_t *hs, http_status_code_t status) +{ + http_msg_t msg; + session_t *ts; + int rv; + + msg.type = HTTP_MSG_REPLY; + msg.code = status; + msg.content_type = HTTP_CONTENT_TEXT_HTML; + msg.data.type = HTTP_MSG_DATA_INLINE; + msg.data.len = vec_len (hs->tx_buf); + + ts = session_get (hs->vpp_session_index, hs->thread_index); + rv = svm_fifo_enqueue (ts->tx_fifo, sizeof (msg), (u8 *) &msg); + ASSERT (rv == sizeof (msg)); + + if (!msg.data.len) + goto done; + + rv = svm_fifo_enqueue (ts->tx_fifo, vec_len (hs->tx_buf), hs->tx_buf); + + if (rv != vec_len (hs->tx_buf)) + { + hs->tx_offset = rv; + svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); + } + else + { + vec_free (hs->tx_buf); + } + +done: + + if (svm_fifo_set_event (ts->tx_fifo)) + session_send_io_evt_to_thread (ts->tx_fifo, SESSION_IO_EVT_TX); +} + +static void +send_data_to_http (void *rpc_args) +{ + hcs_cli_args_t *args = (hcs_cli_args_t *) rpc_args; + hcs_session_t *hs; + + hs = hcs_session_get (args->thread_index, args->hs_index); + if (!hs) + { + vec_free (args->buf); + goto cleanup; + } + + hs->tx_buf = args->buf; + start_send_data (hs, HTTP_STATUS_OK); + +cleanup: + + clib_mem_free (rpc_args); +} + +static uword +hcs_cli_process (vlib_main_t *vm, vlib_node_runtime_t *rt, vlib_frame_t *f) +{ + u8 *request = 0, *reply = 0, *html = 0; + hcs_cli_args_t *args, *rpc_args; + hcs_main_t *hcm = &hcs_main; + hcs_cli_args_t **save_args; + unformat_input_t input; + int i; + + save_args = vlib_node_get_runtime_data (hcm->vlib_main, rt->node_index); + args = *save_args; + + request = args->buf; + + /* Replace slashes with spaces, stop at the end of the path */ + i = 0; + while (i < vec_len (request)) + { + if (request[i] == '/') + request[i] = ' '; + else if (request[i] == ' ') + { + /* vlib_cli_input is vector-based, no need for a NULL */ + _vec_len (request) = i; + break; + } + i++; + } + + /* Generate the html header */ + html = format (0, html_header_template, request /* title */ ); + + /* Run the command */ + unformat_init_vector (&input, vec_dup (request)); + vlib_cli_input (vm, &input, hcs_cli_output, (uword) &reply); + unformat_free (&input); + request = 0; + + /* Generate the html page */ + html = format (html, "%v", reply); + html = format (html, html_footer); + + /* Send it */ + rpc_args = clib_mem_alloc (sizeof (*args)); + clib_memcpy_fast (rpc_args, args, sizeof (*args)); + rpc_args->buf = html; + + session_send_rpc_evt_to_thread_force (args->thread_index, send_data_to_http, + rpc_args); + + vec_free (reply); + vec_free (args->buf); + hcs_cli_process_free (args); + + return (0); +} + +static void +alloc_cli_process (hcs_cli_args_t *args) +{ + hcs_main_t *hcm = &hcs_main; + vlib_main_t *vm = hcm->vlib_main; + hcs_cli_args_t **save_args; + vlib_node_t *n; + char *name; + uword l; + + l = vec_len (hcm->free_http_cli_process_node_indices); + if (l > 0) + { + n = vlib_get_node (vm, hcm->free_http_cli_process_node_indices[l - 1]); + vlib_node_set_state (vm, n->index, VLIB_NODE_STATE_POLLING); + _vec_len (hcm->free_http_cli_process_node_indices) = l - 1; + } + else + { + static vlib_node_registration_t r = { + .function = hcs_cli_process, + .type = VLIB_NODE_TYPE_PROCESS, + .process_log2_n_stack_bytes = 16, + .runtime_data_bytes = sizeof (void *), + }; + + name = (char *) format (0, "http-cli-%d", l); + r.name = name; + vlib_register_node (vm, &r); + vec_free (name); + + n = vlib_get_node (vm, r.index); + } + + /* Save the node index in the args. It won't be zero. */ + args->node_index = n->index; + + /* Save the args (pointer) in the node runtime */ + save_args = vlib_node_get_runtime_data (vm, n->index); + *save_args = clib_mem_alloc (sizeof (*args)); + clib_memcpy_fast (*save_args, args, sizeof (*args)); + + vlib_start_process (vm, n->runtime_index); +} + +static void +alloc_cli_process_callback (void *cb_args) +{ + alloc_cli_process ((hcs_cli_args_t *) cb_args); +} + +static int +hcs_ts_rx_callback (session_t *ts) +{ + hcs_cli_args_t args = {}; + hcs_session_t *hs; + http_msg_t msg; + int rv; + + hs = hcs_session_get (ts->thread_index, ts->opaque); + + /* Read the http message header */ + rv = svm_fifo_dequeue (ts->rx_fifo, sizeof (msg), (u8 *) &msg); + ASSERT (rv == sizeof (msg)); + + if (msg.type != HTTP_MSG_REQUEST || msg.method_type != HTTP_REQ_GET) + { + hs->tx_buf = 0; + start_send_data (hs, HTTP_STATUS_METHOD_NOT_ALLOWED); + return 0; + } + + /* send the command to a new/recycled vlib process */ + vec_validate (args.buf, msg.data.len - 1); + rv = svm_fifo_dequeue (ts->rx_fifo, msg.data.len, args.buf); + ASSERT (rv == msg.data.len); + vec_set_len (args.buf, rv); + + args.hs_index = hs->session_index; + args.thread_index = ts->thread_index; + + /* Send RPC request to main thread */ + if (vlib_get_thread_index () != 0) + vlib_rpc_call_main_thread (alloc_cli_process_callback, (u8 *) &args, + sizeof (args)); + else + alloc_cli_process (&args); + return 0; +} + +static int +hcs_ts_tx_callback (session_t *ts) +{ + hcs_session_t *hs; + u32 to_send; + int rv; + + hs = hcs_session_get (ts->thread_index, ts->opaque); + if (!hs || !hs->tx_buf) + return 0; + + to_send = vec_len (hs->tx_buf) - hs->tx_offset; + rv = svm_fifo_enqueue (ts->tx_fifo, to_send, hs->tx_buf + hs->tx_offset); + + if (rv <= 0) + { + svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); + return 0; + } + + if (rv < to_send) + { + hs->tx_offset += rv; + svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); + } + else + { + vec_free (hs->tx_buf); + } + + if (svm_fifo_set_event (ts->tx_fifo)) + session_send_io_evt_to_thread (ts->tx_fifo, SESSION_IO_EVT_TX); + + return 0; +} + +static int +hcs_ts_accept_callback (session_t *ts) +{ + hcs_session_t *hs; + + hs = hcs_session_alloc (ts->thread_index); + hs->vpp_session_index = ts->session_index; + + ts->opaque = hs->session_index; + ts->session_state = SESSION_STATE_READY; + + return 0; +} + +static int +hcs_ts_connected_callback (u32 app_index, u32 api_context, session_t *s, + session_error_t err) +{ + clib_warning ("called..."); + return -1; +} + +static void +hcs_ts_disconnect_callback (session_t *s) +{ + hcs_main_t *hcm = &hcs_main; + vnet_disconnect_args_t _a = { 0 }, *a = &_a; + + a->handle = session_handle (s); + a->app_index = hcm->app_index; + vnet_disconnect_session (a); +} + +static void +hcs_ts_reset_callback (session_t *s) +{ + hcs_main_t *hcm = &hcs_main; + vnet_disconnect_args_t _a = { 0 }, *a = &_a; + + a->handle = session_handle (s); + a->app_index = hcm->app_index; + vnet_disconnect_session (a); +} + +static void +hcs_ts_cleanup_callback (session_t *s, session_cleanup_ntf_t ntf) +{ + hcs_session_t *hs; + + if (ntf == SESSION_CLEANUP_TRANSPORT) + return; + + hs = hcs_session_get (s->thread_index, s->opaque); + if (!hs) + return; + + vec_free (hs->tx_buf); + hcs_session_free (hs); +} + +static int +hcs_add_segment_callback (u32 client_index, u64 segment_handle) +{ + return 0; +} + +static int +hcs_del_segment_callback (u32 client_index, u64 segment_handle) +{ + return 0; +} + +static session_cb_vft_t hcs_session_cb_vft = { + .session_accept_callback = hcs_ts_accept_callback, + .session_disconnect_callback = hcs_ts_disconnect_callback, + .session_connected_callback = hcs_ts_connected_callback, + .add_segment_callback = hcs_add_segment_callback, + .del_segment_callback = hcs_del_segment_callback, + .builtin_app_rx_callback = hcs_ts_rx_callback, + .builtin_app_tx_callback = hcs_ts_tx_callback, + .session_reset_callback = hcs_ts_reset_callback, + .session_cleanup_callback = hcs_ts_cleanup_callback, +}; + +static int +hcs_attach () +{ + vnet_app_add_cert_key_pair_args_t _ck_pair, *ck_pair = &_ck_pair; + hcs_main_t *hcm = &hcs_main; + u64 options[APP_OPTIONS_N_OPTIONS]; + vnet_app_attach_args_t _a, *a = &_a; + u32 segment_size = 128 << 20; + + clib_memset (a, 0, sizeof (*a)); + clib_memset (options, 0, sizeof (options)); + + if (hcm->private_segment_size) + segment_size = hcm->private_segment_size; + + a->api_client_index = ~0; + a->name = format (0, "http_cli_server"); + a->session_cb_vft = &hcs_session_cb_vft; + a->options = options; + a->options[APP_OPTIONS_SEGMENT_SIZE] = segment_size; + a->options[APP_OPTIONS_ADD_SEGMENT_SIZE] = segment_size; + a->options[APP_OPTIONS_RX_FIFO_SIZE] = + hcm->fifo_size ? hcm->fifo_size : 8 << 10; + a->options[APP_OPTIONS_TX_FIFO_SIZE] = + hcm->fifo_size ? hcm->fifo_size : 32 << 10; + a->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN; + a->options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = hcm->prealloc_fifos; + + if (vnet_application_attach (a)) + { + vec_free (a->name); + clib_warning ("failed to attach server"); + return -1; + } + vec_free (a->name); + hcm->app_index = a->app_index; + + clib_memset (ck_pair, 0, sizeof (*ck_pair)); + ck_pair->cert = (u8 *) test_srv_crt_rsa; + ck_pair->key = (u8 *) test_srv_key_rsa; + ck_pair->cert_len = test_srv_crt_rsa_len; + ck_pair->key_len = test_srv_key_rsa_len; + vnet_app_add_cert_key_pair (ck_pair); + hcm->ckpair_index = ck_pair->index; + + return 0; +} + +static int +hcs_transport_needs_crypto (transport_proto_t proto) +{ + return proto == TRANSPORT_PROTO_TLS || proto == TRANSPORT_PROTO_DTLS || + proto == TRANSPORT_PROTO_QUIC; +} + +static int +hcs_listen () +{ + session_endpoint_cfg_t sep = SESSION_ENDPOINT_CFG_NULL; + hcs_main_t *hcm = &hcs_main; + vnet_listen_args_t _a, *a = &_a; + char *uri = "tcp://0.0.0.0/80"; + u8 need_crypto; + int rv; + + clib_memset (a, 0, sizeof (*a)); + a->app_index = hcm->app_index; + + if (hcm->uri) + uri = (char *) hcm->uri; + + if (parse_uri (uri, &sep)) + return -1; + + need_crypto = hcs_transport_needs_crypto (a->sep_ext.transport_proto); + + sep.transport_proto = TRANSPORT_PROTO_HTTP; + clib_memcpy (&a->sep_ext, &sep, sizeof (sep)); + + if (need_crypto) + { + session_endpoint_alloc_ext_cfg (&a->sep_ext, + TRANSPORT_ENDPT_EXT_CFG_CRYPTO); + a->sep_ext.ext_cfg->crypto.ckpair_index = hcm->ckpair_index; + } + + rv = vnet_listen (a); + + if (need_crypto) + clib_mem_free (a->sep_ext.ext_cfg); + + return rv; +} + +static int +hcs_create (vlib_main_t *vm) +{ + vlib_thread_main_t *vtm = vlib_get_thread_main (); + hcs_main_t *hcm = &hcs_main; + u32 num_threads; + + num_threads = 1 /* main thread */ + vtm->n_threads; + vec_validate (hcm->sessions, num_threads - 1); + + if (hcs_attach ()) + { + clib_warning ("failed to attach server"); + return -1; + } + if (hcs_listen ()) + { + clib_warning ("failed to start listening"); + return -1; + } + + return 0; +} + +static clib_error_t * +hcs_create_command_fn (vlib_main_t *vm, unformat_input_t *input, + vlib_cli_command_t *cmd) +{ + unformat_input_t _line_input, *line_input = &_line_input; + hcs_main_t *hcm = &hcs_main; + u64 seg_size; + int rv; + + hcm->prealloc_fifos = 0; + hcm->private_segment_size = 0; + hcm->fifo_size = 0; + + /* Get a line of input. */ + if (!unformat_user (input, unformat_line_input, line_input)) + goto start_server; + + while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT) + { + if (unformat (line_input, "prealloc-fifos %d", &hcm->prealloc_fifos)) + ; + else if (unformat (line_input, "private-segment-size %U", + unformat_memory_size, &seg_size)) + hcm->private_segment_size = seg_size; + else if (unformat (line_input, "fifo-size %d", &hcm->fifo_size)) + hcm->fifo_size <<= 10; + else if (unformat (line_input, "uri %s", &hcm->uri)) + ; + else + { + unformat_free (line_input); + return clib_error_return (0, "unknown input `%U'", + format_unformat_error, line_input); + } + } + + unformat_free (line_input); + +start_server: + + if (hcm->app_index != (u32) ~0) + return clib_error_return (0, "test http server is already running"); + + vnet_session_enable_disable (vm, 1 /* turn on TCP, etc. */ ); + + rv = hcs_create (vm); + switch (rv) + { + case 0: + break; + default: + return clib_error_return (0, "server_create returned %d", rv); + } + + return 0; +} + +VLIB_CLI_COMMAND (hcs_create_command, static) = { + .path = "http cli server", + .short_help = "http cli server [uri ] [fifo-size ] " + "[private-segment-size ] [prealloc-fifos ]", + .function = hcs_create_command_fn, +}; + +static clib_error_t * +hcs_main_init (vlib_main_t *vm) +{ + hcs_main_t *hcs = &hcs_main; + + hcs->app_index = ~0; + hcs->vlib_main = vm; + return 0; +} + +VLIB_INIT_FUNCTION (hcs_main_init); + +/* +* fd.io coding-style-patch-verification: ON +* +* Local Variables: +* eval: (c-set-style "gnu") +* End: +*/ diff --git a/src/plugins/hs_apps/http_server.c b/src/plugins/hs_apps/http_server.c deleted file mode 100644 index 3083937ec74..00000000000 --- a/src/plugins/hs_apps/http_server.c +++ /dev/null @@ -1,718 +0,0 @@ -/* -* Copyright (c) 2017-2019 Cisco and/or its affiliates. -* Licensed under the Apache License, Version 2.0 (the "License"); -* you may not use this file except in compliance with the License. -* You may obtain a copy of the License at: -* -* http://www.apache.org/licenses/LICENSE-2.0 -* -* Unless required by applicable law or agreed to in writing, software -* distributed under the License is distributed on an "AS IS" BASIS, -* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -* See the License for the specific language governing permissions and -* limitations under the License. -*/ - -#include -#include -#include -#include -#include - -typedef struct -{ - u32 hs_index; - u32 thread_index; - u64 node_index; - u8 *buf; -} http_server_args; - -typedef struct -{ - CLIB_CACHE_LINE_ALIGN_MARK (cacheline0); - u32 session_index; - u32 thread_index; - u8 *tx_buf; - u32 tx_offset; - u32 vpp_session_index; - u64 vpp_session_handle; - u32 timer_handle; -} http_session_t; - -typedef struct -{ - http_session_t **sessions; - u32 *free_http_cli_process_node_indices; - u32 app_index; - - /* Cert key pair for tls */ - u32 ckpair_index; - - u32 prealloc_fifos; - u32 private_segment_size; - u32 fifo_size; - u8 *uri; - u32 is_static; - vlib_main_t *vlib_main; -} http_server_main_t; - -http_server_main_t http_server_main; - -static u8 *static_html; - -static http_session_t * -http_server_session_alloc (u32 thread_index) -{ - http_server_main_t *hsm = &http_server_main; - http_session_t *hs; - pool_get (hsm->sessions[thread_index], hs); - memset (hs, 0, sizeof (*hs)); - hs->session_index = hs - hsm->sessions[thread_index]; - hs->thread_index = thread_index; - hs->timer_handle = ~0; - return hs; -} - -static http_session_t * -http_server_session_get (u32 thread_index, u32 hs_index) -{ - http_server_main_t *hsm = &http_server_main; - if (pool_is_free_index (hsm->sessions[thread_index], hs_index)) - return 0; - return pool_elt_at_index (hsm->sessions[thread_index], hs_index); -} - -static void -http_server_session_free (http_session_t * hs) -{ - http_server_main_t *hsm = &http_server_main; - u32 thread = hs->thread_index; - if (CLIB_DEBUG) - memset (hs, 0xfa, sizeof (*hs)); - pool_put (hsm->sessions[thread], hs); -} - -static void -http_process_free (http_server_args * args) -{ - vlib_node_runtime_t *rt; - vlib_main_t *vm = vlib_get_first_main (); - http_server_main_t *hsm = &http_server_main; - vlib_node_t *n; - u32 node_index; - http_server_args **save_args; - - node_index = args->node_index; - ASSERT (node_index != 0); - - n = vlib_get_node (vm, node_index); - rt = vlib_node_get_runtime (vm, n->index); - save_args = vlib_node_get_runtime_data (vm, n->index); - - /* Reset process session pointer */ - clib_mem_free (*save_args); - *save_args = 0; - - /* Turn off the process node */ - vlib_node_set_state (vm, rt->node_index, VLIB_NODE_STATE_DISABLED); - - /* add node index to the freelist */ - vec_add1 (hsm->free_http_cli_process_node_indices, node_index); -} - -/* Header, including incantation to suppress favicon.ico requests */ -static const char *html_header_template = - "%v" - "" - "
";
-
-static const char *html_footer =
-    "
\r\n"; - -static const char *html_header_static = - "static reply" - "" - "
hello
\r\n"; - -static void -http_cli_output (uword arg, u8 * buffer, uword buffer_bytes) -{ - u8 **output_vecp = (u8 **) arg; - u8 *output_vec; - u32 offset; - - output_vec = *output_vecp; - - offset = vec_len (output_vec); - vec_validate (output_vec, offset + buffer_bytes - 1); - clib_memcpy_fast (output_vec + offset, buffer, buffer_bytes); - - *output_vecp = output_vec; -} - -static void -start_send_data (http_session_t *hs, http_status_code_t status) -{ - http_msg_t msg; - session_t *ts; - int rv; - - msg.type = HTTP_MSG_REPLY; - msg.code = status; - msg.content_type = HTTP_CONTENT_TEXT_HTML; - msg.data.type = HTTP_MSG_DATA_INLINE; - msg.data.len = vec_len (hs->tx_buf); - - ts = session_get (hs->vpp_session_index, hs->thread_index); - rv = svm_fifo_enqueue (ts->tx_fifo, sizeof (msg), (u8 *) &msg); - ASSERT (rv == sizeof (msg)); - - if (!msg.data.len) - goto done; - - rv = svm_fifo_enqueue (ts->tx_fifo, vec_len (hs->tx_buf), hs->tx_buf); - - if (rv != vec_len (hs->tx_buf)) - { - hs->tx_offset = rv; - svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); - } - else - { - vec_free (hs->tx_buf); - } - -done: - - if (svm_fifo_set_event (ts->tx_fifo)) - session_send_io_evt_to_thread (ts->tx_fifo, SESSION_IO_EVT_TX); -} - -static void -send_data_to_http (void *rpc_args) -{ - http_server_args *args = (http_server_args *) rpc_args; - http_session_t *hs; - - hs = http_server_session_get (args->thread_index, args->hs_index); - if (!hs) - { - vec_free (args->buf); - goto cleanup; - } - - hs->tx_buf = args->buf; - start_send_data (hs, HTTP_STATUS_OK); - -cleanup: - - clib_mem_free (rpc_args); -} - -static uword -http_cli_process (vlib_main_t * vm, vlib_node_runtime_t * rt, - vlib_frame_t * f) -{ - u8 *request = 0, *reply = 0, *html = 0; - http_server_main_t *hsm = &http_server_main; - http_server_args **save_args; - http_server_args *args, *rpc_args; - unformat_input_t input; - int i; - - save_args = vlib_node_get_runtime_data (hsm->vlib_main, rt->node_index); - args = *save_args; - - request = args->buf; - - /* Replace slashes with spaces, stop at the end of the path */ - i = 0; - while (i < vec_len (request)) - { - if (request[i] == '/') - request[i] = ' '; - else if (request[i] == ' ') - { - /* vlib_cli_input is vector-based, no need for a NULL */ - _vec_len (request) = i; - break; - } - i++; - } - - /* Generate the html header */ - html = format (0, html_header_template, request /* title */ ); - - /* Run the command */ - unformat_init_vector (&input, vec_dup (request)); - vlib_cli_input (vm, &input, http_cli_output, (uword) & reply); - unformat_free (&input); - request = 0; - - /* Generate the html page */ - html = format (html, "%v", reply); - html = format (html, html_footer); - - /* Send it */ - rpc_args = clib_mem_alloc (sizeof (*args)); - clib_memcpy_fast (rpc_args, args, sizeof (*args)); - rpc_args->buf = html; - - session_send_rpc_evt_to_thread_force (args->thread_index, send_data_to_http, - rpc_args); - - vec_free (reply); - vec_free (args->buf); - http_process_free (args); - - return (0); -} - -static void -alloc_http_process (http_server_args * args) -{ - char *name; - vlib_node_t *n; - http_server_main_t *hsm = &http_server_main; - vlib_main_t *vm = hsm->vlib_main; - uword l = vec_len (hsm->free_http_cli_process_node_indices); - http_server_args **save_args; - - if (vec_len (hsm->free_http_cli_process_node_indices) > 0) - { - n = vlib_get_node (vm, hsm->free_http_cli_process_node_indices[l - 1]); - vlib_node_set_state (vm, n->index, VLIB_NODE_STATE_POLLING); - _vec_len (hsm->free_http_cli_process_node_indices) = l - 1; - } - else - { - static vlib_node_registration_t r = { - .function = http_cli_process, - .type = VLIB_NODE_TYPE_PROCESS, - .process_log2_n_stack_bytes = 16, - .runtime_data_bytes = sizeof (void *), - }; - - name = (char *) format (0, "http-cli-%d", l); - r.name = name; - vlib_register_node (vm, &r); - vec_free (name); - - n = vlib_get_node (vm, r.index); - } - - /* Save the node index in the args. It won't be zero. */ - args->node_index = n->index; - - /* Save the args (pointer) in the node runtime */ - save_args = vlib_node_get_runtime_data (vm, n->index); - *save_args = clib_mem_alloc (sizeof (*args)); - clib_memcpy_fast (*save_args, args, sizeof (*args)); - - vlib_start_process (vm, n->runtime_index); -} - -static void -alloc_http_process_callback (void *cb_args) -{ - alloc_http_process ((http_server_args *) cb_args); -} - -static int -http_server_rx_callback (session_t *ts) -{ - http_server_args args = {}; - http_session_t *hs; - http_msg_t msg; - int rv; - - hs = http_server_session_get (ts->thread_index, ts->opaque); - - /* Read the http message header */ - rv = svm_fifo_dequeue (ts->rx_fifo, sizeof (msg), (u8 *) &msg); - ASSERT (rv == sizeof (msg)); - - if (msg.type != HTTP_MSG_REQUEST || msg.method_type != HTTP_REQ_GET) - { - hs->tx_buf = 0; - start_send_data (hs, HTTP_STATUS_METHOD_NOT_ALLOWED); - return 0; - } - - /* send the command to a new/recycled vlib process */ - vec_validate (args.buf, msg.data.len - 1); - rv = svm_fifo_dequeue (ts->rx_fifo, msg.data.len, args.buf); - ASSERT (rv == msg.data.len); - vec_set_len (args.buf, rv); - - args.hs_index = hs->session_index; - args.thread_index = ts->thread_index; - - /* Send RPC request to main thread */ - if (vlib_get_thread_index () != 0) - vlib_rpc_call_main_thread (alloc_http_process_callback, (u8 *) & args, - sizeof (args)); - else - alloc_http_process (&args); - return 0; -} - -static int -http_server_rx_callback_static (session_t *ts) -{ - http_session_t *hs; - http_msg_t msg; - int rv; - - hs = http_server_session_get (ts->thread_index, ts->opaque); - rv = svm_fifo_dequeue (ts->rx_fifo, sizeof (msg), (u8 *) &msg); - ASSERT (rv == sizeof (msg)); - - svm_fifo_dequeue_drop (ts->rx_fifo, msg.data.len); - - if (msg.type != HTTP_MSG_REQUEST || msg.method_type != HTTP_REQ_GET) - { - hs->tx_buf = 0; - start_send_data (hs, HTTP_STATUS_METHOD_NOT_ALLOWED); - goto done; - } - - hs->tx_buf = vec_dup (static_html); - start_send_data (hs, HTTP_STATUS_OK); - -done: - - return 0; -} - -static int -http_server_tx_callback (session_t *ts) -{ - http_session_t *hs; - u32 to_send; - int rv; - - hs = http_server_session_get (ts->thread_index, ts->opaque); - if (!hs || !hs->tx_buf) - return 0; - - to_send = vec_len (hs->tx_buf) - hs->tx_offset; - rv = svm_fifo_enqueue (ts->tx_fifo, to_send, hs->tx_buf + hs->tx_offset); - - if (rv <= 0) - { - svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); - return 0; - } - - if (rv < to_send) - { - hs->tx_offset += rv; - svm_fifo_add_want_deq_ntf (ts->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF); - } - else - { - vec_free (hs->tx_buf); - } - - if (svm_fifo_set_event (ts->tx_fifo)) - session_send_io_evt_to_thread (ts->tx_fifo, SESSION_IO_EVT_TX); - - return 0; -} - -static int -http_server_session_accept_callback (session_t *ts) -{ - http_session_t *hs; - - hs = http_server_session_alloc (ts->thread_index); - hs->vpp_session_index = ts->session_index; - hs->vpp_session_handle = session_handle (ts); - - ts->opaque = hs->session_index; - ts->session_state = SESSION_STATE_READY; - - return 0; -} - -static void -http_server_session_disconnect_callback (session_t * s) -{ - http_server_main_t *hsm = &http_server_main; - vnet_disconnect_args_t _a = { 0 }, *a = &_a; - - a->handle = session_handle (s); - a->app_index = hsm->app_index; - vnet_disconnect_session (a); -} - -static void -http_server_session_reset_callback (session_t * s) -{ - http_server_main_t *hsm = &http_server_main; - vnet_disconnect_args_t _a = { 0 }, *a = &_a; - - a->handle = session_handle (s); - a->app_index = hsm->app_index; - vnet_disconnect_session (a); -} - -static int -http_server_session_connected_callback (u32 app_index, u32 api_context, - session_t * s, session_error_t err) -{ - clib_warning ("called..."); - return -1; -} - -static int -http_server_add_segment_callback (u32 client_index, u64 segment_handle) -{ - return 0; -} - -static void -http_server_cleanup_callback (session_t * s, session_cleanup_ntf_t ntf) -{ - http_session_t *hs; - - if (ntf == SESSION_CLEANUP_TRANSPORT) - return; - - hs = http_server_session_get (s->thread_index, s->opaque); - if (!hs) - return; - - vec_free (hs->tx_buf); - http_server_session_free (hs); -} - -static session_cb_vft_t http_server_session_cb_vft = { - .session_accept_callback = http_server_session_accept_callback, - .session_disconnect_callback = http_server_session_disconnect_callback, - .session_connected_callback = http_server_session_connected_callback, - .add_segment_callback = http_server_add_segment_callback, - .builtin_app_rx_callback = http_server_rx_callback, - .builtin_app_tx_callback = http_server_tx_callback, - .session_reset_callback = http_server_session_reset_callback, - .session_cleanup_callback = http_server_cleanup_callback, -}; - -static int -http_server_attach () -{ - vnet_app_add_cert_key_pair_args_t _ck_pair, *ck_pair = &_ck_pair; - http_server_main_t *hsm = &http_server_main; - u64 options[APP_OPTIONS_N_OPTIONS]; - vnet_app_attach_args_t _a, *a = &_a; - u32 segment_size = 128 << 20; - - clib_memset (a, 0, sizeof (*a)); - clib_memset (options, 0, sizeof (options)); - - if (hsm->private_segment_size) - segment_size = hsm->private_segment_size; - - a->api_client_index = ~0; - a->name = format (0, "test_http_server"); - a->session_cb_vft = &http_server_session_cb_vft; - a->options = options; - a->options[APP_OPTIONS_SEGMENT_SIZE] = segment_size; - a->options[APP_OPTIONS_ADD_SEGMENT_SIZE] = segment_size; - a->options[APP_OPTIONS_RX_FIFO_SIZE] = - hsm->fifo_size ? hsm->fifo_size : 8 << 10; - a->options[APP_OPTIONS_TX_FIFO_SIZE] = - hsm->fifo_size ? hsm->fifo_size : 32 << 10; - a->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN; - a->options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = hsm->prealloc_fifos; - - if (vnet_application_attach (a)) - { - vec_free (a->name); - clib_warning ("failed to attach server"); - return -1; - } - vec_free (a->name); - hsm->app_index = a->app_index; - - clib_memset (ck_pair, 0, sizeof (*ck_pair)); - ck_pair->cert = (u8 *) test_srv_crt_rsa; - ck_pair->key = (u8 *) test_srv_key_rsa; - ck_pair->cert_len = test_srv_crt_rsa_len; - ck_pair->key_len = test_srv_key_rsa_len; - vnet_app_add_cert_key_pair (ck_pair); - hsm->ckpair_index = ck_pair->index; - - return 0; -} - -static int -http_transport_needs_crypto (transport_proto_t proto) -{ - return proto == TRANSPORT_PROTO_TLS || proto == TRANSPORT_PROTO_DTLS || - proto == TRANSPORT_PROTO_QUIC; -} - -static int -http_server_listen () -{ - session_endpoint_cfg_t sep = SESSION_ENDPOINT_CFG_NULL; - http_server_main_t *hsm = &http_server_main; - vnet_listen_args_t _a, *a = &_a; - char *uri = "tcp://0.0.0.0/80"; - u8 need_crypto; - int rv; - - clib_memset (a, 0, sizeof (*a)); - a->app_index = hsm->app_index; - - if (hsm->uri) - uri = (char *) hsm->uri; - - if (parse_uri (uri, &sep)) - return -1; - - need_crypto = http_transport_needs_crypto (a->sep_ext.transport_proto); - - sep.transport_proto = TRANSPORT_PROTO_HTTP; - clib_memcpy (&a->sep_ext, &sep, sizeof (sep)); - - if (need_crypto) - { - session_endpoint_alloc_ext_cfg (&a->sep_ext, - TRANSPORT_ENDPT_EXT_CFG_CRYPTO); - a->sep_ext.ext_cfg->crypto.ckpair_index = hsm->ckpair_index; - } - - rv = vnet_listen (a); - - if (need_crypto) - clib_mem_free (a->sep_ext.ext_cfg); - - return rv; -} - -static int -http_server_create (vlib_main_t * vm) -{ - vlib_thread_main_t *vtm = vlib_get_thread_main (); - http_server_main_t *hsm = &http_server_main; - u32 num_threads; - - num_threads = 1 /* main thread */ + vtm->n_threads; - vec_validate (hsm->sessions, num_threads - 1); - - if (http_server_attach ()) - { - clib_warning ("failed to attach server"); - return -1; - } - if (http_server_listen ()) - { - clib_warning ("failed to start listening"); - return -1; - } - - return 0; -} - -static clib_error_t * -http_server_create_command_fn (vlib_main_t * vm, - unformat_input_t * input, - vlib_cli_command_t * cmd) -{ - http_server_main_t *hsm = &http_server_main; - unformat_input_t _line_input, *line_input = &_line_input; - u64 seg_size; - int rv; - - hsm->prealloc_fifos = 0; - hsm->private_segment_size = 0; - hsm->fifo_size = 0; - hsm->is_static = 0; - - /* Get a line of input. */ - if (!unformat_user (input, unformat_line_input, line_input)) - goto start_server; - - while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT) - { - if (unformat (line_input, "static")) - hsm->is_static = 1; - else - if (unformat (line_input, "prealloc-fifos %d", &hsm->prealloc_fifos)) - ; - else if (unformat (line_input, "private-segment-size %U", - unformat_memory_size, &seg_size)) - { - if (seg_size >= 0x100000000ULL) - { - vlib_cli_output (vm, "private segment size %llu, too large", - seg_size); - return 0; - } - hsm->private_segment_size = seg_size; - } - else if (unformat (line_input, "fifo-size %d", &hsm->fifo_size)) - hsm->fifo_size <<= 10; - else if (unformat (line_input, "uri %s", &hsm->uri)) - ; - else - return clib_error_return (0, "unknown input `%U'", - format_unformat_error, line_input); - } - unformat_free (line_input); - -start_server: - - if (hsm->app_index != (u32) ~0) - return clib_error_return (0, "test http server is already running"); - - vnet_session_enable_disable (vm, 1 /* turn on TCP, etc. */ ); - - if (hsm->is_static) - { - http_server_session_cb_vft.builtin_app_rx_callback = - http_server_rx_callback_static; - vec_free (static_html); - static_html = format (0, html_header_static); - } - - rv = http_server_create (vm); - switch (rv) - { - case 0: - break; - default: - return clib_error_return (0, "server_create returned %d", rv); - } - return 0; -} - -VLIB_CLI_COMMAND (http_server_create_command, static) = -{ - .path = "test http server", - .short_help = "test http server", - .function = http_server_create_command_fn, -}; - -static clib_error_t * -http_server_main_init (vlib_main_t * vm) -{ - http_server_main_t *hsm = &http_server_main; - - hsm->app_index = ~0; - hsm->vlib_main = vm; - return 0; -} - -VLIB_INIT_FUNCTION (http_server_main_init); - -/* -* fd.io coding-style-patch-verification: ON -* -* Local Variables: -* eval: (c-set-style "gnu") -* End: -*/ -- cgit 1.2.3-korg