Imported WebKit commit 7e538425aa020340619e927792f3d895061fb54b (http://svn.webkit.org/repository/webkit/trunk@116286)

1 files changed, 115 insertions, 0 deletions

diff --git a/Source/WebKit2/WebProcess/soup/WebSoupRequestManager.cpp b/Source/WebKit2/WebProcess/soup/WebSoupRequestManager.cpp
new file mode 100644
index 000000000..8c484c7fe
--- /dev/null
+++ b/Source/WebKit2/WebProcess/soup/WebSoupRequestManager.cpp
@@ -0,0 +1,115 @@
+/*
+ * Copyright (C) 2012 Igalia S.L.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 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
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public License
+ * along with this library; see the file COPYING.LIB.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#include "config.h"
+#include "WebSoupRequestManager.h"
+
+#include "DataReference.h"
+#include "MessageID.h"
+#include "WebKitSoupRequestGeneric.h"
+#include "WebProcess.h"
+#include "WebSoupRequestManagerProxyMessages.h"
+#include <WebCore/ErrorsGtk.h>
+#include <WebCore/ResourceHandle.h>
+#include <WebCore/ResourceRequest.h>
+#include <libsoup/soup-requester.h>
+#include <wtf/gobject/GOwnPtr.h>
+#include <wtf/text/CString.h>
+
+namespace WebKit {
+
+static uint64_t generateSoupRequestID()
+{
+    static uint64_t uniqueSoupRequestID = 1;
+    return uniqueSoupRequestID++;
+}
+
+WebSoupRequestManager::WebSoupRequestManager(WebProcess* process)
+    : m_process(process)
+    , m_schemes(adoptGRef(g_ptr_array_new_with_free_func(g_free)))
+{
+}
+
+WebSoupRequestManager::~WebSoupRequestManager()
+{
+}
+
+void WebSoupRequestManager::didReceiveMessage(CoreIPC::Connection* connection, CoreIPC::MessageID messageID, CoreIPC::ArgumentDecoder* arguments)
+{
+    didReceiveWebSoupRequestManagerMessage(connection, messageID, arguments);
+}
+
+void WebSoupRequestManager::registerURIScheme(const String& scheme)
+{
+    if (m_schemes->len)
+        g_ptr_array_remove_index_fast(m_schemes.get(), m_schemes->len - 1);
+    g_ptr_array_add(m_schemes.get(), g_strdup(scheme.utf8().data()));
+    g_ptr_array_add(m_schemes.get(), 0);
+
+    SoupSession* session = WebCore::ResourceHandle::defaultSession();
+    GRefPtr<SoupRequester> requester = SOUP_REQUESTER(soup_session_get_feature(session, SOUP_TYPE_REQUESTER));
+    if (requester)
+        soup_session_feature_remove_feature(SOUP_SESSION_FEATURE(requester.get()), WEBKIT_TYPE_SOUP_REQUEST_GENERIC);
+    else {
+        requester = adoptGRef(soup_requester_new());
+        soup_session_add_feature(session, SOUP_SESSION_FEATURE(requester.get()));
+    }
+    SoupRequestClass* genericRequestClass = static_cast<SoupRequestClass*>(g_type_class_ref(WEBKIT_TYPE_SOUP_REQUEST_GENERIC));
+    genericRequestClass->schemes = const_cast<const char**>(reinterpret_cast<char**>(m_schemes->pdata));
+    soup_session_feature_add_feature(SOUP_SESSION_FEATURE(requester.get()), WEBKIT_TYPE_SOUP_REQUEST_GENERIC);
+}
+
+void WebSoupRequestManager::handleURIRequest(const CoreIPC::DataReference& requestData, const String& mimeType, uint64_t requestID)
+{
+    GRefPtr<GSimpleAsyncResult> result = adoptGRef(m_requestMap.take(requestID));
+    ASSERT(result.get());
+
+    GRefPtr<WebKitSoupRequestGeneric> request = adoptGRef(WEBKIT_SOUP_REQUEST_GENERIC(g_async_result_get_source_object(G_ASYNC_RESULT(result.get()))));
+    if (requestData.size()) {
+        webkitSoupRequestGenericSetContentLength(request.get(), requestData.size());
+        webkitSoupRequestGenericSetContentType(request.get(), mimeType.utf8().data());
+        GInputStream* dataStream = g_memory_input_stream_new_from_data(requestData.data(), requestData.size(), 0);
+        g_simple_async_result_set_op_res_gpointer(result.get(), dataStream, g_object_unref);
+    } else {
+        GOwnPtr<char> uriString(soup_uri_to_string(soup_request_get_uri(SOUP_REQUEST(request.get())), FALSE));
+        WebCore::ResourceRequest resourceRequest(String::fromUTF8(uriString.get()));
+        WebCore::ResourceError resourceError(WebCore::cannotShowURLError(resourceRequest));
+        g_simple_async_result_set_error(result.get(), g_quark_from_string(resourceError.domain().utf8().data()),
+                                        resourceError.errorCode(), "%s", resourceError.localizedDescription().utf8().data());
+    }
+    g_simple_async_result_complete(result.get());
+}
+
+void WebSoupRequestManager::send(GSimpleAsyncResult* result)
+{
+    GRefPtr<WebKitSoupRequestGeneric> request = adoptGRef(WEBKIT_SOUP_REQUEST_GENERIC(g_async_result_get_source_object(G_ASYNC_RESULT(result))));
+    SoupURI* uri = soup_request_get_uri(SOUP_REQUEST(request.get()));
+    GOwnPtr<char> uriString(soup_uri_to_string(uri, FALSE));
+
+    uint64_t requestID = generateSoupRequestID();
+    m_requestMap.set(requestID, result);
+    m_process->connection()->send(Messages::WebSoupRequestManagerProxy::DidReceiveURIRequest(String::fromUTF8(uriString.get()), requestID), 0);
+}
+
+GInputStream* WebSoupRequestManager::finish(GSimpleAsyncResult* result)
+{
+    return G_INPUT_STREAM(g_object_ref(g_simple_async_result_get_op_res_gpointer(result)));
+}
+
+} // namespace WebKit