WorkerThreadableLoader.cpp [plain text]
#include "config.h"
#include "WorkerThreadableLoader.h"
#include "Document.h"
#include "DocumentThreadableLoader.h"
#include "ResourceError.h"
#include "ResourceRequest.h"
#include "ResourceResponse.h"
#include "SecurityOrigin.h"
#include "ThreadableLoader.h"
#include "WorkerGlobalScope.h"
#include "WorkerLoaderProxy.h"
#include "WorkerThread.h"
#include <wtf/MainThread.h>
#include <wtf/Vector.h>
namespace WebCore {
static const char loadResourceSynchronouslyMode[] = "loadResourceSynchronouslyMode";
WorkerThreadableLoader::WorkerThreadableLoader(WorkerGlobalScope* workerGlobalScope, ThreadableLoaderClient* client, const String& taskMode, const ResourceRequest& request, const ThreadableLoaderOptions& options)
: m_workerGlobalScope(workerGlobalScope)
, m_workerClientWrapper(ThreadableLoaderClientWrapper::create(client))
, m_bridge(*(new MainThreadBridge(m_workerClientWrapper, m_workerGlobalScope->thread().workerLoaderProxy(), taskMode, request, options, workerGlobalScope->url().strippedForUseAsReferrer())))
{
}
WorkerThreadableLoader::~WorkerThreadableLoader()
{
m_bridge.destroy();
}
void WorkerThreadableLoader::loadResourceSynchronously(WorkerGlobalScope* workerGlobalScope, const ResourceRequest& request, ThreadableLoaderClient& client, const ThreadableLoaderOptions& options)
{
WorkerRunLoop& runLoop = workerGlobalScope->thread().runLoop();
String mode = loadResourceSynchronouslyMode;
mode.append(String::number(runLoop.createUniqueId()));
RefPtr<WorkerThreadableLoader> loader = WorkerThreadableLoader::create(workerGlobalScope, &client, mode, request, options);
MessageQueueWaitResult result = MessageQueueMessageReceived;
while (!loader->done() && result != MessageQueueTerminated)
result = runLoop.runInMode(workerGlobalScope, mode);
if (!loader->done() && result == MessageQueueTerminated)
loader->cancel();
}
void WorkerThreadableLoader::cancel()
{
m_bridge.cancel();
}
WorkerThreadableLoader::MainThreadBridge::MainThreadBridge(PassRefPtr<ThreadableLoaderClientWrapper> workerClientWrapper, WorkerLoaderProxy& loaderProxy, const String& taskMode,
const ResourceRequest& request, const ThreadableLoaderOptions& options, const String& outgoingReferrer)
: m_workerClientWrapper(workerClientWrapper)
, m_loaderProxy(loaderProxy)
, m_taskMode(taskMode.isolatedCopy())
{
ASSERT(m_workerClientWrapper.get());
auto* requestData = request.copyData().release();
auto* optionsCopy = options.isolatedCopy().release();
StringCapture capturedOutgoingReferrer(outgoingReferrer);
m_loaderProxy.postTaskToLoader([this, requestData, optionsCopy, capturedOutgoingReferrer](ScriptExecutionContext& context) {
ASSERT(isMainThread());
Document& document = downcast<Document>(context);
auto request = ResourceRequest::adopt(std::unique_ptr<CrossThreadResourceRequestData>(requestData));
request->setHTTPReferrer(capturedOutgoingReferrer.string());
auto options = std::unique_ptr<ThreadableLoaderOptions>(optionsCopy);
m_mainThreadLoader = DocumentThreadableLoader::create(document, *this, *request, *options);
ASSERT(m_mainThreadLoader);
});
}
void WorkerThreadableLoader::MainThreadBridge::destroy()
{
clearClientWrapper();
m_loaderProxy.postTaskToLoader([this] (ScriptExecutionContext& context) {
ASSERT(isMainThread());
ASSERT_UNUSED(context, context.isDocument());
delete this;
});
}
void WorkerThreadableLoader::MainThreadBridge::cancel()
{
m_loaderProxy.postTaskToLoader([this] (ScriptExecutionContext& context) {
ASSERT(isMainThread());
ASSERT_UNUSED(context, context.isDocument());
if (!m_mainThreadLoader)
return;
m_mainThreadLoader->cancel();
m_mainThreadLoader = nullptr;
});
ThreadableLoaderClientWrapper* clientWrapper = m_workerClientWrapper.get();
if (!clientWrapper->done()) {
ResourceError error(String(), 0, String(), String());
error.setIsCancellation(true);
clientWrapper->didFail(error);
}
clearClientWrapper();
}
void WorkerThreadableLoader::MainThreadBridge::clearClientWrapper()
{
m_workerClientWrapper->clearClient();
}
void WorkerThreadableLoader::MainThreadBridge::didSendData(unsigned long long bytesSent, unsigned long long totalBytesToBeSent)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, bytesSent, totalBytesToBeSent] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didSendData(bytesSent, totalBytesToBeSent);
}, m_taskMode);
}
void WorkerThreadableLoader::MainThreadBridge::didReceiveResponse(unsigned long identifier, const ResourceResponse& response)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
auto* responseData = response.copyData().release();
if (!m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, identifier, responseData] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
auto response(ResourceResponse::adopt(std::unique_ptr<CrossThreadResourceResponseData>(responseData)));
workerClientWrapper->didReceiveResponse(identifier, *response);
}, m_taskMode))
delete responseData;
}
void WorkerThreadableLoader::MainThreadBridge::didReceiveData(const char* data, int dataLength)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
Vector<char>* vectorPtr = new Vector<char>(dataLength);
memcpy(vectorPtr->data(), data, dataLength);
if (!m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, vectorPtr] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didReceiveData(vectorPtr->data(), vectorPtr->size());
delete vectorPtr;
}, m_taskMode))
delete vectorPtr;
}
void WorkerThreadableLoader::MainThreadBridge::didFinishLoading(unsigned long identifier, double finishTime)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, identifier, finishTime] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didFinishLoading(identifier, finishTime);
}, m_taskMode);
}
void WorkerThreadableLoader::MainThreadBridge::didFail(const ResourceError& error)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
ResourceError* capturedError = new ResourceError(error.copy());
if (!m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, capturedError] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didFail(*capturedError);
delete capturedError;
}, m_taskMode))
delete capturedError;
}
void WorkerThreadableLoader::MainThreadBridge::didFailAccessControlCheck(const ResourceError& error)
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
ResourceError* capturedError = new ResourceError(error.copy());
if (!m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper, capturedError] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didFailAccessControlCheck(*capturedError);
delete capturedError;
}, m_taskMode))
delete capturedError;
}
void WorkerThreadableLoader::MainThreadBridge::didFailRedirectCheck()
{
RefPtr<ThreadableLoaderClientWrapper> workerClientWrapper = m_workerClientWrapper;
m_loaderProxy.postTaskForModeToWorkerGlobalScope([workerClientWrapper] (ScriptExecutionContext& context) {
ASSERT_UNUSED(context, context.isWorkerGlobalScope());
workerClientWrapper->didFailRedirectCheck();
}, m_taskMode);
}
}