Index: crypto/scoped_test_nss_db.cc |
diff --git a/crypto/scoped_test_nss_db.cc b/crypto/scoped_test_nss_db.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..68a4a914c7cfd03309c7b46f7eb1d409613a00d3 |
--- /dev/null |
+++ b/crypto/scoped_test_nss_db.cc |
@@ -0,0 +1,50 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "crypto/scoped_test_nss_db.h" |
+ |
+#include "base/logging.h" |
+#include "base/threading/thread_restrictions.h" |
+#include "crypto/nss_util.h" |
+#include "crypto/nss_util_internal.h" |
+ |
+namespace crypto { |
+ |
+#if defined(USE_NSS) |
+ScopedTestNSSDB::ScopedTestNSSDB() { |
+ EnsureNSSInit(); |
+ // NSS is allowed to do IO on the current thread since dispatching |
+ // to a dedicated thread would still have the affect of blocking |
+ // the current thread, due to NSS's internal locking requirements |
+ base::ThreadRestrictions::ScopedAllowIO allow_io; |
+ |
+ if (!temp_dir_.CreateUniqueTempDir()) |
+ return; |
+ |
+ const char kTestTPMTokenName[] = "Test DB"; |
Ryan Sleevi
2014/07/22 01:25:32
Fix variable name. It's not a TPM :)
pneubeck (no reviews)
2014/07/22 14:11:05
Done.
|
+ slot_ = OpenUserDB(temp_dir_.path(), kTestTPMTokenName); |
+} |
+ |
+ScopedTestNSSDB::~ScopedTestNSSDB() { |
+ // Don't close when NSS is < 3.15.1, because it would require an additional |
+ // sleep for 1 second after closing the database, due to |
+ // http://bugzil.la/875601. |
+ if (!NSS_VersionCheck("3.15.1") || !slot_) |
+ return; |
+ |
+ // NSS is allowed to do IO on the current thread since dispatching |
+ // to a dedicated thread would still have the affect of blocking |
+ // the current thread, due to NSS's internal locking requirements |
+ base::ThreadRestrictions::ScopedAllowIO allow_io; |
+ |
+ SECStatus status = SECMOD_CloseUserDB(slot_.get()); |
+ if (status != SECSuccess) |
+ PLOG(ERROR) << "SECMOD_CloseUserDB failed: " << PORT_GetError(); |
+ |
+ if (!temp_dir_.Delete()) |
+ LOG(ERROR) << "Could not delete temporary directory."; |
+} |
+#endif |
+ |
+} // namespace crypto |