pub struct FrameDecoder { /* private fields */ }Expand description
Low level Zstandard decoder that can be used to decompress frames with fine control over when and how many bytes are decoded.
This decoder is able to decode frames only partially and gives control over how many bytes/blocks will be decoded at a time (so you don’t have to decode a 10GB file into memory all at once). It reads bytes as needed from a provided source and can be read from to collect partial results.
If you want to just read the whole frame with an io::Read without having to deal with manually calling FrameDecoder::decode_blocks
you can use the provided crate::decoding::StreamingDecoder wich wraps this FrameDecoder.
Workflow is as follows:
use structured_zstd::decoding::BlockDecodingStrategy;
use std::io::{Read, Write};
// no_std environments can use the crate's own Read traits
use structured_zstd::io::{Read, Write};
fn decode_this(mut file: impl Read) {
//Create a new decoder
let mut frame_dec = structured_zstd::decoding::FrameDecoder::new();
let mut result = Vec::new();
// Use reset or init to make the decoder ready to decode the frame from the io::Read
frame_dec.reset(&mut file).unwrap();
// Loop until the frame has been decoded completely
while !frame_dec.is_finished() {
// decode (roughly) batch_size many bytes
frame_dec.decode_blocks(&mut file, BlockDecodingStrategy::UptoBytes(1024)).unwrap();
// read from the decoder to collect bytes from the internal buffer
let bytes_read = frame_dec.read(result.as_mut_slice()).unwrap();
// then do something with it
do_something(&result[0..bytes_read]);
}
// handle the last chunk of data
while frame_dec.can_collect() > 0 {
let x = frame_dec.read(result.as_mut_slice()).unwrap();
do_something(&result[0..x]);
}
}
fn do_something(data: &[u8]) {
std::io::stdout().write_all(data).unwrap();
}Implementations§
Source§impl FrameDecoder
impl FrameDecoder
Sourcepub fn new() -> FrameDecoder ⓘ
pub fn new() -> FrameDecoder ⓘ
This will create a new decoder without allocating anything yet. init()/reset() will allocate all needed buffers if it is the first time this decoder is used else they just reset these buffers with not further allocations
Sourcepub fn init(&mut self, source: impl Read) -> Result<(), FrameDecoderError>
pub fn init(&mut self, source: impl Read) -> Result<(), FrameDecoderError>
init() will allocate all needed buffers if it is the first time this decoder is used else they just reset these buffers with not further allocations
Note that all bytes currently in the decodebuffer from any previous frame will be lost. Collect them with collect()/collect_to_writer()
equivalent to reset()
Sourcepub fn init_with_dict_handle(
&mut self,
source: impl Read,
dict: &DictionaryHandle,
) -> Result<(), FrameDecoderError>
pub fn init_with_dict_handle( &mut self, source: impl Read, dict: &DictionaryHandle, ) -> Result<(), FrameDecoderError>
Initialize the decoder for a new frame using a pre-parsed dictionary handle.
If the frame header has a dictionary ID, this validates it against
dict.id() and returns FrameDecoderError::DictIdMismatch on mismatch.
If the header omits the optional dictionary ID, this still applies the provided dictionary handle.
§Warning
This method always applies dict unless the frame header contains a
non-matching dictionary ID. Callers must only use this API when they
already know the frame was encoded with the provided dictionary, even if
the frame header omits the dictionary ID or encodes an explicit
dictionary ID of 0.
Passing a dictionary for a frame that was not encoded with it can silently corrupt the decoded output.
Sourcepub fn reset(&mut self, source: impl Read) -> Result<(), FrameDecoderError>
pub fn reset(&mut self, source: impl Read) -> Result<(), FrameDecoderError>
reset() will allocate all needed buffers if it is the first time this decoder is used else they just reset these buffers with not further allocations
Note that all bytes currently in the decodebuffer from any previous frame will be lost. Collect them with collect()/collect_to_writer()
equivalent to init()
Sourcepub fn reset_with_dict_handle(
&mut self,
source: impl Read,
dict: &DictionaryHandle,
) -> Result<(), FrameDecoderError>
pub fn reset_with_dict_handle( &mut self, source: impl Read, dict: &DictionaryHandle, ) -> Result<(), FrameDecoderError>
Reset this decoder for a new frame using a pre-parsed dictionary handle.
If the frame header has a dictionary ID, this validates it against
dict.id() and returns FrameDecoderError::DictIdMismatch on mismatch.
If the header omits the optional dictionary ID, this still applies the provided dictionary handle.
§Warning
This method always applies dict unless the frame header contains a
non-matching dictionary ID. Callers must only use this API when they
already know the frame was encoded with the provided dictionary, even if
the frame header omits the dictionary ID or encodes an explicit
dictionary ID of 0.
Passing a dictionary for a frame that was not encoded with it can silently corrupt the decoded output.
Sourcepub fn add_dict(&mut self, dict: Dictionary) -> Result<(), FrameDecoderError>
pub fn add_dict(&mut self, dict: Dictionary) -> Result<(), FrameDecoderError>
Add a dictionary that can be selected dynamically by frame dictionary ID.
Returns FrameDecoderError::DictAlreadyRegistered if the ID is already
registered (either as owned or shared).
Sourcepub fn add_dict_from_bytes(
&mut self,
raw_dictionary: &[u8],
) -> Result<(), FrameDecoderError>
pub fn add_dict_from_bytes( &mut self, raw_dictionary: &[u8], ) -> Result<(), FrameDecoderError>
Parse and add a serialized dictionary blob.
Sourcepub fn add_dict_handle(
&mut self,
dict: DictionaryHandle,
) -> Result<(), FrameDecoderError>
pub fn add_dict_handle( &mut self, dict: DictionaryHandle, ) -> Result<(), FrameDecoderError>
Add a pre-parsed dictionary handle for reuse across decoders.
This API is available on targets with pointer-width atomics
(target_has_atomic = "ptr").
Returns FrameDecoderError::DictAlreadyRegistered if the ID is already
registered (either as owned or shared).
pub fn force_dict(&mut self, dict_id: u32) -> Result<(), FrameDecoderError>
Sourcepub fn content_size(&self) -> u64
pub fn content_size(&self) -> u64
Returns how many bytes the frame contains after decompression
Sourcepub fn get_checksum_from_data(&self) -> Option<u32>
pub fn get_checksum_from_data(&self) -> Option<u32>
Returns the checksum that was read from the data. Only available after all bytes have been read. It is the last 4 bytes of a zstd-frame
Sourcepub fn get_calculated_checksum(&self) -> Option<u32>
pub fn get_calculated_checksum(&self) -> Option<u32>
Returns the checksum that was calculated while decoding. Only a sensible value after all decoded bytes have been collected/read from the FrameDecoder
Sourcepub fn bytes_read_from_source(&self) -> u64
pub fn bytes_read_from_source(&self) -> u64
Counter for how many bytes have been consumed while decoding the frame
Sourcepub fn is_finished(&self) -> bool
pub fn is_finished(&self) -> bool
Whether the current frames last block has been decoded yet If this returns true you can call the drain* functions to get all content (the read() function will drain automatically if this returns true)
Sourcepub fn blocks_decoded(&self) -> usize
pub fn blocks_decoded(&self) -> usize
Counter for how many blocks have already been decoded
Sourcepub fn decode_blocks(
&mut self,
source: impl Read,
strat: BlockDecodingStrategy,
) -> Result<bool, FrameDecoderError>
pub fn decode_blocks( &mut self, source: impl Read, strat: BlockDecodingStrategy, ) -> Result<bool, FrameDecoderError>
Decodes blocks from a reader. It requires that the framedecoder has been initialized first. The Strategy influences how many blocks will be decoded before the function returns This is important if you want to manage memory consumption carefully. If you don’t care about that you can just choose the strategy “All” and have all blocks of the frame decoded into the buffer
Sourcepub fn collect(&mut self) -> Option<Vec<u8>>
pub fn collect(&mut self) -> Option<Vec<u8>>
Collect bytes and retain window_size bytes while decoding is still going on. After decoding of the frame (is_finished() == true) has finished it will collect all remaining bytes
Sourcepub fn collect_to_writer(&mut self, w: impl Write) -> Result<usize, Error>
pub fn collect_to_writer(&mut self, w: impl Write) -> Result<usize, Error>
Collect bytes and retain window_size bytes while decoding is still going on. After decoding of the frame (is_finished() == true) has finished it will collect all remaining bytes
Sourcepub fn can_collect(&self) -> usize
pub fn can_collect(&self) -> usize
How many bytes can currently be collected from the decodebuffer, while decoding is going on this will be lower than the actual decodbuffer size because window_size bytes need to be retained for decoding. After decoding of the frame (is_finished() == true) has finished it will report all remaining bytes
Sourcepub fn decode_from_to(
&mut self,
source: &[u8],
target: &mut [u8],
) -> Result<(usize, usize), FrameDecoderError>
pub fn decode_from_to( &mut self, source: &[u8], target: &mut [u8], ) -> Result<(usize, usize), FrameDecoderError>
Decodes as many blocks as possible from the source slice and reads from the decodebuffer into the target slice The source slice may contain only parts of a frame but must contain at least one full block to make progress
By all means use decode_blocks if you have a io.Reader available. This is just for compatibility with other decompressors which try to serve an old-style c api
Returns (read, written), if read == 0 then the source did not contain a full block and further calls with the same input will not make any progress!
Note that no kind of block can be bigger than 128kb. So to be safe use at least 128*1024 (max block content size) + 3 (block_header size) + 18 (max frame_header size) bytes as your source buffer
You may call this function with an empty source after all bytes have been decoded. This is equivalent to just call decoder.read(&mut target)
Sourcepub fn decode_all(
&mut self,
input: &[u8],
output: &mut [u8],
) -> Result<usize, FrameDecoderError>
pub fn decode_all( &mut self, input: &[u8], output: &mut [u8], ) -> Result<usize, FrameDecoderError>
Decode multiple frames into the output slice.
input must contain an exact number of frames. Skippable frames are allowed and will be
skipped during decode.
output must be large enough to hold the decompressed data. If you don’t know
how large the output will be, use FrameDecoder::decode_blocks instead.
This calls FrameDecoder::init, and all bytes currently in the decoder will be lost.
Returns the number of bytes written to output.
Sourcepub fn decode_all_with_dict_handle(
&mut self,
input: &[u8],
output: &mut [u8],
dict: &DictionaryHandle,
) -> Result<usize, FrameDecoderError>
pub fn decode_all_with_dict_handle( &mut self, input: &[u8], output: &mut [u8], dict: &DictionaryHandle, ) -> Result<usize, FrameDecoderError>
Decode multiple frames into the output slice using a pre-parsed dictionary handle.
input must contain an exact number of frames. Skippable frames are allowed and will be
skipped during decode.
output must be large enough to hold the decompressed data. If you don’t know
how large the output will be, use FrameDecoder::decode_blocks instead.
This calls FrameDecoder::init_with_dict_handle, and all bytes currently in the
decoder will be lost.
§Warning
Each decoded frame is initialized with dict, even when a frame header
omits the optional dictionary ID. Callers must only use this API when
they already know the input frames were encoded with the provided
dictionary; otherwise decoded output can be silently corrupted.
Sourcepub fn decode_all_with_dict_bytes(
&mut self,
input: &[u8],
output: &mut [u8],
raw_dictionary: &[u8],
) -> Result<usize, FrameDecoderError>
pub fn decode_all_with_dict_bytes( &mut self, input: &[u8], output: &mut [u8], raw_dictionary: &[u8], ) -> Result<usize, FrameDecoderError>
Decode multiple frames into the output slice using a serialized dictionary.
§Warning
Each decoded frame is initialized with the parsed dictionary, even when a frame header omits the optional dictionary ID. Callers must only use this API when they already know the input frames were encoded with that dictionary; otherwise decoded output can be silently corrupted.
Sourcepub fn decode_all_to_vec(
&mut self,
input: &[u8],
output: &mut Vec<u8>,
) -> Result<(), FrameDecoderError>
pub fn decode_all_to_vec( &mut self, input: &[u8], output: &mut Vec<u8>, ) -> Result<(), FrameDecoderError>
Decode multiple frames into the extra capacity of the output vector.
input must contain an exact number of frames.
output must have enough extra capacity to hold the decompressed data.
This function will not reallocate or grow the vector. If you don’t know
how large the output will be, use FrameDecoder::decode_blocks instead.
This calls FrameDecoder::init, and all bytes currently in the decoder will be lost.
The length of the output vector is updated to include the decompressed data. The length is not changed if an error occurs.
Trait Implementations§
Source§impl Default for FrameDecoder
impl Default for FrameDecoder
Source§impl Read for FrameDecoder
Read bytes from the decode_buffer that are no longer needed. While the frame is not yet finished
this will retain window_size bytes, else it will drain it completely
impl Read for FrameDecoder
Read bytes from the decode_buffer that are no longer needed. While the frame is not yet finished this will retain window_size bytes, else it will drain it completely
Source§fn read(&mut self, target: &mut [u8]) -> Result<usize, Error>
fn read(&mut self, target: &mut [u8]) -> Result<usize, Error>
1.36.0 · Source§fn read_vectored(&mut self, bufs: &mut [IoSliceMut<'_>]) -> Result<usize, Error>
fn read_vectored(&mut self, bufs: &mut [IoSliceMut<'_>]) -> Result<usize, Error>
read, except that it reads into a slice of buffers. Read moreSource§fn is_read_vectored(&self) -> bool
fn is_read_vectored(&self) -> bool
can_vector)1.0.0 · Source§fn read_to_end(&mut self, buf: &mut Vec<u8>) -> Result<usize, Error>
fn read_to_end(&mut self, buf: &mut Vec<u8>) -> Result<usize, Error>
buf. Read more1.0.0 · Source§fn read_to_string(&mut self, buf: &mut String) -> Result<usize, Error>
fn read_to_string(&mut self, buf: &mut String) -> Result<usize, Error>
buf. Read more1.6.0 · Source§fn read_exact(&mut self, buf: &mut [u8]) -> Result<(), Error>
fn read_exact(&mut self, buf: &mut [u8]) -> Result<(), Error>
buf. Read moreSource§fn read_buf(&mut self, buf: BorrowedCursor<'_>) -> Result<(), Error>
fn read_buf(&mut self, buf: BorrowedCursor<'_>) -> Result<(), Error>
read_buf)Source§fn read_buf_exact(&mut self, cursor: BorrowedCursor<'_>) -> Result<(), Error>
fn read_buf_exact(&mut self, cursor: BorrowedCursor<'_>) -> Result<(), Error>
read_buf)cursor. Read more1.0.0 · Source§fn by_ref(&mut self) -> &mut Selfwhere
Self: Sized,
fn by_ref(&mut self) -> &mut Selfwhere
Self: Sized,
Read. Read more