From d3095900b941d7b08d579f9dc5e9dac45690228b Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Sun, 15 Oct 2023 21:25:49 +0200
Subject: Use `registers` mod

---
 src/trace.rs | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

(limited to 'src/trace.rs')

diff --git a/src/trace.rs b/src/trace.rs
index e576320..399f1c6 100644
--- a/src/trace.rs
+++ b/src/trace.rs
@@ -1,3 +1,4 @@
+use crate::registers::RegistersData;
 use nix::{
     sys::{
         ptrace,
@@ -56,16 +57,12 @@ pub fn trace(pid: Pid, file_to_print: Option<String>) -> anyhow::Result<Vec<u8>>
             WaitStatus::Stopped(pid, signal) => {
                 match signal {
                     Signal::SIGTRAP => {
-                        let regs = ptrace::getregs(pid)?;
                         if have_to_print {
-                            let output = format!(
-                                "{}({:x}, {:x}, {:x}, ...) = {:x}",
-                                regs.orig_rax, regs.rdi, regs.rsi, regs.rdx, regs.rax
-                            );
-                            writeln!(lines, "{output}")?;
+                            let reg = RegistersData::new(ptrace::getregs(pid)?);
+                            writeln!(lines, "{}", reg.output())?;
 
                             if let Some(ref mut f) = f {
-                                writeln!(f, "{output}")?;
+                                writeln!(f, "{}", reg.output())?;
                             }
                         }
                     }
-- 
cgit v1.2.3-18-g5258