png/decoder: Don't panic on overflow in calculation and inform the user.

Fixes the remaining crashes in #221
etemesi254 · Jul 27, 2024 · 7fee844 · 7fee844
1 parent ee1c2b8
commit 7fee844
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/zune-png/src/decoder.rs b/crates/zune-png/src/decoder.rs
@@ -632,7 +632,8 @@ impl<T: ZByteReaderTrait> PngDecoder<T> {
 
         let png_info = self.png_info.clone();
 
-        let image_len = self.inner_buffer_size().unwrap();
+        let image_len = self.inner_buffer_size()
+            .ok_or(PngDecodeErrors::GenericStatic("Output buffer size overflowed a usize (corrupt png?)"))?;
 
         if out.len() < image_len {
             return Err(PngDecodeErrors::TooSmallOutput(image_len, out.len()));