]> git.lizzy.rs Git - rust.git/blob - src/librustc_driver/driver.rs
Replace CrateAnalysis::access_levels with query
[rust.git] / src / librustc_driver / driver.rs
1 use rustc::dep_graph::DepGraph;
2 use rustc::hir;
3 use rustc::hir::lowering::lower_crate;
4 use rustc::hir::map as hir_map;
5 use rustc::lint;
6 use rustc::middle::{self, reachable, resolve_lifetime, stability};
7 use rustc::ty::{self, AllArenas, Resolutions, TyCtxt};
8 use rustc::traits;
9 use rustc::util::common::{install_panic_hook, time, ErrorReported};
10 use rustc::util::profiling::ProfileCategory;
11 use rustc::session::{CompileResult, CrateDisambiguator, Session};
12 use rustc::session::CompileIncomplete;
13 use rustc::session::config::{self, Input, OutputFilenames, OutputType};
14 use rustc::session::search_paths::PathKind;
15 use rustc_allocator as allocator;
16 use rustc_borrowck as borrowck;
17 use rustc_codegen_utils::codegen_backend::CodegenBackend;
18 use rustc_data_structures::fingerprint::Fingerprint;
19 use rustc_data_structures::stable_hasher::StableHasher;
20 use rustc_data_structures::sync::{self, Lock};
21 use rustc_incremental;
22 use rustc_metadata::creader::CrateLoader;
23 use rustc_metadata::cstore::{self, CStore};
24 use rustc_mir as mir;
25 use rustc_passes::{self, ast_validation, hir_stats, loops, rvalue_promotion};
26 use rustc_plugin as plugin;
27 use rustc_plugin::registry::Registry;
28 use rustc_privacy;
29 use rustc_resolve::{MakeGlobMap, Resolver, ResolverArenas};
30 use rustc_traits;
31 use rustc_typeck as typeck;
32 use syntax::{self, ast, attr, diagnostics, visit};
33 use syntax::early_buffered_lints::BufferedEarlyLint;
34 use syntax::ext::base::ExtCtxt;
35 use syntax::fold::Folder;
36 use syntax::parse::{self, PResult};
37 use syntax::util::node_count::NodeCounter;
38 use syntax::util::lev_distance::find_best_match_for_name;
39 use syntax::symbol::Symbol;
40 use syntax_pos::{FileName, hygiene};
41 use syntax_ext;
42
43 use serialize::json;
44
45 use std::any::Any;
46 use std::env;
47 use std::ffi::OsString;
48 use std::fs;
49 use std::io::{self, Write};
50 use std::iter;
51 use std::path::{Path, PathBuf};
52 use std::sync::mpsc;
53
54 use pretty::ReplaceBodyWithLoop;
55 use proc_macro_decls;
56 use profile;
57 use super::Compilation;
58
59 #[cfg(not(parallel_queries))]
60 pub fn spawn_thread_pool<F: FnOnce(config::Options) -> R + sync::Send, R: sync::Send>(
61     opts: config::Options,
62     f: F
63 ) -> R {
64     ty::tls::GCX_PTR.set(&Lock::new(0), || {
65         f(opts)
66     })
67 }
68
69 #[cfg(parallel_queries)]
70 pub fn spawn_thread_pool<F: FnOnce(config::Options) -> R + sync::Send, R: sync::Send>(
71     opts: config::Options,
72     f: F
73 ) -> R {
74     use syntax;
75     use syntax_pos;
76     use rayon::{ThreadPoolBuilder, ThreadPool};
77
78     let gcx_ptr = &Lock::new(0);
79
80     let config = ThreadPoolBuilder::new()
81         .num_threads(Session::query_threads_from_opts(&opts))
82         .deadlock_handler(|| unsafe { ty::query::handle_deadlock() })
83         .stack_size(::STACK_SIZE);
84
85     let with_pool = move |pool: &ThreadPool| {
86         pool.install(move || f(opts))
87     };
88
89     syntax::GLOBALS.with(|syntax_globals| {
90         syntax_pos::GLOBALS.with(|syntax_pos_globals| {
91             // The main handler run for each Rayon worker thread and sets up
92             // the thread local rustc uses. syntax_globals and syntax_pos_globals are
93             // captured and set on the new threads. ty::tls::with_thread_locals sets up
94             // thread local callbacks from libsyntax
95             let main_handler = move |worker: &mut dyn FnMut()| {
96                 syntax::GLOBALS.set(syntax_globals, || {
97                     syntax_pos::GLOBALS.set(syntax_pos_globals, || {
98                         ty::tls::with_thread_locals(|| {
99                             ty::tls::GCX_PTR.set(gcx_ptr, || {
100                                 worker()
101                             })
102                         })
103                     })
104                 })
105             };
106
107             ThreadPool::scoped_pool(config, main_handler, with_pool).unwrap()
108         })
109     })
110 }
111
112 pub fn compile_input(
113     codegen_backend: Box<dyn CodegenBackend>,
114     sess: &Session,
115     cstore: &CStore,
116     input_path: &Option<PathBuf>,
117     input: &Input,
118     outdir: &Option<PathBuf>,
119     output: &Option<PathBuf>,
120     addl_plugins: Option<Vec<String>>,
121     control: &CompileController,
122 ) -> CompileResult {
123     macro_rules! controller_entry_point {
124         ($point: ident, $tsess: expr, $make_state: expr, $phase_result: expr) => {{
125             let state = &mut $make_state;
126             let phase_result: &CompileResult = &$phase_result;
127             if phase_result.is_ok() || control.$point.run_callback_on_error {
128                 (control.$point.callback)(state);
129             }
130
131             if control.$point.stop == Compilation::Stop {
132                 // FIXME: shouldn't this return Err(CompileIncomplete::Stopped)
133                 // if there are no errors?
134                 return $tsess.compile_status();
135             }
136         }}
137     }
138
139     if sess.profile_queries() {
140         profile::begin(sess);
141     }
142
143     // We need nested scopes here, because the intermediate results can keep
144     // large chunks of memory alive and we want to free them as soon as
145     // possible to keep the peak memory usage low
146     let (outputs, ongoing_codegen, dep_graph) = {
147         let krate = match phase_1_parse_input(control, sess, input) {
148             Ok(krate) => krate,
149             Err(mut parse_error) => {
150                 parse_error.emit();
151                 return Err(CompileIncomplete::Errored(ErrorReported));
152             }
153         };
154
155         let (krate, registry) = {
156             let mut compile_state =
157                 CompileState::state_after_parse(input, sess, outdir, output, krate, &cstore);
158             controller_entry_point!(after_parse, sess, compile_state, Ok(()));
159
160             (compile_state.krate.unwrap(), compile_state.registry)
161         };
162
163         let outputs = build_output_filenames(input, outdir, output, &krate.attrs, sess);
164         let crate_name =
165             ::rustc_codegen_utils::link::find_crate_name(Some(sess), &krate.attrs, input);
166         install_panic_hook();
167
168         let ExpansionResult {
169             expanded_crate,
170             defs,
171             analysis,
172             resolutions,
173             mut hir_forest,
174         } = {
175             phase_2_configure_and_expand(
176                 sess,
177                 &cstore,
178                 krate,
179                 registry,
180                 &crate_name,
181                 addl_plugins,
182                 control.make_glob_map,
183                 |expanded_crate| {
184                     let mut state = CompileState::state_after_expand(
185                         input,
186                         sess,
187                         outdir,
188                         output,
189                         &cstore,
190                         expanded_crate,
191                         &crate_name,
192                     );
193                     controller_entry_point!(after_expand, sess, state, Ok(()));
194                     Ok(())
195                 },
196             )?
197         };
198
199         let output_paths = generated_output_paths(sess, &outputs, output.is_some(), &crate_name);
200
201         // Ensure the source file isn't accidentally overwritten during compilation.
202         if let Some(ref input_path) = *input_path {
203             if sess.opts.will_create_output_file() {
204                 if output_contains_path(&output_paths, input_path) {
205                     sess.err(&format!(
206                         "the input file \"{}\" would be overwritten by the generated \
207                          executable",
208                         input_path.display()
209                     ));
210                     return Err(CompileIncomplete::Stopped);
211                 }
212                 if let Some(dir_path) = output_conflicts_with_dir(&output_paths) {
213                     sess.err(&format!(
214                         "the generated executable for the input file \"{}\" conflicts with the \
215                          existing directory \"{}\"",
216                         input_path.display(),
217                         dir_path.display()
218                     ));
219                     return Err(CompileIncomplete::Stopped);
220                 }
221             }
222         }
223
224         write_out_deps(sess, &outputs, &output_paths);
225         if sess.opts.output_types.contains_key(&OutputType::DepInfo)
226             && sess.opts.output_types.len() == 1
227         {
228             return Ok(());
229         }
230
231         if let &Some(ref dir) = outdir {
232             if fs::create_dir_all(dir).is_err() {
233                 sess.err("failed to find or create the directory specified by --out-dir");
234                 return Err(CompileIncomplete::Stopped);
235             }
236         }
237
238         // Construct the HIR map
239         let hir_map = time(sess, "indexing hir", || {
240             hir_map::map_crate(sess, cstore, &mut hir_forest, &defs)
241         });
242
243         {
244             hir_map.dep_graph.assert_ignored();
245             controller_entry_point!(
246                 after_hir_lowering,
247                 sess,
248                 CompileState::state_after_hir_lowering(
249                     input,
250                     sess,
251                     outdir,
252                     output,
253                     &cstore,
254                     &hir_map,
255                     &analysis,
256                     &resolutions,
257                     &expanded_crate,
258                     &hir_map.krate(),
259                     &outputs,
260                     &crate_name
261                 ),
262                 Ok(())
263             );
264         }
265
266         let opt_crate = if control.keep_ast {
267             Some(&expanded_crate)
268         } else {
269             drop(expanded_crate);
270             None
271         };
272
273         let mut arenas = AllArenas::new();
274
275         phase_3_run_analysis_passes(
276             &*codegen_backend,
277             control,
278             sess,
279             cstore,
280             hir_map,
281             analysis,
282             resolutions,
283             &mut arenas,
284             &crate_name,
285             &outputs,
286             |tcx, analysis, rx, result| {
287                 {
288                     // Eventually, we will want to track plugins.
289                     tcx.dep_graph.with_ignore(|| {
290                         let mut state = CompileState::state_after_analysis(
291                             input,
292                             sess,
293                             outdir,
294                             output,
295                             opt_crate,
296                             tcx.hir().krate(),
297                             &analysis,
298                             tcx,
299                             &crate_name,
300                         );
301                         (control.after_analysis.callback)(&mut state);
302                     });
303
304                     if control.after_analysis.stop == Compilation::Stop {
305                         return result.and_then(|_| Err(CompileIncomplete::Stopped));
306                     }
307                 }
308
309                 result?;
310
311                 if log_enabled!(::log::Level::Info) {
312                     println!("Pre-codegen");
313                     tcx.print_debug_stats();
314                 }
315
316                 let ongoing_codegen = phase_4_codegen(&*codegen_backend, tcx, rx);
317
318                 if log_enabled!(::log::Level::Info) {
319                     println!("Post-codegen");
320                     tcx.print_debug_stats();
321                 }
322
323                 if tcx.sess.opts.output_types.contains_key(&OutputType::Mir) {
324                     if let Err(e) = mir::transform::dump_mir::emit_mir(tcx, &outputs) {
325                         sess.err(&format!("could not emit MIR: {}", e));
326                         sess.abort_if_errors();
327                     }
328                 }
329
330                 if tcx.sess.opts.debugging_opts.query_stats {
331                     tcx.queries.print_stats();
332                 }
333
334                 Ok((outputs.clone(), ongoing_codegen, tcx.dep_graph.clone()))
335             },
336         )??
337     };
338
339     if sess.opts.debugging_opts.print_type_sizes {
340         sess.code_stats.borrow().print_type_sizes();
341     }
342
343     codegen_backend.join_codegen_and_link(ongoing_codegen, sess, &dep_graph, &outputs)?;
344
345     if sess.opts.debugging_opts.perf_stats {
346         sess.print_perf_stats();
347     }
348
349     if sess.opts.debugging_opts.self_profile {
350         sess.print_profiler_results();
351     }
352
353     if sess.opts.debugging_opts.profile_json {
354         sess.save_json_results();
355     }
356
357     controller_entry_point!(
358         compilation_done,
359         sess,
360         CompileState::state_when_compilation_done(input, sess, outdir, output),
361         Ok(())
362     );
363
364     Ok(())
365 }
366
367 pub fn source_name(input: &Input) -> FileName {
368     match *input {
369         Input::File(ref ifile) => ifile.clone().into(),
370         Input::Str { ref name, .. } => name.clone(),
371     }
372 }
373
374 /// CompileController is used to customize compilation, it allows compilation to
375 /// be stopped and/or to call arbitrary code at various points in compilation.
376 /// It also allows for various flags to be set to influence what information gets
377 /// collected during compilation.
378 ///
379 /// This is a somewhat higher level controller than a Session - the Session
380 /// controls what happens in each phase, whereas the CompileController controls
381 /// whether a phase is run at all and whether other code (from outside the
382 /// compiler) is run between phases.
383 ///
384 /// Note that if compilation is set to stop and a callback is provided for a
385 /// given entry point, the callback is called before compilation is stopped.
386 ///
387 /// Expect more entry points to be added in the future.
388 pub struct CompileController<'a> {
389     pub after_parse: PhaseController<'a>,
390     pub after_expand: PhaseController<'a>,
391     pub after_hir_lowering: PhaseController<'a>,
392     pub after_analysis: PhaseController<'a>,
393     pub compilation_done: PhaseController<'a>,
394
395     // FIXME we probably want to group the below options together and offer a
396     // better API, rather than this ad-hoc approach.
397     pub make_glob_map: MakeGlobMap,
398     // Whether the compiler should keep the ast beyond parsing.
399     pub keep_ast: bool,
400     // -Zcontinue-parse-after-error
401     pub continue_parse_after_error: bool,
402
403     /// Allows overriding default rustc query providers,
404     /// after `default_provide` has installed them.
405     pub provide: Box<dyn Fn(&mut ty::query::Providers) + 'a>,
406     /// Same as `provide`, but only for non-local crates,
407     /// applied after `default_provide_extern`.
408     pub provide_extern: Box<dyn Fn(&mut ty::query::Providers) + 'a>,
409 }
410
411 impl<'a> CompileController<'a> {
412     pub fn basic() -> CompileController<'a> {
413         CompileController {
414             after_parse: PhaseController::basic(),
415             after_expand: PhaseController::basic(),
416             after_hir_lowering: PhaseController::basic(),
417             after_analysis: PhaseController::basic(),
418             compilation_done: PhaseController::basic(),
419             make_glob_map: MakeGlobMap::No,
420             keep_ast: false,
421             continue_parse_after_error: false,
422             provide: box |_| {},
423             provide_extern: box |_| {},
424         }
425     }
426 }
427
428 /// This implementation makes it easier to create a custom driver when you only want to hook
429 /// into callbacks from `CompileController`.
430 ///
431 /// # Example
432 ///
433 /// ```no_run
434 /// # extern crate rustc_driver;
435 /// # use rustc_driver::driver::CompileController;
436 /// let mut controller = CompileController::basic();
437 /// controller.after_analysis.callback = Box::new(move |_state| {});
438 /// rustc_driver::run_compiler(&[], Box::new(controller), None, None);
439 /// ```
440 impl<'a> ::CompilerCalls<'a> for CompileController<'a> {
441     fn early_callback(
442         &mut self,
443         matches: &::getopts::Matches,
444         sopts: &config::Options,
445         cfg: &ast::CrateConfig,
446         descriptions: &::errors::registry::Registry,
447         output: ::ErrorOutputType,
448     ) -> Compilation {
449         ::RustcDefaultCalls.early_callback(
450             matches,
451             sopts,
452             cfg,
453             descriptions,
454             output,
455         )
456     }
457     fn no_input(
458         &mut self,
459         matches: &::getopts::Matches,
460         sopts: &config::Options,
461         cfg: &ast::CrateConfig,
462         odir: &Option<PathBuf>,
463         ofile: &Option<PathBuf>,
464         descriptions: &::errors::registry::Registry,
465     ) -> Option<(Input, Option<PathBuf>)> {
466         ::RustcDefaultCalls.no_input(
467             matches,
468             sopts,
469             cfg,
470             odir,
471             ofile,
472             descriptions,
473         )
474     }
475     fn late_callback(
476         &mut self,
477         codegen_backend: &dyn (::CodegenBackend),
478         matches: &::getopts::Matches,
479         sess: &Session,
480         cstore: &CStore,
481         input: &Input,
482         odir: &Option<PathBuf>,
483         ofile: &Option<PathBuf>,
484     ) -> Compilation {
485         ::RustcDefaultCalls
486             .late_callback(codegen_backend, matches, sess, cstore, input, odir, ofile)
487     }
488     fn build_controller(
489         self: Box<Self>,
490         _: &Session,
491         _: &::getopts::Matches
492     ) -> CompileController<'a> {
493         *self
494     }
495 }
496
497 pub struct PhaseController<'a> {
498     pub stop: Compilation,
499     // If true then the compiler will try to run the callback even if the phase
500     // ends with an error. Note that this is not always possible.
501     pub run_callback_on_error: bool,
502     pub callback: Box<dyn Fn(&mut CompileState) + 'a>,
503 }
504
505 impl<'a> PhaseController<'a> {
506     pub fn basic() -> PhaseController<'a> {
507         PhaseController {
508             stop: Compilation::Continue,
509             run_callback_on_error: false,
510             callback: box |_| {},
511         }
512     }
513 }
514
515 /// State that is passed to a callback. What state is available depends on when
516 /// during compilation the callback is made. See the various constructor methods
517 /// (`state_*`) in the impl to see which data is provided for any given entry point.
518 pub struct CompileState<'a, 'tcx: 'a> {
519     pub input: &'a Input,
520     pub session: &'tcx Session,
521     pub krate: Option<ast::Crate>,
522     pub registry: Option<Registry<'a>>,
523     pub cstore: Option<&'tcx CStore>,
524     pub crate_name: Option<&'a str>,
525     pub output_filenames: Option<&'a OutputFilenames>,
526     pub out_dir: Option<&'a Path>,
527     pub out_file: Option<&'a Path>,
528     pub expanded_crate: Option<&'a ast::Crate>,
529     pub hir_crate: Option<&'a hir::Crate>,
530     pub hir_map: Option<&'a hir_map::Map<'tcx>>,
531     pub resolutions: Option<&'a Resolutions>,
532     pub analysis: Option<&'a ty::CrateAnalysis>,
533     pub tcx: Option<TyCtxt<'a, 'tcx, 'tcx>>,
534 }
535
536 impl<'a, 'tcx> CompileState<'a, 'tcx> {
537     fn empty(input: &'a Input, session: &'tcx Session, out_dir: &'a Option<PathBuf>) -> Self {
538         CompileState {
539             input,
540             session,
541             out_dir: out_dir.as_ref().map(|s| &**s),
542             out_file: None,
543             krate: None,
544             registry: None,
545             cstore: None,
546             crate_name: None,
547             output_filenames: None,
548             expanded_crate: None,
549             hir_crate: None,
550             hir_map: None,
551             resolutions: None,
552             analysis: None,
553             tcx: None,
554         }
555     }
556
557     fn state_after_parse(
558         input: &'a Input,
559         session: &'tcx Session,
560         out_dir: &'a Option<PathBuf>,
561         out_file: &'a Option<PathBuf>,
562         krate: ast::Crate,
563         cstore: &'tcx CStore,
564     ) -> Self {
565         CompileState {
566             // Initialize the registry before moving `krate`
567             registry: Some(Registry::new(&session, krate.span)),
568             krate: Some(krate),
569             cstore: Some(cstore),
570             out_file: out_file.as_ref().map(|s| &**s),
571             ..CompileState::empty(input, session, out_dir)
572         }
573     }
574
575     fn state_after_expand(
576         input: &'a Input,
577         session: &'tcx Session,
578         out_dir: &'a Option<PathBuf>,
579         out_file: &'a Option<PathBuf>,
580         cstore: &'tcx CStore,
581         expanded_crate: &'a ast::Crate,
582         crate_name: &'a str,
583     ) -> Self {
584         CompileState {
585             crate_name: Some(crate_name),
586             cstore: Some(cstore),
587             expanded_crate: Some(expanded_crate),
588             out_file: out_file.as_ref().map(|s| &**s),
589             ..CompileState::empty(input, session, out_dir)
590         }
591     }
592
593     fn state_after_hir_lowering(
594         input: &'a Input,
595         session: &'tcx Session,
596         out_dir: &'a Option<PathBuf>,
597         out_file: &'a Option<PathBuf>,
598         cstore: &'tcx CStore,
599         hir_map: &'a hir_map::Map<'tcx>,
600         analysis: &'a ty::CrateAnalysis,
601         resolutions: &'a Resolutions,
602         krate: &'a ast::Crate,
603         hir_crate: &'a hir::Crate,
604         output_filenames: &'a OutputFilenames,
605         crate_name: &'a str,
606     ) -> Self {
607         CompileState {
608             crate_name: Some(crate_name),
609             cstore: Some(cstore),
610             hir_map: Some(hir_map),
611             analysis: Some(analysis),
612             resolutions: Some(resolutions),
613             expanded_crate: Some(krate),
614             hir_crate: Some(hir_crate),
615             output_filenames: Some(output_filenames),
616             out_file: out_file.as_ref().map(|s| &**s),
617             ..CompileState::empty(input, session, out_dir)
618         }
619     }
620
621     fn state_after_analysis(
622         input: &'a Input,
623         session: &'tcx Session,
624         out_dir: &'a Option<PathBuf>,
625         out_file: &'a Option<PathBuf>,
626         krate: Option<&'a ast::Crate>,
627         hir_crate: &'a hir::Crate,
628         analysis: &'a ty::CrateAnalysis,
629         tcx: TyCtxt<'a, 'tcx, 'tcx>,
630         crate_name: &'a str,
631     ) -> Self {
632         CompileState {
633             analysis: Some(analysis),
634             tcx: Some(tcx),
635             expanded_crate: krate,
636             hir_crate: Some(hir_crate),
637             crate_name: Some(crate_name),
638             out_file: out_file.as_ref().map(|s| &**s),
639             ..CompileState::empty(input, session, out_dir)
640         }
641     }
642
643     fn state_when_compilation_done(
644         input: &'a Input,
645         session: &'tcx Session,
646         out_dir: &'a Option<PathBuf>,
647         out_file: &'a Option<PathBuf>,
648     ) -> Self {
649         CompileState {
650             out_file: out_file.as_ref().map(|s| &**s),
651             ..CompileState::empty(input, session, out_dir)
652         }
653     }
654 }
655
656 pub fn phase_1_parse_input<'a>(
657     control: &CompileController,
658     sess: &'a Session,
659     input: &Input,
660 ) -> PResult<'a, ast::Crate> {
661     sess.diagnostic()
662         .set_continue_after_error(control.continue_parse_after_error);
663     hygiene::set_default_edition(sess.edition());
664
665     if sess.profile_queries() {
666         profile::begin(sess);
667     }
668
669     sess.profiler(|p| p.start_activity(ProfileCategory::Parsing));
670     let krate = time(sess, "parsing", || match *input {
671         Input::File(ref file) => parse::parse_crate_from_file(file, &sess.parse_sess),
672         Input::Str {
673             ref input,
674             ref name,
675         } => parse::parse_crate_from_source_str(name.clone(), input.clone(), &sess.parse_sess),
676     })?;
677     sess.profiler(|p| p.end_activity(ProfileCategory::Parsing));
678
679     sess.diagnostic().set_continue_after_error(true);
680
681     if sess.opts.debugging_opts.ast_json_noexpand {
682         println!("{}", json::as_json(&krate));
683     }
684
685     if sess.opts.debugging_opts.input_stats {
686         println!(
687             "Lines of code:             {}",
688             sess.source_map().count_lines()
689         );
690         println!("Pre-expansion node count:  {}", count_nodes(&krate));
691     }
692
693     if let Some(ref s) = sess.opts.debugging_opts.show_span {
694         syntax::show_span::run(sess.diagnostic(), s, &krate);
695     }
696
697     if sess.opts.debugging_opts.hir_stats {
698         hir_stats::print_ast_stats(&krate, "PRE EXPANSION AST STATS");
699     }
700
701     Ok(krate)
702 }
703
704 fn count_nodes(krate: &ast::Crate) -> usize {
705     let mut counter = NodeCounter::new();
706     visit::walk_crate(&mut counter, krate);
707     counter.count
708 }
709
710 // For continuing compilation after a parsed crate has been
711 // modified
712
713 pub struct ExpansionResult {
714     pub expanded_crate: ast::Crate,
715     pub defs: hir_map::Definitions,
716     pub analysis: ty::CrateAnalysis,
717     pub resolutions: Resolutions,
718     pub hir_forest: hir_map::Forest,
719 }
720
721 pub struct InnerExpansionResult<'a> {
722     pub expanded_crate: ast::Crate,
723     pub resolver: Resolver<'a>,
724     pub hir_forest: hir_map::Forest,
725 }
726
727 /// Run the "early phases" of the compiler: initial `cfg` processing,
728 /// loading compiler plugins (including those from `addl_plugins`),
729 /// syntax expansion, secondary `cfg` expansion, synthesis of a test
730 /// harness if one is to be provided, injection of a dependency on the
731 /// standard library and prelude, and name resolution.
732 ///
733 /// Returns `None` if we're aborting after handling -W help.
734 pub fn phase_2_configure_and_expand<F>(
735     sess: &Session,
736     cstore: &CStore,
737     krate: ast::Crate,
738     registry: Option<Registry>,
739     crate_name: &str,
740     addl_plugins: Option<Vec<String>>,
741     make_glob_map: MakeGlobMap,
742     after_expand: F,
743 ) -> Result<ExpansionResult, CompileIncomplete>
744 where
745     F: FnOnce(&ast::Crate) -> CompileResult,
746 {
747     // Currently, we ignore the name resolution data structures for the purposes of dependency
748     // tracking. Instead we will run name resolution and include its output in the hash of each
749     // item, much like we do for macro expansion. In other words, the hash reflects not just
750     // its contents but the results of name resolution on those contents. Hopefully we'll push
751     // this back at some point.
752     let mut crate_loader = CrateLoader::new(sess, &cstore, &crate_name);
753     let resolver_arenas = Resolver::arenas();
754     let result = phase_2_configure_and_expand_inner(
755         sess,
756         cstore,
757         krate,
758         registry,
759         crate_name,
760         addl_plugins,
761         make_glob_map,
762         &resolver_arenas,
763         &mut crate_loader,
764         after_expand,
765     );
766     match result {
767         Ok(InnerExpansionResult {
768             expanded_crate,
769             resolver,
770             hir_forest,
771         }) => Ok(ExpansionResult {
772             expanded_crate,
773             defs: resolver.definitions,
774             hir_forest,
775             resolutions: Resolutions {
776                 freevars: resolver.freevars,
777                 export_map: resolver.export_map,
778                 trait_map: resolver.trait_map,
779                 maybe_unused_trait_imports: resolver.maybe_unused_trait_imports,
780                 maybe_unused_extern_crates: resolver.maybe_unused_extern_crates,
781                 extern_prelude: resolver.extern_prelude.iter().map(|(ident, entry)| {
782                     (ident.name, entry.introduced_by_item)
783                 }).collect(),
784             },
785
786             analysis: ty::CrateAnalysis {
787                 name: crate_name.to_string(),
788                 glob_map: if resolver.make_glob_map {
789                     Some(resolver.glob_map)
790                 } else {
791                     None
792                 },
793             },
794         }),
795         Err(x) => Err(x),
796     }
797 }
798
799 /// Same as phase_2_configure_and_expand, but doesn't let you keep the resolver
800 /// around
801 pub fn phase_2_configure_and_expand_inner<'a, F>(
802     sess: &'a Session,
803     cstore: &'a CStore,
804     mut krate: ast::Crate,
805     registry: Option<Registry>,
806     crate_name: &str,
807     addl_plugins: Option<Vec<String>>,
808     make_glob_map: MakeGlobMap,
809     resolver_arenas: &'a ResolverArenas<'a>,
810     crate_loader: &'a mut CrateLoader<'a>,
811     after_expand: F,
812 ) -> Result<InnerExpansionResult<'a>, CompileIncomplete>
813 where
814     F: FnOnce(&ast::Crate) -> CompileResult,
815 {
816     krate = time(sess, "attributes injection", || {
817         syntax::attr::inject(krate, &sess.parse_sess, &sess.opts.debugging_opts.crate_attr)
818     });
819
820     let (mut krate, features) = syntax::config::features(
821         krate,
822         &sess.parse_sess,
823         sess.edition(),
824     );
825     // these need to be set "early" so that expansion sees `quote` if enabled.
826     sess.init_features(features);
827
828     let crate_types = collect_crate_types(sess, &krate.attrs);
829     sess.crate_types.set(crate_types);
830
831     let disambiguator = compute_crate_disambiguator(sess);
832     sess.crate_disambiguator.set(disambiguator);
833     rustc_incremental::prepare_session_directory(sess, &crate_name, disambiguator);
834
835     if sess.opts.incremental.is_some() {
836         time(sess, "garbage collect incremental cache directory", || {
837             if let Err(e) = rustc_incremental::garbage_collect_session_directories(sess) {
838                 warn!(
839                     "Error while trying to garbage collect incremental \
840                      compilation cache directory: {}",
841                     e
842                 );
843             }
844         });
845     }
846
847     // If necessary, compute the dependency graph (in the background).
848     let future_dep_graph = if sess.opts.build_dep_graph() {
849         Some(rustc_incremental::load_dep_graph(sess))
850     } else {
851         None
852     };
853
854     time(sess, "recursion limit", || {
855         middle::recursion_limit::update_limits(sess, &krate);
856     });
857
858     krate = time(sess, "crate injection", || {
859         let alt_std_name = sess.opts.alt_std_name.as_ref().map(|s| &**s);
860         syntax::std_inject::maybe_inject_crates_ref(krate, alt_std_name, sess.edition())
861     });
862
863     let mut addl_plugins = Some(addl_plugins);
864     let registrars = time(sess, "plugin loading", || {
865         plugin::load::load_plugins(
866             sess,
867             &cstore,
868             &krate,
869             crate_name,
870             addl_plugins.take().unwrap(),
871         )
872     });
873
874     let mut registry = registry.unwrap_or_else(|| Registry::new(sess, krate.span));
875
876     time(sess, "plugin registration", || {
877         if sess.features_untracked().rustc_diagnostic_macros {
878             registry.register_macro(
879                 "__diagnostic_used",
880                 diagnostics::plugin::expand_diagnostic_used,
881             );
882             registry.register_macro(
883                 "__register_diagnostic",
884                 diagnostics::plugin::expand_register_diagnostic,
885             );
886             registry.register_macro(
887                 "__build_diagnostic_array",
888                 diagnostics::plugin::expand_build_diagnostic_array,
889             );
890         }
891
892         for registrar in registrars {
893             registry.args_hidden = Some(registrar.args);
894             (registrar.fun)(&mut registry);
895         }
896     });
897
898     let Registry {
899         syntax_exts,
900         early_lint_passes,
901         late_lint_passes,
902         lint_groups,
903         llvm_passes,
904         attributes,
905         ..
906     } = registry;
907
908     sess.track_errors(|| {
909         let mut ls = sess.lint_store.borrow_mut();
910         for pass in early_lint_passes {
911             ls.register_early_pass(Some(sess), true, pass);
912         }
913         for pass in late_lint_passes {
914             ls.register_late_pass(Some(sess), true, pass);
915         }
916
917         for (name, (to, deprecated_name)) in lint_groups {
918             ls.register_group(Some(sess), true, name, deprecated_name, to);
919         }
920
921         *sess.plugin_llvm_passes.borrow_mut() = llvm_passes;
922         *sess.plugin_attributes.borrow_mut() = attributes.clone();
923     })?;
924
925     // Lint plugins are registered; now we can process command line flags.
926     if sess.opts.describe_lints {
927         super::describe_lints(&sess, &sess.lint_store.borrow(), true);
928         return Err(CompileIncomplete::Stopped);
929     }
930
931     time(sess, "pre ast expansion lint checks", || {
932         lint::check_ast_crate(sess, &krate, true)
933     });
934
935     let mut resolver = Resolver::new(
936         sess,
937         cstore,
938         &krate,
939         crate_name,
940         make_glob_map,
941         crate_loader,
942         &resolver_arenas,
943     );
944     syntax_ext::register_builtins(&mut resolver, syntax_exts, sess.features_untracked().quote);
945
946     // Expand all macros
947     sess.profiler(|p| p.start_activity(ProfileCategory::Expansion));
948     krate = time(sess, "expansion", || {
949         // Windows dlls do not have rpaths, so they don't know how to find their
950         // dependencies. It's up to us to tell the system where to find all the
951         // dependent dlls. Note that this uses cfg!(windows) as opposed to
952         // targ_cfg because syntax extensions are always loaded for the host
953         // compiler, not for the target.
954         //
955         // This is somewhat of an inherently racy operation, however, as
956         // multiple threads calling this function could possibly continue
957         // extending PATH far beyond what it should. To solve this for now we
958         // just don't add any new elements to PATH which are already there
959         // within PATH. This is basically a targeted fix at #17360 for rustdoc
960         // which runs rustc in parallel but has been seen (#33844) to cause
961         // problems with PATH becoming too long.
962         let mut old_path = OsString::new();
963         if cfg!(windows) {
964             old_path = env::var_os("PATH").unwrap_or(old_path);
965             let mut new_path = sess.host_filesearch(PathKind::All).search_path_dirs();
966             for path in env::split_paths(&old_path) {
967                 if !new_path.contains(&path) {
968                     new_path.push(path);
969                 }
970             }
971             env::set_var(
972                 "PATH",
973                 &env::join_paths(
974                     new_path
975                         .iter()
976                         .filter(|p| env::join_paths(iter::once(p)).is_ok()),
977                 ).unwrap(),
978             );
979         }
980
981         // Create the config for macro expansion
982         let features = sess.features_untracked();
983         let cfg = syntax::ext::expand::ExpansionConfig {
984             features: Some(&features),
985             recursion_limit: *sess.recursion_limit.get(),
986             trace_mac: sess.opts.debugging_opts.trace_macros,
987             should_test: sess.opts.test,
988             ..syntax::ext::expand::ExpansionConfig::default(crate_name.to_string())
989         };
990
991         let mut ecx = ExtCtxt::new(&sess.parse_sess, cfg, &mut resolver);
992
993         // Expand macros now!
994         let krate = time(sess, "expand crate", || {
995             ecx.monotonic_expander().expand_crate(krate)
996         });
997
998         // The rest is error reporting
999
1000         time(sess, "check unused macros", || {
1001             ecx.check_unused_macros();
1002         });
1003
1004         let mut missing_fragment_specifiers: Vec<_> = ecx.parse_sess
1005             .missing_fragment_specifiers
1006             .borrow()
1007             .iter()
1008             .cloned()
1009             .collect();
1010         missing_fragment_specifiers.sort();
1011
1012         for span in missing_fragment_specifiers {
1013             let lint = lint::builtin::MISSING_FRAGMENT_SPECIFIER;
1014             let msg = "missing fragment specifier";
1015             sess.buffer_lint(lint, ast::CRATE_NODE_ID, span, msg);
1016         }
1017         if cfg!(windows) {
1018             env::set_var("PATH", &old_path);
1019         }
1020         krate
1021     });
1022     sess.profiler(|p| p.end_activity(ProfileCategory::Expansion));
1023
1024     krate = time(sess, "maybe building test harness", || {
1025         syntax::test::modify_for_testing(
1026             &sess.parse_sess,
1027             &mut resolver,
1028             sess.opts.test,
1029             krate,
1030             sess.diagnostic(),
1031             &sess.features_untracked(),
1032         )
1033     });
1034
1035     // If we're actually rustdoc then there's no need to actually compile
1036     // anything, so switch everything to just looping
1037     if sess.opts.actually_rustdoc {
1038         krate = ReplaceBodyWithLoop::new(sess).fold_crate(krate);
1039     }
1040
1041     // If we're in rustdoc we're always compiling as an rlib, but that'll trip a
1042     // bunch of checks in the `modify` function below. For now just skip this
1043     // step entirely if we're rustdoc as it's not too useful anyway.
1044     if !sess.opts.actually_rustdoc {
1045         krate = time(sess, "maybe creating a macro crate", || {
1046             let crate_types = sess.crate_types.borrow();
1047             let num_crate_types = crate_types.len();
1048             let is_proc_macro_crate = crate_types.contains(&config::CrateType::ProcMacro);
1049             let is_test_crate = sess.opts.test;
1050             syntax_ext::proc_macro_decls::modify(
1051                 &sess.parse_sess,
1052                 &mut resolver,
1053                 krate,
1054                 is_proc_macro_crate,
1055                 is_test_crate,
1056                 num_crate_types,
1057                 sess.diagnostic(),
1058             )
1059         });
1060     }
1061
1062     // Expand global allocators, which are treated as an in-tree proc macro
1063     krate = time(sess, "creating allocators", || {
1064         allocator::expand::modify(
1065             &sess.parse_sess,
1066             &mut resolver,
1067             krate,
1068             crate_name.to_string(),
1069             sess.diagnostic(),
1070         )
1071     });
1072
1073     // Add all buffered lints from the `ParseSess` to the `Session`.
1074     sess.parse_sess.buffered_lints.with_lock(|buffered_lints| {
1075         info!("{} parse sess buffered_lints", buffered_lints.len());
1076         for BufferedEarlyLint{id, span, msg, lint_id} in buffered_lints.drain(..) {
1077             let lint = lint::Lint::from_parser_lint_id(lint_id);
1078             sess.buffer_lint(lint, id, span, &msg);
1079         }
1080     });
1081
1082     // Done with macro expansion!
1083
1084     after_expand(&krate)?;
1085
1086     if sess.opts.debugging_opts.input_stats {
1087         println!("Post-expansion node count: {}", count_nodes(&krate));
1088     }
1089
1090     if sess.opts.debugging_opts.hir_stats {
1091         hir_stats::print_ast_stats(&krate, "POST EXPANSION AST STATS");
1092     }
1093
1094     if sess.opts.debugging_opts.ast_json {
1095         println!("{}", json::as_json(&krate));
1096     }
1097
1098     time(sess, "AST validation", || {
1099         ast_validation::check_crate(sess, &krate)
1100     });
1101
1102     time(sess, "name resolution", || -> CompileResult {
1103         resolver.resolve_crate(&krate);
1104         Ok(())
1105     })?;
1106
1107     // Needs to go *after* expansion to be able to check the results of macro expansion.
1108     time(sess, "complete gated feature checking", || {
1109         sess.track_errors(|| {
1110             syntax::feature_gate::check_crate(
1111                 &krate,
1112                 &sess.parse_sess,
1113                 &sess.features_untracked(),
1114                 &attributes,
1115                 sess.opts.unstable_features,
1116             );
1117         })
1118     })?;
1119
1120     // Lower ast -> hir.
1121     // First, we need to collect the dep_graph.
1122     let dep_graph = match future_dep_graph {
1123         None => DepGraph::new_disabled(),
1124         Some(future) => {
1125             let (prev_graph, prev_work_products) =
1126                 time(sess, "blocked while dep-graph loading finishes", || {
1127                     future
1128                         .open()
1129                         .unwrap_or_else(|e| rustc_incremental::LoadResult::Error {
1130                             message: format!("could not decode incremental cache: {:?}", e),
1131                         })
1132                         .open(sess)
1133                 });
1134             DepGraph::new(prev_graph, prev_work_products)
1135         }
1136     };
1137     let hir_forest = time(sess, "lowering ast -> hir", || {
1138         let hir_crate = lower_crate(sess, cstore, &dep_graph, &krate, &mut resolver);
1139
1140         if sess.opts.debugging_opts.hir_stats {
1141             hir_stats::print_hir_stats(&hir_crate);
1142         }
1143
1144         hir_map::Forest::new(hir_crate, &dep_graph)
1145     });
1146
1147     time(sess, "early lint checks", || {
1148         lint::check_ast_crate(sess, &krate, false)
1149     });
1150
1151     // Discard hygiene data, which isn't required after lowering to HIR.
1152     if !sess.opts.debugging_opts.keep_hygiene_data {
1153         syntax::ext::hygiene::clear_markings();
1154     }
1155
1156     Ok(InnerExpansionResult {
1157         expanded_crate: krate,
1158         resolver,
1159         hir_forest,
1160     })
1161 }
1162
1163 pub fn default_provide(providers: &mut ty::query::Providers) {
1164     hir::provide(providers);
1165     borrowck::provide(providers);
1166     mir::provide(providers);
1167     reachable::provide(providers);
1168     resolve_lifetime::provide(providers);
1169     rustc_privacy::provide(providers);
1170     typeck::provide(providers);
1171     ty::provide(providers);
1172     traits::provide(providers);
1173     reachable::provide(providers);
1174     rustc_passes::provide(providers);
1175     rustc_traits::provide(providers);
1176     middle::region::provide(providers);
1177     cstore::provide(providers);
1178     lint::provide(providers);
1179 }
1180
1181 pub fn default_provide_extern(providers: &mut ty::query::Providers) {
1182     cstore::provide_extern(providers);
1183 }
1184
1185 /// Run the resolution, typechecking, region checking and other
1186 /// miscellaneous analysis passes on the crate. Return various
1187 /// structures carrying the results of the analysis.
1188 pub fn phase_3_run_analysis_passes<'tcx, F, R>(
1189     codegen_backend: &dyn CodegenBackend,
1190     control: &CompileController,
1191     sess: &'tcx Session,
1192     cstore: &'tcx CStore,
1193     hir_map: hir_map::Map<'tcx>,
1194     analysis: ty::CrateAnalysis,
1195     resolutions: Resolutions,
1196     arenas: &'tcx mut AllArenas<'tcx>,
1197     name: &str,
1198     output_filenames: &OutputFilenames,
1199     f: F,
1200 ) -> Result<R, CompileIncomplete>
1201 where
1202     F: for<'a> FnOnce(
1203         TyCtxt<'a, 'tcx, 'tcx>,
1204         ty::CrateAnalysis,
1205         mpsc::Receiver<Box<dyn Any + Send>>,
1206         CompileResult,
1207     ) -> R,
1208 {
1209     let query_result_on_disk_cache = time(sess, "load query result cache", || {
1210         rustc_incremental::load_query_result_cache(sess)
1211     });
1212
1213     time(sess, "looking for entry point", || {
1214         middle::entry::find_entry_point(sess, &hir_map, name)
1215     });
1216
1217     sess.plugin_registrar_fn
1218         .set(time(sess, "looking for plugin registrar", || {
1219             plugin::build::find_plugin_registrar(sess.diagnostic(), &hir_map)
1220         }));
1221     sess.proc_macro_decls_static
1222         .set(proc_macro_decls::find(&hir_map));
1223
1224     time(sess, "loop checking", || loops::check_crate(sess, &hir_map));
1225
1226     let mut local_providers = ty::query::Providers::default();
1227     default_provide(&mut local_providers);
1228     codegen_backend.provide(&mut local_providers);
1229     (control.provide)(&mut local_providers);
1230
1231     let mut extern_providers = local_providers;
1232     default_provide_extern(&mut extern_providers);
1233     codegen_backend.provide_extern(&mut extern_providers);
1234     (control.provide_extern)(&mut extern_providers);
1235
1236     let (tx, rx) = mpsc::channel();
1237
1238     TyCtxt::create_and_enter(
1239         sess,
1240         cstore,
1241         local_providers,
1242         extern_providers,
1243         arenas,
1244         resolutions,
1245         hir_map,
1246         query_result_on_disk_cache,
1247         name,
1248         tx,
1249         output_filenames,
1250         |tcx| {
1251             // Do some initialization of the DepGraph that can only be done with the
1252             // tcx available.
1253             rustc_incremental::dep_graph_tcx_init(tcx);
1254
1255             time(sess, "attribute checking", || {
1256                 hir::check_attr::check_crate(tcx)
1257             });
1258
1259             time(sess, "stability checking", || {
1260                 stability::check_unstable_api_usage(tcx)
1261             });
1262
1263             // passes are timed inside typeck
1264             match typeck::check_crate(tcx) {
1265                 Ok(x) => x,
1266                 Err(x) => {
1267                     f(tcx, analysis, rx, Err(x));
1268                     return Err(x);
1269                 }
1270             }
1271
1272             time(sess, "rvalue promotion", || {
1273                 rvalue_promotion::check_crate(tcx)
1274             });
1275
1276             time(sess, "privacy checking", || {
1277                 rustc_privacy::check_crate(tcx)
1278             });
1279
1280             time(sess, "intrinsic checking", || {
1281                 middle::intrinsicck::check_crate(tcx)
1282             });
1283
1284             time(sess, "match checking", || mir::matchck_crate(tcx));
1285
1286             // this must run before MIR dump, because
1287             // "not all control paths return a value" is reported here.
1288             //
1289             // maybe move the check to a MIR pass?
1290             time(sess, "liveness checking", || {
1291                 middle::liveness::check_crate(tcx)
1292             });
1293
1294             time(sess, "borrow checking", || {
1295                 if tcx.use_ast_borrowck() {
1296                     borrowck::check_crate(tcx);
1297                 }
1298             });
1299
1300             time(sess,
1301                  "MIR borrow checking",
1302                  || tcx.par_body_owners(|def_id| { tcx.mir_borrowck(def_id); }));
1303
1304             time(sess, "dumping chalk-like clauses", || {
1305                 rustc_traits::lowering::dump_program_clauses(tcx);
1306             });
1307
1308             time(sess, "MIR effect checking", || {
1309                 for def_id in tcx.body_owners() {
1310                     mir::transform::check_unsafety::check_unsafety(tcx, def_id)
1311                 }
1312             });
1313             // Avoid overwhelming user with errors if type checking failed.
1314             // I'm not sure how helpful this is, to be honest, but it avoids
1315             // a
1316             // lot of annoying errors in the compile-fail tests (basically,
1317             // lint warnings and so on -- kindck used to do this abort, but
1318             // kindck is gone now). -nmatsakis
1319             if sess.err_count() > 0 {
1320                 return Ok(f(tcx, analysis, rx, sess.compile_status()));
1321             }
1322
1323             time(sess, "death checking", || middle::dead::check_crate(tcx));
1324
1325             time(sess, "unused lib feature checking", || {
1326                 stability::check_unused_or_stable_features(tcx)
1327             });
1328
1329             time(sess, "lint checking", || lint::check_crate(tcx));
1330
1331             return Ok(f(tcx, analysis, rx, tcx.sess.compile_status()));
1332         },
1333     )
1334 }
1335
1336 /// Run the codegen backend, after which the AST and analysis can
1337 /// be discarded.
1338 pub fn phase_4_codegen<'a, 'tcx>(
1339     codegen_backend: &dyn CodegenBackend,
1340     tcx: TyCtxt<'a, 'tcx, 'tcx>,
1341     rx: mpsc::Receiver<Box<dyn Any + Send>>,
1342 ) -> Box<dyn Any> {
1343     time(tcx.sess, "resolving dependency formats", || {
1344         ::rustc::middle::dependency_format::calculate(tcx)
1345     });
1346
1347     tcx.sess.profiler(|p| p.start_activity(ProfileCategory::Codegen));
1348     let codegen = time(tcx.sess, "codegen", move || codegen_backend.codegen_crate(tcx, rx));
1349     tcx.sess.profiler(|p| p.end_activity(ProfileCategory::Codegen));
1350     if tcx.sess.profile_queries() {
1351         profile::dump(&tcx.sess, "profile_queries".to_string())
1352     }
1353
1354     codegen
1355 }
1356
1357 fn escape_dep_filename(filename: &FileName) -> String {
1358     // Apparently clang and gcc *only* escape spaces:
1359     // http://llvm.org/klaus/clang/commit/9d50634cfc268ecc9a7250226dd5ca0e945240d4
1360     filename.to_string().replace(" ", "\\ ")
1361 }
1362
1363 // Returns all the paths that correspond to generated files.
1364 fn generated_output_paths(
1365     sess: &Session,
1366     outputs: &OutputFilenames,
1367     exact_name: bool,
1368     crate_name: &str,
1369 ) -> Vec<PathBuf> {
1370     let mut out_filenames = Vec::new();
1371     for output_type in sess.opts.output_types.keys() {
1372         let file = outputs.path(*output_type);
1373         match *output_type {
1374             // If the filename has been overridden using `-o`, it will not be modified
1375             // by appending `.rlib`, `.exe`, etc., so we can skip this transformation.
1376             OutputType::Exe if !exact_name => for crate_type in sess.crate_types.borrow().iter() {
1377                 let p = ::rustc_codegen_utils::link::filename_for_input(
1378                     sess,
1379                     *crate_type,
1380                     crate_name,
1381                     outputs,
1382                 );
1383                 out_filenames.push(p);
1384             },
1385             OutputType::DepInfo if sess.opts.debugging_opts.dep_info_omit_d_target => {
1386                 // Don't add the dep-info output when omitting it from dep-info targets
1387             }
1388             _ => {
1389                 out_filenames.push(file);
1390             }
1391         }
1392     }
1393     out_filenames
1394 }
1395
1396 // Runs `f` on every output file path and returns the first non-None result, or None if `f`
1397 // returns None for every file path.
1398 fn check_output<F, T>(output_paths: &[PathBuf], f: F) -> Option<T>
1399 where
1400     F: Fn(&PathBuf) -> Option<T>,
1401 {
1402     for output_path in output_paths {
1403         if let Some(result) = f(output_path) {
1404             return Some(result);
1405         }
1406     }
1407     None
1408 }
1409
1410 pub fn output_contains_path(output_paths: &[PathBuf], input_path: &PathBuf) -> bool {
1411     let input_path = input_path.canonicalize().ok();
1412     if input_path.is_none() {
1413         return false;
1414     }
1415     let check = |output_path: &PathBuf| {
1416         if output_path.canonicalize().ok() == input_path {
1417             Some(())
1418         } else {
1419             None
1420         }
1421     };
1422     check_output(output_paths, check).is_some()
1423 }
1424
1425 pub fn output_conflicts_with_dir(output_paths: &[PathBuf]) -> Option<PathBuf> {
1426     let check = |output_path: &PathBuf| {
1427         if output_path.is_dir() {
1428             Some(output_path.clone())
1429         } else {
1430             None
1431         }
1432     };
1433     check_output(output_paths, check)
1434 }
1435
1436 fn write_out_deps(sess: &Session, outputs: &OutputFilenames, out_filenames: &[PathBuf]) {
1437     // Write out dependency rules to the dep-info file if requested
1438     if !sess.opts.output_types.contains_key(&OutputType::DepInfo) {
1439         return;
1440     }
1441     let deps_filename = outputs.path(OutputType::DepInfo);
1442
1443     let result = (|| -> io::Result<()> {
1444         // Build a list of files used to compile the output and
1445         // write Makefile-compatible dependency rules
1446         let files: Vec<String> = sess.source_map()
1447             .files()
1448             .iter()
1449             .filter(|fmap| fmap.is_real_file())
1450             .filter(|fmap| !fmap.is_imported())
1451             .map(|fmap| escape_dep_filename(&fmap.name))
1452             .collect();
1453         let mut file = fs::File::create(&deps_filename)?;
1454         for path in out_filenames {
1455             writeln!(file, "{}: {}\n", path.display(), files.join(" "))?;
1456         }
1457
1458         // Emit a fake target for each input file to the compilation. This
1459         // prevents `make` from spitting out an error if a file is later
1460         // deleted. For more info see #28735
1461         for path in files {
1462             writeln!(file, "{}:", path)?;
1463         }
1464         Ok(())
1465     })();
1466
1467     if let Err(e) = result {
1468         sess.fatal(&format!(
1469             "error writing dependencies to `{}`: {}",
1470             deps_filename.display(),
1471             e
1472         ));
1473     }
1474 }
1475
1476 pub fn collect_crate_types(session: &Session, attrs: &[ast::Attribute]) -> Vec<config::CrateType> {
1477     // Unconditionally collect crate types from attributes to make them used
1478     let attr_types: Vec<config::CrateType> = attrs
1479         .iter()
1480         .filter_map(|a| {
1481             if a.check_name("crate_type") {
1482                 match a.value_str() {
1483                     Some(ref n) if *n == "rlib" => Some(config::CrateType::Rlib),
1484                     Some(ref n) if *n == "dylib" => Some(config::CrateType::Dylib),
1485                     Some(ref n) if *n == "cdylib" => Some(config::CrateType::Cdylib),
1486                     Some(ref n) if *n == "lib" => Some(config::default_lib_output()),
1487                     Some(ref n) if *n == "staticlib" => Some(config::CrateType::Staticlib),
1488                     Some(ref n) if *n == "proc-macro" => Some(config::CrateType::ProcMacro),
1489                     Some(ref n) if *n == "bin" => Some(config::CrateType::Executable),
1490                     Some(ref n) => {
1491                         let crate_types = vec![
1492                             Symbol::intern("rlib"),
1493                             Symbol::intern("dylib"),
1494                             Symbol::intern("cdylib"),
1495                             Symbol::intern("lib"),
1496                             Symbol::intern("staticlib"),
1497                             Symbol::intern("proc-macro"),
1498                             Symbol::intern("bin")
1499                         ];
1500
1501                         if let ast::MetaItemKind::NameValue(spanned) = a.meta().unwrap().node {
1502                             let span = spanned.span;
1503                             let lev_candidate = find_best_match_for_name(
1504                                 crate_types.iter(),
1505                                 &n.as_str(),
1506                                 None
1507                             );
1508                             if let Some(candidate) = lev_candidate {
1509                                 session.buffer_lint_with_diagnostic(
1510                                     lint::builtin::UNKNOWN_CRATE_TYPES,
1511                                     ast::CRATE_NODE_ID,
1512                                     span,
1513                                     "invalid `crate_type` value",
1514                                     lint::builtin::BuiltinLintDiagnostics::
1515                                         UnknownCrateTypes(
1516                                             span,
1517                                             "did you mean".to_string(),
1518                                             format!("\"{}\"", candidate)
1519                                         )
1520                                 );
1521                             } else {
1522                                 session.buffer_lint(
1523                                     lint::builtin::UNKNOWN_CRATE_TYPES,
1524                                     ast::CRATE_NODE_ID,
1525                                     span,
1526                                     "invalid `crate_type` value"
1527                                 );
1528                             }
1529                         }
1530                         None
1531                     }
1532                     None => {
1533                         session
1534                             .struct_span_err(a.span, "`crate_type` requires a value")
1535                             .note("for example: `#![crate_type=\"lib\"]`")
1536                             .emit();
1537                         None
1538                     }
1539                 }
1540             } else {
1541                 None
1542             }
1543         })
1544         .collect();
1545
1546     // If we're generating a test executable, then ignore all other output
1547     // styles at all other locations
1548     if session.opts.test {
1549         return vec![config::CrateType::Executable];
1550     }
1551
1552     // Only check command line flags if present. If no types are specified by
1553     // command line, then reuse the empty `base` Vec to hold the types that
1554     // will be found in crate attributes.
1555     let mut base = session.opts.crate_types.clone();
1556     if base.is_empty() {
1557         base.extend(attr_types);
1558         if base.is_empty() {
1559             base.push(::rustc_codegen_utils::link::default_output_for_target(
1560                 session,
1561             ));
1562         } else {
1563             base.sort();
1564             base.dedup();
1565         }
1566     }
1567
1568     base.retain(|crate_type| {
1569         let res = !::rustc_codegen_utils::link::invalid_output_for_target(session, *crate_type);
1570
1571         if !res {
1572             session.warn(&format!(
1573                 "dropping unsupported crate type `{}` for target `{}`",
1574                 *crate_type, session.opts.target_triple
1575             ));
1576         }
1577
1578         res
1579     });
1580
1581     base
1582 }
1583
1584 pub fn compute_crate_disambiguator(session: &Session) -> CrateDisambiguator {
1585     use std::hash::Hasher;
1586
1587     // The crate_disambiguator is a 128 bit hash. The disambiguator is fed
1588     // into various other hashes quite a bit (symbol hashes, incr. comp. hashes,
1589     // debuginfo type IDs, etc), so we don't want it to be too wide. 128 bits
1590     // should still be safe enough to avoid collisions in practice.
1591     let mut hasher = StableHasher::<Fingerprint>::new();
1592
1593     let mut metadata = session.opts.cg.metadata.clone();
1594     // We don't want the crate_disambiguator to dependent on the order
1595     // -C metadata arguments, so sort them:
1596     metadata.sort();
1597     // Every distinct -C metadata value is only incorporated once:
1598     metadata.dedup();
1599
1600     hasher.write(b"metadata");
1601     for s in &metadata {
1602         // Also incorporate the length of a metadata string, so that we generate
1603         // different values for `-Cmetadata=ab -Cmetadata=c` and
1604         // `-Cmetadata=a -Cmetadata=bc`
1605         hasher.write_usize(s.len());
1606         hasher.write(s.as_bytes());
1607     }
1608
1609     // Also incorporate crate type, so that we don't get symbol conflicts when
1610     // linking against a library of the same name, if this is an executable.
1611     let is_exe = session
1612         .crate_types
1613         .borrow()
1614         .contains(&config::CrateType::Executable);
1615     hasher.write(if is_exe { b"exe" } else { b"lib" });
1616
1617     CrateDisambiguator::from(hasher.finish())
1618 }
1619
1620 pub fn build_output_filenames(
1621     input: &Input,
1622     odir: &Option<PathBuf>,
1623     ofile: &Option<PathBuf>,
1624     attrs: &[ast::Attribute],
1625     sess: &Session,
1626 ) -> OutputFilenames {
1627     match *ofile {
1628         None => {
1629             // "-" as input file will cause the parser to read from stdin so we
1630             // have to make up a name
1631             // We want to toss everything after the final '.'
1632             let dirpath = (*odir).as_ref().cloned().unwrap_or_default();
1633
1634             // If a crate name is present, we use it as the link name
1635             let stem = sess.opts
1636                 .crate_name
1637                 .clone()
1638                 .or_else(|| attr::find_crate_name(attrs).map(|n| n.to_string()))
1639                 .unwrap_or_else(|| input.filestem().to_owned());
1640
1641             OutputFilenames {
1642                 out_directory: dirpath,
1643                 out_filestem: stem,
1644                 single_output_file: None,
1645                 extra: sess.opts.cg.extra_filename.clone(),
1646                 outputs: sess.opts.output_types.clone(),
1647             }
1648         }
1649
1650         Some(ref out_file) => {
1651             let unnamed_output_types = sess.opts
1652                 .output_types
1653                 .values()
1654                 .filter(|a| a.is_none())
1655                 .count();
1656             let ofile = if unnamed_output_types > 1 {
1657                 sess.warn(
1658                     "due to multiple output types requested, the explicitly specified \
1659                      output file name will be adapted for each output type",
1660                 );
1661                 None
1662             } else {
1663                 Some(out_file.clone())
1664             };
1665             if *odir != None {
1666                 sess.warn("ignoring --out-dir flag due to -o flag");
1667             }
1668             if !sess.opts.cg.extra_filename.is_empty() {
1669                 sess.warn("ignoring -C extra-filename flag due to -o flag");
1670             }
1671
1672             OutputFilenames {
1673                 out_directory: out_file.parent().unwrap_or_else(|| Path::new("")).to_path_buf(),
1674                 out_filestem: out_file
1675                     .file_stem()
1676                     .unwrap_or_default()
1677                     .to_str()
1678                     .unwrap()
1679                     .to_string(),
1680                 single_output_file: ofile,
1681                 extra: sess.opts.cg.extra_filename.clone(),
1682                 outputs: sess.opts.output_types.clone(),
1683             }
1684         }
1685     }
1686 }