InjectedScriptManager.cpp [plain text]
#include "config.h"
#if ENABLE(INSPECTOR)
#include "InjectedScriptManager.h"
#include "InjectedScript.h"
#include "InjectedScriptHost.h"
#include "InjectedScriptSource.h"
#include "InspectorValues.h"
#include <wtf/PassOwnPtr.h>
#include <wtf/StdLibExtras.h>
using namespace std;
namespace WebCore {
PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
{
return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
}
PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
{
return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerContext));
}
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(long id)
{
return m_idToInjectedScript.get(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 injectedScriptForId(injectedScriptId);
}
return InjectedScript();
}
void InjectedScriptManager::discardInjectedScripts()
{
IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it)
discardInjectedScript(it->second.scriptState());
m_idToInjectedScript.clear();
}
void InjectedScriptManager::discardInjectedScriptsFor(DOMWindow* window)
{
Vector<long> idsToRemove;
IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
ScriptState* scriptState = it->second.scriptState();
if (window != domWindowFromScriptState(scriptState))
continue;
discardInjectedScript(scriptState);
idsToRemove.append(it->first);
}
for (size_t i = 0; i < idsToRemove.size(); i++)
m_idToInjectedScript.remove(idsToRemove[i]);
}
bool InjectedScriptManager::canAccessInspectedWorkerContext(ScriptState*)
{
return true;
}
void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
{
for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != m_idToInjectedScript.end(); ++it)
it->second.releaseObjectGroup(objectGroup);
}
String InjectedScriptManager::injectedScriptSource()
{
return String(reinterpret_cast<const char*>(InjectedScriptSource_js), sizeof(InjectedScriptSource_js));
}
pair<long, ScriptObject> InjectedScriptManager::injectScript(const String& source, ScriptState* scriptState)
{
long id = m_nextInjectedScriptId++;
return std::make_pair(id, createInjectedScript(source, scriptState, id));
}
}
#endif // ENABLE(INSPECTOR)