Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(173)

Side by Side Diff: net/util/sqlite/sqlite_server_bound_cert_store.cc

Issue 381073002: Move sqlite_channel_id_store from chrome/browser/net to net/extras. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Added content/browser/net/quota_policyserver_bound_cert_store... Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
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
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
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698