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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
mod trace;
use crossterm::{
event::{self, Event, KeyCode},
terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
ExecutableCommand,
};
use ratatui::{prelude::*, widgets::*};
use std::{
io::{self, stdout, Write},
str,
};
use crate::trace::{exec, trace};
use clap::Parser;
use fork::{fork, Fork};
use nix::unistd::Pid;
#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Args {
/// Command to execute from ptrace
command: String,
/// Write the output to a file instead of the standard output
#[arg(short = 'f', long = "file")]
file_to_print: Option<String>,
/// If defined, it hides the TUI
#[arg(long = "no-tui", default_value_t = false)]
no_tui: bool,
}
struct UI {
height: usize,
max_lines: usize,
scroll: usize,
}
impl UI {
fn new() -> UI {
UI {
height: 0,
max_lines: 0,
scroll: 0,
}
}
}
/// Create a fork of the program and execute the process in the child. Parent gets the pid
/// value and trace it.
fn main() -> anyhow::Result<()> {
let args = Args::parse();
let pid = match fork() {
Ok(Fork::Child) => return exec(&args.command),
Ok(Fork::Parent(child)) => Pid::from_raw(child as i32),
Err(err) => panic!("fork() failed: {err}"),
};
let output = trace(pid, args.file_to_print)?;
let lines = str::from_utf8(&output)?.trim();
if !args.no_tui {
enable_raw_mode()?;
stdout().execute(EnterAlternateScreen)?;
let mut terminal = Terminal::new(CrosstermBackend::new(stdout()))?;
let mut ui = UI::new();
ui.max_lines = lines.split('\n').count() + 1;
let mut should_quit = false;
while !should_quit {
ui.height = terminal.get_frame().size().height as usize;
terminal.draw(move |frame| {
let size = frame.size();
frame.render_widget(
Paragraph::new(lines)
.block(
Block::default()
.border_style(Style::default().fg(Color::Yellow))
.title(format!("[{pid}]"))
.title(
block::Title::from(format!(
"[lines {}-{}]",
ui.scroll,
ui.scroll + ui.height
))
.position(block::Position::Bottom)
.alignment(Alignment::Right),
)
.borders(Borders::ALL),
)
.scroll((ui.scroll as u16, 0)),
size,
);
})?;
should_quit = handle_events(&mut ui)?;
}
disable_raw_mode()?;
stdout().execute(LeaveAlternateScreen)?;
} else {
writeln!(io::stdout(), "{lines}")?;
}
Ok(())
}
fn handle_events(ui: &mut UI) -> io::Result<bool> {
if event::poll(std::time::Duration::from_millis(50))? {
if let Event::Key(key) = event::read()? {
if key.kind == event::KeyEventKind::Press {
match key.code {
KeyCode::Char('q') => {
return Ok(true);
}
KeyCode::Char('j') | KeyCode::Down => {
if ui.scroll < (ui.max_lines - ui.height + 1) {
ui.scroll += 1;
}
}
KeyCode::Char('J') | KeyCode::Char('G') => {
ui.scroll = ui.max_lines - ui.height + 1;
}
KeyCode::Char('k') | KeyCode::Up => {
if ui.scroll > 1 {
ui.scroll -= 1;
}
}
KeyCode::Char('K') | KeyCode::Char('0') => {
ui.scroll = ui.height;
}
_ => {}
}
}
}
}
Ok(false)
}
|
