isatty.rs.html -- source
Description: Source to the Rust file `library/test/src/helpers/isatty.rs`.
Keywords: rust, rustlang, rust-lang

     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

    //! Helper module which provides a function to test
    //! if stdout is a tty.

    cfg_if::cfg_if! {
        if #[cfg(unix)] {
            pub fn stdout_isatty() -> bool {
                unsafe { libc::isatty(libc::STDOUT_FILENO) != 0 }
            }
        } else if #[cfg(windows)] {
            pub fn stdout_isatty() -> bool {
                type DWORD = u32;
                type BOOL = i32;
                type HANDLE = *mut u8;
                type LPDWORD = *mut u32;
                const STD_OUTPUT_HANDLE: DWORD = -11i32 as DWORD;
                extern "system" {
                    fn GetStdHandle(which: DWORD) -> HANDLE;
                    fn GetConsoleMode(hConsoleHandle: HANDLE, lpMode: LPDWORD) -> BOOL;
                }
                unsafe {
                    let handle = GetStdHandle(STD_OUTPUT_HANDLE);
                    let mut out = 0;
                    GetConsoleMode(handle, &mut out) != 0
                }
            }
        } else {
            // FIXME: Implement isatty on SGX
            pub fn stdout_isatty() -> bool {
                false
            }
        }
    }