// Copyright (c) 2012-2015, The CryptoNote developers, The Bytecoin developers
//
// This file is part of Bytecoin.
//
// Bytecoin is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Bytecoin is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Bytecoin. If not, see .
#pragma once
#include "Chaingen.h"
const uint64_t UNDEF_HEIGHT = static_cast(CryptoNote::UpgradeDetectorBase::UNDEF_HEIGHT);
class CheckBlockPurged : public test_chain_unit_base {
public:
CheckBlockPurged(size_t invalidBlockIdx, uint8_t blockMajorVersion) :
m_invalidBlockIdx(invalidBlockIdx),
m_blockMajorVersion(blockMajorVersion) {
assert(blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_1 || blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_2);
CryptoNote::CurrencyBuilder currencyBuilder(m_logger);
currencyBuilder.upgradeHeight(blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_1 ? UNDEF_HEIGHT : UINT64_C(0));
m_currency = currencyBuilder.currency();
REGISTER_CALLBACK("check_block_purged", CheckBlockPurged::check_block_purged);
REGISTER_CALLBACK("markInvalidBlock", CheckBlockPurged::markInvalidBlock);
}
bool check_block_verification_context(const CryptoNote::block_verification_context& bvc, size_t eventIdx, const CryptoNote::Block& /*blk*/) {
if (m_invalidBlockIdx == eventIdx) {
return bvc.m_verifivation_failed;
} else {
return !bvc.m_verifivation_failed;
}
}
bool check_block_purged(CryptoNote::core& c, size_t eventIdx, const std::vector& events) {
DEFINE_TESTS_ERROR_CONTEXT("CheckBlockPurged::check_block_purged");
CHECK_TEST_CONDITION(m_invalidBlockIdx < eventIdx);
CHECK_EQ(0, c.get_pool_transactions_count());
CHECK_EQ(m_invalidBlockIdx, c.get_current_blockchain_height());
return true;
}
bool markInvalidBlock(CryptoNote::core& c, size_t eventIdx, const std::vector& events) {
m_invalidBlockIdx = eventIdx + 1;
return true;
}
protected:
size_t m_invalidBlockIdx;
const uint8_t m_blockMajorVersion;
};
struct CheckBlockAccepted : public test_chain_unit_base {
CheckBlockAccepted(size_t expectedBlockchainHeight, uint8_t blockMajorVersion) :
m_expectedBlockchainHeight(expectedBlockchainHeight),
m_blockMajorVersion(blockMajorVersion) {
assert(blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_1 || blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_2);
CryptoNote::CurrencyBuilder currencyBuilder(m_logger);
currencyBuilder.upgradeHeight(blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_1 ? UNDEF_HEIGHT : UINT64_C(0));
m_currency = currencyBuilder.currency();
REGISTER_CALLBACK("check_block_accepted", CheckBlockAccepted::check_block_accepted);
}
bool check_block_accepted(CryptoNote::core& c, size_t /*eventIdx*/, const std::vector& /*events*/) {
DEFINE_TESTS_ERROR_CONTEXT("CheckBlockAccepted::check_block_accepted");
CHECK_EQ(0, c.get_pool_transactions_count());
CHECK_EQ(m_expectedBlockchainHeight, c.get_current_blockchain_height());
return true;
}
protected:
size_t m_expectedBlockchainHeight;
const uint8_t m_blockMajorVersion;
};
struct TestBlockMajorVersionAccepted : public CheckBlockAccepted {
TestBlockMajorVersionAccepted(uint8_t blockMajorVersion) :
CheckBlockAccepted(2, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct TestBlockMajorVersionRejected : public CheckBlockPurged {
TestBlockMajorVersionRejected(uint8_t blockAcceptedVersion, uint8_t blockGeneratedVersion) :
CheckBlockPurged(1, blockAcceptedVersion), m_blockGeneratedVersion(blockGeneratedVersion) {}
const uint8_t m_blockGeneratedVersion;
bool generate(std::vector& events) const;
};
struct TestBlockBigMinorVersion : public CheckBlockAccepted {
TestBlockBigMinorVersion(uint8_t blockMajorVersion)
: CheckBlockAccepted(2, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_ts_not_checked : public CheckBlockAccepted
{
gen_block_ts_not_checked(uint8_t blockMajorVersion)
: CheckBlockAccepted(0, blockMajorVersion) {
m_expectedBlockchainHeight = m_currency.timestampCheckWindow();
}
bool generate(std::vector& events) const;
};
struct gen_block_ts_in_past : public CheckBlockPurged
{
gen_block_ts_in_past(uint8_t blockMajorVersion)
: CheckBlockPurged(0, blockMajorVersion) {
m_invalidBlockIdx = m_currency.timestampCheckWindow();
}
bool generate(std::vector& events) const;
};
struct gen_block_ts_in_future_rejected : public CheckBlockPurged
{
gen_block_ts_in_future_rejected(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_ts_in_future_accepted : public CheckBlockAccepted
{
gen_block_ts_in_future_accepted(uint8_t blockMajorVersion)
: CheckBlockAccepted(2, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_invalid_prev_id : public CheckBlockPurged
{
gen_block_invalid_prev_id(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
bool check_block_verification_context(const CryptoNote::block_verification_context& bvc, size_t event_idx, const CryptoNote::Block& /*blk*/);
};
struct gen_block_invalid_nonce : public CheckBlockPurged
{
gen_block_invalid_nonce(uint8_t blockMajorVersion)
: CheckBlockPurged(3, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_no_miner_tx : public CheckBlockPurged
{
gen_block_no_miner_tx(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_unlock_time_is_low : public CheckBlockPurged
{
gen_block_unlock_time_is_low(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_unlock_time_is_high : public CheckBlockPurged
{
gen_block_unlock_time_is_high(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_unlock_time_is_timestamp_in_past : public CheckBlockPurged
{
gen_block_unlock_time_is_timestamp_in_past(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_unlock_time_is_timestamp_in_future : public CheckBlockPurged
{
gen_block_unlock_time_is_timestamp_in_future(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_height_is_low : public CheckBlockPurged
{
gen_block_height_is_low(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_height_is_high : public CheckBlockPurged
{
gen_block_height_is_high(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_has_2_tx_gen_in : public CheckBlockPurged
{
gen_block_miner_tx_has_2_tx_gen_in(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_has_2_in : public CheckBlockPurged
{
gen_block_miner_tx_has_2_in(uint8_t blockMajorVersion)
: CheckBlockPurged(0, blockMajorVersion) {
m_invalidBlockIdx = m_currency.minedMoneyUnlockWindow() + 1;
}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_with_txin_to_key : public CheckBlockPurged
{
gen_block_miner_tx_with_txin_to_key(uint8_t blockMajorVersion)
: CheckBlockPurged(0, blockMajorVersion) {
m_invalidBlockIdx = m_currency.minedMoneyUnlockWindow() + 2;
}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_out_is_small : public CheckBlockPurged
{
gen_block_miner_tx_out_is_small(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_out_is_big : public CheckBlockPurged
{
gen_block_miner_tx_out_is_big(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_has_no_out : public CheckBlockPurged
{
gen_block_miner_tx_has_no_out(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_miner_tx_has_out_to_alice : public CheckBlockAccepted
{
gen_block_miner_tx_has_out_to_alice(uint8_t blockMajorVersion)
: CheckBlockAccepted(2, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_has_invalid_tx : public CheckBlockPurged
{
gen_block_has_invalid_tx(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {}
bool generate(std::vector& events) const;
};
struct gen_block_is_too_big : public CheckBlockPurged
{
gen_block_is_too_big(uint8_t blockMajorVersion)
: CheckBlockPurged(1, blockMajorVersion) {
CryptoNote::CurrencyBuilder currencyBuilder(m_logger);
currencyBuilder.upgradeHeight(blockMajorVersion == CryptoNote::BLOCK_MAJOR_VERSION_1 ? UNDEF_HEIGHT : UINT64_C(0));
currencyBuilder.maxBlockSizeInitial(std::numeric_limits::max() / 2);
m_currency = currencyBuilder.currency();
}
bool generate(std::vector& events) const;
};
struct TestBlockCumulativeSizeExceedsLimit : public CheckBlockPurged {
TestBlockCumulativeSizeExceedsLimit(uint8_t blockMajorVersion)
: CheckBlockPurged(std::numeric_limits::max(), blockMajorVersion) {
}
bool generate(std::vector& events) const;
};
struct gen_block_invalid_binary_format : public test_chain_unit_base
{
gen_block_invalid_binary_format(uint8_t blockMajorVersion);
bool generate(std::vector& events) const;
bool check_block_verification_context(const CryptoNote::block_verification_context& bvc, size_t event_idx, const CryptoNote::Block& /*blk*/);
bool check_all_blocks_purged(CryptoNote::core& c, size_t ev_index, const std::vector& events);
bool corrupt_blocks_boundary(CryptoNote::core& c, size_t ev_index, const std::vector& events);
private:
const uint8_t m_blockMajorVersion;
size_t m_corrupt_blocks_begin_idx;
};
struct TestMaxSizeOfParentBlock : public CheckBlockAccepted {
TestMaxSizeOfParentBlock() : CheckBlockAccepted(2, CryptoNote::BLOCK_MAJOR_VERSION_2) {
}
bool generate(std::vector& events) const;
};
struct TestBigParentBlock : public CheckBlockPurged {
TestBigParentBlock() : CheckBlockPurged(1, CryptoNote::BLOCK_MAJOR_VERSION_2) {
}
bool generate(std::vector& events) const;
};
struct TestBlock2ExtraEmpty : public CheckBlockPurged {
TestBlock2ExtraEmpty() : CheckBlockPurged(1, CryptoNote::BLOCK_MAJOR_VERSION_2) {}
bool generate(std::vector& events) const;
};
struct TestBlock2ExtraWithoutMMTag : public CheckBlockPurged {
TestBlock2ExtraWithoutMMTag() : CheckBlockPurged(1, CryptoNote::BLOCK_MAJOR_VERSION_2) {}
bool generate(std::vector& events) const;
};
struct TestBlock2ExtraWithGarbage : public CheckBlockAccepted {
TestBlock2ExtraWithGarbage() : CheckBlockAccepted(2, CryptoNote::BLOCK_MAJOR_VERSION_2) {}
bool generate(std::vector& events) const;
};