]> git.lizzy.rs Git - rust.git/blob - src/bootstrap/compile.rs
Rollup merge of #94482 - cuishuang:master, r=Dylan-DPC
[rust.git] / src / bootstrap / compile.rs
1 //! Implementation of compiling various phases of the compiler and standard
2 //! library.
3 //!
4 //! This module contains some of the real meat in the rustbuild build system
5 //! which is where Cargo is used to compile the standard library, libtest, and
6 //! the compiler. This module is also responsible for assembling the sysroot as it
7 //! goes along from the output of the previous stage.
8
9 use std::borrow::Cow;
10 use std::collections::HashSet;
11 use std::env;
12 use std::fs;
13 use std::io::prelude::*;
14 use std::io::BufReader;
15 use std::path::{Path, PathBuf};
16 use std::process::{exit, Command, Stdio};
17 use std::str;
18
19 use build_helper::{output, t, up_to_date};
20 use serde::Deserialize;
21
22 use crate::builder::Cargo;
23 use crate::builder::{Builder, Kind, RunConfig, ShouldRun, Step};
24 use crate::cache::{Interned, INTERNER};
25 use crate::config::{LlvmLibunwind, TargetSelection};
26 use crate::dist;
27 use crate::native;
28 use crate::tool::SourceType;
29 use crate::util::{exe, is_debug_info, is_dylib, symlink_dir};
30 use crate::LLVM_TOOLS;
31 use crate::{Compiler, DependencyType, GitRepo, Mode};
32
33 #[derive(Debug, PartialOrd, Ord, Copy, Clone, PartialEq, Eq, Hash)]
34 pub struct Std {
35     pub target: TargetSelection,
36     pub compiler: Compiler,
37 }
38
39 impl Step for Std {
40     type Output = ();
41     const DEFAULT: bool = true;
42
43     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
44         // When downloading stage1, the standard library has already been copied to the sysroot, so
45         // there's no need to rebuild it.
46         let download_rustc = run.builder.config.download_rustc;
47         run.all_krates("test").default_condition(!download_rustc)
48     }
49
50     fn make_run(run: RunConfig<'_>) {
51         run.builder.ensure(Std {
52             compiler: run.builder.compiler(run.builder.top_stage, run.build_triple()),
53             target: run.target,
54         });
55     }
56
57     /// Builds the standard library.
58     ///
59     /// This will build the standard library for a particular stage of the build
60     /// using the `compiler` targeting the `target` architecture. The artifacts
61     /// created will also be linked into the sysroot directory.
62     fn run(self, builder: &Builder<'_>) {
63         let target = self.target;
64         let compiler = self.compiler;
65
66         // These artifacts were already copied (in `impl Step for Sysroot`).
67         // Don't recompile them.
68         // NOTE: the ABI of the beta compiler is different from the ABI of the downloaded compiler,
69         // so its artifacts can't be reused.
70         if builder.config.download_rustc && compiler.stage != 0 {
71             return;
72         }
73
74         if builder.config.keep_stage.contains(&compiler.stage)
75             || builder.config.keep_stage_std.contains(&compiler.stage)
76         {
77             builder.info("Warning: Using a potentially old libstd. This may not behave well.");
78             builder.ensure(StdLink { compiler, target_compiler: compiler, target });
79             return;
80         }
81
82         builder.update_submodule(&Path::new("library").join("stdarch"));
83
84         let mut target_deps = builder.ensure(StartupObjects { compiler, target });
85
86         let compiler_to_use = builder.compiler_for(compiler.stage, compiler.host, target);
87         if compiler_to_use != compiler {
88             builder.ensure(Std { compiler: compiler_to_use, target });
89             builder.info(&format!("Uplifting stage1 std ({} -> {})", compiler_to_use.host, target));
90
91             // Even if we're not building std this stage, the new sysroot must
92             // still contain the third party objects needed by various targets.
93             copy_third_party_objects(builder, &compiler, target);
94             copy_self_contained_objects(builder, &compiler, target);
95
96             builder.ensure(StdLink {
97                 compiler: compiler_to_use,
98                 target_compiler: compiler,
99                 target,
100             });
101             return;
102         }
103
104         target_deps.extend(copy_third_party_objects(builder, &compiler, target));
105         target_deps.extend(copy_self_contained_objects(builder, &compiler, target));
106
107         let mut cargo = builder.cargo(compiler, Mode::Std, SourceType::InTree, target, "build");
108         std_cargo(builder, target, compiler.stage, &mut cargo);
109
110         builder.info(&format!(
111             "Building stage{} std artifacts ({} -> {})",
112             compiler.stage, &compiler.host, target
113         ));
114         run_cargo(
115             builder,
116             cargo,
117             vec![],
118             &libstd_stamp(builder, compiler, target),
119             target_deps,
120             false,
121         );
122
123         builder.ensure(StdLink {
124             compiler: builder.compiler(compiler.stage, builder.config.build),
125             target_compiler: compiler,
126             target,
127         });
128     }
129 }
130
131 fn copy_and_stamp(
132     builder: &Builder<'_>,
133     libdir: &Path,
134     sourcedir: &Path,
135     name: &str,
136     target_deps: &mut Vec<(PathBuf, DependencyType)>,
137     dependency_type: DependencyType,
138 ) {
139     let target = libdir.join(name);
140     builder.copy(&sourcedir.join(name), &target);
141
142     target_deps.push((target, dependency_type));
143 }
144
145 fn copy_llvm_libunwind(builder: &Builder<'_>, target: TargetSelection, libdir: &Path) -> PathBuf {
146     let libunwind_path = builder.ensure(native::Libunwind { target });
147     let libunwind_source = libunwind_path.join("libunwind.a");
148     let libunwind_target = libdir.join("libunwind.a");
149     builder.copy(&libunwind_source, &libunwind_target);
150     libunwind_target
151 }
152
153 /// Copies third party objects needed by various targets.
154 fn copy_third_party_objects(
155     builder: &Builder<'_>,
156     compiler: &Compiler,
157     target: TargetSelection,
158 ) -> Vec<(PathBuf, DependencyType)> {
159     let mut target_deps = vec![];
160
161     // FIXME: remove this in 2021
162     if target == "x86_64-fortanix-unknown-sgx" {
163         if env::var_os("X86_FORTANIX_SGX_LIBS").is_some() {
164             builder.info("Warning: X86_FORTANIX_SGX_LIBS environment variable is ignored, libunwind is now compiled as part of rustbuild");
165         }
166     }
167
168     if builder.config.sanitizers_enabled(target) && compiler.stage != 0 {
169         // The sanitizers are only copied in stage1 or above,
170         // to avoid creating dependency on LLVM.
171         target_deps.extend(
172             copy_sanitizers(builder, &compiler, target)
173                 .into_iter()
174                 .map(|d| (d, DependencyType::Target)),
175         );
176     }
177
178     if target == "x86_64-fortanix-unknown-sgx"
179         || builder.config.llvm_libunwind == LlvmLibunwind::InTree
180             && (target.contains("linux") || target.contains("fuchsia"))
181     {
182         let libunwind_path =
183             copy_llvm_libunwind(builder, target, &builder.sysroot_libdir(*compiler, target));
184         target_deps.push((libunwind_path, DependencyType::Target));
185     }
186
187     target_deps
188 }
189
190 /// Copies third party objects needed by various targets for self-contained linkage.
191 fn copy_self_contained_objects(
192     builder: &Builder<'_>,
193     compiler: &Compiler,
194     target: TargetSelection,
195 ) -> Vec<(PathBuf, DependencyType)> {
196     let libdir_self_contained = builder.sysroot_libdir(*compiler, target).join("self-contained");
197     t!(fs::create_dir_all(&libdir_self_contained));
198     let mut target_deps = vec![];
199
200     // Copies the libc and CRT objects.
201     //
202     // rustc historically provides a more self-contained installation for musl targets
203     // not requiring the presence of a native musl toolchain. For example, it can fall back
204     // to using gcc from a glibc-targeting toolchain for linking.
205     // To do that we have to distribute musl startup objects as a part of Rust toolchain
206     // and link with them manually in the self-contained mode.
207     if target.contains("musl") {
208         let srcdir = builder.musl_libdir(target).unwrap_or_else(|| {
209             panic!("Target {:?} does not have a \"musl-libdir\" key", target.triple)
210         });
211         for &obj in &["libc.a", "crt1.o", "Scrt1.o", "rcrt1.o", "crti.o", "crtn.o"] {
212             copy_and_stamp(
213                 builder,
214                 &libdir_self_contained,
215                 &srcdir,
216                 obj,
217                 &mut target_deps,
218                 DependencyType::TargetSelfContained,
219             );
220         }
221         let crt_path = builder.ensure(native::CrtBeginEnd { target });
222         for &obj in &["crtbegin.o", "crtbeginS.o", "crtend.o", "crtendS.o"] {
223             let src = crt_path.join(obj);
224             let target = libdir_self_contained.join(obj);
225             builder.copy(&src, &target);
226             target_deps.push((target, DependencyType::TargetSelfContained));
227         }
228
229         if !target.starts_with("s390x") {
230             let libunwind_path = copy_llvm_libunwind(builder, target, &libdir_self_contained);
231             target_deps.push((libunwind_path, DependencyType::TargetSelfContained));
232         }
233     } else if target.ends_with("-wasi") {
234         let srcdir = builder
235             .wasi_root(target)
236             .unwrap_or_else(|| {
237                 panic!("Target {:?} does not have a \"wasi-root\" key", target.triple)
238             })
239             .join("lib/wasm32-wasi");
240         for &obj in &["libc.a", "crt1-command.o", "crt1-reactor.o"] {
241             copy_and_stamp(
242                 builder,
243                 &libdir_self_contained,
244                 &srcdir,
245                 obj,
246                 &mut target_deps,
247                 DependencyType::TargetSelfContained,
248             );
249         }
250     } else if target.contains("windows-gnu") {
251         for obj in ["crt2.o", "dllcrt2.o"].iter() {
252             let src = compiler_file(builder, builder.cc(target), target, obj);
253             let target = libdir_self_contained.join(obj);
254             builder.copy(&src, &target);
255             target_deps.push((target, DependencyType::TargetSelfContained));
256         }
257     }
258
259     target_deps
260 }
261
262 /// Configure cargo to compile the standard library, adding appropriate env vars
263 /// and such.
264 pub fn std_cargo(builder: &Builder<'_>, target: TargetSelection, stage: u32, cargo: &mut Cargo) {
265     if let Some(target) = env::var_os("MACOSX_STD_DEPLOYMENT_TARGET") {
266         cargo.env("MACOSX_DEPLOYMENT_TARGET", target);
267     }
268
269     // Determine if we're going to compile in optimized C intrinsics to
270     // the `compiler-builtins` crate. These intrinsics live in LLVM's
271     // `compiler-rt` repository, but our `src/llvm-project` submodule isn't
272     // always checked out, so we need to conditionally look for this. (e.g. if
273     // an external LLVM is used we skip the LLVM submodule checkout).
274     //
275     // Note that this shouldn't affect the correctness of `compiler-builtins`,
276     // but only its speed. Some intrinsics in C haven't been translated to Rust
277     // yet but that's pretty rare. Other intrinsics have optimized
278     // implementations in C which have only had slower versions ported to Rust,
279     // so we favor the C version where we can, but it's not critical.
280     //
281     // If `compiler-rt` is available ensure that the `c` feature of the
282     // `compiler-builtins` crate is enabled and it's configured to learn where
283     // `compiler-rt` is located.
284     let compiler_builtins_root = builder.src.join("src/llvm-project/compiler-rt");
285     let compiler_builtins_c_feature = if compiler_builtins_root.exists() {
286         // Note that `libprofiler_builtins/build.rs` also computes this so if
287         // you're changing something here please also change that.
288         cargo.env("RUST_COMPILER_RT_ROOT", &compiler_builtins_root);
289         " compiler-builtins-c"
290     } else {
291         ""
292     };
293
294     if builder.no_std(target) == Some(true) {
295         let mut features = "compiler-builtins-mem".to_string();
296         if !target.starts_with("bpf") {
297             features.push_str(compiler_builtins_c_feature);
298         }
299
300         // for no-std targets we only compile a few no_std crates
301         cargo
302             .args(&["-p", "alloc"])
303             .arg("--manifest-path")
304             .arg(builder.src.join("library/alloc/Cargo.toml"))
305             .arg("--features")
306             .arg(features);
307     } else {
308         let mut features = builder.std_features(target);
309         features.push_str(compiler_builtins_c_feature);
310
311         cargo
312             .arg("--features")
313             .arg(features)
314             .arg("--manifest-path")
315             .arg(builder.src.join("library/test/Cargo.toml"));
316
317         // Help the libc crate compile by assisting it in finding various
318         // sysroot native libraries.
319         if target.contains("musl") {
320             if let Some(p) = builder.musl_libdir(target) {
321                 let root = format!("native={}", p.to_str().unwrap());
322                 cargo.rustflag("-L").rustflag(&root);
323             }
324         }
325
326         if target.ends_with("-wasi") {
327             if let Some(p) = builder.wasi_root(target) {
328                 let root = format!("native={}/lib/wasm32-wasi", p.to_str().unwrap());
329                 cargo.rustflag("-L").rustflag(&root);
330             }
331         }
332     }
333
334     // By default, rustc uses `-Cembed-bitcode=yes`, and Cargo overrides that
335     // with `-Cembed-bitcode=no` for non-LTO builds. However, libstd must be
336     // built with bitcode so that the produced rlibs can be used for both LTO
337     // builds (which use bitcode) and non-LTO builds (which use object code).
338     // So we override the override here!
339     //
340     // But we don't bother for the stage 0 compiler because it's never used
341     // with LTO.
342     if stage >= 1 {
343         cargo.rustflag("-Cembed-bitcode=yes");
344     }
345
346     // By default, rustc does not include unwind tables unless they are required
347     // for a particular target. They are not required by RISC-V targets, but
348     // compiling the standard library with them means that users can get
349     // backtraces without having to recompile the standard library themselves.
350     //
351     // This choice was discussed in https://github.com/rust-lang/rust/pull/69890
352     if target.contains("riscv") {
353         cargo.rustflag("-Cforce-unwind-tables=yes");
354     }
355
356     let html_root =
357         format!("-Zcrate-attr=doc(html_root_url=\"{}/\")", builder.doc_rust_lang_org_channel(),);
358     cargo.rustflag(&html_root);
359     cargo.rustdocflag(&html_root);
360 }
361
362 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
363 struct StdLink {
364     pub compiler: Compiler,
365     pub target_compiler: Compiler,
366     pub target: TargetSelection,
367 }
368
369 impl Step for StdLink {
370     type Output = ();
371
372     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
373         run.never()
374     }
375
376     /// Link all libstd rlibs/dylibs into the sysroot location.
377     ///
378     /// Links those artifacts generated by `compiler` to the `stage` compiler's
379     /// sysroot for the specified `host` and `target`.
380     ///
381     /// Note that this assumes that `compiler` has already generated the libstd
382     /// libraries for `target`, and this method will find them in the relevant
383     /// output directory.
384     fn run(self, builder: &Builder<'_>) {
385         let compiler = self.compiler;
386         let target_compiler = self.target_compiler;
387         let target = self.target;
388         builder.info(&format!(
389             "Copying stage{} std from stage{} ({} -> {} / {})",
390             target_compiler.stage, compiler.stage, &compiler.host, target_compiler.host, target
391         ));
392         let libdir = builder.sysroot_libdir(target_compiler, target);
393         let hostdir = builder.sysroot_libdir(target_compiler, compiler.host);
394         add_to_sysroot(builder, &libdir, &hostdir, &libstd_stamp(builder, compiler, target));
395     }
396 }
397
398 /// Copies sanitizer runtime libraries into target libdir.
399 fn copy_sanitizers(
400     builder: &Builder<'_>,
401     compiler: &Compiler,
402     target: TargetSelection,
403 ) -> Vec<PathBuf> {
404     let runtimes: Vec<native::SanitizerRuntime> = builder.ensure(native::Sanitizers { target });
405
406     if builder.config.dry_run {
407         return Vec::new();
408     }
409
410     let mut target_deps = Vec::new();
411     let libdir = builder.sysroot_libdir(*compiler, target);
412
413     for runtime in &runtimes {
414         let dst = libdir.join(&runtime.name);
415         builder.copy(&runtime.path, &dst);
416
417         if target == "x86_64-apple-darwin" || target == "aarch64-apple-darwin" {
418             // Update the library’s install name to reflect that it has has been renamed.
419             apple_darwin_update_library_name(&dst, &format!("@rpath/{}", &runtime.name));
420             // Upon renaming the install name, the code signature of the file will invalidate,
421             // so we will sign it again.
422             apple_darwin_sign_file(&dst);
423         }
424
425         target_deps.push(dst);
426     }
427
428     target_deps
429 }
430
431 fn apple_darwin_update_library_name(library_path: &Path, new_name: &str) {
432     let status = Command::new("install_name_tool")
433         .arg("-id")
434         .arg(new_name)
435         .arg(library_path)
436         .status()
437         .expect("failed to execute `install_name_tool`");
438     assert!(status.success());
439 }
440
441 fn apple_darwin_sign_file(file_path: &Path) {
442     let status = Command::new("codesign")
443         .arg("-f") // Force to rewrite the existing signature
444         .arg("-s")
445         .arg("-")
446         .arg(file_path)
447         .status()
448         .expect("failed to execute `codesign`");
449     assert!(status.success());
450 }
451
452 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
453 pub struct StartupObjects {
454     pub compiler: Compiler,
455     pub target: TargetSelection,
456 }
457
458 impl Step for StartupObjects {
459     type Output = Vec<(PathBuf, DependencyType)>;
460
461     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
462         run.path("library/rtstartup")
463     }
464
465     fn make_run(run: RunConfig<'_>) {
466         run.builder.ensure(StartupObjects {
467             compiler: run.builder.compiler(run.builder.top_stage, run.build_triple()),
468             target: run.target,
469         });
470     }
471
472     /// Builds and prepare startup objects like rsbegin.o and rsend.o
473     ///
474     /// These are primarily used on Windows right now for linking executables/dlls.
475     /// They don't require any library support as they're just plain old object
476     /// files, so we just use the nightly snapshot compiler to always build them (as
477     /// no other compilers are guaranteed to be available).
478     fn run(self, builder: &Builder<'_>) -> Vec<(PathBuf, DependencyType)> {
479         let for_compiler = self.compiler;
480         let target = self.target;
481         if !target.contains("windows-gnu") {
482             return vec![];
483         }
484
485         let mut target_deps = vec![];
486
487         let src_dir = &builder.src.join("library").join("rtstartup");
488         let dst_dir = &builder.native_dir(target).join("rtstartup");
489         let sysroot_dir = &builder.sysroot_libdir(for_compiler, target);
490         t!(fs::create_dir_all(dst_dir));
491
492         for file in &["rsbegin", "rsend"] {
493             let src_file = &src_dir.join(file.to_string() + ".rs");
494             let dst_file = &dst_dir.join(file.to_string() + ".o");
495             if !up_to_date(src_file, dst_file) {
496                 let mut cmd = Command::new(&builder.initial_rustc);
497                 cmd.env("RUSTC_BOOTSTRAP", "1");
498                 if !builder.local_rebuild {
499                     // a local_rebuild compiler already has stage1 features
500                     cmd.arg("--cfg").arg("bootstrap");
501                 }
502                 builder.run(
503                     cmd.arg("--target")
504                         .arg(target.rustc_target_arg())
505                         .arg("--emit=obj")
506                         .arg("-o")
507                         .arg(dst_file)
508                         .arg(src_file),
509                 );
510             }
511
512             let target = sysroot_dir.join((*file).to_string() + ".o");
513             builder.copy(dst_file, &target);
514             target_deps.push((target, DependencyType::Target));
515         }
516
517         target_deps
518     }
519 }
520
521 #[derive(Debug, PartialOrd, Ord, Copy, Clone, PartialEq, Eq, Hash)]
522 pub struct Rustc {
523     pub target: TargetSelection,
524     pub compiler: Compiler,
525 }
526
527 impl Step for Rustc {
528     type Output = ();
529     const ONLY_HOSTS: bool = true;
530     const DEFAULT: bool = false;
531
532     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
533         run.never()
534     }
535
536     fn make_run(run: RunConfig<'_>) {
537         run.builder.ensure(Rustc {
538             compiler: run.builder.compiler(run.builder.top_stage, run.build_triple()),
539             target: run.target,
540         });
541     }
542
543     /// Builds the compiler.
544     ///
545     /// This will build the compiler for a particular stage of the build using
546     /// the `compiler` targeting the `target` architecture. The artifacts
547     /// created will also be linked into the sysroot directory.
548     fn run(self, builder: &Builder<'_>) {
549         let compiler = self.compiler;
550         let target = self.target;
551
552         // NOTE: the ABI of the beta compiler is different from the ABI of the downloaded compiler,
553         // so its artifacts can't be reused.
554         if builder.config.download_rustc && compiler.stage != 0 {
555             // Copy the existing artifacts instead of rebuilding them.
556             // NOTE: this path is only taken for tools linking to rustc-dev.
557             builder.ensure(Sysroot { compiler });
558             return;
559         }
560
561         builder.ensure(Std { compiler, target });
562
563         if builder.config.keep_stage.contains(&compiler.stage) {
564             builder.info("Warning: Using a potentially old librustc. This may not behave well.");
565             builder.info("Warning: Use `--keep-stage-std` if you want to rebuild the compiler when it changes");
566             builder.ensure(RustcLink { compiler, target_compiler: compiler, target });
567             return;
568         }
569
570         let compiler_to_use = builder.compiler_for(compiler.stage, compiler.host, target);
571         if compiler_to_use != compiler {
572             builder.ensure(Rustc { compiler: compiler_to_use, target });
573             builder
574                 .info(&format!("Uplifting stage1 rustc ({} -> {})", builder.config.build, target));
575             builder.ensure(RustcLink {
576                 compiler: compiler_to_use,
577                 target_compiler: compiler,
578                 target,
579             });
580             return;
581         }
582
583         // Ensure that build scripts and proc macros have a std / libproc_macro to link against.
584         builder.ensure(Std {
585             compiler: builder.compiler(self.compiler.stage, builder.config.build),
586             target: builder.config.build,
587         });
588
589         let mut cargo = builder.cargo(compiler, Mode::Rustc, SourceType::InTree, target, "build");
590         rustc_cargo(builder, &mut cargo, target);
591
592         if builder.config.rust_profile_use.is_some()
593             && builder.config.rust_profile_generate.is_some()
594         {
595             panic!("Cannot use and generate PGO profiles at the same time");
596         }
597
598         let is_collecting = if let Some(path) = &builder.config.rust_profile_generate {
599             if compiler.stage == 1 {
600                 cargo.rustflag(&format!("-Cprofile-generate={}", path));
601                 // Apparently necessary to avoid overflowing the counters during
602                 // a Cargo build profile
603                 cargo.rustflag("-Cllvm-args=-vp-counters-per-site=4");
604                 true
605             } else {
606                 false
607             }
608         } else if let Some(path) = &builder.config.rust_profile_use {
609             if compiler.stage == 1 {
610                 cargo.rustflag(&format!("-Cprofile-use={}", path));
611                 cargo.rustflag("-Cllvm-args=-pgo-warn-missing-function");
612                 true
613             } else {
614                 false
615             }
616         } else {
617             false
618         };
619         if is_collecting {
620             // Ensure paths to Rust sources are relative, not absolute.
621             cargo.rustflag(&format!(
622                 "-Cllvm-args=-static-func-strip-dirname-prefix={}",
623                 builder.config.src.components().count()
624             ));
625         }
626
627         builder.info(&format!(
628             "Building stage{} compiler artifacts ({} -> {})",
629             compiler.stage, &compiler.host, target
630         ));
631         run_cargo(
632             builder,
633             cargo,
634             vec![],
635             &librustc_stamp(builder, compiler, target),
636             vec![],
637             false,
638         );
639
640         builder.ensure(RustcLink {
641             compiler: builder.compiler(compiler.stage, builder.config.build),
642             target_compiler: compiler,
643             target,
644         });
645     }
646 }
647
648 pub fn rustc_cargo(builder: &Builder<'_>, cargo: &mut Cargo, target: TargetSelection) {
649     cargo
650         .arg("--features")
651         .arg(builder.rustc_features(builder.kind))
652         .arg("--manifest-path")
653         .arg(builder.src.join("compiler/rustc/Cargo.toml"));
654     rustc_cargo_env(builder, cargo, target);
655 }
656
657 pub fn rustc_cargo_env(builder: &Builder<'_>, cargo: &mut Cargo, target: TargetSelection) {
658     // Set some configuration variables picked up by build scripts and
659     // the compiler alike
660     cargo
661         .env("CFG_RELEASE", builder.rust_release())
662         .env("CFG_RELEASE_CHANNEL", &builder.config.channel)
663         .env("CFG_VERSION", builder.rust_version());
664
665     if let Some(backend) = builder.config.rust_codegen_backends.get(0) {
666         cargo.env("CFG_DEFAULT_CODEGEN_BACKEND", backend);
667     }
668
669     let libdir_relative = builder.config.libdir_relative().unwrap_or_else(|| Path::new("lib"));
670     let target_config = builder.config.target_config.get(&target);
671
672     cargo.env("CFG_LIBDIR_RELATIVE", libdir_relative);
673
674     if let Some(ref ver_date) = builder.rust_info.commit_date() {
675         cargo.env("CFG_VER_DATE", ver_date);
676     }
677     if let Some(ref ver_hash) = builder.rust_info.sha() {
678         cargo.env("CFG_VER_HASH", ver_hash);
679     }
680     if !builder.unstable_features() {
681         cargo.env("CFG_DISABLE_UNSTABLE_FEATURES", "1");
682     }
683
684     // Prefer the current target's own default_linker, else a globally
685     // specified one.
686     if let Some(s) = target_config.and_then(|c| c.default_linker.as_ref()) {
687         cargo.env("CFG_DEFAULT_LINKER", s);
688     } else if let Some(ref s) = builder.config.rustc_default_linker {
689         cargo.env("CFG_DEFAULT_LINKER", s);
690     }
691
692     if builder.config.rustc_parallel {
693         cargo.rustflag("--cfg=parallel_compiler");
694         cargo.rustdocflag("--cfg=parallel_compiler");
695     }
696     if builder.config.rust_verify_llvm_ir {
697         cargo.env("RUSTC_VERIFY_LLVM_IR", "1");
698     }
699
700     // Pass down configuration from the LLVM build into the build of
701     // rustc_llvm and rustc_codegen_llvm.
702     //
703     // Note that this is disabled if LLVM itself is disabled or we're in a check
704     // build. If we are in a check build we still go ahead here presuming we've
705     // detected that LLVM is alreay built and good to go which helps prevent
706     // busting caches (e.g. like #71152).
707     if builder.config.llvm_enabled()
708         && (builder.kind != Kind::Check
709             || crate::native::prebuilt_llvm_config(builder, target).is_ok())
710     {
711         if builder.is_rust_llvm(target) {
712             cargo.env("LLVM_RUSTLLVM", "1");
713         }
714         let llvm_config = builder.ensure(native::Llvm { target });
715         cargo.env("LLVM_CONFIG", &llvm_config);
716         if let Some(s) = target_config.and_then(|c| c.llvm_config.as_ref()) {
717             cargo.env("CFG_LLVM_ROOT", s);
718         }
719         // Some LLVM linker flags (-L and -l) may be needed to link rustc_llvm.
720         if let Some(ref s) = builder.config.llvm_ldflags {
721             cargo.env("LLVM_LINKER_FLAGS", s);
722         }
723         // Building with a static libstdc++ is only supported on linux right now,
724         // not for MSVC or macOS
725         if builder.config.llvm_static_stdcpp
726             && !target.contains("freebsd")
727             && !target.contains("msvc")
728             && !target.contains("apple")
729         {
730             let file = compiler_file(builder, builder.cxx(target).unwrap(), target, "libstdc++.a");
731             cargo.env("LLVM_STATIC_STDCPP", file);
732         }
733         if builder.config.llvm_link_shared {
734             cargo.env("LLVM_LINK_SHARED", "1");
735         }
736         if builder.config.llvm_use_libcxx {
737             cargo.env("LLVM_USE_LIBCXX", "1");
738         }
739         if builder.config.llvm_optimize && !builder.config.llvm_release_debuginfo {
740             cargo.env("LLVM_NDEBUG", "1");
741         }
742     }
743 }
744
745 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
746 struct RustcLink {
747     pub compiler: Compiler,
748     pub target_compiler: Compiler,
749     pub target: TargetSelection,
750 }
751
752 impl Step for RustcLink {
753     type Output = ();
754
755     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
756         run.never()
757     }
758
759     /// Same as `std_link`, only for librustc
760     fn run(self, builder: &Builder<'_>) {
761         let compiler = self.compiler;
762         let target_compiler = self.target_compiler;
763         let target = self.target;
764         builder.info(&format!(
765             "Copying stage{} rustc from stage{} ({} -> {} / {})",
766             target_compiler.stage, compiler.stage, &compiler.host, target_compiler.host, target
767         ));
768         add_to_sysroot(
769             builder,
770             &builder.sysroot_libdir(target_compiler, target),
771             &builder.sysroot_libdir(target_compiler, compiler.host),
772             &librustc_stamp(builder, compiler, target),
773         );
774     }
775 }
776
777 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
778 pub struct CodegenBackend {
779     pub target: TargetSelection,
780     pub compiler: Compiler,
781     pub backend: Interned<String>,
782 }
783
784 impl Step for CodegenBackend {
785     type Output = ();
786     const ONLY_HOSTS: bool = true;
787     // Only the backends specified in the `codegen-backends` entry of `config.toml` are built.
788     const DEFAULT: bool = true;
789
790     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
791         run.path("compiler/rustc_codegen_cranelift")
792     }
793
794     fn make_run(run: RunConfig<'_>) {
795         for &backend in &run.builder.config.rust_codegen_backends {
796             if backend == "llvm" {
797                 continue; // Already built as part of rustc
798             }
799
800             run.builder.ensure(CodegenBackend {
801                 target: run.target,
802                 compiler: run.builder.compiler(run.builder.top_stage, run.build_triple()),
803                 backend,
804             });
805         }
806     }
807
808     fn run(self, builder: &Builder<'_>) {
809         let compiler = self.compiler;
810         let target = self.target;
811         let backend = self.backend;
812
813         builder.ensure(Rustc { compiler, target });
814
815         if builder.config.keep_stage.contains(&compiler.stage) {
816             builder.info(
817                 "Warning: Using a potentially old codegen backend. \
818                 This may not behave well.",
819             );
820             // Codegen backends are linked separately from this step today, so we don't do
821             // anything here.
822             return;
823         }
824
825         let compiler_to_use = builder.compiler_for(compiler.stage, compiler.host, target);
826         if compiler_to_use != compiler {
827             builder.ensure(CodegenBackend { compiler: compiler_to_use, target, backend });
828             return;
829         }
830
831         let out_dir = builder.cargo_out(compiler, Mode::Codegen, target);
832
833         let mut cargo = builder.cargo(compiler, Mode::Codegen, SourceType::InTree, target, "build");
834         cargo
835             .arg("--manifest-path")
836             .arg(builder.src.join(format!("compiler/rustc_codegen_{}/Cargo.toml", backend)));
837         rustc_cargo_env(builder, &mut cargo, target);
838
839         let tmp_stamp = out_dir.join(".tmp.stamp");
840
841         builder.info(&format!(
842             "Building stage{} codegen backend {} ({} -> {})",
843             compiler.stage, backend, &compiler.host, target
844         ));
845         let files = run_cargo(builder, cargo, vec![], &tmp_stamp, vec![], false);
846         if builder.config.dry_run {
847             return;
848         }
849         let mut files = files.into_iter().filter(|f| {
850             let filename = f.file_name().unwrap().to_str().unwrap();
851             is_dylib(filename) && filename.contains("rustc_codegen_")
852         });
853         let codegen_backend = match files.next() {
854             Some(f) => f,
855             None => panic!("no dylibs built for codegen backend?"),
856         };
857         if let Some(f) = files.next() {
858             panic!(
859                 "codegen backend built two dylibs:\n{}\n{}",
860                 codegen_backend.display(),
861                 f.display()
862             );
863         }
864         let stamp = codegen_backend_stamp(builder, compiler, target, backend);
865         let codegen_backend = codegen_backend.to_str().unwrap();
866         t!(fs::write(&stamp, &codegen_backend));
867     }
868 }
869
870 /// Creates the `codegen-backends` folder for a compiler that's about to be
871 /// assembled as a complete compiler.
872 ///
873 /// This will take the codegen artifacts produced by `compiler` and link them
874 /// into an appropriate location for `target_compiler` to be a functional
875 /// compiler.
876 fn copy_codegen_backends_to_sysroot(
877     builder: &Builder<'_>,
878     compiler: Compiler,
879     target_compiler: Compiler,
880 ) {
881     let target = target_compiler.host;
882
883     // Note that this step is different than all the other `*Link` steps in
884     // that it's not assembling a bunch of libraries but rather is primarily
885     // moving the codegen backend into place. The codegen backend of rustc is
886     // not linked into the main compiler by default but is rather dynamically
887     // selected at runtime for inclusion.
888     //
889     // Here we're looking for the output dylib of the `CodegenBackend` step and
890     // we're copying that into the `codegen-backends` folder.
891     let dst = builder.sysroot_codegen_backends(target_compiler);
892     t!(fs::create_dir_all(&dst), dst);
893
894     if builder.config.dry_run {
895         return;
896     }
897
898     for backend in builder.config.rust_codegen_backends.iter() {
899         if backend == "llvm" {
900             continue; // Already built as part of rustc
901         }
902
903         let stamp = codegen_backend_stamp(builder, compiler, target, *backend);
904         let dylib = t!(fs::read_to_string(&stamp));
905         let file = Path::new(&dylib);
906         let filename = file.file_name().unwrap().to_str().unwrap();
907         // change `librustc_codegen_cranelift-xxxxxx.so` to
908         // `librustc_codegen_cranelift-release.so`
909         let target_filename = {
910             let dash = filename.find('-').unwrap();
911             let dot = filename.find('.').unwrap();
912             format!("{}-{}{}", &filename[..dash], builder.rust_release(), &filename[dot..])
913         };
914         builder.copy(&file, &dst.join(target_filename));
915     }
916 }
917
918 /// Cargo's output path for the standard library in a given stage, compiled
919 /// by a particular compiler for the specified target.
920 pub fn libstd_stamp(builder: &Builder<'_>, compiler: Compiler, target: TargetSelection) -> PathBuf {
921     builder.cargo_out(compiler, Mode::Std, target).join(".libstd.stamp")
922 }
923
924 /// Cargo's output path for librustc in a given stage, compiled by a particular
925 /// compiler for the specified target.
926 pub fn librustc_stamp(
927     builder: &Builder<'_>,
928     compiler: Compiler,
929     target: TargetSelection,
930 ) -> PathBuf {
931     builder.cargo_out(compiler, Mode::Rustc, target).join(".librustc.stamp")
932 }
933
934 /// Cargo's output path for librustc_codegen_llvm in a given stage, compiled by a particular
935 /// compiler for the specified target and backend.
936 fn codegen_backend_stamp(
937     builder: &Builder<'_>,
938     compiler: Compiler,
939     target: TargetSelection,
940     backend: Interned<String>,
941 ) -> PathBuf {
942     builder
943         .cargo_out(compiler, Mode::Codegen, target)
944         .join(format!(".librustc_codegen_{}.stamp", backend))
945 }
946
947 pub fn compiler_file(
948     builder: &Builder<'_>,
949     compiler: &Path,
950     target: TargetSelection,
951     file: &str,
952 ) -> PathBuf {
953     let mut cmd = Command::new(compiler);
954     cmd.args(builder.cflags(target, GitRepo::Rustc));
955     cmd.arg(format!("-print-file-name={}", file));
956     let out = output(&mut cmd);
957     PathBuf::from(out.trim())
958 }
959
960 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
961 pub struct Sysroot {
962     pub compiler: Compiler,
963 }
964
965 impl Step for Sysroot {
966     type Output = Interned<PathBuf>;
967
968     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
969         run.never()
970     }
971
972     /// Returns the sysroot for the `compiler` specified that *this build system
973     /// generates*.
974     ///
975     /// That is, the sysroot for the stage0 compiler is not what the compiler
976     /// thinks it is by default, but it's the same as the default for stages
977     /// 1-3.
978     fn run(self, builder: &Builder<'_>) -> Interned<PathBuf> {
979         let compiler = self.compiler;
980         let sysroot = if compiler.stage == 0 {
981             builder.out.join(&compiler.host.triple).join("stage0-sysroot")
982         } else {
983             builder.out.join(&compiler.host.triple).join(format!("stage{}", compiler.stage))
984         };
985         let _ = fs::remove_dir_all(&sysroot);
986         t!(fs::create_dir_all(&sysroot));
987
988         // If we're downloading a compiler from CI, we can use the same compiler for all stages other than 0.
989         if builder.config.download_rustc && compiler.stage != 0 {
990             assert_eq!(
991                 builder.config.build, compiler.host,
992                 "Cross-compiling is not yet supported with `download-rustc`",
993             );
994             // Copy the compiler into the correct sysroot.
995             let ci_rustc_dir =
996                 builder.config.out.join(&*builder.config.build.triple).join("ci-rustc");
997             builder.cp_r(&ci_rustc_dir, &sysroot);
998             return INTERNER.intern_path(sysroot);
999         }
1000
1001         // Symlink the source root into the same location inside the sysroot,
1002         // where `rust-src` component would go (`$sysroot/lib/rustlib/src/rust`),
1003         // so that any tools relying on `rust-src` also work for local builds,
1004         // and also for translating the virtual `/rustc/$hash` back to the real
1005         // directory (for running tests with `rust.remap-debuginfo = true`).
1006         let sysroot_lib_rustlib_src = sysroot.join("lib/rustlib/src");
1007         t!(fs::create_dir_all(&sysroot_lib_rustlib_src));
1008         let sysroot_lib_rustlib_src_rust = sysroot_lib_rustlib_src.join("rust");
1009         if let Err(e) = symlink_dir(&builder.config, &builder.src, &sysroot_lib_rustlib_src_rust) {
1010             eprintln!(
1011                 "warning: creating symbolic link `{}` to `{}` failed with {}",
1012                 sysroot_lib_rustlib_src_rust.display(),
1013                 builder.src.display(),
1014                 e,
1015             );
1016             if builder.config.rust_remap_debuginfo {
1017                 eprintln!(
1018                     "warning: some `src/test/ui` tests will fail when lacking `{}`",
1019                     sysroot_lib_rustlib_src_rust.display(),
1020                 );
1021             }
1022         }
1023
1024         INTERNER.intern_path(sysroot)
1025     }
1026 }
1027
1028 #[derive(Debug, Copy, PartialOrd, Ord, Clone, PartialEq, Eq, Hash)]
1029 pub struct Assemble {
1030     /// The compiler which we will produce in this step. Assemble itself will
1031     /// take care of ensuring that the necessary prerequisites to do so exist,
1032     /// that is, this target can be a stage2 compiler and Assemble will build
1033     /// previous stages for you.
1034     pub target_compiler: Compiler,
1035 }
1036
1037 impl Step for Assemble {
1038     type Output = Compiler;
1039     const ONLY_HOSTS: bool = true;
1040
1041     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1042         run.path("compiler/rustc")
1043     }
1044
1045     fn make_run(run: RunConfig<'_>) {
1046         run.builder.ensure(Assemble {
1047             target_compiler: run.builder.compiler(run.builder.top_stage + 1, run.target),
1048         });
1049     }
1050
1051     /// Prepare a new compiler from the artifacts in `stage`
1052     ///
1053     /// This will assemble a compiler in `build/$host/stage$stage`. The compiler
1054     /// must have been previously produced by the `stage - 1` builder.build
1055     /// compiler.
1056     fn run(self, builder: &Builder<'_>) -> Compiler {
1057         let target_compiler = self.target_compiler;
1058
1059         if target_compiler.stage == 0 {
1060             assert_eq!(
1061                 builder.config.build, target_compiler.host,
1062                 "Cannot obtain compiler for non-native build triple at stage 0"
1063             );
1064             // The stage 0 compiler for the build triple is always pre-built.
1065             return target_compiler;
1066         }
1067
1068         // Get the compiler that we'll use to bootstrap ourselves.
1069         //
1070         // Note that this is where the recursive nature of the bootstrap
1071         // happens, as this will request the previous stage's compiler on
1072         // downwards to stage 0.
1073         //
1074         // Also note that we're building a compiler for the host platform. We
1075         // only assume that we can run `build` artifacts, which means that to
1076         // produce some other architecture compiler we need to start from
1077         // `build` to get there.
1078         //
1079         // FIXME: It may be faster if we build just a stage 1 compiler and then
1080         //        use that to bootstrap this compiler forward.
1081         let build_compiler = builder.compiler(target_compiler.stage - 1, builder.config.build);
1082
1083         // If we're downloading a compiler from CI, we can use the same compiler for all stages other than 0.
1084         if builder.config.download_rustc {
1085             builder.ensure(Sysroot { compiler: target_compiler });
1086             return target_compiler;
1087         }
1088
1089         // Build the libraries for this compiler to link to (i.e., the libraries
1090         // it uses at runtime). NOTE: Crates the target compiler compiles don't
1091         // link to these. (FIXME: Is that correct? It seems to be correct most
1092         // of the time but I think we do link to these for stage2/bin compilers
1093         // when not performing a full bootstrap).
1094         builder.ensure(Rustc { compiler: build_compiler, target: target_compiler.host });
1095
1096         for &backend in builder.config.rust_codegen_backends.iter() {
1097             if backend == "llvm" {
1098                 continue; // Already built as part of rustc
1099             }
1100
1101             builder.ensure(CodegenBackend {
1102                 compiler: build_compiler,
1103                 target: target_compiler.host,
1104                 backend,
1105             });
1106         }
1107
1108         let lld_install = if builder.config.lld_enabled {
1109             Some(builder.ensure(native::Lld { target: target_compiler.host }))
1110         } else {
1111             None
1112         };
1113
1114         let stage = target_compiler.stage;
1115         let host = target_compiler.host;
1116         builder.info(&format!("Assembling stage{} compiler ({})", stage, host));
1117
1118         // Link in all dylibs to the libdir
1119         let stamp = librustc_stamp(builder, build_compiler, target_compiler.host);
1120         let proc_macros = builder
1121             .read_stamp_file(&stamp)
1122             .into_iter()
1123             .filter_map(|(path, dependency_type)| {
1124                 if dependency_type == DependencyType::Host {
1125                     Some(path.file_name().unwrap().to_owned().into_string().unwrap())
1126                 } else {
1127                     None
1128                 }
1129             })
1130             .collect::<HashSet<_>>();
1131
1132         let sysroot = builder.sysroot(target_compiler);
1133         let rustc_libdir = builder.rustc_libdir(target_compiler);
1134         t!(fs::create_dir_all(&rustc_libdir));
1135         let src_libdir = builder.sysroot_libdir(build_compiler, host);
1136         for f in builder.read_dir(&src_libdir) {
1137             let filename = f.file_name().into_string().unwrap();
1138             if (is_dylib(&filename) || is_debug_info(&filename)) && !proc_macros.contains(&filename)
1139             {
1140                 builder.copy(&f.path(), &rustc_libdir.join(&filename));
1141             }
1142         }
1143
1144         copy_codegen_backends_to_sysroot(builder, build_compiler, target_compiler);
1145
1146         // We prepend this bin directory to the user PATH when linking Rust binaries. To
1147         // avoid shadowing the system LLD we rename the LLD we provide to `rust-lld`.
1148         let libdir = builder.sysroot_libdir(target_compiler, target_compiler.host);
1149         let libdir_bin = libdir.parent().unwrap().join("bin");
1150         t!(fs::create_dir_all(&libdir_bin));
1151         if let Some(lld_install) = lld_install {
1152             let src_exe = exe("lld", target_compiler.host);
1153             let dst_exe = exe("rust-lld", target_compiler.host);
1154             builder.copy(&lld_install.join("bin").join(&src_exe), &libdir_bin.join(&dst_exe));
1155             // for `-Z gcc-ld=lld`
1156             let gcc_ld_dir = libdir_bin.join("gcc-ld");
1157             t!(fs::create_dir(&gcc_ld_dir));
1158             for flavor in ["ld", "ld64"] {
1159                 let lld_wrapper_exe = builder.ensure(crate::tool::LldWrapper {
1160                     compiler: build_compiler,
1161                     target: target_compiler.host,
1162                     flavor_feature: flavor,
1163                 });
1164                 builder.copy(&lld_wrapper_exe, &gcc_ld_dir.join(exe(flavor, target_compiler.host)));
1165             }
1166         }
1167
1168         if builder.config.rust_codegen_backends.contains(&INTERNER.intern_str("llvm")) {
1169             let llvm_config_bin = builder.ensure(native::Llvm { target: target_compiler.host });
1170             if !builder.config.dry_run {
1171                 let llvm_bin_dir = output(Command::new(llvm_config_bin).arg("--bindir"));
1172                 let llvm_bin_dir = Path::new(llvm_bin_dir.trim());
1173
1174                 // Since we've already built the LLVM tools, install them to the sysroot.
1175                 // This is the equivalent of installing the `llvm-tools-preview` component via
1176                 // rustup, and lets developers use a locally built toolchain to
1177                 // build projects that expect llvm tools to be present in the sysroot
1178                 // (e.g. the `bootimage` crate).
1179                 for tool in LLVM_TOOLS {
1180                     let tool_exe = exe(tool, target_compiler.host);
1181                     let src_path = llvm_bin_dir.join(&tool_exe);
1182                     // When using `donwload-ci-llvm`, some of the tools
1183                     // may not exist, so skip trying to copy them.
1184                     if src_path.exists() {
1185                         builder.copy(&src_path, &libdir_bin.join(&tool_exe));
1186                     }
1187                 }
1188             }
1189         }
1190
1191         // Ensure that `libLLVM.so` ends up in the newly build compiler directory,
1192         // so that it can be found when the newly built `rustc` is run.
1193         dist::maybe_install_llvm_runtime(builder, target_compiler.host, &sysroot);
1194         dist::maybe_install_llvm_target(builder, target_compiler.host, &sysroot);
1195
1196         // Link the compiler binary itself into place
1197         let out_dir = builder.cargo_out(build_compiler, Mode::Rustc, host);
1198         let rustc = out_dir.join(exe("rustc-main", host));
1199         let bindir = sysroot.join("bin");
1200         t!(fs::create_dir_all(&bindir));
1201         let compiler = builder.rustc(target_compiler);
1202         builder.copy(&rustc, &compiler);
1203
1204         target_compiler
1205     }
1206 }
1207
1208 /// Link some files into a rustc sysroot.
1209 ///
1210 /// For a particular stage this will link the file listed in `stamp` into the
1211 /// `sysroot_dst` provided.
1212 pub fn add_to_sysroot(
1213     builder: &Builder<'_>,
1214     sysroot_dst: &Path,
1215     sysroot_host_dst: &Path,
1216     stamp: &Path,
1217 ) {
1218     let self_contained_dst = &sysroot_dst.join("self-contained");
1219     t!(fs::create_dir_all(&sysroot_dst));
1220     t!(fs::create_dir_all(&sysroot_host_dst));
1221     t!(fs::create_dir_all(&self_contained_dst));
1222     for (path, dependency_type) in builder.read_stamp_file(stamp) {
1223         let dst = match dependency_type {
1224             DependencyType::Host => sysroot_host_dst,
1225             DependencyType::Target => sysroot_dst,
1226             DependencyType::TargetSelfContained => self_contained_dst,
1227         };
1228         builder.copy(&path, &dst.join(path.file_name().unwrap()));
1229     }
1230 }
1231
1232 pub fn run_cargo(
1233     builder: &Builder<'_>,
1234     cargo: Cargo,
1235     tail_args: Vec<String>,
1236     stamp: &Path,
1237     additional_target_deps: Vec<(PathBuf, DependencyType)>,
1238     is_check: bool,
1239 ) -> Vec<PathBuf> {
1240     if builder.config.dry_run {
1241         return Vec::new();
1242     }
1243
1244     // `target_root_dir` looks like $dir/$target/release
1245     let target_root_dir = stamp.parent().unwrap();
1246     // `target_deps_dir` looks like $dir/$target/release/deps
1247     let target_deps_dir = target_root_dir.join("deps");
1248     // `host_root_dir` looks like $dir/release
1249     let host_root_dir = target_root_dir
1250         .parent()
1251         .unwrap() // chop off `release`
1252         .parent()
1253         .unwrap() // chop off `$target`
1254         .join(target_root_dir.file_name().unwrap());
1255
1256     // Spawn Cargo slurping up its JSON output. We'll start building up the
1257     // `deps` array of all files it generated along with a `toplevel` array of
1258     // files we need to probe for later.
1259     let mut deps = Vec::new();
1260     let mut toplevel = Vec::new();
1261     let ok = stream_cargo(builder, cargo, tail_args, &mut |msg| {
1262         let (filenames, crate_types) = match msg {
1263             CargoMessage::CompilerArtifact {
1264                 filenames,
1265                 target: CargoTarget { crate_types },
1266                 ..
1267             } => (filenames, crate_types),
1268             _ => return,
1269         };
1270         for filename in filenames {
1271             // Skip files like executables
1272             if !(filename.ends_with(".rlib")
1273                 || filename.ends_with(".lib")
1274                 || filename.ends_with(".a")
1275                 || is_debug_info(&filename)
1276                 || is_dylib(&filename)
1277                 || (is_check && filename.ends_with(".rmeta")))
1278             {
1279                 continue;
1280             }
1281
1282             let filename = Path::new(&*filename);
1283
1284             // If this was an output file in the "host dir" we don't actually
1285             // worry about it, it's not relevant for us
1286             if filename.starts_with(&host_root_dir) {
1287                 // Unless it's a proc macro used in the compiler
1288                 if crate_types.iter().any(|t| t == "proc-macro") {
1289                     deps.push((filename.to_path_buf(), DependencyType::Host));
1290                 }
1291                 continue;
1292             }
1293
1294             // If this was output in the `deps` dir then this is a precise file
1295             // name (hash included) so we start tracking it.
1296             if filename.starts_with(&target_deps_dir) {
1297                 deps.push((filename.to_path_buf(), DependencyType::Target));
1298                 continue;
1299             }
1300
1301             // Otherwise this was a "top level artifact" which right now doesn't
1302             // have a hash in the name, but there's a version of this file in
1303             // the `deps` folder which *does* have a hash in the name. That's
1304             // the one we'll want to we'll probe for it later.
1305             //
1306             // We do not use `Path::file_stem` or `Path::extension` here,
1307             // because some generated files may have multiple extensions e.g.
1308             // `std-<hash>.dll.lib` on Windows. The aforementioned methods only
1309             // split the file name by the last extension (`.lib`) while we need
1310             // to split by all extensions (`.dll.lib`).
1311             let expected_len = t!(filename.metadata()).len();
1312             let filename = filename.file_name().unwrap().to_str().unwrap();
1313             let mut parts = filename.splitn(2, '.');
1314             let file_stem = parts.next().unwrap().to_owned();
1315             let extension = parts.next().unwrap().to_owned();
1316
1317             toplevel.push((file_stem, extension, expected_len));
1318         }
1319     });
1320
1321     if !ok {
1322         exit(1);
1323     }
1324
1325     // Ok now we need to actually find all the files listed in `toplevel`. We've
1326     // got a list of prefix/extensions and we basically just need to find the
1327     // most recent file in the `deps` folder corresponding to each one.
1328     let contents = t!(target_deps_dir.read_dir())
1329         .map(|e| t!(e))
1330         .map(|e| (e.path(), e.file_name().into_string().unwrap(), t!(e.metadata())))
1331         .collect::<Vec<_>>();
1332     for (prefix, extension, expected_len) in toplevel {
1333         let candidates = contents.iter().filter(|&&(_, ref filename, ref meta)| {
1334             meta.len() == expected_len
1335                 && filename
1336                     .strip_prefix(&prefix[..])
1337                     .map(|s| s.starts_with('-') && s.ends_with(&extension[..]))
1338                     .unwrap_or(false)
1339         });
1340         let max = candidates.max_by_key(|&&(_, _, ref metadata)| {
1341             metadata.modified().expect("mtime should be available on all relevant OSes")
1342         });
1343         let path_to_add = match max {
1344             Some(triple) => triple.0.to_str().unwrap(),
1345             None => panic!("no output generated for {:?} {:?}", prefix, extension),
1346         };
1347         if is_dylib(path_to_add) {
1348             let candidate = format!("{}.lib", path_to_add);
1349             let candidate = PathBuf::from(candidate);
1350             if candidate.exists() {
1351                 deps.push((candidate, DependencyType::Target));
1352             }
1353         }
1354         deps.push((path_to_add.into(), DependencyType::Target));
1355     }
1356
1357     deps.extend(additional_target_deps);
1358     deps.sort();
1359     let mut new_contents = Vec::new();
1360     for (dep, dependency_type) in deps.iter() {
1361         new_contents.extend(match *dependency_type {
1362             DependencyType::Host => b"h",
1363             DependencyType::Target => b"t",
1364             DependencyType::TargetSelfContained => b"s",
1365         });
1366         new_contents.extend(dep.to_str().unwrap().as_bytes());
1367         new_contents.extend(b"\0");
1368     }
1369     t!(fs::write(&stamp, &new_contents));
1370     deps.into_iter().map(|(d, _)| d).collect()
1371 }
1372
1373 pub fn stream_cargo(
1374     builder: &Builder<'_>,
1375     cargo: Cargo,
1376     tail_args: Vec<String>,
1377     cb: &mut dyn FnMut(CargoMessage<'_>),
1378 ) -> bool {
1379     let mut cargo = Command::from(cargo);
1380     if builder.config.dry_run {
1381         return true;
1382     }
1383     // Instruct Cargo to give us json messages on stdout, critically leaving
1384     // stderr as piped so we can get those pretty colors.
1385     let mut message_format = if builder.config.json_output {
1386         String::from("json")
1387     } else {
1388         String::from("json-render-diagnostics")
1389     };
1390     if let Some(s) = &builder.config.rustc_error_format {
1391         message_format.push_str(",json-diagnostic-");
1392         message_format.push_str(s);
1393     }
1394     cargo.arg("--message-format").arg(message_format).stdout(Stdio::piped());
1395
1396     for arg in tail_args {
1397         cargo.arg(arg);
1398     }
1399
1400     builder.verbose(&format!("running: {:?}", cargo));
1401     let mut child = match cargo.spawn() {
1402         Ok(child) => child,
1403         Err(e) => panic!("failed to execute command: {:?}\nerror: {}", cargo, e),
1404     };
1405
1406     // Spawn Cargo slurping up its JSON output. We'll start building up the
1407     // `deps` array of all files it generated along with a `toplevel` array of
1408     // files we need to probe for later.
1409     let stdout = BufReader::new(child.stdout.take().unwrap());
1410     for line in stdout.lines() {
1411         let line = t!(line);
1412         match serde_json::from_str::<CargoMessage<'_>>(&line) {
1413             Ok(msg) => {
1414                 if builder.config.json_output {
1415                     // Forward JSON to stdout.
1416                     println!("{}", line);
1417                 }
1418                 cb(msg)
1419             }
1420             // If this was informational, just print it out and continue
1421             Err(_) => println!("{}", line),
1422         }
1423     }
1424
1425     // Make sure Cargo actually succeeded after we read all of its stdout.
1426     let status = t!(child.wait());
1427     if builder.is_verbose() && !status.success() {
1428         eprintln!(
1429             "command did not execute successfully: {:?}\n\
1430                   expected success, got: {}",
1431             cargo, status
1432         );
1433     }
1434     status.success()
1435 }
1436
1437 #[derive(Deserialize)]
1438 pub struct CargoTarget<'a> {
1439     crate_types: Vec<Cow<'a, str>>,
1440 }
1441
1442 #[derive(Deserialize)]
1443 #[serde(tag = "reason", rename_all = "kebab-case")]
1444 pub enum CargoMessage<'a> {
1445     CompilerArtifact {
1446         package_id: Cow<'a, str>,
1447         features: Vec<Cow<'a, str>>,
1448         filenames: Vec<Cow<'a, str>>,
1449         target: CargoTarget<'a>,
1450     },
1451     BuildScriptExecuted {
1452         package_id: Cow<'a, str>,
1453     },
1454     BuildFinished {
1455         success: bool,
1456     },
1457 }