]> git.lizzy.rs Git - rust.git/blob - src/librustc_interface/passes.rs
Label unmarked time
[rust.git] / src / librustc_interface / passes.rs
1 use crate::interface::{Compiler, Result};
2 use crate::proc_macro_decls;
3 use crate::util;
4
5 use log::{info, log_enabled, warn};
6 use rustc::arena::Arena;
7 use rustc::dep_graph::DepGraph;
8 use rustc::hir::map;
9 use rustc::lint;
10 use rustc::middle;
11 use rustc::middle::cstore::{CrateStore, MetadataLoader, MetadataLoaderDyn};
12 use rustc::session::config::{self, CrateType, Input, OutputFilenames, OutputType};
13 use rustc::session::config::{PpMode, PpSourceMode};
14 use rustc::session::search_paths::PathKind;
15 use rustc::session::Session;
16 use rustc::traits;
17 use rustc::ty::steal::Steal;
18 use rustc::ty::{self, AllArenas, GlobalCtxt, ResolverOutputs, TyCtxt};
19 use rustc::util::common::ErrorReported;
20 use rustc_builtin_macros;
21 use rustc_codegen_ssa::back::link::emit_metadata;
22 use rustc_codegen_utils::codegen_backend::CodegenBackend;
23 use rustc_codegen_utils::link::filename_for_metadata;
24 use rustc_data_structures::sync::{par_iter, Lrc, Once, ParallelIterator, WorkerLocal};
25 use rustc_data_structures::{box_region_allow_access, declare_box_region_type, parallel};
26 use rustc_errors::PResult;
27 use rustc_expand::base::ExtCtxt;
28 use rustc_hir::def_id::{CrateNum, LOCAL_CRATE};
29 use rustc_incremental;
30 use rustc_mir as mir;
31 use rustc_parse::{parse_crate_from_file, parse_crate_from_source_str};
32 use rustc_passes::{self, ast_validation, hir_stats, layout_test};
33 use rustc_plugin_impl as plugin;
34 use rustc_privacy;
35 use rustc_resolve::{Resolver, ResolverArenas};
36 use rustc_span::symbol::Symbol;
37 use rustc_span::FileName;
38 use rustc_traits;
39 use rustc_typeck as typeck;
40 use syntax::early_buffered_lints::BufferedEarlyLint;
41 use syntax::mut_visit::MutVisitor;
42 use syntax::util::node_count::NodeCounter;
43 use syntax::{self, ast, visit};
44
45 use rustc_serialize::json;
46 use tempfile::Builder as TempFileBuilder;
47
48 use std::any::Any;
49 use std::cell::RefCell;
50 use std::ffi::OsString;
51 use std::io::{self, Write};
52 use std::path::PathBuf;
53 use std::rc::Rc;
54 use std::{env, fs, iter, mem};
55
56 pub fn parse<'a>(sess: &'a Session, input: &Input) -> PResult<'a, ast::Crate> {
57     let krate = sess.time("parse_crate", || match input {
58         Input::File(file) => parse_crate_from_file(file, &sess.parse_sess),
59         Input::Str { input, name } => {
60             parse_crate_from_source_str(name.clone(), input.clone(), &sess.parse_sess)
61         }
62     })?;
63
64     if sess.opts.debugging_opts.ast_json_noexpand {
65         println!("{}", json::as_json(&krate));
66     }
67
68     if sess.opts.debugging_opts.input_stats {
69         println!("Lines of code:             {}", sess.source_map().count_lines());
70         println!("Pre-expansion node count:  {}", count_nodes(&krate));
71     }
72
73     if let Some(ref s) = sess.opts.debugging_opts.show_span {
74         syntax::show_span::run(sess.diagnostic(), s, &krate);
75     }
76
77     if sess.opts.debugging_opts.hir_stats {
78         hir_stats::print_ast_stats(&krate, "PRE EXPANSION AST STATS");
79     }
80
81     Ok(krate)
82 }
83
84 fn count_nodes(krate: &ast::Crate) -> usize {
85     let mut counter = NodeCounter::new();
86     visit::walk_crate(&mut counter, krate);
87     counter.count
88 }
89
90 declare_box_region_type!(
91     pub BoxedResolver,
92     for(),
93     (&mut Resolver<'_>) -> (Result<ast::Crate>, ResolverOutputs)
94 );
95
96 /// Runs the "early phases" of the compiler: initial `cfg` processing, loading compiler plugins,
97 /// syntax expansion, secondary `cfg` expansion, synthesis of a test
98 /// harness if one is to be provided, injection of a dependency on the
99 /// standard library and prelude, and name resolution.
100 ///
101 /// Returns `None` if we're aborting after handling -W help.
102 pub fn configure_and_expand(
103     sess: Lrc<Session>,
104     lint_store: Lrc<lint::LintStore>,
105     metadata_loader: Box<MetadataLoaderDyn>,
106     krate: ast::Crate,
107     crate_name: &str,
108 ) -> Result<(ast::Crate, BoxedResolver)> {
109     // Currently, we ignore the name resolution data structures for the purposes of dependency
110     // tracking. Instead we will run name resolution and include its output in the hash of each
111     // item, much like we do for macro expansion. In other words, the hash reflects not just
112     // its contents but the results of name resolution on those contents. Hopefully we'll push
113     // this back at some point.
114     let crate_name = crate_name.to_string();
115     let (result, resolver) = BoxedResolver::new(static move || {
116         let sess = &*sess;
117         let resolver_arenas = Resolver::arenas();
118         let res = configure_and_expand_inner(
119             sess,
120             &lint_store,
121             krate,
122             &crate_name,
123             &resolver_arenas,
124             &*metadata_loader,
125         );
126         let mut resolver = match res {
127             Err(v) => {
128                 yield BoxedResolver::initial_yield(Err(v));
129                 panic!()
130             }
131             Ok((krate, resolver)) => {
132                 yield BoxedResolver::initial_yield(Ok(krate));
133                 resolver
134             }
135         };
136         box_region_allow_access!(for(), (&mut Resolver<'_>), (&mut resolver));
137         resolver.into_outputs()
138     });
139     result.map(|k| (k, resolver))
140 }
141
142 impl BoxedResolver {
143     pub fn to_resolver_outputs(resolver: Rc<RefCell<BoxedResolver>>) -> ResolverOutputs {
144         match Rc::try_unwrap(resolver) {
145             Ok(resolver) => resolver.into_inner().complete(),
146             Err(resolver) => resolver.borrow_mut().access(|resolver| resolver.clone_outputs()),
147         }
148     }
149 }
150
151 pub fn register_plugins<'a>(
152     sess: &'a Session,
153     metadata_loader: &'a dyn MetadataLoader,
154     register_lints: impl Fn(&Session, &mut lint::LintStore),
155     mut krate: ast::Crate,
156     crate_name: &str,
157 ) -> Result<(ast::Crate, Lrc<lint::LintStore>)> {
158     krate = sess.time("attributes_injection", || {
159         rustc_builtin_macros::cmdline_attrs::inject(
160             krate,
161             &sess.parse_sess,
162             &sess.opts.debugging_opts.crate_attr,
163         )
164     });
165
166     let (krate, features) = sess.time("compute_features", || {
167         rustc_expand::config::features(
168             krate,
169             &sess.parse_sess,
170             sess.edition(),
171             &sess.opts.debugging_opts.allow_features,
172         )
173     });
174     // these need to be set "early" so that expansion sees `quote` if enabled.
175     sess.init_features(features);
176
177     let crate_types = util::collect_crate_types(sess, &krate.attrs);
178     sess.crate_types.set(crate_types);
179
180     let disambiguator =
181         sess.time("compute_crate_disambiguator", || util::compute_crate_disambiguator(sess));
182     sess.crate_disambiguator.set(disambiguator);
183     rustc_incremental::prepare_session_directory(sess, &crate_name, disambiguator);
184
185     if sess.opts.incremental.is_some() {
186         sess.time("incr_comp_garbage_collect_session_directories", || {
187             if let Err(e) = rustc_incremental::garbage_collect_session_directories(sess) {
188                 warn!(
189                     "Error while trying to garbage collect incremental \
190                      compilation cache directory: {}",
191                     e
192                 );
193             }
194         });
195     }
196
197     sess.time("recursion_limit", || {
198         middle::recursion_limit::update_limits(sess, &krate);
199     });
200
201     let mut lint_store = rustc_lint::new_lint_store(
202         sess.opts.debugging_opts.no_interleave_lints,
203         sess.unstable_options(),
204     );
205     register_lints(&sess, &mut lint_store);
206
207     let registrars =
208         sess.time("plugin_loading", || plugin::load::load_plugins(sess, metadata_loader, &krate));
209     sess.time("plugin_registration", || {
210         let mut registry = plugin::Registry { lint_store: &mut lint_store };
211         for registrar in registrars {
212             registrar(&mut registry);
213         }
214     });
215
216     Ok((krate, Lrc::new(lint_store)))
217 }
218
219 fn configure_and_expand_inner<'a>(
220     sess: &'a Session,
221     lint_store: &'a lint::LintStore,
222     mut krate: ast::Crate,
223     crate_name: &str,
224     resolver_arenas: &'a ResolverArenas<'a>,
225     metadata_loader: &'a MetadataLoaderDyn,
226 ) -> Result<(ast::Crate, Resolver<'a>)> {
227     sess.time("pre_AST_expansion_lint_checks", || {
228         rustc_lint::check_ast_crate(
229             sess,
230             lint_store,
231             &krate,
232             true,
233             None,
234             rustc_lint::BuiltinCombinedPreExpansionLintPass::new(),
235         );
236     });
237
238     let mut resolver = Resolver::new(sess, &krate, crate_name, metadata_loader, &resolver_arenas);
239     rustc_builtin_macros::register_builtin_macros(&mut resolver, sess.edition());
240
241     krate = sess.time("crate_injection", || {
242         let alt_std_name = sess.opts.alt_std_name.as_ref().map(|s| Symbol::intern(s));
243         let (krate, name) = rustc_builtin_macros::standard_library_imports::inject(
244             krate,
245             &mut resolver,
246             &sess.parse_sess,
247             alt_std_name,
248         );
249         if let Some(name) = name {
250             sess.parse_sess.injected_crate_name.set(name);
251         }
252         krate
253     });
254
255     util::check_attr_crate_type(&krate.attrs, &mut resolver.lint_buffer());
256
257     // Expand all macros
258     krate = sess.time("macro_expand_crate", || {
259         // Windows dlls do not have rpaths, so they don't know how to find their
260         // dependencies. It's up to us to tell the system where to find all the
261         // dependent dlls. Note that this uses cfg!(windows) as opposed to
262         // targ_cfg because syntax extensions are always loaded for the host
263         // compiler, not for the target.
264         //
265         // This is somewhat of an inherently racy operation, however, as
266         // multiple threads calling this function could possibly continue
267         // extending PATH far beyond what it should. To solve this for now we
268         // just don't add any new elements to PATH which are already there
269         // within PATH. This is basically a targeted fix at #17360 for rustdoc
270         // which runs rustc in parallel but has been seen (#33844) to cause
271         // problems with PATH becoming too long.
272         let mut old_path = OsString::new();
273         if cfg!(windows) {
274             old_path = env::var_os("PATH").unwrap_or(old_path);
275             let mut new_path = sess.host_filesearch(PathKind::All).search_path_dirs();
276             for path in env::split_paths(&old_path) {
277                 if !new_path.contains(&path) {
278                     new_path.push(path);
279                 }
280             }
281             env::set_var(
282                 "PATH",
283                 &env::join_paths(
284                     new_path.iter().filter(|p| env::join_paths(iter::once(p)).is_ok()),
285                 )
286                 .unwrap(),
287             );
288         }
289
290         // Create the config for macro expansion
291         let features = sess.features_untracked();
292         let cfg = rustc_expand::expand::ExpansionConfig {
293             features: Some(&features),
294             recursion_limit: *sess.recursion_limit.get(),
295             trace_mac: sess.opts.debugging_opts.trace_macros,
296             should_test: sess.opts.test,
297             ..rustc_expand::expand::ExpansionConfig::default(crate_name.to_string())
298         };
299
300         let mut ecx = ExtCtxt::new(&sess.parse_sess, cfg, &mut resolver);
301
302         // Expand macros now!
303         let krate = sess.time("expand_crate", || ecx.monotonic_expander().expand_crate(krate));
304
305         // The rest is error reporting
306
307         sess.time("check_unused_macros", || {
308             ecx.check_unused_macros();
309         });
310
311         let mut missing_fragment_specifiers: Vec<_> =
312             ecx.parse_sess.missing_fragment_specifiers.borrow().iter().cloned().collect();
313         missing_fragment_specifiers.sort();
314
315         for span in missing_fragment_specifiers {
316             let lint = lint::builtin::MISSING_FRAGMENT_SPECIFIER;
317             let msg = "missing fragment specifier";
318             resolver.lint_buffer().buffer_lint(lint, ast::CRATE_NODE_ID, span, msg);
319         }
320         if cfg!(windows) {
321             env::set_var("PATH", &old_path);
322         }
323         krate
324     });
325
326     sess.time("maybe_building_test_harness", || {
327         rustc_builtin_macros::test_harness::inject(
328             &sess.parse_sess,
329             &mut resolver,
330             sess.opts.test,
331             &mut krate,
332             sess.diagnostic(),
333             &sess.features_untracked(),
334             sess.panic_strategy(),
335             sess.target.target.options.panic_strategy,
336             sess.opts.debugging_opts.panic_abort_tests,
337         )
338     });
339
340     // If we're actually rustdoc then there's no need to actually compile
341     // anything, so switch everything to just looping
342     let mut should_loop = sess.opts.actually_rustdoc;
343     if let Some(PpMode::PpmSource(PpSourceMode::PpmEveryBodyLoops)) = sess.opts.pretty {
344         should_loop |= true;
345     }
346     if should_loop {
347         util::ReplaceBodyWithLoop::new(&mut resolver).visit_crate(&mut krate);
348     }
349
350     let has_proc_macro_decls = sess.time("AST_validation", || {
351         ast_validation::check_crate(sess, &krate, &mut resolver.lint_buffer())
352     });
353
354     let crate_types = sess.crate_types.borrow();
355     let is_proc_macro_crate = crate_types.contains(&config::CrateType::ProcMacro);
356
357     // For backwards compatibility, we don't try to run proc macro injection
358     // if rustdoc is run on a proc macro crate without '--crate-type proc-macro' being
359     // specified. This should only affect users who manually invoke 'rustdoc', as
360     // 'cargo doc' will automatically pass the proper '--crate-type' flags.
361     // However, we do emit a warning, to let such users know that they should
362     // start passing '--crate-type proc-macro'
363     if has_proc_macro_decls && sess.opts.actually_rustdoc && !is_proc_macro_crate {
364         let mut msg = sess.diagnostic().struct_warn(
365             &"Trying to document proc macro crate \
366             without passing '--crate-type proc-macro to rustdoc",
367         );
368
369         msg.warn("The generated documentation may be incorrect");
370         msg.emit()
371     } else {
372         krate = sess.time("maybe_create_a_macro_crate", || {
373             let num_crate_types = crate_types.len();
374             let is_test_crate = sess.opts.test;
375             rustc_builtin_macros::proc_macro_harness::inject(
376                 &sess.parse_sess,
377                 &mut resolver,
378                 krate,
379                 is_proc_macro_crate,
380                 has_proc_macro_decls,
381                 is_test_crate,
382                 num_crate_types,
383                 sess.diagnostic(),
384             )
385         });
386     }
387
388     // Done with macro expansion!
389
390     if sess.opts.debugging_opts.input_stats {
391         println!("Post-expansion node count: {}", count_nodes(&krate));
392     }
393
394     if sess.opts.debugging_opts.hir_stats {
395         hir_stats::print_ast_stats(&krate, "POST EXPANSION AST STATS");
396     }
397
398     if sess.opts.debugging_opts.ast_json {
399         println!("{}", json::as_json(&krate));
400     }
401
402     resolver.resolve_crate(&krate);
403
404     // Needs to go *after* expansion to be able to check the results of macro expansion.
405     sess.time("complete_gated_feature_checking", || {
406         syntax::feature_gate::check_crate(
407             &krate,
408             &sess.parse_sess,
409             &sess.features_untracked(),
410             sess.opts.unstable_features,
411         );
412     });
413
414     // Add all buffered lints from the `ParseSess` to the `Session`.
415     sess.parse_sess.buffered_lints.with_lock(|buffered_lints| {
416         info!("{} parse sess buffered_lints", buffered_lints.len());
417         for BufferedEarlyLint { id, span, msg, lint_id } in buffered_lints.drain(..) {
418             resolver.lint_buffer().buffer_lint(lint_id, id, span, &msg);
419         }
420     });
421
422     Ok((krate, resolver))
423 }
424
425 pub fn lower_to_hir<'res, 'tcx>(
426     sess: &'tcx Session,
427     lint_store: &lint::LintStore,
428     resolver: &'res mut Resolver<'_>,
429     dep_graph: &'res DepGraph,
430     krate: &'res ast::Crate,
431     arena: &'tcx Arena<'tcx>,
432 ) -> Result<map::Forest<'tcx>> {
433     // Lower AST to HIR.
434     let hir_crate = rustc_ast_lowering::lower_crate(
435         sess,
436         &dep_graph,
437         &krate,
438         resolver,
439         rustc_parse::nt_to_tokenstream,
440         arena,
441     );
442
443     if sess.opts.debugging_opts.hir_stats {
444         hir_stats::print_hir_stats(&hir_crate);
445     }
446
447     let hir_forest = map::Forest::new(hir_crate, &dep_graph);
448
449     sess.time("early_lint_checks", || {
450         rustc_lint::check_ast_crate(
451             sess,
452             lint_store,
453             &krate,
454             false,
455             Some(std::mem::take(resolver.lint_buffer())),
456             rustc_lint::BuiltinCombinedEarlyLintPass::new(),
457         )
458     });
459
460     // Discard hygiene data, which isn't required after lowering to HIR.
461     if !sess.opts.debugging_opts.keep_hygiene_data {
462         rustc_span::hygiene::clear_syntax_context_map();
463     }
464
465     Ok(hir_forest)
466 }
467
468 // Returns all the paths that correspond to generated files.
469 fn generated_output_paths(
470     sess: &Session,
471     outputs: &OutputFilenames,
472     exact_name: bool,
473     crate_name: &str,
474 ) -> Vec<PathBuf> {
475     let mut out_filenames = Vec::new();
476     for output_type in sess.opts.output_types.keys() {
477         let file = outputs.path(*output_type);
478         match *output_type {
479             // If the filename has been overridden using `-o`, it will not be modified
480             // by appending `.rlib`, `.exe`, etc., so we can skip this transformation.
481             OutputType::Exe if !exact_name => {
482                 for crate_type in sess.crate_types.borrow().iter() {
483                     let p = ::rustc_codegen_utils::link::filename_for_input(
484                         sess,
485                         *crate_type,
486                         crate_name,
487                         outputs,
488                     );
489                     out_filenames.push(p);
490                 }
491             }
492             OutputType::DepInfo if sess.opts.debugging_opts.dep_info_omit_d_target => {
493                 // Don't add the dep-info output when omitting it from dep-info targets
494             }
495             _ => {
496                 out_filenames.push(file);
497             }
498         }
499     }
500     out_filenames
501 }
502
503 // Runs `f` on every output file path and returns the first non-None result, or None if `f`
504 // returns None for every file path.
505 fn check_output<F, T>(output_paths: &[PathBuf], f: F) -> Option<T>
506 where
507     F: Fn(&PathBuf) -> Option<T>,
508 {
509     for output_path in output_paths {
510         if let Some(result) = f(output_path) {
511             return Some(result);
512         }
513     }
514     None
515 }
516
517 fn output_contains_path(output_paths: &[PathBuf], input_path: &PathBuf) -> bool {
518     let input_path = input_path.canonicalize().ok();
519     if input_path.is_none() {
520         return false;
521     }
522     let check = |output_path: &PathBuf| {
523         if output_path.canonicalize().ok() == input_path { Some(()) } else { None }
524     };
525     check_output(output_paths, check).is_some()
526 }
527
528 fn output_conflicts_with_dir(output_paths: &[PathBuf]) -> Option<PathBuf> {
529     let check = |output_path: &PathBuf| output_path.is_dir().then(|| output_path.clone());
530     check_output(output_paths, check)
531 }
532
533 fn escape_dep_filename(filename: &FileName) -> String {
534     // Apparently clang and gcc *only* escape spaces:
535     // http://llvm.org/klaus/clang/commit/9d50634cfc268ecc9a7250226dd5ca0e945240d4
536     filename.to_string().replace(" ", "\\ ")
537 }
538
539 fn write_out_deps(
540     sess: &Session,
541     boxed_resolver: &Steal<Rc<RefCell<BoxedResolver>>>,
542     outputs: &OutputFilenames,
543     out_filenames: &[PathBuf],
544 ) {
545     // Write out dependency rules to the dep-info file if requested
546     if !sess.opts.output_types.contains_key(&OutputType::DepInfo) {
547         return;
548     }
549     let deps_filename = outputs.path(OutputType::DepInfo);
550
551     let result = (|| -> io::Result<()> {
552         // Build a list of files used to compile the output and
553         // write Makefile-compatible dependency rules
554         let mut files: Vec<String> = sess
555             .source_map()
556             .files()
557             .iter()
558             .filter(|fmap| fmap.is_real_file())
559             .filter(|fmap| !fmap.is_imported())
560             .map(|fmap| escape_dep_filename(&fmap.unmapped_path.as_ref().unwrap_or(&fmap.name)))
561             .collect();
562
563         if sess.binary_dep_depinfo() {
564             boxed_resolver.borrow().borrow_mut().access(|resolver| {
565                 for cnum in resolver.cstore().crates_untracked() {
566                     let source = resolver.cstore().crate_source_untracked(cnum);
567                     if let Some((path, _)) = source.dylib {
568                         files.push(escape_dep_filename(&FileName::Real(path)));
569                     }
570                     if let Some((path, _)) = source.rlib {
571                         files.push(escape_dep_filename(&FileName::Real(path)));
572                     }
573                     if let Some((path, _)) = source.rmeta {
574                         files.push(escape_dep_filename(&FileName::Real(path)));
575                     }
576                 }
577             });
578         }
579
580         let mut file = fs::File::create(&deps_filename)?;
581         for path in out_filenames {
582             writeln!(file, "{}: {}\n", path.display(), files.join(" "))?;
583         }
584
585         // Emit a fake target for each input file to the compilation. This
586         // prevents `make` from spitting out an error if a file is later
587         // deleted. For more info see #28735
588         for path in files {
589             writeln!(file, "{}:", path)?;
590         }
591         Ok(())
592     })();
593
594     match result {
595         Ok(_) => {
596             if sess.opts.json_artifact_notifications {
597                 sess.parse_sess
598                     .span_diagnostic
599                     .emit_artifact_notification(&deps_filename, "dep-info");
600             }
601         }
602         Err(e) => sess.fatal(&format!(
603             "error writing dependencies to `{}`: {}",
604             deps_filename.display(),
605             e
606         )),
607     }
608 }
609
610 pub fn prepare_outputs(
611     sess: &Session,
612     compiler: &Compiler,
613     krate: &ast::Crate,
614     boxed_resolver: &Steal<Rc<RefCell<BoxedResolver>>>,
615     crate_name: &str,
616 ) -> Result<OutputFilenames> {
617     let _timer = sess.timer("prepare_outputs");
618
619     // FIXME: rustdoc passes &[] instead of &krate.attrs here
620     let outputs = util::build_output_filenames(
621         &compiler.input,
622         &compiler.output_dir,
623         &compiler.output_file,
624         &krate.attrs,
625         sess,
626     );
627
628     let output_paths =
629         generated_output_paths(sess, &outputs, compiler.output_file.is_some(), &crate_name);
630
631     // Ensure the source file isn't accidentally overwritten during compilation.
632     if let Some(ref input_path) = compiler.input_path {
633         if sess.opts.will_create_output_file() {
634             if output_contains_path(&output_paths, input_path) {
635                 sess.err(&format!(
636                     "the input file \"{}\" would be overwritten by the generated \
637                         executable",
638                     input_path.display()
639                 ));
640                 return Err(ErrorReported);
641             }
642             if let Some(dir_path) = output_conflicts_with_dir(&output_paths) {
643                 sess.err(&format!(
644                     "the generated executable for the input file \"{}\" conflicts with the \
645                         existing directory \"{}\"",
646                     input_path.display(),
647                     dir_path.display()
648                 ));
649                 return Err(ErrorReported);
650             }
651         }
652     }
653
654     write_out_deps(sess, boxed_resolver, &outputs, &output_paths);
655
656     let only_dep_info = sess.opts.output_types.contains_key(&OutputType::DepInfo)
657         && sess.opts.output_types.len() == 1;
658
659     if !only_dep_info {
660         if let Some(ref dir) = compiler.output_dir {
661             if fs::create_dir_all(dir).is_err() {
662                 sess.err("failed to find or create the directory specified by `--out-dir`");
663                 return Err(ErrorReported);
664             }
665         }
666     }
667
668     Ok(outputs)
669 }
670
671 pub fn default_provide(providers: &mut ty::query::Providers<'_>) {
672     providers.analysis = analysis;
673     proc_macro_decls::provide(providers);
674     plugin::build::provide(providers);
675     rustc::hir::provide(providers);
676     mir::provide(providers);
677     rustc_privacy::provide(providers);
678     typeck::provide(providers);
679     ty::provide(providers);
680     traits::provide(providers);
681     rustc_passes::provide(providers);
682     rustc_resolve::provide(providers);
683     rustc_traits::provide(providers);
684     rustc_metadata::provide(providers);
685     rustc_lint::provide(providers);
686     rustc_codegen_utils::provide(providers);
687     rustc_codegen_ssa::provide(providers);
688 }
689
690 pub fn default_provide_extern(providers: &mut ty::query::Providers<'_>) {
691     rustc_metadata::provide_extern(providers);
692     rustc_codegen_ssa::provide_extern(providers);
693 }
694
695 pub struct QueryContext<'tcx>(&'tcx GlobalCtxt<'tcx>);
696
697 impl<'tcx> QueryContext<'tcx> {
698     pub fn enter<F, R>(&mut self, f: F) -> R
699     where
700         F: FnOnce(TyCtxt<'tcx>) -> R,
701     {
702         ty::tls::enter_global(self.0, |tcx| f(tcx))
703     }
704
705     pub fn print_stats(&self) {
706         self.0.queries.print_stats()
707     }
708 }
709
710 pub fn create_global_ctxt<'tcx>(
711     compiler: &'tcx Compiler,
712     lint_store: Lrc<lint::LintStore>,
713     hir_forest: &'tcx map::Forest<'tcx>,
714     mut resolver_outputs: ResolverOutputs,
715     outputs: OutputFilenames,
716     crate_name: &str,
717     global_ctxt: &'tcx Once<GlobalCtxt<'tcx>>,
718     all_arenas: &'tcx AllArenas,
719     arena: &'tcx WorkerLocal<Arena<'tcx>>,
720 ) -> QueryContext<'tcx> {
721     let sess = &compiler.session();
722     let defs = mem::take(&mut resolver_outputs.definitions);
723
724     // Construct the HIR map.
725     let hir_map = map::map_crate(sess, &*resolver_outputs.cstore, &hir_forest, defs);
726
727     let query_result_on_disk_cache = rustc_incremental::load_query_result_cache(sess);
728
729     let codegen_backend = sess.time("load_codegen_backend", || compiler.codegen_backend());
730
731     let (local_providers, extern_providers) = sess.time("load_codegen_backend", || {
732         let mut local_providers = ty::query::Providers::default();
733         default_provide(&mut local_providers);
734         codegen_backend.provide(&mut local_providers);
735
736         let mut extern_providers = local_providers;
737         default_provide_extern(&mut extern_providers);
738         codegen_backend.provide_extern(&mut extern_providers);
739
740         if let Some(callback) = compiler.override_queries {
741             callback(sess, &mut local_providers, &mut extern_providers);
742         }
743
744         (local_providers, extern_providers)
745     });
746
747     let gcx = sess.time("setup_global_ctxt", || {
748         global_ctxt.init_locking(|| {
749             TyCtxt::create_global_ctxt(
750                 sess,
751                 lint_store,
752                 local_providers,
753                 extern_providers,
754                 &all_arenas,
755                 arena,
756                 resolver_outputs,
757                 hir_map,
758                 query_result_on_disk_cache,
759                 &crate_name,
760                 &outputs,
761             )
762         })
763     });
764
765     // Do some initialization of the DepGraph that can only be done with the tcx available.
766     ty::tls::enter_global(&gcx, |tcx| {
767         tcx.sess.time("dep_graph_tcx_init", || rustc_incremental::dep_graph_tcx_init(tcx));
768     });
769
770     QueryContext(gcx)
771 }
772
773 /// Runs the resolution, type-checking, region checking and other
774 /// miscellaneous analysis passes on the crate.
775 fn analysis(tcx: TyCtxt<'_>, cnum: CrateNum) -> Result<()> {
776     assert_eq!(cnum, LOCAL_CRATE);
777
778     let sess = tcx.sess;
779     let mut entry_point = None;
780
781     sess.time("misc_checking_1", || {
782         parallel!(
783             {
784                 entry_point = sess
785                     .time("looking_for_entry_point", || rustc_passes::entry::find_entry_point(tcx));
786
787                 sess.time("looking_for_plugin_registrar", || {
788                     plugin::build::find_plugin_registrar(tcx)
789                 });
790
791                 sess.time("looking_for_derive_registrar", || proc_macro_decls::find(tcx));
792             },
793             {
794                 par_iter(&tcx.hir().krate().modules).for_each(|(&module, _)| {
795                     let local_def_id = tcx.hir().local_def_id(module);
796                     tcx.ensure().check_mod_loops(local_def_id);
797                     tcx.ensure().check_mod_attrs(local_def_id);
798                     tcx.ensure().check_mod_unstable_api_usage(local_def_id);
799                     tcx.ensure().check_mod_const_bodies(local_def_id);
800                 });
801             }
802         );
803     });
804
805     // passes are timed inside typeck
806     typeck::check_crate(tcx)?;
807
808     sess.time("misc_checking_2", || {
809         parallel!(
810             {
811                 sess.time("match_checking", || {
812                     tcx.par_body_owners(|def_id| {
813                         tcx.ensure().check_match(def_id);
814                     });
815                 });
816             },
817             {
818                 sess.time("liveness_and_intrinsic_checking", || {
819                     par_iter(&tcx.hir().krate().modules).for_each(|(&module, _)| {
820                         // this must run before MIR dump, because
821                         // "not all control paths return a value" is reported here.
822                         //
823                         // maybe move the check to a MIR pass?
824                         let local_def_id = tcx.hir().local_def_id(module);
825
826                         tcx.ensure().check_mod_liveness(local_def_id);
827                         tcx.ensure().check_mod_intrinsics(local_def_id);
828                     });
829                 });
830             }
831         );
832     });
833
834     sess.time("MIR_borrow_checking", || {
835         tcx.par_body_owners(|def_id| tcx.ensure().mir_borrowck(def_id));
836     });
837
838     sess.time("dumping_chalk_like_clauses", || {
839         rustc_traits::lowering::dump_program_clauses(tcx);
840     });
841
842     sess.time("MIR_effect_checking", || {
843         for def_id in tcx.body_owners() {
844             mir::transform::check_unsafety::check_unsafety(tcx, def_id)
845         }
846     });
847
848     sess.time("layout_testing", || layout_test::test_layout(tcx));
849
850     // Avoid overwhelming user with errors if borrow checking failed.
851     // I'm not sure how helpful this is, to be honest, but it avoids a
852     // lot of annoying errors in the compile-fail tests (basically,
853     // lint warnings and so on -- kindck used to do this abort, but
854     // kindck is gone now). -nmatsakis
855     if sess.has_errors() {
856         return Err(ErrorReported);
857     }
858
859     sess.time("misc_checking_3", || {
860         parallel!(
861             {
862                 tcx.ensure().privacy_access_levels(LOCAL_CRATE);
863
864                 parallel!(
865                     {
866                         tcx.ensure().check_private_in_public(LOCAL_CRATE);
867                     },
868                     {
869                         sess.time("death_checking", || rustc_passes::dead::check_crate(tcx));
870                     },
871                     {
872                         sess.time("unused_lib_feature_checking", || {
873                             rustc_passes::stability::check_unused_or_stable_features(tcx)
874                         });
875                     },
876                     {
877                         sess.time("lint_checking", || {
878                             rustc_lint::check_crate(tcx, || {
879                                 rustc_lint::BuiltinCombinedLateLintPass::new()
880                             });
881                         });
882                     }
883                 );
884             },
885             {
886                 sess.time("privacy_checking_modules", || {
887                     par_iter(&tcx.hir().krate().modules).for_each(|(&module, _)| {
888                         tcx.ensure().check_mod_privacy(tcx.hir().local_def_id(module));
889                     });
890                 });
891             }
892         );
893     });
894
895     Ok(())
896 }
897
898 fn encode_and_write_metadata(
899     tcx: TyCtxt<'_>,
900     outputs: &OutputFilenames,
901 ) -> (middle::cstore::EncodedMetadata, bool) {
902     #[derive(PartialEq, Eq, PartialOrd, Ord)]
903     enum MetadataKind {
904         None,
905         Uncompressed,
906         Compressed,
907     }
908
909     let metadata_kind = tcx
910         .sess
911         .crate_types
912         .borrow()
913         .iter()
914         .map(|ty| match *ty {
915             CrateType::Executable | CrateType::Staticlib | CrateType::Cdylib => MetadataKind::None,
916
917             CrateType::Rlib => MetadataKind::Uncompressed,
918
919             CrateType::Dylib | CrateType::ProcMacro => MetadataKind::Compressed,
920         })
921         .max()
922         .unwrap_or(MetadataKind::None);
923
924     let metadata = match metadata_kind {
925         MetadataKind::None => middle::cstore::EncodedMetadata::new(),
926         MetadataKind::Uncompressed | MetadataKind::Compressed => tcx.encode_metadata(),
927     };
928
929     let _prof_timer = tcx.sess.prof.generic_activity("write_crate_metadata");
930
931     let need_metadata_file = tcx.sess.opts.output_types.contains_key(&OutputType::Metadata);
932     if need_metadata_file {
933         let crate_name = &tcx.crate_name(LOCAL_CRATE).as_str();
934         let out_filename = filename_for_metadata(tcx.sess, crate_name, outputs);
935         // To avoid races with another rustc process scanning the output directory,
936         // we need to write the file somewhere else and atomically move it to its
937         // final destination, with an `fs::rename` call. In order for the rename to
938         // always succeed, the temporary file needs to be on the same filesystem,
939         // which is why we create it inside the output directory specifically.
940         let metadata_tmpdir = TempFileBuilder::new()
941             .prefix("rmeta")
942             .tempdir_in(out_filename.parent().unwrap())
943             .unwrap_or_else(|err| tcx.sess.fatal(&format!("couldn't create a temp dir: {}", err)));
944         let metadata_filename = emit_metadata(tcx.sess, &metadata, &metadata_tmpdir);
945         if let Err(e) = fs::rename(&metadata_filename, &out_filename) {
946             tcx.sess.fatal(&format!("failed to write {}: {}", out_filename.display(), e));
947         }
948         if tcx.sess.opts.json_artifact_notifications {
949             tcx.sess
950                 .parse_sess
951                 .span_diagnostic
952                 .emit_artifact_notification(&out_filename, "metadata");
953         }
954     }
955
956     let need_metadata_module = metadata_kind == MetadataKind::Compressed;
957
958     (metadata, need_metadata_module)
959 }
960
961 /// Runs the codegen backend, after which the AST and analysis can
962 /// be discarded.
963 pub fn start_codegen<'tcx>(
964     codegen_backend: &dyn CodegenBackend,
965     tcx: TyCtxt<'tcx>,
966     outputs: &OutputFilenames,
967 ) -> Box<dyn Any> {
968     if log_enabled!(::log::Level::Info) {
969         println!("Pre-codegen");
970         tcx.print_debug_stats();
971     }
972
973     let (metadata, need_metadata_module) = encode_and_write_metadata(tcx, outputs);
974
975     let codegen = tcx.sess.time("codegen_crate", move || {
976         codegen_backend.codegen_crate(tcx, metadata, need_metadata_module)
977     });
978
979     if log_enabled!(::log::Level::Info) {
980         println!("Post-codegen");
981         tcx.print_debug_stats();
982     }
983
984     if tcx.sess.opts.output_types.contains_key(&OutputType::Mir) {
985         if let Err(e) = mir::transform::dump_mir::emit_mir(tcx, outputs) {
986             tcx.sess.err(&format!("could not emit MIR: {}", e));
987             tcx.sess.abort_if_errors();
988         }
989     }
990
991     codegen
992 }