| //! Code for building the standard library. |
| |
| use crate::compiler::UnitInterner; |
| use crate::compiler::unit_dependencies::IsArtifact; |
| use crate::compiler::{CompileKind, CompileMode, RustcTargetData, Unit}; |
| use crate::ops::{self, Packages}; |
| use crate::resolver::HasDevUnits; |
| use crate::resolver::Resolve; |
| use crate::resolver::features::{CliFeatures, FeaturesFor, ResolvedFeatures}; |
| use crate::util::errors::CargoResult; |
| use crate::workspace::profiles::{Profiles, UnitFor}; |
| use crate::workspace::{PackageId, PackageSet, Workspace}; |
| |
| use crate::util::data_structures::{HashMap, HashSet}; |
| use std::path::PathBuf; |
| |
| use super::BuildConfig; |
| |
| fn std_crates<'a>(crates: &'a [String], default: &'static str, units: &[Unit]) -> HashSet<&'a str> { |
| let mut crates = HashSet::from_iter(crates.iter().map(|s| s.as_str())); |
| // This is a temporary hack until there is a more principled way to |
| // declare dependencies in Cargo.toml. |
| if crates.is_empty() { |
| crates.insert(default); |
| } |
| if crates.contains("std") { |
| crates.insert("core"); |
| crates.insert("alloc"); |
| crates.insert("proc_macro"); |
| crates.insert("panic_unwind"); |
| crates.insert("compiler_builtins"); |
| // Only build libtest if it looks like it is needed (libtest depends on libstd) |
| // If we know what units we're building, we can filter for libtest depending on the jobs. |
| if units |
| .iter() |
| .any(|unit| unit.mode.is_rustc_test() && unit.target.harness()) |
| { |
| crates.insert("test"); |
| } |
| } else if crates.contains("core") { |
| crates.insert("compiler_builtins"); |
| } |
| |
| crates |
| } |
| |
| /// Resolve the standard library dependencies. |
| /// |
| /// * `crates` is the arg value from `-Zbuild-std`. |
| pub fn resolve_std<'gctx>( |
| ws: &Workspace<'gctx>, |
| target_data: &mut RustcTargetData<'gctx>, |
| build_config: &BuildConfig, |
| crates: &[String], |
| kinds: &[CompileKind], |
| ) -> CargoResult<(PackageSet<'gctx>, Resolve, ResolvedFeatures)> { |
| let src_path = detect_sysroot_src_path(target_data)?; |
| let std_ws_manifest_path = src_path.join("Cargo.toml"); |
| let gctx = ws.gctx(); |
| // TODO: Consider doing something to enforce --locked? Or to prevent the |
| // lock file from being written, such as setting ephemeral. |
| let mut std_ws = Workspace::new(&std_ws_manifest_path, gctx)?; |
| // Don't require optional dependencies in this workspace, aka std's own |
| // `[dev-dependencies]`. No need for us to generate a `Resolve` which has |
| // those included because we'll never use them anyway. |
| std_ws.set_require_optional_deps(false); |
| let specs = { |
| // If there is anything looks like needing std, resolve with it. |
| // If not, we assume only `core` maye be needed, as `core the most fundamental crate. |
| // |
| // This may need a UI overhaul if `build-std` wants to fully support multi-targets. |
| let maybe_std = kinds |
| .iter() |
| .any(|kind| target_data.info(*kind).maybe_support_std()); |
| let mut crates = std_crates(crates, if maybe_std { "std" } else { "core" }, &[]); |
| // `sysroot` is not in the default set because it is optional, but it needs |
| // to be part of the resolve in case we do need it or `libtest`. |
| crates.insert("sysroot"); |
| let specs = Packages::Packages(crates.into_iter().map(Into::into).collect()); |
| specs.to_package_id_specs(&std_ws)? |
| }; |
| let features = match &gctx.cli_unstable().build_std_features { |
| Some(list) => list.clone(), |
| None => vec![ |
| "panic-unwind".to_string(), |
| "backtrace".to_string(), |
| "default".to_string(), |
| ], |
| }; |
| let cli_features = CliFeatures::from_command_line( |
| &features, /*all_features*/ false, /*uses_default_features*/ false, |
| )?; |
| let dry_run = false; |
| let mut resolve = ops::resolve_ws_with_opts( |
| &std_ws, |
| target_data, |
| &build_config.requested_kinds, |
| &cli_features, |
| &specs, |
| HasDevUnits::No, |
| crate::resolver::features::ForceAllTargets::No, |
| dry_run, |
| )?; |
| debug_assert_eq!(resolve.specs_and_features.len(), 1); |
| Ok(( |
| resolve.pkg_set, |
| resolve.targeted_resolve, |
| resolve |
| .specs_and_features |
| .pop() |
| .expect("resolve should have a single spec with resolved features") |
| .resolved_features, |
| )) |
| } |
| |
| /// Generates a map of root units for the standard library for each kind requested. |
| /// |
| /// * `crates` is the arg value from `-Zbuild-std`. |
| /// * `units` is the root units of the build. |
| pub fn generate_std_roots( |
| crates: &[String], |
| units: &[Unit], |
| std_resolve: &Resolve, |
| std_features: &ResolvedFeatures, |
| kinds: &[CompileKind], |
| package_set: &PackageSet<'_>, |
| interner: &UnitInterner, |
| profiles: &Profiles, |
| target_data: &RustcTargetData<'_>, |
| ) -> CargoResult<HashMap<CompileKind, Vec<Unit>>> { |
| // Generate a map of Units for each kind requested. |
| let mut ret = HashMap::default(); |
| let (maybe_std, maybe_core): (Vec<&CompileKind>, Vec<_>) = kinds |
| .iter() |
| .partition(|kind| target_data.info(**kind).maybe_support_std()); |
| for (default_crate, kinds) in [("core", maybe_core), ("std", maybe_std)] { |
| if kinds.is_empty() { |
| continue; |
| } |
| generate_roots( |
| &mut ret, |
| default_crate, |
| crates, |
| units, |
| std_resolve, |
| std_features, |
| &kinds, |
| package_set, |
| interner, |
| profiles, |
| target_data, |
| )?; |
| } |
| |
| Ok(ret) |
| } |
| |
| fn generate_roots( |
| ret: &mut HashMap<CompileKind, Vec<Unit>>, |
| default: &'static str, |
| crates: &[String], |
| units: &[Unit], |
| std_resolve: &Resolve, |
| std_features: &ResolvedFeatures, |
| kinds: &[&CompileKind], |
| package_set: &PackageSet<'_>, |
| interner: &UnitInterner, |
| profiles: &Profiles, |
| target_data: &RustcTargetData<'_>, |
| ) -> CargoResult<()> { |
| let std_ids = std_crates(crates, default, units) |
| .iter() |
| .map(|crate_name| std_resolve.query(crate_name)) |
| .collect::<CargoResult<Vec<PackageId>>>()?; |
| let std_pkgs = package_set.get_many(std_ids)?; |
| |
| for pkg in std_pkgs { |
| let lib = pkg |
| .targets() |
| .iter() |
| .find(|t| t.is_lib()) |
| .expect("std has a lib"); |
| // I don't think we need to bother with Check here, the difference |
| // in time is minimal, and the difference in caching is |
| // significant. |
| let mode = CompileMode::Build; |
| let features = std_features.activated_features(pkg.package_id(), FeaturesFor::NormalOrDev); |
| for kind in kinds { |
| let kind = **kind; |
| let list = ret.entry(kind).or_insert_with(Vec::new); |
| let unit_for = UnitFor::new_normal(kind); |
| let profile = profiles.get_profile( |
| pkg.package_id(), |
| /*is_member*/ false, |
| /*is_local*/ false, |
| unit_for, |
| kind, |
| ); |
| list.push(interner.intern( |
| pkg, |
| lib, |
| profile, |
| kind, |
| mode, |
| features.clone(), |
| target_data.info(kind).rustflags.clone(), |
| target_data.info(kind).rustdocflags.clone(), |
| target_data.target_config(kind).links_overrides.clone(), |
| /*is_std*/ true, |
| /*dep_hash*/ 0, |
| IsArtifact::No, |
| None, |
| false, |
| )); |
| } |
| } |
| Ok(()) |
| } |
| |
| fn detect_sysroot_src_path(target_data: &RustcTargetData<'_>) -> CargoResult<PathBuf> { |
| if let Some(s) = target_data.gctx.get_env_os("__CARGO_TESTS_ONLY_SRC_ROOT") { |
| return Ok(s.into()); |
| } |
| |
| // NOTE: This is temporary until we figure out how to acquire the source. |
| let src_path = target_data |
| .info(CompileKind::Host) |
| .sysroot |
| .join("lib") |
| .join("rustlib") |
| .join("src") |
| .join("rust") |
| .join("library"); |
| let lock = src_path.join("Cargo.lock"); |
| if !lock.exists() { |
| let msg = format!( |
| "{:?} does not exist, unable to build with the standard \ |
| library, try:\n rustup component add rust-src", |
| lock |
| ); |
| match target_data.gctx.get_env("RUSTUP_TOOLCHAIN") { |
| Ok(rustup_toolchain) => { |
| anyhow::bail!("{} --toolchain {}", msg, rustup_toolchain); |
| } |
| Err(_) => { |
| anyhow::bail!(msg); |
| } |
| } |
| } |
| Ok(src_path) |
| } |