/* -*- Mode: C; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
* Copyright © 2016 Igalia S.L.
*
* This file is part of Epiphany.
*
* Epiphany is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Epiphany 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Epiphany. If not, see .
*/
#include "config.h"
#include "ephy-view-source-handler.h"
#include "ephy-embed-container.h"
#include "ephy-embed-shell.h"
#include "ephy-web-view.h"
#include
#include
#include
struct _EphyViewSourceHandler {
GObject parent_instance;
GList *outstanding_requests;
};
G_DEFINE_TYPE (EphyViewSourceHandler, ephy_view_source_handler, G_TYPE_OBJECT)
typedef struct {
EphyViewSourceHandler *source_handler;
WebKitURISchemeRequest *scheme_request;
WebKitWebView *web_view;
GCancellable *cancellable;
guint load_changed_id;
} EphyViewSourceRequest;
static EphyViewSourceRequest *
ephy_view_source_request_new (EphyViewSourceHandler *handler,
WebKitURISchemeRequest *request)
{
EphyViewSourceRequest *view_source_request;
view_source_request = g_new (EphyViewSourceRequest, 1);
view_source_request->source_handler = g_object_ref (handler);
view_source_request->scheme_request = g_object_ref (request);
view_source_request->web_view = NULL; /* created only if required */
view_source_request->cancellable = g_cancellable_new ();
view_source_request->load_changed_id = 0;
return view_source_request;
}
static void
ephy_view_source_request_free (EphyViewSourceRequest *request)
{
if (request->load_changed_id > 0)
g_signal_handler_disconnect (request->web_view, request->load_changed_id);
g_object_unref (request->source_handler);
g_object_unref (request->scheme_request);
g_clear_object (&request->web_view);
g_cancellable_cancel (request->cancellable);
g_object_unref (request->cancellable);
g_free (request);
}
static void
finish_uri_scheme_request (EphyViewSourceRequest *request,
gchar *data,
GError *error)
{
GInputStream *stream;
gssize data_length;
g_assert ((data && !error) || (!data && error));
if (error) {
webkit_uri_scheme_request_finish_error (request->scheme_request, error);
} else {
data_length = MIN (strlen (data), G_MAXSSIZE);
stream = g_memory_input_stream_new_from_data (data, data_length, g_free);
webkit_uri_scheme_request_finish (request->scheme_request, stream, data_length, "text/html");
g_object_unref (stream);
}
request->source_handler->outstanding_requests =
g_list_remove (request->source_handler->outstanding_requests,
request);
ephy_view_source_request_free (request);
}
static void
web_resource_data_cb (WebKitWebResource *resource,
GAsyncResult *result,
EphyViewSourceRequest *request)
{
g_autofree guchar *data = NULL;
g_autofree char *escaped_str = NULL;
g_autoptr (GError) error = NULL;
g_autofree char *html = NULL;
gsize length;
data = webkit_web_resource_get_data_finish (resource, result, &length, &error);
if (!data) {
finish_uri_scheme_request (request, NULL, error);
return;
}
/* Warning: data is not a string, so we pass length here because it's not NUL-terminated. */
escaped_str = g_markup_escape_text ((const char *)data, length);
html = g_strdup_printf (""
" "
" "
" "
" %s"
""
""
" "
" "
" "
" %s
"
"",
webkit_web_resource_get_uri (resource),
escaped_str);
finish_uri_scheme_request (request, g_steal_pointer (&html), NULL);
}
static void
ephy_view_source_request_begin_get_source_from_web_view (EphyViewSourceRequest *request,
WebKitWebView *web_view)
{
WebKitWebResource *resource = webkit_web_view_get_main_resource (web_view);
g_assert (resource);
webkit_web_resource_get_data (resource,
request->cancellable,
(GAsyncReadyCallback)(web_resource_data_cb),
request);
}
static void
load_changed_cb (WebKitWebView *web_view,
WebKitLoadEvent load_event,
EphyViewSourceRequest *request)
{
if (load_event == WEBKIT_LOAD_FINISHED) {
g_signal_handler_disconnect (request->web_view, request->load_changed_id);
request->load_changed_id = 0;
ephy_view_source_request_begin_get_source_from_web_view (request, web_view);
}
}
static void
ephy_view_source_request_begin_get_source_from_uri (EphyViewSourceRequest *request,
const char *uri)
{
EphyEmbedShell *shell = ephy_embed_shell_get_default ();
WebKitWebContext *context = ephy_embed_shell_get_web_context (shell);
request->web_view = WEBKIT_WEB_VIEW (g_object_ref_sink (webkit_web_view_new_with_context (context)));
g_assert (request->load_changed_id == 0);
request->load_changed_id = g_signal_connect (request->web_view, "load-changed",
G_CALLBACK (load_changed_cb),
request);
webkit_web_view_load_uri (request->web_view, uri);
}
static gint
embed_is_displaying_matching_uri (EphyEmbed *embed,
GUri *uri)
{
EphyWebView *web_view;
g_autoptr (GUri) view_uri = NULL;
g_autofree char *modified_uri = NULL;
g_autofree char *uri_string = NULL;
if (ephy_embed_has_load_pending (embed))
return -1;
web_view = ephy_embed_get_web_view (embed);
if (ephy_web_view_is_loading (web_view))
return -1;
view_uri = g_uri_parse (ephy_web_view_get_address (web_view),
G_URI_FLAGS_NONE, NULL);
if (!view_uri)
return -1;
modified_uri = g_uri_to_string_partial (view_uri, G_URI_HIDE_FRAGMENT);
uri_string = g_uri_to_string (uri);
return strcmp (modified_uri, uri_string);
}
static WebKitWebView *
get_web_view_matching_uri (GUri *uri)
{
EphyEmbedShell *shell;
GtkWindow *window;
GList *embeds = NULL;
GList *found;
EphyEmbed *embed = NULL;
shell = ephy_embed_shell_get_default ();
window = gtk_application_get_active_window (GTK_APPLICATION (shell));
if (!EPHY_IS_EMBED_CONTAINER (window))
goto out;
embeds = ephy_embed_container_get_children (EPHY_EMBED_CONTAINER (window));
found = g_list_find_custom (embeds, uri, (GCompareFunc)embed_is_displaying_matching_uri);
if (found)
embed = found->data;
out:
g_list_free (embeds);
return embed ? WEBKIT_WEB_VIEW (ephy_embed_get_web_view (embed)) : NULL;
}
static void
ephy_view_source_request_start (EphyViewSourceRequest *request)
{
g_autoptr (GUri) uri = NULL;
g_autoptr (GUri) converted_uri = NULL;
const char *original_uri;
WebKitWebView *web_view;
request->source_handler->outstanding_requests =
g_list_prepend (request->source_handler->outstanding_requests, request);
original_uri = webkit_uri_scheme_request_get_uri (request->scheme_request);
uri = g_uri_parse (original_uri, G_URI_FLAGS_ENCODED | G_URI_FLAGS_SCHEME_NORMALIZE, NULL);
if (!uri || !g_uri_get_fragment (uri)) {
/* Can't assert because user could theoretically input something weird */
GError *error = g_error_new (WEBKIT_NETWORK_ERROR,
WEBKIT_NETWORK_ERROR_FAILED,
_("%s is not a valid URI"),
original_uri);
finish_uri_scheme_request (request, NULL, error);
g_error_free (error);
return;
}
/* Convert e.g. ephy-source://gnome.org#https to https://gnome.org */
converted_uri = g_uri_build (g_uri_get_flags (uri),
g_uri_get_fragment (uri),
g_uri_get_userinfo (uri),
g_uri_get_host (uri),
g_uri_get_port (uri),
g_uri_get_path (uri),
g_uri_get_query (uri),
NULL);
g_assert (converted_uri);
web_view = get_web_view_matching_uri (converted_uri);
if (web_view)
ephy_view_source_request_begin_get_source_from_web_view (request, WEBKIT_WEB_VIEW (web_view));
else {
g_autofree char *modified_uri = NULL;
modified_uri = g_uri_to_string (converted_uri);
ephy_view_source_request_begin_get_source_from_uri (request, modified_uri);
}
}
static void
cancel_outstanding_request (EphyViewSourceRequest *request)
{
g_cancellable_cancel (request->cancellable);
}
static void
ephy_view_source_handler_dispose (GObject *object)
{
EphyViewSourceHandler *handler = EPHY_VIEW_SOURCE_HANDLER (object);
if (handler->outstanding_requests) {
g_list_foreach (handler->outstanding_requests, (GFunc)cancel_outstanding_request, NULL);
g_list_free (handler->outstanding_requests);
handler->outstanding_requests = NULL;
}
G_OBJECT_CLASS (ephy_view_source_handler_parent_class)->dispose (object);
}
static void
ephy_view_source_handler_init (EphyViewSourceHandler *handler)
{
}
static void
ephy_view_source_handler_class_init (EphyViewSourceHandlerClass *klass)
{
GObjectClass *object_class = G_OBJECT_CLASS (klass);
object_class->dispose = ephy_view_source_handler_dispose;
}
EphyViewSourceHandler *
ephy_view_source_handler_new (void)
{
return EPHY_VIEW_SOURCE_HANDLER (g_object_new (EPHY_TYPE_VIEW_SOURCE_HANDLER, NULL));
}
void
ephy_view_source_handler_handle_request (EphyViewSourceHandler *handler,
WebKitURISchemeRequest *scheme_request)
{
EphyViewSourceRequest *view_source_request;
view_source_request = ephy_view_source_request_new (handler, scheme_request);
ephy_view_source_request_start (view_source_request);
}