1 /* GIO - GLib Input, Output and Streaming Library
3 * Copyright (C) 2010 Collabora, Ltd.
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 of the License, or (at your option) any later version.
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.
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18 * Boston, MA 02111-1307, USA.
20 * Author: Nicolas Dufresne <nicolas.dufresne@collabora.co.uk>
25 #include "gsocks4aproxy.h"
29 #include "gasyncresult.h"
30 #include "giomodule.h"
31 #include "giomodule-priv.h"
32 #include "giostream.h"
33 #include "ginetaddress.h"
34 #include "ginputstream.h"
36 #include "goutputstream.h"
38 #include "gproxyaddress.h"
39 #include "gsimpleasyncresult.h"
41 #define SOCKS4_VERSION 4
43 #define SOCKS4_CMD_CONNECT 1
44 #define SOCKS4_CMD_BIND 2
46 #define SOCKS4_MAX_LEN 255
48 #define SOCKS4_REP_VERSION 0
49 #define SOCKS4_REP_GRANTED 90
50 #define SOCKS4_REP_REJECTED 91
51 #define SOCKS4_REP_NO_IDENT 92
52 #define SOCKS4_REP_BAD_IDENT 93
54 static void g_socks4a_proxy_iface_init (GProxyInterface *proxy_iface);
56 #define g_socks4a_proxy_get_type _g_socks4a_proxy_get_type
57 G_DEFINE_TYPE_WITH_CODE (GSocks4aProxy, g_socks4a_proxy, G_TYPE_OBJECT,
58 G_IMPLEMENT_INTERFACE (G_TYPE_PROXY,
59 g_socks4a_proxy_iface_init)
60 _g_io_modules_ensure_extension_points_registered ();
61 g_io_extension_point_implement (G_PROXY_EXTENSION_POINT_NAME,
67 g_socks4a_proxy_finalize (GObject *object)
70 G_OBJECT_CLASS (g_socks4a_proxy_parent_class)->finalize (object);
74 g_socks4a_proxy_init (GSocks4aProxy *proxy)
76 proxy->supports_hostname = TRUE;
80 * +----+----+----+----+----+----+----+----+----+----+....+----+------+....+------+
81 * | VN | CD | DSTPORT | DSTIP | USERID |NULL| HOST | | NULL |
82 * +----+----+----+----+----+----+----+----+----+----+....+----+------+....+------+
83 * 1 1 2 4 variable 1 variable
85 #define SOCKS4_CONN_MSG_LEN (9 + SOCKS4_MAX_LEN * 2)
87 set_connect_msg (guint8 *msg,
88 const gchar *hostname,
99 msg[len++] = SOCKS4_VERSION;
100 msg[len++] = SOCKS4_CMD_CONNECT;
103 guint16 hp = g_htons (port);
104 memcpy (msg + len, &hp, 2);
108 is_ip = g_hostname_is_ip_address (hostname);
115 addr = g_inet_address_new_from_string (ip);
116 addr_len = g_inet_address_get_native_size (addr);
120 g_set_error (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED,
121 _("SOCKSv4 does not support IPv6 address '%s'"),
123 g_object_unref (addr);
127 memcpy (msg + len, g_inet_address_to_bytes (addr), addr_len);
130 g_object_unref (addr);
134 gsize user_len = strlen (username);
136 if (user_len > SOCKS4_MAX_LEN)
138 g_set_error (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED,
139 _("SOCKSv4 implementation limits username to %i characters"),
144 memcpy (msg + len, username, user_len);
152 gsize host_len = strlen (hostname);
154 if (host_len > SOCKS4_MAX_LEN)
156 g_set_error (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED,
157 _("SOCKSv4a implementation limits hostname to %i characters"),
162 memcpy (msg + len, hostname, host_len);
171 * +----+----+----+----+----+----+----+----+
172 * | VN | CD | DSTPORT | DSTIP |
173 * +----+----+----+----+----+----+----+----+
176 #define SOCKS4_CONN_REP_LEN 8
178 parse_connect_reply (const guint8 *data, GError **error)
180 if (data[0] != SOCKS4_REP_VERSION)
182 g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED,
183 _("The server is not a SOCKSv4 proxy server."));
187 if (data[1] != SOCKS4_REP_GRANTED)
189 g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED,
190 _("Connection through SOCKSv4 server was rejected"));
198 g_socks4a_proxy_connect (GProxy *proxy,
199 GIOStream *io_stream,
200 GProxyAddress *proxy_address,
201 GCancellable *cancellable,
206 const gchar *hostname;
208 const gchar *username;
210 hostname = g_proxy_address_get_destination_hostname (proxy_address);
211 port = g_proxy_address_get_destination_port (proxy_address);
212 username = g_proxy_address_get_username (proxy_address);
214 in = g_io_stream_get_input_stream (io_stream);
215 out = g_io_stream_get_output_stream (io_stream);
217 /* Send SOCKS4 connection request */
219 guint8 msg[SOCKS4_CONN_MSG_LEN];
222 len = set_connect_msg (msg, hostname, port, username, error);
227 if (!g_output_stream_write_all (out, msg, len, NULL,
232 /* Read SOCKS4 response */
234 guint8 data[SOCKS4_CONN_REP_LEN];
236 if (!g_input_stream_read_all (in, data, SOCKS4_CONN_REP_LEN, NULL,
240 if (!parse_connect_reply (data, error))
244 return g_object_ref (io_stream);
253 GSimpleAsyncResult *simple;
254 GIOStream *io_stream;
255 GProxyAddress *proxy_address;
256 GCancellable *cancellable;
265 static void connect_msg_write_cb (GObject *source,
266 GAsyncResult *result,
268 static void connect_reply_read_cb (GObject *source,
269 GAsyncResult *result,
273 free_connect_data (ConnectAsyncData *data)
276 g_object_unref (data->io_stream);
278 if (data->proxy_address)
279 g_object_unref (data->proxy_address);
281 if (data->cancellable)
282 g_object_unref (data->cancellable);
284 g_slice_free (ConnectAsyncData, data);
288 complete_async_from_error (ConnectAsyncData *data, GError *error)
290 GSimpleAsyncResult *simple = data->simple;
291 g_simple_async_result_take_error (data->simple, error);
292 g_simple_async_result_set_op_res_gpointer (simple, NULL, NULL);
293 g_simple_async_result_complete (simple);
294 g_object_unref (simple);
298 do_read (GAsyncReadyCallback callback, ConnectAsyncData *data)
301 in = g_io_stream_get_input_stream (data->io_stream);
302 g_input_stream_read_async (in,
303 data->buffer + data->offset,
304 data->length - data->offset,
305 G_PRIORITY_DEFAULT, data->cancellable,
310 do_write (GAsyncReadyCallback callback, ConnectAsyncData *data)
313 out = g_io_stream_get_output_stream (data->io_stream);
314 g_output_stream_write_async (out,
315 data->buffer + data->offset,
316 data->length - data->offset,
317 G_PRIORITY_DEFAULT, data->cancellable,
324 g_socks4a_proxy_connect_async (GProxy *proxy,
325 GIOStream *io_stream,
326 GProxyAddress *proxy_address,
327 GCancellable *cancellable,
328 GAsyncReadyCallback callback,
331 GError *error = NULL;
332 GSimpleAsyncResult *simple;
333 ConnectAsyncData *data;
334 const gchar *hostname;
336 const gchar *username;
338 simple = g_simple_async_result_new (G_OBJECT (proxy),
340 g_socks4a_proxy_connect_async);
342 data = g_slice_new0 (ConnectAsyncData);
344 data->simple = simple;
345 data->io_stream = g_object_ref (io_stream);
348 data->cancellable = g_object_ref (cancellable);
350 g_simple_async_result_set_op_res_gpointer (simple, data,
351 (GDestroyNotify) free_connect_data);
353 hostname = g_proxy_address_get_destination_hostname (proxy_address);
354 port = g_proxy_address_get_destination_port (proxy_address);
355 username = g_proxy_address_get_username (proxy_address);
357 data->buffer = g_malloc0 (SOCKS4_CONN_MSG_LEN);
358 data->length = set_connect_msg (data->buffer,
359 hostname, port, username,
363 if (data->length < 0)
365 g_simple_async_result_take_error (data->simple, error);
366 g_simple_async_result_set_op_res_gpointer (simple, NULL, NULL);
367 g_simple_async_result_complete_in_idle (simple);
368 g_object_unref (simple);
372 do_write (connect_msg_write_cb, data);
377 connect_msg_write_cb (GObject *source,
378 GAsyncResult *result,
381 GError *error = NULL;
382 ConnectAsyncData *data = user_data;
385 written = g_output_stream_write_finish (G_OUTPUT_STREAM (source),
390 complete_async_from_error (data, error);
394 data->offset += written;
396 if (data->offset == data->length)
398 g_free (data->buffer);
400 data->buffer = g_malloc0 (SOCKS4_CONN_REP_LEN);
401 data->length = SOCKS4_CONN_REP_LEN;
404 do_read (connect_reply_read_cb, data);
408 do_write (connect_msg_write_cb, data);
413 connect_reply_read_cb (GObject *source,
414 GAsyncResult *result,
417 GError *error = NULL;
418 ConnectAsyncData *data = user_data;
421 read = g_input_stream_read_finish (G_INPUT_STREAM (source),
426 complete_async_from_error (data, error);
430 data->offset += read;
432 if (data->offset == data->length)
434 if (!parse_connect_reply (data->buffer, &error))
436 complete_async_from_error (data, error);
440 GSimpleAsyncResult *simple = data->simple;
441 g_simple_async_result_complete (simple);
442 g_object_unref (simple);
447 do_read (connect_reply_read_cb, data);
451 static GIOStream *g_socks4a_proxy_connect_finish (GProxy *proxy,
452 GAsyncResult *result,
456 g_socks4a_proxy_connect_finish (GProxy *proxy,
457 GAsyncResult *result,
460 GSimpleAsyncResult *simple = G_SIMPLE_ASYNC_RESULT (result);
461 ConnectAsyncData *data = g_simple_async_result_get_op_res_gpointer (simple);
463 if (g_simple_async_result_propagate_error (simple, error))
466 return g_object_ref (data->io_stream);
470 g_socks4a_proxy_supports_hostname (GProxy *proxy)
472 return G_SOCKS4A_PROXY (proxy)->supports_hostname;
476 g_socks4a_proxy_class_init (GSocks4aProxyClass *class)
478 GObjectClass *object_class;
480 object_class = (GObjectClass *) class;
481 object_class->finalize = g_socks4a_proxy_finalize;
485 g_socks4a_proxy_iface_init (GProxyInterface *proxy_iface)
487 proxy_iface->connect = g_socks4a_proxy_connect;
488 proxy_iface->connect_async = g_socks4a_proxy_connect_async;
489 proxy_iface->connect_finish = g_socks4a_proxy_connect_finish;
490 proxy_iface->supports_hostname = g_socks4a_proxy_supports_hostname;