procinsh 0.1.2

A local, read-only Linux x86-64 process inspector
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use super::{ProcessId, check_identity, permission_help};
use anyhow::{Result, bail, ensure};
use serde::Serialize;

pub const MAX_READ: usize = 65536;

#[derive(Serialize)]
pub struct MemoryRead {
    pub process_id: ProcessId,
    #[serde(serialize_with = "super::maps::hex")]
    pub address: u64,
    pub requested_length: usize,
    pub bytes: Vec<u8>,
    pub partial: bool,
    pub captured_at: u64,
}

pub fn read_raw(pid: i32, address: u64, length: usize) -> Result<Vec<u8>> {
    ensure!(
        (1..=MAX_READ).contains(&length),
        "length must be 1..=65536 bytes"
    );
    ensure!(
        address.checked_add(length as u64).is_some(),
        "address range overflow"
    );
    let mut bytes = vec![0u8; length];
    let local = libc::iovec {
        iov_base: bytes.as_mut_ptr().cast(),
        iov_len: length,
    };
    let remote = libc::iovec {
        iov_base: address as *mut libc::c_void,
        iov_len: length,
    };
    // SAFETY: local points at an allocated buffer; the kernel validates remote memory.
    let count = unsafe { libc::process_vm_readv(pid, &local, 1, &remote, 1, 0) };
    if count < 0 {
        bail!(permission_help(
            "process_vm_readv",
            std::io::Error::last_os_error()
        ));
    }
    bytes.truncate(count as usize);
    Ok(bytes)
}

pub fn read(id: ProcessId, address: u64, length: usize) -> Result<MemoryRead> {
    check_identity(id)?;
    let bytes = read_raw(id.pid, address, length)?;
    check_identity(id)?;
    Ok(MemoryRead {
        process_id: id,
        address,
        requested_length: length,
        partial: bytes.len() != length,
        bytes,
        captured_at: super::timestamp_ms(),
    })
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn reads_own_memory_and_rejects_invalid_ranges() {
        let value = *b"procinsh";
        let id = super::super::identity(std::process::id() as i32).unwrap();
        assert_eq!(
            read(id, value.as_ptr() as u64, value.len()).unwrap().bytes,
            value
        );
        assert!(read_raw(id.pid, 0, MAX_READ + 1).is_err());
        assert!(read_raw(id.pid, u64::MAX, 16).is_err());
        assert!(
            read(
                ProcessId {
                    start_time_ticks: id.start_time_ticks + 1,
                    ..id
                },
                value.as_ptr() as u64,
                1
            )
            .is_err()
        );
    }
}