InspectorDOMStorageAgent.cpp [plain text]
#include "config.h"
#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 "InstrumentingAgents.h"
#include "Page.h"
#include "PageGroup.h"
#include "SecurityOrigin.h"
#include "Storage.h"
#include "StorageNamespace.h"
#include "StorageNamespaceProvider.h"
#include "VoidCallback.h"
#include <inspector/InspectorFrontendDispatchers.h>
#include <inspector/InspectorValues.h>
#include <wtf/Vector.h>
using namespace Inspector;
namespace WebCore {
InspectorDOMStorageAgent::InspectorDOMStorageAgent(WebAgentContext& context, InspectorPageAgent* pageAgent)
: InspectorAgentBase(ASCIILiteral("DOMStorage"), context)
, m_frontendDispatcher(std::make_unique<Inspector::DOMStorageFrontendDispatcher>(context.frontendRouter))
, m_backendDispatcher(Inspector::DOMStorageBackendDispatcher::create(context.backendDispatcher, this))
, m_pageAgent(pageAgent)
{
m_instrumentingAgents.setInspectorDOMStorageAgent(this);
}
InspectorDOMStorageAgent::~InspectorDOMStorageAgent()
{
m_instrumentingAgents.setInspectorDOMStorageAgent(nullptr);
}
void InspectorDOMStorageAgent::didCreateFrontendAndBackend(Inspector::FrontendRouter*, Inspector::BackendDispatcher*)
{
}
void InspectorDOMStorageAgent::willDestroyFrontendAndBackend(Inspector::DisconnectReason)
{
ErrorString unused;
disable(unused);
}
void InspectorDOMStorageAgent::enable(ErrorString&)
{
m_enabled = true;
}
void InspectorDOMStorageAgent::disable(ErrorString&)
{
m_enabled = false;
}
void InspectorDOMStorageAgent::getDOMStorageItems(ErrorString& errorString, const InspectorObject& storageId, RefPtr<Inspector::Protocol::Array<Inspector::Protocol::Array<String>>>& items)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = ASCIILiteral("No StorageArea for given storageId");
return;
}
auto storageItems = Inspector::Protocol::Array<Inspector::Protocol::Array<String>>::create();
for (unsigned i = 0; i < storageArea->length(); ++i) {
String key = storageArea->key(i);
String value = storageArea->item(key);
auto entry = Inspector::Protocol::Array<String>::create();
entry->addItem(key);
entry->addItem(value);
storageItems->addItem(WTFMove(entry));
}
items = WTFMove(storageItems);
}
void InspectorDOMStorageAgent::setDOMStorageItem(ErrorString& errorString, const InspectorObject& storageId, const String& key, const String& value)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = ASCIILiteral("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 InspectorObject& storageId, const String& key)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = ASCIILiteral("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();
}
RefPtr<Inspector::Protocol::DOMStorage::StorageId> InspectorDOMStorageAgent::storageId(SecurityOrigin* securityOrigin, bool isLocalStorage)
{
return Inspector::Protocol::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_enabled)
return;
RefPtr<Inspector::Protocol::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);
}
RefPtr<StorageArea> InspectorDOMStorageAgent::findStorageArea(ErrorString& errorString, const InspectorObject& storageId, Frame*& targetFrame)
{
String securityOrigin;
bool isLocalStorage = false;
bool success = storageId.getString(ASCIILiteral("securityOrigin"), securityOrigin);
if (success)
success = storageId.getBoolean(ASCIILiteral("isLocalStorage"), isLocalStorage);
if (!success) {
errorString = ASCIILiteral("Invalid storageId format");
targetFrame = nullptr;
return nullptr;
}
targetFrame = m_pageAgent->findFrameWithSecurityOrigin(securityOrigin);
if (!targetFrame) {
errorString = ASCIILiteral("Frame not found for the given security origin");
return nullptr;
}
if (!isLocalStorage)
return m_pageAgent->page().sessionStorage()->storageArea(targetFrame->document()->securityOrigin());
return m_pageAgent->page().storageNamespaceProvider().localStorageArea(*targetFrame->document());
}
}