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
43 namespace blink {
44
45 #if ENABLE(OILPAN)
~CallbackData()46 InjectedScriptManager::CallbackData::~CallbackData()
47 {
48 WrapperPersistent<InjectedScriptHost>::destroy(hostPtr);
49 }
50 #endif
51
createForPage()52 PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
53 {
54 return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
55 }
56
createForWorker()57 PassOwnPtrWillBeRawPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
58 {
59 return adoptPtrWillBeNoop(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope));
60 }
61
InjectedScriptManager(InspectedStateAccessCheck accessCheck)62 InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck)
63 : m_nextInjectedScriptId(1)
64 , m_injectedScriptHost(InjectedScriptHost::create())
65 , m_inspectedStateAccessCheck(accessCheck)
66 {
67 }
68
~InjectedScriptManager()69 InjectedScriptManager::~InjectedScriptManager()
70 {
71 }
72
trace(Visitor * visitor)73 void InjectedScriptManager::trace(Visitor* visitor)
74 {
75 visitor->trace(m_injectedScriptHost);
76 }
77
disconnect()78 void InjectedScriptManager::disconnect()
79 {
80 m_injectedScriptHost->disconnect();
81 m_injectedScriptHost.clear();
82 }
83
injectedScriptHost()84 InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
85 {
86 return m_injectedScriptHost.get();
87 }
88
injectedScriptForId(int id)89 InjectedScript 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
injectedScriptIdFor(ScriptState * scriptState)101 int 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
injectedScriptForObjectId(const String & objectId)111 InjectedScript 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
discardInjectedScripts()123 void InjectedScriptManager::discardInjectedScripts()
124 {
125 m_idToInjectedScript.clear();
126 m_scriptStateToId.clear();
127 }
128
discardInjectedScriptsFor(LocalDOMWindow * window)129 void 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
canAccessInspectedWorkerGlobalScope(ScriptState *)155 bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*)
156 {
157 return true;
158 }
159
releaseObjectGroup(const String & objectGroup)160 void 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
injectedScriptSource()171 String InjectedScriptManager::injectedScriptSource()
172 {
173 const blink::WebData& injectedScriptSourceResource = blink::Platform::current()->loadResource("InjectedScriptSource.js");
174 return String(injectedScriptSourceResource.data(), injectedScriptSourceResource.size());
175 }
176
injectedScriptFor(ScriptState * inspectedScriptState)177 InjectedScript 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