SchedulableLoader.cpp [plain text]
#include "config.h"
#include "SchedulableLoader.h"
#if ENABLE(NETWORK_PROCESS)
#include "NetworkBlobRegistry.h"
#include "NetworkConnectionToWebProcess.h"
#include "NetworkResourceLoadParameters.h"
#include <WebCore/FormData.h>
using namespace WebCore;
namespace WebKit {
SchedulableLoader::SchedulableLoader(const NetworkResourceLoadParameters& parameters, NetworkConnectionToWebProcess* connection)
: m_identifier(parameters.identifier)
, m_webPageID(parameters.webPageID)
, m_webFrameID(parameters.webFrameID)
, m_request(parameters.request)
, m_priority(parameters.priority)
, m_contentSniffingPolicy(parameters.contentSniffingPolicy)
, m_allowStoredCredentials(parameters.allowStoredCredentials)
, m_clientCredentialPolicy(parameters.clientCredentialPolicy)
, m_inPrivateBrowsingMode(parameters.inPrivateBrowsingMode)
, m_shouldClearReferrerOnHTTPSToHTTPRedirect(parameters.shouldClearReferrerOnHTTPSToHTTPRedirect)
, m_isLoadingMainResource(parameters.isMainResource)
, m_sandboxExtensionsAreConsumed(false)
, m_connection(connection)
{
ASSERT((m_webPageID && m_webFrameID) || m_clientCredentialPolicy == DoNotAskClientForAnyCredentials);
for (size_t i = 0, count = parameters.requestBodySandboxExtensions.size(); i < count; ++i) {
if (RefPtr<SandboxExtension> extension = SandboxExtension::create(parameters.requestBodySandboxExtensions[i]))
m_requestBodySandboxExtensions.append(extension);
}
#if ENABLE(BLOB)
if (m_request.httpBody()) {
const Vector<FormDataElement>& elements = m_request.httpBody()->elements();
for (size_t i = 0, count = elements.size(); i < count; ++i) {
if (elements[i].m_type == FormDataElement::encodedBlob) {
Vector<RefPtr<SandboxExtension>> blobElementExtensions = NetworkBlobRegistry::shared().sandboxExtensions(elements[i].m_url);
m_requestBodySandboxExtensions.appendVector(blobElementExtensions);
}
}
}
if (m_request.url().protocolIs("blob")) {
ASSERT(!SandboxExtension::create(parameters.resourceSandboxExtension));
m_resourceSandboxExtensions = NetworkBlobRegistry::shared().sandboxExtensions(m_request.url());
} else
#endif
if (RefPtr<SandboxExtension> resourceSandboxExtension = SandboxExtension::create(parameters.resourceSandboxExtension))
m_resourceSandboxExtensions.append(resourceSandboxExtension);
}
SchedulableLoader::~SchedulableLoader()
{
ASSERT(!m_hostRecord);
}
void SchedulableLoader::consumeSandboxExtensions()
{
for (size_t i = 0, count = m_requestBodySandboxExtensions.size(); i < count; ++i)
m_requestBodySandboxExtensions[i]->consume();
for (size_t i = 0, count = m_resourceSandboxExtensions.size(); i < count; ++i)
m_resourceSandboxExtensions[i]->consume();
m_sandboxExtensionsAreConsumed = true;
}
void SchedulableLoader::invalidateSandboxExtensions()
{
if (m_sandboxExtensionsAreConsumed) {
for (size_t i = 0, count = m_requestBodySandboxExtensions.size(); i < count; ++i)
m_requestBodySandboxExtensions[i]->revoke();
for (size_t i = 0, count = m_resourceSandboxExtensions.size(); i < count; ++i)
m_resourceSandboxExtensions[i]->revoke();
}
m_requestBodySandboxExtensions.clear();
m_resourceSandboxExtensions.clear();
m_sandboxExtensionsAreConsumed = false;
}
}
#endif // ENABLE(NETWORK_PROCESS)