Coverage Report

Created: 2025-03-18 19:34

/root/bitcoin/src/streams.cpp
Line
Count
Source (jump to first uncovered line)
1
// Copyright (c) 2009-present The Bitcoin Core developers
2
// Distributed under the MIT software license, see the accompanying
3
// file COPYING or https://opensource.org/license/mit/.
4
5
#include <memusage.h>
6
#include <span.h>
7
#include <streams.h>
8
#include <util/fs_helpers.h>
9
10
#include <array>
11
12
AutoFile::AutoFile(std::FILE* file, std::vector<std::byte> data_xor)
13
0
    : m_file{file}, m_xor{std::move(data_xor)}
14
0
{
15
0
    if (!IsNull()) {
  Branch (15:9): [True: 0, False: 0]

16
0
        auto pos{std::ftell(m_file)};
17
0
        if (pos >= 0) m_position = pos;
  Branch (17:13): [True: 0, False: 0]

18
0
    }
19
0
}
20
21
std::size_t AutoFile::detail_fread(Span<std::byte> dst)
22
0
{
23
0
    if (!m_file) throw std::ios_base::failure("AutoFile::read: file handle is nullptr");
  Branch (23:9): [True: 0, False: 0]

24
0
    size_t ret = std::fread(dst.data(), 1, dst.size(), m_file);
25
0
    if (!m_xor.empty()) {
  Branch (25:9): [True: 0, False: 0]

26
0
        if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::read: position unknown");
  Branch (26:13): [True: 0, False: 0]

27
0
        util::Xor(dst.subspan(0, ret), m_xor, *m_position);
28
0
    }
29
0
    if (m_position.has_value()) *m_position += ret;
  Branch (29:9): [True: 0, False: 0]

30
0
    return ret;
31
0
}
32
33
void AutoFile::seek(int64_t offset, int origin)
34
0
{
35
0
    if (IsNull()) {
  Branch (35:9): [True: 0, False: 0]

36
0
        throw std::ios_base::failure("AutoFile::seek: file handle is nullptr");
37
0
    }
38
0
    if (std::fseek(m_file, offset, origin) != 0) {
  Branch (38:9): [True: 0, False: 0]

39
0
        throw std::ios_base::failure(feof() ? "AutoFile::seek: end of file" : "AutoFile::seek: fseek failed");
  Branch (39:38): [True: 0, False: 0]

40
0
    }
41
0
    if (origin == SEEK_SET) {
  Branch (41:9): [True: 0, False: 0]

42
0
        m_position = offset;
43
0
    } else if (origin == SEEK_CUR && m_position.has_value()) {
  Branch (43:16): [True: 0, False: 0]
  Branch (43:38): [True: 0, False: 0]

44
0
        *m_position += offset;
45
0
    } else {
46
0
        int64_t r{std::ftell(m_file)};
47
0
        if (r < 0) {
  Branch (47:13): [True: 0, False: 0]

48
0
            throw std::ios_base::failure("AutoFile::seek: ftell failed");
49
0
        }
50
0
        m_position = r;
51
0
    }
52
0
}
53
54
int64_t AutoFile::tell()
55
0
{
56
0
    if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::tell: position unknown");
  Branch (56:9): [True: 0, False: 0]

57
0
    return *m_position;
58
0
}
59
60
void AutoFile::read(Span<std::byte> dst)
61
0
{
62
0
    if (detail_fread(dst) != dst.size()) {
  Branch (62:9): [True: 0, False: 0]

63
0
        throw std::ios_base::failure(feof() ? "AutoFile::read: end of file" : "AutoFile::read: fread failed");
  Branch (63:38): [True: 0, False: 0]

64
0
    }
65
0
}
66
67
void AutoFile::ignore(size_t nSize)
68
0
{
69
0
    if (!m_file) throw std::ios_base::failure("AutoFile::ignore: file handle is nullptr");
  Branch (69:9): [True: 0, False: 0]

70
0
    unsigned char data[4096];
71
0
    while (nSize > 0) {
  Branch (71:12): [True: 0, False: 0]

72
0
        size_t nNow = std::min<size_t>(nSize, sizeof(data));
73
0
        if (std::fread(data, 1, nNow, m_file) != nNow) {
  Branch (73:13): [True: 0, False: 0]

74
0
            throw std::ios_base::failure(feof() ? "AutoFile::ignore: end of file" : "AutoFile::ignore: fread failed");
  Branch (74:42): [True: 0, False: 0]

75
0
        }
76
0
        nSize -= nNow;
77
0
        if (m_position.has_value()) *m_position += nNow;
  Branch (77:13): [True: 0, False: 0]

78
0
    }
79
0
}
80
81
void AutoFile::write(Span<const std::byte> src)
82
0
{
83
0
    if (!m_file) throw std::ios_base::failure("AutoFile::write: file handle is nullptr");
  Branch (83:9): [True: 0, False: 0]

84
0
    if (m_xor.empty()) {
  Branch (84:9): [True: 0, False: 0]

85
0
        if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
  Branch (85:13): [True: 0, False: 0]

86
0
            throw std::ios_base::failure("AutoFile::write: write failed");
87
0
        }
88
0
        if (m_position.has_value()) *m_position += src.size();
  Branch (88:13): [True: 0, False: 0]

89
0
    } else {
90
0
        if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::write: position unknown");
  Branch (90:13): [True: 0, False: 0]

91
0
        std::array<std::byte, 4096> buf;
92
0
        while (src.size() > 0) {
  Branch (92:16): [True: 0, False: 0]

93
0
            auto buf_now{Span{buf}.first(std::min<size_t>(src.size(), buf.size()))};
94
0
            std::copy(src.begin(), src.begin() + buf_now.size(), buf_now.begin());
95
0
            util::Xor(buf_now, m_xor, *m_position);
96
0
            if (std::fwrite(buf_now.data(), 1, buf_now.size(), m_file) != buf_now.size()) {
  Branch (96:17): [True: 0, False: 0]

97
0
                throw std::ios_base::failure{"XorFile::write: failed"};
98
0
            }
99
0
            src = src.subspan(buf_now.size());
100
0
            *m_position += buf_now.size();
101
0
        }
102
0
    }
103
0
}
104
105
bool AutoFile::Commit()
106
0
{
107
0
    return ::FileCommit(m_file);
108
0
}
109
110
bool AutoFile::Truncate(unsigned size)
111
0
{
112
0
    return ::TruncateFile(m_file, size);
113
0
}
114
115
size_t DataStream::GetMemoryUsage() const noexcept
116
0
{
117
0
    return sizeof(*this) + memusage::DynamicUsage(vch);
118
0
}