[V8] Add a "context type" parameter to GetTemplate and ConfigureV8SomethingTemplate...
[WebKit-https.git] / Source / WebCore / bindings / v8 / WorkerScriptController.cpp
1 /*
2  * Copyright (C) 2009, 2012 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32
33 #if ENABLE(WORKERS)
34
35 #include "WorkerScriptController.h"
36
37 #include "DOMTimer.h"
38 #include "ScriptCallStack.h"
39 #include "ScriptRunner.h"
40 #include "ScriptSourceCode.h"
41 #include "ScriptValue.h"
42 #include "V8DedicatedWorkerContext.h"
43 #include "V8Initializer.h"
44 #include "V8SharedWorkerContext.h"
45 #include "V8WorkerContext.h"
46 #include "WorkerContext.h"
47 #include "WorkerObjectProxy.h"
48 #include "WorkerThread.h"
49 #include "WrapperTypeInfo.h"
50 #include <v8.h>
51
52 #if PLATFORM(CHROMIUM)
53 #include <public/Platform.h>
54 #include <public/WebWorkerRunLoop.h>
55 #endif
56
57 namespace WebCore {
58
59 WorkerScriptController::WorkerScriptController(WorkerContext* workerContext)
60     : m_workerContext(workerContext)
61     , m_isolate(v8::Isolate::New())
62     , m_executionForbidden(false)
63     , m_executionScheduledToTerminate(false)
64 {
65     m_isolate->Enter();
66     V8PerIsolateData* data = V8PerIsolateData::create(m_isolate);
67     m_domDataStore = adoptPtr(new DOMDataStore(WorkerWorld));
68     data->setDOMDataStore(m_domDataStore.get());
69
70     V8Initializer::initializeWorker(m_isolate);
71 }
72
73 WorkerScriptController::~WorkerScriptController()
74 {
75     m_domDataStore.clear();
76 #if PLATFORM(CHROMIUM)
77     // The corresponding call to didStartWorkerRunLoop is in
78     // WorkerThread::workerThread().
79     // See http://webkit.org/b/83104#c14 for why this is here.
80     WebKit::Platform::current()->didStopWorkerRunLoop(WebKit::WebWorkerRunLoop(&m_workerContext->thread()->runLoop()));
81 #endif
82     disposeContext();
83     V8PerIsolateData::dispose(m_isolate);
84     m_isolate->Exit();
85     m_isolate->Dispose();
86 }
87
88 void WorkerScriptController::disposeContext()
89 {
90     m_perContextData.clear();
91     m_context.clear();
92 }
93
94 bool WorkerScriptController::initializeContextIfNeeded()
95 {
96     if (!m_context.isEmpty())
97         return true;
98
99     v8::Persistent<v8::ObjectTemplate> globalTemplate;
100     m_context.adopt(v8::Context::New(0, globalTemplate));
101     if (m_context.isEmpty())
102         return false;
103
104     // Starting from now, use local context only.
105     v8::Local<v8::Context> context = v8::Local<v8::Context>::New(m_context.get());
106
107     v8::Context::Scope scope(context);
108
109     m_perContextData = V8PerContextData::create(m_context.get());
110     if (!m_perContextData->init()) {
111         disposeContext();
112         return false;
113     }
114
115     // Set DebugId for the new context.
116     context->SetEmbedderData(0, v8::String::NewSymbol("worker"));
117
118     // Create a new JS object and use it as the prototype for the shadow global object.
119     WrapperTypeInfo* contextType = &V8DedicatedWorkerContext::info;
120 #if ENABLE(SHARED_WORKERS)
121     if (!m_workerContext->isDedicatedWorkerContext())
122         contextType = &V8SharedWorkerContext::info;
123 #endif
124     v8::Handle<v8::Function> workerContextConstructor = m_perContextData->constructorForType(contextType);
125     v8::Local<v8::Object> jsWorkerContext = V8ObjectConstructor::newInstance(workerContextConstructor);
126     if (jsWorkerContext.IsEmpty()) {
127         disposeContext();
128         return false;
129     }
130
131     V8DOMWrapper::associateObjectWithWrapper(PassRefPtr<WorkerContext>(m_workerContext), contextType, jsWorkerContext, m_isolate, WrapperConfiguration::Dependent);
132
133     // Insert the object instance as the prototype of the shadow object.
134     v8::Handle<v8::Object> globalObject = v8::Handle<v8::Object>::Cast(m_context->Global()->GetPrototype());
135     globalObject->SetPrototype(jsWorkerContext);
136
137     return true;
138 }
139
140 ScriptValue WorkerScriptController::evaluate(const String& script, const String& fileName, const TextPosition& scriptStartPosition, WorkerContextExecutionState* state)
141 {
142     V8GCController::checkMemoryUsage();
143
144     v8::HandleScope handleScope;
145
146     if (!initializeContextIfNeeded())
147         return ScriptValue();
148
149     if (!m_disableEvalPending.isEmpty()) {
150         m_context->AllowCodeGenerationFromStrings(false);
151         m_context->SetErrorMessageForCodeGenerationFromStrings(v8String(m_disableEvalPending, m_context->GetIsolate()));
152         m_disableEvalPending = String();
153     }
154
155     v8::Isolate* isolate = m_context->GetIsolate();
156     v8::Context::Scope scope(m_context.get());
157
158     v8::TryCatch block;
159
160     v8::Handle<v8::String> scriptString = v8String(script, isolate);
161     v8::Handle<v8::Script> compiledScript = ScriptSourceCode::compileScript(scriptString, fileName, scriptStartPosition, 0, isolate);
162     v8::Local<v8::Value> result = ScriptRunner::runCompiledScript(compiledScript, m_workerContext);
163
164     if (!block.CanContinue()) {
165         m_workerContext->script()->forbidExecution();
166         return ScriptValue();
167     }
168
169     if (block.HasCaught()) {
170         v8::Local<v8::Message> message = block.Message();
171         state->hadException = true;
172         state->errorMessage = toWebCoreString(message->Get());
173         state->lineNumber = message->GetLineNumber();
174         state->sourceURL = toWebCoreString(message->GetScriptResourceName());
175         if (m_workerContext->sanitizeScriptError(state->errorMessage, state->lineNumber, state->sourceURL))
176             state->exception = throwError(v8GeneralError, state->errorMessage.utf8().data(), isolate);
177         else
178             state->exception = ScriptValue(block.Exception());
179
180         block.Reset();
181     } else
182         state->hadException = false;
183
184     if (result.IsEmpty() || result->IsUndefined())
185         return ScriptValue();
186
187     return ScriptValue(result);
188 }
189
190 void WorkerScriptController::evaluate(const ScriptSourceCode& sourceCode, ScriptValue* exception)
191 {
192     if (isExecutionForbidden())
193         return;
194
195     WorkerContextExecutionState state;
196     evaluate(sourceCode.source(), sourceCode.url().string(), sourceCode.startPosition(), &state);
197     if (state.hadException) {
198         if (exception)
199             *exception = state.exception;
200         else
201             m_workerContext->reportException(state.errorMessage, state.lineNumber, state.sourceURL, 0);
202     }
203 }
204
205 void WorkerScriptController::scheduleExecutionTermination()
206 {
207     // The mutex provides a memory barrier to ensure that once
208     // termination is scheduled, isExecutionTerminating will
209     // accurately reflect that state when called from another thread.
210     {
211         MutexLocker locker(m_scheduledTerminationMutex);
212         m_executionScheduledToTerminate = true;
213     }
214     v8::V8::TerminateExecution(m_isolate);
215 }
216
217 bool WorkerScriptController::isExecutionTerminating() const
218 {
219     // See comments in scheduleExecutionTermination regarding mutex usage.
220     MutexLocker locker(m_scheduledTerminationMutex);
221     return m_executionScheduledToTerminate;
222 }
223
224 void WorkerScriptController::forbidExecution()
225 {
226     ASSERT(m_workerContext->isContextThread());
227     m_executionForbidden = true;
228 }
229
230 bool WorkerScriptController::isExecutionForbidden() const
231 {
232     ASSERT(m_workerContext->isContextThread());
233     return m_executionForbidden;
234 }
235
236 void WorkerScriptController::disableEval(const String& errorMessage)
237 {
238     m_disableEvalPending = errorMessage;
239 }
240
241 void WorkerScriptController::setException(const ScriptValue& exception)
242 {
243     throwError(*exception.v8Value(), m_context->GetIsolate());
244 }
245
246 WorkerScriptController* WorkerScriptController::controllerForContext()
247 {
248     // Happens on frame destruction, check otherwise GetCurrent() will crash.
249     if (!v8::Context::InContext())
250         return 0;
251     v8::Handle<v8::Context> context = v8::Context::GetCurrent();
252     v8::Handle<v8::Object> global = context->Global();
253     global = global->FindInstanceInPrototypeChain(V8WorkerContext::GetTemplate(context->GetIsolate(), WorkerWorld));
254     // Return 0 if the current executing context is not the worker context.
255     if (global.IsEmpty())
256         return 0;
257     WorkerContext* workerContext = V8WorkerContext::toNative(global);
258     return workerContext->script();
259 }
260
261 } // namespace WebCore
262
263 #endif // ENABLE(WORKERS)