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
extern crate libc;
use std::env;
use std::ffi::CString;
use std::ptr;
use std::os::unix::ffi::OsStrExt;
const DEFAULT_PAGER_ENV: &'static str = "PAGER";
#[derive(Debug, Default)]
pub struct Pager {
env: String,
ok: bool,
}
impl Pager {
pub fn new() -> Self {
Pager {
env: String::from(DEFAULT_PAGER_ENV),
ok: true,
}
}
pub fn env(&mut self, env: &str) -> &Self {
self.env = String::from(env);
self
}
pub fn ok(&self) -> bool {
self.ok
}
pub fn setup(&mut self) {
if let Some(pager) = getenv(&self.env) {
let (pager_stdin, main_stdout) = pipe();
let pid = fork();
match pid {
-1 => {
close(pager_stdin);
close(main_stdout);
self.ok = false
}
0 => {
dup2(main_stdout, libc::STDOUT_FILENO);
close(pager_stdin);
}
_ => {
let argv = vec![pager.as_ptr(), ptr::null()];
dup2(pager_stdin, libc::STDIN_FILENO);
close(main_stdout);
execvp(argv);
}
}
}
}
}
fn getenv(var: &str) -> Option<CString> {
let value = env::var_os(&var).unwrap();
let value = value.as_os_str().as_bytes();
let value = CString::new(value);
value.ok()
}
fn fork() -> libc::pid_t {
unsafe { libc::fork() }
}
fn execvp(argv: Vec<*const libc::c_char>) {
assert!(unsafe { libc::execvp(argv[0], argv.as_ptr()) } > -1);
}
fn dup2(fd1: i32, fd2: i32) {
assert!(unsafe { libc::dup2(fd1, fd2) } > -1);
}
fn close(fd: i32) {
assert_eq!(unsafe { libc::close(fd) }, 0);
}
fn pipe() -> (i32, i32) {
let mut fds = [0; 2];
assert_eq!(unsafe { libc::pipe(fds.as_mut_ptr()) }, 0);
(fds[0], fds[1])
}