jay-compositor 1.15.0

The Jay compositor
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

use crate::utils::errorfmt::ErrorFmt;
use bstr::ByteSlice;
use jay_algorithms::oserror::OsErrorExt;
use std::os::unix::ffi::OsStrExt;
use std::rc::Rc;
use uapi::OwnedFd;
use uapi::c;

pub struct PidInfo {
    pub uid: c::uid_t,
    pub pid: c::pid_t,
    pub comm: String,
    pub exe: Rc<String>,
}

pub fn get_pid_info(uid: c::uid_t, pid: c::pid_t) -> PidInfo {
    let comm = match std::fs::read(format!("/proc/{pid}/comm")) {
        Ok(name) => name.trim_ascii_end().as_bstr().to_string(),
        Err(e) => {
            log::warn!("Could not read `comm` of pid {}: {}", pid, ErrorFmt(e));
            "Unknown".to_string()
        }
    };
    let exe = match std::fs::read_link(format!("/proc/{pid}/exe")) {
        Ok(name) => name
            .as_os_str()
            .as_bytes()
            .trim_ascii_end()
            .as_bstr()
            .to_string(),
        Err(e) => {
            log::warn!("Could not read `exe` of pid {}: {}", pid, ErrorFmt(e));
            "Unknown".to_string()
        }
    };
    PidInfo {
        uid,
        pid,
        comm,
        exe: Rc::new(exe),
    }
}

pub fn get_socket_creds(socket: &OwnedFd) -> Option<(c::uid_t, c::pid_t)> {
    let mut cred = c::ucred {
        pid: 0,
        uid: 0,
        gid: 0,
    };
    match uapi::getsockopt(socket.raw(), c::SOL_SOCKET, c::SO_PEERCRED, &mut cred).to_os_error() {
        Ok(_) => Some((cred.uid, cred.pid)),
        Err(e) => {
            log::error!(
                "Cannot determine peer credentials of new connection: {}",
                ErrorFmt(e),
            );
            None
        }
    }
}