Fix panic attempting to encode zero sized webp

image-rs · Aug 4, 2023 · cb92511 · cb92511
1 parent 814596e
commit cb92511
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/codecs/webp/encoder.rs b/src/codecs/webp/encoder.rs
@@ -94,7 +94,11 @@ impl<W: Write> WebPEncoder<W> {
         };
 
         // Validate dimensions upfront to avoid panics.
-        if !SampleLayout::row_major_packed(color.channel_count(), width, height).fits(data.len()) {
+        if width == 0
+            || height == 0
+            || !SampleLayout::row_major_packed(color.channel_count(), width, height)
+                .fits(data.len())
+        {
             return Err(ImageError::Parameter(ParameterError::from_kind(
                 ParameterErrorKind::DimensionMismatch,
             )));