| /* GIO - GLib Input, Output and Streaming Library |
| * |
| * Copyright (C) 2010 Collabora, Ltd. |
| * Copyright (C) 2014 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/>. |
| * |
| * Author: Nicolas Dufresne <nicolas.dufresne@collabora.co.uk> |
| * Marc-André Lureau <marcandre.lureau@redhat.com> |
| */ |
| |
| #include "config.h" |
| |
| #include "ghttpproxy.h" |
| |
| #include <string.h> |
| #include <stdlib.h> |
| |
| #include "giomodule.h" |
| #include "giomodule-priv.h" |
| #include "giostream.h" |
| #include "ginputstream.h" |
| #include "glibintl.h" |
| #include "goutputstream.h" |
| #include "gproxy.h" |
| #include "gproxyaddress.h" |
| #include "gsocketconnectable.h" |
| #include "gtask.h" |
| #include "gtlsclientconnection.h" |
| #include "gtlsconnection.h" |
| |
| |
| struct _GHttpProxy |
| { |
| GObject parent; |
| }; |
| |
| struct _GHttpProxyClass |
| { |
| GObjectClass parent_class; |
| }; |
| |
| static void g_http_proxy_iface_init (GProxyInterface *proxy_iface); |
| |
| #define g_http_proxy_get_type _g_http_proxy_get_type |
| G_DEFINE_TYPE_WITH_CODE (GHttpProxy, g_http_proxy, G_TYPE_OBJECT, |
| G_IMPLEMENT_INTERFACE (G_TYPE_PROXY, |
| g_http_proxy_iface_init) |
| _g_io_modules_ensure_extension_points_registered (); |
| g_io_extension_point_implement (G_PROXY_EXTENSION_POINT_NAME, |
| g_define_type_id, |
| "http", |
| 0)) |
| |
| static void |
| g_http_proxy_init (GHttpProxy *proxy) |
| { |
| } |
| |
| static gchar * |
| create_request (GProxyAddress *proxy_address, |
| gboolean *has_cred, |
| GError **error) |
| { |
| const gchar *hostname; |
| gint port; |
| const gchar *username; |
| const gchar *password; |
| GString *request; |
| gchar *ascii_hostname; |
| |
| if (has_cred) |
| *has_cred = FALSE; |
| |
| hostname = g_proxy_address_get_destination_hostname (proxy_address); |
| ascii_hostname = g_hostname_to_ascii (hostname); |
| if (!ascii_hostname) |
| { |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED, |
| _("Invalid hostname")); |
| return NULL; |
| } |
| port = g_proxy_address_get_destination_port (proxy_address); |
| username = g_proxy_address_get_username (proxy_address); |
| password = g_proxy_address_get_password (proxy_address); |
| |
| request = g_string_new (NULL); |
| |
| g_string_append_printf (request, |
| "CONNECT %s:%i HTTP/1.0\r\n" |
| "Host: %s:%i\r\n" |
| "Proxy-Connection: keep-alive\r\n" |
| "User-Agent: GLib/%i.%i\r\n", |
| ascii_hostname, port, |
| ascii_hostname, port, |
| GLIB_MAJOR_VERSION, GLIB_MINOR_VERSION); |
| g_free (ascii_hostname); |
| |
| if (username != NULL && password != NULL) |
| { |
| gchar *cred; |
| gchar *base64_cred; |
| |
| if (has_cred) |
| *has_cred = TRUE; |
| |
| cred = g_strdup_printf ("%s:%s", username, password); |
| base64_cred = g_base64_encode ((guchar *) cred, strlen (cred)); |
| g_free (cred); |
| g_string_append_printf (request, |
| "Proxy-Authorization: Basic %s\r\n", |
| base64_cred); |
| g_free (base64_cred); |
| } |
| |
| g_string_append (request, "\r\n"); |
| |
| return g_string_free (request, FALSE); |
| } |
| |
| static gboolean |
| check_reply (const gchar *buffer, |
| gboolean has_cred, |
| GError **error) |
| { |
| gint err_code; |
| const gchar *ptr = buffer + 7; |
| |
| if (strncmp (buffer, "HTTP/1.", 7) != 0 || (*ptr != '0' && *ptr != '1')) |
| { |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED, |
| _("Bad HTTP proxy reply")); |
| return FALSE; |
| } |
| |
| ptr++; |
| while (*ptr == ' ') |
| ptr++; |
| |
| err_code = atoi (ptr); |
| |
| if (err_code < 200 || err_code >= 300) |
| { |
| switch (err_code) |
| { |
| case 403: |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_NOT_ALLOWED, |
| _("HTTP proxy connection not allowed")); |
| break; |
| case 407: |
| if (has_cred) |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_AUTH_FAILED, |
| _("HTTP proxy authentication failed")); |
| else |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_NEED_AUTH, |
| _("HTTP proxy authentication required")); |
| break; |
| default: |
| g_set_error (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED, |
| _("HTTP proxy connection failed: %i"), err_code); |
| } |
| |
| return FALSE; |
| } |
| |
| return TRUE; |
| } |
| |
| #define HTTP_END_MARKER "\r\n\r\n" |
| |
| static GIOStream * |
| g_http_proxy_connect (GProxy *proxy, |
| GIOStream *io_stream, |
| GProxyAddress *proxy_address, |
| GCancellable *cancellable, |
| GError **error) |
| { |
| GInputStream *in; |
| GOutputStream *out; |
| gchar *buffer = NULL; |
| gsize buffer_length; |
| gssize bytes_read; |
| gboolean has_cred; |
| GIOStream *tlsconn = NULL; |
| |
| if (G_IS_HTTPS_PROXY (proxy)) |
| { |
| tlsconn = g_tls_client_connection_new (io_stream, |
| G_SOCKET_CONNECTABLE (proxy_address), |
| error); |
| if (!tlsconn) |
| goto error; |
| |
| #ifdef DEBUG |
| { |
| GTlsCertificateFlags tls_validation_flags = G_TLS_CERTIFICATE_VALIDATE_ALL; |
| |
| tls_validation_flags &= ~(G_TLS_CERTIFICATE_UNKNOWN_CA | G_TLS_CERTIFICATE_BAD_IDENTITY); |
| g_tls_client_connection_set_validation_flags (G_TLS_CLIENT_CONNECTION (tlsconn), |
| tls_validation_flags); |
| } |
| #endif |
| |
| if (!g_tls_connection_handshake (G_TLS_CONNECTION (tlsconn), cancellable, error)) |
| goto error; |
| |
| io_stream = tlsconn; |
| } |
| |
| in = g_io_stream_get_input_stream (io_stream); |
| out = g_io_stream_get_output_stream (io_stream); |
| |
| buffer = create_request (proxy_address, &has_cred, error); |
| if (!buffer) |
| goto error; |
| if (!g_output_stream_write_all (out, buffer, strlen (buffer), NULL, |
| cancellable, error)) |
| goto error; |
| |
| g_free (buffer); |
| |
| bytes_read = 0; |
| buffer_length = 1024; |
| buffer = g_malloc (buffer_length); |
| |
| /* Read byte-by-byte instead of using GDataInputStream |
| * since we do not want to read beyond the end marker |
| */ |
| do |
| { |
| gsize nread; |
| |
| nread = g_input_stream_read (in, buffer + bytes_read, 1, cancellable, error); |
| if (nread == -1) |
| goto error; |
| |
| if (nread == 0) |
| break; |
| |
| ++bytes_read; |
| |
| if (bytes_read == buffer_length) |
| { |
| buffer_length = 2 * buffer_length; |
| buffer = g_realloc (buffer, buffer_length); |
| } |
| |
| *(buffer + bytes_read) = '\0'; |
| |
| if (g_str_has_suffix (buffer, HTTP_END_MARKER)) |
| break; |
| } |
| while (TRUE); |
| |
| if (bytes_read == 0) |
| { |
| g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PROXY_FAILED, |
| _("HTTP proxy server closed connection unexpectedly.")); |
| goto error; |
| } |
| |
| if (!check_reply (buffer, has_cred, error)) |
| goto error; |
| |
| g_free (buffer); |
| |
| g_object_ref (io_stream); |
| g_clear_object (&tlsconn); |
| |
| return io_stream; |
| |
| error: |
| g_clear_object (&tlsconn); |
| g_free (buffer); |
| return NULL; |
| } |
| |
| typedef struct |
| { |
| GIOStream *io_stream; |
| GProxyAddress *proxy_address; |
| } ConnectAsyncData; |
| |
| static void |
| free_connect_data (ConnectAsyncData *data) |
| { |
| g_object_unref (data->io_stream); |
| g_object_unref (data->proxy_address); |
| g_slice_free (ConnectAsyncData, data); |
| } |
| |
| static void |
| connect_thread (GTask *task, |
| gpointer source_object, |
| gpointer task_data, |
| GCancellable *cancellable) |
| { |
| GProxy *proxy = source_object; |
| ConnectAsyncData *data = task_data; |
| GIOStream *res; |
| GError *error = NULL; |
| |
| res = g_http_proxy_connect (proxy, data->io_stream, data->proxy_address, |
| cancellable, &error); |
| |
| if (res == NULL) |
| g_task_return_error (task, error); |
| else |
| g_task_return_pointer (task, res, g_object_unref); |
| } |
| |
| static void |
| g_http_proxy_connect_async (GProxy *proxy, |
| GIOStream *io_stream, |
| GProxyAddress *proxy_address, |
| GCancellable *cancellable, |
| GAsyncReadyCallback callback, |
| gpointer user_data) |
| { |
| ConnectAsyncData *data; |
| GTask *task; |
| |
| data = g_slice_new0 (ConnectAsyncData); |
| data->io_stream = g_object_ref (io_stream); |
| data->proxy_address = g_object_ref (proxy_address); |
| |
| task = g_task_new (proxy, cancellable, callback, user_data); |
| g_task_set_source_tag (task, g_http_proxy_connect_async); |
| g_task_set_task_data (task, data, (GDestroyNotify) free_connect_data); |
| |
| g_task_run_in_thread (task, connect_thread); |
| g_object_unref (task); |
| } |
| |
| static GIOStream * |
| g_http_proxy_connect_finish (GProxy *proxy, |
| GAsyncResult *result, |
| GError **error) |
| { |
| return g_task_propagate_pointer (G_TASK (result), error); |
| } |
| |
| static gboolean |
| g_http_proxy_supports_hostname (GProxy *proxy) |
| { |
| return TRUE; |
| } |
| |
| static void |
| g_http_proxy_class_init (GHttpProxyClass *class) |
| { |
| } |
| |
| static void |
| g_http_proxy_iface_init (GProxyInterface *proxy_iface) |
| { |
| proxy_iface->connect = g_http_proxy_connect; |
| proxy_iface->connect_async = g_http_proxy_connect_async; |
| proxy_iface->connect_finish = g_http_proxy_connect_finish; |
| proxy_iface->supports_hostname = g_http_proxy_supports_hostname; |
| } |
| |
| struct _GHttpsProxy |
| { |
| GHttpProxy parent; |
| }; |
| |
| struct _GHttpsProxyClass |
| { |
| GHttpProxyClass parent_class; |
| }; |
| |
| #define g_https_proxy_get_type _g_https_proxy_get_type |
| G_DEFINE_TYPE_WITH_CODE (GHttpsProxy, g_https_proxy, G_TYPE_HTTP_PROXY, |
| G_IMPLEMENT_INTERFACE (G_TYPE_PROXY, |
| g_http_proxy_iface_init) |
| _g_io_modules_ensure_extension_points_registered (); |
| g_io_extension_point_implement (G_PROXY_EXTENSION_POINT_NAME, |
| g_define_type_id, |
| "https", |
| 0)) |
| |
| static void |
| g_https_proxy_init (GHttpsProxy *proxy) |
| { |
| } |
| |
| static void |
| g_https_proxy_class_init (GHttpsProxyClass *class) |
| { |
| } |