Make use of CompressorImpl::extra_size

Dr-Emann · Mar 20, 2024 · 0a30154 · 0a30154
1 parent 18ff70b
commit 0a30154
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/crates/applesauce-core/src/compressor/mod.rs b/crates/applesauce-core/src/compressor/mod.rs
@@ -21,7 +21,6 @@ mod zlib;
 pub(crate) trait CompressorImpl {
     /// The offset to start data at, for the specified number of blocks
     #[must_use]
-
     fn header_size(block_count: u64) -> u64;
 
     #[must_use]

diff --git a/crates/applesauce-core/src/compressor/zlib.rs b/crates/applesauce-core/src/compressor/zlib.rs
@@ -131,6 +131,15 @@ impl super::CompressorImpl for Zlib {
             u32::try_from(writer.stream_position()?).map_err(|_| io::ErrorKind::InvalidInput)?;
         writer.write_all(&ZLIB_TRAILER)?;
 
+        // This is logically a non-modifying operation, even if it takes &mut self, and can fail
+        #[allow(clippy::debug_assert_with_mut_call)]
+        {
+            debug_assert_eq!(
+                writer.stream_position()?,
+                u64::from(data_end) + Self::extra_size(block_count.into())
+            );
+        }
+
         writer.rewind()?;
         writer.write_all(&header(data_end))?;