From 7f59516611b65dee598966034d526ff1e6e00cf5 Mon Sep 17 00:00:00 2001
From: Joel Klinghed <the_jk@spawned.biz>
Date: Mon, 15 Sep 2025 20:56:50 +0200
Subject: decompress: Return better io error for BUF_ERROR

Use new MaxTooSmall. As the comment notes tho, it might be that we
are lacking input as well, but until I figure out how to test for
that case and determine the cause, lets at least return a more
specific error.
---
 src/decompress_z.cc | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'src/decompress_z.cc')

diff --git a/src/decompress_z.cc b/src/decompress_z.cc
index f9f87ae..e888e77 100644
--- a/src/decompress_z.cc
+++ b/src/decompress_z.cc
@@ -62,9 +62,16 @@ class DecompressReader : public io::Reader {
       if (!in_eof_)
         buffer_->consume(stream_.next_in - rptr);
     } else {
-      return std::unexpected(
-          ret == Z_DATA_ERROR
-          ? io::ReadError::InvalidData : io::ReadError::Error);
+      switch (ret) {
+        case Z_DATA_ERROR:
+          return std::unexpected(io::ReadError::InvalidData);
+        case Z_BUF_ERROR:
+          // Tricky, could also be because of too little input.
+          // TODO: Improve logic here if we can
+          return std::unexpected(io::ReadError::MaxTooSmall);
+        default:
+          return std::unexpected(io::ReadError::Error);
+      }
     }
     return got;
   }
-- 
cgit v1.3