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
|
use std::io::{self, Write};
fn main() {
if let Err(err) = try_main() {
eprintln!("{err}");
std::process::exit(1);
}
}
fn try_main() -> io::Result<()> {
let mut tee_stderr = false;
let mut echo_stdin = false;
let mut echo_env = false;
let mut fail = false;
let mut suicide = false;
let mut args = std::env::args().skip(1).peekable();
while let Some(arg) = args.peek() {
match arg.as_str() {
"-e" => tee_stderr = true,
"-i" => echo_stdin = true,
"-$" => echo_env = true,
"-f" => fail = true,
"-s" => suicide = true,
_ => break,
}
args.next();
}
let stdin = io::stdin();
let stdout = io::stdout();
let stderr = io::stderr();
let mut stdin = stdin.lock();
let mut stdout = stdout.lock();
let mut stderr = stderr.lock();
macro_rules! w {
($($tt:tt)*) => {
write!(stdout, $($tt)*)?;
if tee_stderr {
write!(stderr, $($tt)*)?;
}
}
}
if echo_stdin {
io::copy(&mut stdin, &mut stdout)?;
} else if echo_env {
for key in args {
if let Some(v) = std::env::var_os(&key) {
w!("{}={}\n", key, v.to_string_lossy());
}
}
} else {
let mut space = "";
for arg in args {
w!("{}{}", space, arg);
space = " ";
}
w!("\n");
}
if fail {
return Err(io::ErrorKind::Other.into());
}
if suicide {
#[cfg(unix)]
unsafe {
let pid = signals::getpid();
if pid > 0 {
signals::kill(pid, 9);
}
}
}
Ok(())
}
#[cfg(unix)]
mod signals {
use std::os::raw::c_int;
extern "C" {
pub fn kill(pid: c_int, sig: c_int) -> c_int;
pub fn getpid() -> c_int;
}
}
|
