run_parts/
exec.rs

1use failure::{self, Error};
2use io_mux::{Mux, TaggedData};
3
4use std::io::{self, Write};
5use std::os::unix::process::ExitStatusExt;
6use std::path::Path;
7use std::process::Command;
8
9use crate::{Opt, Report, Status};
10
11pub fn exec(opt: &Opt, fp: &Path, status: &mut Status) -> Result<(), Error> {
12    let mut mux = Mux::new()?;
13    let mut report = Report::new(opt, fp);
14    let mut child = Command::new(fp.to_str().unwrap())
15        .args(&opt.arg)
16        .stdout(mux.make_untagged_sender()?)
17        .stderr(mux.make_tagged_sender("e")?)
18        .spawn()?;
19    let mut done_sender = mux.make_tagged_sender("d")?;
20    std::thread::spawn(move || match child.wait() {
21        Ok(status) => {
22            let exit_code = if let Some(code) = status.code() {
23                code as u8
24            } else {
25                status.signal().unwrap() as u8 + 128
26            };
27            let _ = done_sender.write_all(&[exit_code]);
28        }
29        Err(e) => {
30            let _ = writeln!(done_sender, "Error: {:?}", e);
31        }
32    });
33
34    let stdout = io::stdout();
35    let mut stdout = stdout.lock();
36
37    let stderr = io::stderr();
38    let mut stderr = stderr.lock();
39
40    loop {
41        let TaggedData { tag, data } = mux.read()?;
42        match (tag.as_deref(), data) {
43            (Some("d"), &[exit_code]) => {
44                status.exit_code = exit_code as i32;
45                break;
46            }
47            (Some("d"), error) => {
48                std::io::stderr().write_all(error)?;
49                status.exit_code = exitcode::SOFTWARE;
50                break;
51            }
52            (None, _) => write(&mut stdout, data, report.out_report()),
53            (_, _) => write(&mut stderr, data, report.err_report()),
54        }
55    }
56    Ok(())
57}
58
59fn write(w: &mut dyn Write, data: &[u8], report: Option<&String>) {
60    if let Some(report) = report {
61        w.write_all(report.as_bytes()).unwrap();
62        w.write_all(b":\n").unwrap();
63    }
64    w.write_all(data).unwrap();
65}
run_parts/exec.rs

run_parts/
exec.rs
