From aa890fa634785784bfce19d897828fa6c348556c Mon Sep 17 00:00:00 2001
From: Chris Hennick <4961925+Pr0methean@users.noreply.github.com>
Date: Sun, 5 May 2024 18:28:17 -0700
Subject: [PATCH] feat: Add method `decompressed_size()` so non-recursive ZIP
 bombs can be detected

---
 src/read.rs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/read.rs b/src/read.rs
index 10b9eaaf..55dc322c 100644
--- a/src/read.rs
+++ b/src/read.rs
@@ -356,6 +356,19 @@ impl<R> ZipArchive<R> {
             comment: comment.into_boxed_slice().into(),
         })
     }
+
+    /// Total size of the files in the archive, if it can be known. Doesn't include directories or 
+    /// metadata.
+    pub fn decompressed_size(&self) -> Option<u128> {
+        let mut total = 0u128;
+        for file in self.shared.files.values() {
+            if file.using_data_descriptor {
+                return None;
+            }
+            total = total.checked_add(file.uncompressed_size as u128)?;
+        }
+        Some(total)
+    }
 }
 
 impl<R: Read + Seek> ZipArchive<R> {