builder.rs 60.8 KB
Newer Older
1
use std::any::Any;
2
use std::cell::{Cell, RefCell};
3 4
use std::collections::BTreeSet;
use std::env;
5
use std::ffi::OsStr;
6
use std::fmt::Debug;
7
use std::fs;
8
use std::hash::Hash;
9
use std::ops::Deref;
10 11
use std::path::{Path, PathBuf};
use std::process::Command;
S
Santiago Pastorino 已提交
12
use std::time::{Duration, Instant};
13

14
use build_helper::{output, t};
15

L
ljedrz 已提交
16 17 18
use crate::cache::{Cache, Interned, INTERNER};
use crate::check;
use crate::compile;
19
use crate::config::TargetSelection;
L
ljedrz 已提交
20 21
use crate::dist;
use crate::doc;
J
Joshua Nelson 已提交
22
use crate::flags::{Color, Subcommand};
L
ljedrz 已提交
23 24
use crate::install;
use crate::native;
25
use crate::run;
L
ljedrz 已提交
26
use crate::test;
27
use crate::tool::{self, SourceType};
28
use crate::util::{self, add_dylib_path, add_link_lib_path, exe, libdir};
M
Mark Rousskov 已提交
29
use crate::{Build, DocTests, GitRepo, Mode};
L
ljedrz 已提交
30 31

pub use crate::Compiler;
32

33 34 35 36 37
pub struct Builder<'a> {
    pub build: &'a Build,
    pub top_stage: u32,
    pub kind: Kind,
    cache: Cache,
38
    stack: RefCell<Vec<Box<dyn Any>>>,
39
    time_spent_on_dependencies: Cell<Duration>,
40
    pub paths: Vec<PathBuf>,
41 42 43 44 45 46 47 48 49 50
}

impl<'a> Deref for Builder<'a> {
    type Target = Build;

    fn deref(&self) -> &Self::Target {
        self.build
    }
}

51
pub trait Step: 'static + Clone + Debug + PartialEq + Eq + Hash {
52 53
    /// `PathBuf` when directories are created or to return a `Compiler` once
    /// it's been assembled.
54
    type Output: Clone;
55

R
Ralf Jung 已提交
56 57
    /// Whether this step is run by default as part of its respective phase.
    /// `true` here can still be overwritten by `should_run` calling `default_condition`.
58 59
    const DEFAULT: bool = false;

60
    /// If true, then this rule should be skipped if --target was specified, but --host was not
61 62
    const ONLY_HOSTS: bool = false;

A
Alexander Regueiro 已提交
63
    /// Primary function to execute this rule. Can call `builder.ensure()`
64
    /// with other steps to run those.
T
Taiki Endo 已提交
65
    fn run(self, builder: &Builder<'_>) -> Self::Output;
66

67 68
    /// When bootstrap is passed a set of paths, this controls whether this rule
    /// will execute. However, it does not get called in a "default" context
A
Alexander Regueiro 已提交
69
    /// when we are not passed any paths; in that case, `make_run` is called
70
    /// directly.
T
Taiki Endo 已提交
71
    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_>;
72

A
Alexander Regueiro 已提交
73
    /// Builds up a "root" rule, either as a default rule or from a path passed
74 75 76 77 78
    /// to us.
    ///
    /// When path is `None`, we are executing in a context where no paths were
    /// passed. When `./x.py build` is run, for example, this rule could get
    /// called if it is in the correct list below with a path of `None`.
T
Taiki Endo 已提交
79
    fn make_run(_run: RunConfig<'_>) {
80 81 82 83 84 85
        // It is reasonable to not have an implementation of make_run for rules
        // who do not want to get called from the root context. This means that
        // they are likely dependencies (e.g., sysroot creation) or similar, and
        // as such calling them from ./x.py isn't logical.
        unimplemented!()
    }
86 87
}

88 89
pub struct RunConfig<'a> {
    pub builder: &'a Builder<'a>,
90
    pub target: TargetSelection,
91
    pub path: PathBuf,
92 93
}

94 95 96 97 98 99
impl RunConfig<'_> {
    pub fn build_triple(&self) -> TargetSelection {
        self.builder.build.build
    }
}

100 101 102
struct StepDescription {
    default: bool,
    only_hosts: bool,
T
Taiki Endo 已提交
103 104
    should_run: fn(ShouldRun<'_>) -> ShouldRun<'_>,
    make_run: fn(RunConfig<'_>),
105 106 107
    name: &'static str,
}

108
/// Collection of paths used to match a task rule.
109
#[derive(Debug, Clone, PartialOrd, Ord, PartialEq, Eq)]
110
pub enum PathSet {
111 112 113 114 115
    /// A collection of individual paths.
    ///
    /// These are generally matched as a path suffix. For example, a
    /// command-line value of `libstd` will match if `src/libstd` is in the
    /// set.
S
Santiago Pastorino 已提交
116
    Set(BTreeSet<PathBuf>),
117 118 119 120 121 122
    /// A "suite" of paths.
    ///
    /// These can match as a path suffix (like `Set`), or as a prefix. For
    /// example, a command-line value of `src/test/ui/abi/variadic-ffi.rs`
    /// will match `src/test/ui`. A command-line value of `ui` would also
    /// match `src/test/ui`.
S
Santiago Pastorino 已提交
123
    Suite(PathBuf),
124 125 126
}

impl PathSet {
127
    fn empty() -> PathSet {
128
        PathSet::Set(BTreeSet::new())
129 130
    }

131 132 133
    fn one<P: Into<PathBuf>>(path: P) -> PathSet {
        let mut set = BTreeSet::new();
        set.insert(path.into());
134
        PathSet::Set(set)
135 136 137
    }

    fn has(&self, needle: &Path) -> bool {
138 139
        match self {
            PathSet::Set(set) => set.iter().any(|p| p.ends_with(needle)),
140
            PathSet::Suite(suite) => suite.ends_with(needle),
141
        }
142 143
    }

T
Taiki Endo 已提交
144
    fn path(&self, builder: &Builder<'_>) -> PathBuf {
145
        match self {
M
Mark Rousskov 已提交
146
            PathSet::Set(set) => set.iter().next().unwrap_or(&builder.build.src).to_path_buf(),
S
Santiago Pastorino 已提交
147
            PathSet::Suite(path) => PathBuf::from(path),
148
        }
149
    }
150 151 152 153 154 155 156 157 158
}

impl StepDescription {
    fn from<S: Step>() -> StepDescription {
        StepDescription {
            default: S::DEFAULT,
            only_hosts: S::ONLY_HOSTS,
            should_run: S::should_run,
            make_run: S::make_run,
M
Mark Rousskov 已提交
159
            name: std::any::type_name::<S>(),
160 161 162
        }
    }

T
Taiki Endo 已提交
163
    fn maybe_run(&self, builder: &Builder<'_>, pathset: &PathSet) {
164 165 166 167
        if builder.config.exclude.iter().any(|e| pathset.has(e)) {
            eprintln!("Skipping {:?} because it is excluded", pathset);
            return;
        } else if !builder.config.exclude.is_empty() {
S
Santiago Pastorino 已提交
168 169 170 171
            eprintln!(
                "{:?} not skipped for {:?} -- not in {:?}",
                pathset, self.name, builder.config.exclude
            );
172
        }
173

M
Mark Simulacrum 已提交
174
        // Determine the targets participating in this rule.
T
Tyler Mandry 已提交
175
        let targets = if self.only_hosts { &builder.hosts } else { &builder.targets };
176

177 178 179
        for target in targets {
            let run = RunConfig { builder, path: pathset.path(builder), target: *target };
            (self.make_run)(run);
180 181 182
        }
    }

T
Taiki Endo 已提交
183
    fn run(v: &[StepDescription], builder: &Builder<'_>, paths: &[PathBuf]) {
M
Mark Rousskov 已提交
184 185
        let should_runs =
            v.iter().map(|desc| (desc.should_run)(ShouldRun::new(builder))).collect::<Vec<_>>();
186 187 188

        // sanity checks on rules
        for (desc, should_run) in v.iter().zip(&should_runs) {
S
Santiago Pastorino 已提交
189 190 191 192 193
            assert!(
                !should_run.paths.is_empty(),
                "{:?} should have at least one pathset",
                desc.name
            );
194 195
        }

196 197
        if paths.is_empty() || builder.config.include_default_paths {
            for (desc, should_run) in v.iter().zip(&should_runs) {
198
                if desc.default && should_run.is_really_default {
199 200 201
                    for pathset in &should_run.paths {
                        desc.maybe_run(builder, pathset);
                    }
202 203
                }
            }
204
        }
205

206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
        for path in paths {
            // strip CurDir prefix if present
            let path = match path.strip_prefix(".") {
                Ok(p) => p,
                Err(_) => path,
            };

            let mut attempted_run = false;
            for (desc, should_run) in v.iter().zip(&should_runs) {
                if let Some(suite) = should_run.is_suite_path(path) {
                    attempted_run = true;
                    desc.maybe_run(builder, suite);
                } else if let Some(pathset) = should_run.pathset_for_path(path) {
                    attempted_run = true;
                    desc.maybe_run(builder, pathset);
221 222
                }
            }
223 224 225 226

            if !attempted_run {
                panic!("error: no rules matched {}", path.display());
            }
227 228 229 230
        }
    }
}

231 232
#[derive(Clone)]
pub struct ShouldRun<'a> {
233
    pub builder: &'a Builder<'a>,
234
    // use a BTreeSet to maintain sort order
235
    paths: BTreeSet<PathSet>,
236 237

    // If this is a default rule, this is an additional constraint placed on
238
    // its run. Generally something like compiler docs being enabled.
239
    is_really_default: bool,
240 241 242
}

impl<'a> ShouldRun<'a> {
T
Taiki Endo 已提交
243
    fn new(builder: &'a Builder<'_>) -> ShouldRun<'a> {
244
        ShouldRun {
245
            builder,
246
            paths: BTreeSet::new(),
247
            is_really_default: true, // by default no additional conditions
248 249 250
        }
    }

251 252 253 254 255
    pub fn default_condition(mut self, cond: bool) -> Self {
        self.is_really_default = cond;
        self
    }

256 257 258 259 260 261 262 263 264
    /// Indicates it should run if the command-line selects the given crate or
    /// any of its (local) dependencies.
    ///
    /// Compared to `krate`, this treats the dependencies as aliases for the
    /// same job. Generally it is preferred to use `krate`, and treat each
    /// individual path separately. For example `./x.py test src/liballoc`
    /// (which uses `krate`) will test just `liballoc`. However, `./x.py check
    /// src/liballoc` (which uses `all_krates`) will check all of `libtest`.
    /// `all_krates` should probably be removed at some point.
265 266
    pub fn all_krates(mut self, name: &str) -> Self {
        let mut set = BTreeSet::new();
267
        for krate in self.builder.in_tree_crates(name, None) {
268 269
            let path = krate.local_path(self.builder);
            set.insert(path);
270
        }
271
        self.paths.insert(PathSet::Set(set));
272 273 274
        self
    }

275 276 277 278
    /// Indicates it should run if the command-line selects the given crate or
    /// any of its (local) dependencies.
    ///
    /// `make_run` will be called separately for each matching command-line path.
279
    pub fn krate(mut self, name: &str) -> Self {
280
        for krate in self.builder.in_tree_crates(name, None) {
281 282
            let path = krate.local_path(self.builder);
            self.paths.insert(PathSet::one(path));
283 284 285 286
        }
        self
    }

287 288 289 290 291 292 293
    // single, non-aliased path
    pub fn path(self, path: &str) -> Self {
        self.paths(&[path])
    }

    // multiple aliases for the same job
    pub fn paths(mut self, paths: &[&str]) -> Self {
M
Mark Rousskov 已提交
294
        self.paths.insert(PathSet::Set(paths.iter().map(PathBuf::from).collect()));
295 296 297 298
        self
    }

    pub fn is_suite_path(&self, path: &Path) -> Option<&PathSet> {
S
Santiago Pastorino 已提交
299 300 301
        self.paths.iter().find(|pathset| match pathset {
            PathSet::Suite(p) => path.starts_with(p),
            PathSet::Set(_) => false,
302 303 304 305 306
        })
    }

    pub fn suite_path(mut self, suite: &str) -> Self {
        self.paths.insert(PathSet::Suite(PathBuf::from(suite)));
307 308 309 310
        self
    }

    // allows being more explicit about why should_run in Step returns the value passed to it
311 312
    pub fn never(mut self) -> ShouldRun<'a> {
        self.paths.insert(PathSet::empty());
313 314 315
        self
    }

316 317
    fn pathset_for_path(&self, path: &Path) -> Option<&PathSet> {
        self.paths.iter().find(|pathset| pathset.has(path))
318 319 320
    }
}

321 322 323
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Kind {
    Build,
324
    Check,
L
ljedrz 已提交
325 326
    Clippy,
    Fix,
A
Adam Perry 已提交
327
    Format,
328 329 330 331 332
    Test,
    Bench,
    Dist,
    Doc,
    Install,
333
    Run,
334 335
}

336 337 338
impl<'a> Builder<'a> {
    fn get_step_descriptions(kind: Kind) -> Vec<StepDescription> {
        macro_rules! describe {
T
Taiki Endo 已提交
339
            ($($rule:ty),+ $(,)?) => {{
340 341
                vec![$(StepDescription::from::<$rule>()),+]
            }};
342
        }
343
        match kind {
S
Santiago Pastorino 已提交
344 345 346
            Kind::Build => describe!(
                compile::Std,
                compile::Rustc,
347
                compile::CodegenBackend,
S
Santiago Pastorino 已提交
348 349 350 351 352 353 354 355 356 357 358 359 360 361
                compile::StartupObjects,
                tool::BuildManifest,
                tool::Rustbook,
                tool::ErrorIndex,
                tool::UnstableBookGen,
                tool::Tidy,
                tool::Linkchecker,
                tool::CargoTest,
                tool::Compiletest,
                tool::RemoteTestServer,
                tool::RemoteTestClient,
                tool::RustInstaller,
                tool::Cargo,
                tool::Rls,
A
Aleksey Kladov 已提交
362
                tool::RustAnalyzer,
R
Rich Kadel 已提交
363
                tool::RustDemangler,
S
Santiago Pastorino 已提交
364 365
                tool::Rustdoc,
                tool::Clippy,
366
                tool::CargoClippy,
S
Santiago Pastorino 已提交
367
                native::Llvm,
368
                native::Sanitizers,
S
Santiago Pastorino 已提交
369 370
                tool::Rustfmt,
                tool::Miri,
371
                tool::CargoMiri,
S
Santiago Pastorino 已提交
372 373
                native::Lld
            ),
J
Joshua Nelson 已提交
374
            Kind::Check | Kind::Clippy { .. } | Kind::Fix | Kind::Format => describe!(
375 376 377 378 379 380 381
                check::Std,
                check::Rustc,
                check::Rustdoc,
                check::CodegenBackend,
                check::Clippy,
                check::Bootstrap
            ),
S
Santiago Pastorino 已提交
382
            Kind::Test => describe!(
383
                crate::toolstate::ToolStateCheck,
384
                test::ExpandYamlAnchors,
S
Santiago Pastorino 已提交
385 386 387 388 389 390 391
                test::Tidy,
                test::Ui,
                test::CompileFail,
                test::RunPassValgrind,
                test::MirOpt,
                test::Codegen,
                test::CodegenUnits,
392
                test::Assembly,
S
Santiago Pastorino 已提交
393 394 395 396 397 398 399 400 401
                test::Incremental,
                test::Debuginfo,
                test::UiFullDeps,
                test::Rustdoc,
                test::Pretty,
                test::Crate,
                test::CrateLibrustc,
                test::CrateRustdoc,
                test::Linkcheck,
402
                test::TierCheck,
S
Santiago Pastorino 已提交
403 404 405 406 407
                test::Cargotest,
                test::Cargo,
                test::Rls,
                test::ErrorIndex,
                test::Distcheck,
408
                test::RunMakeFullDeps,
S
Santiago Pastorino 已提交
409 410 411 412 413 414 415
                test::Nomicon,
                test::Reference,
                test::RustdocBook,
                test::RustByExample,
                test::TheBook,
                test::UnstableBook,
                test::RustcBook,
416
                test::RustcGuide,
417
                test::EmbeddedBook,
E
Eric Huss 已提交
418
                test::EditionGuide,
S
Santiago Pastorino 已提交
419 420 421
                test::Rustfmt,
                test::Miri,
                test::Clippy,
422
                test::CompiletestTest,
G
Guillaume Gomez 已提交
423
                test::RustdocJSStd,
G
Guillaume Gomez 已提交
424
                test::RustdocJSNotStd,
S
Santiago Pastorino 已提交
425
                test::RustdocTheme,
J
John Kåre Alsaker 已提交
426
                test::RustdocUi,
427 428
                // Run bootstrap close to the end as it's unlikely to fail
                test::Bootstrap,
429
                // Run run-make last, since these won't pass without make on Windows
S
Santiago Pastorino 已提交
430 431
                test::RunMake,
            ),
432
            Kind::Bench => describe!(test::Crate, test::CrateLibrustc),
S
Santiago Pastorino 已提交
433 434 435 436 437 438 439 440 441 442 443 444 445 446
            Kind::Doc => describe!(
                doc::UnstableBook,
                doc::UnstableBookGen,
                doc::TheBook,
                doc::Standalone,
                doc::Std,
                doc::Rustc,
                doc::Rustdoc,
                doc::ErrorIndex,
                doc::Nomicon,
                doc::Reference,
                doc::RustdocBook,
                doc::RustByExample,
                doc::RustcBook,
S
Steve Klabnik 已提交
447
                doc::CargoBook,
J
James Munns 已提交
448
                doc::EmbeddedBook,
S
Steve Klabnik 已提交
449
                doc::EditionGuide,
S
Santiago Pastorino 已提交
450 451 452 453 454 455 456 457
            ),
            Kind::Dist => describe!(
                dist::Docs,
                dist::RustcDocs,
                dist::Mingw,
                dist::Rustc,
                dist::DebuggerScripts,
                dist::Std,
J
Josh Stone 已提交
458
                dist::RustcDev,
S
Santiago Pastorino 已提交
459 460 461 462 463
                dist::Analysis,
                dist::Src,
                dist::PlainSourceTarball,
                dist::Cargo,
                dist::Rls,
A
Aleksey Kladov 已提交
464
                dist::RustAnalyzer,
S
Santiago Pastorino 已提交
465
                dist::Rustfmt,
466
                dist::Clippy,
467
                dist::Miri,
468
                dist::LlvmTools,
469
                dist::RustDev,
S
Santiago Pastorino 已提交
470
                dist::Extended,
471
                dist::BuildManifest,
S
Santiago Pastorino 已提交
472 473 474 475 476 477
            ),
            Kind::Install => describe!(
                install::Docs,
                install::Std,
                install::Cargo,
                install::Rls,
A
Aleksey Kladov 已提交
478
                install::RustAnalyzer,
S
Santiago Pastorino 已提交
479
                install::Rustfmt,
480
                install::Clippy,
481
                install::Miri,
S
Santiago Pastorino 已提交
482 483 484 485
                install::Analysis,
                install::Src,
                install::Rustc
            ),
486
            Kind::Run => describe!(run::ExpandYamlAnchors, run::BuildManifest),
487 488
        }
    }
489

490 491 492 493 494 495 496 497 498 499 500
    pub fn get_help(build: &Build, subcommand: &str) -> Option<String> {
        let kind = match subcommand {
            "build" => Kind::Build,
            "doc" => Kind::Doc,
            "test" => Kind::Test,
            "bench" => Kind::Bench,
            "dist" => Kind::Dist,
            "install" => Kind::Install,
            _ => return None,
        };

J
Joshua Nelson 已提交
501
        let builder = Self::new_internal(build, kind, vec![]);
502 503
        let builder = &builder;
        let mut should_run = ShouldRun::new(builder);
504 505
        for desc in Builder::get_step_descriptions(builder.kind) {
            should_run = (desc.should_run)(should_run);
506 507
        }
        let mut help = String::from("Available paths:\n");
508 509 510
        let mut add_path = |path: &Path| {
            help.push_str(&format!("    ./x.py {} {}\n", subcommand, path.display()));
        };
511
        for pathset in should_run.paths {
512 513 514 515 516 517 518 519 520
            match pathset {
                PathSet::Set(set) => {
                    for path in set {
                        add_path(&path);
                    }
                }
                PathSet::Suite(path) => {
                    add_path(&path.join("..."));
                }
521
            }
522 523 524 525
        }
        Some(help)
    }

J
Joshua Nelson 已提交
526 527 528
    fn new_internal(build: &Build, kind: Kind, paths: Vec<PathBuf>) -> Builder<'_> {
        Builder {
            build,
529
            top_stage: build.config.stage,
J
Joshua Nelson 已提交
530 531 532 533 534 535 536 537
            kind,
            cache: Cache::new(),
            stack: RefCell::new(Vec::new()),
            time_spent_on_dependencies: Cell::new(Duration::new(0, 0)),
            paths,
        }
    }

T
Taiki Endo 已提交
538
    pub fn new(build: &Build) -> Builder<'_> {
M
Mark Simulacrum 已提交
539
        let (kind, paths) = match build.config.cmd {
540
            Subcommand::Build { ref paths } => (Kind::Build, &paths[..]),
541
            Subcommand::Check { ref paths, all_targets: _ } => (Kind::Check, &paths[..]),
J
Joshua Nelson 已提交
542
            Subcommand::Clippy { ref paths, .. } => (Kind::Clippy, &paths[..]),
L
ljedrz 已提交
543
            Subcommand::Fix { ref paths } => (Kind::Fix, &paths[..]),
544
            Subcommand::Doc { ref paths, .. } => (Kind::Doc, &paths[..]),
545 546 547 548
            Subcommand::Test { ref paths, .. } => (Kind::Test, &paths[..]),
            Subcommand::Bench { ref paths, .. } => (Kind::Bench, &paths[..]),
            Subcommand::Dist { ref paths } => (Kind::Dist, &paths[..]),
            Subcommand::Install { ref paths } => (Kind::Install, &paths[..]),
549
            Subcommand::Run { ref paths } => (Kind::Run, &paths[..]),
J
Joshua Nelson 已提交
550 551
            Subcommand::Format { .. } | Subcommand::Clean { .. } | Subcommand::Setup { .. } => {
                panic!()
552
            }
553
        };
554

555
        Self::new_internal(build, kind, paths.to_owned())
556 557
    }

558
    pub fn execute_cli(&self) {
M
Mark Simulacrum 已提交
559
        self.run_step_descriptions(&Builder::get_step_descriptions(self.kind), &self.paths);
560 561 562 563
    }

    pub fn default_doc(&self, paths: Option<&[PathBuf]>) {
        let paths = paths.unwrap_or(&[]);
M
Mark Simulacrum 已提交
564 565 566 567 568
        self.run_step_descriptions(&Builder::get_step_descriptions(Kind::Doc), paths);
    }

    fn run_step_descriptions(&self, v: &[StepDescription], paths: &[PathBuf]) {
        StepDescription::run(v, self, paths);
569 570
    }

B
Bastien Orivel 已提交
571
    /// Obtain a compiler at a given stage and for a given host. Explicitly does
572 573 574
    /// not take `Compiler` since all `Compiler` instances are meant to be
    /// obtained through this function, since it ensures that they are valid
    /// (i.e., built and assembled).
575
    pub fn compiler(&self, stage: u32, host: TargetSelection) -> Compiler {
M
Mark Rousskov 已提交
576
        self.ensure(compile::Assemble { target_compiler: Compiler { stage, host } })
577 578
    }

579 580 581 582 583 584 585 586 587 588 589
    /// Similar to `compiler`, except handles the full-bootstrap option to
    /// silently use the stage1 compiler instead of a stage2 compiler if one is
    /// requested.
    ///
    /// Note that this does *not* have the side effect of creating
    /// `compiler(stage, host)`, unlike `compiler` above which does have such
    /// a side effect. The returned compiler here can only be used to compile
    /// new artifacts, it can't be used to rely on the presence of a particular
    /// sysroot.
    ///
    /// See `force_use_stage1` for documentation on what each argument is.
A
Alex Crichton 已提交
590 591 592
    pub fn compiler_for(
        &self,
        stage: u32,
593 594
        host: TargetSelection,
        target: TargetSelection,
A
Alex Crichton 已提交
595
    ) -> Compiler {
596 597 598 599 600 601 602
        if self.build.force_use_stage1(Compiler { stage, host }, target) {
            self.compiler(1, self.config.build)
        } else {
            self.compiler(stage, host)
        }
    }

603
    pub fn sysroot(&self, compiler: Compiler) -> Interned<PathBuf> {
604 605 606 607 608
        self.ensure(compile::Sysroot { compiler })
    }

    /// Returns the libdir where the standard library and other artifacts are
    /// found for a compiler's sysroot.
609
    pub fn sysroot_libdir(&self, compiler: Compiler, target: TargetSelection) -> Interned<PathBuf> {
610 611 612
        #[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
        struct Libdir {
            compiler: Compiler,
613
            target: TargetSelection,
614
        }
615 616
        impl Step for Libdir {
            type Output = Interned<PathBuf>;
617

T
Taiki Endo 已提交
618
            fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
619
                run.never()
620 621
            }

T
Taiki Endo 已提交
622
            fn run(self, builder: &Builder<'_>) -> Interned<PathBuf> {
O
O01eg 已提交
623
                let lib = builder.sysroot_libdir_relative(self.compiler);
S
Santiago Pastorino 已提交
624 625 626 627
                let sysroot = builder
                    .sysroot(self.compiler)
                    .join(lib)
                    .join("rustlib")
628
                    .join(self.target.triple)
S
Santiago Pastorino 已提交
629
                    .join("lib");
630 631
                let _ = fs::remove_dir_all(&sysroot);
                t!(fs::create_dir_all(&sysroot));
632
                INTERNER.intern_path(sysroot)
633 634 635 636 637
            }
        }
        self.ensure(Libdir { compiler, target })
    }

638 639 640 641
    pub fn sysroot_codegen_backends(&self, compiler: Compiler) -> PathBuf {
        self.sysroot_libdir(compiler, compiler.host).with_file_name("codegen-backends")
    }

642 643 644 645 646 647 648
    /// Returns the compiler's libdir where it stores the dynamic libraries that
    /// it itself links against.
    ///
    /// For example this returns `<sysroot>/lib` on Unix and `<sysroot>/bin` on
    /// Windows.
    pub fn rustc_libdir(&self, compiler: Compiler) -> PathBuf {
        if compiler.is_snapshot(self) {
649
            self.rustc_snapshot_libdir()
650
        } else {
O
O01eg 已提交
651
            match self.config.libdir_relative() {
M
Mark Rousskov 已提交
652 653 654
                Some(relative_libdir) if compiler.stage >= 1 => {
                    self.sysroot(compiler).join(relative_libdir)
                }
655
                _ => self.sysroot(compiler).join(libdir(compiler.host)),
O
O01eg 已提交
656 657 658 659 660 661 662 663 664 665 666
            }
        }
    }

    /// Returns the compiler's relative libdir where it stores the dynamic libraries that
    /// it itself links against.
    ///
    /// For example this returns `lib` on Unix and `bin` on
    /// Windows.
    pub fn libdir_relative(&self, compiler: Compiler) -> &Path {
        if compiler.is_snapshot(self) {
667
            libdir(self.config.build).as_ref()
O
O01eg 已提交
668 669
        } else {
            match self.config.libdir_relative() {
M
Mark Rousskov 已提交
670
                Some(relative_libdir) if compiler.stage >= 1 => relative_libdir,
671
                _ => libdir(compiler.host).as_ref(),
O
O01eg 已提交
672
            }
673 674 675
        }
    }

O
O01eg 已提交
676 677 678 679 680 681
    /// Returns the compiler's relative libdir where the standard library and other artifacts are
    /// found for a compiler's sysroot.
    ///
    /// For example this returns `lib` on Unix and Windows.
    pub fn sysroot_libdir_relative(&self, compiler: Compiler) -> &Path {
        match self.config.libdir_relative() {
M
Mark Rousskov 已提交
682
            Some(relative_libdir) if compiler.stage >= 1 => relative_libdir,
O
O01eg 已提交
683
            _ if compiler.stage == 0 => &self.build.initial_libdir,
M
Mark Rousskov 已提交
684
            _ => Path::new("lib"),
O
O01eg 已提交
685 686 687
        }
    }

688 689
    /// Adds the compiler's directory of dynamic libraries to `cmd`'s dynamic
    /// library lookup path.
690
    pub fn add_rustc_lib_path(&self, compiler: Compiler, cmd: &mut Command) {
691 692 693 694
        // Windows doesn't need dylib path munging because the dlls for the
        // compiler live next to the compiler and the system will find them
        // automatically.
        if cfg!(windows) {
S
Santiago Pastorino 已提交
695
            return;
696 697
        }

698
        add_dylib_path(vec![self.rustc_libdir(compiler)], cmd);
699 700
    }

A
Alexander Regueiro 已提交
701
    /// Gets a path to the compiler specified.
702 703 704 705
    pub fn rustc(&self, compiler: Compiler) -> PathBuf {
        if compiler.is_snapshot(self) {
            self.initial_rustc.clone()
        } else {
706
            self.sysroot(compiler).join("bin").join(exe("rustc", compiler.host))
707 708 709
        }
    }

710 711 712 713 714 715 716 717 718
    /// Gets the paths to all of the compiler's codegen backends.
    fn codegen_backends(&self, compiler: Compiler) -> impl Iterator<Item = PathBuf> {
        fs::read_dir(self.sysroot_codegen_backends(compiler))
            .into_iter()
            .flatten()
            .filter_map(Result::ok)
            .map(|entry| entry.path())
    }

M
Mark Rousskov 已提交
719 720
    pub fn rustdoc(&self, compiler: Compiler) -> PathBuf {
        self.ensure(tool::Rustdoc { compiler })
M
Mark Simulacrum 已提交
721 722
    }

M
Mark Rousskov 已提交
723
    pub fn rustdoc_cmd(&self, compiler: Compiler) -> Command {
M
Mark Simulacrum 已提交
724
        let mut cmd = Command::new(&self.out.join("bootstrap/debug/rustdoc"));
O
Oliver Schneider 已提交
725
        cmd.env("RUSTC_STAGE", compiler.stage.to_string())
S
Santiago Pastorino 已提交
726
            .env("RUSTC_SYSROOT", self.sysroot(compiler))
M
Mark Rousskov 已提交
727 728 729
            // Note that this is *not* the sysroot_libdir because rustdoc must be linked
            // equivalently to rustc.
            .env("RUSTDOC_LIBDIR", self.rustc_libdir(compiler))
S
Santiago Pastorino 已提交
730
            .env("CFG_RELEASE_CHANNEL", &self.config.channel)
M
Mark Rousskov 已提交
731
            .env("RUSTDOC_REAL", self.rustdoc(compiler))
732
            .env("RUSTC_BOOTSTRAP", "1")
733
            .arg("-Winvalid_codeblock_attributes");
G
Guillaume Gomez 已提交
734 735 736
        if self.config.deny_warnings {
            cmd.arg("-Dwarnings");
        }
737 738 739 740 741

        // Remove make-related flags that can cause jobserver problems.
        cmd.env_remove("MAKEFLAGS");
        cmd.env_remove("MFLAGS");

742
        if let Some(linker) = self.linker(compiler.host) {
743 744
            cmd.env("RUSTDOC_LINKER", linker);
        }
745
        if self.is_fuse_ld_lld(compiler.host) {
746
            cmd.env("RUSTDOC_FUSE_LD_LLD", "1");
O
Oliver Schneider 已提交
747
        }
M
Mark Simulacrum 已提交
748
        cmd
749 750
    }

751 752 753 754
    /// Return the path to `llvm-config` for the target, if it exists.
    ///
    /// Note that this returns `None` if LLVM is disabled, or if we're in a
    /// check build or dry-run, where there's no need to build all of LLVM.
755
    fn llvm_config(&self, target: TargetSelection) -> Option<PathBuf> {
756 757 758 759 760 761 762 763 764
        if self.config.llvm_enabled() && self.kind != Kind::Check && !self.config.dry_run {
            let llvm_config = self.ensure(native::Llvm { target });
            if llvm_config.is_file() {
                return Some(llvm_config);
            }
        }
        None
    }

765 766 767 768 769 770 771
    /// Prepares an invocation of `cargo` to be run.
    ///
    /// This will create a `Command` that represents a pending execution of
    /// Cargo. This cargo will be configured to use `compiler` as the actual
    /// rustc compiler, its output will be scoped by `mode`'s output directory,
    /// it will pass the `--target` flag for the specified `target`, and will be
    /// executing the Cargo command `cmd`.
S
Santiago Pastorino 已提交
772 773 774 775
    pub fn cargo(
        &self,
        compiler: Compiler,
        mode: Mode,
776
        source_type: SourceType,
777
        target: TargetSelection,
S
Santiago Pastorino 已提交
778
        cmd: &str,
779
    ) -> Cargo {
M
Mark Simulacrum 已提交
780 781
        let mut cargo = Command::new(&self.initial_cargo);
        let out_dir = self.stage_out(compiler, mode);
782

783 784 785 786 787 788
        // Codegen backends are not yet tracked by -Zbinary-dep-depinfo,
        // so we need to explicitly clear out if they've been updated.
        for backend in self.codegen_backends(compiler) {
            self.clear_if_dirty(&out_dir, &backend);
        }

789
        if cmd == "doc" || cmd == "rustdoc" {
790
            let my_out = match mode {
791
                // This is the intended out directory for compiler documentation.
792
                Mode::Rustc | Mode::ToolRustc => self.compiler_doc_out(target),
793
                Mode::Std => out_dir.join(target.triple).join("doc"),
E
Eric Huss 已提交
794
                _ => panic!("doc mode {:?} not expected", mode),
795
            };
M
Mark Rousskov 已提交
796
            let rustdoc = self.rustdoc(compiler);
797 798 799
            self.clear_if_dirty(&my_out, &rustdoc);
        }

J
Jonas Schievink 已提交
800
        cargo.env("CARGO_TARGET_DIR", &out_dir).arg(cmd);
801 802

        let profile_var = |name: &str| {
M
Mark Rousskov 已提交
803
            let profile = if self.config.rust_optimize { "RELEASE" } else { "DEV" };
804 805
            format!("CARGO_PROFILE_{}_{}", profile, name)
        };
806

807
        // See comment in rustc_llvm/build.rs for why this is necessary, largely llvm-config
808 809 810 811 812 813
        // needs to not accidentally link to libLLVM in stage0/lib.
        cargo.env("REAL_LIBRARY_PATH_VAR", &util::dylib_path_var());
        if let Some(e) = env::var_os(util::dylib_path_var()) {
            cargo.env("REAL_LIBRARY_PATH", e);
        }

J
Joshua Nelson 已提交
814 815 816 817 818 819 820 821 822 823
        match self.build.config.color {
            Color::Always => {
                cargo.arg("--color=always");
            }
            Color::Never => {
                cargo.arg("--color=never");
            }
            Color::Auto => {} // nothing to do
        }

824
        if cmd != "install" {
825
            cargo.arg("--target").arg(target.rustc_target_arg());
826 827 828
        } else {
            assert_eq!(target, compiler.host);
        }
829

L
ljedrz 已提交
830
        // Set a flag for `check`/`clippy`/`fix`, so that certain build
831
        // scripts can do less work (i.e. not building/requiring LLVM).
L
ljedrz 已提交
832
        if cmd == "check" || cmd == "clippy" || cmd == "fix" {
833
            // If we've not yet built LLVM, or it's stale, then bust
834
            // the rustc_llvm cache. That will always work, even though it
835
            // may mean that on the next non-check build we'll need to rebuild
836
            // rustc_llvm. But if LLVM is stale, that'll be a tiny amount
837 838 839 840 841
            // of work comparitively, and we'd likely need to rebuild it anyway,
            // so that's okay.
            if crate::native::prebuilt_llvm_config(self, target).is_err() {
                cargo.env("RUST_CHECK", "1");
            }
V
varkor 已提交
842 843
        }

844
        let stage = if compiler.stage == 0 && self.local_rebuild {
845
            // Assume the local-rebuild rustc already has stage1 features.
846
            1
847
        } else {
848 849
            compiler.stage
        };
850

851
        let mut rustflags = Rustflags::new(target);
852
        if stage != 0 {
853 854 855
            if let Ok(s) = env::var("CARGOFLAGS_NOT_BOOTSTRAP") {
                cargo.args(s.split_whitespace());
            }
856 857
            rustflags.env("RUSTFLAGS_NOT_BOOTSTRAP");
        } else {
858 859 860
            if let Ok(s) = env::var("CARGOFLAGS_BOOTSTRAP") {
                cargo.args(s.split_whitespace());
            }
861
            rustflags.env("RUSTFLAGS_BOOTSTRAP");
J
Joshua Nelson 已提交
862
            if cmd == "clippy" {
863 864
                // clippy overwrites sysroot if we pass it to cargo.
                // Pass it directly to clippy instead.
J
Joshua Nelson 已提交
865 866 867 868 869 870 871 872 873 874 875 876 877
                // NOTE: this can't be fixed in clippy because we explicitly don't set `RUSTC`,
                // so it has no way of knowing the sysroot.
                rustflags.arg("--sysroot");
                rustflags.arg(
                    self.sysroot(compiler)
                        .as_os_str()
                        .to_str()
                        .expect("sysroot must be valid UTF-8"),
                );
                // Only run clippy on a very limited subset of crates (in particular, not build scripts).
                cargo.arg("-Zunstable-options");
                // Explicitly does *not* set `--cfg=bootstrap`, since we're using a nightly clippy.
                let host_version = Command::new("rustc").arg("--version").output().map_err(|_| ());
878
                let output = host_version.and_then(|output| {
879
                    if output.status.success() {
J
Joshua Nelson 已提交
880 881 882 883
                        Ok(output)
                    } else {
                        Err(())
                    }
884
                }).unwrap_or_else(|_| {
J
Joshua Nelson 已提交
885
                    eprintln!(
886
                        "error: `x.py clippy` requires a host `rustc` toolchain with the `clippy` component"
J
Joshua Nelson 已提交
887
                    );
888
                    eprintln!("help: try `rustup component add clippy`");
J
Joshua Nelson 已提交
889
                    std::process::exit(1);
890 891 892
                });
                if !t!(std::str::from_utf8(&output.stdout)).contains("nightly") {
                    rustflags.arg("--cfg=bootstrap");
J
Joshua Nelson 已提交
893 894 895 896
                }
            } else {
                rustflags.arg("--cfg=bootstrap");
            }
897 898
        }

899 900 901 902
        if self.config.rust_new_symbol_mangling {
            rustflags.arg("-Zsymbol-mangling-version=v0");
        }

903 904 905
        // FIXME: It might be better to use the same value for both `RUSTFLAGS` and `RUSTDOCFLAGS`,
        // but this breaks CI. At the very least, stage0 `rustdoc` needs `--cfg bootstrap`. See
        // #71458.
906
        let mut rustdocflags = rustflags.clone();
907

908 909 910 911
        if let Ok(s) = env::var("CARGOFLAGS") {
            cargo.args(s.split_whitespace());
        }

J
John Kåre Alsaker 已提交
912
        match mode {
M
Mark Rousskov 已提交
913
            Mode::Std | Mode::ToolBootstrap | Mode::ToolStd => {}
914
            Mode::Rustc | Mode::Codegen | Mode::ToolRustc => {
J
John Kåre Alsaker 已提交
915 916 917
                // Build proc macros both for the host and the target
                if target != compiler.host && cmd != "check" {
                    cargo.arg("-Zdual-proc-macros");
918
                    rustflags.arg("-Zdual-proc-macros");
J
John Kåre Alsaker 已提交
919
                }
M
Mark Rousskov 已提交
920
            }
J
John Kåre Alsaker 已提交
921 922
        }

923 924 925 926 927 928 929 930 931 932 933 934 935
        // This tells Cargo (and in turn, rustc) to output more complete
        // dependency information.  Most importantly for rustbuild, this
        // includes sysroot artifacts, like libstd, which means that we don't
        // need to track those in rustbuild (an error prone process!). This
        // feature is currently unstable as there may be some bugs and such, but
        // it represents a big improvement in rustbuild's reliability on
        // rebuilds, so we're using it here.
        //
        // For some additional context, see #63470 (the PR originally adding
        // this), as well as #63012 which is the tracking issue for this
        // feature on the rustc side.
        cargo.arg("-Zbinary-dep-depinfo");

936 937 938 939
        cargo.arg("-j").arg(self.jobs().to_string());
        // Remove make-related flags to ensure Cargo can correctly set things up
        cargo.env_remove("MAKEFLAGS");
        cargo.env_remove("MFLAGS");
940

941 942
        // FIXME: Temporary fix for https://github.com/rust-lang/cargo/issues/3005
        // Force cargo to output binaries with disambiguating hashes in the name
943 944
        let mut metadata = if compiler.stage == 0 {
            // Treat stage0 like a special channel, whether it's a normal prior-
945 946
            // release rustc or a local rebuild with the same version, so we
            // never mix these libraries by accident.
947
            "bootstrap".to_string()
948
        } else {
949
            self.config.channel.to_string()
950
        };
951 952 953 954 955 956 957 958 959 960 961 962 963
        // We want to make sure that none of the dependencies between
        // std/test/rustc unify with one another. This is done for weird linkage
        // reasons but the gist of the problem is that if librustc, libtest, and
        // libstd all depend on libc from crates.io (which they actually do) we
        // want to make sure they all get distinct versions. Things get really
        // weird if we try to unify all these dependencies right now, namely
        // around how many times the library is linked in dynamic libraries and
        // such. If rustc were a static executable or if we didn't ship dylibs
        // this wouldn't be a problem, but we do, so it is. This is in general
        // just here to make sure things build right. If you can remove this and
        // things still build right, please do!
        match mode {
            Mode::Std => metadata.push_str("std"),
M
Mark Rousskov 已提交
964 965 966 967 968 969 970 971 972 973 974
            // When we're building rustc tools, they're built with a search path
            // that contains things built during the rustc build. For example,
            // bitflags is built during the rustc build, and is a dependency of
            // rustdoc as well. We're building rustdoc in a different target
            // directory, though, which means that Cargo will rebuild the
            // dependency. When we go on to build rustdoc, we'll look for
            // bitflags, and find two different copies: one built during the
            // rustc step and one that we just built. This isn't always a
            // problem, somehow -- not really clear why -- but we know that this
            // fixes things.
            Mode::ToolRustc => metadata.push_str("tool-rustc"),
975 976
            // Same for codegen backends.
            Mode::Codegen => metadata.push_str("codegen"),
M
Mark Rousskov 已提交
977
            _ => {}
978
        }
979
        cargo.env("__CARGO_DEFAULT_LIB_METADATA", &metadata);
980

L
ljedrz 已提交
981
        if cmd == "clippy" {
982
            rustflags.arg("-Zforce-unstable-if-unmarked");
983 984
        }

J
Jonas Schievink 已提交
985
        rustflags.arg("-Zmacro-backtrace");
986

K
kennytm 已提交
987
        let want_rustdoc = self.doc_tests != DocTests::No;
K
kennytm 已提交
988

989 990 991 992 993 994
        // We synthetically interpret a stage0 compiler used to build tools as a
        // "raw" compiler in that it's the exact snapshot we download. Normally
        // the stage0 build means it uses libraries build by the stage0
        // compiler, but for tools we just use the precompiled libraries that
        // we've downloaded
        let use_snapshot = mode == Mode::ToolBootstrap;
995
        assert!(!use_snapshot || stage == 0 || self.local_rebuild);
996 997

        let maybe_sysroot = self.sysroot(compiler);
M
Mark Rousskov 已提交
998
        let sysroot = if use_snapshot { self.rustc_snapshot_sysroot() } else { &maybe_sysroot };
M
Mark Rousskov 已提交
999
        let libdir = self.rustc_libdir(compiler);
1000

1001 1002 1003 1004 1005 1006 1007 1008
        // Clear the output directory if the real rustc we're using has changed;
        // Cargo cannot detect this as it thinks rustc is bootstrap/debug/rustc.
        //
        // Avoid doing this during dry run as that usually means the relevant
        // compiler is not yet linked/copied properly.
        //
        // Only clear out the directory if we're compiling std; otherwise, we
        // should let Cargo take care of things for us (via depdep info)
M
Mark Rousskov 已提交
1009
        if !self.config.dry_run && mode == Mode::Std && cmd == "build" {
1010 1011 1012
            self.clear_if_dirty(&out_dir, &self.rustc(compiler));
        }

1013 1014
        // Customize the compiler we're running. Specify the compiler to cargo
        // as our shim and then pass it some various options used to configure
M
Mark Simulacrum 已提交
1015
        // how the actual compiler itself is called.
1016 1017 1018
        //
        // These variables are primarily all read by
        // src/bootstrap/bin/{rustc.rs,rustdoc.rs}
S
Santiago Pastorino 已提交
1019 1020 1021 1022
        cargo
            .env("RUSTBUILD_NATIVE_DIR", self.native_dir(target))
            .env("RUSTC_REAL", self.rustc(compiler))
            .env("RUSTC_STAGE", stage.to_string())
1023 1024
            .env("RUSTC_SYSROOT", &sysroot)
            .env("RUSTC_LIBDIR", &libdir)
S
Santiago Pastorino 已提交
1025 1026 1027
            .env("RUSTDOC", self.out.join("bootstrap/debug/rustdoc"))
            .env(
                "RUSTDOC_REAL",
1028
                if cmd == "doc" || cmd == "rustdoc" || (cmd == "test" && want_rustdoc) {
M
Mark Rousskov 已提交
1029
                    self.rustdoc(compiler)
S
Santiago Pastorino 已提交
1030 1031 1032 1033
                } else {
                    PathBuf::from("/path/to/nowhere/rustdoc/not/required")
                },
            )
1034 1035
            .env("RUSTC_ERROR_METADATA_DST", self.extended_error_dir())
            .env("RUSTC_BREAK_ON_ICE", "1");
J
Joshua Nelson 已提交
1036 1037 1038 1039 1040
        // Clippy support is a hack and uses the default `cargo-clippy` in path.
        // Don't override RUSTC so that the `cargo-clippy` in path will be run.
        if cmd != "clippy" {
            cargo.env("RUSTC", self.out.join("bootstrap/debug/rustc"));
        }
1041

1042 1043 1044 1045 1046 1047 1048 1049 1050
        // Dealing with rpath here is a little special, so let's go into some
        // detail. First off, `-rpath` is a linker option on Unix platforms
        // which adds to the runtime dynamic loader path when looking for
        // dynamic libraries. We use this by default on Unix platforms to ensure
        // that our nightlies behave the same on Windows, that is they work out
        // of the box. This can be disabled, of course, but basically that's why
        // we're gated on RUSTC_RPATH here.
        //
        // Ok, so the astute might be wondering "why isn't `-C rpath` used
K
KRAAI, MATTHEW [VISUS] 已提交
1051
        // here?" and that is indeed a good question to ask. This codegen
1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
        // option is the compiler's current interface to generating an rpath.
        // Unfortunately it doesn't quite suffice for us. The flag currently
        // takes no value as an argument, so the compiler calculates what it
        // should pass to the linker as `-rpath`. This unfortunately is based on
        // the **compile time** directory structure which when building with
        // Cargo will be very different than the runtime directory structure.
        //
        // All that's a really long winded way of saying that if we use
        // `-Crpath` then the executables generated have the wrong rpath of
        // something like `$ORIGIN/deps` when in fact the way we distribute
        // rustc requires the rpath to be `$ORIGIN/../lib`.
        //
        // So, all in all, to set up the correct rpath we pass the linker
        // argument manually via `-C link-args=-Wl,-rpath,...`. Plus isn't it
        // fun to pass a flag to a tool to pass a flag to pass a flag to a tool
        // to change a flag in a binary?
1068
        if self.config.rust_rpath && util::use_host_linker(target) {
1069 1070 1071 1072 1073 1074 1075 1076
            let rpath = if target.contains("apple") {
                // Note that we need to take one extra step on macOS to also pass
                // `-Wl,-instal_name,@rpath/...` to get things to work right. To
                // do that we pass a weird flag to the compiler to get it to do
                // so. Note that this is definitely a hack, and we should likely
                // flesh out rpath support more fully in the future.
                rustflags.arg("-Zosx-rpath-install-name");
                Some("-Wl,-rpath,@loader_path/../lib")
1077
            } else if !target.contains("windows") {
1078 1079 1080 1081 1082 1083 1084 1085 1086
                Some("-Wl,-rpath,$ORIGIN/../lib")
            } else {
                None
            };
            if let Some(rpath) = rpath {
                rustflags.arg(&format!("-Clink-args={}", rpath));
            }
        }

1087
        if let Some(host_linker) = self.linker(compiler.host) {
O
Oliver Schneider 已提交
1088 1089
            cargo.env("RUSTC_HOST_LINKER", host_linker);
        }
1090
        if self.is_fuse_ld_lld(compiler.host) {
1091 1092
            cargo.env("RUSTC_HOST_FUSE_LD_LLD", "1");
        }
1093

1094
        if let Some(target_linker) = self.linker(target) {
1095
            let target = crate::envify(&target.triple);
1096
            cargo.env(&format!("CARGO_TARGET_{}_LINKER", target), target_linker);
O
Oliver Schneider 已提交
1097
        }
1098
        if self.is_fuse_ld_lld(target) {
1099 1100 1101
            rustflags.arg("-Clink-args=-fuse-ld=lld");
        }

L
ljedrz 已提交
1102
        if !(["build", "check", "clippy", "fix", "rustc"].contains(&cmd)) && want_rustdoc {
M
Mark Rousskov 已提交
1103
            cargo.env("RUSTDOC_LIBDIR", self.rustc_libdir(compiler));
1104
        }
1105

1106
        let debuginfo_level = match mode {
1107
            Mode::Rustc | Mode::Codegen => self.config.rust_debuginfo_level_rustc,
1108
            Mode::Std => self.config.rust_debuginfo_level_std,
M
Mark Rousskov 已提交
1109 1110 1111
            Mode::ToolBootstrap | Mode::ToolStd | Mode::ToolRustc => {
                self.config.rust_debuginfo_level_tools
            }
1112
        };
1113
        cargo.env(profile_var("DEBUG"), debuginfo_level.to_string());
1114 1115 1116 1117 1118 1119 1120 1121
        cargo.env(
            profile_var("DEBUG_ASSERTIONS"),
            if mode == Mode::Std {
                self.config.rust_debug_assertions_std.to_string()
            } else {
                self.config.rust_debug_assertions.to_string()
            },
        );
1122

1123 1124 1125 1126 1127
        if self.config.cmd.bless() {
            // Bless `expect!` tests.
            cargo.env("UPDATE_EXPECT", "1");
        }

1128
        if !mode.is_tool() {
O
Oliver Schneider 已提交
1129
            cargo.env("RUSTC_FORCE_UNSTABLE", "1");
1130 1131
        }

1132
        if let Some(x) = self.crt_static(target) {
1133 1134 1135 1136 1137
            if x {
                rustflags.arg("-Ctarget-feature=+crt-static");
            } else {
                rustflags.arg("-Ctarget-feature=-crt-static");
            }
1138 1139
        }

1140 1141 1142 1143
        if let Some(x) = self.crt_static(compiler.host) {
            cargo.env("RUSTC_HOST_CRT_STATIC", x.to_string());
        }

1144 1145
        if let Some(map_to) = self.build.debuginfo_map_to(GitRepo::Rustc) {
            let map = format!("{}={}", self.build.src.display(), map_to);
1146
            cargo.env("RUSTC_DEBUGINFO_MAP", map);
1147 1148 1149 1150

            // `rustc` needs to know the virtual `/rustc/$hash` we're mapping to,
            // in order to opportunistically reverse it later.
            cargo.env("CFG_VIRTUAL_RUST_SOURCE_BASE_DIR", map_to);
1151 1152
        }

1153 1154
        // Enable usage of unstable features
        cargo.env("RUSTC_BOOTSTRAP", "1");
M
Mark Simulacrum 已提交
1155
        self.add_rust_test_threads(&mut cargo);
1156 1157 1158

        // Almost all of the crates that we compile as part of the bootstrap may
        // have a build script, including the standard library. To compile a
M
Mark Simulacrum 已提交
1159
        // build script, however, it itself needs a standard library! This
1160
        // introduces a bit of a pickle when we're compiling the standard
M
Mark Simulacrum 已提交
1161
        // library itself.
1162 1163
        //
        // To work around this we actually end up using the snapshot compiler
M
Mark Simulacrum 已提交
1164
        // (stage0) for compiling build scripts of the standard library itself.
1165 1166 1167 1168 1169
        // The stage0 compiler is guaranteed to have a libstd available for use.
        //
        // For other crates, however, we know that we've already got a standard
        // library up and running, so we can use the normal compiler to compile
        // build scripts in that situation.
B
bjorn3 已提交
1170
        if mode == Mode::Std {
S
Santiago Pastorino 已提交
1171 1172 1173
            cargo
                .env("RUSTC_SNAPSHOT", &self.initial_rustc)
                .env("RUSTC_SNAPSHOT_LIBDIR", self.rustc_snapshot_libdir());
1174
        } else {
S
Santiago Pastorino 已提交
1175 1176 1177
            cargo
                .env("RUSTC_SNAPSHOT", self.rustc(compiler))
                .env("RUSTC_SNAPSHOT_LIBDIR", self.rustc_libdir(compiler));
1178 1179
        }

1180 1181 1182 1183
        // Tools that use compiler libraries may inherit the `-lLLVM` link
        // requirement, but the `-L` library path is not propagated across
        // separate Cargo projects. We can add LLVM's library path to the
        // platform-specific environment variable as a workaround.
1184 1185 1186 1187 1188
        if mode == Mode::ToolRustc {
            if let Some(llvm_config) = self.llvm_config(target) {
                let llvm_libdir = output(Command::new(&llvm_config).arg("--libdir"));
                add_link_lib_path(vec![llvm_libdir.trim().into()], &mut cargo);
            }
1189 1190
        }

1191 1192 1193 1194 1195 1196 1197 1198
        // Compile everything except libraries and proc macros with the more
        // efficient initial-exec TLS model. This doesn't work with `dlopen`,
        // so we can't use it by default in general, but we can use it for tools
        // and our own internal libraries.
        if !mode.must_support_dlopen() {
            rustflags.arg("-Ztls-model=initial-exec");
        }

1199
        if self.config.incremental {
1200
            cargo.env("CARGO_INCREMENTAL", "1");
1201 1202 1203
        } else {
            // Don't rely on any default setting for incr. comp. in Cargo
            cargo.env("CARGO_INCREMENTAL", "0");
1204 1205
        }

M
Mark Simulacrum 已提交
1206
        if let Some(ref on_fail) = self.config.on_fail {
1207 1208 1209
            cargo.env("RUSTC_ON_FAIL", on_fail);
        }

1210 1211 1212 1213
        if self.config.print_step_timings {
            cargo.env("RUSTC_PRINT_STEP_TIMINGS", "1");
        }

J
John Kåre Alsaker 已提交
1214 1215 1216 1217
        if self.config.backtrace_on_ice {
            cargo.env("RUSTC_BACKTRACE_ON_ICE", "1");
        }

L
ljedrz 已提交
1218
        cargo.env("RUSTC_VERBOSE", self.verbosity.to_string());
1219

1220
        if source_type == SourceType::InTree {
1221
            let mut lint_flags = Vec::new();
1222 1223 1224
            // When extending this list, add the new lints to the RUSTFLAGS of the
            // build_bootstrap function of src/bootstrap/bootstrap.py as well as
            // some code doesn't go through this `rustc` wrapper.
1225 1226
            lint_flags.push("-Wrust_2018_idioms");
            lint_flags.push("-Wunused_lifetimes");
1227 1228

            if self.config.deny_warnings {
1229
                lint_flags.push("-Dwarnings");
1230
                rustdocflags.arg("-Dwarnings");
1231
            }
1232

1233 1234 1235 1236
            // FIXME(#58633) hide "unused attribute" errors in incremental
            // builds of the standard library, as the underlying checks are
            // not yet properly integrated with incremental recompilation.
            if mode == Mode::Std && compiler.stage == 0 && self.config.incremental {
1237
                lint_flags.push("-Aunused-attributes");
1238
            }
1239 1240 1241 1242 1243 1244 1245 1246 1247
            // This does not use RUSTFLAGS due to caching issues with Cargo.
            // Clippy is treated as an "in tree" tool, but shares the same
            // cache as other "submodule" tools. With these options set in
            // RUSTFLAGS, that causes *every* shared dependency to be rebuilt.
            // By injecting this into the rustc wrapper, this circumvents
            // Cargo's fingerprint detection. This is fine because lint flags
            // are always ignored in dependencies. Eventually this should be
            // fixed via better support from Cargo.
            cargo.env("RUSTC_LINT_FLAGS", lint_flags.join(" "));
G
Guillaume Gomez 已提交
1248

1249
            rustdocflags.arg("-Winvalid_codeblock_attributes");
1250 1251
        }

1252
        if mode == Mode::Rustc {
A
Alex Crichton 已提交
1253 1254
            rustflags.arg("-Zunstable-options");
            rustflags.arg("-Wrustc::internal");
1255 1256
        }

O
Oliver Schneider 已提交
1257 1258 1259 1260 1261
        // Throughout the build Cargo can execute a number of build scripts
        // compiling C/C++ code and we need to pass compilers, archivers, flags, etc
        // obtained previously to those build scripts.
        // Build scripts use either the `cc` crate or `configure/make` so we pass
        // the options through environment variables that are fetched and understood by both.
1262 1263
        //
        // FIXME: the guard against msvc shouldn't need to be here
1264 1265 1266 1267 1268
        if target.contains("msvc") {
            if let Some(ref cl) = self.config.llvm_clang_cl {
                cargo.env("CC", cl).env("CXX", cl);
            }
        } else {
1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285
            let ccache = self.config.ccache.as_ref();
            let ccacheify = |s: &Path| {
                let ccache = match ccache {
                    Some(ref s) => s,
                    None => return s.display().to_string(),
                };
                // FIXME: the cc-rs crate only recognizes the literal strings
                // `ccache` and `sccache` when doing caching compilations, so we
                // mirror that here. It should probably be fixed upstream to
                // accept a new env var or otherwise work with custom ccache
                // vars.
                match &ccache[..] {
                    "ccache" | "sccache" => format!("{} {}", ccache, s.display()),
                    _ => s.display().to_string(),
                }
            };
            let cc = ccacheify(&self.cc(target));
1286
            cargo.env(format!("CC_{}", target.triple), &cc);
O
Oliver Schneider 已提交
1287

1288
            let cflags = self.cflags(target, GitRepo::Rustc).join(" ");
1289
            cargo.env(format!("CFLAGS_{}", target.triple), &cflags);
O
Oliver Schneider 已提交
1290 1291 1292

            if let Some(ar) = self.ar(target) {
                let ranlib = format!("{} s", ar.display());
1293 1294 1295
                cargo
                    .env(format!("AR_{}", target.triple), ar)
                    .env(format!("RANLIB_{}", target.triple), ranlib);
O
Oliver Schneider 已提交
1296
            }
1297

M
Mark Simulacrum 已提交
1298
            if let Ok(cxx) = self.cxx(target) {
1299
                let cxx = ccacheify(&cxx);
S
Santiago Pastorino 已提交
1300
                cargo
1301 1302
                    .env(format!("CXX_{}", target.triple), &cxx)
                    .env(format!("CXXFLAGS_{}", target.triple), cflags);
1303 1304 1305
            }
        }

M
Mark Rousskov 已提交
1306
        if mode == Mode::Std && self.config.extended && compiler.is_final_stage(self) {
1307
            rustflags.arg("-Zsave-analysis");
M
Mark Rousskov 已提交
1308 1309 1310
            cargo.env(
                "RUST_SAVE_ANALYSIS_CONFIG",
                "{\"output_file\": null,\"full_docs\": false,\
1311
                       \"pub_only\": true,\"reachable_only\": false,\
M
Mark Rousskov 已提交
1312 1313
                       \"distro_crate\": true,\"signatures\": false,\"borrow_data\": false}",
            );
1314 1315
        }

A
Andrew Paverd 已提交
1316
        // If Control Flow Guard is enabled, pass the `control-flow-guard` flag to rustc
1317 1318 1319 1320 1321 1322 1323 1324 1325 1326
        // when compiling the standard library, since this might be linked into the final outputs
        // produced by rustc. Since this mitigation is only available on Windows, only enable it
        // for the standard library in case the compiler is run on a non-Windows platform.
        // This is not needed for stage 0 artifacts because these will only be used for building
        // the stage 1 compiler.
        if cfg!(windows)
            && mode == Mode::Std
            && self.config.control_flow_guard
            && compiler.stage >= 1
        {
1327
            rustflags.arg("-Ccontrol-flow-guard");
1328 1329
        }

O
Oliver Schneider 已提交
1330
        // For `cargo doc` invocations, make rustdoc print the Rust version into the docs
1331 1332 1333 1334 1335
        // This replaces spaces with newlines because RUSTDOCFLAGS does not
        // support arguments with regular spaces. Hopefully someday Cargo will
        // have space support.
        let rust_version = self.rust_version().replace(' ', "\n");
        rustdocflags.arg("--crate-version").arg(&rust_version);
O
Oliver Schneider 已提交
1336

1337 1338 1339
        // Environment variables *required* throughout the build
        //
        // FIXME: should update code to not require this env var
1340
        cargo.env("CFG_COMPILER_HOST_TRIPLE", target.triple);
1341

1342
        // Set this for all builds to make sure doc builds also get it.
1343
        cargo.env("CFG_RELEASE_CHANNEL", &self.config.channel);
1344

1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362
        // This one's a bit tricky. As of the time of this writing the compiler
        // links to the `winapi` crate on crates.io. This crate provides raw
        // bindings to Windows system functions, sort of like libc does for
        // Unix. This crate also, however, provides "import libraries" for the
        // MinGW targets. There's an import library per dll in the windows
        // distribution which is what's linked to. These custom import libraries
        // are used because the winapi crate can reference Windows functions not
        // present in the MinGW import libraries.
        //
        // For example MinGW may ship libdbghelp.a, but it may not have
        // references to all the functions in the dbghelp dll. Instead the
        // custom import library for dbghelp in the winapi crates has all this
        // information.
        //
        // Unfortunately for us though the import libraries are linked by
        // default via `-ldylib=winapi_foo`. That is, they're linked with the
        // `dylib` type with a `winapi_` prefix (so the winapi ones don't
        // conflict with the system MinGW ones). This consequently means that
I
Irina Popa 已提交
1363
        // the binaries we ship of things like rustc_codegen_llvm (aka the rustc_codegen_llvm
1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
        // DLL) when linked against *again*, for example with procedural macros
        // or plugins, will trigger the propagation logic of `-ldylib`, passing
        // `-lwinapi_foo` to the linker again. This isn't actually available in
        // our distribution, however, so the link fails.
        //
        // To solve this problem we tell winapi to not use its bundled import
        // libraries. This means that it will link to the system MinGW import
        // libraries by default, and the `-ldylib=foo` directives will still get
        // passed to the final linker, but they'll look like `-lfoo` which can
        // be resolved because MinGW has the import library. The downside is we
        // don't get newer functions from Windows, but we don't use any of them
        // anyway.
C
Collins Abitekaniza 已提交
1376
        if !mode.is_tool() {
A
Alex Crichton 已提交
1377 1378
            cargo.env("WINAPI_NO_BUNDLED_LIBRARIES", "1");
        }
1379

C
comex 已提交
1380
        for _ in 1..self.verbosity {
1381 1382
            cargo.arg("-v");
        }
1383

1384
        match (mode, self.config.rust_codegen_units_std, self.config.rust_codegen_units) {
M
Mark Rousskov 已提交
1385
            (Mode::Std, Some(n), _) | (_, _, Some(n)) => {
1386
                cargo.env(profile_var("CODEGEN_UNITS"), n.to_string());
1387 1388 1389 1390
            }
            _ => {
                // Don't set anything
            }
1391 1392
        }

O
Oliver Schneider 已提交
1393
        if self.config.rust_optimize {
1394 1395
            // FIXME: cargo bench/install do not accept `--release`
            if cmd != "bench" && cmd != "install" {
O
Oliver Schneider 已提交
1396 1397
                cargo.arg("--release");
            }
1398
        }
1399

M
Mark Simulacrum 已提交
1400
        if self.config.locked_deps {
1401 1402
            cargo.arg("--locked");
        }
M
Mark Simulacrum 已提交
1403
        if self.config.vendor || self.is_sudo {
1404 1405 1406
            cargo.arg("--frozen");
        }

1407
        // Try to use a sysroot-relative bindir, in case it was configured absolutely.
1408
        cargo.env("RUSTC_INSTALL_BINDIR", self.config.bindir_relative());
1409

M
Mark Simulacrum 已提交
1410
        self.ci_env.force_coloring_in_ci(&mut cargo);
1411

1412 1413 1414
        // When we build Rust dylibs they're all intended for intermediate
        // usage, so make sure we pass the -Cprefer-dynamic flag instead of
        // linking all deps statically into the dylib.
1415
        if matches!(mode, Mode::Std | Mode::Rustc) {
A
Alex Crichton 已提交
1416
            rustflags.arg("-Cprefer-dynamic");
1417 1418
        }

1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433
        // When building incrementally we default to a lower ThinLTO import limit
        // (unless explicitly specified otherwise). This will produce a somewhat
        // slower code but give way better compile times.
        {
            let limit = match self.config.rust_thin_lto_import_instr_limit {
                Some(limit) => Some(limit),
                None if self.config.incremental => Some(10),
                _ => None,
            };

            if let Some(limit) = limit {
                rustflags.arg(&format!("-Cllvm-args=-import-instr-limit={}", limit));
            }
        }

1434
        Cargo { command: cargo, rustflags, rustdocflags }
1435 1436
    }

V
varkor 已提交
1437
    /// Ensure that a given step is built, returning its output. This will
1438 1439
    /// cache the step, so it is safe (and good!) to call this as often as
    /// needed to ensure that all dependencies are built.
1440
    pub fn ensure<S: Step>(&'a self, step: S) -> S::Output {
1441 1442
        {
            let mut stack = self.stack.borrow_mut();
1443 1444
            for stack_step in stack.iter() {
                // should skip
M
Mark Rousskov 已提交
1445
                if stack_step.downcast_ref::<S>().map_or(true, |stack_step| *stack_step != step) {
1446
                    continue;
1447
                }
1448
                let mut out = String::new();
1449
                out += &format!("\n\nCycle in build detected when adding {:?}\n", step);
1450 1451 1452 1453 1454
                for el in stack.iter().rev() {
                    out += &format!("\t{:?}\n", el);
                }
                panic!(out);
            }
1455
            if let Some(out) = self.cache.get(&step) {
1456
                self.verbose(&format!("{}c {:?}", "  ".repeat(stack.len()), step));
1457 1458 1459

                return out;
            }
1460
            self.verbose(&format!("{}> {:?}", "  ".repeat(stack.len()), step));
1461
            stack.push(Box::new(step.clone()));
1462
        }
1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473

        let (out, dur) = {
            let start = Instant::now();
            let zero = Duration::new(0, 0);
            let parent = self.time_spent_on_dependencies.replace(zero);
            let out = step.clone().run(self);
            let dur = start.elapsed();
            let deps = self.time_spent_on_dependencies.replace(parent + dur);
            (out, dur - deps)
        };

M
Mark Rousskov 已提交
1474
        if self.config.print_step_timings && !self.config.dry_run {
1475
            println!("[TIMING] {:?} -- {}.{:03}", step, dur.as_secs(), dur.subsec_millis());
1476 1477
        }

1478 1479
        {
            let mut stack = self.stack.borrow_mut();
1480 1481
            let cur_step = stack.pop().expect("step stack empty");
            assert_eq!(cur_step.downcast_ref(), Some(&step));
1482
        }
M
Mark Rousskov 已提交
1483
        self.verbose(&format!("{}< {:?}", "  ".repeat(self.stack.borrow().len()), step));
1484 1485
        self.cache.put(step, out.clone());
        out
1486 1487
    }
}
M
Mark Simulacrum 已提交
1488 1489

#[cfg(test)]
C
chansuke 已提交
1490
mod tests;
1491

1492
#[derive(Debug, Clone)]
1493 1494 1495
struct Rustflags(String);

impl Rustflags {
1496
    fn new(target: TargetSelection) -> Rustflags {
1497
        let mut ret = Rustflags(String::new());
1498

A
Alex Crichton 已提交
1499
        // Inherit `RUSTFLAGS` by default ...
1500
        ret.env("RUSTFLAGS");
1501 1502

        // ... and also handle target-specific env RUSTFLAGS if they're
1503
        // configured.
1504
        let target_specific = format!("CARGO_TARGET_{}_RUSTFLAGS", crate::envify(&target.triple));
1505 1506
        ret.env(&target_specific);

A
Alex Crichton 已提交
1507
        ret
1508 1509 1510 1511
    }

    fn env(&mut self, env: &str) {
        if let Ok(s) = env::var(env) {
1512
            for part in s.split(' ') {
1513 1514 1515 1516 1517 1518
                self.arg(part);
            }
        }
    }

    fn arg(&mut self, arg: &str) -> &mut Self {
1519
        assert_eq!(arg.split(' ').count(), 1);
1520
        if !self.0.is_empty() {
1521 1522 1523 1524 1525 1526
            self.0.push_str(" ");
        }
        self.0.push_str(arg);
        self
    }
}
1527 1528 1529 1530 1531

#[derive(Debug)]
pub struct Cargo {
    command: Command,
    rustflags: Rustflags,
1532
    rustdocflags: Rustflags,
1533 1534 1535
}

impl Cargo {
M
Mark Rousskov 已提交
1536 1537 1538 1539
    pub fn rustdocflag(&mut self, arg: &str) -> &mut Cargo {
        self.rustdocflags.arg(arg);
        self
    }
1540 1541 1542 1543 1544
    pub fn rustflag(&mut self, arg: &str) -> &mut Cargo {
        self.rustflags.arg(arg);
        self
    }

1545 1546 1547 1548 1549 1550
    pub fn arg(&mut self, arg: impl AsRef<OsStr>) -> &mut Cargo {
        self.command.arg(arg.as_ref());
        self
    }

    pub fn args<I, S>(&mut self, args: I) -> &mut Cargo
M
Mark Rousskov 已提交
1551 1552 1553
    where
        I: IntoIterator<Item = S>,
        S: AsRef<OsStr>,
1554 1555 1556 1557 1558 1559 1560 1561
    {
        for arg in args {
            self.arg(arg.as_ref());
        }
        self
    }

    pub fn env(&mut self, key: impl AsRef<OsStr>, value: impl AsRef<OsStr>) -> &mut Cargo {
M
Mark Rousskov 已提交
1562 1563 1564
        // These are managed through rustflag/rustdocflag interfaces.
        assert_ne!(key.as_ref(), "RUSTFLAGS");
        assert_ne!(key.as_ref(), "RUSTDOCFLAGS");
1565 1566 1567
        self.command.env(key.as_ref(), value.as_ref());
        self
    }
1568 1569 1570 1571

    pub fn add_rustc_lib_path(&mut self, builder: &Builder<'_>, compiler: Compiler) {
        builder.add_rustc_lib_path(compiler, &mut self.command);
    }
1572 1573 1574 1575
}

impl From<Cargo> for Command {
    fn from(mut cargo: Cargo) -> Command {
1576 1577 1578 1579 1580 1581 1582 1583 1584 1585
        let rustflags = &cargo.rustflags.0;
        if !rustflags.is_empty() {
            cargo.command.env("RUSTFLAGS", rustflags);
        }

        let rustdocflags = &cargo.rustdocflags.0;
        if !rustdocflags.is_empty() {
            cargo.command.env("RUSTDOCFLAGS", rustdocflags);
        }

1586 1587 1588
        cargo.command
    }
}