/rust/registry/src/index.crates.io-1949cf8c6b5b557f/rustix-1.1.2/src/termios/tty.rs
Line | Count | Source |
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 | 0 | pub fn isatty<Fd: AsFd>(fd: Fd) -> bool { |
23 | 0 | backend::termios::syscalls::isatty(fd.as_fd()) |
24 | 0 | } |
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 | 0 | pub fn ttyname<Fd: AsFd, B: Into<Vec<u8>>>(fd: Fd, reuse: B) -> io::Result<CString> { |
46 | 0 | _ttyname(fd.as_fd(), reuse.into()) |
47 | 0 | } |
48 | | |
49 | | #[cfg(not(any(target_os = "fuchsia", target_os = "wasi")))] |
50 | | #[cfg(feature = "alloc")] |
51 | | #[cfg(feature = "fs")] |
52 | | #[allow(unsafe_code)] |
53 | 0 | fn _ttyname(fd: BorrowedFd<'_>, mut buffer: Vec<u8>) -> io::Result<CString> { |
54 | 0 | buffer.clear(); |
55 | 0 | buffer.reserve(SMALL_PATH_BUFFER_SIZE); |
56 | | |
57 | | loop { |
58 | 0 | match backend::termios::syscalls::ttyname(fd, buffer.spare_capacity_mut()) { |
59 | 0 | Err(io::Errno::RANGE) => { |
60 | 0 | // Use `Vec` reallocation strategy to grow capacity |
61 | 0 | // exponentially. |
62 | 0 | buffer.reserve(buffer.capacity() + 1); |
63 | 0 | } |
64 | 0 | Ok(len) => { |
65 | | // SAFETY: Assume the backend returns the length of the string |
66 | | // excluding the NUL. |
67 | 0 | unsafe { |
68 | 0 | buffer.set_len(len + 1); |
69 | 0 | } |
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 | 0 | return Ok(CString::from_vec_with_nul_unchecked(buffer)); |
85 | | } |
86 | | } |
87 | 0 | Err(errno) => return Err(errno), |
88 | | } |
89 | | } |
90 | 0 | } |