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