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