OLD | NEW |
1 // Copyright (c) 2012 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_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ | 5 #ifndef CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ |
6 #define CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ | 6 #define CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ |
7 | 7 |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include <map> | 10 #include <map> |
11 #include <memory> | 11 #include <memory> |
12 #include <string> | 12 #include <string> |
13 #include <utility> | 13 #include <utility> |
14 #include <vector> | 14 #include <vector> |
15 | 15 |
16 #include "base/id_map.h" | |
17 #include "base/macros.h" | 16 #include "base/macros.h" |
18 #include "base/memory/ref_counted.h" | 17 #include "base/memory/ref_counted.h" |
19 #include "content/browser/blob_storage/chrome_blob_storage_context.h" | 18 #include "content/browser/blob_storage/chrome_blob_storage_context.h" |
20 #include "content/common/indexed_db/indexed_db.mojom.h" | 19 #include "content/common/indexed_db/indexed_db.mojom.h" |
21 #include "content/public/browser/browser_associated_interface.h" | 20 #include "content/public/browser/browser_associated_interface.h" |
22 #include "content/public/browser/browser_message_filter.h" | 21 #include "content/public/browser/browser_message_filter.h" |
23 #include "net/url_request/url_request_context_getter.h" | 22 #include "net/url_request/url_request_context_getter.h" |
24 #include "storage/browser/blob/blob_data_handle.h" | 23 #include "storage/browser/blob/blob_data_handle.h" |
25 #include "url/gurl.h" | 24 #include "url/gurl.h" |
26 | 25 |
27 struct IndexedDBHostMsg_DatabaseObserve_Params; | 26 struct IndexedDBHostMsg_DatabaseObserve_Params; |
28 struct IndexedDBMsg_Observation; | 27 struct IndexedDBMsg_Observation; |
29 struct IndexedDBMsg_ObserverChanges; | 28 struct IndexedDBMsg_ObserverChanges; |
30 | 29 |
31 namespace url { | 30 namespace url { |
32 class Origin; | 31 class Origin; |
33 } | 32 } |
34 | 33 |
35 namespace content { | 34 namespace content { |
36 class IndexedDBBlobInfo; | 35 class IndexedDBBlobInfo; |
37 class IndexedDBCallbacks; | 36 class IndexedDBCallbacks; |
38 class IndexedDBConnection; | 37 class IndexedDBConnection; |
39 class IndexedDBContextImpl; | 38 class IndexedDBContextImpl; |
40 class IndexedDBCursor; | |
41 class IndexedDBDatabaseCallbacks; | 39 class IndexedDBDatabaseCallbacks; |
42 class IndexedDBKey; | |
43 class IndexedDBObservation; | 40 class IndexedDBObservation; |
44 class IndexedDBObserverChanges; | 41 class IndexedDBObserverChanges; |
45 | 42 |
46 // Handles all IndexedDB related messages from a particular renderer process. | 43 // Handles all IndexedDB related messages from a particular renderer process. |
47 class IndexedDBDispatcherHost | 44 class IndexedDBDispatcherHost |
48 : public BrowserMessageFilter, | 45 : public BrowserMessageFilter, |
49 public BrowserAssociatedInterface<::indexed_db::mojom::Factory>, | 46 public BrowserAssociatedInterface<::indexed_db::mojom::Factory>, |
50 public ::indexed_db::mojom::Factory { | 47 public ::indexed_db::mojom::Factory { |
51 public: | 48 public: |
52 // Only call the constructor from the UI thread. | 49 // Only call the constructor from the UI thread. |
53 IndexedDBDispatcherHost(int ipc_process_id, | 50 IndexedDBDispatcherHost(int ipc_process_id, |
54 net::URLRequestContextGetter* request_context_getter, | 51 net::URLRequestContextGetter* request_context_getter, |
55 IndexedDBContextImpl* indexed_db_context, | 52 IndexedDBContextImpl* indexed_db_context, |
56 ChromeBlobStorageContext* blob_storage_context); | 53 ChromeBlobStorageContext* blob_storage_context); |
57 | 54 |
58 static IndexedDBMsg_ObserverChanges ConvertObserverChanges( | 55 static IndexedDBMsg_ObserverChanges ConvertObserverChanges( |
59 std::unique_ptr<IndexedDBObserverChanges> changes); | 56 std::unique_ptr<IndexedDBObserverChanges> changes); |
60 static IndexedDBMsg_Observation ConvertObservation( | 57 static IndexedDBMsg_Observation ConvertObservation( |
61 const IndexedDBObservation* observation); | 58 const IndexedDBObservation* observation); |
62 | 59 |
63 // BrowserMessageFilter implementation. | 60 // BrowserMessageFilter implementation. |
64 void OnChannelClosing() override; | 61 void OnChannelClosing() override; |
65 void OnDestruct() const override; | 62 void OnDestruct() const override; |
66 base::TaskRunner* OverrideTaskRunnerForMessage( | |
67 const IPC::Message& message) override; | |
68 bool OnMessageReceived(const IPC::Message& message) override; | 63 bool OnMessageReceived(const IPC::Message& message) override; |
69 | 64 |
70 void FinishTransaction(int64_t host_transaction_id, bool committed); | 65 void FinishTransaction(int64_t host_transaction_id, bool committed); |
71 | 66 |
72 // A shortcut for accessing our context. | 67 // A shortcut for accessing our context. |
73 IndexedDBContextImpl* context() const { return indexed_db_context_.get(); } | 68 IndexedDBContextImpl* context() const { return indexed_db_context_.get(); } |
74 storage::BlobStorageContext* blob_storage_context() const { | 69 storage::BlobStorageContext* blob_storage_context() const { |
75 return blob_storage_context_->context(); | 70 return blob_storage_context_->context(); |
76 } | 71 } |
77 int ipc_process_id() const { return ipc_process_id_; } | 72 int ipc_process_id() const { return ipc_process_id_; } |
78 | 73 |
79 // IndexedDBCallbacks call these methods to add the results into the | |
80 // applicable map. See below for more details. | |
81 int32_t Add(IndexedDBCursor* cursor); | |
82 | |
83 bool RegisterTransactionId(int64_t host_transaction_id, | 74 bool RegisterTransactionId(int64_t host_transaction_id, |
84 const url::Origin& origin); | 75 const url::Origin& origin); |
85 bool GetTransactionSize(int64_t host_transaction_id, | 76 bool GetTransactionSize(int64_t host_transaction_id, |
86 int64_t* transaction_size); | 77 int64_t* transaction_size); |
87 void AddToTransaction(int64_t host_transaction_id, int64_t value_length); | 78 void AddToTransaction(int64_t host_transaction_id, int64_t value_length); |
88 | 79 |
89 IndexedDBCursor* GetCursorFromId(int32_t ipc_cursor_id); | |
90 | |
91 // These are called to map a 32-bit front-end (renderer-specific) transaction | 80 // These are called to map a 32-bit front-end (renderer-specific) transaction |
92 // id to and from a back-end ("host") transaction id that encodes the process | 81 // id to and from a back-end ("host") transaction id that encodes the process |
93 // id in the high 32 bits. The mapping is host-specific and ids are validated. | 82 // id in the high 32 bits. The mapping is host-specific and ids are validated. |
94 int64_t HostTransactionId(int64_t transaction_id); | 83 int64_t HostTransactionId(int64_t transaction_id); |
95 int64_t RendererTransactionId(int64_t host_transaction_id); | 84 int64_t RendererTransactionId(int64_t host_transaction_id); |
96 | 85 |
97 // These are called to decode a host transaction ID, for diagnostic purposes. | 86 // These are called to decode a host transaction ID, for diagnostic purposes. |
98 static uint32_t TransactionIdToRendererTransactionId( | 87 static uint32_t TransactionIdToRendererTransactionId( |
99 int64_t host_transaction_id); | 88 int64_t host_transaction_id); |
100 static uint32_t TransactionIdToProcessId(int64_t host_transaction_id); | 89 static uint32_t TransactionIdToProcessId(int64_t host_transaction_id); |
101 | 90 |
102 std::string HoldBlobData(const IndexedDBBlobInfo& blob_info); | 91 std::string HoldBlobData(const IndexedDBBlobInfo& blob_info); |
103 void DropBlobData(const std::string& uuid); | 92 void DropBlobData(const std::string& uuid); |
104 | 93 |
105 // True if the channel is closing/closed and outstanding requests | 94 // True if the channel is closing/closed and outstanding requests |
106 // can be abandoned. Only access on IndexedDB thread. | 95 // can be abandoned. Only access on IndexedDB thread. |
107 bool IsOpen() const; | 96 bool IsOpen() const; |
108 | 97 |
109 private: | 98 private: |
110 // Friends to enable OnDestruct() delegation. | 99 // Friends to enable OnDestruct() delegation. |
111 friend class BrowserThread; | 100 friend class BrowserThread; |
112 friend class base::DeleteHelper<IndexedDBDispatcherHost>; | 101 friend class base::DeleteHelper<IndexedDBDispatcherHost>; |
113 | 102 |
114 // Used in nested classes. | 103 // Used in nested classes. |
115 typedef std::map<int64_t, int64_t> TransactionIDToDatabaseIDMap; | |
116 typedef std::map<int64_t, int64_t> TransactionIDToSizeMap; | 104 typedef std::map<int64_t, int64_t> TransactionIDToSizeMap; |
117 typedef std::map<int64_t, url::Origin> TransactionIDToOriginMap; | 105 typedef std::map<int64_t, url::Origin> TransactionIDToOriginMap; |
118 | 106 |
119 // IDMap for RefCounted types | |
120 template <typename RefCountedType> | |
121 class RefIDMap { | |
122 public: | |
123 typedef int32_t KeyType; | |
124 | |
125 RefIDMap() {} | |
126 ~RefIDMap() {} | |
127 | |
128 KeyType Add(RefCountedType* data) { | |
129 return map_.Add(new scoped_refptr<RefCountedType>(data)); | |
130 } | |
131 | |
132 RefCountedType* Lookup(KeyType id) { | |
133 scoped_refptr<RefCountedType>* ptr = map_.Lookup(id); | |
134 if (ptr == NULL) | |
135 return NULL; | |
136 return ptr->get(); | |
137 } | |
138 | |
139 void Remove(KeyType id) { map_.Remove(id); } | |
140 | |
141 void set_check_on_null_data(bool value) { | |
142 map_.set_check_on_null_data(value); | |
143 } | |
144 | |
145 private: | |
146 IDMap<scoped_refptr<RefCountedType>, IDMapOwnPointer> map_; | |
147 | |
148 DISALLOW_COPY_AND_ASSIGN(RefIDMap); | |
149 }; | |
150 | |
151 class CursorDispatcherHost { | |
152 public: | |
153 explicit CursorDispatcherHost(IndexedDBDispatcherHost* parent); | |
154 ~CursorDispatcherHost(); | |
155 | |
156 bool OnMessageReceived(const IPC::Message& message); | |
157 | |
158 void OnAdvance(int32_t ipc_object_store_id, | |
159 int32_t ipc_thread_id, | |
160 int32_t ipc_callbacks_id, | |
161 uint32_t count); | |
162 void OnContinue(int32_t ipc_object_store_id, | |
163 int32_t ipc_thread_id, | |
164 int32_t ipc_callbacks_id, | |
165 const IndexedDBKey& key, | |
166 const IndexedDBKey& primary_key); | |
167 void OnPrefetch(int32_t ipc_cursor_id, | |
168 int32_t ipc_thread_id, | |
169 int32_t ipc_callbacks_id, | |
170 int n); | |
171 void OnPrefetchReset(int32_t ipc_cursor_id, | |
172 int used_prefetches, | |
173 int unused_prefetches); | |
174 void OnDestroyed(int32_t ipc_cursor_id); | |
175 | |
176 IndexedDBDispatcherHost* parent_; | |
177 RefIDMap<IndexedDBCursor> map_; | |
178 | |
179 private: | |
180 DISALLOW_COPY_AND_ASSIGN(CursorDispatcherHost); | |
181 }; | |
182 | |
183 ~IndexedDBDispatcherHost() override; | 107 ~IndexedDBDispatcherHost() override; |
184 | 108 |
185 // Helper templates. | |
186 template <class ReturnType> | |
187 ReturnType* GetOrTerminateProcess(RefIDMap<ReturnType>* map, | |
188 int32_t ipc_return_object_id); | |
189 | |
190 template <typename MapType> | |
191 void DestroyObject(MapType* map, int32_t ipc_object_id); | |
192 | |
193 // indexed_db::mojom::Factory implementation: | 109 // indexed_db::mojom::Factory implementation: |
194 void GetDatabaseNames( | 110 void GetDatabaseNames( |
195 ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info, | 111 ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info, |
196 const url::Origin& origin) override; | 112 const url::Origin& origin) override; |
197 void Open(int32_t worker_thread, | 113 void Open(int32_t worker_thread, |
198 ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info, | 114 ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info, |
199 ::indexed_db::mojom::DatabaseCallbacksAssociatedPtrInfo | 115 ::indexed_db::mojom::DatabaseCallbacksAssociatedPtrInfo |
200 database_callbacks_info, | 116 database_callbacks_info, |
201 const url::Origin& origin, | 117 const url::Origin& origin, |
202 const base::string16& name, | 118 const base::string16& name, |
(...skipping 10 matching lines...) Expand all Loading... |
213 scoped_refptr<IndexedDBCallbacks> callbacks, | 129 scoped_refptr<IndexedDBCallbacks> callbacks, |
214 scoped_refptr<IndexedDBDatabaseCallbacks> database_callbacks, | 130 scoped_refptr<IndexedDBDatabaseCallbacks> database_callbacks, |
215 const url::Origin& origin, | 131 const url::Origin& origin, |
216 const base::string16& name, | 132 const base::string16& name, |
217 int64_t version, | 133 int64_t version, |
218 int64_t transaction_id); | 134 int64_t transaction_id); |
219 void DeleteDatabaseOnIDBThread(scoped_refptr<IndexedDBCallbacks> callbacks, | 135 void DeleteDatabaseOnIDBThread(scoped_refptr<IndexedDBCallbacks> callbacks, |
220 const url::Origin& origin, | 136 const url::Origin& origin, |
221 const base::string16& name); | 137 const base::string16& name); |
222 | 138 |
223 // Message processing. Most of the work is delegated to the dispatcher hosts | |
224 // below. | |
225 void OnAckReceivedBlobs(const std::vector<std::string>& uuids); | |
226 | |
227 void ResetDispatcherHosts(); | 139 void ResetDispatcherHosts(); |
228 | 140 |
229 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; | 141 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; |
230 scoped_refptr<IndexedDBContextImpl> indexed_db_context_; | 142 scoped_refptr<IndexedDBContextImpl> indexed_db_context_; |
231 scoped_refptr<ChromeBlobStorageContext> blob_storage_context_; | 143 scoped_refptr<ChromeBlobStorageContext> blob_storage_context_; |
232 | 144 |
233 // Maps blob uuid to a pair (handle, ref count). Entry is added and/or count | 145 // Maps blob uuid to a pair (handle, ref count). Entry is added and/or count |
234 // is incremented in HoldBlobData(), and count is decremented and/or entry | 146 // is incremented in HoldBlobData(), and count is decremented and/or entry |
235 // removed in DropBlobData(). | 147 // removed in DropBlobData(). |
236 std::map<std::string, | 148 std::map<std::string, |
237 std::pair<std::unique_ptr<storage::BlobDataHandle>, int>> | 149 std::pair<std::unique_ptr<storage::BlobDataHandle>, int>> |
238 blob_data_handle_map_; | 150 blob_data_handle_map_; |
239 | 151 |
240 // Only access on IndexedDB thread. | 152 // Only access on IndexedDB thread. |
241 bool is_open_ = true; | 153 bool is_open_ = true; |
242 TransactionIDToSizeMap transaction_size_map_; | 154 TransactionIDToSizeMap transaction_size_map_; |
243 TransactionIDToOriginMap transaction_origin_map_; | 155 TransactionIDToOriginMap transaction_origin_map_; |
244 std::unique_ptr<CursorDispatcherHost> cursor_dispatcher_host_; | |
245 | 156 |
246 // Used to set file permissions for blob storage. | 157 // Used to set file permissions for blob storage. |
247 int ipc_process_id_; | 158 int ipc_process_id_; |
248 | 159 |
249 DISALLOW_IMPLICIT_CONSTRUCTORS(IndexedDBDispatcherHost); | 160 DISALLOW_IMPLICIT_CONSTRUCTORS(IndexedDBDispatcherHost); |
250 }; | 161 }; |
251 | 162 |
252 } // namespace content | 163 } // namespace content |
253 | 164 |
254 #endif // CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ | 165 #endif // CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_DISPATCHER_HOST_H_ |
OLD | NEW |