Commit 8748b2dc authored by Max Kellermann's avatar Max Kellermann

fs/io/BufferedReader: add ReadFull(size_t)

parent 5f66af37
......@@ -53,6 +53,19 @@ BufferedReader::Fill(bool need_more)
return true;
}
void *
BufferedReader::ReadFull(size_t size)
{
while (true) {
auto r = Read();
if (r.size >= size)
return r.data;
if (!Fill(true))
throw std::runtime_error("Premature end of file");
}
}
size_t
BufferedReader::ReadFromBuffer(WritableBuffer<void> dest)
{
......
......@@ -51,6 +51,14 @@ public:
return buffer.Read().ToVoid();
}
/**
* Read a buffer of exactly the given size (without consuming
* it). Throws std::runtime_error if not enough data is
* available.
*/
gcc_pure
void *ReadFull(size_t size);
void Consume(size_t n) {
buffer.Consume(n);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment