mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-04 10:07:27 -05:00
be55f545d5
This is an extraction of ArgsManager related functions from util/system into their own common file. Config file related functions are moved to common/config.cpp. The background of this commit is an ongoing effort to decouple the libbitcoinkernel library from the ArgsManager. The ArgsManager belongs into the common library, since the kernel library should not depend on it. See doc/design/libraries.md for more information on this rationale.
96 lines
3 KiB
C++
96 lines
3 KiB
C++
// 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.
|
|
|
|
#include <node/utxo_snapshot.h>
|
|
|
|
#include <common/args.h>
|
|
#include <logging.h>
|
|
#include <streams.h>
|
|
#include <sync.h>
|
|
#include <tinyformat.h>
|
|
#include <txdb.h>
|
|
#include <uint256.h>
|
|
#include <util/fs.h>
|
|
#include <validation.h>
|
|
|
|
#include <cassert>
|
|
#include <cstdio>
|
|
#include <optional>
|
|
#include <string>
|
|
|
|
namespace node {
|
|
|
|
bool WriteSnapshotBaseBlockhash(Chainstate& snapshot_chainstate)
|
|
{
|
|
AssertLockHeld(::cs_main);
|
|
assert(snapshot_chainstate.m_from_snapshot_blockhash);
|
|
|
|
const std::optional<fs::path> chaindir = snapshot_chainstate.CoinsDB().StoragePath();
|
|
assert(chaindir); // Sanity check that chainstate isn't in-memory.
|
|
const fs::path write_to = *chaindir / node::SNAPSHOT_BLOCKHASH_FILENAME;
|
|
|
|
FILE* file{fsbridge::fopen(write_to, "wb")};
|
|
AutoFile afile{file};
|
|
if (afile.IsNull()) {
|
|
LogPrintf("[snapshot] failed to open base blockhash file for writing: %s\n",
|
|
fs::PathToString(write_to));
|
|
return false;
|
|
}
|
|
afile << *snapshot_chainstate.m_from_snapshot_blockhash;
|
|
|
|
if (afile.fclose() != 0) {
|
|
LogPrintf("[snapshot] failed to close base blockhash file %s after writing\n",
|
|
fs::PathToString(write_to));
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
std::optional<uint256> ReadSnapshotBaseBlockhash(fs::path chaindir)
|
|
{
|
|
if (!fs::exists(chaindir)) {
|
|
LogPrintf("[snapshot] cannot read base blockhash: no chainstate dir " /* Continued */
|
|
"exists at path %s\n", fs::PathToString(chaindir));
|
|
return std::nullopt;
|
|
}
|
|
const fs::path read_from = chaindir / node::SNAPSHOT_BLOCKHASH_FILENAME;
|
|
const std::string read_from_str = fs::PathToString(read_from);
|
|
|
|
if (!fs::exists(read_from)) {
|
|
LogPrintf("[snapshot] snapshot chainstate dir is malformed! no base blockhash file " /* Continued */
|
|
"exists at path %s. Try deleting %s and calling loadtxoutset again?\n",
|
|
fs::PathToString(chaindir), read_from_str);
|
|
return std::nullopt;
|
|
}
|
|
|
|
uint256 base_blockhash;
|
|
FILE* file{fsbridge::fopen(read_from, "rb")};
|
|
AutoFile afile{file};
|
|
if (afile.IsNull()) {
|
|
LogPrintf("[snapshot] failed to open base blockhash file for reading: %s\n",
|
|
read_from_str);
|
|
return std::nullopt;
|
|
}
|
|
afile >> base_blockhash;
|
|
|
|
if (std::fgetc(afile.Get()) != EOF) {
|
|
LogPrintf("[snapshot] warning: unexpected trailing data in %s\n", read_from_str);
|
|
} else if (std::ferror(afile.Get())) {
|
|
LogPrintf("[snapshot] warning: i/o error reading %s\n", read_from_str);
|
|
}
|
|
return base_blockhash;
|
|
}
|
|
|
|
std::optional<fs::path> FindSnapshotChainstateDir()
|
|
{
|
|
fs::path possible_dir =
|
|
gArgs.GetDataDirNet() / fs::u8path(strprintf("chainstate%s", SNAPSHOT_CHAINSTATE_SUFFIX));
|
|
|
|
if (fs::exists(possible_dir)) {
|
|
return possible_dir;
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
} // namespace node
|