builder.rs 33.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
// Copyright 2017 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

11 12 13 14
use std::any::Any;
use std::cell::RefCell;
use std::collections::BTreeSet;
use std::env;
15
use std::fmt::Debug;
16
use std::fs;
17
use std::hash::Hash;
18
use std::ops::Deref;
19 20 21 22 23 24 25 26
use std::path::{Path, PathBuf};
use std::process::Command;

use compile;
use install;
use dist;
use util::{exe, libdir, add_lib_path};
use {Build, Mode};
27
use cache::{INTERNER, Interned, Cache};
28
use check;
29
use test;
30 31
use flags::Subcommand;
use doc;
M
Mark Simulacrum 已提交
32
use tool;
A
Alex Crichton 已提交
33
use native;
34

35 36
pub use Compiler;

37 38 39 40 41
pub struct Builder<'a> {
    pub build: &'a Build,
    pub top_stage: u32,
    pub kind: Kind,
    cache: Cache,
42
    stack: RefCell<Vec<Box<Any>>>,
43 44 45 46 47 48 49 50 51 52
}

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

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

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

58 59
    const DEFAULT: bool = false;

60
    /// Run this rule for all hosts without cross compiling.
61 62 63 64 65 66 67 68
    const ONLY_HOSTS: bool = false;

    /// Run this rule for all targets, but only with the native host.
    const ONLY_BUILD_TARGETS: bool = false;

    /// Only run this step with the build triple as host and target.
    const ONLY_BUILD: bool = false;

69 70
    /// Primary function to execute this rule. Can call `builder.ensure(...)`
    /// with other steps to run those.
71
    fn run(self, builder: &Builder) -> Self::Output;
72

73 74 75 76
    /// 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
    /// when we are not passed any paths; in that case, make_run is called
    /// directly.
77
    fn should_run(run: ShouldRun) -> ShouldRun;
78

79 80 81 82 83 84
    /// Build up a "root" rule, either as a default rule or from a path passed
    /// 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`.
85
    fn make_run(_run: RunConfig) {
86 87 88 89 90 91
        // 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!()
    }
92 93
}

94 95 96 97
pub struct RunConfig<'a> {
    pub builder: &'a Builder<'a>,
    pub host: Interned<String>,
    pub target: Interned<String>,
98
    pub path: PathBuf,
99 100
}

101 102 103 104 105 106
struct StepDescription {
    default: bool,
    only_hosts: bool,
    only_build_targets: bool,
    only_build: bool,
    should_run: fn(ShouldRun) -> ShouldRun,
107
    make_run: fn(RunConfig),
108 109 110 111 112 113 114 115 116
    name: &'static str,
}

#[derive(Debug, Clone, PartialOrd, Ord, PartialEq, Eq)]
struct PathSet {
    set: BTreeSet<PathBuf>,
}

impl PathSet {
117 118 119 120
    fn empty() -> PathSet {
        PathSet { set: BTreeSet::new() }
    }

121 122 123 124 125 126 127 128 129 130
    fn one<P: Into<PathBuf>>(path: P) -> PathSet {
        let mut set = BTreeSet::new();
        set.insert(path.into());
        PathSet { set }
    }

    fn has(&self, needle: &Path) -> bool {
        self.set.iter().any(|p| p.ends_with(needle))
    }

131 132
    fn path(&self, builder: &Builder) -> PathBuf {
        self.set.iter().next().unwrap_or(&builder.build.src).to_path_buf()
133
    }
134 135 136 137 138 139 140 141 142 143 144
}

impl StepDescription {
    fn from<S: Step>() -> StepDescription {
        StepDescription {
            default: S::DEFAULT,
            only_hosts: S::ONLY_HOSTS,
            only_build_targets: S::ONLY_BUILD_TARGETS,
            only_build: S::ONLY_BUILD,
            should_run: S::should_run,
            make_run: S::make_run,
145
            name: unsafe { ::std::intrinsics::type_name::<S>() },
146 147 148
        }
    }

149 150 151 152 153 154 155
    fn maybe_run(&self, builder: &Builder, pathset: &PathSet) {
        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() {
            eprintln!("{:?} not skipped for {:?} -- not in {:?}", pathset,
                self.name, builder.config.exclude);
156
        }
157 158
        let build = builder.build;
        let hosts = if self.only_build_targets || self.only_build {
M
Mark Simulacrum 已提交
159
            build.build_triple()
160 161 162 163
        } else {
            &build.hosts
        };

M
Mark Simulacrum 已提交
164
        // Determine the targets participating in this rule.
165
        let targets = if self.only_hosts {
166
            if build.config.run_host_only {
167 168
                &[]
            } else if self.only_build {
M
Mark Simulacrum 已提交
169
                build.build_triple()
170
            } else {
M
Mark Simulacrum 已提交
171
                &build.hosts
172 173 174 175 176 177 178
            }
        } else {
            &build.targets
        };

        for host in hosts {
            for target in targets {
179 180
                let run = RunConfig {
                    builder,
181
                    path: pathset.path(builder),
182 183 184 185
                    host: *host,
                    target: *target,
                };
                (self.make_run)(run);
186 187 188 189 190
            }
        }
    }

    fn run(v: &[StepDescription], builder: &Builder, paths: &[PathBuf]) {
191 192 193
        let should_runs = v.iter().map(|desc| {
            (desc.should_run)(ShouldRun::new(builder))
        }).collect::<Vec<_>>();
194 195 196 197 198 199 200

        // sanity checks on rules
        for (desc, should_run) in v.iter().zip(&should_runs) {
            assert!(!should_run.paths.is_empty(),
                "{:?} should have at least one pathset", desc.name);
        }

201
        if paths.is_empty() {
202 203
            for (desc, should_run) in v.iter().zip(should_runs) {
                if desc.default && should_run.is_really_default {
204 205 206
                    for pathset in &should_run.paths {
                        desc.maybe_run(builder, pathset);
                    }
207 208 209 210 211
                }
            }
        } else {
            for path in paths {
                let mut attempted_run = false;
212
                for (desc, should_run) in v.iter().zip(&should_runs) {
213
                    if let Some(pathset) = should_run.pathset_for_path(path) {
214
                        attempted_run = true;
215
                        desc.maybe_run(builder, pathset);
216 217 218 219
                    }
                }

                if !attempted_run {
220
                    panic!("Error: no rules matched {}.", path.display());
221 222 223 224 225 226
                }
            }
        }
    }
}

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

    // If this is a default rule, this is an additional constraint placed on
    // it's run. Generally something like compiler docs being enabled.
    is_really_default: bool,
236 237 238 239 240
}

impl<'a> ShouldRun<'a> {
    fn new(builder: &'a Builder) -> ShouldRun<'a> {
        ShouldRun {
241
            builder,
242
            paths: BTreeSet::new(),
243
            is_really_default: true, // by default no additional conditions
244 245 246
        }
    }

247 248 249 250 251
    pub fn default_condition(mut self, cond: bool) -> Self {
        self.is_really_default = cond;
        self
    }

252 253 254 255 256 257 258 259 260 261 262 263
    // Unlike `krate` this will create just one pathset. As such, it probably shouldn't actually
    // ever be used, but as we transition to having all rules properly handle passing krate(...) by
    // actually doing something different for every crate passed.
    pub fn all_krates(mut self, name: &str) -> Self {
        let mut set = BTreeSet::new();
        for krate in self.builder.in_tree_crates(name) {
            set.insert(PathBuf::from(&krate.path));
        }
        self.paths.insert(PathSet { set });
        self
    }

264
    pub fn krate(mut self, name: &str) -> Self {
265 266
        for krate in self.builder.in_tree_crates(name) {
            self.paths.insert(PathSet::one(&krate.path));
267 268 269 270
        }
        self
    }

271 272 273 274 275 276 277 278 279 280
    // 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 {
        self.paths.insert(PathSet {
            set: paths.iter().map(PathBuf::from).collect(),
        });
281 282 283 284
        self
    }

    // allows being more explicit about why should_run in Step returns the value passed to it
285 286
    pub fn never(mut self) -> ShouldRun<'a> {
        self.paths.insert(PathSet::empty());
287 288 289
        self
    }

290 291
    fn pathset_for_path(&self, path: &Path) -> Option<&PathSet> {
        self.paths.iter().find(|pathset| pathset.has(path))
292 293 294
    }
}

295 296 297
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Kind {
    Build,
298
    Check,
299 300 301 302 303 304 305
    Test,
    Bench,
    Dist,
    Doc,
    Install,
}

306 307 308 309 310 311
impl<'a> Builder<'a> {
    fn get_step_descriptions(kind: Kind) -> Vec<StepDescription> {
        macro_rules! describe {
            ($($rule:ty),+ $(,)*) => {{
                vec![$(StepDescription::from::<$rule>()),+]
            }};
312
        }
313 314 315 316 317
        match kind {
            Kind::Build => describe!(compile::Std, compile::Test, compile::Rustc,
                compile::StartupObjects, tool::BuildManifest, tool::Rustbook, tool::ErrorIndex,
                tool::UnstableBookGen, tool::Tidy, tool::Linkchecker, tool::CargoTest,
                tool::Compiletest, tool::RemoteTestServer, tool::RemoteTestClient,
O
Oliver Schneider 已提交
318
                tool::RustInstaller, tool::Cargo, tool::Rls, tool::Rustdoc, tool::Clippy,
319
                native::Llvm, tool::Rustfmt, tool::Miri),
320
            Kind::Check => describe!(check::Std, check::Test, check::Rustc),
321 322 323 324 325 326 327 328 329 330
            Kind::Test => describe!(test::Tidy, test::Bootstrap, test::Ui, test::RunPass,
                test::CompileFail, test::ParseFail, test::RunFail, test::RunPassValgrind,
                test::MirOpt, test::Codegen, test::CodegenUnits, test::Incremental, test::Debuginfo,
                test::UiFullDeps, test::RunPassFullDeps, test::RunFailFullDeps,
                test::CompileFailFullDeps, test::IncrementalFullDeps, test::Rustdoc, test::Pretty,
                test::RunPassPretty, test::RunFailPretty, test::RunPassValgrindPretty,
                test::RunPassFullDepsPretty, test::RunFailFullDepsPretty, test::RunMake,
                test::Crate, test::CrateLibrustc, test::Rustdoc, test::Linkcheck, test::Cargotest,
                test::Cargo, test::Rls, test::Docs, test::ErrorIndex, test::Distcheck,
                test::Rustfmt, test::Miri, test::Clippy, test::RustdocJS, test::RustdocTheme),
331
            Kind::Bench => describe!(test::Crate, test::CrateLibrustc),
332 333
            Kind::Doc => describe!(doc::UnstableBook, doc::UnstableBookGen, doc::TheBook,
                doc::Standalone, doc::Std, doc::Test, doc::Rustc, doc::ErrorIndex, doc::Nomicon,
P
projektir 已提交
334
                doc::Reference, doc::Rustdoc, doc::RustByExample, doc::CargoBook),
335 336
            Kind::Dist => describe!(dist::Docs, dist::Mingw, dist::Rustc, dist::DebuggerScripts,
                dist::Std, dist::Analysis, dist::Src, dist::PlainSourceTarball, dist::Cargo,
337
                dist::Rls, dist::Rustfmt, dist::Extended, dist::HashSign),
338
            Kind::Install => describe!(install::Docs, install::Std, install::Cargo, install::Rls,
O
Oliver Schneider 已提交
339
                install::Rustfmt, install::Analysis, install::Src, install::Rustc),
340 341
        }
    }
342

343 344 345 346 347 348 349 350 351 352 353 354
    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,
        };

        let builder = Builder {
355
            build,
M
Mark Simulacrum 已提交
356
            top_stage: build.config.stage.unwrap_or(2),
357
            kind,
358 359 360 361 362 363
            cache: Cache::new(),
            stack: RefCell::new(Vec::new()),
        };

        let builder = &builder;
        let mut should_run = ShouldRun::new(builder);
364 365
        for desc in Builder::get_step_descriptions(builder.kind) {
            should_run = (desc.should_run)(should_run);
366 367
        }
        let mut help = String::from("Available paths:\n");
368 369 370 371
        for pathset in should_run.paths {
            for path in pathset.set {
                help.push_str(format!("    ./x.py {} {}\n", subcommand, path.display()).as_str());
            }
372 373 374 375
        }
        Some(help)
    }

376
    pub fn run(build: &Build) {
M
Mark Simulacrum 已提交
377
        let (kind, paths) = match build.config.cmd {
378
            Subcommand::Build { ref paths } => (Kind::Build, &paths[..]),
379
            Subcommand::Check { ref paths } => (Kind::Check, &paths[..]),
380 381 382 383 384
            Subcommand::Doc { ref paths } => (Kind::Doc, &paths[..]),
            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[..]),
O
Oliver Schneider 已提交
385
            Subcommand::Clean { .. } => panic!(),
386 387
        };

388 389 390 391
        if let Some(path) = paths.get(0) {
            if path == Path::new("nonexistent/path/to/trigger/cargo/metadata") {
                return;
            }
392 393
        }

394
        let builder = Builder {
395
            build,
M
Mark Simulacrum 已提交
396
            top_stage: build.config.stage.unwrap_or(2),
397
            kind,
398 399 400 401
            cache: Cache::new(),
            stack: RefCell::new(Vec::new()),
        };

402 403 404 405 406 407
        if kind == Kind::Dist {
            assert!(!build.config.test_miri, "Do not distribute with miri enabled.\n\
                The distributed libraries would include all MIR (increasing binary size).
                The distributed MIR would include validation statements.");
        }

408
        StepDescription::run(&Builder::get_step_descriptions(builder.kind), &builder, paths);
409 410 411 412
    }

    pub fn default_doc(&self, paths: Option<&[PathBuf]>) {
        let paths = paths.unwrap_or(&[]);
413
        StepDescription::run(&Builder::get_step_descriptions(Kind::Doc), self, paths);
414 415
    }

B
Bastien Orivel 已提交
416
    /// Obtain a compiler at a given stage and for a given host. Explicitly does
417 418 419
    /// 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).
420
    pub fn compiler(&self, stage: u32, host: Interned<String>) -> Compiler {
421 422 423
        self.ensure(compile::Assemble { target_compiler: Compiler { stage, host } })
    }

424
    pub fn sysroot(&self, compiler: Compiler) -> Interned<PathBuf> {
425 426 427 428 429
        self.ensure(compile::Sysroot { compiler })
    }

    /// Returns the libdir where the standard library and other artifacts are
    /// found for a compiler's sysroot.
430 431 432 433 434 435 436
    pub fn sysroot_libdir(
        &self, compiler: Compiler, target: Interned<String>
    ) -> Interned<PathBuf> {
        #[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
        struct Libdir {
            compiler: Compiler,
            target: Interned<String>,
437
        }
438 439
        impl Step for Libdir {
            type Output = Interned<PathBuf>;
440

441 442
            fn should_run(run: ShouldRun) -> ShouldRun {
                run.never()
443 444
            }

445
            fn run(self, builder: &Builder) -> Interned<PathBuf> {
446
                let compiler = self.compiler;
447 448 449
                let config = &builder.build.config;
                let lib = if compiler.stage >= 1 && config.libdir_relative().is_some() {
                    builder.build.config.libdir_relative().unwrap()
450
                } else {
451
                    Path::new("lib")
452 453 454
                };
                let sysroot = builder.sysroot(self.compiler).join(lib)
                    .join("rustlib").join(self.target).join("lib");
455 456
                let _ = fs::remove_dir_all(&sysroot);
                t!(fs::create_dir_all(&sysroot));
457
                INTERNER.intern_path(sysroot)
458 459 460 461 462
            }
        }
        self.ensure(Libdir { compiler, target })
    }

463 464 465 466 467
    pub fn sysroot_codegen_backends(&self, compiler: Compiler) -> PathBuf {
        self.sysroot_libdir(compiler, compiler.host)
            .with_file_name("codegen-backends")
    }

468 469 470 471 472 473 474 475 476
    /// 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) {
            self.build.rustc_snapshot_libdir()
        } else {
477
            self.sysroot(compiler).join(libdir(&compiler.host))
478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493
        }
    }

    /// Adds the compiler's directory of dynamic libraries to `cmd`'s dynamic
    /// library lookup path.
    pub fn add_rustc_lib_path(&self, compiler: Compiler, cmd: &mut Command) {
        // 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) {
            return
        }

        add_lib_path(vec![self.rustc_libdir(compiler)], cmd);
    }

494 495 496 497 498
    /// Get a path to the compiler specified.
    pub fn rustc(&self, compiler: Compiler) -> PathBuf {
        if compiler.is_snapshot(self) {
            self.initial_rustc.clone()
        } else {
499
            self.sysroot(compiler).join("bin").join(exe("rustc", &compiler.host))
500 501 502
        }
    }

503 504
    pub fn rustdoc(&self, host: Interned<String>) -> PathBuf {
        self.ensure(tool::Rustdoc { host })
M
Mark Simulacrum 已提交
505 506
    }

507
    pub fn rustdoc_cmd(&self, host: Interned<String>) -> Command {
M
Mark Simulacrum 已提交
508
        let mut cmd = Command::new(&self.out.join("bootstrap/debug/rustdoc"));
509
        let compiler = self.compiler(self.top_stage, host);
O
Oliver Schneider 已提交
510 511
        cmd.env("RUSTC_STAGE", compiler.stage.to_string())
           .env("RUSTC_SYSROOT", self.sysroot(compiler))
512
           .env("RUSTDOC_LIBDIR", self.sysroot_libdir(compiler, self.build.build))
O
Oliver Schneider 已提交
513 514
           .env("CFG_RELEASE_CHANNEL", &self.build.config.channel)
           .env("RUSTDOC_REAL", self.rustdoc(host))
A
Alex Crichton 已提交
515 516
           .env("RUSTDOC_CRATE_VERSION", self.build.rust_version())
           .env("RUSTC_BOOTSTRAP", "1");
O
Oliver Schneider 已提交
517 518 519
        if let Some(linker) = self.build.linker(host) {
            cmd.env("RUSTC_TARGET_LINKER", linker);
        }
M
Mark Simulacrum 已提交
520
        cmd
521 522
    }

523 524 525 526 527 528 529
    /// 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`.
530
    pub fn cargo(&self,
M
Mark Simulacrum 已提交
531 532
             compiler: Compiler,
             mode: Mode,
533
             target: Interned<String>,
M
Mark Simulacrum 已提交
534 535 536
             cmd: &str) -> Command {
        let mut cargo = Command::new(&self.initial_cargo);
        let out_dir = self.stage_out(compiler, mode);
537 538
        cargo.env("CARGO_TARGET_DIR", out_dir)
             .arg(cmd)
539 540
             .arg("--target")
             .arg(target);
541

542 543 544 545 546 547
        // If we were invoked from `make` then that's already got a jobserver
        // set up for us so no need to tell Cargo about jobs all over again.
        if env::var_os("MAKEFLAGS").is_none() && env::var_os("MFLAGS").is_none() {
             cargo.arg("-j").arg(self.jobs().to_string());
        }

548 549
        // FIXME: Temporary fix for https://github.com/rust-lang/cargo/issues/3005
        // Force cargo to output binaries with disambiguating hashes in the name
M
Mark Simulacrum 已提交
550
        cargo.env("__CARGO_DEFAULT_LIB_METADATA", &self.config.channel);
551 552

        let stage;
M
Mark Simulacrum 已提交
553
        if compiler.stage == 0 && self.local_rebuild {
554 555 556 557 558 559
            // Assume the local-rebuild rustc already has stage1 features.
            stage = 1;
        } else {
            stage = compiler.stage;
        }

560 561 562 563 564 565 566 567 568 569 570 571
        let mut extra_args = env::var(&format!("RUSTFLAGS_STAGE_{}", stage)).unwrap_or_default();
        if stage != 0 {
            let s = env::var("RUSTFLAGS_STAGE_NOT_0").unwrap_or_default();
            extra_args.push_str(" ");
            extra_args.push_str(&s);
        }

        if !extra_args.is_empty() {
            cargo.env("RUSTFLAGS",
                format!("{} {}", env::var("RUSTFLAGS").unwrap_or_default(), extra_args));
        }

572 573
        // 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 已提交
574
        // how the actual compiler itself is called.
575 576 577
        //
        // These variables are primarily all read by
        // src/bootstrap/bin/{rustc.rs,rustdoc.rs}
M
Mark Simulacrum 已提交
578 579
        cargo.env("RUSTBUILD_NATIVE_DIR", self.native_dir(target))
             .env("RUSTC", self.out.join("bootstrap/debug/rustc"))
580
             .env("RUSTC_REAL", self.rustc(compiler))
581 582
             .env("RUSTC_STAGE", stage.to_string())
             .env("RUSTC_DEBUG_ASSERTIONS",
M
Mark Simulacrum 已提交
583
                  self.config.rust_debug_assertions.to_string())
584 585
             .env("RUSTC_SYSROOT", self.sysroot(compiler))
             .env("RUSTC_LIBDIR", self.rustc_libdir(compiler))
M
Mark Simulacrum 已提交
586 587
             .env("RUSTC_RPATH", self.config.rust_rpath.to_string())
             .env("RUSTDOC", self.out.join("bootstrap/debug/rustdoc"))
M
Mark Simulacrum 已提交
588
             .env("RUSTDOC_REAL", if cmd == "doc" || cmd == "test" {
589
                 self.rustdoc(compiler.host)
M
Mark Simulacrum 已提交
590 591 592
             } else {
                 PathBuf::from("/path/to/nowhere/rustdoc/not/required")
             })
593 594
             .env("TEST_MIRI", self.config.test_miri.to_string())
             .env("RUSTC_ERROR_METADATA_DST", self.extended_error_dir());
595

O
Oliver Schneider 已提交
596 597 598 599 600 601
        if let Some(host_linker) = self.build.linker(compiler.host) {
            cargo.env("RUSTC_HOST_LINKER", host_linker);
        }
        if let Some(target_linker) = self.build.linker(target) {
            cargo.env("RUSTC_TARGET_LINKER", target_linker);
        }
P
penpalperson 已提交
602 603 604
        if let Some(ref error_format) = self.config.rustc_error_format {
            cargo.env("RUSTC_ERROR_FORMAT", error_format);
        }
605
        if cmd != "build" && cmd != "check" {
606 607
            cargo.env("RUSTDOC_LIBDIR", self.rustc_libdir(self.compiler(2, self.build.build)));
        }
608 609

        if mode != Mode::Tool {
610 611 612 613 614
            // Tools don't get debuginfo right now, e.g. cargo and rls don't
            // get compiled with debuginfo.
            // Adding debuginfo increases their sizes by a factor of 3-4.
            cargo.env("RUSTC_DEBUGINFO", self.config.rust_debuginfo.to_string());
            cargo.env("RUSTC_DEBUGINFO_LINES", self.config.rust_debuginfo_lines.to_string());
O
Oliver Schneider 已提交
615
            cargo.env("RUSTC_FORCE_UNSTABLE", "1");
616 617 618

            // Currently the compiler depends on crates from crates.io, and
            // then other crates can depend on the compiler (e.g. proc-macro
M
Mark Simulacrum 已提交
619
            // crates). Let's say, for example that rustc itself depends on the
620 621
            // bitflags crate. If an external crate then depends on the
            // bitflags crate as well, we need to make sure they don't
B
Bastien Orivel 已提交
622
            // conflict, even if they pick the same version of bitflags. We'll
623 624 625 626 627 628 629 630 631 632 633 634 635
            // want to make sure that e.g. a plugin and rustc each get their
            // own copy of bitflags.

            // Cargo ensures that this works in general through the -C metadata
            // flag. This flag will frob the symbols in the binary to make sure
            // they're different, even though the source code is the exact
            // same. To solve this problem for the compiler we extend Cargo's
            // already-passed -C metadata flag with our own. Our rustc.rs
            // wrapper around the actual rustc will detect -C metadata being
            // passed and frob it with this extra string we're passing in.
            cargo.env("RUSTC_METADATA_SUFFIX", "rustc");
        }

636 637 638 639
        if let Some(x) = self.crt_static(target) {
            cargo.env("RUSTC_CRT_STATIC", x.to_string());
        }

640 641
        // Enable usage of unstable features
        cargo.env("RUSTC_BOOTSTRAP", "1");
M
Mark Simulacrum 已提交
642
        self.add_rust_test_threads(&mut cargo);
643 644 645

        // 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 已提交
646
        // build script, however, it itself needs a standard library! This
647
        // introduces a bit of a pickle when we're compiling the standard
M
Mark Simulacrum 已提交
648
        // library itself.
649 650
        //
        // To work around this we actually end up using the snapshot compiler
M
Mark Simulacrum 已提交
651
        // (stage0) for compiling build scripts of the standard library itself.
652 653 654 655 656
        // 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.
657 658
        //
        // If LLVM support is disabled we need to use the snapshot compiler to compile
659
        // build scripts, as the new compiler doesn't support executables.
660
        if mode == Mode::Libstd || !self.build.config.llvm_enabled {
M
Mark Simulacrum 已提交
661 662
            cargo.env("RUSTC_SNAPSHOT", &self.initial_rustc)
                 .env("RUSTC_SNAPSHOT_LIBDIR", self.rustc_snapshot_libdir());
663
        } else {
664
            cargo.env("RUSTC_SNAPSHOT", self.rustc(compiler))
665 666 667 668 669 670
                 .env("RUSTC_SNAPSHOT_LIBDIR", self.rustc_libdir(compiler));
        }

        // Ignore incremental modes except for stage0, since we're
        // not guaranteeing correctness across builds if the compiler
        // is changing under your feet.`
M
Mark Simulacrum 已提交
671
        if self.config.incremental && compiler.stage == 0 {
672
            cargo.env("CARGO_INCREMENTAL", "1");
673 674
        }

M
Mark Simulacrum 已提交
675
        if let Some(ref on_fail) = self.config.on_fail {
676 677 678
            cargo.env("RUSTC_ON_FAIL", on_fail);
        }

M
Mark Simulacrum 已提交
679
        cargo.env("RUSTC_VERBOSE", format!("{}", self.verbosity));
680

O
Oliver Schneider 已提交
681 682 683 684 685
        // 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.
686 687 688
        //
        // FIXME: the guard against msvc shouldn't need to be here
        if !target.contains("msvc") {
689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707
            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));
            cargo.env(format!("CC_{}", target), &cc)
                 .env("CC", &cc);
O
Oliver Schneider 已提交
708 709 710 711 712 713 714 715 716 717 718 719

            let cflags = self.cflags(target).join(" ");
            cargo.env(format!("CFLAGS_{}", target), cflags.clone())
                 .env("CFLAGS", cflags.clone());

            if let Some(ar) = self.ar(target) {
                let ranlib = format!("{} s", ar.display());
                cargo.env(format!("AR_{}", target), ar)
                     .env("AR", ar)
                     .env(format!("RANLIB_{}", target), ranlib.clone())
                     .env("RANLIB", ranlib);
            }
720

M
Mark Simulacrum 已提交
721
            if let Ok(cxx) = self.cxx(target) {
722 723 724
                let cxx = ccacheify(&cxx);
                cargo.env(format!("CXX_{}", target), &cxx)
                     .env("CXX", &cxx)
O
Oliver Schneider 已提交
725 726
                     .env(format!("CXXFLAGS_{}", target), cflags.clone())
                     .env("CXXFLAGS", cflags);
727 728 729
            }
        }

M
Mark Simulacrum 已提交
730
        if mode == Mode::Libstd && self.config.extended && compiler.is_final_stage(self) {
731 732 733
            cargo.env("RUSTC_SAVE_ANALYSIS", "api".to_string());
        }

O
Oliver Schneider 已提交
734 735 736
        // For `cargo doc` invocations, make rustdoc print the Rust version into the docs
        cargo.env("RUSTDOC_CRATE_VERSION", self.build.rust_version());

737 738 739 740 741
        // Environment variables *required* throughout the build
        //
        // FIXME: should update code to not require this env var
        cargo.env("CFG_COMPILER_HOST_TRIPLE", target);

742 743 744
        // Set this for all builds to make sure doc builds also get it.
        cargo.env("CFG_RELEASE_CHANNEL", &self.build.config.channel);

745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777
        // 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
        // the binaries we ship of things like rustc_trans (aka the rustc_trans
        // 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.
        cargo.env("WINAPI_NO_BUNDLED_LIBRARIES", "1");

O
Oliver Schneider 已提交
778
        if self.is_very_verbose() {
779 780
            cargo.arg("-v");
        }
781 782 783 784 785 786 787

        // This must be kept before the thinlto check, as we set codegen units
        // to 1 forcibly there.
        if let Some(n) = self.config.rust_codegen_units {
            cargo.env("RUSTC_CODEGEN_UNITS", n.to_string());
        }

O
Oliver Schneider 已提交
788 789 790 791 792 793 794
        if self.config.rust_optimize {
            // FIXME: cargo bench does not accept `--release`
            if cmd != "bench" {
                cargo.arg("--release");
            }

            if self.config.rust_codegen_units.is_none() &&
795 796
               self.build.is_rust_llvm(compiler.host) &&
               self.config.rust_thinlto {
O
Oliver Schneider 已提交
797
                cargo.env("RUSTC_THINLTO", "1");
798 799 800 801 802
            } else if self.config.rust_codegen_units.is_none() {
                // Generally, if ThinLTO has been disabled for some reason, we
                // want to set the codegen units to 1. However, we shouldn't do
                // this if the option was specifically set by the user.
                cargo.env("RUSTC_CODEGEN_UNITS", "1");
O
Oliver Schneider 已提交
803
            }
804
        }
805

M
Mark Simulacrum 已提交
806
        if self.config.locked_deps {
807 808
            cargo.arg("--locked");
        }
M
Mark Simulacrum 已提交
809
        if self.config.vendor || self.is_sudo {
810 811 812
            cargo.arg("--frozen");
        }

M
Mark Simulacrum 已提交
813
        self.ci_env.force_coloring_in_ci(&mut cargo);
814 815 816 817

        cargo
    }

818 819 820
    /// Ensure that a given step is built, returning it's output. This will
    /// cache the step, so it is safe (and good!) to call this as often as
    /// needed to ensure that all dependencies are built.
821
    pub fn ensure<S: Step>(&'a self, step: S) -> S::Output {
822 823
        {
            let mut stack = self.stack.borrow_mut();
824 825 826 827
            for stack_step in stack.iter() {
                // should skip
                if stack_step.downcast_ref::<S>().map_or(true, |stack_step| *stack_step != step) {
                    continue;
828
                }
829
                let mut out = String::new();
830
                out += &format!("\n\nCycle in build detected when adding {:?}\n", step);
831 832 833 834 835
                for el in stack.iter().rev() {
                    out += &format!("\t{:?}\n", el);
                }
                panic!(out);
            }
836 837
            if let Some(out) = self.cache.get(&step) {
                self.build.verbose(&format!("{}c {:?}", "  ".repeat(stack.len()), step));
838 839 840

                return out;
            }
841
            self.build.verbose(&format!("{}> {:?}", "  ".repeat(stack.len()), step));
842
            stack.push(Box::new(step.clone()));
843
        }
844
        let out = step.clone().run(self);
845 846
        {
            let mut stack = self.stack.borrow_mut();
847 848
            let cur_step = stack.pop().expect("step stack empty");
            assert_eq!(cur_step.downcast_ref(), Some(&step));
849
        }
850 851 852
        self.build.verbose(&format!("{}< {:?}", "  ".repeat(self.stack.borrow().len()), step));
        self.cache.put(step, out.clone());
        out
853 854
    }
}