Replace version check with probing Backtrace api

This commit is contained in:
David Tolnay 2019-10-18 10:43:43 -04:00
parent d7c37d6c4d
commit 5ada2978c5
No known key found for this signature in database
GPG Key ID: F9BA143B95FF6D82

View File

@ -1,35 +1,59 @@
use std::env; use std::env;
use std::process::Command; use std::fs;
use std::str::{self, FromStr}; use std::path::Path;
use std::process::{Command, ExitStatus};
// This code exercises the surface area that we expect of the std Backtrace
// type. If the current toolchain is able to compile it, we go ahead and use
// backtrace in anyhow.
const PROBE: &str = r#"
#![feature(backtrace)]
#![allow(dead_code)]
use std::backtrace::{Backtrace, BacktraceStatus};
use std::error::Error;
use std::fmt::{self, Display};
#[derive(Debug)]
struct E;
impl Display for E {
fn fmt(&self, _formatter: &mut fmt::Formatter) -> fmt::Result {
unimplemented!()
}
}
impl Error for E {
fn backtrace(&self) -> Option<&Backtrace> {
let backtrace = Backtrace::capture();
match backtrace.status() {
BacktraceStatus::Captured | BacktraceStatus::Disabled | _ => {}
}
unimplemented!()
}
}
"#;
fn main() { fn main() {
let compiler = match rustc_version() { match compile_probe() {
Some(compiler) => compiler, Some(status) if status.success() => println!("cargo:rustc-cfg=backtrace"),
None => return, _ => {}
};
if compiler.minor >= 40 && compiler.nightly {
println!("cargo:rustc-cfg=backtrace");
} }
} }
struct Compiler { fn compile_probe() -> Option<ExitStatus> {
minor: u32,
nightly: bool,
}
fn rustc_version() -> Option<Compiler> {
let rustc = env::var_os("RUSTC")?; let rustc = env::var_os("RUSTC")?;
let output = Command::new(rustc).arg("--version").output().ok()?; let out_dir = env::var_os("OUT_DIR")?;
let version = str::from_utf8(&output.stdout).ok()?; let probefile = Path::new(&out_dir).join("lib.rs");
fs::write(&probefile, PROBE).ok()?;
let mut pieces = version.split('.'); Command::new(rustc)
if pieces.next() != Some("rustc 1") { .arg("--edition=2018")
return None; .arg("--crate-name=anyhow_build")
} .arg("--crate-type=lib")
.arg("--emit=metadata")
let next = pieces.next()?; .arg("--out-dir")
let minor = u32::from_str(next).ok()?; .arg(out_dir)
let nightly = version.contains("nightly") || version.contains("dev"); .arg(probefile)
Some(Compiler { minor, nightly }) .status()
.ok()
} }