InspectorDOMStorageAgent.cpp [plain text]
#include "config.h"
#if ENABLE(INSPECTOR)
#include "InspectorDOMStorageAgent.h"
#include "DOMWindow.h"
#include "Database.h"
#include "Document.h"
#include "ExceptionCode.h"
#include "ExceptionCodeDescription.h"
#include "Frame.h"
#include "InspectorPageAgent.h"
#include "InspectorWebFrontendDispatchers.h"
#include "InstrumentingAgents.h"
#include "Page.h"
#include "PageGroup.h"
#include "SecurityOrigin.h"
#include "Storage.h"
#include "StorageArea.h"
#include "StorageNamespace.h"
#include "VoidCallback.h"
#include <inspector/InspectorValues.h>
#include <wtf/Vector.h>
using namespace Inspector;
namespace WebCore {
InspectorDOMStorageAgent::InspectorDOMStorageAgent(InstrumentingAgents* instrumentingAgents, InspectorPageAgent* pageAgent)
: InspectorAgentBase(ASCIILiteral("DOMStorage"), instrumentingAgents)
, m_pageAgent(pageAgent)
, m_enabled(false)
{
m_instrumentingAgents->setInspectorDOMStorageAgent(this);
}
InspectorDOMStorageAgent::~InspectorDOMStorageAgent()
{
m_instrumentingAgents->setInspectorDOMStorageAgent(nullptr);
m_instrumentingAgents = nullptr;
}
void InspectorDOMStorageAgent::didCreateFrontendAndBackend(Inspector::InspectorFrontendChannel* frontendChannel, InspectorBackendDispatcher* backendDispatcher)
{
m_frontendDispatcher = std::make_unique<InspectorDOMStorageFrontendDispatcher>(frontendChannel);
m_backendDispatcher = InspectorDOMStorageBackendDispatcher::create(backendDispatcher, this);
}
void InspectorDOMStorageAgent::willDestroyFrontendAndBackend(InspectorDisconnectReason)
{
m_frontendDispatcher = nullptr;
m_backendDispatcher.clear();
disable(nullptr);
}
void InspectorDOMStorageAgent::enable(ErrorString*)
{
m_enabled = true;
}
void InspectorDOMStorageAgent::disable(ErrorString*)
{
m_enabled = false;
}
void InspectorDOMStorageAgent::getDOMStorageItems(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, RefPtr<Inspector::TypeBuilder::Array<Inspector::TypeBuilder::Array<String>>>& items)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
if (errorString)
*errorString = "No StorageArea for given storageId";
return;
}
RefPtr<Inspector::TypeBuilder::Array<Inspector::TypeBuilder::Array<String>>> storageItems = Inspector::TypeBuilder::Array<Inspector::TypeBuilder::Array<String>>::create();
for (unsigned i = 0; i < storageArea->length(); ++i) {
String key = storageArea->key(i);
String value = storageArea->item(key);
RefPtr<Inspector::TypeBuilder::Array<String>> entry = Inspector::TypeBuilder::Array<String>::create();
entry->addItem(key);
entry->addItem(value);
storageItems->addItem(entry.release());
}
items = storageItems.release();
}
void InspectorDOMStorageAgent::setDOMStorageItem(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, const String& key, const String& value)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(nullptr, storageId, frame);
if (!storageArea) {
*errorString = "Storage not found";
return;
}
bool quotaException = false;
storageArea->setItem(frame, key, value, quotaException);
if (quotaException)
*errorString = ExceptionCodeDescription(QUOTA_EXCEEDED_ERR).name;
}
void InspectorDOMStorageAgent::removeDOMStorageItem(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, const String& key)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(nullptr, storageId, frame);
if (!storageArea) {
*errorString = "Storage not found";
return;
}
storageArea->removeItem(frame, key);
}
String InspectorDOMStorageAgent::storageId(Storage* storage)
{
ASSERT(storage);
Document* document = storage->frame()->document();
ASSERT(document);
DOMWindow* window = document->domWindow();
ASSERT(window);
RefPtr<SecurityOrigin> securityOrigin = document->securityOrigin();
bool isLocalStorage = window->optionalLocalStorage() == storage;
return storageId(securityOrigin.get(), isLocalStorage)->toJSONString();
}
PassRefPtr<Inspector::TypeBuilder::DOMStorage::StorageId> InspectorDOMStorageAgent::storageId(SecurityOrigin* securityOrigin, bool isLocalStorage)
{
return Inspector::TypeBuilder::DOMStorage::StorageId::create()
.setSecurityOrigin(securityOrigin->toRawString())
.setIsLocalStorage(isLocalStorage).release();
}
void InspectorDOMStorageAgent::didDispatchDOMStorageEvent(const String& key, const String& oldValue, const String& newValue, StorageType storageType, SecurityOrigin* securityOrigin, Page*)
{
if (!m_frontendDispatcher || !m_enabled)
return;
RefPtr<Inspector::TypeBuilder::DOMStorage::StorageId> id = storageId(securityOrigin, storageType == LocalStorage);
if (key.isNull())
m_frontendDispatcher->domStorageItemsCleared(id);
else if (newValue.isNull())
m_frontendDispatcher->domStorageItemRemoved(id, key);
else if (oldValue.isNull())
m_frontendDispatcher->domStorageItemAdded(id, key, newValue);
else
m_frontendDispatcher->domStorageItemUpdated(id, key, oldValue, newValue);
}
PassRefPtr<StorageArea> InspectorDOMStorageAgent::findStorageArea(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, Frame*& targetFrame)
{
String securityOrigin;
bool isLocalStorage = false;
bool success = storageId->getString("securityOrigin", &securityOrigin);
if (success)
success = storageId->getBoolean("isLocalStorage", &isLocalStorage);
if (!success) {
if (errorString)
*errorString = "Invalid storageId format";
targetFrame = nullptr;
return nullptr;
}
targetFrame = m_pageAgent->findFrameWithSecurityOrigin(securityOrigin);
if (!targetFrame) {
if (errorString)
*errorString = "Frame not found for the given security origin";
return nullptr;
}
Page* page = m_pageAgent->page();
if (isLocalStorage)
return page->group().localStorage()->storageArea(targetFrame->document()->securityOrigin());
return page->sessionStorage()->storageArea(targetFrame->document()->securityOrigin());
}
}
#endif // ENABLE(INSPECTOR)