1/* 2 * Copyright (C) 2008 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 14 * its contributors may be used to endorse or promote products derived 15 * from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28#include "config.h" 29#include "OriginQuotaManager.h" 30 31#if ENABLE(DATABASE) 32 33#include "Database.h" 34#include "OriginUsageRecord.h" 35 36namespace WebCore { 37 38OriginQuotaManager::OriginQuotaManager() 39#ifndef NDEBUG 40 : m_usageRecordGuardLocked(false) 41#endif 42{ 43} 44 45void OriginQuotaManager::lock() 46{ 47 m_usageRecordGuard.lock(); 48#ifndef NDEBUG 49 m_usageRecordGuardLocked = true; 50#endif 51} 52 53void OriginQuotaManager::unlock() 54{ 55#ifndef NDEBUG 56 m_usageRecordGuardLocked = false; 57#endif 58 m_usageRecordGuard.unlock(); 59} 60 61void OriginQuotaManager::trackOrigin(PassRefPtr<SecurityOrigin> origin) 62{ 63 ASSERT(m_usageRecordGuardLocked); 64 ASSERT(!m_usageMap.contains(origin.get())); 65 66 m_usageMap.set(origin, new OriginUsageRecord); 67} 68 69bool OriginQuotaManager::tracksOrigin(SecurityOrigin* origin) const 70{ 71 ASSERT(m_usageRecordGuardLocked); 72 return m_usageMap.contains(origin); 73} 74 75void OriginQuotaManager::addDatabase(SecurityOrigin* origin, const String& databaseIdentifier, const String& fullPath) 76{ 77 ASSERT(m_usageRecordGuardLocked); 78 79 OriginUsageRecord* usageRecord = m_usageMap.get(origin); 80 ASSERT(usageRecord); 81 82 usageRecord->addDatabase(databaseIdentifier.threadsafeCopy(), fullPath.threadsafeCopy()); 83} 84 85void OriginQuotaManager::removeDatabase(SecurityOrigin* origin, const String& databaseIdentifier) 86{ 87 ASSERT(m_usageRecordGuardLocked); 88 89 if (OriginUsageRecord* usageRecord = m_usageMap.get(origin)) 90 usageRecord->removeDatabase(databaseIdentifier); 91} 92 93void OriginQuotaManager::removeOrigin(SecurityOrigin* origin) 94{ 95 ASSERT(m_usageRecordGuardLocked); 96 97 if (OriginUsageRecord* usageRecord = m_usageMap.get(origin)) { 98 m_usageMap.remove(origin); 99 delete usageRecord; 100 } 101} 102 103void OriginQuotaManager::markDatabase(Database* database) 104{ 105 ASSERT(database); 106 ASSERT(m_usageRecordGuardLocked); 107 OriginUsageRecord* usageRecord = m_usageMap.get(database->securityOrigin()); 108 ASSERT(usageRecord); 109 110 usageRecord->markDatabase(database->stringIdentifier()); 111} 112 113unsigned long long OriginQuotaManager::diskUsage(SecurityOrigin* origin) const 114{ 115 ASSERT(m_usageRecordGuardLocked); 116 117 OriginUsageRecord* usageRecord = m_usageMap.get(origin); 118 ASSERT(usageRecord); 119 120 return usageRecord->diskUsage(); 121} 122 123} 124 125#endif // ENABLE(DATABASE) 126