use nix::{
fcntl::{FcntlArg, OFlag},
sys::epoll,
};
use std::os::unix::io::{AsRawFd, RawFd};
mod _pick_device;
fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut d = _pick_device::pick_device();
println!("{}", d);
let raw_fd = d.as_raw_fd();
nix::fcntl::fcntl(raw_fd, FcntlArg::F_SETFL(OFlag::O_NONBLOCK))?;
let epoll_fd = Epoll::new(epoll::epoll_create1(
epoll::EpollCreateFlags::EPOLL_CLOEXEC,
)?);
let mut event = epoll::EpollEvent::new(epoll::EpollFlags::EPOLLIN, 0);
epoll::epoll_ctl(
epoll_fd.as_raw_fd(),
epoll::EpollOp::EpollCtlAdd,
raw_fd,
Some(&mut event),
)?;
let mut events = [epoll::EpollEvent::empty(); 2];
println!("Events:");
loop {
match d.fetch_events() {
Ok(iterator) => {
for ev in iterator {
println!("{:?}", ev);
}
}
Err(e) if e.kind() == std::io::ErrorKind::WouldBlock => {
epoll::epoll_wait(epoll_fd.as_raw_fd(), &mut events, -1)?;
}
Err(e) => {
eprintln!("{}", e);
break;
}
}
}
Ok(())
}
struct Epoll(RawFd);
impl Epoll {
pub(crate) fn new(fd: RawFd) -> Self {
Epoll(fd)
}
}
impl AsRawFd for Epoll {
fn as_raw_fd(&self) -> RawFd {
self.0
}
}
impl Drop for Epoll {
fn drop(&mut self) {
let _ = nix::unistd::close(self.0);
}
}