extern crate rustc_driver; extern crate rustc_hir; extern crate rustc_interface; use rustc_driver::Compilation; use rustc_hir::def_id::LOCAL_CRATE; 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() }