1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_ 6 #define CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_ 7 8 #include <map> 9 #include <set> 10 #include <vector> 11 12 #include "base/compiler_specific.h" 13 #include "base/files/file_path.h" 14 #include "base/gtest_prod_util.h" 15 #include "base/memory/scoped_ptr.h" 16 #include "content/browser/browser_main_loop.h" 17 #include "content/browser/indexed_db/indexed_db_factory.h" 18 #include "content/public/browser/indexed_db_context.h" 19 #include "url/gurl.h" 20 #include "webkit/common/quota/quota_types.h" 21 22 class GURL; 23 24 namespace base { 25 class ListValue; 26 class FilePath; 27 class SequencedTaskRunner; 28 } 29 30 namespace quota { 31 class QuotaManagerProxy; 32 class SpecialStoragePolicy; 33 } 34 35 namespace content { 36 37 class IndexedDBConnection; 38 39 class CONTENT_EXPORT IndexedDBContextImpl 40 : NON_EXPORTED_BASE(public IndexedDBContext) { 41 public: 42 // If |data_path| is empty, nothing will be saved to disk. 43 IndexedDBContextImpl(const base::FilePath& data_path, 44 quota::SpecialStoragePolicy* special_storage_policy, 45 quota::QuotaManagerProxy* quota_manager_proxy, 46 base::SequencedTaskRunner* task_runner); 47 48 IndexedDBFactory* GetIDBFactory(); 49 50 // The indexed db directory. 51 static const base::FilePath::CharType kIndexedDBDirectory[]; 52 53 // Disables the exit-time deletion of session-only data. 54 void SetForceKeepSessionState() { force_keep_session_state_ = true; } 55 56 // IndexedDBContext implementation: 57 virtual base::TaskRunner* TaskRunner() const OVERRIDE; 58 virtual std::vector<IndexedDBInfo> GetAllOriginsInfo() OVERRIDE; 59 virtual int64 GetOriginDiskUsage(const GURL& origin_url) OVERRIDE; 60 virtual void DeleteForOrigin(const GURL& origin_url) OVERRIDE; 61 virtual base::FilePath GetFilePathForTesting( 62 const std::string& origin_id) const OVERRIDE; 63 virtual void SetTaskRunnerForTesting(base::SequencedTaskRunner* task_runner) 64 OVERRIDE; 65 66 // Methods called by IndexedDBDispatcherHost for quota support. 67 void ConnectionOpened(const GURL& origin_url, IndexedDBConnection* db); 68 void ConnectionClosed(const GURL& origin_url, IndexedDBConnection* db); 69 void TransactionComplete(const GURL& origin_url); 70 bool WouldBeOverQuota(const GURL& origin_url, int64 additional_bytes); 71 bool IsOverQuota(const GURL& origin_url); 72 73 quota::QuotaManagerProxy* quota_manager_proxy(); 74 75 std::vector<GURL> GetAllOrigins(); 76 base::Time GetOriginLastModified(const GURL& origin_url); 77 base::ListValue* GetAllOriginsDetails(); 78 // ForceClose takes a value rather than a reference since it may release the 79 // owning object. 80 void ForceClose(const GURL origin_url); 81 base::FilePath GetFilePath(const GURL& origin_url) const; 82 base::FilePath data_path() const { return data_path_; } 83 bool IsInOriginSet(const GURL& origin_url) { 84 std::set<GURL>* set = GetOriginSet(); 85 return set->find(origin_url) != set->end(); 86 } 87 size_t GetConnectionCount(const GURL& origin_url); 88 89 // For unit tests allow to override the |data_path_|. 90 void set_data_path_for_testing(const base::FilePath& data_path) { 91 data_path_ = data_path; 92 } 93 94 protected: 95 virtual ~IndexedDBContextImpl(); 96 97 private: 98 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ClearLocalState); 99 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ClearSessionOnlyDatabases); 100 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, SetForceKeepSessionState); 101 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ForceCloseOpenDatabasesOnDelete); 102 friend class IndexedDBQuotaClientTest; 103 104 typedef std::map<GURL, int64> OriginToSizeMap; 105 class IndexedDBGetUsageAndQuotaCallback; 106 107 base::FilePath GetIndexedDBFilePath(const std::string& origin_id) const; 108 int64 ReadUsageFromDisk(const GURL& origin_url) const; 109 void EnsureDiskUsageCacheInitialized(const GURL& origin_url); 110 void QueryDiskAndUpdateQuotaUsage(const GURL& origin_url); 111 void GotUsageAndQuota(const GURL& origin_url, 112 quota::QuotaStatusCode, 113 int64 usage, 114 int64 quota); 115 void GotUpdatedQuota(const GURL& origin_url, int64 usage, int64 quota); 116 void QueryAvailableQuota(const GURL& origin_url); 117 118 std::set<GURL>* GetOriginSet(); 119 bool AddToOriginSet(const GURL& origin_url) { 120 return GetOriginSet()->insert(origin_url).second; 121 } 122 void RemoveFromOriginSet(const GURL& origin_url) { 123 GetOriginSet()->erase(origin_url); 124 } 125 126 // Only for testing. 127 void ResetCaches(); 128 129 scoped_refptr<IndexedDBFactory> factory_; 130 base::FilePath data_path_; 131 // If true, nothing (not even session-only data) should be deleted on exit. 132 bool force_keep_session_state_; 133 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_; 134 scoped_refptr<quota::QuotaManagerProxy> quota_manager_proxy_; 135 base::SequencedTaskRunner* task_runner_; 136 scoped_ptr<std::set<GURL> > origin_set_; 137 OriginToSizeMap origin_size_map_; 138 OriginToSizeMap space_available_map_; 139 typedef std::set<IndexedDBConnection*> ConnectionSet; 140 std::map<GURL, ConnectionSet> connections_; 141 142 DISALLOW_COPY_AND_ASSIGN(IndexedDBContextImpl); 143 }; 144 145 } // namespace content 146 147 #endif // CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_ 148
