LCOV - code coverage report
Current view: top level - src - streams.cpp (source / functions) Coverage Total Hit
Test: fuzz_coverage.info Lines: 89.0 % 73 65
Test Date: 2025-07-28 04:06:55 Functions: 81.8 % 11 9
Branches: 68.3 % 104 71

             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         [ +  + ]:      340207 : AutoFile::AutoFile(std::FILE* file, const Obfuscation& obfuscation) : m_file{file}, m_obfuscation{obfuscation}
      14                 :             : {
      15         [ +  + ]:      340207 :     if (!IsNull()) {
      16                 :      310009 :         auto pos{std::ftell(m_file)};
      17         [ +  + ]:      310009 :         if (pos >= 0) m_position = pos;
      18                 :             :     }
      19                 :      340207 : }
      20                 :             : 
      21                 :    32008238 : std::size_t AutoFile::detail_fread(std::span<std::byte> dst)
      22                 :             : {
      23   [ +  +  +  - ]:    32008238 :     if (!m_file) throw std::ios_base::failure("AutoFile::read: file handle is nullptr");
      24         [ +  - ]:    32007264 :     const size_t ret = std::fread(dst.data(), 1, dst.size(), m_file);
      25         [ +  + ]:    32007264 :     if (m_obfuscation) {
      26   [ +  +  +  - ]:      667664 :         if (!m_position) throw std::ios_base::failure("AutoFile::read: position unknown");
      27         [ -  + ]:      665824 :         m_obfuscation(dst.subspan(0, ret), *m_position);
      28                 :             :     }
      29         [ +  + ]:    32005424 :     if (m_position) *m_position += ret;
      30                 :    32005424 :     return ret;
      31                 :             : }
      32                 :             : 
      33                 :         272 : void AutoFile::seek(int64_t offset, int origin)
      34                 :             : {
      35         [ -  + ]:         272 :     if (IsNull()) {
      36         [ #  # ]:           0 :         throw std::ios_base::failure("AutoFile::seek: file handle is nullptr");
      37                 :             :     }
      38         [ -  + ]:         272 :     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         [ +  + ]:         272 :     if (origin == SEEK_SET) {
      42                 :         111 :         m_position = offset;
      43   [ +  +  +  - ]:         161 :     } else if (origin == SEEK_CUR && m_position.has_value()) {
      44                 :          91 :         *m_position += offset;
      45                 :             :     } else {
      46                 :          70 :         int64_t r{std::ftell(m_file)};
      47         [ -  + ]:          70 :         if (r < 0) {
      48         [ #  # ]:           0 :             throw std::ios_base::failure("AutoFile::seek: ftell failed");
      49                 :             :         }
      50                 :          70 :         m_position = r;
      51                 :             :     }
      52                 :         272 : }
      53                 :             : 
      54                 :          70 : int64_t AutoFile::tell()
      55                 :             : {
      56   [ -  +  -  - ]:          70 :     if (!m_position.has_value()) throw std::ios_base::failure("AutoFile::tell: position unknown");
      57                 :          70 :     return *m_position;
      58                 :             : }
      59                 :             : 
      60                 :    31514057 : void AutoFile::read(std::span<std::byte> dst)
      61                 :             : {
      62         [ +  + ]:    31514057 :     if (detail_fread(dst) != dst.size()) {
      63   [ +  +  +  - ]:        4981 :         throw std::ios_base::failure(feof() ? "AutoFile::read: end of file" : "AutoFile::read: fread failed");
      64                 :             :     }
      65                 :    31508454 : }
      66                 :             : 
      67                 :        1646 : void AutoFile::ignore(size_t nSize)
      68                 :             : {
      69   [ +  +  +  - ]:        1646 :     if (!m_file) throw std::ios_base::failure("AutoFile::ignore: file handle is nullptr");
      70                 :             :     unsigned char data[4096];
      71         [ +  + ]:        3049 :     while (nSize > 0) {
      72         [ +  + ]:        2196 :         size_t nNow = std::min<size_t>(nSize, sizeof(data));
      73   [ -  +  +  + ]:        4392 :         if (std::fread(data, 1, nNow, m_file) != nNow) {
      74   [ +  +  +  - ]:         904 :             throw std::ios_base::failure(feof() ? "AutoFile::ignore: end of file" : "AutoFile::ignore: fread failed");
      75                 :             :         }
      76                 :        1611 :         nSize -= nNow;
      77         [ +  + ]:        1611 :         if (m_position.has_value()) *m_position += nNow;
      78                 :             :     }
      79                 :         853 : }
      80                 :             : 
      81                 :     1898201 : void AutoFile::write(std::span<const std::byte> src)
      82                 :             : {
      83   [ +  +  +  - ]:     1898201 :     if (!m_file) throw std::ios_base::failure("AutoFile::write: file handle is nullptr");
      84         [ +  + ]:     1896801 :     if (!m_obfuscation) {
      85         [ +  + ]:     1367939 :         if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
      86         [ +  - ]:        2573 :             throw std::ios_base::failure("AutoFile::write: write failed");
      87                 :             :         }
      88                 :     1365366 :         m_was_written = true;
      89         [ +  + ]:     1365366 :         if (m_position.has_value()) *m_position += src.size();
      90                 :             :     } else {
      91                 :             :         std::array<std::byte, 4096> buf;
      92         [ +  + ]:     1056265 :         while (src.size()) {
      93         [ +  - ]:     1057266 :             auto buf_now{std::span{buf}.first(std::min<size_t>(src.size(), buf.size()))};
      94                 :      528633 :             std::copy_n(src.begin(), buf_now.size(), buf_now.begin());
      95                 :      528633 :             write_buffer(buf_now);
      96                 :      527403 :             src = src.subspan(buf_now.size());
      97                 :             :         }
      98                 :             :     }
      99                 :     1892998 : }
     100                 :             : 
     101                 :      818117 : void AutoFile::write_buffer(std::span<std::byte> src)
     102                 :             : {
     103   [ -  +  -  - ]:      818117 :     if (!m_file) throw std::ios_base::failure("AutoFile::write_buffer: file handle is nullptr");
     104         [ +  - ]:      818117 :     if (m_obfuscation) {
     105   [ +  +  +  - ]:      818117 :         if (!m_position) throw std::ios_base::failure("AutoFile::write_buffer: obfuscation position unknown");
     106                 :      817629 :         m_obfuscation(src, *m_position); // obfuscate in-place
     107                 :             :     }
     108         [ +  + ]:      817629 :     if (std::fwrite(src.data(), 1, src.size(), m_file) != src.size()) {
     109         [ +  - ]:         742 :         throw std::ios_base::failure("AutoFile::write_buffer: write failed");
     110                 :             :     }
     111                 :      816887 :     m_was_written = true;
     112         [ +  - ]:      816887 :     if (m_position) *m_position += src.size();
     113                 :      816887 : }
     114                 :             : 
     115                 :           0 : bool AutoFile::Commit()
     116                 :             : {
     117                 :           0 :     return ::FileCommit(m_file);
     118                 :             : }
     119                 :             : 
     120                 :           0 : bool AutoFile::Truncate(unsigned size)
     121                 :             : {
     122                 :           0 :     m_was_written = true;
     123                 :           0 :     return ::TruncateFile(m_file, size);
     124                 :             : }
     125                 :             : 
     126                 :      325333 : size_t DataStream::GetMemoryUsage() const noexcept
     127                 :             : {
     128         [ +  + ]:      325333 :     return sizeof(*this) + memusage::DynamicUsage(vch);
     129                 :             : }
        

Generated by: LCOV version 2.0-1