| 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 #include "chrome/browser/net/sqlite_server_bound_cert_store.h" | 5 #include "net/util/sqlite/sqlite_server_bound_cert_store.h" |
| 6 | 6 |
| 7 #include <list> | 7 #include <list> |
| 8 #include <set> | 8 #include <set> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/file_util.h" | 12 #include "base/file_util.h" |
| 13 #include "base/files/file_path.h" | 13 #include "base/files/file_path.h" |
| 14 #include "base/location.h" |
| 14 #include "base/logging.h" | 15 #include "base/logging.h" |
| 15 #include "base/memory/scoped_ptr.h" | 16 #include "base/memory/scoped_ptr.h" |
| 16 #include "base/metrics/histogram.h" | 17 #include "base/metrics/histogram.h" |
| 18 #include "base/sequenced_task_runner.h" |
| 17 #include "base/strings/string_util.h" | 19 #include "base/strings/string_util.h" |
| 18 #include "base/threading/thread.h" | |
| 19 #include "base/threading/thread_restrictions.h" | |
| 20 #include "net/cert/x509_certificate.h" | 20 #include "net/cert/x509_certificate.h" |
| 21 #include "net/cookies/cookie_util.h" | 21 #include "net/cookies/cookie_util.h" |
| 22 #include "net/ssl/ssl_client_cert_type.h" | 22 #include "net/ssl/ssl_client_cert_type.h" |
| 23 #include "sql/error_delegate_util.h" | 23 #include "sql/error_delegate_util.h" |
| 24 #include "sql/meta_table.h" | 24 #include "sql/meta_table.h" |
| 25 #include "sql/statement.h" | 25 #include "sql/statement.h" |
| 26 #include "sql/transaction.h" | 26 #include "sql/transaction.h" |
| 27 #include "third_party/sqlite/sqlite3.h" | |
| 28 #include "url/gurl.h" | 27 #include "url/gurl.h" |
| 29 #include "webkit/browser/quota/special_storage_policy.h" | 28 |
| 29 namespace net { |
| 30 | 30 |
| 31 // This class is designed to be shared between any calling threads and the | 31 // This class is designed to be shared between any calling threads and the |
| 32 // background task runner. It batches operations and commits them on a timer. | 32 // background task runner. It batches operations and commits them on a timer. |
| 33 class SQLiteServerBoundCertStore::Backend | 33 class SQLiteServerBoundCertStore::Backend |
| 34 : public base::RefCountedThreadSafe<SQLiteServerBoundCertStore::Backend> { | 34 : public base::RefCountedThreadSafe<SQLiteServerBoundCertStore::Backend> { |
| 35 public: | 35 public: |
| 36 Backend( | 36 Backend( |
| 37 const base::FilePath& path, | 37 const base::FilePath& path, |
| 38 const scoped_refptr<base::SequencedTaskRunner>& background_task_runner, | 38 const scoped_refptr<base::SequencedTaskRunner>& background_task_runner) |
| 39 quota::SpecialStoragePolicy* special_storage_policy) | |
| 40 : path_(path), | 39 : path_(path), |
| 41 num_pending_(0), | 40 num_pending_(0), |
| 42 force_keep_session_state_(false), | 41 force_keep_session_state_(false), |
| 43 background_task_runner_(background_task_runner), | 42 background_task_runner_(background_task_runner), |
| 44 special_storage_policy_(special_storage_policy), | |
| 45 corruption_detected_(false) {} | 43 corruption_detected_(false) {} |
| 46 | 44 |
| 47 // Creates or loads the SQLite database. | 45 // Creates or loads the SQLite database. |
| 48 void Load(const LoadedCallback& loaded_callback); | 46 void Load(const LoadedCallback& loaded_callback); |
| 49 | 47 |
| 50 // Batch a server bound cert addition. | 48 // Batch a server bound cert addition. |
| 51 void AddServerBoundCert( | 49 void AddServerBoundCert( |
| 52 const net::DefaultServerBoundCertStore::ServerBoundCert& cert); | 50 const DefaultServerBoundCertStore::ServerBoundCert& cert); |
| 53 | 51 |
| 54 // Batch a server bound cert deletion. | 52 // Batch a server bound cert deletion. |
| 55 void DeleteServerBoundCert( | 53 void DeleteServerBoundCert( |
| 56 const net::DefaultServerBoundCertStore::ServerBoundCert& cert); | 54 const DefaultServerBoundCertStore::ServerBoundCert& cert); |
| 57 | 55 |
| 58 // Commit any pending operations and close the database. This must be called | 56 // Commit any pending operations and close the database. This must be called |
| 59 // before the object is destructed. | 57 // before the object is destructed. |
| 60 void Close(); | 58 void Close(); |
| 61 | 59 |
| 62 void SetForceKeepSessionState(); | 60 void SetForceKeepSessionState(); |
| 63 | 61 |
| 64 private: | 62 private: |
| 65 void LoadOnDBThread( | 63 void LoadInBackground( |
| 66 ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert>* certs); | 64 ScopedVector<DefaultServerBoundCertStore::ServerBoundCert>* certs); |
| 67 | 65 |
| 68 friend class base::RefCountedThreadSafe<SQLiteServerBoundCertStore::Backend>; | 66 friend class base::RefCountedThreadSafe<SQLiteServerBoundCertStore::Backend>; |
| 69 | 67 |
| 70 // You should call Close() before destructing this object. | 68 // You should call Close() before destructing this object. |
| 71 ~Backend() { | 69 ~Backend() { |
| 72 DCHECK(!db_.get()) << "Close should have already been called."; | 70 DCHECK(!db_.get()) << "Close should have already been called."; |
| 73 DCHECK(num_pending_ == 0 && pending_.empty()); | 71 DCHECK(num_pending_ == 0 && pending_.empty()); |
| 74 } | 72 } |
| 75 | 73 |
| 76 // Database upgrade statements. | 74 // Database upgrade statements. |
| 77 bool EnsureDatabaseVersion(); | 75 bool EnsureDatabaseVersion(); |
| 78 | 76 |
| 79 class PendingOperation { | 77 class PendingOperation { |
| 80 public: | 78 public: |
| 81 typedef enum { | 79 typedef enum { CERT_ADD, CERT_DELETE } OperationType; |
| 82 CERT_ADD, | |
| 83 CERT_DELETE | |
| 84 } OperationType; | |
| 85 | 80 |
| 86 PendingOperation( | 81 PendingOperation(OperationType op, |
| 87 OperationType op, | 82 const DefaultServerBoundCertStore::ServerBoundCert& cert) |
| 88 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) | |
| 89 : op_(op), cert_(cert) {} | 83 : op_(op), cert_(cert) {} |
| 90 | 84 |
| 91 OperationType op() const { return op_; } | 85 OperationType op() const { return op_; } |
| 92 const net::DefaultServerBoundCertStore::ServerBoundCert& cert() const { | 86 const DefaultServerBoundCertStore::ServerBoundCert& cert() const { |
| 93 return cert_; | 87 return cert_; |
| 94 } | 88 } |
| 95 | 89 |
| 96 private: | 90 private: |
| 97 OperationType op_; | 91 OperationType op_; |
| 98 net::DefaultServerBoundCertStore::ServerBoundCert cert_; | 92 DefaultServerBoundCertStore::ServerBoundCert cert_; |
| 99 }; | 93 }; |
| 100 | 94 |
| 101 private: | 95 private: |
| 102 // Batch a server bound cert operation (add or delete). | 96 // Batch a server bound cert operation (add or delete). |
| 103 void BatchOperation( | 97 void BatchOperation(PendingOperation::OperationType op, |
| 104 PendingOperation::OperationType op, | 98 const DefaultServerBoundCertStore::ServerBoundCert& cert); |
| 105 const net::DefaultServerBoundCertStore::ServerBoundCert& cert); | |
| 106 // Commit our pending operations to the database. | 99 // Commit our pending operations to the database. |
| 107 void Commit(); | 100 void Commit(); |
| 108 // Close() executed on the background thread. | 101 // Close() executed on the background task runner. |
| 109 void InternalBackgroundClose(); | 102 void InternalBackgroundClose(); |
| 110 | 103 |
| 111 void DeleteCertificatesOnShutdown(); | 104 void DeleteCertificatesOnShutdown(); |
| 112 | 105 |
| 113 void DatabaseErrorCallback(int error, sql::Statement* stmt); | 106 void DatabaseErrorCallback(int error, sql::Statement* stmt); |
| 114 void KillDatabase(); | 107 void KillDatabase(); |
| 115 | 108 |
| 116 base::FilePath path_; | 109 base::FilePath path_; |
| 117 scoped_ptr<sql::Connection> db_; | 110 scoped_ptr<sql::Connection> db_; |
| 118 sql::MetaTable meta_table_; | 111 sql::MetaTable meta_table_; |
| 119 | 112 |
| 120 typedef std::list<PendingOperation*> PendingOperationsList; | 113 typedef std::list<PendingOperation*> PendingOperationsList; |
| 121 PendingOperationsList pending_; | 114 PendingOperationsList pending_; |
| 122 PendingOperationsList::size_type num_pending_; | 115 PendingOperationsList::size_type num_pending_; |
| 123 // True if the persistent store should skip clear on exit rules. | 116 // True if the persistent store should skip clear on exit rules. |
| 124 bool force_keep_session_state_; | 117 bool force_keep_session_state_; |
| 125 // Guard |pending_|, |num_pending_| and |force_keep_session_state_|. | 118 // Guard |pending_|, |num_pending_| and |force_keep_session_state_|. |
| 126 base::Lock lock_; | 119 base::Lock lock_; |
| 127 | 120 |
| 128 // Cache of origins we have certificates stored for. | 121 // Cache of origins we have certificates stored for. |
| 129 std::set<std::string> cert_origins_; | 122 std::set<std::string> cert_origins_; |
| 130 | 123 |
| 131 scoped_refptr<base::SequencedTaskRunner> background_task_runner_; | 124 scoped_refptr<base::SequencedTaskRunner> background_task_runner_; |
| 132 | 125 |
| 133 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_; | |
| 134 | |
| 135 // Indicates if the kill-database callback has been scheduled. | 126 // Indicates if the kill-database callback has been scheduled. |
| 136 bool corruption_detected_; | 127 bool corruption_detected_; |
| 137 | 128 |
| 138 DISALLOW_COPY_AND_ASSIGN(Backend); | 129 DISALLOW_COPY_AND_ASSIGN(Backend); |
| 139 }; | 130 }; |
| 140 | 131 |
| 141 // Version number of the database. | 132 // Version number of the database. |
| 142 static const int kCurrentVersionNumber = 4; | 133 static const int kCurrentVersionNumber = 4; |
| 143 static const int kCompatibleVersionNumber = 1; | 134 static const int kCompatibleVersionNumber = 1; |
| 144 | 135 |
| 145 namespace { | 136 namespace { |
| 146 | 137 |
| 147 // Initializes the certs table, returning true on success. | 138 // Initializes the certs table, returning true on success. |
| 148 bool InitTable(sql::Connection* db) { | 139 bool InitTable(sql::Connection* db) { |
| 149 // The table is named "origin_bound_certs" for backwards compatability before | 140 // The table is named "origin_bound_certs" for backwards compatability before |
| 150 // we renamed this class to SQLiteServerBoundCertStore. Likewise, the primary | 141 // we renamed this class to SQLiteServerBoundCertStore. Likewise, the primary |
| 151 // key is "origin", but now can be other things like a plain domain. | 142 // key is "origin", but now can be other things like a plain domain. |
| 152 if (!db->DoesTableExist("origin_bound_certs")) { | 143 if (!db->DoesTableExist("origin_bound_certs")) { |
| 153 if (!db->Execute("CREATE TABLE origin_bound_certs (" | 144 if (!db->Execute( |
| 154 "origin TEXT NOT NULL UNIQUE PRIMARY KEY," | 145 "CREATE TABLE origin_bound_certs (" |
| 155 "private_key BLOB NOT NULL," | 146 "origin TEXT NOT NULL UNIQUE PRIMARY KEY," |
| 156 "cert BLOB NOT NULL," | 147 "private_key BLOB NOT NULL," |
| 157 "cert_type INTEGER," | 148 "cert BLOB NOT NULL," |
| 158 "expiration_time INTEGER," | 149 "cert_type INTEGER," |
| 159 "creation_time INTEGER)")) | 150 "expiration_time INTEGER," |
| 151 "creation_time INTEGER)")) |
| 160 return false; | 152 return false; |
| 161 } | 153 } |
| 162 | 154 |
| 163 return true; | 155 return true; |
| 164 } | 156 } |
| 165 | 157 |
| 166 } // namespace | 158 } // namespace |
| 167 | 159 |
| 168 void SQLiteServerBoundCertStore::Backend::Load( | 160 void SQLiteServerBoundCertStore::Backend::Load( |
| 169 const LoadedCallback& loaded_callback) { | 161 const LoadedCallback& loaded_callback) { |
| 170 // This function should be called only once per instance. | 162 // This function should be called only once per instance. |
| 171 DCHECK(!db_.get()); | 163 DCHECK(!db_.get()); |
| 172 scoped_ptr<ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert> > | 164 scoped_ptr<ScopedVector<DefaultServerBoundCertStore::ServerBoundCert> > certs( |
| 173 certs(new ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert>( | 165 new ScopedVector<DefaultServerBoundCertStore::ServerBoundCert>()); |
| 174 )); | 166 ScopedVector<DefaultServerBoundCertStore::ServerBoundCert>* certs_ptr = |
| 175 ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert>* certs_ptr = | |
| 176 certs.get(); | 167 certs.get(); |
| 177 | 168 |
| 178 background_task_runner_->PostTaskAndReply( | 169 background_task_runner_->PostTaskAndReply( |
| 179 FROM_HERE, | 170 FROM_HERE, |
| 180 base::Bind(&Backend::LoadOnDBThread, this, certs_ptr), | 171 base::Bind(&Backend::LoadInBackground, this, certs_ptr), |
| 181 base::Bind(loaded_callback, base::Passed(&certs))); | 172 base::Bind(loaded_callback, base::Passed(&certs))); |
| 182 } | 173 } |
| 183 | 174 |
| 184 void SQLiteServerBoundCertStore::Backend::LoadOnDBThread( | 175 void SQLiteServerBoundCertStore::Backend::LoadInBackground( |
| 185 ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert>* certs) { | 176 ScopedVector<DefaultServerBoundCertStore::ServerBoundCert>* certs) { |
| 186 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); | 177 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); |
| 187 | 178 |
| 188 // This method should be called only once per instance. | 179 // This method should be called only once per instance. |
| 189 DCHECK(!db_.get()); | 180 DCHECK(!db_.get()); |
| 190 | 181 |
| 191 base::TimeTicks start = base::TimeTicks::Now(); | 182 base::TimeTicks start = base::TimeTicks::Now(); |
| 192 | 183 |
| 193 // Ensure the parent directory for storing certs is created before reading | 184 // Ensure the parent directory for storing certs is created before reading |
| 194 // from it. | 185 // from it. |
| 195 const base::FilePath dir = path_.DirName(); | 186 const base::FilePath dir = path_.DirName(); |
| 196 if (!base::PathExists(dir) && !base::CreateDirectory(dir)) | 187 if (!base::PathExists(dir) && !base::CreateDirectory(dir)) |
| 197 return; | 188 return; |
| 198 | 189 |
| 199 int64 db_size = 0; | 190 int64 db_size = 0; |
| 200 if (base::GetFileSize(path_, &db_size)) | 191 if (base::GetFileSize(path_, &db_size)) |
| 201 UMA_HISTOGRAM_COUNTS("DomainBoundCerts.DBSizeInKB", db_size / 1024 ); | 192 UMA_HISTOGRAM_COUNTS("DomainBoundCerts.DBSizeInKB", db_size / 1024); |
| 202 | 193 |
| 203 db_.reset(new sql::Connection); | 194 db_.reset(new sql::Connection); |
| 204 db_->set_histogram_tag("DomainBoundCerts"); | 195 db_->set_histogram_tag("DomainBoundCerts"); |
| 205 | 196 |
| 206 // Unretained to avoid a ref loop with db_. | 197 // Unretained to avoid a ref loop with db_. |
| 207 db_->set_error_callback( | 198 db_->set_error_callback( |
| 208 base::Bind(&SQLiteServerBoundCertStore::Backend::DatabaseErrorCallback, | 199 base::Bind(&SQLiteServerBoundCertStore::Backend::DatabaseErrorCallback, |
| 209 base::Unretained(this))); | 200 base::Unretained(this))); |
| 210 | 201 |
| 211 if (!db_->Open(path_)) { | 202 if (!db_->Open(path_)) { |
| (...skipping 21 matching lines...) Expand all Loading... |
| 233 "creation_time FROM origin_bound_certs")); | 224 "creation_time FROM origin_bound_certs")); |
| 234 if (!smt.is_valid()) { | 225 if (!smt.is_valid()) { |
| 235 if (corruption_detected_) | 226 if (corruption_detected_) |
| 236 KillDatabase(); | 227 KillDatabase(); |
| 237 meta_table_.Reset(); | 228 meta_table_.Reset(); |
| 238 db_.reset(); | 229 db_.reset(); |
| 239 return; | 230 return; |
| 240 } | 231 } |
| 241 | 232 |
| 242 while (smt.Step()) { | 233 while (smt.Step()) { |
| 243 net::SSLClientCertType type = | 234 SSLClientCertType type = static_cast<SSLClientCertType>(smt.ColumnInt(3)); |
| 244 static_cast<net::SSLClientCertType>(smt.ColumnInt(3)); | 235 if (type != CLIENT_CERT_ECDSA_SIGN) |
| 245 if (type != net::CLIENT_CERT_ECDSA_SIGN) | |
| 246 continue; | 236 continue; |
| 247 std::string private_key_from_db, cert_from_db; | 237 std::string private_key_from_db, cert_from_db; |
| 248 smt.ColumnBlobAsString(1, &private_key_from_db); | 238 smt.ColumnBlobAsString(1, &private_key_from_db); |
| 249 smt.ColumnBlobAsString(2, &cert_from_db); | 239 smt.ColumnBlobAsString(2, &cert_from_db); |
| 250 scoped_ptr<net::DefaultServerBoundCertStore::ServerBoundCert> cert( | 240 scoped_ptr<DefaultServerBoundCertStore::ServerBoundCert> cert( |
| 251 new net::DefaultServerBoundCertStore::ServerBoundCert( | 241 new DefaultServerBoundCertStore::ServerBoundCert( |
| 252 smt.ColumnString(0), // origin | 242 smt.ColumnString(0), // origin |
| 253 base::Time::FromInternalValue(smt.ColumnInt64(5)), | 243 base::Time::FromInternalValue(smt.ColumnInt64(5)), |
| 254 base::Time::FromInternalValue(smt.ColumnInt64(4)), | 244 base::Time::FromInternalValue(smt.ColumnInt64(4)), |
| 255 private_key_from_db, | 245 private_key_from_db, |
| 256 cert_from_db)); | 246 cert_from_db)); |
| 257 cert_origins_.insert(cert->server_identifier()); | 247 cert_origins_.insert(cert->server_identifier()); |
| 258 certs->push_back(cert.release()); | 248 certs->push_back(cert.release()); |
| 259 } | 249 } |
| 260 | 250 |
| 261 UMA_HISTOGRAM_COUNTS_10000("DomainBoundCerts.DBLoadedCount", certs->size()); | 251 UMA_HISTOGRAM_COUNTS_10000("DomainBoundCerts.DBLoadedCount", certs->size()); |
| 262 base::TimeDelta load_time = base::TimeTicks::Now() - start; | 252 base::TimeDelta load_time = base::TimeTicks::Now() - start; |
| 263 UMA_HISTOGRAM_CUSTOM_TIMES("DomainBoundCerts.DBLoadTime", | 253 UMA_HISTOGRAM_CUSTOM_TIMES("DomainBoundCerts.DBLoadTime", |
| 264 load_time, | 254 load_time, |
| 265 base::TimeDelta::FromMilliseconds(1), | 255 base::TimeDelta::FromMilliseconds(1), |
| 266 base::TimeDelta::FromMinutes(1), | 256 base::TimeDelta::FromMinutes(1), |
| 267 50); | 257 50); |
| 268 DVLOG(1) << "loaded " << certs->size() << " in " << load_time.InMilliseconds() | 258 DVLOG(1) << "loaded " << certs->size() << " in " << load_time.InMilliseconds() |
| 269 << " ms"; | 259 << " ms"; |
| 270 } | 260 } |
| 271 | 261 |
| 272 bool SQLiteServerBoundCertStore::Backend::EnsureDatabaseVersion() { | 262 bool SQLiteServerBoundCertStore::Backend::EnsureDatabaseVersion() { |
| 273 // Version check. | 263 // Version check. |
| 274 if (!meta_table_.Init( | 264 if (!meta_table_.Init( |
| 275 db_.get(), kCurrentVersionNumber, kCompatibleVersionNumber)) { | 265 db_.get(), kCurrentVersionNumber, kCompatibleVersionNumber)) { |
| 276 return false; | 266 return false; |
| 277 } | 267 } |
| 278 | 268 |
| 279 if (meta_table_.GetCompatibleVersionNumber() > kCurrentVersionNumber) { | 269 if (meta_table_.GetCompatibleVersionNumber() > kCurrentVersionNumber) { |
| 280 LOG(WARNING) << "Server bound cert database is too new."; | 270 LOG(WARNING) << "Server bound cert database is too new."; |
| 281 return false; | 271 return false; |
| 282 } | 272 } |
| 283 | 273 |
| 284 int cur_version = meta_table_.GetVersionNumber(); | 274 int cur_version = meta_table_.GetVersionNumber(); |
| 285 if (cur_version == 1) { | 275 if (cur_version == 1) { |
| 286 sql::Transaction transaction(db_.get()); | 276 sql::Transaction transaction(db_.get()); |
| 287 if (!transaction.Begin()) | 277 if (!transaction.Begin()) |
| 288 return false; | 278 return false; |
| 289 if (!db_->Execute("ALTER TABLE origin_bound_certs ADD COLUMN cert_type " | 279 if (!db_->Execute( |
| 290 "INTEGER")) { | 280 "ALTER TABLE origin_bound_certs ADD COLUMN cert_type " |
| 281 "INTEGER")) { |
| 291 LOG(WARNING) << "Unable to update server bound cert database to " | 282 LOG(WARNING) << "Unable to update server bound cert database to " |
| 292 << "version 2."; | 283 << "version 2."; |
| 293 return false; | 284 return false; |
| 294 } | 285 } |
| 295 // All certs in version 1 database are rsa_sign, which are unsupported. | 286 // All certs in version 1 database are rsa_sign, which are unsupported. |
| 296 // Just discard them all. | 287 // Just discard them all. |
| 297 if (!db_->Execute("DELETE from origin_bound_certs")) { | 288 if (!db_->Execute("DELETE from origin_bound_certs")) { |
| 298 LOG(WARNING) << "Unable to update server bound cert database to " | 289 LOG(WARNING) << "Unable to update server bound cert database to " |
| 299 << "version 2."; | 290 << "version 2."; |
| 300 return false; | 291 return false; |
| 301 } | 292 } |
| 302 ++cur_version; | 293 ++cur_version; |
| 303 meta_table_.SetVersionNumber(cur_version); | 294 meta_table_.SetVersionNumber(cur_version); |
| 304 meta_table_.SetCompatibleVersionNumber( | 295 meta_table_.SetCompatibleVersionNumber( |
| 305 std::min(cur_version, kCompatibleVersionNumber)); | 296 std::min(cur_version, kCompatibleVersionNumber)); |
| 306 transaction.Commit(); | 297 transaction.Commit(); |
| 307 } | 298 } |
| 308 | 299 |
| 309 if (cur_version <= 3) { | 300 if (cur_version <= 3) { |
| 310 sql::Transaction transaction(db_.get()); | 301 sql::Transaction transaction(db_.get()); |
| 311 if (!transaction.Begin()) | 302 if (!transaction.Begin()) |
| 312 return false; | 303 return false; |
| 313 | 304 |
| 314 if (cur_version == 2) { | 305 if (cur_version == 2) { |
| 315 if (!db_->Execute("ALTER TABLE origin_bound_certs ADD COLUMN " | 306 if (!db_->Execute( |
| 316 "expiration_time INTEGER")) { | 307 "ALTER TABLE origin_bound_certs ADD COLUMN " |
| 308 "expiration_time INTEGER")) { |
| 317 LOG(WARNING) << "Unable to update server bound cert database to " | 309 LOG(WARNING) << "Unable to update server bound cert database to " |
| 318 << "version 4."; | 310 << "version 4."; |
| 319 return false; | 311 return false; |
| 320 } | 312 } |
| 321 } | 313 } |
| 322 | 314 |
| 323 if (!db_->Execute("ALTER TABLE origin_bound_certs ADD COLUMN " | 315 if (!db_->Execute( |
| 324 "creation_time INTEGER")) { | 316 "ALTER TABLE origin_bound_certs ADD COLUMN " |
| 317 "creation_time INTEGER")) { |
| 325 LOG(WARNING) << "Unable to update server bound cert database to " | 318 LOG(WARNING) << "Unable to update server bound cert database to " |
| 326 << "version 4."; | 319 << "version 4."; |
| 327 return false; | 320 return false; |
| 328 } | 321 } |
| 329 | 322 |
| 330 sql::Statement smt(db_->GetUniqueStatement( | 323 sql::Statement smt( |
| 331 "SELECT origin, cert FROM origin_bound_certs")); | 324 db_->GetUniqueStatement("SELECT origin, cert FROM origin_bound_certs")); |
| 332 sql::Statement update_expires_smt(db_->GetUniqueStatement( | 325 sql::Statement update_expires_smt(db_->GetUniqueStatement( |
| 333 "UPDATE origin_bound_certs SET expiration_time = ? WHERE origin = ?")); | 326 "UPDATE origin_bound_certs SET expiration_time = ? WHERE origin = ?")); |
| 334 sql::Statement update_creation_smt(db_->GetUniqueStatement( | 327 sql::Statement update_creation_smt(db_->GetUniqueStatement( |
| 335 "UPDATE origin_bound_certs SET creation_time = ? WHERE origin = ?")); | 328 "UPDATE origin_bound_certs SET creation_time = ? WHERE origin = ?")); |
| 336 if (!smt.is_valid() || | 329 if (!smt.is_valid() || !update_expires_smt.is_valid() || |
| 337 !update_expires_smt.is_valid() || | |
| 338 !update_creation_smt.is_valid()) { | 330 !update_creation_smt.is_valid()) { |
| 339 LOG(WARNING) << "Unable to update server bound cert database to " | 331 LOG(WARNING) << "Unable to update server bound cert database to " |
| 340 << "version 4."; | 332 << "version 4."; |
| 341 return false; | 333 return false; |
| 342 } | 334 } |
| 343 | 335 |
| 344 while (smt.Step()) { | 336 while (smt.Step()) { |
| 345 std::string origin = smt.ColumnString(0); | 337 std::string origin = smt.ColumnString(0); |
| 346 std::string cert_from_db; | 338 std::string cert_from_db; |
| 347 smt.ColumnBlobAsString(1, &cert_from_db); | 339 smt.ColumnBlobAsString(1, &cert_from_db); |
| 348 // Parse the cert and extract the real value and then update the DB. | 340 // Parse the cert and extract the real value and then update the DB. |
| 349 scoped_refptr<net::X509Certificate> cert( | 341 scoped_refptr<X509Certificate> cert(X509Certificate::CreateFromBytes( |
| 350 net::X509Certificate::CreateFromBytes( | 342 cert_from_db.data(), cert_from_db.size())); |
| 351 cert_from_db.data(), cert_from_db.size())); | |
| 352 if (cert.get()) { | 343 if (cert.get()) { |
| 353 if (cur_version == 2) { | 344 if (cur_version == 2) { |
| 354 update_expires_smt.Reset(true); | 345 update_expires_smt.Reset(true); |
| 355 update_expires_smt.BindInt64(0, | 346 update_expires_smt.BindInt64(0, |
| 356 cert->valid_expiry().ToInternalValue()); | 347 cert->valid_expiry().ToInternalValue()); |
| 357 update_expires_smt.BindString(1, origin); | 348 update_expires_smt.BindString(1, origin); |
| 358 if (!update_expires_smt.Run()) { | 349 if (!update_expires_smt.Run()) { |
| 359 LOG(WARNING) << "Unable to update server bound cert database to " | 350 LOG(WARNING) << "Unable to update server bound cert database to " |
| 360 << "version 4."; | 351 << "version 4."; |
| 361 return false; | 352 return false; |
| (...skipping 20 matching lines...) Expand all Loading... |
| 382 meta_table_.SetVersionNumber(cur_version); | 373 meta_table_.SetVersionNumber(cur_version); |
| 383 meta_table_.SetCompatibleVersionNumber( | 374 meta_table_.SetCompatibleVersionNumber( |
| 384 std::min(cur_version, kCompatibleVersionNumber)); | 375 std::min(cur_version, kCompatibleVersionNumber)); |
| 385 transaction.Commit(); | 376 transaction.Commit(); |
| 386 } | 377 } |
| 387 | 378 |
| 388 // Put future migration cases here. | 379 // Put future migration cases here. |
| 389 | 380 |
| 390 // When the version is too old, we just try to continue anyway, there should | 381 // When the version is too old, we just try to continue anyway, there should |
| 391 // not be a released product that makes a database too old for us to handle. | 382 // not be a released product that makes a database too old for us to handle. |
| 392 LOG_IF(WARNING, cur_version < kCurrentVersionNumber) << | 383 LOG_IF(WARNING, cur_version < kCurrentVersionNumber) |
| 393 "Server bound cert database version " << cur_version << | 384 << "Server bound cert database version " << cur_version |
| 394 " is too old to handle."; | 385 << " is too old to handle."; |
| 395 | 386 |
| 396 return true; | 387 return true; |
| 397 } | 388 } |
| 398 | 389 |
| 399 void SQLiteServerBoundCertStore::Backend::DatabaseErrorCallback( | 390 void SQLiteServerBoundCertStore::Backend::DatabaseErrorCallback( |
| 400 int error, | 391 int error, |
| 401 sql::Statement* stmt) { | 392 sql::Statement* stmt) { |
| 402 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); | 393 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); |
| 403 | 394 |
| 404 if (!sql::IsErrorCatastrophic(error)) | 395 if (!sql::IsErrorCatastrophic(error)) |
| (...skipping 20 matching lines...) Expand all Loading... |
| 425 // This Backend will now be in-memory only. In a future run the database | 416 // This Backend will now be in-memory only. In a future run the database |
| 426 // will be recreated. Hopefully things go better then! | 417 // will be recreated. Hopefully things go better then! |
| 427 bool success = db_->RazeAndClose(); | 418 bool success = db_->RazeAndClose(); |
| 428 UMA_HISTOGRAM_BOOLEAN("DomainBoundCerts.KillDatabaseResult", success); | 419 UMA_HISTOGRAM_BOOLEAN("DomainBoundCerts.KillDatabaseResult", success); |
| 429 meta_table_.Reset(); | 420 meta_table_.Reset(); |
| 430 db_.reset(); | 421 db_.reset(); |
| 431 } | 422 } |
| 432 } | 423 } |
| 433 | 424 |
| 434 void SQLiteServerBoundCertStore::Backend::AddServerBoundCert( | 425 void SQLiteServerBoundCertStore::Backend::AddServerBoundCert( |
| 435 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) { | 426 const DefaultServerBoundCertStore::ServerBoundCert& cert) { |
| 436 BatchOperation(PendingOperation::CERT_ADD, cert); | 427 BatchOperation(PendingOperation::CERT_ADD, cert); |
| 437 } | 428 } |
| 438 | 429 |
| 439 void SQLiteServerBoundCertStore::Backend::DeleteServerBoundCert( | 430 void SQLiteServerBoundCertStore::Backend::DeleteServerBoundCert( |
| 440 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) { | 431 const DefaultServerBoundCertStore::ServerBoundCert& cert) { |
| 441 BatchOperation(PendingOperation::CERT_DELETE, cert); | 432 BatchOperation(PendingOperation::CERT_DELETE, cert); |
| 442 } | 433 } |
| 443 | 434 |
| 444 void SQLiteServerBoundCertStore::Backend::BatchOperation( | 435 void SQLiteServerBoundCertStore::Backend::BatchOperation( |
| 445 PendingOperation::OperationType op, | 436 PendingOperation::OperationType op, |
| 446 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) { | 437 const DefaultServerBoundCertStore::ServerBoundCert& cert) { |
| 447 // Commit every 30 seconds. | 438 // Commit every 30 seconds. |
| 448 static const int kCommitIntervalMs = 30 * 1000; | 439 static const int kCommitIntervalMs = 30 * 1000; |
| 449 // Commit right away if we have more than 512 outstanding operations. | 440 // Commit right away if we have more than 512 outstanding operations. |
| 450 static const size_t kCommitAfterBatchSize = 512; | 441 static const size_t kCommitAfterBatchSize = 512; |
| 451 | 442 |
| 452 // We do a full copy of the cert here, and hopefully just here. | 443 // We do a full copy of the cert here, and hopefully just here. |
| 453 scoped_ptr<PendingOperation> po(new PendingOperation(op, cert)); | 444 scoped_ptr<PendingOperation> po(new PendingOperation(op, cert)); |
| 454 | 445 |
| 455 PendingOperationsList::size_type num_pending; | 446 PendingOperationsList::size_type num_pending; |
| 456 { | 447 { |
| (...skipping 22 matching lines...) Expand all Loading... |
| 479 { | 470 { |
| 480 base::AutoLock locked(lock_); | 471 base::AutoLock locked(lock_); |
| 481 pending_.swap(ops); | 472 pending_.swap(ops); |
| 482 num_pending_ = 0; | 473 num_pending_ = 0; |
| 483 } | 474 } |
| 484 | 475 |
| 485 // Maybe an old timer fired or we are already Close()'ed. | 476 // Maybe an old timer fired or we are already Close()'ed. |
| 486 if (!db_.get() || ops.empty()) | 477 if (!db_.get() || ops.empty()) |
| 487 return; | 478 return; |
| 488 | 479 |
| 489 sql::Statement add_smt(db_->GetCachedStatement(SQL_FROM_HERE, | 480 sql::Statement add_smt(db_->GetCachedStatement( |
| 481 SQL_FROM_HERE, |
| 490 "INSERT INTO origin_bound_certs (origin, private_key, cert, cert_type, " | 482 "INSERT INTO origin_bound_certs (origin, private_key, cert, cert_type, " |
| 491 "expiration_time, creation_time) VALUES (?,?,?,?,?,?)")); | 483 "expiration_time, creation_time) VALUES (?,?,?,?,?,?)")); |
| 492 if (!add_smt.is_valid()) | 484 if (!add_smt.is_valid()) |
| 493 return; | 485 return; |
| 494 | 486 |
| 495 sql::Statement del_smt(db_->GetCachedStatement(SQL_FROM_HERE, | 487 sql::Statement del_smt(db_->GetCachedStatement( |
| 496 "DELETE FROM origin_bound_certs WHERE origin=?")); | 488 SQL_FROM_HERE, "DELETE FROM origin_bound_certs WHERE origin=?")); |
| 497 if (!del_smt.is_valid()) | 489 if (!del_smt.is_valid()) |
| 498 return; | 490 return; |
| 499 | 491 |
| 500 sql::Transaction transaction(db_.get()); | 492 sql::Transaction transaction(db_.get()); |
| 501 if (!transaction.Begin()) | 493 if (!transaction.Begin()) |
| 502 return; | 494 return; |
| 503 | 495 |
| 504 for (PendingOperationsList::iterator it = ops.begin(); | 496 for (PendingOperationsList::iterator it = ops.begin(); it != ops.end(); |
| 505 it != ops.end(); ++it) { | 497 ++it) { |
| 506 // Free the certs as we commit them to the database. | 498 // Free the certs as we commit them to the database. |
| 507 scoped_ptr<PendingOperation> po(*it); | 499 scoped_ptr<PendingOperation> po(*it); |
| 508 switch (po->op()) { | 500 switch (po->op()) { |
| 509 case PendingOperation::CERT_ADD: { | 501 case PendingOperation::CERT_ADD: { |
| 510 cert_origins_.insert(po->cert().server_identifier()); | 502 cert_origins_.insert(po->cert().server_identifier()); |
| 511 add_smt.Reset(true); | 503 add_smt.Reset(true); |
| 512 add_smt.BindString(0, po->cert().server_identifier()); | 504 add_smt.BindString(0, po->cert().server_identifier()); |
| 513 const std::string& private_key = po->cert().private_key(); | 505 const std::string& private_key = po->cert().private_key(); |
| 514 add_smt.BindBlob(1, private_key.data(), private_key.size()); | 506 add_smt.BindBlob(1, private_key.data(), private_key.size()); |
| 515 const std::string& cert = po->cert().cert(); | 507 const std::string& cert = po->cert().cert(); |
| 516 add_smt.BindBlob(2, cert.data(), cert.size()); | 508 add_smt.BindBlob(2, cert.data(), cert.size()); |
| 517 add_smt.BindInt(3, net::CLIENT_CERT_ECDSA_SIGN); | 509 add_smt.BindInt(3, CLIENT_CERT_ECDSA_SIGN); |
| 518 add_smt.BindInt64(4, po->cert().expiration_time().ToInternalValue()); | 510 add_smt.BindInt64(4, po->cert().expiration_time().ToInternalValue()); |
| 519 add_smt.BindInt64(5, po->cert().creation_time().ToInternalValue()); | 511 add_smt.BindInt64(5, po->cert().creation_time().ToInternalValue()); |
| 520 if (!add_smt.Run()) | 512 if (!add_smt.Run()) |
| 521 NOTREACHED() << "Could not add a server bound cert to the DB."; | 513 NOTREACHED() << "Could not add a server bound cert to the DB."; |
| 522 break; | 514 break; |
| 523 } | 515 } |
| 524 case PendingOperation::CERT_DELETE: | 516 case PendingOperation::CERT_DELETE: |
| 525 cert_origins_.erase(po->cert().server_identifier()); | 517 cert_origins_.erase(po->cert().server_identifier()); |
| 526 del_smt.Reset(true); | 518 del_smt.Reset(true); |
| 527 del_smt.BindString(0, po->cert().server_identifier()); | 519 del_smt.BindString(0, po->cert().server_identifier()); |
| 528 if (!del_smt.Run()) | 520 if (!del_smt.Run()) |
| 529 NOTREACHED() << "Could not delete a server bound cert from the DB."; | 521 NOTREACHED() << "Could not delete a server bound cert from the DB."; |
| 530 break; | 522 break; |
| 531 | 523 |
| 532 default: | 524 default: |
| 533 NOTREACHED(); | 525 NOTREACHED(); |
| 534 break; | 526 break; |
| 535 } | 527 } |
| 536 } | 528 } |
| 537 transaction.Commit(); | 529 transaction.Commit(); |
| 538 } | 530 } |
| 539 | 531 |
| 540 // Fire off a close message to the background thread. We could still have a | 532 // Fire off a close message to the background task runner. We could still have a |
| 541 // pending commit timer that will be holding a reference on us, but if/when | 533 // pending commit timer that will be holding a reference on us, but if/when |
| 542 // this fires we will already have been cleaned up and it will be ignored. | 534 // this fires we will already have been cleaned up and it will be ignored. |
| 543 void SQLiteServerBoundCertStore::Backend::Close() { | 535 void SQLiteServerBoundCertStore::Backend::Close() { |
| 544 // Must close the backend on the background thread. | 536 // Must close the backend on the background task runner. |
| 545 background_task_runner_->PostTask( | 537 background_task_runner_->PostTask( |
| 546 FROM_HERE, base::Bind(&Backend::InternalBackgroundClose, this)); | 538 FROM_HERE, base::Bind(&Backend::InternalBackgroundClose, this)); |
| 547 } | 539 } |
| 548 | 540 |
| 549 void SQLiteServerBoundCertStore::Backend::InternalBackgroundClose() { | 541 void SQLiteServerBoundCertStore::Backend::InternalBackgroundClose() { |
| 550 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); | 542 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); |
| 551 // Commit any pending operations | 543 // Commit any pending operations |
| 552 Commit(); | 544 Commit(); |
| 553 | 545 |
| 554 if (!force_keep_session_state_ && | |
| 555 special_storage_policy_.get() && | |
| 556 special_storage_policy_->HasSessionOnlyOrigins()) { | |
| 557 DeleteCertificatesOnShutdown(); | |
| 558 } | |
| 559 | |
| 560 db_.reset(); | 546 db_.reset(); |
| 561 } | 547 } |
| 562 | 548 |
| 563 void SQLiteServerBoundCertStore::Backend::DeleteCertificatesOnShutdown() { | |
| 564 DCHECK(background_task_runner_->RunsTasksOnCurrentThread()); | |
| 565 | |
| 566 if (!db_.get()) | |
| 567 return; | |
| 568 | |
| 569 if (cert_origins_.empty()) | |
| 570 return; | |
| 571 | |
| 572 if (!special_storage_policy_.get()) | |
| 573 return; | |
| 574 | |
| 575 sql::Statement del_smt(db_->GetCachedStatement( | |
| 576 SQL_FROM_HERE, "DELETE FROM origin_bound_certs WHERE origin=?")); | |
| 577 if (!del_smt.is_valid()) { | |
| 578 LOG(WARNING) << "Unable to delete certificates on shutdown."; | |
| 579 return; | |
| 580 } | |
| 581 | |
| 582 sql::Transaction transaction(db_.get()); | |
| 583 if (!transaction.Begin()) { | |
| 584 LOG(WARNING) << "Unable to delete certificates on shutdown."; | |
| 585 return; | |
| 586 } | |
| 587 | |
| 588 for (std::set<std::string>::iterator it = cert_origins_.begin(); | |
| 589 it != cert_origins_.end(); ++it) { | |
| 590 const GURL url(net::cookie_util::CookieOriginToURL(*it, true)); | |
| 591 if (!url.is_valid() || !special_storage_policy_->IsStorageSessionOnly(url)) | |
| 592 continue; | |
| 593 del_smt.Reset(true); | |
| 594 del_smt.BindString(0, *it); | |
| 595 if (!del_smt.Run()) | |
| 596 NOTREACHED() << "Could not delete a certificate from the DB."; | |
| 597 } | |
| 598 | |
| 599 if (!transaction.Commit()) | |
| 600 LOG(WARNING) << "Unable to delete certificates on shutdown."; | |
| 601 } | |
| 602 | |
| 603 void SQLiteServerBoundCertStore::Backend::SetForceKeepSessionState() { | 549 void SQLiteServerBoundCertStore::Backend::SetForceKeepSessionState() { |
| 604 base::AutoLock locked(lock_); | 550 base::AutoLock locked(lock_); |
| 605 force_keep_session_state_ = true; | 551 force_keep_session_state_ = true; |
| 606 } | 552 } |
| 607 | 553 |
| 608 SQLiteServerBoundCertStore::SQLiteServerBoundCertStore( | 554 SQLiteServerBoundCertStore::SQLiteServerBoundCertStore( |
| 609 const base::FilePath& path, | 555 const base::FilePath& path, |
| 610 const scoped_refptr<base::SequencedTaskRunner>& background_task_runner, | 556 const scoped_refptr<base::SequencedTaskRunner>& background_task_runner) |
| 611 quota::SpecialStoragePolicy* special_storage_policy) | 557 : backend_(new Backend(path, background_task_runner)) { |
| 612 : backend_(new Backend(path, | 558 } |
| 613 background_task_runner, | |
| 614 special_storage_policy)) {} | |
| 615 | 559 |
| 616 void SQLiteServerBoundCertStore::Load( | 560 void SQLiteServerBoundCertStore::Load(const LoadedCallback& loaded_callback) { |
| 617 const LoadedCallback& loaded_callback) { | |
| 618 backend_->Load(loaded_callback); | 561 backend_->Load(loaded_callback); |
| 619 } | 562 } |
| 620 | 563 |
| 621 void SQLiteServerBoundCertStore::AddServerBoundCert( | 564 void SQLiteServerBoundCertStore::AddServerBoundCert( |
| 622 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) { | 565 const DefaultServerBoundCertStore::ServerBoundCert& cert) { |
| 623 backend_->AddServerBoundCert(cert); | 566 backend_->AddServerBoundCert(cert); |
| 624 } | 567 } |
| 625 | 568 |
| 626 void SQLiteServerBoundCertStore::DeleteServerBoundCert( | 569 void SQLiteServerBoundCertStore::DeleteServerBoundCert( |
| 627 const net::DefaultServerBoundCertStore::ServerBoundCert& cert) { | 570 const DefaultServerBoundCertStore::ServerBoundCert& cert) { |
| 628 backend_->DeleteServerBoundCert(cert); | 571 backend_->DeleteServerBoundCert(cert); |
| 629 } | 572 } |
| 630 | 573 |
| 631 void SQLiteServerBoundCertStore::SetForceKeepSessionState() { | 574 void SQLiteServerBoundCertStore::SetForceKeepSessionState() { |
| 632 backend_->SetForceKeepSessionState(); | 575 backend_->SetForceKeepSessionState(); |
| 633 } | 576 } |
| 634 | 577 |
| 635 SQLiteServerBoundCertStore::~SQLiteServerBoundCertStore() { | 578 SQLiteServerBoundCertStore::~SQLiteServerBoundCertStore() { |
| 636 backend_->Close(); | 579 backend_->Close(); |
| 637 // We release our reference to the Backend, though it will probably still have | 580 // We release our reference to the Backend, though it will probably still have |
| 638 // a reference if the background thread has not run Close() yet. | 581 // a reference if the background task runner has not run Close() yet. |
| 639 } | 582 } |
| 583 |
| 584 } // namespace net |
| OLD | NEW |