| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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/cert/ct_ev_whitelist.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 | |
| 11 namespace net { | |
| 12 | |
| 13 namespace ct { | |
| 14 | |
| 15 namespace internal { | |
| 16 | |
| 17 const uint8 kSomeData[] = {0xd5, 0xe2, 0xaf, 0xe5, 0xbb, 0x10, 0x7c, 0xd1}; | |
| 18 | |
| 19 TEST(BitStreamReaderTest, CanReadSingleByte) { | |
| 20 BitStreamReader reader(reinterpret_cast<const char*>(kSomeData), 1); | |
| 21 uint64 v(0); | |
| 22 | |
| 23 EXPECT_EQ(8u, reader.BitsLeft()); | |
| 24 EXPECT_TRUE(reader.ReadBits(8, &v)); | |
| 25 EXPECT_EQ(UINT64_C(0xd5), v); | |
| 26 | |
| 27 EXPECT_FALSE(reader.ReadBits(1, &v)); | |
| 28 EXPECT_EQ(0u, reader.BitsLeft()); | |
| 29 } | |
| 30 | |
| 31 TEST(BitStreamReaderTest, CanReadSingleBits) { | |
| 32 const uint64 expected_bits[] = {1, 1, 0, 1, 0, 1, 0, 1, | |
| 33 1, 1, 1, 0, 0, 0, 1, 0}; | |
| 34 BitStreamReader reader(reinterpret_cast<const char*>(kSomeData), 2); | |
| 35 EXPECT_EQ(16u, reader.BitsLeft()); | |
| 36 uint64 v(0); | |
| 37 | |
| 38 for (int i = 0; i < 16; ++i) { | |
| 39 EXPECT_TRUE(reader.ReadBits(1, &v)); | |
| 40 EXPECT_EQ(expected_bits[i], v); | |
| 41 } | |
| 42 EXPECT_EQ(0u, reader.BitsLeft()); | |
| 43 } | |
| 44 | |
| 45 TEST(BitStreamReaderTest, CanReadBitGroups) { | |
| 46 BitStreamReader reader(reinterpret_cast<const char*>(kSomeData), 3); | |
| 47 EXPECT_EQ(24u, reader.BitsLeft()); | |
| 48 uint64 v(0); | |
| 49 uint64 res(0); | |
| 50 | |
| 51 EXPECT_TRUE(reader.ReadBits(5, &v)); | |
| 52 res |= v << 19; | |
| 53 EXPECT_EQ(19u, reader.BitsLeft()); | |
| 54 EXPECT_TRUE(reader.ReadBits(13, &v)); | |
| 55 res |= v << 6; | |
| 56 EXPECT_EQ(6u, reader.BitsLeft()); | |
| 57 EXPECT_TRUE(reader.ReadBits(6, &v)); | |
| 58 res |= v; | |
| 59 EXPECT_EQ(UINT64_C(0xd5e2af), res); | |
| 60 | |
| 61 EXPECT_FALSE(reader.ReadBits(1, &v)); | |
| 62 } | |
| 63 | |
| 64 TEST(BitStreamReaderTest, CanRead64Bit) { | |
| 65 BitStreamReader reader(reinterpret_cast<const char*>(kSomeData), 8); | |
| 66 EXPECT_EQ(64u, reader.BitsLeft()); | |
| 67 uint64 v(0); | |
| 68 | |
| 69 EXPECT_TRUE(reader.ReadBits(64, &v)); | |
| 70 EXPECT_EQ(UINT64_C(0xd5e2afe5bb107cd1), v); | |
| 71 } | |
| 72 | |
| 73 TEST(BitStreamReaderTest, CanReadUnaryEncodedNumbers) { | |
| 74 BitStreamReader reader(reinterpret_cast<const char*>(kSomeData), 3); | |
| 75 const uint64 expected_values[] = {2, 1, 1, 4, 0, 0, 1, 1, 1, 4}; | |
| 76 uint64 v(0); | |
| 77 for (int i = 0; i < 10; ++i) { | |
| 78 EXPECT_TRUE(reader.ReadUnaryEncoding(&v)); | |
| 79 EXPECT_EQ(expected_values[i], v) << "Values differ at position " << i; | |
| 80 } | |
| 81 } | |
| 82 | |
| 83 } // namespace internal | |
| 84 | |
| 85 namespace { | |
| 86 | |
| 87 const uint8 kFirstHashRaw[] = {0x00, 0x00, 0x03, 0xd7, 0xfc, 0x18, 0x02, 0xcb}; | |
| 88 std::string GetFirstHash() { | |
| 89 return std::string(reinterpret_cast<const char*>(kFirstHashRaw), 8); | |
| 90 } | |
| 91 | |
| 92 // Second hash: Diff from first hash is > 2^47 | |
| 93 const uint8 kSecondHashRaw[] = {0x00, 0x01, 0x05, 0xd2, 0x58, 0x47, 0xa7, 0xbf}; | |
| 94 std::string GetSecondHash() { | |
| 95 return std::string(reinterpret_cast<const char*>(kSecondHashRaw), 8); | |
| 96 } | |
| 97 | |
| 98 // Third hash: Diff from 2nd hash is < 2^47 | |
| 99 const uint8 kThirdHashRaw[] = {0x00, 0x01, 0x48, 0x45, 0x8c, 0x53, 0x03, 0x94}; | |
| 100 std::string GetThirdHash() { | |
| 101 return std::string(reinterpret_cast<const char*>(kThirdHashRaw), 8); | |
| 102 } | |
| 103 | |
| 104 const uint8 kWhitelistData[] = { | |
| 105 0x00, 0x00, 0x03, 0xd7, 0xfc, 0x18, 0x02, 0xcb, // First hash | |
| 106 0xc0, 0x7e, 0x97, 0x0b, 0xe9, 0x3d, 0x10, 0x9c, | |
| 107 0xcd, 0x02, 0xd6, 0xf5, 0x40, | |
| 108 }; | |
| 109 | |
| 110 } // namespace | |
| 111 | |
| 112 TEST(CTEVWhitelistTest, UncompressFailsForTooShortList) { | |
| 113 // This list does not contain enough bytes even for the first hash. | |
| 114 std::set<std::string> res; | |
| 115 EXPECT_FALSE(internal::UncompressEVWhitelist( | |
| 116 std::string(reinterpret_cast<const char*>(kWhitelistData), 7), &res)); | |
| 117 } | |
| 118 | |
| 119 TEST(CTEVWhitelistTest, UncompressFailsForTruncatedList) { | |
| 120 // This list is missing bits for the second part of the diff. | |
| 121 std::set<std::string> res; | |
| 122 EXPECT_FALSE(internal::UncompressEVWhitelist( | |
| 123 std::string(reinterpret_cast<const char*>(kWhitelistData), 14), &res)); | |
| 124 } | |
| 125 | |
| 126 TEST(CTEVWhitelistTest, UncompressesWhitelistCorrectly) { | |
| 127 std::set<std::string> res; | |
| 128 ASSERT_TRUE(internal::UncompressEVWhitelist( | |
| 129 std::string(reinterpret_cast<const char*>(kWhitelistData), | |
| 130 arraysize(kWhitelistData)), | |
| 131 &res)); | |
| 132 | |
| 133 // Ensure first hash is found | |
| 134 EXPECT_TRUE(res.find(GetFirstHash()) != res.end()); | |
| 135 // Ensure second hash is found | |
| 136 EXPECT_TRUE(res.find(GetSecondHash()) != res.end()); | |
| 137 // Ensure last hash is found | |
| 138 EXPECT_TRUE(res.find(GetThirdHash()) != res.end()); | |
| 139 // Ensure that there are exactly 3 hashes. | |
| 140 EXPECT_EQ(3u, res.size()); | |
| 141 } | |
| 142 | |
| 143 TEST(CTEVWhitelistTest, CanFindHashInSetList) { | |
| 144 std::set<std::string> whitelist_data; | |
| 145 whitelist_data.insert(GetFirstHash()); | |
| 146 internal::SetEVWhitelistData(whitelist_data); | |
| 147 | |
| 148 EXPECT_TRUE(IsCertificateHashInWhitelist(GetFirstHash())); | |
| 149 } | |
| 150 | |
| 151 TEST(CTEVWhitelistTest, CannotFindOldHashAfterSetList) { | |
| 152 std::set<std::string> whitelist_data; | |
| 153 whitelist_data.insert(GetFirstHash()); | |
| 154 internal::SetEVWhitelistData(whitelist_data); | |
| 155 EXPECT_TRUE(IsCertificateHashInWhitelist(GetFirstHash())); | |
| 156 | |
| 157 std::set<std::string> new_whitelist_data; | |
| 158 new_whitelist_data.insert(GetSecondHash()); | |
| 159 internal::SetEVWhitelistData(new_whitelist_data); | |
| 160 EXPECT_TRUE(IsCertificateHashInWhitelist(GetSecondHash())); | |
| 161 EXPECT_FALSE(IsCertificateHashInWhitelist(GetFirstHash())); | |
| 162 } | |
| 163 | |
| 164 TEST(CTEVWhitelistTest, CorrectlyIdentifiesWhitelistIsInvalid) { | |
| 165 std::set<std::string> whitelist_data; | |
| 166 internal::SetEVWhitelistData(whitelist_data); | |
| 167 EXPECT_FALSE(HasValidEVWhitelist()); | |
| 168 } | |
| 169 | |
| 170 TEST(CTEVWhitelistTest, CorrectlyIdentifiesWhitelistIsValid) { | |
| 171 std::set<std::string> whitelist_data; | |
| 172 whitelist_data.insert(GetFirstHash()); | |
| 173 internal::SetEVWhitelistData(whitelist_data); | |
| 174 EXPECT_TRUE(HasValidEVWhitelist()); | |
| 175 } | |
| 176 | |
| 177 } // namespace ct | |
| 178 | |
| 179 } // namespace net | |
| OLD | NEW |