notkshitij/keepassxc-proxy-rust
1
0
Fork 0
Code Releases 1 Activity

remove length fix, change max buffer size

Browse Source
This commit is contained in:
Andy Brandt
2017-12-01 23:08:42 -06:00
parent c7a53e654b
commit 7a2298fa12
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+4 -7
View File
@@ -10,6 +10,8 @@ mod proxy_socket;
use proxy_socket::ProxySocket; use proxy_socket::ProxySocket;
const BUFFER_SIZE: u32 = 1024 * 16;
fn valid_length(length: u32) -> bool { fn valid_length(length: u32) -> bool {
return length > 0 && length <= 4096; // 1024 ^ 2 is the maximum return length > 0 && length <= 4096; // 1024 ^ 2 is the maximum
} }
@@ -38,14 +40,9 @@ fn read_body<T: Read + Write>(length: u32, socket: &mut ProxySocket<T>) {
} }
fn read_response<T: Read>(socket: &mut ProxySocket<T>) { fn read_response<T: Read>(socket: &mut ProxySocket<T>) {
let mut buf = vec![0; 1024 * 1024]; let mut buf = vec![0; BUFFER_SIZE as usize];
if let Ok(len) = socket.read(&mut buf) { if let Ok(len) = socket.read(&mut buf) {
// for some reason the length is 1 byte too long in linux? write_response(&buf[0..len]);
let mut adjust = 0;
if cfg!(not(windows)) {
adjust = 1;
}
write_response(&buf[0..len - adjust]);
} }
} }