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
117
118
119
120
|
use serde_derive::{ Serialize, Deserialize };
use crate::backends::{ Backend, mk_tmp_dir, RunStatus };
use std::path::{ Path, PathBuf };
use std::io::{ Result, Error, ErrorKind };
use std::process::{ Command };
use std::collections::hash_map::DefaultHasher;
use std::hash::{ Hash, Hasher };
use nix::{ sys::wait, unistd::Pid };
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct ClangBackend {
template: Option<String>,
#[serde(default)]
args: Vec<String>,
#[serde(default = "default_cc")]
cc: String,
}
fn default_cc() -> String {
"clang++".to_string()
}
fn get_binary_by_filename(fname: &Path) -> Result<PathBuf> {
let hashed_fname = {
let mut hasher = DefaultHasher::new();
fname.hash(&mut hasher);
format!("{:x}", hasher.finish())
};
Ok(mk_tmp_dir()?.join(hashed_fname))
}
impl ClangBackend {
fn build(&self, fname: &Path) -> Result<PathBuf> {
let binary_fname = get_binary_by_filename(fname)?;
let get_mtime = |file| {
std::fs::metadata(file)?
.modified()
};
let src_mod = get_mtime(fname);
let binary_mod = get_mtime(&binary_fname);
if src_mod.is_err() || binary_mod.is_err() || src_mod.unwrap() > binary_mod.unwrap() {
let clang_status = Command::new(&self.cc)
.arg("-x").arg("c++")
.arg(fname.as_os_str())
.arg("-o").arg(&binary_fname)
.args(&self.args)
.status()?;
if !clang_status.success() {
return Err(Error::new(ErrorKind::Other,
"could not compile"));
}
}
Ok(binary_fname)
}
}
impl Backend for ClangBackend {
fn get_template(&self) -> Option<&str> {
match self.template {
Some(ref t) => Some(t),
None => None
}
}
fn run(&self, fname: &Path) -> Result<RunStatus> {
let binary_fname = self.build(fname)?;
let binary_proc = Command::new(&binary_fname)
.spawn()?;
get_status(binary_proc)
}
}
#[cfg(unix)]
fn get_status(proc: std::process::Child) -> Result<RunStatus> {
let pid = Pid::from_raw(proc.id() as i32);
loop {
let status_result = wait::waitpid(Some(pid), None)
.map_err(|err| Error::new(ErrorKind::Other, err));
let status = status_result?;
match status {
wait::WaitStatus::Exited(ret_pid, code) => {
assert_eq!(ret_pid, pid);
if code == 0 {
return Ok(RunStatus::Success);
} else {
return Ok(RunStatus::ErrorCode(code));
}
},
wait::WaitStatus::Signaled(ret_pid, sig, coredump) => {
assert_eq!(ret_pid, pid);
return Ok(RunStatus::Signal(sig, coredump));
},
_ => continue,
}
}
}
#[cfg(not(unix))]
fn get_status(proc: std::process::Child) -> Result<RunStatus> {
compile_error!("currently only unix supported");
}
|
