utxo_snapshot.h

Go to the documentation of this file.

// Copyright (c) 2009-2010 Satoshi Nakamoto
// Copyright (c) 2009-2019 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_NODE_UTXO_SNAPSHOT_H
#define BITCOIN_NODE_UTXO_SNAPSHOT_H
 
#include <chainparams.h>
#include <kernel/chainparams.h>
#include <primitives/blockhash.h>
#include <serialize.h>
#include <util/chaintype.h>
#include <util/check.h>
#include <util/fs.h>
#include <validation.h>
 
#include <optional>
 
// UTXO set snapshot magic bytes
static constexpr std::array<uint8_t, 5> SNAPSHOT_MAGIC_BYTES = {
    {'u', 't', 'x', 'o', 0xff}};
 
struct BlockHash;
 
namespace node {
class SnapshotMetadata {
    inline static const uint16_t VERSION{2};
    const std::set<uint16_t> m_supported_versions{VERSION};
    const CMessageHeader::MessageMagic m_network_magic;
 
public:
    BlockHash m_base_blockhash;
 
    uint64_t m_coins_count = 0;
 
    SnapshotMetadata(const CMessageHeader::MessageMagic network_magic)
        : m_network_magic(network_magic) {}
    SnapshotMetadata(const CMessageHeader::MessageMagic network_magic,
                     const BlockHash &base_blockhash, uint64_t coins_count)
        : m_network_magic(network_magic), m_base_blockhash(base_blockhash),
          m_coins_count(coins_count) {}
 
    template <typename Stream> inline void Serialize(Stream &s) const {
        s << SNAPSHOT_MAGIC_BYTES;
        s << VERSION;
        s << m_network_magic;
        s << m_base_blockhash;
        s << m_coins_count;
    }
 
    template <typename Stream> inline void Unserialize(Stream &s) {
        // Read the snapshot magic bytes
        std::array<uint8_t, SNAPSHOT_MAGIC_BYTES.size()> snapshot_magic;
        s >> snapshot_magic;
        if (snapshot_magic != SNAPSHOT_MAGIC_BYTES) {
            throw std::ios_base::failure(
                "Invalid UTXO set snapshot magic bytes. Please check if this "
                "is indeed a snapshot file or if you are using an outdated "
                "snapshot format.");
        }
 
        // Read the version
        uint16_t version;
        s >> version;
        if (m_supported_versions.find(version) == m_supported_versions.end()) {
            throw std::ios_base::failure(
                strprintf("Version of snapshot %s does not match any of the "
                          "supported versions.",
                          version));
        }
 
        // Read the network magic (pchMessageStart)
        CMessageHeader::MessageMagic message;
        s >> message;
        if (!std::equal(message.begin(), message.end(),
                        m_network_magic.data())) {
            auto metadata_network{GetNetworkForMagic(message)};
            if (metadata_network) {
                std::string network_string{
                    ChainTypeToString(metadata_network.value())};
                auto node_network{GetNetworkForMagic(m_network_magic)};
                std::string node_network_string{
                    ChainTypeToString(node_network.value())};
                throw std::ios_base::failure(
                    strprintf("The network of the snapshot (%s) does not match "
                              "the network of this node (%s).",
                              network_string, node_network_string));
            } else {
                throw std::ios_base::failure(
                    "This snapshot has been created for an unrecognized "
                    "network. This could be a new testnet or possibly caused "
                    "by data corruption.");
            }
        }
 
        s >> m_base_blockhash;
        s >> m_coins_count;
    }
};
 
const fs::path SNAPSHOT_BLOCKHASH_FILENAME{"base_blockhash"};
 
bool WriteSnapshotBaseBlockhash(Chainstate &snapshot_chainstate)
    EXCLUSIVE_LOCKS_REQUIRED(::cs_main);
 
std::optional<BlockHash> ReadSnapshotBaseBlockhash(const fs::path &chaindir)
    EXCLUSIVE_LOCKS_REQUIRED(::cs_main);
 
constexpr std::string_view SNAPSHOT_CHAINSTATE_SUFFIX = "_snapshot";
 
std::optional<fs::path> FindSnapshotChainstateDir(const fs::path &data_dir);
 
} // namespace node
 
#endif // BITCOIN_NODE_UTXO_SNAPSHOT_H