1 // Copyright 2016 The Chromium Authors
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/merkle_audit_proof.h"
6
7 #include "base/check.h"
8 #include "testing/gmock/include/gmock/gmock.h"
9 #include "testing/gtest/include/gtest/gtest.h"
10
11 namespace net::ct {
12 namespace {
13
TEST(MerkleAuditProofTest,CalculatesAuditPathLengthCorrectly)14 TEST(MerkleAuditProofTest, CalculatesAuditPathLengthCorrectly) {
15 // Test all leaves up to a tree size of 4:
16 EXPECT_EQ(0u, CalculateAuditPathLength(0, 1));
17 EXPECT_EQ(1u, CalculateAuditPathLength(0, 2));
18 EXPECT_EQ(1u, CalculateAuditPathLength(1, 2));
19 EXPECT_EQ(2u, CalculateAuditPathLength(0, 3));
20 EXPECT_EQ(2u, CalculateAuditPathLength(1, 3));
21 EXPECT_EQ(1u, CalculateAuditPathLength(2, 3));
22 EXPECT_EQ(2u, CalculateAuditPathLength(0, 4));
23 EXPECT_EQ(2u, CalculateAuditPathLength(1, 4));
24 EXPECT_EQ(2u, CalculateAuditPathLength(2, 4));
25 EXPECT_EQ(2u, CalculateAuditPathLength(3, 4));
26 // Boundary cases for a larger tree size:
27 EXPECT_EQ(9u, CalculateAuditPathLength(0, 257));
28 EXPECT_EQ(9u, CalculateAuditPathLength(255, 257));
29 EXPECT_EQ(1u, CalculateAuditPathLength(256, 257));
30 // Example from CT over DNS draft RFC:
31 EXPECT_EQ(20u, CalculateAuditPathLength(123456, 999999));
32 // Test data from
33 // https://github.com/google/certificate-transparency/blob/af98904302724c29aa6659ca372d41c9687de2b7/python/ct/crypto/merkle_test.py:
34 EXPECT_EQ(22u, CalculateAuditPathLength(848049, 3630887));
35 }
36
TEST(MerkleAuditProofDeathTest,DiesIfLeafIndexIsGreaterThanOrEqualToTreeSize)37 TEST(MerkleAuditProofDeathTest, DiesIfLeafIndexIsGreaterThanOrEqualToTreeSize) {
38 #ifdef OFFICIAL_BUILD
39 // The official build does not print the reason a CHECK failed.
40 const char kErrorRegex[] = "";
41 #else
42 const char kErrorRegex[] = "leaf_index < tree_size";
43 #endif
44
45 EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(0, 0), kErrorRegex);
46 EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(10, 10), kErrorRegex);
47 EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(11, 10), kErrorRegex);
48 }
49
50 } // namespace
51 } // namespace net::ct
52