arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 03:30:02 +00:00
Code Issues Releases Wiki Activity

feat(core): introduce skip function for UzlibContext

Browse Source 
[no changelog]
This commit is contained in:
cepetr 2024-02-21 08:43:13 +01:00 committed by cepetr
parent 9575c81bd3
commit a1165d9326
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
core/embed/rust/src/trezorhal/uzlib.rs
View File

@ -40,7 +40,7 @@ impl<'a> UzlibContext<'a> {
ctx
}
/// returns Ok(true) if DONE
/// Returns `Ok(true)` if all data was read.
pub fn uncompress(&mut self, dest_buf: &mut [u8]) -> Result<bool, ()> {
unsafe {
self.uncomp.dest = dest_buf.as_mut_ptr();
@ -55,4 +55,18 @@ impl<'a> UzlibContext<'a> {
}
}
}
/// Skips a specified number of bytes (`nbytes`) in the uncompressed stream.
///
/// Returns `Ok(true)` if all data was read.
pub fn skip(&mut self, nbytes: usize) -> Result<bool, ()> {
let mut result = false; // false => OK, true => DONE
let mut sink = [0u8; 256];
for i in (0..nbytes).step_by(sink.len()) {
let chunk_len = core::cmp::min(sink.len(), nbytes - i);
let chunk = &mut sink[0..chunk_len];
result = self.uncompress(chunk)?;
}
Ok(result)
}
}