use { super::util::{NameGen, TestResult}, anyhow::Context, futures::io::{AsyncBufReadExt, AsyncWriteExt, BufReader}, interprocess::os::windows::named_pipe::{ tokio::{DuplexBytePipeStream, PipeListenerOptionsExt}, PipeListenerOptions, }, std::{convert::TryInto, ffi::OsStr, io, sync::Arc, time::Duration}, tokio::{sync::oneshot::Sender, task, time::sleep, try_join}, }; static SERVER_MSG: &str = "Hello from server!\n"; static CLIENT_MSG: &str = "Hello from client!\n"; pub async fn server(name_sender: Sender, num_clients: u32) -> TestResult { async fn handle_conn(conn: DuplexBytePipeStream) -> TestResult { let (reader, mut writer) = conn.split(); let mut buffer = String::with_capacity(128); let mut reader = BufReader::new(reader); let read = async { reader .read_line(&mut buffer) .await .context("Pipe receive failed") }; let write = async { writer .write_all(SERVER_MSG.as_bytes()) .await .context("Pipe send failed") }; try_join!(read, write)?; assert_eq!(buffer, CLIENT_MSG); Ok(()) } let (name, listener) = NameGen::new(true) .find_map(|nm| { let rnm: &OsStr = nm.as_ref(); let l = match PipeListenerOptions::new() .name(rnm) .create_tokio::() { Ok(l) => l, Err(e) if e.kind() == io::ErrorKind::AddrInUse => return None, Err(e) => return Some(Err(e)), }; Some(Ok((nm, l))) }) .unwrap() .context("Listener bind failed")?; let _ = name_sender.send(name); let mut tasks = Vec::with_capacity(num_clients.try_into().unwrap()); for _ in 0..num_clients { let conn = match listener.accept().await { Ok(c) => c, Err(e) => { eprintln!("Incoming connection failed: {}", e); continue; } }; let task = task::spawn(handle_conn(conn)); tasks.push(task); } for task in tasks { task.await .context("Server task panicked")? .context("Server task returned early with error")?; } Ok(()) } pub async fn client(name: Arc) -> TestResult { let mut buffer = String::with_capacity(128); let (reader, mut writer) = loop { match DuplexBytePipeStream::connect(name.as_str()) { Err(e) if e.kind() == io::ErrorKind::WouldBlock => { sleep(Duration::from_millis(10)).await; continue; } not_busy => break not_busy, } } .context("Connect failed")? .split(); let mut reader = BufReader::new(reader); let read = async { reader .read_line(&mut buffer) .await .context("Pipe receive failed") }; let write = async { writer .write_all(CLIENT_MSG.as_bytes()) .await .context("Pipe send failed") }; try_join!(read, write)?; assert_eq!(buffer, SERVER_MSG); Ok(()) }