WebCore:
[WebKit-https.git] / WebCore / workers / WorkerContext.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 "WorkerContext.h"
33
34 #include "ActiveDOMObject.h"
35 #include "DOMTimer.h"
36 #include "DOMWindow.h"
37 #include "Event.h"
38 #include "EventException.h"
39 #include "MessagePort.h"
40 #include "NotImplemented.h"
41 #include "ScriptSourceCode.h"
42 #include "ScriptValue.h"
43 #include "SecurityOrigin.h"
44 #include "WorkerLocation.h"
45 #include "WorkerNavigator.h"
46 #include "WorkerObjectProxy.h"
47 #include "WorkerScriptLoader.h"
48 #include "WorkerThread.h"
49 #include "WorkerThreadableLoader.h"
50 #include "XMLHttpRequestException.h"
51 #include <wtf/RefPtr.h>
52
53 namespace WebCore {
54
55 WorkerContext::WorkerContext(const KURL& url, const String& userAgent, WorkerThread* thread)
56     : m_url(url)
57     , m_userAgent(userAgent)
58     , m_script(new WorkerScriptController(this))
59     , m_thread(thread)
60     , m_closing(false)
61 {
62     setSecurityOrigin(SecurityOrigin::create(url));
63 }
64
65 WorkerContext::~WorkerContext()
66 {
67 }
68
69 ScriptExecutionContext* WorkerContext::scriptExecutionContext() const
70 {
71     return const_cast<WorkerContext*>(this);
72 }
73
74 const KURL& WorkerContext::virtualURL() const
75 {
76     return m_url;
77 }
78
79 KURL WorkerContext::virtualCompleteURL(const String& url) const
80 {
81     return completeURL(url);
82 }
83
84 KURL WorkerContext::completeURL(const String& url) const
85 {
86     // Always return a null URL when passed a null string.
87     // FIXME: Should we change the KURL constructor to have this behavior?
88     if (url.isNull())
89         return KURL();
90     // Always use UTF-8 in Workers.
91     return KURL(m_url, url);
92 }
93
94 String WorkerContext::userAgent(const KURL&) const
95 {
96     return m_userAgent;
97 }
98
99 WorkerLocation* WorkerContext::location() const
100 {
101     if (!m_location)
102         m_location = WorkerLocation::create(m_url);
103     return m_location.get();
104 }
105
106 void WorkerContext::close()
107 {
108     if (m_closing)
109         return;
110
111     m_closing = true;
112     m_thread->stop();
113 }
114
115 WorkerNavigator* WorkerContext::navigator() const
116 {
117     if (!m_navigator)
118         m_navigator = WorkerNavigator::create(m_userAgent);
119     return m_navigator.get();
120 }
121
122 bool WorkerContext::hasPendingActivity() const
123 {
124     ActiveDOMObjectsMap& activeObjects = activeDOMObjects();
125     ActiveDOMObjectsMap::const_iterator activeObjectsEnd = activeObjects.end();
126     for (ActiveDOMObjectsMap::const_iterator iter = activeObjects.begin(); iter != activeObjectsEnd; ++iter) {
127         if (iter->first->hasPendingActivity())
128             return true;
129     }
130
131     // Keep the worker active as long as there is a MessagePort with pending activity or that is remotely entangled.
132     HashSet<MessagePort*>::const_iterator messagePortsEnd = messagePorts().end();
133     for (HashSet<MessagePort*>::const_iterator iter = messagePorts().begin(); iter != messagePortsEnd; ++iter) {
134         if ((*iter)->hasPendingActivity() || ((*iter)->isEntangled() && !(*iter)->locallyEntangledPort()))
135             return true;
136     }
137
138     return false;
139 }
140
141 void WorkerContext::addMessage(MessageDestination destination, MessageSource source, MessageType type, MessageLevel level, const String& message, unsigned lineNumber, const String& sourceURL)
142 {
143     m_thread->workerObjectProxy().postConsoleMessageToWorkerObject(destination, source, type, level, message, lineNumber, sourceURL);
144 }
145
146 void WorkerContext::resourceRetrievedByXMLHttpRequest(unsigned long, const ScriptString&)
147 {
148     // FIXME: The implementation is pending the fixes in https://bugs.webkit.org/show_bug.cgi?id=23175
149     notImplemented();
150 }
151
152 void WorkerContext::scriptImported(unsigned long, const String&)
153 {
154     // FIXME: The implementation is pending the fixes in https://bugs.webkit.org/show_bug.cgi?id=23175
155     notImplemented();
156 }
157
158 void WorkerContext::addEventListener(const AtomicString& eventType, PassRefPtr<EventListener> eventListener, bool)
159 {
160     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
161     if (iter == m_eventListeners.end()) {
162         ListenerVector listeners;
163         listeners.append(eventListener);
164         m_eventListeners.add(eventType, listeners);
165     } else {
166         ListenerVector& listeners = iter->second;
167         for (ListenerVector::iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
168             if (*listenerIter == eventListener)
169                 return;
170         }
171         
172         listeners.append(eventListener);
173         m_eventListeners.add(eventType, listeners);
174     }    
175 }
176
177 void WorkerContext::removeEventListener(const AtomicString& eventType, EventListener* eventListener, bool)
178 {
179     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
180     if (iter == m_eventListeners.end())
181         return;
182     
183     ListenerVector& listeners = iter->second;
184     for (ListenerVector::const_iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
185         if (*listenerIter == eventListener) {
186             listeners.remove(listenerIter - listeners.begin());
187             return;
188         }
189     }
190 }
191
192 bool WorkerContext::dispatchEvent(PassRefPtr<Event> event, ExceptionCode& ec)
193 {
194     if (!event || event->type().isEmpty()) {
195         ec = EventException::UNSPECIFIED_EVENT_TYPE_ERR;
196         return true;
197     }
198     
199     ListenerVector listenersCopy = m_eventListeners.get(event->type());
200     for (ListenerVector::const_iterator listenerIter = listenersCopy.begin(); listenerIter != listenersCopy.end(); ++listenerIter) {
201         event->setTarget(this);
202         event->setCurrentTarget(this);
203         listenerIter->get()->handleEvent(event.get(), false);
204     }
205     
206     return !event->defaultPrevented();
207 }
208
209 void WorkerContext::postTask(PassRefPtr<Task> task)
210 {
211     thread()->runLoop().postTask(task);
212 }
213
214 int WorkerContext::setTimeout(ScheduledAction* action, int timeout)
215 {
216     return DOMTimer::install(scriptExecutionContext(), action, timeout, true);
217 }
218
219 void WorkerContext::clearTimeout(int timeoutId)
220 {
221     DOMTimer::removeById(scriptExecutionContext(), timeoutId);
222 }
223
224 int WorkerContext::setInterval(ScheduledAction* action, int timeout)
225 {
226     return DOMTimer::install(scriptExecutionContext(), action, timeout, false);
227 }
228
229 void WorkerContext::clearInterval(int timeoutId)
230 {
231     DOMTimer::removeById(scriptExecutionContext(), timeoutId);
232 }
233
234 void WorkerContext::importScripts(const Vector<String>& urls, const String& callerURL, int callerLine, ExceptionCode& ec)
235 {
236     ec = 0;
237     Vector<String>::const_iterator urlsEnd = urls.end();
238     Vector<KURL> completedURLs;
239     for (Vector<String>::const_iterator it = urls.begin(); it != urlsEnd; ++it) {
240         const KURL& url = scriptExecutionContext()->completeURL(*it);
241         if (!url.isValid()) {
242             ec = SYNTAX_ERR;
243             return;
244         }
245         completedURLs.append(url);
246     }
247     String securityOrigin = scriptExecutionContext()->securityOrigin()->toString();
248     Vector<KURL>::const_iterator end = completedURLs.end();
249
250     for (Vector<KURL>::const_iterator it = completedURLs.begin(); it != end; ++it) {
251         WorkerScriptLoader scriptLoader;
252         scriptLoader.loadSynchronously(scriptExecutionContext(), *it, DoNotCompleteURL, AllowCrossOriginLoad);
253
254         // If the fetching attempt failed, throw a NETWORK_ERR exception and abort all these steps.
255         if (scriptLoader.failed()) {
256             ec = XMLHttpRequestException::NETWORK_ERR;
257             return;
258         }
259
260         scriptExecutionContext()->scriptImported(scriptLoader.identifier(), scriptLoader.script());
261         scriptExecutionContext()->addMessage(InspectorControllerDestination, JSMessageSource, LogMessageType, LogMessageLevel, "Worker script imported: \"" + *it + "\".", callerLine, callerURL);
262
263         ScriptValue exception;
264         m_script->evaluate(ScriptSourceCode(scriptLoader.script(), *it), &exception);
265         if (!exception.hasNoValue()) {
266             m_script->setException(exception);
267             return;
268         }
269     }
270 }
271
272 } // namespace WebCore
273
274 #endif // ENABLE(WORKERS)