socketcan_alt/
aio.rs

1use crate::{CmsgIter, Frame, Timestamping};
2use std::ffi::CStr;
3use std::io::{ErrorKind, Result};
4use std::os::unix::io::{AsRawFd, RawFd};
5use tokio::io::unix::AsyncFd;
6
7pub struct Socket(AsyncFd<crate::Socket>);
8
9impl Socket {
10    pub fn bind<I>(ifname: I) -> Result<Self>
11    where
12        I: AsRef<CStr>,
13    {
14        let socket = crate::Socket::bind(ifname)?;
15        socket.set_nonblocking(true)?;
16        Ok(Self(AsyncFd::new(socket)?))
17    }
18
19    pub fn set_timestamping(&self, timestamping: Timestamping) -> Result<()> {
20        self.0.get_ref().set_timestamping(timestamping)
21    }
22
23    pub fn set_recv_own_msgs(&self, enable: bool) -> Result<()> {
24        self.0.get_ref().set_recv_own_msgs(enable)
25    }
26
27    pub fn set_fd_frames(&self, enable: bool) -> Result<()> {
28        self.0.get_ref().set_fd_frames(enable)
29    }
30
31    pub async fn recv(&self) -> Result<Frame> {
32        loop {
33            if let Ok(v) = self.0.readable().await?.try_io(|s| s.get_ref().recv()) {
34                break v;
35            }
36        }
37    }
38
39    #[allow(clippy::needless_lifetimes)]
40    pub async fn recv_msg<'a>(
41        &self,
42        cmsg_buf: &'a mut [u8],
43    ) -> Result<(Frame, Option<CmsgIter<'a>>)> {
44        let mut cmsg_buf = Some(cmsg_buf);
45        loop {
46            let mut guard = self.0.readable().await?;
47            match self.0.get_ref()._recv_msg(cmsg_buf.take().unwrap()) {
48                Err((e, b)) if e.kind() == ErrorKind::WouldBlock => {
49                    cmsg_buf = Some(b);
50                    guard.clear_ready();
51                }
52                r => break r.map_err(|(e, _)| e),
53            }
54        }
55    }
56
57    pub async fn send(&self, frame: &Frame) -> Result<()> {
58        loop {
59            if let Ok(v) = self.0.writable().await?.try_io(|s| s.get_ref().send(frame)) {
60                break v;
61            }
62        }
63    }
64}
65
66impl AsRawFd for Socket {
67    fn as_raw_fd(&self) -> RawFd {
68        self.0.as_raw_fd()
69    }
70}
71
72#[cfg(test)]
73mod tests;
socketcan_alt/aio.rs

socketcan_alt/
aio.rs
