src/third_party/WebKit/Source/bindings/v8/custom/V8InjectedScriptManager.cpp

   1 /*
   2  * Copyright (C) 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 #include "core/inspector/InjectedScriptManager.h"
  33
  34 #include "bindings/core/v8/V8InjectedScriptHost.h"
  35 #include "bindings/core/v8/V8Window.h"
  36 #include "bindings/v8/BindingSecurity.h"
  37 #include "bindings/v8/ScopedPersistent.h"
  38 #include "bindings/v8/ScriptDebugServer.h"
  39 #include "bindings/v8/ScriptValue.h"
  40 #include "bindings/v8/V8Binding.h"
  41 #include "bindings/v8/V8ObjectConstructor.h"
  42 #include "bindings/v8/V8ScriptRunner.h"
  43 #include "core/inspector/InjectedScriptHost.h"
  44 #include "core/frame/LocalDOMWindow.h"
  45 #include "wtf/RefPtr.h"
  46
  47 namespace WebCore {
  48
  49 struct InjectedScriptManager::CallbackData {
  50     ScopedPersistent<v8::Object> handle;
  51     RefPtr<InjectedScriptHost> host;
  52 };
  53
  54 static v8::Local<v8::Object> createInjectedScriptHostV8Wrapper(InjectedScriptHost* host, v8::Isolate* isolate)
  55 {
  56     v8::Local<v8::Function> function = V8InjectedScriptHost::domTemplate(isolate)->GetFunction();
  57     if (function.IsEmpty()) {
  58         // Return if allocation failed.
  59         return v8::Local<v8::Object>();
  60     }
  61     v8::Local<v8::Object> instanceTemplate = V8ObjectConstructor::newInstance(isolate, function);
  62     if (instanceTemplate.IsEmpty()) {
  63         // Avoid setting the wrapper if allocation failed.
  64         return v8::Local<v8::Object>();
  65     }
  66     V8DOMWrapper::setNativeInfo(instanceTemplate, &V8InjectedScriptHost::wrapperTypeInfo, host);
  67     // Create a weak reference to the v8 wrapper of InspectorBackend to deref
  68     // InspectorBackend when the wrapper is garbage collected.
  69     InjectedScriptManager::CallbackData* data = new InjectedScriptManager::CallbackData;
  70     data->host = host;
  71     data->handle.set(isolate, instanceTemplate);
  72     data->handle.setWeak(data, &InjectedScriptManager::setWeakCallback);
  73     return instanceTemplate;
  74 }
  75
  76 ScriptValue InjectedScriptManager::createInjectedScript(const String& scriptSource, ScriptState* inspectedScriptState, int id)
  77 {
  78     v8::Isolate* isolate = inspectedScriptState->isolate();
  79     ScriptState::Scope scope(inspectedScriptState);
  80
  81     // Call custom code to create InjectedScripHost wrapper specific for the context
  82     // instead of calling toV8() that would create the
  83     // wrapper in the current context.
  84     // FIXME: make it possible to use generic bindings factory for InjectedScriptHost.
  85     v8::Local<v8::Object> scriptHostWrapper = createInjectedScriptHostV8Wrapper(m_injectedScriptHost.get(), inspectedScriptState->isolate());
  86     if (scriptHostWrapper.IsEmpty())
  87         return ScriptValue();
  88
  89     // Inject javascript into the context. The compiled script is supposed to evaluate into
  90     // a single anonymous function(it's anonymous to avoid cluttering the global object with
  91     // inspector's stuff) the function is called a few lines below with InjectedScriptHost wrapper,
  92     // injected script id and explicit reference to the inspected global object. The function is expected
  93     // to create and configure InjectedScript instance that is going to be used by the inspector.
  94     v8::Local<v8::Value> value = V8ScriptRunner::compileAndRunInternalScript(v8String(isolate, scriptSource), isolate);
  95     ASSERT(!value.IsEmpty());
  96     ASSERT(value->IsFunction());
  97
  98     v8::Local<v8::Object> windowGlobal = inspectedScriptState->context()->Global();
  99     v8::Handle<v8::Value> info[] = { scriptHostWrapper, windowGlobal, v8::Number::New(inspectedScriptState->isolate(), id) };
 100     v8::Local<v8::Value> injectedScriptValue = V8ScriptRunner::callInternalFunction(v8::Local<v8::Function>::Cast(value), windowGlobal, WTF_ARRAY_LENGTH(info), info, inspectedScriptState->isolate());
 101     return ScriptValue(inspectedScriptState, injectedScriptValue);
 102 }
 103
 104 bool InjectedScriptManager::canAccessInspectedWindow(ScriptState* scriptState)
 105 {
 106     ScriptState::Scope scope(scriptState);
 107     v8::Local<v8::Object> global = scriptState->context()->Global();
 108     if (global.IsEmpty())
 109         return false;
 110     v8::Handle<v8::Object> holder = V8Window::findInstanceInPrototypeChain(global, scriptState->isolate());
 111     if (holder.IsEmpty())
 112         return false;
 113     LocalFrame* frame = V8Window::toNative(holder)->frame();
 114
 115     return BindingSecurity::shouldAllowAccessToFrame(scriptState->isolate(), frame, DoNotReportSecurityError);
 116 }
 117
 118 void InjectedScriptManager::setWeakCallback(const v8::WeakCallbackData<v8::Object, InjectedScriptManager::CallbackData>& data)
 119 {
 120     data.GetParameter()->handle.clear();
 121     data.GetParameter()->host.clear();
 122     delete data.GetParameter();
 123 }
 124
 125 } // namespace WebCore