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