| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 "net/ssl/ssl_client_auth_cache.h" | 5 #include "net/ssl/ssl_client_auth_cache.h" |
| 6 | 6 |
| 7 #include "base/time.h" | 7 #include "base/time.h" |
| 8 #include "net/cert/x509_certificate.h" | 8 #include "net/cert/x509_certificate.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 | 10 |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 std::string server3("foo3:443"); | 27 std::string server3("foo3:443"); |
| 28 scoped_refptr<X509Certificate> cert3( | 28 scoped_refptr<X509Certificate> cert3( |
| 29 new X509Certificate("foo3", "CA", start_date, expiration_date)); | 29 new X509Certificate("foo3", "CA", start_date, expiration_date)); |
| 30 | 30 |
| 31 scoped_refptr<X509Certificate> cached_cert; | 31 scoped_refptr<X509Certificate> cached_cert; |
| 32 // Lookup non-existent client certificate. | 32 // Lookup non-existent client certificate. |
| 33 cached_cert = NULL; | 33 cached_cert = NULL; |
| 34 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); | 34 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); |
| 35 | 35 |
| 36 // Add client certificate for server1. | 36 // Add client certificate for server1. |
| 37 cache.Add(server1, cert1); | 37 cache.Add(server1, cert1.get()); |
| 38 cached_cert = NULL; | 38 cached_cert = NULL; |
| 39 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 39 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 40 EXPECT_EQ(cert1, cached_cert); | 40 EXPECT_EQ(cert1, cached_cert); |
| 41 | 41 |
| 42 // Add client certificate for server2. | 42 // Add client certificate for server2. |
| 43 cache.Add(server2, cert2); | 43 cache.Add(server2, cert2.get()); |
| 44 cached_cert = NULL; | 44 cached_cert = NULL; |
| 45 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 45 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 46 EXPECT_EQ(cert1, cached_cert.get()); | 46 EXPECT_EQ(cert1, cached_cert.get()); |
| 47 cached_cert = NULL; | 47 cached_cert = NULL; |
| 48 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); | 48 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); |
| 49 EXPECT_EQ(cert2, cached_cert); | 49 EXPECT_EQ(cert2, cached_cert); |
| 50 | 50 |
| 51 // Overwrite the client certificate for server1. | 51 // Overwrite the client certificate for server1. |
| 52 cache.Add(server1, cert3); | 52 cache.Add(server1, cert3.get()); |
| 53 cached_cert = NULL; | 53 cached_cert = NULL; |
| 54 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 54 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 55 EXPECT_EQ(cert3, cached_cert); | 55 EXPECT_EQ(cert3, cached_cert); |
| 56 cached_cert = NULL; | 56 cached_cert = NULL; |
| 57 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); | 57 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); |
| 58 EXPECT_EQ(cert2, cached_cert); | 58 EXPECT_EQ(cert2, cached_cert); |
| 59 | 59 |
| 60 // Remove client certificate of server1. | 60 // Remove client certificate of server1. |
| 61 cache.Remove(server1); | 61 cache.Remove(server1); |
| 62 cached_cert = NULL; | 62 cached_cert = NULL; |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 118 // declined to send a certificate to |server1|. | 118 // declined to send a certificate to |server1|. |
| 119 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 119 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 120 EXPECT_EQ(NULL, cached_cert.get()); | 120 EXPECT_EQ(NULL, cached_cert.get()); |
| 121 | 121 |
| 122 // Remove the existing cached certificate. | 122 // Remove the existing cached certificate. |
| 123 cache.Remove(server1); | 123 cache.Remove(server1); |
| 124 cached_cert = NULL; | 124 cached_cert = NULL; |
| 125 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); | 125 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); |
| 126 | 126 |
| 127 // Add a new preference for a specific certificate. | 127 // Add a new preference for a specific certificate. |
| 128 cache.Add(server1, cert1); | 128 cache.Add(server1, cert1.get()); |
| 129 cached_cert = NULL; | 129 cached_cert = NULL; |
| 130 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 130 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 131 EXPECT_EQ(cert1, cached_cert); | 131 EXPECT_EQ(cert1, cached_cert); |
| 132 | 132 |
| 133 // Replace the specific preference with a NULL certificate. | 133 // Replace the specific preference with a NULL certificate. |
| 134 cache.Add(server1, NULL); | 134 cache.Add(server1, NULL); |
| 135 cached_cert = NULL; | 135 cached_cert = NULL; |
| 136 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 136 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 137 EXPECT_EQ(NULL, cached_cert.get()); | 137 EXPECT_EQ(NULL, cached_cert.get()); |
| 138 } | 138 } |
| 139 | 139 |
| 140 // Check that the OnCertAdded() method removes all cache entries. | 140 // Check that the OnCertAdded() method removes all cache entries. |
| 141 TEST(SSLClientAuthCacheTest, OnCertAdded) { | 141 TEST(SSLClientAuthCacheTest, OnCertAdded) { |
| 142 SSLClientAuthCache cache; | 142 SSLClientAuthCache cache; |
| 143 base::Time start_date = base::Time::Now(); | 143 base::Time start_date = base::Time::Now(); |
| 144 base::Time expiration_date = start_date + base::TimeDelta::FromDays(1); | 144 base::Time expiration_date = start_date + base::TimeDelta::FromDays(1); |
| 145 | 145 |
| 146 std::string server1("foo:443"); | 146 std::string server1("foo:443"); |
| 147 scoped_refptr<X509Certificate> cert1( | 147 scoped_refptr<X509Certificate> cert1( |
| 148 new X509Certificate("foo", "CA", start_date, expiration_date)); | 148 new X509Certificate("foo", "CA", start_date, expiration_date)); |
| 149 | 149 |
| 150 cache.Add(server1, cert1); | 150 cache.Add(server1, cert1.get()); |
| 151 | 151 |
| 152 std::string server2("foo2:443"); | 152 std::string server2("foo2:443"); |
| 153 cache.Add(server2, NULL); | 153 cache.Add(server2, NULL); |
| 154 | 154 |
| 155 scoped_refptr<X509Certificate> cached_cert; | 155 scoped_refptr<X509Certificate> cached_cert; |
| 156 | 156 |
| 157 // Demonstrate the set up is correct. | 157 // Demonstrate the set up is correct. |
| 158 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); | 158 EXPECT_TRUE(cache.Lookup(server1, &cached_cert)); |
| 159 EXPECT_EQ(cert1, cached_cert); | 159 EXPECT_EQ(cert1, cached_cert); |
| 160 | 160 |
| 161 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); | 161 EXPECT_TRUE(cache.Lookup(server2, &cached_cert)); |
| 162 EXPECT_EQ(NULL, cached_cert.get()); | 162 EXPECT_EQ(NULL, cached_cert.get()); |
| 163 | 163 |
| 164 cache.OnCertAdded(NULL); | 164 cache.OnCertAdded(NULL); |
| 165 | 165 |
| 166 // Check that we no longer have entries for either server. | 166 // Check that we no longer have entries for either server. |
| 167 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); | 167 EXPECT_FALSE(cache.Lookup(server1, &cached_cert)); |
| 168 EXPECT_FALSE(cache.Lookup(server2, &cached_cert)); | 168 EXPECT_FALSE(cache.Lookup(server2, &cached_cert)); |
| 169 } | 169 } |
| 170 | 170 |
| 171 } // namespace net | 171 } // namespace net |
| OLD | NEW |