NetworkResourceLoader.cpp [plain text]
#include "config.h"
#include "NetworkResourceLoader.h"
#if ENABLE(NETWORK_PROCESS)
#include "AuthenticationManager.h"
#include "DataReference.h"
#include "Logging.h"
#include "NetworkConnectionToWebProcess.h"
#include "NetworkProcess.h"
#include "NetworkProcessConnectionMessages.h"
#include "NetworkResourceLoadParameters.h"
#include "PlatformCertificateInfo.h"
#include "RemoteNetworkingContext.h"
#include "ShareableResource.h"
#include "SharedMemory.h"
#include "WebCoreArgumentCoders.h"
#include "WebErrors.h"
#include "WebResourceLoaderMessages.h"
#include <WebCore/NotImplemented.h>
#include <WebCore/ResourceBuffer.h>
#include <WebCore/ResourceHandle.h>
#include <wtf/CurrentTime.h>
#include <wtf/MainThread.h>
using namespace WebCore;
namespace WebKit {
NetworkResourceLoader::NetworkResourceLoader(const NetworkResourceLoadParameters& loadParameters, NetworkConnectionToWebProcess* connection)
: SchedulableLoader(loadParameters, connection)
, m_bytesReceived(0)
, m_handleConvertedToDownload(false)
{
ASSERT(isMainThread());
}
NetworkResourceLoader::~NetworkResourceLoader()
{
ASSERT(isMainThread());
ASSERT(!m_handle);
}
void NetworkResourceLoader::start()
{
ASSERT(isMainThread());
ref();
m_networkingContext = RemoteNetworkingContext::create(false, false, inPrivateBrowsingMode(), shouldClearReferrerOnHTTPSToHTTPRedirect());
consumeSandboxExtensions();
m_handle = ResourceHandle::create(m_networkingContext.get(), request(), this, false , contentSniffingPolicy() == SniffContent);
}
void NetworkResourceLoader::cleanup()
{
ASSERT(isMainThread());
invalidateSandboxExtensions();
if (FormData* formData = request().httpBody())
formData->removeGeneratedFilesIfNeeded();
NetworkProcess::shared().networkResourceLoadScheduler().scheduleRemoveLoader(this);
if (m_handle) {
m_handle = 0;
deref();
}
}
template<typename U> bool NetworkResourceLoader::sendAbortingOnFailure(const U& message, unsigned messageSendFlags)
{
bool result = messageSenderConnection()->send(message, messageSenderDestinationID(), messageSendFlags);
if (!result)
abort();
return result;
}
void NetworkResourceLoader::didConvertHandleToDownload()
{
ASSERT(m_handle);
m_handleConvertedToDownload = true;
}
void NetworkResourceLoader::abort()
{
ASSERT(isMainThread());
if (m_handle && !m_handleConvertedToDownload)
m_handle->cancel();
cleanup();
}
void NetworkResourceLoader::didReceiveResponseAsync(ResourceHandle* handle, const ResourceResponse& response)
{
ASSERT_UNUSED(handle, handle == m_handle);
if (FormData* formData = request().httpBody())
formData->removeGeneratedFilesIfNeeded();
sendAbortingOnFailure(Messages::WebResourceLoader::DidReceiveResponseWithCertificateInfo(response, PlatformCertificateInfo(response), isLoadingMainResource()));
if (!m_handle)
return;
if (!isLoadingMainResource()) {
m_handle->continueDidReceiveResponse();
}
}
void NetworkResourceLoader::didReceiveData(ResourceHandle*, const char* data, int length, int encodedDataLength)
{
ASSERT_NOT_REACHED();
}
void NetworkResourceLoader::didReceiveBuffer(ResourceHandle* handle, PassRefPtr<SharedBuffer> buffer, int encodedDataLength)
{
ASSERT_UNUSED(handle, handle == m_handle);
m_bytesReceived += buffer->size();
#if __MAC_OS_X_VERSION_MIN_REQUIRED >= 1090
ShareableResource::Handle shareableResourceHandle;
tryGetShareableHandleFromSharedBuffer(shareableResourceHandle, buffer.get());
if (!shareableResourceHandle.isNull()) {
abort();
send(Messages::WebResourceLoader::DidReceiveResource(shareableResourceHandle, currentTime()));
return;
}
#endif // __MAC_OS_X_VERSION_MIN_REQUIRED >= 1090
CoreIPC::DataReference dataReference(reinterpret_cast<const uint8_t*>(buffer->data()), buffer->size());
sendAbortingOnFailure(Messages::WebResourceLoader::DidReceiveData(dataReference, encodedDataLength));
}
void NetworkResourceLoader::didFinishLoading(ResourceHandle* handle, double finishTime)
{
ASSERT_UNUSED(handle, handle == m_handle);
send(Messages::WebResourceLoader::DidFinishResourceLoad(finishTime));
cleanup();
}
void NetworkResourceLoader::didFail(ResourceHandle* handle, const ResourceError& error)
{
ASSERT_UNUSED(handle, handle == m_handle);
send(Messages::WebResourceLoader::DidFailResourceLoad(error));
cleanup();
}
void NetworkResourceLoader::willSendRequestAsync(ResourceHandle* handle, const ResourceRequest& request, const ResourceResponse& redirectResponse)
{
ASSERT_UNUSED(handle, handle == m_handle);
ASSERT(!redirectResponse.isNull());
ASSERT(isMainThread());
m_suggestedRequestForWillSendRequest = request;
sendAbortingOnFailure(Messages::WebResourceLoader::WillSendRequest(request, redirectResponse), CoreIPC::DispatchMessageEvenWhenWaitingForSyncReply);
}
void NetworkResourceLoader::continueWillSendRequest(const ResourceRequest& newRequest)
{
m_suggestedRequestForWillSendRequest.updateFromDelegatePreservingOldHTTPBody(newRequest.nsURLRequest(DoNotUpdateHTTPBody));
RunLoop::main()->dispatch(bind(&NetworkResourceLoadScheduler::receivedRedirect, &NetworkProcess::shared().networkResourceLoadScheduler(), this, m_suggestedRequestForWillSendRequest.url()));
m_handle->continueWillSendRequest(m_suggestedRequestForWillSendRequest);
m_suggestedRequestForWillSendRequest = ResourceRequest();
}
void NetworkResourceLoader::continueDidReceiveResponse()
{
if (!m_handle)
return;
m_handle->continueDidReceiveResponse();
}
void NetworkResourceLoader::didSendData(ResourceHandle* handle, unsigned long long bytesSent, unsigned long long totalBytesToBeSent)
{
ASSERT_UNUSED(handle, handle == m_handle);
send(Messages::WebResourceLoader::DidSendData(bytesSent, totalBytesToBeSent));
}
void NetworkResourceLoader::wasBlocked(ResourceHandle* handle)
{
ASSERT_UNUSED(handle, handle == m_handle);
didFail(handle, WebKit::blockedError(request()));
}
void NetworkResourceLoader::cannotShowURL(ResourceHandle* handle)
{
ASSERT_UNUSED(handle, handle == m_handle);
didFail(handle, WebKit::cannotShowURLError(request()));
}
bool NetworkResourceLoader::shouldUseCredentialStorage(ResourceHandle* handle)
{
ASSERT_UNUSED(handle, handle == m_handle || !m_handle);
return allowStoredCredentials() == AllowStoredCredentials;
}
void NetworkResourceLoader::shouldUseCredentialStorageAsync(ResourceHandle* handle)
{
ASSERT_UNUSED(handle, handle == m_handle);
handle->continueShouldUseCredentialStorage(shouldUseCredentialStorage(handle));
}
void NetworkResourceLoader::didReceiveAuthenticationChallenge(ResourceHandle* handle, const AuthenticationChallenge& challenge)
{
ASSERT_UNUSED(handle, handle == m_handle);
if (clientCredentialPolicy() == DoNotAskClientForAnyCredentials) {
challenge.authenticationClient()->receivedRequestToContinueWithoutCredential(challenge);
return;
}
NetworkProcess::shared().authenticationManager().didReceiveAuthenticationChallenge(webPageID(), webFrameID(), challenge);
}
void NetworkResourceLoader::didCancelAuthenticationChallenge(ResourceHandle* handle, const AuthenticationChallenge& challenge)
{
ASSERT_UNUSED(handle, handle == m_handle);
notImplemented();
}
CoreIPC::Connection* NetworkResourceLoader::messageSenderConnection()
{
return connectionToWebProcess()->connection();
}
uint64_t NetworkResourceLoader::messageSenderDestinationID()
{
return identifier();
}
#if USE(PROTECTION_SPACE_AUTH_CALLBACK)
void NetworkResourceLoader::canAuthenticateAgainstProtectionSpaceAsync(ResourceHandle* handle, const ProtectionSpace& protectionSpace)
{
ASSERT(isMainThread());
ASSERT_UNUSED(handle, handle == m_handle);
sendAbortingOnFailure(Messages::WebResourceLoader::CanAuthenticateAgainstProtectionSpace(protectionSpace), CoreIPC::DispatchMessageEvenWhenWaitingForSyncReply);
}
void NetworkResourceLoader::continueCanAuthenticateAgainstProtectionSpace(bool result)
{
m_handle->continueCanAuthenticateAgainstProtectionSpace(result);
}
#endif
#if USE(NETWORK_CFDATA_ARRAY_CALLBACK)
bool NetworkResourceLoader::supportsDataArray()
{
notImplemented();
return false;
}
void NetworkResourceLoader::didReceiveDataArray(ResourceHandle*, CFArrayRef)
{
ASSERT_NOT_REACHED();
notImplemented();
}
#endif
#if PLATFORM(MAC)
void NetworkResourceLoader::willStopBufferingData(ResourceHandle*, const char*, int)
{
notImplemented();
}
#endif // PLATFORM(MAC)
}
#endif // ENABLE(NETWORK_PROCESS)