1 files changed, 76 insertions, 0 deletions
diff --git a/net/cert/test_root_certs.cc b/net/cert/test_root_certs.cc
new file mode 100644
index 0000000..3c77ea5
--- /dev/null
+++ b/net/cert/test_root_certs.cc
@@ -0,0 +1,76 @@
+// Copyright (c) 2012 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 "net/cert/test_root_certs.h"
+
+#include <string>
+
+#include "base/file_util.h"
+#include "base/files/file_path.h"
+#include "base/logging.h"
+#include "net/cert/x509_certificate.h"
+
+namespace net {
+
+namespace {
+
+bool g_has_instance = false;
+
+base::LazyInstance<TestRootCerts>::Leaky
+    g_test_root_certs = LAZY_INSTANCE_INITIALIZER;
+
+CertificateList LoadCertificates(const base::FilePath& filename) {
+  std::string raw_cert;
+  if (!file_util::ReadFileToString(filename, &raw_cert)) {
+    LOG(ERROR) << "Can't load certificate " << filename.value();
+    return CertificateList();
+  }
+
+  return X509Certificate::CreateCertificateListFromBytes(
+      raw_cert.data(), raw_cert.length(), X509Certificate::FORMAT_AUTO);
+}
+
+}  // namespace
+
+// static
+TestRootCerts* TestRootCerts::GetInstance() {
+  return g_test_root_certs.Pointer();
+}
+
+bool TestRootCerts::HasInstance() {
+  return g_has_instance;
+}
+
+bool TestRootCerts::AddFromFile(const base::FilePath& file) {
+  CertificateList root_certs = LoadCertificates(file);
+  if (root_certs.empty() || root_certs.size() > 1)
+    return false;
+
+  return Add(root_certs.front());
+}
+
+TestRootCerts::TestRootCerts() {
+  Init();
+  g_has_instance = true;
+}
+
+ScopedTestRoot::ScopedTestRoot() {}
+
+ScopedTestRoot::ScopedTestRoot(X509Certificate* cert) {
+  Reset(cert);
+}
+
+ScopedTestRoot::~ScopedTestRoot() {
+  Reset(NULL);
+}
+
+void ScopedTestRoot::Reset(X509Certificate* cert) {
+  if (cert_)
+    TestRootCerts::GetInstance()->Clear();
+  if (cert)
+    TestRootCerts::GetInstance()->Add(cert);
+  cert_ = cert;
+}
+
+}  // namespace net