bootstrap/core/build_steps/
llvm.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
11use std::env::consts::EXE_EXTENSION;
12use std::ffi::{OsStr, OsString};
13use std::path::{Path, PathBuf};
14use std::sync::OnceLock;
15use std::{env, fs};
16
17use build_helper::git::PathFreshness;
18
19use crate::core::builder::{Builder, RunConfig, ShouldRun, Step, StepMetadata};
20use crate::core::config::{Config, TargetSelection};
21use crate::utils::build_stamp::{BuildStamp, generate_smart_stamp_hash};
22use crate::utils::exec::command;
23use crate::utils::helpers::{
24    self, exe, get_clang_cl_resource_dir, t, unhashed_basename, up_to_date,
25};
26use crate::{CLang, GitRepo, Kind, trace};
27
28#[derive(Clone)]
29pub struct LlvmResult {
30    /// Path to llvm-config binary.
31    /// NB: This is always the host llvm-config!
32    pub host_llvm_config: PathBuf,
33    /// Path to LLVM cmake directory for the target.
34    pub llvm_cmake_dir: PathBuf,
35}
36
37pub struct Meta {
38    stamp: BuildStamp,
39    res: LlvmResult,
40    out_dir: PathBuf,
41    root: String,
42}
43
44pub enum LlvmBuildStatus {
45    AlreadyBuilt(LlvmResult),
46    ShouldBuild(Meta),
47}
48
49impl LlvmBuildStatus {
50    pub fn should_build(&self) -> bool {
51        match self {
52            LlvmBuildStatus::AlreadyBuilt(_) => false,
53            LlvmBuildStatus::ShouldBuild(_) => true,
54        }
55    }
56
57    #[cfg(test)]
58    pub fn llvm_result(&self) -> &LlvmResult {
59        match self {
60            LlvmBuildStatus::AlreadyBuilt(res) => res,
61            LlvmBuildStatus::ShouldBuild(meta) => &meta.res,
62        }
63    }
64}
65
66/// Linker flags to pass to LLVM's CMake invocation.
67#[derive(Debug, Clone, Default)]
68struct LdFlags {
69    /// CMAKE_EXE_LINKER_FLAGS
70    exe: OsString,
71    /// CMAKE_SHARED_LINKER_FLAGS
72    shared: OsString,
73    /// CMAKE_MODULE_LINKER_FLAGS
74    module: OsString,
75}
76
77impl LdFlags {
78    fn push_all(&mut self, s: impl AsRef<OsStr>) {
79        let s = s.as_ref();
80        self.exe.push(" ");
81        self.exe.push(s);
82        self.shared.push(" ");
83        self.shared.push(s);
84        self.module.push(" ");
85        self.module.push(s);
86    }
87}
88
89/// This returns whether we've already previously built LLVM.
90///
91/// It's used to avoid busting caches during x.py check -- if we've already built
92/// LLVM, it's fine for us to not try to avoid doing so.
93///
94/// This will return the llvm-config if it can get it (but it will not build it
95/// if not).
96pub fn prebuilt_llvm_config(
97    builder: &Builder<'_>,
98    target: TargetSelection,
99    // Certain commands (like `x test mir-opt --bless`) may call this function with different targets,
100    // which could bypass the CI LLVM early-return even if `builder.config.llvm_from_ci` is true.
101    // This flag should be `true` only if the caller needs the LLVM sources (e.g., if it will build LLVM).
102    handle_submodule_when_needed: bool,
103) -> LlvmBuildStatus {
104    builder.config.maybe_download_ci_llvm();
105
106    // If we're using a custom LLVM bail out here, but we can only use a
107    // custom LLVM for the build triple.
108    if let Some(config) = builder.config.target_config.get(&target)
109        && let Some(ref s) = config.llvm_config
110    {
111        check_llvm_version(builder, s);
112        let host_llvm_config = s.to_path_buf();
113        let mut llvm_cmake_dir = host_llvm_config.clone();
114        llvm_cmake_dir.pop();
115        llvm_cmake_dir.pop();
116        llvm_cmake_dir.push("lib");
117        llvm_cmake_dir.push("cmake");
118        llvm_cmake_dir.push("llvm");
119        return LlvmBuildStatus::AlreadyBuilt(LlvmResult { host_llvm_config, llvm_cmake_dir });
120    }
121
122    if handle_submodule_when_needed {
123        // If submodules are disabled, this does nothing.
124        builder.config.update_submodule("src/llvm-project");
125    }
126
127    let root = "src/llvm-project/llvm";
128    let out_dir = builder.llvm_out(target);
129
130    let build_llvm_config = if let Some(build_llvm_config) = builder
131        .config
132        .target_config
133        .get(&builder.config.host_target)
134        .and_then(|config| config.llvm_config.clone())
135    {
136        build_llvm_config
137    } else {
138        let mut llvm_config_ret_dir = builder.llvm_out(builder.config.host_target);
139        llvm_config_ret_dir.push("bin");
140        llvm_config_ret_dir.join(exe("llvm-config", builder.config.host_target))
141    };
142
143    let llvm_cmake_dir = out_dir.join("lib/cmake/llvm");
144    let res = LlvmResult { host_llvm_config: build_llvm_config, llvm_cmake_dir };
145
146    static STAMP_HASH_MEMO: OnceLock<String> = OnceLock::new();
147    let smart_stamp_hash = STAMP_HASH_MEMO.get_or_init(|| {
148        generate_smart_stamp_hash(
149            builder,
150            &builder.config.src.join("src/llvm-project"),
151            builder.in_tree_llvm_info.sha().unwrap_or_default(),
152        )
153    });
154
155    let stamp = BuildStamp::new(&out_dir).with_prefix("llvm").add_stamp(smart_stamp_hash);
156
157    if stamp.is_up_to_date() {
158        if stamp.stamp().is_empty() {
159            builder.info(
160                "Could not determine the LLVM submodule commit hash. \
161                     Assuming that an LLVM rebuild is not necessary.",
162            );
163            builder.info(&format!(
164                "To force LLVM to rebuild, remove the file `{}`",
165                stamp.path().display()
166            ));
167        }
168        return LlvmBuildStatus::AlreadyBuilt(res);
169    }
170
171    LlvmBuildStatus::ShouldBuild(Meta { stamp, res, out_dir, root: root.into() })
172}
173
174/// Paths whose changes invalidate LLVM downloads.
175pub const LLVM_INVALIDATION_PATHS: &[&str] = &[
176    "src/llvm-project",
177    "src/bootstrap/download-ci-llvm-stamp",
178    // the LLVM shared object file is named `LLVM-<LLVM-version>-rust-{version}-nightly`
179    "src/version",
180];
181
182/// Detect whether LLVM sources have been modified locally or not.
183pub(crate) fn detect_llvm_freshness(config: &Config, is_git: bool) -> PathFreshness {
184    if is_git {
185        config.check_path_modifications(LLVM_INVALIDATION_PATHS)
186    } else if let Some(info) = crate::utils::channel::read_commit_info_file(&config.src) {
187        PathFreshness::LastModifiedUpstream { upstream: info.sha.trim().to_owned() }
188    } else {
189        PathFreshness::MissingUpstream
190    }
191}
192
193/// Returns whether the CI-found LLVM is currently usable.
194///
195/// This checks the build triple platform to confirm we're usable at all, and if LLVM
196/// with/without assertions is available.
197pub(crate) fn is_ci_llvm_available_for_target(
198    host_target: &TargetSelection,
199    asserts: bool,
200) -> bool {
201    // This is currently all tier 1 targets and tier 2 targets with host tools
202    // (since others may not have CI artifacts)
203    // https://doc.rust-lang.org/rustc/platform-support.html#tier-1
204    let supported_platforms = [
205        // tier 1
206        ("aarch64-unknown-linux-gnu", false),
207        ("aarch64-apple-darwin", false),
208        ("aarch64-pc-windows-msvc", false),
209        ("i686-pc-windows-msvc", false),
210        ("i686-unknown-linux-gnu", false),
211        ("x86_64-unknown-linux-gnu", true),
212        ("x86_64-apple-darwin", true),
213        ("x86_64-pc-windows-gnu", false),
214        ("x86_64-pc-windows-msvc", true),
215        // tier 2 with host tools
216        ("aarch64-unknown-linux-musl", false),
217        ("aarch64-pc-windows-gnullvm", false),
218        ("arm-unknown-linux-gnueabi", false),
219        ("arm-unknown-linux-gnueabihf", false),
220        ("armv7-unknown-linux-gnueabihf", false),
221        ("i686-pc-windows-gnu", false),
222        ("loongarch64-unknown-linux-gnu", false),
223        ("loongarch64-unknown-linux-musl", false),
224        ("powerpc-unknown-linux-gnu", false),
225        ("powerpc64-unknown-linux-gnu", false),
226        ("powerpc64le-unknown-linux-gnu", false),
227        ("powerpc64le-unknown-linux-musl", false),
228        ("riscv64gc-unknown-linux-gnu", false),
229        ("s390x-unknown-linux-gnu", false),
230        ("x86_64-pc-windows-gnullvm", false),
231        ("x86_64-unknown-freebsd", false),
232        ("x86_64-unknown-illumos", false),
233        ("x86_64-unknown-linux-musl", false),
234        ("x86_64-unknown-netbsd", false),
235    ];
236
237    if !supported_platforms.contains(&(&*host_target.triple, asserts))
238        && (asserts || !supported_platforms.contains(&(&*host_target.triple, true)))
239    {
240        return false;
241    }
242
243    true
244}
245
246#[derive(Debug, Clone, Hash, PartialEq, Eq)]
247pub struct Llvm {
248    pub target: TargetSelection,
249}
250
251impl Step for Llvm {
252    type Output = LlvmResult;
253
254    const IS_HOST: bool = true;
255
256    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
257        run.path("src/llvm-project").path("src/llvm-project/llvm")
258    }
259
260    fn make_run(run: RunConfig<'_>) {
261        run.builder.ensure(Llvm { target: run.target });
262    }
263
264    /// Compile LLVM for `target`.
265    fn run(self, builder: &Builder<'_>) -> LlvmResult {
266        let target = self.target;
267        let target_native = if self.target.starts_with("riscv") {
268            // RISC-V target triples in Rust is not named the same as C compiler target triples.
269            // This converts Rust RISC-V target triples to C compiler triples.
270            let idx = target.triple.find('-').unwrap();
271
272            format!("riscv{}{}", &target.triple[5..7], &target.triple[idx..])
273        } else if self.target.starts_with("powerpc") && self.target.ends_with("freebsd") {
274            // FreeBSD 13 had incompatible ABI changes on all PowerPC platforms.
275            // Set the version suffix to 13.0 so the correct target details are used.
276            format!("{}{}", self.target, "13.0")
277        } else {
278            target.to_string()
279        };
280
281        // If LLVM has already been built or been downloaded through download-ci-llvm, we avoid building it again.
282        let Meta { stamp, res, out_dir, root } = match prebuilt_llvm_config(builder, target, true) {
283            LlvmBuildStatus::AlreadyBuilt(p) => return p,
284            LlvmBuildStatus::ShouldBuild(m) => m,
285        };
286
287        if builder.llvm_link_shared() && target.is_windows() && !target.is_windows_gnullvm() {
288            panic!("shared linking to LLVM is not currently supported on {}", target.triple);
289        }
290
291        let _guard = builder.msg_unstaged(Kind::Build, "LLVM", target);
292        t!(stamp.remove());
293        let _time = helpers::timeit(builder);
294        t!(fs::create_dir_all(&out_dir));
295
296        // https://llvm.org/docs/CMake.html
297        let mut cfg = cmake::Config::new(builder.src.join(root));
298        let mut ldflags = LdFlags::default();
299
300        let profile = match (builder.config.llvm_optimize, builder.config.llvm_release_debuginfo) {
301            (false, _) => "Debug",
302            (true, false) => "Release",
303            (true, true) => "RelWithDebInfo",
304        };
305
306        // NOTE: remember to also update `bootstrap.example.toml` when changing the
307        // defaults!
308        let llvm_targets = match &builder.config.llvm_targets {
309            Some(s) => s,
310            None => {
311                "AArch64;AMDGPU;ARM;BPF;Hexagon;LoongArch;MSP430;Mips;NVPTX;PowerPC;RISCV;\
312                     Sparc;SystemZ;WebAssembly;X86"
313            }
314        };
315
316        let llvm_exp_targets = match builder.config.llvm_experimental_targets {
317            Some(ref s) => s,
318            None => "AVR;M68k;CSKY;Xtensa",
319        };
320
321        let assertions = if builder.config.llvm_assertions { "ON" } else { "OFF" };
322        let plugins = if builder.config.llvm_plugins { "ON" } else { "OFF" };
323        let enable_tests = if builder.config.llvm_tests { "ON" } else { "OFF" };
324        let enable_warnings = if builder.config.llvm_enable_warnings { "ON" } else { "OFF" };
325
326        cfg.out_dir(&out_dir)
327            .profile(profile)
328            .define("LLVM_ENABLE_ASSERTIONS", assertions)
329            .define("LLVM_UNREACHABLE_OPTIMIZE", "OFF")
330            .define("LLVM_ENABLE_PLUGINS", plugins)
331            .define("LLVM_TARGETS_TO_BUILD", llvm_targets)
332            .define("LLVM_EXPERIMENTAL_TARGETS_TO_BUILD", llvm_exp_targets)
333            .define("LLVM_INCLUDE_EXAMPLES", "OFF")
334            .define("LLVM_INCLUDE_DOCS", "OFF")
335            .define("LLVM_INCLUDE_BENCHMARKS", "OFF")
336            .define("LLVM_INCLUDE_TESTS", enable_tests)
337            .define("LLVM_ENABLE_LIBEDIT", "OFF")
338            .define("LLVM_ENABLE_BINDINGS", "OFF")
339            .define("LLVM_ENABLE_Z3_SOLVER", "OFF")
340            .define("LLVM_PARALLEL_COMPILE_JOBS", builder.jobs().to_string())
341            .define("LLVM_TARGET_ARCH", target_native.split('-').next().unwrap())
342            .define("LLVM_DEFAULT_TARGET_TRIPLE", target_native)
343            .define("LLVM_ENABLE_WARNINGS", enable_warnings);
344
345        // Parts of our test suite rely on the `FileCheck` tool, which is built by default in
346        // `build/$TARGET/llvm/build/bin` is but *not* then installed to `build/$TARGET/llvm/bin`.
347        // This flag makes sure `FileCheck` is copied in the final binaries directory.
348        cfg.define("LLVM_INSTALL_UTILS", "ON");
349
350        if builder.config.llvm_profile_generate {
351            cfg.define("LLVM_BUILD_INSTRUMENTED", "IR");
352            if let Ok(llvm_profile_dir) = std::env::var("LLVM_PROFILE_DIR") {
353                cfg.define("LLVM_PROFILE_DATA_DIR", llvm_profile_dir);
354            }
355            cfg.define("LLVM_BUILD_RUNTIME", "No");
356        }
357        if let Some(path) = builder.config.llvm_profile_use.as_ref() {
358            cfg.define("LLVM_PROFDATA_FILE", path);
359        }
360
361        // Libraries for ELF section compression and profraw files merging.
362        if !target.is_msvc() {
363            cfg.define("LLVM_ENABLE_ZLIB", "ON");
364        } else {
365            cfg.define("LLVM_ENABLE_ZLIB", "OFF");
366        }
367
368        // Are we compiling for iOS/tvOS/watchOS/visionOS?
369        if target.contains("apple-ios")
370            || target.contains("apple-tvos")
371            || target.contains("apple-watchos")
372            || target.contains("apple-visionos")
373        {
374            // Prevent cmake from adding -bundle to CFLAGS automatically, which leads to a compiler error because "-bitcode_bundle" also gets added.
375            cfg.define("LLVM_ENABLE_PLUGINS", "OFF");
376            // Zlib fails to link properly, leading to a compiler error.
377            cfg.define("LLVM_ENABLE_ZLIB", "OFF");
378        }
379
380        // This setting makes the LLVM tools link to the dynamic LLVM library,
381        // which saves both memory during parallel links and overall disk space
382        // for the tools. We don't do this on every platform as it doesn't work
383        // equally well everywhere.
384        if builder.llvm_link_shared() {
385            cfg.define("LLVM_LINK_LLVM_DYLIB", "ON");
386        }
387
388        if (target.starts_with("csky")
389            || target.starts_with("riscv")
390            || target.starts_with("sparc-"))
391            && !target.contains("freebsd")
392            && !target.contains("openbsd")
393            && !target.contains("netbsd")
394        {
395            // CSKY and RISC-V GCC erroneously requires linking against
396            // `libatomic` when using 1-byte and 2-byte C++
397            // atomics but the LLVM build system check cannot
398            // detect this. Therefore it is set manually here.
399            // Some BSD uses Clang as its system compiler and
400            // provides no libatomic in its base system so does
401            // not want this. 32-bit SPARC requires linking against
402            // libatomic as well.
403            ldflags.exe.push(" -latomic");
404            ldflags.shared.push(" -latomic");
405        }
406
407        if target.starts_with("mips") && target.contains("netbsd") {
408            // LLVM wants 64-bit atomics, while mipsel is 32-bit only, so needs -latomic
409            ldflags.exe.push(" -latomic");
410            ldflags.shared.push(" -latomic");
411        }
412
413        if target.starts_with("arm64ec") {
414            // MSVC linker requires the -machine:arm64ec flag to be passed to
415            // know it's linking as Arm64EC (vs Arm64X).
416            ldflags.exe.push(" -machine:arm64ec");
417            ldflags.shared.push(" -machine:arm64ec");
418        }
419
420        if target.is_msvc() {
421            cfg.define("CMAKE_MSVC_RUNTIME_LIBRARY", "MultiThreaded");
422            cfg.static_crt(true);
423        }
424
425        if target.starts_with("i686") {
426            cfg.define("LLVM_BUILD_32_BITS", "ON");
427        }
428
429        if target.starts_with("x86_64") && target.contains("ohos") {
430            cfg.define("LLVM_TOOL_LLVM_RTDYLD_BUILD", "OFF");
431        }
432
433        let mut enabled_llvm_projects = Vec::new();
434
435        if helpers::forcing_clang_based_tests() {
436            enabled_llvm_projects.push("clang");
437        }
438
439        if builder.config.llvm_polly {
440            enabled_llvm_projects.push("polly");
441        }
442
443        if builder.config.llvm_clang {
444            enabled_llvm_projects.push("clang");
445        }
446
447        // We want libxml to be disabled.
448        // See https://github.com/rust-lang/rust/pull/50104
449        cfg.define("LLVM_ENABLE_LIBXML2", "OFF");
450
451        let mut enabled_llvm_runtimes = Vec::new();
452
453        if helpers::forcing_clang_based_tests() {
454            enabled_llvm_runtimes.push("compiler-rt");
455        }
456
457        // This is an experimental flag, which likely builds more than necessary.
458        // We will optimize it when we get closer to releasing it on nightly.
459        if builder.config.llvm_offload {
460            enabled_llvm_runtimes.push("offload");
461            //FIXME(ZuseZ4): LLVM intends to drop the offload dependency on openmp.
462            //Remove this line once they achieved it.
463            enabled_llvm_runtimes.push("openmp");
464            enabled_llvm_projects.push("compiler-rt");
465        }
466
467        if !enabled_llvm_projects.is_empty() {
468            enabled_llvm_projects.sort();
469            enabled_llvm_projects.dedup();
470            cfg.define("LLVM_ENABLE_PROJECTS", enabled_llvm_projects.join(";"));
471        }
472
473        if !enabled_llvm_runtimes.is_empty() {
474            enabled_llvm_runtimes.sort();
475            enabled_llvm_runtimes.dedup();
476            cfg.define("LLVM_ENABLE_RUNTIMES", enabled_llvm_runtimes.join(";"));
477        }
478
479        if let Some(num_linkers) = builder.config.llvm_link_jobs
480            && num_linkers > 0
481        {
482            cfg.define("LLVM_PARALLEL_LINK_JOBS", num_linkers.to_string());
483        }
484
485        // https://llvm.org/docs/HowToCrossCompileLLVM.html
486        if !builder.config.is_host_target(target) {
487            let LlvmResult { host_llvm_config, .. } =
488                builder.ensure(Llvm { target: builder.config.host_target });
489            if !builder.config.dry_run() {
490                let llvm_bindir = command(&host_llvm_config)
491                    .arg("--bindir")
492                    .cached()
493                    .run_capture_stdout(builder)
494                    .stdout();
495                let host_bin = Path::new(llvm_bindir.trim());
496                cfg.define(
497                    "LLVM_TABLEGEN",
498                    host_bin.join("llvm-tblgen").with_extension(EXE_EXTENSION),
499                );
500                // LLVM_NM is required for cross compiling using MSVC
501                cfg.define("LLVM_NM", host_bin.join("llvm-nm").with_extension(EXE_EXTENSION));
502            }
503            cfg.define("LLVM_CONFIG_PATH", host_llvm_config);
504            if builder.config.llvm_clang {
505                let build_bin =
506                    builder.llvm_out(builder.config.host_target).join("build").join("bin");
507                let clang_tblgen = build_bin.join("clang-tblgen").with_extension(EXE_EXTENSION);
508                if !builder.config.dry_run() && !clang_tblgen.exists() {
509                    panic!("unable to find {}", clang_tblgen.display());
510                }
511                cfg.define("CLANG_TABLEGEN", clang_tblgen);
512            }
513        }
514
515        let llvm_version_suffix = if let Some(ref suffix) = builder.config.llvm_version_suffix {
516            // Allow version-suffix="" to not define a version suffix at all.
517            if !suffix.is_empty() { Some(suffix.to_string()) } else { None }
518        } else if builder.config.channel == "dev" {
519            // Changes to a version suffix require a complete rebuild of the LLVM.
520            // To avoid rebuilds during a time of version bump, don't include rustc
521            // release number on the dev channel.
522            Some("-rust-dev".to_string())
523        } else {
524            Some(format!("-rust-{}-{}", builder.version, builder.config.channel))
525        };
526        if let Some(ref suffix) = llvm_version_suffix {
527            cfg.define("LLVM_VERSION_SUFFIX", suffix);
528        }
529
530        configure_cmake(builder, target, &mut cfg, true, ldflags, &[]);
531        configure_llvm(builder, target, &mut cfg);
532
533        for (key, val) in &builder.config.llvm_build_config {
534            cfg.define(key, val);
535        }
536
537        if builder.config.dry_run() {
538            return res;
539        }
540
541        cfg.build();
542
543        // Helper to find the name of LLVM's shared library on darwin and linux.
544        let find_llvm_lib_name = |extension| {
545            let major = get_llvm_version_major(builder, &res.host_llvm_config);
546            match &llvm_version_suffix {
547                Some(version_suffix) => format!("libLLVM-{major}{version_suffix}.{extension}"),
548                None => format!("libLLVM-{major}.{extension}"),
549            }
550        };
551
552        // FIXME(ZuseZ4): Do we need that for Enzyme too?
553        // When building LLVM with LLVM_LINK_LLVM_DYLIB for macOS, an unversioned
554        // libLLVM.dylib will be built. However, llvm-config will still look
555        // for a versioned path like libLLVM-14.dylib. Manually create a symbolic
556        // link to make llvm-config happy.
557        if builder.llvm_link_shared() && target.contains("apple-darwin") {
558            let lib_name = find_llvm_lib_name("dylib");
559            let lib_llvm = out_dir.join("build").join("lib").join(lib_name);
560            if !lib_llvm.exists() {
561                t!(builder.symlink_file("libLLVM.dylib", &lib_llvm));
562            }
563        }
564
565        // When building LLVM as a shared library on linux, it can contain unexpected debuginfo:
566        // some can come from the C++ standard library. Unless we're explicitly requesting LLVM to
567        // be built with debuginfo, strip it away after the fact, to make dist artifacts smaller.
568        if builder.llvm_link_shared()
569            && builder.config.llvm_optimize
570            && !builder.config.llvm_release_debuginfo
571        {
572            // Find the name of the LLVM shared library that we just built.
573            let lib_name = find_llvm_lib_name("so");
574
575            // If the shared library exists in LLVM's `/build/lib/` or `/lib/` folders, strip its
576            // debuginfo.
577            crate::core::build_steps::compile::strip_debug(
578                builder,
579                target,
580                &out_dir.join("lib").join(&lib_name),
581            );
582            crate::core::build_steps::compile::strip_debug(
583                builder,
584                target,
585                &out_dir.join("build").join("lib").join(&lib_name),
586            );
587        }
588
589        t!(stamp.write());
590
591        res
592    }
593
594    fn metadata(&self) -> Option<StepMetadata> {
595        Some(StepMetadata::build("llvm", self.target))
596    }
597}
598
599pub fn get_llvm_version(builder: &Builder<'_>, llvm_config: &Path) -> String {
600    command(llvm_config)
601        .arg("--version")
602        .cached()
603        .run_capture_stdout(builder)
604        .stdout()
605        .trim()
606        .to_owned()
607}
608
609pub fn get_llvm_version_major(builder: &Builder<'_>, llvm_config: &Path) -> u8 {
610    let version = get_llvm_version(builder, llvm_config);
611    let major_str = version.split_once('.').expect("Failed to parse LLVM version").0;
612    major_str.parse().unwrap()
613}
614
615fn check_llvm_version(builder: &Builder<'_>, llvm_config: &Path) {
616    if builder.config.dry_run() {
617        return;
618    }
619
620    let version = get_llvm_version(builder, llvm_config);
621    let mut parts = version.split('.').take(2).filter_map(|s| s.parse::<u32>().ok());
622    if let (Some(major), Some(_minor)) = (parts.next(), parts.next())
623        && major >= 20
624    {
625        return;
626    }
627    panic!("\n\nbad LLVM version: {version}, need >=20\n\n")
628}
629
630fn configure_cmake(
631    builder: &Builder<'_>,
632    target: TargetSelection,
633    cfg: &mut cmake::Config,
634    use_compiler_launcher: bool,
635    mut ldflags: LdFlags,
636    suppressed_compiler_flag_prefixes: &[&str],
637) {
638    // Do not print installation messages for up-to-date files.
639    // LLVM and LLD builds can produce a lot of those and hit CI limits on log size.
640    cfg.define("CMAKE_INSTALL_MESSAGE", "LAZY");
641
642    // Do not allow the user's value of DESTDIR to influence where
643    // LLVM will install itself. LLVM must always be installed in our
644    // own build directories.
645    cfg.env("DESTDIR", "");
646
647    if builder.ninja() {
648        cfg.generator("Ninja");
649    }
650    cfg.target(&target.triple).host(&builder.config.host_target.triple);
651
652    if !builder.config.is_host_target(target) {
653        cfg.define("CMAKE_CROSSCOMPILING", "True");
654
655        // NOTE: Ideally, we wouldn't have to do this, and `cmake-rs` would just handle it for us.
656        // But it currently determines this based on the `CARGO_CFG_TARGET_OS` environment variable,
657        // which isn't set when compiling outside `build.rs` (like bootstrap is).
658        //
659        // So for now, we define `CMAKE_SYSTEM_NAME` ourselves, to panicking in `cmake-rs`.
660        if target.contains("netbsd") {
661            cfg.define("CMAKE_SYSTEM_NAME", "NetBSD");
662        } else if target.contains("dragonfly") {
663            cfg.define("CMAKE_SYSTEM_NAME", "DragonFly");
664        } else if target.contains("openbsd") {
665            cfg.define("CMAKE_SYSTEM_NAME", "OpenBSD");
666        } else if target.contains("freebsd") {
667            cfg.define("CMAKE_SYSTEM_NAME", "FreeBSD");
668        } else if target.is_windows() {
669            cfg.define("CMAKE_SYSTEM_NAME", "Windows");
670        } else if target.contains("haiku") {
671            cfg.define("CMAKE_SYSTEM_NAME", "Haiku");
672        } else if target.contains("solaris") || target.contains("illumos") {
673            cfg.define("CMAKE_SYSTEM_NAME", "SunOS");
674        } else if target.contains("linux") {
675            cfg.define("CMAKE_SYSTEM_NAME", "Linux");
676        } else if target.contains("darwin") {
677            // macOS
678            cfg.define("CMAKE_SYSTEM_NAME", "Darwin");
679        } else if target.contains("ios") {
680            cfg.define("CMAKE_SYSTEM_NAME", "iOS");
681        } else if target.contains("tvos") {
682            cfg.define("CMAKE_SYSTEM_NAME", "tvOS");
683        } else if target.contains("visionos") {
684            cfg.define("CMAKE_SYSTEM_NAME", "visionOS");
685        } else if target.contains("watchos") {
686            cfg.define("CMAKE_SYSTEM_NAME", "watchOS");
687        } else if target.contains("none") {
688            // "none" should be the last branch
689            cfg.define("CMAKE_SYSTEM_NAME", "Generic");
690        } else {
691            builder.info(&format!(
692                "could not determine CMAKE_SYSTEM_NAME from the target `{target}`, build may fail",
693            ));
694            // Fallback, set `CMAKE_SYSTEM_NAME` anyhow to avoid the logic `cmake-rs` tries, and
695            // to avoid CMAKE_SYSTEM_NAME being inferred from the host.
696            cfg.define("CMAKE_SYSTEM_NAME", "Generic");
697        }
698
699        // When cross-compiling we should also set CMAKE_SYSTEM_VERSION, but in
700        // that case like CMake we cannot easily determine system version either.
701        //
702        // Since, the LLVM itself makes rather limited use of version checks in
703        // CMakeFiles (and then only in tests), and so far no issues have been
704        // reported, the system version is currently left unset.
705
706        if target.contains("apple") {
707            if !target.contains("darwin") {
708                // FIXME(madsmtm): compiler-rt's CMake setup is kinda weird, it seems like they do
709                // version testing etc. for macOS (i.e. Darwin), even while building for iOS?
710                //
711                // So for now we set it to "Darwin" on all Apple platforms.
712                cfg.define("CMAKE_SYSTEM_NAME", "Darwin");
713
714                // These two defines prevent CMake from automatically trying to add a MacOSX sysroot, which leads to a compiler error.
715                cfg.define("CMAKE_OSX_SYSROOT", "/");
716                cfg.define("CMAKE_OSX_DEPLOYMENT_TARGET", "");
717            }
718
719            // Make sure that CMake does not build universal binaries on macOS.
720            // Explicitly specify the one single target architecture.
721            if target.starts_with("aarch64") {
722                // macOS uses a different name for building arm64
723                cfg.define("CMAKE_OSX_ARCHITECTURES", "arm64");
724            } else if target.starts_with("i686") {
725                // macOS uses a different name for building i386
726                cfg.define("CMAKE_OSX_ARCHITECTURES", "i386");
727            } else {
728                cfg.define("CMAKE_OSX_ARCHITECTURES", target.triple.split('-').next().unwrap());
729            }
730        }
731    }
732
733    let sanitize_cc = |cc: &Path| {
734        if target.is_msvc() {
735            OsString::from(cc.to_str().unwrap().replace('\\', "/"))
736        } else {
737            cc.as_os_str().to_owned()
738        }
739    };
740
741    // MSVC with CMake uses msbuild by default which doesn't respect these
742    // vars that we'd otherwise configure. In that case we just skip this
743    // entirely.
744    if target.is_msvc() && !builder.ninja() {
745        return;
746    }
747
748    let (cc, cxx) = match builder.config.llvm_clang_cl {
749        Some(ref cl) => (cl.into(), cl.into()),
750        None => (builder.cc(target), builder.cxx(target).unwrap()),
751    };
752
753    // If ccache is configured we inform the build a little differently how
754    // to invoke ccache while also invoking our compilers.
755    if use_compiler_launcher && let Some(ref ccache) = builder.config.ccache {
756        cfg.define("CMAKE_C_COMPILER_LAUNCHER", ccache)
757            .define("CMAKE_CXX_COMPILER_LAUNCHER", ccache);
758    }
759    cfg.define("CMAKE_C_COMPILER", sanitize_cc(&cc))
760        .define("CMAKE_CXX_COMPILER", sanitize_cc(&cxx))
761        .define("CMAKE_ASM_COMPILER", sanitize_cc(&cc));
762
763    cfg.build_arg("-j").build_arg(builder.jobs().to_string());
764    // FIXME(madsmtm): Allow `cmake-rs` to select flags by itself by passing
765    // our flags via `.cflag`/`.cxxflag` instead.
766    //
767    // Needs `suppressed_compiler_flag_prefixes` to be gone, and hence
768    // https://github.com/llvm/llvm-project/issues/88780 to be fixed.
769    let mut cflags: OsString = builder
770        .cc_handled_clags(target, CLang::C)
771        .into_iter()
772        .chain(builder.cc_unhandled_cflags(target, GitRepo::Llvm, CLang::C))
773        .filter(|flag| {
774            !suppressed_compiler_flag_prefixes
775                .iter()
776                .any(|suppressed_prefix| flag.starts_with(suppressed_prefix))
777        })
778        .collect::<Vec<String>>()
779        .join(" ")
780        .into();
781    if let Some(ref s) = builder.config.llvm_cflags {
782        cflags.push(" ");
783        cflags.push(s);
784    }
785    if target.contains("ohos") {
786        cflags.push(" -D_LINUX_SYSINFO_H");
787    }
788    if builder.config.llvm_clang_cl.is_some() {
789        cflags.push(format!(" --target={target}"));
790    }
791    cfg.define("CMAKE_C_FLAGS", cflags);
792    let mut cxxflags: OsString = builder
793        .cc_handled_clags(target, CLang::Cxx)
794        .into_iter()
795        .chain(builder.cc_unhandled_cflags(target, GitRepo::Llvm, CLang::Cxx))
796        .filter(|flag| {
797            !suppressed_compiler_flag_prefixes
798                .iter()
799                .any(|suppressed_prefix| flag.starts_with(suppressed_prefix))
800        })
801        .collect::<Vec<String>>()
802        .join(" ")
803        .into();
804    if let Some(ref s) = builder.config.llvm_cxxflags {
805        cxxflags.push(" ");
806        cxxflags.push(s);
807    }
808    if target.contains("ohos") {
809        cxxflags.push(" -D_LINUX_SYSINFO_H");
810    }
811    if builder.config.llvm_clang_cl.is_some() {
812        cxxflags.push(format!(" --target={target}"));
813    }
814    cfg.define("CMAKE_CXX_FLAGS", cxxflags);
815    if let Some(ar) = builder.ar(target)
816        && ar.is_absolute()
817    {
818        // LLVM build breaks if `CMAKE_AR` is a relative path, for some reason it
819        // tries to resolve this path in the LLVM build directory.
820        cfg.define("CMAKE_AR", sanitize_cc(&ar));
821    }
822
823    if let Some(ranlib) = builder.ranlib(target)
824        && ranlib.is_absolute()
825    {
826        // LLVM build breaks if `CMAKE_RANLIB` is a relative path, for some reason it
827        // tries to resolve this path in the LLVM build directory.
828        cfg.define("CMAKE_RANLIB", sanitize_cc(&ranlib));
829    }
830
831    if let Some(ref flags) = builder.config.llvm_ldflags {
832        ldflags.push_all(flags);
833    }
834
835    if let Some(flags) = get_var("LDFLAGS", &builder.config.host_target.triple, &target.triple) {
836        ldflags.push_all(&flags);
837    }
838
839    // For distribution we want the LLVM tools to be *statically* linked to libstdc++.
840    // We also do this if the user explicitly requested static libstdc++.
841    if builder.config.llvm_static_stdcpp
842        && !target.is_msvc()
843        && !target.contains("netbsd")
844        && !target.contains("solaris")
845    {
846        if target.contains("apple") || target.is_windows() {
847            ldflags.push_all("-static-libstdc++");
848        } else {
849            ldflags.push_all("-Wl,-Bsymbolic -static-libstdc++");
850        }
851    }
852
853    cfg.define("CMAKE_SHARED_LINKER_FLAGS", &ldflags.shared);
854    cfg.define("CMAKE_MODULE_LINKER_FLAGS", &ldflags.module);
855    cfg.define("CMAKE_EXE_LINKER_FLAGS", &ldflags.exe);
856
857    if env::var_os("SCCACHE_ERROR_LOG").is_some() {
858        cfg.env("RUSTC_LOG", "sccache=warn");
859    }
860}
861
862fn configure_llvm(builder: &Builder<'_>, target: TargetSelection, cfg: &mut cmake::Config) {
863    // ThinLTO is only available when building with LLVM, enabling LLD is required.
864    // Apple's linker ld64 supports ThinLTO out of the box though, so don't use LLD on Darwin.
865    if builder.config.llvm_thin_lto {
866        cfg.define("LLVM_ENABLE_LTO", "Thin");
867        if !target.contains("apple") {
868            cfg.define("LLVM_ENABLE_LLD", "ON");
869        }
870    }
871
872    // Libraries for ELF section compression.
873    if builder.config.llvm_libzstd {
874        cfg.define("LLVM_ENABLE_ZSTD", "FORCE_ON");
875        cfg.define("LLVM_USE_STATIC_ZSTD", "TRUE");
876    } else {
877        cfg.define("LLVM_ENABLE_ZSTD", "OFF");
878    }
879
880    if let Some(ref linker) = builder.config.llvm_use_linker {
881        cfg.define("LLVM_USE_LINKER", linker);
882    }
883
884    if builder.config.llvm_allow_old_toolchain {
885        cfg.define("LLVM_TEMPORARILY_ALLOW_OLD_TOOLCHAIN", "YES");
886    }
887}
888
889// Adapted from https://github.com/alexcrichton/cc-rs/blob/fba7feded71ee4f63cfe885673ead6d7b4f2f454/src/lib.rs#L2347-L2365
890fn get_var(var_base: &str, host: &str, target: &str) -> Option<OsString> {
891    let kind = if host == target { "HOST" } else { "TARGET" };
892    let target_u = target.replace('-', "_");
893    env::var_os(format!("{var_base}_{target}"))
894        .or_else(|| env::var_os(format!("{var_base}_{target_u}")))
895        .or_else(|| env::var_os(format!("{kind}_{var_base}")))
896        .or_else(|| env::var_os(var_base))
897}
898
899#[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
900pub struct Enzyme {
901    pub target: TargetSelection,
902}
903
904impl Step for Enzyme {
905    type Output = PathBuf;
906    const IS_HOST: bool = true;
907
908    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
909        run.path("src/tools/enzyme/enzyme")
910    }
911
912    fn make_run(run: RunConfig<'_>) {
913        run.builder.ensure(Enzyme { target: run.target });
914    }
915
916    /// Compile Enzyme for `target`.
917    fn run(self, builder: &Builder<'_>) -> PathBuf {
918        builder.require_submodule(
919            "src/tools/enzyme",
920            Some("The Enzyme sources are required for autodiff."),
921        );
922        if builder.config.dry_run() {
923            let out_dir = builder.enzyme_out(self.target);
924            return out_dir;
925        }
926        let target = self.target;
927
928        let LlvmResult { host_llvm_config, .. } = builder.ensure(Llvm { target: self.target });
929
930        static STAMP_HASH_MEMO: OnceLock<String> = OnceLock::new();
931        let smart_stamp_hash = STAMP_HASH_MEMO.get_or_init(|| {
932            generate_smart_stamp_hash(
933                builder,
934                &builder.config.src.join("src/tools/enzyme"),
935                builder.enzyme_info.sha().unwrap_or_default(),
936            )
937        });
938
939        let out_dir = builder.enzyme_out(target);
940        let stamp = BuildStamp::new(&out_dir).with_prefix("enzyme").add_stamp(smart_stamp_hash);
941
942        trace!("checking build stamp to see if we need to rebuild enzyme artifacts");
943        if stamp.is_up_to_date() {
944            trace!(?out_dir, "enzyme build artifacts are up to date");
945            if stamp.stamp().is_empty() {
946                builder.info(
947                    "Could not determine the Enzyme submodule commit hash. \
948                     Assuming that an Enzyme rebuild is not necessary.",
949                );
950                builder.info(&format!(
951                    "To force Enzyme to rebuild, remove the file `{}`",
952                    stamp.path().display()
953                ));
954            }
955            return out_dir;
956        }
957
958        trace!(?target, "(re)building enzyme artifacts");
959        builder.info(&format!("Building Enzyme for {target}"));
960        t!(stamp.remove());
961        let _time = helpers::timeit(builder);
962        t!(fs::create_dir_all(&out_dir));
963
964        builder
965            .config
966            .update_submodule(Path::new("src").join("tools").join("enzyme").to_str().unwrap());
967        let mut cfg = cmake::Config::new(builder.src.join("src/tools/enzyme/enzyme/"));
968        configure_cmake(builder, target, &mut cfg, true, LdFlags::default(), &[]);
969
970        // Re-use the same flags as llvm to control the level of debug information
971        // generated by Enzyme.
972        // FIXME(ZuseZ4): Find a nicer way to use Enzyme Debug builds.
973        let profile = match (builder.config.llvm_optimize, builder.config.llvm_release_debuginfo) {
974            (false, _) => "Debug",
975            (true, false) => "Release",
976            (true, true) => "RelWithDebInfo",
977        };
978        trace!(?profile);
979
980        cfg.out_dir(&out_dir)
981            .profile(profile)
982            .env("LLVM_CONFIG_REAL", &host_llvm_config)
983            .define("LLVM_ENABLE_ASSERTIONS", "ON")
984            .define("ENZYME_EXTERNAL_SHARED_LIB", "ON")
985            .define("ENZYME_BC_LOADER", "OFF")
986            .define("LLVM_DIR", builder.llvm_out(target));
987
988        cfg.build();
989
990        t!(stamp.write());
991        out_dir
992    }
993}
994
995#[derive(Debug, Clone, Hash, PartialEq, Eq)]
996pub struct Lld {
997    pub target: TargetSelection,
998}
999
1000impl Step for Lld {
1001    type Output = PathBuf;
1002    const IS_HOST: bool = true;
1003
1004    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1005        run.path("src/llvm-project/lld")
1006    }
1007
1008    fn make_run(run: RunConfig<'_>) {
1009        run.builder.ensure(Lld { target: run.target });
1010    }
1011
1012    /// Compile LLD for `target`.
1013    fn run(self, builder: &Builder<'_>) -> PathBuf {
1014        if builder.config.dry_run() {
1015            return PathBuf::from("lld-out-dir-test-gen");
1016        }
1017        let target = self.target;
1018
1019        let LlvmResult { host_llvm_config, llvm_cmake_dir } = builder.ensure(Llvm { target });
1020
1021        // The `dist` step packages LLD next to LLVM's binaries for download-ci-llvm. The root path
1022        // we usually expect here is `./build/$triple/ci-llvm/`, with the binaries in its `bin`
1023        // subfolder. We check if that's the case, and if LLD's binary already exists there next to
1024        // `llvm-config`: if so, we can use it instead of building LLVM/LLD from source.
1025        let ci_llvm_bin = host_llvm_config.parent().unwrap();
1026        if ci_llvm_bin.is_dir() && ci_llvm_bin.file_name().unwrap() == "bin" {
1027            let lld_path = ci_llvm_bin.join(exe("lld", target));
1028            if lld_path.exists() {
1029                // The following steps copying `lld` as `rust-lld` to the sysroot, expect it in the
1030                // `bin` subfolder of this step's out dir.
1031                return ci_llvm_bin.parent().unwrap().to_path_buf();
1032            }
1033        }
1034
1035        let out_dir = builder.lld_out(target);
1036
1037        let lld_stamp = BuildStamp::new(&out_dir).with_prefix("lld");
1038        if lld_stamp.path().exists() {
1039            return out_dir;
1040        }
1041
1042        let _guard = builder.msg_unstaged(Kind::Build, "LLD", target);
1043        let _time = helpers::timeit(builder);
1044        t!(fs::create_dir_all(&out_dir));
1045
1046        let mut cfg = cmake::Config::new(builder.src.join("src/llvm-project/lld"));
1047        let mut ldflags = LdFlags::default();
1048
1049        // When building LLD as part of a build with instrumentation on windows, for example
1050        // when doing PGO on CI, cmake or clang-cl don't automatically link clang's
1051        // profiler runtime in. In that case, we need to manually ask cmake to do it, to avoid
1052        // linking errors, much like LLVM's cmake setup does in that situation.
1053        if builder.config.llvm_profile_generate
1054            && target.is_msvc()
1055            && let Some(clang_cl_path) = builder.config.llvm_clang_cl.as_ref()
1056        {
1057            // Find clang's runtime library directory and push that as a search path to the
1058            // cmake linker flags.
1059            let clang_rt_dir = get_clang_cl_resource_dir(builder, clang_cl_path);
1060            ldflags.push_all(format!("/libpath:{}", clang_rt_dir.display()));
1061        }
1062
1063        // LLD is built as an LLVM tool, but is distributed outside of the `llvm-tools` component,
1064        // which impacts where it expects to find LLVM's shared library. This causes #80703.
1065        //
1066        // LLD is distributed at "$root/lib/rustlib/$host/bin/rust-lld", but the `libLLVM-*.so` it
1067        // needs is distributed at "$root/lib". The default rpath of "$ORIGIN/../lib" points at the
1068        // lib path for LLVM tools, not the one for rust binaries.
1069        //
1070        // (The `llvm-tools` component copies the .so there for the other tools, and with that
1071        // component installed, one can successfully invoke `rust-lld` directly without rustup's
1072        // `LD_LIBRARY_PATH` overrides)
1073        //
1074        if builder.config.rpath_enabled(target)
1075            && helpers::use_host_linker(target)
1076            && builder.config.llvm_link_shared()
1077            && target.contains("linux")
1078        {
1079            // So we inform LLD where it can find LLVM's libraries by adding an rpath entry to the
1080            // expected parent `lib` directory.
1081            //
1082            // Be careful when changing this path, we need to ensure it's quoted or escaped:
1083            // `$ORIGIN` would otherwise be expanded when the `LdFlags` are passed verbatim to
1084            // cmake.
1085            ldflags.push_all("-Wl,-rpath,'$ORIGIN/../../../'");
1086        }
1087
1088        configure_cmake(builder, target, &mut cfg, true, ldflags, &[]);
1089        configure_llvm(builder, target, &mut cfg);
1090
1091        // Re-use the same flags as llvm to control the level of debug information
1092        // generated for lld.
1093        let profile = match (builder.config.llvm_optimize, builder.config.llvm_release_debuginfo) {
1094            (false, _) => "Debug",
1095            (true, false) => "Release",
1096            (true, true) => "RelWithDebInfo",
1097        };
1098
1099        cfg.out_dir(&out_dir)
1100            .profile(profile)
1101            .define("LLVM_CMAKE_DIR", llvm_cmake_dir)
1102            .define("LLVM_INCLUDE_TESTS", "OFF");
1103
1104        if !builder.config.is_host_target(target) {
1105            // Use the host llvm-tblgen binary.
1106            cfg.define(
1107                "LLVM_TABLEGEN_EXE",
1108                host_llvm_config.with_file_name("llvm-tblgen").with_extension(EXE_EXTENSION),
1109            );
1110        }
1111
1112        cfg.build();
1113
1114        t!(lld_stamp.write());
1115        out_dir
1116    }
1117}
1118
1119#[derive(Debug, Clone, PartialEq, Eq, Hash)]
1120pub struct Sanitizers {
1121    pub target: TargetSelection,
1122}
1123
1124impl Step for Sanitizers {
1125    type Output = Vec<SanitizerRuntime>;
1126
1127    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1128        run.alias("sanitizers")
1129    }
1130
1131    fn make_run(run: RunConfig<'_>) {
1132        run.builder.ensure(Sanitizers { target: run.target });
1133    }
1134
1135    /// Builds sanitizer runtime libraries.
1136    fn run(self, builder: &Builder<'_>) -> Self::Output {
1137        let compiler_rt_dir = builder.src.join("src/llvm-project/compiler-rt");
1138        if !compiler_rt_dir.exists() {
1139            return Vec::new();
1140        }
1141
1142        let out_dir = builder.native_dir(self.target).join("sanitizers");
1143        let runtimes = supported_sanitizers(&out_dir, self.target, &builder.config.channel);
1144
1145        if builder.config.dry_run() || runtimes.is_empty() {
1146            return runtimes;
1147        }
1148
1149        let LlvmResult { host_llvm_config, .. } =
1150            builder.ensure(Llvm { target: builder.config.host_target });
1151
1152        static STAMP_HASH_MEMO: OnceLock<String> = OnceLock::new();
1153        let smart_stamp_hash = STAMP_HASH_MEMO.get_or_init(|| {
1154            generate_smart_stamp_hash(
1155                builder,
1156                &builder.config.src.join("src/llvm-project/compiler-rt"),
1157                builder.in_tree_llvm_info.sha().unwrap_or_default(),
1158            )
1159        });
1160
1161        let stamp = BuildStamp::new(&out_dir).with_prefix("sanitizers").add_stamp(smart_stamp_hash);
1162
1163        if stamp.is_up_to_date() {
1164            if stamp.stamp().is_empty() {
1165                builder.info(&format!(
1166                    "Rebuild sanitizers by removing the file `{}`",
1167                    stamp.path().display()
1168                ));
1169            }
1170
1171            return runtimes;
1172        }
1173
1174        let _guard = builder.msg_unstaged(Kind::Build, "sanitizers", self.target);
1175        t!(stamp.remove());
1176        let _time = helpers::timeit(builder);
1177
1178        let mut cfg = cmake::Config::new(&compiler_rt_dir);
1179        cfg.profile("Release");
1180        cfg.define("CMAKE_C_COMPILER_TARGET", self.target.triple);
1181        cfg.define("COMPILER_RT_BUILD_BUILTINS", "OFF");
1182        cfg.define("COMPILER_RT_BUILD_CRT", "OFF");
1183        cfg.define("COMPILER_RT_BUILD_LIBFUZZER", "OFF");
1184        cfg.define("COMPILER_RT_BUILD_PROFILE", "OFF");
1185        cfg.define("COMPILER_RT_BUILD_SANITIZERS", "ON");
1186        cfg.define("COMPILER_RT_BUILD_XRAY", "OFF");
1187        cfg.define("COMPILER_RT_DEFAULT_TARGET_ONLY", "ON");
1188        cfg.define("COMPILER_RT_USE_LIBCXX", "OFF");
1189        cfg.define("LLVM_CONFIG_PATH", &host_llvm_config);
1190
1191        if self.target.contains("ohos") {
1192            cfg.define("COMPILER_RT_USE_BUILTINS_LIBRARY", "ON");
1193        }
1194
1195        // On Darwin targets the sanitizer runtimes are build as universal binaries.
1196        // Unfortunately sccache currently lacks support to build them successfully.
1197        // Disable compiler launcher on Darwin targets to avoid potential issues.
1198        let use_compiler_launcher = !self.target.contains("apple-darwin");
1199        // Since v1.0.86, the cc crate adds -mmacosx-version-min to the default
1200        // flags on MacOS. A long-standing bug in the CMake rules for compiler-rt
1201        // causes architecture detection to be skipped when this flag is present,
1202        // and compilation fails. https://github.com/llvm/llvm-project/issues/88780
1203        let suppressed_compiler_flag_prefixes: &[&str] =
1204            if self.target.contains("apple-darwin") { &["-mmacosx-version-min="] } else { &[] };
1205        configure_cmake(
1206            builder,
1207            self.target,
1208            &mut cfg,
1209            use_compiler_launcher,
1210            LdFlags::default(),
1211            suppressed_compiler_flag_prefixes,
1212        );
1213
1214        t!(fs::create_dir_all(&out_dir));
1215        cfg.out_dir(out_dir);
1216
1217        for runtime in &runtimes {
1218            cfg.build_target(&runtime.cmake_target);
1219            cfg.build();
1220        }
1221        t!(stamp.write());
1222
1223        runtimes
1224    }
1225}
1226
1227#[derive(Clone, Debug)]
1228pub struct SanitizerRuntime {
1229    /// CMake target used to build the runtime.
1230    pub cmake_target: String,
1231    /// Path to the built runtime library.
1232    pub path: PathBuf,
1233    /// Library filename that will be used rustc.
1234    pub name: String,
1235}
1236
1237/// Returns sanitizers available on a given target.
1238fn supported_sanitizers(
1239    out_dir: &Path,
1240    target: TargetSelection,
1241    channel: &str,
1242) -> Vec<SanitizerRuntime> {
1243    let darwin_libs = |os: &str, components: &[&str]| -> Vec<SanitizerRuntime> {
1244        components
1245            .iter()
1246            .map(move |c| SanitizerRuntime {
1247                cmake_target: format!("clang_rt.{c}_{os}_dynamic"),
1248                path: out_dir.join(format!("build/lib/darwin/libclang_rt.{c}_{os}_dynamic.dylib")),
1249                name: format!("librustc-{channel}_rt.{c}.dylib"),
1250            })
1251            .collect()
1252    };
1253
1254    let common_libs = |os: &str, arch: &str, components: &[&str]| -> Vec<SanitizerRuntime> {
1255        components
1256            .iter()
1257            .map(move |c| SanitizerRuntime {
1258                cmake_target: format!("clang_rt.{c}-{arch}"),
1259                path: out_dir.join(format!("build/lib/{os}/libclang_rt.{c}-{arch}.a")),
1260                name: format!("librustc-{channel}_rt.{c}.a"),
1261            })
1262            .collect()
1263    };
1264
1265    match &*target.triple {
1266        "aarch64-apple-darwin" => darwin_libs("osx", &["asan", "lsan", "tsan", "rtsan"]),
1267        "aarch64-apple-ios" => darwin_libs("ios", &["asan", "tsan", "rtsan"]),
1268        "aarch64-apple-ios-sim" => darwin_libs("iossim", &["asan", "tsan", "rtsan"]),
1269        "aarch64-apple-ios-macabi" => darwin_libs("osx", &["asan", "lsan", "tsan"]),
1270        "aarch64-unknown-fuchsia" => common_libs("fuchsia", "aarch64", &["asan"]),
1271        "aarch64-unknown-linux-gnu" => {
1272            common_libs("linux", "aarch64", &["asan", "lsan", "msan", "tsan", "hwasan", "rtsan"])
1273        }
1274        "aarch64-unknown-linux-ohos" => {
1275            common_libs("linux", "aarch64", &["asan", "lsan", "msan", "tsan", "hwasan"])
1276        }
1277        "loongarch64-unknown-linux-gnu" | "loongarch64-unknown-linux-musl" => {
1278            common_libs("linux", "loongarch64", &["asan", "lsan", "msan", "tsan"])
1279        }
1280        "x86_64-apple-darwin" => darwin_libs("osx", &["asan", "lsan", "tsan", "rtsan"]),
1281        "x86_64-unknown-fuchsia" => common_libs("fuchsia", "x86_64", &["asan"]),
1282        "x86_64-apple-ios" => darwin_libs("iossim", &["asan", "tsan"]),
1283        "x86_64-apple-ios-macabi" => darwin_libs("osx", &["asan", "lsan", "tsan"]),
1284        "x86_64-unknown-freebsd" => common_libs("freebsd", "x86_64", &["asan", "msan", "tsan"]),
1285        "x86_64-unknown-netbsd" => {
1286            common_libs("netbsd", "x86_64", &["asan", "lsan", "msan", "tsan"])
1287        }
1288        "x86_64-unknown-illumos" => common_libs("illumos", "x86_64", &["asan"]),
1289        "x86_64-pc-solaris" => common_libs("solaris", "x86_64", &["asan"]),
1290        "x86_64-unknown-linux-gnu" => common_libs(
1291            "linux",
1292            "x86_64",
1293            &["asan", "dfsan", "lsan", "msan", "safestack", "tsan", "rtsan"],
1294        ),
1295        "x86_64-unknown-linux-musl" => {
1296            common_libs("linux", "x86_64", &["asan", "lsan", "msan", "tsan"])
1297        }
1298        "s390x-unknown-linux-gnu" => {
1299            common_libs("linux", "s390x", &["asan", "lsan", "msan", "tsan"])
1300        }
1301        "s390x-unknown-linux-musl" => {
1302            common_libs("linux", "s390x", &["asan", "lsan", "msan", "tsan"])
1303        }
1304        "x86_64-unknown-linux-ohos" => {
1305            common_libs("linux", "x86_64", &["asan", "lsan", "msan", "tsan"])
1306        }
1307        _ => Vec::new(),
1308    }
1309}
1310
1311#[derive(Debug, Clone, PartialEq, Eq, Hash)]
1312pub struct CrtBeginEnd {
1313    pub target: TargetSelection,
1314}
1315
1316impl Step for CrtBeginEnd {
1317    type Output = PathBuf;
1318
1319    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1320        run.path("src/llvm-project/compiler-rt/lib/crt")
1321    }
1322
1323    fn make_run(run: RunConfig<'_>) {
1324        if run.target.needs_crt_begin_end() {
1325            run.builder.ensure(CrtBeginEnd { target: run.target });
1326        }
1327    }
1328
1329    /// Build crtbegin.o/crtend.o for musl target.
1330    fn run(self, builder: &Builder<'_>) -> Self::Output {
1331        builder.require_submodule(
1332            "src/llvm-project",
1333            Some("The LLVM sources are required for the CRT from `compiler-rt`."),
1334        );
1335
1336        let out_dir = builder.native_dir(self.target).join("crt");
1337
1338        if builder.config.dry_run() {
1339            return out_dir;
1340        }
1341
1342        let crtbegin_src = builder.src.join("src/llvm-project/compiler-rt/lib/builtins/crtbegin.c");
1343        let crtend_src = builder.src.join("src/llvm-project/compiler-rt/lib/builtins/crtend.c");
1344        if up_to_date(&crtbegin_src, &out_dir.join("crtbeginS.o"))
1345            && up_to_date(&crtend_src, &out_dir.join("crtendS.o"))
1346        {
1347            return out_dir;
1348        }
1349
1350        let _guard = builder.msg_unstaged(Kind::Build, "crtbegin.o and crtend.o", self.target);
1351        t!(fs::create_dir_all(&out_dir));
1352
1353        let mut cfg = cc::Build::new();
1354
1355        if let Some(ar) = builder.ar(self.target) {
1356            cfg.archiver(ar);
1357        }
1358        cfg.compiler(builder.cc(self.target));
1359        cfg.cargo_metadata(false)
1360            .out_dir(&out_dir)
1361            .target(&self.target.triple)
1362            .host(&builder.config.host_target.triple)
1363            .warnings(false)
1364            .debug(false)
1365            .opt_level(3)
1366            .file(crtbegin_src)
1367            .file(crtend_src);
1368
1369        // Those flags are defined in src/llvm-project/compiler-rt/lib/builtins/CMakeLists.txt
1370        // Currently only consumer of those objects is musl, which use .init_array/.fini_array
1371        // instead of .ctors/.dtors
1372        cfg.flag("-std=c11")
1373            .define("CRT_HAS_INITFINI_ARRAY", None)
1374            .define("EH_USE_FRAME_REGISTRY", None);
1375
1376        let objs = cfg.compile_intermediates();
1377        assert_eq!(objs.len(), 2);
1378        for obj in objs {
1379            let base_name = unhashed_basename(&obj);
1380            assert!(base_name == "crtbegin" || base_name == "crtend");
1381            t!(fs::copy(&obj, out_dir.join(format!("{base_name}S.o"))));
1382            t!(fs::rename(&obj, out_dir.join(format!("{base_name}.o"))));
1383        }
1384
1385        out_dir
1386    }
1387}
1388
1389#[derive(Debug, Clone, PartialEq, Eq, Hash)]
1390pub struct Libunwind {
1391    pub target: TargetSelection,
1392}
1393
1394impl Step for Libunwind {
1395    type Output = PathBuf;
1396
1397    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
1398        run.path("src/llvm-project/libunwind")
1399    }
1400
1401    fn make_run(run: RunConfig<'_>) {
1402        run.builder.ensure(Libunwind { target: run.target });
1403    }
1404
1405    /// Build libunwind.a
1406    fn run(self, builder: &Builder<'_>) -> Self::Output {
1407        builder.require_submodule(
1408            "src/llvm-project",
1409            Some("The LLVM sources are required for libunwind."),
1410        );
1411
1412        if builder.config.dry_run() {
1413            return PathBuf::new();
1414        }
1415
1416        let out_dir = builder.native_dir(self.target).join("libunwind");
1417        let root = builder.src.join("src/llvm-project/libunwind");
1418
1419        if up_to_date(&root, &out_dir.join("libunwind.a")) {
1420            return out_dir;
1421        }
1422
1423        let _guard = builder.msg_unstaged(Kind::Build, "libunwind.a", self.target);
1424        t!(fs::create_dir_all(&out_dir));
1425
1426        let mut cc_cfg = cc::Build::new();
1427        let mut cpp_cfg = cc::Build::new();
1428
1429        cpp_cfg.cpp(true);
1430        cpp_cfg.cpp_set_stdlib(None);
1431        cpp_cfg.flag("-nostdinc++");
1432        cpp_cfg.flag("-fno-exceptions");
1433        cpp_cfg.flag("-fno-rtti");
1434        cpp_cfg.flag_if_supported("-fvisibility-global-new-delete-hidden");
1435
1436        for cfg in [&mut cc_cfg, &mut cpp_cfg].iter_mut() {
1437            if let Some(ar) = builder.ar(self.target) {
1438                cfg.archiver(ar);
1439            }
1440            cfg.target(&self.target.triple);
1441            cfg.host(&builder.config.host_target.triple);
1442            cfg.warnings(false);
1443            cfg.debug(false);
1444            // get_compiler() need set opt_level first.
1445            cfg.opt_level(3);
1446            cfg.flag("-fstrict-aliasing");
1447            cfg.flag("-funwind-tables");
1448            cfg.flag("-fvisibility=hidden");
1449            cfg.define("_LIBUNWIND_DISABLE_VISIBILITY_ANNOTATIONS", None);
1450            cfg.define("_LIBUNWIND_IS_NATIVE_ONLY", "1");
1451            cfg.include(root.join("include"));
1452            cfg.cargo_metadata(false);
1453            cfg.out_dir(&out_dir);
1454
1455            if self.target.contains("x86_64-fortanix-unknown-sgx") {
1456                cfg.static_flag(true);
1457                cfg.flag("-fno-stack-protector");
1458                cfg.flag("-ffreestanding");
1459                cfg.flag("-fexceptions");
1460
1461                // easiest way to undefine since no API available in cc::Build to undefine
1462                cfg.flag("-U_FORTIFY_SOURCE");
1463                cfg.define("_FORTIFY_SOURCE", "0");
1464                cfg.define("RUST_SGX", "1");
1465                cfg.define("__NO_STRING_INLINES", None);
1466                cfg.define("__NO_MATH_INLINES", None);
1467                cfg.define("_LIBUNWIND_IS_BAREMETAL", None);
1468                cfg.define("NDEBUG", None);
1469            }
1470            if self.target.is_windows() {
1471                cfg.define("_LIBUNWIND_HIDE_SYMBOLS", "1");
1472            }
1473        }
1474
1475        cc_cfg.compiler(builder.cc(self.target));
1476        if let Ok(cxx) = builder.cxx(self.target) {
1477            cpp_cfg.compiler(cxx);
1478        } else {
1479            cc_cfg.compiler(builder.cc(self.target));
1480        }
1481
1482        // Don't set this for clang
1483        // By default, Clang builds C code in GNU C17 mode.
1484        // By default, Clang builds C++ code according to the C++98 standard,
1485        // with many C++11 features accepted as extensions.
1486        if cc_cfg.get_compiler().is_like_gnu() {
1487            cc_cfg.flag("-std=c99");
1488        }
1489        if cpp_cfg.get_compiler().is_like_gnu() {
1490            cpp_cfg.flag("-std=c++11");
1491        }
1492
1493        if self.target.contains("x86_64-fortanix-unknown-sgx") || self.target.contains("musl") {
1494            // use the same GCC C compiler command to compile C++ code so we do not need to setup the
1495            // C++ compiler env variables on the builders.
1496            // Don't set this for clang++, as clang++ is able to compile this without libc++.
1497            if cpp_cfg.get_compiler().is_like_gnu() {
1498                cpp_cfg.cpp(false);
1499                cpp_cfg.compiler(builder.cc(self.target));
1500            }
1501        }
1502
1503        let mut c_sources = vec![
1504            "Unwind-sjlj.c",
1505            "UnwindLevel1-gcc-ext.c",
1506            "UnwindLevel1.c",
1507            "UnwindRegistersRestore.S",
1508            "UnwindRegistersSave.S",
1509        ];
1510
1511        let cpp_sources = vec!["Unwind-EHABI.cpp", "Unwind-seh.cpp", "libunwind.cpp"];
1512        let cpp_len = cpp_sources.len();
1513
1514        if self.target.contains("x86_64-fortanix-unknown-sgx") {
1515            c_sources.push("UnwindRustSgx.c");
1516        }
1517
1518        for src in c_sources {
1519            cc_cfg.file(root.join("src").join(src).canonicalize().unwrap());
1520        }
1521
1522        for src in &cpp_sources {
1523            cpp_cfg.file(root.join("src").join(src).canonicalize().unwrap());
1524        }
1525
1526        cpp_cfg.compile("unwind-cpp");
1527
1528        // FIXME: https://github.com/alexcrichton/cc-rs/issues/545#issuecomment-679242845
1529        let mut count = 0;
1530        let mut files = fs::read_dir(&out_dir)
1531            .unwrap()
1532            .map(|entry| entry.unwrap().path().canonicalize().unwrap())
1533            .collect::<Vec<_>>();
1534        files.sort();
1535        for file in files {
1536            if file.is_file() && file.extension() == Some(OsStr::new("o")) {
1537                // Object file name without the hash prefix is "Unwind-EHABI", "Unwind-seh" or "libunwind".
1538                let base_name = unhashed_basename(&file);
1539                if cpp_sources.iter().any(|f| *base_name == f[..f.len() - 4]) {
1540                    cc_cfg.object(&file);
1541                    count += 1;
1542                }
1543            }
1544        }
1545        assert_eq!(cpp_len, count, "Can't get object files from {out_dir:?}");
1546
1547        cc_cfg.compile("unwind");
1548        out_dir
1549    }
1550}