InspectorDOMStorageAgent.cpp [plain text]
#include "config.h"
#include "InspectorDOMStorageAgent.h"
#include "DOMException.h"
#include "DOMWindow.h"
#include "Database.h"
#include "Document.h"
#include "Frame.h"
#include "InspectorPageAgent.h"
#include "InstrumentingAgents.h"
#include "Page.h"
#include "SecurityOrigin.h"
#include "SecurityOriginData.h"
#include "Storage.h"
#include "StorageNamespace.h"
#include "StorageNamespaceProvider.h"
#include "StorageType.h"
#include "VoidCallback.h"
#include <JavaScriptCore/InspectorFrontendDispatchers.h>
#include <wtf/JSONValues.h>
namespace WebCore {
using namespace Inspector;
InspectorDOMStorageAgent::InspectorDOMStorageAgent(WebAgentContext& context, InspectorPageAgent* pageAgent)
: InspectorAgentBase("DOMStorage"_s, 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 JSON::Object& storageId, RefPtr<JSON::ArrayOf<JSON::ArrayOf<String>>>& items)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = "No StorageArea for given storageId"_s;
return;
}
auto storageItems = JSON::ArrayOf<JSON::ArrayOf<String>>::create();
for (unsigned i = 0; i < storageArea->length(); ++i) {
String key = storageArea->key(i);
String value = storageArea->item(key);
auto entry = JSON::ArrayOf<String>::create();
entry->addItem(key);
entry->addItem(value);
storageItems->addItem(WTFMove(entry));
}
items = WTFMove(storageItems);
}
void InspectorDOMStorageAgent::setDOMStorageItem(ErrorString& errorString, const JSON::Object& storageId, const String& key, const String& value)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = "Storage not found"_s;
return;
}
bool quotaException = false;
storageArea->setItem(frame, key, value, quotaException);
if (quotaException)
errorString = DOMException::name(QuotaExceededError);
}
void InspectorDOMStorageAgent::removeDOMStorageItem(ErrorString& errorString, const JSON::Object& storageId, const String& key)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
errorString = "Storage not found"_s;
return;
}
storageArea->removeItem(frame, key);
}
String InspectorDOMStorageAgent::storageId(Storage& storage)
{
Document* document = storage.frame()->document();
ASSERT(document);
DOMWindow* window = document->domWindow();
ASSERT(window);
Ref<SecurityOrigin> securityOrigin = document->securityOrigin();
bool isLocalStorage = window->optionalLocalStorage() == &storage;
return storageId(securityOrigin.ptr(), 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)
{
if (!m_enabled)
return;
RefPtr<Inspector::Protocol::DOMStorage::StorageId> id = storageId(securityOrigin, storageType == StorageType::Local);
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 JSON::Object& storageId, Frame*& targetFrame)
{
String securityOrigin;
bool isLocalStorage = false;
bool success = storageId.getString("securityOrigin"_s, securityOrigin);
if (success)
success = storageId.getBoolean("isLocalStorage"_s, isLocalStorage);
if (!success) {
errorString = "Invalid storageId format"_s;
targetFrame = nullptr;
return nullptr;
}
targetFrame = m_pageAgent->findFrameWithSecurityOrigin(securityOrigin);
if (!targetFrame) {
errorString = "Frame not found for the given security origin"_s;
return nullptr;
}
if (!isLocalStorage)
return m_pageAgent->page().sessionStorage()->storageArea(targetFrame->document()->securityOrigin().data());
return m_pageAgent->page().storageNamespaceProvider().localStorageArea(*targetFrame->document());
}
}