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 |