1 | //! Functions which operate on file descriptors which might be terminals. |
2 | |
3 | use crate::backend; |
4 | #[cfg (feature = "alloc" )] |
5 | #[cfg (feature = "fs" )] |
6 | #[cfg (not(any(target_os = "fuchsia" , target_os = "wasi" )))] |
7 | use crate::path::SMALL_PATH_BUFFER_SIZE; |
8 | use backend::fd::AsFd; |
9 | #[cfg (feature = "alloc" )] |
10 | #[cfg (not(any(target_os = "fuchsia" , target_os = "wasi" )))] |
11 | use {crate::ffi::CString, crate::io, alloc::vec::Vec, backend::fd::BorrowedFd}; |
12 | |
13 | /// `isatty(fd)`—Tests whether a file descriptor refers to a terminal. |
14 | /// |
15 | /// # References |
16 | /// - [POSIX] |
17 | /// - [Linux] |
18 | /// |
19 | /// [POSIX]: https://pubs.opengroup.org/onlinepubs/9799919799/functions/isatty.html |
20 | /// [Linux]: https://man7.org/linux/man-pages/man3/isatty.3.html |
21 | #[inline ] |
22 | pub fn isatty<Fd: AsFd>(fd: Fd) -> bool { |
23 | backend::termios::syscalls::isatty(fd.as_fd()) |
24 | } |
25 | |
26 | /// `ttyname_r(fd)`—Returns the name of the tty open on `fd`. |
27 | /// |
28 | /// If `reuse` already has available capacity, reuse it if possible. |
29 | /// |
30 | /// On Linux, this function depends on procfs being mounted on /proc. |
31 | /// |
32 | /// # References |
33 | /// - [POSIX] |
34 | /// - [Linux] |
35 | /// |
36 | /// [POSIX]: https://pubs.opengroup.org/onlinepubs/9799919799/functions/ttyname.html |
37 | /// [Linux]: https://man7.org/linux/man-pages/man3/ttyname.3.html |
38 | #[cfg (not(any(target_os = "fuchsia" , target_os = "wasi" )))] |
39 | #[cfg (feature = "alloc" )] |
40 | #[cfg (feature = "fs" )] |
41 | #[doc (alias = "ttyname_r" )] |
42 | #[cfg_attr (docsrs, doc(cfg(feature = "fs" )))] |
43 | #[cfg_attr (docsrs, doc(cfg(feature = "alloc" )))] |
44 | #[inline ] |
45 | pub fn ttyname<Fd: AsFd, B: Into<Vec<u8>>>(fd: Fd, reuse: B) -> io::Result<CString> { |
46 | _ttyname(fd.as_fd(), buffer:reuse.into()) |
47 | } |
48 | |
49 | #[cfg (not(any(target_os = "fuchsia" , target_os = "wasi" )))] |
50 | #[cfg (feature = "alloc" )] |
51 | #[cfg (feature = "fs" )] |
52 | #[allow (unsafe_code)] |
53 | fn _ttyname(fd: BorrowedFd<'_>, mut buffer: Vec<u8>) -> io::Result<CString> { |
54 | buffer.clear(); |
55 | buffer.reserve(SMALL_PATH_BUFFER_SIZE); |
56 | |
57 | loop { |
58 | match backend::termios::syscalls::ttyname(fd, buffer.spare_capacity_mut()) { |
59 | Err(io::Errno::RANGE) => { |
60 | // Use `Vec` reallocation strategy to grow capacity |
61 | // exponentially. |
62 | buffer.reserve(buffer.capacity() + 1); |
63 | } |
64 | Ok(len) => { |
65 | // SAFETY: Assume the backend returns the length of the string |
66 | // excluding the NUL. |
67 | unsafe { |
68 | buffer.set_len(len + 1); |
69 | } |
70 | |
71 | // SAFETY: |
72 | // - “ttyname_r stores this pathname in the buffer buf” |
73 | // - [POSIX definition 3.271: Pathname]: “A string that is |
74 | // used to identify a file.” |
75 | // - [POSIX definition 3.375: String]: “A contiguous sequence |
76 | // of bytes terminated by and including the first null byte.” |
77 | // |
78 | // Thus, there will be a single NUL byte at the end of the |
79 | // string. |
80 | // |
81 | // [POSIX definition 3.271: Pathname]: https://pubs.opengroup.org/onlinepubs/9799919799/basedefs/V1_chap03.html#tag_03_271 |
82 | // [POSIX definition 3.375: String]: https://pubs.opengroup.org/onlinepubs/9799919799/basedefs/V1_chap03.html#tag_03_375 |
83 | unsafe { |
84 | return Ok(CString::from_vec_with_nul_unchecked(buffer)); |
85 | } |
86 | } |
87 | Err(errno) => return Err(errno), |
88 | } |
89 | } |
90 | } |
91 | |