]> git.lizzy.rs Git - rust.git/blob - src/bootstrap/step.rs
Auto merge of #41433 - estebank:constructor, r=michaelwoerister
[rust.git] / src / bootstrap / step.rs
1 // Copyright 2016 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! Definition of steps of the build system.
12 //!
13 //! This is where some of the real meat of rustbuild is located, in how we
14 //! define targets and the dependencies amongst them. This file can sort of be
15 //! viewed as just defining targets in a makefile which shell out to predefined
16 //! functions elsewhere about how to execute the target.
17 //!
18 //! The primary function here you're likely interested in is the `build_rules`
19 //! function. This will create a `Rules` structure which basically just lists
20 //! everything that rustbuild can do. Each rule has a human-readable name, a
21 //! path associated with it, some dependencies, and then a closure of how to
22 //! actually perform the rule.
23 //!
24 //! All steps below are defined in self-contained units, so adding a new target
25 //! to the build system should just involve adding the meta information here
26 //! along with the actual implementation elsewhere. You can find more comments
27 //! about how to define rules themselves below.
28
29 use std::collections::{BTreeMap, HashSet, HashMap};
30 use std::mem;
31
32 use check::{self, TestKind};
33 use compile;
34 use dist;
35 use doc;
36 use flags::Subcommand;
37 use install;
38 use native;
39 use {Compiler, Build, Mode};
40
41 pub fn run(build: &Build) {
42     let rules = build_rules(build);
43     let steps = rules.plan();
44     rules.run(&steps);
45 }
46
47 pub fn build_rules<'a>(build: &'a Build) -> Rules {
48     let mut rules = Rules::new(build);
49
50     // This is the first rule that we're going to define for rustbuild, which is
51     // used to compile LLVM itself. All rules are added through the `rules`
52     // structure created above and are configured through a builder-style
53     // interface.
54     //
55     // First up we see the `build` method. This represents a rule that's part of
56     // the top-level `build` subcommand. For example `./x.py build` is what this
57     // is associating with. Note that this is normally only relevant if you flag
58     // a rule as `default`, which we'll talk about later.
59     //
60     // Next up we'll see two arguments to this method:
61     //
62     // * `llvm` - this is the "human readable" name of this target. This name is
63     //            not accessed anywhere outside this file itself (e.g. not in
64     //            the CLI nor elsewhere in rustbuild). The purpose of this is to
65     //            easily define dependencies between rules. That is, other rules
66     //            will depend on this with the name "llvm".
67     // * `src/llvm` - this is the relevant path to the rule that we're working
68     //                with. This path is the engine behind how commands like
69     //                `./x.py build src/llvm` work. This should typically point
70     //                to the relevant component, but if there's not really a
71     //                path to be assigned here you can pass something like
72     //                `path/to/nowhere` to ignore it.
73     //
74     // After we create the rule with the `build` method we can then configure
75     // various aspects of it. For example this LLVM rule uses `.host(true)` to
76     // flag that it's a rule only for host targets. In other words, LLVM isn't
77     // compiled for targets configured through `--target` (e.g. those we're just
78     // building a standard library for).
79     //
80     // Next up the `dep` method will add a dependency to this rule. The closure
81     // is yielded the step that represents executing the `llvm` rule itself
82     // (containing information like stage, host, target, ...) and then it must
83     // return a target that the step depends on. Here LLVM is actually
84     // interesting where a cross-compiled LLVM depends on the host LLVM, but
85     // otherwise it has no dependencies.
86     //
87     // To handle this we do a bit of dynamic dispatch to see what the dependency
88     // is. If we're building a LLVM for the build triple, then we don't actually
89     // have any dependencies! To do that we return a dependency on the `Step::noop()`
90     // target which does nothing.
91     //
92     // If we're build a cross-compiled LLVM, however, we need to assemble the
93     // libraries from the previous compiler. This step has the same name as
94     // ours (llvm) but we want it for a different target, so we use the
95     // builder-style methods on `Step` to configure this target to the build
96     // triple.
97     //
98     // Finally, to finish off this rule, we define how to actually execute it.
99     // That logic is all defined in the `native` module so we just delegate to
100     // the relevant function there. The argument to the closure passed to `run`
101     // is a `Step` (defined below) which encapsulates information like the
102     // stage, target, host, etc.
103     rules.build("llvm", "src/llvm")
104          .host(true)
105          .dep(move |s| {
106              if s.target == build.config.build {
107                  Step::noop()
108              } else {
109                  s.target(&build.config.build)
110              }
111          })
112          .run(move |s| native::llvm(build, s.target));
113
114     // Ok! After that example rule  that's hopefully enough to explain what's
115     // going on here. You can check out the API docs below and also see a bunch
116     // more examples of rules directly below as well.
117
118     // the compiler with no target libraries ready to go
119     rules.build("rustc", "src/rustc")
120          .dep(|s| s.name("create-sysroot").target(s.host))
121          .dep(move |s| {
122              if s.stage == 0 {
123                  Step::noop()
124              } else {
125                  s.name("librustc")
126                   .host(&build.config.build)
127                   .stage(s.stage - 1)
128              }
129          })
130          .run(move |s| compile::assemble_rustc(build, s.stage, s.target));
131
132     // Helper for loading an entire DAG of crates, rooted at `name`
133     let krates = |name: &str| {
134         let mut ret = Vec::new();
135         let mut list = vec![name];
136         let mut visited = HashSet::new();
137         while let Some(krate) = list.pop() {
138             let default = krate == name;
139             let krate = &build.crates[krate];
140             let path = krate.path.strip_prefix(&build.src)
141                 // This handles out of tree paths
142                 .unwrap_or(&krate.path);
143             ret.push((krate, path.to_str().unwrap(), default));
144             for dep in krate.deps.iter() {
145                 if visited.insert(dep) && dep != "build_helper" {
146                     list.push(dep);
147                 }
148             }
149         }
150         return ret
151     };
152
153     // ========================================================================
154     // Crate compilations
155     //
156     // Tools used during the build system but not shipped
157     rules.build("create-sysroot", "path/to/nowhere")
158          .run(move |s| compile::create_sysroot(build, &s.compiler()));
159
160     // These rules are "pseudo rules" that don't actually do any work
161     // themselves, but represent a complete sysroot with the relevant compiler
162     // linked into place.
163     //
164     // That is, depending on "libstd" means that when the rule is completed then
165     // the `stage` sysroot for the compiler `host` will be available with a
166     // standard library built for `target` linked in place. Not all rules need
167     // the compiler itself to be available, just the standard library, so
168     // there's a distinction between the two.
169     rules.build("libstd", "src/libstd")
170          .dep(|s| s.name("rustc").target(s.host))
171          .dep(|s| s.name("libstd-link"));
172     rules.build("libtest", "src/libtest")
173          .dep(|s| s.name("libstd"))
174          .dep(|s| s.name("libtest-link"))
175          .default(true);
176     rules.build("librustc", "src/librustc")
177          .dep(|s| s.name("libtest"))
178          .dep(|s| s.name("librustc-link"))
179          .host(true)
180          .default(true);
181
182     // Helper method to define the rules to link a crate into its place in the
183     // sysroot.
184     //
185     // The logic here is a little subtle as there's a few cases to consider.
186     // Not all combinations of (stage, host, target) actually require something
187     // to be compiled, but rather libraries could get propagated from a
188     // different location. For example:
189     //
190     // * Any crate with a `host` that's not the build triple will not actually
191     //   compile something. A different `host` means that the build triple will
192     //   actually compile the libraries, and then we'll copy them over from the
193     //   build triple to the `host` directory.
194     //
195     // * Some crates aren't even compiled by the build triple, but may be copied
196     //   from previous stages. For example if we're not doing a full bootstrap
197     //   then we may just depend on the stage1 versions of libraries to be
198     //   available to get linked forward.
199     //
200     // * Finally, there are some cases, however, which do indeed comiple crates
201     //   and link them into place afterwards.
202     //
203     // The rule definition below mirrors these three cases. The `dep` method
204     // calculates the correct dependency which either comes from stage1, a
205     // different compiler, or from actually building the crate itself (the `dep`
206     // rule). The `run` rule then mirrors these three cases and links the cases
207     // forward into the compiler sysroot specified from the correct location.
208     fn crate_rule<'a, 'b>(build: &'a Build,
209                           rules: &'b mut Rules<'a>,
210                           krate: &'a str,
211                           dep: &'a str,
212                           link: fn(&Build, &Compiler, &Compiler, &str))
213                           -> RuleBuilder<'a, 'b> {
214         let mut rule = rules.build(&krate, "path/to/nowhere");
215         rule.dep(move |s| {
216                 if build.force_use_stage1(&s.compiler(), s.target) {
217                     s.host(&build.config.build).stage(1)
218                 } else if s.host == build.config.build {
219                     s.name(dep)
220                 } else {
221                     s.host(&build.config.build)
222                 }
223             })
224             .run(move |s| {
225                 if build.force_use_stage1(&s.compiler(), s.target) {
226                     link(build,
227                          &s.stage(1).host(&build.config.build).compiler(),
228                          &s.compiler(),
229                          s.target)
230                 } else if s.host == build.config.build {
231                     link(build, &s.compiler(), &s.compiler(), s.target)
232                 } else {
233                     link(build,
234                          &s.host(&build.config.build).compiler(),
235                          &s.compiler(),
236                          s.target)
237                 }
238             });
239             return rule
240     }
241
242     // Similar to the `libstd`, `libtest`, and `librustc` rules above, except
243     // these rules only represent the libraries being available in the sysroot,
244     // not the compiler itself. This is done as not all rules need a compiler in
245     // the sysroot, but may just need the libraries.
246     //
247     // All of these rules use the helper definition above.
248     crate_rule(build,
249                &mut rules,
250                "libstd-link",
251                "build-crate-std",
252                compile::std_link)
253         .dep(|s| s.name("startup-objects"))
254         .dep(|s| s.name("create-sysroot").target(s.host));
255     crate_rule(build,
256                &mut rules,
257                "libtest-link",
258                "build-crate-test",
259                compile::test_link)
260         .dep(|s| s.name("libstd-link"));
261     crate_rule(build,
262                &mut rules,
263                "librustc-link",
264                "build-crate-rustc-main",
265                compile::rustc_link)
266         .dep(|s| s.name("libtest-link"));
267
268     for (krate, path, _default) in krates("std") {
269         rules.build(&krate.build_step, path)
270              .dep(|s| s.name("startup-objects"))
271              .dep(move |s| s.name("rustc").host(&build.config.build).target(s.host))
272              .run(move |s| compile::std(build, s.target, &s.compiler()));
273     }
274     for (krate, path, _default) in krates("test") {
275         rules.build(&krate.build_step, path)
276              .dep(|s| s.name("libstd-link"))
277              .run(move |s| compile::test(build, s.target, &s.compiler()));
278     }
279     for (krate, path, _default) in krates("rustc-main") {
280         rules.build(&krate.build_step, path)
281              .dep(|s| s.name("libtest-link"))
282              .dep(move |s| s.name("llvm").host(&build.config.build).stage(0))
283              .dep(|s| s.name("may-run-build-script"))
284              .run(move |s| compile::rustc(build, s.target, &s.compiler()));
285     }
286
287     // Crates which have build scripts need to rely on this rule to ensure that
288     // the necessary prerequisites for a build script are linked and located in
289     // place.
290     rules.build("may-run-build-script", "path/to/nowhere")
291          .dep(move |s| {
292              s.name("libstd-link")
293               .host(&build.config.build)
294               .target(&build.config.build)
295          });
296     rules.build("startup-objects", "src/rtstartup")
297          .dep(|s| s.name("create-sysroot").target(s.host))
298          .run(move |s| compile::build_startup_objects(build, &s.compiler(), s.target));
299
300     // ========================================================================
301     // Test targets
302     //
303     // Various unit tests and tests suites we can run
304     {
305         let mut suite = |name, path, mode, dir| {
306             rules.test(name, path)
307                  .dep(|s| s.name("libtest"))
308                  .dep(|s| s.name("tool-compiletest").target(s.host).stage(0))
309                  .dep(|s| s.name("test-helpers"))
310                  .dep(|s| s.name("emulator-copy-libs"))
311                  .default(mode != "pretty") // pretty tests don't run everywhere
312                  .run(move |s| {
313                      check::compiletest(build, &s.compiler(), s.target, mode, dir)
314                  });
315         };
316
317         suite("check-ui", "src/test/ui", "ui", "ui");
318         suite("check-rpass", "src/test/run-pass", "run-pass", "run-pass");
319         suite("check-cfail", "src/test/compile-fail", "compile-fail", "compile-fail");
320         suite("check-pfail", "src/test/parse-fail", "parse-fail", "parse-fail");
321         suite("check-rfail", "src/test/run-fail", "run-fail", "run-fail");
322         suite("check-rpass-valgrind", "src/test/run-pass-valgrind",
323               "run-pass-valgrind", "run-pass-valgrind");
324         suite("check-mir-opt", "src/test/mir-opt", "mir-opt", "mir-opt");
325         if build.config.codegen_tests {
326             suite("check-codegen", "src/test/codegen", "codegen", "codegen");
327         }
328         suite("check-codegen-units", "src/test/codegen-units", "codegen-units",
329               "codegen-units");
330         suite("check-incremental", "src/test/incremental", "incremental",
331               "incremental");
332     }
333
334     if build.config.build.contains("msvc") {
335         // nothing to do for debuginfo tests
336     } else {
337         rules.test("check-debuginfo-lldb", "src/test/debuginfo-lldb")
338              .dep(|s| s.name("libtest"))
339              .dep(|s| s.name("tool-compiletest").target(s.host).stage(0))
340              .dep(|s| s.name("test-helpers"))
341              .dep(|s| s.name("debugger-scripts"))
342              .run(move |s| check::compiletest(build, &s.compiler(), s.target,
343                                          "debuginfo-lldb", "debuginfo"));
344         rules.test("check-debuginfo-gdb", "src/test/debuginfo-gdb")
345              .dep(|s| s.name("libtest"))
346              .dep(|s| s.name("tool-compiletest").target(s.host).stage(0))
347              .dep(|s| s.name("test-helpers"))
348              .dep(|s| s.name("debugger-scripts"))
349              .dep(|s| s.name("emulator-copy-libs"))
350              .run(move |s| check::compiletest(build, &s.compiler(), s.target,
351                                          "debuginfo-gdb", "debuginfo"));
352         let mut rule = rules.test("check-debuginfo", "src/test/debuginfo");
353         rule.default(true);
354         if build.config.build.contains("apple") {
355             rule.dep(|s| s.name("check-debuginfo-lldb"));
356         } else {
357             rule.dep(|s| s.name("check-debuginfo-gdb"));
358         }
359     }
360
361     rules.test("debugger-scripts", "src/etc/lldb_batchmode.py")
362          .run(move |s| dist::debugger_scripts(build, &build.sysroot(&s.compiler()),
363                                          s.target));
364
365     {
366         let mut suite = |name, path, mode, dir| {
367             rules.test(name, path)
368                  .dep(|s| s.name("librustc"))
369                  .dep(|s| s.name("test-helpers"))
370                  .dep(|s| s.name("tool-compiletest").target(s.host).stage(0))
371                  .default(mode != "pretty")
372                  .host(true)
373                  .run(move |s| {
374                      check::compiletest(build, &s.compiler(), s.target, mode, dir)
375                  });
376         };
377
378         suite("check-ui-full", "src/test/ui-fulldeps", "ui", "ui-fulldeps");
379         suite("check-rpass-full", "src/test/run-pass-fulldeps",
380               "run-pass", "run-pass-fulldeps");
381         suite("check-rfail-full", "src/test/run-fail-fulldeps",
382               "run-fail", "run-fail-fulldeps");
383         suite("check-cfail-full", "src/test/compile-fail-fulldeps",
384               "compile-fail", "compile-fail-fulldeps");
385         suite("check-rmake", "src/test/run-make", "run-make", "run-make");
386         suite("check-rustdoc", "src/test/rustdoc", "rustdoc", "rustdoc");
387         suite("check-pretty", "src/test/pretty", "pretty", "pretty");
388         suite("check-pretty-rpass", "src/test/run-pass/pretty", "pretty",
389               "run-pass");
390         suite("check-pretty-rfail", "src/test/run-fail/pretty", "pretty",
391               "run-fail");
392         suite("check-pretty-valgrind", "src/test/run-pass-valgrind/pretty", "pretty",
393               "run-pass-valgrind");
394         suite("check-pretty-rpass-full", "src/test/run-pass-fulldeps/pretty",
395               "pretty", "run-pass-fulldeps");
396         suite("check-pretty-rfail-full", "src/test/run-fail-fulldeps/pretty",
397               "pretty", "run-fail-fulldeps");
398     }
399
400     for (krate, path, _default) in krates("std") {
401         rules.test(&krate.test_step, path)
402              .dep(|s| s.name("libtest"))
403              .dep(|s| s.name("emulator-copy-libs"))
404              .run(move |s| check::krate(build, &s.compiler(), s.target,
405                                         Mode::Libstd, TestKind::Test,
406                                         Some(&krate.name)));
407     }
408     rules.test("check-std-all", "path/to/nowhere")
409          .dep(|s| s.name("libtest"))
410          .dep(|s| s.name("emulator-copy-libs"))
411          .default(true)
412          .run(move |s| check::krate(build, &s.compiler(), s.target,
413                                     Mode::Libstd, TestKind::Test, None));
414
415     // std benchmarks
416     for (krate, path, _default) in krates("std") {
417         rules.bench(&krate.bench_step, path)
418              .dep(|s| s.name("libtest"))
419              .dep(|s| s.name("emulator-copy-libs"))
420              .run(move |s| check::krate(build, &s.compiler(), s.target,
421                                         Mode::Libstd, TestKind::Bench,
422                                         Some(&krate.name)));
423     }
424     rules.bench("bench-std-all", "path/to/nowhere")
425          .dep(|s| s.name("libtest"))
426          .dep(|s| s.name("emulator-copy-libs"))
427          .default(true)
428          .run(move |s| check::krate(build, &s.compiler(), s.target,
429                                     Mode::Libstd, TestKind::Bench, None));
430
431     for (krate, path, _default) in krates("test") {
432         rules.test(&krate.test_step, path)
433              .dep(|s| s.name("libtest"))
434              .dep(|s| s.name("emulator-copy-libs"))
435              .run(move |s| check::krate(build, &s.compiler(), s.target,
436                                         Mode::Libtest, TestKind::Test,
437                                         Some(&krate.name)));
438     }
439     rules.test("check-test-all", "path/to/nowhere")
440          .dep(|s| s.name("libtest"))
441          .dep(|s| s.name("emulator-copy-libs"))
442          .default(true)
443          .run(move |s| check::krate(build, &s.compiler(), s.target,
444                                     Mode::Libtest, TestKind::Test, None));
445     for (krate, path, _default) in krates("rustc-main") {
446         rules.test(&krate.test_step, path)
447              .dep(|s| s.name("librustc"))
448              .dep(|s| s.name("emulator-copy-libs"))
449              .host(true)
450              .run(move |s| check::krate(build, &s.compiler(), s.target,
451                                         Mode::Librustc, TestKind::Test,
452                                         Some(&krate.name)));
453     }
454     rules.test("check-rustc-all", "path/to/nowhere")
455          .dep(|s| s.name("librustc"))
456          .dep(|s| s.name("emulator-copy-libs"))
457          .default(true)
458          .host(true)
459          .run(move |s| check::krate(build, &s.compiler(), s.target,
460                                     Mode::Librustc, TestKind::Test, None));
461
462     rules.test("check-linkchecker", "src/tools/linkchecker")
463          .dep(|s| s.name("tool-linkchecker").stage(0))
464          .dep(|s| s.name("default:doc"))
465          .default(true)
466          .host(true)
467          .run(move |s| check::linkcheck(build, s.target));
468     rules.test("check-cargotest", "src/tools/cargotest")
469          .dep(|s| s.name("tool-cargotest").stage(0))
470          .dep(|s| s.name("librustc"))
471          .host(true)
472          .run(move |s| check::cargotest(build, s.stage, s.target));
473     rules.test("check-cargo", "cargo")
474          .dep(|s| s.name("tool-cargo"))
475          .host(true)
476          .run(move |s| check::cargo(build, s.stage, s.target));
477     rules.test("check-tidy", "src/tools/tidy")
478          .dep(|s| s.name("tool-tidy").stage(0))
479          .default(true)
480          .host(true)
481          .only_build(true)
482          .run(move |s| check::tidy(build, s.target));
483     rules.test("check-error-index", "src/tools/error_index_generator")
484          .dep(|s| s.name("libstd"))
485          .dep(|s| s.name("tool-error-index").host(s.host).stage(0))
486          .default(true)
487          .host(true)
488          .run(move |s| check::error_index(build, &s.compiler()));
489     rules.test("check-docs", "src/doc")
490          .dep(|s| s.name("libtest"))
491          .default(true)
492          .host(true)
493          .run(move |s| check::docs(build, &s.compiler()));
494     rules.test("check-distcheck", "distcheck")
495          .dep(|s| s.name("dist-src"))
496          .run(move |_| check::distcheck(build));
497
498     rules.build("test-helpers", "src/rt/rust_test_helpers.c")
499          .run(move |s| native::test_helpers(build, s.target));
500     rules.build("openssl", "path/to/nowhere")
501          .run(move |s| native::openssl(build, s.target));
502
503     // Some test suites are run inside emulators, and most of our test binaries
504     // are linked dynamically which means we need to ship the standard library
505     // and such to the emulator ahead of time. This step represents this and is
506     // a dependency of all test suites.
507     //
508     // Most of the time this step is a noop (the `check::emulator_copy_libs`
509     // only does work if necessary). For some steps such as shipping data to
510     // QEMU we have to build our own tools so we've got conditional dependencies
511     // on those programs as well. Note that the QEMU client is built for the
512     // build target (us) and the server is built for the target.
513     rules.test("emulator-copy-libs", "path/to/nowhere")
514          .dep(|s| s.name("libtest"))
515          .dep(move |s| {
516              if build.qemu_rootfs(s.target).is_some() {
517                 s.name("tool-qemu-test-client").target(s.host).stage(0)
518              } else {
519                  Step::noop()
520              }
521          })
522          .dep(move |s| {
523              if build.qemu_rootfs(s.target).is_some() {
524                 s.name("tool-qemu-test-server")
525              } else {
526                  Step::noop()
527              }
528          })
529          .run(move |s| check::emulator_copy_libs(build, &s.compiler(), s.target));
530
531     rules.test("check-bootstrap", "src/bootstrap")
532          .default(true)
533          .host(true)
534          .only_build(true)
535          .run(move |_| check::bootstrap(build));
536
537     // ========================================================================
538     // Build tools
539     //
540     // Tools used during the build system but not shipped
541     rules.build("tool-rustbook", "src/tools/rustbook")
542          .dep(|s| s.name("maybe-clean-tools"))
543          .dep(|s| s.name("librustc-tool"))
544          .run(move |s| compile::tool(build, s.stage, s.target, "rustbook"));
545     rules.build("tool-error-index", "src/tools/error_index_generator")
546          .dep(|s| s.name("maybe-clean-tools"))
547          .dep(|s| s.name("librustc-tool"))
548          .run(move |s| compile::tool(build, s.stage, s.target, "error_index_generator"));
549     rules.build("tool-tidy", "src/tools/tidy")
550          .dep(|s| s.name("maybe-clean-tools"))
551          .dep(|s| s.name("libstd-tool"))
552          .run(move |s| compile::tool(build, s.stage, s.target, "tidy"));
553     rules.build("tool-linkchecker", "src/tools/linkchecker")
554          .dep(|s| s.name("maybe-clean-tools"))
555          .dep(|s| s.name("libstd-tool"))
556          .run(move |s| compile::tool(build, s.stage, s.target, "linkchecker"));
557     rules.build("tool-cargotest", "src/tools/cargotest")
558          .dep(|s| s.name("maybe-clean-tools"))
559          .dep(|s| s.name("libstd-tool"))
560          .run(move |s| compile::tool(build, s.stage, s.target, "cargotest"));
561     rules.build("tool-compiletest", "src/tools/compiletest")
562          .dep(|s| s.name("maybe-clean-tools"))
563          .dep(|s| s.name("libtest-tool"))
564          .run(move |s| compile::tool(build, s.stage, s.target, "compiletest"));
565     rules.build("tool-build-manifest", "src/tools/build-manifest")
566          .dep(|s| s.name("maybe-clean-tools"))
567          .dep(|s| s.name("libstd-tool"))
568          .run(move |s| compile::tool(build, s.stage, s.target, "build-manifest"));
569     rules.build("tool-qemu-test-server", "src/tools/qemu-test-server")
570          .dep(|s| s.name("maybe-clean-tools"))
571          .dep(|s| s.name("libstd-tool"))
572          .run(move |s| compile::tool(build, s.stage, s.target, "qemu-test-server"));
573     rules.build("tool-qemu-test-client", "src/tools/qemu-test-client")
574          .dep(|s| s.name("maybe-clean-tools"))
575          .dep(|s| s.name("libstd-tool"))
576          .run(move |s| compile::tool(build, s.stage, s.target, "qemu-test-client"));
577     rules.build("tool-cargo", "cargo")
578          .dep(|s| s.name("maybe-clean-tools"))
579          .dep(|s| s.name("libstd-tool"))
580          .dep(|s| s.stage(0).host(s.target).name("openssl"))
581          .dep(move |s| {
582              // Cargo depends on procedural macros, which requires a full host
583              // compiler to be available, so we need to depend on that.
584              s.name("librustc-link")
585               .target(&build.config.build)
586               .host(&build.config.build)
587          })
588          .run(move |s| compile::tool(build, s.stage, s.target, "cargo"));
589     rules.build("tool-rls", "rls")
590          .host(true)
591          .dep(|s| s.name("librustc-tool"))
592          .dep(|s| s.stage(0).host(s.target).name("openssl"))
593          .dep(move |s| {
594              // rls, like cargo, uses procedural macros
595              s.name("librustc-link")
596               .target(&build.config.build)
597               .host(&build.config.build)
598          })
599          .run(move |s| compile::tool(build, s.stage, s.target, "rls"));
600
601     // "pseudo rule" which represents completely cleaning out the tools dir in
602     // one stage. This needs to happen whenever a dependency changes (e.g.
603     // libstd, libtest, librustc) and all of the tool compilations above will
604     // be sequenced after this rule.
605     rules.build("maybe-clean-tools", "path/to/nowhere")
606          .after("librustc-tool")
607          .after("libtest-tool")
608          .after("libstd-tool");
609
610     rules.build("librustc-tool", "path/to/nowhere")
611          .dep(|s| s.name("librustc"))
612          .run(move |s| compile::maybe_clean_tools(build, s.stage, s.target, Mode::Librustc));
613     rules.build("libtest-tool", "path/to/nowhere")
614          .dep(|s| s.name("libtest"))
615          .run(move |s| compile::maybe_clean_tools(build, s.stage, s.target, Mode::Libtest));
616     rules.build("libstd-tool", "path/to/nowhere")
617          .dep(|s| s.name("libstd"))
618          .run(move |s| compile::maybe_clean_tools(build, s.stage, s.target, Mode::Libstd));
619
620     // ========================================================================
621     // Documentation targets
622     rules.doc("doc-book", "src/doc/book")
623          .dep(move |s| {
624              s.name("tool-rustbook")
625               .host(&build.config.build)
626               .target(&build.config.build)
627               .stage(0)
628          })
629          .default(build.config.docs)
630          .run(move |s| doc::book(build, s.target, "book"));
631     rules.doc("doc-nomicon", "src/doc/nomicon")
632          .dep(move |s| {
633              s.name("tool-rustbook")
634               .host(&build.config.build)
635               .target(&build.config.build)
636               .stage(0)
637          })
638          .default(build.config.docs)
639          .run(move |s| doc::rustbook(build, s.target, "nomicon"));
640     rules.doc("doc-reference", "src/doc/reference")
641          .dep(move |s| {
642              s.name("tool-rustbook")
643               .host(&build.config.build)
644               .target(&build.config.build)
645               .stage(0)
646          })
647          .default(build.config.docs)
648          .run(move |s| doc::rustbook(build, s.target, "reference"));
649     rules.doc("doc-unstable-book", "src/doc/unstable-book")
650          .dep(move |s| {
651              s.name("tool-rustbook")
652               .host(&build.config.build)
653               .target(&build.config.build)
654               .stage(0)
655          })
656          .default(build.config.docs)
657          .run(move |s| doc::rustbook(build, s.target, "unstable-book"));
658     rules.doc("doc-standalone", "src/doc")
659          .dep(move |s| {
660              s.name("rustc")
661               .host(&build.config.build)
662               .target(&build.config.build)
663               .stage(0)
664          })
665          .default(build.config.docs)
666          .run(move |s| doc::standalone(build, s.target));
667     rules.doc("doc-error-index", "src/tools/error_index_generator")
668          .dep(move |s| s.name("tool-error-index").target(&build.config.build).stage(0))
669          .dep(move |s| s.name("librustc-link"))
670          .default(build.config.docs)
671          .host(true)
672          .run(move |s| doc::error_index(build, s.target));
673     for (krate, path, default) in krates("std") {
674         rules.doc(&krate.doc_step, path)
675              .dep(|s| s.name("libstd-link"))
676              .default(default && build.config.docs)
677              .run(move |s| doc::std(build, s.stage, s.target));
678     }
679     for (krate, path, default) in krates("test") {
680         rules.doc(&krate.doc_step, path)
681              .dep(|s| s.name("libtest-link"))
682              // Needed so rustdoc generates relative links to std.
683              .dep(|s| s.name("doc-crate-std"))
684              .default(default && build.config.compiler_docs)
685              .run(move |s| doc::test(build, s.stage, s.target));
686     }
687     for (krate, path, default) in krates("rustc-main") {
688         rules.doc(&krate.doc_step, path)
689              .dep(|s| s.name("librustc-link"))
690              // Needed so rustdoc generates relative links to std.
691              .dep(|s| s.name("doc-crate-std"))
692              .host(true)
693              .default(default && build.config.docs)
694              .run(move |s| doc::rustc(build, s.stage, s.target));
695     }
696
697     // ========================================================================
698     // Distribution targets
699     rules.dist("dist-rustc", "src/librustc")
700          .dep(move |s| s.name("rustc").host(&build.config.build))
701          .host(true)
702          .only_host_build(true)
703          .default(true)
704          .run(move |s| dist::rustc(build, s.stage, s.target));
705     rules.dist("dist-std", "src/libstd")
706          .dep(move |s| {
707              // We want to package up as many target libraries as possible
708              // for the `rust-std` package, so if this is a host target we
709              // depend on librustc and otherwise we just depend on libtest.
710              if build.config.host.iter().any(|t| t == s.target) {
711                  s.name("librustc-link")
712              } else {
713                  s.name("libtest-link")
714              }
715          })
716          .default(true)
717          .only_host_build(true)
718          .run(move |s| dist::std(build, &s.compiler(), s.target));
719     rules.dist("dist-mingw", "path/to/nowhere")
720          .default(true)
721          .only_host_build(true)
722          .run(move |s| {
723              if s.target.contains("pc-windows-gnu") {
724                  dist::mingw(build, s.target)
725              }
726          });
727     rules.dist("dist-src", "src")
728          .default(true)
729          .host(true)
730          .only_build(true)
731          .only_host_build(true)
732          .run(move |_| dist::rust_src(build));
733     rules.dist("dist-docs", "src/doc")
734          .default(true)
735          .only_host_build(true)
736          .dep(|s| s.name("default:doc"))
737          .run(move |s| dist::docs(build, s.stage, s.target));
738     rules.dist("dist-analysis", "analysis")
739          .default(build.config.extended)
740          .dep(|s| s.name("dist-std"))
741          .only_host_build(true)
742          .run(move |s| dist::analysis(build, &s.compiler(), s.target));
743     rules.dist("dist-rls", "rls")
744          .host(true)
745          .only_host_build(true)
746          .dep(|s| s.name("tool-rls"))
747          .run(move |s| dist::rls(build, s.stage, s.target));
748     rules.dist("install", "path/to/nowhere")
749          .dep(|s| s.name("default:dist"))
750          .run(move |s| install::install(build, s.stage, s.target));
751     rules.dist("dist-cargo", "cargo")
752          .host(true)
753          .only_host_build(true)
754          .dep(|s| s.name("tool-cargo"))
755          .run(move |s| dist::cargo(build, s.stage, s.target));
756     rules.dist("dist-extended", "extended")
757          .default(build.config.extended)
758          .host(true)
759          .only_host_build(true)
760          .dep(|d| d.name("dist-std"))
761          .dep(|d| d.name("dist-rustc"))
762          .dep(|d| d.name("dist-mingw"))
763          .dep(|d| d.name("dist-docs"))
764          .dep(|d| d.name("dist-cargo"))
765          .dep(|d| d.name("dist-rls"))
766          .dep(|d| d.name("dist-analysis"))
767          .run(move |s| dist::extended(build, s.stage, s.target));
768
769     rules.dist("dist-sign", "hash-and-sign")
770          .host(true)
771          .only_build(true)
772          .only_host_build(true)
773          .dep(move |s| s.name("tool-build-manifest").target(&build.config.build).stage(0))
774          .run(move |_| dist::hash_and_sign(build));
775
776     rules.verify();
777     return rules;
778 }
779
780 #[derive(PartialEq, Eq, Hash, Clone, Debug)]
781 struct Step<'a> {
782     /// Human readable name of the rule this step is executing. Possible names
783     /// are all defined above in `build_rules`.
784     name: &'a str,
785
786     /// The stage this step is executing in. This is typically 0, 1, or 2.
787     stage: u32,
788
789     /// This step will likely involve a compiler, and the target that compiler
790     /// itself is built for is called the host, this variable. Typically this is
791     /// the target of the build machine itself.
792     host: &'a str,
793
794     /// The target that this step represents generating. If you're building a
795     /// standard library for a new suite of targets, for example, this'll be set
796     /// to those targets.
797     target: &'a str,
798 }
799
800 impl<'a> Step<'a> {
801     fn noop() -> Step<'a> {
802         Step { name: "", stage: 0, host: "", target: "" }
803     }
804
805     /// Creates a new step which is the same as this, except has a new name.
806     fn name(&self, name: &'a str) -> Step<'a> {
807         Step { name: name, ..*self }
808     }
809
810     /// Creates a new step which is the same as this, except has a new stage.
811     fn stage(&self, stage: u32) -> Step<'a> {
812         Step { stage: stage, ..*self }
813     }
814
815     /// Creates a new step which is the same as this, except has a new host.
816     fn host(&self, host: &'a str) -> Step<'a> {
817         Step { host: host, ..*self }
818     }
819
820     /// Creates a new step which is the same as this, except has a new target.
821     fn target(&self, target: &'a str) -> Step<'a> {
822         Step { target: target, ..*self }
823     }
824
825     /// Returns the `Compiler` structure that this step corresponds to.
826     fn compiler(&self) -> Compiler<'a> {
827         Compiler::new(self.stage, self.host)
828     }
829 }
830
831 struct Rule<'a> {
832     /// The human readable name of this target, defined in `build_rules`.
833     name: &'a str,
834
835     /// The path associated with this target, used in the `./x.py` driver for
836     /// easy and ergonomic specification of what to do.
837     path: &'a str,
838
839     /// The "kind" of top-level command that this rule is associated with, only
840     /// relevant if this is a default rule.
841     kind: Kind,
842
843     /// List of dependencies this rule has. Each dependency is a function from a
844     /// step that's being executed to another step that should be executed.
845     deps: Vec<Box<Fn(&Step<'a>) -> Step<'a> + 'a>>,
846
847     /// How to actually execute this rule. Takes a step with contextual
848     /// information and then executes it.
849     run: Box<Fn(&Step<'a>) + 'a>,
850
851     /// Whether or not this is a "default" rule. That basically means that if
852     /// you run, for example, `./x.py test` whether it's included or not.
853     default: bool,
854
855     /// Whether or not this is a "host" rule, or in other words whether this is
856     /// only intended for compiler hosts and not for targets that are being
857     /// generated.
858     host: bool,
859
860     /// Whether this rule is only for steps where the host is the build triple,
861     /// not anything in hosts or targets.
862     only_host_build: bool,
863
864     /// Whether this rule is only for the build triple, not anything in hosts or
865     /// targets.
866     only_build: bool,
867
868     /// A list of "order only" dependencies. This rules does not actually
869     /// depend on these rules, but if they show up in the dependency graph then
870     /// this rule must be executed after all these rules.
871     after: Vec<&'a str>,
872 }
873
874 #[derive(PartialEq)]
875 enum Kind {
876     Build,
877     Test,
878     Bench,
879     Dist,
880     Doc,
881 }
882
883 impl<'a> Rule<'a> {
884     fn new(name: &'a str, path: &'a str, kind: Kind) -> Rule<'a> {
885         Rule {
886             name: name,
887             deps: Vec::new(),
888             run: Box::new(|_| ()),
889             path: path,
890             kind: kind,
891             default: false,
892             host: false,
893             only_host_build: false,
894             only_build: false,
895             after: Vec::new(),
896         }
897     }
898 }
899
900 /// Builder pattern returned from the various methods on `Rules` which will add
901 /// the rule to the internal list on `Drop`.
902 struct RuleBuilder<'a: 'b, 'b> {
903     rules: &'b mut Rules<'a>,
904     rule: Rule<'a>,
905 }
906
907 impl<'a, 'b> RuleBuilder<'a, 'b> {
908     fn dep<F>(&mut self, f: F) -> &mut Self
909         where F: Fn(&Step<'a>) -> Step<'a> + 'a,
910     {
911         self.rule.deps.push(Box::new(f));
912         self
913     }
914
915     fn after(&mut self, step: &'a str) -> &mut Self {
916         self.rule.after.push(step);
917         self
918     }
919
920     fn run<F>(&mut self, f: F) -> &mut Self
921         where F: Fn(&Step<'a>) + 'a,
922     {
923         self.rule.run = Box::new(f);
924         self
925     }
926
927     fn default(&mut self, default: bool) -> &mut Self {
928         self.rule.default = default;
929         self
930     }
931
932     fn host(&mut self, host: bool) -> &mut Self {
933         self.rule.host = host;
934         self
935     }
936
937     fn only_build(&mut self, only_build: bool) -> &mut Self {
938         self.rule.only_build = only_build;
939         self
940     }
941
942     fn only_host_build(&mut self, only_host_build: bool) -> &mut Self {
943         self.rule.only_host_build = only_host_build;
944         self
945     }
946 }
947
948 impl<'a, 'b> Drop for RuleBuilder<'a, 'b> {
949     fn drop(&mut self) {
950         let rule = mem::replace(&mut self.rule, Rule::new("", "", Kind::Build));
951         let prev = self.rules.rules.insert(rule.name, rule);
952         if let Some(prev) = prev {
953             panic!("duplicate rule named: {}", prev.name);
954         }
955     }
956 }
957
958 pub struct Rules<'a> {
959     build: &'a Build,
960     sbuild: Step<'a>,
961     rules: BTreeMap<&'a str, Rule<'a>>,
962 }
963
964 impl<'a> Rules<'a> {
965     fn new(build: &'a Build) -> Rules<'a> {
966         Rules {
967             build: build,
968             sbuild: Step {
969                 stage: build.flags.stage.unwrap_or(2),
970                 target: &build.config.build,
971                 host: &build.config.build,
972                 name: "",
973             },
974             rules: BTreeMap::new(),
975         }
976     }
977
978     /// Creates a new rule of `Kind::Build` with the specified human readable
979     /// name and path associated with it.
980     ///
981     /// The builder returned should be configured further with information such
982     /// as how to actually run this rule.
983     fn build<'b>(&'b mut self, name: &'a str, path: &'a str)
984                  -> RuleBuilder<'a, 'b> {
985         self.rule(name, path, Kind::Build)
986     }
987
988     /// Same as `build`, but for `Kind::Test`.
989     fn test<'b>(&'b mut self, name: &'a str, path: &'a str)
990                 -> RuleBuilder<'a, 'b> {
991         self.rule(name, path, Kind::Test)
992     }
993
994     /// Same as `build`, but for `Kind::Bench`.
995     fn bench<'b>(&'b mut self, name: &'a str, path: &'a str)
996                 -> RuleBuilder<'a, 'b> {
997         self.rule(name, path, Kind::Bench)
998     }
999
1000     /// Same as `build`, but for `Kind::Doc`.
1001     fn doc<'b>(&'b mut self, name: &'a str, path: &'a str)
1002                -> RuleBuilder<'a, 'b> {
1003         self.rule(name, path, Kind::Doc)
1004     }
1005
1006     /// Same as `build`, but for `Kind::Dist`.
1007     fn dist<'b>(&'b mut self, name: &'a str, path: &'a str)
1008                 -> RuleBuilder<'a, 'b> {
1009         self.rule(name, path, Kind::Dist)
1010     }
1011
1012     fn rule<'b>(&'b mut self,
1013                 name: &'a str,
1014                 path: &'a str,
1015                 kind: Kind) -> RuleBuilder<'a, 'b> {
1016         RuleBuilder {
1017             rules: self,
1018             rule: Rule::new(name, path, kind),
1019         }
1020     }
1021
1022     /// Verify the dependency graph defined by all our rules are correct, e.g.
1023     /// everything points to a valid something else.
1024     fn verify(&self) {
1025         for rule in self.rules.values() {
1026             for dep in rule.deps.iter() {
1027                 let dep = dep(&self.sbuild.name(rule.name));
1028                 if self.rules.contains_key(&dep.name) || dep.name.starts_with("default:") {
1029                     continue
1030                 }
1031                 if dep == Step::noop() {
1032                     continue
1033                 }
1034                 panic!("\
1035
1036 invalid rule dependency graph detected, was a rule added and maybe typo'd?
1037
1038     `{}` depends on `{}` which does not exist
1039
1040 ", rule.name, dep.name);
1041             }
1042         }
1043     }
1044
1045     pub fn get_help(&self, command: &str) -> Option<String> {
1046         let kind = match command {
1047             "build" => Kind::Build,
1048             "doc" => Kind::Doc,
1049             "test" => Kind::Test,
1050             "bench" => Kind::Bench,
1051             "dist" => Kind::Dist,
1052             _ => return None,
1053         };
1054         let rules = self.rules.values().filter(|r| r.kind == kind);
1055         let rules = rules.filter(|r| !r.path.contains("nowhere"));
1056         let mut rules = rules.collect::<Vec<_>>();
1057         rules.sort_by_key(|r| r.path);
1058
1059         let mut help_string = String::from("Available paths:\n");
1060         for rule in rules {
1061             help_string.push_str(format!("    ./x.py {} {}\n", command, rule.path).as_str());
1062         }
1063         Some(help_string)
1064     }
1065
1066     /// Construct the top-level build steps that we're going to be executing,
1067     /// given the subcommand that our build is performing.
1068     fn plan(&self) -> Vec<Step<'a>> {
1069         // Ok, the logic here is pretty subtle, and involves quite a few
1070         // conditionals. The basic idea here is to:
1071         //
1072         // 1. First, filter all our rules to the relevant ones. This means that
1073         //    the command specified corresponds to one of our `Kind` variants,
1074         //    and we filter all rules based on that.
1075         //
1076         // 2. Next, we determine which rules we're actually executing. If a
1077         //    number of path filters were specified on the command line we look
1078         //    for those, otherwise we look for anything tagged `default`.
1079         //    Here we also compute the priority of each rule based on how early
1080         //    in the command line the matching path filter showed up.
1081         //
1082         // 3. Finally, we generate some steps with host and target information.
1083         //
1084         // The last step is by far the most complicated and subtle. The basic
1085         // thinking here is that we want to take the cartesian product of
1086         // specified hosts and targets and build rules with that. The list of
1087         // hosts and targets, if not specified, come from the how this build was
1088         // configured. If the rule we're looking at is a host-only rule the we
1089         // ignore the list of targets and instead consider the list of hosts
1090         // also the list of targets.
1091         //
1092         // Once the host and target lists are generated we take the cartesian
1093         // product of the two and then create a step based off them. Note that
1094         // the stage each step is associated was specified with the `--step`
1095         // flag on the command line.
1096         let (kind, paths) = match self.build.flags.cmd {
1097             Subcommand::Build { ref paths } => (Kind::Build, &paths[..]),
1098             Subcommand::Doc { ref paths } => (Kind::Doc, &paths[..]),
1099             Subcommand::Test { ref paths, test_args: _ } => (Kind::Test, &paths[..]),
1100             Subcommand::Bench { ref paths, test_args: _ } => (Kind::Bench, &paths[..]),
1101             Subcommand::Dist { ref paths, install } => {
1102                 if install {
1103                     return vec![self.sbuild.name("install")]
1104                 } else {
1105                     (Kind::Dist, &paths[..])
1106                 }
1107             }
1108             Subcommand::Clean => panic!(),
1109         };
1110
1111         let mut rules: Vec<_> = self.rules.values().filter_map(|rule| {
1112             if rule.kind != kind {
1113                 return None;
1114             }
1115
1116             if paths.len() == 0 && rule.default {
1117                 Some((rule, 0))
1118             } else {
1119                 paths.iter().position(|path| path.ends_with(rule.path))
1120                      .map(|priority| (rule, priority))
1121             }
1122         }).collect();
1123
1124         rules.sort_by_key(|&(_, priority)| priority);
1125
1126         rules.into_iter().flat_map(|(rule, _)| {
1127             let hosts = if rule.only_host_build || rule.only_build {
1128                 &self.build.config.host[..1]
1129             } else if self.build.flags.host.len() > 0 {
1130                 &self.build.flags.host
1131             } else {
1132                 &self.build.config.host
1133             };
1134             let targets = if self.build.flags.target.len() > 0 {
1135                 &self.build.flags.target
1136             } else {
1137                 &self.build.config.target
1138             };
1139             // Determine the actual targets participating in this rule.
1140             // NOTE: We should keep the full projection from build triple to
1141             // the hosts for the dist steps, now that the hosts array above is
1142             // truncated to avoid duplication of work in that case. Therefore
1143             // the original non-shadowed hosts array is used below.
1144             let arr = if rule.host {
1145                 // If --target was specified but --host wasn't specified,
1146                 // don't run any host-only tests. Also, respect any `--host`
1147                 // overrides as done for `hosts`.
1148                 if self.build.flags.host.len() > 0 {
1149                     &self.build.flags.host[..]
1150                 } else if self.build.flags.target.len() > 0 {
1151                     &[]
1152                 } else if rule.only_build {
1153                     &self.build.config.host[..1]
1154                 } else {
1155                     &self.build.config.host[..]
1156                 }
1157             } else {
1158                 targets
1159             };
1160
1161             hosts.iter().flat_map(move |host| {
1162                 arr.iter().map(move |target| {
1163                     self.sbuild.name(rule.name).target(target).host(host)
1164                 })
1165             })
1166         }).collect()
1167     }
1168
1169     /// Execute all top-level targets indicated by `steps`.
1170     ///
1171     /// This will take the list returned by `plan` and then execute each step
1172     /// along with all required dependencies as it goes up the chain.
1173     fn run(&self, steps: &[Step<'a>]) {
1174         self.build.verbose("bootstrap top targets:");
1175         for step in steps.iter() {
1176             self.build.verbose(&format!("\t{:?}", step));
1177         }
1178
1179         // Using `steps` as the top-level targets, make a topological ordering
1180         // of what we need to do.
1181         let order = self.expand(steps);
1182
1183         // Print out what we're doing for debugging
1184         self.build.verbose("bootstrap build plan:");
1185         for step in order.iter() {
1186             self.build.verbose(&format!("\t{:?}", step));
1187         }
1188
1189         // And finally, iterate over everything and execute it.
1190         for step in order.iter() {
1191             if self.build.flags.keep_stage.map_or(false, |s| step.stage <= s) {
1192                 self.build.verbose(&format!("keeping step {:?}", step));
1193                 continue;
1194             }
1195             self.build.verbose(&format!("executing step {:?}", step));
1196             (self.rules[step.name].run)(step);
1197         }
1198     }
1199
1200     /// From the top level targets `steps` generate a topological ordering of
1201     /// all steps needed to run those steps.
1202     fn expand(&self, steps: &[Step<'a>]) -> Vec<Step<'a>> {
1203         // First up build a graph of steps and their dependencies. The `nodes`
1204         // map is a map from step to a unique number. The `edges` map is a
1205         // map from these unique numbers to a list of other numbers,
1206         // representing dependencies.
1207         let mut nodes = HashMap::new();
1208         nodes.insert(Step::noop(), 0);
1209         let mut edges = HashMap::new();
1210         edges.insert(0, HashSet::new());
1211         for step in steps {
1212             self.build_graph(step.clone(), &mut nodes, &mut edges);
1213         }
1214
1215         // Now that we've built up the actual dependency graph, draw more
1216         // dependency edges to satisfy the `after` dependencies field for each
1217         // rule.
1218         self.satisfy_after_deps(&nodes, &mut edges);
1219
1220         // And finally, perform a topological sort to return a list of steps to
1221         // execute.
1222         let mut order = Vec::new();
1223         let mut visited = HashSet::new();
1224         visited.insert(0);
1225         let idx_to_node = nodes.iter().map(|p| (*p.1, p.0)).collect::<HashMap<_, _>>();
1226         for idx in 0..nodes.len() {
1227             self.topo_sort(idx, &idx_to_node, &edges, &mut visited, &mut order);
1228         }
1229         return order
1230     }
1231
1232     /// Builds the dependency graph rooted at `step`.
1233     ///
1234     /// The `nodes` and `edges` maps are filled out according to the rule
1235     /// described by `step.name`.
1236     fn build_graph(&self,
1237                    step: Step<'a>,
1238                    nodes: &mut HashMap<Step<'a>, usize>,
1239                    edges: &mut HashMap<usize, HashSet<usize>>) -> usize {
1240         use std::collections::hash_map::Entry;
1241
1242         let idx = nodes.len();
1243         match nodes.entry(step.clone()) {
1244             Entry::Vacant(e) => { e.insert(idx); }
1245             Entry::Occupied(e) => return *e.get(),
1246         }
1247
1248         let mut deps = Vec::new();
1249         for dep in self.rules[step.name].deps.iter() {
1250             let dep = dep(&step);
1251             if dep.name.starts_with("default:") {
1252                 let kind = match &dep.name[8..] {
1253                     "doc" => Kind::Doc,
1254                     "dist" => Kind::Dist,
1255                     kind => panic!("unknown kind: `{}`", kind),
1256                 };
1257                 let host = self.build.config.host.iter().any(|h| h == dep.target);
1258                 let rules = self.rules.values().filter(|r| r.default);
1259                 for rule in rules.filter(|r| r.kind == kind && (!r.host || host)) {
1260                     deps.push(self.build_graph(dep.name(rule.name), nodes, edges));
1261                 }
1262             } else {
1263                 deps.push(self.build_graph(dep, nodes, edges));
1264             }
1265         }
1266
1267         edges.entry(idx).or_insert(HashSet::new()).extend(deps);
1268         return idx
1269     }
1270
1271     /// Given a dependency graph with a finished list of `nodes`, fill out more
1272     /// dependency `edges`.
1273     ///
1274     /// This is the step which satisfies all `after` listed dependencies in
1275     /// `Rule` above.
1276     fn satisfy_after_deps(&self,
1277                           nodes: &HashMap<Step<'a>, usize>,
1278                           edges: &mut HashMap<usize, HashSet<usize>>) {
1279         // Reverse map from the name of a step to the node indices that it
1280         // appears at.
1281         let mut name_to_idx = HashMap::new();
1282         for (step, &idx) in nodes {
1283             name_to_idx.entry(step.name).or_insert(Vec::new()).push(idx);
1284         }
1285
1286         for (step, idx) in nodes {
1287             if *step == Step::noop() {
1288                 continue
1289             }
1290             for after in self.rules[step.name].after.iter() {
1291                 // This is the critical piece of an `after` dependency. If the
1292                 // dependency isn't actually in our graph then no edge is drawn,
1293                 // only if it's already present do we draw the edges.
1294                 if let Some(idxs) = name_to_idx.get(after) {
1295                     edges.get_mut(idx).unwrap()
1296                          .extend(idxs.iter().cloned());
1297                 }
1298             }
1299         }
1300     }
1301
1302     fn topo_sort(&self,
1303                  cur: usize,
1304                  nodes: &HashMap<usize, &Step<'a>>,
1305                  edges: &HashMap<usize, HashSet<usize>>,
1306                  visited: &mut HashSet<usize>,
1307                  order: &mut Vec<Step<'a>>) {
1308         if !visited.insert(cur) {
1309             return
1310         }
1311         for dep in edges[&cur].iter() {
1312             self.topo_sort(*dep, nodes, edges, visited, order);
1313         }
1314         order.push(nodes[&cur].clone());
1315     }
1316 }
1317
1318 #[cfg(test)]
1319 mod tests {
1320     use std::env;
1321
1322     use Build;
1323     use config::Config;
1324     use flags::Flags;
1325
1326     macro_rules! a {
1327         ($($a:expr),*) => (vec![$($a.to_string()),*])
1328     }
1329
1330     fn build(args: &[&str],
1331              extra_host: &[&str],
1332              extra_target: &[&str]) -> Build {
1333         let mut args = args.iter().map(|s| s.to_string()).collect::<Vec<_>>();
1334         args.push("--build".to_string());
1335         args.push("A".to_string());
1336         let flags = Flags::parse(&args);
1337
1338         let mut config = Config::default();
1339         config.docs = true;
1340         config.build = "A".to_string();
1341         config.host = vec![config.build.clone()];
1342         config.host.extend(extra_host.iter().map(|s| s.to_string()));
1343         config.target = config.host.clone();
1344         config.target.extend(extra_target.iter().map(|s| s.to_string()));
1345
1346         let mut build = Build::new(flags, config);
1347         let cwd = env::current_dir().unwrap();
1348         build.crates.insert("std".to_string(), ::Crate {
1349             name: "std".to_string(),
1350             deps: Vec::new(),
1351             path: cwd.join("src/std"),
1352             doc_step: "doc-crate-std".to_string(),
1353             build_step: "build-crate-std".to_string(),
1354             test_step: "test-crate-std".to_string(),
1355             bench_step: "bench-crate-std".to_string(),
1356             version: String::new(),
1357         });
1358         build.crates.insert("test".to_string(), ::Crate {
1359             name: "test".to_string(),
1360             deps: Vec::new(),
1361             path: cwd.join("src/test"),
1362             doc_step: "doc-crate-test".to_string(),
1363             build_step: "build-crate-test".to_string(),
1364             test_step: "test-crate-test".to_string(),
1365             bench_step: "bench-crate-test".to_string(),
1366             version: String::new(),
1367         });
1368         build.crates.insert("rustc-main".to_string(), ::Crate {
1369             name: "rustc-main".to_string(),
1370             deps: Vec::new(),
1371             version: String::new(),
1372             path: cwd.join("src/rustc-main"),
1373             doc_step: "doc-crate-rustc-main".to_string(),
1374             build_step: "build-crate-rustc-main".to_string(),
1375             test_step: "test-crate-rustc-main".to_string(),
1376             bench_step: "bench-crate-rustc-main".to_string(),
1377         });
1378         return build
1379     }
1380
1381     #[test]
1382     fn dist_baseline() {
1383         let build = build(&["dist"], &[], &[]);
1384         let rules = super::build_rules(&build);
1385         let plan = rules.plan();
1386         println!("rules: {:#?}", plan);
1387         assert!(plan.iter().all(|s| s.stage == 2));
1388         assert!(plan.iter().all(|s| s.host == "A" ));
1389         assert!(plan.iter().all(|s| s.target == "A" ));
1390
1391         let step = super::Step {
1392             name: "",
1393             stage: 2,
1394             host: &build.config.build,
1395             target: &build.config.build,
1396         };
1397
1398         assert!(plan.contains(&step.name("dist-docs")));
1399         assert!(plan.contains(&step.name("dist-mingw")));
1400         assert!(plan.contains(&step.name("dist-rustc")));
1401         assert!(plan.contains(&step.name("dist-std")));
1402         assert!(plan.contains(&step.name("dist-src")));
1403     }
1404
1405     #[test]
1406     fn dist_with_targets() {
1407         let build = build(&["dist"], &[], &["B"]);
1408         let rules = super::build_rules(&build);
1409         let plan = rules.plan();
1410         println!("rules: {:#?}", plan);
1411         assert!(plan.iter().all(|s| s.stage == 2));
1412         assert!(plan.iter().all(|s| s.host == "A" ));
1413
1414         let step = super::Step {
1415             name: "",
1416             stage: 2,
1417             host: &build.config.build,
1418             target: &build.config.build,
1419         };
1420
1421         assert!(plan.contains(&step.name("dist-docs")));
1422         assert!(plan.contains(&step.name("dist-mingw")));
1423         assert!(plan.contains(&step.name("dist-rustc")));
1424         assert!(plan.contains(&step.name("dist-std")));
1425         assert!(plan.contains(&step.name("dist-src")));
1426
1427         assert!(plan.contains(&step.target("B").name("dist-docs")));
1428         assert!(plan.contains(&step.target("B").name("dist-mingw")));
1429         assert!(!plan.contains(&step.target("B").name("dist-rustc")));
1430         assert!(plan.contains(&step.target("B").name("dist-std")));
1431         assert!(!plan.contains(&step.target("B").name("dist-src")));
1432     }
1433
1434     #[test]
1435     fn dist_with_hosts() {
1436         let build = build(&["dist"], &["B"], &[]);
1437         let rules = super::build_rules(&build);
1438         let plan = rules.plan();
1439         println!("rules: {:#?}", plan);
1440         assert!(plan.iter().all(|s| s.stage == 2));
1441
1442         let step = super::Step {
1443             name: "",
1444             stage: 2,
1445             host: &build.config.build,
1446             target: &build.config.build,
1447         };
1448
1449         assert!(!plan.iter().any(|s| s.host == "B"));
1450
1451         assert!(plan.contains(&step.name("dist-docs")));
1452         assert!(plan.contains(&step.name("dist-mingw")));
1453         assert!(plan.contains(&step.name("dist-rustc")));
1454         assert!(plan.contains(&step.name("dist-std")));
1455         assert!(plan.contains(&step.name("dist-src")));
1456
1457         assert!(plan.contains(&step.target("B").name("dist-docs")));
1458         assert!(plan.contains(&step.target("B").name("dist-mingw")));
1459         assert!(plan.contains(&step.target("B").name("dist-rustc")));
1460         assert!(plan.contains(&step.target("B").name("dist-std")));
1461         assert!(!plan.contains(&step.target("B").name("dist-src")));
1462     }
1463
1464     #[test]
1465     fn dist_with_targets_and_hosts() {
1466         let build = build(&["dist"], &["B"], &["C"]);
1467         let rules = super::build_rules(&build);
1468         let plan = rules.plan();
1469         println!("rules: {:#?}", plan);
1470         assert!(plan.iter().all(|s| s.stage == 2));
1471
1472         let step = super::Step {
1473             name: "",
1474             stage: 2,
1475             host: &build.config.build,
1476             target: &build.config.build,
1477         };
1478
1479         assert!(!plan.iter().any(|s| s.host == "B"));
1480         assert!(!plan.iter().any(|s| s.host == "C"));
1481
1482         assert!(plan.contains(&step.name("dist-docs")));
1483         assert!(plan.contains(&step.name("dist-mingw")));
1484         assert!(plan.contains(&step.name("dist-rustc")));
1485         assert!(plan.contains(&step.name("dist-std")));
1486         assert!(plan.contains(&step.name("dist-src")));
1487
1488         assert!(plan.contains(&step.target("B").name("dist-docs")));
1489         assert!(plan.contains(&step.target("B").name("dist-mingw")));
1490         assert!(plan.contains(&step.target("B").name("dist-rustc")));
1491         assert!(plan.contains(&step.target("B").name("dist-std")));
1492         assert!(!plan.contains(&step.target("B").name("dist-src")));
1493
1494         assert!(plan.contains(&step.target("C").name("dist-docs")));
1495         assert!(plan.contains(&step.target("C").name("dist-mingw")));
1496         assert!(!plan.contains(&step.target("C").name("dist-rustc")));
1497         assert!(plan.contains(&step.target("C").name("dist-std")));
1498         assert!(!plan.contains(&step.target("C").name("dist-src")));
1499     }
1500
1501     #[test]
1502     fn dist_target_with_target_flag() {
1503         let build = build(&["dist", "--target=C"], &["B"], &["C"]);
1504         let rules = super::build_rules(&build);
1505         let plan = rules.plan();
1506         println!("rules: {:#?}", plan);
1507         assert!(plan.iter().all(|s| s.stage == 2));
1508
1509         let step = super::Step {
1510             name: "",
1511             stage: 2,
1512             host: &build.config.build,
1513             target: &build.config.build,
1514         };
1515
1516         assert!(!plan.iter().any(|s| s.target == "A"));
1517         assert!(!plan.iter().any(|s| s.target == "B"));
1518         assert!(!plan.iter().any(|s| s.host == "B"));
1519         assert!(!plan.iter().any(|s| s.host == "C"));
1520
1521         assert!(plan.contains(&step.target("C").name("dist-docs")));
1522         assert!(plan.contains(&step.target("C").name("dist-mingw")));
1523         assert!(!plan.contains(&step.target("C").name("dist-rustc")));
1524         assert!(plan.contains(&step.target("C").name("dist-std")));
1525         assert!(!plan.contains(&step.target("C").name("dist-src")));
1526     }
1527
1528     #[test]
1529     fn dist_host_with_target_flag() {
1530         let build = build(&["dist", "--host=B", "--target=B"], &["B"], &["C"]);
1531         let rules = super::build_rules(&build);
1532         let plan = rules.plan();
1533         println!("rules: {:#?}", plan);
1534         assert!(plan.iter().all(|s| s.stage == 2));
1535
1536         let step = super::Step {
1537             name: "",
1538             stage: 2,
1539             host: &build.config.build,
1540             target: &build.config.build,
1541         };
1542
1543         assert!(!plan.iter().any(|s| s.target == "A"));
1544         assert!(!plan.iter().any(|s| s.target == "C"));
1545         assert!(!plan.iter().any(|s| s.host == "B"));
1546         assert!(!plan.iter().any(|s| s.host == "C"));
1547
1548         assert!(plan.contains(&step.target("B").name("dist-docs")));
1549         assert!(plan.contains(&step.target("B").name("dist-mingw")));
1550         assert!(plan.contains(&step.target("B").name("dist-rustc")));
1551         assert!(plan.contains(&step.target("B").name("dist-std")));
1552         assert!(plan.contains(&step.target("B").name("dist-src")));
1553
1554         let all = rules.expand(&plan);
1555         println!("all rules: {:#?}", all);
1556         assert!(!all.contains(&step.name("rustc")));
1557         assert!(!all.contains(&step.name("build-crate-test").stage(1)));
1558
1559         // all stage0 compiles should be for the build target, A
1560         for step in all.iter().filter(|s| s.stage == 0) {
1561             if !step.name.contains("build-crate") {
1562                 continue
1563             }
1564             println!("step: {:?}", step);
1565             assert!(step.host != "B");
1566             assert!(step.target != "B");
1567             assert!(step.host != "C");
1568             assert!(step.target != "C");
1569         }
1570     }
1571
1572     #[test]
1573     fn build_default() {
1574         let build = build(&["build"], &["B"], &["C"]);
1575         let rules = super::build_rules(&build);
1576         let plan = rules.plan();
1577         println!("rules: {:#?}", plan);
1578         assert!(plan.iter().all(|s| s.stage == 2));
1579
1580         let step = super::Step {
1581             name: "",
1582             stage: 2,
1583             host: &build.config.build,
1584             target: &build.config.build,
1585         };
1586
1587         // rustc built for all for of (A, B) x (A, B)
1588         assert!(plan.contains(&step.name("librustc")));
1589         assert!(plan.contains(&step.target("B").name("librustc")));
1590         assert!(plan.contains(&step.host("B").target("A").name("librustc")));
1591         assert!(plan.contains(&step.host("B").target("B").name("librustc")));
1592
1593         // rustc never built for C
1594         assert!(!plan.iter().any(|s| {
1595             s.name.contains("rustc") && (s.host == "C" || s.target == "C")
1596         }));
1597
1598         // test built for everything
1599         assert!(plan.contains(&step.name("libtest")));
1600         assert!(plan.contains(&step.target("B").name("libtest")));
1601         assert!(plan.contains(&step.host("B").target("A").name("libtest")));
1602         assert!(plan.contains(&step.host("B").target("B").name("libtest")));
1603         assert!(plan.contains(&step.host("A").target("C").name("libtest")));
1604         assert!(plan.contains(&step.host("B").target("C").name("libtest")));
1605
1606         let all = rules.expand(&plan);
1607         println!("all rules: {:#?}", all);
1608         assert!(all.contains(&step.name("rustc")));
1609         assert!(all.contains(&step.name("libstd")));
1610     }
1611
1612     #[test]
1613     fn build_filtered() {
1614         let build = build(&["build", "--target=C"], &["B"], &["C"]);
1615         let rules = super::build_rules(&build);
1616         let plan = rules.plan();
1617         println!("rules: {:#?}", plan);
1618         assert!(plan.iter().all(|s| s.stage == 2));
1619
1620         assert!(!plan.iter().any(|s| s.name.contains("rustc")));
1621         assert!(plan.iter().all(|s| {
1622             !s.name.contains("test") || s.target == "C"
1623         }));
1624     }
1625
1626     #[test]
1627     fn test_default() {
1628         let build = build(&["test"], &[], &[]);
1629         let rules = super::build_rules(&build);
1630         let plan = rules.plan();
1631         println!("rules: {:#?}", plan);
1632         assert!(plan.iter().all(|s| s.stage == 2));
1633         assert!(plan.iter().all(|s| s.host == "A"));
1634         assert!(plan.iter().all(|s| s.target == "A"));
1635
1636         assert!(plan.iter().any(|s| s.name.contains("-ui")));
1637         assert!(plan.iter().any(|s| s.name.contains("cfail")));
1638         assert!(plan.iter().any(|s| s.name.contains("cfail-full")));
1639         assert!(plan.iter().any(|s| s.name.contains("codegen-units")));
1640         assert!(plan.iter().any(|s| s.name.contains("debuginfo")));
1641         assert!(plan.iter().any(|s| s.name.contains("docs")));
1642         assert!(plan.iter().any(|s| s.name.contains("error-index")));
1643         assert!(plan.iter().any(|s| s.name.contains("incremental")));
1644         assert!(plan.iter().any(|s| s.name.contains("linkchecker")));
1645         assert!(plan.iter().any(|s| s.name.contains("mir-opt")));
1646         assert!(plan.iter().any(|s| s.name.contains("pfail")));
1647         assert!(plan.iter().any(|s| s.name.contains("rfail")));
1648         assert!(plan.iter().any(|s| s.name.contains("rfail-full")));
1649         assert!(plan.iter().any(|s| s.name.contains("rmake")));
1650         assert!(plan.iter().any(|s| s.name.contains("rpass")));
1651         assert!(plan.iter().any(|s| s.name.contains("rpass-full")));
1652         assert!(plan.iter().any(|s| s.name.contains("rustc-all")));
1653         assert!(plan.iter().any(|s| s.name.contains("rustdoc")));
1654         assert!(plan.iter().any(|s| s.name.contains("std-all")));
1655         assert!(plan.iter().any(|s| s.name.contains("test-all")));
1656         assert!(plan.iter().any(|s| s.name.contains("tidy")));
1657         assert!(plan.iter().any(|s| s.name.contains("valgrind")));
1658     }
1659
1660     #[test]
1661     fn test_with_a_target() {
1662         let build = build(&["test", "--target=C"], &[], &["C"]);
1663         let rules = super::build_rules(&build);
1664         let plan = rules.plan();
1665         println!("rules: {:#?}", plan);
1666         assert!(plan.iter().all(|s| s.stage == 2));
1667         assert!(plan.iter().all(|s| s.host == "A"));
1668         assert!(plan.iter().all(|s| s.target == "C"));
1669
1670         assert!(plan.iter().any(|s| s.name.contains("-ui")));
1671         assert!(!plan.iter().any(|s| s.name.contains("ui-full")));
1672         assert!(plan.iter().any(|s| s.name.contains("cfail")));
1673         assert!(!plan.iter().any(|s| s.name.contains("cfail-full")));
1674         assert!(plan.iter().any(|s| s.name.contains("codegen-units")));
1675         assert!(plan.iter().any(|s| s.name.contains("debuginfo")));
1676         assert!(!plan.iter().any(|s| s.name.contains("docs")));
1677         assert!(!plan.iter().any(|s| s.name.contains("error-index")));
1678         assert!(plan.iter().any(|s| s.name.contains("incremental")));
1679         assert!(!plan.iter().any(|s| s.name.contains("linkchecker")));
1680         assert!(plan.iter().any(|s| s.name.contains("mir-opt")));
1681         assert!(plan.iter().any(|s| s.name.contains("pfail")));
1682         assert!(plan.iter().any(|s| s.name.contains("rfail")));
1683         assert!(!plan.iter().any(|s| s.name.contains("rfail-full")));
1684         assert!(!plan.iter().any(|s| s.name.contains("rmake")));
1685         assert!(plan.iter().any(|s| s.name.contains("rpass")));
1686         assert!(!plan.iter().any(|s| s.name.contains("rpass-full")));
1687         assert!(!plan.iter().any(|s| s.name.contains("rustc-all")));
1688         assert!(!plan.iter().any(|s| s.name.contains("rustdoc")));
1689         assert!(plan.iter().any(|s| s.name.contains("std-all")));
1690         assert!(plan.iter().any(|s| s.name.contains("test-all")));
1691         assert!(!plan.iter().any(|s| s.name.contains("tidy")));
1692         assert!(plan.iter().any(|s| s.name.contains("valgrind")));
1693     }
1694 }