mfio-rt 0.1.0

mfio based async runtime
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use nix::libc;
use nix::sys::socket::*;
use std::net::SocketAddr;
use std::os::fd::RawFd;

// From mio
#[allow(dead_code)]
pub(crate) fn new_for_addr(
    address: SocketAddr,
    nonblock: bool,
) -> std::io::Result<(AddressFamily, RawFd)> {
    let domain = match address {
        SocketAddr::V4(_) => AddressFamily::Inet,
        SocketAddr::V6(_) => AddressFamily::Inet6,
    };
    new_socket(domain, SockType::Stream, nonblock).map(|v| (domain, v))
}

/// Create a new non-blocking socket.
pub(crate) fn new_socket(
    domain: AddressFamily,
    socket_type: SockType,
    nonblock: bool,
) -> std::io::Result<RawFd> {
    #[cfg(any(
        target_os = "android",
        target_os = "dragonfly",
        target_os = "freebsd",
        target_os = "illumos",
        target_os = "linux",
        target_os = "netbsd",
        target_os = "openbsd",
    ))]
    let flags = SockFlag::SOCK_CLOEXEC.union(SockFlag::from_bits_truncate(if nonblock {
        libc::SOCK_NONBLOCK
    } else {
        0
    }));
    #[cfg(not(any(
        target_os = "android",
        target_os = "dragonfly",
        target_os = "freebsd",
        target_os = "illumos",
        target_os = "linux",
        target_os = "netbsd",
        target_os = "openbsd",
    )))]
    let flags = SockFlag::empty();

    let fd = socket(domain, socket_type, flags, None)?;

    // Mimick `libstd` and set `SO_NOSIGPIPE` on apple systems.
    #[cfg(any(
        target_os = "ios",
        target_os = "macos",
        target_os = "tvos",
        target_os = "watchos",
    ))]
    if let Err(e) = nix::errno::Errno::result(unsafe {
        libc::setsockopt(
            fd,
            libc::SOL_SOCKET,
            libc::SO_NOSIGPIPE,
            &1 as *const libc::c_int as *const libc::c_void,
            core::mem::size_of::<libc::c_int>() as libc::socklen_t,
        )
    }) {
        let _ = nix::unistd::close(fd);
        return Err(e.into());
    }

    // Darwin doesn't have SOCK_NONBLOCK or SOCK_CLOEXEC.
    #[cfg(any(
        target_os = "ios",
        target_os = "macos",
        target_os = "tvos",
        target_os = "watchos",
    ))]
    if nonblock {
        use nix::fcntl::*;

        if let Err(e) = set_nonblock(fd) {
            let _ = nix::unistd::close(fd);
            return Err(e.into());
        }

        if let Err(e) = fcntl(fd, FcntlArg::F_SETFD(FdFlag::FD_CLOEXEC)) {
            let _ = nix::unistd::close(fd);
            return Err(e.into());
        }
    }

    Ok(fd)
}

pub(crate) fn set_nonblock(fd: RawFd) -> std::result::Result<(), nix::errno::Errno> {
    use nix::fcntl::*;

    let flags = fcntl(fd, FcntlArg::F_GETFL)?;
    fcntl(
        fd,
        FcntlArg::F_SETFL(OFlag::from_bits_truncate(flags).union(OFlag::O_NONBLOCK)),
    )?;

    Ok(())
}