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 29 #ifndef OriginQuotaManager_h 30 #define OriginQuotaManager_h 31 32 #if ENABLE(DATABASE) 33 34 #include "SecurityOriginHash.h" 35 #include <wtf/HashMap.h> 36 #include <wtf/Threading.h> 37 #include <wtf/text/StringHash.h> 38 39 namespace WebCore { 40 41 class AbstractDatabase; 42 class OriginUsageRecord; 43 44 class OriginQuotaManager { 45 WTF_MAKE_NONCOPYABLE(OriginQuotaManager); WTF_MAKE_FAST_ALLOCATED; 46 public: 47 OriginQuotaManager(); 48 49 bool tryLock(); 50 void lock(); 51 void unlock(); 52 53 void trackOrigin(PassRefPtr<SecurityOrigin>); 54 bool tracksOrigin(SecurityOrigin*) const; 55 void addDatabase(SecurityOrigin*, const String& databaseIdentifier, const String& fullPath); 56 void removeDatabase(SecurityOrigin*, const String& databaseIdentifier); 57 void removeOrigin(SecurityOrigin*); 58 59 void markDatabase(AbstractDatabase*); // Mark dirtiness of a specific database. 60 unsigned long long diskUsage(SecurityOrigin*) const; 61 62 private: 63 mutable Mutex m_usageRecordGuard; 64 #ifndef NDEBUG 65 bool m_usageRecordGuardLocked; 66 #endif 67 68 typedef HashMap<RefPtr<SecurityOrigin>, OriginUsageRecord*, SecurityOriginHash> OriginUsageMap; 69 OriginUsageMap m_usageMap; 70 }; 71 72 } // namespace WebCore 73 74 #endif // ENABLE(DATABASE) 75 76 #endif // OriginQuotaManager_h 77