| 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 bool DefaultServerBoundCertStore::GetServerBoundCert( | |
| 240 const std::string& server_identifier, | |
| 241 SSLClientCertType* type, | |
| 242 base::Time* expiration_time, | |
| 243 std::string* private_key_result, | |
| 244 std::string* cert_result, | |
| 245 const GetCertCallback& callback) { | |
| 246 DCHECK(CalledOnValidThread()); | |
| 247 InitIfNecessary(); | |
| 248 | |
| 249 if (!loaded_) { | |
| 250 EnqueueTask(scoped_ptr<Task>( | |
| 251 new GetServerBoundCertTask(server_identifier, callback))); | |
| 252 return false; | |
| 253 } | |
| 254 | |
| 255 ServerBoundCertMap::iterator it = server_bound_certs_.find(server_identifier); | |
| 256 | |
| 257 if (it == server_bound_certs_.end()) { | |
| 258 *type = CLIENT_CERT_INVALID_TYPE; | |
| 259 return true; | |
| 260 } | |
| 261 | |
| 262 ServerBoundCert* cert = it->second; | |
| 263 *type = cert->type(); | |
| 264 *expiration_time = cert->expiration_time(); | |
| 265 *private_key_result = cert->private_key(); | |
| 266 *cert_result = cert->cert(); | |
| 267 | |
| 268 return true; | |
| 269 } | |
| 270 | |
| 271 void DefaultServerBoundCertStore::SetServerBoundCert( | |
| 272 const std::string& server_identifier, | |
| 273 SSLClientCertType type, | |
| 274 base::Time creation_time, | |
| 275 base::Time expiration_time, | |
| 276 const std::string& private_key, | |
| 277 const std::string& cert) { | |
| 278 RunOrEnqueueTask(scoped_ptr<Task>(new SetServerBoundCertTask( | |
| 279 server_identifier, type, creation_time, expiration_time, private_key, | |
| 280 cert))); | |
| 281 } | |
| 282 | |
| 283 void DefaultServerBoundCertStore::DeleteServerBoundCert( | |
| 284 const std::string& server_identifier, | |
| 285 const base::Closure& callback) { | |
| 286 RunOrEnqueueTask(scoped_ptr<Task>( | |
| 287 new DeleteServerBoundCertTask(server_identifier, callback))); | |
| 288 } | |
| 289 | |
| 290 void DefaultServerBoundCertStore::DeleteAllCreatedBetween( | |
| 291 base::Time delete_begin, | |
| 292 base::Time delete_end, | |
| 293 const base::Closure& callback) { | |
| 294 RunOrEnqueueTask(scoped_ptr<Task>( | |
| 295 new DeleteAllCreatedBetweenTask(delete_begin, delete_end, callback))); | |
| 296 } | |
| 297 | |
| 298 void DefaultServerBoundCertStore::DeleteAll( | |
| 299 const base::Closure& callback) { | |
| 300 DeleteAllCreatedBetween(base::Time(), base::Time(), callback); | |
| 301 } | |
| 302 | |
| 303 void DefaultServerBoundCertStore::GetAllServerBoundCerts( | |
| 304 const GetCertListCallback& callback) { | |
| 305 RunOrEnqueueTask(scoped_ptr<Task>(new GetAllServerBoundCertsTask(callback))); | |
| 306 } | |
| 307 | |
| 308 int DefaultServerBoundCertStore::GetCertCount() { | |
| 309 DCHECK(CalledOnValidThread()); | |
| 310 | |
| 311 return server_bound_certs_.size(); | |
| 312 } | |
| 313 | |
| 314 void DefaultServerBoundCertStore::SetForceKeepSessionState() { | |
| 315 DCHECK(CalledOnValidThread()); | |
| 316 InitIfNecessary(); | |
| 317 | |
| 318 if (store_) | |
| 319 store_->SetForceKeepSessionState(); | |
| 320 } | |
| 321 | |
| 322 DefaultServerBoundCertStore::~DefaultServerBoundCertStore() { | |
| 323 DeleteAllInMemory(); | |
| 324 } | |
| 325 | |
| 326 void DefaultServerBoundCertStore::DeleteAllInMemory() { | |
| 327 DCHECK(CalledOnValidThread()); | |
| 328 | |
| 329 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
| 330 it != server_bound_certs_.end(); ++it) { | |
| 331 delete it->second; | |
| 332 } | |
| 333 server_bound_certs_.clear(); | |
| 334 } | |
| 335 | |
| 336 void DefaultServerBoundCertStore::InitStore() { | |
| 337 DCHECK(CalledOnValidThread()); | |
| 338 DCHECK(store_) << "Store must exist to initialize"; | |
| 339 DCHECK(!loaded_); | |
| 340 | |
| 341 store_->Load(base::Bind(&DefaultServerBoundCertStore::OnLoaded, | |
| 342 weak_ptr_factory_.GetWeakPtr())); | |
| 343 } | |
| 344 | |
| 345 void DefaultServerBoundCertStore::OnLoaded( | |
| 346 scoped_ptr<ScopedVector<ServerBoundCert> > certs) { | |
| 347 DCHECK(CalledOnValidThread()); | |
| 348 | |
| 349 for (std::vector<ServerBoundCert*>::const_iterator it = certs->begin(); | |
| 350 it != certs->end(); ++it) { | |
| 351 DCHECK(server_bound_certs_.find((*it)->server_identifier()) == | |
| 352 server_bound_certs_.end()); | |
| 353 server_bound_certs_[(*it)->server_identifier()] = *it; | |
| 354 } | |
| 355 certs->weak_clear(); | |
| 356 | |
| 357 loaded_ = true; | |
| 358 | |
| 359 base::TimeDelta wait_time; | |
| 360 if (!waiting_tasks_.empty()) | |
| 361 wait_time = base::TimeTicks::Now() - waiting_tasks_start_time_; | |
| 362 DVLOG(1) << "Task delay " << wait_time.InMilliseconds(); | |
| 363 UMA_HISTOGRAM_CUSTOM_TIMES("DomainBoundCerts.TaskMaxWaitTime", | |
| 364 wait_time, | |
| 365 base::TimeDelta::FromMilliseconds(1), | |
| 366 base::TimeDelta::FromMinutes(1), | |
| 367 50); | |
| 368 UMA_HISTOGRAM_COUNTS_100("DomainBoundCerts.TaskWaitCount", | |
| 369 waiting_tasks_.size()); | |
| 370 | |
| 371 | |
| 372 for (ScopedVector<Task>::iterator i = waiting_tasks_.begin(); | |
| 373 i != waiting_tasks_.end(); ++i) | |
| 374 (*i)->Run(this); | |
| 375 waiting_tasks_.clear(); | |
| 376 } | |
| 377 | |
| 378 void DefaultServerBoundCertStore::SyncSetServerBoundCert( | |
| 379 const std::string& server_identifier, | |
| 380 SSLClientCertType type, | |
| 381 base::Time creation_time, | |
| 382 base::Time expiration_time, | |
| 383 const std::string& private_key, | |
| 384 const std::string& cert) { | |
| 385 DCHECK(CalledOnValidThread()); | |
| 386 DCHECK(loaded_); | |
| 387 | |
| 388 InternalDeleteServerBoundCert(server_identifier); | |
| 389 InternalInsertServerBoundCert( | |
| 390 server_identifier, | |
| 391 new ServerBoundCert( | |
| 392 server_identifier, type, creation_time, expiration_time, private_key, | |
| 393 cert)); | |
| 394 } | |
| 395 | |
| 396 void DefaultServerBoundCertStore::SyncDeleteServerBoundCert( | |
| 397 const std::string& server_identifier) { | |
| 398 DCHECK(CalledOnValidThread()); | |
| 399 DCHECK(loaded_); | |
| 400 InternalDeleteServerBoundCert(server_identifier); | |
| 401 } | |
| 402 | |
| 403 void DefaultServerBoundCertStore::SyncDeleteAllCreatedBetween( | |
| 404 base::Time delete_begin, | |
| 405 base::Time delete_end) { | |
| 406 DCHECK(CalledOnValidThread()); | |
| 407 DCHECK(loaded_); | |
| 408 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
| 409 it != server_bound_certs_.end();) { | |
| 410 ServerBoundCertMap::iterator cur = it; | |
| 411 ++it; | |
| 412 ServerBoundCert* cert = cur->second; | |
| 413 if ((delete_begin.is_null() || cert->creation_time() >= delete_begin) && | |
| 414 (delete_end.is_null() || cert->creation_time() < delete_end)) { | |
| 415 if (store_) | |
| 416 store_->DeleteServerBoundCert(*cert); | |
| 417 delete cert; | |
| 418 server_bound_certs_.erase(cur); | |
| 419 } | |
| 420 } | |
| 421 } | |
| 422 | |
| 423 void DefaultServerBoundCertStore::SyncGetAllServerBoundCerts( | |
| 424 ServerBoundCertList* cert_list) { | |
| 425 DCHECK(CalledOnValidThread()); | |
| 426 DCHECK(loaded_); | |
| 427 for (ServerBoundCertMap::iterator it = server_bound_certs_.begin(); | |
| 428 it != server_bound_certs_.end(); ++it) | |
| 429 cert_list->push_back(*it->second); | |
| 430 } | |
| 431 | |
| 432 void DefaultServerBoundCertStore::EnqueueTask(scoped_ptr<Task> task) { | |
| 433 DCHECK(CalledOnValidThread()); | |
| 434 DCHECK(!loaded_); | |
| 435 if (waiting_tasks_.empty()) | |
| 436 waiting_tasks_start_time_ = base::TimeTicks::Now(); | |
| 437 waiting_tasks_.push_back(task.release()); | |
| 438 } | |
| 439 | |
| 440 void DefaultServerBoundCertStore::RunOrEnqueueTask(scoped_ptr<Task> task) { | |
| 441 DCHECK(CalledOnValidThread()); | |
| 442 InitIfNecessary(); | |
| 443 | |
| 444 if (!loaded_) { | |
| 445 EnqueueTask(task.Pass()); | |
| 446 return; | |
| 447 } | |
| 448 | |
| 449 task->Run(this); | |
| 450 } | |
| 451 | |
| 452 void DefaultServerBoundCertStore::InternalDeleteServerBoundCert( | |
| 453 const std::string& server_identifier) { | |
| 454 DCHECK(CalledOnValidThread()); | |
| 455 DCHECK(loaded_); | |
| 456 | |
| 457 ServerBoundCertMap::iterator it = server_bound_certs_.find(server_identifier); | |
| 458 if (it == server_bound_certs_.end()) | |
| 459 return; // There is nothing to delete. | |
| 460 | |
| 461 ServerBoundCert* cert = it->second; | |
| 462 if (store_) | |
| 463 store_->DeleteServerBoundCert(*cert); | |
| 464 server_bound_certs_.erase(it); | |
| 465 delete cert; | |
| 466 } | |
| 467 | |
| 468 void DefaultServerBoundCertStore::InternalInsertServerBoundCert( | |
| 469 const std::string& server_identifier, | |
| 470 ServerBoundCert* cert) { | |
| 471 DCHECK(CalledOnValidThread()); | |
| 472 DCHECK(loaded_); | |
| 473 | |
| 474 if (store_) | |
| 475 store_->AddServerBoundCert(*cert); | |
| 476 server_bound_certs_[server_identifier] = cert; | |
| 477 } | |
| 478 | |
| 479 DefaultServerBoundCertStore::PersistentStore::PersistentStore() {} | |
| 480 | |
| 481 DefaultServerBoundCertStore::PersistentStore::~PersistentStore() {} | |
| 482 | |
| 483 } // namespace net | |
| OLD | NEW |