diff options
Diffstat (limited to 'src/trace.rs')
| -rw-r--r-- | src/trace.rs | 25 |
1 files changed, 21 insertions, 4 deletions
diff --git a/src/trace.rs b/src/trace.rs index 04b3ecd..e826c53 100644 --- a/src/trace.rs +++ b/src/trace.rs @@ -6,7 +6,12 @@ use nix::{ }, unistd::Pid, }; -use std::{os::unix::process::CommandExt, process::Command}; +use std::{ + fs::File, + io::{self, Write}, + os::unix::process::CommandExt, + process::Command, +}; /// Exec the `command` value tracing it with `ptrace` lib pub fn exec(command: &String) -> anyhow::Result<()> { @@ -25,7 +30,7 @@ pub fn exec(command: &String) -> anyhow::Result<()> { } /// Trace a process with `pid` ID -pub fn trace(pid: Pid) -> anyhow::Result<()> { +pub fn trace(pid: Pid, file_to_print: Option<String>) -> anyhow::Result<()> { // Since you have to do 2 syscalls (start and end) you have to alternate the print value, // because it could be equals except for the `rax` register. let mut have_to_print = true; @@ -33,6 +38,13 @@ pub fn trace(pid: Pid) -> anyhow::Result<()> { // First wait for the parent process _ = waitpid(pid, None)?; + // If `fiole_to_print` is not None, create a new file with that value for redirecting all the + // output (also in stdout) + let mut f = None; + if let Some(filename) = file_to_print { + f = Some(File::create(filename)?); + } + loop { have_to_print ^= true; ptrace::syscall(pid, None)?; @@ -49,10 +61,15 @@ pub fn trace(pid: Pid) -> anyhow::Result<()> { Signal::SIGTRAP => { let regs = ptrace::getregs(pid)?; if have_to_print { - println!( + let output = format!( "{}({:x}, {:x}, {:x}, ...) = {:x}", - regs.orig_rax, regs.rdi, regs.rsi, regs.rdx, regs.rax, + regs.orig_rax, regs.rdi, regs.rsi, regs.rdx, regs.rax ); + writeln!(io::stdout(), "{output}")?; + + if let Some(ref mut f) = f { + writeln!(f, "{output}")?; + } } } _ => {} |