]> git.lizzy.rs Git - rust.git/blob - src/bootstrap/lib.rs
Rollup merge of #98391 - joboet:sgx_parker, r=m-ou-se
[rust.git] / src / bootstrap / lib.rs
1 //! Implementation of rustbuild, the Rust build system.
2 //!
3 //! This module, and its descendants, are the implementation of the Rust build
4 //! system. Most of this build system is backed by Cargo but the outer layer
5 //! here serves as the ability to orchestrate calling Cargo, sequencing Cargo
6 //! builds, building artifacts like LLVM, etc. The goals of rustbuild are:
7 //!
8 //! * To be an easily understandable, easily extensible, and maintainable build
9 //!   system.
10 //! * Leverage standard tools in the Rust ecosystem to build the compiler, aka
11 //!   crates.io and Cargo.
12 //! * A standard interface to build across all platforms, including MSVC
13 //!
14 //! ## Architecture
15 //!
16 //! The build system defers most of the complicated logic managing invocations
17 //! of rustc and rustdoc to Cargo itself. However, moving through various stages
18 //! and copying artifacts is still necessary for it to do. Each time rustbuild
19 //! is invoked, it will iterate through the list of predefined steps and execute
20 //! each serially in turn if it matches the paths passed or is a default rule.
21 //! For each step rustbuild relies on the step internally being incremental and
22 //! parallel. Note, though, that the `-j` parameter to rustbuild gets forwarded
23 //! to appropriate test harnesses and such.
24 //!
25 //! Most of the "meaty" steps that matter are backed by Cargo, which does indeed
26 //! have its own parallelism and incremental management. Later steps, like
27 //! tests, aren't incremental and simply run the entire suite currently.
28 //! However, compiletest itself tries to avoid running tests when the artifacts
29 //! that are involved (mainly the compiler) haven't changed.
30 //!
31 //! When you execute `x.py build`, the steps executed are:
32 //!
33 //! * First, the python script is run. This will automatically download the
34 //!   stage0 rustc and cargo according to `src/stage0.json`, or use the cached
35 //!   versions if they're available. These are then used to compile rustbuild
36 //!   itself (using Cargo). Finally, control is then transferred to rustbuild.
37 //!
38 //! * Rustbuild takes over, performs sanity checks, probes the environment,
39 //!   reads configuration, and starts executing steps as it reads the command
40 //!   line arguments (paths) or going through the default rules.
41 //!
42 //!   The build output will be something like the following:
43 //!
44 //!   Building stage0 std artifacts
45 //!   Copying stage0 std
46 //!   Building stage0 test artifacts
47 //!   Copying stage0 test
48 //!   Building stage0 compiler artifacts
49 //!   Copying stage0 rustc
50 //!   Assembling stage1 compiler
51 //!   Building stage1 std artifacts
52 //!   Copying stage1 std
53 //!   Building stage1 test artifacts
54 //!   Copying stage1 test
55 //!   Building stage1 compiler artifacts
56 //!   Copying stage1 rustc
57 //!   Assembling stage2 compiler
58 //!   Uplifting stage1 std
59 //!   Uplifting stage1 test
60 //!   Uplifting stage1 rustc
61 //!
62 //! Let's disect that a little:
63 //!
64 //! ## Building stage0 {std,test,compiler} artifacts
65 //!
66 //! These steps use the provided (downloaded, usually) compiler to compile the
67 //! local Rust source into libraries we can use.
68 //!
69 //! ## Copying stage0 {std,test,rustc}
70 //!
71 //! This copies the build output from Cargo into
72 //! `build/$HOST/stage0-sysroot/lib/rustlib/$ARCH/lib`. FIXME: this step's
73 //! documentation should be expanded -- the information already here may be
74 //! incorrect.
75 //!
76 //! ## Assembling stage1 compiler
77 //!
78 //! This copies the libraries we built in "building stage0 ... artifacts" into
79 //! the stage1 compiler's lib directory. These are the host libraries that the
80 //! compiler itself uses to run. These aren't actually used by artifacts the new
81 //! compiler generates. This step also copies the rustc and rustdoc binaries we
82 //! generated into build/$HOST/stage/bin.
83 //!
84 //! The stage1/bin/rustc is a fully functional compiler, but it doesn't yet have
85 //! any libraries to link built binaries or libraries to. The next 3 steps will
86 //! provide those libraries for it; they are mostly equivalent to constructing
87 //! the stage1/bin compiler so we don't go through them individually.
88 //!
89 //! ## Uplifting stage1 {std,test,rustc}
90 //!
91 //! This step copies the libraries from the stage1 compiler sysroot into the
92 //! stage2 compiler. This is done to avoid rebuilding the compiler; libraries
93 //! we'd build in this step should be identical (in function, if not necessarily
94 //! identical on disk) so there's no need to recompile the compiler again. Note
95 //! that if you want to, you can enable the full-bootstrap option to change this
96 //! behavior.
97 //!
98 //! Each step is driven by a separate Cargo project and rustbuild orchestrates
99 //! copying files between steps and otherwise preparing for Cargo to run.
100 //!
101 //! ## Further information
102 //!
103 //! More documentation can be found in each respective module below, and you can
104 //! also check out the `src/bootstrap/README.md` file for more information.
105
106 use std::cell::{Cell, RefCell};
107 use std::collections::{HashMap, HashSet};
108 use std::env;
109 use std::fs::{self, File};
110 use std::io;
111 use std::path::{Path, PathBuf};
112 use std::process::Command;
113 use std::str;
114
115 use channel::GitInfo;
116 use config::{DryRun, Target};
117 use filetime::FileTime;
118 use once_cell::sync::OnceCell;
119
120 use crate::builder::Kind;
121 use crate::config::{LlvmLibunwind, TargetSelection};
122 use crate::util::{exe, libdir, mtime, output, run, run_suppressed, try_run_suppressed, CiEnv};
123
124 mod bolt;
125 mod builder;
126 mod cache;
127 mod cc_detect;
128 mod channel;
129 mod check;
130 mod clean;
131 mod compile;
132 mod config;
133 mod dist;
134 mod doc;
135 mod download;
136 mod flags;
137 mod format;
138 mod install;
139 mod metadata;
140 mod native;
141 mod run;
142 mod sanity;
143 mod setup;
144 mod tarball;
145 mod test;
146 mod tool;
147 mod toolstate;
148 pub mod util;
149
150 #[cfg(feature = "build-metrics")]
151 mod metrics;
152
153 #[cfg(windows)]
154 mod job;
155
156 #[cfg(all(unix, not(target_os = "haiku")))]
157 mod job {
158     pub unsafe fn setup(build: &mut crate::Build) {
159         if build.config.low_priority {
160             libc::setpriority(libc::PRIO_PGRP as _, 0, 10);
161         }
162     }
163 }
164
165 #[cfg(any(target_os = "haiku", target_os = "hermit", not(any(unix, windows))))]
166 mod job {
167     pub unsafe fn setup(_build: &mut crate::Build) {}
168 }
169
170 pub use crate::builder::PathSet;
171 use crate::cache::{Interned, INTERNER};
172 pub use crate::config::Config;
173 pub use crate::flags::Subcommand;
174
175 const LLVM_TOOLS: &[&str] = &[
176     "llvm-cov",      // used to generate coverage report
177     "llvm-nm",       // used to inspect binaries; it shows symbol names, their sizes and visibility
178     "llvm-objcopy",  // used to transform ELFs into binary format which flashing tools consume
179     "llvm-objdump",  // used to disassemble programs
180     "llvm-profdata", // used to inspect and merge files generated by profiles
181     "llvm-readobj",  // used to get information from ELFs/objects that the other tools don't provide
182     "llvm-size",     // used to prints the size of the linker sections of a program
183     "llvm-strip",    // used to discard symbols from binary files to reduce their size
184     "llvm-ar",       // used for creating and modifying archive files
185     "llvm-as",       // used to convert LLVM assembly to LLVM bitcode
186     "llvm-dis",      // used to disassemble LLVM bitcode
187     "llc",           // used to compile LLVM bytecode
188     "opt",           // used to optimize LLVM bytecode
189 ];
190
191 /// LLD file names for all flavors.
192 const LLD_FILE_NAMES: &[&str] = &["ld.lld", "ld64.lld", "lld-link", "wasm-ld"];
193
194 pub const VERSION: usize = 2;
195
196 /// Extra --check-cfg to add when building
197 /// (Mode restriction, config name, config values (if any))
198 const EXTRA_CHECK_CFGS: &[(Option<Mode>, &'static str, Option<&[&'static str]>)] = &[
199     (None, "bootstrap", None),
200     (Some(Mode::Rustc), "parallel_compiler", None),
201     (Some(Mode::ToolRustc), "parallel_compiler", None),
202     (Some(Mode::Codegen), "parallel_compiler", None),
203     (Some(Mode::Std), "stdarch_intel_sde", None),
204     (Some(Mode::Std), "no_fp_fmt_parse", None),
205     (Some(Mode::Std), "no_global_oom_handling", None),
206     (Some(Mode::Std), "no_rc", None),
207     (Some(Mode::Std), "no_sync", None),
208     (Some(Mode::Std), "freebsd12", None),
209     (Some(Mode::Std), "backtrace_in_libstd", None),
210     /* Extra values not defined in the built-in targets yet, but used in std */
211     (Some(Mode::Std), "target_env", Some(&["libnx"])),
212     (Some(Mode::Std), "target_os", Some(&["watchos"])),
213     (
214         Some(Mode::Std),
215         "target_arch",
216         Some(&["asmjs", "spirv", "nvptx", "nvptx64", "le32", "xtensa"]),
217     ),
218     /* Extra names used by dependencies */
219     // FIXME: Used by rustfmt is their test but is invalid (neither cargo nor bootstrap ever set
220     // this config) should probably by removed or use a allow attribute.
221     (Some(Mode::ToolRustc), "release", None),
222     // FIXME: Used by stdarch in their test, should use a allow attribute instead.
223     (Some(Mode::Std), "dont_compile_me", None),
224     // FIXME: Used by serde_json, but we should not be triggering on external dependencies.
225     (Some(Mode::Rustc), "no_btreemap_remove_entry", None),
226     (Some(Mode::ToolRustc), "no_btreemap_remove_entry", None),
227     // FIXME: Used by crossbeam-utils, but we should not be triggering on external dependencies.
228     (Some(Mode::Rustc), "crossbeam_loom", None),
229     (Some(Mode::ToolRustc), "crossbeam_loom", None),
230     // FIXME: Used by proc-macro2, but we should not be triggering on external dependencies.
231     (Some(Mode::Rustc), "span_locations", None),
232     (Some(Mode::ToolRustc), "span_locations", None),
233     // Can be passed in RUSTFLAGS to prevent direct syscalls in rustix.
234     (None, "rustix_use_libc", None),
235 ];
236
237 /// A structure representing a Rust compiler.
238 ///
239 /// Each compiler has a `stage` that it is associated with and a `host` that
240 /// corresponds to the platform the compiler runs on. This structure is used as
241 /// a parameter to many methods below.
242 #[derive(Eq, PartialOrd, Ord, PartialEq, Clone, Copy, Hash, Debug)]
243 pub struct Compiler {
244     stage: u32,
245     host: TargetSelection,
246 }
247
248 #[derive(PartialEq, Eq, Copy, Clone, Debug)]
249 pub enum DocTests {
250     /// Run normal tests and doc tests (default).
251     Yes,
252     /// Do not run any doc tests.
253     No,
254     /// Only run doc tests.
255     Only,
256 }
257
258 pub enum GitRepo {
259     Rustc,
260     Llvm,
261 }
262
263 /// Global configuration for the build system.
264 ///
265 /// This structure transitively contains all configuration for the build system.
266 /// All filesystem-encoded configuration is in `config`, all flags are in
267 /// `flags`, and then parsed or probed information is listed in the keys below.
268 ///
269 /// This structure is a parameter of almost all methods in the build system,
270 /// although most functions are implemented as free functions rather than
271 /// methods specifically on this structure itself (to make it easier to
272 /// organize).
273 pub struct Build {
274     /// User-specified configuration from `config.toml`.
275     config: Config,
276
277     // Version information
278     version: String,
279
280     // Properties derived from the above configuration
281     src: PathBuf,
282     out: PathBuf,
283     bootstrap_out: PathBuf,
284     cargo_info: channel::GitInfo,
285     rust_analyzer_info: channel::GitInfo,
286     clippy_info: channel::GitInfo,
287     miri_info: channel::GitInfo,
288     rustfmt_info: channel::GitInfo,
289     in_tree_llvm_info: channel::GitInfo,
290     local_rebuild: bool,
291     fail_fast: bool,
292     doc_tests: DocTests,
293     verbosity: usize,
294
295     // Targets for which to build
296     build: TargetSelection,
297     hosts: Vec<TargetSelection>,
298     targets: Vec<TargetSelection>,
299
300     initial_rustc: PathBuf,
301     initial_cargo: PathBuf,
302     initial_lld: PathBuf,
303     initial_libdir: PathBuf,
304
305     // Runtime state filled in later on
306     // C/C++ compilers and archiver for all targets
307     cc: HashMap<TargetSelection, cc::Tool>,
308     cxx: HashMap<TargetSelection, cc::Tool>,
309     ar: HashMap<TargetSelection, PathBuf>,
310     ranlib: HashMap<TargetSelection, PathBuf>,
311     // Miscellaneous
312     // allow bidirectional lookups: both name -> path and path -> name
313     crates: HashMap<Interned<String>, Crate>,
314     crate_paths: HashMap<PathBuf, Interned<String>>,
315     is_sudo: bool,
316     ci_env: CiEnv,
317     delayed_failures: RefCell<Vec<String>>,
318     prerelease_version: Cell<Option<u32>>,
319     tool_artifacts:
320         RefCell<HashMap<TargetSelection, HashMap<String, (&'static str, PathBuf, Vec<String>)>>>,
321
322     #[cfg(feature = "build-metrics")]
323     metrics: metrics::BuildMetrics,
324 }
325
326 #[derive(Debug)]
327 struct Crate {
328     name: Interned<String>,
329     deps: HashSet<Interned<String>>,
330     path: PathBuf,
331 }
332
333 impl Crate {
334     fn local_path(&self, build: &Build) -> PathBuf {
335         self.path.strip_prefix(&build.config.src).unwrap().into()
336     }
337 }
338
339 /// When building Rust various objects are handled differently.
340 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
341 pub enum DependencyType {
342     /// Libraries originating from proc-macros.
343     Host,
344     /// Typical Rust libraries.
345     Target,
346     /// Non Rust libraries and objects shipped to ease usage of certain targets.
347     TargetSelfContained,
348 }
349
350 /// The various "modes" of invoking Cargo.
351 ///
352 /// These entries currently correspond to the various output directories of the
353 /// build system, with each mod generating output in a different directory.
354 #[derive(Debug, Hash, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
355 pub enum Mode {
356     /// Build the standard library, placing output in the "stageN-std" directory.
357     Std,
358
359     /// Build librustc, and compiler libraries, placing output in the "stageN-rustc" directory.
360     Rustc,
361
362     /// Build a codegen backend for rustc, placing the output in the "stageN-codegen" directory.
363     Codegen,
364
365     /// Build a tool, placing output in the "stage0-bootstrap-tools"
366     /// directory. This is for miscellaneous sets of tools that are built
367     /// using the bootstrap stage0 compiler in its entirety (target libraries
368     /// and all). Typically these tools compile with stable Rust.
369     ToolBootstrap,
370
371     /// Build a tool which uses the locally built std, placing output in the
372     /// "stageN-tools" directory. Its usage is quite rare, mainly used by
373     /// compiletest which needs libtest.
374     ToolStd,
375
376     /// Build a tool which uses the locally built rustc and the target std,
377     /// placing the output in the "stageN-tools" directory. This is used for
378     /// anything that needs a fully functional rustc, such as rustdoc, clippy,
379     /// cargo, rls, rustfmt, miri, etc.
380     ToolRustc,
381 }
382
383 impl Mode {
384     pub fn is_tool(&self) -> bool {
385         matches!(self, Mode::ToolBootstrap | Mode::ToolRustc | Mode::ToolStd)
386     }
387
388     pub fn must_support_dlopen(&self) -> bool {
389         matches!(self, Mode::Std | Mode::Codegen)
390     }
391 }
392
393 pub enum CLang {
394     C,
395     Cxx,
396 }
397
398 macro_rules! forward {
399     ( $( $fn:ident( $($param:ident: $ty:ty),* ) $( -> $ret:ty)? ),+ $(,)? ) => {
400         impl Build {
401             $( fn $fn(&self, $($param: $ty),* ) $( -> $ret)? {
402                 self.config.$fn( $($param),* )
403             } )+
404         }
405     }
406 }
407
408 forward! {
409     verbose(msg: &str),
410     is_verbose() -> bool,
411     create(path: &Path, s: &str),
412     remove(f: &Path),
413     tempdir() -> PathBuf,
414     try_run(cmd: &mut Command) -> bool,
415     llvm_link_shared() -> bool,
416     download_rustc() -> bool,
417     initial_rustfmt() -> Option<PathBuf>,
418 }
419
420 impl Build {
421     /// Creates a new set of build configuration from the `flags` on the command
422     /// line and the filesystem `config`.
423     ///
424     /// By default all build output will be placed in the current directory.
425     pub fn new(mut config: Config) -> Build {
426         let src = config.src.clone();
427         let out = config.out.clone();
428
429         #[cfg(unix)]
430         // keep this consistent with the equivalent check in x.py:
431         // https://github.com/rust-lang/rust/blob/a8a33cf27166d3eabaffc58ed3799e054af3b0c6/src/bootstrap/bootstrap.py#L796-L797
432         let is_sudo = match env::var_os("SUDO_USER") {
433             Some(_sudo_user) => {
434                 let uid = unsafe { libc::getuid() };
435                 uid == 0
436             }
437             None => false,
438         };
439         #[cfg(not(unix))]
440         let is_sudo = false;
441
442         let ignore_git = config.ignore_git;
443         let rust_info = channel::GitInfo::new(ignore_git, &src);
444         let cargo_info = channel::GitInfo::new(ignore_git, &src.join("src/tools/cargo"));
445         let rust_analyzer_info =
446             channel::GitInfo::new(ignore_git, &src.join("src/tools/rust-analyzer"));
447         let clippy_info = channel::GitInfo::new(ignore_git, &src.join("src/tools/clippy"));
448         let miri_info = channel::GitInfo::new(ignore_git, &src.join("src/tools/miri"));
449         let rustfmt_info = channel::GitInfo::new(ignore_git, &src.join("src/tools/rustfmt"));
450
451         // we always try to use git for LLVM builds
452         let in_tree_llvm_info = channel::GitInfo::new(false, &src.join("src/llvm-project"));
453
454         let initial_target_libdir_str = if config.dry_run() {
455             "/dummy/lib/path/to/lib/".to_string()
456         } else {
457             output(
458                 Command::new(&config.initial_rustc)
459                     .arg("--target")
460                     .arg(config.build.rustc_target_arg())
461                     .arg("--print")
462                     .arg("target-libdir"),
463             )
464         };
465         let initial_target_dir = Path::new(&initial_target_libdir_str).parent().unwrap();
466         let initial_lld = initial_target_dir.join("bin").join("rust-lld");
467
468         let initial_sysroot = if config.dry_run() {
469             "/dummy".to_string()
470         } else {
471             output(Command::new(&config.initial_rustc).arg("--print").arg("sysroot"))
472         };
473         let initial_libdir = initial_target_dir
474             .parent()
475             .unwrap()
476             .parent()
477             .unwrap()
478             .strip_prefix(initial_sysroot.trim())
479             .unwrap()
480             .to_path_buf();
481
482         let version = std::fs::read_to_string(src.join("src").join("version"))
483             .expect("failed to read src/version");
484         let version = version.trim();
485
486         let bootstrap_out = std::env::current_exe()
487             .expect("could not determine path to running process")
488             .parent()
489             .unwrap()
490             .to_path_buf();
491         if !bootstrap_out.join(exe("rustc", config.build)).exists() && !cfg!(test) {
492             // this restriction can be lifted whenever https://github.com/rust-lang/rfcs/pull/3028 is implemented
493             panic!(
494                 "`rustc` not found in {}, run `cargo build --bins` before `cargo run`",
495                 bootstrap_out.display()
496             )
497         }
498
499         if rust_info.is_from_tarball() && config.description.is_none() {
500             config.description = Some("built from a source tarball".to_owned());
501         }
502
503         let mut build = Build {
504             initial_rustc: config.initial_rustc.clone(),
505             initial_cargo: config.initial_cargo.clone(),
506             initial_lld,
507             initial_libdir,
508             local_rebuild: config.local_rebuild,
509             fail_fast: config.cmd.fail_fast(),
510             doc_tests: config.cmd.doc_tests(),
511             verbosity: config.verbose,
512
513             build: config.build,
514             hosts: config.hosts.clone(),
515             targets: config.targets.clone(),
516
517             config,
518             version: version.to_string(),
519             src,
520             out,
521             bootstrap_out,
522
523             cargo_info,
524             rust_analyzer_info,
525             clippy_info,
526             miri_info,
527             rustfmt_info,
528             in_tree_llvm_info,
529             cc: HashMap::new(),
530             cxx: HashMap::new(),
531             ar: HashMap::new(),
532             ranlib: HashMap::new(),
533             crates: HashMap::new(),
534             crate_paths: HashMap::new(),
535             is_sudo,
536             ci_env: CiEnv::current(),
537             delayed_failures: RefCell::new(Vec::new()),
538             prerelease_version: Cell::new(None),
539             tool_artifacts: Default::default(),
540
541             #[cfg(feature = "build-metrics")]
542             metrics: metrics::BuildMetrics::init(),
543         };
544
545         // If local-rust is the same major.minor as the current version, then force a
546         // local-rebuild
547         let local_version_verbose =
548             output(Command::new(&build.initial_rustc).arg("--version").arg("--verbose"));
549         let local_release = local_version_verbose
550             .lines()
551             .filter_map(|x| x.strip_prefix("release:"))
552             .next()
553             .unwrap()
554             .trim();
555         if local_release.split('.').take(2).eq(version.split('.').take(2)) {
556             build.verbose(&format!("auto-detected local-rebuild {}", local_release));
557             build.local_rebuild = true;
558         }
559
560         build.verbose("finding compilers");
561         cc_detect::find(&mut build);
562         // When running `setup`, the profile is about to change, so any requirements we have now may
563         // be different on the next invocation. Don't check for them until the next time x.py is
564         // run. This is ok because `setup` never runs any build commands, so it won't fail if commands are missing.
565         //
566         // Similarly, for `setup` we don't actually need submodules or cargo metadata.
567         if !matches!(build.config.cmd, Subcommand::Setup { .. }) {
568             build.verbose("running sanity check");
569             sanity::check(&mut build);
570
571             // Make sure we update these before gathering metadata so we don't get an error about missing
572             // Cargo.toml files.
573             let rust_submodules = [
574                 "src/tools/rust-installer",
575                 "src/tools/cargo",
576                 "library/backtrace",
577                 "library/stdarch",
578             ];
579             for s in rust_submodules {
580                 build.update_submodule(Path::new(s));
581             }
582             // Now, update all existing submodules.
583             build.update_existing_submodules();
584
585             build.verbose("learning about cargo");
586             metadata::build(&mut build);
587         }
588
589         build
590     }
591
592     // modified from `check_submodule` and `update_submodule` in bootstrap.py
593     /// Given a path to the directory of a submodule, update it.
594     ///
595     /// `relative_path` should be relative to the root of the git repository, not an absolute path.
596     pub(crate) fn update_submodule(&self, relative_path: &Path) {
597         fn dir_is_empty(dir: &Path) -> bool {
598             t!(std::fs::read_dir(dir)).next().is_none()
599         }
600
601         if !self.config.submodules(&self.rust_info()) {
602             return;
603         }
604
605         let absolute_path = self.config.src.join(relative_path);
606
607         // NOTE: The check for the empty directory is here because when running x.py the first time,
608         // the submodule won't be checked out. Check it out now so we can build it.
609         if !channel::GitInfo::new(false, &absolute_path).is_managed_git_subrepository()
610             && !dir_is_empty(&absolute_path)
611         {
612             return;
613         }
614
615         // check_submodule
616         let checked_out_hash =
617             output(Command::new("git").args(&["rev-parse", "HEAD"]).current_dir(&absolute_path));
618         // update_submodules
619         let recorded = output(
620             Command::new("git")
621                 .args(&["ls-tree", "HEAD"])
622                 .arg(relative_path)
623                 .current_dir(&self.config.src),
624         );
625         let actual_hash = recorded
626             .split_whitespace()
627             .nth(2)
628             .unwrap_or_else(|| panic!("unexpected output `{}`", recorded));
629
630         // update_submodule
631         if actual_hash == checked_out_hash.trim_end() {
632             // already checked out
633             return;
634         }
635
636         println!("Updating submodule {}", relative_path.display());
637         self.run(
638             Command::new("git")
639                 .args(&["submodule", "-q", "sync"])
640                 .arg(relative_path)
641                 .current_dir(&self.config.src),
642         );
643
644         // Try passing `--progress` to start, then run git again without if that fails.
645         let update = |progress: bool| {
646             let mut git = Command::new("git");
647             git.args(&["submodule", "update", "--init", "--recursive", "--depth=1"]);
648             if progress {
649                 git.arg("--progress");
650             }
651             git.arg(relative_path).current_dir(&self.config.src);
652             git
653         };
654         // NOTE: doesn't use `try_run` because this shouldn't print an error if it fails.
655         if !update(true).status().map_or(false, |status| status.success()) {
656             self.run(&mut update(false));
657         }
658
659         // Save any local changes, but avoid running `git stash pop` if there are none (since it will exit with an error).
660         let has_local_modifications = !self.try_run(
661             Command::new("git")
662                 .args(&["diff-index", "--quiet", "HEAD"])
663                 .current_dir(&absolute_path),
664         );
665         if has_local_modifications {
666             self.run(Command::new("git").args(&["stash", "push"]).current_dir(&absolute_path));
667         }
668
669         self.run(Command::new("git").args(&["reset", "-q", "--hard"]).current_dir(&absolute_path));
670         self.run(Command::new("git").args(&["clean", "-qdfx"]).current_dir(&absolute_path));
671
672         if has_local_modifications {
673             self.run(Command::new("git").args(&["stash", "pop"]).current_dir(absolute_path));
674         }
675     }
676
677     /// If any submodule has been initialized already, sync it unconditionally.
678     /// This avoids contributors checking in a submodule change by accident.
679     pub fn update_existing_submodules(&self) {
680         // Avoid running git when there isn't a git checkout.
681         if !self.config.submodules(&self.rust_info()) {
682             return;
683         }
684         let output = output(
685             self.config
686                 .git()
687                 .args(&["config", "--file"])
688                 .arg(&self.config.src.join(".gitmodules"))
689                 .args(&["--get-regexp", "path"]),
690         );
691         for line in output.lines() {
692             // Look for `submodule.$name.path = $path`
693             // Sample output: `submodule.src/rust-installer.path src/tools/rust-installer`
694             let submodule = Path::new(line.splitn(2, ' ').nth(1).unwrap());
695             // Don't update the submodule unless it's already been cloned.
696             if channel::GitInfo::new(false, submodule).is_managed_git_subrepository() {
697                 self.update_submodule(submodule);
698             }
699         }
700     }
701
702     /// Executes the entire build, as configured by the flags and configuration.
703     pub fn build(&mut self) {
704         unsafe {
705             job::setup(self);
706         }
707
708         if let Subcommand::Format { check, paths } = &self.config.cmd {
709             return format::format(&builder::Builder::new(&self), *check, &paths);
710         }
711
712         if let Subcommand::Clean { all } = self.config.cmd {
713             return clean::clean(self, all);
714         }
715
716         if let Subcommand::Setup { profile } = &self.config.cmd {
717             return setup::setup(&self.config, *profile);
718         }
719
720         // Download rustfmt early so that it can be used in rust-analyzer configs.
721         let _ = &builder::Builder::new(&self).initial_rustfmt();
722
723         {
724             let builder = builder::Builder::new(&self);
725             if let Some(path) = builder.paths.get(0) {
726                 if path == Path::new("nonexistent/path/to/trigger/cargo/metadata") {
727                     return;
728                 }
729             }
730         }
731
732         if !self.config.dry_run() {
733             {
734                 self.config.dry_run = DryRun::SelfCheck;
735                 let builder = builder::Builder::new(&self);
736                 builder.execute_cli();
737             }
738             self.config.dry_run = DryRun::Disabled;
739             let builder = builder::Builder::new(&self);
740             builder.execute_cli();
741         } else {
742             let builder = builder::Builder::new(&self);
743             builder.execute_cli();
744         }
745
746         // Check for postponed failures from `test --no-fail-fast`.
747         let failures = self.delayed_failures.borrow();
748         if failures.len() > 0 {
749             eprintln!("\n{} command(s) did not execute successfully:\n", failures.len());
750             for failure in failures.iter() {
751                 eprintln!("  - {}\n", failure);
752             }
753             detail_exit(1);
754         }
755
756         #[cfg(feature = "build-metrics")]
757         self.metrics.persist(self);
758     }
759
760     /// Clear out `dir` if `input` is newer.
761     ///
762     /// After this executes, it will also ensure that `dir` exists.
763     fn clear_if_dirty(&self, dir: &Path, input: &Path) -> bool {
764         let stamp = dir.join(".stamp");
765         let mut cleared = false;
766         if mtime(&stamp) < mtime(input) {
767             self.verbose(&format!("Dirty - {}", dir.display()));
768             let _ = fs::remove_dir_all(dir);
769             cleared = true;
770         } else if stamp.exists() {
771             return cleared;
772         }
773         t!(fs::create_dir_all(dir));
774         t!(File::create(stamp));
775         cleared
776     }
777
778     fn rust_info(&self) -> &GitInfo {
779         &self.config.rust_info
780     }
781
782     /// Gets the space-separated set of activated features for the standard
783     /// library.
784     fn std_features(&self, target: TargetSelection) -> String {
785         let mut features = "panic-unwind".to_string();
786
787         match self.config.llvm_libunwind(target) {
788             LlvmLibunwind::InTree => features.push_str(" llvm-libunwind"),
789             LlvmLibunwind::System => features.push_str(" system-llvm-libunwind"),
790             LlvmLibunwind::No => {}
791         }
792         if self.config.backtrace {
793             features.push_str(" backtrace");
794         }
795         if self.config.profiler_enabled(target) {
796             features.push_str(" profiler");
797         }
798         features
799     }
800
801     /// Gets the space-separated set of activated features for the compiler.
802     fn rustc_features(&self, kind: Kind) -> String {
803         let mut features = vec![];
804         if self.config.jemalloc {
805             features.push("jemalloc");
806         }
807         if self.config.llvm_enabled() || kind == Kind::Check {
808             features.push("llvm");
809         }
810         // keep in sync with `bootstrap/compile.rs:rustc_cargo_env`
811         if self.config.rustc_parallel {
812             features.push("rustc_use_parallel_compiler");
813         }
814
815         // If debug logging is on, then we want the default for tracing:
816         // https://github.com/tokio-rs/tracing/blob/3dd5c03d907afdf2c39444a29931833335171554/tracing/src/level_filters.rs#L26
817         // which is everything (including debug/trace/etc.)
818         // if its unset, if debug_assertions is on, then debug_logging will also be on
819         // as well as tracing *ignoring* this feature when debug_assertions is on
820         if !self.config.rust_debug_logging {
821             features.push("max_level_info");
822         }
823
824         features.join(" ")
825     }
826
827     /// Component directory that Cargo will produce output into (e.g.
828     /// release/debug)
829     fn cargo_dir(&self) -> &'static str {
830         if self.config.rust_optimize { "release" } else { "debug" }
831     }
832
833     fn tools_dir(&self, compiler: Compiler) -> PathBuf {
834         let out = self
835             .out
836             .join(&*compiler.host.triple)
837             .join(format!("stage{}-tools-bin", compiler.stage));
838         t!(fs::create_dir_all(&out));
839         out
840     }
841
842     /// Returns the root directory for all output generated in a particular
843     /// stage when running with a particular host compiler.
844     ///
845     /// The mode indicates what the root directory is for.
846     fn stage_out(&self, compiler: Compiler, mode: Mode) -> PathBuf {
847         let suffix = match mode {
848             Mode::Std => "-std",
849             Mode::Rustc => "-rustc",
850             Mode::Codegen => "-codegen",
851             Mode::ToolBootstrap => "-bootstrap-tools",
852             Mode::ToolStd | Mode::ToolRustc => "-tools",
853         };
854         self.out.join(&*compiler.host.triple).join(format!("stage{}{}", compiler.stage, suffix))
855     }
856
857     /// Returns the root output directory for all Cargo output in a given stage,
858     /// running a particular compiler, whether or not we're building the
859     /// standard library, and targeting the specified architecture.
860     fn cargo_out(&self, compiler: Compiler, mode: Mode, target: TargetSelection) -> PathBuf {
861         self.stage_out(compiler, mode).join(&*target.triple).join(self.cargo_dir())
862     }
863
864     /// Root output directory for LLVM compiled for `target`
865     ///
866     /// Note that if LLVM is configured externally then the directory returned
867     /// will likely be empty.
868     fn llvm_out(&self, target: TargetSelection) -> PathBuf {
869         self.out.join(&*target.triple).join("llvm")
870     }
871
872     fn lld_out(&self, target: TargetSelection) -> PathBuf {
873         self.out.join(&*target.triple).join("lld")
874     }
875
876     /// Output directory for all documentation for a target
877     fn doc_out(&self, target: TargetSelection) -> PathBuf {
878         self.out.join(&*target.triple).join("doc")
879     }
880
881     /// Output directory for all JSON-formatted documentation for a target
882     fn json_doc_out(&self, target: TargetSelection) -> PathBuf {
883         self.out.join(&*target.triple).join("json-doc")
884     }
885
886     fn test_out(&self, target: TargetSelection) -> PathBuf {
887         self.out.join(&*target.triple).join("test")
888     }
889
890     /// Output directory for all documentation for a target
891     fn compiler_doc_out(&self, target: TargetSelection) -> PathBuf {
892         self.out.join(&*target.triple).join("compiler-doc")
893     }
894
895     /// Output directory for some generated md crate documentation for a target (temporary)
896     fn md_doc_out(&self, target: TargetSelection) -> Interned<PathBuf> {
897         INTERNER.intern_path(self.out.join(&*target.triple).join("md-doc"))
898     }
899
900     /// Returns `true` if no custom `llvm-config` is set for the specified target.
901     ///
902     /// If no custom `llvm-config` was specified then Rust's llvm will be used.
903     fn is_rust_llvm(&self, target: TargetSelection) -> bool {
904         match self.config.target_config.get(&target) {
905             Some(Target { llvm_has_rust_patches: Some(patched), .. }) => *patched,
906             Some(Target { llvm_config, .. }) => {
907                 // If the user set llvm-config we assume Rust is not patched,
908                 // but first check to see if it was configured by llvm-from-ci.
909                 (self.config.llvm_from_ci && target == self.config.build) || llvm_config.is_none()
910             }
911             None => true,
912         }
913     }
914
915     /// Returns the path to `FileCheck` binary for the specified target
916     fn llvm_filecheck(&self, target: TargetSelection) -> PathBuf {
917         let target_config = self.config.target_config.get(&target);
918         if let Some(s) = target_config.and_then(|c| c.llvm_filecheck.as_ref()) {
919             s.to_path_buf()
920         } else if let Some(s) = target_config.and_then(|c| c.llvm_config.as_ref()) {
921             let llvm_bindir = output(Command::new(s).arg("--bindir"));
922             let filecheck = Path::new(llvm_bindir.trim()).join(exe("FileCheck", target));
923             if filecheck.exists() {
924                 filecheck
925             } else {
926                 // On Fedora the system LLVM installs FileCheck in the
927                 // llvm subdirectory of the libdir.
928                 let llvm_libdir = output(Command::new(s).arg("--libdir"));
929                 let lib_filecheck =
930                     Path::new(llvm_libdir.trim()).join("llvm").join(exe("FileCheck", target));
931                 if lib_filecheck.exists() {
932                     lib_filecheck
933                 } else {
934                     // Return the most normal file name, even though
935                     // it doesn't exist, so that any error message
936                     // refers to that.
937                     filecheck
938                 }
939             }
940         } else {
941             let base = self.llvm_out(target).join("build");
942             let base = if !self.ninja() && target.contains("msvc") {
943                 if self.config.llvm_optimize {
944                     if self.config.llvm_release_debuginfo {
945                         base.join("RelWithDebInfo")
946                     } else {
947                         base.join("Release")
948                     }
949                 } else {
950                     base.join("Debug")
951                 }
952             } else {
953                 base
954             };
955             base.join("bin").join(exe("FileCheck", target))
956         }
957     }
958
959     /// Directory for libraries built from C/C++ code and shared between stages.
960     fn native_dir(&self, target: TargetSelection) -> PathBuf {
961         self.out.join(&*target.triple).join("native")
962     }
963
964     /// Root output directory for rust_test_helpers library compiled for
965     /// `target`
966     fn test_helpers_out(&self, target: TargetSelection) -> PathBuf {
967         self.native_dir(target).join("rust-test-helpers")
968     }
969
970     /// Adds the `RUST_TEST_THREADS` env var if necessary
971     fn add_rust_test_threads(&self, cmd: &mut Command) {
972         if env::var_os("RUST_TEST_THREADS").is_none() {
973             cmd.env("RUST_TEST_THREADS", self.jobs().to_string());
974         }
975     }
976
977     /// Returns the libdir of the snapshot compiler.
978     fn rustc_snapshot_libdir(&self) -> PathBuf {
979         self.rustc_snapshot_sysroot().join(libdir(self.config.build))
980     }
981
982     /// Returns the sysroot of the snapshot compiler.
983     fn rustc_snapshot_sysroot(&self) -> &Path {
984         static SYSROOT_CACHE: OnceCell<PathBuf> = once_cell::sync::OnceCell::new();
985         SYSROOT_CACHE.get_or_init(|| {
986             let mut rustc = Command::new(&self.initial_rustc);
987             rustc.args(&["--print", "sysroot"]);
988             output(&mut rustc).trim().into()
989         })
990     }
991
992     /// Runs a command, printing out nice contextual information if it fails.
993     fn run(&self, cmd: &mut Command) {
994         if self.config.dry_run() {
995             return;
996         }
997         self.verbose(&format!("running: {:?}", cmd));
998         run(cmd, self.is_verbose())
999     }
1000
1001     /// Runs a command, printing out nice contextual information if it fails.
1002     fn run_quiet(&self, cmd: &mut Command) {
1003         if self.config.dry_run() {
1004             return;
1005         }
1006         self.verbose(&format!("running: {:?}", cmd));
1007         run_suppressed(cmd)
1008     }
1009
1010     /// Runs a command, printing out nice contextual information if it fails.
1011     /// Exits if the command failed to execute at all, otherwise returns its
1012     /// `status.success()`.
1013     fn try_run_quiet(&self, cmd: &mut Command) -> bool {
1014         if self.config.dry_run() {
1015             return true;
1016         }
1017         self.verbose(&format!("running: {:?}", cmd));
1018         try_run_suppressed(cmd)
1019     }
1020
1021     pub fn is_verbose_than(&self, level: usize) -> bool {
1022         self.verbosity > level
1023     }
1024
1025     /// Prints a message if this build is configured in more verbose mode than `level`.
1026     fn verbose_than(&self, level: usize, msg: &str) {
1027         if self.is_verbose_than(level) {
1028             println!("{}", msg);
1029         }
1030     }
1031
1032     fn info(&self, msg: &str) {
1033         match self.config.dry_run {
1034             DryRun::SelfCheck => return,
1035             DryRun::Disabled | DryRun::UserSelected => {
1036                 println!("{}", msg);
1037             }
1038         }
1039     }
1040
1041     /// Returns the number of parallel jobs that have been configured for this
1042     /// build.
1043     fn jobs(&self) -> u32 {
1044         self.config.jobs.unwrap_or_else(|| {
1045             std::thread::available_parallelism().map_or(1, std::num::NonZeroUsize::get) as u32
1046         })
1047     }
1048
1049     fn debuginfo_map_to(&self, which: GitRepo) -> Option<String> {
1050         if !self.config.rust_remap_debuginfo {
1051             return None;
1052         }
1053
1054         match which {
1055             GitRepo::Rustc => {
1056                 let sha = self.rust_sha().unwrap_or(&self.version);
1057                 Some(format!("/rustc/{}", sha))
1058             }
1059             GitRepo::Llvm => Some(String::from("/rustc/llvm")),
1060         }
1061     }
1062
1063     /// Returns the path to the C compiler for the target specified.
1064     fn cc(&self, target: TargetSelection) -> &Path {
1065         self.cc[&target].path()
1066     }
1067
1068     /// Returns a list of flags to pass to the C compiler for the target
1069     /// specified.
1070     fn cflags(&self, target: TargetSelection, which: GitRepo, c: CLang) -> Vec<String> {
1071         let base = match c {
1072             CLang::C => &self.cc[&target],
1073             CLang::Cxx => &self.cxx[&target],
1074         };
1075
1076         // Filter out -O and /O (the optimization flags) that we picked up from
1077         // cc-rs because the build scripts will determine that for themselves.
1078         let mut base = base
1079             .args()
1080             .iter()
1081             .map(|s| s.to_string_lossy().into_owned())
1082             .filter(|s| !s.starts_with("-O") && !s.starts_with("/O"))
1083             .collect::<Vec<String>>();
1084
1085         // If we're compiling on macOS then we add a few unconditional flags
1086         // indicating that we want libc++ (more filled out than libstdc++) and
1087         // we want to compile for 10.7. This way we can ensure that
1088         // LLVM/etc are all properly compiled.
1089         if target.contains("apple-darwin") {
1090             base.push("-stdlib=libc++".into());
1091         }
1092
1093         // Work around an apparently bad MinGW / GCC optimization,
1094         // See: https://lists.llvm.org/pipermail/cfe-dev/2016-December/051980.html
1095         // See: https://gcc.gnu.org/bugzilla/show_bug.cgi?id=78936
1096         if &*target.triple == "i686-pc-windows-gnu" {
1097             base.push("-fno-omit-frame-pointer".into());
1098         }
1099
1100         if let Some(map_to) = self.debuginfo_map_to(which) {
1101             let map = format!("{}={}", self.src.display(), map_to);
1102             let cc = self.cc(target);
1103             if cc.ends_with("clang") || cc.ends_with("gcc") {
1104                 base.push(format!("-fdebug-prefix-map={}", map));
1105             } else if cc.ends_with("clang-cl.exe") {
1106                 base.push("-Xclang".into());
1107                 base.push(format!("-fdebug-prefix-map={}", map));
1108             }
1109         }
1110         base
1111     }
1112
1113     /// Returns the path to the `ar` archive utility for the target specified.
1114     fn ar(&self, target: TargetSelection) -> Option<&Path> {
1115         self.ar.get(&target).map(|p| &**p)
1116     }
1117
1118     /// Returns the path to the `ranlib` utility for the target specified.
1119     fn ranlib(&self, target: TargetSelection) -> Option<&Path> {
1120         self.ranlib.get(&target).map(|p| &**p)
1121     }
1122
1123     /// Returns the path to the C++ compiler for the target specified.
1124     fn cxx(&self, target: TargetSelection) -> Result<&Path, String> {
1125         match self.cxx.get(&target) {
1126             Some(p) => Ok(p.path()),
1127             None => {
1128                 Err(format!("target `{}` is not configured as a host, only as a target", target))
1129             }
1130         }
1131     }
1132
1133     /// Returns the path to the linker for the given target if it needs to be overridden.
1134     fn linker(&self, target: TargetSelection) -> Option<&Path> {
1135         if let Some(linker) = self.config.target_config.get(&target).and_then(|c| c.linker.as_ref())
1136         {
1137             Some(linker)
1138         } else if target.contains("vxworks") {
1139             // need to use CXX compiler as linker to resolve the exception functions
1140             // that are only existed in CXX libraries
1141             Some(self.cxx[&target].path())
1142         } else if target != self.config.build
1143             && util::use_host_linker(target)
1144             && !target.contains("msvc")
1145         {
1146             Some(self.cc(target))
1147         } else if self.config.use_lld && !self.is_fuse_ld_lld(target) && self.build == target {
1148             Some(&self.initial_lld)
1149         } else {
1150             None
1151         }
1152     }
1153
1154     // LLD is used through `-fuse-ld=lld` rather than directly.
1155     // Only MSVC targets use LLD directly at the moment.
1156     fn is_fuse_ld_lld(&self, target: TargetSelection) -> bool {
1157         self.config.use_lld && !target.contains("msvc")
1158     }
1159
1160     fn lld_flags(&self, target: TargetSelection) -> impl Iterator<Item = String> {
1161         let mut options = [None, None];
1162
1163         if self.config.use_lld {
1164             if self.is_fuse_ld_lld(target) {
1165                 options[0] = Some("-Clink-arg=-fuse-ld=lld".to_string());
1166             }
1167
1168             let no_threads = util::lld_flag_no_threads(target.contains("windows"));
1169             options[1] = Some(format!("-Clink-arg=-Wl,{}", no_threads));
1170         }
1171
1172         IntoIterator::into_iter(options).flatten()
1173     }
1174
1175     /// Returns if this target should statically link the C runtime, if specified
1176     fn crt_static(&self, target: TargetSelection) -> Option<bool> {
1177         if target.contains("pc-windows-msvc") {
1178             Some(true)
1179         } else {
1180             self.config.target_config.get(&target).and_then(|t| t.crt_static)
1181         }
1182     }
1183
1184     /// Returns the "musl root" for this `target`, if defined
1185     fn musl_root(&self, target: TargetSelection) -> Option<&Path> {
1186         self.config
1187             .target_config
1188             .get(&target)
1189             .and_then(|t| t.musl_root.as_ref())
1190             .or_else(|| self.config.musl_root.as_ref())
1191             .map(|p| &**p)
1192     }
1193
1194     /// Returns the "musl libdir" for this `target`.
1195     fn musl_libdir(&self, target: TargetSelection) -> Option<PathBuf> {
1196         let t = self.config.target_config.get(&target)?;
1197         if let libdir @ Some(_) = &t.musl_libdir {
1198             return libdir.clone();
1199         }
1200         self.musl_root(target).map(|root| root.join("lib"))
1201     }
1202
1203     /// Returns the sysroot for the wasi target, if defined
1204     fn wasi_root(&self, target: TargetSelection) -> Option<&Path> {
1205         self.config.target_config.get(&target).and_then(|t| t.wasi_root.as_ref()).map(|p| &**p)
1206     }
1207
1208     /// Returns `true` if this is a no-std `target`, if defined
1209     fn no_std(&self, target: TargetSelection) -> Option<bool> {
1210         self.config.target_config.get(&target).map(|t| t.no_std)
1211     }
1212
1213     /// Returns `true` if the target will be tested using the `remote-test-client`
1214     /// and `remote-test-server` binaries.
1215     fn remote_tested(&self, target: TargetSelection) -> bool {
1216         self.qemu_rootfs(target).is_some()
1217             || target.contains("android")
1218             || env::var_os("TEST_DEVICE_ADDR").is_some()
1219     }
1220
1221     /// Returns the root of the "rootfs" image that this target will be using,
1222     /// if one was configured.
1223     ///
1224     /// If `Some` is returned then that means that tests for this target are
1225     /// emulated with QEMU and binaries will need to be shipped to the emulator.
1226     fn qemu_rootfs(&self, target: TargetSelection) -> Option<&Path> {
1227         self.config.target_config.get(&target).and_then(|t| t.qemu_rootfs.as_ref()).map(|p| &**p)
1228     }
1229
1230     /// Path to the python interpreter to use
1231     fn python(&self) -> &Path {
1232         if self.config.build.ends_with("apple-darwin") {
1233             // Force /usr/bin/python3 on macOS for LLDB tests because we're loading the
1234             // LLDB plugin's compiled module which only works with the system python
1235             // (namely not Homebrew-installed python)
1236             Path::new("/usr/bin/python3")
1237         } else {
1238             self.config
1239                 .python
1240                 .as_ref()
1241                 .expect("python is required for running LLDB or rustdoc tests")
1242         }
1243     }
1244
1245     /// Temporary directory that extended error information is emitted to.
1246     fn extended_error_dir(&self) -> PathBuf {
1247         self.out.join("tmp/extended-error-metadata")
1248     }
1249
1250     /// Tests whether the `compiler` compiling for `target` should be forced to
1251     /// use a stage1 compiler instead.
1252     ///
1253     /// Currently, by default, the build system does not perform a "full
1254     /// bootstrap" by default where we compile the compiler three times.
1255     /// Instead, we compile the compiler two times. The final stage (stage2)
1256     /// just copies the libraries from the previous stage, which is what this
1257     /// method detects.
1258     ///
1259     /// Here we return `true` if:
1260     ///
1261     /// * The build isn't performing a full bootstrap
1262     /// * The `compiler` is in the final stage, 2
1263     /// * We're not cross-compiling, so the artifacts are already available in
1264     ///   stage1
1265     ///
1266     /// When all of these conditions are met the build will lift artifacts from
1267     /// the previous stage forward.
1268     fn force_use_stage1(&self, compiler: Compiler, target: TargetSelection) -> bool {
1269         !self.config.full_bootstrap
1270             && compiler.stage >= 2
1271             && (self.hosts.iter().any(|h| *h == target) || target == self.build)
1272     }
1273
1274     /// Given `num` in the form "a.b.c" return a "release string" which
1275     /// describes the release version number.
1276     ///
1277     /// For example on nightly this returns "a.b.c-nightly", on beta it returns
1278     /// "a.b.c-beta.1" and on stable it just returns "a.b.c".
1279     fn release(&self, num: &str) -> String {
1280         match &self.config.channel[..] {
1281             "stable" => num.to_string(),
1282             "beta" => {
1283                 if self.rust_info().is_managed_git_subrepository() && !self.config.ignore_git {
1284                     format!("{}-beta.{}", num, self.beta_prerelease_version())
1285                 } else {
1286                     format!("{}-beta", num)
1287                 }
1288             }
1289             "nightly" => format!("{}-nightly", num),
1290             _ => format!("{}-dev", num),
1291         }
1292     }
1293
1294     fn beta_prerelease_version(&self) -> u32 {
1295         if let Some(s) = self.prerelease_version.get() {
1296             return s;
1297         }
1298
1299         // Figure out how many merge commits happened since we branched off master.
1300         // That's our beta number!
1301         // (Note that we use a `..` range, not the `...` symmetric difference.)
1302         let count =
1303             output(self.config.git().arg("rev-list").arg("--count").arg("--merges").arg(format!(
1304                 "refs/remotes/origin/{}..HEAD",
1305                 self.config.stage0_metadata.config.nightly_branch
1306             )));
1307         let n = count.trim().parse().unwrap();
1308         self.prerelease_version.set(Some(n));
1309         n
1310     }
1311
1312     /// Returns the value of `release` above for Rust itself.
1313     fn rust_release(&self) -> String {
1314         self.release(&self.version)
1315     }
1316
1317     /// Returns the "package version" for a component given the `num` release
1318     /// number.
1319     ///
1320     /// The package version is typically what shows up in the names of tarballs.
1321     /// For channels like beta/nightly it's just the channel name, otherwise
1322     /// it's the `num` provided.
1323     fn package_vers(&self, num: &str) -> String {
1324         match &self.config.channel[..] {
1325             "stable" => num.to_string(),
1326             "beta" => "beta".to_string(),
1327             "nightly" => "nightly".to_string(),
1328             _ => format!("{}-dev", num),
1329         }
1330     }
1331
1332     /// Returns the value of `package_vers` above for Rust itself.
1333     fn rust_package_vers(&self) -> String {
1334         self.package_vers(&self.version)
1335     }
1336
1337     /// Returns the `version` string associated with this compiler for Rust
1338     /// itself.
1339     ///
1340     /// Note that this is a descriptive string which includes the commit date,
1341     /// sha, version, etc.
1342     fn rust_version(&self) -> String {
1343         let mut version = self.rust_info().version(self, &self.version);
1344         if let Some(ref s) = self.config.description {
1345             version.push_str(" (");
1346             version.push_str(s);
1347             version.push(')');
1348         }
1349         version
1350     }
1351
1352     /// Returns the full commit hash.
1353     fn rust_sha(&self) -> Option<&str> {
1354         self.rust_info().sha()
1355     }
1356
1357     /// Returns the `a.b.c` version that the given package is at.
1358     fn release_num(&self, package: &str) -> String {
1359         let toml_file_name = self.src.join(&format!("src/tools/{}/Cargo.toml", package));
1360         let toml = t!(fs::read_to_string(&toml_file_name));
1361         for line in toml.lines() {
1362             if let Some(stripped) =
1363                 line.strip_prefix("version = \"").and_then(|s| s.strip_suffix("\""))
1364             {
1365                 return stripped.to_owned();
1366             }
1367         }
1368
1369         panic!("failed to find version in {}'s Cargo.toml", package)
1370     }
1371
1372     /// Returns `true` if unstable features should be enabled for the compiler
1373     /// we're building.
1374     fn unstable_features(&self) -> bool {
1375         match &self.config.channel[..] {
1376             "stable" | "beta" => false,
1377             "nightly" | _ => true,
1378         }
1379     }
1380
1381     /// Returns a Vec of all the dependencies of the given root crate,
1382     /// including transitive dependencies and the root itself. Only includes
1383     /// "local" crates (those in the local source tree, not from a registry).
1384     fn in_tree_crates(&self, root: &str, target: Option<TargetSelection>) -> Vec<&Crate> {
1385         let mut ret = Vec::new();
1386         let mut list = vec![INTERNER.intern_str(root)];
1387         let mut visited = HashSet::new();
1388         while let Some(krate) = list.pop() {
1389             let krate = &self.crates[&krate];
1390             ret.push(krate);
1391             for dep in &krate.deps {
1392                 if !self.crates.contains_key(dep) {
1393                     // Ignore non-workspace members.
1394                     continue;
1395                 }
1396                 // Don't include optional deps if their features are not
1397                 // enabled. Ideally this would be computed from `cargo
1398                 // metadata --features â€¦`, but that is somewhat slow. In
1399                 // the future, we may want to consider just filtering all
1400                 // build and dev dependencies in metadata::build.
1401                 if visited.insert(dep)
1402                     && (dep != "profiler_builtins"
1403                         || target
1404                             .map(|t| self.config.profiler_enabled(t))
1405                             .unwrap_or_else(|| self.config.any_profiler_enabled()))
1406                     && (dep != "rustc_codegen_llvm" || self.config.llvm_enabled())
1407                 {
1408                     list.push(*dep);
1409                 }
1410             }
1411         }
1412         ret
1413     }
1414
1415     fn read_stamp_file(&self, stamp: &Path) -> Vec<(PathBuf, DependencyType)> {
1416         if self.config.dry_run() {
1417             return Vec::new();
1418         }
1419
1420         let mut paths = Vec::new();
1421         let contents = t!(fs::read(stamp), &stamp);
1422         // This is the method we use for extracting paths from the stamp file passed to us. See
1423         // run_cargo for more information (in compile.rs).
1424         for part in contents.split(|b| *b == 0) {
1425             if part.is_empty() {
1426                 continue;
1427             }
1428             let dependency_type = match part[0] as char {
1429                 'h' => DependencyType::Host,
1430                 's' => DependencyType::TargetSelfContained,
1431                 't' => DependencyType::Target,
1432                 _ => unreachable!(),
1433             };
1434             let path = PathBuf::from(t!(str::from_utf8(&part[1..])));
1435             paths.push((path, dependency_type));
1436         }
1437         paths
1438     }
1439
1440     /// Copies a file from `src` to `dst`
1441     pub fn copy(&self, src: &Path, dst: &Path) {
1442         self.copy_internal(src, dst, false);
1443     }
1444
1445     fn copy_internal(&self, src: &Path, dst: &Path, dereference_symlinks: bool) {
1446         if self.config.dry_run() {
1447             return;
1448         }
1449         self.verbose_than(1, &format!("Copy {:?} to {:?}", src, dst));
1450         if src == dst {
1451             return;
1452         }
1453         let _ = fs::remove_file(&dst);
1454         let metadata = t!(src.symlink_metadata());
1455         let mut src = src.to_path_buf();
1456         if metadata.file_type().is_symlink() {
1457             if dereference_symlinks {
1458                 src = t!(fs::canonicalize(src));
1459             } else {
1460                 let link = t!(fs::read_link(src));
1461                 t!(self.symlink_file(link, dst));
1462                 return;
1463             }
1464         }
1465         if let Ok(()) = fs::hard_link(&src, dst) {
1466             // Attempt to "easy copy" by creating a hard link
1467             // (symlinks don't work on windows), but if that fails
1468             // just fall back to a slow `copy` operation.
1469         } else {
1470             if let Err(e) = fs::copy(&src, dst) {
1471                 panic!("failed to copy `{}` to `{}`: {}", src.display(), dst.display(), e)
1472             }
1473             t!(fs::set_permissions(dst, metadata.permissions()));
1474             let atime = FileTime::from_last_access_time(&metadata);
1475             let mtime = FileTime::from_last_modification_time(&metadata);
1476             t!(filetime::set_file_times(dst, atime, mtime));
1477         }
1478     }
1479
1480     /// Copies the `src` directory recursively to `dst`. Both are assumed to exist
1481     /// when this function is called.
1482     pub fn cp_r(&self, src: &Path, dst: &Path) {
1483         if self.config.dry_run() {
1484             return;
1485         }
1486         for f in self.read_dir(src) {
1487             let path = f.path();
1488             let name = path.file_name().unwrap();
1489             let dst = dst.join(name);
1490             if t!(f.file_type()).is_dir() {
1491                 t!(fs::create_dir_all(&dst));
1492                 self.cp_r(&path, &dst);
1493             } else {
1494                 let _ = fs::remove_file(&dst);
1495                 self.copy(&path, &dst);
1496             }
1497         }
1498     }
1499
1500     /// Copies the `src` directory recursively to `dst`. Both are assumed to exist
1501     /// when this function is called. Unwanted files or directories can be skipped
1502     /// by returning `false` from the filter function.
1503     pub fn cp_filtered(&self, src: &Path, dst: &Path, filter: &dyn Fn(&Path) -> bool) {
1504         // Immediately recurse with an empty relative path
1505         self.recurse_(src, dst, Path::new(""), filter)
1506     }
1507
1508     // Inner function does the actual work
1509     fn recurse_(&self, src: &Path, dst: &Path, relative: &Path, filter: &dyn Fn(&Path) -> bool) {
1510         for f in self.read_dir(src) {
1511             let path = f.path();
1512             let name = path.file_name().unwrap();
1513             let dst = dst.join(name);
1514             let relative = relative.join(name);
1515             // Only copy file or directory if the filter function returns true
1516             if filter(&relative) {
1517                 if t!(f.file_type()).is_dir() {
1518                     let _ = fs::remove_dir_all(&dst);
1519                     self.create_dir(&dst);
1520                     self.recurse_(&path, &dst, &relative, filter);
1521                 } else {
1522                     let _ = fs::remove_file(&dst);
1523                     self.copy(&path, &dst);
1524                 }
1525             }
1526         }
1527     }
1528
1529     fn copy_to_folder(&self, src: &Path, dest_folder: &Path) {
1530         let file_name = src.file_name().unwrap();
1531         let dest = dest_folder.join(file_name);
1532         self.copy(src, &dest);
1533     }
1534
1535     fn install(&self, src: &Path, dstdir: &Path, perms: u32) {
1536         if self.config.dry_run() {
1537             return;
1538         }
1539         let dst = dstdir.join(src.file_name().unwrap());
1540         self.verbose_than(1, &format!("Install {:?} to {:?}", src, dst));
1541         t!(fs::create_dir_all(dstdir));
1542         if !src.exists() {
1543             panic!("Error: File \"{}\" not found!", src.display());
1544         }
1545         self.copy_internal(src, &dst, true);
1546         chmod(&dst, perms);
1547     }
1548
1549     fn read(&self, path: &Path) -> String {
1550         if self.config.dry_run() {
1551             return String::new();
1552         }
1553         t!(fs::read_to_string(path))
1554     }
1555
1556     fn create_dir(&self, dir: &Path) {
1557         if self.config.dry_run() {
1558             return;
1559         }
1560         t!(fs::create_dir_all(dir))
1561     }
1562
1563     fn remove_dir(&self, dir: &Path) {
1564         if self.config.dry_run() {
1565             return;
1566         }
1567         t!(fs::remove_dir_all(dir))
1568     }
1569
1570     fn read_dir(&self, dir: &Path) -> impl Iterator<Item = fs::DirEntry> {
1571         let iter = match fs::read_dir(dir) {
1572             Ok(v) => v,
1573             Err(_) if self.config.dry_run() => return vec![].into_iter(),
1574             Err(err) => panic!("could not read dir {:?}: {:?}", dir, err),
1575         };
1576         iter.map(|e| t!(e)).collect::<Vec<_>>().into_iter()
1577     }
1578
1579     fn symlink_file<P: AsRef<Path>, Q: AsRef<Path>>(&self, src: P, link: Q) -> io::Result<()> {
1580         #[cfg(unix)]
1581         use std::os::unix::fs::symlink as symlink_file;
1582         #[cfg(windows)]
1583         use std::os::windows::fs::symlink_file;
1584         if !self.config.dry_run() { symlink_file(src.as_ref(), link.as_ref()) } else { Ok(()) }
1585     }
1586
1587     /// Returns if config.ninja is enabled, and checks for ninja existence,
1588     /// exiting with a nicer error message if not.
1589     fn ninja(&self) -> bool {
1590         let mut cmd_finder = crate::sanity::Finder::new();
1591
1592         if self.config.ninja_in_file {
1593             // Some Linux distros rename `ninja` to `ninja-build`.
1594             // CMake can work with either binary name.
1595             if cmd_finder.maybe_have("ninja-build").is_none()
1596                 && cmd_finder.maybe_have("ninja").is_none()
1597             {
1598                 eprintln!(
1599                     "
1600 Couldn't find required command: ninja (or ninja-build)
1601
1602 You should install ninja as described at
1603 <https://github.com/ninja-build/ninja/wiki/Pre-built-Ninja-packages>,
1604 or set `ninja = false` in the `[llvm]` section of `config.toml`.
1605 Alternatively, set `download-ci-llvm = true` in that `[llvm]` section
1606 to download LLVM rather than building it.
1607 "
1608                 );
1609                 detail_exit(1);
1610             }
1611         }
1612
1613         // If ninja isn't enabled but we're building for MSVC then we try
1614         // doubly hard to enable it. It was realized in #43767 that the msbuild
1615         // CMake generator for MSVC doesn't respect configuration options like
1616         // disabling LLVM assertions, which can often be quite important!
1617         //
1618         // In these cases we automatically enable Ninja if we find it in the
1619         // environment.
1620         if !self.config.ninja_in_file && self.config.build.contains("msvc") {
1621             if cmd_finder.maybe_have("ninja").is_some() {
1622                 return true;
1623             }
1624         }
1625
1626         self.config.ninja_in_file
1627     }
1628 }
1629
1630 #[cfg(unix)]
1631 fn chmod(path: &Path, perms: u32) {
1632     use std::os::unix::fs::*;
1633     t!(fs::set_permissions(path, fs::Permissions::from_mode(perms)));
1634 }
1635 #[cfg(windows)]
1636 fn chmod(_path: &Path, _perms: u32) {}
1637
1638 /// If code is not 0 (successful exit status), exit status is 101 (rust's default error code.)
1639 /// If the test is running and code is an error code, it will cause a panic.
1640 fn detail_exit(code: i32) -> ! {
1641     // if in test and code is an error code, panic with status code provided
1642     if cfg!(test) && code != 0 {
1643         panic!("status code: {}", code);
1644     } else {
1645         //otherwise,exit with provided status code
1646         std::process::exit(code);
1647     }
1648 }
1649
1650 impl Compiler {
1651     pub fn with_stage(mut self, stage: u32) -> Compiler {
1652         self.stage = stage;
1653         self
1654     }
1655
1656     /// Returns `true` if this is a snapshot compiler for `build`'s configuration
1657     pub fn is_snapshot(&self, build: &Build) -> bool {
1658         self.stage == 0 && self.host == build.build
1659     }
1660
1661     /// Returns if this compiler should be treated as a final stage one in the
1662     /// current build session.
1663     /// This takes into account whether we're performing a full bootstrap or
1664     /// not; don't directly compare the stage with `2`!
1665     pub fn is_final_stage(&self, build: &Build) -> bool {
1666         let final_stage = if build.config.full_bootstrap { 2 } else { 1 };
1667         self.stage >= final_stage
1668     }
1669 }
1670
1671 fn envify(s: &str) -> String {
1672     s.chars()
1673         .map(|c| match c {
1674             '-' => '_',
1675             c => c,
1676         })
1677         .flat_map(|c| c.to_uppercase())
1678         .collect()
1679 }