PageRuntimeAgent.cpp [plain text]
#include "config.h"
#include "PageRuntimeAgent.h"
#include "Document.h"
#include "Frame.h"
#include "InspectorPageAgent.h"
#include "InstrumentingAgents.h"
#include "JSDOMWindowBase.h"
#include "MainFrame.h"
#include "Page.h"
#include "PageConsoleClient.h"
#include "ScriptController.h"
#include "ScriptState.h"
#include "SecurityOrigin.h"
#include <inspector/InjectedScript.h>
#include <inspector/InjectedScriptManager.h>
using Inspector::Protocol::Runtime::ExecutionContextDescription;
using namespace Inspector;
namespace WebCore {
PageRuntimeAgent::PageRuntimeAgent(PageAgentContext& context, InspectorPageAgent* pageAgent)
: InspectorRuntimeAgent(context)
, m_frontendDispatcher(std::make_unique<Inspector::RuntimeFrontendDispatcher>(context.frontendRouter))
, m_backendDispatcher(Inspector::RuntimeBackendDispatcher::create(context.backendDispatcher, this))
, m_pageAgent(pageAgent)
, m_inspectedPage(context.inspectedPage)
{
}
void PageRuntimeAgent::didCreateFrontendAndBackend(Inspector::FrontendRouter*, Inspector::BackendDispatcher*)
{
}
void PageRuntimeAgent::willDestroyFrontendAndBackend(Inspector::DisconnectReason reason)
{
String unused;
disable(unused);
InspectorRuntimeAgent::willDestroyFrontendAndBackend(reason);
}
void PageRuntimeAgent::enable(ErrorString& errorString)
{
if (enabled())
return;
InspectorRuntimeAgent::enable(errorString);
if (m_mainWorldContextCreated)
reportExecutionContextCreation();
}
void PageRuntimeAgent::disable(ErrorString& errorString)
{
if (!enabled())
return;
InspectorRuntimeAgent::disable(errorString);
}
void PageRuntimeAgent::didCreateMainWorldContext(Frame& frame)
{
m_mainWorldContextCreated = true;
if (!enabled())
return;
String frameId = m_pageAgent->frameId(&frame);
JSC::ExecState* scriptState = mainWorldExecState(&frame);
notifyContextCreated(frameId, scriptState, nullptr, true);
}
InjectedScript PageRuntimeAgent::injectedScriptForEval(ErrorString& errorString, const int* executionContextId)
{
if (!executionContextId) {
JSC::ExecState* scriptState = mainWorldExecState(&m_inspectedPage.mainFrame());
InjectedScript result = injectedScriptManager().injectedScriptFor(scriptState);
if (result.hasNoValue())
errorString = ASCIILiteral("Internal error: main world execution context not found.");
return result;
}
InjectedScript injectedScript = injectedScriptManager().injectedScriptForId(*executionContextId);
if (injectedScript.hasNoValue())
errorString = ASCIILiteral("Execution context with given id not found.");
return injectedScript;
}
void PageRuntimeAgent::muteConsole()
{
PageConsoleClient::mute();
}
void PageRuntimeAgent::unmuteConsole()
{
PageConsoleClient::unmute();
}
void PageRuntimeAgent::reportExecutionContextCreation()
{
Vector<std::pair<JSC::ExecState*, SecurityOrigin*>> isolatedContexts;
for (Frame* frame = &m_inspectedPage.mainFrame(); frame; frame = frame->tree().traverseNext()) {
if (!frame->script().canExecuteScripts(NotAboutToExecuteScript))
continue;
String frameId = m_pageAgent->frameId(frame);
JSC::ExecState* scriptState = mainWorldExecState(frame);
notifyContextCreated(frameId, scriptState, nullptr, true);
frame->script().collectIsolatedContexts(isolatedContexts);
if (isolatedContexts.isEmpty())
continue;
for (auto& context : isolatedContexts)
notifyContextCreated(frameId, context.first, context.second, false);
isolatedContexts.clear();
}
}
void PageRuntimeAgent::notifyContextCreated(const String& frameId, JSC::ExecState* scriptState, SecurityOrigin* securityOrigin, bool isPageContext)
{
ASSERT(securityOrigin || isPageContext);
InjectedScript result = injectedScriptManager().injectedScriptFor(scriptState);
if (result.hasNoValue())
return;
int executionContextId = injectedScriptManager().injectedScriptIdFor(scriptState);
String name = securityOrigin ? securityOrigin->toRawString() : String();
m_frontendDispatcher->executionContextCreated(ExecutionContextDescription::create()
.setId(executionContextId)
.setIsPageContext(isPageContext)
.setName(name)
.setFrameId(frameId)
.release());
}
}