Unreviewed, rolling out r144924.
[WebKit-https.git] / Source / WebCore / 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
33 #if ENABLE(INSPECTOR)
34
35 #include "InjectedScriptManager.h"
36
37 #include "InjectedScript.h"
38 #include "InjectedScriptHost.h"
39 #include "InjectedScriptSource.h"
40 #include "InspectorValues.h"
41 #include "ScriptObject.h"
42 #include <wtf/PassOwnPtr.h>
43
44 namespace WebCore {
45
46 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
47 {
48     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
49 }
50
51 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
52 {
53     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerContext));
54 }
55
56 InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck)
57     : m_nextInjectedScriptId(1)
58     , m_injectedScriptHost(InjectedScriptHost::create())
59     , m_inspectedStateAccessCheck(accessCheck)
60 {
61 }
62
63 InjectedScriptManager::~InjectedScriptManager()
64 {
65 }
66
67 void InjectedScriptManager::disconnect()
68 {
69     m_injectedScriptHost->disconnect();
70     m_injectedScriptHost.clear();
71 }
72
73 InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
74 {
75     return m_injectedScriptHost.get();
76 }
77
78 InjectedScript InjectedScriptManager::injectedScriptForId(int id)
79 {
80     IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id);
81     if (it != m_idToInjectedScript.end())
82         return it->value;
83     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
84         if (it->value == id)
85             return injectedScriptFor(it->key);
86     }
87     return InjectedScript();
88 }
89
90 int InjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState)
91 {
92     ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState);
93     if (it != m_scriptStateToId.end())
94         return it->value;
95     int id = m_nextInjectedScriptId++;
96     m_scriptStateToId.set(scriptState, id);
97     return id;
98 }
99
100 InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
101 {
102     RefPtr<InspectorValue> parsedObjectId = InspectorValue::parseJSON(objectId);
103     if (parsedObjectId && parsedObjectId->type() == InspectorValue::TypeObject) {
104         long injectedScriptId = 0;
105         bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId);
106         if (success)
107             return m_idToInjectedScript.get(injectedScriptId);
108     }
109     return InjectedScript();
110 }
111
112 void InjectedScriptManager::discardInjectedScripts()
113 {
114     m_idToInjectedScript.clear();
115     m_scriptStateToId.clear();
116 }
117
118 void InjectedScriptManager::discardInjectedScriptsFor(DOMWindow* window)
119 {
120     if (m_scriptStateToId.isEmpty())
121         return;
122
123     Vector<long> idsToRemove;
124     IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
125     for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
126         ScriptState* scriptState = it->value.scriptState();
127         if (window != domWindowFromScriptState(scriptState))
128             continue;
129         m_scriptStateToId.remove(scriptState);
130         idsToRemove.append(it->key);
131     }
132
133     for (size_t i = 0; i < idsToRemove.size(); i++)
134         m_idToInjectedScript.remove(idsToRemove[i]);
135
136     // Now remove script states that have id but no injected script.
137     Vector<ScriptState*> scriptStatesToRemove;
138     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
139         ScriptState* scriptState = it->key;
140         if (window == domWindowFromScriptState(scriptState))
141             scriptStatesToRemove.append(scriptState);
142     }
143     for (size_t i = 0; i < scriptStatesToRemove.size(); i++)
144         m_scriptStateToId.remove(scriptStatesToRemove[i]);
145 }
146
147 bool InjectedScriptManager::canAccessInspectedWorkerContext(ScriptState*)
148 {
149     return true;
150 }
151
152 void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
153 {
154     for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != m_idToInjectedScript.end(); ++it)
155         it->value.releaseObjectGroup(objectGroup);
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
184 #endif // ENABLE(INSPECTOR)