embassy/tests/utils/src/bin/saturate_serial.rs
2023-05-01 22:42:36 +02:00

54 lines
1.6 KiB
Rust

use std::path::Path;
use std::time::Duration;
use std::{env, io, process, thread};
use rand::random;
use serial::SerialPort;
pub fn main() {
if let Some(port_name) = env::args().nth(1) {
let idles = env::args().position(|x| x == "--idles").is_some();
println!("Saturating port {:?} with 115200 8N1", port_name);
println!("Idles: {}", idles);
println!("Process ID: {}", process::id());
let mut port = serial::open(&port_name).unwrap();
if saturate(&mut port, idles).is_err() {
eprintln!("Unable to saturate port");
}
} else {
let path = env::args().next().unwrap();
let basepath = Path::new(&path).with_extension("");
let basename = basepath.file_name().unwrap();
eprintln!("USAGE: {} <port-name>", basename.to_string_lossy());
}
}
fn saturate<T: SerialPort>(port: &mut T, idles: bool) -> io::Result<()> {
port.reconfigure(&|settings| {
settings.set_baud_rate(serial::Baud115200)?;
settings.set_char_size(serial::Bits8);
settings.set_parity(serial::ParityNone);
settings.set_stop_bits(serial::Stop1);
Ok(())
})?;
let mut written = 0;
loop {
let len = random::<usize>() % 0x1000;
let buf: Vec<u8> = (written..written + len).map(|x| x as u8).collect();
port.write_all(&buf)?;
if idles {
let micros = (random::<usize>() % 1000) as u64;
println!("Sleeping {}us", micros);
port.flush().unwrap();
thread::sleep(Duration::from_micros(micros));
}
written += len;
println!("Written: {}", written);
}
}