netbird/IronRDP
0
0
Fork 0
mirror of https://github.com/netbirdio/IronRDP.git synced 2026-05-22 18:43:12 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix(server): keep looping on compress_stream2() > 0 (#888)

Browse Source 
Fixes: 87df67fdc7 ("feat: add QOIZ image codec")

---------

Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com>
Co-authored-by: Benoît Cortier <3809077+CBenoit@users.noreply.github.com>
Changelog: ignore
This commit is contained in:
Marc-Andre Lureau
2025-07-25 17:11:17 +00:00
committed by GitHub
co-authored by Benoît Cortier
parent 721c680f9b
commit aafd45229b
1 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/ironrdp-server/src/encoder/mod.rs
+16 -11
View File
@@ -504,19 +504,24 @@ impl BitmapUpdateHandler for QoizHandler {
let qoi = qoi_encode(bitmap)?;
let mut inb = zstd_safe::InBuffer::around(&qoi);
let mut data = vec![0; qoi.len()];
let mut outb = zstd_safe::OutBuffer::around(data.as_mut_slice());
let mut outb;
let mut pos = 0;
let mut zctxt = self.zctxt.lock().unwrap();
let res = zctxt
.compress_stream2(
&mut outb,
&mut inb,
zstd_safe::zstd_sys::ZSTD_EndDirective::ZSTD_e_flush,
)
.map_err(zstd_safe::get_error_name)
.unwrap();
if res != 0 {
return Err(anyhow!("Failed to zstd compress"));
loop {
outb = zstd_safe::OutBuffer::around_pos(data.as_mut_slice(), pos);
let res = zctxt
.compress_stream2(
&mut outb,
&mut inb,
zstd_safe::zstd_sys::ZSTD_EndDirective::ZSTD_e_flush,
)
.map_err(|code| anyhow!("failed to zstd compress: {}", zstd_safe::get_error_name(code)))?;
if res == 0 {
break;
}
pos = outb.pos();
data.resize(data.len() + res, 0);
}
set_surface(bitmap, self.codec_id, outb.as_slice())