feat(decompression-plz): added PartialDecompressError to hold if the extra is raw in case of partial decompression

Author: darkseid-is

decompression-plz/src/decompression/dstruct.rs

@@ -491,6 +491,7 @@ mod tests {
             partial_body,
             header_index,
             compression_index,
+            ..
         } = e.reason
         {
             assert_eq!(partial_body, INPUT);

decompression-plz/src/decompression/multi/error.rs

@@ -24,6 +24,10 @@ impl MultiDecompressError {
         &self.reason
     }
 
+    pub fn reason_as_mut(&mut self) -> &mut MultiDecompressErrorReason {
+        &mut self.reason
+    }
+
     pub fn is_corrupt(&self) -> bool {
         matches!(self.reason, MultiDecompressErrorReason::Corrupt)
     }
@@ -42,6 +46,7 @@ impl MultiDecompressError {
             partial_body,
             header_index,
             compression_index,
+            is_extra_raw: false,
         };
         self.reason = reason;
         self
@@ -64,11 +69,22 @@ pub enum MultiDecompressErrorReason {
         partial_body: BytesMut,
         header_index: usize,
         compression_index: usize,
+        is_extra_raw: bool,
     },
 }
 
 impl MultiDecompressErrorReason {
     pub fn is_partial(&self) -> bool {
         matches!(self, MultiDecompressErrorReason::Partial { .. })
     }
+
+    pub fn set_extra_is_raw(&mut self) {
+        if let MultiDecompressErrorReason::Partial {
+            is_extra_raw,
+            ..
+        } = self
+        {
+            *is_extra_raw = true;
+        }
+    }
 }

decompression-plz/src/decompression/multi/mod.rs

@@ -43,6 +43,7 @@ where
                             partial_body: output,
                             header_index,
                             compression_index,
+                            is_extra_raw: false,
                         }
                     };
                     return Err(MultiDecompressError::new(reason, e));
@@ -143,6 +144,7 @@ mod tests {
             partial_body,
             header_index,
             compression_index,
+            ..
         } = result.reason
         {
             assert_eq!(header_index, 0);
@@ -174,6 +176,7 @@ mod tests {
             partial_body,
             header_index,
             compression_index,
+            ..
         } = result.reason
         {
             assert_eq!(header_index, 0);
@@ -198,6 +201,7 @@ mod tests {
             partial_body,
             header_index,
             compression_index,
+            ..
         } = result.reason
         {
             assert_eq!(header_index, 1);
@@ -226,6 +230,7 @@ mod tests {
             partial_body,
             header_index,
             compression_index,
+            ..
         } = result.reason
         {
             assert_eq!(header_index, 5);

decompression-plz/src/decompression/state.rs

@@ -165,7 +165,12 @@ impl<'a> DecompressionState<'a> {
                             main_decompressed,
                         )
                     }
-                    Err(e) => return Err(e),
+                    Err(mut e) => {
+                        if e.reason().is_partial() {
+                            e.reason_as_mut().set_extra_is_raw();
+                        }
+                        return Err(e);
+                    }
                 }
             }
             DecompressionState::EndMainOnly(_)

decompression-plz/src/state.rs

@@ -140,10 +140,13 @@ where
                 ref mut partial_body,
                 header_index,
                 compression_index,
+                is_extra_raw,
             } = e.reason
             {
                 decode_struct.body = partial_body.split();
-                decode_struct.extra_body = None;
+                if !is_extra_raw {
+                    decode_struct.extra_body = None;
+                }
                 for (index, einfo) in encoding_info.iter().rev().enumerate() {
                     if index > header_index {
                         decode_struct