OLD | NEW |
| (Empty) |
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 #include "net/base/default_server_bound_cert_store.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/message_loop.h" | |
9 #include "base/metrics/histogram.h" | |
10 | |
11 namespace net { | |
12 | |
13 // -------------------------------------------------------------------------- | |
14 // Task | |
15 class DefaultServerBoundCertStore::Task { | |
16 public: | |
17 virtual ~Task(); | |
18 | |
19 // Runs the task and invokes the client callback on the thread that | |
20 // originally constructed the task. | |
21 virtual void Run(DefaultServerBoundCertStore* store) = 0; | |
22 | |
23 protected: | |
24 void InvokeCallback(base::Closure callback) const; | |
25 }; | |
26 | |
27 DefaultServerBoundCertStore::Task::~Task() { | |
28 } | |
29 | |
30 void DefaultServerBoundCertStore::Task::InvokeCallback( | |
31 base::Closure callback) const { | |
32 if (!callback.is_null()) | |
33 callback.Run(); | |
34 } | |
35 | |
36 // -------------------------------------------------------------------------- | |
37 // GetServerBoundCertTask | |
38 class DefaultServerBoundCertStore::GetServerBoundCertTask | |
39 : public DefaultServerBoundCertStore::Task { | |
40 public: | |
41 GetServerBoundCertTask(const std::string& server_identifier, | |
42 const GetCertCallback& callback); | |
43 virtual ~GetServerBoundCertTask(); | |
44 virtual void Run(DefaultServerBoundCertStore* store) OVERRIDE; | |
45 | |
46 private: | |
47 std::string server_identifier_; | |
48 GetCertCallback callback_; | |
49 }; | |
50 | |
51 DefaultServerBoundCertStore::GetServerBoundCertTask::GetServerBoundCertTask( | |
52 const std::string& server_identifier, | |
53 const GetCertCallback& callback) | |
54 : server_identifier_(server_identifier), | |
55 callback_(callback) { | |
56 } | |
57 | |
58 DefaultServerBoundCertStore::GetServerBoundCertTask::~GetServerBoundCertTask() { | |
59 } | |
60 | |
61 void DefaultServerBoundCertStore::GetServerBoundCertTask::Run( | |
62 DefaultServerBoundCertStore* store) { | |
63 SSLClientCertType type = CLIENT_CERT_INVALID_TYPE; | |
64 base::Time expiration_time; | |
65 std::string private_key_result; | |
66 std::string cert_result; | |
67 bool was_sync = store->GetServerBoundCert( | |
68 server_identifier_, &type, &expiration_time, &private_key_result, | |
69 &cert_result, GetCertCallback()); | |
70 DCHECK(was_sync); | |
71 | |
72 InvokeCallback(base::Bind(callback_, server_identifier_, type, | |
73 expiration_time, private_key_result, cert_result)); | |
74 } | |
75 | |
76 // -------------------------------------------------------------------------- | |
77 // SetServerBoundCertTask | |
78 class DefaultServerBoundCertStore::SetServerBoundCertTask | |
79 : public DefaultServerBoundCertStore::Task { | |
80 public: | |
81 SetServerBoundCertTask(const std::string& server_identifier, | |
82 SSLClientCertType type, | |
83 base::Time creation_time, | |
84 base::Time expiration_time, | |
85 const std::string& private_key, | |
86 const std::string& cert); | |
87 virtual ~SetServerBoundCertTask(); | |
88 virtual void Run(DefaultServerBoundCertStore* store) OVERRIDE; | |
89 | |
90 private: | |
91 std::string server_identifier_; | |
92 SSLClientCertType type_; | |
93 base::Time creation_time_; | |
94 base::Time expiration_time_; | |
95 std::string private_key_; | |
96 std::string cert_; | |
97 }; | |
98 | |
99 DefaultServerBoundCertStore::SetServerBoundCertTask::SetServerBoundCertTask( | |
100 const std::string& server_identifier, | |
101 SSLClientCertType type, | |
102 base::Time creation_time, | |
103 base::Time expiration_time, | |
104 const std::string& private_key, | |
105 const std::string& cert) | |
106 : server_identifier_(server_identifier), | |
107 type_(type), | |
108 creation_time_(creation_time), | |
109 expiration_time_(expiration_time), | |
110 private_key_(private_key), | |
111 cert_(cert) { | |
112 } | |
113 | |
114 DefaultServerBoundCertStore::SetServerBoundCertTask::~SetServerBoundCertTask() { | |
115 } | |
116 | |
117 void DefaultServerBoundCertStore::SetServerBoundCertTask::Run( | |
118 DefaultServerBoundCertStore* store) { | |
119 store->SyncSetServerBoundCert(server_identifier_, type_, creation_time_, | |
120 expiration_time_, private_key_, cert_); | |
121 } | |
122 | |
123 // -------------------------------------------------------------------------- | |
124 // DeleteServerBoundCertTask | |
125 class DefaultServerBoundCertStore::DeleteServerBoundCertTask | |
126 : public DefaultServerBoundCertStore::Task { | |
127 public: | |
128 DeleteServerBoundCertTask(const std::string& server_identifier, | |
129 const base::Closure& callback); | |
130 virtual ~DeleteServerBoundCertTask(); | |
131 virtual void Run(DefaultServerBoundCertStore* store) OVERRIDE; | |
132 | |
133 private: | |
134 std::string server_identifier_; | |
135 base::Closure callback_; | |
136 }; | |
137 | |
138 DefaultServerBoundCertStore::DeleteServerBoundCertTask:: | |
139 DeleteServerBoundCertTask( | |
140 const std::string& server_identifier, | |
141 const base::Closure& callback) | |
142 : server_identifier_(server_identifier), | |
143 callback_(callback) { | |
144 } | |
145 | |
146 DefaultServerBoundCertStore::DeleteServerBoundCertTask:: | |
147 ~DeleteServerBoundCertTask() { | |
148 } | |
149 | |
150 void DefaultServerBoundCertStore::DeleteServerBoundCertTask::Run( | |
151 DefaultServerBoundCertStore* store) { | |
152 store->SyncDeleteServerBoundCert(server_identifier_); | |
153 | |
154 InvokeCallback(callback_); | |
155 } | |
156 | |
157 // -------------------------------------------------------------------------- | |
158 // DeleteAllCreatedBetweenTask | |
159 class DefaultServerBoundCertStore::DeleteAllCreatedBetweenTask | |
160 : public DefaultServerBoundCertStore::Task { | |
161 public: | |
162 DeleteAllCreatedBetweenTask(base::Time delete_begin, | |
163 base::Time delete_end, | |
164 const base::Closure& callback); | |
165 virtual ~DeleteAllCreatedBetweenTask(); | |
166 virtual void Run(DefaultServerBoundCertStore* store) OVERRIDE; | |
167 | |
168 private: | |
169 base::Time delete_begin_; | |
170 base::Time delete_end_; | |
171 base::Closure callback_; | |
172 }; | |
173 | |
174 DefaultServerBoundCertStore::DeleteAllCreatedBetweenTask:: | |
175 DeleteAllCreatedBetweenTask( | |
176 base::Time delete_begin, | |
177 base::Time delete_end, | |
178 const base::Closure& callback) | |
179 : delete_begin_(delete_begin), | |
180 delete_end_(delete_end), | |
181 callback_(callback) { | |
182 } | |
183 | |
184 DefaultServerBoundCertStore::DeleteAllCreatedBetweenTask:: | |
185 ~DeleteAllCreatedBetweenTask() { | |
186 } | |
187 | |
188 void DefaultServerBoundCertStore::DeleteAllCreatedBetweenTask::Run( | |
189 DefaultServerBoundCertStore* store) { | |
190 store->SyncDeleteAllCreatedBetween(delete_begin_, delete_end_); | |
191 | |
192 InvokeCallback(callback_); | |
193 } | |
194 | |
195 // -------------------------------------------------------------------------- | |
196 // GetAllServerBoundCertsTask | |
197 class DefaultServerBoundCertStore::GetAllServerBoundCertsTask | |
198 : public DefaultServerBoundCertStore::Task { | |
199 public: | |
200 explicit GetAllServerBoundCertsTask(const GetCertListCallback& callback); | |
201 virtual ~GetAllServerBoundCertsTask(); | |
202 virtual void Run(DefaultServerBoundCertStore* store) OVERRIDE; | |
203 | |
204 private: | |
205 std::string server_identifier_; | |
206 GetCertListCallback callback_; | |
207 }; | |
208 | |
209 DefaultServerBoundCertStore::GetAllServerBoundCertsTask:: | |
210 GetAllServerBoundCertsTask(const GetCertListCallback& callback) | |
211 : callback_(callback) { | |
212 } | |
213 | |
214 DefaultServerBoundCertStore::GetAllServerBoundCertsTask:: | |
215 ~GetAllServerBoundCertsTask() { | |
216 } | |
217 | |
218 void DefaultServerBoundCertStore::GetAllServerBoundCertsTask::Run( | |
219 DefaultServerBoundCertStore* store) { | |
220 ServerBoundCertList cert_list; | |
221 store->SyncGetAllServerBoundCerts(&cert_list); | |
222 | |
223 InvokeCallback(base::Bind(callback_, cert_list)); | |
224 } | |
225 | |
226 // -------------------------------------------------------------------------- | |
227 // DefaultServerBoundCertStore | |
228 | |
229 // static | |
230 const size_t DefaultServerBoundCertStore::kMaxCerts = 3300; | |
231 | |
232 DefaultServerBoundCertStore::DefaultServerBoundCertStore( | |
233 PersistentStore* store) | |
234 : initialized_(false), | |
235 loaded_(false), | |
236 store_(store), | |
237 ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {} | |
238 | |
239 void DefaultServerBoundCertStore::FlushStore( | |
240 const base::Closure& completion_task) { | |
241 DCHECK(CalledOnValidThread()); | |
242 | |
243 if (initialized_ && store_) | |
244 store_->Flush(completion_task); | |
245 else if (!completion_task.is_null()) | |
246 MessageLoop::current()->PostTask(FROM_HERE, completion_task); | |
247 } | |
248 | |
249 bool DefaultServerBoundCertStore::GetServerBoundCert( | |
250 const std::string& server_identifier, | |
251 SSLClientCertType* type, | |
252 base::Time* expiration_time, | |
253 std::string* private_key_result, | |
254 std::string* cert_result, | |
255 const GetCertCallback& callback) { | |
256 DCHECK(CalledOnValidThread()); | |
257 InitIfNecessary(); | |
258 | |
259 if (!loaded_) { | |
260 EnqueueTask(scoped_ptr<Task>( | |
261 new GetServerBoundCertTask(server_identifier, callback))); | |
262 return false; | |
263 } | |
264 | |
265 ServerBoundCertMap::iterator it = server_bound_certs_.find(server_identifier); | |
266 | |
267 if (it == server_bound_certs_.end()) { | |
268 *type = CLIENT_CERT_INVALID_TYPE; | |
269 return true; | |
270 } | |
271 | |
272 ServerBoundCert* cert = it->second; | |
273 *type = cert->type(); | |
274 *expiration_time = cert->expiration_time(); | |
275 *private_key_result = cert->private_key(); | |
276 *cert_result = cert->cert(); | |
277 | |
278 return true; | |
279 } | |
280 | |
281 void DefaultServerBoundCertStore::SetServerBoundCert( | |
282 const std::string& server_identifier, | |
283 SSLClientCertType type, | |
284 base::Time creation_time, | |
285 base::Time expiration_time, | |
286 const std::string& private_key, | |
287 const std::string& cert) { | |
288 RunOrEnqueueTask(scoped_ptr<Task>(new SetServerBoundCertTask( | |
289 server_identifier, type, creation_time, expiration_time, private_key, | |
290 cert))); | |
291 } | |
292 | |
293 void DefaultServerBoundCertStore::DeleteServerBoundCert( | |
294 const std::string& server_identifier, | |
295 const base::Closure& callback) { | |
296 RunOrEnqueueTask(scoped_ptr<Task>( | |
297 new DeleteServerBoundCertTask(server_identifier, callback))); | |
298 } | |
299 | |
300 void DefaultServerBoundCertStore::DeleteAllCreatedBetween( | |
301 base::Time delete_begin, | |
302 base::Time delete_end, | |
303 const base::Closure& callback) { | |
304 RunOrEnqueueTask(scoped_ptr<Task>( | |
305 new DeleteAllCreatedBetweenTask(delete_begin, delete_end, callback))); | |
306 } | |
307 | |
308 void DefaultServerBoundCertStore::DeleteAll( | |
309 const base::Closure& callback) { | |
310 DeleteAllCreatedBetween(base::Time(), base::Time(), callback); | |
311 } | |
312 | |
313 void DefaultServerBoundCertStore::GetAllServerBoundCerts( | |
314 const GetCertListCallback& callback) { | |
315 RunOrEnqueueTask(scoped_ptr<Task>(new GetAllServerBoundCertsTask(callback))); | |
316 } | |
317 | |
318 int DefaultServerBoundCertStore::GetCertCount() { | |
319 DCHECK(CalledOnValidThread()); | |
320 | |
321 return server_bound_certs_.size(); | |
322 } | |
323 | |
324 void DefaultServerBoundCertStore::SetForceKeepSessionState() { | |
325 DCHECK(CalledOnValidThread()); | |
326 InitIfNecessary(); | |
327 | |
328 if (store_) | |
329 store_->SetForceKeepSessionState(); | |
330 } | |
331 | |
332 DefaultServerBoundCertStore::~DefaultServerBoundCertStore() { | |
333 DeleteAllInMemory(); | |
334 } | |
335 | |
336 void DefaultServerBoundCertStore::DeleteAllInMemory() { | |
337 DCHECK(CalledOnValidThread()); | |
338 | |
339 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
340 it != server_bound_certs_.end(); ++it) { | |
341 delete it->second; | |
342 } | |
343 server_bound_certs_.clear(); | |
344 } | |
345 | |
346 void DefaultServerBoundCertStore::InitStore() { | |
347 DCHECK(CalledOnValidThread()); | |
348 DCHECK(store_) << "Store must exist to initialize"; | |
349 DCHECK(!loaded_); | |
350 | |
351 store_->Load(base::Bind(&DefaultServerBoundCertStore::OnLoaded, | |
352 weak_ptr_factory_.GetWeakPtr())); | |
353 } | |
354 | |
355 void DefaultServerBoundCertStore::OnLoaded( | |
356 scoped_ptr<ScopedVector<ServerBoundCert> > certs) { | |
357 DCHECK(CalledOnValidThread()); | |
358 | |
359 for (std::vector<ServerBoundCert*>::const_iterator it = certs->begin(); | |
360 it != certs->end(); ++it) { | |
361 DCHECK(server_bound_certs_.find((*it)->server_identifier()) == | |
362 server_bound_certs_.end()); | |
363 server_bound_certs_[(*it)->server_identifier()] = *it; | |
364 } | |
365 certs->weak_clear(); | |
366 | |
367 loaded_ = true; | |
368 | |
369 base::TimeDelta wait_time; | |
370 if (!waiting_tasks_.empty()) | |
371 wait_time = base::TimeTicks::Now() - waiting_tasks_start_time_; | |
372 DVLOG(1) << "Task delay " << wait_time.InMilliseconds(); | |
373 UMA_HISTOGRAM_CUSTOM_TIMES("DomainBoundCerts.TaskMaxWaitTime", | |
374 wait_time, | |
375 base::TimeDelta::FromMilliseconds(1), | |
376 base::TimeDelta::FromMinutes(1), | |
377 50); | |
378 UMA_HISTOGRAM_COUNTS_100("DomainBoundCerts.TaskWaitCount", | |
379 waiting_tasks_.size()); | |
380 | |
381 | |
382 for (ScopedVector<Task>::iterator i = waiting_tasks_.begin(); | |
383 i != waiting_tasks_.end(); ++i) | |
384 (*i)->Run(this); | |
385 waiting_tasks_.clear(); | |
386 } | |
387 | |
388 void DefaultServerBoundCertStore::SyncSetServerBoundCert( | |
389 const std::string& server_identifier, | |
390 SSLClientCertType type, | |
391 base::Time creation_time, | |
392 base::Time expiration_time, | |
393 const std::string& private_key, | |
394 const std::string& cert) { | |
395 DCHECK(CalledOnValidThread()); | |
396 DCHECK(loaded_); | |
397 | |
398 InternalDeleteServerBoundCert(server_identifier); | |
399 InternalInsertServerBoundCert( | |
400 server_identifier, | |
401 new ServerBoundCert( | |
402 server_identifier, type, creation_time, expiration_time, private_key, | |
403 cert)); | |
404 } | |
405 | |
406 void DefaultServerBoundCertStore::SyncDeleteServerBoundCert( | |
407 const std::string& server_identifier) { | |
408 DCHECK(CalledOnValidThread()); | |
409 DCHECK(loaded_); | |
410 InternalDeleteServerBoundCert(server_identifier); | |
411 } | |
412 | |
413 void DefaultServerBoundCertStore::SyncDeleteAllCreatedBetween( | |
414 base::Time delete_begin, | |
415 base::Time delete_end) { | |
416 DCHECK(CalledOnValidThread()); | |
417 DCHECK(loaded_); | |
418 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
419 it != server_bound_certs_.end();) { | |
420 ServerBoundCertMap::iterator cur = it; | |
421 ++it; | |
422 ServerBoundCert* cert = cur->second; | |
423 if ((delete_begin.is_null() || cert->creation_time() >= delete_begin) && | |
424 (delete_end.is_null() || cert->creation_time() < delete_end)) { | |
425 if (store_) | |
426 store_->DeleteServerBoundCert(*cert); | |
427 delete cert; | |
428 server_bound_certs_.erase(cur); | |
429 } | |
430 } | |
431 } | |
432 | |
433 void DefaultServerBoundCertStore::SyncGetAllServerBoundCerts( | |
434 ServerBoundCertList* cert_list) { | |
435 DCHECK(CalledOnValidThread()); | |
436 DCHECK(loaded_); | |
437 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
438 it != server_bound_certs_.end(); ++it) | |
439 cert_list->push_back(*it->second); | |
440 } | |
441 | |
442 void DefaultServerBoundCertStore::EnqueueTask(scoped_ptr<Task> task) { | |
443 DCHECK(CalledOnValidThread()); | |
444 DCHECK(!loaded_); | |
445 if (waiting_tasks_.empty()) | |
446 waiting_tasks_start_time_ = base::TimeTicks::Now(); | |
447 waiting_tasks_.push_back(task.release()); | |
448 } | |
449 | |
450 void DefaultServerBoundCertStore::RunOrEnqueueTask(scoped_ptr<Task> task) { | |
451 DCHECK(CalledOnValidThread()); | |
452 InitIfNecessary(); | |
453 | |
454 if (!loaded_) { | |
455 EnqueueTask(task.Pass()); | |
456 return; | |
457 } | |
458 | |
459 task->Run(this); | |
460 } | |
461 | |
462 void DefaultServerBoundCertStore::InternalDeleteServerBoundCert( | |
463 const std::string& server_identifier) { | |
464 DCHECK(CalledOnValidThread()); | |
465 DCHECK(loaded_); | |
466 | |
467 ServerBoundCertMap::iterator it = server_bound_certs_.find(server_identifier); | |
468 if (it == server_bound_certs_.end()) | |
469 return; // There is nothing to delete. | |
470 | |
471 ServerBoundCert* cert = it->second; | |
472 if (store_) | |
473 store_->DeleteServerBoundCert(*cert); | |
474 server_bound_certs_.erase(it); | |
475 delete cert; | |
476 } | |
477 | |
478 void DefaultServerBoundCertStore::InternalInsertServerBoundCert( | |
479 const std::string& server_identifier, | |
480 ServerBoundCert* cert) { | |
481 DCHECK(CalledOnValidThread()); | |
482 DCHECK(loaded_); | |
483 | |
484 if (store_) | |
485 store_->AddServerBoundCert(*cert); | |
486 server_bound_certs_[server_identifier] = cert; | |
487 } | |
488 | |
489 DefaultServerBoundCertStore::PersistentStore::PersistentStore() {} | |
490 | |
491 DefaultServerBoundCertStore::PersistentStore::~PersistentStore() {} | |
492 | |
493 } // namespace net | |
OLD | NEW |