PageRuntimeAgent.cpp [plain text]
#include "config.h"
#include "PageRuntimeAgent.h"
#if ENABLE(INSPECTOR)
#include "Document.h"
#include "Frame.h"
#include "InspectorPageAgent.h"
#include "InstrumentingAgents.h"
#include "JSDOMWindowBase.h"
#include "MainFrame.h"
#include "Page.h"
#include "PageConsole.h"
#include "ScriptController.h"
#include "ScriptState.h"
#include "SecurityOrigin.h"
#include <inspector/InjectedScript.h>
#include <inspector/InjectedScriptManager.h>
using Inspector::TypeBuilder::Runtime::ExecutionContextDescription;
using namespace Inspector;
namespace WebCore {
PageRuntimeAgent::PageRuntimeAgent(InjectedScriptManager* injectedScriptManager, Page* page, InspectorPageAgent* pageAgent)
: InspectorRuntimeAgent(injectedScriptManager)
, m_inspectedPage(page)
, m_pageAgent(pageAgent)
, m_mainWorldContextCreated(false)
{
}
void PageRuntimeAgent::didCreateFrontendAndBackend(Inspector::InspectorFrontendChannel* frontendChannel, InspectorBackendDispatcher* backendDispatcher)
{
m_frontendDispatcher = std::make_unique<InspectorRuntimeFrontendDispatcher>(frontendChannel);
m_backendDispatcher = InspectorRuntimeBackendDispatcher::create(backendDispatcher, this);
}
void PageRuntimeAgent::willDestroyFrontendAndBackend(InspectorDisconnectReason)
{
m_frontendDispatcher = nullptr;
m_backendDispatcher.clear();
String errorString;
disable(&errorString);
}
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;
ASSERT(m_frontendDispatcher);
String frameId = m_pageAgent->frameId(frame);
JSC::ExecState* scriptState = mainWorldExecState(frame);
notifyContextCreated(frameId, scriptState, nullptr, true);
}
void PageRuntimeAgent::didCreateIsolatedContext(Frame* frame, JSC::ExecState* scriptState, SecurityOrigin* origin)
{
if (!enabled())
return;
ASSERT(m_frontendDispatcher);
String frameId = m_pageAgent->frameId(frame);
notifyContextCreated(frameId, scriptState, origin, false);
}
JSC::VM& PageRuntimeAgent::globalVM()
{
return JSDOMWindowBase::commonVM();
}
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()
{
PageConsole::mute();
}
void PageRuntimeAgent::unmuteConsole()
{
PageConsole::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 (size_t i = 0; i< isolatedContexts.size(); i++)
notifyContextCreated(frameId, isolatedContexts[i].first, isolatedContexts[i].second, false);
isolatedContexts.clear();
}
}
void PageRuntimeAgent::notifyContextCreated(const String& frameId, JSC::ExecState* scriptState, SecurityOrigin* securityOrigin, bool isPageContext)
{
ASSERT(securityOrigin || isPageContext);
int executionContextId = injectedScriptManager()->injectedScriptIdFor(scriptState);
String name = securityOrigin ? securityOrigin->toRawString() : "";
m_frontendDispatcher->executionContextCreated(ExecutionContextDescription::create()
.setId(executionContextId)
.setIsPageContext(isPageContext)
.setName(name)
.setFrameId(frameId)
.release());
}
}
#endif // ENABLE(INSPECTOR)