From 7a2298fa129d099b9c8cf03bca8635f3a88a579b Mon Sep 17 00:00:00 2001 From: Andy Brandt Date: Fri, 1 Dec 2017 23:08:42 -0600 Subject: [PATCH] remove length fix, change max buffer size --- src/main.rs | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/src/main.rs b/src/main.rs index bd73f4e..ad6578c 100644 --- a/src/main.rs +++ b/src/main.rs @@ -10,6 +10,8 @@ mod proxy_socket; use proxy_socket::ProxySocket; +const BUFFER_SIZE: u32 = 1024 * 16; + fn valid_length(length: u32) -> bool { return length > 0 && length <= 4096; // 1024 ^ 2 is the maximum } @@ -38,14 +40,9 @@ fn read_body(length: u32, socket: &mut ProxySocket) { } fn read_response(socket: &mut ProxySocket) { - let mut buf = vec![0; 1024 * 1024]; + let mut buf = vec![0; BUFFER_SIZE as usize]; if let Ok(len) = socket.read(&mut buf) { - // for some reason the length is 1 byte too long in linux? - let mut adjust = 0; - if cfg!(not(windows)) { - adjust = 1; - } - write_response(&buf[0..len - adjust]); + write_response(&buf[0..len]); } }