OriginQuotaManager.cpp [plain text]
#include "config.h"
#include "OriginQuotaManager.h"
#if ENABLE(SQL_DATABASE)
#include "AbstractDatabase.h"
#include "OriginUsageRecord.h"
namespace WebCore {
OriginQuotaManager::OriginQuotaManager()
#ifndef NDEBUG
: m_usageRecordGuardLocked(false)
#endif
{
}
bool OriginQuotaManager::tryLock()
{
bool locked = m_usageRecordGuard.tryLock();
#ifndef NDEBUG
if (locked)
m_usageRecordGuardLocked = true;
else
ASSERT(m_usageRecordGuardLocked);
#endif
return locked;
}
void OriginQuotaManager::lock()
{
m_usageRecordGuard.lock();
#ifndef NDEBUG
m_usageRecordGuardLocked = true;
#endif
}
void OriginQuotaManager::unlock()
{
#ifndef NDEBUG
m_usageRecordGuardLocked = false;
#endif
m_usageRecordGuard.unlock();
}
void OriginQuotaManager::trackOrigin(PassRefPtr<SecurityOrigin> origin)
{
ASSERT(m_usageRecordGuardLocked);
ASSERT(!m_usageMap.contains(origin.get()));
m_usageMap.set(origin->isolatedCopy(), new OriginUsageRecord);
}
bool OriginQuotaManager::tracksOrigin(SecurityOrigin* origin) const
{
ASSERT(m_usageRecordGuardLocked);
return m_usageMap.contains(origin);
}
void OriginQuotaManager::addDatabase(SecurityOrigin* origin, const String& databaseIdentifier, const String& fullPath)
{
ASSERT(m_usageRecordGuardLocked);
OriginUsageRecord* usageRecord = m_usageMap.get(origin);
ASSERT(usageRecord);
usageRecord->addDatabase(databaseIdentifier.isolatedCopy(), fullPath.isolatedCopy());
}
void OriginQuotaManager::removeDatabase(SecurityOrigin* origin, const String& databaseIdentifier)
{
ASSERT(m_usageRecordGuardLocked);
if (OriginUsageRecord* usageRecord = m_usageMap.get(origin))
usageRecord->removeDatabase(databaseIdentifier);
}
void OriginQuotaManager::removeOrigin(SecurityOrigin* origin)
{
ASSERT(m_usageRecordGuardLocked);
if (OriginUsageRecord* usageRecord = m_usageMap.get(origin)) {
m_usageMap.remove(origin);
delete usageRecord;
}
}
void OriginQuotaManager::markDatabase(AbstractDatabase* database)
{
ASSERT(database);
ASSERT(m_usageRecordGuardLocked);
OriginUsageRecord* usageRecord = m_usageMap.get(database->securityOrigin());
ASSERT(usageRecord);
usageRecord->markDatabase(database->stringIdentifier());
}
unsigned long long OriginQuotaManager::diskUsage(SecurityOrigin* origin) const
{
ASSERT(m_usageRecordGuardLocked);
OriginUsageRecord* usageRecord = m_usageMap.get(origin);
ASSERT(usageRecord);
return usageRecord->diskUsage();
}
}
#endif // ENABLE(SQL_DATABASE)