Source code

Revision control

Copy as Markdown

Other Tools

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "BTVerifier.h"
#include "CTTestUtils.h"
#include "gtest/gtest.h"
namespace mozilla {
namespace ct {
using namespace pkix;
class BTSerializationTest : public ::testing::Test {
public:
void SetUp() override {
mTestInclusionProof = GetTestInclusionProof();
mTestInclusionProofUnexpectedData = GetTestInclusionProofUnexpectedData();
mTestInclusionProofInvalidHashSize = GetTestInclusionProofInvalidHashSize();
mTestInclusionProofInvalidHash = GetTestInclusionProofInvalidHash();
mTestInclusionProofMissingLogId = GetTestInclusionProofMissingLogId();
mTestInclusionProofNullPathLength = GetTestInclusionProofNullPathLength();
mTestInclusionProofPathLengthTooSmall =
GetTestInclusionProofPathLengthTooSmall();
mTestInclusionProofPathLengthTooLarge =
GetTestInclusionProofPathLengthTooLarge();
mTestInclusionProofNullTreeSize = GetTestInclusionProofNullTreeSize();
mTestInclusionProofLeafIndexOutOfBounds =
GetTestInclusionProofLeafIndexOutOfBounds();
mTestInclusionProofExtraData = GetTestInclusionProofExtraData();
}
protected:
Buffer mTestInclusionProof;
Buffer mTestInclusionProofUnexpectedData;
Buffer mTestInclusionProofInvalidHashSize;
Buffer mTestInclusionProofInvalidHash;
Buffer mTestInclusionProofMissingLogId;
Buffer mTestInclusionProofNullPathLength;
Buffer mTestInclusionProofPathLengthTooSmall;
Buffer mTestInclusionProofPathLengthTooLarge;
Buffer mTestInclusionProofNullTreeSize;
Buffer mTestInclusionProofLeafIndexOutOfBounds;
Buffer mTestInclusionProofExtraData;
};
TEST_F(BTSerializationTest, DecodesInclusionProof) {
const uint64_t expectedTreeSize = 4;
const uint64_t expectedLeafIndex = 2;
const uint64_t expectedInclusionPathElements = 2;
Buffer expectedLogId = {0x01, 0x00};
Input encodedProofInput = InputForBuffer(mTestInclusionProof);
InclusionProofDataV2 ipr;
ASSERT_EQ(Success, DecodeInclusionProof(encodedProofInput, ipr));
EXPECT_EQ(expectedLogId, ipr.logId);
EXPECT_EQ(expectedTreeSize, ipr.treeSize);
EXPECT_EQ(expectedLeafIndex, ipr.leafIndex);
EXPECT_EQ(expectedInclusionPathElements, ipr.inclusionPath.size());
EXPECT_EQ(GetTestNodeHash0(), ipr.inclusionPath[0]);
EXPECT_EQ(GetTestNodeHash1(), ipr.inclusionPath[1]);
}
TEST_F(BTSerializationTest, FailsDecodingInclusionProofUnexpectedData) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofUnexpectedData);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingInvalidHashSize) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofInvalidHashSize);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingInvalidHash) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofInvalidHash);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingMissingLogId) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofMissingLogId);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingNullPathLength) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofNullPathLength);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingPathLengthTooSmall) {
Input encodedProofInput =
InputForBuffer(mTestInclusionProofPathLengthTooSmall);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingPathLengthTooLarge) {
Input encodedProofInput =
InputForBuffer(mTestInclusionProofPathLengthTooLarge);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingNullTreeSize) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofNullTreeSize);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingLeafIndexOutOfBounds) {
Input encodedProofInput =
InputForBuffer(mTestInclusionProofLeafIndexOutOfBounds);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
TEST_F(BTSerializationTest, FailsDecodingExtraData) {
Input encodedProofInput = InputForBuffer(mTestInclusionProofExtraData);
InclusionProofDataV2 ipr;
ASSERT_EQ(pkix::Result::ERROR_BAD_DER,
DecodeInclusionProof(encodedProofInput, ipr));
}
} // namespace ct
} // namespace mozilla