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
12using base::StringPiece;
13using std::string;
14using std::vector;
15
16namespace net {
17namespace test {
18
19TEST(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
31TEST(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
46TEST(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
70TEST(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
91TEST(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