handle empty input

PSeitz · PSeitz · commit 6e9b9fb119ff · 2023-05-29T00:24:00.000+08:00
closes #116
diff --git a/src/frame/compress.rs b/src/frame/compress.rs
@@ -86,6 +86,8 @@ pub struct FrameEncoder<W: io::Write> {
     dst: Vec<u8>,
     /// Whether we have an open frame in the output.
     is_frame_open: bool,
+    /// Whether we have an frame closed in the output.
+    data_to_frame_written: bool,
     /// The frame information to be used in this encoder.
     frame_info: FrameInfo,
 }
@@ -140,6 +142,7 @@ impl<W: io::Write> FrameEncoder<W> {
             content_len: 0,
             dst: Vec::new(),
             is_frame_open: false,
+            data_to_frame_written: false,
             frame_info,
             src_start: 0,
             src_end: 0,
@@ -169,8 +172,15 @@ impl<W: io::Write> FrameEncoder<W> {
     /// terminator.
     pub fn try_finish(&mut self) -> Result<(), Error> {
         match self.flush() {
-            Ok(()) if self.is_frame_open => self.end_frame(),
-            Ok(()) => Ok(()),
+            Ok(()) => {
+                // Empty input special case
+                if !self.is_frame_open && !self.data_to_frame_written {
+                    self.begin_frame(0)?;
+                }
+                self.end_frame()?;
+                self.data_to_frame_written = true;
+                Ok(())
+            }
             Err(err) => Err(err.into()),
         }
     }
diff --git a/tests/tests.rs b/tests/tests.rs
@@ -95,6 +95,7 @@ fn test_roundtrip(bytes: impl AsRef<[u8]>) {
         let mut frame_info = lz4_flex::frame::FrameInfo::new();
         frame_info.block_mode = *bm;
         let compressed_flex = lz4_flex_frame_compress_with(frame_info, bytes).unwrap();
+        dbg!(&compressed_flex);
         let decompressed = lz4_flex_frame_decompress(&compressed_flex).unwrap();
         assert_eq!(decompressed, bytes);
     }

Original file line number	Diff line number	Diff line change
`@@ -95,6 +95,7 @@ fn test_roundtrip(bytes: impl AsRef<[u8]>) {`
`95`	`95`	`let mut frame_info = lz4_flex::frame::FrameInfo::new();`
`96`	`96`	`frame_info.block_mode = *bm;`
`97`	`97`	`let compressed_flex = lz4_flex_frame_compress_with(frame_info, bytes).unwrap();`
	`98`	`+ dbg!(&compressed_flex);`
`98`	`99`	`let decompressed = lz4_flex_frame_decompress(&compressed_flex).unwrap();`
`99`	`100`	`assert_eq!(decompressed, bytes);`
`100`	`101`	`}`