]> git.lizzy.rs Git - rust.git/blob - src/librustc_interface/passes.rs
Remove `-Z continue-parse-after-error`
[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("parsing", || 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) = rustc_expand::config::features(
167         krate,
168         &sess.parse_sess,
169         sess.edition(),
170         &sess.opts.debugging_opts.allow_features,
171     );
172     // these need to be set "early" so that expansion sees `quote` if enabled.
173     sess.init_features(features);
174
175     let crate_types = util::collect_crate_types(sess, &krate.attrs);
176     sess.crate_types.set(crate_types);
177
178     let disambiguator = util::compute_crate_disambiguator(sess);
179     sess.crate_disambiguator.set(disambiguator);
180     rustc_incremental::prepare_session_directory(sess, &crate_name, disambiguator);
181
182     if sess.opts.incremental.is_some() {
183         sess.time("garbage-collect incremental cache directory", || {
184             let _prof_timer =
185                 sess.prof.generic_activity("incr_comp_garbage_collect_session_directories");
186             if let Err(e) = rustc_incremental::garbage_collect_session_directories(sess) {
187                 warn!(
188                     "Error while trying to garbage collect incremental \
189                      compilation cache directory: {}",
190                     e
191                 );
192             }
193         });
194     }
195
196     sess.time("recursion limit", || {
197         middle::recursion_limit::update_limits(sess, &krate);
198     });
199
200     let mut lint_store = rustc_lint::new_lint_store(
201         sess.opts.debugging_opts.no_interleave_lints,
202         sess.unstable_options(),
203     );
204     register_lints(&sess, &mut lint_store);
205
206     let registrars =
207         sess.time("plugin loading", || plugin::load::load_plugins(sess, metadata_loader, &krate));
208     sess.time("plugin registration", || {
209         let mut registry = plugin::Registry { lint_store: &mut lint_store };
210         for registrar in registrars {
211             registrar(&mut registry);
212         }
213     });
214
215     Ok((krate, Lrc::new(lint_store)))
216 }
217
218 fn configure_and_expand_inner<'a>(
219     sess: &'a Session,
220     lint_store: &'a lint::LintStore,
221     mut krate: ast::Crate,
222     crate_name: &str,
223     resolver_arenas: &'a ResolverArenas<'a>,
224     metadata_loader: &'a MetadataLoaderDyn,
225 ) -> Result<(ast::Crate, Resolver<'a>)> {
226     sess.time("pre-AST-expansion lint checks", || {
227         rustc_lint::check_ast_crate(
228             sess,
229             lint_store,
230             &krate,
231             true,
232             None,
233             rustc_lint::BuiltinCombinedPreExpansionLintPass::new(),
234         );
235     });
236
237     let mut resolver = Resolver::new(sess, &krate, crate_name, metadata_loader, &resolver_arenas);
238     rustc_builtin_macros::register_builtin_macros(&mut resolver, sess.edition());
239
240     krate = sess.time("crate injection", || {
241         let alt_std_name = sess.opts.alt_std_name.as_ref().map(|s| Symbol::intern(s));
242         let (krate, name) = rustc_builtin_macros::standard_library_imports::inject(
243             krate,
244             &mut resolver,
245             &sess.parse_sess,
246             alt_std_name,
247         );
248         if let Some(name) = name {
249             sess.parse_sess.injected_crate_name.set(name);
250         }
251         krate
252     });
253
254     util::check_attr_crate_type(&krate.attrs, &mut resolver.lint_buffer());
255
256     // Expand all macros
257     krate = sess.time("expansion", || {
258         // Windows dlls do not have rpaths, so they don't know how to find their
259         // dependencies. It's up to us to tell the system where to find all the
260         // dependent dlls. Note that this uses cfg!(windows) as opposed to
261         // targ_cfg because syntax extensions are always loaded for the host
262         // compiler, not for the target.
263         //
264         // This is somewhat of an inherently racy operation, however, as
265         // multiple threads calling this function could possibly continue
266         // extending PATH far beyond what it should. To solve this for now we
267         // just don't add any new elements to PATH which are already there
268         // within PATH. This is basically a targeted fix at #17360 for rustdoc
269         // which runs rustc in parallel but has been seen (#33844) to cause
270         // problems with PATH becoming too long.
271         let mut old_path = OsString::new();
272         if cfg!(windows) {
273             old_path = env::var_os("PATH").unwrap_or(old_path);
274             let mut new_path = sess.host_filesearch(PathKind::All).search_path_dirs();
275             for path in env::split_paths(&old_path) {
276                 if !new_path.contains(&path) {
277                     new_path.push(path);
278                 }
279             }
280             env::set_var(
281                 "PATH",
282                 &env::join_paths(
283                     new_path.iter().filter(|p| env::join_paths(iter::once(p)).is_ok()),
284                 )
285                 .unwrap(),
286             );
287         }
288
289         // Create the config for macro expansion
290         let features = sess.features_untracked();
291         let cfg = rustc_expand::expand::ExpansionConfig {
292             features: Some(&features),
293             recursion_limit: *sess.recursion_limit.get(),
294             trace_mac: sess.opts.debugging_opts.trace_macros,
295             should_test: sess.opts.test,
296             ..rustc_expand::expand::ExpansionConfig::default(crate_name.to_string())
297         };
298
299         let mut ecx = ExtCtxt::new(&sess.parse_sess, cfg, &mut resolver);
300
301         // Expand macros now!
302         let krate = sess.time("expand crate", || ecx.monotonic_expander().expand_crate(krate));
303
304         // The rest is error reporting
305
306         sess.time("check unused macros", || {
307             ecx.check_unused_macros();
308         });
309
310         let mut missing_fragment_specifiers: Vec<_> =
311             ecx.parse_sess.missing_fragment_specifiers.borrow().iter().cloned().collect();
312         missing_fragment_specifiers.sort();
313
314         for span in missing_fragment_specifiers {
315             let lint = lint::builtin::MISSING_FRAGMENT_SPECIFIER;
316             let msg = "missing fragment specifier";
317             resolver.lint_buffer().buffer_lint(lint, ast::CRATE_NODE_ID, span, msg);
318         }
319         if cfg!(windows) {
320             env::set_var("PATH", &old_path);
321         }
322         krate
323     });
324
325     sess.time("maybe building test harness", || {
326         rustc_builtin_macros::test_harness::inject(
327             &sess.parse_sess,
328             &mut resolver,
329             sess.opts.test,
330             &mut krate,
331             sess.diagnostic(),
332             &sess.features_untracked(),
333             sess.panic_strategy(),
334             sess.target.target.options.panic_strategy,
335             sess.opts.debugging_opts.panic_abort_tests,
336         )
337     });
338
339     // If we're actually rustdoc then there's no need to actually compile
340     // anything, so switch everything to just looping
341     let mut should_loop = sess.opts.actually_rustdoc;
342     if let Some(PpMode::PpmSource(PpSourceMode::PpmEveryBodyLoops)) = sess.opts.pretty {
343         should_loop |= true;
344     }
345     if should_loop {
346         util::ReplaceBodyWithLoop::new(&mut resolver).visit_crate(&mut krate);
347     }
348
349     let has_proc_macro_decls = sess.time("AST validation", || {
350         ast_validation::check_crate(sess, &krate, &mut resolver.lint_buffer())
351     });
352
353     let crate_types = sess.crate_types.borrow();
354     let is_proc_macro_crate = crate_types.contains(&config::CrateType::ProcMacro);
355
356     // For backwards compatibility, we don't try to run proc macro injection
357     // if rustdoc is run on a proc macro crate without '--crate-type proc-macro' being
358     // specified. This should only affect users who manually invoke 'rustdoc', as
359     // 'cargo doc' will automatically pass the proper '--crate-type' flags.
360     // However, we do emit a warning, to let such users know that they should
361     // start passing '--crate-type proc-macro'
362     if has_proc_macro_decls && sess.opts.actually_rustdoc && !is_proc_macro_crate {
363         let mut msg = sess.diagnostic().struct_warn(
364             &"Trying to document proc macro crate \
365             without passing '--crate-type proc-macro to rustdoc",
366         );
367
368         msg.warn("The generated documentation may be incorrect");
369         msg.emit()
370     } else {
371         krate = sess.time("maybe creating a macro crate", || {
372             let num_crate_types = crate_types.len();
373             let is_test_crate = sess.opts.test;
374             rustc_builtin_macros::proc_macro_harness::inject(
375                 &sess.parse_sess,
376                 &mut resolver,
377                 krate,
378                 is_proc_macro_crate,
379                 has_proc_macro_decls,
380                 is_test_crate,
381                 num_crate_types,
382                 sess.diagnostic(),
383             )
384         });
385     }
386
387     // Done with macro expansion!
388
389     if sess.opts.debugging_opts.input_stats {
390         println!("Post-expansion node count: {}", count_nodes(&krate));
391     }
392
393     if sess.opts.debugging_opts.hir_stats {
394         hir_stats::print_ast_stats(&krate, "POST EXPANSION AST STATS");
395     }
396
397     if sess.opts.debugging_opts.ast_json {
398         println!("{}", json::as_json(&krate));
399     }
400
401     sess.time("name resolution", || {
402         resolver.resolve_crate(&krate);
403     });
404
405     // Needs to go *after* expansion to be able to check the results of macro expansion.
406     sess.time("complete gated feature checking", || {
407         syntax::feature_gate::check_crate(
408             &krate,
409             &sess.parse_sess,
410             &sess.features_untracked(),
411             sess.opts.unstable_features,
412         );
413     });
414
415     // Add all buffered lints from the `ParseSess` to the `Session`.
416     sess.parse_sess.buffered_lints.with_lock(|buffered_lints| {
417         info!("{} parse sess buffered_lints", buffered_lints.len());
418         for BufferedEarlyLint { id, span, msg, lint_id } in buffered_lints.drain(..) {
419             resolver.lint_buffer().buffer_lint(lint_id, id, span, &msg);
420         }
421     });
422
423     Ok((krate, resolver))
424 }
425
426 pub fn lower_to_hir<'res, 'tcx>(
427     sess: &'tcx Session,
428     lint_store: &lint::LintStore,
429     resolver: &'res mut Resolver<'_>,
430     dep_graph: &'res DepGraph,
431     krate: &'res ast::Crate,
432     arena: &'tcx Arena<'tcx>,
433 ) -> Result<map::Forest<'tcx>> {
434     // Lower AST to HIR.
435     let hir_forest = sess.time("lowering AST -> HIR", || {
436         let hir_crate = rustc_ast_lowering::lower_crate(
437             sess,
438             &dep_graph,
439             &krate,
440             resolver,
441             rustc_parse::nt_to_tokenstream,
442             arena,
443         );
444
445         if sess.opts.debugging_opts.hir_stats {
446             hir_stats::print_hir_stats(&hir_crate);
447         }
448
449         map::Forest::new(hir_crate, &dep_graph)
450     });
451
452     sess.time("early lint checks", || {
453         rustc_lint::check_ast_crate(
454             sess,
455             lint_store,
456             &krate,
457             false,
458             Some(std::mem::take(resolver.lint_buffer())),
459             rustc_lint::BuiltinCombinedEarlyLintPass::new(),
460         )
461     });
462
463     // Discard hygiene data, which isn't required after lowering to HIR.
464     if !sess.opts.debugging_opts.keep_hygiene_data {
465         rustc_span::hygiene::clear_syntax_context_map();
466     }
467
468     Ok(hir_forest)
469 }
470
471 // Returns all the paths that correspond to generated files.
472 fn generated_output_paths(
473     sess: &Session,
474     outputs: &OutputFilenames,
475     exact_name: bool,
476     crate_name: &str,
477 ) -> Vec<PathBuf> {
478     let mut out_filenames = Vec::new();
479     for output_type in sess.opts.output_types.keys() {
480         let file = outputs.path(*output_type);
481         match *output_type {
482             // If the filename has been overridden using `-o`, it will not be modified
483             // by appending `.rlib`, `.exe`, etc., so we can skip this transformation.
484             OutputType::Exe if !exact_name => {
485                 for crate_type in sess.crate_types.borrow().iter() {
486                     let p = ::rustc_codegen_utils::link::filename_for_input(
487                         sess,
488                         *crate_type,
489                         crate_name,
490                         outputs,
491                     );
492                     out_filenames.push(p);
493                 }
494             }
495             OutputType::DepInfo if sess.opts.debugging_opts.dep_info_omit_d_target => {
496                 // Don't add the dep-info output when omitting it from dep-info targets
497             }
498             _ => {
499                 out_filenames.push(file);
500             }
501         }
502     }
503     out_filenames
504 }
505
506 // Runs `f` on every output file path and returns the first non-None result, or None if `f`
507 // returns None for every file path.
508 fn check_output<F, T>(output_paths: &[PathBuf], f: F) -> Option<T>
509 where
510     F: Fn(&PathBuf) -> Option<T>,
511 {
512     for output_path in output_paths {
513         if let Some(result) = f(output_path) {
514             return Some(result);
515         }
516     }
517     None
518 }
519
520 fn output_contains_path(output_paths: &[PathBuf], input_path: &PathBuf) -> bool {
521     let input_path = input_path.canonicalize().ok();
522     if input_path.is_none() {
523         return false;
524     }
525     let check = |output_path: &PathBuf| {
526         if output_path.canonicalize().ok() == input_path { Some(()) } else { None }
527     };
528     check_output(output_paths, check).is_some()
529 }
530
531 fn output_conflicts_with_dir(output_paths: &[PathBuf]) -> Option<PathBuf> {
532     let check = |output_path: &PathBuf| output_path.is_dir().then(|| output_path.clone());
533     check_output(output_paths, check)
534 }
535
536 fn escape_dep_filename(filename: &FileName) -> String {
537     // Apparently clang and gcc *only* escape spaces:
538     // http://llvm.org/klaus/clang/commit/9d50634cfc268ecc9a7250226dd5ca0e945240d4
539     filename.to_string().replace(" ", "\\ ")
540 }
541
542 fn write_out_deps(
543     sess: &Session,
544     boxed_resolver: &Steal<Rc<RefCell<BoxedResolver>>>,
545     outputs: &OutputFilenames,
546     out_filenames: &[PathBuf],
547 ) {
548     // Write out dependency rules to the dep-info file if requested
549     if !sess.opts.output_types.contains_key(&OutputType::DepInfo) {
550         return;
551     }
552     let deps_filename = outputs.path(OutputType::DepInfo);
553
554     let result = (|| -> io::Result<()> {
555         // Build a list of files used to compile the output and
556         // write Makefile-compatible dependency rules
557         let mut files: Vec<String> = sess
558             .source_map()
559             .files()
560             .iter()
561             .filter(|fmap| fmap.is_real_file())
562             .filter(|fmap| !fmap.is_imported())
563             .map(|fmap| escape_dep_filename(&fmap.unmapped_path.as_ref().unwrap_or(&fmap.name)))
564             .collect();
565
566         if sess.binary_dep_depinfo() {
567             boxed_resolver.borrow().borrow_mut().access(|resolver| {
568                 for cnum in resolver.cstore().crates_untracked() {
569                     let source = resolver.cstore().crate_source_untracked(cnum);
570                     if let Some((path, _)) = source.dylib {
571                         files.push(escape_dep_filename(&FileName::Real(path)));
572                     }
573                     if let Some((path, _)) = source.rlib {
574                         files.push(escape_dep_filename(&FileName::Real(path)));
575                     }
576                     if let Some((path, _)) = source.rmeta {
577                         files.push(escape_dep_filename(&FileName::Real(path)));
578                     }
579                 }
580             });
581         }
582
583         let mut file = fs::File::create(&deps_filename)?;
584         for path in out_filenames {
585             writeln!(file, "{}: {}\n", path.display(), files.join(" "))?;
586         }
587
588         // Emit a fake target for each input file to the compilation. This
589         // prevents `make` from spitting out an error if a file is later
590         // deleted. For more info see #28735
591         for path in files {
592             writeln!(file, "{}:", path)?;
593         }
594         Ok(())
595     })();
596
597     match result {
598         Ok(_) => {
599             if sess.opts.json_artifact_notifications {
600                 sess.parse_sess
601                     .span_diagnostic
602                     .emit_artifact_notification(&deps_filename, "dep-info");
603             }
604         }
605         Err(e) => sess.fatal(&format!(
606             "error writing dependencies to `{}`: {}",
607             deps_filename.display(),
608             e
609         )),
610     }
611 }
612
613 pub fn prepare_outputs(
614     sess: &Session,
615     compiler: &Compiler,
616     krate: &ast::Crate,
617     boxed_resolver: &Steal<Rc<RefCell<BoxedResolver>>>,
618     crate_name: &str,
619 ) -> Result<OutputFilenames> {
620     // FIXME: rustdoc passes &[] instead of &krate.attrs here
621     let outputs = util::build_output_filenames(
622         &compiler.input,
623         &compiler.output_dir,
624         &compiler.output_file,
625         &krate.attrs,
626         sess,
627     );
628
629     let output_paths =
630         generated_output_paths(sess, &outputs, compiler.output_file.is_some(), &crate_name);
631
632     // Ensure the source file isn't accidentally overwritten during compilation.
633     if let Some(ref input_path) = compiler.input_path {
634         if sess.opts.will_create_output_file() {
635             if output_contains_path(&output_paths, input_path) {
636                 sess.err(&format!(
637                     "the input file \"{}\" would be overwritten by the generated \
638                         executable",
639                     input_path.display()
640                 ));
641                 return Err(ErrorReported);
642             }
643             if let Some(dir_path) = output_conflicts_with_dir(&output_paths) {
644                 sess.err(&format!(
645                     "the generated executable for the input file \"{}\" conflicts with the \
646                         existing directory \"{}\"",
647                     input_path.display(),
648                     dir_path.display()
649                 ));
650                 return Err(ErrorReported);
651             }
652         }
653     }
654
655     write_out_deps(sess, boxed_resolver, &outputs, &output_paths);
656
657     let only_dep_info = sess.opts.output_types.contains_key(&OutputType::DepInfo)
658         && sess.opts.output_types.len() == 1;
659
660     if !only_dep_info {
661         if let Some(ref dir) = compiler.output_dir {
662             if fs::create_dir_all(dir).is_err() {
663                 sess.err("failed to find or create the directory specified by `--out-dir`");
664                 return Err(ErrorReported);
665             }
666         }
667     }
668
669     Ok(outputs)
670 }
671
672 pub fn default_provide(providers: &mut ty::query::Providers<'_>) {
673     providers.analysis = analysis;
674     proc_macro_decls::provide(providers);
675     plugin::build::provide(providers);
676     rustc::hir::provide(providers);
677     mir::provide(providers);
678     rustc_privacy::provide(providers);
679     typeck::provide(providers);
680     ty::provide(providers);
681     traits::provide(providers);
682     rustc_passes::provide(providers);
683     rustc_resolve::provide(providers);
684     rustc_traits::provide(providers);
685     rustc_metadata::provide(providers);
686     rustc_lint::provide(providers);
687     rustc_codegen_utils::provide(providers);
688     rustc_codegen_ssa::provide(providers);
689 }
690
691 pub fn default_provide_extern(providers: &mut ty::query::Providers<'_>) {
692     rustc_metadata::provide_extern(providers);
693     rustc_codegen_ssa::provide_extern(providers);
694 }
695
696 pub struct QueryContext<'tcx>(&'tcx GlobalCtxt<'tcx>);
697
698 impl<'tcx> QueryContext<'tcx> {
699     pub fn enter<F, R>(&mut self, f: F) -> R
700     where
701         F: FnOnce(TyCtxt<'tcx>) -> R,
702     {
703         ty::tls::enter_global(self.0, |tcx| f(tcx))
704     }
705
706     pub fn print_stats(&self) {
707         self.0.queries.print_stats()
708     }
709 }
710
711 pub fn create_global_ctxt<'tcx>(
712     compiler: &'tcx Compiler,
713     lint_store: Lrc<lint::LintStore>,
714     hir_forest: &'tcx map::Forest<'tcx>,
715     mut resolver_outputs: ResolverOutputs,
716     outputs: OutputFilenames,
717     crate_name: &str,
718     global_ctxt: &'tcx Once<GlobalCtxt<'tcx>>,
719     all_arenas: &'tcx AllArenas,
720     arena: &'tcx WorkerLocal<Arena<'tcx>>,
721 ) -> QueryContext<'tcx> {
722     let sess = &compiler.session();
723     let defs = mem::take(&mut resolver_outputs.definitions);
724
725     // Construct the HIR map.
726     let hir_map = sess.time("indexing HIR", || {
727         map::map_crate(sess, &*resolver_outputs.cstore, &hir_forest, defs)
728     });
729
730     let query_result_on_disk_cache =
731         sess.time("load query result cache", || rustc_incremental::load_query_result_cache(sess));
732
733     let codegen_backend = compiler.codegen_backend();
734     let mut local_providers = ty::query::Providers::default();
735     default_provide(&mut local_providers);
736     codegen_backend.provide(&mut local_providers);
737
738     let mut extern_providers = local_providers;
739     default_provide_extern(&mut extern_providers);
740     codegen_backend.provide_extern(&mut extern_providers);
741
742     if let Some(callback) = compiler.override_queries {
743         callback(sess, &mut local_providers, &mut extern_providers);
744     }
745
746     let gcx = global_ctxt.init_locking(|| {
747         TyCtxt::create_global_ctxt(
748             sess,
749             lint_store,
750             local_providers,
751             extern_providers,
752             &all_arenas,
753             arena,
754             resolver_outputs,
755             hir_map,
756             query_result_on_disk_cache,
757             &crate_name,
758             &outputs,
759         )
760     });
761
762     // Do some initialization of the DepGraph that can only be done with the tcx available.
763     ty::tls::enter_global(&gcx, |tcx| {
764         tcx.sess.time("dep graph tcx init", || rustc_incremental::dep_graph_tcx_init(tcx));
765     });
766
767     QueryContext(gcx)
768 }
769
770 /// Runs the resolution, type-checking, region checking and other
771 /// miscellaneous analysis passes on the crate.
772 fn analysis(tcx: TyCtxt<'_>, cnum: CrateNum) -> Result<()> {
773     assert_eq!(cnum, LOCAL_CRATE);
774
775     let sess = tcx.sess;
776     let mut entry_point = None;
777
778     sess.time("misc checking 1", || {
779         parallel!(
780             {
781                 entry_point = sess
782                     .time("looking for entry point", || rustc_passes::entry::find_entry_point(tcx));
783
784                 sess.time("looking for plugin registrar", || {
785                     plugin::build::find_plugin_registrar(tcx)
786                 });
787
788                 sess.time("looking for derive registrar", || proc_macro_decls::find(tcx));
789             },
790             {
791                 par_iter(&tcx.hir().krate().modules).for_each(|(&module, _)| {
792                     let local_def_id = tcx.hir().local_def_id(module);
793                     tcx.ensure().check_mod_loops(local_def_id);
794                     tcx.ensure().check_mod_attrs(local_def_id);
795                     tcx.ensure().check_mod_unstable_api_usage(local_def_id);
796                     tcx.ensure().check_mod_const_bodies(local_def_id);
797                 });
798             }
799         );
800     });
801
802     // passes are timed inside typeck
803     typeck::check_crate(tcx)?;
804
805     sess.time("misc checking 2", || {
806         parallel!(
807             {
808                 sess.time("match checking", || {
809                     tcx.par_body_owners(|def_id| {
810                         tcx.ensure().check_match(def_id);
811                     });
812                 });
813             },
814             {
815                 sess.time("liveness checking + intrinsic checking", || {
816                     par_iter(&tcx.hir().krate().modules).for_each(|(&module, _)| {
817                         // this must run before MIR dump, because
818                         // "not all control paths return a value" is reported here.
819                         //
820                         // maybe move the check to a MIR pass?
821                         let local_def_id = tcx.hir().local_def_id(module);
822
823                         tcx.ensure().check_mod_liveness(local_def_id);
824                         tcx.ensure().check_mod_intrinsics(local_def_id);
825                     });
826                 });
827             }
828         );
829     });
830
831     sess.time("MIR borrow checking", || {
832         tcx.par_body_owners(|def_id| tcx.ensure().mir_borrowck(def_id));
833     });
834
835     sess.time("dumping Chalk-like clauses", || {
836         rustc_traits::lowering::dump_program_clauses(tcx);
837     });
838
839     sess.time("MIR effect checking", || {
840         for def_id in tcx.body_owners() {
841             mir::transform::check_unsafety::check_unsafety(tcx, def_id)
842         }
843     });
844
845     sess.time("layout testing", || layout_test::test_layout(tcx));
846
847     // Avoid overwhelming user with errors if borrow checking failed.
848     // I'm not sure how helpful this is, to be honest, but it avoids a
849     // lot of annoying errors in the compile-fail tests (basically,
850     // lint warnings and so on -- kindck used to do this abort, but
851     // kindck is gone now). -nmatsakis
852     if sess.has_errors() {
853         return Err(ErrorReported);
854     }
855
856     sess.time("misc checking 3", || {
857         parallel!(
858             {
859                 sess.time("privacy access levels", || {
860                     tcx.ensure().privacy_access_levels(LOCAL_CRATE);
861                 });
862                 parallel!(
863                     {
864                         sess.time("private in public", || {
865                             tcx.ensure().check_private_in_public(LOCAL_CRATE);
866                         });
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 need_metadata_file = tcx.sess.opts.output_types.contains_key(&OutputType::Metadata);
930     if need_metadata_file {
931         let crate_name = &tcx.crate_name(LOCAL_CRATE).as_str();
932         let out_filename = filename_for_metadata(tcx.sess, crate_name, outputs);
933         // To avoid races with another rustc process scanning the output directory,
934         // we need to write the file somewhere else and atomically move it to its
935         // final destination, with an `fs::rename` call. In order for the rename to
936         // always succeed, the temporary file needs to be on the same filesystem,
937         // which is why we create it inside the output directory specifically.
938         let metadata_tmpdir = TempFileBuilder::new()
939             .prefix("rmeta")
940             .tempdir_in(out_filename.parent().unwrap())
941             .unwrap_or_else(|err| tcx.sess.fatal(&format!("couldn't create a temp dir: {}", err)));
942         let metadata_filename = emit_metadata(tcx.sess, &metadata, &metadata_tmpdir);
943         if let Err(e) = fs::rename(&metadata_filename, &out_filename) {
944             tcx.sess.fatal(&format!("failed to write {}: {}", out_filename.display(), e));
945         }
946         if tcx.sess.opts.json_artifact_notifications {
947             tcx.sess
948                 .parse_sess
949                 .span_diagnostic
950                 .emit_artifact_notification(&out_filename, "metadata");
951         }
952     }
953
954     let need_metadata_module = metadata_kind == MetadataKind::Compressed;
955
956     (metadata, need_metadata_module)
957 }
958
959 /// Runs the codegen backend, after which the AST and analysis can
960 /// be discarded.
961 pub fn start_codegen<'tcx>(
962     codegen_backend: &dyn CodegenBackend,
963     tcx: TyCtxt<'tcx>,
964     outputs: &OutputFilenames,
965 ) -> Box<dyn Any> {
966     if log_enabled!(::log::Level::Info) {
967         println!("Pre-codegen");
968         tcx.print_debug_stats();
969     }
970
971     let (metadata, need_metadata_module) =
972         tcx.sess.time("metadata encoding and writing", || encode_and_write_metadata(tcx, outputs));
973
974     let codegen = tcx.sess.time("codegen", move || {
975         codegen_backend.codegen_crate(tcx, metadata, need_metadata_module)
976     });
977
978     if log_enabled!(::log::Level::Info) {
979         println!("Post-codegen");
980         tcx.print_debug_stats();
981     }
982
983     if tcx.sess.opts.output_types.contains_key(&OutputType::Mir) {
984         if let Err(e) = mir::transform::dump_mir::emit_mir(tcx, outputs) {
985             tcx.sess.err(&format!("could not emit MIR: {}", e));
986             tcx.sess.abort_if_errors();
987         }
988     }
989
990     codegen
991 }