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