ptrace-inject 0.1.1

Shared library injection for *nix using ptrace.
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

use clap::Parser;
use color_eyre::{eyre::eyre, Result};
use ptrace_inject::{Injector, Process};

#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[command(flatten)]
    verbose: clap_verbosity_flag::Verbosity<clap_verbosity_flag::WarnLevel>,

    /// Name of a process to attach to.
    #[arg(short, long, group = "target")]
    name: Option<String>,

    /// PID of a process to attach to.
    #[arg(short, long, group = "target")]
    pid: Option<u32>,

    /// Path to the library to inject.
    #[arg(requires = "target")]
    library: std::path::PathBuf,
}

impl Args {
    fn run(self) -> Result<()> {
        pretty_env_logger::formatted_builder()
            .filter_level(self.verbose.log_level_filter())
            .init();
        let process = if let Some(name) = self.name {
            Process::by_name(&name)?
                .ok_or_else(|| eyre!("could not find process with name {name:?}"))?
        } else if let Some(pid) = self.pid {
            Process::get(pid)?
        } else {
            panic!("no target specified, but clap should have caught this");
        };
        Injector::attach(process)?.inject(&self.library)
    }
}

fn main() -> Result<()> {
    color_eyre::install()?;
    Args::parse().run()
}