| /* |
| * Copyright (C) 2007, 2008 Apple Inc. All rights reserved. |
| * Copyright (C) 2008 Matt Lilek <webkit@mattlilek.com> |
| * Copyright (C) 2012 Google Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of |
| * its contributors may be used to endorse or promote products derived |
| * from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY |
| * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY |
| * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
| * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| |
| #if ENABLE(INSPECTOR) |
| |
| #include "InjectedScriptManager.h" |
| |
| #include "InjectedScript.h" |
| #include "InjectedScriptHost.h" |
| #include "InjectedScriptSource.h" |
| #include "InspectorValues.h" |
| #include "ScriptObject.h" |
| #include <wtf/PassOwnPtr.h> |
| |
| namespace WebCore { |
| |
| PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForPage() |
| { |
| return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow)); |
| } |
| |
| PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForWorker() |
| { |
| return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope)); |
| } |
| |
| InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck) |
| : m_nextInjectedScriptId(1) |
| , m_injectedScriptHost(InjectedScriptHost::create()) |
| , m_inspectedStateAccessCheck(accessCheck) |
| { |
| } |
| |
| InjectedScriptManager::~InjectedScriptManager() |
| { |
| } |
| |
| void InjectedScriptManager::disconnect() |
| { |
| m_injectedScriptHost->disconnect(); |
| m_injectedScriptHost.clear(); |
| } |
| |
| InjectedScriptHost* InjectedScriptManager::injectedScriptHost() |
| { |
| return m_injectedScriptHost.get(); |
| } |
| |
| InjectedScript InjectedScriptManager::injectedScriptForId(int id) |
| { |
| IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id); |
| if (it != m_idToInjectedScript.end()) |
| return it->value; |
| for (ExecStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) { |
| if (it->value == id) |
| return injectedScriptFor(it->key); |
| } |
| return InjectedScript(); |
| } |
| |
| int InjectedScriptManager::injectedScriptIdFor(JSC::ExecState* scriptState) |
| { |
| ExecStateToId::iterator it = m_scriptStateToId.find(scriptState); |
| if (it != m_scriptStateToId.end()) |
| return it->value; |
| int id = m_nextInjectedScriptId++; |
| m_scriptStateToId.set(scriptState, id); |
| return id; |
| } |
| |
| InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId) |
| { |
| RefPtr<InspectorValue> parsedObjectId = InspectorValue::parseJSON(objectId); |
| if (parsedObjectId && parsedObjectId->type() == InspectorValue::TypeObject) { |
| long injectedScriptId = 0; |
| bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId); |
| if (success) |
| return m_idToInjectedScript.get(injectedScriptId); |
| } |
| return InjectedScript(); |
| } |
| |
| void InjectedScriptManager::discardInjectedScripts() |
| { |
| m_idToInjectedScript.clear(); |
| m_scriptStateToId.clear(); |
| } |
| |
| void InjectedScriptManager::discardInjectedScriptsFor(DOMWindow* window) |
| { |
| if (m_scriptStateToId.isEmpty()) |
| return; |
| |
| Vector<long> idsToRemove; |
| IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end(); |
| for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) { |
| JSC::ExecState* scriptState = it->value.scriptState(); |
| if (window != domWindowFromExecState(scriptState)) |
| continue; |
| m_scriptStateToId.remove(scriptState); |
| idsToRemove.append(it->key); |
| } |
| |
| for (size_t i = 0; i < idsToRemove.size(); i++) |
| m_idToInjectedScript.remove(idsToRemove[i]); |
| |
| // Now remove script states that have id but no injected script. |
| Vector<JSC::ExecState*> scriptStatesToRemove; |
| for (ExecStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) { |
| JSC::ExecState* scriptState = it->key; |
| if (window == domWindowFromExecState(scriptState)) |
| scriptStatesToRemove.append(scriptState); |
| } |
| for (size_t i = 0; i < scriptStatesToRemove.size(); i++) |
| m_scriptStateToId.remove(scriptStatesToRemove[i]); |
| } |
| |
| bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(JSC::ExecState*) |
| { |
| return true; |
| } |
| |
| void InjectedScriptManager::releaseObjectGroup(const String& objectGroup) |
| { |
| for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != m_idToInjectedScript.end(); ++it) |
| it->value.releaseObjectGroup(objectGroup); |
| } |
| |
| String InjectedScriptManager::injectedScriptSource() |
| { |
| return String(reinterpret_cast<const char*>(InjectedScriptSource_js), sizeof(InjectedScriptSource_js)); |
| } |
| |
| InjectedScript InjectedScriptManager::injectedScriptFor(JSC::ExecState* inspectedExecState) |
| { |
| ExecStateToId::iterator it = m_scriptStateToId.find(inspectedExecState); |
| if (it != m_scriptStateToId.end()) { |
| IdToInjectedScriptMap::iterator it1 = m_idToInjectedScript.find(it->value); |
| if (it1 != m_idToInjectedScript.end()) |
| return it1->value; |
| } |
| |
| if (!m_inspectedStateAccessCheck(inspectedExecState)) |
| return InjectedScript(); |
| |
| int id = injectedScriptIdFor(inspectedExecState); |
| ScriptObject injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedExecState, id); |
| InjectedScript result(injectedScriptObject, m_inspectedStateAccessCheck); |
| m_idToInjectedScript.set(id, result); |
| return result; |
| } |
| |
| } // namespace WebCore |
| |
| #endif // ENABLE(INSPECTOR) |