From 026a49ffa7229a905a4e829b6843a583b605a880 Mon Sep 17 00:00:00 2001
From: Chris Hennick <hennickc@amazon.com>
Date: Wed, 28 Feb 2024 17:22:43 -0800
Subject: [PATCH] Truncate huge files during read fuzz rather than skipping
 them

---
 fuzz/fuzz_targets/fuzz_read.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/fuzz/fuzz_targets/fuzz_read.rs b/fuzz/fuzz_targets/fuzz_read.rs
index 13a4ac29..dde824d6 100644
--- a/fuzz/fuzz_targets/fuzz_read.rs
+++ b/fuzz/fuzz_targets/fuzz_read.rs
@@ -1,15 +1,14 @@
 #![no_main]
 use libfuzzer_sys::fuzz_target;
+use std::io::Read;
 
 fn decompress_all(data: &[u8]) -> Result<(), Box<dyn std::error::Error>> {
     let reader = std::io::Cursor::new(data);
     let mut zip = zip_next::ZipArchive::new(reader)?;
 
     for i in 0..zip.len() {
-        let mut file = zip.by_index(i)?;
-        if file.size() <= 1 << 24 {
-            let _ = std::io::copy(&mut file, &mut std::io::sink());
-        }
+        let file = zip.by_index(i)?;
+        std::io::copy(&mut file.take(1 << 24), &mut std::io::sink())?;
     }
 
     Ok(())