Home | History | Annotate | Download | only in storage 
      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 "AbstractDatabase.h"
     34 #include "OriginUsageRecord.h"
     35 
     36 namespace WebCore {
     37 
     38 OriginQuotaManager::OriginQuotaManager()
     39 #ifndef NDEBUG
     40     : m_usageRecordGuardLocked(false)
     41 #endif
     42 {
     43 }
     44 
     45 bool OriginQuotaManager::tryLock()
     46 {
     47     bool locked = m_usageRecordGuard.tryLock();
     48 #ifndef NDEBUG
     49     if (locked)
     50         m_usageRecordGuardLocked = true;
     51     else
     52         ASSERT(m_usageRecordGuardLocked);
     53 #endif
     54     return locked;
     55 }
     56 
     57 void OriginQuotaManager::lock()
     58 {
     59     m_usageRecordGuard.lock();
     60 #ifndef NDEBUG
     61     m_usageRecordGuardLocked = true;
     62 #endif
     63 }
     64 
     65 void OriginQuotaManager::unlock()
     66 {
     67 #ifndef NDEBUG
     68     m_usageRecordGuardLocked = false;
     69 #endif
     70     m_usageRecordGuard.unlock();
     71 }
     72 
     73 void OriginQuotaManager::trackOrigin(PassRefPtr<SecurityOrigin> origin)
     74 {
     75     ASSERT(m_usageRecordGuardLocked);
     76     ASSERT(!m_usageMap.contains(origin.get()));
     77 
     78     m_usageMap.set(origin->threadsafeCopy(), new OriginUsageRecord);
     79 }
     80 
     81 bool OriginQuotaManager::tracksOrigin(SecurityOrigin* origin) const
     82 {
     83     ASSERT(m_usageRecordGuardLocked);
     84     return m_usageMap.contains(origin);
     85 }
     86 
     87 void OriginQuotaManager::addDatabase(SecurityOrigin* origin, const String& databaseIdentifier, const String& fullPath)
     88 {
     89     ASSERT(m_usageRecordGuardLocked);
     90 
     91     OriginUsageRecord* usageRecord = m_usageMap.get(origin);
     92     ASSERT(usageRecord);
     93 
     94     usageRecord->addDatabase(databaseIdentifier.threadsafeCopy(), fullPath.threadsafeCopy());
     95 }
     96 
     97 void OriginQuotaManager::removeDatabase(SecurityOrigin* origin, const String& databaseIdentifier)
     98 {
     99     ASSERT(m_usageRecordGuardLocked);
    100 
    101     if (OriginUsageRecord* usageRecord = m_usageMap.get(origin))
    102         usageRecord->removeDatabase(databaseIdentifier);
    103 }
    104 
    105 void OriginQuotaManager::removeOrigin(SecurityOrigin* origin)
    106 {
    107     ASSERT(m_usageRecordGuardLocked);
    108 
    109     if (OriginUsageRecord* usageRecord = m_usageMap.get(origin)) {
    110         m_usageMap.remove(origin);
    111         delete usageRecord;
    112     }
    113 }
    114 
    115 void OriginQuotaManager::markDatabase(AbstractDatabase* database)
    116 {
    117     ASSERT(database);
    118     ASSERT(m_usageRecordGuardLocked);
    119     OriginUsageRecord* usageRecord = m_usageMap.get(database->securityOrigin());
    120     ASSERT(usageRecord);
    121 
    122     usageRecord->markDatabase(database->stringIdentifier());
    123 }
    124 
    125 unsigned long long OriginQuotaManager::diskUsage(SecurityOrigin* origin) const
    126 {
    127     ASSERT(m_usageRecordGuardLocked);
    128 
    129     OriginUsageRecord* usageRecord = m_usageMap.get(origin);
    130     ASSERT(usageRecord);
    131 
    132     return usageRecord->diskUsage();
    133 }
    134 
    135 }
    136 
    137 #endif // ENABLE(DATABASE)
    138