Branch data     Line data    Source code

       1                 :             : // Copyright (c) 2019-present The Bitcoin Core developers
       2                 :             : // Distributed under the MIT software license, see the accompanying
       3                 :             : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :             : 
       5                 :             : #include <util/string.h>
       6                 :             : 
       7                 :             : #include <iterator>
       8                 :             : #include <memory>
       9                 :             : #include <regex>
      10                 :             : #include <stdexcept>
      11                 :             : #include <string>
      12                 :             : 
      13                 :             : namespace util {
      14                 :         255 : void ReplaceAll(std::string& in_out, const std::string& search, const std::string& substitute)
      15                 :             : {
      16         [ +  + ]:         255 :     if (search.empty()) return;
      17         [ +  - ]:         253 :     in_out = std::regex_replace(in_out, std::regex(search), substitute);
      18                 :             : }
      19                 :             : 
      20                 :          10 : LineReader::LineReader(std::span<const std::byte> buffer, size_t max_line_length)
      21                 :          10 :     : start(buffer.begin()), end(buffer.end()), max_line_length(max_line_length), it(buffer.begin()) {}
      22                 :             : 
      23                 :          22 : std::optional<std::string> LineReader::ReadLine()
      24                 :             : {
      25         [ +  + ]:          22 :     if (it == end) {
      26                 :           6 :         return std::nullopt;
      27                 :             :     }
      28                 :             : 
      29                 :          16 :     auto line_start = it;
      30                 :          16 :     size_t count = 0;
      31         [ +  + ]:         182 :     while (it != end) {
      32                 :             :         // Read a character from the incoming buffer and increment the iterator
      33                 :         181 :         auto c = static_cast<char>(*it);
      34         [ +  + ]:         181 :         ++it;
      35                 :         181 :         ++count;
      36                 :             :         // If the character we just consumed was \n, the line is terminated.
      37                 :             :         // The \n itself does not count against max_line_length.
      38         [ +  + ]:         181 :         if (c == '\n') {
      39                 :          11 :             const std::string_view untrimmed_line(reinterpret_cast<const char*>(std::to_address(line_start)), count);
      40                 :          11 :             const std::string_view line = TrimStringView(untrimmed_line); // delete leading and trailing whitespace including \r and \n
      41                 :          22 :             return std::string(line);
      42                 :             :         }
      43                 :             :         // If the character we just consumed gives us a line length greater
      44                 :             :         // than max_line_length, and we are not at the end of the line (or buffer) yet,
      45                 :             :         // that means the line we are currently reading is too long, and we throw.
      46         [ +  + ]:         170 :         if (count > max_line_length) {
      47                 :             :             // Reset iterator
      48                 :           4 :             it = line_start;
      49         [ +  - ]:           4 :             throw std::runtime_error("max_line_length exceeded by LineReader");
      50                 :             :         }
      51                 :             :     }
      52                 :             :     // End of buffer reached without finding a \n or exceeding max_line_length.
      53                 :             :     // Reset the iterator so the rest of the buffer can be read granularly
      54                 :             :     // with ReadLength() and return null to indicate a line was not found.
      55                 :           1 :     it = line_start;
      56                 :           1 :     return std::nullopt;
      57                 :             : }
      58                 :             : 
      59                 :             : // Ignores max_line_length but won't overflow
      60                 :          11 : std::string LineReader::ReadLength(size_t len)
      61                 :             : {
      62         [ +  + ]:          11 :     if (len == 0) return "";
      63   [ +  +  +  - ]:          10 :     if (Remaining() < len) throw std::runtime_error("Not enough data in buffer");
      64                 :           9 :     std::string out(reinterpret_cast<const char*>(std::to_address(it)), len);
      65                 :           9 :     it += len;
      66                 :           9 :     return out;
      67                 :           9 : }
      68                 :             : 
      69                 :          14 : size_t LineReader::Remaining() const
      70                 :             : {
      71                 :          14 :     return std::distance(it, end);
      72                 :             : }
      73                 :             : } // namespace util