/* GStaticProxyResolver tests * * Copyright 2011, 2013 Red Hat, Inc. * * 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, see * . */ #include static void async_result_cb (GObject *obj, GAsyncResult *result, gpointer user_data) { GAsyncResult **result_out = user_data; *result_out = g_object_ref (result); } static void test_uris (void) { GProxyResolver *resolver; const gchar *ignore_hosts[2] = { "127.0.0.1", NULL }; gchar **proxies; GError *error = NULL; const gchar *uri; gchar *str = NULL; GAsyncResult *result = NULL; /* Valid URI. */ uri = "http://%E0%B4%A8%E0%B4%B2:80/"; resolver = g_simple_proxy_resolver_new (NULL, (char **) ignore_hosts); proxies = g_proxy_resolver_lookup (resolver, uri, NULL, &error); g_assert_no_error (error); g_strfreev (proxies); g_proxy_resolver_lookup_async (resolver, uri, NULL, async_result_cb, &result); while (result == NULL) g_main_context_iteration (NULL, TRUE); proxies = g_proxy_resolver_lookup_finish (resolver, result, &error); g_assert_no_error (error); g_strfreev (proxies); g_clear_object (&result); g_object_unref (resolver); /* Invalid URI. */ uri = "%E0%B4%A8%E0%B4%B2"; str = g_strdup_printf ("Invalid URI ā€˜%sā€™", uri); resolver = g_simple_proxy_resolver_new (NULL, (char **) ignore_hosts); proxies = g_proxy_resolver_lookup (resolver, uri, NULL, &error); g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT); g_assert_cmpstr (error->message, ==, str); g_clear_error (&error); g_assert_null (proxies); g_clear_object (&result); g_proxy_resolver_lookup_async (resolver, uri, NULL, async_result_cb, &result); while (result == NULL) g_main_context_iteration (NULL, TRUE); proxies = g_proxy_resolver_lookup_finish (resolver, result, &error); g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT); g_assert_cmpstr (error->message, ==, str); g_clear_error (&error); g_assert_null (proxies); g_object_unref (result); g_object_unref (resolver); g_free (str); resolver = g_simple_proxy_resolver_new ("default://", (char **) ignore_hosts); g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver), "http", "http://proxy.example.com"); g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver), "ftp", "ftp://proxy.example.com"); proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "http://proxy.example.com"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "HTTP://uppercase.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "http://proxy.example.com"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "htt://missing-letter.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "default://"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "https://extra-letter.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "default://"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "ftp://five.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "ftp://proxy.example.com"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "direct://"); g_strfreev (proxies); g_object_unref (resolver); } static void test_socks (void) { GProxyResolver *resolver; const gchar *ignore_hosts[2] = { "127.0.0.1", NULL }; gchar **proxies; GError *error = NULL; resolver = g_simple_proxy_resolver_new ("socks://proxy.example.com", (char **) ignore_hosts); proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 3); g_assert_cmpstr (proxies[0], ==, "socks5://proxy.example.com"); g_assert_cmpstr (proxies[1], ==, "socks4a://proxy.example.com"); g_assert_cmpstr (proxies[2], ==, "socks4://proxy.example.com"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "direct://"); g_strfreev (proxies); g_object_unref (resolver); resolver = g_simple_proxy_resolver_new ("default-proxy://", (char **) ignore_hosts); g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver), "http", "socks://proxy.example.com"); proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 3); g_assert_cmpstr (proxies[0], ==, "socks5://proxy.example.com"); g_assert_cmpstr (proxies[1], ==, "socks4a://proxy.example.com"); g_assert_cmpstr (proxies[2], ==, "socks4://proxy.example.com"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "ftp://two.example.com/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "default-proxy://"); g_strfreev (proxies); proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/", NULL, &error); g_assert_no_error (error); g_assert_cmpint (g_strv_length (proxies), ==, 1); g_assert_cmpstr (proxies[0], ==, "direct://"); g_strfreev (proxies); g_object_unref (resolver); } static const char *ignore_hosts[] = { ".bbb.xx", "*.ccc.xx", "ddd.xx", "*.eee.xx:8000", "127.0.0.0/24", "10.0.0.1:8000", "::1", "fe80::/10", NULL }; static const struct { const char *uri; const char *proxy; } ignore_tests[] = { { "http://aaa.xx/", "http://localhost:8080" }, { "http://aaa.xx:8000/", "http://localhost:8080" }, { "http://www.aaa.xx/", "http://localhost:8080" }, { "http://www.aaa.xx:8000/", "http://localhost:8080" }, { "https://aaa.xx/", "http://localhost:8080" }, { "http://bbb.xx/", "direct://" }, { "http://www.bbb.xx/", "direct://" }, { "http://bbb.xx:8000/", "direct://" }, { "http://www.bbb.xx:8000/", "direct://" }, { "https://bbb.xx/", "direct://" }, { "http://nobbb.xx/", "http://localhost:8080" }, { "http://www.nobbb.xx/", "http://localhost:8080" }, { "http://nobbb.xx:8000/", "http://localhost:8080" }, { "http://www.nobbb.xx:8000/", "http://localhost:8080" }, { "https://nobbb.xx/", "http://localhost:8080" }, { "http://ccc.xx/", "direct://" }, { "http://www.ccc.xx/", "direct://" }, { "http://ccc.xx:8000/", "direct://" }, { "http://www.ccc.xx:8000/", "direct://" }, { "https://ccc.xx/", "direct://" }, { "http://ddd.xx/", "direct://" }, { "http://ddd.xx:8000/", "direct://" }, { "http://www.ddd.xx/", "direct://" }, { "http://www.ddd.xx:8000/", "direct://" }, { "https://ddd.xx/", "direct://" }, { "http://eee.xx/", "http://localhost:8080" }, { "http://eee.xx:8000/", "direct://" }, { "http://www.eee.xx/", "http://localhost:8080" }, { "http://www.eee.xx:8000/", "direct://" }, { "https://eee.xx/", "http://localhost:8080" }, { "http://1.2.3.4/", "http://localhost:8080" }, { "http://127.0.0.1/", "direct://" }, { "http://127.0.0.2/", "direct://" }, { "http://127.0.0.255/", "direct://" }, { "http://127.0.1.0/", "http://localhost:8080" }, { "http://10.0.0.1/", "http://localhost:8080" }, { "http://10.0.0.1:8000/", "direct://" }, { "http://[::1]/", "direct://" }, { "http://[::1]:80/", "direct://" }, { "http://[::1:1]/", "http://localhost:8080" }, { "http://[::1:1]:80/", "http://localhost:8080" }, { "http://[fe80::1]/", "direct://" }, { "http://[fe80::1]:80/", "direct://" }, { "http://[fec0::1]/", "http://localhost:8080" }, { "http://[fec0::1]:80/", "http://localhost:8080" } }; static const int n_ignore_tests = G_N_ELEMENTS (ignore_tests); static void test_ignore (void) { GProxyResolver *resolver; GError *error = NULL; char **proxies; int i; resolver = g_simple_proxy_resolver_new ("http://localhost:8080", (char **)ignore_hosts); for (i = 0; i < n_ignore_tests; i++) { proxies = g_proxy_resolver_lookup (resolver, ignore_tests[i].uri, NULL, &error); g_assert_no_error (error); g_assert_cmpstr (proxies[0], ==, ignore_tests[i].proxy); g_strfreev (proxies); } g_object_unref (resolver); } int main (int argc, char *argv[]) { g_test_init (&argc, &argv, NULL); g_test_add_func ("/static-proxy/uri", test_uris); g_test_add_func ("/static-proxy/socks", test_socks); g_test_add_func ("/static-proxy/ignore", test_ignore); return g_test_run(); }