diff --git a/Cargo.toml b/Cargo.toml index d9e1a23b..8f4b3aa0 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -5,6 +5,7 @@ members = [ "xtask", ] exclude = [ + "extras/bench-server", "extras/esp-wifishark", ] resolver = "2" diff --git a/extras/bench-server/.gitignore b/extras/bench-server/.gitignore new file mode 100644 index 00000000..ea8c4bf7 --- /dev/null +++ b/extras/bench-server/.gitignore @@ -0,0 +1 @@ +/target diff --git a/extras/bench-server/Cargo.toml b/extras/bench-server/Cargo.toml new file mode 100644 index 00000000..56fee5a1 --- /dev/null +++ b/extras/bench-server/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "bench-server" +version = "0.1.0" +edition = "2021" + +[dependencies] +pretty_env_logger = "0.5.0" +log = "0.4.0" \ No newline at end of file diff --git a/extras/bench-server/src/main.rs b/extras/bench-server/src/main.rs new file mode 100644 index 00000000..21ab6486 --- /dev/null +++ b/extras/bench-server/src/main.rs @@ -0,0 +1,87 @@ +use std::io::{Read, Write}; +use std::net::{TcpListener, TcpStream}; +use std::thread::spawn; +use std::time::Duration; + +use log::info; + +fn main() { + pretty_env_logger::init(); + spawn(|| rx_listen()); + spawn(|| rxtx_listen()); + tx_listen(); +} + +fn tx_listen() { + let listener = TcpListener::bind("0.0.0.0:4321").unwrap(); + loop { + let (socket, addr) = listener.accept().unwrap(); + info!("tx: received connection from: {}", addr); + spawn(|| tx_conn(socket)); + } +} + +fn tx_conn(mut socket: TcpStream) { + socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); + socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); + + let buf = [0; 1024]; + loop { + if let Err(e) = socket.write_all(&buf) { + info!("tx: failed to write to socket; err = {:?}", e); + return; + } + } +} + +fn rx_listen() { + let listener = TcpListener::bind("0.0.0.0:4322").unwrap(); + loop { + let (socket, addr) = listener.accept().unwrap(); + info!("rx: received connection from: {}", addr); + spawn(|| rx_conn(socket)); + } +} + +fn rx_conn(mut socket: TcpStream) { + socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); + socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); + + let mut buf = [0; 1024]; + loop { + if let Err(e) = socket.read_exact(&mut buf) { + info!("rx: failed to read from socket; err = {:?}", e); + return; + } + } +} + +fn rxtx_listen() { + let listener = TcpListener::bind("0.0.0.0:4323").unwrap(); + loop { + let (socket, addr) = listener.accept().unwrap(); + info!("rxtx: received connection from: {}", addr); + spawn(|| rxtx_conn(socket)); + } +} + +fn rxtx_conn(mut socket: TcpStream) { + socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); + socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); + + let mut buf = [0; 1024]; + loop { + match socket.read(&mut buf) { + Ok(n) => { + if let Err(e) = socket.write_all(&buf[..n]) { + info!("rxtx: failed to write to socket; err = {:?}", e); + return; + } + } + Err(e) => { + info!("rxtx: failed to read from socket; err = {:?}", e); + return; + } + } + } +} \ No newline at end of file