1 // Copyright (c) 2013 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/quic/crypto/cert_compressor.h"
6
7 #include "base/strings/string_number_conversions.h"
8 #include "net/quic/quic_utils.h"
9 #include "net/quic/test_tools/crypto_test_utils.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 using base::StringPiece;
13 using std::string;
14 using std::vector;
15
16 namespace net {
17 namespace test {
18
TEST(CertCompressor,EmptyChain)19 TEST(CertCompressor, EmptyChain) {
20 vector<string> chain;
21 const string compressed =
22 CertCompressor::CompressChain(chain, StringPiece(), StringPiece(), NULL);
23 EXPECT_EQ("00", base::HexEncode(compressed.data(), compressed.size()));
24
25 vector<string> chain2, cached_certs;
26 ASSERT_TRUE(
27 CertCompressor::DecompressChain(compressed, cached_certs, NULL, &chain2));
28 EXPECT_EQ(chain.size(), chain2.size());
29 }
30
TEST(CertCompressor,Compressed)31 TEST(CertCompressor, Compressed) {
32 vector<string> chain;
33 chain.push_back("testcert");
34 const string compressed =
35 CertCompressor::CompressChain(chain, StringPiece(), StringPiece(), NULL);
36 ASSERT_GE(compressed.size(), 2u);
37 EXPECT_EQ("0100", base::HexEncode(compressed.substr(0, 2).data(), 2));
38
39 vector<string> chain2, cached_certs;
40 ASSERT_TRUE(
41 CertCompressor::DecompressChain(compressed, cached_certs, NULL, &chain2));
42 EXPECT_EQ(chain.size(), chain2.size());
43 EXPECT_EQ(chain[0], chain2[0]);
44 }
45
TEST(CertCompressor,Common)46 TEST(CertCompressor, Common) {
47 vector<string> chain;
48 chain.push_back("testcert");
49 static const uint64 set_hash = 42;
50 scoped_ptr<CommonCertSets> common_sets(
51 CryptoTestUtils::MockCommonCertSets(chain[0], set_hash, 1));
52 const string compressed = CertCompressor::CompressChain(
53 chain,
54 StringPiece(reinterpret_cast<const char*>(&set_hash), sizeof(set_hash)),
55 StringPiece(), common_sets.get());
56 const string common("03" /* common */
57 "2A00000000000000" /* set hash 42 */
58 "01000000" /* index 1 */
59 "00" /* end of list */);
60 EXPECT_EQ(common.data(),
61 base::HexEncode(compressed.data(), compressed.size()));
62
63 vector<string> chain2, cached_certs;
64 ASSERT_TRUE(CertCompressor::DecompressChain(compressed, cached_certs,
65 common_sets.get(), &chain2));
66 EXPECT_EQ(chain.size(), chain2.size());
67 EXPECT_EQ(chain[0], chain2[0]);
68 }
69
TEST(CertCompressor,Cached)70 TEST(CertCompressor, Cached) {
71 vector<string> chain;
72 chain.push_back("testcert");
73 uint64 hash = QuicUtils::FNV1a_64_Hash(chain[0].data(), chain[0].size());
74 StringPiece hash_bytes(reinterpret_cast<char*>(&hash), sizeof(hash));
75 const string compressed =
76 CertCompressor::CompressChain(chain, StringPiece(), hash_bytes, NULL);
77
78 EXPECT_EQ("02" /* cached */ +
79 base::HexEncode(hash_bytes.data(), hash_bytes.size()) +
80 "00" /* end of list */,
81 base::HexEncode(compressed.data(), compressed.size()));
82
83 vector<string> cached_certs, chain2;
84 cached_certs.push_back(chain[0]);
85 ASSERT_TRUE(
86 CertCompressor::DecompressChain(compressed, cached_certs, NULL, &chain2));
87 EXPECT_EQ(chain.size(), chain2.size());
88 EXPECT_EQ(chain[0], chain2[0]);
89 }
90
TEST(CertCompressor,BadInputs)91 TEST(CertCompressor, BadInputs) {
92 vector<string> cached_certs, chain;
93
94 /* bad entry type */
95 const string bad_entry("04");
96 EXPECT_FALSE(CertCompressor::DecompressChain(
97 base::HexEncode(bad_entry.data(), bad_entry.size()),
98 cached_certs, NULL, &chain));
99
100 /* no terminator */
101 const string no_terminator("01");
102 EXPECT_FALSE(CertCompressor::DecompressChain(
103 base::HexEncode(no_terminator.data(), no_terminator.size()),
104 cached_certs, NULL, &chain));
105
106 /* hash truncated */
107 const string hash_truncated("0200");
108 EXPECT_FALSE(CertCompressor::DecompressChain(
109 base::HexEncode(hash_truncated.data(), hash_truncated.size()),
110 cached_certs, NULL, &chain));
111
112 /* hash and index truncated */
113 const string hash_and_index_truncated("0300");
114 EXPECT_FALSE(CertCompressor::DecompressChain(
115 base::HexEncode(hash_and_index_truncated.data(),
116 hash_and_index_truncated.size()),
117 cached_certs, NULL, &chain));
118
119 /* without a CommonCertSets */
120 const string without_a_common_cert_set(
121 "03" "0000000000000000" "00000000");
122 EXPECT_FALSE(CertCompressor::DecompressChain(
123 base::HexEncode(without_a_common_cert_set.data(),
124 without_a_common_cert_set.size()),
125 cached_certs, NULL, &chain));
126
127 scoped_ptr<CommonCertSets> common_sets(
128 CryptoTestUtils::MockCommonCertSets("foo", 42, 1));
129
130 /* incorrect hash and index */
131 const string incorrect_hash_and_index(
132 "03" "a200000000000000" "00000000");
133 EXPECT_FALSE(CertCompressor::DecompressChain(
134 base::HexEncode(incorrect_hash_and_index.data(),
135 incorrect_hash_and_index.size()),
136 cached_certs, NULL, &chain));
137 }
138
139 } // namespace test
140 } // namespace net
141