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/test/cert_test_util.h"
6
7#include "base/files/file_path.h"
8#include "base/files/file_util.h"
9#include "base/path_service.h"
10#include "net/cert/ev_root_ca_metadata.h"
11#include "net/cert/x509_certificate.h"
12#include "testing/gtest/include/gtest/gtest.h"
13
14namespace net {
15
16CertificateList CreateCertificateListFromFile(
17    const base::FilePath& certs_dir,
18    const std::string& cert_file,
19    int format) {
20  base::FilePath cert_path = certs_dir.AppendASCII(cert_file);
21  std::string cert_data;
22  if (!base::ReadFileToString(cert_path, &cert_data))
23    return CertificateList();
24  return X509Certificate::CreateCertificateListFromBytes(cert_data.data(),
25                                                         cert_data.size(),
26                                                         format);
27}
28
29scoped_refptr<X509Certificate> CreateCertificateChainFromFile(
30    const base::FilePath& certs_dir,
31    const std::string& cert_file,
32    int format) {
33  CertificateList certs = CreateCertificateListFromFile(
34      certs_dir, cert_file, format);
35  if (certs.empty())
36    return NULL;
37
38  X509Certificate::OSCertHandles intermediates;
39  for (size_t i = 1; i < certs.size(); ++i)
40    intermediates.push_back(certs[i]->os_cert_handle());
41
42  scoped_refptr<X509Certificate> result(X509Certificate::CreateFromHandle(
43        certs[0]->os_cert_handle(), intermediates));
44  return result;
45}
46
47scoped_refptr<X509Certificate> ImportCertFromFile(
48    const base::FilePath& certs_dir,
49    const std::string& cert_file) {
50  base::FilePath cert_path = certs_dir.AppendASCII(cert_file);
51  std::string cert_data;
52  if (!base::ReadFileToString(cert_path, &cert_data))
53    return NULL;
54
55  CertificateList certs_in_file =
56      X509Certificate::CreateCertificateListFromBytes(
57          cert_data.data(), cert_data.size(), X509Certificate::FORMAT_AUTO);
58  if (certs_in_file.empty())
59    return NULL;
60  return certs_in_file[0];
61}
62
63ScopedTestEVPolicy::ScopedTestEVPolicy(EVRootCAMetadata* ev_root_ca_metadata,
64                                       const SHA1HashValue& fingerprint,
65                                       const char* policy)
66    : fingerprint_(fingerprint),
67      ev_root_ca_metadata_(ev_root_ca_metadata) {
68  EXPECT_TRUE(ev_root_ca_metadata->AddEVCA(fingerprint, policy));
69}
70
71ScopedTestEVPolicy::~ScopedTestEVPolicy() {
72  EXPECT_TRUE(ev_root_ca_metadata_->RemoveEVCA(fingerprint_));
73}
74
75}  // namespace net
76