kai
/
pounder_test
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix TCP command interface rx buffer wrapping 

Previously, if a message spanned the end of the
receive ring buffer, the last byte in the buffer
would be missing from the to-be-decoded message,
leading to a parse error or panic.

With this commit, >3M commands were exchanged
over one socket without further issues.
This commit is contained in:
David Nadlinger 2019-11-13 01:50:18 +00:00
parent bc85cd478b
commit 2297286360
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main.rs
View File

@ -852,7 +852,7 @@ impl Server {
self.discard = true; self.discard = true;
self.data.clear(); self.data.clear();
} else if !self.discard && len > 0 { } else if !self.discard && len > 0 {
self.data.extend_from_slice(&buf[..len - 1]).unwrap(); self.data.extend_from_slice(&buf[..len]).unwrap();
} }
(len, found) (len, found)
}).unwrap(); }).unwrap();
@ -862,7 +862,7 @@ impl Server {
json_reply(socket, &Response { code: 520, message: "command buffer overflow" }); json_reply(socket, &Response { code: 520, message: "command buffer overflow" });
self.data.clear(); self.data.clear();
} else { } else {
let r = from_slice::<T>(&self.data); let r = from_slice::<T>(&self.data[..self.data.len() - 1]);
self.data.clear(); self.data.clear();
match r { match r {
Ok(res) => { Ok(res) => {