/* 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
* <http://www.gnu.org/licenses/>.
*/
#include <gio/gio.h>
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;
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, 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, 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 (resolver);
g_free (str);
resolver = g_simple_proxy_resolver_new ("default://", 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;
gchar *ignore_hosts[2] = { "127.0.0.1", NULL };
gchar **proxies;
GError *error = NULL;
resolver = g_simple_proxy_resolver_new ("socks://proxy.example.com", 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://", 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();
}