Limit max memory allocation by fuzz target (#29)

image-rs · Dec 26, 2023 · bb34e55 · bb34e55
1 parent cfd582c
commit bb34e55
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/fuzz/fuzz_targets/decode_still.rs b/fuzz/fuzz_targets/decode_still.rs
@@ -8,7 +8,10 @@ fuzz_target!(|input: &[u8]| {
     if let Ok(mut decoder) = decoder {
         let (width, height) = decoder.dimensions();
         let bytes_per_pixel = if decoder.has_alpha() { 4 } else { 3 };
-        let mut data = vec![0; width as usize * height as usize * bytes_per_pixel];
-        let _ = decoder.read_image(&mut data);
+        let total_bytes = width as usize * height as usize * bytes_per_pixel;
+        if total_bytes <= 1024 * 1024 * 1024 {
+            let mut data = vec![0; total_bytes];
+            let _ = decoder.read_image(&mut data);
+        }
     }
 });