1/*
2 * Copyright (C) 2008, 2010 Apple Inc. All Rights Reserved.
3 * Copyright (C) 2009 Google Inc. All Rights Reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
18 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 *
26 */
27
28#include "config.h"
29#include "core/workers/Worker.h"
30
31#include "bindings/core/v8/ExceptionState.h"
32#include "core/dom/Document.h"
33#include "core/events/MessageEvent.h"
34#include "core/fetch/ResourceFetcher.h"
35#include "core/inspector/InspectorInstrumentation.h"
36#include "core/frame/LocalDOMWindow.h"
37#include "core/frame/UseCounter.h"
38#include "core/workers/WorkerGlobalScopeProxy.h"
39#include "core/workers/WorkerGlobalScopeProxyProvider.h"
40#include "core/workers/WorkerScriptLoader.h"
41#include "core/workers/WorkerThread.h"
42#include "wtf/MainThread.h"
43
44namespace blink {
45
46inline Worker::Worker(ExecutionContext* context)
47    : AbstractWorker(context)
48    , m_contextProxy(0)
49{
50}
51
52PassRefPtrWillBeRawPtr<Worker> Worker::create(ExecutionContext* context, const String& url, ExceptionState& exceptionState)
53{
54    ASSERT(isMainThread());
55    Document* document = toDocument(context);
56    UseCounter::count(context, UseCounter::WorkerStart);
57    if (!document->page()) {
58        exceptionState.throwDOMException(InvalidAccessError, "The context provided is invalid.");
59        return nullptr;
60    }
61    WorkerGlobalScopeProxyProvider* proxyProvider = WorkerGlobalScopeProxyProvider::from(*document->page());
62    ASSERT(proxyProvider);
63
64    RefPtrWillBeRawPtr<Worker> worker = adoptRefWillBeNoop(new Worker(context));
65
66    worker->suspendIfNeeded();
67
68    KURL scriptURL = worker->resolveURL(url, exceptionState);
69    if (scriptURL.isEmpty())
70        return nullptr;
71
72    worker->m_scriptLoader = WorkerScriptLoader::create();
73    worker->m_scriptLoader->loadAsynchronously(*context, scriptURL, DenyCrossOriginRequests, worker.get());
74    worker->m_contextProxy = proxyProvider->createWorkerGlobalScopeProxy(worker.get());
75    return worker.release();
76}
77
78Worker::~Worker()
79{
80    ASSERT(isMainThread());
81    if (!m_contextProxy)
82        return;
83    ASSERT(executionContext()); // The context is protected by worker context proxy, so it cannot be destroyed while a Worker exists.
84    m_contextProxy->workerObjectDestroyed();
85}
86
87const AtomicString& Worker::interfaceName() const
88{
89    return EventTargetNames::Worker;
90}
91
92void Worker::postMessage(ExecutionContext*, PassRefPtr<SerializedScriptValue> message, const MessagePortArray* ports, ExceptionState& exceptionState)
93{
94    ASSERT(m_contextProxy);
95    // Disentangle the port in preparation for sending it to the remote context.
96    OwnPtr<MessagePortChannelArray> channels = MessagePort::disentanglePorts(ports, exceptionState);
97    if (exceptionState.hadException())
98        return;
99    m_contextProxy->postMessageToWorkerGlobalScope(message, channels.release());
100}
101
102void Worker::terminate()
103{
104    if (m_contextProxy)
105        m_contextProxy->terminateWorkerGlobalScope();
106}
107
108void Worker::stop()
109{
110    terminate();
111}
112
113bool Worker::hasPendingActivity() const
114{
115    // The worker context does not exist while loading, so we must ensure that the worker object is not collected, nor are its event listeners.
116    return (m_contextProxy && m_contextProxy->hasPendingActivity()) || m_scriptLoader;
117}
118
119void Worker::didReceiveResponse(unsigned long identifier, const ResourceResponse&)
120{
121    InspectorInstrumentation::didReceiveScriptResponse(executionContext(), identifier);
122}
123
124void Worker::notifyFinished()
125{
126    if (m_scriptLoader->failed()) {
127        dispatchEvent(Event::createCancelable(EventTypeNames::error));
128    } else {
129        ASSERT(m_contextProxy);
130        WorkerThreadStartMode startMode = DontPauseWorkerGlobalScopeOnStart;
131        if (InspectorInstrumentation::shouldPauseDedicatedWorkerOnStart(executionContext()))
132            startMode = PauseWorkerGlobalScopeOnStart;
133        m_contextProxy->startWorkerGlobalScope(m_scriptLoader->url(), executionContext()->userAgent(m_scriptLoader->url()), m_scriptLoader->script(), startMode);
134        InspectorInstrumentation::scriptImported(executionContext(), m_scriptLoader->identifier(), m_scriptLoader->script());
135    }
136    m_scriptLoader = nullptr;
137}
138
139void Worker::trace(Visitor* visitor)
140{
141    AbstractWorker::trace(visitor);
142}
143
144} // namespace blink
145