| 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 |
| 11 #include "base/atomic_sequence_num.h" |
| 12 #include "base/basictypes.h" |
| 12 #include "base/file_path.h" | 13 #include "base/file_path.h" |
| 13 #include "base/gtest_prod_util.h" | |
| 14 #include "base/memory/ref_counted.h" | 14 #include "base/memory/ref_counted.h" |
| 15 #include "base/string16.h" | 15 #include "base/observer_list.h" |
| 16 #include "base/time.h" | |
| 17 #include "content/common/content_export.h" | |
| 18 | 16 |
| 19 class DOMStorageArea; | 17 class FilePath; |
| 20 class DOMStorageMessageFilter; | 18 class GURL; |
| 21 class DOMStorageNamespace; | 19 class NullableString16; |
| 22 class WebKitContext; | 20 |
| 21 namespace base { |
| 22 class Time; |
| 23 } |
| 23 | 24 |
| 24 namespace quota { | 25 namespace quota { |
| 25 class SpecialStoragePolicy; | 26 class SpecialStoragePolicy; |
| 26 } | 27 } |
| 27 | 28 |
| 28 // This is owned by WebKitContext and is all the dom storage information that's | 29 namespace dom_storage { |
| 29 // shared by all the DOMStorageMessageFilters that share the same browser | 30 |
| 30 // context. The specifics of responsibilities are fairly well documented here | 31 class DomStorageArea; |
| 31 // and in StorageNamespace and StorageArea. Everything is only to be accessed | 32 class DomStorageNamespace; |
| 32 // on the WebKit thread unless noted otherwise. | 33 class DomStorageSession; |
| 34 class DomStorageTaskRunner; |
| 35 |
| 36 // The Context is the root of an object containment hierachy for |
| 37 // Namespaces and Areas related to the owning profile. |
| 38 // One instance is allocated in the main process for each profile, |
| 39 // instance methods should be called serially in the background as |
| 40 // determined by the task_runner. Specifcally not on chrome's non-blocking |
| 41 // IO thread since these methods can result in blocking file io. |
| 33 // | 42 // |
| 34 // NOTE: Virtual methods facilitate mocking functions for testing. | 43 // In general terms, the DomStorage object relationships are... |
| 35 class CONTENT_EXPORT DOMStorageContext { | 44 // Contexts (per-profile) own Namespaces which own Areas which share Maps. |
| 45 // Hosts(per-renderer) refer to Namespaces and Areas open in it's renderer. |
| 46 // Sessions (per-tab) cause the creation and deletion of session Namespaces. |
| 47 // |
| 48 // Session Namespaces are cloned by initially making a shallow copy of |
| 49 // all contained Areas, the shallow copies refer to the same refcounted Map, |
| 50 // and does a deep copy-on-write if needed. |
| 51 // |
| 52 // Classes intended to be used by an embedder are DomStorageContext, |
| 53 // DomStorageHost, and DomStorageSession. The other classes are for |
| 54 // internal consumption. |
| 55 class DomStorageContext |
| 56 : public base::RefCountedThreadSafe<DomStorageContext> { |
| 36 public: | 57 public: |
| 37 DOMStorageContext(WebKitContext* webkit_context, | 58 // An interface for observing LocalStorage events on the |
| 38 quota::SpecialStoragePolicy* special_storage_policy); | 59 // background thread. |
| 39 virtual ~DOMStorageContext(); | 60 class EventObserver { |
| 61 public: |
| 62 virtual void OnDomStorageItemSet( |
| 63 const DomStorageArea* area, |
| 64 const string16& key, |
| 65 const string16& new_value, |
| 66 const NullableString16& old_value, // may be null on initial insert |
| 67 const GURL& page_url) = 0; |
| 68 virtual void OnDomStorageItemRemoved( |
| 69 const DomStorageArea* area, |
| 70 const string16& key, |
| 71 const string16& old_value, |
| 72 const GURL& page_url) = 0; |
| 73 virtual void OnDomStorageAreaCleared( |
| 74 const DomStorageArea* area, |
| 75 const GURL& page_url) = 0; |
| 76 virtual ~EventObserver() {} |
| 77 }; |
| 40 | 78 |
| 41 // Invalid storage id. No storage session will ever report this value. | 79 DomStorageContext(const FilePath& directory, // empty for incognito profiles |
| 42 // Used in DOMStorageMessageFilter::OnStorageAreaId when coping with | 80 quota::SpecialStoragePolicy* special_storage_policy, |
| 43 // interactions with non-existent storage sessions. | 81 DomStorageTaskRunner* task_runner); |
| 44 static const int64 kInvalidStorageId = -1; | 82 DomStorageTaskRunner* task_runner() const { return task_runner_; } |
| 83 DomStorageNamespace* GetStorageNamespace(int64 namespace_id); |
| 45 | 84 |
| 46 // Allocate a new storage area id. Only call on the WebKit thread. | 85 // Methods to add, remove, and notify EventObservers. |
| 47 int64 AllocateStorageAreaId(); | 86 void AddEventObserver(EventObserver* observer); |
| 87 void RemoveEventObserver(EventObserver* observer); |
| 88 void NotifyItemSet( |
| 89 const DomStorageArea* area, |
| 90 const string16& key, |
| 91 const string16& new_value, |
| 92 const NullableString16& old_value, |
| 93 const GURL& page_url); |
| 94 void NotifyItemRemoved( |
| 95 const DomStorageArea* area, |
| 96 const string16& key, |
| 97 const string16& old_value, |
| 98 const GURL& page_url); |
| 99 void NotifyAreaCleared( |
| 100 const DomStorageArea* area, |
| 101 const GURL& page_url); |
| 48 | 102 |
| 49 // Allocate a new session storage id. Only call on the UI or IO thread. | 103 private: |
| 50 int64 AllocateSessionStorageNamespaceId(); | 104 friend class DomStorageSession; |
| 105 friend class base::RefCountedThreadSafe<DomStorageContext>; |
| 106 typedef std::map<int64, scoped_refptr<DomStorageNamespace> > |
| 107 StorageNamespaceMap; |
| 51 | 108 |
| 52 // Clones a session storage namespace and returns the cloned namespaces' id. | 109 ~DomStorageContext(); |
| 53 // Only call on the IO thread. | |
| 54 int64 CloneSessionStorage(int64 original_id); | |
| 55 | 110 |
| 56 // Various storage area methods. The storage area is owned by one of the | 111 // May be called on any thread. |
| 57 // namespaces that's owned by this class. | 112 int64 AllocateSessionId() { |
| 58 void RegisterStorageArea(DOMStorageArea* storage_area); | 113 return session_id_sequence_.GetNext(); |
| 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 | |
| 77 // Tells storage namespaces to purge any memory they do not need. | |
| 78 virtual void PurgeMemory(); | |
| 79 | |
| 80 // Delete any local storage files that have been touched since the cutoff | |
| 81 // date that's supplied. Protected origins, per the SpecialStoragePolicy, | |
| 82 // are not deleted by this method. | |
| 83 void DeleteDataModifiedSince(const base::Time& cutoff); | |
| 84 | |
| 85 // Deletes a single local storage file. | |
| 86 void DeleteLocalStorageFile(const FilePath& file_path); | |
| 87 | |
| 88 // Deletes the local storage file for the given origin. | |
| 89 void DeleteLocalStorageForOrigin(const string16& origin_id); | |
| 90 | |
| 91 // Deletes all local storage files. | |
| 92 void DeleteAllLocalStorageFiles(); | |
| 93 | |
| 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) { | |
| 104 clear_local_state_on_exit_ = clear_local_state; | |
| 105 } | 114 } |
| 106 | 115 |
| 107 // Disables the exit-time deletion for all data (also session-only data). | 116 // Must be called on the background thread. |
| 108 void SaveSessionState() { | 117 void CreateSessionNamespace(int64 namespace_id); |
| 109 save_session_state_ = true; | 118 void DeleteSessionNamespace(int64 namespace_id); |
| 110 } | 119 void CloneSessionNamespace(int64 existing_id, int64 new_id); |
| 111 | 120 |
| 112 void set_data_path_for_testing(const FilePath& data_path) { | 121 // Collection of namespaces keyed by id. |
| 113 data_path_ = data_path; | 122 StorageNamespaceMap namespaces_; |
| 114 } | |
| 115 | 123 |
| 116 private: | 124 // Where localstorage data is stored, maybe empty for the incognito use case. |
| 125 FilePath directory_; |
| 117 | 126 |
| 118 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SessionOnly); | 127 // Used to schedule sequenced background tasks. |
| 119 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SaveSessionState); | 128 scoped_refptr<DomStorageTaskRunner> task_runner_; |
| 120 | 129 |
| 121 // Get the local storage instance. The object is owned by this class. | 130 // List of objects observing local storage events. |
| 122 DOMStorageNamespace* CreateLocalStorage(); | 131 ObserverList<EventObserver> event_observers_; |
| 123 | 132 |
| 124 // Get a new session storage namespace. The object is owned by this class. | 133 // We use a 32 bit identifier for per tab storage sessions. |
| 125 DOMStorageNamespace* CreateSessionStorage(int64 namespace_id); | 134 // At a tab per second, this range is large enough for 68 years. |
| 126 | 135 base::AtomicSequenceNumber session_id_sequence_; |
| 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; | |
| 170 StorageNamespaceMap storage_namespace_map_; | |
| 171 | |
| 172 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_; | |
| 173 | |
| 174 DISALLOW_IMPLICIT_CONSTRUCTORS(DOMStorageContext); | |
| 175 }; | 136 }; |
| 176 | 137 |
| 177 #endif // CONTENT_BROWSER_IN_PROCESS_WEBKIT_DOM_STORAGE_CONTEXT_H_ | 138 } // namespace dom_storage |
| 139 |
| 140 #endif // WEBKIT_DOM_STORAGE_DOM_STORAGE_CONTEXT_H_ |
| OLD | NEW |