| #![allow(unknown_lints)] |
| #![allow(unexpected_cfgs)] |
| |
| use std::env; |
| use std::ffi::OsString; |
| use std::iter; |
| use std::path::Path; |
| use std::process::{self, Command, Stdio}; |
| use std::str; |
| |
| fn main() { |
| let rustc = rustc_minor_version().unwrap_or(u32::MAX); |
| |
| if rustc >= 80 { |
| println!("cargo:rustc-check-cfg=cfg(fuzzing)"); |
| println!("cargo:rustc-check-cfg=cfg(no_is_available)"); |
| println!("cargo:rustc-check-cfg=cfg(no_literal_byte_character)"); |
| println!("cargo:rustc-check-cfg=cfg(no_literal_c_string)"); |
| println!("cargo:rustc-check-cfg=cfg(no_source_text)"); |
| println!("cargo:rustc-check-cfg=cfg(proc_macro_span)"); |
| println!("cargo:rustc-check-cfg=cfg(procmacro2_backtrace)"); |
| println!("cargo:rustc-check-cfg=cfg(procmacro2_nightly_testing)"); |
| println!("cargo:rustc-check-cfg=cfg(procmacro2_semver_exempt)"); |
| println!("cargo:rustc-check-cfg=cfg(randomize_layout)"); |
| println!("cargo:rustc-check-cfg=cfg(span_locations)"); |
| println!("cargo:rustc-check-cfg=cfg(super_unstable)"); |
| println!("cargo:rustc-check-cfg=cfg(wrap_proc_macro)"); |
| } |
| |
| let docs_rs = env::var_os("DOCS_RS").is_some(); |
| let semver_exempt = cfg!(procmacro2_semver_exempt) || docs_rs; |
| if semver_exempt { |
| // https://github.com/dtolnay/proc-macro2/issues/147 |
| println!("cargo:rustc-cfg=procmacro2_semver_exempt"); |
| } |
| |
| if semver_exempt || cfg!(feature = "span-locations") { |
| // Provide methods Span::start and Span::end which give the line/column |
| // location of a token. This is behind a cfg because tracking location |
| // inside spans is a performance hit. |
| println!("cargo:rustc-cfg=span_locations"); |
| } |
| |
| if rustc < 57 { |
| // Do not use proc_macro::is_available() to detect whether the proc |
| // macro API is available vs needs to be polyfilled. Instead, use the |
| // proc macro API unconditionally and catch the panic that occurs if it |
| // isn't available. |
| println!("cargo:rustc-cfg=no_is_available"); |
| } |
| |
| if rustc < 66 { |
| // Do not call libproc_macro's Span::source_text. Always return None. |
| println!("cargo:rustc-cfg=no_source_text"); |
| } |
| |
| if rustc < 79 { |
| // Do not call Literal::byte_character nor Literal::c_string. They can |
| // be emulated by way of Literal::from_str. |
| println!("cargo:rustc-cfg=no_literal_byte_character"); |
| println!("cargo:rustc-cfg=no_literal_c_string"); |
| } |
| |
| if !cfg!(feature = "proc-macro") { |
| println!("cargo:rerun-if-changed=build.rs"); |
| return; |
| } |
| |
| println!("cargo:rerun-if-changed=build/probe.rs"); |
| |
| let proc_macro_span; |
| let consider_rustc_bootstrap; |
| if compile_probe(false) { |
| // This is a nightly or dev compiler, so it supports unstable features |
| // regardless of RUSTC_BOOTSTRAP. No need to rerun build script if |
| // RUSTC_BOOTSTRAP is changed. |
| proc_macro_span = true; |
| consider_rustc_bootstrap = false; |
| } else if let Some(rustc_bootstrap) = env::var_os("RUSTC_BOOTSTRAP") { |
| if compile_probe(true) { |
| // This is a stable or beta compiler for which the user has set |
| // RUSTC_BOOTSTRAP to turn on unstable features. Rerun build script |
| // if they change it. |
| proc_macro_span = true; |
| consider_rustc_bootstrap = true; |
| } else if rustc_bootstrap == "1" { |
| // This compiler does not support the proc macro Span API in the |
| // form that proc-macro2 expects. No need to pay attention to |
| // RUSTC_BOOTSTRAP. |
| proc_macro_span = false; |
| consider_rustc_bootstrap = false; |
| } else { |
| // This is a stable or beta compiler for which RUSTC_BOOTSTRAP is |
| // set to restrict the use of unstable features by this crate. |
| proc_macro_span = false; |
| consider_rustc_bootstrap = true; |
| } |
| } else { |
| // Without RUSTC_BOOTSTRAP, this compiler does not support the proc |
| // macro Span API in the form that proc-macro2 expects, but try again if |
| // the user turns on unstable features. |
| proc_macro_span = false; |
| consider_rustc_bootstrap = true; |
| } |
| |
| if proc_macro_span || !semver_exempt { |
| // Wrap types from libproc_macro rather than polyfilling the whole API. |
| // Enabled as long as procmacro2_semver_exempt is not set, because we |
| // can't emulate the unstable API without emulating everything else. |
| // Also enabled unconditionally on nightly, in which case the |
| // procmacro2_semver_exempt surface area is implemented by using the |
| // nightly-only proc_macro API. |
| println!("cargo:rustc-cfg=wrap_proc_macro"); |
| } |
| |
| if proc_macro_span { |
| // Enable non-dummy behavior of Span::start and Span::end methods which |
| // requires an unstable compiler feature. Enabled when building with |
| // nightly, unless `-Z allow-feature` in RUSTFLAGS disallows unstable |
| // features. |
| println!("cargo:rustc-cfg=proc_macro_span"); |
| } |
| |
| if semver_exempt && proc_macro_span { |
| // Implement the semver exempt API in terms of the nightly-only |
| // proc_macro API. |
| println!("cargo:rustc-cfg=super_unstable"); |
| } |
| |
| if consider_rustc_bootstrap { |
| println!("cargo:rerun-if-env-changed=RUSTC_BOOTSTRAP"); |
| } |
| } |
| |
| fn compile_probe(rustc_bootstrap: bool) -> bool { |
| if env::var_os("RUSTC_STAGE").is_some() { |
| // We are running inside rustc bootstrap. This is a highly non-standard |
| // environment with issues such as: |
| // |
| // https://github.com/rust-lang/cargo/issues/11138 |
| // https://github.com/rust-lang/rust/issues/114839 |
| // |
| // Let's just not use nightly features here. |
| return false; |
| } |
| |
| let rustc = cargo_env_var("RUSTC"); |
| let out_dir = cargo_env_var("OUT_DIR"); |
| let probefile = Path::new("build").join("probe.rs"); |
| |
| let rustc_wrapper = env::var_os("RUSTC_WRAPPER").filter(|wrapper| !wrapper.is_empty()); |
| let rustc_workspace_wrapper = |
| env::var_os("RUSTC_WORKSPACE_WRAPPER").filter(|wrapper| !wrapper.is_empty()); |
| let mut rustc = rustc_wrapper |
| .into_iter() |
| .chain(rustc_workspace_wrapper) |
| .chain(iter::once(rustc)); |
| let mut cmd = Command::new(rustc.next().unwrap()); |
| cmd.args(rustc); |
| |
| if !rustc_bootstrap { |
| cmd.env_remove("RUSTC_BOOTSTRAP"); |
| } |
| |
| cmd.stderr(Stdio::null()) |
| .arg("--edition=2021") |
| .arg("--crate-name=proc_macro2") |
| .arg("--crate-type=lib") |
| .arg("--cap-lints=allow") |
| .arg("--emit=dep-info,metadata") |
| .arg("--out-dir") |
| .arg(out_dir) |
| .arg(probefile); |
| |
| if let Some(target) = env::var_os("TARGET") { |
| cmd.arg("--target").arg(target); |
| } |
| |
| // If Cargo wants to set RUSTFLAGS, use that. |
| if let Ok(rustflags) = env::var("CARGO_ENCODED_RUSTFLAGS") { |
| if !rustflags.is_empty() { |
| for arg in rustflags.split('\x1f') { |
| cmd.arg(arg); |
| } |
| } |
| } |
| |
| match cmd.status() { |
| Ok(status) => status.success(), |
| Err(_) => false, |
| } |
| } |
| |
| fn rustc_minor_version() -> Option<u32> { |
| let rustc = cargo_env_var("RUSTC"); |
| let output = Command::new(rustc).arg("--version").output().ok()?; |
| let version = str::from_utf8(&output.stdout).ok()?; |
| let mut pieces = version.split('.'); |
| if pieces.next() != Some("rustc 1") { |
| return None; |
| } |
| pieces.next()?.parse().ok() |
| } |
| |
| fn cargo_env_var(key: &str) -> OsString { |
| env::var_os(key).unwrap_or_else(|| { |
| eprintln!( |
| "Environment variable ${} is not set during execution of build script", |
| key, |
| ); |
| process::exit(1); |
| }) |
| } |