StorageNamespaceImpl.cpp [plain text]
#include "StorageNamespaceImpl.h"
#include "StorageAreaImpl.h"
#include "StorageSyncManager.h"
#include "StorageTracker.h"
#include <WebCore/StorageMap.h>
#include <wtf/MainThread.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/text/StringHash.h>
namespace WebCore {
static HashMap<String, StorageNamespaceImpl*>& localStorageNamespaceMap()
{
static NeverDestroyed<HashMap<String, StorageNamespaceImpl*>> localStorageNamespaceMap;
return localStorageNamespaceMap;
}
RefPtr<StorageNamespaceImpl> StorageNamespaceImpl::createSessionStorageNamespace(unsigned quota)
{
return adoptRef(new StorageNamespaceImpl(SessionStorage, String(), quota));
}
RefPtr<StorageNamespaceImpl> StorageNamespaceImpl::getOrCreateLocalStorageNamespace(const String& databasePath, unsigned quota)
{
ASSERT(!databasePath.isNull());
auto& slot = localStorageNamespaceMap().add(databasePath, nullptr).iterator->value;
if (slot)
return slot;
RefPtr<StorageNamespaceImpl> storageNamespace = adoptRef(new StorageNamespaceImpl(LocalStorage, databasePath, quota));
slot = storageNamespace.get();
return storageNamespace;
}
StorageNamespaceImpl::StorageNamespaceImpl(StorageType storageType, const String& path, unsigned quota)
: m_storageType(storageType)
, m_path(path.isolatedCopy())
, m_syncManager(0)
, m_quota(quota)
, m_isShutdown(false)
{
if (m_storageType == LocalStorage && !m_path.isEmpty())
m_syncManager = StorageSyncManager::create(m_path);
}
StorageNamespaceImpl::~StorageNamespaceImpl()
{
ASSERT(isMainThread());
if (m_storageType == LocalStorage) {
ASSERT(localStorageNamespaceMap().get(m_path) == this);
localStorageNamespaceMap().remove(m_path);
}
if (!m_isShutdown)
close();
}
RefPtr<StorageNamespace> StorageNamespaceImpl::copy(Page*)
{
ASSERT(isMainThread());
ASSERT(!m_isShutdown);
ASSERT(m_storageType == SessionStorage);
RefPtr<StorageNamespaceImpl> newNamespace = adoptRef(new StorageNamespaceImpl(m_storageType, m_path, m_quota));
StorageAreaMap::iterator end = m_storageAreaMap.end();
for (StorageAreaMap::iterator i = m_storageAreaMap.begin(); i != end; ++i)
newNamespace->m_storageAreaMap.set(i->key, i->value->copy());
return newNamespace;
}
RefPtr<StorageArea> StorageNamespaceImpl::storageArea(RefPtr<SecurityOrigin>&& origin)
{
ASSERT(isMainThread());
ASSERT(!m_isShutdown);
RefPtr<StorageAreaImpl> storageArea;
if ((storageArea = m_storageAreaMap.get(origin)))
return storageArea;
storageArea = StorageAreaImpl::create(m_storageType, origin.copyRef(), m_syncManager, m_quota);
m_storageAreaMap.set(WTFMove(origin), storageArea);
return storageArea;
}
void StorageNamespaceImpl::close()
{
ASSERT(isMainThread());
if (m_isShutdown)
return;
if (m_storageType == SessionStorage) {
ASSERT(!m_syncManager);
return;
}
StorageAreaMap::iterator end = m_storageAreaMap.end();
for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
it->value->close();
if (m_syncManager)
m_syncManager->close();
m_isShutdown = true;
}
void StorageNamespaceImpl::clearOriginForDeletion(SecurityOrigin* origin)
{
ASSERT(isMainThread());
RefPtr<StorageAreaImpl> storageArea = m_storageAreaMap.get(origin);
if (storageArea)
storageArea->clearForOriginDeletion();
}
void StorageNamespaceImpl::clearAllOriginsForDeletion()
{
ASSERT(isMainThread());
StorageAreaMap::iterator end = m_storageAreaMap.end();
for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
it->value->clearForOriginDeletion();
}
void StorageNamespaceImpl::sync()
{
ASSERT(isMainThread());
StorageAreaMap::iterator end = m_storageAreaMap.end();
for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
it->value->sync();
}
void StorageNamespaceImpl::closeIdleLocalStorageDatabases()
{
ASSERT(isMainThread());
StorageAreaMap::iterator end = m_storageAreaMap.end();
for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
it->value->closeDatabaseIfIdle();
}
}