summaryrefslogtreecommitdiffstats
path: root/third_party/rust/nix/src/features.rs
blob: 39d17604e1a1092a96916b3b62b90ba7b4fb3e39 (plain)
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
//! Feature tests for OS functionality
pub use self::os::*;

#[cfg(any(target_os = "linux", target_os = "android"))]
mod os {
    use crate::sys::utsname::uname;
    use crate::Result;
    use std::os::unix::ffi::OsStrExt;

    // Features:
    // * atomic cloexec on socket: 2.6.27
    // * pipe2: 2.6.27
    // * accept4: 2.6.28

    static VERS_UNKNOWN: usize = 1;
    static VERS_2_6_18: usize = 2;
    static VERS_2_6_27: usize = 3;
    static VERS_2_6_28: usize = 4;
    static VERS_3: usize = 5;

    #[inline]
    fn digit(dst: &mut usize, b: u8) {
        *dst *= 10;
        *dst += (b - b'0') as usize;
    }

    fn parse_kernel_version() -> Result<usize> {
        let u = uname()?;

        let mut curr: usize = 0;
        let mut major: usize = 0;
        let mut minor: usize = 0;
        let mut patch: usize = 0;

        for &b in u.release().as_bytes() {
            if curr >= 3 {
                break;
            }

            match b {
                b'.' | b'-' => {
                    curr += 1;
                }
                b'0'..=b'9' => match curr {
                    0 => digit(&mut major, b),
                    1 => digit(&mut minor, b),
                    _ => digit(&mut patch, b),
                },
                _ => break,
            }
        }

        Ok(if major >= 3 {
            VERS_3
        } else if major >= 2 {
            if minor >= 7 {
                VERS_UNKNOWN
            } else if minor >= 6 {
                if patch >= 28 {
                    VERS_2_6_28
                } else if patch >= 27 {
                    VERS_2_6_27
                } else {
                    VERS_2_6_18
                }
            } else {
                VERS_UNKNOWN
            }
        } else {
            VERS_UNKNOWN
        })
    }

    fn kernel_version() -> Result<usize> {
        static mut KERNEL_VERS: usize = 0;

        unsafe {
            if KERNEL_VERS == 0 {
                KERNEL_VERS = parse_kernel_version()?;
            }

            Ok(KERNEL_VERS)
        }
    }

    /// Check if the OS supports atomic close-on-exec for sockets
    pub fn socket_atomic_cloexec() -> bool {
        kernel_version()
            .map(|version| version >= VERS_2_6_27)
            .unwrap_or(false)
    }

    #[test]
    pub fn test_parsing_kernel_version() {
        assert!(kernel_version().unwrap() > 0);
    }
}

#[cfg(any(
        target_os = "dragonfly",    // Since ???
        target_os = "freebsd",      // Since 10.0
        target_os = "illumos",      // Since ???
        target_os = "netbsd",       // Since 6.0
        target_os = "openbsd",      // Since 5.7
        target_os = "redox",        // Since 1-july-2020
))]
mod os {
    /// Check if the OS supports atomic close-on-exec for sockets
    pub const fn socket_atomic_cloexec() -> bool {
        true
    }
}

#[cfg(any(
    target_os = "macos",
    target_os = "ios",
    target_os = "fuchsia",
    target_os = "haiku",
    target_os = "solaris"
))]
mod os {
    /// Check if the OS supports atomic close-on-exec for sockets
    pub const fn socket_atomic_cloexec() -> bool {
        false
    }
}