diff --git a/src/hash.h b/src/hash.h index b18a031268..b2ef29fcdb 100644 --- a/src/hash.h +++ b/src/hash.h @@ -6,6 +6,7 @@ #ifndef BITCOIN_HASH_H #define BITCOIN_HASH_H +#include #include #include #include @@ -165,6 +166,39 @@ public: }; /** Reads data from an underlying stream, while hashing the read data. */ +template +class HashVerifier : public HashWriter +{ +private: + Source& m_source; + +public: + explicit HashVerifier(Source& source LIFETIMEBOUND) : m_source{source} {} + + void read(Span dst) + { + m_source.read(dst); + this->write(dst); + } + + void ignore(size_t num_bytes) + { + std::byte data[1024]; + while (num_bytes > 0) { + size_t now = std::min(num_bytes, 1024); + read({data, now}); + num_bytes -= now; + } + } + + template + HashVerifier& operator>>(T&& obj) + { + ::Unserialize(*this, obj); + return *this; + } +}; + template class CHashVerifier : public CHashWriter {