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