Skip to main content

cargo/core/compiler/
custom_build.rs

1//! How to execute a build script and parse its output.
2//!
3//! ## Preparing a build script run
4//!
5//! A [build script] is an optional Rust script Cargo will run before building
6//! your package. As of this writing, two kinds of special [`Unit`]s will be
7//! constructed when there is a build script in a package.
8//!
9//! * Build script compilation --- This unit is generally the same as units
10//!   that would compile other Cargo targets. It will recursively creates units
11//!   of its dependencies. One biggest difference is that the [`Unit`] of
12//!   compiling a build script is flagged as [`TargetKind::CustomBuild`].
13//! * Build script execution --- During the construction of the [`UnitGraph`],
14//!   Cargo inserts a [`Unit`] with [`CompileMode::RunCustomBuild`]. This unit
15//!   depends on the unit of compiling the associated build script, to ensure
16//!   the executable is available before running. The [`Work`] of running the
17//!   build script is prepared in the function [`prepare`].
18//!
19//! ## Running a build script
20//!
21//! When running a build script, Cargo is aware of the progress and the result
22//! of a build script. Standard output is the chosen interprocess communication
23//! between Cargo and build script processes. A set of strings is defined for
24//! that purpose. These strings, a.k.a. instructions, are interpreted by
25//! [`BuildOutput::parse`] and stored in [`BuildRunner::build_script_outputs`].
26//! The entire execution work is constructed by [`build_work`].
27//!
28//! [build script]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html
29//! [`TargetKind::CustomBuild`]: crate::core::manifest::TargetKind::CustomBuild
30//! [`UnitGraph`]: super::unit_graph::UnitGraph
31//! [`CompileMode::RunCustomBuild`]: crate::core::compiler::CompileMode::RunCustomBuild
32//! [instructions]: https://doc.rust-lang.org/cargo/reference/build-scripts.html#outputs-of-the-build-script
33
34use super::{BuildRunner, Job, Unit, Work, fingerprint, get_dynamic_search_path};
35use crate::core::compiler::CompileMode;
36use crate::core::compiler::artifact;
37use crate::core::compiler::build_runner::UnitHash;
38use crate::core::compiler::job_queue::JobState;
39use crate::core::{PackageId, Target, profiles::ProfileRoot};
40use crate::util::errors::CargoResult;
41use crate::util::internal;
42use crate::util::machine_message::{self, Message};
43use anyhow::{Context as _, bail};
44use cargo_platform::Cfg;
45use cargo_util::paths;
46use cargo_util_schemas::manifest::RustVersion;
47use std::collections::hash_map::{Entry, HashMap};
48use std::collections::{BTreeSet, HashSet};
49use std::path::{Path, PathBuf};
50use std::str;
51use std::sync::{Arc, Mutex};
52
53/// A build script instruction that tells Cargo to display an error after the
54/// build script has finished running. Read [the doc] for more.
55///
56/// [the doc]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#cargo-error
57const CARGO_ERROR_SYNTAX: &str = "cargo::error=";
58/// Deprecated: A build script instruction that tells Cargo to display a warning after the
59/// build script has finished running. Read [the doc] for more.
60///
61/// [the doc]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#cargo-warning
62const OLD_CARGO_WARNING_SYNTAX: &str = "cargo:warning=";
63/// A build script instruction that tells Cargo to display a warning after the
64/// build script has finished running. Read [the doc] for more.
65///
66/// [the doc]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#cargo-warning
67const NEW_CARGO_WARNING_SYNTAX: &str = "cargo::warning=";
68
69#[derive(Clone, Debug, Hash, PartialEq, Eq, PartialOrd, Ord)]
70pub enum Severity {
71    Error,
72    Warning,
73}
74
75pub type LogMessage = (Severity, String);
76
77/// Represents a path added to the library search path.
78///
79/// We need to keep track of requests to add search paths within the cargo build directory
80/// separately from paths outside of Cargo. The reason is that we want to give precedence to linking
81/// against libraries within the Cargo build directory even if a similar library exists in the
82/// system (e.g. crate A adds `/usr/lib` to the search path and then a later build of crate B adds
83/// `target/debug/...` to satisfy its request to link against the library B that it built, but B is
84/// also found in `/usr/lib`).
85///
86/// There's some nuance here because we want to preserve relative order of paths of the same type.
87/// For example, if the build process would in declaration order emit the following linker line:
88/// ```bash
89/// -L/usr/lib -Ltarget/debug/build/crate1/libs -L/lib -Ltarget/debug/build/crate2/libs)
90/// ```
91///
92/// we want the linker to actually receive:
93/// ```bash
94/// -Ltarget/debug/build/crate1/libs -Ltarget/debug/build/crate2/libs) -L/usr/lib -L/lib
95/// ```
96///
97/// so that the library search paths within the crate artifacts directory come first but retain
98/// relative ordering while the system library paths come after while still retaining relative
99/// ordering among them; ordering is the order they are emitted within the build process,
100/// not lexicographic order.
101///
102/// WARNING: Even though this type implements PartialOrd + Ord, this is a lexicographic ordering.
103/// The linker line will require an explicit sorting algorithm. PartialOrd + Ord is derived because
104/// BuildOutput requires it but that ordering is different from the one for the linker search path,
105/// at least today. It may be worth reconsidering & perhaps it's ok if BuildOutput doesn't have
106/// a lexicographic ordering for the library_paths? I'm not sure the consequence of that.
107#[derive(Clone, Debug, Hash, PartialEq, Eq, PartialOrd, Ord)]
108pub enum LibraryPath {
109    /// The path is pointing within the output folder of the crate and takes priority over
110    /// external paths when passed to the linker.
111    CargoArtifact(PathBuf),
112    /// The path is pointing outside of the crate's build location. The linker will always
113    /// receive such paths after `CargoArtifact`.
114    External(PathBuf),
115}
116
117impl LibraryPath {
118    fn new(p: PathBuf, script_out_dir: &Path) -> Self {
119        let search_path = get_dynamic_search_path(&p);
120        if search_path.starts_with(script_out_dir) {
121            Self::CargoArtifact(p)
122        } else {
123            Self::External(p)
124        }
125    }
126
127    pub fn into_path_buf(self) -> PathBuf {
128        match self {
129            LibraryPath::CargoArtifact(p) | LibraryPath::External(p) => p,
130        }
131    }
132}
133
134impl AsRef<PathBuf> for LibraryPath {
135    fn as_ref(&self) -> &PathBuf {
136        match self {
137            LibraryPath::CargoArtifact(p) | LibraryPath::External(p) => p,
138        }
139    }
140}
141
142/// Contains the parsed output of a custom build script.
143#[derive(Clone, Debug, Hash, Default, PartialEq, Eq, PartialOrd, Ord)]
144pub struct BuildOutput {
145    /// Paths to pass to rustc with the `-L` flag.
146    pub library_paths: Vec<LibraryPath>,
147    /// Names and link kinds of libraries, suitable for the `-l` flag.
148    pub library_links: Vec<String>,
149    /// Linker arguments suitable to be passed to `-C link-arg=<args>`
150    pub linker_args: Vec<(LinkArgTarget, String)>,
151    /// Various `--cfg` flags to pass to the compiler.
152    pub cfgs: Vec<String>,
153    /// Various `--check-cfg` flags to pass to the compiler.
154    pub check_cfgs: Vec<String>,
155    /// Additional environment variables to run the compiler with.
156    pub env: Vec<(String, String)>,
157    /// Metadata to pass to the immediate dependencies.
158    pub metadata: Vec<(String, String)>,
159    /// Paths to trigger a rerun of this build script.
160    /// May be absolute or relative paths (relative to package root).
161    pub rerun_if_changed: Vec<PathBuf>,
162    /// Environment variables which, when changed, will cause a rebuild.
163    pub rerun_if_env_changed: Vec<String>,
164    /// Errors and warnings generated by this build.
165    ///
166    /// These are only displayed if this is a "local" package, `-vv` is used, or
167    /// there is a build error for any target in this package. Note that any log
168    /// message of severity `Error` will by itself cause a build error, and will
169    /// cause all log messages to be displayed.
170    pub log_messages: Vec<LogMessage>,
171}
172
173/// Map of packages to build script output.
174///
175/// This initially starts out as empty. Overridden build scripts get
176/// inserted during `build_map`. The rest of the entries are added
177/// immediately after each build script runs.
178///
179/// The [`UnitHash`] is the unique metadata hash for the `RunCustomBuild` Unit of
180/// the package. It needs a unique key, since the build script can be run
181/// multiple times with different profiles or features. We can't embed a
182/// `Unit` because this structure needs to be shareable between threads.
183#[derive(Default)]
184pub struct BuildScriptOutputs {
185    outputs: HashMap<UnitHash, BuildOutput>,
186}
187
188/// Linking information for a `Unit`.
189///
190/// See [`build_map`] for more details.
191#[derive(Default)]
192pub struct BuildScripts {
193    /// List of build script outputs this Unit needs to include for linking. Each
194    /// element is an index into `BuildScriptOutputs`.
195    ///
196    /// Cargo will use this `to_link` vector to add `-L` flags to compiles as we
197    /// propagate them upwards towards the final build. Note, however, that we
198    /// need to preserve the ordering of `to_link` to be topologically sorted.
199    /// This will ensure that build scripts which print their paths properly will
200    /// correctly pick up the files they generated (if there are duplicates
201    /// elsewhere).
202    ///
203    /// To preserve this ordering, the (id, metadata) is stored in two places, once
204    /// in the `Vec` and once in `seen_to_link` for a fast lookup. We maintain
205    /// this as we're building interactively below to ensure that the memory
206    /// usage here doesn't blow up too much.
207    ///
208    /// For more information, see #2354.
209    pub to_link: Vec<(PackageId, UnitHash)>,
210    /// This is only used while constructing `to_link` to avoid duplicates.
211    seen_to_link: HashSet<(PackageId, UnitHash)>,
212    /// Host-only dependencies that have build scripts. Each element is an
213    /// index into `BuildScriptOutputs`.
214    ///
215    /// This is the set of transitive dependencies that are host-only
216    /// (proc-macro, plugin, build-dependency) that contain a build script.
217    /// Any `BuildOutput::library_paths` path relative to `target` will be
218    /// added to `LD_LIBRARY_PATH` so that the compiler can find any dynamic
219    /// libraries a build script may have generated.
220    pub plugins: BTreeSet<(PackageId, UnitHash)>,
221}
222
223/// Dependency information as declared by a build script that might trigger
224/// a recompile of itself.
225#[derive(Debug)]
226pub struct BuildDeps {
227    /// Absolute path to the file in the target directory that stores the
228    /// output of the build script.
229    pub build_script_output: PathBuf,
230    /// Files that trigger a rebuild if they change.
231    pub rerun_if_changed: Vec<PathBuf>,
232    /// Environment variables that trigger a rebuild if they change.
233    pub rerun_if_env_changed: Vec<String>,
234}
235
236/// Represents one of the instructions from `cargo::rustc-link-arg-*` build
237/// script instruction family.
238///
239/// In other words, indicates targets that custom linker arguments applies to.
240///
241/// See the [build script documentation][1] for more.
242///
243/// [1]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#cargorustc-link-argflag
244#[derive(Clone, Hash, Debug, PartialEq, Eq, PartialOrd, Ord)]
245pub enum LinkArgTarget {
246    /// Represents `cargo::rustc-link-arg=FLAG`.
247    All,
248    /// Represents `cargo::rustc-cdylib-link-arg=FLAG`.
249    Cdylib,
250    /// Represents `cargo::rustc-link-arg-bins=FLAG`.
251    Bin,
252    /// Represents `cargo::rustc-link-arg-bin=BIN=FLAG`.
253    SingleBin(String),
254    /// Represents `cargo::rustc-link-arg-tests=FLAG`.
255    Test,
256    /// Represents `cargo::rustc-link-arg-benches=FLAG`.
257    Bench,
258    /// Represents `cargo::rustc-link-arg-examples=FLAG`.
259    Example,
260}
261
262impl LinkArgTarget {
263    /// Checks if this link type applies to a given [`Target`].
264    pub fn applies_to(&self, target: &Target, mode: CompileMode) -> bool {
265        let is_test = mode.is_any_test();
266        match self {
267            LinkArgTarget::All => true,
268            LinkArgTarget::Cdylib => !is_test && target.is_cdylib(),
269            LinkArgTarget::Bin => target.is_bin(),
270            LinkArgTarget::SingleBin(name) => target.is_bin() && target.name() == name,
271            LinkArgTarget::Test => target.is_test(),
272            LinkArgTarget::Bench => target.is_bench(),
273            LinkArgTarget::Example => target.is_exe_example(),
274        }
275    }
276}
277
278/// Prepares a `Work` that executes the target as a custom build script.
279#[tracing::instrument(skip_all)]
280pub fn prepare(build_runner: &mut BuildRunner<'_, '_>, unit: &Unit) -> CargoResult<Job> {
281    let metadata = build_runner.get_run_build_script_metadata(unit);
282    if build_runner
283        .build_script_outputs
284        .lock()
285        .unwrap()
286        .contains_key(metadata)
287    {
288        // The output is already set, thus the build script is overridden.
289        fingerprint::prepare_target(build_runner, unit, false)
290    } else {
291        build_work(build_runner, unit)
292    }
293}
294
295/// Emits the output of a build script as a [`machine_message::BuildScript`]
296/// JSON string to standard output.
297fn emit_build_output(
298    state: &JobState<'_, '_>,
299    output: &BuildOutput,
300    out_dir: &Path,
301    package_id: PackageId,
302) -> CargoResult<()> {
303    let library_paths = output
304        .library_paths
305        .iter()
306        .map(|l| l.as_ref().display().to_string())
307        .collect::<Vec<_>>();
308
309    let msg = machine_message::BuildScript {
310        package_id: package_id.to_spec(),
311        linked_libs: &output.library_links,
312        linked_paths: &library_paths,
313        cfgs: &output.cfgs,
314        env: &output.env,
315        out_dir,
316    }
317    .to_json_string();
318    state.stdout(msg)?;
319    Ok(())
320}
321
322/// Constructs the unit of work of running a build script.
323///
324/// The construction includes:
325///
326/// * Set environment variables for the build script run.
327/// * Create the output dir (`OUT_DIR`) for the build script output.
328/// * Determine if the build script needs a re-run.
329/// * Run the build script and store its output.
330fn build_work(build_runner: &mut BuildRunner<'_, '_>, unit: &Unit) -> CargoResult<Job> {
331    assert!(unit.mode.is_run_custom_build());
332    let bcx = &build_runner.bcx;
333    let dependencies = build_runner.unit_deps(unit);
334    let build_script_unit = dependencies
335        .iter()
336        .find(|d| !d.unit.mode.is_run_custom_build() && d.unit.target.is_custom_build())
337        .map(|d| &d.unit)
338        .expect("running a script not depending on an actual script");
339    let script_dir = build_runner.files().build_script_dir(build_script_unit);
340
341    let script_out_dir = if bcx.gctx.cli_unstable().build_dir_new_layout {
342        build_runner.files().out_dir_new_layout(unit)
343    } else {
344        build_runner.files().build_script_out_dir(unit)
345    };
346
347    if let Some(deps) = unit.pkg.manifest().metabuild() {
348        prepare_metabuild(build_runner, build_script_unit, deps)?;
349    }
350
351    // Building the command to execute
352    let to_exec = script_dir.join(unit.target.name());
353
354    // Start preparing the process to execute, starting out with some
355    // environment variables. Note that the profile-related environment
356    // variables are not set with this the build script's profile but rather the
357    // package's library profile.
358    // NOTE: if you add any profile flags, be sure to update
359    // `Profiles::get_profile_run_custom_build` so that those flags get
360    // carried over.
361    let to_exec = to_exec.into_os_string();
362    let mut cmd = build_runner.compilation.host_process(to_exec, &unit.pkg)?;
363    let debug = unit.profile.debuginfo.is_turned_on();
364    cmd.env("OUT_DIR", &script_out_dir)
365        .env("CARGO_MANIFEST_DIR", unit.pkg.root())
366        .env("CARGO_MANIFEST_PATH", unit.pkg.manifest_path())
367        .env("NUM_JOBS", &bcx.jobs().to_string())
368        .env("TARGET", bcx.target_data.short_name(&unit.kind))
369        .env("DEBUG", debug.to_string())
370        .env("OPT_LEVEL", &unit.profile.opt_level)
371        .env(
372            "PROFILE",
373            match unit.profile.root {
374                ProfileRoot::Release => "release",
375                ProfileRoot::Debug => "debug",
376            },
377        )
378        .env("HOST", &bcx.host_triple())
379        .env("RUSTC", &bcx.rustc().path)
380        .env("RUSTDOC", &*bcx.gctx.rustdoc()?)
381        .inherit_jobserver(&build_runner.jobserver);
382
383    // Find all artifact dependencies and make their file and containing directory discoverable using environment variables.
384    for (var, value) in artifact::get_env(build_runner, unit, dependencies)? {
385        cmd.env(&var, value);
386    }
387
388    if let Some(linker) = &build_runner.compilation.target_linker(unit.kind) {
389        cmd.env("RUSTC_LINKER", linker);
390    }
391
392    if let Some(links) = unit.pkg.manifest().links() {
393        cmd.env("CARGO_MANIFEST_LINKS", links);
394    }
395
396    if let Some(trim_paths) = unit.profile.trim_paths.as_ref() {
397        cmd.env("CARGO_TRIM_PATHS", trim_paths.to_string());
398    }
399
400    // Be sure to pass along all enabled features for this package, this is the
401    // last piece of statically known information that we have.
402    for feat in &unit.features {
403        cmd.env(&format!("CARGO_FEATURE_{}", super::envify(feat)), "1");
404    }
405
406    let mut cfg_map = HashMap::new();
407    cfg_map.insert(
408        "feature",
409        unit.features.iter().map(|s| s.as_str()).collect::<Vec<_>>(),
410    );
411    // Manually inject debug_assertions based on the profile setting.
412    // The cfg query from rustc doesn't include profile settings and would always be true,
413    // so we override it with the actual profile setting.
414    if unit.profile.debug_assertions {
415        cfg_map.insert("debug_assertions", Vec::new());
416    }
417    for cfg in bcx.target_data.cfg(unit.kind) {
418        match *cfg {
419            Cfg::Name(ref n) => {
420                // Skip debug_assertions from rustc query; we use the profile setting instead
421                if n.as_str() == "debug_assertions" {
422                    continue;
423                }
424                cfg_map.insert(n.as_str(), Vec::new());
425            }
426            Cfg::KeyPair(ref k, ref v) => {
427                let values = cfg_map.entry(k.as_str()).or_default();
428                values.push(v.as_str());
429            }
430        }
431    }
432    for (k, v) in cfg_map {
433        // FIXME: We should handle raw-idents somehow instead of predenting they
434        // don't exist here
435        let k = format!("CARGO_CFG_{}", super::envify(k));
436        cmd.env(&k, v.join(","));
437    }
438
439    // Also inform the build script of the rustc compiler context.
440    if let Some(wrapper) = bcx.rustc().wrapper.as_ref() {
441        cmd.env("RUSTC_WRAPPER", wrapper);
442    } else {
443        cmd.env_remove("RUSTC_WRAPPER");
444    }
445    cmd.env_remove("RUSTC_WORKSPACE_WRAPPER");
446    if build_runner.bcx.ws.is_member(&unit.pkg) {
447        if let Some(wrapper) = bcx.rustc().workspace_wrapper.as_ref() {
448            cmd.env("RUSTC_WORKSPACE_WRAPPER", wrapper);
449        }
450    }
451    cmd.env("CARGO_ENCODED_RUSTFLAGS", unit.rustflags.join("\x1f"));
452    cmd.env_remove("RUSTFLAGS");
453
454    if build_runner.bcx.ws.gctx().extra_verbose() {
455        cmd.display_env_vars();
456    }
457
458    let any_build_script_metadata = bcx.gctx.cli_unstable().any_build_script_metadata;
459
460    // Gather the set of native dependencies that this package has along with
461    // some other variables to close over.
462    //
463    // This information will be used at build-time later on to figure out which
464    // sorts of variables need to be discovered at that time.
465    let lib_deps = dependencies
466        .iter()
467        .filter_map(|dep| {
468            if dep.unit.mode.is_run_custom_build() {
469                let dep_metadata = build_runner.get_run_build_script_metadata(&dep.unit);
470
471                let dep_name = dep.dep_name.unwrap_or(dep.unit.pkg.name());
472
473                Some((
474                    dep_name,
475                    dep.unit
476                        .pkg
477                        .manifest()
478                        .links()
479                        .map(|links| links.to_string()),
480                    dep.unit.pkg.package_id(),
481                    dep_metadata,
482                ))
483            } else {
484                None
485            }
486        })
487        .collect::<Vec<_>>();
488    let library_name = unit.pkg.library().map(|t| t.crate_name());
489    let pkg_descr = unit.pkg.to_string();
490    let build_script_outputs = Arc::clone(&build_runner.build_script_outputs);
491    let id = unit.pkg.package_id();
492    let run_files = BuildScriptRunFiles::for_unit(build_runner, unit);
493    let host_target_root = build_runner.files().host_dest().map(|v| v.to_path_buf());
494    let all = (
495        id,
496        library_name.clone(),
497        pkg_descr.clone(),
498        Arc::clone(&build_script_outputs),
499        run_files.stdout.clone(),
500        script_out_dir.clone(),
501    );
502    let build_scripts = build_runner.build_scripts.get(unit).cloned();
503    let json_messages = bcx.build_config.emit_json();
504    let extra_verbose = bcx.gctx.extra_verbose();
505    let (prev_output, prev_script_out_dir) = prev_build_output(build_runner, unit);
506    let metadata_hash = build_runner.get_run_build_script_metadata(unit);
507
508    paths::create_dir_all(&script_dir)?;
509    paths::create_dir_all(&script_out_dir)?;
510    paths::create_dir_all(&run_files.root)?;
511
512    let nightly_features_allowed = build_runner.bcx.gctx.nightly_features_allowed;
513    let targets: Vec<Target> = unit.pkg.targets().to_vec();
514    let msrv = unit.pkg.rust_version().cloned();
515    // Need a separate copy for the fresh closure.
516    let targets_fresh = targets.clone();
517    let msrv_fresh = msrv.clone();
518
519    let env_profile_name = unit.profile.name.to_uppercase();
520    let built_with_debuginfo = build_runner
521        .bcx
522        .unit_graph
523        .get(unit)
524        .and_then(|deps| deps.iter().find(|dep| dep.unit.target == unit.target))
525        .map(|dep| dep.unit.profile.debuginfo.is_turned_on())
526        .unwrap_or(false);
527
528    // Prepare the unit of "dirty work" which will actually run the custom build
529    // command.
530    //
531    // Note that this has to do some extra work just before running the command
532    // to determine extra environment variables and such.
533    let dirty = Work::new(move |state| {
534        // Make sure that OUT_DIR exists.
535        //
536        // If we have an old build directory, then just move it into place,
537        // otherwise create it!
538        paths::create_dir_all(&script_out_dir)
539            .context("failed to create script output directory for build command")?;
540
541        // For all our native lib dependencies, pick up their metadata to pass
542        // along to this custom build command. We're also careful to augment our
543        // dynamic library search path in case the build script depended on any
544        // native dynamic libraries.
545        {
546            let build_script_outputs = build_script_outputs.lock().unwrap();
547            for (name, links, dep_id, dep_metadata) in lib_deps {
548                let script_output = build_script_outputs.get(dep_metadata).ok_or_else(|| {
549                    internal(format!(
550                        "failed to locate build state for env vars: {}/{}",
551                        dep_id, dep_metadata
552                    ))
553                })?;
554                let data = &script_output.metadata;
555                for (key, value) in data.iter() {
556                    if let Some(ref links) = links {
557                        cmd.env(
558                            &format!("DEP_{}_{}", super::envify(&links), super::envify(key)),
559                            value,
560                        );
561                    }
562                    if any_build_script_metadata {
563                        cmd.env(
564                            &format!("CARGO_DEP_{}_{}", super::envify(&name), super::envify(key)),
565                            value,
566                        );
567                    }
568                }
569            }
570            if let Some(build_scripts) = build_scripts
571                && let Some(ref host_target_root) = host_target_root
572            {
573                super::add_plugin_deps(
574                    &mut cmd,
575                    &build_script_outputs,
576                    &build_scripts,
577                    host_target_root,
578                )?;
579            }
580        }
581
582        // And now finally, run the build command itself!
583        state.running(&cmd);
584        let timestamp = paths::set_invocation_time(&run_files.root)?;
585        let prefix = format!("[{} {}] ", id.name(), id.version());
586        let mut log_messages_in_case_of_panic = Vec::new();
587        let span = tracing::debug_span!("build_script", process = cmd.to_string());
588        let output = span.in_scope(|| {
589            cmd.exec_with_streaming(
590                &mut |stdout| {
591                    if let Some(error) = stdout.strip_prefix(CARGO_ERROR_SYNTAX) {
592                        log_messages_in_case_of_panic.push((Severity::Error, error.to_owned()));
593                    }
594                    if let Some(warning) = stdout
595                        .strip_prefix(OLD_CARGO_WARNING_SYNTAX)
596                        .or(stdout.strip_prefix(NEW_CARGO_WARNING_SYNTAX))
597                    {
598                        log_messages_in_case_of_panic.push((Severity::Warning, warning.to_owned()));
599                    }
600                    if extra_verbose {
601                        state.stdout(format!("{}{}", prefix, stdout))?;
602                    }
603                    Ok(())
604                },
605                &mut |stderr| {
606                    if extra_verbose {
607                        state.stderr(format!("{}{}", prefix, stderr))?;
608                    }
609                    Ok(())
610                },
611                true,
612            )
613            .with_context(|| {
614                let mut build_error_context =
615                    format!("failed to run custom build command for `{}`", pkg_descr);
616
617                // If we're opting into backtraces, mention that build dependencies' backtraces can
618                // be improved by requesting debuginfo to be built, if we're not building with
619                // debuginfo already.
620                #[expect(clippy::disallowed_methods, reason = "consistency with rustc")]
621                if let Ok(show_backtraces) = std::env::var("RUST_BACKTRACE") {
622                    if !built_with_debuginfo && show_backtraces != "0" {
623                        build_error_context.push_str(&format!(
624                            "\n\
625                            note: To improve backtraces for build dependencies, set the \
626                            CARGO_PROFILE_{env_profile_name}_BUILD_OVERRIDE_DEBUG=true environment \
627                            variable to enable debug information generation.",
628                        ));
629                    }
630                }
631
632                build_error_context
633            })
634        });
635
636        // If the build failed
637        if let Err(error) = output {
638            insert_log_messages_in_build_outputs(
639                build_script_outputs,
640                id,
641                metadata_hash,
642                log_messages_in_case_of_panic,
643            );
644            return Err(error);
645        }
646        // ... or it logged any errors
647        else if log_messages_in_case_of_panic
648            .iter()
649            .any(|(severity, _)| *severity == Severity::Error)
650        {
651            insert_log_messages_in_build_outputs(
652                build_script_outputs,
653                id,
654                metadata_hash,
655                log_messages_in_case_of_panic,
656            );
657            anyhow::bail!("build script logged errors");
658        }
659
660        let output = output.unwrap();
661
662        // After the build command has finished running, we need to be sure to
663        // remember all of its output so we can later discover precisely what it
664        // was, even if we don't run the build command again (due to freshness).
665        //
666        // This is also the location where we provide feedback into the build
667        // state informing what variables were discovered via our script as
668        // well.
669        paths::write(&run_files.stdout, &output.stdout)?;
670        // This mtime shift allows Cargo to detect if a source file was
671        // modified in the middle of the build.
672        paths::set_file_time_no_err(run_files.stdout, timestamp);
673        paths::write(&run_files.stderr, &output.stderr)?;
674        paths::write(&run_files.root_output, paths::path2bytes(&script_out_dir)?)?;
675        let parsed_output = BuildOutput::parse(
676            &output.stdout,
677            library_name,
678            &pkg_descr,
679            &script_out_dir,
680            &script_out_dir,
681            nightly_features_allowed,
682            &targets,
683            &msrv,
684        )?;
685
686        if json_messages {
687            emit_build_output(state, &parsed_output, script_out_dir.as_path(), id)?;
688        }
689        build_script_outputs
690            .lock()
691            .unwrap()
692            .insert(id, metadata_hash, parsed_output);
693        Ok(())
694    });
695
696    // Now that we've prepared our work-to-do, we need to prepare the fresh work
697    // itself to run when we actually end up just discarding what we calculated
698    // above.
699    let fresh = Work::new(move |state| {
700        let (id, library_name, pkg_descr, build_script_outputs, output_file, script_out_dir) = all;
701        let output = match prev_output {
702            Some(output) => output,
703            None => BuildOutput::parse_file(
704                &output_file,
705                library_name,
706                &pkg_descr,
707                &prev_script_out_dir,
708                &script_out_dir,
709                nightly_features_allowed,
710                &targets_fresh,
711                &msrv_fresh,
712            )?,
713        };
714
715        if json_messages {
716            emit_build_output(state, &output, script_out_dir.as_path(), id)?;
717        }
718
719        build_script_outputs
720            .lock()
721            .unwrap()
722            .insert(id, metadata_hash, output);
723        Ok(())
724    });
725
726    let mut job = fingerprint::prepare_target(build_runner, unit, false)?;
727    if job.freshness().is_dirty() {
728        job.before(dirty);
729    } else {
730        job.before(fresh);
731    }
732    Ok(job)
733}
734
735/// When a build script run fails, store only log messages, and nuke other
736/// outputs, as they are likely broken.
737fn insert_log_messages_in_build_outputs(
738    build_script_outputs: Arc<Mutex<BuildScriptOutputs>>,
739    id: PackageId,
740    metadata_hash: UnitHash,
741    log_messages: Vec<LogMessage>,
742) {
743    let build_output_with_only_log_messages = BuildOutput {
744        log_messages,
745        ..BuildOutput::default()
746    };
747    build_script_outputs.lock().unwrap().insert(
748        id,
749        metadata_hash,
750        build_output_with_only_log_messages,
751    );
752}
753
754impl BuildOutput {
755    /// Like [`BuildOutput::parse`] but from a file path.
756    pub fn parse_file(
757        path: &Path,
758        library_name: Option<String>,
759        pkg_descr: &str,
760        script_out_dir_when_generated: &Path,
761        script_out_dir: &Path,
762        nightly_features_allowed: bool,
763        targets: &[Target],
764        msrv: &Option<RustVersion>,
765    ) -> CargoResult<BuildOutput> {
766        let contents = paths::read_bytes(path)?;
767        BuildOutput::parse(
768            &contents,
769            library_name,
770            pkg_descr,
771            script_out_dir_when_generated,
772            script_out_dir,
773            nightly_features_allowed,
774            targets,
775            msrv,
776        )
777    }
778
779    /// Parses the output instructions of a build script.
780    ///
781    /// * `pkg_descr` --- for error messages
782    /// * `library_name` --- for determining if `RUSTC_BOOTSTRAP` should be allowed
783    pub fn parse(
784        input: &[u8],
785        // Takes String instead of InternedString so passing `unit.pkg.name()` will give a compile error.
786        library_name: Option<String>,
787        pkg_descr: &str,
788        script_out_dir_when_generated: &Path,
789        script_out_dir: &Path,
790        nightly_features_allowed: bool,
791        targets: &[Target],
792        msrv: &Option<RustVersion>,
793    ) -> CargoResult<BuildOutput> {
794        let mut library_paths = Vec::new();
795        let mut library_links = Vec::new();
796        let mut linker_args = Vec::new();
797        let mut cfgs = Vec::new();
798        let mut check_cfgs = Vec::new();
799        let mut env = Vec::new();
800        let mut metadata = Vec::new();
801        let mut rerun_if_changed = Vec::new();
802        let mut rerun_if_env_changed = Vec::new();
803        let mut log_messages = Vec::new();
804        let whence = format!("build script of `{}`", pkg_descr);
805        // Old syntax:
806        //    cargo:rustc-flags=VALUE
807        //    cargo:KEY=VALUE (for other unreserved keys)
808        // New syntax:
809        //    cargo::rustc-flags=VALUE
810        //    cargo::metadata=KEY=VALUE (for other unreserved keys)
811        // Due to backwards compatibility, no new keys can be added to this old format.
812        const RESERVED_PREFIXES: &[&str] = &[
813            "rustc-flags=",
814            "rustc-link-lib=",
815            "rustc-link-search=",
816            "rustc-link-arg-cdylib=",
817            "rustc-cdylib-link-arg=",
818            "rustc-link-arg-bins=",
819            "rustc-link-arg-bin=",
820            "rustc-link-arg-tests=",
821            "rustc-link-arg-benches=",
822            "rustc-link-arg-examples=",
823            "rustc-link-arg=",
824            "rustc-cfg=",
825            "rustc-check-cfg=",
826            "rustc-env=",
827            "warning=",
828            "rerun-if-changed=",
829            "rerun-if-env-changed=",
830        ];
831        const DOCS_LINK_SUGGESTION: &str = "See https://doc.rust-lang.org/cargo/reference/build-scripts.html#outputs-of-the-build-script \
832                for more information about build script outputs.";
833
834        fn has_reserved_prefix(flag: &str) -> bool {
835            RESERVED_PREFIXES
836                .iter()
837                .any(|reserved_prefix| flag.starts_with(reserved_prefix))
838        }
839
840        fn check_minimum_supported_rust_version_for_new_syntax(
841            pkg_descr: &str,
842            msrv: &Option<RustVersion>,
843            flag: &str,
844        ) -> CargoResult<()> {
845            if let Some(msrv) = msrv {
846                let new_syntax_added_in = RustVersion::new(1, 77, 0);
847                if !new_syntax_added_in.is_compatible_with(&msrv.to_partial()) {
848                    let old_syntax_suggestion = if has_reserved_prefix(flag) {
849                        format!(
850                            "Switch to the old `cargo:{flag}` syntax (note the single colon).\n"
851                        )
852                    } else if flag.starts_with("metadata=") {
853                        let old_format_flag = flag.strip_prefix("metadata=").unwrap();
854                        format!(
855                            "Switch to the old `cargo:{old_format_flag}` syntax instead of `cargo::{flag}` (note the single colon).\n"
856                        )
857                    } else {
858                        String::new()
859                    };
860
861                    bail!(
862                        "the `cargo::` syntax for build script output instructions was added in \
863                        Rust 1.77.0, but the minimum supported Rust version of `{pkg_descr}` is {msrv}.\n\
864                        {old_syntax_suggestion}\
865                        {DOCS_LINK_SUGGESTION}"
866                    );
867                }
868            }
869
870            Ok(())
871        }
872
873        fn parse_directive<'a>(
874            whence: &str,
875            line: &str,
876            data: &'a str,
877            old_syntax: bool,
878        ) -> CargoResult<(&'a str, &'a str)> {
879            let mut iter = data.splitn(2, "=");
880            let key = iter.next();
881            let value = iter.next();
882            match (key, value) {
883                (Some(a), Some(b)) => Ok((a, b.trim_end())),
884                _ => bail!(
885                    "invalid output in {whence}: `{line}`\n\
886                    Expected a line with `{syntax}KEY=VALUE` with an `=` character, \
887                    but none was found.\n\
888                    {DOCS_LINK_SUGGESTION}",
889                    syntax = if old_syntax { "cargo:" } else { "cargo::" },
890                ),
891            }
892        }
893
894        fn parse_metadata<'a>(
895            whence: &str,
896            line: &str,
897            data: &'a str,
898            old_syntax: bool,
899        ) -> CargoResult<(&'a str, &'a str)> {
900            let mut iter = data.splitn(2, "=");
901            let key = iter.next();
902            let value = iter.next();
903            match (key, value) {
904                (Some(a), Some(b)) => Ok((a, b.trim_end())),
905                _ => bail!(
906                    "invalid output in {whence}: `{line}`\n\
907                    Expected a line with `{syntax}KEY=VALUE` with an `=` character, \
908                    but none was found.\n\
909                    {DOCS_LINK_SUGGESTION}",
910                    syntax = if old_syntax {
911                        "cargo:"
912                    } else {
913                        "cargo::metadata="
914                    },
915                ),
916            }
917        }
918
919        for line in input.split(|b| *b == b'\n') {
920            let line = match str::from_utf8(line) {
921                Ok(line) => line.trim(),
922                Err(..) => continue,
923            };
924            let mut old_syntax = false;
925            let (key, value) = if let Some(data) = line.strip_prefix("cargo::") {
926                check_minimum_supported_rust_version_for_new_syntax(pkg_descr, msrv, data)?;
927                // For instance, `cargo::rustc-flags=foo` or `cargo::metadata=foo=bar`.
928                parse_directive(whence.as_str(), line, data, old_syntax)?
929            } else if let Some(data) = line.strip_prefix("cargo:") {
930                old_syntax = true;
931                // For instance, `cargo:rustc-flags=foo`.
932                if has_reserved_prefix(data) {
933                    parse_directive(whence.as_str(), line, data, old_syntax)?
934                } else {
935                    // For instance, `cargo:foo=bar`.
936                    ("metadata", data)
937                }
938            } else {
939                // Skip this line since it doesn't start with "cargo:" or "cargo::".
940                continue;
941            };
942            // This will rewrite paths if the target directory has been moved.
943            let value = value.replace(
944                script_out_dir_when_generated.to_str().unwrap(),
945                script_out_dir.to_str().unwrap(),
946            );
947
948            let syntax_prefix = if old_syntax { "cargo:" } else { "cargo::" };
949            macro_rules! check_and_add_target {
950                ($target_kind: expr, $is_target_kind: expr, $link_type: expr) => {
951                    if !targets.iter().any(|target| $is_target_kind(target)) {
952                        bail!(
953                            "invalid instruction `{}{}` from {}\n\
954                                The package {} does not have a {} target.",
955                            syntax_prefix,
956                            key,
957                            whence,
958                            pkg_descr,
959                            $target_kind
960                        );
961                    }
962                    linker_args.push(($link_type, value));
963                };
964            }
965
966            // Keep in sync with TargetConfig::parse_links_overrides.
967            match key {
968                "rustc-flags" => {
969                    let (paths, links) = BuildOutput::parse_rustc_flags(&value, &whence)?;
970                    library_links.extend(links.into_iter());
971                    library_paths.extend(
972                        paths
973                            .into_iter()
974                            .map(|p| LibraryPath::new(p, script_out_dir)),
975                    );
976                }
977                "rustc-link-lib" => library_links.push(value.to_string()),
978                "rustc-link-search" => {
979                    library_paths.push(LibraryPath::new(PathBuf::from(value), script_out_dir))
980                }
981                "rustc-link-arg-cdylib" | "rustc-cdylib-link-arg" => {
982                    if !targets.iter().any(|target| target.is_cdylib()) {
983                        log_messages.push((
984                            Severity::Warning,
985                            format!(
986                                "{}{} was specified in the build script of {}, \
987                             but that package does not contain a cdylib target\n\
988                             \n\
989                             Allowing this was an unintended change in the 1.50 \
990                             release, and may become an error in the future. \
991                             For more information, see \
992                             <https://github.com/rust-lang/cargo/issues/9562>.",
993                                syntax_prefix, key, pkg_descr
994                            ),
995                        ));
996                    }
997                    linker_args.push((LinkArgTarget::Cdylib, value))
998                }
999                "rustc-link-arg-bins" => {
1000                    check_and_add_target!("bin", Target::is_bin, LinkArgTarget::Bin);
1001                }
1002                "rustc-link-arg-bin" => {
1003                    let (bin_name, arg) = value.split_once('=').ok_or_else(|| {
1004                        anyhow::format_err!(
1005                            "invalid instruction `{}{}={}` from {}\n\
1006                                The instruction should have the form {}{}=BIN=ARG",
1007                            syntax_prefix,
1008                            key,
1009                            value,
1010                            whence,
1011                            syntax_prefix,
1012                            key
1013                        )
1014                    })?;
1015                    if !targets
1016                        .iter()
1017                        .any(|target| target.is_bin() && target.name() == bin_name)
1018                    {
1019                        bail!(
1020                            "invalid instruction `{}{}` from {}\n\
1021                                The package {} does not have a bin target with the name `{}`.",
1022                            syntax_prefix,
1023                            key,
1024                            whence,
1025                            pkg_descr,
1026                            bin_name
1027                        );
1028                    }
1029                    linker_args.push((
1030                        LinkArgTarget::SingleBin(bin_name.to_owned()),
1031                        arg.to_string(),
1032                    ));
1033                }
1034                "rustc-link-arg-tests" => {
1035                    check_and_add_target!("test", Target::is_test, LinkArgTarget::Test);
1036                }
1037                "rustc-link-arg-benches" => {
1038                    check_and_add_target!("benchmark", Target::is_bench, LinkArgTarget::Bench);
1039                }
1040                "rustc-link-arg-examples" => {
1041                    check_and_add_target!("example", Target::is_example, LinkArgTarget::Example);
1042                }
1043                "rustc-link-arg" => {
1044                    linker_args.push((LinkArgTarget::All, value));
1045                }
1046                "rustc-cfg" => cfgs.push(value.to_string()),
1047                "rustc-check-cfg" => check_cfgs.push(value.to_string()),
1048                "rustc-env" => {
1049                    let (key, val) = BuildOutput::parse_rustc_env(&value, &whence)?;
1050                    // Build scripts aren't allowed to set RUSTC_BOOTSTRAP.
1051                    // See https://github.com/rust-lang/cargo/issues/7088.
1052                    if key == "RUSTC_BOOTSTRAP" {
1053                        // If RUSTC_BOOTSTRAP is already set, the user of Cargo knows about
1054                        // bootstrap and still wants to override the channel. Give them a way to do
1055                        // so, but still emit a warning that the current crate shouldn't be trying
1056                        // to set RUSTC_BOOTSTRAP.
1057                        // If this is a nightly build, setting RUSTC_BOOTSTRAP wouldn't affect the
1058                        // behavior, so still only give a warning.
1059                        // NOTE: cargo only allows nightly features on RUSTC_BOOTSTRAP=1, but we
1060                        // want setting any value of RUSTC_BOOTSTRAP to downgrade this to a warning
1061                        // (so that `RUSTC_BOOTSTRAP=library_name` will work)
1062                        let rustc_bootstrap_allows = |name: Option<&str>| {
1063                            let name = match name {
1064                                // as of 2021, no binaries on crates.io use RUSTC_BOOTSTRAP, so
1065                                // fine-grained opt-outs aren't needed. end-users can always use
1066                                // RUSTC_BOOTSTRAP=1 from the top-level if it's really a problem.
1067                                None => return false,
1068                                Some(n) => n,
1069                            };
1070                            #[expect(
1071                                clippy::disallowed_methods,
1072                                reason = "consistency with rustc, not specified behavior"
1073                            )]
1074                            std::env::var("RUSTC_BOOTSTRAP")
1075                                .map_or(false, |var| var.split(',').any(|s| s == name))
1076                        };
1077                        if nightly_features_allowed
1078                            || rustc_bootstrap_allows(library_name.as_deref())
1079                        {
1080                            log_messages.push((Severity::Warning, format!("cannot set `RUSTC_BOOTSTRAP={}` from {}.\n\
1081                                note: crates cannot set `RUSTC_BOOTSTRAP` themselves, as doing so would subvert the stability guarantees of Rust for your project.",
1082                                val, whence
1083                            )));
1084                        } else {
1085                            // Setting RUSTC_BOOTSTRAP would change the behavior of the crate.
1086                            // Abort with an error.
1087                            bail!(
1088                                "cannot set `RUSTC_BOOTSTRAP={}` from {}.\n\
1089                                note: crates cannot set `RUSTC_BOOTSTRAP` themselves, as doing so would subvert the stability guarantees of Rust for your project.\n\
1090                                help: If you're sure you want to do this in your project, set the environment variable `RUSTC_BOOTSTRAP={}` before running cargo instead.",
1091                                val,
1092                                whence,
1093                                library_name.as_deref().unwrap_or("1"),
1094                            );
1095                        }
1096                    } else {
1097                        env.push((key, val));
1098                    }
1099                }
1100                "error" => log_messages.push((Severity::Error, value.to_string())),
1101                "warning" => log_messages.push((Severity::Warning, value.to_string())),
1102                "rerun-if-changed" => rerun_if_changed.push(PathBuf::from(value)),
1103                "rerun-if-env-changed" => rerun_if_env_changed.push(value.to_string()),
1104                "metadata" => {
1105                    let (key, value) = parse_metadata(whence.as_str(), line, &value, old_syntax)?;
1106                    metadata.push((key.to_owned(), value.to_owned()));
1107                }
1108                _ => bail!(
1109                    "invalid output in {whence}: `{line}`\n\
1110                    Unknown key: `{key}`.\n\
1111                    {DOCS_LINK_SUGGESTION}",
1112                ),
1113            }
1114        }
1115
1116        Ok(BuildOutput {
1117            library_paths,
1118            library_links,
1119            linker_args,
1120            cfgs,
1121            check_cfgs,
1122            env,
1123            metadata,
1124            rerun_if_changed,
1125            rerun_if_env_changed,
1126            log_messages,
1127        })
1128    }
1129
1130    /// Parses [`cargo::rustc-flags`] instruction.
1131    ///
1132    /// [`cargo::rustc-flags`]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#cargorustc-flagsflags
1133    pub fn parse_rustc_flags(
1134        value: &str,
1135        whence: &str,
1136    ) -> CargoResult<(Vec<PathBuf>, Vec<String>)> {
1137        let value = value.trim();
1138        let mut flags_iter = value
1139            .split(|c: char| c.is_whitespace())
1140            .filter(|w| w.chars().any(|c| !c.is_whitespace()));
1141        let (mut library_paths, mut library_links) = (Vec::new(), Vec::new());
1142
1143        while let Some(flag) = flags_iter.next() {
1144            if flag.starts_with("-l") || flag.starts_with("-L") {
1145                // Check if this flag has no space before the value as is
1146                // common with tools like pkg-config
1147                // e.g. -L/some/dir/local/lib or -licui18n
1148                let (flag, mut value) = flag.split_at(2);
1149                if value.is_empty() {
1150                    value = match flags_iter.next() {
1151                        Some(v) => v,
1152                        None => bail! {
1153                            "flag in rustc-flags has no value in {}: {}",
1154                            whence,
1155                            value
1156                        },
1157                    }
1158                }
1159
1160                match flag {
1161                    "-l" => library_links.push(value.to_string()),
1162                    "-L" => library_paths.push(PathBuf::from(value)),
1163
1164                    // This was already checked above
1165                    _ => unreachable!(),
1166                };
1167            } else {
1168                bail!(
1169                    "only `-l` and `-L` flags are allowed in {}: `{}`",
1170                    whence,
1171                    value
1172                )
1173            }
1174        }
1175        Ok((library_paths, library_links))
1176    }
1177
1178    /// Parses [`cargo::rustc-env`] instruction.
1179    ///
1180    /// [`cargo::rustc-env`]: https://doc.rust-lang.org/nightly/cargo/reference/build-scripts.html#rustc-env
1181    pub fn parse_rustc_env(value: &str, whence: &str) -> CargoResult<(String, String)> {
1182        match value.split_once('=') {
1183            Some((n, v)) => Ok((n.to_owned(), v.to_owned())),
1184            _ => bail!("Variable rustc-env has no value in {whence}: {value}"),
1185        }
1186    }
1187}
1188
1189/// Prepares the Rust script for the unstable feature [metabuild].
1190///
1191/// [metabuild]: https://doc.rust-lang.org/nightly/cargo/reference/unstable.html#metabuild
1192fn prepare_metabuild(
1193    build_runner: &BuildRunner<'_, '_>,
1194    unit: &Unit,
1195    deps: &[String],
1196) -> CargoResult<()> {
1197    let mut output = Vec::new();
1198    let available_deps = build_runner.unit_deps(unit);
1199    // Filter out optional dependencies, and look up the actual lib name.
1200    let meta_deps: Vec<_> = deps
1201        .iter()
1202        .filter_map(|name| {
1203            available_deps
1204                .iter()
1205                .find(|d| d.unit.pkg.name().as_str() == name.as_str())
1206                .map(|d| d.unit.target.crate_name())
1207        })
1208        .collect();
1209    output.push("fn main() {\n".to_string());
1210    for dep in &meta_deps {
1211        output.push(format!("    {}::metabuild();\n", dep));
1212    }
1213    output.push("}\n".to_string());
1214    let output = output.join("");
1215    let path = unit
1216        .pkg
1217        .manifest()
1218        .metabuild_path(build_runner.bcx.ws.build_dir());
1219    paths::create_dir_all(path.parent().unwrap())?;
1220    paths::write_if_changed(path, &output)?;
1221    Ok(())
1222}
1223
1224impl BuildDeps {
1225    /// Creates a build script dependency information from a previous
1226    /// build script output path and the content.
1227    pub fn new(output_file: &Path, output: Option<&BuildOutput>) -> BuildDeps {
1228        BuildDeps {
1229            build_script_output: output_file.to_path_buf(),
1230            rerun_if_changed: output
1231                .map(|p| &p.rerun_if_changed)
1232                .cloned()
1233                .unwrap_or_default(),
1234            rerun_if_env_changed: output
1235                .map(|p| &p.rerun_if_env_changed)
1236                .cloned()
1237                .unwrap_or_default(),
1238        }
1239    }
1240}
1241
1242/// Computes several maps in [`BuildRunner`].
1243///
1244/// - [`build_scripts`]: A map that tracks which build scripts each package
1245///   depends on.
1246/// - [`build_explicit_deps`]: Dependency statements emitted by build scripts
1247///   from a previous run.
1248/// - [`build_script_outputs`]: Pre-populates this with any overridden build
1249///   scripts.
1250///
1251/// The important one here is [`build_scripts`], which for each `(package,
1252/// metadata)` stores a [`BuildScripts`] object which contains a list of
1253/// dependencies with build scripts that the unit should consider when linking.
1254/// For example this lists all dependencies' `-L` flags which need to be
1255/// propagated transitively.
1256///
1257/// The given set of units to this function is the initial set of
1258/// targets/profiles which are being built.
1259///
1260/// [`build_scripts`]: BuildRunner::build_scripts
1261/// [`build_explicit_deps`]: BuildRunner::build_explicit_deps
1262/// [`build_script_outputs`]: BuildRunner::build_script_outputs
1263pub fn build_map(build_runner: &mut BuildRunner<'_, '_>) -> CargoResult<()> {
1264    let mut ret = HashMap::new();
1265    for unit in &build_runner.bcx.roots {
1266        build(&mut ret, build_runner, unit)?;
1267    }
1268    build_runner
1269        .build_scripts
1270        .extend(ret.into_iter().map(|(k, v)| (k, Arc::new(v))));
1271    return Ok(());
1272
1273    // Recursive function to build up the map we're constructing. This function
1274    // memoizes all of its return values as it goes along.
1275    fn build<'a>(
1276        out: &'a mut HashMap<Unit, BuildScripts>,
1277        build_runner: &mut BuildRunner<'_, '_>,
1278        unit: &Unit,
1279    ) -> CargoResult<&'a BuildScripts> {
1280        // Do a quick pre-flight check to see if we've already calculated the
1281        // set of dependencies.
1282        if out.contains_key(unit) {
1283            return Ok(&out[unit]);
1284        }
1285
1286        // If there is a build script override, pre-fill the build output.
1287        if unit.mode.is_run_custom_build() {
1288            if let Some(links) = unit.pkg.manifest().links() {
1289                if let Some(output) = unit.links_overrides.get(links) {
1290                    let metadata = build_runner.get_run_build_script_metadata(unit);
1291                    build_runner.build_script_outputs.lock().unwrap().insert(
1292                        unit.pkg.package_id(),
1293                        metadata,
1294                        output.clone(),
1295                    );
1296                }
1297            }
1298        }
1299
1300        let mut ret = BuildScripts::default();
1301
1302        // If a package has a build script, add itself as something to inspect for linking.
1303        if !unit.target.is_custom_build() && unit.pkg.has_custom_build() {
1304            let script_metas = build_runner
1305                .find_build_script_metadatas(unit)
1306                .expect("has_custom_build should have RunCustomBuild");
1307            for script_meta in script_metas {
1308                add_to_link(&mut ret, unit.pkg.package_id(), script_meta);
1309            }
1310        }
1311
1312        if unit.mode.is_run_custom_build() {
1313            parse_previous_explicit_deps(build_runner, unit);
1314        }
1315
1316        // We want to invoke the compiler deterministically to be cache-friendly
1317        // to rustc invocation caching schemes, so be sure to generate the same
1318        // set of build script dependency orderings via sorting the targets that
1319        // come out of the `Context`.
1320        let mut dependencies: Vec<Unit> = build_runner
1321            .unit_deps(unit)
1322            .iter()
1323            .map(|d| d.unit.clone())
1324            .collect();
1325        dependencies.sort_by_key(|u| u.pkg.package_id());
1326
1327        for dep_unit in dependencies.iter() {
1328            let dep_scripts = build(out, build_runner, dep_unit)?;
1329
1330            if dep_unit.target.for_host() {
1331                ret.plugins.extend(dep_scripts.to_link.iter().cloned());
1332            } else if dep_unit.target.is_linkable() {
1333                for &(pkg, metadata) in dep_scripts.to_link.iter() {
1334                    add_to_link(&mut ret, pkg, metadata);
1335                }
1336            }
1337        }
1338
1339        match out.entry(unit.clone()) {
1340            Entry::Vacant(entry) => Ok(entry.insert(ret)),
1341            Entry::Occupied(_) => panic!("cyclic dependencies in `build_map`"),
1342        }
1343    }
1344
1345    // When adding an entry to 'to_link' we only actually push it on if the
1346    // script hasn't seen it yet (e.g., we don't push on duplicates).
1347    fn add_to_link(scripts: &mut BuildScripts, pkg: PackageId, metadata: UnitHash) {
1348        if scripts.seen_to_link.insert((pkg, metadata)) {
1349            scripts.to_link.push((pkg, metadata));
1350        }
1351    }
1352
1353    /// Load any dependency declarations from a previous build script run.
1354    fn parse_previous_explicit_deps(build_runner: &mut BuildRunner<'_, '_>, unit: &Unit) {
1355        let run_files = BuildScriptRunFiles::for_unit(build_runner, unit);
1356        let (prev_output, _) = prev_build_output(build_runner, unit);
1357        let deps = BuildDeps::new(&run_files.stdout, prev_output.as_ref());
1358        build_runner.build_explicit_deps.insert(unit.clone(), deps);
1359    }
1360}
1361
1362/// Returns the previous parsed `BuildOutput`, if any, from a previous
1363/// execution.
1364///
1365/// Also returns the directory containing the output, typically used later in
1366/// processing.
1367fn prev_build_output(
1368    build_runner: &mut BuildRunner<'_, '_>,
1369    unit: &Unit,
1370) -> (Option<BuildOutput>, PathBuf) {
1371    let script_out_dir = if build_runner.bcx.gctx.cli_unstable().build_dir_new_layout {
1372        build_runner.files().out_dir_new_layout(unit)
1373    } else {
1374        build_runner.files().build_script_out_dir(unit)
1375    };
1376    let run_files = BuildScriptRunFiles::for_unit(build_runner, unit);
1377
1378    let prev_script_out_dir = paths::read_bytes(&run_files.root_output)
1379        .and_then(|bytes| paths::bytes2path(&bytes))
1380        .unwrap_or_else(|_| script_out_dir.clone());
1381
1382    (
1383        BuildOutput::parse_file(
1384            &run_files.stdout,
1385            unit.pkg.library().map(|t| t.crate_name()),
1386            &unit.pkg.to_string(),
1387            &prev_script_out_dir,
1388            &script_out_dir,
1389            build_runner.bcx.gctx.nightly_features_allowed,
1390            unit.pkg.targets(),
1391            &unit.pkg.rust_version().cloned(),
1392        )
1393        .ok(),
1394        prev_script_out_dir,
1395    )
1396}
1397
1398impl BuildScriptOutputs {
1399    /// Inserts a new entry into the map.
1400    fn insert(&mut self, pkg_id: PackageId, metadata: UnitHash, parsed_output: BuildOutput) {
1401        match self.outputs.entry(metadata) {
1402            Entry::Vacant(entry) => {
1403                entry.insert(parsed_output);
1404            }
1405            Entry::Occupied(entry) => panic!(
1406                "build script output collision for {}/{}\n\
1407                old={:?}\nnew={:?}",
1408                pkg_id,
1409                metadata,
1410                entry.get(),
1411                parsed_output
1412            ),
1413        }
1414    }
1415
1416    /// Returns `true` if the given key already exists.
1417    fn contains_key(&self, metadata: UnitHash) -> bool {
1418        self.outputs.contains_key(&metadata)
1419    }
1420
1421    /// Gets the build output for the given key.
1422    pub fn get(&self, meta: UnitHash) -> Option<&BuildOutput> {
1423        self.outputs.get(&meta)
1424    }
1425
1426    /// Returns an iterator over all entries.
1427    pub fn iter(&self) -> impl Iterator<Item = (&UnitHash, &BuildOutput)> {
1428        self.outputs.iter()
1429    }
1430}
1431
1432/// Files with information about a running build script.
1433struct BuildScriptRunFiles {
1434    /// The directory containing files related to running a build script.
1435    root: PathBuf,
1436    /// The stdout produced by the build script
1437    stdout: PathBuf,
1438    /// The stderr produced by the build script
1439    stderr: PathBuf,
1440    /// A file that contains the path to the `out` dir of the build script.
1441    /// This is used for detect if the directory was moved since the previous run.
1442    root_output: PathBuf,
1443}
1444
1445impl BuildScriptRunFiles {
1446    pub fn for_unit(build_runner: &BuildRunner<'_, '_>, unit: &Unit) -> Self {
1447        let root = build_runner.files().build_script_run_dir(unit);
1448        let stdout = if build_runner.bcx.gctx.cli_unstable().build_dir_new_layout {
1449            root.join("stdout")
1450        } else {
1451            root.join("output")
1452        };
1453        let stderr = root.join("stderr");
1454        let root_output = root.join("root-output");
1455        Self {
1456            root,
1457            stdout,
1458            stderr,
1459            root_output,
1460        }
1461    }
1462}