WebResourceLoader.cpp [plain text]
#include "config.h"
#include "WebResourceLoader.h"
#if ENABLE(NETWORK_PROCESS)
#include "DataReference.h"
#include "Logging.h"
#include "NetworkProcessConnection.h"
#include "NetworkResourceLoaderMessages.h"
#include "WebCoreArgumentCoders.h"
#include "WebErrors.h"
#include "WebProcess.h"
#include <WebCore/ApplicationCacheHost.h>
#include <WebCore/CertificateInfo.h>
#include <WebCore/DocumentLoader.h>
#include <WebCore/ResourceError.h>
#include <WebCore/ResourceLoader.h>
using namespace WebCore;
namespace WebKit {
Ref<WebResourceLoader> WebResourceLoader::create(PassRefPtr<ResourceLoader> coreLoader)
{
return adoptRef(*new WebResourceLoader(coreLoader));
}
WebResourceLoader::WebResourceLoader(PassRefPtr<WebCore::ResourceLoader> coreLoader)
: m_coreLoader(coreLoader)
{
}
WebResourceLoader::~WebResourceLoader()
{
}
IPC::Connection* WebResourceLoader::messageSenderConnection()
{
return WebProcess::singleton().networkConnection()->connection();
}
uint64_t WebResourceLoader::messageSenderDestinationID()
{
return m_coreLoader->identifier();
}
void WebResourceLoader::cancelResourceLoader()
{
m_coreLoader->cancel();
}
void WebResourceLoader::detachFromCoreLoader()
{
m_coreLoader = nullptr;
}
void WebResourceLoader::willSendRequest(const ResourceRequest& proposedRequest, const ResourceResponse& redirectResponse)
{
LOG(Network, "(WebProcess) WebResourceLoader::willSendRequest to '%s'", proposedRequest.url().string().utf8().data());
RefPtr<WebResourceLoader> protect(this);
ResourceRequest newRequest = proposedRequest;
if (m_coreLoader->documentLoader()->applicationCacheHost()->maybeLoadFallbackForRedirect(m_coreLoader.get(), newRequest, redirectResponse))
return;
m_coreLoader->willSendRequest(WTF::move(newRequest), redirectResponse, [protect](ResourceRequest& request) {
if (!protect->m_coreLoader)
return;
if (!request.isNull())
protect->send(Messages::NetworkResourceLoader::ContinueWillSendRequest(request));
else
protect->m_coreLoader->cancel();
});
}
void WebResourceLoader::didSendData(uint64_t bytesSent, uint64_t totalBytesToBeSent)
{
m_coreLoader->didSendData(bytesSent, totalBytesToBeSent);
}
void WebResourceLoader::didReceiveResponse(const ResourceResponse& response, bool needsContinueDidReceiveResponseMessage)
{
LOG(Network, "(WebProcess) WebResourceLoader::didReceiveResponseWithCertificateInfo for '%s'. Status %d.", m_coreLoader->url().string().utf8().data(), response.httpStatusCode());
Ref<WebResourceLoader> protect(*this);
if (m_coreLoader->documentLoader()->applicationCacheHost()->maybeLoadFallbackForResponse(m_coreLoader.get(), response))
return;
bool shoudCallCoreLoaderDidReceiveResponse = true;
#if USE(QUICK_LOOK)
bool isMainLoad = m_coreLoader->documentLoader()->mainResourceLoader() == m_coreLoader;
if (isMainLoad && QuickLookHandle::shouldCreateForMIMEType(response.mimeType())) {
m_coreLoader->documentLoader()->setQuickLookHandle(QuickLookHandle::create(*m_coreLoader, response));
shoudCallCoreLoaderDidReceiveResponse = false;
}
#endif
if (shoudCallCoreLoaderDidReceiveResponse)
m_coreLoader->didReceiveResponse(response);
if (!m_coreLoader)
return;
if (needsContinueDidReceiveResponseMessage)
send(Messages::NetworkResourceLoader::ContinueDidReceiveResponse());
}
void WebResourceLoader::didReceiveData(const IPC::DataReference& data, int64_t encodedDataLength)
{
LOG(Network, "(WebProcess) WebResourceLoader::didReceiveData of size %i for '%s'", (int)data.size(), m_coreLoader->url().string().utf8().data());
#if USE(QUICK_LOOK)
if (QuickLookHandle* quickLookHandle = m_coreLoader->documentLoader()->quickLookHandle()) {
if (quickLookHandle->didReceiveData(adoptCF(CFDataCreate(kCFAllocatorDefault, data.data(), data.size())).get()))
return;
}
#endif
m_coreLoader->didReceiveData(reinterpret_cast<const char*>(data.data()), data.size(), encodedDataLength, DataPayloadBytes);
}
void WebResourceLoader::didFinishResourceLoad(double finishTime)
{
LOG(Network, "(WebProcess) WebResourceLoader::didFinishResourceLoad for '%s'", m_coreLoader->url().string().utf8().data());
#if USE(QUICK_LOOK)
if (QuickLookHandle* quickLookHandle = m_coreLoader->documentLoader()->quickLookHandle()) {
if (quickLookHandle->didFinishLoading())
return;
}
#endif
m_coreLoader->didFinishLoading(finishTime);
}
void WebResourceLoader::didFailResourceLoad(const ResourceError& error)
{
LOG(Network, "(WebProcess) WebResourceLoader::didFailResourceLoad for '%s'", m_coreLoader->url().string().utf8().data());
#if USE(QUICK_LOOK)
if (QuickLookHandle* quickLookHandle = m_coreLoader->documentLoader()->quickLookHandle())
quickLookHandle->didFail();
#endif
if (m_coreLoader->documentLoader()->applicationCacheHost()->maybeLoadFallbackForError(m_coreLoader.get(), error))
return;
m_coreLoader->didFail(error);
}
#if ENABLE(SHAREABLE_RESOURCE)
void WebResourceLoader::didReceiveResource(const ShareableResource::Handle& handle, double finishTime)
{
LOG(Network, "(WebProcess) WebResourceLoader::didReceiveResource for '%s'", m_coreLoader->url().string().utf8().data());
RefPtr<SharedBuffer> buffer = handle.tryWrapInSharedBuffer();
#if USE(QUICK_LOOK)
if (QuickLookHandle* quickLookHandle = m_coreLoader->documentLoader()->quickLookHandle()) {
if (buffer) {
if (quickLookHandle->didReceiveData(buffer->existingCFData())) {
quickLookHandle->didFinishLoading();
return;
}
} else
quickLookHandle->didFail();
}
#endif
if (!buffer) {
LOG_ERROR("Unable to create buffer from ShareableResource sent from the network process.");
m_coreLoader->didFail(internalError(m_coreLoader->request().url()));
return;
}
Ref<WebResourceLoader> protect(*this);
if (buffer->size())
m_coreLoader->didReceiveBuffer(buffer.get(), buffer->size(), DataPayloadWholeResource);
if (!m_coreLoader)
return;
m_coreLoader->didFinishLoading(finishTime);
}
#endif
#if USE(PROTECTION_SPACE_AUTH_CALLBACK)
void WebResourceLoader::canAuthenticateAgainstProtectionSpace(const ProtectionSpace& protectionSpace)
{
Ref<WebResourceLoader> protect(*this);
bool result = m_coreLoader->canAuthenticateAgainstProtectionSpace(protectionSpace);
if (!m_coreLoader)
return;
send(Messages::NetworkResourceLoader::ContinueCanAuthenticateAgainstProtectionSpace(result));
}
#endif
}
#endif // ENABLE(NETWORK_PROCESS)