OrgRust
/
tcpproxy
mirror of https://github.com/neosmart/tcpproxy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Make per-client buffer size a constant

This commit is contained in:
Mahmoud Al-Qudsi 2022-07-08 17:09:41 -05:00
parent 6a7f1b395c
commit 512832b304
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main.rs
View File

@ -8,6 +8,7 @@ use tokio::sync::broadcast;
type BoxedError = Box<dyn std::error::Error + Sync + Send + 'static>;
static DEBUG: AtomicBool = AtomicBool::new(false);
const BUF_SIZE: usize = 1024;
fn print_usage(program: &str, opts: Options) {
let program_path = std::path::PathBuf::from(program);
@ -98,7 +99,7 @@ async fn forward(bind_ip: &str, local_port: i32, remote: &str) -> Result<(), Box
W: tokio::io::AsyncWrite + Unpin,
{
let mut copied = 0;
let mut buf = [0u8; 1024];
let mut buf = [0u8; BUF_SIZE];
loop {
let bytes_read;
tokio::select! {