fix(decompression-plz): chunked encoding remove header

Author: darkseid-is

decompression-plz/src/state.rs

@@ -52,17 +52,20 @@ where
                     ds.chunked_to_raw();
                     // remove chunked TE
                     // Chunked TE must be the last
-                    encoding_infos
-                        .last_mut()
-                        .unwrap()
-                        .encodings_as_mut()
-                        .pop();
+                    let last_info = encoding_infos.last_mut().unwrap();
+                    last_info.encodings_as_mut().pop();
+
+                    // if after removing TE it is empty, remove the header
+                    if last_info.encodings().is_empty() {
+                        ds.message
+                            .remove_header_on_position(last_info.header_index);
+
+                        // remove the last encoding_info
+                        encoding_infos.pop();
+                    }
                 }
                 // If only chunked was present then Vec<EncodingInfo> is empty
-                let mut next_state = if is_empty_encodings(&encoding_infos) {
-                    ds.message.header_map_as_mut().remove_header_on_position(
-                        encoding_infos[0].header_index,
-                    );
+                let mut next_state = if encoding_infos.is_empty() {
                     if ds.content_encoding_is_some() {
                         let encodings = ds.get_content_encoding();
                         Self::ContentEncoding(ds, encodings)
@@ -199,8 +202,3 @@ where
         }
     }
 }
-
-// Only chunked was present
-fn is_empty_encodings(einfo_vec: &[EncodingInfo]) -> bool {
-    einfo_vec.len() == 1 && einfo_vec[0].encodings().is_empty()
-}