StorageAreaImpl.cpp [plain text]
#include "config.h"
#include "StorageAreaImpl.h"
#if ENABLE(DOM_STORAGE)
#include "DOMWindow.h"
#include "EventNames.h"
#include "ExceptionCode.h"
#include "Frame.h"
#include "Page.h"
#include "PageGroup.h"
#include "SecurityOrigin.h"
#include "Settings.h"
#include "StorageEvent.h"
#include "StorageAreaSync.h"
#include "StorageMap.h"
#include "StorageSyncManager.h"
namespace WebCore {
PassRefPtr<StorageArea> StorageAreaImpl::create(StorageType storageType, SecurityOrigin* origin, PassRefPtr<StorageSyncManager> syncManager)
{
return adoptRef(new StorageAreaImpl(storageType, origin, syncManager));
}
StorageAreaImpl::~StorageAreaImpl()
{
}
StorageAreaImpl::StorageAreaImpl(StorageType storageType, SecurityOrigin* origin, PassRefPtr<StorageSyncManager> syncManager)
: m_storageType(storageType)
, m_securityOrigin(origin)
, m_storageMap(StorageMap::create())
, m_storageSyncManager(syncManager)
#ifndef NDEBUG
, m_isShutdown(false)
#endif
{
ASSERT(m_securityOrigin);
ASSERT(m_storageMap);
if (m_storageSyncManager) {
m_storageAreaSync = StorageAreaSync::create(m_storageSyncManager, this);
ASSERT(m_storageAreaSync);
}
}
PassRefPtr<StorageArea> StorageAreaImpl::copy(SecurityOrigin* origin)
{
ASSERT(!m_isShutdown);
return adoptRef(new StorageAreaImpl(origin, this));
}
StorageAreaImpl::StorageAreaImpl(SecurityOrigin* origin, StorageAreaImpl* area)
: m_storageType(area->m_storageType)
, m_securityOrigin(origin)
, m_storageMap(area->m_storageMap)
, m_storageSyncManager(area->m_storageSyncManager)
#ifndef NDEBUG
, m_isShutdown(area->m_isShutdown)
#endif
{
ASSERT(m_securityOrigin);
ASSERT(m_storageMap);
ASSERT(!m_isShutdown);
}
unsigned StorageAreaImpl::length() const
{
ASSERT(!m_isShutdown);
return m_storageMap->length();
}
String StorageAreaImpl::key(unsigned index, ExceptionCode& ec) const
{
ASSERT(!m_isShutdown);
blockUntilImportComplete();
String key;
if (!m_storageMap->key(index, key)) {
ec = INDEX_SIZE_ERR;
return String();
}
return key;
}
String StorageAreaImpl::getItem(const String& key) const
{
ASSERT(!m_isShutdown);
blockUntilImportComplete();
return m_storageMap->getItem(key);
}
void StorageAreaImpl::setItem(const String& key, const String& value, ExceptionCode& ec, Frame* frame)
{
ASSERT(!m_isShutdown);
ASSERT(!value.isNull());
blockUntilImportComplete();
if (frame->page()->settings()->privateBrowsingEnabled()) {
ec = QUOTA_EXCEEDED_ERR;
return;
}
String oldValue;
RefPtr<StorageMap> newMap = m_storageMap->setItem(key, value, oldValue);
if (newMap)
m_storageMap = newMap.release();
if (oldValue != value) {
if (m_storageAreaSync)
m_storageAreaSync->scheduleItemForSync(key, value);
dispatchStorageEvent(key, oldValue, value, frame);
}
}
void StorageAreaImpl::removeItem(const String& key, Frame* frame)
{
ASSERT(!m_isShutdown);
blockUntilImportComplete();
if (frame->page()->settings()->privateBrowsingEnabled())
return;
String oldValue;
RefPtr<StorageMap> newMap = m_storageMap->removeItem(key, oldValue);
if (newMap)
m_storageMap = newMap.release();
if (!oldValue.isNull()) {
if (m_storageAreaSync)
m_storageAreaSync->scheduleItemForSync(key, String());
dispatchStorageEvent(key, oldValue, String(), frame);
}
}
void StorageAreaImpl::clear(Frame* frame)
{
ASSERT(!m_isShutdown);
blockUntilImportComplete();
if (frame->page()->settings()->privateBrowsingEnabled())
return;
m_storageMap = StorageMap::create();
if (m_storageAreaSync)
m_storageAreaSync->scheduleClear();
dispatchStorageEvent(String(), String(), String(), frame);
}
bool StorageAreaImpl::contains(const String& key) const
{
ASSERT(!m_isShutdown);
blockUntilImportComplete();
return m_storageMap->contains(key);
}
void StorageAreaImpl::importItem(const String& key, const String& value)
{
ASSERT(!m_isShutdown);
m_storageMap->importItem(key, value);
}
SecurityOrigin* StorageAreaImpl::securityOrigin()
{
return m_securityOrigin.get();
}
void StorageAreaImpl::close()
{
if (m_storageAreaSync)
m_storageAreaSync->scheduleFinalSync();
#ifndef NDEBUG
m_isShutdown = true;
#endif
}
void StorageAreaImpl::blockUntilImportComplete() const
{
if (m_storageAreaSync)
m_storageAreaSync->blockUntilImportComplete();
}
void StorageAreaImpl::dispatchStorageEvent(const String& key, const String& oldValue, const String& newValue, Frame* sourceFrame)
{
Vector<RefPtr<Frame> > frames;
Page* page = sourceFrame->page();
if (!page)
return;
if (m_storageType == SessionStorage) {
for (Frame* frame = page->mainFrame(); frame; frame = frame->tree()->traverseNext()) {
if (frame->document()->securityOrigin()->equal(securityOrigin()))
frames.append(frame);
}
for (unsigned i = 0; i < frames.size(); ++i)
frames[i]->document()->dispatchWindowEvent(StorageEvent::create(eventNames().storageEvent, key, oldValue, newValue, sourceFrame->document()->documentURI(), sourceFrame->domWindow(), frames[i]->domWindow()->sessionStorage()));
} else {
const HashSet<Page*>& pages = page->group().pages();
HashSet<Page*>::const_iterator end = pages.end();
for (HashSet<Page*>::const_iterator it = pages.begin(); it != end; ++it) {
for (Frame* frame = (*it)->mainFrame(); frame; frame = frame->tree()->traverseNext()) {
if (frame->document()->securityOrigin()->equal(securityOrigin()))
frames.append(frame);
}
}
for (unsigned i = 0; i < frames.size(); ++i)
frames[i]->document()->dispatchWindowEvent(StorageEvent::create(eventNames().storageEvent, key, oldValue, newValue, sourceFrame->document()->documentURI(), sourceFrame->domWindow(), frames[i]->domWindow()->localStorage()));
}
}
}
#endif // ENABLE(DOM_STORAGE)