#include "config.h"
#include "Worker.h"
#include "DOMWindow.h"
#include "CachedResourceLoader.h"
#include "Document.h"
#include "EventException.h"
#include "EventListener.h"
#include "EventNames.h"
#include "ExceptionCode.h"
#include "Frame.h"
#include "FrameLoader.h"
#include "InspectorInstrumentation.h"
#include "MessageEvent.h"
#include "NetworkStateNotifier.h"
#include "TextEncoding.h"
#include "WorkerGlobalScopeProxy.h"
#include "WorkerScriptLoader.h"
#include "WorkerThread.h"
#include <wtf/HashSet.h>
#include <wtf/MainThread.h>
namespace WebCore {
static HashSet<Worker*>* allWorkers;
void networkStateChanged(bool isOnLine)
{
HashSet<Worker*>::iterator end = allWorkers->end();
for (HashSet<Worker*>::iterator it = allWorkers->begin(); it != end; ++it)
(*it)->notifyNetworkStateChange(isOnLine);
}
inline Worker::Worker(ScriptExecutionContext& context)
: AbstractWorker(context)
, m_contextProxy(WorkerGlobalScopeProxy::create(this))
{
if (!allWorkers) {
allWorkers = new HashSet<Worker*>;
networkStateNotifier().addNetworkStateChangeListener(networkStateChanged);
}
HashSet<Worker*>::AddResult addResult = allWorkers->add(this);
ASSERT_UNUSED(addResult, addResult.isNewEntry);
}
PassRefPtr<Worker> Worker::create(ScriptExecutionContext& context, const String& url, ExceptionCode& ec)
{
ASSERT(isMainThread());
ASSERT_WITH_SECURITY_IMPLICATION(context.isDocument());
RefPtr<Worker> worker = adoptRef(new Worker(context));
worker->suspendIfNeeded();
URL scriptURL = worker->resolveURL(url, ec);
if (scriptURL.isEmpty())
return 0;
worker->setPendingActivity(worker.get());
worker->m_scriptLoader = WorkerScriptLoader::create();
worker->m_scriptLoader->loadAsynchronously(&context, scriptURL, DenyCrossOriginRequests, worker.get());
return worker.release();
}
Worker::~Worker()
{
ASSERT(isMainThread());
ASSERT(scriptExecutionContext()); allWorkers->remove(this);
m_contextProxy->workerObjectDestroyed();
}
void Worker::postMessage(PassRefPtr<SerializedScriptValue> message, MessagePort* port, ExceptionCode& ec)
{
MessagePortArray ports;
if (port)
ports.append(port);
postMessage(message, &ports, ec);
}
void Worker::postMessage(PassRefPtr<SerializedScriptValue> message, const MessagePortArray* ports, ExceptionCode& ec)
{
std::unique_ptr<MessagePortChannelArray> channels = MessagePort::disentanglePorts(ports, ec);
if (ec)
return;
m_contextProxy->postMessageToWorkerGlobalScope(message, WTF::move(channels));
}
void Worker::terminate()
{
m_contextProxy->terminateWorkerGlobalScope();
}
bool Worker::canSuspend() const
{
return false;
}
void Worker::stop()
{
terminate();
}
bool Worker::hasPendingActivity() const
{
return m_contextProxy->hasPendingActivity() || ActiveDOMObject::hasPendingActivity();
}
void Worker::notifyNetworkStateChange(bool isOnLine)
{
m_contextProxy->notifyNetworkStateChange(isOnLine);
}
void Worker::didReceiveResponse(unsigned long identifier, const ResourceResponse&)
{
InspectorInstrumentation::didReceiveScriptResponse(scriptExecutionContext(), identifier);
}
void Worker::notifyFinished()
{
if (m_scriptLoader->failed())
dispatchEvent(Event::create(eventNames().errorEvent, false, true));
else {
WorkerThreadStartMode startMode = DontPauseWorkerGlobalScopeOnStart;
if (InspectorInstrumentation::shouldPauseDedicatedWorkerOnStart(scriptExecutionContext()))
startMode = PauseWorkerGlobalScopeOnStart;
m_contextProxy->startWorkerGlobalScope(m_scriptLoader->url(), scriptExecutionContext()->userAgent(m_scriptLoader->url()), m_scriptLoader->script(), startMode);
InspectorInstrumentation::scriptImported(scriptExecutionContext(), m_scriptLoader->identifier(), m_scriptLoader->script());
}
m_scriptLoader = nullptr;
unsetPendingActivity(this);
}
}