mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-12 12:52:35 -03:00
f64aa9c411
Add more fs::path operator/ and operator+ overloads to prevent unsafe string->path conversions on Windows that would cause strings to be decoded according to the current Windows locale & code page instead of the correct string encoding. Update application code to deal with loss of implicit string->path conversions by calling fs::u8path or fs::PathFromString explicitly, or by just changing variable types from std::string to fs::path to avoid conversions altoghther, or make them happen earlier. In all cases, there's no change in behavior either (1) because strings only contained ASCII characters and would be decoded the same regardless of what encoding was used, or (2) because of the 1:1 mapping between paths and strings using the PathToString and PathFromString functions. Co-authored-by: Hennadii Stepanov <32963518+hebasto@users.noreply.github.com>
99 lines
3 KiB
C++
99 lines
3 KiB
C++
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
// Copyright (c) 2009-2021 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 <stdexcept>
|
|
|
|
#include <flatfile.h>
|
|
#include <logging.h>
|
|
#include <tinyformat.h>
|
|
#include <util/system.h>
|
|
|
|
FlatFileSeq::FlatFileSeq(fs::path dir, const char* prefix, size_t chunk_size) :
|
|
m_dir(std::move(dir)),
|
|
m_prefix(prefix),
|
|
m_chunk_size(chunk_size)
|
|
{
|
|
if (chunk_size == 0) {
|
|
throw std::invalid_argument("chunk_size must be positive");
|
|
}
|
|
}
|
|
|
|
std::string FlatFilePos::ToString() const
|
|
{
|
|
return strprintf("FlatFilePos(nFile=%i, nPos=%i)", nFile, nPos);
|
|
}
|
|
|
|
fs::path FlatFileSeq::FileName(const FlatFilePos& pos) const
|
|
{
|
|
return m_dir / fs::u8path(strprintf("%s%05u.dat", m_prefix, pos.nFile));
|
|
}
|
|
|
|
FILE* FlatFileSeq::Open(const FlatFilePos& pos, bool read_only)
|
|
{
|
|
if (pos.IsNull()) {
|
|
return nullptr;
|
|
}
|
|
fs::path path = FileName(pos);
|
|
fs::create_directories(path.parent_path());
|
|
FILE* file = fsbridge::fopen(path, read_only ? "rb": "rb+");
|
|
if (!file && !read_only)
|
|
file = fsbridge::fopen(path, "wb+");
|
|
if (!file) {
|
|
LogPrintf("Unable to open file %s\n", fs::PathToString(path));
|
|
return nullptr;
|
|
}
|
|
if (pos.nPos && fseek(file, pos.nPos, SEEK_SET)) {
|
|
LogPrintf("Unable to seek to position %u of %s\n", pos.nPos, fs::PathToString(path));
|
|
fclose(file);
|
|
return nullptr;
|
|
}
|
|
return file;
|
|
}
|
|
|
|
size_t FlatFileSeq::Allocate(const FlatFilePos& pos, size_t add_size, bool& out_of_space)
|
|
{
|
|
out_of_space = false;
|
|
|
|
unsigned int n_old_chunks = (pos.nPos + m_chunk_size - 1) / m_chunk_size;
|
|
unsigned int n_new_chunks = (pos.nPos + add_size + m_chunk_size - 1) / m_chunk_size;
|
|
if (n_new_chunks > n_old_chunks) {
|
|
size_t old_size = pos.nPos;
|
|
size_t new_size = n_new_chunks * m_chunk_size;
|
|
size_t inc_size = new_size - old_size;
|
|
|
|
if (CheckDiskSpace(m_dir, inc_size)) {
|
|
FILE *file = Open(pos);
|
|
if (file) {
|
|
LogPrint(BCLog::VALIDATION, "Pre-allocating up to position 0x%x in %s%05u.dat\n", new_size, m_prefix, pos.nFile);
|
|
AllocateFileRange(file, pos.nPos, inc_size);
|
|
fclose(file);
|
|
return inc_size;
|
|
}
|
|
} else {
|
|
out_of_space = true;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
bool FlatFileSeq::Flush(const FlatFilePos& pos, bool finalize)
|
|
{
|
|
FILE* file = Open(FlatFilePos(pos.nFile, 0)); // Avoid fseek to nPos
|
|
if (!file) {
|
|
return error("%s: failed to open file %d", __func__, pos.nFile);
|
|
}
|
|
if (finalize && !TruncateFile(file, pos.nPos)) {
|
|
fclose(file);
|
|
return error("%s: failed to truncate file %d", __func__, pos.nFile);
|
|
}
|
|
if (!FileCommit(file)) {
|
|
fclose(file);
|
|
return error("%s: failed to commit file %d", __func__, pos.nFile);
|
|
}
|
|
DirectoryCommit(m_dir);
|
|
|
|
fclose(file);
|
|
return true;
|
|
}
|