9a9018bdd82917a39b30953f0f8741832f3190fe
[WebKit-https.git] / Source / JavaScriptCore / inspector / InjectedScriptManager.cpp
1 /*
2  * Copyright (C) 2007, 2008, 2013 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 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 "InjectedScriptManager.h"
33
34 #if ENABLE(INSPECTOR)
35
36 #include "Completion.h"
37 #include "InjectedScriptHost.h"
38 #include "InjectedScriptSource.h"
39 #include "InspectorValues.h"
40 #include "JSInjectedScriptHost.h"
41 #include "JSLock.h"
42 #include "ScriptObject.h"
43 #include "SourceCode.h"
44
45 using namespace JSC;
46
47 namespace Inspector {
48
49 InjectedScriptManager::InjectedScriptManager(InspectorEnvironment& environment, PassRefPtr<InjectedScriptHost> injectedScriptHost)
50     : m_environment(environment)
51     , m_injectedScriptHost(injectedScriptHost)
52     , m_nextInjectedScriptId(1)
53 {
54 }
55
56 InjectedScriptManager::~InjectedScriptManager()
57 {
58 }
59
60 void InjectedScriptManager::disconnect()
61 {
62     discardInjectedScripts();
63 }
64
65 InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
66 {
67     return m_injectedScriptHost.get();
68 }
69
70 InjectedScript InjectedScriptManager::injectedScriptForId(int id)
71 {
72     auto it = m_idToInjectedScript.find(id);
73     if (it != m_idToInjectedScript.end())
74         return it->value;
75
76     for (auto it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
77         if (it->value == id)
78             return injectedScriptFor(it->key);
79     }
80
81     return InjectedScript();
82 }
83
84 int InjectedScriptManager::injectedScriptIdFor(ExecState* scriptState)
85 {
86     auto it = m_scriptStateToId.find(scriptState);
87     if (it != m_scriptStateToId.end())
88         return it->value;
89
90     int id = m_nextInjectedScriptId++;
91     m_scriptStateToId.set(scriptState, id);
92     return id;
93 }
94
95 InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
96 {
97     RefPtr<InspectorValue> parsedObjectId;
98     if (!InspectorValue::parseJSON(objectId, parsedObjectId))
99         return InjectedScript();
100
101     RefPtr<InspectorObject> resultObject;
102     if (!parsedObjectId->asObject(resultObject))
103         return InjectedScript();
104
105     long injectedScriptId = 0;
106     if (!resultObject->getInteger(ASCIILiteral("injectedScriptId"), injectedScriptId))
107         return InjectedScript();
108
109     return m_idToInjectedScript.get(injectedScriptId);
110 }
111
112 void InjectedScriptManager::discardInjectedScripts()
113 {
114     m_injectedScriptHost->clearAllWrappers();
115     m_idToInjectedScript.clear();
116     m_scriptStateToId.clear();
117 }
118
119 void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
120 {
121     for (auto it = m_idToInjectedScript.begin(); it != m_idToInjectedScript.end(); ++it)
122         it->value.releaseObjectGroup(objectGroup);
123 }
124
125 String InjectedScriptManager::injectedScriptSource()
126 {
127     return StringImpl::createWithoutCopying(InjectedScriptSource_js, sizeof(InjectedScriptSource_js));
128 }
129
130 Deprecated::ScriptObject InjectedScriptManager::createInjectedScript(const String& source, ExecState* scriptState, int id)
131 {
132     JSLockHolder lock(scriptState);
133
134     SourceCode sourceCode = makeSource(source);
135     JSGlobalObject* globalObject = scriptState->lexicalGlobalObject();
136     JSValue globalThisValue = scriptState->globalThisValue();
137
138     JSValue evaluationException;
139     InspectorEvaluateHandler evaluateHandler = m_environment.evaluateHandler();
140     JSValue functionValue = evaluateHandler(scriptState, sourceCode, globalThisValue, &evaluationException);
141     if (evaluationException)
142         return Deprecated::ScriptObject();
143
144     CallData callData;
145     CallType callType = getCallData(functionValue, callData);
146     if (callType == CallTypeNone)
147         return Deprecated::ScriptObject();
148
149     MarkedArgumentBuffer args;
150     args.append(m_injectedScriptHost->jsWrapper(scriptState, globalObject));
151     args.append(globalThisValue);
152     args.append(jsNumber(id));
153
154     JSValue result = JSC::call(scriptState, functionValue, callType, callData, globalThisValue, args);
155     if (result.isObject())
156         return Deprecated::ScriptObject(scriptState, result.getObject());
157
158     return Deprecated::ScriptObject();
159 }
160
161 InjectedScript InjectedScriptManager::injectedScriptFor(ExecState* inspectedExecState)
162 {
163     auto it = m_scriptStateToId.find(inspectedExecState);
164     if (it != m_scriptStateToId.end()) {
165         auto it1 = m_idToInjectedScript.find(it->value);
166         if (it1 != m_idToInjectedScript.end())
167             return it1->value;
168     }
169
170     if (!m_environment.canAccessInspectedScriptState(inspectedExecState))
171         return InjectedScript();
172
173     int id = injectedScriptIdFor(inspectedExecState);
174     Deprecated::ScriptObject injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedExecState, id);
175     InjectedScript result(injectedScriptObject, &m_environment);
176     m_idToInjectedScript.set(id, result);
177     didCreateInjectedScript(result);
178     return result;
179 }
180
181 void InjectedScriptManager::didCreateInjectedScript(InjectedScript)
182 {
183     // Intentionally empty. This allows for subclasses to inject additional scripts.
184 }
185
186 } // namespace Inspector
187
188 #endif // ENABLE(INSPECTOR)