265 lines
8.6 KiB
Rust
265 lines
8.6 KiB
Rust
#![feature(rustc_private, i128_type)]
|
|
|
|
extern crate getopts;
|
|
extern crate miri;
|
|
extern crate rustc;
|
|
extern crate rustc_driver;
|
|
extern crate rustc_errors;
|
|
extern crate env_logger;
|
|
extern crate log_settings;
|
|
extern crate syntax;
|
|
extern crate log;
|
|
|
|
use rustc::session::Session;
|
|
use rustc::middle::cstore::CrateStore;
|
|
use rustc_driver::{Compilation, CompilerCalls, RustcDefaultCalls};
|
|
use rustc_driver::driver::{CompileState, CompileController};
|
|
use rustc::session::config::{self, Input, ErrorOutputType};
|
|
use rustc::hir::{self, itemlikevisit};
|
|
use rustc::ty::TyCtxt;
|
|
use syntax::ast::{self, MetaItemKind, NestedMetaItemKind};
|
|
use std::path::PathBuf;
|
|
|
|
struct MiriCompilerCalls {
|
|
default: RustcDefaultCalls,
|
|
}
|
|
|
|
impl<'a> CompilerCalls<'a> for MiriCompilerCalls {
|
|
fn early_callback(
|
|
&mut self,
|
|
matches: &getopts::Matches,
|
|
sopts: &config::Options,
|
|
cfg: &ast::CrateConfig,
|
|
descriptions: &rustc_errors::registry::Registry,
|
|
output: ErrorOutputType,
|
|
) -> Compilation {
|
|
self.default.early_callback(
|
|
matches,
|
|
sopts,
|
|
cfg,
|
|
descriptions,
|
|
output,
|
|
)
|
|
}
|
|
fn no_input(
|
|
&mut self,
|
|
matches: &getopts::Matches,
|
|
sopts: &config::Options,
|
|
cfg: &ast::CrateConfig,
|
|
odir: &Option<PathBuf>,
|
|
ofile: &Option<PathBuf>,
|
|
descriptions: &rustc_errors::registry::Registry,
|
|
) -> Option<(Input, Option<PathBuf>)> {
|
|
self.default.no_input(
|
|
matches,
|
|
sopts,
|
|
cfg,
|
|
odir,
|
|
ofile,
|
|
descriptions,
|
|
)
|
|
}
|
|
fn late_callback(
|
|
&mut self,
|
|
matches: &getopts::Matches,
|
|
sess: &Session,
|
|
cstore: &CrateStore,
|
|
input: &Input,
|
|
odir: &Option<PathBuf>,
|
|
ofile: &Option<PathBuf>,
|
|
) -> Compilation {
|
|
self.default.late_callback(matches, sess, cstore, input, odir, ofile)
|
|
}
|
|
fn build_controller(
|
|
&mut self,
|
|
sess: &Session,
|
|
matches: &getopts::Matches,
|
|
) -> CompileController<'a> {
|
|
let mut control = self.default.build_controller(sess, matches);
|
|
control.after_hir_lowering.callback = Box::new(after_hir_lowering);
|
|
control.after_analysis.callback = Box::new(after_analysis);
|
|
if sess.target.target != sess.host {
|
|
// only fully compile targets on the host. linking will fail for cross-compilation.
|
|
control.after_analysis.stop = Compilation::Stop;
|
|
}
|
|
control
|
|
}
|
|
}
|
|
|
|
fn after_hir_lowering(state: &mut CompileState) {
|
|
let attr = (
|
|
String::from("miri"),
|
|
syntax::feature_gate::AttributeType::Whitelisted,
|
|
);
|
|
state.session.plugin_attributes.borrow_mut().push(attr);
|
|
}
|
|
|
|
fn after_analysis<'a, 'tcx>(state: &mut CompileState<'a, 'tcx>) {
|
|
state.session.abort_if_errors();
|
|
|
|
let tcx = state.tcx.unwrap();
|
|
let limits = resource_limits_from_attributes(state);
|
|
|
|
if std::env::args().any(|arg| arg == "--test") {
|
|
struct Visitor<'a, 'tcx: 'a>(
|
|
miri::ResourceLimits,
|
|
TyCtxt<'a, 'tcx, 'tcx>,
|
|
&'a CompileState<'a, 'tcx>
|
|
);
|
|
impl<'a, 'tcx: 'a, 'hir> itemlikevisit::ItemLikeVisitor<'hir> for Visitor<'a, 'tcx> {
|
|
fn visit_item(&mut self, i: &'hir hir::Item) {
|
|
if let hir::Item_::ItemFn(_, _, _, _, _, body_id) = i.node {
|
|
if i.attrs.iter().any(|attr| {
|
|
attr.name().map_or(false, |n| n == "test")
|
|
})
|
|
{
|
|
let did = self.1.hir.body_owner_def_id(body_id);
|
|
println!(
|
|
"running test: {}",
|
|
self.1.def_path_debug_str(did),
|
|
);
|
|
miri::eval_main(self.1, did, None, self.0);
|
|
self.2.session.abort_if_errors();
|
|
}
|
|
}
|
|
}
|
|
fn visit_trait_item(&mut self, _trait_item: &'hir hir::TraitItem) {}
|
|
fn visit_impl_item(&mut self, _impl_item: &'hir hir::ImplItem) {}
|
|
}
|
|
state.hir_crate.unwrap().visit_all_item_likes(
|
|
&mut Visitor(limits, tcx, state),
|
|
);
|
|
} else if let Some((entry_node_id, _)) = *state.session.entry_fn.borrow() {
|
|
let entry_def_id = tcx.hir.local_def_id(entry_node_id);
|
|
let start_wrapper = tcx.lang_items().start_fn().and_then(|start_fn| {
|
|
if tcx.is_mir_available(start_fn) {
|
|
Some(start_fn)
|
|
} else {
|
|
None
|
|
}
|
|
});
|
|
miri::eval_main(tcx, entry_def_id, start_wrapper, limits);
|
|
|
|
state.session.abort_if_errors();
|
|
} else {
|
|
println!("no main function found, assuming auxiliary build");
|
|
}
|
|
}
|
|
|
|
fn resource_limits_from_attributes(state: &CompileState) -> miri::ResourceLimits {
|
|
let mut limits = miri::ResourceLimits::default();
|
|
let krate = state.hir_crate.as_ref().unwrap();
|
|
let err_msg = "miri attributes need to be in the form `miri(key = value)`";
|
|
let extract_int = |lit: &syntax::ast::Lit| -> u128 {
|
|
match lit.node {
|
|
syntax::ast::LitKind::Int(i, _) => i,
|
|
_ => {
|
|
state.session.span_fatal(
|
|
lit.span,
|
|
"expected an integer literal",
|
|
)
|
|
}
|
|
}
|
|
};
|
|
|
|
for attr in krate.attrs.iter().filter(|a| {
|
|
a.name().map_or(false, |n| n == "miri")
|
|
})
|
|
{
|
|
if let Some(items) = attr.meta_item_list() {
|
|
for item in items {
|
|
if let NestedMetaItemKind::MetaItem(ref inner) = item.node {
|
|
if let MetaItemKind::NameValue(ref value) = inner.node {
|
|
match &inner.name().as_str()[..] {
|
|
"memory_size" => limits.memory_size = extract_int(value) as u64,
|
|
"step_limit" => limits.step_limit = extract_int(value) as u64,
|
|
"stack_limit" => limits.stack_limit = extract_int(value) as usize,
|
|
_ => state.session.span_err(item.span, "unknown miri attribute"),
|
|
}
|
|
} else {
|
|
state.session.span_err(inner.span, err_msg);
|
|
}
|
|
} else {
|
|
state.session.span_err(item.span, err_msg);
|
|
}
|
|
}
|
|
} else {
|
|
state.session.span_err(attr.span, err_msg);
|
|
}
|
|
}
|
|
limits
|
|
}
|
|
|
|
fn init_logger() {
|
|
let format = |record: &log::LogRecord| {
|
|
if record.level() == log::LogLevel::Trace {
|
|
// prepend frame number
|
|
let indentation = log_settings::settings().indentation;
|
|
format!(
|
|
"{indentation}:{lvl}:{module}: {text}",
|
|
lvl = record.level(),
|
|
module = record.location().module_path(),
|
|
indentation = indentation,
|
|
text = record.args(),
|
|
)
|
|
} else {
|
|
format!(
|
|
"{lvl}:{module}: {text}",
|
|
lvl = record.level(),
|
|
module = record.location().module_path(),
|
|
text = record.args(),
|
|
)
|
|
}
|
|
};
|
|
|
|
let mut builder = env_logger::LogBuilder::new();
|
|
builder.format(format).filter(
|
|
None,
|
|
log::LogLevelFilter::Info,
|
|
);
|
|
|
|
if std::env::var("MIRI_LOG").is_ok() {
|
|
builder.parse(&std::env::var("MIRI_LOG").unwrap());
|
|
}
|
|
|
|
builder.init().unwrap();
|
|
}
|
|
|
|
fn find_sysroot() -> String {
|
|
if let Ok(sysroot) = std::env::var("MIRI_SYSROOT") {
|
|
return sysroot;
|
|
}
|
|
|
|
// 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()
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
init_logger();
|
|
let mut args: Vec<String> = std::env::args().collect();
|
|
|
|
let sysroot_flag = String::from("--sysroot");
|
|
if !args.contains(&sysroot_flag) {
|
|
args.push(sysroot_flag);
|
|
args.push(find_sysroot());
|
|
}
|
|
|
|
// Make sure we always have all the MIR (e.g. for auxilary builds in unit tests).
|
|
args.push("-Zalways-encode-mir".to_owned());
|
|
|
|
rustc_driver::run_compiler(&args, &mut MiriCompilerCalls {
|
|
default: RustcDefaultCalls,
|
|
}, None, None);
|
|
}
|