sudo-rs 0.2.14

A memory safe implementation of sudo and su.
Documentation 
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

use std::{
    ffi::OsString,
    fs, io,
    os::unix::fs::{FileTypeExt, MetadataExt},
    path::Path,
};

use crate::system::{DeviceId, Process, WithProcess, term::Terminal};

pub(super) fn ttyname_from_dev() -> io::Result<Option<OsString>> {
    let Ok(Some(tty_dev)) = Process::tty_device_id(WithProcess::Current) else {
        return Ok(None);
    };

    let tty_name = dev_check(Path::new("/dev/console"), tty_dev)
        .or_else(|| ttyname_from_stdioe(tty_dev))
        .or_else(|| find_tty_in_dir(Path::new("/dev/pts"), tty_dev))
        .or_else(|| find_tty_in_dir(Path::new("/dev"), tty_dev));

    if tty_name.is_some() {
        Ok(tty_name)
    } else {
        Err(io::ErrorKind::NotFound.into())
    }
}

fn is_our_tty(metadata: fs::Metadata, tty_dev: DeviceId) -> bool {
    metadata.file_type().is_char_device() && metadata.rdev() == tty_dev.inner()
}

fn ttyname_from_stdioe(tty_dev: DeviceId) -> Option<OsString> {
    [
        io::stdin().ttyname(),
        io::stdout().ttyname(),
        io::stderr().ttyname(),
    ]
    .iter()
    .flatten()
    .find_map(|ttyname| dev_check(ttyname.as_ref(), tty_dev))
}

fn dev_check(path: &Path, tty_dev: DeviceId) -> Option<OsString> {
    let metadata = fs::metadata(path).ok()?;

    is_our_tty(metadata, tty_dev).then(|| path.into())
}

fn find_tty_in_dir(dir: &Path, tty_dev: DeviceId) -> Option<OsString> {
    for entry in fs::read_dir(dir).ok()?.filter_map(|entry| entry.ok()) {
        if let Ok(metadata) = entry.metadata() {
            if is_our_tty(metadata, tty_dev) {
                return Some(entry.path().into());
            }
        }
    }

    None
}