70 строки
2.2 KiB
Rust
70 строки
2.2 KiB
Rust
|
use {TryWrite};
|
||
|
|
use mio::{Events, Poll, PollOpt, Ready, Token};
|
||
|
|
use mio_uds_windows::{UnixListener, UnixStream};
|
||
|
|
use tempdir::TempDir;
|
||
|
|
|
||
|
|
const LISTEN: Token = Token(0);
|
||
|
|
const CLIENT: Token = Token(1);
|
||
|
|
const SERVER: Token = Token(2);
|
||
|
|
|
||
|
|
struct MyHandler {
|
||
|
|
listener: UnixListener,
|
||
|
|
connected: UnixStream,
|
||
|
|
accepted: Option<UnixStream>,
|
||
|
|
shutdown: bool,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn local_addr_ready() {
|
||
|
|
let dir = TempDir::new("uds").unwrap();
|
||
|
|
|
||
|
|
let server = UnixListener::bind(dir.path().join("foo")).unwrap();
|
||
|
|
let addr = server.local_addr().unwrap();
|
||
|
|
|
||
|
|
let poll = Poll::new().unwrap();
|
||
|
|
poll.register(&server, LISTEN, Ready::readable(),
|
||
|
|
PollOpt::edge()).unwrap();
|
||
|
|
|
||
|
|
let sock = UnixStream::connect(&addr.as_pathname().unwrap()).unwrap();
|
||
|
|
poll.register(&sock, CLIENT, Ready::readable(),
|
||
|
|
PollOpt::edge()).unwrap();
|
||
|
|
|
||
|
|
let mut events = Events::with_capacity(1024);
|
||
|
|
|
||
|
|
let mut handler = MyHandler {
|
||
|
|
listener: server,
|
||
|
|
connected: sock,
|
||
|
|
accepted: None,
|
||
|
|
shutdown: false,
|
||
|
|
};
|
||
|
|
|
||
|
|
while !handler.shutdown {
|
||
|
|
poll.poll(&mut events, None).unwrap();
|
||
|
|
|
||
|
|
for event in &events {
|
||
|
|
match event.token() {
|
||
|
|
LISTEN => {
|
||
|
|
let sock = handler.listener.accept().unwrap().unwrap().0;
|
||
|
|
poll.register(&sock,
|
||
|
|
SERVER,
|
||
|
|
Ready::writable(),
|
||
|
|
PollOpt::edge()).unwrap();
|
||
|
|
handler.accepted = Some(sock);
|
||
|
|
}
|
||
|
|
SERVER => {
|
||
|
|
handler.accepted.as_ref().unwrap().peer_addr().unwrap();
|
||
|
|
handler.accepted.as_ref().unwrap().local_addr().unwrap();
|
||
|
|
handler.accepted.as_mut().unwrap().try_write(&[1, 2, 3]).unwrap();
|
||
|
|
handler.accepted = None;
|
||
|
|
}
|
||
|
|
CLIENT => {
|
||
|
|
handler.connected.peer_addr().unwrap();
|
||
|
|
handler.connected.local_addr().unwrap();
|
||
|
|
handler.shutdown = true;
|
||
|
|
}
|
||
|
|
_ => panic!("unexpected token"),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|