ttrpc 0.9.0

A Rust version of ttrpc.
Documentation 
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

use std::convert::TryFrom;
use std::io::{Error as IoError, Result as IoResult};
use std::os::fd::{FromRawFd as _, RawFd};
use std::os::unix::net::{
    SocketAddr, UnixListener as StdUnixListener, UnixStream as StdUnixStream,
};

use async_stream::stream;
use tokio::net::{UnixListener, UnixStream};

use super::{Listener, Socket};

impl Listener {
    pub fn bind_unix(addr: impl AsRef<str>) -> IoResult<Self> {
        let addr = parse_unix_addr(addr)?;
        let listener = StdUnixListener::bind_addr(&addr)?;
        Self::try_from(listener)
    }

    /// # Safety
    /// The file descriptor must represent a unix listener.
    pub unsafe fn from_raw_unix_listener_fd(fd: std::os::fd::RawFd) -> IoResult<Self> {
        let listener = unsafe { StdUnixListener::from_raw_fd(fd) };
        Self::try_from(listener)
    }
}

impl Socket {
    pub async fn connect_unix(addr: impl AsRef<str>) -> IoResult<Self> {
        let addr = parse_unix_addr(addr)?;
        let socket = StdUnixStream::connect_addr(&addr)?;
        Self::try_from(socket)
    }

    /// # Safety
    /// The file descriptor must represent a unix socket.
    pub unsafe fn from_raw_unix_socket_fd(fd: RawFd) -> IoResult<Self> {
        let socket = unsafe { StdUnixStream::from_raw_fd(fd) };
        Self::try_from(socket)
    }
}

impl From<UnixListener> for Listener {
    fn from(listener: UnixListener) -> Self {
        Self::new(stream! {
            loop {
                yield listener.accept().await.map(|(socket, _)| socket);
            }
        })
    }
}

impl TryFrom<StdUnixListener> for Listener {
    type Error = IoError;
    fn try_from(listener: StdUnixListener) -> IoResult<Self> {
        listener.set_nonblocking(true)?;
        Ok(Self::from(UnixListener::from_std(listener)?))
    }
}

impl From<UnixStream> for Socket {
    fn from(socket: UnixStream) -> Self {
        Self::new(socket)
    }
}

impl TryFrom<StdUnixStream> for Socket {
    type Error = IoError;
    fn try_from(socket: StdUnixStream) -> IoResult<Self> {
        socket.set_nonblocking(true)?;
        Ok(Self::from(UnixStream::from_std(socket)?))
    }
}

fn parse_unix_addr(addr: impl AsRef<str>) -> IoResult<SocketAddr> {
    let addr = addr.as_ref();

    #[cfg(any(target_os = "linux", target_os = "android"))]
    if let Some(addr) = addr.strip_prefix('@') {
        use std::os::linux::net::SocketAddrExt as _;
        return SocketAddr::from_abstract_name(addr);
    }

    #[cfg(not(any(target_os = "linux", target_os = "android")))]
    if addr.starts_with('@') {
        return Err(io_other!(
            "Abstract unix domain socket is not support on this platform",
        ));
    }

    SocketAddr::from_pathname(addr)
}