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 39namespace WebCore { 40 41class AbstractDatabase; 42class OriginUsageRecord; 43 44class OriginQuotaManager { 45 WTF_MAKE_NONCOPYABLE(OriginQuotaManager); WTF_MAKE_FAST_ALLOCATED; 46public: 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 62private: 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