summaryrefslogtreecommitdiff
path: root/tests/server-test.c
diff options
context:
space:
mode:
authorSeonah Moon <seonah1.moon@samsung.com>2019-12-09 14:52:00 +0900
committerSeonah Moon <seonah1.moon@samsung.com>2019-12-09 14:52:24 +0900
commit58894334cd3f0b89c865304e8e9d6eea3cd20a55 (patch)
tree6fdb22df40c774e8db791bc7cd27b39680a12fb0 /tests/server-test.c
parent9f52ffa8b526e717e324fe0835ab794478650f11 (diff)
downloadlibsoup-58894334cd3f0b89c865304e8e9d6eea3cd20a55.tar.gz
libsoup-58894334cd3f0b89c865304e8e9d6eea3cd20a55.tar.bz2
libsoup-58894334cd3f0b89c865304e8e9d6eea3cd20a55.zip
Imported Upstream version 2.62.2upstream/2.62.2
Change-Id: Id151dafaa6ac9f569d76f08282b5c9e4c3b19621
Diffstat (limited to 'tests/server-test.c')
-rw-r--r--tests/server-test.c1150
1 files changed, 1087 insertions, 63 deletions
diff --git a/tests/server-test.c b/tests/server-test.c
index 0c980908..cf132b33 100644
--- a/tests/server-test.c
+++ b/tests/server-test.c
@@ -5,8 +5,13 @@
#include "test-utils.h"
-SoupServer *server, *ssl_server;
-SoupURI *base_uri, *ssl_base_uri;
+#include <gio/gnetworking.h>
+
+typedef struct {
+ SoupServer *server;
+ SoupURI *base_uri, *ssl_base_uri;
+ GSList *handlers;
+} ServerData;
static void
server_callback (SoupServer *server, SoupMessage *msg,
@@ -33,6 +38,58 @@ server_callback (SoupServer *server, SoupMessage *msg,
}
static void
+server_setup_nohandler (ServerData *sd, gconstpointer test_data)
+{
+ sd->server = soup_test_server_new (SOUP_TEST_SERVER_IN_THREAD);
+ sd->base_uri = soup_test_server_get_uri (sd->server, "http", NULL);
+ if (tls_available)
+ sd->ssl_base_uri = soup_test_server_get_uri (sd->server, "https", NULL);
+}
+
+static void
+server_add_handler (ServerData *sd,
+ const char *path,
+ SoupServerCallback callback,
+ gpointer user_data,
+ GDestroyNotify destroy)
+{
+ soup_server_add_handler (sd->server, path, callback, user_data, destroy);
+ sd->handlers = g_slist_prepend (sd->handlers, g_strdup (path));
+}
+
+static void
+server_add_early_handler (ServerData *sd,
+ const char *path,
+ SoupServerCallback callback,
+ gpointer user_data,
+ GDestroyNotify destroy)
+{
+ soup_server_add_early_handler (sd->server, path, callback, user_data, destroy);
+ sd->handlers = g_slist_prepend (sd->handlers, g_strdup (path));
+}
+
+static void
+server_setup (ServerData *sd, gconstpointer test_data)
+{
+ server_setup_nohandler (sd, test_data);
+ server_add_handler (sd, NULL, server_callback, NULL, NULL);
+}
+
+static void
+server_teardown (ServerData *sd, gconstpointer test_data)
+{
+ GSList *iter;
+
+ for (iter = sd->handlers; iter; iter = iter->next)
+ soup_server_remove_handler (sd->server, iter->data);
+ g_slist_free_full (sd->handlers, g_free);
+
+ g_clear_pointer (&sd->server, soup_test_server_quit_unref);
+ g_clear_pointer (&sd->base_uri, soup_uri_free);
+ g_clear_pointer (&sd->ssl_base_uri, soup_uri_free);
+}
+
+static void
server_star_callback (SoupServer *server, SoupMessage *msg,
const char *path, GHashTable *query,
SoupClientContext *context, gpointer data)
@@ -58,7 +115,7 @@ server_star_callback (SoupServer *server, SoupMessage *msg,
* all other URIs. #590751
*/
static void
-do_star_test (void)
+do_star_test (ServerData *sd, gconstpointer test_data)
{
SoupSession *session;
SoupMessage *msg;
@@ -68,7 +125,7 @@ do_star_test (void)
g_test_bug ("590751");
session = soup_test_session_new (SOUP_TYPE_SESSION_SYNC, NULL);
- star_uri = soup_uri_copy (base_uri);
+ star_uri = soup_uri_copy (sd->base_uri);
soup_uri_set_path (star_uri, "*");
debug_printf (1, " Testing with no handler\n");
@@ -81,7 +138,7 @@ do_star_test (void)
g_assert_cmpstr (handled_by, ==, NULL);
g_object_unref (msg);
- soup_server_add_handler (server, "*", server_star_callback, NULL, NULL);
+ server_add_handler (sd, "*", server_star_callback, NULL, NULL);
debug_printf (1, " Testing with handler\n");
msg = soup_message_new_from_uri ("OPTIONS", star_uri);
@@ -169,8 +226,10 @@ do_one_server_aliases_test (SoupURI *uri,
}
static void
-do_server_aliases_test (void)
+do_server_aliases_test (ServerData *sd, gconstpointer test_data)
{
+ char *http_aliases[] = { "dav", NULL };
+ char *https_aliases[] = { "davs", NULL };
char *http_good[] = { "http", "dav", NULL };
char *http_bad[] = { "https", "davs", "fred", NULL };
char *https_good[] = { "https", "davs", NULL };
@@ -179,21 +238,26 @@ do_server_aliases_test (void)
g_test_bug ("703694");
+ g_object_set (G_OBJECT (sd->server),
+ SOUP_SERVER_HTTP_ALIASES, http_aliases,
+ SOUP_SERVER_HTTPS_ALIASES, https_aliases,
+ NULL);
+
for (i = 0; http_good[i]; i++)
- do_one_server_aliases_test (base_uri, http_good[i], TRUE);
+ do_one_server_aliases_test (sd->base_uri, http_good[i], TRUE);
for (i = 0; http_bad[i]; i++)
- do_one_server_aliases_test (base_uri, http_bad[i], FALSE);
+ do_one_server_aliases_test (sd->base_uri, http_bad[i], FALSE);
if (tls_available) {
for (i = 0; https_good[i]; i++)
- do_one_server_aliases_test (ssl_base_uri, https_good[i], TRUE);
+ do_one_server_aliases_test (sd->ssl_base_uri, https_good[i], TRUE);
for (i = 0; https_bad[i]; i++)
- do_one_server_aliases_test (ssl_base_uri, https_bad[i], FALSE);
+ do_one_server_aliases_test (sd->ssl_base_uri, https_bad[i], FALSE);
}
}
static void
-do_dot_dot_test (void)
+do_dot_dot_test (ServerData *sd, gconstpointer test_data)
{
SoupSession *session;
SoupMessage *msg;
@@ -203,7 +267,7 @@ do_dot_dot_test (void)
session = soup_test_session_new (SOUP_TYPE_SESSION_SYNC, NULL);
- uri = soup_uri_new_with_base (base_uri, "/..%2ftest");
+ uri = soup_uri_new_with_base (sd->base_uri, "/..%2ftest");
msg = soup_message_new_from_uri ("GET", uri);
soup_uri_free (uri);
@@ -220,10 +284,13 @@ ipv6_server_callback (SoupServer *server, SoupMessage *msg,
SoupClientContext *context, gpointer data)
{
const char *host;
+ GSocketAddress *addr;
char expected_host[128];
+ addr = soup_client_context_get_local_address (context);
g_snprintf (expected_host, sizeof (expected_host),
- "[::1]:%d", soup_server_get_port (server));
+ "[::1]:%d",
+ g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)));
host = soup_message_headers_get_one (msg->request_headers, "Host");
g_assert_cmpstr (host, ==, expected_host);
@@ -235,95 +302,1052 @@ ipv6_server_callback (SoupServer *server, SoupMessage *msg,
}
static void
-do_ipv6_test (void)
+do_ipv6_test (ServerData *sd, gconstpointer test_data)
{
- SoupServer *ipv6_server;
- SoupURI *ipv6_uri;
- SoupAddress *ipv6_addr;
SoupSession *session;
SoupMessage *msg;
+ GError *error = NULL;
g_test_bug ("666399");
- ipv6_addr = soup_address_new ("::1", SOUP_ADDRESS_ANY_PORT);
- soup_address_resolve_sync (ipv6_addr, NULL);
- ipv6_server = soup_server_new (SOUP_SERVER_INTERFACE, ipv6_addr,
- NULL);
- g_object_unref (ipv6_addr);
- if (!ipv6_server) {
- debug_printf (1, " skipping due to lack of IPv6 support\n");
+ sd->server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+ server_add_handler (sd, NULL, ipv6_server_callback, NULL, NULL);
+
+ if (!soup_server_listen_local (sd->server, 0,
+ SOUP_SERVER_LISTEN_IPV6_ONLY,
+ &error)) {
+#if GLIB_CHECK_VERSION (2, 41, 0)
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED);
+#endif
+ g_test_skip ("no IPv6 support");
return;
}
- soup_server_add_handler (ipv6_server, NULL, ipv6_server_callback, NULL, NULL);
- soup_server_run_async (ipv6_server);
-
- ipv6_uri = soup_uri_new ("http://[::1]/");
- soup_uri_set_port (ipv6_uri, soup_server_get_port (ipv6_server));
+ sd->base_uri = soup_test_server_get_uri (sd->server, "http", "::1");
session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
debug_printf (1, " HTTP/1.1\n");
- msg = soup_message_new_from_uri ("GET", ipv6_uri);
+ msg = soup_message_new_from_uri ("GET", sd->base_uri);
soup_session_send_message (session, msg);
soup_test_assert_message_status (msg, SOUP_STATUS_OK);
g_object_unref (msg);
debug_printf (1, " HTTP/1.0\n");
- msg = soup_message_new_from_uri ("GET", ipv6_uri);
+ msg = soup_message_new_from_uri ("GET", sd->base_uri);
soup_message_set_http_version (msg, SOUP_HTTP_1_0);
soup_session_send_message (session, msg);
soup_test_assert_message_status (msg, SOUP_STATUS_OK);
g_object_unref (msg);
- soup_uri_free (ipv6_uri);
soup_test_session_abort_unref (session);
- soup_test_server_quit_unref (ipv6_server);
}
-int
-main (int argc, char **argv)
+static void
+multi_server_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
{
- char *http_aliases[] = { "dav", NULL };
- char *https_aliases[] = { "davs", NULL };
- int ret;
+ GSocketAddress *addr;
+ GInetSocketAddress *iaddr;
+ SoupURI *uri;
+ char *uristr, *addrstr;
- test_init (argc, argv, NULL);
+ addr = soup_client_context_get_local_address (context);
+ iaddr = G_INET_SOCKET_ADDRESS (addr);
- server = soup_test_server_new (TRUE);
- soup_server_add_handler (server, NULL, server_callback, NULL, NULL);
- base_uri = soup_uri_new ("http://127.0.0.1/");
- soup_uri_set_port (base_uri, soup_server_get_port (server));
+ uri = soup_message_get_uri (msg);
+ uristr = soup_uri_to_string (uri, FALSE);
- g_object_set (G_OBJECT (server),
- SOUP_SERVER_HTTP_ALIASES, http_aliases,
- NULL);
+ addrstr = g_inet_address_to_string (g_inet_socket_address_get_address (iaddr));
+ g_assert_cmpstr (addrstr, ==, uri->host);
+ g_free (addrstr);
- if (tls_available) {
- ssl_server = soup_test_server_new_ssl (TRUE);
- soup_server_add_handler (ssl_server, NULL, server_callback, NULL, NULL);
- ssl_base_uri = soup_uri_new ("https://127.0.0.1/");
- soup_uri_set_port (ssl_base_uri, soup_server_get_port (ssl_server));
- g_object_set (G_OBJECT (ssl_server),
- SOUP_SERVER_HTTPS_ALIASES, https_aliases,
- NULL);
+ g_assert_cmpint (g_inet_socket_address_get_port (iaddr), ==, uri->port);
+
+ /* FIXME ssl */
+
+ soup_message_set_response (msg, "text/plain",
+ SOUP_MEMORY_TAKE, uristr, strlen (uristr));
+ soup_message_set_status (msg, SOUP_STATUS_OK);
+}
+
+static void
+do_multi_test (ServerData *sd, SoupURI *uri1, SoupURI *uri2)
+{
+ char *uristr;
+ SoupSession *session;
+ SoupMessage *msg;
+
+ server_add_handler (sd, NULL, multi_server_callback, NULL, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+
+ uristr = soup_uri_to_string (uri1, FALSE);
+ msg = soup_message_new ("GET", uristr);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ g_assert_cmpstr (msg->response_body->data, ==, uristr);
+ g_object_unref (msg);
+ g_free (uristr);
+
+ uristr = soup_uri_to_string (uri2, FALSE);
+ msg = soup_message_new ("GET", uristr);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ g_assert_cmpstr (msg->response_body->data, ==, uristr);
+ g_object_unref (msg);
+ g_free (uristr);
+
+ soup_test_session_abort_unref (session);
+
+ soup_uri_free (uri1);
+ soup_uri_free (uri2);
+}
+
+static void
+do_multi_port_test (ServerData *sd, gconstpointer test_data)
+{
+ GSList *uris;
+ SoupURI *uri1, *uri2;
+ GError *error = NULL;
+
+ sd->server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+
+ if (!soup_server_listen_local (sd->server, 0, SOUP_SERVER_LISTEN_IPV4_ONLY, &error)) {
+ g_assert_no_error (error);
+ g_error_free (error);
+ return;
+ }
+ if (!soup_server_listen_local (sd->server, 0, SOUP_SERVER_LISTEN_IPV4_ONLY, &error)) {
+ g_assert_no_error (error);
+ g_error_free (error);
+ return;
}
- g_test_add_func ("/server/OPTIONS *", do_star_test);
- g_test_add_func ("/server/aliases", do_server_aliases_test);
- g_test_add_func ("/server/..-in-path", do_dot_dot_test);
- g_test_add_func ("/server/ipv6", do_ipv6_test);
+ uris = soup_server_get_uris (sd->server);
+ g_assert_cmpint (g_slist_length (uris), ==, 2);
+ uri1 = uris->data;
+ uri2 = uris->next->data;
+ g_slist_free (uris);
- ret = g_test_run ();
+ g_assert_cmpint (uri1->port, !=, uri2->port);
+
+ do_multi_test (sd, uri1, uri2);
+}
+
+static void
+do_multi_scheme_test (ServerData *sd, gconstpointer test_data)
+{
+ GSList *uris;
+ SoupURI *uri1, *uri2;
+ GError *error = NULL;
+
+ SOUP_TEST_SKIP_IF_NO_TLS;
+
+ sd->server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+
+ if (!soup_server_listen_local (sd->server, 0, SOUP_SERVER_LISTEN_IPV4_ONLY, &error)) {
+ g_assert_no_error (error);
+ g_error_free (error);
+ return;
+ }
+ if (!soup_server_listen_local (sd->server, 0,
+ SOUP_SERVER_LISTEN_IPV4_ONLY | SOUP_SERVER_LISTEN_HTTPS,
+ &error)) {
+ g_assert_no_error (error);
+ g_error_free (error);
+ return;
+ }
+
+ uris = soup_server_get_uris (sd->server);
+ g_assert_cmpint (g_slist_length (uris), ==, 2);
+ uri1 = uris->data;
+ uri2 = uris->next->data;
+ g_slist_free (uris);
+
+ g_assert_cmpstr (uri1->scheme, !=, uri2->scheme);
+
+ do_multi_test (sd, uri1, uri2);
+}
+
+static void
+do_multi_family_test (ServerData *sd, gconstpointer test_data)
+{
+ GSList *uris;
+ SoupURI *uri1, *uri2;
+ GError *error = NULL;
+
+ sd->server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+
+ if (!soup_server_listen_local (sd->server, 0, 0, &error)) {
+ g_assert_no_error (error);
+ g_error_free (error);
+ return;
+ }
+
+ uris = soup_server_get_uris (sd->server);
+ if (g_slist_length (uris) == 1) {
+ gboolean ipv6_works;
+
+ /* No IPv6? Double-check */
+ ipv6_works = soup_server_listen_local (sd->server, 0,
+ SOUP_SERVER_LISTEN_IPV6_ONLY,
+ NULL);
+ if (ipv6_works)
+ g_assert_false (ipv6_works);
+ else
+ g_test_skip ("no IPv6 support");
+ return;
+ }
+
+ g_assert_cmpint (g_slist_length (uris), ==, 2);
+ uri1 = uris->data;
+ uri2 = uris->next->data;
+ g_slist_free (uris);
+
+ g_assert_cmpstr (uri1->host, !=, uri2->host);
+ g_assert_cmpint (uri1->port, ==, uri2->port);
+
+ do_multi_test (sd, uri1, uri2);
+}
+
+static void
+do_gsocket_import_test (void)
+{
+ GSocket *gsock;
+ GSocketAddress *gaddr;
+ SoupServer *server;
+ GSList *listeners;
+ SoupURI *uri;
+ SoupSession *session;
+ SoupMessage *msg;
+ GError *error = NULL;
+
+ gsock = g_socket_new (G_SOCKET_FAMILY_IPV4,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ g_assert_no_error (error);
+
+ gaddr = g_inet_socket_address_new_from_string ("127.0.0.1", 0);
+ g_socket_bind (gsock, gaddr, TRUE, &error);
+ g_object_unref (gaddr);
+ g_assert_no_error (error);
+ g_socket_listen (gsock, &error);
+ g_assert_no_error (error);
+
+ gaddr = g_socket_get_local_address (gsock, &error);
+ g_assert_no_error (error);
+ g_object_unref (gaddr);
+
+ server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+ soup_server_add_handler (server, NULL, server_callback, NULL, NULL);
+
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 0);
+ g_slist_free (listeners);
+
+ soup_server_listen_socket (server, gsock, 0, &error);
+ g_assert_no_error (error);
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 1);
+ g_slist_free (listeners);
+
+ uri = soup_test_server_get_uri (server, "http", "127.0.0.1");
+ g_assert_nonnull (uri);
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 1);
+ g_slist_free (listeners);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+ msg = soup_message_new_from_uri ("GET", uri);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ g_object_unref (msg);
+
+ soup_test_session_abort_unref (session);
+
+ soup_uri_free (uri);
+ soup_test_server_quit_unref (server);
+
+ g_assert_false (g_socket_is_connected (gsock));
+ g_object_unref (gsock);
+}
- soup_uri_free (base_uri);
+static void
+do_fd_import_test (void)
+{
+ GSocket *gsock;
+ GSocketAddress *gaddr;
+ SoupServer *server;
+ GSList *listeners;
+ SoupURI *uri;
+ SoupSession *session;
+ SoupMessage *msg;
+ int type;
+ GError *error = NULL;
+
+ gsock = g_socket_new (G_SOCKET_FAMILY_IPV4,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ g_assert_no_error (error);
+
+ gaddr = g_inet_socket_address_new_from_string ("127.0.0.1", 0);
+ g_socket_bind (gsock, gaddr, TRUE, &error);
+ g_object_unref (gaddr);
+ g_assert_no_error (error);
+ g_socket_listen (gsock, &error);
+ g_assert_no_error (error);
+
+ gaddr = g_socket_get_local_address (gsock, &error);
+ g_assert_no_error (error);
+ g_object_unref (gaddr);
+
+ server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+ soup_server_add_handler (server, NULL, server_callback, NULL, NULL);
+
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 0);
+ g_slist_free (listeners);
+
+ soup_server_listen_fd (server, g_socket_get_fd (gsock), 0, &error);
+ g_assert_no_error (error);
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 1);
+ g_slist_free (listeners);
+
+ uri = soup_test_server_get_uri (server, "http", "127.0.0.1");
+ g_assert_nonnull (uri);
+ listeners = soup_server_get_listeners (server);
+ g_assert_cmpint (g_slist_length (listeners), ==, 1);
+ g_slist_free (listeners);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+ msg = soup_message_new_from_uri ("GET", uri);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ g_object_unref (msg);
+
+ soup_test_session_abort_unref (session);
+
+ soup_uri_free (uri);
soup_test_server_quit_unref (server);
- if (tls_available) {
- soup_uri_free (ssl_base_uri);
- soup_test_server_quit_unref (ssl_server);
+ /* @server should have closed our socket, although @gsock doesn't
+ * know this.
+ */
+ g_socket_get_option (gsock, SOL_SOCKET, SO_TYPE, &type, &error);
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_FAILED);
+ g_clear_error (&error);
+ g_object_unref (gsock);
+}
+
+typedef struct
+{
+ GIOStream parent;
+ GInputStream *input_stream;
+ GOutputStream *output_stream;
+} GTestIOStream;
+
+typedef struct
+{
+ GIOStreamClass parent_class;
+} GTestIOStreamClass;
+
+static GType g_test_io_stream_get_type (void);
+G_DEFINE_TYPE (GTestIOStream, g_test_io_stream, G_TYPE_IO_STREAM);
+
+
+static GInputStream *
+get_input_stream (GIOStream *io_stream)
+{
+ GTestIOStream *self = (GTestIOStream *) io_stream;
+
+ return self->input_stream;
+}
+
+static GOutputStream *
+get_output_stream (GIOStream *io_stream)
+{
+ GTestIOStream *self = (GTestIOStream *) io_stream;
+
+ return self->output_stream;
+}
+
+static void
+finalize (GObject *object)
+{
+ GTestIOStream *self = (GTestIOStream *) object;
+
+ if (self->input_stream != NULL)
+ g_object_unref (self->input_stream);
+
+ if (self->output_stream != NULL)
+ g_object_unref (self->output_stream);
+
+ G_OBJECT_CLASS (g_test_io_stream_parent_class)->finalize (object);
+}
+
+static void
+g_test_io_stream_class_init (GTestIOStreamClass *klass)
+{
+ GObjectClass *object_class = G_OBJECT_CLASS (klass);
+ GIOStreamClass *io_class = G_IO_STREAM_CLASS (klass);
+
+ object_class->finalize = finalize;
+
+ io_class->get_input_stream = get_input_stream;
+ io_class->get_output_stream = get_output_stream;
+}
+
+static void
+g_test_io_stream_init (GTestIOStream *self)
+{
+}
+
+static GIOStream *
+g_test_io_stream_new (GInputStream *input, GOutputStream *output)
+{
+ GTestIOStream *self;
+
+ self = g_object_new (g_test_io_stream_get_type (), NULL);
+ self->input_stream = g_object_ref (input);
+ self->output_stream = g_object_ref (output);
+
+ return G_IO_STREAM (self);
+}
+
+static void
+mem_server_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ GSocketAddress *addr;
+ GSocket *sock;
+ const char *host;
+
+ addr = soup_client_context_get_local_address (context);
+ g_assert_nonnull (addr);
+
+ addr = soup_client_context_get_remote_address (context);
+ g_assert_nonnull (addr);
+
+ sock = soup_client_context_get_gsocket (context);
+ g_assert_null (sock);
+
+ host = soup_client_context_get_host (context);
+ g_assert_cmpstr (host, ==, "127.0.0.1");
+
+ server_callback (server, msg, path, query, context, data);
+}
+
+static void
+do_iostream_accept_test (void)
+{
+ GError *error = NULL;
+ SoupServer *server;
+ GInputStream *input;
+ GOutputStream *output;
+ GIOStream *stream;
+ GSocketAddress *addr;
+ const char req[] = "GET / HTTP/1.0\r\n\r\n";
+ gchar *reply;
+ gsize reply_size;
+
+ server = soup_test_server_new (SOUP_TEST_SERVER_NO_DEFAULT_LISTENER);
+ soup_server_add_handler (server, NULL, mem_server_callback, NULL, NULL);
+
+ input = g_memory_input_stream_new_from_data (req, sizeof(req), NULL);
+ output = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
+ stream = g_test_io_stream_new (input, output);
+
+ addr = g_inet_socket_address_new_from_string ("127.0.0.1", 0);
+
+ soup_server_accept_iostream (server, stream, addr, addr, &error);
+ g_assert_no_error (error);
+
+ soup_test_server_quit_unref (server);
+
+ reply = g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (output));
+ reply_size = g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (output));
+ g_assert_true (reply_size > 0);
+ g_assert_true (g_str_has_prefix (reply, "HTTP/1.0 200 OK"));
+
+ g_clear_object (&addr);
+ g_clear_object (&stream);
+ g_clear_object (&input);
+ g_clear_object (&output);
+ g_clear_error (&error);
+}
+
+typedef struct {
+ SoupServer *server;
+ SoupMessage *smsg;
+ gboolean handler_called;
+ gboolean paused;
+} UnhandledServerData;
+
+static gboolean
+idle_unpause_message (gpointer user_data)
+{
+ UnhandledServerData *usd = user_data;
+
+ soup_server_unpause_message (usd->server, usd->smsg);
+ return FALSE;
+}
+
+static void
+unhandled_server_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ UnhandledServerData *usd = data;
+
+ usd->handler_called = TRUE;
+
+ if (soup_message_headers_get_one (msg->request_headers, "X-Test-Server-Pause")) {
+ usd->paused = TRUE;
+ usd->server = server;
+ usd->smsg = msg;
+ soup_server_pause_message (server, msg);
+ g_idle_add (idle_unpause_message, usd);
+ }
+}
+
+static void
+do_fail_404_test (ServerData *sd, gconstpointer test_data)
+{
+ SoupSession *session;
+ SoupMessage *msg;
+ UnhandledServerData usd;
+
+ usd.handler_called = usd.paused = FALSE;
+
+ server_add_handler (sd, "/not-a-match", unhandled_server_callback, &usd, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+ msg = soup_message_new_from_uri ("GET", sd->base_uri);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_NOT_FOUND);
+ g_object_unref (msg);
+
+ g_assert_false (usd.handler_called);
+ g_assert_false (usd.paused);
+
+ soup_test_session_abort_unref (session);
+}
+
+static void
+do_fail_500_test (ServerData *sd, gconstpointer pause)
+{
+ SoupSession *session;
+ SoupMessage *msg;
+ UnhandledServerData usd;
+
+ usd.handler_called = usd.paused = FALSE;
+
+ server_add_handler (sd, NULL, unhandled_server_callback, &usd, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+ msg = soup_message_new_from_uri ("GET", sd->base_uri);
+ if (pause)
+ soup_message_headers_append (msg->request_headers, "X-Test-Server-Pause", "true");
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR);
+ g_object_unref (msg);
+
+ g_assert_true (usd.handler_called);
+ if (pause)
+ g_assert_true (usd.paused);
+ else
+ g_assert_false (usd.paused);
+
+ soup_test_session_abort_unref (session);
+}
+
+static void
+stream_got_chunk (SoupMessage *msg, SoupBuffer *chunk, gpointer user_data)
+{
+ GChecksum *checksum = user_data;
+
+ g_checksum_update (checksum, (const guchar *)chunk->data, chunk->length);
+}
+
+static void
+stream_got_body (SoupMessage *msg, gpointer user_data)
+{
+ GChecksum *checksum = user_data;
+ const char *md5 = g_checksum_get_string (checksum);
+
+ soup_message_set_status (msg, SOUP_STATUS_OK);
+ soup_message_set_response (msg, "text/plain", SOUP_MEMORY_COPY,
+ md5, strlen (md5));
+ g_checksum_free (checksum);
+}
+
+static void
+early_stream_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ GChecksum *checksum;
+
+ if (msg->method != SOUP_METHOD_POST) {
+ soup_message_set_status (msg, SOUP_STATUS_METHOD_NOT_ALLOWED);
+ return;
}
+ checksum = g_checksum_new (G_CHECKSUM_MD5);
+ g_signal_connect (msg, "got-chunk",
+ G_CALLBACK (stream_got_chunk), checksum);
+ g_signal_connect (msg, "got-body",
+ G_CALLBACK (stream_got_body), checksum);
+
+ soup_message_body_set_accumulate (msg->request_body, TRUE);
+}
+
+static void
+do_early_stream_test (ServerData *sd, gconstpointer test_data)
+{
+ SoupSession *session;
+ SoupMessage *msg;
+ SoupBuffer *index;
+ char *md5;
+
+ server_add_early_handler (sd, NULL, early_stream_callback, NULL, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_SYNC, NULL);
+
+ msg = soup_message_new_from_uri ("POST", sd->base_uri);
+
+ index = soup_test_get_index ();
+ soup_message_body_append (msg->request_body, SOUP_MEMORY_COPY,
+ index->data, index->length);
+ soup_session_send_message (session, msg);
+
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+
+ md5 = g_compute_checksum_for_data (G_CHECKSUM_MD5,
+ (guchar *) index->data, index->length);
+ g_assert_cmpstr (md5, ==, msg->response_body->data);
+ g_free (md5);
+
+ g_object_unref (msg);
+ soup_test_session_abort_unref (session);
+}
+
+static void
+early_respond_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ if (!strcmp (path, "/"))
+ soup_message_set_status (msg, SOUP_STATUS_FORBIDDEN);
+}
+
+static void
+do_early_respond_test (ServerData *sd, gconstpointer test_data)
+{
+ SoupSession *session;
+ SoupMessage *msg;
+ SoupURI *uri2;
+
+ server_add_early_handler (sd, NULL, early_respond_callback, NULL, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_SYNC, NULL);
+
+ /* The early handler will intercept, and the normal handler will be skipped */
+ msg = soup_message_new_from_uri ("GET", sd->base_uri);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_FORBIDDEN);
+ g_assert_cmpint (msg->response_body->length, ==, 0);
+ g_object_unref (msg);
+
+ /* The early handler will ignore this one */
+ uri2 = soup_uri_new_with_base (sd->base_uri, "/subdir");
+ msg = soup_message_new_from_uri ("GET", uri2);
+ soup_session_send_message (session, msg);
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ g_assert_cmpstr (msg->response_body->data, ==, "index");
+ g_object_unref (msg);
+ soup_uri_free (uri2);
+
+ soup_test_session_abort_unref (session);
+}
+
+static void
+early_multi_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ soup_message_headers_append (msg->response_headers, "X-Early", "yes");
+}
+
+static void
+do_early_multi_test (ServerData *sd, gconstpointer test_data)
+{
+ SoupSession *session;
+ SoupMessage *msg;
+ SoupURI *uri;
+ struct {
+ const char *path;
+ gboolean expect_normal, expect_early;
+ } multi_tests[] = {
+ { "/", FALSE, FALSE },
+ { "/normal", TRUE, FALSE },
+ { "/normal/subdir", TRUE, FALSE },
+ { "/normal/early", FALSE, TRUE },
+ { "/normal/early/subdir", FALSE, TRUE },
+ { "/early", FALSE, TRUE },
+ { "/early/subdir", FALSE, TRUE },
+ { "/early/normal", TRUE, FALSE },
+ { "/early/normal/subdir", TRUE, FALSE },
+ { "/both", TRUE, TRUE },
+ { "/both/subdir", TRUE, TRUE }
+ };
+ int i;
+ const char *header;
+
+ server_add_handler (sd, "/normal", server_callback, NULL, NULL);
+ server_add_early_handler (sd, "/normal/early", early_multi_callback, NULL, NULL);
+ server_add_early_handler (sd, "/early", early_multi_callback, NULL, NULL);
+ server_add_handler (sd, "/early/normal", server_callback, NULL, NULL);
+ server_add_handler (sd, "/both", server_callback, NULL, NULL);
+ server_add_early_handler (sd, "/both", early_multi_callback, NULL, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION_SYNC, NULL);
+
+ for (i = 0; i < G_N_ELEMENTS (multi_tests); i++) {
+ uri = soup_uri_new_with_base (sd->base_uri, multi_tests[i].path);
+ msg = soup_message_new_from_uri ("GET", uri);
+ soup_uri_free (uri);
+
+ soup_session_send_message (session, msg);
+
+ /* The normal handler sets status to OK. The early handler doesn't
+ * touch status, meaning that if it runs and the normal handler doesn't,
+ * then SoupServer will set the status to INTERNAL_SERVER_ERROR
+ * (since a handler ran, but didn't set the status). If neither handler
+ * runs then SoupServer will set the status to NOT_FOUND.
+ */
+ if (multi_tests[i].expect_normal)
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ else if (multi_tests[i].expect_early)
+ soup_test_assert_message_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR);
+ else
+ soup_test_assert_message_status (msg, SOUP_STATUS_NOT_FOUND);
+
+ header = soup_message_headers_get_one (msg->response_headers, "X-Early");
+ if (multi_tests[i].expect_early)
+ g_assert_cmpstr (header, ==, "yes");
+ else
+ g_assert_cmpstr (header, ==, NULL);
+ if (multi_tests[i].expect_normal)
+ g_assert_cmpstr (msg->response_body->data, ==, "index");
+ else
+ g_assert_cmpint (msg->response_body->length, ==, 0);
+
+ g_object_unref (msg);
+ }
+
+ soup_test_session_abort_unref (session);
+}
+
+typedef struct {
+ GIOStream *iostream;
+ GInputStream *istream;
+ GOutputStream *ostream;
+
+ gssize nread, nwrote;
+ guchar *buffer;
+} TunnelEnd;
+
+typedef struct {
+ SoupServer *self;
+ SoupMessage *msg;
+ SoupClientContext *context;
+ GCancellable *cancellable;
+
+ TunnelEnd client, server;
+} Tunnel;
+
+#define BUFSIZE 8192
+
+static void tunnel_read_cb (GObject *object,
+ GAsyncResult *result,
+ gpointer user_data);
+
+static void
+tunnel_close (Tunnel *tunnel)
+{
+ if (tunnel->cancellable) {
+ g_cancellable_cancel (tunnel->cancellable);
+ g_object_unref (tunnel->cancellable);
+ }
+
+ if (tunnel->client.iostream) {
+ g_io_stream_close (tunnel->client.iostream, NULL, NULL);
+ g_object_unref (tunnel->client.iostream);
+ }
+ if (tunnel->server.iostream) {
+ g_io_stream_close (tunnel->server.iostream, NULL, NULL);
+ g_object_unref (tunnel->server.iostream);
+ }
+
+ g_free (tunnel->client.buffer);
+ g_free (tunnel->server.buffer);
+
+ g_clear_object (&tunnel->self);
+ g_clear_object (&tunnel->msg);
+
+ g_free (tunnel);
+}
+
+static void
+tunnel_wrote_cb (GObject *object,
+ GAsyncResult *result,
+ gpointer user_data)
+{
+ Tunnel *tunnel = user_data;
+ TunnelEnd *write_end, *read_end;
+ GError *error = NULL;
+ gssize nwrote;
+
+ nwrote = g_output_stream_write_finish (G_OUTPUT_STREAM (object), result, &error);
+ if (nwrote <= 0) {
+ if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
+ g_error_free (error);
+ return;
+ } else if (error) {
+ g_print ("Tunnel write failed: %s\n", error->message);
+ g_error_free (error);
+ }
+ tunnel_close (tunnel);
+ return;
+ }
+
+ if (object == (GObject *)tunnel->client.ostream) {
+ write_end = &tunnel->client;
+ read_end = &tunnel->server;
+ } else {
+ write_end = &tunnel->server;
+ read_end = &tunnel->client;
+ }
+
+ write_end->nwrote += nwrote;
+ if (write_end->nwrote < read_end->nread) {
+ g_output_stream_write_async (write_end->ostream,
+ read_end->buffer + write_end->nwrote,
+ read_end->nread - write_end->nwrote,
+ G_PRIORITY_DEFAULT, tunnel->cancellable,
+ tunnel_wrote_cb, tunnel);
+ } else {
+ g_input_stream_read_async (read_end->istream,
+ read_end->buffer, BUFSIZE,
+ G_PRIORITY_DEFAULT, tunnel->cancellable,
+ tunnel_read_cb, tunnel);
+ }
+}
+
+static void
+tunnel_read_cb (GObject *object,
+ GAsyncResult *result,
+ gpointer user_data)
+{
+ Tunnel *tunnel = user_data;
+ TunnelEnd *read_end, *write_end;
+ GError *error = NULL;
+ gssize nread;
+
+ nread = g_input_stream_read_finish (G_INPUT_STREAM (object), result, &error);
+ if (nread <= 0) {
+ if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
+ g_error_free (error);
+ return;
+ } else if (error) {
+ g_print ("Tunnel read failed: %s\n", error->message);
+ g_error_free (error);
+ }
+ tunnel_close (tunnel);
+ return;
+ }
+
+ if (object == (GObject *)tunnel->client.istream) {
+ read_end = &tunnel->client;
+ write_end = &tunnel->server;
+ } else {
+ read_end = &tunnel->server;
+ write_end = &tunnel->client;
+ }
+
+ read_end->nread = nread;
+ write_end->nwrote = 0;
+ g_output_stream_write_async (write_end->ostream,
+ read_end->buffer, read_end->nread,
+ G_PRIORITY_DEFAULT, tunnel->cancellable,
+ tunnel_wrote_cb, tunnel);
+}
+
+static void
+start_tunnel (SoupMessage *msg, gpointer user_data)
+{
+ Tunnel *tunnel = user_data;
+
+ tunnel->client.iostream = soup_client_context_steal_connection (tunnel->context);
+ tunnel->client.istream = g_io_stream_get_input_stream (tunnel->client.iostream);
+ tunnel->client.ostream = g_io_stream_get_output_stream (tunnel->client.iostream);
+ g_clear_object (&tunnel->self);
+ g_clear_object (&tunnel->msg);
+
+ tunnel->client.buffer = g_malloc (BUFSIZE);
+ tunnel->server.buffer = g_malloc (BUFSIZE);
+
+ tunnel->cancellable = g_cancellable_new ();
+
+ g_input_stream_read_async (tunnel->client.istream,
+ tunnel->client.buffer, BUFSIZE,
+ G_PRIORITY_DEFAULT, tunnel->cancellable,
+ tunnel_read_cb, tunnel);
+ g_input_stream_read_async (tunnel->server.istream,
+ tunnel->server.buffer, BUFSIZE,
+ G_PRIORITY_DEFAULT, tunnel->cancellable,
+ tunnel_read_cb, tunnel);
+}
+
+
+static void
+tunnel_connected_cb (GObject *object,
+ GAsyncResult *result,
+ gpointer user_data)
+{
+ Tunnel *tunnel = user_data;
+ GError *error = NULL;
+
+ tunnel->server.iostream = (GIOStream *)
+ g_socket_client_connect_to_host_finish (G_SOCKET_CLIENT (object), result, &error);
+ if (!tunnel->server.iostream) {
+ soup_message_set_status (tunnel->msg, SOUP_STATUS_BAD_GATEWAY);
+ soup_message_set_response (tunnel->msg, "text/plain",
+ SOUP_MEMORY_COPY,
+ error->message, strlen (error->message));
+ g_error_free (error);
+ soup_server_unpause_message (tunnel->self, tunnel->msg);
+ tunnel_close (tunnel);
+ return;
+ }
+
+ tunnel->server.istream = g_io_stream_get_input_stream (tunnel->server.iostream);
+ tunnel->server.ostream = g_io_stream_get_output_stream (tunnel->server.iostream);
+
+ soup_message_set_status (tunnel->msg, SOUP_STATUS_OK);
+ soup_server_unpause_message (tunnel->self, tunnel->msg);
+ g_signal_connect (tunnel->msg, "wrote-body",
+ G_CALLBACK (start_tunnel), tunnel);
+}
+
+static void
+proxy_server_callback (SoupServer *server, SoupMessage *msg,
+ const char *path, GHashTable *query,
+ SoupClientContext *context, gpointer data)
+{
+ GSocketClient *sclient;
+ SoupURI *dest_uri;
+ Tunnel *tunnel;
+
+ if (msg->method != SOUP_METHOD_CONNECT) {
+ soup_message_set_status (msg, SOUP_STATUS_NOT_IMPLEMENTED);
+ return;
+ }
+
+ soup_server_pause_message (server, msg);
+
+ tunnel = g_new0 (Tunnel, 1);
+ tunnel->self = g_object_ref (server);
+ tunnel->msg = g_object_ref (msg);
+ tunnel->context = context;
+
+ dest_uri = soup_message_get_uri (msg);
+ sclient = g_socket_client_new ();
+ g_socket_client_connect_to_host_async (sclient, dest_uri->host, dest_uri->port,
+ NULL, tunnel_connected_cb, tunnel);
+ g_object_unref (sclient);
+}
+
+static void
+do_steal_connect_test (ServerData *sd, gconstpointer test_data)
+{
+ SoupServer *proxy;
+ SoupURI *proxy_uri;
+ SoupSession *session;
+ SoupMessage *msg;
+ const char *handled_by;
+
+ SOUP_TEST_SKIP_IF_NO_TLS;
+
+ proxy = soup_test_server_new (SOUP_TEST_SERVER_IN_THREAD);
+ proxy_uri = soup_test_server_get_uri (proxy, SOUP_URI_SCHEME_HTTP, "127.0.0.1");
+ soup_server_add_handler (proxy, NULL, proxy_server_callback, NULL, NULL);
+
+ session = soup_test_session_new (SOUP_TYPE_SESSION,
+ SOUP_SESSION_PROXY_URI, proxy_uri,
+ NULL);
+ msg = soup_message_new_from_uri ("GET", sd->ssl_base_uri);
+ soup_session_send_message (session, msg);
+
+ soup_test_assert_message_status (msg, SOUP_STATUS_OK);
+ handled_by = soup_message_headers_get_one (msg->response_headers, "X-Handled-By");
+ g_assert_cmpstr (handled_by, ==, "server_callback");
+
+ g_object_unref (msg);
+ soup_test_session_abort_unref (session);
+
+ soup_test_server_quit_unref (proxy);
+ soup_uri_free (proxy_uri);
+}
+
+int
+main (int argc, char **argv)
+{
+ int ret;
+
+ test_init (argc, argv, NULL);
+
+ g_test_add ("/server/OPTIONS *", ServerData, NULL,
+ server_setup, do_star_test, server_teardown);
+ g_test_add ("/server/aliases", ServerData, NULL,
+ server_setup, do_server_aliases_test, server_teardown);
+ g_test_add ("/server/..-in-path", ServerData, NULL,
+ server_setup, do_dot_dot_test, server_teardown);
+ g_test_add ("/server/ipv6", ServerData, NULL,
+ NULL, do_ipv6_test, server_teardown);
+ g_test_add ("/server/multi/port", ServerData, NULL,
+ NULL, do_multi_port_test, server_teardown);
+ g_test_add ("/server/multi/scheme", ServerData, NULL,
+ NULL, do_multi_scheme_test, server_teardown);
+ g_test_add ("/server/multi/family", ServerData, NULL,
+ NULL, do_multi_family_test, server_teardown);
+ g_test_add_func ("/server/import/gsocket", do_gsocket_import_test);
+ g_test_add_func ("/server/import/fd", do_fd_import_test);
+ g_test_add_func ("/server/accept/iostream", do_iostream_accept_test);
+ g_test_add ("/server/fail/404", ServerData, NULL,
+ server_setup_nohandler, do_fail_404_test, server_teardown);
+ g_test_add ("/server/fail/500", ServerData, GINT_TO_POINTER (FALSE),
+ server_setup_nohandler, do_fail_500_test, server_teardown);
+ g_test_add ("/server/fail/500-pause", ServerData, GINT_TO_POINTER (TRUE),
+ server_setup_nohandler, do_fail_500_test, server_teardown);
+ g_test_add ("/server/early/stream", ServerData, NULL,
+ server_setup_nohandler, do_early_stream_test, server_teardown);
+ g_test_add ("/server/early/respond", ServerData, NULL,
+ server_setup, do_early_respond_test, server_teardown);
+ g_test_add ("/server/early/multi", ServerData, NULL,
+ server_setup_nohandler, do_early_multi_test, server_teardown);
+ g_test_add ("/server/steal/CONNECT", ServerData, NULL,
+ server_setup, do_steal_connect_test, server_teardown);
+
+ ret = g_test_run ();
+
test_cleanup ();
return ret;
}