blob: e13cc729e8bed484b575739a221d0c557a41a7c6 [file] [log] [blame]
/*
* Copyright (C) 2007-2021 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 "InjectedScriptHost.h"
#include "InjectedScriptSource.h"
#include "JSLock.h"
#include "JSObjectInlines.h"
#include "SourceCode.h"
#include <wtf/JSONValues.h>
namespace Inspector {
using namespace JSC;
InjectedScriptManager::InjectedScriptManager(InspectorEnvironment& environment, Ref<InjectedScriptHost>&& injectedScriptHost)
: m_environment(environment)
, m_injectedScriptHost(WTFMove(injectedScriptHost))
, m_nextInjectedScriptId(1)
{
}
InjectedScriptManager::~InjectedScriptManager()
{
}
void InjectedScriptManager::connect()
{
}
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(JSGlobalObject* globalObject)
{
auto it = m_scriptStateToId.find(globalObject);
if (it != m_scriptStateToId.end())
return it->value;
int id = m_nextInjectedScriptId++;
m_scriptStateToId.set(globalObject, id);
return id;
}
InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
{
auto parsedObjectId = JSON::Value::parseJSON(objectId);
if (!parsedObjectId)
return InjectedScript();
auto resultObject = parsedObjectId->asObject();
if (!resultObject)
return InjectedScript();
auto injectedScriptId = resultObject->getInteger("injectedScriptId"_s);
if (!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::clearEventValue()
{
for (auto& injectedScript : m_idToInjectedScript.values())
injectedScript.clearEventValue();
}
void InjectedScriptManager::clearExceptionValue()
{
for (auto& injectedScript : m_idToInjectedScript.values())
injectedScript.clearExceptionValue();
}
String InjectedScriptManager::injectedScriptSource()
{
return StringImpl::createWithoutCopying(InjectedScriptSource_js, sizeof(InjectedScriptSource_js));
}
Expected<JSObject*, NakedPtr<Exception>> InjectedScriptManager::createInjectedScript(const String& source, JSGlobalObject* globalObject, int id)
{
VM& vm = globalObject->vm();
JSLockHolder lock(vm);
auto scope = DECLARE_CATCH_SCOPE(vm);
SourceCode sourceCode = makeSource(source, { });
JSValue globalThisValue = globalObject->globalThis();
NakedPtr<Exception> evaluationException;
InspectorEvaluateHandler evaluateHandler = m_environment.evaluateHandler();
JSValue functionValue = evaluateHandler(globalObject, sourceCode, globalThisValue, evaluationException);
if (evaluationException)
return makeUnexpected(evaluationException);
auto callData = getCallData(vm, functionValue);
if (callData.type == CallData::Type::None)
return nullptr;
MarkedArgumentBuffer args;
args.append(m_injectedScriptHost->wrapper(globalObject));
args.append(globalThisValue);
args.append(jsNumber(id));
ASSERT(!args.hasOverflowed());
JSValue result = JSC::call(globalObject, functionValue, callData, globalThisValue, args);
scope.clearException();
return result.getObject();
}
InjectedScript InjectedScriptManager::injectedScriptFor(JSGlobalObject* globalObject)
{
auto it = m_scriptStateToId.find(globalObject);
if (it != m_scriptStateToId.end()) {
auto it1 = m_idToInjectedScript.find(it->value);
if (it1 != m_idToInjectedScript.end())
return it1->value;
}
if (!m_environment.canAccessInspectedScriptState(globalObject))
return InjectedScript();
int id = injectedScriptIdFor(globalObject);
auto createResult = createInjectedScript(injectedScriptSource(), globalObject, id);
if (!createResult) {
auto& error = createResult.error();
ASSERT(error);
if (globalObject->vm().isTerminationException(error))
return InjectedScript();
unsigned line = 0;
unsigned column = 0;
auto& stack = error->stack();
if (stack.size() > 0)
stack[0].computeLineAndColumn(line, column);
WTFLogAlways("Error when creating injected script: %s (%d:%d)\n", error->value().toWTFString(globalObject).utf8().data(), line, column);
WTFLogAlways("%s\n", injectedScriptSource().utf8().data());
RELEASE_ASSERT_NOT_REACHED();
}
if (!createResult.value()) {
WTFLogAlways("Missing injected script object");
WTFLogAlways("%s\n", injectedScriptSource().utf8().data());
RELEASE_ASSERT_NOT_REACHED();
}
InjectedScript result({ globalObject, createResult.value() }, &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