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
#![deny(missing_docs)]
use std::{borrow::Cow, io, ops::Deref, path::PathBuf, process::Command, string::FromUtf8Error};
#[derive(Debug)]
pub struct LlvmTools {
bin_dir: PathBuf,
}
impl LlvmTools {
pub fn new() -> Result<Self, Error> {
let sysroot = {
let mut rustc_cmd = Command::new("rustc");
rustc_cmd.arg("--print").arg("sysroot");
let rustc_cmd_string = format!("{:?}", rustc_cmd);
let out = rustc_cmd
.output()
.map_err(|e| Error::CommandExecution(rustc_cmd_string, e))?;
if !out.status.success() {
return Err(Error::RetrieveSysroot(String::from_utf8(out.stderr)));
}
let sysroot_string =
String::from_utf8(out.stdout).map_err(Error::SysrootInvalidUtf8)?;
PathBuf::from(sysroot_string.trim())
};
let example_tool_name = exe("llvm-objdump");
let rustlib = sysroot.join("lib").join("rustlib");
for entry in rustlib.read_dir().map_err(Error::ReadDirFailed)? {
let bin_dir = entry.map_err(Error::ReadEntryFailed)?.path().join("bin");
let tool_path = bin_dir.join(example_tool_name.deref());
if tool_path.exists() {
return Ok(Self { bin_dir });
}
}
Err(Error::NotFound)
}
pub fn tool(&self, tool_name: &str) -> Option<PathBuf> {
let tool_path = self.bin_dir.join(&tool_name);
if tool_path.exists() {
Some(tool_path)
} else {
None
}
}
}
#[derive(Debug)]
pub enum Error {
CommandExecution(String, io::Error),
RetrieveSysroot(Result<String, FromUtf8Error>),
SysrootInvalidUtf8(FromUtf8Error),
ReadDirFailed(io::Error),
ReadEntryFailed(io::Error),
NotFound,
}
pub fn exe(executable_name: &str) -> Cow<str> {
if cfg!(target_os = "windows") {
Cow::Owned(format!("{}.exe", executable_name))
} else {
Cow::Borrowed(executable_name)
}
}