benches/helpers/miri_helper.rs

   1 extern crate getopts;
   2 extern crate miri;
   3 extern crate rustc;
   4 extern crate rustc_driver;
   5 extern crate test;
   6
   7 use self::miri::{eval_main, run_mir_passes};
   8 use self::rustc::session::Session;
   9 use self::rustc_driver::{driver, CompilerCalls, Compilation};
  10 use std::cell::RefCell;
  11 use std::rc::Rc;
  12 use test::Bencher;
  13
  14 pub struct MiriCompilerCalls<'a>(Rc<RefCell<&'a mut Bencher>>);
  15
  16 fn find_sysroot() -> String {
  17     // Taken from https://github.com/Manishearth/rust-clippy/pull/911.
  18     let home = option_env!("RUSTUP_HOME").or(option_env!("MULTIRUST_HOME"));
  19     let toolchain = option_env!("RUSTUP_TOOLCHAIN").or(option_env!("MULTIRUST_TOOLCHAIN"));
  20     match (home, toolchain) {
  21         (Some(home), Some(toolchain)) => format!("{}/toolchains/{}", home, toolchain),
  22         _ => option_env!("RUST_SYSROOT")
  23             .expect("need to specify RUST_SYSROOT env var or use rustup or multirust")
  24             .to_owned(),
  25     }
  26 }
  27
  28 pub fn run(filename: &str, bencher: &mut Bencher) {
  29     let args = &[
  30         "miri".to_string(),
  31         format!("benches/helpers/{}.rs", filename),
  32         "--sysroot".to_string(),
  33         find_sysroot()
  34     ];
  35     let compiler_calls = &mut MiriCompilerCalls(Rc::new(RefCell::new(bencher)));
  36     rustc_driver::run_compiler(args, compiler_calls);
  37 }
  38
  39 impl<'a> CompilerCalls<'a> for MiriCompilerCalls<'a> {
  40     fn build_controller(
  41         &mut self,
  42         _: &Session,
  43         _: &getopts::Matches
  44     ) -> driver::CompileController<'a> {
  45         let mut control: driver::CompileController<'a> = driver::CompileController::basic();
  46
  47         let bencher = self.0.clone();
  48
  49         control.after_analysis.stop = Compilation::Stop;
  50         control.after_analysis.callback = Box::new(move |state| {
  51             state.session.abort_if_errors();
  52
  53             let tcx = state.tcx.unwrap();
  54             let mir_map = state.mir_map.unwrap();
  55             let (node_id, _) = state.session.entry_fn.borrow()
  56                 .expect("no main or start function found");
  57
  58             let mut mir_map = MirMap { map: mir_map.map.clone() };
  59             run_mir_passes(tcx, &mut mir_map);
  60             bencher.borrow_mut().iter(|| { eval_main(tcx, &mir_map, node_id); });
  61
  62             state.session.abort_if_errors();
  63         });
  64
  65         control
  66     }
  67 }