1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
//! The same as v6.rs, but with `poll` calls.
#![cfg(not(any(target_os = "redox", target_os = "wasi")))]
use rustix::io::{poll, PollFd, PollFlags};
use rustix::net::{
accept, bind_v6, connect_v6, getsockname, listen, recv, send, socket, AddressFamily, Ipv6Addr,
Protocol, RecvFlags, SendFlags, SocketAddrAny, SocketAddrV6, SocketType,
};
use std::sync::{Arc, Condvar, Mutex};
use std::thread;
const BUFFER_SIZE: usize = 20;
fn server(ready: Arc<(Mutex<u16>, Condvar)>) {
let connection_socket = socket(
AddressFamily::INET6,
SocketType::STREAM,
Protocol::default(),
)
.unwrap();
let name = SocketAddrV6::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 0, 0, 0);
bind_v6(&connection_socket, &name).unwrap();
let who = match getsockname(&connection_socket).unwrap() {
SocketAddrAny::V6(addr) => addr,
_ => panic!(),
};
listen(&connection_socket, 1).unwrap();
{
let (lock, cvar) = &*ready;
let mut port = lock.lock().unwrap();
*port = who.port();
cvar.notify_all();
}
let mut buffer = vec![0; BUFFER_SIZE];
let data_socket = accept(&connection_socket).unwrap();
let mut fds = [PollFd::new(&data_socket, PollFlags::IN)];
assert_eq!(poll(&mut fds, -1).unwrap(), 1);
assert!(fds[0].revents().intersects(PollFlags::IN));
assert!(!fds[0].revents().intersects(PollFlags::OUT));
let expected_nread = rustix::io::ioctl_fionread(&data_socket).unwrap();
let nread = recv(&data_socket, &mut buffer, RecvFlags::empty()).unwrap();
assert_eq!(String::from_utf8_lossy(&buffer[..nread]), "hello, world");
assert_eq!(expected_nread, nread as u64);
let mut fds = [PollFd::new(&data_socket, PollFlags::OUT)];
assert_eq!(poll(&mut fds, -1).unwrap(), 1);
assert!(!fds[0].revents().intersects(PollFlags::IN));
assert!(fds[0].revents().intersects(PollFlags::OUT));
send(&data_socket, b"goodnight, moon", SendFlags::empty()).unwrap();
}
fn client(ready: Arc<(Mutex<u16>, Condvar)>) {
let port = {
let (lock, cvar) = &*ready;
let mut port = lock.lock().unwrap();
while *port == 0 {
port = cvar.wait(port).unwrap();
}
*port
};
let addr = SocketAddrV6::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), port, 0, 0);
let mut buffer = vec![0; BUFFER_SIZE];
let data_socket = socket(
AddressFamily::INET6,
SocketType::STREAM,
Protocol::default(),
)
.unwrap();
connect_v6(&data_socket, &addr).unwrap();
let mut fds = [PollFd::new(&data_socket, PollFlags::OUT)];
assert_eq!(poll(&mut fds, -1).unwrap(), 1);
assert!(!fds[0].revents().intersects(PollFlags::IN));
assert!(fds[0].revents().intersects(PollFlags::OUT));
send(&data_socket, b"hello, world", SendFlags::empty()).unwrap();
let mut fds = [PollFd::new(&data_socket, PollFlags::IN)];
assert_eq!(poll(&mut fds, -1).unwrap(), 1);
assert!(fds[0].revents().intersects(PollFlags::IN));
assert!(!fds[0].revents().intersects(PollFlags::OUT));
let expected_nread = rustix::io::ioctl_fionread(&data_socket).unwrap();
let nread = recv(&data_socket, &mut buffer, RecvFlags::empty()).unwrap();
assert_eq!(String::from_utf8_lossy(&buffer[..nread]), "goodnight, moon");
assert_eq!(expected_nread, nread as u64);
}
#[test]
fn test_poll() {
let ready = Arc::new((Mutex::new(0_u16), Condvar::new()));
let ready_clone = Arc::clone(&ready);
let server = thread::Builder::new()
.name("server".to_string())
.spawn(move || {
server(ready);
})
.unwrap();
let client = thread::Builder::new()
.name("client".to_string())
.spawn(move || {
client(ready_clone);
})
.unwrap();
client.join().unwrap();
server.join().unwrap();
}
|