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