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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

/// Helper macro to execute a system call that returns an `io::Result`.
//
// Macro must be defined before any modules that uses them.
#[allow(unused_macros)]
macro_rules! syscall {
    ($fn: ident ( $($arg: expr),* $(,)* ) ) => {{
        let res = unsafe { libc::$fn($($arg, )*) };
        if res == -1 {
            Err(std::io::Error::last_os_error())
        } else {
            Ok(res)
        }
    }};
}

cfg_os_poll! {
    mod selector;
    pub(crate) use self::selector::{event, Event, Events, Selector};

    mod sourcefd;
    #[cfg(feature = "os-ext")]
    pub use self::sourcefd::SourceFd;

    mod waker;
    pub(crate) use self::waker::Waker;

    cfg_net! {
        mod net;

        pub(crate) mod tcp;
        pub(crate) mod udp;
        pub(crate) mod uds;
        pub use self::uds::SocketAddr;
    }

    cfg_io_source! {
        // Both `kqueue` and `epoll` don't need to hold any user space state.
        #[cfg(not(any(mio_unsupported_force_poll_poll, target_os = "solaris", target_os = "vita")))]
        mod stateless_io_source {
            use std::io;
            use std::os::unix::io::RawFd;
            use crate::Registry;
            use crate::Token;
            use crate::Interest;

            pub(crate) struct IoSourceState;

            impl IoSourceState {
                pub fn new() -> IoSourceState {
                    IoSourceState
                }

                pub fn do_io<T, F, R>(&self, f: F, io: &T) -> io::Result<R>
                where
                    F: FnOnce(&T) -> io::Result<R>,
                {
                    // We don't hold state, so we can just call the function and
                    // return.
                    f(io)
                }

                pub fn register(
                    &mut self,
                    registry: &Registry,
                    token: Token,
                    interests: Interest,
                    fd: RawFd,
                ) -> io::Result<()> {
                    // Pass through, we don't have any state
                    registry.selector().register(fd, token, interests)
                }

                pub fn reregister(
                    &mut self,
                    registry: &Registry,
                    token: Token,
                    interests: Interest,
                    fd: RawFd,
                ) -> io::Result<()> {
                    // Pass through, we don't have any state
                    registry.selector().reregister(fd, token, interests)
                }

                pub fn deregister(&mut self, registry: &Registry, fd: RawFd) -> io::Result<()> {
                    // Pass through, we don't have any state
                    registry.selector().deregister(fd)
                }
            }
        }

        #[cfg(not(any(mio_unsupported_force_poll_poll, target_os = "solaris",target_os = "vita")))]
        pub(crate) use self::stateless_io_source::IoSourceState;

        #[cfg(any(mio_unsupported_force_poll_poll, target_os = "solaris", target_os = "vita"))]
        pub(crate) use self::selector::IoSourceState;
    }

    #[cfg(any(
        // For the public `pipe` module, must match `cfg_os_ext` macro.
        feature = "os-ext",
        // For the `Waker` type based on a pipe.
        mio_unsupported_force_waker_pipe,
        target_os = "aix",
        target_os = "dragonfly",
        target_os = "illumos",
        target_os = "netbsd",
        target_os = "openbsd",
        target_os = "redox",
        target_os = "solaris",
        target_os = "vita",
    ))]
    pub(crate) mod pipe;
}

cfg_not_os_poll! {
    cfg_net! {
        mod uds;
        pub use self::uds::SocketAddr;
    }

    cfg_any_os_ext! {
        mod sourcefd;
        #[cfg(feature = "os-ext")]
        pub use self::sourcefd::SourceFd;
    }
}