mirror of
https://github.com/jiangcuo/nix.git
synced 2026-01-25 17:35:28 +00:00
ea888247b7
* Relax lifetime requirements for PollFd::new Fixes #2118 * Take BorrowedFd as the argument for PollFd::new &AsFd didn't work because there are 'static types, like std::fs::File, which implement AsFd. The created BorrowedFd type within the PollFd::new method would have a very brief lifetime, but the PhantomData would capture the lifetime of the std::fs::File. Taking BorrowFd<'fd> argument makes the lifetime explicit. * fix legacy comment --------- Co-authored-by: Steve Lau <stevelauc@outlook.com>
79 lines
2.3 KiB
Rust
79 lines
2.3 KiB
Rust
use nix::{
|
|
errno::Errno,
|
|
poll::{poll, PollFd, PollFlags},
|
|
unistd::{pipe, write},
|
|
};
|
|
use std::os::unix::io::{AsFd, BorrowedFd};
|
|
|
|
macro_rules! loop_while_eintr {
|
|
($poll_expr: expr) => {
|
|
loop {
|
|
match $poll_expr {
|
|
Ok(nfds) => break nfds,
|
|
Err(Errno::EINTR) => (),
|
|
Err(e) => panic!("{}", e),
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
#[test]
|
|
fn test_poll() {
|
|
let (r, w) = pipe().unwrap();
|
|
let mut fds = [PollFd::new(r.as_fd(), PollFlags::POLLIN)];
|
|
|
|
// Poll an idle pipe. Should timeout
|
|
let nfds = loop_while_eintr!(poll(&mut fds, 100));
|
|
assert_eq!(nfds, 0);
|
|
assert!(!fds[0].revents().unwrap().contains(PollFlags::POLLIN));
|
|
|
|
write(&w, b".").unwrap();
|
|
|
|
// Poll a readable pipe. Should return an event.
|
|
let nfds = poll(&mut fds, 100).unwrap();
|
|
assert_eq!(nfds, 1);
|
|
assert!(fds[0].revents().unwrap().contains(PollFlags::POLLIN));
|
|
}
|
|
|
|
// ppoll(2) is the same as poll except for how it handles timeouts and signals.
|
|
// Repeating the test for poll(2) should be sufficient to check that our
|
|
// bindings are correct.
|
|
#[cfg(any(
|
|
target_os = "android",
|
|
target_os = "dragonfly",
|
|
target_os = "freebsd",
|
|
target_os = "linux"
|
|
))]
|
|
#[test]
|
|
fn test_ppoll() {
|
|
use nix::poll::ppoll;
|
|
use nix::sys::signal::SigSet;
|
|
use nix::sys::time::{TimeSpec, TimeValLike};
|
|
|
|
let timeout = TimeSpec::milliseconds(1);
|
|
let (r, w) = pipe().unwrap();
|
|
let mut fds = [PollFd::new(r.as_fd(), PollFlags::POLLIN)];
|
|
|
|
// Poll an idle pipe. Should timeout
|
|
let sigset = SigSet::empty();
|
|
let nfds = loop_while_eintr!(ppoll(&mut fds, Some(timeout), Some(sigset)));
|
|
assert_eq!(nfds, 0);
|
|
assert!(!fds[0].revents().unwrap().contains(PollFlags::POLLIN));
|
|
|
|
write(&w, b".").unwrap();
|
|
|
|
// Poll a readable pipe. Should return an event.
|
|
let nfds = ppoll(&mut fds, Some(timeout), None).unwrap();
|
|
assert_eq!(nfds, 1);
|
|
assert!(fds[0].revents().unwrap().contains(PollFlags::POLLIN));
|
|
}
|
|
|
|
#[test]
|
|
fn test_pollfd_events() {
|
|
let fd_zero = unsafe { BorrowedFd::borrow_raw(0) };
|
|
let mut pfd = PollFd::new(fd_zero.as_fd(), PollFlags::POLLIN);
|
|
assert_eq!(pfd.events(), PollFlags::POLLIN);
|
|
pfd.set_events(PollFlags::POLLOUT);
|
|
assert_eq!(pfd.events(), PollFlags::POLLOUT);
|
|
}
|