blob: fe1133c7feb5edabc4c36a85e0a205a4b00ce62b [file] [log] [blame]
/*
* Copyright (C) 2007-2017 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 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"
#include "InjectedScriptManager.h"
#include "CatchScope.h"
#include "Completion.h"
#include "InjectedScriptHost.h"
#include "InjectedScriptSource.h"
#include "JSCInlines.h"
#include "JSInjectedScriptHost.h"
#include "JSLock.h"
#include "ScriptObject.h"
#include "SourceCode.h"
#include <wtf/JSONValues.h>
using namespace JSC;
namespace Inspector {
InjectedScriptManager::InjectedScriptManager(InspectorEnvironment& environment, Ref<InjectedScriptHost>&& injectedScriptHost)
: m_environment(environment)
, m_injectedScriptHost(WTFMove(injectedScriptHost))
, m_nextInjectedScriptId(1)
{
}
InjectedScriptManager::~InjectedScriptManager()
{
}
void InjectedScriptManager::disconnect()
{
discardInjectedScripts();
}
void InjectedScriptManager::discardInjectedScripts()
{
m_injectedScriptHost->clearAllWrappers();
m_idToInjectedScript.clear();
m_scriptStateToId.clear();
}
InjectedScriptHost& InjectedScriptManager::injectedScriptHost()
{
return m_injectedScriptHost.get();
}
InjectedScript InjectedScriptManager::injectedScriptForId(int id)
{
auto it = m_idToInjectedScript.find(id);
if (it != m_idToInjectedScript.end())
return it->value;
for (auto it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
if (it->value == id)
return injectedScriptFor(it->key);
}
return InjectedScript();
}
int InjectedScriptManager::injectedScriptIdFor(ExecState* scriptState)
{
auto 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<JSON::Value> parsedObjectId;
if (!JSON::Value::parseJSON(objectId, parsedObjectId))
return InjectedScript();
RefPtr<JSON::Object> resultObject;
if (!parsedObjectId->asObject(resultObject))
return InjectedScript();
long injectedScriptId = 0;
if (!resultObject->getInteger(ASCIILiteral("injectedScriptId"), injectedScriptId))
return InjectedScript();
return m_idToInjectedScript.get(injectedScriptId);
}
void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
{
for (auto& injectedScript : m_idToInjectedScript.values())
injectedScript.releaseObjectGroup(objectGroup);
}
void InjectedScriptManager::clearExceptionValue()
{
for (auto& injectedScript : m_idToInjectedScript.values())
injectedScript.clearExceptionValue();
}
String InjectedScriptManager::injectedScriptSource()
{
return StringImpl::createWithoutCopying(InjectedScriptSource_js, sizeof(InjectedScriptSource_js));
}
JSC::JSObject* InjectedScriptManager::createInjectedScript(const String& source, ExecState* scriptState, int id)
{
VM& vm = scriptState->vm();
JSLockHolder lock(vm);
auto scope = DECLARE_CATCH_SCOPE(vm);
SourceCode sourceCode = makeSource(source, { });
JSGlobalObject* globalObject = scriptState->lexicalGlobalObject();
JSValue globalThisValue = scriptState->globalThisValue();
NakedPtr<Exception> evaluationException;
InspectorEvaluateHandler evaluateHandler = m_environment.evaluateHandler();
JSValue functionValue = evaluateHandler(scriptState, sourceCode, globalThisValue, evaluationException);
if (evaluationException)
return nullptr;
CallData callData;
CallType callType = getCallData(functionValue, callData);
if (callType == CallType::None)
return nullptr;
MarkedArgumentBuffer args;
args.append(m_injectedScriptHost->wrapper(scriptState, globalObject));
args.append(globalThisValue);
args.append(jsNumber(id));
ASSERT(!args.hasOverflowed());
JSValue result = JSC::call(scriptState, functionValue, callType, callData, globalThisValue, args);
scope.clearException();
return result.getObject();
}
InjectedScript InjectedScriptManager::injectedScriptFor(ExecState* inspectedExecState)
{
auto it = m_scriptStateToId.find(inspectedExecState);
if (it != m_scriptStateToId.end()) {
auto it1 = m_idToInjectedScript.find(it->value);
if (it1 != m_idToInjectedScript.end())
return it1->value;
}
if (!m_environment.canAccessInspectedScriptState(inspectedExecState))
return InjectedScript();
int id = injectedScriptIdFor(inspectedExecState);
auto injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedExecState, id);
if (!injectedScriptObject) {
WTFLogAlways("Failed to parse/execute InjectedScriptSource.js!");
WTFLogAlways("%s\n", injectedScriptSource().ascii().data());
RELEASE_ASSERT_NOT_REACHED();
}
InjectedScript result({ inspectedExecState, injectedScriptObject }, &m_environment);
m_idToInjectedScript.set(id, result);
didCreateInjectedScript(result);
return result;
}
void InjectedScriptManager::didCreateInjectedScript(const InjectedScript&)
{
// Intentionally empty. This allows for subclasses to inject additional scripts.
}
} // namespace Inspector