60 lines
1.8 KiB
Rust
60 lines
1.8 KiB
Rust
extern crate rustc_driver;
|
|
extern crate rustc_hir;
|
|
extern crate rustc_interface;
|
|
|
|
use rustc_hir::def_id::LOCAL_CRATE;
|
|
use rustc_driver::Compilation;
|
|
use rustc_interface::{interface, Queries};
|
|
|
|
use crate::test::Bencher;
|
|
|
|
struct MiriCompilerCalls<'a> {
|
|
bencher: &'a mut Bencher,
|
|
}
|
|
|
|
impl rustc_driver::Callbacks for MiriCompilerCalls<'_> {
|
|
fn after_analysis<'tcx>(
|
|
&mut self,
|
|
compiler: &interface::Compiler,
|
|
queries: &'tcx Queries<'tcx>,
|
|
) -> Compilation {
|
|
compiler.session().abort_if_errors();
|
|
|
|
queries.global_ctxt().unwrap().peek_mut().enter(|tcx| {
|
|
let (entry_def_id, _) =
|
|
tcx.entry_fn(LOCAL_CRATE).expect("no main or start function found");
|
|
|
|
self.bencher.iter(|| {
|
|
let config = miri::MiriConfig::default();
|
|
miri::eval_main(tcx, entry_def_id, config);
|
|
});
|
|
});
|
|
|
|
compiler.session().abort_if_errors();
|
|
|
|
Compilation::Stop
|
|
}
|
|
}
|
|
|
|
fn find_sysroot() -> String {
|
|
// Taken from https://github.com/Manishearth/rust-clippy/pull/911.
|
|
let home = option_env!("RUSTUP_HOME").or(option_env!("MULTIRUST_HOME"));
|
|
let toolchain = option_env!("RUSTUP_TOOLCHAIN").or(option_env!("MULTIRUST_TOOLCHAIN"));
|
|
match (home, toolchain) {
|
|
(Some(home), Some(toolchain)) => format!("{}/toolchains/{}", home, toolchain),
|
|
_ => option_env!("RUST_SYSROOT")
|
|
.expect("need to specify RUST_SYSROOT env var or use rustup or multirust")
|
|
.to_owned(),
|
|
}
|
|
}
|
|
|
|
pub fn run(filename: &str, bencher: &mut Bencher) {
|
|
let args = &[
|
|
"miri".to_string(),
|
|
format!("benches/helpers/{}.rs", filename),
|
|
"--sysroot".to_string(),
|
|
find_sysroot(),
|
|
];
|
|
rustc_driver::RunCompiler::new(args, &mut MiriCompilerCalls { bencher }).run().unwrap()
|
|
}
|