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
|
use std::os::unix::process::CommandExt as _;
pub struct Command {
inner: std::process::Command,
stdin: Option<std::process::Stdio>,
stdout: Option<std::process::Stdio>,
stderr: Option<std::process::Stdio>,
}
impl Command {
pub fn new<S: AsRef<std::ffi::OsStr>>(program: S) -> Self {
Self {
inner: std::process::Command::new(program),
stdin: None,
stdout: None,
stderr: None,
}
}
pub fn arg<S: AsRef<std::ffi::OsStr>>(&mut self, arg: S) -> &mut Self {
self.inner.arg(arg);
self
}
pub fn args<I, S>(&mut self, args: I) -> &mut Self
where
I: IntoIterator<Item = S>,
S: AsRef<std::ffi::OsStr>,
{
self.inner.args(args);
self
}
pub fn env<K, V>(&mut self, key: K, val: V) -> &mut Self
where
K: AsRef<std::ffi::OsStr>,
V: AsRef<std::ffi::OsStr>,
{
self.inner.env(key, val);
self
}
pub fn envs<I, K, V>(&mut self, vars: I) -> &mut Self
where
I: IntoIterator<Item = (K, V)>,
K: AsRef<std::ffi::OsStr>,
V: AsRef<std::ffi::OsStr>,
{
self.inner.envs(vars);
self
}
pub fn env_remove<K: AsRef<std::ffi::OsStr>>(
&mut self,
key: K,
) -> &mut Self {
self.inner.env_remove(key);
self
}
pub fn env_clear(&mut self) -> &mut Self {
self.inner.env_clear();
self
}
pub fn current_dir<P: AsRef<std::path::Path>>(
&mut self,
dir: P,
) -> &mut Self {
self.inner.current_dir(dir);
self
}
pub fn stdin<T: Into<std::process::Stdio>>(
&mut self,
cfg: Option<T>,
) -> &mut Self {
self.stdin = cfg.map(Into::into);
self
}
pub fn stdout<T: Into<std::process::Stdio>>(
&mut self,
cfg: Option<T>,
) -> &mut Self {
self.stdout = cfg.map(Into::into);
self
}
pub fn stderr<T: Into<std::process::Stdio>>(
&mut self,
cfg: Option<T>,
) -> &mut Self {
self.stderr = cfg.map(Into::into);
self
}
pub fn spawn(
&mut self,
pty: &crate::blocking::Pty,
) -> crate::Result<std::process::Child> {
let (stdin, stdout, stderr, pre_exec) =
crate::sys::setup_subprocess(pty, pty.pts()?)?;
self.inner.stdin(self.stdin.take().unwrap_or(stdin));
self.inner.stdout(self.stdout.take().unwrap_or(stdout));
self.inner.stderr(self.stderr.take().unwrap_or(stderr));
// safe because setsid() and close() are async-signal-safe functions
// and ioctl() is a raw syscall (which is inherently
// async-signal-safe).
unsafe { self.inner.pre_exec(pre_exec) };
Ok(self.inner.spawn()?)
}
}
|