Branch data Line data Source code
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 : : #include <util/obfuscation.h>
10 : :
11 : : #include <array>
12 : :
13 [ + + ]: 371995 : AutoFile::AutoFile(std::FILE* file, const Obfuscation& obfuscation) : m_file{file}, m_obfuscation{obfuscation}
14 : : {
15 [ + + ]: 371995 : if (!IsNull()) {
16 : 345280 : auto pos{std::ftell(m_file)};
17 [ + + ]: 345280 : if (pos >= 0) m_position = pos;
18 : : }
19 : 371995 : }
20 : :
21 : 38110989 : std::size_t AutoFile::detail_fread(std::span<std::byte> dst)
22 : : {
23 [ + + + - ]: 38111691 : if (!m_file) throw std::ios_base::failure("AutoFile::read: file handle is nullptr");
24 [ + - ]: 38110287 : const size_t ret = std::fread(dst.data(), 1, dst.size(), m_file);
25 [ + + ]: 38110287 : if (m_obfuscation) {
26 [ + + + - ]: 887000 : if (!m_position) throw std::ios_base::failure("AutoFile::read: position unknown");
27 [ - + ]: 885810 : m_obfuscation(dst.subspan(0, ret), *m_position);
28 : : }
29 [ + + ]: 38109692 : if (m_position) *m_position += ret;
30 : 38109692 : return ret;
31 : : }
32 : :
33 : 410 : void AutoFile::seek(int64_t offset, int origin)
34 : : {
35 [ - + ]: 410 : if (IsNull()) {
36 [ # # ]: 0 : throw std::ios_base::failure("AutoFile::seek: file handle is nullptr");
37 : : }
38 [ - + ]: 410 : if (std::fseek(m_file, offset, origin) != 0) {
39 [ # # # # ]: 0 : throw std::ios_base::failure(feof() ? "AutoFile::seek: end of file" : "AutoFile::seek: fseek failed");
40 : : }
41 [ + + ]: 410 : if (origin == SEEK_SET) {
42 : 224 : m_position = offset;
43 [ + + + - ]: 186 : } else if (origin == SEEK_CUR && m_position.has_value()) {
44 : 101 : *m_position += offset;
45 : : } else {
46 : 85 : int64_t r{std::ftell(m_file)};
47 [ - + ]: 85 : if (r < 0) {
48 [ # # ]: 0 : throw std::ios_base::failure("AutoFile::seek: ftell failed");
49 : : }
50 : 85 : m_position = r;
51 : : }
52 : 410 : }
53 : :
54 : 170 : int64_t AutoFile::tell()
55 : : {
56 [ - + - - ]: 170 : if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::tell: position unknown");
57 : 170 : return *m_position;
58 : : }
59 : :
60 : 85 : int64_t AutoFile::size()
61 : : {
62 [ - + ]: 85 : if (IsNull()) {
63 [ # # ]: 0 : throw std::ios_base::failure("AutoFile::size: file handle is nullptr");
64 : : }
65 : : // Temporarily save the current position
66 : 85 : int64_t current_pos = tell();
67 : 85 : seek(0, SEEK_END);
68 : 85 : int64_t file_size = tell();
69 : : // Restore the original position
70 : 85 : seek(current_pos, SEEK_SET);
71 : 85 : return file_size;
72 : : }
73 : :
74 : 37662495 : void AutoFile::read(std::span<std::byte> dst)
75 : : {
76 [ + + ]: 37662495 : if (detail_fread(dst) != dst.size()) {
77 [ + + + - ]: 13159 : throw std::ios_base::failure(feof() ? "AutoFile::read: end of file" : "AutoFile::read: fread failed");
78 : : }
79 : 37655645 : }
80 : :
81 : 2038 : void AutoFile::ignore(size_t nSize)
82 : : {
83 [ + + + - ]: 2261 : if (!m_file) throw std::ios_base::failure("AutoFile::ignore: file handle is nullptr");
84 : : unsigned char data[4096];
85 [ + + ]: 5265 : while (nSize > 0) {
86 [ + + ]: 4022 : size_t nNow = std::min<size_t>(nSize, sizeof(data));
87 [ - + + + ]: 8044 : if (std::fread(data, 1, nNow, m_file) != nNow) {
88 [ + + + - ]: 1413 : throw std::ios_base::failure(feof() ? "AutoFile::ignore: end of file" : "AutoFile::ignore: fread failed");
89 : : }
90 : 3450 : nSize -= nNow;
91 [ + + ]: 3450 : if (m_position.has_value()) *m_position += nNow;
92 : : }
93 : 1243 : }
94 : :
95 : 2467199 : void AutoFile::write(std::span<const std::byte> src)
96 : : {
97 [ + + + - ]: 2468500 : if (!m_file) throw std::ios_base::failure("AutoFile::write: file handle is nullptr");
98 [ + + ]: 2465898 : if (!m_obfuscation) {
99 [ + + ]: 1930809 : if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
100 [ + - ]: 6300 : throw std::ios_base::failure("AutoFile::write: write failed");
101 : : }
102 : 1927659 : m_was_written = true;
103 [ + + ]: 1927659 : if (m_position.has_value()) *m_position += src.size();
104 : : } else {
105 : : std::array<std::byte, 4096> buf;
106 [ + + ]: 1068971 : while (src.size()) {
107 [ + - ]: 1069658 : auto buf_now{std::span{buf}.first(std::min<size_t>(src.size(), buf.size()))};
108 : 534829 : std::copy_n(src.begin(), buf_now.size(), buf_now.begin());
109 : 534829 : write_buffer(buf_now);
110 : 533882 : src = src.subspan(buf_now.size());
111 : : }
112 : : }
113 : 2461801 : }
114 : :
115 : 860835 : void AutoFile::write_buffer(std::span<std::byte> src)
116 : : {
117 [ - + - - ]: 860835 : if (!m_file) throw std::ios_base::failure("AutoFile::write_buffer: file handle is nullptr");
118 [ + - ]: 860835 : if (m_obfuscation) {
119 [ + + + - ]: 861144 : if (!m_position) throw std::ios_base::failure("AutoFile::write_buffer: obfuscation position unknown");
120 : 860526 : m_obfuscation(src, *m_position); // obfuscate in-place
121 : : }
122 [ + + ]: 860526 : if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
123 [ + - ]: 1276 : throw std::ios_base::failure("AutoFile::write_buffer: write failed");
124 : : }
125 : 859888 : m_was_written = true;
126 [ + - ]: 859888 : if (m_position) *m_position += src.size();
127 : 859888 : }
128 : :
129 : 0 : bool AutoFile::Commit()
130 : : {
131 : 0 : return ::FileCommit(m_file);
132 : : }
133 : :
134 : 0 : bool AutoFile::Truncate(unsigned size)
135 : : {
136 : 0 : m_was_written = true;
137 : 0 : return ::TruncateFile(m_file, size);
138 : : }
139 : :
140 : 295161 : size_t DataStream::GetMemoryUsage() const noexcept
141 : : {
142 [ - + ]: 295161 : return sizeof(*this) + memusage::DynamicUsage(vch);
143 : : }
|
