[libsoup/carlosgc/http2-early-hints: 3/4] get: add support for PUT and POST methods

[Carlos Garcia Campos <carlosgc src gnome org>]

commit 8e03778377dac2fdef5dcbdefc861e1fe4cf15ab
Author: Carlos Garcia Campos <cgarcia igalia com>
Date:   Tue Mar 22 16:27:55 2022 +0100

    get: add support for PUT and POST methods
    
    Add command line options to send data. Add also an option to ignore TLS
    errors to make it easier to test with local servers.

 examples/get.c | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 71 insertions(+), 4 deletions(-)
---
diff --git a/examples/get.c b/examples/get.c
index 657914ca..02c86f72 100644
--- a/examples/get.c
+++ b/examples/get.c
@@ -11,12 +11,19 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+#include <gio/gio.h>
+#ifdef G_OS_UNIX
+#include <gio/gunixinputstream.h>
+#endif
+
 #include <libsoup/soup.h>
 
 static SoupSession *session;
 static GMainLoop *loop;
-static gboolean debug, head, quiet;
-static const gchar *output_file_path = NULL;
+static gboolean debug, head, expect_continue, quiet, ignore_tls;
+static const gchar *method;
+static const gchar *output_file_path;
+static const gchar *input_file_path;
 
 #define OUTPUT_BUFFER_SIZE 8192
 
@@ -110,6 +117,14 @@ on_request_sent (GObject *source, GAsyncResult *result, gpointer user_data)
         g_object_unref (in);
 }
 
+static gboolean
+accept_certificate (SoupMessage         *msg,
+                    GTlsCertificate     *certificate,
+                    GTlsCertificateFlags errors)
+{
+        return TRUE;
+}
+
 static const char *ca_file, *proxy;
 static char *client_cert_file, *client_key_file;
 static char *user_agent;
@@ -126,21 +141,33 @@ static GOptionEntry entries[] = {
        { "key", 0, 0,
          G_OPTION_ARG_STRING, &client_key_file,
          "Use FILE as the TLS client key file", "FILE" },
+        { "ignore-tls", 0, 0,
+          G_OPTION_ARG_NONE, &ignore_tls,
+          "Ignore TLS certificate errors", NULL },
        { "debug", 'd', 0,
          G_OPTION_ARG_NONE, &debug,
          "Show HTTP headers", NULL },
         { "user-agent", 'u', 0,
           G_OPTION_ARG_STRING, &user_agent,
           "User agent string", "STRING" },
+        { "method", 'm', 0,
+          G_OPTION_ARG_STRING, &method,
+          "HTTP method to use", "STRING" },
        { "head", 'h', 0,
           G_OPTION_ARG_NONE, &head,
-          "Do HEAD rather than GET", NULL },
+          "Do HEAD rather than GET (equivalent to --method=HEAD)", NULL },
+        { "expect-continue", 0, 0,
+          G_OPTION_ARG_NONE, &expect_continue,
+          "Include Expects: 100-continue header in the request", NULL },
        { "ntlm", 'n', 0,
          G_OPTION_ARG_NONE, &ntlm,
          "Use NTLM authentication", NULL },
        { "output", 'o', 0,
          G_OPTION_ARG_STRING, &output_file_path,
          "Write the received data to FILE instead of stdout", "FILE" },
+        { "input", 'i', 0,
+          G_OPTION_ARG_STRING, &input_file_path,
+          "Read data from FILE when method is PUT or POST", "FILE" },
        { "proxy", 'p', 0,
          G_OPTION_ARG_STRING, &proxy,
          "Use URL as an HTTP proxy", "URL" },
@@ -264,8 +291,48 @@ main (int argc, char **argv)
                g_object_unref (resolver);
        }
 
+        if (!method)
+                method = head ? "HEAD" : "GET";
+        msg = soup_message_new (method, url);
+
+        if (g_strcmp0 (method, "PUT") == 0 || g_strcmp0 (method, "POST") == 0) {
+                GInputStream *stream;
+
+                if (input_file_path) {
+                        GFile *input_file = g_file_new_for_commandline_arg (input_file_path);
+                        stream = G_INPUT_STREAM (g_file_read (input_file, NULL, &error));
+                        if (!stream) {
+                                g_printerr ("Failed to open input file \"%s\": %s\n", input_file_path, 
error->message);
+                                g_error_free (error);
+                                g_object_unref (session);
+                                exit (1);
+                        }
+                } else {
+#ifdef G_OS_UNIX
+                        stream = g_unix_input_stream_new (0, FALSE);
+#else
+                        g_printerr ("Input file is required for method %s\n", method);
+                        g_object_unref (session);
+                        exit (1);
+#endif
+                }
+
+                soup_message_set_request_body (msg, NULL, stream, -1);
+                g_object_unref (stream);
+
+                if (expect_continue) {
+                        soup_message_headers_set_expectations (soup_message_get_request_headers (msg),
+                                                               SOUP_EXPECTATION_CONTINUE);
+                }
+        }
+
+        if (ignore_tls) {
+                g_signal_connect (msg, "accept-certificate",
+                                  G_CALLBACK (accept_certificate),
+                                  NULL);
+        }
+
         /* Send the request */
-       msg = soup_message_new (head ? "HEAD" : "GET", url);
         soup_message_set_tls_client_certificate (msg, client_cert);
         soup_session_send_async (session, msg, G_PRIORITY_DEFAULT, NULL,
                                 on_request_sent, NULL);