1 // Copyright 2017 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/ssl/client_cert_identity.h"
6 
7 #include <memory>
8 
9 #include "crypto/rsa_private_key.h"
10 #include "net/cert/x509_util.h"
11 #include "net/ssl/client_cert_identity_test_util.h"
12 #include "net/ssl/ssl_private_key.h"
13 #include "testing/gtest/include/gtest/gtest.h"
14 
15 namespace net {
16 
TEST(ClientCertIdentitySorter,SortClientCertificates)17 TEST(ClientCertIdentitySorter, SortClientCertificates) {
18   ClientCertIdentityList certs;
19 
20   std::unique_ptr<crypto::RSAPrivateKey> key(
21       crypto::RSAPrivateKey::Create(1024));
22   ASSERT_TRUE(key);
23 
24   scoped_refptr<X509Certificate> cert;
25   std::string der_cert;
26 
27   ASSERT_TRUE(x509_util::CreateSelfSignedCert(
28       key->key(), x509_util::DIGEST_SHA256, "CN=expired", 1,
29       base::Time::UnixEpoch(), base::Time::UnixEpoch(), {}, &der_cert));
30   cert = X509Certificate::CreateFromBytes(der_cert.data(), der_cert.size());
31   ASSERT_TRUE(cert);
32   certs.push_back(std::make_unique<FakeClientCertIdentity>(cert, nullptr));
33 
34   const base::Time now = base::Time::Now();
35 
36   ASSERT_TRUE(x509_util::CreateSelfSignedCert(
37       key->key(), x509_util::DIGEST_SHA256, "CN=not yet valid", 2,
38       now + base::TimeDelta::FromDays(10), now + base::TimeDelta::FromDays(15),
39       {}, &der_cert));
40   cert = X509Certificate::CreateFromBytes(der_cert.data(), der_cert.size());
41   ASSERT_TRUE(cert);
42   certs.push_back(std::make_unique<FakeClientCertIdentity>(cert, nullptr));
43 
44   ASSERT_TRUE(x509_util::CreateSelfSignedCert(
45       key->key(), x509_util::DIGEST_SHA256, "CN=older cert", 3,
46       now - base::TimeDelta::FromDays(5), now + base::TimeDelta::FromDays(5),
47       {}, &der_cert));
48   cert = X509Certificate::CreateFromBytes(der_cert.data(), der_cert.size());
49   ASSERT_TRUE(cert);
50   certs.push_back(std::make_unique<FakeClientCertIdentity>(cert, nullptr));
51 
52   ASSERT_TRUE(x509_util::CreateSelfSignedCert(
53       key->key(), x509_util::DIGEST_SHA256, "CN=newer cert", 2,
54       now - base::TimeDelta::FromDays(3), now + base::TimeDelta::FromDays(5),
55       {}, &der_cert));
56   cert = X509Certificate::CreateFromBytes(der_cert.data(), der_cert.size());
57   ASSERT_TRUE(cert);
58   certs.push_back(std::make_unique<FakeClientCertIdentity>(cert, nullptr));
59 
60   std::sort(certs.begin(), certs.end(), ClientCertIdentitySorter());
61 
62   ASSERT_EQ(4u, certs.size());
63   ASSERT_TRUE(certs[0].get());
64   EXPECT_EQ("newer cert", certs[0]->certificate()->subject().common_name);
65   ASSERT_TRUE(certs[1].get());
66   EXPECT_EQ("older cert", certs[1]->certificate()->subject().common_name);
67   ASSERT_TRUE(certs[2].get());
68   EXPECT_EQ("not yet valid", certs[2]->certificate()->subject().common_name);
69   ASSERT_TRUE(certs[3].get());
70   EXPECT_EQ("expired", certs[3]->certificate()->subject().common_name);
71 }
72 
73 }  // namespace net
74