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
|
use nix::pty::{ForkptyResult, forkpty, PtyMaster};
use nix::unistd::read;
use std::vec::Vec;
use std::io::{self, Read, Write};
use std::process::Command;
use std::thread;
use std::time::Duration;
use vte::{Params, Parser, Perform};
struct Performer;
impl Perform for Performer {
fn print(&mut self, c: char) {
println!("[print] {:?}", c);
}
fn execute(&mut self, byte: u8) {
println!("[execute] {:02x}", byte);
}
fn hook(&mut self, params: &Params, intermediates: &[u8], ignore: bool, c: char) {
println!(
"[hook] params={:?}, intermediates={:?}, ignore={:?}, char={:?}",
params, intermediates, ignore, c
);
}
fn put(&mut self, byte: u8) {
println!("[put] {:02x}", byte);
}
fn unhook(&mut self) {
println!("[unhook]");
}
fn osc_dispatch(&mut self, params: &[&[u8]], bell_terminated: bool) {
println!("[osc_dispatch] params={:?} bell_terminated={}", params, bell_terminated);
}
fn csi_dispatch(&mut self, params: &Params, intermediates: &[u8], ignore: bool, c: char) {
println!(
"[csi_dispatch] params={:#?}, intermediates={:?}, ignore={:?}, char={:?}",
params, intermediates, ignore, c
);
}
fn esc_dispatch(&mut self, intermediates: &[u8], ignore: bool, byte: u8) {
println!(
"[esc_dispatch] intermediates={:?}, ignore={:?}, byte={:02x}",
intermediates, ignore, byte
);
}
}
fn main() {
// TODO get rid of hard coded `sh` path
let mut pty = spawn_pty("/home/andromeda/.nix-profile/bin/sh").unwrap();
println!("{:?}", &pty);
let input = io::stdin();
let mut handle = input.lock();
let mut statemachine = Parser::new();
let mut performer = Performer;
let mut buf = [0u8; 2048];
while true {
match read(&pty, &mut buf) {
Ok(0) => break,
Ok(n) => {
statemachine.advance(&mut performer, &buf[..n]);
},
Err(_e) => break,
};
};
}
// forks a new pty and returns file descriptor of the master
fn spawn_pty(shell: &str) -> Option<PtyMaster> {
// SAFETY safe unless os out of PTYs; incredibly unlikely
match unsafe {forkpty(None, None)} {
Ok(fork_pty_res) => match fork_pty_res {
ForkptyResult::Parent {master, ..} => {
// SAFETY `master` is a valid PtyMaster
return Some(unsafe{PtyMaster::from_owned_fd(master)});
}
ForkptyResult::Child =>{
Command::new(shell).spawn();
// TODO come up with an elegant solution
thread::sleep(Duration::MAX);
return None;
}
},
Err(_e) => return None,
};
}
|
