src/bootstrap/native.rs

   1 //! Compilation of native dependencies like LLVM.
   2 //!
   3 //! Native projects like LLVM unfortunately aren't suited just yet for
   4 //! compilation in build scripts that Cargo has. This is because the
   5 //! compilation takes a *very* long time but also because we don't want to
   6 //! compile LLVM 3 times as part of a normal bootstrap (we want it cached).
   7 //!
   8 //! LLVM and compiler-rt are essentially just wired up to everything else to
   9 //! ensure that they're always in place if needed.
  10
  11 use std::env;
  12 use std::env::consts::EXE_EXTENSION;
  13 use std::ffi::{OsStr, OsString};
  14 use std::fs::{self, File};
  15 use std::io;
  16 use std::path::{Path, PathBuf};
  17 use std::process::Command;
  18
  19 use crate::builder::{Builder, RunConfig, ShouldRun, Step};
  20 use crate::config::TargetSelection;
  21 use crate::util::get_clang_cl_resource_dir;
  22 use crate::util::{self, exe, output, program_out_of_date, t, up_to_date};
  23 use crate::{CLang, GitRepo};
  24
  25 pub struct Meta {
  26     stamp: HashStamp,
  27     build_llvm_config: PathBuf,
  28     out_dir: PathBuf,
  29     root: String,
  30 }
  31
  32 // Linker flags to pass to LLVM's CMake invocation.
  33 #[derive(Debug, Clone, Default)]
  34 struct LdFlags {
  35     // CMAKE_EXE_LINKER_FLAGS
  36     exe: OsString,
  37     // CMAKE_SHARED_LINKER_FLAGS
  38     shared: OsString,
  39     // CMAKE_MODULE_LINKER_FLAGS
  40     module: OsString,
  41 }
  42
  43 impl LdFlags {
  44     fn push_all(&mut self, s: impl AsRef<OsStr>) {
  45         let s = s.as_ref();
  46         self.exe.push(" ");
  47         self.exe.push(s);
  48         self.shared.push(" ");
  49         self.shared.push(s);
  50         self.module.push(" ");
  51         self.module.push(s);
  52     }
  53 }
  54
  55 // This returns whether we've already previously built LLVM.
  56 //
  57 // It's used to avoid busting caches during x.py check -- if we've already built
  58 // LLVM, it's fine for us to not try to avoid doing so.
  59 //
  60 // This will return the llvm-config if it can get it (but it will not build it
  61 // if not).
  62 pub fn prebuilt_llvm_config(
  63     builder: &Builder<'_>,
  64     target: TargetSelection,
  65 ) -> Result<PathBuf, Meta> {
  66     maybe_download_ci_llvm(builder);
  67
  68     // If we're using a custom LLVM bail out here, but we can only use a
  69     // custom LLVM for the build triple.
  70     if let Some(config) = builder.config.target_config.get(&target) {
  71         if let Some(ref s) = config.llvm_config {
  72             check_llvm_version(builder, s);
  73             return Ok(s.to_path_buf());
  74         }
  75     }
  76
  77     let root = "src/llvm-project/llvm";
  78     let out_dir = builder.llvm_out(target);
  79
  80     let mut llvm_config_ret_dir = builder.llvm_out(builder.config.build);
  81     if !builder.config.build.contains("msvc") || builder.ninja() {
  82         llvm_config_ret_dir.push("build");
  83     }
  84     llvm_config_ret_dir.push("bin");
  85
  86     let build_llvm_config = llvm_config_ret_dir.join(exe("llvm-config", builder.config.build));
  87
  88     let stamp = out_dir.join("llvm-finished-building");
  89     let stamp = HashStamp::new(stamp, builder.in_tree_llvm_info.sha());
  90
  91     if builder.config.llvm_skip_rebuild && stamp.path.exists() {
  92         builder.info(
  93             "Warning: \
  94                 Using a potentially stale build of LLVM; \
  95                 This may not behave well.",
  96         );
  97         return Ok(build_llvm_config);
  98     }
  99
 100     if stamp.is_done() {
 101         if stamp.hash.is_none() {
 102             builder.info(
 103                 "Could not determine the LLVM submodule commit hash. \
 104                      Assuming that an LLVM rebuild is not necessary.",
 105             );
 106             builder.info(&format!(
 107                 "To force LLVM to rebuild, remove the file `{}`",
 108                 stamp.path.display()
 109             ));
 110         }
 111         return Ok(build_llvm_config);
 112     }
 113
 114     Err(Meta { stamp, build_llvm_config, out_dir, root: root.into() })
 115 }
 116
 117 pub(crate) fn maybe_download_ci_llvm(builder: &Builder<'_>) {
 118     let config = &builder.config;
 119     if !config.llvm_from_ci {
 120         return;
 121     }
 122     let mut rev_list = config.git();
 123     rev_list.args(&[
 124         PathBuf::from("rev-list"),
 125         format!("--author={}", builder.config.stage0_metadata.config.git_merge_commit_email).into(),
 126         "-n1".into(),
 127         "--first-parent".into(),
 128         "HEAD".into(),
 129         "--".into(),
 130         builder.src.join("src/llvm-project"),
 131         builder.src.join("src/bootstrap/download-ci-llvm-stamp"),
 132         // the LLVM shared object file is named `LLVM-12-rust-{version}-nightly`
 133         builder.src.join("src/version"),
 134     ]);
 135     let llvm_sha = output(&mut rev_list);
 136     let llvm_sha = llvm_sha.trim();
 137
 138     if llvm_sha == "" {
 139         eprintln!("error: could not find commit hash for downloading LLVM");
 140         eprintln!("help: maybe your repository history is too shallow?");
 141         eprintln!("help: consider disabling `download-ci-llvm`");
 142         eprintln!("help: or fetch enough history to include one upstream commit");
 143         panic!();
 144     }
 145
 146     let llvm_root = config.ci_llvm_root();
 147     let llvm_stamp = llvm_root.join(".llvm-stamp");
 148     let key = format!("{}{}", llvm_sha, config.llvm_assertions);
 149     if program_out_of_date(&llvm_stamp, &key) && !config.dry_run {
 150         download_ci_llvm(builder, &llvm_sha);
 151         for entry in t!(fs::read_dir(llvm_root.join("bin"))) {
 152             builder.fix_bin_or_dylib(&t!(entry).path());
 153         }
 154
 155         // Update the timestamp of llvm-config to force rustc_llvm to be
 156         // rebuilt. This is a hacky workaround for a deficiency in Cargo where
 157         // the rerun-if-changed directive doesn't handle changes very well.
 158         // https://github.com/rust-lang/cargo/issues/10791
 159         // Cargo only compares the timestamp of the file relative to the last
 160         // time `rustc_llvm` build script ran. However, the timestamps of the
 161         // files in the tarball are in the past, so it doesn't trigger a
 162         // rebuild.
 163         let now = filetime::FileTime::from_system_time(std::time::SystemTime::now());
 164         let llvm_config = llvm_root.join("bin").join(exe("llvm-config", builder.config.build));
 165         t!(filetime::set_file_times(&llvm_config, now, now));
 166
 167         let llvm_lib = llvm_root.join("lib");
 168         for entry in t!(fs::read_dir(&llvm_lib)) {
 169             let lib = t!(entry).path();
 170             if lib.extension().map_or(false, |ext| ext == "so") {
 171                 builder.fix_bin_or_dylib(&lib);
 172             }
 173         }
 174         t!(fs::write(llvm_stamp, key));
 175     }
 176 }
 177
 178 fn download_ci_llvm(builder: &Builder<'_>, llvm_sha: &str) {
 179     let llvm_assertions = builder.config.llvm_assertions;
 180
 181     let cache_prefix = format!("llvm-{}-{}", llvm_sha, llvm_assertions);
 182     let cache_dst = builder.out.join("cache");
 183     let rustc_cache = cache_dst.join(cache_prefix);
 184     if !rustc_cache.exists() {
 185         t!(fs::create_dir_all(&rustc_cache));
 186     }
 187     let base = if llvm_assertions {
 188         &builder.config.stage0_metadata.config.artifacts_with_llvm_assertions_server
 189     } else {
 190         &builder.config.stage0_metadata.config.artifacts_server
 191     };
 192     let channel = builder.config.artifact_channel(llvm_sha);
 193     let filename = format!("rust-dev-{}-{}.tar.xz", channel, builder.build.build.triple);
 194     let tarball = rustc_cache.join(&filename);
 195     if !tarball.exists() {
 196         let help_on_error = "error: failed to download llvm from ci
 197
 198 help: old builds get deleted after a certain time
 199 help: if trying to compile an old commit of rustc, disable `download-ci-llvm` in config.toml:
 200
 201 [llvm]
 202 download-ci-llvm = false
 203 ";
 204         builder.download_component(
 205             &format!("{base}/{llvm_sha}/{filename}"),
 206             &tarball,
 207             help_on_error,
 208         );
 209     }
 210     let llvm_root = builder.config.ci_llvm_root();
 211     builder.unpack(&tarball, &llvm_root, "rust-dev");
 212 }
 213
 214 #[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
 215 pub struct Llvm {
 216     pub target: TargetSelection,
 217 }
 218
 219 impl Step for Llvm {
 220     type Output = PathBuf; // path to llvm-config
 221
 222     const ONLY_HOSTS: bool = true;
 223
 224     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
 225         run.path("src/llvm-project").path("src/llvm-project/llvm")
 226     }
 227
 228     fn make_run(run: RunConfig<'_>) {
 229         run.builder.ensure(Llvm { target: run.target });
 230     }
 231
 232     /// Compile LLVM for `target`.
 233     fn run(self, builder: &Builder<'_>) -> PathBuf {
 234         let target = self.target;
 235         let target_native = if self.target.starts_with("riscv") {
 236             // RISC-V target triples in Rust is not named the same as C compiler target triples.
 237             // This converts Rust RISC-V target triples to C compiler triples.
 238             let idx = target.triple.find('-').unwrap();
 239
 240             format!("riscv{}{}", &target.triple[5..7], &target.triple[idx..])
 241         } else if self.target.starts_with("powerpc") && self.target.ends_with("freebsd") {
 242             // FreeBSD 13 had incompatible ABI changes on all PowerPC platforms.
 243             // Set the version suffix to 13.0 so the correct target details are used.
 244             format!("{}{}", self.target, "13.0")
 245         } else {
 246             target.to_string()
 247         };
 248
 249         let Meta { stamp, build_llvm_config, out_dir, root } =
 250             match prebuilt_llvm_config(builder, target) {
 251                 Ok(p) => return p,
 252                 Err(m) => m,
 253             };
 254
 255         builder.update_submodule(&Path::new("src").join("llvm-project"));
 256         if builder.llvm_link_shared() && target.contains("windows") {
 257             panic!("shared linking to LLVM is not currently supported on {}", target.triple);
 258         }
 259
 260         builder.info(&format!("Building LLVM for {}", target));
 261         t!(stamp.remove());
 262         let _time = util::timeit(&builder);
 263         t!(fs::create_dir_all(&out_dir));
 264
 265         // https://llvm.org/docs/CMake.html
 266         let mut cfg = cmake::Config::new(builder.src.join(root));
 267         let mut ldflags = LdFlags::default();
 268
 269         let profile = match (builder.config.llvm_optimize, builder.config.llvm_release_debuginfo) {
 270             (false, _) => "Debug",
 271             (true, false) => "Release",
 272             (true, true) => "RelWithDebInfo",
 273         };
 274
 275         // NOTE: remember to also update `config.toml.example` when changing the
 276         // defaults!
 277         let llvm_targets = match &builder.config.llvm_targets {
 278             Some(s) => s,
 279             None => {
 280                 "AArch64;ARM;BPF;Hexagon;MSP430;Mips;NVPTX;PowerPC;RISCV;\
 281                      Sparc;SystemZ;WebAssembly;X86"
 282             }
 283         };
 284
 285         let llvm_exp_targets = match builder.config.llvm_experimental_targets {
 286             Some(ref s) => s,
 287             None => "AVR;M68k",
 288         };
 289
 290         let assertions = if builder.config.llvm_assertions { "ON" } else { "OFF" };
 291         let plugins = if builder.config.llvm_plugins { "ON" } else { "OFF" };
 292         let enable_tests = if builder.config.llvm_tests { "ON" } else { "OFF" };
 293
 294         cfg.out_dir(&out_dir)
 295             .profile(profile)
 296             .define("LLVM_ENABLE_ASSERTIONS", assertions)
 297             .define("LLVM_ENABLE_PLUGINS", plugins)
 298             .define("LLVM_TARGETS_TO_BUILD", llvm_targets)
 299             .define("LLVM_EXPERIMENTAL_TARGETS_TO_BUILD", llvm_exp_targets)
 300             .define("LLVM_INCLUDE_EXAMPLES", "OFF")
 301             .define("LLVM_INCLUDE_DOCS", "OFF")
 302             .define("LLVM_INCLUDE_BENCHMARKS", "OFF")
 303             .define("LLVM_INCLUDE_TESTS", enable_tests)
 304             .define("LLVM_ENABLE_TERMINFO", "OFF")
 305             .define("LLVM_ENABLE_LIBEDIT", "OFF")
 306             .define("LLVM_ENABLE_BINDINGS", "OFF")
 307             .define("LLVM_ENABLE_Z3_SOLVER", "OFF")
 308             .define("LLVM_PARALLEL_COMPILE_JOBS", builder.jobs().to_string())
 309             .define("LLVM_TARGET_ARCH", target_native.split('-').next().unwrap())
 310             .define("LLVM_DEFAULT_TARGET_TRIPLE", target_native);
 311
 312         // Parts of our test suite rely on the `FileCheck` tool, which is built by default in
 313         // `build/$TARGET/llvm/build/bin` is but *not* then installed to `build/$TARGET/llvm/bin`.
 314         // This flag makes sure `FileCheck` is copied in the final binaries directory.
 315         cfg.define("LLVM_INSTALL_UTILS", "ON");
 316
 317         if builder.config.llvm_profile_generate {
 318             cfg.define("LLVM_BUILD_INSTRUMENTED", "IR");
 319             if let Ok(llvm_profile_dir) = std::env::var("LLVM_PROFILE_DIR") {
 320                 cfg.define("LLVM_PROFILE_DATA_DIR", llvm_profile_dir);
 321             }
 322             cfg.define("LLVM_BUILD_RUNTIME", "No");
 323         }
 324         if let Some(path) = builder.config.llvm_profile_use.as_ref() {
 325             cfg.define("LLVM_PROFDATA_FILE", &path);
 326         }
 327
 328         if target != "aarch64-apple-darwin" && !target.contains("windows") {
 329             cfg.define("LLVM_ENABLE_ZLIB", "ON");
 330         } else {
 331             cfg.define("LLVM_ENABLE_ZLIB", "OFF");
 332         }
 333
 334         // Are we compiling for iOS/tvOS/watchOS?
 335         if target.contains("apple-ios")
 336             || target.contains("apple-tvos")
 337             || target.contains("apple-watchos")
 338         {
 339             // These two defines prevent CMake from automatically trying to add a MacOSX sysroot, which leads to a compiler error.
 340             cfg.define("CMAKE_OSX_SYSROOT", "/");
 341             cfg.define("CMAKE_OSX_DEPLOYMENT_TARGET", "");
 342             // Prevent cmake from adding -bundle to CFLAGS automatically, which leads to a compiler error because "-bitcode_bundle" also gets added.
 343             cfg.define("LLVM_ENABLE_PLUGINS", "OFF");
 344             // Zlib fails to link properly, leading to a compiler error.
 345             cfg.define("LLVM_ENABLE_ZLIB", "OFF");
 346         }
 347
 348         if builder.config.llvm_thin_lto {
 349             cfg.define("LLVM_ENABLE_LTO", "Thin");
 350             if !target.contains("apple") {
 351                 cfg.define("LLVM_ENABLE_LLD", "ON");
 352             }
 353         }
 354
 355         // This setting makes the LLVM tools link to the dynamic LLVM library,
 356         // which saves both memory during parallel links and overall disk space
 357         // for the tools. We don't do this on every platform as it doesn't work
 358         // equally well everywhere.
 359         //
 360         // If we're not linking rustc to a dynamic LLVM, though, then don't link
 361         // tools to it.
 362         let llvm_link_shared =
 363             builder.llvm_link_tools_dynamically(target) && builder.llvm_link_shared();
 364         if llvm_link_shared {
 365             cfg.define("LLVM_LINK_LLVM_DYLIB", "ON");
 366         }
 367
 368         if target.starts_with("riscv") && !target.contains("freebsd") {
 369             // RISC-V GCC erroneously requires linking against
 370             // `libatomic` when using 1-byte and 2-byte C++
 371             // atomics but the LLVM build system check cannot
 372             // detect this. Therefore it is set manually here.
 373             // FreeBSD uses Clang as its system compiler and
 374             // provides no libatomic in its base system so does
 375             // not want this.
 376             ldflags.exe.push(" -latomic");
 377             ldflags.shared.push(" -latomic");
 378         }
 379
 380         if target.contains("msvc") {
 381             cfg.define("LLVM_USE_CRT_DEBUG", "MT");
 382             cfg.define("LLVM_USE_CRT_RELEASE", "MT");
 383             cfg.define("LLVM_USE_CRT_RELWITHDEBINFO", "MT");
 384             cfg.static_crt(true);
 385         }
 386
 387         if target.starts_with("i686") {
 388             cfg.define("LLVM_BUILD_32_BITS", "ON");
 389         }
 390
 391         let mut enabled_llvm_projects = Vec::new();
 392
 393         if util::forcing_clang_based_tests() {
 394             enabled_llvm_projects.push("clang");
 395             enabled_llvm_projects.push("compiler-rt");
 396         }
 397
 398         if builder.config.llvm_polly {
 399             enabled_llvm_projects.push("polly");
 400         }
 401
 402         if builder.config.llvm_clang {
 403             enabled_llvm_projects.push("clang");
 404         }
 405
 406         // We want libxml to be disabled.
 407         // See https://github.com/rust-lang/rust/pull/50104
 408         cfg.define("LLVM_ENABLE_LIBXML2", "OFF");
 409
 410         if !enabled_llvm_projects.is_empty() {
 411             enabled_llvm_projects.sort();
 412             enabled_llvm_projects.dedup();
 413             cfg.define("LLVM_ENABLE_PROJECTS", enabled_llvm_projects.join(";"));
 414         }
 415
 416         if let Some(num_linkers) = builder.config.llvm_link_jobs {
 417             if num_linkers > 0 {
 418                 cfg.define("LLVM_PARALLEL_LINK_JOBS", num_linkers.to_string());
 419             }
 420         }
 421
 422         // Workaround for ppc32 lld limitation
 423         if target == "powerpc-unknown-freebsd" {
 424             ldflags.exe.push(" -fuse-ld=bfd");
 425         }
 426
 427         // https://llvm.org/docs/HowToCrossCompileLLVM.html
 428         if target != builder.config.build {
 429             builder.ensure(Llvm { target: builder.config.build });
 430             // FIXME: if the llvm root for the build triple is overridden then we
 431             //        should use llvm-tblgen from there, also should verify that it
 432             //        actually exists most of the time in normal installs of LLVM.
 433             let host_bin = builder.llvm_out(builder.config.build).join("bin");
 434             cfg.define("LLVM_TABLEGEN", host_bin.join("llvm-tblgen").with_extension(EXE_EXTENSION));
 435             // LLVM_NM is required for cross compiling using MSVC
 436             cfg.define("LLVM_NM", host_bin.join("llvm-nm").with_extension(EXE_EXTENSION));
 437             cfg.define(
 438                 "LLVM_CONFIG_PATH",
 439                 host_bin.join("llvm-config").with_extension(EXE_EXTENSION),
 440             );
 441             if builder.config.llvm_clang {
 442                 let build_bin = builder.llvm_out(builder.config.build).join("build").join("bin");
 443                 let clang_tblgen = build_bin.join("clang-tblgen").with_extension(EXE_EXTENSION);
 444                 if !builder.config.dry_run && !clang_tblgen.exists() {
 445                     panic!("unable to find {}", clang_tblgen.display());
 446                 }
 447                 cfg.define("CLANG_TABLEGEN", clang_tblgen);
 448             }
 449         }
 450
 451         let llvm_version_suffix = if let Some(ref suffix) = builder.config.llvm_version_suffix {
 452             // Allow version-suffix="" to not define a version suffix at all.
 453             if !suffix.is_empty() { Some(suffix.to_string()) } else { None }
 454         } else if builder.config.channel == "dev" {
 455             // Changes to a version suffix require a complete rebuild of the LLVM.
 456             // To avoid rebuilds during a time of version bump, don't include rustc
 457             // release number on the dev channel.
 458             Some("-rust-dev".to_string())
 459         } else {
 460             Some(format!("-rust-{}-{}", builder.version, builder.config.channel))
 461         };
 462         if let Some(ref suffix) = llvm_version_suffix {
 463             cfg.define("LLVM_VERSION_SUFFIX", suffix);
 464         }
 465
 466         if let Some(ref linker) = builder.config.llvm_use_linker {
 467             cfg.define("LLVM_USE_LINKER", linker);
 468         }
 469
 470         if builder.config.llvm_allow_old_toolchain {
 471             cfg.define("LLVM_TEMPORARILY_ALLOW_OLD_TOOLCHAIN", "YES");
 472         }
 473
 474         configure_cmake(builder, target, &mut cfg, true, ldflags);
 475
 476         for (key, val) in &builder.config.llvm_build_config {
 477             cfg.define(key, val);
 478         }
 479
 480         // FIXME: we don't actually need to build all LLVM tools and all LLVM
 481         //        libraries here, e.g., we just want a few components and a few
 482         //        tools. Figure out how to filter them down and only build the right
 483         //        tools and libs on all platforms.
 484
 485         if builder.config.dry_run {
 486             return build_llvm_config;
 487         }
 488
 489         cfg.build();
 490
 491         // When building LLVM with LLVM_LINK_LLVM_DYLIB for macOS, an unversioned
 492         // libLLVM.dylib will be built. However, llvm-config will still look
 493         // for a versioned path like libLLVM-14.dylib. Manually create a symbolic
 494         // link to make llvm-config happy.
 495         if llvm_link_shared && target.contains("apple-darwin") {
 496             let mut cmd = Command::new(&build_llvm_config);
 497             let version = output(cmd.arg("--version"));
 498             let major = version.split('.').next().unwrap();
 499             let lib_name = match llvm_version_suffix {
 500                 Some(s) => format!("libLLVM-{}{}.dylib", major, s),
 501                 None => format!("libLLVM-{}.dylib", major),
 502             };
 503
 504             let lib_llvm = out_dir.join("build").join("lib").join(lib_name);
 505             if !lib_llvm.exists() {
 506                 t!(builder.symlink_file("libLLVM.dylib", &lib_llvm));
 507             }
 508         }
 509
 510         t!(stamp.write());
 511
 512         build_llvm_config
 513     }
 514 }
 515
 516 fn check_llvm_version(builder: &Builder<'_>, llvm_config: &Path) {
 517     if !builder.config.llvm_version_check {
 518         return;
 519     }
 520
 521     if builder.config.dry_run {
 522         return;
 523     }
 524
 525     let mut cmd = Command::new(llvm_config);
 526     let version = output(cmd.arg("--version"));
 527     let mut parts = version.split('.').take(2).filter_map(|s| s.parse::<u32>().ok());
 528     if let (Some(major), Some(_minor)) = (parts.next(), parts.next()) {
 529         if major >= 12 {
 530             return;
 531         }
 532     }
 533     panic!("\n\nbad LLVM version: {}, need >=12.0\n\n", version)
 534 }
 535
 536 fn configure_cmake(
 537     builder: &Builder<'_>,
 538     target: TargetSelection,
 539     cfg: &mut cmake::Config,
 540     use_compiler_launcher: bool,
 541     mut ldflags: LdFlags,
 542 ) {
 543     // Do not print installation messages for up-to-date files.
 544     // LLVM and LLD builds can produce a lot of those and hit CI limits on log size.
 545     cfg.define("CMAKE_INSTALL_MESSAGE", "LAZY");
 546
 547     // Do not allow the user's value of DESTDIR to influence where
 548     // LLVM will install itself. LLVM must always be installed in our
 549     // own build directories.
 550     cfg.env("DESTDIR", "");
 551
 552     if builder.ninja() {
 553         cfg.generator("Ninja");
 554     }
 555     cfg.target(&target.triple).host(&builder.config.build.triple);
 556
 557     if target != builder.config.build {
 558         cfg.define("CMAKE_CROSSCOMPILING", "True");
 559
 560         if target.contains("netbsd") {
 561             cfg.define("CMAKE_SYSTEM_NAME", "NetBSD");
 562         } else if target.contains("freebsd") {
 563             cfg.define("CMAKE_SYSTEM_NAME", "FreeBSD");
 564         } else if target.contains("windows") {
 565             cfg.define("CMAKE_SYSTEM_NAME", "Windows");
 566         } else if target.contains("haiku") {
 567             cfg.define("CMAKE_SYSTEM_NAME", "Haiku");
 568         } else if target.contains("solaris") || target.contains("illumos") {
 569             cfg.define("CMAKE_SYSTEM_NAME", "SunOS");
 570         }
 571         // When cross-compiling we should also set CMAKE_SYSTEM_VERSION, but in
 572         // that case like CMake we cannot easily determine system version either.
 573         //
 574         // Since, the LLVM itself makes rather limited use of version checks in
 575         // CMakeFiles (and then only in tests), and so far no issues have been
 576         // reported, the system version is currently left unset.
 577
 578         if target.contains("darwin") {
 579             // Make sure that CMake does not build universal binaries on macOS.
 580             // Explicitly specifiy the one single target architecture.
 581             if target.starts_with("aarch64") {
 582                 // macOS uses a different name for building arm64
 583                 cfg.define("CMAKE_OSX_ARCHITECTURES", "arm64");
 584             } else {
 585                 cfg.define("CMAKE_OSX_ARCHITECTURES", target.triple.split('-').next().unwrap());
 586             }
 587         }
 588     }
 589
 590     let sanitize_cc = |cc: &Path| {
 591         if target.contains("msvc") {
 592             OsString::from(cc.to_str().unwrap().replace("\\", "/"))
 593         } else {
 594             cc.as_os_str().to_owned()
 595         }
 596     };
 597
 598     // MSVC with CMake uses msbuild by default which doesn't respect these
 599     // vars that we'd otherwise configure. In that case we just skip this
 600     // entirely.
 601     if target.contains("msvc") && !builder.ninja() {
 602         return;
 603     }
 604
 605     let (cc, cxx) = match builder.config.llvm_clang_cl {
 606         Some(ref cl) => (cl.as_ref(), cl.as_ref()),
 607         None => (builder.cc(target), builder.cxx(target).unwrap()),
 608     };
 609
 610     // Handle msvc + ninja + ccache specially (this is what the bots use)
 611     if target.contains("msvc") && builder.ninja() && builder.config.ccache.is_some() {
 612         let mut wrap_cc = env::current_exe().expect("failed to get cwd");
 613         wrap_cc.set_file_name("sccache-plus-cl.exe");
 614
 615         cfg.define("CMAKE_C_COMPILER", sanitize_cc(&wrap_cc))
 616             .define("CMAKE_CXX_COMPILER", sanitize_cc(&wrap_cc));
 617         cfg.env("SCCACHE_PATH", builder.config.ccache.as_ref().unwrap())
 618             .env("SCCACHE_TARGET", target.triple)
 619             .env("SCCACHE_CC", &cc)
 620             .env("SCCACHE_CXX", &cxx);
 621
 622         // Building LLVM on MSVC can be a little ludicrous at times. We're so far
 623         // off the beaten path here that I'm not really sure this is even half
 624         // supported any more. Here we're trying to:
 625         //
 626         // * Build LLVM on MSVC
 627         // * Build LLVM with `clang-cl` instead of `cl.exe`
 628         // * Build a project with `sccache`
 629         // * Build for 32-bit as well
 630         // * Build with Ninja
 631         //
 632         // For `cl.exe` there are different binaries to compile 32/64 bit which
 633         // we use but for `clang-cl` there's only one which internally
 634         // multiplexes via flags. As a result it appears that CMake's detection
 635         // of a compiler's architecture and such on MSVC **doesn't** pass any
 636         // custom flags we pass in CMAKE_CXX_FLAGS below. This means that if we
 637         // use `clang-cl.exe` it's always diagnosed as a 64-bit compiler which
 638         // definitely causes problems since all the env vars are pointing to
 639         // 32-bit libraries.
 640         //
 641         // To hack around this... again... we pass an argument that's
 642         // unconditionally passed in the sccache shim. This'll get CMake to
 643         // correctly diagnose it's doing a 32-bit compilation and LLVM will
 644         // internally configure itself appropriately.
 645         if builder.config.llvm_clang_cl.is_some() && target.contains("i686") {
 646             cfg.env("SCCACHE_EXTRA_ARGS", "-m32");
 647         }
 648     } else {
 649         // If ccache is configured we inform the build a little differently how
 650         // to invoke ccache while also invoking our compilers.
 651         if use_compiler_launcher {
 652             if let Some(ref ccache) = builder.config.ccache {
 653                 cfg.define("CMAKE_C_COMPILER_LAUNCHER", ccache)
 654                     .define("CMAKE_CXX_COMPILER_LAUNCHER", ccache);
 655             }
 656         }
 657         cfg.define("CMAKE_C_COMPILER", sanitize_cc(cc))
 658             .define("CMAKE_CXX_COMPILER", sanitize_cc(cxx))
 659             .define("CMAKE_ASM_COMPILER", sanitize_cc(cc));
 660     }
 661
 662     cfg.build_arg("-j").build_arg(builder.jobs().to_string());
 663     let mut cflags: OsString = builder.cflags(target, GitRepo::Llvm, CLang::C).join(" ").into();
 664     if let Some(ref s) = builder.config.llvm_cflags {
 665         cflags.push(" ");
 666         cflags.push(s);
 667     }
 668     // Some compiler features used by LLVM (such as thread locals) will not work on a min version below iOS 10.
 669     if target.contains("apple-ios") {
 670         if target.contains("86-") {
 671             cflags.push(" -miphonesimulator-version-min=10.0");
 672         } else {
 673             cflags.push(" -miphoneos-version-min=10.0");
 674         }
 675     }
 676     if builder.config.llvm_clang_cl.is_some() {
 677         cflags.push(&format!(" --target={}", target));
 678     }
 679     cfg.define("CMAKE_C_FLAGS", cflags);
 680     let mut cxxflags: OsString = builder.cflags(target, GitRepo::Llvm, CLang::Cxx).join(" ").into();
 681     if let Some(ref s) = builder.config.llvm_cxxflags {
 682         cxxflags.push(" ");
 683         cxxflags.push(s);
 684     }
 685     if builder.config.llvm_clang_cl.is_some() {
 686         cxxflags.push(&format!(" --target={}", target));
 687     }
 688     cfg.define("CMAKE_CXX_FLAGS", cxxflags);
 689     if let Some(ar) = builder.ar(target) {
 690         if ar.is_absolute() {
 691             // LLVM build breaks if `CMAKE_AR` is a relative path, for some reason it
 692             // tries to resolve this path in the LLVM build directory.
 693             cfg.define("CMAKE_AR", sanitize_cc(ar));
 694         }
 695     }
 696
 697     if let Some(ranlib) = builder.ranlib(target) {
 698         if ranlib.is_absolute() {
 699             // LLVM build breaks if `CMAKE_RANLIB` is a relative path, for some reason it
 700             // tries to resolve this path in the LLVM build directory.
 701             cfg.define("CMAKE_RANLIB", sanitize_cc(ranlib));
 702         }
 703     }
 704
 705     if let Some(ref flags) = builder.config.llvm_ldflags {
 706         ldflags.push_all(flags);
 707     }
 708
 709     if let Some(flags) = get_var("LDFLAGS", &builder.config.build.triple, &target.triple) {
 710         ldflags.push_all(&flags);
 711     }
 712
 713     // For distribution we want the LLVM tools to be *statically* linked to libstdc++.
 714     // We also do this if the user explicitly requested static libstdc++.
 715     if builder.config.llvm_static_stdcpp {
 716         if !target.contains("msvc") && !target.contains("netbsd") && !target.contains("solaris") {
 717             if target.contains("apple") || target.contains("windows") {
 718                 ldflags.push_all("-static-libstdc++");
 719             } else {
 720                 ldflags.push_all("-Wl,-Bsymbolic -static-libstdc++");
 721             }
 722         }
 723     }
 724
 725     cfg.define("CMAKE_SHARED_LINKER_FLAGS", &ldflags.shared);
 726     cfg.define("CMAKE_MODULE_LINKER_FLAGS", &ldflags.module);
 727     cfg.define("CMAKE_EXE_LINKER_FLAGS", &ldflags.exe);
 728
 729     if env::var_os("SCCACHE_ERROR_LOG").is_some() {
 730         cfg.env("RUSTC_LOG", "sccache=warn");
 731     }
 732 }
 733
 734 // Adapted from https://github.com/alexcrichton/cc-rs/blob/fba7feded71ee4f63cfe885673ead6d7b4f2f454/src/lib.rs#L2347-L2365
 735 fn get_var(var_base: &str, host: &str, target: &str) -> Option<OsString> {
 736     let kind = if host == target { "HOST" } else { "TARGET" };
 737     let target_u = target.replace("-", "_");
 738     env::var_os(&format!("{}_{}", var_base, target))
 739         .or_else(|| env::var_os(&format!("{}_{}", var_base, target_u)))
 740         .or_else(|| env::var_os(&format!("{}_{}", kind, var_base)))
 741         .or_else(|| env::var_os(var_base))
 742 }
 743
 744 #[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
 745 pub struct Lld {
 746     pub target: TargetSelection,
 747 }
 748
 749 impl Step for Lld {
 750     type Output = PathBuf;
 751     const ONLY_HOSTS: bool = true;
 752
 753     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
 754         run.path("src/llvm-project/lld")
 755     }
 756
 757     fn make_run(run: RunConfig<'_>) {
 758         run.builder.ensure(Lld { target: run.target });
 759     }
 760
 761     /// Compile LLD for `target`.
 762     fn run(self, builder: &Builder<'_>) -> PathBuf {
 763         if builder.config.dry_run {
 764             return PathBuf::from("lld-out-dir-test-gen");
 765         }
 766         let target = self.target;
 767
 768         let llvm_config = builder.ensure(Llvm { target: self.target });
 769
 770         let out_dir = builder.lld_out(target);
 771         let done_stamp = out_dir.join("lld-finished-building");
 772         if done_stamp.exists() {
 773             return out_dir;
 774         }
 775
 776         builder.info(&format!("Building LLD for {}", target));
 777         let _time = util::timeit(&builder);
 778         t!(fs::create_dir_all(&out_dir));
 779
 780         let mut cfg = cmake::Config::new(builder.src.join("src/llvm-project/lld"));
 781         let mut ldflags = LdFlags::default();
 782
 783         // When building LLD as part of a build with instrumentation on windows, for example
 784         // when doing PGO on CI, cmake or clang-cl don't automatically link clang's
 785         // profiler runtime in. In that case, we need to manually ask cmake to do it, to avoid
 786         // linking errors, much like LLVM's cmake setup does in that situation.
 787         if builder.config.llvm_profile_generate && target.contains("msvc") {
 788             if let Some(clang_cl_path) = builder.config.llvm_clang_cl.as_ref() {
 789                 // Find clang's runtime library directory and push that as a search path to the
 790                 // cmake linker flags.
 791                 let clang_rt_dir = get_clang_cl_resource_dir(clang_cl_path);
 792                 ldflags.push_all(&format!("/libpath:{}", clang_rt_dir.display()));
 793             }
 794         }
 795
 796         configure_cmake(builder, target, &mut cfg, true, ldflags);
 797
 798         // This is an awful, awful hack. Discovered when we migrated to using
 799         // clang-cl to compile LLVM/LLD it turns out that LLD, when built out of
 800         // tree, will execute `llvm-config --cmakedir` and then tell CMake about
 801         // that directory for later processing. Unfortunately if this path has
 802         // forward slashes in it (which it basically always does on Windows)
 803         // then CMake will hit a syntax error later on as... something isn't
 804         // escaped it seems?
 805         //
 806         // Instead of attempting to fix this problem in upstream CMake and/or
 807         // LLVM/LLD we just hack around it here. This thin wrapper will take the
 808         // output from llvm-config and replace all instances of `\` with `/` to
 809         // ensure we don't hit the same bugs with escaping. It means that you
 810         // can't build on a system where your paths require `\` on Windows, but
 811         // there's probably a lot of reasons you can't do that other than this.
 812         let llvm_config_shim = env::current_exe().unwrap().with_file_name("llvm-config-wrapper");
 813
 814         // Re-use the same flags as llvm to control the level of debug information
 815         // generated for lld.
 816         let profile = match (builder.config.llvm_optimize, builder.config.llvm_release_debuginfo) {
 817             (false, _) => "Debug",
 818             (true, false) => "Release",
 819             (true, true) => "RelWithDebInfo",
 820         };
 821
 822         cfg.out_dir(&out_dir)
 823             .profile(profile)
 824             .env("LLVM_CONFIG_REAL", &llvm_config)
 825             .define("LLVM_CONFIG_PATH", llvm_config_shim)
 826             .define("LLVM_INCLUDE_TESTS", "OFF");
 827
 828         if builder.config.llvm_allow_old_toolchain {
 829             cfg.define("LLVM_TEMPORARILY_ALLOW_OLD_TOOLCHAIN", "YES");
 830         }
 831
 832         // While we're using this horrible workaround to shim the execution of
 833         // llvm-config, let's just pile on more. I can't seem to figure out how
 834         // to build LLD as a standalone project and also cross-compile it at the
 835         // same time. It wants a natively executable `llvm-config` to learn
 836         // about LLVM, but then it learns about all the host configuration of
 837         // LLVM and tries to link to host LLVM libraries.
 838         //
 839         // To work around that we tell our shim to replace anything with the
 840         // build target with the actual target instead. This'll break parts of
 841         // LLD though which try to execute host tools, such as llvm-tblgen, so
 842         // we specifically tell it where to find those. This is likely super
 843         // brittle and will break over time. If anyone knows better how to
 844         // cross-compile LLD it would be much appreciated to fix this!
 845         if target != builder.config.build {
 846             cfg.env("LLVM_CONFIG_SHIM_REPLACE", &builder.config.build.triple)
 847                 .env("LLVM_CONFIG_SHIM_REPLACE_WITH", &target.triple)
 848                 .define(
 849                     "LLVM_TABLEGEN_EXE",
 850                     llvm_config.with_file_name("llvm-tblgen").with_extension(EXE_EXTENSION),
 851                 );
 852         }
 853
 854         // Explicitly set C++ standard, because upstream doesn't do so
 855         // for standalone builds.
 856         cfg.define("CMAKE_CXX_STANDARD", "14");
 857
 858         cfg.build();
 859
 860         t!(File::create(&done_stamp));
 861         out_dir
 862     }
 863 }
 864
 865 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
 866 pub struct TestHelpers {
 867     pub target: TargetSelection,
 868 }
 869
 870 impl Step for TestHelpers {
 871     type Output = ();
 872
 873     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
 874         run.path("src/test/auxiliary/rust_test_helpers.c")
 875     }
 876
 877     fn make_run(run: RunConfig<'_>) {
 878         run.builder.ensure(TestHelpers { target: run.target })
 879     }
 880
 881     /// Compiles the `rust_test_helpers.c` library which we used in various
 882     /// `run-pass` tests for ABI testing.
 883     fn run(self, builder: &Builder<'_>) {
 884         if builder.config.dry_run {
 885             return;
 886         }
 887         // The x86_64-fortanix-unknown-sgx target doesn't have a working C
 888         // toolchain. However, some x86_64 ELF objects can be linked
 889         // without issues. Use this hack to compile the test helpers.
 890         let target = if self.target == "x86_64-fortanix-unknown-sgx" {
 891             TargetSelection::from_user("x86_64-unknown-linux-gnu")
 892         } else {
 893             self.target
 894         };
 895         let dst = builder.test_helpers_out(target);
 896         let src = builder.src.join("src/test/auxiliary/rust_test_helpers.c");
 897         if up_to_date(&src, &dst.join("librust_test_helpers.a")) {
 898             return;
 899         }
 900
 901         builder.info("Building test helpers");
 902         t!(fs::create_dir_all(&dst));
 903         let mut cfg = cc::Build::new();
 904         // FIXME: Workaround for https://github.com/emscripten-core/emscripten/issues/9013
 905         if target.contains("emscripten") {
 906             cfg.pic(false);
 907         }
 908
 909         // We may have found various cross-compilers a little differently due to our
 910         // extra configuration, so inform cc of these compilers. Note, though, that
 911         // on MSVC we still need cc's detection of env vars (ugh).
 912         if !target.contains("msvc") {
 913             if let Some(ar) = builder.ar(target) {
 914                 cfg.archiver(ar);
 915             }
 916             cfg.compiler(builder.cc(target));
 917         }
 918         cfg.cargo_metadata(false)
 919             .out_dir(&dst)
 920             .target(&target.triple)
 921             .host(&builder.config.build.triple)
 922             .opt_level(0)
 923             .warnings(false)
 924             .debug(false)
 925             .file(builder.src.join("src/test/auxiliary/rust_test_helpers.c"))
 926             .compile("rust_test_helpers");
 927     }
 928 }
 929
 930 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
 931 pub struct Sanitizers {
 932     pub target: TargetSelection,
 933 }
 934
 935 impl Step for Sanitizers {
 936     type Output = Vec<SanitizerRuntime>;
 937
 938     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
 939         run.alias("sanitizers")
 940     }
 941
 942     fn make_run(run: RunConfig<'_>) {
 943         run.builder.ensure(Sanitizers { target: run.target });
 944     }
 945
 946     /// Builds sanitizer runtime libraries.
 947     fn run(self, builder: &Builder<'_>) -> Self::Output {
 948         let compiler_rt_dir = builder.src.join("src/llvm-project/compiler-rt");
 949         if !compiler_rt_dir.exists() {
 950             return Vec::new();
 951         }
 952
 953         let out_dir = builder.native_dir(self.target).join("sanitizers");
 954         let runtimes = supported_sanitizers(&out_dir, self.target, &builder.config.channel);
 955         if runtimes.is_empty() {
 956             return runtimes;
 957         }
 958
 959         let llvm_config = builder.ensure(Llvm { target: builder.config.build });
 960         if builder.config.dry_run {
 961             return runtimes;
 962         }
 963
 964         let stamp = out_dir.join("sanitizers-finished-building");
 965         let stamp = HashStamp::new(stamp, builder.in_tree_llvm_info.sha());
 966
 967         if stamp.is_done() {
 968             if stamp.hash.is_none() {
 969                 builder.info(&format!(
 970                     "Rebuild sanitizers by removing the file `{}`",
 971                     stamp.path.display()
 972                 ));
 973             }
 974             return runtimes;
 975         }
 976
 977         builder.info(&format!("Building sanitizers for {}", self.target));
 978         t!(stamp.remove());
 979         let _time = util::timeit(&builder);
 980
 981         let mut cfg = cmake::Config::new(&compiler_rt_dir);
 982         cfg.profile("Release");
 983         cfg.define("CMAKE_C_COMPILER_TARGET", self.target.triple);
 984         cfg.define("COMPILER_RT_BUILD_BUILTINS", "OFF");
 985         cfg.define("COMPILER_RT_BUILD_CRT", "OFF");
 986         cfg.define("COMPILER_RT_BUILD_LIBFUZZER", "OFF");
 987         cfg.define("COMPILER_RT_BUILD_PROFILE", "OFF");
 988         cfg.define("COMPILER_RT_BUILD_SANITIZERS", "ON");
 989         cfg.define("COMPILER_RT_BUILD_XRAY", "OFF");
 990         cfg.define("COMPILER_RT_DEFAULT_TARGET_ONLY", "ON");
 991         cfg.define("COMPILER_RT_USE_LIBCXX", "OFF");
 992         cfg.define("LLVM_CONFIG_PATH", &llvm_config);
 993
 994         // On Darwin targets the sanitizer runtimes are build as universal binaries.
 995         // Unfortunately sccache currently lacks support to build them successfully.
 996         // Disable compiler launcher on Darwin targets to avoid potential issues.
 997         let use_compiler_launcher = !self.target.contains("apple-darwin");
 998         configure_cmake(builder, self.target, &mut cfg, use_compiler_launcher, LdFlags::default());
 999
1000         t!(fs::create_dir_all(&out_dir));
1001         cfg.out_dir(out_dir);
1002
1003         for runtime in &runtimes {
1004             cfg.build_target(&runtime.cmake_target);
1005             cfg.build();
1006         }
1007         t!(stamp.write());
1008
1009         runtimes
1010     }
1011 }
1012
1013 #[derive(Clone, Debug)]
1014 pub struct SanitizerRuntime {
1015     /// CMake target used to build the runtime.
1016     pub cmake_target: String,
1017     /// Path to the built runtime library.
1018     pub path: PathBuf,
1019     /// Library filename that will be used rustc.
1020     pub name: String,
1021 }
1022
1023 /// Returns sanitizers available on a given target.
1024 fn supported_sanitizers(
1025     out_dir: &Path,
1026     target: TargetSelection,
1027     channel: &str,
1028 ) -> Vec<SanitizerRuntime> {
1029     let darwin_libs = |os: &str, components: &[&str]| -> Vec<SanitizerRuntime> {
1030         components
1031             .iter()
1032             .map(move |c| SanitizerRuntime {
1033                 cmake_target: format!("clang_rt.{}_{}_dynamic", c, os),
1034                 path: out_dir
1035                     .join(&format!("build/lib/darwin/libclang_rt.{}_{}_dynamic.dylib", c, os)),
1036                 name: format!("librustc-{}_rt.{}.dylib", channel, c),
1037             })
1038             .collect()
1039     };
1040
1041     let common_libs = |os: &str, arch: &str, components: &[&str]| -> Vec<SanitizerRuntime> {
1042         components
1043             .iter()
1044             .map(move |c| SanitizerRuntime {
1045                 cmake_target: format!("clang_rt.{}-{}", c, arch),
1046                 path: out_dir.join(&format!("build/lib/{}/libclang_rt.{}-{}.a", os, c, arch)),
1047                 name: format!("librustc-{}_rt.{}.a", channel, c),
1048             })
1049             .collect()
1050     };
1051
1052     match &*target.triple {
1053         "aarch64-apple-darwin" => darwin_libs("osx", &["asan", "lsan", "tsan"]),
1054         "aarch64-fuchsia" => common_libs("fuchsia", "aarch64", &["asan"]),
1055         "aarch64-unknown-linux-gnu" => {
1056             common_libs("linux", "aarch64", &["asan", "lsan", "msan", "tsan", "hwasan"])
1057         }
1058         "x86_64-apple-darwin" => darwin_libs("osx", &["asan", "lsan", "tsan"]),
1059         "x86_64-fuchsia" => common_libs("fuchsia", "x86_64", &["asan"]),
1060         "x86_64-unknown-freebsd" => common_libs("freebsd", "x86_64", &["asan", "msan", "tsan"]),
1061         "x86_64-unknown-netbsd" => {
1062             common_libs("netbsd", "x86_64", &["asan", "lsan", "msan", "tsan"])
1063         }
1064         "x86_64-unknown-illumos" => common_libs("illumos", "x86_64", &["asan"]),
1065         "x86_64-pc-solaris" => common_libs("solaris", "x86_64", &["asan"]),
1066         "x86_64-unknown-linux-gnu" => {
1067             common_libs("linux", "x86_64", &["asan", "lsan", "msan", "tsan"])
1068         }
1069         "x86_64-unknown-linux-musl" => {
1070             common_libs("linux", "x86_64", &["asan", "lsan", "msan", "tsan"])
1071         }
1072         _ => Vec::new(),
1073     }
1074 }
1075
1076 struct HashStamp {
1077     path: PathBuf,
1078     hash: Option<Vec<u8>>,
1079 }
1080
1081 impl HashStamp {
1082     fn new(path: PathBuf, hash: Option<&str>) -> Self {
1083         HashStamp { path, hash: hash.map(|s| s.as_bytes().to_owned()) }
1084     }
1085
1086     fn is_done(&self) -> bool {
1087         match fs::read(&self.path) {
1088             Ok(h) => self.hash.as_deref().unwrap_or(b"") == h.as_slice(),
1089             Err(e) if e.kind() == io::ErrorKind::NotFound => false,
1090             Err(e) => {
1091                 panic!("failed to read stamp file `{}`: {}", self.path.display(), e);
1092             }
1093         }
1094     }
1095
1096     fn remove(&self) -> io::Result<()> {
1097         match fs::remove_file(&self.path) {
1098             Ok(()) => Ok(()),
1099             Err(e) => {
1100                 if e.kind() == io::ErrorKind::NotFound {
1101                     Ok(())
1102                 } else {
1103                     Err(e)
1104                 }
1105             }
1106         }
1107     }
1108
1109     fn write(&self) -> io::Result<()> {
1110         fs::write(&self.path, self.hash.as_deref().unwrap_or(b""))
1111     }
1112 }
1113
1114 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
1115 pub struct CrtBeginEnd {
1116     pub target: TargetSelection,
1117 }
1118
1119 impl Step for CrtBeginEnd {
1120     type Output = PathBuf;
1121
1122     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1123         run.path("src/llvm-project/compiler-rt/lib/crt")
1124     }
1125
1126     fn make_run(run: RunConfig<'_>) {
1127         run.builder.ensure(CrtBeginEnd { target: run.target });
1128     }
1129
1130     /// Build crtbegin.o/crtend.o for musl target.
1131     fn run(self, builder: &Builder<'_>) -> Self::Output {
1132         let out_dir = builder.native_dir(self.target).join("crt");
1133
1134         if builder.config.dry_run {
1135             return out_dir;
1136         }
1137
1138         let crtbegin_src = builder.src.join("src/llvm-project/compiler-rt/lib/crt/crtbegin.c");
1139         let crtend_src = builder.src.join("src/llvm-project/compiler-rt/lib/crt/crtend.c");
1140         if up_to_date(&crtbegin_src, &out_dir.join("crtbegin.o"))
1141             && up_to_date(&crtend_src, &out_dir.join("crtendS.o"))
1142         {
1143             return out_dir;
1144         }
1145
1146         builder.info("Building crtbegin.o and crtend.o");
1147         t!(fs::create_dir_all(&out_dir));
1148
1149         let mut cfg = cc::Build::new();
1150
1151         if let Some(ar) = builder.ar(self.target) {
1152             cfg.archiver(ar);
1153         }
1154         cfg.compiler(builder.cc(self.target));
1155         cfg.cargo_metadata(false)
1156             .out_dir(&out_dir)
1157             .target(&self.target.triple)
1158             .host(&builder.config.build.triple)
1159             .warnings(false)
1160             .debug(false)
1161             .opt_level(3)
1162             .file(crtbegin_src)
1163             .file(crtend_src);
1164
1165         // Those flags are defined in src/llvm-project/compiler-rt/lib/crt/CMakeLists.txt
1166         // Currently only consumer of those objects is musl, which use .init_array/.fini_array
1167         // instead of .ctors/.dtors
1168         cfg.flag("-std=c11")
1169             .define("CRT_HAS_INITFINI_ARRAY", None)
1170             .define("EH_USE_FRAME_REGISTRY", None);
1171
1172         cfg.compile("crt");
1173
1174         t!(fs::copy(out_dir.join("crtbegin.o"), out_dir.join("crtbeginS.o")));
1175         t!(fs::copy(out_dir.join("crtend.o"), out_dir.join("crtendS.o")));
1176         out_dir
1177     }
1178 }
1179
1180 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
1181 pub struct Libunwind {
1182     pub target: TargetSelection,
1183 }
1184
1185 impl Step for Libunwind {
1186     type Output = PathBuf;
1187
1188     fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1189         run.path("src/llvm-project/libunwind")
1190     }
1191
1192     fn make_run(run: RunConfig<'_>) {
1193         run.builder.ensure(Libunwind { target: run.target });
1194     }
1195
1196     /// Build linunwind.a
1197     fn run(self, builder: &Builder<'_>) -> Self::Output {
1198         if builder.config.dry_run {
1199             return PathBuf::new();
1200         }
1201
1202         let out_dir = builder.native_dir(self.target).join("libunwind");
1203         let root = builder.src.join("src/llvm-project/libunwind");
1204
1205         if up_to_date(&root, &out_dir.join("libunwind.a")) {
1206             return out_dir;
1207         }
1208
1209         builder.info(&format!("Building libunwind.a for {}", self.target.triple));
1210         t!(fs::create_dir_all(&out_dir));
1211
1212         let mut cc_cfg = cc::Build::new();
1213         let mut cpp_cfg = cc::Build::new();
1214
1215         cpp_cfg.cpp(true);
1216         cpp_cfg.cpp_set_stdlib(None);
1217         cpp_cfg.flag("-nostdinc++");
1218         cpp_cfg.flag("-fno-exceptions");
1219         cpp_cfg.flag("-fno-rtti");
1220         cpp_cfg.flag_if_supported("-fvisibility-global-new-delete-hidden");
1221
1222         for cfg in [&mut cc_cfg, &mut cpp_cfg].iter_mut() {
1223             if let Some(ar) = builder.ar(self.target) {
1224                 cfg.archiver(ar);
1225             }
1226             cfg.target(&self.target.triple);
1227             cfg.host(&builder.config.build.triple);
1228             cfg.warnings(false);
1229             cfg.debug(false);
1230             // get_compiler() need set opt_level first.
1231             cfg.opt_level(3);
1232             cfg.flag("-fstrict-aliasing");
1233             cfg.flag("-funwind-tables");
1234             cfg.flag("-fvisibility=hidden");
1235             cfg.define("_LIBUNWIND_DISABLE_VISIBILITY_ANNOTATIONS", None);
1236             cfg.include(root.join("include"));
1237             cfg.cargo_metadata(false);
1238             cfg.out_dir(&out_dir);
1239
1240             if self.target.contains("x86_64-fortanix-unknown-sgx") {
1241                 cfg.static_flag(true);
1242                 cfg.flag("-fno-stack-protector");
1243                 cfg.flag("-ffreestanding");
1244                 cfg.flag("-fexceptions");
1245
1246                 // easiest way to undefine since no API available in cc::Build to undefine
1247                 cfg.flag("-U_FORTIFY_SOURCE");
1248                 cfg.define("_FORTIFY_SOURCE", "0");
1249                 cfg.define("RUST_SGX", "1");
1250                 cfg.define("__NO_STRING_INLINES", None);
1251                 cfg.define("__NO_MATH_INLINES", None);
1252                 cfg.define("_LIBUNWIND_IS_BAREMETAL", None);
1253                 cfg.define("__LIBUNWIND_IS_NATIVE_ONLY", None);
1254                 cfg.define("NDEBUG", None);
1255             }
1256             if self.target.contains("windows") {
1257                 cfg.define("_LIBUNWIND_HIDE_SYMBOLS", "1");
1258                 cfg.define("_LIBUNWIND_IS_NATIVE_ONLY", "1");
1259             }
1260         }
1261
1262         cc_cfg.compiler(builder.cc(self.target));
1263         if let Ok(cxx) = builder.cxx(self.target) {
1264             cpp_cfg.compiler(cxx);
1265         } else {
1266             cc_cfg.compiler(builder.cc(self.target));
1267         }
1268
1269         // Don't set this for clang
1270         // By default, Clang builds C code in GNU C17 mode.
1271         // By default, Clang builds C++ code according to the C++98 standard,
1272         // with many C++11 features accepted as extensions.
1273         if cc_cfg.get_compiler().is_like_gnu() {
1274             cc_cfg.flag("-std=c99");
1275         }
1276         if cpp_cfg.get_compiler().is_like_gnu() {
1277             cpp_cfg.flag("-std=c++11");
1278         }
1279
1280         if self.target.contains("x86_64-fortanix-unknown-sgx") || self.target.contains("musl") {
1281             // use the same GCC C compiler command to compile C++ code so we do not need to setup the
1282             // C++ compiler env variables on the builders.
1283             // Don't set this for clang++, as clang++ is able to compile this without libc++.
1284             if cpp_cfg.get_compiler().is_like_gnu() {
1285                 cpp_cfg.cpp(false);
1286                 cpp_cfg.compiler(builder.cc(self.target));
1287             }
1288         }
1289
1290         let mut c_sources = vec![
1291             "Unwind-sjlj.c",
1292             "UnwindLevel1-gcc-ext.c",
1293             "UnwindLevel1.c",
1294             "UnwindRegistersRestore.S",
1295             "UnwindRegistersSave.S",
1296         ];
1297
1298         let cpp_sources = vec!["Unwind-EHABI.cpp", "Unwind-seh.cpp", "libunwind.cpp"];
1299         let cpp_len = cpp_sources.len();
1300
1301         if self.target.contains("x86_64-fortanix-unknown-sgx") {
1302             c_sources.push("UnwindRustSgx.c");
1303         }
1304
1305         for src in c_sources {
1306             cc_cfg.file(root.join("src").join(src).canonicalize().unwrap());
1307         }
1308
1309         for src in &cpp_sources {
1310             cpp_cfg.file(root.join("src").join(src).canonicalize().unwrap());
1311         }
1312
1313         cpp_cfg.compile("unwind-cpp");
1314
1315         // FIXME: https://github.com/alexcrichton/cc-rs/issues/545#issuecomment-679242845
1316         let mut count = 0;
1317         for entry in fs::read_dir(&out_dir).unwrap() {
1318             let file = entry.unwrap().path().canonicalize().unwrap();
1319             if file.is_file() && file.extension() == Some(OsStr::new("o")) {
1320                 // file name starts with "Unwind-EHABI", "Unwind-seh" or "libunwind"
1321                 let file_name = file.file_name().unwrap().to_str().expect("UTF-8 file name");
1322                 if cpp_sources.iter().any(|f| file_name.starts_with(&f[..f.len() - 4])) {
1323                     cc_cfg.object(&file);
1324                     count += 1;
1325                 }
1326             }
1327         }
1328         assert_eq!(cpp_len, count, "Can't get object files from {:?}", &out_dir);
1329
1330         cc_cfg.compile("unwind");
1331         out_dir
1332     }
1333 }