WebURLSchemeTask.cpp [plain text]
#include "config.h"
#include "WebURLSchemeTask.h"
#include "DataReference.h"
#include "SharedBufferDataReference.h"
#include "WebErrors.h"
#include "WebPageMessages.h"
#include "WebPageProxy.h"
#include "WebURLSchemeHandler.h"
namespace WebKit {
using namespace WebCore;
Ref<WebURLSchemeTask> WebURLSchemeTask::create(WebURLSchemeHandler& handler, WebPageProxy& page, WebProcessProxy& process, uint64_t resourceIdentifier, ResourceRequest&& request, SyncLoadCompletionHandler&& syncCompletionHandler)
{
return adoptRef(*new WebURLSchemeTask(handler, page, process, resourceIdentifier, WTFMove(request), WTFMove(syncCompletionHandler)));
}
WebURLSchemeTask::WebURLSchemeTask(WebURLSchemeHandler& handler, WebPageProxy& page, WebProcessProxy& process, uint64_t resourceIdentifier, ResourceRequest&& request, SyncLoadCompletionHandler&& syncCompletionHandler)
: m_urlSchemeHandler(handler)
, m_page(&page)
, m_process(makeRef(process))
, m_identifier(resourceIdentifier)
, m_pageIdentifier(page.pageID())
, m_request(WTFMove(request))
, m_syncCompletionHandler(WTFMove(syncCompletionHandler))
{
}
auto WebURLSchemeTask::didPerformRedirection(WebCore::ResourceResponse&& response, WebCore::ResourceRequest&& request) -> ExceptionType
{
if (m_stopped)
return ExceptionType::TaskAlreadyStopped;
if (m_completed)
return ExceptionType::CompleteAlreadyCalled;
if (m_dataSent)
return ExceptionType::DataAlreadySent;
if (m_responseSent)
return ExceptionType::RedirectAfterResponse;
if (isSync())
m_syncResponse = response;
m_request = request;
m_process->send(Messages::WebPage::URLSchemeTaskDidPerformRedirection(m_urlSchemeHandler->identifier(), m_identifier, response, request), m_page->pageID());
return ExceptionType::None;
}
auto WebURLSchemeTask::didReceiveResponse(const ResourceResponse& response) -> ExceptionType
{
if (m_stopped)
return ExceptionType::TaskAlreadyStopped;
if (m_completed)
return ExceptionType::CompleteAlreadyCalled;
if (m_dataSent)
return ExceptionType::DataAlreadySent;
m_responseSent = true;
response.includeCertificateInfo();
if (isSync())
m_syncResponse = response;
m_process->send(Messages::WebPage::URLSchemeTaskDidReceiveResponse(m_urlSchemeHandler->identifier(), m_identifier, response), m_page->pageID());
return ExceptionType::None;
}
auto WebURLSchemeTask::didReceiveData(Ref<SharedBuffer>&& buffer) -> ExceptionType
{
if (m_stopped)
return ExceptionType::TaskAlreadyStopped;
if (m_completed)
return ExceptionType::CompleteAlreadyCalled;
if (!m_responseSent)
return ExceptionType::NoResponseSent;
m_dataSent = true;
if (isSync()) {
if (m_syncData)
m_syncData->append(WTFMove(buffer));
else
m_syncData = WTFMove(buffer);
return ExceptionType::None;
}
m_process->send(Messages::WebPage::URLSchemeTaskDidReceiveData(m_urlSchemeHandler->identifier(), m_identifier, { buffer }), m_page->pageID());
return ExceptionType::None;
}
auto WebURLSchemeTask::didComplete(const ResourceError& error) -> ExceptionType
{
if (m_stopped)
return ExceptionType::TaskAlreadyStopped;
if (m_completed)
return ExceptionType::CompleteAlreadyCalled;
if (!m_responseSent && error.isNull())
return ExceptionType::NoResponseSent;
m_completed = true;
if (isSync()) {
IPC::DataReference data;
if (m_syncData)
data = { reinterpret_cast<const uint8_t*>(m_syncData->data()), m_syncData->size() };
m_syncCompletionHandler(m_syncResponse, error, data);
m_syncData = nullptr;
}
m_process->send(Messages::WebPage::URLSchemeTaskDidComplete(m_urlSchemeHandler->identifier(), m_identifier, error), m_page->pageID());
m_urlSchemeHandler->taskCompleted(*this);
return ExceptionType::None;
}
void WebURLSchemeTask::pageDestroyed()
{
ASSERT(m_page);
m_page = nullptr;
m_process = nullptr;
m_stopped = true;
if (isSync())
m_syncCompletionHandler({ }, failedCustomProtocolSyncLoad(m_request), { });
}
void WebURLSchemeTask::stop()
{
ASSERT(!m_stopped);
m_stopped = true;
if (isSync())
m_syncCompletionHandler({ }, failedCustomProtocolSyncLoad(m_request), { });
}
}