2 files changed, 76 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..893b9f5
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,27 @@
+mod trace;
+
+use crate::trace::{exec, trace};
+use clap::Parser;
+use std::process::Command;
+
+#[derive(Parser)]
+#[command(author, version, about, long_about = None)]
+struct Args {
+    command: String,
+}
+
+fn main() -> anyhow::Result<()> {
+    let args = Args::parse();
+    let params = args.command.split(' ').collect::<Vec<&str>>();
+
+    let mut command = Command::new(params[0]);
+    if params.len() > 1 {
+        for arg in &params[1..] {
+            command.arg(arg);
+        }
+    }
+    let pid = exec(&mut command)?;
+    trace(pid)?;
+
+    Ok(())
+}
diff --git a/src/trace.rs b/src/trace.rs
new file mode 100644
index 0000000..0031fcd
--- /dev/null
+++ b/src/trace.rs
@@ -0,0 +1,49 @@
+use nix::{
+    sys::{
+        ptrace,
+        signal::Signal,
+        wait::{waitpid, WaitStatus},
+    },
+    unistd::Pid,
+};
+use std::{os::unix::process::CommandExt, process::Command};
+
+pub fn exec(command: &mut Command) -> anyhow::Result<Pid> {
+    unsafe {
+        command.pre_exec(|| ptrace::traceme().map_err(|e| e.into()));
+    }
+    let child = command.spawn()?;
+    Ok(Pid::from_raw(child.id() as i32))
+}
+
+pub fn trace(pid: Pid) -> anyhow::Result<()> {
+    let mut have_to_print = true;
+    loop {
+        have_to_print ^= true;
+        ptrace::syscall(pid, None)?;
+        let status = waitpid(pid, None)?;
+
+        match status {
+            WaitStatus::Exited(_pid, _) => {
+                break;
+            }
+            WaitStatus::Stopped(pid, signal) => {
+                match signal {
+                    Signal::SIGTRAP => {
+                        let regs = ptrace::getregs(pid)?;
+                        if have_to_print {
+                            println!(
+                                "{}({:x}, {:x}, {:x}, ...) = {:x}",
+                                regs.orig_rax, regs.rdi, regs.rsi, regs.rdx, regs.rax,
+                            );
+                        }
+                    }
+                    _ => {}
+                };
+            }
+            _ => {}
+        };
+    }
+
+    Ok(())
+}