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 "OriginUsageRecord.h" 30 31 #if ENABLE(DATABASE) 32 33 #include "SQLiteFileSystem.h" 34 35 namespace WebCore { 36 37 OriginUsageRecord::OriginUsageRecord() 38 : m_cachedDiskUsageIsValid(false) 39 { 40 } 41 42 void OriginUsageRecord::addDatabase(const String& identifier, const String& fullPath) 43 { 44 ASSERT(!m_databaseMap.contains(identifier)); 45 ASSERT_ARG(identifier, identifier.impl()->hasOneRef() || identifier.isEmpty()); 46 ASSERT_ARG(fullPath, fullPath.impl()->hasOneRef() || fullPath.isEmpty()); 47 48 m_databaseMap.set(identifier, DatabaseEntry(fullPath)); 49 m_unknownSet.add(identifier); 50 51 m_cachedDiskUsageIsValid = false; 52 } 53 54 void OriginUsageRecord::removeDatabase(const String& identifier) 55 { 56 ASSERT(m_databaseMap.contains(identifier)); 57 58 m_databaseMap.remove(identifier); 59 m_unknownSet.remove(identifier); 60 m_cachedDiskUsageIsValid = false; 61 } 62 63 void OriginUsageRecord::markDatabase(const String& identifier) 64 { 65 ASSERT(m_databaseMap.contains(identifier)); 66 ASSERT_ARG(identifier, identifier.impl()->hasOneRef() || identifier.isEmpty()); 67 68 m_unknownSet.add(identifier); 69 m_cachedDiskUsageIsValid = false; 70 } 71 72 unsigned long long OriginUsageRecord::diskUsage() 73 { 74 // Use the last cached usage value if we have it. 75 if (m_cachedDiskUsageIsValid) 76 return m_cachedDiskUsage; 77 78 // stat() for the sizes known to be dirty. 79 HashSet<String>::iterator iUnknown = m_unknownSet.begin(); 80 HashSet<String>::iterator endUnknown = m_unknownSet.end(); 81 for (; iUnknown != endUnknown; ++iUnknown) { 82 const String& path = m_databaseMap.get(*iUnknown).filename; 83 ASSERT(!path.isEmpty()); 84 85 // When we can't determine the file size, we'll just have to assume the file is missing/inaccessible. 86 long long size = SQLiteFileSystem::getDatabaseFileSize(path); 87 m_databaseMap.set(*iUnknown, DatabaseEntry(path, size)); 88 } 89 m_unknownSet.clear(); 90 91 // Recalculate the cached usage value. 92 m_cachedDiskUsage = 0; 93 HashMap<String, DatabaseEntry>::iterator iDatabase = m_databaseMap.begin(); 94 HashMap<String, DatabaseEntry>::iterator endDatabase = m_databaseMap.end(); 95 for (; iDatabase != endDatabase; ++iDatabase) 96 m_cachedDiskUsage += iDatabase->second.size; 97 98 m_cachedDiskUsageIsValid = true; 99 return m_cachedDiskUsage; 100 } 101 102 } 103 104 #endif 105
