net/quic/crypto/cert_compressor_test.cc

   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
  19 TEST(CertCompressor, EmptyChain) {
  20   vector<string> chain;
  21   const string compressed = CertCompressor::CompressChain(
  22       chain, StringPiece(), StringPiece(), nullptr);
  23   EXPECT_EQ("00", base::HexEncode(compressed.data(), compressed.size()));
  24
  25   vector<string> chain2, cached_certs;
  26   ASSERT_TRUE(CertCompressor::DecompressChain(compressed, cached_certs, nullptr,
  27                                               &chain2));
  28   EXPECT_EQ(chain.size(), chain2.size());
  29 }
  30
  31 TEST(CertCompressor, Compressed) {
  32   vector<string> chain;
  33   chain.push_back("testcert");
  34   const string compressed = CertCompressor::CompressChain(
  35       chain, StringPiece(), StringPiece(), nullptr);
  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(CertCompressor::DecompressChain(compressed, cached_certs, nullptr,
  41                                               &chain2));
  42   EXPECT_EQ(chain.size(), chain2.size());
  43   EXPECT_EQ(chain[0], chain2[0]);
  44 }
  45
  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
  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, nullptr);
  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(CertCompressor::DecompressChain(compressed, cached_certs, nullptr,
  86                                               &chain2));
  87   EXPECT_EQ(chain.size(), chain2.size());
  88   EXPECT_EQ(chain[0], chain2[0]);
  89 }
  90
  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, nullptr, &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, nullptr, &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, nullptr, &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, nullptr, &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, nullptr, &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, nullptr, &chain));
 137 }
 138
 139 }  // namespace test
 140 }  // namespace net