WebURLSchemeHandlerProxy.cpp [plain text]
#include "config.h"
#include "WebURLSchemeHandlerProxy.h"
#include "DataReference.h"
#include "URLSchemeTaskParameters.h"
#include "WebCoreArgumentCoders.h"
#include "WebErrors.h"
#include "WebFrame.h"
#include "WebLoaderStrategy.h"
#include "WebPage.h"
#include "WebPageProxyMessages.h"
#include "WebProcess.h"
#include <WebCore/ResourceError.h>
#include <WebCore/ResourceLoader.h>
#include <WebCore/ResourceRequest.h>
#include <WebCore/ResourceResponse.h>
namespace WebKit {
using namespace WebCore;
WebURLSchemeHandlerProxy::WebURLSchemeHandlerProxy(WebPage& page, uint64_t identifier)
: m_webPage(page)
, m_identifier(identifier)
{
}
WebURLSchemeHandlerProxy::~WebURLSchemeHandlerProxy()
{
ASSERT(m_tasks.isEmpty());
}
void WebURLSchemeHandlerProxy::startNewTask(ResourceLoader& loader, WebFrame& webFrame)
{
auto result = m_tasks.add(loader.identifier(), WebURLSchemeTaskProxy::create(*this, loader, webFrame));
ASSERT(result.isNewEntry);
WebProcess::singleton().webLoaderStrategy().addURLSchemeTaskProxy(*result.iterator->value);
result.iterator->value->startLoading();
}
void WebURLSchemeHandlerProxy::loadSynchronously(ResourceLoadIdentifier loadIdentifier, WebFrame& webFrame, const ResourceRequest& request, ResourceResponse& response, ResourceError& error, Vector<char>& data)
{
data.shrink(0);
if (!m_webPage.sendSync(Messages::WebPageProxy::LoadSynchronousURLSchemeTask(URLSchemeTaskParameters { m_identifier, loadIdentifier, request, webFrame.info() }), Messages::WebPageProxy::LoadSynchronousURLSchemeTask::Reply(response, error, data))) {
error = failedCustomProtocolSyncLoad(request);
return;
}
}
void WebURLSchemeHandlerProxy::stopAllTasks()
{
while (!m_tasks.isEmpty())
m_tasks.begin()->value->stopLoading();
}
void WebURLSchemeHandlerProxy::taskDidPerformRedirection(uint64_t taskIdentifier, WebCore::ResourceResponse&& redirectResponse, WebCore::ResourceRequest&& newRequest)
{
auto* task = m_tasks.get(taskIdentifier);
if (!task)
return;
task->didPerformRedirection(WTFMove(redirectResponse), WTFMove(newRequest));
}
void WebURLSchemeHandlerProxy::taskDidReceiveResponse(uint64_t taskIdentifier, const ResourceResponse& response)
{
auto* task = m_tasks.get(taskIdentifier);
if (!task)
return;
task->didReceiveResponse(response);
}
void WebURLSchemeHandlerProxy::taskDidReceiveData(uint64_t taskIdentifier, size_t size, const uint8_t* data)
{
auto* task = m_tasks.get(taskIdentifier);
if (!task)
return;
task->didReceiveData(size, data);
}
void WebURLSchemeHandlerProxy::taskDidComplete(uint64_t taskIdentifier, const ResourceError& error)
{
if (auto task = removeTask(taskIdentifier))
task->didComplete(error);
}
void WebURLSchemeHandlerProxy::taskDidStopLoading(WebURLSchemeTaskProxy& task)
{
ASSERT(m_tasks.get(task.identifier()) == &task);
removeTask(task.identifier());
}
RefPtr<WebURLSchemeTaskProxy> WebURLSchemeHandlerProxy::removeTask(uint64_t identifier)
{
auto task = m_tasks.take(identifier);
if (!task)
return nullptr;
WebProcess::singleton().webLoaderStrategy().removeURLSchemeTaskProxy(*task);
return task;
}
}