src/third_party/WebKit/Source/core/inspector/InjectedScriptManager.cpp

   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 "InjectedScriptSource.h"
  35 #include "bindings/v8/ScriptObject.h"
  36 #include "core/inspector/InjectedScript.h"
  37 #include "core/inspector/InjectedScriptHost.h"
  38 #include "core/inspector/JSONParser.h"
  39 #include "platform/JSONValues.h"
  40 #include "wtf/PassOwnPtr.h"
  41
  42 namespace WebCore {
  43
  44 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
  45 {
  46     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
  47 }
  48
  49 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
  50 {
  51     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope));
  52 }
  53
  54 InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck)
  55     : m_nextInjectedScriptId(1)
  56     , m_injectedScriptHost(InjectedScriptHost::create())
  57     , m_inspectedStateAccessCheck(accessCheck)
  58 {
  59 }
  60
  61 InjectedScriptManager::~InjectedScriptManager()
  62 {
  63 }
  64
  65 void InjectedScriptManager::disconnect()
  66 {
  67     m_injectedScriptHost->disconnect();
  68     m_injectedScriptHost.clear();
  69 }
  70
  71 InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
  72 {
  73     return m_injectedScriptHost.get();
  74 }
  75
  76 InjectedScript InjectedScriptManager::injectedScriptForId(int id)
  77 {
  78     IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id);
  79     if (it != m_idToInjectedScript.end())
  80         return it->value;
  81     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
  82         if (it->value == id)
  83             return injectedScriptFor(it->key.get());
  84     }
  85     return InjectedScript();
  86 }
  87
  88 int InjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState)
  89 {
  90     ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState);
  91     if (it != m_scriptStateToId.end())
  92         return it->value;
  93     int id = m_nextInjectedScriptId++;
  94     m_scriptStateToId.set(scriptState, id);
  95     return id;
  96 }
  97
  98 InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
  99 {
 100     RefPtr<JSONValue> parsedObjectId = parseJSON(objectId);
 101     if (parsedObjectId && parsedObjectId->type() == JSONValue::TypeObject) {
 102         long injectedScriptId = 0;
 103         bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId);
 104         if (success)
 105             return m_idToInjectedScript.get(injectedScriptId);
 106     }
 107     return InjectedScript();
 108 }
 109
 110 void InjectedScriptManager::discardInjectedScripts()
 111 {
 112     m_idToInjectedScript.clear();
 113     m_scriptStateToId.clear();
 114 }
 115
 116 void InjectedScriptManager::discardInjectedScriptsFor(DOMWindow* window)
 117 {
 118     if (m_scriptStateToId.isEmpty())
 119         return;
 120
 121     Vector<long> idsToRemove;
 122     IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
 123     for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
 124         ScriptState* scriptState = it->value.scriptState();
 125         if (window != scriptState->domWindow())
 126             continue;
 127         m_scriptStateToId.remove(scriptState);
 128         idsToRemove.append(it->key);
 129     }
 130     m_idToInjectedScript.removeAll(idsToRemove);
 131
 132     // Now remove script states that have id but no injected script.
 133     Vector<ScriptState*> scriptStatesToRemove;
 134     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
 135         ScriptState* scriptState = it->key.get();
 136         if (window == scriptState->domWindow())
 137             scriptStatesToRemove.append(scriptState);
 138     }
 139     m_scriptStateToId.removeAll(scriptStatesToRemove);
 140 }
 141
 142 bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*)
 143 {
 144     return true;
 145 }
 146
 147 void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
 148 {
 149     Vector<int> keys;
 150     keys.appendRange(m_idToInjectedScript.keys().begin(), m_idToInjectedScript.keys().end());
 151     for (Vector<int>::iterator k = keys.begin(); k != keys.end(); ++k) {
 152         IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(*k);
 153         if (s != m_idToInjectedScript.end())
 154             s->value.releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here.
 155     }
 156 }
 157
 158 String InjectedScriptManager::injectedScriptSource()
 159 {
 160     return String(reinterpret_cast<const char*>(InjectedScriptSource_js), sizeof(InjectedScriptSource_js));
 161 }
 162
 163 InjectedScript InjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState)
 164 {
 165     ScriptStateToId::iterator it = m_scriptStateToId.find(inspectedScriptState);
 166     if (it != m_scriptStateToId.end()) {
 167         IdToInjectedScriptMap::iterator it1 = m_idToInjectedScript.find(it->value);
 168         if (it1 != m_idToInjectedScript.end())
 169             return it1->value;
 170     }
 171
 172     if (!m_inspectedStateAccessCheck(inspectedScriptState))
 173         return InjectedScript();
 174
 175     int id = injectedScriptIdFor(inspectedScriptState);
 176     ScriptObject injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedScriptState, id);
 177     InjectedScript result(injectedScriptObject, m_inspectedStateAccessCheck);
 178     m_idToInjectedScript.set(id, result);
 179     return result;
 180 }
 181
 182 } // namespace WebCore
 183