File tree Expand file tree Collapse file tree 7 files changed +14
-9
lines changed
Expand file tree Collapse file tree 7 files changed +14
-9
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ use body_plz::variants::Body;
22use bytes:: BytesMut ;
33use header_plz:: const_headers:: CONTENT_LENGTH ;
44
5- use crate :: dtraits :: DecompressTrait ;
5+ use crate :: decompress_trait :: DecompressTrait ;
66
77pub fn add_body_and_update_cl < T > ( message : & mut T , body : BytesMut )
88where
Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ use bytes::BytesMut;
22use header_plz:: body_headers:: BodyHeader ;
33use header_plz:: body_headers:: encoding_info:: EncodingInfo ;
44
5- use crate :: dtraits :: DecompressTrait ;
5+ use crate :: decompress_trait :: DecompressTrait ;
66
77pub struct DecodeStruct < ' a , T > {
88 pub message : & ' a mut T ,
File renamed without changes.
Original file line number Diff line number Diff line change @@ -470,7 +470,7 @@ mod tests {
470470 assert_eq ! ( err. reason, MultiDecompressErrorReason :: Corrupt ) ;
471471 // TODO: FIX test
472472 // matches!(err.error, assert_error);
473-
473+ assert ! ( !ds . writer . get_ref ( ) . is_empty ( ) ) ;
474474 assert_eq ! ( ds. encoding_info, original_info) ;
475475 }
476476 }
@@ -507,6 +507,7 @@ mod tests {
507507 panic ! ( "Expected Partial error, got: {:?}" , e. reason) ;
508508 }
509509
510+ assert ! ( ds. writer. get_ref( ) . is_empty( ) ) ;
510511 assert_eq ! ( ds. encoding_info, verify_encodings) ;
511512 }
512513
Original file line number Diff line number Diff line change @@ -3,6 +3,7 @@ use crate::decompression::{
33} ;
44use bytes:: { BufMut , BytesMut , buf:: Writer } ;
55use header_plz:: body_headers:: encoding_info:: EncodingInfo ;
6+ use tracing:: error;
67
78/*
891. (Main + Extra) - compressed ie. compresssed together
@@ -144,9 +145,12 @@ impl<'a> DecompressionState<'a> {
144145 main_plus_extra_decompressed,
145146 )
146147 }
147- Err ( _) => DecompressionState :: ExtraRawMainTry (
148- decompression_struct,
149- ) ,
148+ Err ( e) => {
149+ error ! ( "[-] decompressing main + extra| {}" , e. reason) ;
150+ DecompressionState :: ExtraRawMainTry (
151+ decompression_struct,
152+ )
153+ }
150154 }
151155 }
152156 /* Main - try decompress
Original file line number Diff line number Diff line change @@ -5,9 +5,9 @@ use bytes::BytesMut;
55use crate :: state:: DecodeState ;
66pub mod content_length;
77mod decode_struct;
8+ mod decompress_trait;
89mod decompression;
9- mod dtraits;
10- pub use dtraits:: DecompressTrait ;
10+ pub use decompress_trait:: DecompressTrait ;
1111mod state;
1212
1313pub fn decompress < T > (
Original file line number Diff line number Diff line change @@ -7,10 +7,10 @@ use tracing::error;
77use crate :: {
88 content_length:: add_body_and_update_cl,
99 decode_struct:: DecodeStruct ,
10+ decompress_trait:: DecompressTrait ,
1011 decompression:: {
1112 multi:: error:: MultiDecompressErrorReason , state:: decompression_runner,
1213 } ,
13- dtraits:: DecompressTrait ,
1414} ;
1515
1616pub enum DecodeState < ' a , T > {
You can’t perform that action at this time.
0 commit comments