WebURLSchemeTask.cpp [plain text]
#include "config.h"
#include "WebURLSchemeTask.h"
#include "DataReference.h"
#include "WebPageMessages.h"
#include "WebPageProxy.h"
#include "WebURLSchemeHandler.h"
using namespace WebCore;
namespace WebKit {
Ref<WebURLSchemeTask> WebURLSchemeTask::create(WebURLSchemeHandler& handler, WebPageProxy& page, uint64_t resourceIdentifier, ResourceRequest&& request)
{
return adoptRef(*new WebURLSchemeTask(handler, page, resourceIdentifier, WTFMove(request)));
}
WebURLSchemeTask::WebURLSchemeTask(WebURLSchemeHandler& handler, WebPageProxy& page, uint64_t resourceIdentifier, ResourceRequest&& request)
: m_urlSchemeHandler(handler)
, m_page(&page)
, m_identifier(resourceIdentifier)
, m_pageIdentifier(page.pageID())
, m_request(WTFMove(request))
{
}
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;
m_request = request;
m_page->send(Messages::WebPage::URLSchemeTaskDidPerformRedirection(m_urlSchemeHandler->identifier(), m_identifier, response, request));
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();
m_page->send(Messages::WebPage::URLSchemeTaskDidReceiveResponse(m_urlSchemeHandler->identifier(), m_identifier, response));
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;
m_page->send(Messages::WebPage::URLSchemeTaskDidReceiveData(m_urlSchemeHandler->identifier(), m_identifier, IPC::SharedBufferDataReference(buffer.ptr())));
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;
m_page->send(Messages::WebPage::URLSchemeTaskDidComplete(m_urlSchemeHandler->identifier(), m_identifier, error));
m_urlSchemeHandler->taskCompleted(*this);
return ExceptionType::None;
}
void WebURLSchemeTask::pageDestroyed()
{
ASSERT(m_page);
m_page = nullptr;
m_stopped = true;
}
void WebURLSchemeTask::stop()
{
ASSERT(!m_stopped);
m_stopped = true;
}
}