| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_CONTEXT_H_ | 5 #ifndef WEBKIT_DOM_STORAGE_DOM_STORAGE_CONTEXT_H_ |
| 6 #define CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_CONTEXT_H_ | 6 #define WEBKIT_DOM_STORAGE_DOM_STORAGE_CONTEXT_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include <map> | 9 #include <map> |
| 10 #include <set> | |
| 11 | 10 |
| 12 #include "base/file_path.h" | 11 #include "base/basictypes.h" |
| 13 #include "base/gtest_prod_util.h" | |
| 14 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 15 #include "base/string16.h" | |
| 16 #include "base/time.h" | |
| 17 #include "content/common/content_export.h" | |
| 18 | 13 |
| 19 class DOMStorageArea; | 14 |
| 20 class DOMStorageMessageFilter; | 15 class FilePath; |
| 21 class DOMStorageNamespace; | 16 class GURL; |
| 22 class WebKitContext; | 17 |
| 18 namespace base { |
| 19 class Time; |
| 20 } |
| 23 | 21 |
| 24 namespace quota { | 22 namespace quota { |
| 25 class SpecialStoragePolicy; | 23 class SpecialStoragePolicy; |
| 26 } | 24 } |
| 27 | 25 |
| 28 // This is owned by WebKitContext and is all the dom storage information that's | 26 namespace dom_storage { |
| 29 // shared by all the DOMStorageMessageFilters that share the same browser | 27 |
| 30 // context. The specifics of responsibilities are fairly well documented here | 28 class DOMStorageArea; |
| 31 // and in StorageNamespace and StorageArea. Everything is only to be accessed | 29 class DOMStorageNamespace; |
| 32 // on the WebKit thread unless noted otherwise. | 30 class DOMStorageSession; |
| 33 // | 31 |
| 34 // NOTE: Virtual methods facilitate mocking functions for testing. | 32 // One instance is allocated in the main process for each profile, |
| 35 class CONTENT_EXPORT DOMStorageContext { | 33 // should be called only on the background thread. |
| 34 class DOMStorageContext |
| 35 : public base::RefCountedThreadSafe<DOMStorageContext> { |
| 36 public: | 36 public: |
| 37 DOMStorageContext(WebKitContext* webkit_context, | 37 DOMStorageContext(const FilePath& directory, // empty for incognito profiles |
| 38 quota::SpecialStoragePolicy* special_storage_policy); | 38 quota::SpecialStoragePolicy* special_storage_policy); |
| 39 virtual ~DOMStorageContext(); | |
| 40 | 39 |
| 41 // Invalid storage id. No storage session will ever report this value. | 40 DOMStorageNamespace* GetStorageNamespace(int64 namespace_id); |
| 42 // Used in DOMStorageMessageFilter::OnStorageAreaId when coping with | |
| 43 // interactions with non-existent storage sessions. | |
| 44 static const int64 kInvalidStorageId = -1; | |
| 45 | 41 |
| 46 // Allocate a new storage area id. Only call on the WebKit thread. | 42 void NotifyEventListeners(); |
| 47 int64 AllocateStorageAreaId(); | |
| 48 | |
| 49 // Allocate a new session storage id. Only call on the UI or IO thread. | |
| 50 int64 AllocateSessionStorageNamespaceId(); | |
| 51 | |
| 52 // Clones a session storage namespace and returns the cloned namespaces' id. | |
| 53 // Only call on the IO thread. | |
| 54 int64 CloneSessionStorage(int64 original_id); | |
| 55 | |
| 56 // Various storage area methods. The storage area is owned by one of the | |
| 57 // namespaces that's owned by this class. | |
| 58 void RegisterStorageArea(DOMStorageArea* storage_area); | |
| 59 void UnregisterStorageArea(DOMStorageArea* storage_area); | |
| 60 DOMStorageArea* GetStorageArea(int64 id); | |
| 61 | |
| 62 // Called on WebKit thread when a session storage namespace can be deleted. | |
| 63 void DeleteSessionStorageNamespace(int64 namespace_id); | |
| 64 | |
| 65 // Get a namespace from an id. What's returned is owned by this class. If | |
| 66 // allocation_allowed is true, then this function will create the storage | |
| 67 // namespace if it hasn't been already. | |
| 68 DOMStorageNamespace* GetStorageNamespace(int64 id, bool allocation_allowed); | |
| 69 | |
| 70 // Sometimes an event from one DOM storage message filter requires | |
| 71 // communication to all of them. | |
| 72 typedef std::set<DOMStorageMessageFilter*> MessageFilterSet; | |
| 73 void RegisterMessageFilter(DOMStorageMessageFilter* message_filter); | |
| 74 void UnregisterMessageFilter(DOMStorageMessageFilter* MessageFilter); | |
| 75 const MessageFilterSet* GetMessageFilterSet() const; | |
| 76 | 43 |
| 77 // Tells storage namespaces to purge any memory they do not need. | 44 // Tells storage namespaces to purge any memory they do not need. |
| 78 virtual void PurgeMemory(); | 45 void PurgeMemory(); |
| 79 | 46 |
| 80 // Delete any local storage files that have been touched since the cutoff | 47 // Delete any local storage files that have been touched since the cutoff |
| 81 // date that's supplied. Protected origins, per the SpecialStoragePolicy, | 48 // date that's supplied. Protected origins, per the SpecialStoragePolicy, |
| 82 // are not deleted by this method. | 49 // are not deleted by this method. |
| 83 void DeleteDataModifiedSince(const base::Time& cutoff); | 50 void DeleteDataModifiedSince(const base::Time& cutoff); |
| 84 | 51 |
| 85 // Deletes a single local storage file. | 52 // Delete any local storage files which are allowed to be stored only until |
| 86 void DeleteLocalStorageFile(const FilePath& file_path); | 53 // the end of the session. Protected origins, per the SpecialStoragePolicy, |
| 54 // are not deleted by this method. |
| 55 void DeleteSessionOnlyData(); |
| 87 | 56 |
| 88 // Deletes the local storage file for the given origin. | 57 // Deletes the local storage file for the given origin. |
| 89 void DeleteLocalStorageForOrigin(const string16& origin_id); | 58 void DeleteLocalStorageForOrigin(const GURL& origin); |
| 90 | 59 |
| 91 // Deletes all local storage files. | 60 // Deletes all local storage files. |
| 92 void DeleteAllLocalStorageFiles(); | 61 void DeleteAllLocalStorageFiles(); |
| 93 | 62 |
| 94 // The local storage directory. | |
| 95 static const FilePath::CharType kLocalStorageDirectory[]; | |
| 96 | |
| 97 // The local storage file extension. | |
| 98 static const FilePath::CharType kLocalStorageExtension[]; | |
| 99 | |
| 100 // Get the file name of the local storage file for the given origin. | |
| 101 FilePath GetLocalStorageFilePath(const string16& origin_id) const; | |
| 102 | |
| 103 void set_clear_local_state_on_exit_(bool clear_local_state) { | 63 void set_clear_local_state_on_exit_(bool clear_local_state) { |
| 104 clear_local_state_on_exit_ = clear_local_state; | 64 clear_local_state_on_exit_ = clear_local_state; |
| 105 } | 65 } |
| 106 | 66 |
| 107 // Disables the exit-time deletion for all data (also session-only data). | 67 private: |
| 108 void SaveSessionState() { | 68 friend class DOMStorageSession; |
| 109 save_session_state_ = true; | 69 friend class base::RefCountedThreadSafe<DOMStorageContext>; |
| 110 } | |
| 111 | 70 |
| 112 void set_data_path_for_testing(const FilePath& data_path) { | 71 ~DOMStorageContext(); |
| 113 data_path_ = data_path; | |
| 114 } | |
| 115 | 72 |
| 116 private: | 73 // These three methods may be called on any thread, tasks |
| 74 // are scheduled on the background thread to perform the work. |
| 75 int64 CreateSessionStorageNamespace(); |
| 76 void DeleteSessionStorageNamespace(int64 namespace_id); |
| 77 int64 CloneSessionStorageNamespace(int64 existing_id); |
| 117 | 78 |
| 118 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SessionOnly); | |
| 119 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SaveSessionState); | |
| 120 | |
| 121 // Get the local storage instance. The object is owned by this class. | |
| 122 DOMStorageNamespace* CreateLocalStorage(); | |
| 123 | |
| 124 // Get a new session storage namespace. The object is owned by this class. | |
| 125 DOMStorageNamespace* CreateSessionStorage(int64 namespace_id); | |
| 126 | |
| 127 // Used internally to register storage namespaces we create. | |
| 128 void RegisterStorageNamespace(DOMStorageNamespace* storage_namespace); | |
| 129 | |
| 130 // The WebKit thread half of CloneSessionStorage above. Static because | |
| 131 // DOMStorageContext isn't ref counted thus we can't use a runnable method. | |
| 132 // That said, we know this is safe because this class is destroyed on the | |
| 133 // WebKit thread, so there's no way it could be destroyed before this is run. | |
| 134 static void CompleteCloningSessionStorage(DOMStorageContext* context, | |
| 135 int64 existing_id, int64 clone_id); | |
| 136 | |
| 137 // The last used storage_area_id and storage_namespace_id's. For the storage | |
| 138 // namespaces, IDs allocated on the UI thread are positive and count up while | |
| 139 // IDs allocated on the IO thread are negative and count down. This allows us | |
| 140 // to allocate unique IDs on both without any locking. All storage area ids | |
| 141 // are allocated on the WebKit thread. | |
| 142 int64 last_storage_area_id_; | |
| 143 int64 last_session_storage_namespace_id_on_ui_thread_; | |
| 144 int64 last_session_storage_namespace_id_on_io_thread_; | |
| 145 | |
| 146 // True if the destructor should delete its files. | |
| 147 bool clear_local_state_on_exit_; | |
| 148 | |
| 149 // If true, nothing (not even session-only data) should be deleted on exit. | |
| 150 bool save_session_state_; | |
| 151 | |
| 152 // Path where the browser context data is stored. | |
| 153 // TODO(pastarmovj): Keep in mind that unlike indexed db data_path_ variable | |
| 154 // this one still has to point to the upper level dir because of the | |
| 155 // MigrateLocalStorageDirectory function. Once this function disappears we can | |
| 156 // make it point directly to the dom storage path. | |
| 157 FilePath data_path_; | |
| 158 | |
| 159 // All the DOMStorageMessageFilters that are attached to us. ONLY USE ON THE | |
| 160 // IO THREAD! | |
| 161 MessageFilterSet message_filter_set_; | |
| 162 | |
| 163 // Maps ids to StorageAreas. We do NOT own these objects. StorageNamespace | |
| 164 // (which does own them) will notify us when we should remove the entries. | |
| 165 typedef std::map<int64, DOMStorageArea*> StorageAreaMap; | |
| 166 StorageAreaMap storage_area_map_; | |
| 167 | |
| 168 // Maps ids to StorageNamespaces. We own these objects. | |
| 169 typedef std::map<int64, DOMStorageNamespace*> StorageNamespaceMap; | 79 typedef std::map<int64, DOMStorageNamespace*> StorageNamespaceMap; |
| 170 StorageNamespaceMap storage_namespace_map_; | 80 StorageNamespaceMap storage_namespace_map_; |
| 171 | 81 bool clear_local_state_on_exit_; |
| 172 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_; | |
| 173 | |
| 174 DISALLOW_IMPLICIT_CONSTRUCTORS(DOMStorageContext); | |
| 175 }; | 82 }; |
| 176 | 83 |
| 177 #endif // CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_CONTEXT_H_ | 84 } // namespace dom_storage |
| 85 |
| 86 #endif // WEBKIT_DOM_STORAGE_DOM_STORAGE_CONTEXT_H_ |
| OLD | NEW |