WebSoupRequestManager.cpp [plain text]
#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)));
}
}