http: client POST method

Type: improvement

Change-Id: Iaa70abcee02866f9a6426a6e8e4709eeba0e8114
Signed-off-by: Matus Fabian <matfabia@cisco.com>

1 files changed, 569 insertions, 0 deletions

diff --git a/src/plugins/hs_apps/http_simple_post.c b/src/plugins/hs_apps/http_simple_post.c
new file mode 100644
index 00000000000..b0b9a38db2d
--- /dev/null
+++ b/src/plugins/hs_apps/http_simple_post.c
@@ -0,0 +1,569 @@
+/* SPDX-License-Identifier: Apache-2.0
+ * Copyright(c) 2024 Cisco Systems, Inc.
+ */
+
+#include <vnet/session/application.h>
+#include <vnet/session/application_interface.h>
+#include <vnet/session/session.h>
+#include <http/http.h>
+#include <http/http_header_names.h>
+#include <http/http_content_types.h>
+#include <vppinfra/unix.h>
+
+typedef struct
+{
+  CLIB_CACHE_LINE_ALIGN_MARK (cacheline0);
+  u32 session_index;
+  u32 thread_index;
+  u32 vpp_session_index;
+  u8 is_closed;
+} hsp_session_t;
+
+typedef struct
+{
+  hsp_session_t *sessions;
+  u32 thread_index;
+} hsp_worker_t;
+
+typedef struct
+{
+  u32 app_index;
+  vlib_main_t *vlib_main;
+  u32 cli_node_index;
+  u8 attached;
+  u8 *uri;
+  session_endpoint_cfg_t connect_sep;
+  u8 *target;
+  u8 *headers_buf;
+  u8 *data;
+  u32 data_offset;
+  hsp_worker_t *wrk;
+  u8 *http_response;
+  u8 is_file;
+  u8 use_ptr;
+} hsp_main_t;
+
+typedef enum
+{
+  HSP_CONNECT_FAILED = 1,
+  HSP_TRANSPORT_CLOSED,
+  HSP_REPLY_RECEIVED,
+} hsp_cli_signal_t;
+
+static hsp_main_t hsp_main;
+
+static inline hsp_worker_t *
+hsp_worker_get (u32 thread_index)
+{
+  return &hsp_main.wrk[thread_index];
+}
+
+static inline hsp_session_t *
+hsp_session_get (u32 session_index, u32 thread_index)
+{
+  hsp_worker_t *wrk = hsp_worker_get (thread_index);
+  return pool_elt_at_index (wrk->sessions, session_index);
+}
+
+static hsp_session_t *
+hsp_session_alloc (hsp_worker_t *wrk)
+{
+  hsp_session_t *s;
+
+  pool_get_zero (wrk->sessions, s);
+  s->session_index = s - wrk->sessions;
+  s->thread_index = wrk->thread_index;
+
+  return s;
+}
+
+static int
+hsp_session_connected_callback (u32 app_index, u32 hsp_session_index,
+				session_t *s, session_error_t err)
+{
+  hsp_main_t *hspm = &hsp_main;
+  hsp_session_t *hsp_session, *new_hsp_session;
+  hsp_worker_t *wrk;
+  http_header_t *headers = 0;
+  http_msg_t msg;
+  int rv;
+
+  if (err)
+    {
+      clib_warning ("hsp_session_index[%d] connected error: %U",
+		    hsp_session_index, format_session_error, err);
+      vlib_process_signal_event_mt (hspm->vlib_main, hspm->cli_node_index,
+				    HSP_CONNECT_FAILED, 0);
+      return -1;
+    }
+
+  hsp_session = hsp_session_get (hsp_session_index, 0);
+  wrk = hsp_worker_get (s->thread_index);
+  new_hsp_session = hsp_session_alloc (wrk);
+  clib_memcpy_fast (new_hsp_session, hsp_session, sizeof (*hsp_session));
+  hsp_session->vpp_session_index = s->session_index;
+
+  if (hspm->is_file)
+    {
+      http_add_header (
+	&headers, http_header_name_token (HTTP_HEADER_CONTENT_TYPE),
+	http_content_type_token (HTTP_CONTENT_APP_OCTET_STREAM));
+    }
+  else
+    {
+      http_add_header (
+	&headers, http_header_name_token (HTTP_HEADER_CONTENT_TYPE),
+	http_content_type_token (HTTP_CONTENT_APP_X_WWW_FORM_URLENCODED));
+    }
+  hspm->headers_buf = http_serialize_headers (headers);
+  vec_free (headers);
+
+  msg.type = HTTP_MSG_REQUEST;
+  msg.method_type = HTTP_REQ_POST;
+  /* request target */
+  msg.data.target_form = HTTP_TARGET_ORIGIN_FORM;
+  msg.data.target_path_len = vec_len (hspm->target);
+  /* custom headers */
+  msg.data.headers_len = vec_len (hspm->headers_buf);
+  /* request body */
+  msg.data.body_len = vec_len (hspm->data);
+  /* total length */
+  msg.data.len =
+    msg.data.target_path_len + msg.data.headers_len + msg.data.body_len;
+
+  if (hspm->use_ptr)
+    {
+      uword target = pointer_to_uword (hspm->target);
+      uword headers = pointer_to_uword (hspm->headers_buf);
+      uword body = pointer_to_uword (hspm->data);
+      msg.data.type = HTTP_MSG_DATA_PTR;
+      svm_fifo_seg_t segs[4] = {
+	{ (u8 *) &msg, sizeof (msg) },
+	{ (u8 *) &target, sizeof (target) },
+	{ (u8 *) &headers, sizeof (headers) },
+	{ (u8 *) &body, sizeof (body) },
+      };
+
+      rv =
+	svm_fifo_enqueue_segments (s->tx_fifo, segs, 4, 0 /* allow partial */);
+      ASSERT (rv == (sizeof (msg) + sizeof (target) + sizeof (headers) +
+		     sizeof (body)));
+      goto done;
+    }
+
+  msg.data.type = HTTP_MSG_DATA_INLINE;
+  msg.data.target_path_offset = 0;
+  msg.data.headers_offset = msg.data.target_path_len;
+  msg.data.body_offset = msg.data.headers_offset + msg.data.headers_len;
+
+  rv = svm_fifo_enqueue (s->tx_fifo, sizeof (msg), (u8 *) &msg);
+  ASSERT (rv == sizeof (msg));
+
+  rv = svm_fifo_enqueue (s->tx_fifo, vec_len (hspm->target), hspm->target);
+  ASSERT (rv == vec_len (hspm->target));
+
+  rv = svm_fifo_enqueue (s->tx_fifo, vec_len (hspm->headers_buf),
+			 hspm->headers_buf);
+  ASSERT (rv == msg.data.headers_len);
+
+  rv = svm_fifo_enqueue (s->tx_fifo, vec_len (hspm->data), hspm->data);
+  if (rv != vec_len (hspm->data))
+    {
+      hspm->data_offset = (rv > 0) ? rv : 0;
+      svm_fifo_add_want_deq_ntf (s->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
+    }
+
+done:
+  if (svm_fifo_set_event (s->tx_fifo))
+    session_program_tx_io_evt (s->handle, SESSION_IO_EVT_TX);
+
+  return 0;
+}
+
+static void
+hsp_session_disconnect_callback (session_t *s)
+{
+  hsp_main_t *hspm = &hsp_main;
+  vnet_disconnect_args_t _a = { 0 }, *a = &_a;
+  int rv;
+
+  a->handle = session_handle (s);
+  a->app_index = hspm->app_index;
+  if ((rv = vnet_disconnect_session (a)))
+    clib_warning ("warning: disconnect returned: %U", format_session_error,
+		  rv);
+}
+
+static void
+hsp_session_transport_closed_callback (session_t *s)
+{
+  hsp_main_t *hspm = &hsp_main;
+
+  vlib_process_signal_event_mt (hspm->vlib_main, hspm->cli_node_index,
+				HSP_TRANSPORT_CLOSED, 0);
+}
+
+static void
+hsp_session_reset_callback (session_t *s)
+{
+  hsp_main_t *hspm = &hsp_main;
+  hsp_session_t *hsp_session;
+  vnet_disconnect_args_t _a = { 0 }, *a = &_a;
+  int rv;
+
+  hsp_session = hsp_session_get (s->opaque, s->thread_index);
+  hsp_session->is_closed = 1;
+
+  a->handle = session_handle (s);
+  a->app_index = hspm->app_index;
+  if ((rv = vnet_disconnect_session (a)))
+    clib_warning ("warning: disconnect returned: %U", format_session_error,
+		  rv);
+}
+
+static int
+hsp_rx_callback (session_t *s)
+{
+  hsp_main_t *hspm = &hsp_main;
+  hsp_session_t *hsp_session;
+  http_msg_t msg;
+  int rv;
+
+  hsp_session = hsp_session_get (s->opaque, s->thread_index);
+
+  if (hsp_session->is_closed)
+    {
+      clib_warning ("hsp_session_index[%d] is closed", s->opaque);
+      return -1;
+    }
+
+  rv = svm_fifo_dequeue (s->rx_fifo, sizeof (msg), (u8 *) &msg);
+  ASSERT (rv == sizeof (msg));
+
+  if (msg.type != HTTP_MSG_REPLY)
+    {
+      clib_warning ("unexpected msg type %d", msg.type);
+      return -1;
+    }
+
+  svm_fifo_dequeue_drop_all (s->rx_fifo);
+
+  if (msg.code == HTTP_STATUS_OK)
+    hspm->http_response = format (0, "request success");
+  else
+    hspm->http_response = format (0, "request failed");
+
+  hsp_session_disconnect_callback (s);
+  vlib_process_signal_event_mt (hspm->vlib_main, hspm->cli_node_index,
+				HSP_REPLY_RECEIVED, 0);
+  return 0;
+}
+
+static int
+hsp_tx_callback (session_t *s)
+{
+  hsp_main_t *hspm = &hsp_main;
+  u32 to_send;
+  int rv;
+
+  to_send = vec_len (hspm->data) - hspm->data_offset;
+  rv = svm_fifo_enqueue (s->tx_fifo, to_send, hspm->data + hspm->data_offset);
+
+  if (rv <= 0)
+    {
+      svm_fifo_add_want_deq_ntf (s->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
+      return 0;
+    }
+
+  if (rv < to_send)
+    {
+      hspm->data_offset += rv;
+      svm_fifo_add_want_deq_ntf (s->tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
+    }
+
+  if (svm_fifo_set_event (s->tx_fifo))
+    session_program_tx_io_evt (s->handle, SESSION_IO_EVT_TX);
+
+  return 0;
+}
+
+static session_cb_vft_t hsp_session_cb_vft = {
+  .session_connected_callback = hsp_session_connected_callback,
+  .session_disconnect_callback = hsp_session_disconnect_callback,
+  .session_transport_closed_callback = hsp_session_transport_closed_callback,
+  .session_reset_callback = hsp_session_reset_callback,
+  .builtin_app_rx_callback = hsp_rx_callback,
+  .builtin_app_tx_callback = hsp_tx_callback,
+};
+
+static clib_error_t *
+hsp_attach ()
+{
+  hsp_main_t *hspm = &hsp_main;
+  vnet_app_attach_args_t _a, *a = &_a;
+  u64 options[18];
+  int rv;
+
+  clib_memset (a, 0, sizeof (*a));
+  clib_memset (options, 0, sizeof (options));
+
+  a->api_client_index = APP_INVALID_INDEX;
+  a->name = format (0, "http_simple_post");
+  a->session_cb_vft = &hsp_session_cb_vft;
+  a->options = options;
+  a->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN;
+
+  if ((rv = vnet_application_attach (a)))
+    return clib_error_return (0, "attach returned: %U", format_session_error,
+			      rv);
+
+  hspm->app_index = a->app_index;
+  vec_free (a->name);
+  hspm->attached = 1;
+
+  return 0;
+}
+
+static int
+hsp_connect_rpc (void *rpc_args)
+{
+  vnet_connect_args_t *a = rpc_args;
+  int rv;
+
+  rv = vnet_connect (a);
+  if (rv)
+    clib_warning (0, "connect returned: %U", format_session_error, rv);
+
+  vec_free (a);
+  return rv;
+}
+
+static void
+hsp_connect ()
+{
+  hsp_main_t *hspm = &hsp_main;
+  vnet_connect_args_t *a = 0;
+  hsp_worker_t *wrk;
+  hsp_session_t *hsp_session;
+
+  vec_validate (a, 0);
+  clib_memset (a, 0, sizeof (a[0]));
+
+  clib_memcpy (&a->sep_ext, &hspm->connect_sep, sizeof (hspm->connect_sep));
+  a->app_index = hspm->app_index;
+
+  /* allocate http session on main thread */
+  wrk = hsp_worker_get (0);
+  hsp_session = hsp_session_alloc (wrk);
+  a->api_context = hsp_session->session_index;
+
+  session_send_rpc_evt_to_thread_force (transport_cl_thread (),
+					hsp_connect_rpc, a);
+}
+
+static clib_error_t *
+hsp_run (vlib_main_t *vm)
+{
+  hsp_main_t *hspm = &hsp_main;
+  vlib_thread_main_t *vtm = vlib_get_thread_main ();
+  u32 num_threads;
+  hsp_worker_t *wrk;
+  uword event_type, *event_data = 0;
+  clib_error_t *err;
+
+  num_threads = 1 /* main thread */ + vtm->n_threads;
+  vec_validate (hspm->wrk, num_threads);
+  vec_foreach (wrk, hspm->wrk)
+    wrk->thread_index = wrk - hspm->wrk;
+
+  if ((err = hsp_attach ()))
+    return clib_error_return (0, "http simple post attach: %U",
+			      format_clib_error, err);
+
+  hsp_connect ();
+
+  vlib_process_wait_for_event_or_clock (vm, 10);
+  event_type = vlib_process_get_events (vm, &event_data);
+  switch (event_type)
+    {
+    case ~0:
+      err = clib_error_return (0, "error: timeout");
+      break;
+    case HSP_CONNECT_FAILED:
+      err = clib_error_return (0, "error: failed to connect");
+      break;
+    case HSP_TRANSPORT_CLOSED:
+      err = clib_error_return (0, "error: transport closed");
+      break;
+    case HSP_REPLY_RECEIVED:
+      vlib_cli_output (vm, "%v", hspm->http_response);
+      break;
+    default:
+      err = clib_error_return (0, "error: unexpected event %d", event_type);
+      break;
+    }
+
+  vec_free (event_data);
+  return err;
+}
+
+static int
+hsp_detach ()
+{
+  hsp_main_t *hspm = &hsp_main;
+  vnet_app_detach_args_t _da, *da = &_da;
+  int rv;
+
+  if (!hspm->attached)
+    return 0;
+
+  da->app_index = hspm->app_index;
+  da->api_client_index = APP_INVALID_INDEX;
+  rv = vnet_application_detach (da);
+  hspm->attached = 0;
+  hspm->app_index = APP_INVALID_INDEX;
+
+  return rv;
+}
+
+static void
+hcc_worker_cleanup (hsp_worker_t *wrk)
+{
+  pool_free (wrk->sessions);
+}
+
+static void
+hsp_cleanup ()
+{
+  hsp_main_t *hspm = &hsp_main;
+  hsp_worker_t *wrk;
+
+  vec_foreach (wrk, hspm->wrk)
+    hcc_worker_cleanup (wrk);
+
+  vec_free (hspm->uri);
+  vec_free (hspm->target);
+  vec_free (hspm->headers_buf);
+  vec_free (hspm->data);
+  vec_free (hspm->http_response);
+  vec_free (hspm->wrk);
+}
+
+static clib_error_t *
+hsp_command_fn (vlib_main_t *vm, unformat_input_t *input,
+		vlib_cli_command_t *cmd)
+{
+  hsp_main_t *hspm = &hsp_main;
+  clib_error_t *err = 0;
+  unformat_input_t _line_input, *line_input = &_line_input;
+  u8 *path = 0;
+  u8 *file_data;
+  int rv;
+
+  if (hspm->attached)
+    return clib_error_return (0, "failed: already running!");
+
+  hspm->use_ptr = 0;
+
+  /* Get a line of input. */
+  if (!unformat_user (input, unformat_line_input, line_input))
+    return clib_error_return (0, "expected required arguments");
+
+  while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat (line_input, "uri %s", &hspm->uri))
+	;
+      else if (unformat (line_input, "data %v", &hspm->data))
+	hspm->is_file = 0;
+      else if (unformat (line_input, "target %s", &hspm->target))
+	;
+      else if (unformat (line_input, "file %s", &path))
+	hspm->is_file = 1;
+      else if (unformat (line_input, "use-ptr"))
+	hspm->use_ptr = 1;
+      else
+	{
+	  err = clib_error_return (0, "unknown input `%U'",
+				   format_unformat_error, line_input);
+	  goto done;
+	}
+    }
+
+  if (!hspm->uri)
+    {
+      err = clib_error_return (0, "URI not defined");
+      goto done;
+    }
+  if (!hspm->target)
+    {
+      err = clib_error_return (0, "target not defined");
+      goto done;
+    }
+  if (!hspm->data)
+    {
+      if (path)
+	{
+	  err = clib_file_contents ((char *) path, &file_data);
+	  if (err)
+	    goto done;
+	  hspm->data = file_data;
+	}
+      else
+	{
+	  err = clib_error_return (0, "data not defined");
+	  goto done;
+	}
+    }
+
+  if ((rv = parse_uri ((char *) hspm->uri, &hspm->connect_sep)))
+    {
+      err =
+	clib_error_return (0, "URI parse error: %U", format_session_error, rv);
+      goto done;
+    }
+
+  vlib_worker_thread_barrier_sync (vm);
+  vnet_session_enable_disable (vm, 1 /* turn on TCP, etc. */);
+  vlib_worker_thread_barrier_release (vm);
+
+  hspm->cli_node_index =
+    vlib_get_current_process (vm)->node_runtime.node_index;
+
+  err = hsp_run (vm);
+
+  if ((rv = hsp_detach ()))
+    {
+      /* don't override last error */
+      if (!err)
+	err = clib_error_return (0, "detach returned: %U",
+				 format_session_error, rv);
+      else
+	clib_warning ("warning: detach returned: %U", format_session_error,
+		      rv);
+    }
+
+done:
+  hsp_cleanup ();
+  unformat_free (line_input);
+  return err;
+}
+
+VLIB_CLI_COMMAND (hsp_command, static) = {
+  .path = "http post",
+  .short_help = "uri http://<ip-addr> target <origin-form> "
+		"[data <form-urlencoded> | file <file-path>] [use-ptr]",
+  .function = hsp_command_fn,
+  .is_mp_safe = 1,
+};
+
+static clib_error_t *
+hsp_main_init (vlib_main_t *vm)
+{
+  hsp_main_t *hspm = &hsp_main;
+
+  hspm->app_index = APP_INVALID_INDEX;
+  hspm->vlib_main = vm;
+  return 0;
+}
+
+VLIB_INIT_FUNCTION (hsp_main_init);
\ No newline at end of file