qwqVictor/wasm-rdp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix vnc substract overflow

Browse Source
This commit is contained in:
Jovi Hsu 2021-11-15 18:49:27 +08:00
parent 2de7b4b9ec
commit 85abc9d7f8
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
frontend/src/protocal/vnc.rs
View File

@ -1,5 +1,3 @@
use super::common::*; use super::common::*;
use super::des; use super::des;
use yew::services::ConsoleService; use yew::services::ConsoleService;
@ -74,12 +72,12 @@ impl ProtocalImpl for VncHandler {
fn do_input(&mut self, input: Vec<u8>) { fn do_input(&mut self, input: Vec<u8>) {
self.buf_num += input.len(); self.buf_num += input.len();
ConsoleService::info(&format!( // ConsoleService::info(&format!(
"VNC input {}, left {}, require {}", // "VNC input {}, left {}, require {}",
input.len(), // input.len(),
self.buf_num, // self.buf_num,
self.require // self.require
)); // ));
self.reader.append(input); self.reader.append(input);
while self.buf_num >= self.require { while self.buf_num >= self.require {
self.handle_input(); self.handle_input();
@ -326,8 +324,8 @@ impl VncHandler {
fn handle_framebuffer_update(&mut self) { fn handle_framebuffer_update(&mut self) {
let _padding = self.read_u8(); let _padding = self.read_u8();
self.num_rects_left = self.read_u16(); self.num_rects_left = self.read_u16();
ConsoleService::log(&format!("VNC: {} rects", self.num_rects_left)); // ConsoleService::log(&format!("VNC: {} rects", self.num_rects_left));
self.require = 12; // the length of the first rectangle self.require = 12; // the length of the first rectangle hdr
} }
//Each rectangle consists of: //Each rectangle consists of:
@ -386,12 +384,15 @@ impl VncHandler {
data: image_data, data: image_data,
})); }));
self.num_rects_left -= 1; self.num_rects_left -= 1;
if 0 == self.num_rects_left {
self.during_update = false;
self.require = 1;
} else {
self.require = 12; // the length of the next rectangle hdr
}
} }
if 0 == self.num_rects_left {
self.during_update = false; // ConsoleService::log(&format!("{} rects left", self.num_rects_left));
self.require = 1;
}
ConsoleService::log(&format!("{} rects left", self.num_rects_left));
} }
fn handle_set_colour_map(&mut self) { fn handle_set_colour_map(&mut self) {