diff options
| author | Dario Nieuwenhuis <[email protected]> | 2023-06-06 02:28:00 +0200 |
|---|---|---|
| committer | Dario Nieuwenhuis <[email protected]> | 2023-06-06 02:50:57 +0200 |
| commit | 6701606e4c7811082808271b67caa1cdc9387163 (patch) | |
| tree | fe47f8b76fe4e74aed0bc7f959d2f04954303c29 /tests/perf-server | |
| parent | 79b982c941576597bd09d585a9bdaa7fce47688a (diff) | |
cyw43: add perf HIL test.
Diffstat (limited to 'tests/perf-server')
| -rw-r--r-- | tests/perf-server/Cargo.toml | 8 | ||||
| -rwxr-xr-x | tests/perf-server/deploy.sh | 11 | ||||
| -rw-r--r-- | tests/perf-server/perf-server.service | 16 | ||||
| -rw-r--r-- | tests/perf-server/src/main.rs | 90 |
4 files changed, 125 insertions, 0 deletions
diff --git a/tests/perf-server/Cargo.toml b/tests/perf-server/Cargo.toml new file mode 100644 index 000000000..532039050 --- /dev/null +++ b/tests/perf-server/Cargo.toml | |||
| @@ -0,0 +1,8 @@ | |||
| 1 | [package] | ||
| 2 | name = "perf-server" | ||
| 3 | version = "0.1.0" | ||
| 4 | edition = "2021" | ||
| 5 | |||
| 6 | [dependencies] | ||
| 7 | log = "0.4.17" | ||
| 8 | pretty_env_logger = "0.4.0" | ||
diff --git a/tests/perf-server/deploy.sh b/tests/perf-server/deploy.sh new file mode 100755 index 000000000..032e99c30 --- /dev/null +++ b/tests/perf-server/deploy.sh | |||
| @@ -0,0 +1,11 @@ | |||
| 1 | #!/bin/bash | ||
| 2 | |||
| 3 | set -euxo pipefail | ||
| 4 | |||
| 5 | [email protected] | ||
| 6 | |||
| 7 | cargo build --release | ||
| 8 | ssh $HOST -- systemctl stop perf-server | ||
| 9 | scp target/release/perf-server $HOST:/root | ||
| 10 | scp perf-server.service $HOST:/etc/systemd/system/ | ||
| 11 | ssh $HOST -- 'systemctl daemon-reload; systemctl restart perf-server' \ No newline at end of file | ||
diff --git a/tests/perf-server/perf-server.service b/tests/perf-server/perf-server.service new file mode 100644 index 000000000..c14c5d16f --- /dev/null +++ b/tests/perf-server/perf-server.service | |||
| @@ -0,0 +1,16 @@ | |||
| 1 | [Unit] | ||
| 2 | Description=perf-server | ||
| 3 | After=network.target | ||
| 4 | StartLimitIntervalSec=0 | ||
| 5 | |||
| 6 | [Service] | ||
| 7 | Type=simple | ||
| 8 | Restart=always | ||
| 9 | RestartSec=1 | ||
| 10 | User=root | ||
| 11 | ExecStart=/root/perf-server | ||
| 12 | Environment=RUST_BACKTRACE=1 | ||
| 13 | Environment=RUST_LOG=info | ||
| 14 | |||
| 15 | [Install] | ||
| 16 | WantedBy=multi-user.target | ||
diff --git a/tests/perf-server/src/main.rs b/tests/perf-server/src/main.rs new file mode 100644 index 000000000..f6e7efc59 --- /dev/null +++ b/tests/perf-server/src/main.rs | |||
| @@ -0,0 +1,90 @@ | |||
| 1 | use std::io::{Read, Write}; | ||
| 2 | use std::net::{TcpListener, TcpStream}; | ||
| 3 | use std::thread::spawn; | ||
| 4 | use std::time::Duration; | ||
| 5 | |||
| 6 | use log::info; | ||
| 7 | |||
| 8 | fn main() { | ||
| 9 | pretty_env_logger::init(); | ||
| 10 | spawn(|| rx_listen()); | ||
| 11 | spawn(|| rxtx_listen()); | ||
| 12 | tx_listen(); | ||
| 13 | } | ||
| 14 | |||
| 15 | fn tx_listen() { | ||
| 16 | info!("tx: listening on 0.0.0.0:4321"); | ||
| 17 | let listener = TcpListener::bind("0.0.0.0:4321").unwrap(); | ||
| 18 | loop { | ||
| 19 | let (socket, addr) = listener.accept().unwrap(); | ||
| 20 | info!("tx: received connection from: {}", addr); | ||
| 21 | spawn(|| tx_conn(socket)); | ||
| 22 | } | ||
| 23 | } | ||
| 24 | |||
| 25 | fn tx_conn(mut socket: TcpStream) { | ||
| 26 | socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 27 | socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 28 | |||
| 29 | let buf = [0; 1024]; | ||
| 30 | loop { | ||
| 31 | if let Err(e) = socket.write_all(&buf) { | ||
| 32 | info!("tx: failed to write to socket; err = {:?}", e); | ||
| 33 | return; | ||
| 34 | } | ||
| 35 | } | ||
| 36 | } | ||
| 37 | |||
| 38 | fn rx_listen() { | ||
| 39 | info!("rx: listening on 0.0.0.0:4322"); | ||
| 40 | let listener = TcpListener::bind("0.0.0.0:4322").unwrap(); | ||
| 41 | loop { | ||
| 42 | let (socket, addr) = listener.accept().unwrap(); | ||
| 43 | info!("rx: received connection from: {}", addr); | ||
| 44 | spawn(|| rx_conn(socket)); | ||
| 45 | } | ||
| 46 | } | ||
| 47 | |||
| 48 | fn rx_conn(mut socket: TcpStream) { | ||
| 49 | socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 50 | socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 51 | |||
| 52 | let mut buf = [0; 1024]; | ||
| 53 | loop { | ||
| 54 | if let Err(e) = socket.read_exact(&mut buf) { | ||
| 55 | info!("rx: failed to read from socket; err = {:?}", e); | ||
| 56 | return; | ||
| 57 | } | ||
| 58 | } | ||
| 59 | } | ||
| 60 | |||
| 61 | fn rxtx_listen() { | ||
| 62 | info!("rxtx: listening on 0.0.0.0:4323"); | ||
| 63 | let listener = TcpListener::bind("0.0.0.0:4323").unwrap(); | ||
| 64 | loop { | ||
| 65 | let (socket, addr) = listener.accept().unwrap(); | ||
| 66 | info!("rxtx: received connection from: {}", addr); | ||
| 67 | spawn(|| rxtx_conn(socket)); | ||
| 68 | } | ||
| 69 | } | ||
| 70 | |||
| 71 | fn rxtx_conn(mut socket: TcpStream) { | ||
| 72 | socket.set_read_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 73 | socket.set_write_timeout(Some(Duration::from_secs(30))).unwrap(); | ||
| 74 | |||
| 75 | let mut buf = [0; 1024]; | ||
| 76 | loop { | ||
| 77 | match socket.read(&mut buf) { | ||
| 78 | Ok(n) => { | ||
| 79 | if let Err(e) = socket.write_all(&buf[..n]) { | ||
| 80 | info!("rxtx: failed to write to socket; err = {:?}", e); | ||
| 81 | return; | ||
| 82 | } | ||
| 83 | } | ||
| 84 | Err(e) => { | ||
| 85 | info!("rxtx: failed to read from socket; err = {:?}", e); | ||
| 86 | return; | ||
| 87 | } | ||
| 88 | } | ||
| 89 | } | ||
| 90 | } | ||
