1/* 2 * Copyright (C) 2007, 2008 Apple Inc. All rights reserved. 3 * Copyright (C) 2008 Matt Lilek <webkit@mattlilek.com> 4 * Copyright (C) 2012 Google Inc. All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 16 * its contributors may be used to endorse or promote products derived 17 * from this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 20 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 21 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 22 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 23 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 24 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 26 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 28 * 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/ScriptValue.h" 35#include "core/inspector/InjectedScript.h" 36#include "core/inspector/InjectedScriptHost.h" 37#include "core/inspector/JSONParser.h" 38#include "platform/JSONValues.h" 39#include "public/platform/Platform.h" 40#include "public/platform/WebData.h" 41#include "wtf/PassOwnPtr.h" 42 43namespace blink { 44 45#if ENABLE(OILPAN) 46InjectedScriptManager::CallbackData::~CallbackData() 47{ 48 WrapperPersistent<InjectedScriptHost>::destroy(hostPtr); 49} 50#endif 51 52PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForPage() 53{ 54 return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow)); 55} 56 57PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForWorker() 58{ 59 return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope)); 60} 61 62InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck) 63 : m_nextInjectedScriptId(1) 64 , m_injectedScriptHost(InjectedScriptHost::create()) 65 , m_inspectedStateAccessCheck(accessCheck) 66{ 67} 68 69InjectedScriptManager::~InjectedScriptManager() 70{ 71} 72 73void InjectedScriptManager::trace(Visitor* visitor) 74{ 75 visitor->trace(m_injectedScriptHost); 76} 77 78void InjectedScriptManager::disconnect() 79{ 80 m_injectedScriptHost->disconnect(); 81 m_injectedScriptHost.clear(); 82} 83 84InjectedScriptHost* InjectedScriptManager::injectedScriptHost() 85{ 86 return m_injectedScriptHost.get(); 87} 88 89InjectedScript InjectedScriptManager::injectedScriptForId(int id) 90{ 91 IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id); 92 if (it != m_idToInjectedScript.end()) 93 return it->value; 94 for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) { 95 if (it->value == id) 96 return injectedScriptFor(it->key.get()); 97 } 98 return InjectedScript(); 99} 100 101int InjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState) 102{ 103 ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState); 104 if (it != m_scriptStateToId.end()) 105 return it->value; 106 int id = m_nextInjectedScriptId++; 107 m_scriptStateToId.set(scriptState, id); 108 return id; 109} 110 111InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId) 112{ 113 RefPtr<JSONValue> parsedObjectId = parseJSON(objectId); 114 if (parsedObjectId && parsedObjectId->type() == JSONValue::TypeObject) { 115 long injectedScriptId = 0; 116 bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId); 117 if (success) 118 return m_idToInjectedScript.get(injectedScriptId); 119 } 120 return InjectedScript(); 121} 122 123void InjectedScriptManager::discardInjectedScripts() 124{ 125 m_idToInjectedScript.clear(); 126 m_scriptStateToId.clear(); 127} 128 129void InjectedScriptManager::discardInjectedScriptsFor(LocalDOMWindow* window) 130{ 131 if (m_scriptStateToId.isEmpty()) 132 return; 133 134 Vector<long> idsToRemove; 135 IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end(); 136 for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) { 137 ScriptState* scriptState = it->value.scriptState(); 138 if (window != scriptState->domWindow()) 139 continue; 140 m_scriptStateToId.remove(scriptState); 141 idsToRemove.append(it->key); 142 } 143 m_idToInjectedScript.removeAll(idsToRemove); 144 145 // Now remove script states that have id but no injected script. 146 Vector<ScriptState*> scriptStatesToRemove; 147 for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) { 148 ScriptState* scriptState = it->key.get(); 149 if (window == scriptState->domWindow()) 150 scriptStatesToRemove.append(scriptState); 151 } 152 m_scriptStateToId.removeAll(scriptStatesToRemove); 153} 154 155bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*) 156{ 157 return true; 158} 159 160void InjectedScriptManager::releaseObjectGroup(const String& objectGroup) 161{ 162 Vector<int> keys; 163 keys.appendRange(m_idToInjectedScript.keys().begin(), m_idToInjectedScript.keys().end()); 164 for (Vector<int>::iterator k = keys.begin(); k != keys.end(); ++k) { 165 IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(*k); 166 if (s != m_idToInjectedScript.end()) 167 s->value.releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here. 168 } 169} 170 171String InjectedScriptManager::injectedScriptSource() 172{ 173 const blink::WebData& injectedScriptSourceResource = blink::Platform::current()->loadResource("InjectedScriptSource.js"); 174 return String(injectedScriptSourceResource.data(), injectedScriptSourceResource.size()); 175} 176 177InjectedScript InjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState) 178{ 179 ScriptStateToId::iterator it = m_scriptStateToId.find(inspectedScriptState); 180 if (it != m_scriptStateToId.end()) { 181 IdToInjectedScriptMap::iterator it1 = m_idToInjectedScript.find(it->value); 182 if (it1 != m_idToInjectedScript.end()) 183 return it1->value; 184 } 185 186 if (!m_inspectedStateAccessCheck(inspectedScriptState)) 187 return InjectedScript(); 188 189 int id = injectedScriptIdFor(inspectedScriptState); 190 ScriptValue injectedScriptValue = createInjectedScript(injectedScriptSource(), inspectedScriptState, id); 191 InjectedScript result(injectedScriptValue, m_inspectedStateAccessCheck); 192 m_idToInjectedScript.set(id, result); 193 return result; 194} 195 196} // namespace blink 197 198