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 <v8.h>
  50
  51 #if PLATFORM(CHROMIUM)
  52 #include <public/Platform.h>
  53 #include <public/WebWorkerRunLoop.h>
  54 #endif
  55
  56 namespace WebCore {
  57
  58 WorkerScriptController::WorkerScriptController(WorkerContext* workerContext)
  59     : m_workerContext(workerContext)
  60     , m_isolate(v8::Isolate::New())
  61     , m_executionForbidden(false)
  62     , m_executionScheduledToTerminate(false)
  63 {
  64     m_isolate->Enter();
  65     V8PerIsolateData* data = V8PerIsolateData::create(m_isolate);
  66     m_domDataStore = adoptPtr(new DOMDataStore(DOMDataStore::Worker));
  67     data->setDOMDataStore(m_domDataStore.get());
  68
  69     V8Initializer::initializeWorker(m_isolate);
  70 }
  71
  72 WorkerScriptController::~WorkerScriptController()
  73 {
  74     m_domDataStore.clear();
  75 #if PLATFORM(CHROMIUM)
  76     // The corresponding call to didStartWorkerRunLoop is in
  77     // WorkerThread::workerThread().
  78     // See http://webkit.org/b/83104#c14 for why this is here.
  79     WebKit::Platform::current()->didStopWorkerRunLoop(WebKit::WebWorkerRunLoop(&m_workerContext->thread()->runLoop()));
  80 #endif
  81     disposeContext();
  82     V8PerIsolateData::dispose(m_isolate);
  83     m_isolate->Exit();
  84     m_isolate->Dispose();
  85 }
  86
  87 void WorkerScriptController::disposeContext()
  88 {
  89     m_perContextData.clear();
  90     m_context.clear();
  91 }
  92
  93 bool WorkerScriptController::initializeContextIfNeeded()
  94 {
  95     if (!m_context.isEmpty())
  96         return true;
  97
  98     v8::Persistent<v8::ObjectTemplate> globalTemplate;
  99     m_context.adopt(v8::Context::New(0, globalTemplate));
 100     if (m_context.isEmpty())
 101         return false;
 102
 103     // Starting from now, use local context only.
 104     v8::Local<v8::Context> context = v8::Local<v8::Context>::New(m_context.get());
 105
 106     v8::Context::Scope scope(context);
 107
 108     m_perContextData = V8PerContextData::create(m_context.get());
 109     if (!m_perContextData->init()) {
 110         disposeContext();
 111         return false;
 112     }
 113
 114     // Set DebugId for the new context.
 115     context->SetEmbedderData(0, v8::String::NewSymbol("worker"));
 116
 117     // Create a new JS object and use it as the prototype for the shadow global object.
 118     WrapperTypeInfo* contextType = &V8DedicatedWorkerContext::info;
 119 #if ENABLE(SHARED_WORKERS)
 120     if (!m_workerContext->isDedicatedWorkerContext())
 121         contextType = &V8SharedWorkerContext::info;
 122 #endif
 123     v8::Handle<v8::Function> workerContextConstructor = m_perContextData->constructorForType(contextType);
 124     v8::Local<v8::Object> jsWorkerContext = V8ObjectConstructor::newInstance(workerContextConstructor);
 125     if (jsWorkerContext.IsEmpty()) {
 126         disposeContext();
 127         return false;
 128     }
 129
 130     V8DOMWrapper::associateObjectWithWrapper(PassRefPtr<WorkerContext>(m_workerContext), contextType, jsWorkerContext, m_isolate, WrapperConfiguration::Dependent);
 131
 132     // Insert the object instance as the prototype of the shadow object.
 133     v8::Handle<v8::Object> globalObject = v8::Handle<v8::Object>::Cast(m_context->Global()->GetPrototype());
 134     globalObject->SetPrototype(jsWorkerContext);
 135
 136     return true;
 137 }
 138
 139 ScriptValue WorkerScriptController::evaluate(const String& script, const String& fileName, const TextPosition& scriptStartPosition, WorkerContextExecutionState* state)
 140 {
 141     V8GCController::checkMemoryUsage();
 142
 143     v8::HandleScope handleScope;
 144
 145     if (!initializeContextIfNeeded())
 146         return ScriptValue();
 147
 148     if (!m_disableEvalPending.isEmpty()) {
 149         m_context->AllowCodeGenerationFromStrings(false);
 150         m_context->SetErrorMessageForCodeGenerationFromStrings(v8String(m_disableEvalPending, m_context->GetIsolate()));
 151         m_disableEvalPending = String();
 152     }
 153
 154     v8::Isolate* isolate = m_context->GetIsolate();
 155     v8::Context::Scope scope(m_context.get());
 156
 157     v8::TryCatch block;
 158
 159     v8::Handle<v8::String> scriptString = v8String(script, isolate);
 160     v8::Handle<v8::Script> compiledScript = ScriptSourceCode::compileScript(scriptString, fileName, scriptStartPosition, 0, isolate);
 161     v8::Local<v8::Value> result = ScriptRunner::runCompiledScript(compiledScript, m_workerContext);
 162
 163     if (!block.CanContinue()) {
 164         m_workerContext->script()->forbidExecution();
 165         return ScriptValue();
 166     }
 167
 168     if (block.HasCaught()) {
 169         v8::Local<v8::Message> message = block.Message();
 170         state->hadException = true;
 171         state->errorMessage = toWebCoreString(message->Get());
 172         state->lineNumber = message->GetLineNumber();
 173         state->sourceURL = toWebCoreString(message->GetScriptResourceName());
 174         if (m_workerContext->sanitizeScriptError(state->errorMessage, state->lineNumber, state->sourceURL))
 175             state->exception = throwError(v8GeneralError, state->errorMessage.utf8().data(), isolate);
 176         else
 177             state->exception = ScriptValue(block.Exception());
 178
 179         block.Reset();
 180     } else
 181         state->hadException = false;
 182
 183     if (result.IsEmpty() || result->IsUndefined())
 184         return ScriptValue();
 185
 186     return ScriptValue(result);
 187 }
 188
 189 void WorkerScriptController::evaluate(const ScriptSourceCode& sourceCode, ScriptValue* exception)
 190 {
 191     if (isExecutionForbidden())
 192         return;
 193
 194     WorkerContextExecutionState state;
 195     evaluate(sourceCode.source(), sourceCode.url().string(), sourceCode.startPosition(), &state);
 196     if (state.hadException) {
 197         if (exception)
 198             *exception = state.exception;
 199         else
 200             m_workerContext->reportException(state.errorMessage, state.lineNumber, state.sourceURL, 0);
 201     }
 202 }
 203
 204 void WorkerScriptController::scheduleExecutionTermination()
 205 {
 206     // The mutex provides a memory barrier to ensure that once
 207     // termination is scheduled, isExecutionTerminating will
 208     // accurately reflect that state when called from another thread.
 209     {
 210         MutexLocker locker(m_scheduledTerminationMutex);
 211         m_executionScheduledToTerminate = true;
 212     }
 213     v8::V8::TerminateExecution(m_isolate);
 214 }
 215
 216 bool WorkerScriptController::isExecutionTerminating() const
 217 {
 218     // See comments in scheduleExecutionTermination regarding mutex usage.
 219     MutexLocker locker(m_scheduledTerminationMutex);
 220     return m_executionScheduledToTerminate;
 221 }
 222
 223 void WorkerScriptController::forbidExecution()
 224 {
 225     ASSERT(m_workerContext->isContextThread());
 226     m_executionForbidden = true;
 227 }
 228
 229 bool WorkerScriptController::isExecutionForbidden() const
 230 {
 231     ASSERT(m_workerContext->isContextThread());
 232     return m_executionForbidden;
 233 }
 234
 235 void WorkerScriptController::disableEval(const String& errorMessage)
 236 {
 237     m_disableEvalPending = errorMessage;
 238 }
 239
 240 void WorkerScriptController::setException(const ScriptValue& exception)
 241 {
 242     throwError(*exception.v8Value(), m_context->GetIsolate());
 243 }
 244
 245 WorkerScriptController* WorkerScriptController::controllerForContext()
 246 {
 247     // Happens on frame destruction, check otherwise GetCurrent() will crash.
 248     if (!v8::Context::InContext())
 249         return 0;
 250     v8::Handle<v8::Context> context = v8::Context::GetCurrent();
 251     v8::Handle<v8::Object> global = context->Global();
 252     global = global->FindInstanceInPrototypeChain(V8WorkerContext::GetTemplate(context->GetIsolate()));
 253     // Return 0 if the current executing context is not the worker context.
 254     if (global.IsEmpty())
 255         return 0;
 256     WorkerContext* workerContext = V8WorkerContext::toNative(global);
 257     return workerContext->script();
 258 }
 259
 260 } // namespace WebCore
 261
 262 #endif // ENABLE(WORKERS)