summaryrefslogtreecommitdiff
path: root/tests/px-manager-test.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/px-manager-test.c')
-rw-r--r--tests/px-manager-test.c314
1 files changed, 314 insertions, 0 deletions
diff --git a/tests/px-manager-test.c b/tests/px-manager-test.c
new file mode 100644
index 0000000..bfd3112
--- /dev/null
+++ b/tests/px-manager-test.c
@@ -0,0 +1,314 @@
+/* px-manager-test.c
+ *
+ * Copyright 2022-2023 The Libproxy Team
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ *
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ */
+
+#include "px-manager.h"
+#include "px-manager-helper.h"
+
+#include <libsoup/soup.h>
+
+#define SERVER_PORT 1983
+
+typedef struct {
+ GMainLoop *loop;
+ PxManager *manager;
+} Fixture;
+
+static void
+server_callback (SoupServer *server,
+ SoupServerMessage *msg,
+ const char *path,
+ GHashTable *query,
+ gpointer data)
+{
+ g_print ("%s: path %s\n", __FUNCTION__, path);
+ soup_server_message_set_status (SOUP_SERVER_MESSAGE (msg), SOUP_STATUS_OK, NULL);
+
+ if (g_strcmp0 (path, "/test.pac") == 0) {
+ g_autofree char *pac = g_test_build_filename (G_TEST_DIST, "data", "px-manager-sample.pac", NULL);
+ g_autofree char *pac_data = NULL;
+ g_autoptr (GError) error = NULL;
+ gsize len;
+
+ if (!g_file_get_contents (pac, &pac_data, &len, &error)) {
+ g_warning ("Could not read pac file: %s", error ? error->message : "");
+ return;
+ }
+ soup_server_message_set_response (msg, "text/plain", SOUP_MEMORY_COPY, pac_data, len);
+ }
+}
+
+static void
+fixture_setup (Fixture *fixture,
+ gconstpointer data)
+{
+ g_autofree char *path = NULL;
+
+ fixture->loop = g_main_loop_new (NULL, FALSE);
+
+ if (data)
+ path = g_test_build_filename (G_TEST_DIST, "data", data, NULL);
+
+ fixture->manager = px_test_manager_new ("config-sysconfig", path);
+}
+
+static void
+fixture_teardown (Fixture *fixture,
+ gconstpointer data)
+{
+ g_clear_object (&fixture->manager);
+}
+
+static gpointer
+download_pac (gpointer data)
+{
+ Fixture *self = data;
+ GBytes *pac;
+
+ pac = px_manager_pac_download (self->manager, "http://127.0.0.1:1983/test.pac");
+ g_assert_nonnull (pac);
+
+ g_main_loop_quit (self->loop);
+
+ return NULL;
+}
+
+static void
+test_pac_download (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GThread) thread = NULL;
+
+ thread = g_thread_new ("test", (GThreadFunc)download_pac, self);
+ g_main_loop_run (self->loop);
+}
+
+static void
+test_get_proxies_direct (Fixture *self,
+ const void *user_data)
+{
+ g_auto (GStrv) config = NULL;
+
+ config = px_manager_get_proxies_sync (self->manager, "", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "direct://");
+
+ config = px_manager_get_proxies_sync (self->manager, "nonsense", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "direct://");
+
+ config = px_manager_get_proxies_sync (self->manager, "https://www.example.com", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "direct://");
+}
+
+static void
+test_get_proxies_nonpac (Fixture *self,
+ const void *user_data)
+{
+ g_auto (GStrv) config = NULL;
+
+ config = px_manager_get_proxies_sync (self->manager, "https://www.example.com", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "http://127.0.0.1:1983");
+}
+
+static gpointer
+get_proxies_pac (gpointer data)
+{
+ Fixture *self = data;
+ g_auto (GStrv) config = NULL;
+
+ config = px_manager_get_proxies_sync (self->manager, "https://www.example.com", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "http://127.0.0.1:1983");
+
+ config = px_manager_get_proxies_sync (self->manager, "https://192.168.10.4", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "socks://127.0.0.1:1983");
+
+ config = px_manager_get_proxies_sync (self->manager, "https://192.168.10.5", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "socks4://127.0.0.1:1983");
+
+ config = px_manager_get_proxies_sync (self->manager, "https://192.168.10.6", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "socks4a://127.0.0.1:1983");
+
+ config = px_manager_get_proxies_sync (self->manager, "https://192.168.10.7", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "socks5://127.0.0.1:1983");
+
+ g_main_loop_quit (self->loop);
+
+ return NULL;
+}
+
+static void
+test_get_proxies_pac (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GThread) thread = NULL;
+
+ thread = g_thread_new ("test", (GThreadFunc)get_proxies_pac, self);
+ g_main_loop_run (self->loop);
+}
+
+static gpointer
+get_wpad (gpointer data)
+{
+ Fixture *self = data;
+ g_auto (GStrv) config = NULL;
+
+ config = px_manager_get_proxies_sync (self->manager, "https://www.example.com", NULL);
+ g_assert_nonnull (config);
+ g_assert_cmpstr (config[0], ==, "direct://");
+
+ g_main_loop_quit (self->loop);
+
+ return NULL;
+}
+
+static void
+test_get_wpad (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GThread) thread = NULL;
+
+ thread = g_thread_new ("test", (GThreadFunc)get_wpad, self);
+ g_main_loop_run (self->loop);
+}
+
+static void
+test_ignore_domain (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GUri) uri = g_uri_parse("http://10.10.1.12", G_URI_FLAGS_PARSE_RELAXED, NULL);
+ char **ignore_list = g_malloc0 (sizeof (char *) * 2);
+ gboolean ret;
+
+ /* Invalid test */
+ ignore_list[0] = g_strdup ("10.10.1.13");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_assert_false (ret);
+
+ /* Valid test */
+ ignore_list[0] = g_strdup ("10.10.1.12");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_assert_true (ret);
+
+ g_free (ignore_list[0]);
+ g_free (ignore_list);
+}
+
+static void
+test_ignore_domain_port (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GUri) uri = g_uri_parse("http://10.10.1.12:22", G_URI_FLAGS_PARSE_RELAXED, NULL);
+ char **ignore_list = g_malloc0 (sizeof (char *) * 2);
+ gboolean ret;
+
+ /* Invalid test */
+ ignore_list[0] = g_strdup ("10.10.1.13");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_free (ignore_list[0]);
+ g_assert_false (ret);
+
+ /* Invalid test */
+ ignore_list[0] = g_strdup ("10.10.1.12:24");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_free (ignore_list[0]);
+ g_assert_false (ret);
+
+ /* Valid test */
+ ignore_list[0] = g_strdup ("10.10.1.12:22");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_assert_true (ret);
+
+ g_free (ignore_list[0]);
+ g_free (ignore_list);
+}
+
+static void
+test_ignore_hostname (Fixture *self,
+ const void *user_data)
+{
+ g_autoptr (GUri) uri = g_uri_parse("http://18.10.1.12", G_URI_FLAGS_PARSE_RELAXED, NULL);
+ char **ignore_list = g_malloc0 (sizeof (char *) * 2);
+ gboolean ret;
+
+ /* Invalid test */
+ ignore_list[0] = g_strdup ("<local>");
+ ignore_list[1] = NULL;
+
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_assert_false (ret);
+ g_uri_unref (uri);
+
+ /* Valid test */
+ uri = g_uri_parse("http://127.0.0.1", G_URI_FLAGS_PARSE_RELAXED, NULL);
+ ret = px_manager_is_ignore (uri, ignore_list);
+ g_assert_false (ret);
+
+ g_free (ignore_list[0]);
+ g_free (ignore_list);
+}
+
+int
+main (int argc,
+ char **argv)
+{
+ SoupServer *server = NULL;
+ g_autoptr (GError) error = NULL;
+
+ g_test_init (&argc, &argv, NULL);
+
+ server = soup_server_new (NULL, NULL);
+ if (!soup_server_listen_local (server, SERVER_PORT, SOUP_SERVER_LISTEN_IPV4_ONLY, &error)) {
+ g_warning ("Could not create local server: %s", error ? error->message : "");
+ return -1;
+ }
+
+ soup_server_add_handler (server, NULL, server_callback, NULL, NULL);
+
+ g_test_add ("/pac/download", Fixture, "px-manager-direct", fixture_setup, test_pac_download, fixture_teardown);
+ g_test_add ("/pac/get_proxies_direct", Fixture, "px-manager-direct", fixture_setup, test_get_proxies_direct, fixture_teardown);
+ g_test_add ("/pac/get_proxies_nonpac", Fixture, "px-manager-nonpac", fixture_setup, test_get_proxies_nonpac, fixture_teardown);
+ g_test_add ("/pac/get_proxies_pac", Fixture, "px-manager-pac", fixture_setup, test_get_proxies_pac, fixture_teardown);
+ g_test_add ("/pac/wpad", Fixture, "px-manager-wpad", fixture_setup, test_get_wpad, fixture_teardown);
+
+ g_test_add ("/ignore/domain", Fixture, "px-manager-ignore", fixture_setup, test_ignore_domain, fixture_teardown);
+ g_test_add ("/ignore/domain_port", Fixture, "px-manager-ignore", fixture_setup, test_ignore_domain_port, fixture_teardown);
+ g_test_add ("/ignore/hostname", Fixture, "px-manager-ignore", fixture_setup, test_ignore_hostname, fixture_teardown);
+
+ return g_test_run ();
+}
+
View Lorry Controller Status