2e03e3d8303e0db7ecc6cdb15a64ef04bbdcd6e9
[WebKit-https.git] / WebCore / workers / Worker.cpp
1 /*
2  * Copyright (C) 2008 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
30 #if ENABLE(WORKERS)
31
32 #include "Worker.h"
33
34 #include "DOMWindow.h"
35 #include "DocLoader.h"
36 #include "Document.h"
37 #include "EventException.h"
38 #include "EventListener.h"
39 #include "EventNames.h"
40 #include "ExceptionCode.h"
41 #include "Frame.h"
42 #include "FrameLoader.h"
43 #include "MessageEvent.h"
44 #include "SecurityOrigin.h"
45 #include "TextEncoding.h"
46 #include "WorkerContextProxy.h"
47 #include "WorkerScriptLoader.h"
48 #include "WorkerThread.h"
49 #include <wtf/MainThread.h>
50
51 namespace WebCore {
52
53 Worker::Worker(const String& url, ScriptExecutionContext* context, ExceptionCode& ec)
54     : ActiveDOMObject(context, this)
55     , m_contextProxy(WorkerContextProxy::create(this))
56 {
57     m_scriptURL = context->completeURL(url);
58     if (url.isEmpty() || !m_scriptURL.isValid()) {
59         ec = SYNTAX_ERR;
60         return;
61     }
62
63     if (!context->securityOrigin()->canAccess(SecurityOrigin::create(m_scriptURL).get())) {
64         ec = SECURITY_ERR;
65         return;
66     }
67
68     m_scriptLoader = new WorkerScriptLoader();
69     m_scriptLoader->loadAsynchronously(scriptExecutionContext(), m_scriptURL, DenyCrossOriginRedirect, this);
70     setPendingActivity(this);  // The worker context does not exist while loading, so we must ensure that the worker object is not collected, as well as its event listeners.
71 }
72
73 Worker::~Worker()
74 {
75     ASSERT(isMainThread());
76     ASSERT(scriptExecutionContext()); // The context is protected by worker context proxy, so it cannot be destroyed while a Worker exists.
77     m_contextProxy->workerObjectDestroyed();
78 }
79
80 void Worker::postMessage(const String& message, ExceptionCode& ec)
81 {
82     postMessage(message, 0, ec);
83 }
84
85 void Worker::postMessage(const String& message, MessagePort* messagePort, ExceptionCode& ec)
86 {
87     // Disentangle the port in preparation for sending it to the remote context.
88     OwnPtr<MessagePortChannel> channel = messagePort ? messagePort->disentangle(ec) : 0;
89     if (ec)
90         return;
91     m_contextProxy->postMessageToWorkerContext(message, channel.release());
92 }
93
94 void Worker::terminate()
95 {
96     m_contextProxy->terminateWorkerContext();
97 }
98
99 bool Worker::canSuspend() const
100 {
101     // FIXME: It is not currently possible to suspend a worker, so pages with workers can not go into page cache.
102     return false;
103 }
104
105 void Worker::stop()
106 {
107     terminate();
108 }
109
110 bool Worker::hasPendingActivity() const
111 {
112     return m_contextProxy->hasPendingActivity() || ActiveDOMObject::hasPendingActivity();
113 }
114
115 void Worker::notifyFinished()
116 {
117     if (m_scriptLoader->failed())
118         dispatchErrorEvent();
119     else
120         m_contextProxy->startWorkerContext(m_scriptURL, scriptExecutionContext()->userAgent(m_scriptURL), m_scriptLoader->script());
121
122     m_scriptLoader = 0;
123
124     unsetPendingActivity(this);
125 }
126
127 void Worker::dispatchErrorEvent()
128 {
129     RefPtr<Event> evt = Event::create(eventNames().errorEvent, false, true);
130     if (m_onErrorListener) {
131         evt->setTarget(this);
132         evt->setCurrentTarget(this);
133         m_onErrorListener->handleEvent(evt.get(), true);
134     }
135
136     ExceptionCode ec = 0;
137     dispatchEvent(evt.release(), ec);
138     ASSERT(!ec);
139 }
140
141 void Worker::addEventListener(const AtomicString& eventType, PassRefPtr<EventListener> eventListener, bool)
142 {
143     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
144     if (iter == m_eventListeners.end()) {
145         ListenerVector listeners;
146         listeners.append(eventListener);
147         m_eventListeners.add(eventType, listeners);
148     } else {
149         ListenerVector& listeners = iter->second;
150         for (ListenerVector::iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
151             if (*listenerIter == eventListener)
152                 return;
153         }
154         
155         listeners.append(eventListener);
156         m_eventListeners.add(eventType, listeners);
157     }    
158 }
159
160 void Worker::removeEventListener(const AtomicString& eventType, EventListener* eventListener, bool)
161 {
162     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
163     if (iter == m_eventListeners.end())
164         return;
165     
166     ListenerVector& listeners = iter->second;
167     for (ListenerVector::const_iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
168         if (*listenerIter == eventListener) {
169             listeners.remove(listenerIter - listeners.begin());
170             return;
171         }
172     }
173 }
174
175 bool Worker::dispatchEvent(PassRefPtr<Event> event, ExceptionCode& ec)
176 {
177     if (!event || event->type().isEmpty()) {
178         ec = EventException::UNSPECIFIED_EVENT_TYPE_ERR;
179         return true;
180     }
181
182     ListenerVector listenersCopy = m_eventListeners.get(event->type());
183     for (ListenerVector::const_iterator listenerIter = listenersCopy.begin(); listenerIter != listenersCopy.end(); ++listenerIter) {
184         event->setTarget(this);
185         event->setCurrentTarget(this);
186         listenerIter->get()->handleEvent(event.get(), false);
187     }
188
189     return !event->defaultPrevented();
190 }
191
192 void Worker::dispatchMessage(const String& message, PassRefPtr<MessagePort> port)
193 {
194     RefPtr<Event> evt = MessageEvent::create(message, "", "", 0, port);
195
196     if (m_onMessageListener.get()) {
197         evt->setTarget(this);
198         evt->setCurrentTarget(this);
199         m_onMessageListener->handleEvent(evt.get(), false);
200     }
201
202     ExceptionCode ec = 0;
203     dispatchEvent(evt.release(), ec);
204     ASSERT(!ec);
205 }
206
207 } // namespace WebCore
208
209 #endif // ENABLE(WORKERS)