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 : :
10 : : #include <array>
11 : :
12 : 192673 : AutoFile::AutoFile(std::FILE* file, std::vector<std::byte> data_xor)
13 [ + + ]: 192673 : : m_file{file}, m_xor{std::move(data_xor)}
14 : : {
15 [ + + ]: 192673 : if (!IsNull()) {
16 [ + - ]: 190663 : auto pos{std::ftell(m_file)};
17 [ + + ]: 190663 : if (pos >= 0) m_position = pos;
18 : : }
19 : 192673 : }
20 : :
21 : 9941901 : std::size_t AutoFile::detail_fread(Span<std::byte> dst)
22 : : {
23 [ + + + - ]: 9941901 : if (!m_file) throw std::ios_base::failure("AutoFile::read: file handle is nullptr");
24 [ + - ]: 9941410 : size_t ret = std::fread(dst.data(), 1, dst.size(), m_file);
25 [ + + ]: 9941410 : if (!m_xor.empty()) {
26 [ + + + - ]: 595616 : if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::read: position unknown");
27 : 595164 : util::Xor(dst.subspan(0, ret), m_xor, *m_position);
28 : : }
29 [ + + ]: 9940958 : if (m_position.has_value()) *m_position += ret;
30 : 9940958 : return ret;
31 : : }
32 : :
33 : 223 : void AutoFile::seek(int64_t offset, int origin)
34 : : {
35 [ - + ]: 223 : if (IsNull()) {
36 [ # # ]: 0 : throw std::ios_base::failure("AutoFile::seek: file handle is nullptr");
37 : : }
38 [ - + ]: 223 : 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 [ + + ]: 223 : if (origin == SEEK_SET) {
42 : 89 : m_position = offset;
43 [ + + + - ]: 134 : } else if (origin == SEEK_CUR && m_position.has_value()) {
44 : 74 : *m_position += offset;
45 : : } else {
46 : 60 : int64_t r{std::ftell(m_file)};
47 [ - + ]: 60 : if (r < 0) {
48 [ # # ]: 0 : throw std::ios_base::failure("AutoFile::seek: ftell failed");
49 : : }
50 : 60 : m_position = r;
51 : : }
52 : 223 : }
53 : :
54 : 174835 : int64_t AutoFile::tell()
55 : : {
56 [ - + - - ]: 174835 : if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::tell: position unknown");
57 : 174835 : return *m_position;
58 : : }
59 : :
60 : 9556613 : void AutoFile::read(Span<std::byte> dst)
61 : : {
62 [ + + ]: 9556613 : if (detail_fread(dst) != dst.size()) {
63 [ + + + - ]: 3602 : throw std::ios_base::failure(feof() ? "AutoFile::read: end of file" : "AutoFile::read: fread failed");
64 : : }
65 : 9552958 : }
66 : :
67 : 820 : void AutoFile::ignore(size_t nSize)
68 : : {
69 [ + + + - ]: 820 : if (!m_file) throw std::ios_base::failure("AutoFile::ignore: file handle is nullptr");
70 : : unsigned char data[4096];
71 [ + + ]: 1534 : while (nSize > 0) {
72 [ + + ]: 1180 : size_t nNow = std::min<size_t>(nSize, sizeof(data));
73 [ - + + + ]: 2360 : if (std::fread(data, 1, nNow, m_file) != nNow) {
74 [ + + + - ]: 478 : throw std::ios_base::failure(feof() ? "AutoFile::ignore: end of file" : "AutoFile::ignore: fread failed");
75 : : }
76 : 835 : nSize -= nNow;
77 [ + + ]: 835 : if (m_position.has_value()) *m_position += nNow;
78 : : }
79 : 354 : }
80 : :
81 : 4117513 : void AutoFile::write(Span<const std::byte> src)
82 : : {
83 [ + + + - ]: 4117513 : if (!m_file) throw std::ios_base::failure("AutoFile::write: file handle is nullptr");
84 [ + + ]: 4116512 : if (m_xor.empty()) {
85 [ + + ]: 949251 : if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
86 [ + - ]: 2342 : throw std::ios_base::failure("AutoFile::write: write failed");
87 : : }
88 [ + + ]: 946909 : if (m_position.has_value()) *m_position += src.size();
89 : : } else {
90 [ + + + - ]: 3167261 : if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::write: position unknown");
91 : : std::array<std::byte, 4096> buf;
92 [ + + ]: 6333403 : while (src.size() > 0) {
93 [ + - ]: 6333508 : auto buf_now{Span{buf}.first(std::min<size_t>(src.size(), buf.size()))};
94 : 3166754 : std::copy(src.begin(), src.begin() + buf_now.size(), buf_now.begin());
95 : 3166754 : util::Xor(buf_now, m_xor, *m_position);
96 [ + + ]: 3166754 : if (std::fwrite(buf_now.data(), 1, buf_now.size(), m_file) != buf_now.size()) {
97 [ + - ]: 176 : throw std::ios_base::failure{"XorFile::write: failed"};
98 : : }
99 : 3166578 : src = src.subspan(buf_now.size());
100 : 3166578 : *m_position += buf_now.size();
101 : : }
102 : : }
103 : 4113558 : }
104 : :
105 : 0 : bool AutoFile::Commit()
106 : : {
107 : 0 : return ::FileCommit(m_file);
108 : : }
109 : :
110 : 0 : bool AutoFile::Truncate(unsigned size)
111 : : {
112 : 0 : return ::TruncateFile(m_file, size);
113 : : }
114 : :
115 : 192615 : size_t DataStream::GetMemoryUsage() const noexcept
116 : : {
117 [ + + ]: 192615 : return sizeof(*this) + memusage::DynamicUsage(vch);
118 : : }
|
