bitcoin-chainstate.cpp

Go to the documentation of this file.

// Copyright (c) 2022 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
//
// The bitcoin-chainstate executable serves to surface the dependencies required
// by a program wishing to use Bitcoin ABC's consensus engine as it is right
// now.
//
// DEVELOPER NOTE: Since this is a "demo-only", experimental, etc. executable,
//                 it may diverge from Bitcoin ABC's coding style.
//
// It is part of the libbitcoinkernel project.
 
#include <kernel/chainparams.h>
#include <kernel/chainstatemanager_opts.h>
#include <kernel/validation_cache_sizes.h>
 
#include <chainparams.h>
#include <common/args.h>
#include <config.h>
#include <consensus/validation.h>
#include <core_io.h>
#include <init/common.h>
#include <node/blockstorage.h>
#include <node/caches.h>
#include <node/chainstate.h>
#include <scheduler.h>
#include <script/scriptcache.h>
#include <script/sigcache.h>
#include <util/thread.h>
#include <util/translation.h>
#include <validation.h>
#include <validationinterface.h>
 
#include <cstdint>
#include <filesystem>
#include <functional>
#include <iosfwd>
#include <memory>
 
int main(int argc, char *argv[]) {
    // SETUP: Argument parsing and handling
    if (argc != 2) {
        std::cerr << "Usage: " << argv[0] << " DATADIR" << std::endl
                  << "Display DATADIR information, and process hex-encoded "
                     "blocks on standard input."
                  << std::endl
                  << std::endl
                  << "IMPORTANT: THIS EXECUTABLE IS EXPERIMENTAL, FOR TESTING "
                     "ONLY, AND EXPECTED TO"
                  << std::endl
                  << "           BREAK IN FUTURE VERSIONS. DO NOT USE ON YOUR "
                     "ACTUAL DATADIR."
                  << std::endl;
        return 1;
    }
    std::filesystem::path abs_datadir = std::filesystem::absolute(argv[1]);
    std::filesystem::create_directories(abs_datadir);
    gArgs.ForceSetArg("-datadir", abs_datadir.string());
 
    // SETUP: Misc Globals
    SelectParams(CBaseChainParams::MAIN);
 
    auto chainparams = CChainParams::Main(CChainParams::ChainOptions{});
    auto &config = const_cast<Config &>(GetConfig());
    config.SetChainParams(*chainparams);
 
    // ECC_Start, etc.
    init::SetGlobals();
 
    // Necessary for CheckInputScripts (eventually called by ProcessNewBlock),
    // which will try the script cache first and fall back to actually
    // performing the check with the signature cache.
    kernel::ValidationCacheSizes validation_cache_sizes{};
    Assert(InitSignatureCache(validation_cache_sizes.signature_cache_bytes));
    Assert(InitScriptExecutionCache(
        validation_cache_sizes.script_execution_cache_bytes));
 
    // SETUP: Scheduling and Background Signals
    CScheduler scheduler{};
    // Start the lightweight task scheduler thread
    scheduler.m_service_thread = std::thread(util::TraceThread, "scheduler",
                                             [&] { scheduler.serviceQueue(); });
 
    // Gather some entropy once per minute.
    scheduler.scheduleEvery(
        [] {
            RandAddPeriodic();
            return true;
        },
        std::chrono::minutes{1});
 
    GetMainSignals().RegisterBackgroundSignalScheduler(scheduler);
 
    class KernelNotifications : public kernel::Notifications {
    public:
        void blockTip(SynchronizationState, CBlockIndex &) override {
            std::cout << "Block tip changed" << std::endl;
        }
        void headerTip(SynchronizationState, int64_t height, int64_t timestamp,
                       bool presync) override {
            std::cout << "Header tip changed: " << height << ", " << timestamp
                      << ", " << presync << std::endl;
        }
        void progress(const bilingual_str &title, int progress_percent,
                      bool resume_possible) override {
            std::cout << "Progress: " << title.original << ", "
                      << progress_percent << ", " << resume_possible
                      << std::endl;
        }
        void warning(const std::string &warning) override {
            std::cout << "Warning: " << warning << std::endl;
        }
    };
    auto notifications = std::make_unique<KernelNotifications>();
 
    // SETUP: Chainstate
    const ChainstateManager::Options chainman_opts{
        .config = config,
        .datadir = gArgs.GetDataDirNet(),
        .adjusted_time_callback = NodeClock::now,
        .notifications = *notifications,
    };
    const node::BlockManager::Options blockman_opts{
        .chainparams = chainman_opts.config.GetChainParams(),
        .blocks_dir = gArgs.GetBlocksDirPath(),
    };
    ChainstateManager chainman{chainman_opts, blockman_opts};
 
    node::CacheSizes cache_sizes;
    cache_sizes.block_tree_db = 2 << 20;
    cache_sizes.coins_db = 2 << 22;
    cache_sizes.coins = (450 << 20) - (2 << 20) - (2 << 22);
    node::ChainstateLoadOptions options;
    options.check_interrupt = [] { return false; };
    auto [status, error] = node::LoadChainstate(chainman, cache_sizes, options);
    if (status != node::ChainstateLoadStatus::SUCCESS) {
        std::cerr << "Failed to load Chain state from your datadir."
                  << std::endl;
        goto epilogue;
    }
    std::tie(status, error) = node::VerifyLoadedChainstate(chainman, options);
    if (status != node::ChainstateLoadStatus::SUCCESS) {
        std::cerr << "Failed to verify loaded Chain state from your datadir."
                  << std::endl;
        goto epilogue;
    }
 
    for (Chainstate *chainstate :
         WITH_LOCK(::cs_main, return chainman.GetAll())) {
        BlockValidationState state;
        if (!chainstate->ActivateBestChain(state, nullptr)) {
            std::cerr << "Failed to connect best block (" << state.ToString()
                      << ")" << std::endl;
            goto epilogue;
        }
    }
 
    // Main program logic starts here
    std::cout
        << "Hello! I'm going to print out some information about your datadir."
        << std::endl;
    {
        LOCK(chainman.GetMutex());
        std::cout << "\t"
                  << "Path: " << gArgs.GetDataDirNet() << std::endl
                  << "\t"
                  << "Reindexing: " << std::boolalpha << node::fReindex.load()
                  << std::noboolalpha << std::endl
                  << "\t"
                  << "Snapshot Active: " << std::boolalpha
                  << chainman.IsSnapshotActive() << std::noboolalpha
                  << std::endl
                  << "\t"
                  << "Active Height: " << chainman.ActiveHeight() << std::endl
                  << "\t"
                  << "Active IBD: " << std::boolalpha
                  << chainman.ActiveChainstate().IsInitialBlockDownload()
                  << std::noboolalpha << std::endl;
        CBlockIndex *tip = chainman.ActiveTip();
        if (tip) {
            std::cout << "\t" << tip->ToString() << std::endl;
        }
    }
 
    for (std::string line; std::getline(std::cin, line);) {
        if (line.empty()) {
            std::cerr << "Empty line found" << std::endl;
            break;
        }
 
        std::shared_ptr<CBlock> blockptr = std::make_shared<CBlock>();
        CBlock &block = *blockptr;
 
        if (!DecodeHexBlk(block, line)) {
            std::cerr << "Block decode failed" << std::endl;
            break;
        }
 
        if (block.vtx.empty() || !block.vtx[0]->IsCoinBase()) {
            std::cerr << "Block does not start with a coinbase" << std::endl;
            break;
        }
 
        BlockHash hash = block.GetHash();
        {
            LOCK(cs_main);
            const CBlockIndex *pindex =
                chainman.m_blockman.LookupBlockIndex(hash);
            if (pindex) {
                if (pindex->IsValid(BlockValidity::SCRIPTS)) {
                    std::cerr << "Duplicate" << std::endl;
                    break;
                }
                if (pindex->nStatus.hasFailed()) {
                    std::cerr << "Duplicate-invalid" << std::endl;
                    break;
                }
            }
        }
 
        // Adapted from rpc/mining.cpp
        class submitblock_StateCatcher final : public CValidationInterface {
        public:
            BlockHash hash;
            bool found;
            BlockValidationState state;
 
            explicit submitblock_StateCatcher(const BlockHash &hashIn)
                : hash(hashIn), found(false), state() {}
 
        protected:
            void BlockChecked(const CBlock &block,
                              const BlockValidationState &stateIn) override {
                if (block.GetHash() != hash) {
                    return;
                }
                found = true;
                state = stateIn;
            }
        };
 
        bool new_block;
        auto sc = std::make_shared<submitblock_StateCatcher>(block.GetHash());
        RegisterSharedValidationInterface(sc);
        bool accepted = chainman.ProcessNewBlock(blockptr,
                                                 /*force_processing=*/true,
                                                 /*min_pow_checked=*/true,
                                                 /*new_block=*/&new_block);
        UnregisterSharedValidationInterface(sc);
        if (!new_block && accepted) {
            std::cerr << "Duplicate" << std::endl;
            break;
        }
        if (!sc->found) {
            std::cerr << "Inconclusive" << std::endl;
            break;
        }
        std::cout << sc->state.ToString() << std::endl;
        switch (sc->state.GetResult()) {
            case BlockValidationResult::BLOCK_RESULT_UNSET:
                std::cerr << "Initial value. Block has not yet been rejected"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_HEADER_LOW_WORK:
                std::cerr
                    << "the block header may be on a too-little-work chain"
                    << std::endl;
                break;
            case BlockValidationResult::BLOCK_CONSENSUS:
                std::cerr << "Invalid by consensus rules (excluding any below "
                             "reasons)"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_CACHED_INVALID:
                std::cerr << "This block was cached as being invalid and we "
                             "didn't store the reason why"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_INVALID_HEADER:
                std::cerr << "Invalid proof of work or time too old"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_MUTATED:
                std::cerr << "The block's data didn't match the data committed "
                             "to by the PoW"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_MISSING_PREV:
                std::cerr << "We don't have the previous block the checked one "
                             "is built on"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_INVALID_PREV:
                std::cerr << "A block this one builds on is invalid"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_TIME_FUTURE:
                std::cerr << "Block timestamp was > 2 hours in the future (or "
                             "our clock is bad)"
                          << std::endl;
                break;
            case BlockValidationResult::BLOCK_CHECKPOINT:
                std::cerr << "The block failed to meet one of our checkpoints"
                          << std::endl;
                break;
        }
    }
 
epilogue:
    // Without this precise shutdown sequence, there will be a lot of nullptr
    // dereferencing and UB.
    scheduler.stop();
    if (chainman.m_load_block.joinable()) {
        chainman.m_load_block.join();
    }
    StopScriptCheckWorkerThreads();
 
    GetMainSignals().FlushBackgroundCallbacks();
    {
        LOCK(cs_main);
        for (Chainstate *chainstate : chainman.GetAll()) {
            if (chainstate->CanFlushToDisk()) {
                chainstate->ForceFlushStateToDisk();
                chainstate->ResetCoinsViews();
            }
        }
    }
    GetMainSignals().UnregisterBackgroundSignalScheduler();
 
    init::UnsetGlobals();
}