1 /* GStaticProxyResolver tests
2 *
3 * Copyright 2011, 2013 Red Hat, Inc.
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, see
17 * <http://www.gnu.org/licenses/>.
18 */
19
20 #include <gio/gio.h>
21
22 static void
async_result_cb(GObject * obj,GAsyncResult * result,gpointer user_data)23 async_result_cb (GObject *obj,
24 GAsyncResult *result,
25 gpointer user_data)
26 {
27 GAsyncResult **result_out = user_data;
28 *result_out = g_object_ref (result);
29 }
30
31 static void
test_uris(void)32 test_uris (void)
33 {
34 GProxyResolver *resolver;
35 const gchar *ignore_hosts[2] = { "127.0.0.1", NULL };
36 gchar **proxies;
37 GError *error = NULL;
38 const gchar *uri;
39 gchar *str = NULL;
40 GAsyncResult *result = NULL;
41
42 /* Valid URI. */
43 uri = "http://%E0%B4%A8%E0%B4%B2:80/";
44 resolver = g_simple_proxy_resolver_new (NULL, (char **) ignore_hosts);
45
46 proxies = g_proxy_resolver_lookup (resolver, uri, NULL, &error);
47 g_assert_no_error (error);
48 g_strfreev (proxies);
49
50 g_proxy_resolver_lookup_async (resolver, uri, NULL, async_result_cb, &result);
51 while (result == NULL)
52 g_main_context_iteration (NULL, TRUE);
53 proxies = g_proxy_resolver_lookup_finish (resolver, result, &error);
54 g_assert_no_error (error);
55 g_strfreev (proxies);
56 g_clear_object (&result);
57
58 g_object_unref (resolver);
59
60 /* Invalid URI. */
61 uri = "%E0%B4%A8%E0%B4%B2";
62 str = g_strdup_printf ("Invalid URI ‘%s’", uri);
63 resolver = g_simple_proxy_resolver_new (NULL, (char **) ignore_hosts);
64
65 proxies = g_proxy_resolver_lookup (resolver, uri, NULL, &error);
66 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT);
67 g_assert_cmpstr (error->message, ==, str);
68 g_clear_error (&error);
69 g_assert_null (proxies);
70 g_clear_object (&result);
71
72 g_proxy_resolver_lookup_async (resolver, uri, NULL, async_result_cb, &result);
73 while (result == NULL)
74 g_main_context_iteration (NULL, TRUE);
75 proxies = g_proxy_resolver_lookup_finish (resolver, result, &error);
76 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT);
77 g_assert_cmpstr (error->message, ==, str);
78 g_clear_error (&error);
79 g_assert_null (proxies);
80 g_object_unref (result);
81
82 g_object_unref (resolver);
83 g_free (str);
84
85 resolver = g_simple_proxy_resolver_new ("default://", (char **) ignore_hosts);
86 g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver),
87 "http", "http://proxy.example.com");
88 g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver),
89 "ftp", "ftp://proxy.example.com");
90
91 proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/",
92 NULL, &error);
93 g_assert_no_error (error);
94 g_assert_cmpint (g_strv_length (proxies), ==, 1);
95 g_assert_cmpstr (proxies[0], ==, "http://proxy.example.com");
96 g_strfreev (proxies);
97
98 proxies = g_proxy_resolver_lookup (resolver, "HTTP://uppercase.example.com/",
99 NULL, &error);
100 g_assert_no_error (error);
101 g_assert_cmpint (g_strv_length (proxies), ==, 1);
102 g_assert_cmpstr (proxies[0], ==, "http://proxy.example.com");
103 g_strfreev (proxies);
104
105 proxies = g_proxy_resolver_lookup (resolver, "htt://missing-letter.example.com/",
106 NULL, &error);
107 g_assert_no_error (error);
108 g_assert_cmpint (g_strv_length (proxies), ==, 1);
109 g_assert_cmpstr (proxies[0], ==, "default://");
110 g_strfreev (proxies);
111
112 proxies = g_proxy_resolver_lookup (resolver, "https://extra-letter.example.com/",
113 NULL, &error);
114 g_assert_no_error (error);
115 g_assert_cmpint (g_strv_length (proxies), ==, 1);
116 g_assert_cmpstr (proxies[0], ==, "default://");
117 g_strfreev (proxies);
118
119 proxies = g_proxy_resolver_lookup (resolver, "ftp://five.example.com/",
120 NULL, &error);
121 g_assert_no_error (error);
122 g_assert_cmpint (g_strv_length (proxies), ==, 1);
123 g_assert_cmpstr (proxies[0], ==, "ftp://proxy.example.com");
124 g_strfreev (proxies);
125
126 proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/",
127 NULL, &error);
128 g_assert_no_error (error);
129 g_assert_cmpint (g_strv_length (proxies), ==, 1);
130 g_assert_cmpstr (proxies[0], ==, "direct://");
131 g_strfreev (proxies);
132
133 g_object_unref (resolver);
134 }
135
136 static void
test_socks(void)137 test_socks (void)
138 {
139 GProxyResolver *resolver;
140 const gchar *ignore_hosts[2] = { "127.0.0.1", NULL };
141 gchar **proxies;
142 GError *error = NULL;
143
144 resolver = g_simple_proxy_resolver_new ("socks://proxy.example.com", (char **) ignore_hosts);
145
146 proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/",
147 NULL, &error);
148 g_assert_no_error (error);
149 g_assert_cmpint (g_strv_length (proxies), ==, 3);
150 g_assert_cmpstr (proxies[0], ==, "socks5://proxy.example.com");
151 g_assert_cmpstr (proxies[1], ==, "socks4a://proxy.example.com");
152 g_assert_cmpstr (proxies[2], ==, "socks4://proxy.example.com");
153 g_strfreev (proxies);
154
155 proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/",
156 NULL, &error);
157 g_assert_no_error (error);
158 g_assert_cmpint (g_strv_length (proxies), ==, 1);
159 g_assert_cmpstr (proxies[0], ==, "direct://");
160 g_strfreev (proxies);
161
162 g_object_unref (resolver);
163
164 resolver = g_simple_proxy_resolver_new ("default-proxy://", (char **) ignore_hosts);
165 g_simple_proxy_resolver_set_uri_proxy (G_SIMPLE_PROXY_RESOLVER (resolver),
166 "http", "socks://proxy.example.com");
167
168 proxies = g_proxy_resolver_lookup (resolver, "http://one.example.com/",
169 NULL, &error);
170 g_assert_no_error (error);
171 g_assert_cmpint (g_strv_length (proxies), ==, 3);
172 g_assert_cmpstr (proxies[0], ==, "socks5://proxy.example.com");
173 g_assert_cmpstr (proxies[1], ==, "socks4a://proxy.example.com");
174 g_assert_cmpstr (proxies[2], ==, "socks4://proxy.example.com");
175 g_strfreev (proxies);
176
177 proxies = g_proxy_resolver_lookup (resolver, "ftp://two.example.com/",
178 NULL, &error);
179 g_assert_no_error (error);
180 g_assert_cmpint (g_strv_length (proxies), ==, 1);
181 g_assert_cmpstr (proxies[0], ==, "default-proxy://");
182 g_strfreev (proxies);
183
184 proxies = g_proxy_resolver_lookup (resolver, "http://127.0.0.1/",
185 NULL, &error);
186 g_assert_no_error (error);
187 g_assert_cmpint (g_strv_length (proxies), ==, 1);
188 g_assert_cmpstr (proxies[0], ==, "direct://");
189 g_strfreev (proxies);
190
191 g_object_unref (resolver);
192 }
193
194 static const char *ignore_hosts[] = {
195 ".bbb.xx",
196 "*.ccc.xx",
197 "ddd.xx",
198 "*.eee.xx:8000",
199 "127.0.0.0/24",
200 "10.0.0.1:8000",
201 "::1",
202 "fe80::/10",
203 NULL
204 };
205
206 static const struct {
207 const char *uri;
208 const char *proxy;
209 } ignore_tests[] = {
210 { "http://aaa.xx/", "http://localhost:8080" },
211 { "http://aaa.xx:8000/", "http://localhost:8080" },
212 { "http://www.aaa.xx/", "http://localhost:8080" },
213 { "http://www.aaa.xx:8000/", "http://localhost:8080" },
214 { "https://aaa.xx/", "http://localhost:8080" },
215 { "http://bbb.xx/", "direct://" },
216 { "http://www.bbb.xx/", "direct://" },
217 { "http://bbb.xx:8000/", "direct://" },
218 { "http://www.bbb.xx:8000/", "direct://" },
219 { "https://bbb.xx/", "direct://" },
220 { "http://nobbb.xx/", "http://localhost:8080" },
221 { "http://www.nobbb.xx/", "http://localhost:8080" },
222 { "http://nobbb.xx:8000/", "http://localhost:8080" },
223 { "http://www.nobbb.xx:8000/", "http://localhost:8080" },
224 { "https://nobbb.xx/", "http://localhost:8080" },
225 { "http://ccc.xx/", "direct://" },
226 { "http://www.ccc.xx/", "direct://" },
227 { "http://ccc.xx:8000/", "direct://" },
228 { "http://www.ccc.xx:8000/", "direct://" },
229 { "https://ccc.xx/", "direct://" },
230 { "http://ddd.xx/", "direct://" },
231 { "http://ddd.xx:8000/", "direct://" },
232 { "http://www.ddd.xx/", "direct://" },
233 { "http://www.ddd.xx:8000/", "direct://" },
234 { "https://ddd.xx/", "direct://" },
235 { "http://eee.xx/", "http://localhost:8080" },
236 { "http://eee.xx:8000/", "direct://" },
237 { "http://www.eee.xx/", "http://localhost:8080" },
238 { "http://www.eee.xx:8000/", "direct://" },
239 { "https://eee.xx/", "http://localhost:8080" },
240 { "http://1.2.3.4/", "http://localhost:8080" },
241 { "http://127.0.0.1/", "direct://" },
242 { "http://127.0.0.2/", "direct://" },
243 { "http://127.0.0.255/", "direct://" },
244 { "http://127.0.1.0/", "http://localhost:8080" },
245 { "http://10.0.0.1/", "http://localhost:8080" },
246 { "http://10.0.0.1:8000/", "direct://" },
247 { "http://[::1]/", "direct://" },
248 { "http://[::1]:80/", "direct://" },
249 { "http://[::1:1]/", "http://localhost:8080" },
250 { "http://[::1:1]:80/", "http://localhost:8080" },
251 { "http://[fe80::1]/", "direct://" },
252 { "http://[fe80::1]:80/", "direct://" },
253 { "http://[fec0::1]/", "http://localhost:8080" },
254 { "http://[fec0::1]:80/", "http://localhost:8080" }
255 };
256 static const int n_ignore_tests = G_N_ELEMENTS (ignore_tests);
257
258 static void
test_ignore(void)259 test_ignore (void)
260 {
261 GProxyResolver *resolver;
262 GError *error = NULL;
263 char **proxies;
264 int i;
265
266 resolver = g_simple_proxy_resolver_new ("http://localhost:8080",
267 (char **)ignore_hosts);
268
269 for (i = 0; i < n_ignore_tests; i++)
270 {
271 proxies = g_proxy_resolver_lookup (resolver, ignore_tests[i].uri,
272 NULL, &error);
273 g_assert_no_error (error);
274
275 g_assert_cmpstr (proxies[0], ==, ignore_tests[i].proxy);
276 g_strfreev (proxies);
277 }
278
279 g_object_unref (resolver);
280 }
281
282 int
main(int argc,char * argv[])283 main (int argc,
284 char *argv[])
285 {
286 g_test_init (&argc, &argv, NULL);
287
288 g_test_add_func ("/static-proxy/uri", test_uris);
289 g_test_add_func ("/static-proxy/socks", test_socks);
290 g_test_add_func ("/static-proxy/ignore", test_ignore);
291
292 return g_test_run();
293 }
294