Skip to content

Commit f566f27

Browse files
committed
refactor(decompression-plz): renamed dtraits to decompress_trait
1 parent d3e15ff commit f566f27

File tree

7 files changed

+14
-9
lines changed

7 files changed

+14
-9
lines changed

decompression-plz/src/content_length.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ use body_plz::variants::Body;
22
use bytes::BytesMut;
33
use header_plz::const_headers::CONTENT_LENGTH;
44

5-
use crate::dtraits::DecompressTrait;
5+
use crate::decompress_trait::DecompressTrait;
66

77
pub fn add_body_and_update_cl<T>(message: &mut T, body: BytesMut)
88
where

decompression-plz/src/decode_struct.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ use bytes::BytesMut;
22
use header_plz::body_headers::BodyHeader;
33
use header_plz::body_headers::encoding_info::EncodingInfo;
44

5-
use crate::dtraits::DecompressTrait;
5+
use crate::decompress_trait::DecompressTrait;
66

77
pub struct DecodeStruct<'a, T> {
88
pub message: &'a mut T,

decompression-plz/src/decompression/dstruct.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff 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

decompression-plz/src/decompression/state.rs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ use crate::decompression::{
33
};
44
use bytes::{BufMut, BytesMut, buf::Writer};
55
use header_plz::body_headers::encoding_info::EncodingInfo;
6+
use tracing::error;
67

78
/*
89
1. (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

decompression-plz/src/lib.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@ use bytes::BytesMut;
55
use crate::state::DecodeState;
66
pub mod content_length;
77
mod decode_struct;
8+
mod decompress_trait;
89
mod decompression;
9-
mod dtraits;
10-
pub use dtraits::DecompressTrait;
10+
pub use decompress_trait::DecompressTrait;
1111
mod state;
1212

1313
pub fn decompress<T>(

decompression-plz/src/state.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@ use tracing::error;
77
use 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

1616
pub enum DecodeState<'a, T> {

0 commit comments

Comments
 (0)