]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_session/src/session.rs
Rollup merge of #82420 - sunfishcode:wasi-docs, r=alexcrichton
[rust.git] / compiler / rustc_session / src / session.rs
1 use crate::cgu_reuse_tracker::CguReuseTracker;
2 use crate::code_stats::CodeStats;
3 pub use crate::code_stats::{DataTypeKind, FieldInfo, SizeKind, VariantInfo};
4 use crate::config::{self, CrateType, OutputType, PrintRequest, SanitizerSet, SwitchWithOptPath};
5 use crate::filesearch;
6 use crate::lint::{self, LintId};
7 use crate::parse::ParseSess;
8 use crate::search_paths::{PathKind, SearchPath};
9
10 pub use rustc_ast::attr::MarkedAttrs;
11 pub use rustc_ast::crate_disambiguator::CrateDisambiguator;
12 pub use rustc_ast::Attribute;
13 use rustc_data_structures::flock;
14 use rustc_data_structures::fx::{FxHashMap, FxHashSet};
15 use rustc_data_structures::jobserver::{self, Client};
16 use rustc_data_structures::profiling::{duration_to_secs_str, SelfProfiler, SelfProfilerRef};
17 use rustc_data_structures::sync::{
18     self, AtomicU64, AtomicUsize, Lock, Lrc, OnceCell, OneThread, Ordering, Ordering::SeqCst,
19 };
20 use rustc_errors::annotate_snippet_emitter_writer::AnnotateSnippetEmitterWriter;
21 use rustc_errors::emitter::{Emitter, EmitterWriter, HumanReadableErrorType};
22 use rustc_errors::json::JsonEmitter;
23 use rustc_errors::registry::Registry;
24 use rustc_errors::{Applicability, Diagnostic, DiagnosticBuilder, DiagnosticId, ErrorReported};
25 use rustc_lint_defs::FutureBreakage;
26 use rustc_span::edition::Edition;
27 use rustc_span::source_map::{FileLoader, MultiSpan, RealFileLoader, SourceMap, Span};
28 use rustc_span::{sym, SourceFileHashAlgorithm, Symbol};
29 use rustc_target::asm::InlineAsmArch;
30 use rustc_target::spec::{CodeModel, PanicStrategy, RelocModel, RelroLevel};
31 use rustc_target::spec::{SplitDebuginfo, Target, TargetTriple, TlsModel};
32
33 use std::cell::{self, RefCell};
34 use std::env;
35 use std::fmt;
36 use std::io::Write;
37 use std::num::NonZeroU32;
38 use std::ops::{Div, Mul};
39 use std::path::PathBuf;
40 use std::str::FromStr;
41 use std::sync::Arc;
42 use std::time::Duration;
43
44 pub trait SessionLintStore: sync::Send + sync::Sync {
45     fn name_to_lint(&self, lint_name: &str) -> LintId;
46 }
47
48 pub struct OptimizationFuel {
49     /// If `-zfuel=crate=n` is specified, initially set to `n`, otherwise `0`.
50     remaining: u64,
51     /// We're rejecting all further optimizations.
52     out_of_fuel: bool,
53 }
54
55 /// The behavior of the CTFE engine when an error occurs with regards to backtraces.
56 #[derive(Clone, Copy)]
57 pub enum CtfeBacktrace {
58     /// Do nothing special, return the error as usual without a backtrace.
59     Disabled,
60     /// Capture a backtrace at the point the error is created and return it in the error
61     /// (to be printed later if/when the error ever actually gets shown to the user).
62     Capture,
63     /// Capture a backtrace at the point the error is created and immediately print it out.
64     Immediate,
65 }
66
67 /// New-type wrapper around `usize` for representing limits. Ensures that comparisons against
68 /// limits are consistent throughout the compiler.
69 #[derive(Clone, Copy, Debug)]
70 pub struct Limit(pub usize);
71
72 impl Limit {
73     /// Create a new limit from a `usize`.
74     pub fn new(value: usize) -> Self {
75         Limit(value)
76     }
77
78     /// Check that `value` is within the limit. Ensures that the same comparisons are used
79     /// throughout the compiler, as mismatches can cause ICEs, see #72540.
80     #[inline]
81     pub fn value_within_limit(&self, value: usize) -> bool {
82         value <= self.0
83     }
84 }
85
86 impl fmt::Display for Limit {
87     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
88         write!(f, "{}", self.0)
89     }
90 }
91
92 impl Div<usize> for Limit {
93     type Output = Limit;
94
95     fn div(self, rhs: usize) -> Self::Output {
96         Limit::new(self.0 / rhs)
97     }
98 }
99
100 impl Mul<usize> for Limit {
101     type Output = Limit;
102
103     fn mul(self, rhs: usize) -> Self::Output {
104         Limit::new(self.0 * rhs)
105     }
106 }
107
108 /// Represents the data associated with a compilation
109 /// session for a single crate.
110 pub struct Session {
111     pub target: Target,
112     pub host: Target,
113     pub opts: config::Options,
114     pub host_tlib_path: SearchPath,
115     /// `None` if the host and target are the same.
116     pub target_tlib_path: Option<SearchPath>,
117     pub parse_sess: ParseSess,
118     pub sysroot: PathBuf,
119     /// The name of the root source file of the crate, in the local file system.
120     /// `None` means that there is no source file.
121     pub local_crate_source_file: Option<PathBuf>,
122     /// The directory the compiler has been executed in plus a flag indicating
123     /// if the value stored here has been affected by path remapping.
124     pub working_dir: (PathBuf, bool),
125
126     /// Set of `(DiagnosticId, Option<Span>, message)` tuples tracking
127     /// (sub)diagnostics that have been set once, but should not be set again,
128     /// in order to avoid redundantly verbose output (Issue #24690, #44953).
129     pub one_time_diagnostics: Lock<FxHashSet<(DiagnosticMessageId, Option<Span>, String)>>,
130     crate_types: OnceCell<Vec<CrateType>>,
131     /// The `crate_disambiguator` is constructed out of all the `-C metadata`
132     /// arguments passed to the compiler. Its value together with the crate-name
133     /// forms a unique global identifier for the crate. It is used to allow
134     /// multiple crates with the same name to coexist. See the
135     /// `rustc_codegen_llvm::back::symbol_names` module for more information.
136     pub crate_disambiguator: OnceCell<CrateDisambiguator>,
137
138     features: OnceCell<rustc_feature::Features>,
139
140     lint_store: OnceCell<Lrc<dyn SessionLintStore>>,
141
142     /// The maximum recursion limit for potentially infinitely recursive
143     /// operations such as auto-dereference and monomorphization.
144     pub recursion_limit: OnceCell<Limit>,
145
146     /// The maximum length of types during monomorphization.
147     pub type_length_limit: OnceCell<Limit>,
148
149     /// The maximum blocks a const expression can evaluate.
150     pub const_eval_limit: OnceCell<Limit>,
151
152     incr_comp_session: OneThread<RefCell<IncrCompSession>>,
153     /// Used for incremental compilation tests. Will only be populated if
154     /// `-Zquery-dep-graph` is specified.
155     pub cgu_reuse_tracker: CguReuseTracker,
156
157     /// Used by `-Z self-profile`.
158     pub prof: SelfProfilerRef,
159
160     /// Some measurements that are being gathered during compilation.
161     pub perf_stats: PerfStats,
162
163     /// Data about code being compiled, gathered during compilation.
164     pub code_stats: CodeStats,
165
166     /// If `-zfuel=crate=n` is specified, `Some(crate)`.
167     optimization_fuel_crate: Option<String>,
168
169     /// Tracks fuel info if `-zfuel=crate=n` is specified.
170     optimization_fuel: Lock<OptimizationFuel>,
171
172     // The next two are public because the driver needs to read them.
173     /// If `-zprint-fuel=crate`, `Some(crate)`.
174     pub print_fuel_crate: Option<String>,
175     /// Always set to zero and incremented so that we can print fuel expended by a crate.
176     pub print_fuel: AtomicU64,
177
178     /// Loaded up early on in the initialization of this `Session` to avoid
179     /// false positives about a job server in our environment.
180     pub jobserver: Client,
181
182     /// Cap lint level specified by a driver specifically.
183     pub driver_lint_caps: FxHashMap<lint::LintId, lint::Level>,
184
185     /// `Span`s of trait methods that weren't found to avoid emitting object safety errors
186     pub trait_methods_not_found: Lock<FxHashSet<Span>>,
187
188     /// Mapping from ident span to path span for paths that don't exist as written, but that
189     /// exist under `std`. For example, wrote `str::from_utf8` instead of `std::str::from_utf8`.
190     pub confused_type_with_std_module: Lock<FxHashMap<Span, Span>>,
191
192     /// Path for libraries that will take preference over libraries shipped by Rust.
193     /// Used by windows-gnu targets to priortize system mingw-w64 libraries.
194     pub system_library_path: OneThread<RefCell<Option<Option<PathBuf>>>>,
195
196     /// Tracks the current behavior of the CTFE engine when an error occurs.
197     /// Options range from returning the error without a backtrace to returning an error
198     /// and immediately printing the backtrace to stderr.
199     pub ctfe_backtrace: Lock<CtfeBacktrace>,
200
201     /// This tracks where `-Zunleash-the-miri-inside-of-you` was used to get around a
202     /// const check, optionally with the relevant feature gate.  We use this to
203     /// warn about unleashing, but with a single diagnostic instead of dozens that
204     /// drown everything else in noise.
205     miri_unleashed_features: Lock<Vec<(Span, Option<Symbol>)>>,
206
207     /// Base directory containing the `src/` for the Rust standard library, and
208     /// potentially `rustc` as well, if we can can find it. Right now it's always
209     /// `$sysroot/lib/rustlib/src/rust` (i.e. the `rustup` `rust-src` component).
210     ///
211     /// This directory is what the virtual `/rustc/$hash` is translated back to,
212     /// if Rust was built with path remapping to `/rustc/$hash` enabled
213     /// (the `rust.remap-debuginfo` option in `config.toml`).
214     pub real_rust_source_base_dir: Option<PathBuf>,
215
216     /// Architecture to use for interpreting asm!.
217     pub asm_arch: Option<InlineAsmArch>,
218
219     /// Set of enabled features for the current target.
220     pub target_features: FxHashSet<Symbol>,
221
222     known_attrs: Lock<MarkedAttrs>,
223     used_attrs: Lock<MarkedAttrs>,
224
225     /// `Span`s for `if` conditions that we have suggested turning into `if let`.
226     pub if_let_suggestions: Lock<FxHashSet<Span>>,
227 }
228
229 pub struct PerfStats {
230     /// The accumulated time spent on computing symbol hashes.
231     pub symbol_hash_time: Lock<Duration>,
232     /// Total number of values canonicalized queries constructed.
233     pub queries_canonicalized: AtomicUsize,
234     /// Number of times this query is invoked.
235     pub normalize_generic_arg_after_erasing_regions: AtomicUsize,
236     /// Number of times this query is invoked.
237     pub normalize_projection_ty: AtomicUsize,
238 }
239
240 /// Enum to support dispatch of one-time diagnostics (in `Session.diag_once`).
241 enum DiagnosticBuilderMethod {
242     Note,
243     SpanNote,
244     SpanSuggestion(String), // suggestion
245                             // Add more variants as needed to support one-time diagnostics.
246 }
247
248 /// Trait implemented by error types. This should not be implemented manually. Instead, use
249 /// `#[derive(SessionDiagnostic)]` -- see [rustc_macros::SessionDiagnostic].
250 pub trait SessionDiagnostic<'a> {
251     /// Write out as a diagnostic out of `sess`.
252     #[must_use]
253     fn into_diagnostic(self, sess: &'a Session) -> DiagnosticBuilder<'a>;
254 }
255
256 /// Diagnostic message ID, used by `Session.one_time_diagnostics` to avoid
257 /// emitting the same message more than once.
258 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
259 pub enum DiagnosticMessageId {
260     ErrorId(u16), // EXXXX error code as integer
261     LintId(lint::LintId),
262     StabilityId(Option<NonZeroU32>), // issue number
263 }
264
265 impl From<&'static lint::Lint> for DiagnosticMessageId {
266     fn from(lint: &'static lint::Lint) -> Self {
267         DiagnosticMessageId::LintId(lint::LintId::of(lint))
268     }
269 }
270
271 impl Session {
272     pub fn miri_unleashed_feature(&self, span: Span, feature_gate: Option<Symbol>) {
273         self.miri_unleashed_features.lock().push((span, feature_gate));
274     }
275
276     fn check_miri_unleashed_features(&self) {
277         let unleashed_features = self.miri_unleashed_features.lock();
278         if !unleashed_features.is_empty() {
279             let mut must_err = false;
280             // Create a diagnostic pointing at where things got unleashed.
281             let mut diag = self.struct_warn("skipping const checks");
282             for &(span, feature_gate) in unleashed_features.iter() {
283                 // FIXME: `span_label` doesn't do anything, so we use "help" as a hack.
284                 if let Some(feature_gate) = feature_gate {
285                     diag.span_help(span, &format!("skipping check for `{}` feature", feature_gate));
286                     // The unleash flag must *not* be used to just "hack around" feature gates.
287                     must_err = true;
288                 } else {
289                     diag.span_help(span, "skipping check that does not even have a feature gate");
290                 }
291             }
292             diag.emit();
293             // If we should err, make sure we did.
294             if must_err && !self.has_errors() {
295                 // We have skipped a feature gate, and not run into other errors... reject.
296                 self.err(
297                     "`-Zunleash-the-miri-inside-of-you` may not be used to circumvent feature \
298                      gates, except when testing error paths in the CTFE engine",
299                 );
300             }
301         }
302     }
303
304     /// Invoked all the way at the end to finish off diagnostics printing.
305     pub fn finish_diagnostics(&self, registry: &Registry) {
306         self.check_miri_unleashed_features();
307         self.diagnostic().print_error_count(registry);
308         self.emit_future_breakage();
309     }
310
311     fn emit_future_breakage(&self) {
312         if !self.opts.debugging_opts.emit_future_incompat_report {
313             return;
314         }
315
316         let diags = self.diagnostic().take_future_breakage_diagnostics();
317         if diags.is_empty() {
318             return;
319         }
320         // If any future-breakage lints were registered, this lint store
321         // should be available
322         let lint_store = self.lint_store.get().expect("`lint_store` not initialized!");
323         let diags_and_breakage: Vec<(FutureBreakage, Diagnostic)> = diags
324             .into_iter()
325             .map(|diag| {
326                 let lint_name = match &diag.code {
327                     Some(DiagnosticId::Lint { name, has_future_breakage: true }) => name,
328                     _ => panic!("Unexpected code in diagnostic {:?}", diag),
329                 };
330                 let lint = lint_store.name_to_lint(&lint_name);
331                 let future_breakage =
332                     lint.lint.future_incompatible.unwrap().future_breakage.unwrap();
333                 (future_breakage, diag)
334             })
335             .collect();
336         self.parse_sess.span_diagnostic.emit_future_breakage_report(diags_and_breakage);
337     }
338
339     pub fn local_crate_disambiguator(&self) -> CrateDisambiguator {
340         self.crate_disambiguator.get().copied().unwrap()
341     }
342
343     pub fn crate_types(&self) -> &[CrateType] {
344         self.crate_types.get().unwrap().as_slice()
345     }
346
347     pub fn init_crate_types(&self, crate_types: Vec<CrateType>) {
348         self.crate_types.set(crate_types).expect("`crate_types` was initialized twice")
349     }
350
351     #[inline]
352     pub fn recursion_limit(&self) -> Limit {
353         self.recursion_limit.get().copied().unwrap()
354     }
355
356     #[inline]
357     pub fn type_length_limit(&self) -> Limit {
358         self.type_length_limit.get().copied().unwrap()
359     }
360
361     pub fn const_eval_limit(&self) -> Limit {
362         self.const_eval_limit.get().copied().unwrap()
363     }
364
365     pub fn struct_span_warn<S: Into<MultiSpan>>(&self, sp: S, msg: &str) -> DiagnosticBuilder<'_> {
366         self.diagnostic().struct_span_warn(sp, msg)
367     }
368     pub fn struct_span_warn_with_code<S: Into<MultiSpan>>(
369         &self,
370         sp: S,
371         msg: &str,
372         code: DiagnosticId,
373     ) -> DiagnosticBuilder<'_> {
374         self.diagnostic().struct_span_warn_with_code(sp, msg, code)
375     }
376     pub fn struct_warn(&self, msg: &str) -> DiagnosticBuilder<'_> {
377         self.diagnostic().struct_warn(msg)
378     }
379     pub fn struct_span_allow<S: Into<MultiSpan>>(&self, sp: S, msg: &str) -> DiagnosticBuilder<'_> {
380         self.diagnostic().struct_span_allow(sp, msg)
381     }
382     pub fn struct_allow(&self, msg: &str) -> DiagnosticBuilder<'_> {
383         self.diagnostic().struct_allow(msg)
384     }
385     pub fn struct_span_err<S: Into<MultiSpan>>(&self, sp: S, msg: &str) -> DiagnosticBuilder<'_> {
386         self.diagnostic().struct_span_err(sp, msg)
387     }
388     pub fn struct_span_err_with_code<S: Into<MultiSpan>>(
389         &self,
390         sp: S,
391         msg: &str,
392         code: DiagnosticId,
393     ) -> DiagnosticBuilder<'_> {
394         self.diagnostic().struct_span_err_with_code(sp, msg, code)
395     }
396     // FIXME: This method should be removed (every error should have an associated error code).
397     pub fn struct_err(&self, msg: &str) -> DiagnosticBuilder<'_> {
398         self.diagnostic().struct_err(msg)
399     }
400     pub fn struct_err_with_code(&self, msg: &str, code: DiagnosticId) -> DiagnosticBuilder<'_> {
401         self.diagnostic().struct_err_with_code(msg, code)
402     }
403     pub fn struct_span_fatal<S: Into<MultiSpan>>(&self, sp: S, msg: &str) -> DiagnosticBuilder<'_> {
404         self.diagnostic().struct_span_fatal(sp, msg)
405     }
406     pub fn struct_span_fatal_with_code<S: Into<MultiSpan>>(
407         &self,
408         sp: S,
409         msg: &str,
410         code: DiagnosticId,
411     ) -> DiagnosticBuilder<'_> {
412         self.diagnostic().struct_span_fatal_with_code(sp, msg, code)
413     }
414     pub fn struct_fatal(&self, msg: &str) -> DiagnosticBuilder<'_> {
415         self.diagnostic().struct_fatal(msg)
416     }
417
418     pub fn span_fatal<S: Into<MultiSpan>>(&self, sp: S, msg: &str) -> ! {
419         self.diagnostic().span_fatal(sp, msg).raise()
420     }
421     pub fn span_fatal_with_code<S: Into<MultiSpan>>(
422         &self,
423         sp: S,
424         msg: &str,
425         code: DiagnosticId,
426     ) -> ! {
427         self.diagnostic().span_fatal_with_code(sp, msg, code).raise()
428     }
429     pub fn fatal(&self, msg: &str) -> ! {
430         self.diagnostic().fatal(msg).raise()
431     }
432     pub fn span_err_or_warn<S: Into<MultiSpan>>(&self, is_warning: bool, sp: S, msg: &str) {
433         if is_warning {
434             self.span_warn(sp, msg);
435         } else {
436             self.span_err(sp, msg);
437         }
438     }
439     pub fn span_err<S: Into<MultiSpan>>(&self, sp: S, msg: &str) {
440         self.diagnostic().span_err(sp, msg)
441     }
442     pub fn span_err_with_code<S: Into<MultiSpan>>(&self, sp: S, msg: &str, code: DiagnosticId) {
443         self.diagnostic().span_err_with_code(sp, &msg, code)
444     }
445     pub fn err(&self, msg: &str) {
446         self.diagnostic().err(msg)
447     }
448     pub fn emit_err<'a>(&'a self, err: impl SessionDiagnostic<'a>) {
449         err.into_diagnostic(self).emit()
450     }
451     pub fn err_count(&self) -> usize {
452         self.diagnostic().err_count()
453     }
454     pub fn has_errors(&self) -> bool {
455         self.diagnostic().has_errors()
456     }
457     pub fn has_errors_or_delayed_span_bugs(&self) -> bool {
458         self.diagnostic().has_errors_or_delayed_span_bugs()
459     }
460     pub fn abort_if_errors(&self) {
461         self.diagnostic().abort_if_errors();
462     }
463     pub fn compile_status(&self) -> Result<(), ErrorReported> {
464         if self.has_errors() {
465             self.diagnostic().emit_stashed_diagnostics();
466             Err(ErrorReported)
467         } else {
468             Ok(())
469         }
470     }
471     // FIXME(matthewjasper) Remove this method, it should never be needed.
472     pub fn track_errors<F, T>(&self, f: F) -> Result<T, ErrorReported>
473     where
474         F: FnOnce() -> T,
475     {
476         let old_count = self.err_count();
477         let result = f();
478         let errors = self.err_count() - old_count;
479         if errors == 0 { Ok(result) } else { Err(ErrorReported) }
480     }
481     pub fn span_warn<S: Into<MultiSpan>>(&self, sp: S, msg: &str) {
482         self.diagnostic().span_warn(sp, msg)
483     }
484     pub fn span_warn_with_code<S: Into<MultiSpan>>(&self, sp: S, msg: &str, code: DiagnosticId) {
485         self.diagnostic().span_warn_with_code(sp, msg, code)
486     }
487     pub fn warn(&self, msg: &str) {
488         self.diagnostic().warn(msg)
489     }
490     pub fn opt_span_warn<S: Into<MultiSpan>>(&self, opt_sp: Option<S>, msg: &str) {
491         match opt_sp {
492             Some(sp) => self.span_warn(sp, msg),
493             None => self.warn(msg),
494         }
495     }
496     /// Delay a span_bug() call until abort_if_errors()
497     #[track_caller]
498     pub fn delay_span_bug<S: Into<MultiSpan>>(&self, sp: S, msg: &str) {
499         self.diagnostic().delay_span_bug(sp, msg)
500     }
501
502     /// Used for code paths of expensive computations that should only take place when
503     /// warnings or errors are emitted. If no messages are emitted ("good path"), then
504     /// it's likely a bug.
505     pub fn delay_good_path_bug(&self, msg: &str) {
506         if self.opts.debugging_opts.print_type_sizes
507             || self.opts.debugging_opts.query_dep_graph
508             || self.opts.debugging_opts.dump_mir.is_some()
509             || self.opts.debugging_opts.unpretty.is_some()
510             || self.opts.output_types.contains_key(&OutputType::Mir)
511             || std::env::var_os("RUSTC_LOG").is_some()
512         {
513             return;
514         }
515
516         self.diagnostic().delay_good_path_bug(msg)
517     }
518
519     pub fn note_without_error(&self, msg: &str) {
520         self.diagnostic().note_without_error(msg)
521     }
522     pub fn span_note_without_error<S: Into<MultiSpan>>(&self, sp: S, msg: &str) {
523         self.diagnostic().span_note_without_error(sp, msg)
524     }
525     pub fn struct_note_without_error(&self, msg: &str) -> DiagnosticBuilder<'_> {
526         self.diagnostic().struct_note_without_error(msg)
527     }
528
529     pub fn diagnostic(&self) -> &rustc_errors::Handler {
530         &self.parse_sess.span_diagnostic
531     }
532
533     /// Analogous to calling methods on the given `DiagnosticBuilder`, but
534     /// deduplicates on lint ID, span (if any), and message for this `Session`
535     fn diag_once<'a, 'b>(
536         &'a self,
537         diag_builder: &'b mut DiagnosticBuilder<'a>,
538         method: DiagnosticBuilderMethod,
539         msg_id: DiagnosticMessageId,
540         message: &str,
541         span_maybe: Option<Span>,
542     ) {
543         let id_span_message = (msg_id, span_maybe, message.to_owned());
544         let fresh = self.one_time_diagnostics.borrow_mut().insert(id_span_message);
545         if fresh {
546             match method {
547                 DiagnosticBuilderMethod::Note => {
548                     diag_builder.note(message);
549                 }
550                 DiagnosticBuilderMethod::SpanNote => {
551                     let span = span_maybe.expect("`span_note` needs a span");
552                     diag_builder.span_note(span, message);
553                 }
554                 DiagnosticBuilderMethod::SpanSuggestion(suggestion) => {
555                     let span = span_maybe.expect("`span_suggestion_*` needs a span");
556                     diag_builder.span_suggestion(
557                         span,
558                         message,
559                         suggestion,
560                         Applicability::Unspecified,
561                     );
562                 }
563             }
564         }
565     }
566
567     pub fn diag_span_note_once<'a, 'b>(
568         &'a self,
569         diag_builder: &'b mut DiagnosticBuilder<'a>,
570         msg_id: DiagnosticMessageId,
571         span: Span,
572         message: &str,
573     ) {
574         self.diag_once(
575             diag_builder,
576             DiagnosticBuilderMethod::SpanNote,
577             msg_id,
578             message,
579             Some(span),
580         );
581     }
582
583     pub fn diag_note_once<'a, 'b>(
584         &'a self,
585         diag_builder: &'b mut DiagnosticBuilder<'a>,
586         msg_id: DiagnosticMessageId,
587         message: &str,
588     ) {
589         self.diag_once(diag_builder, DiagnosticBuilderMethod::Note, msg_id, message, None);
590     }
591
592     pub fn diag_span_suggestion_once<'a, 'b>(
593         &'a self,
594         diag_builder: &'b mut DiagnosticBuilder<'a>,
595         msg_id: DiagnosticMessageId,
596         span: Span,
597         message: &str,
598         suggestion: String,
599     ) {
600         self.diag_once(
601             diag_builder,
602             DiagnosticBuilderMethod::SpanSuggestion(suggestion),
603             msg_id,
604             message,
605             Some(span),
606         );
607     }
608
609     #[inline]
610     pub fn source_map(&self) -> &SourceMap {
611         self.parse_sess.source_map()
612     }
613     pub fn verbose(&self) -> bool {
614         self.opts.debugging_opts.verbose
615     }
616     pub fn time_passes(&self) -> bool {
617         self.opts.debugging_opts.time_passes || self.opts.debugging_opts.time
618     }
619     pub fn instrument_mcount(&self) -> bool {
620         self.opts.debugging_opts.instrument_mcount
621     }
622     pub fn time_llvm_passes(&self) -> bool {
623         self.opts.debugging_opts.time_llvm_passes
624     }
625     pub fn meta_stats(&self) -> bool {
626         self.opts.debugging_opts.meta_stats
627     }
628     pub fn asm_comments(&self) -> bool {
629         self.opts.debugging_opts.asm_comments
630     }
631     pub fn verify_llvm_ir(&self) -> bool {
632         self.opts.debugging_opts.verify_llvm_ir || option_env!("RUSTC_VERIFY_LLVM_IR").is_some()
633     }
634     pub fn borrowck_stats(&self) -> bool {
635         self.opts.debugging_opts.borrowck_stats
636     }
637     pub fn print_llvm_passes(&self) -> bool {
638         self.opts.debugging_opts.print_llvm_passes
639     }
640     pub fn binary_dep_depinfo(&self) -> bool {
641         self.opts.debugging_opts.binary_dep_depinfo
642     }
643
644     /// Gets the features enabled for the current compilation session.
645     /// DO NOT USE THIS METHOD if there is a TyCtxt available, as it circumvents
646     /// dependency tracking. Use tcx.features() instead.
647     #[inline]
648     pub fn features_untracked(&self) -> &rustc_feature::Features {
649         self.features.get().unwrap()
650     }
651
652     pub fn init_features(&self, features: rustc_feature::Features) {
653         match self.features.set(features) {
654             Ok(()) => {}
655             Err(_) => panic!("`features` was initialized twice"),
656         }
657     }
658
659     pub fn init_lint_store(&self, lint_store: Lrc<dyn SessionLintStore>) {
660         self.lint_store
661             .set(lint_store)
662             .map_err(|_| ())
663             .expect("`lint_store` was initialized twice");
664     }
665
666     /// Calculates the flavor of LTO to use for this compilation.
667     pub fn lto(&self) -> config::Lto {
668         // If our target has codegen requirements ignore the command line
669         if self.target.requires_lto {
670             return config::Lto::Fat;
671         }
672
673         // If the user specified something, return that. If they only said `-C
674         // lto` and we've for whatever reason forced off ThinLTO via the CLI,
675         // then ensure we can't use a ThinLTO.
676         match self.opts.cg.lto {
677             config::LtoCli::Unspecified => {
678                 // The compiler was invoked without the `-Clto` flag. Fall
679                 // through to the default handling
680             }
681             config::LtoCli::No => {
682                 // The user explicitly opted out of any kind of LTO
683                 return config::Lto::No;
684             }
685             config::LtoCli::Yes | config::LtoCli::Fat | config::LtoCli::NoParam => {
686                 // All of these mean fat LTO
687                 return config::Lto::Fat;
688             }
689             config::LtoCli::Thin => {
690                 return if self.opts.cli_forced_thinlto_off {
691                     config::Lto::Fat
692                 } else {
693                     config::Lto::Thin
694                 };
695             }
696         }
697
698         // Ok at this point the target doesn't require anything and the user
699         // hasn't asked for anything. Our next decision is whether or not
700         // we enable "auto" ThinLTO where we use multiple codegen units and
701         // then do ThinLTO over those codegen units. The logic below will
702         // either return `No` or `ThinLocal`.
703
704         // If processing command line options determined that we're incompatible
705         // with ThinLTO (e.g., `-C lto --emit llvm-ir`) then return that option.
706         if self.opts.cli_forced_thinlto_off {
707             return config::Lto::No;
708         }
709
710         // If `-Z thinlto` specified process that, but note that this is mostly
711         // a deprecated option now that `-C lto=thin` exists.
712         if let Some(enabled) = self.opts.debugging_opts.thinlto {
713             if enabled {
714                 return config::Lto::ThinLocal;
715             } else {
716                 return config::Lto::No;
717             }
718         }
719
720         // If there's only one codegen unit and LTO isn't enabled then there's
721         // no need for ThinLTO so just return false.
722         if self.codegen_units() == 1 {
723             return config::Lto::No;
724         }
725
726         // Now we're in "defaults" territory. By default we enable ThinLTO for
727         // optimized compiles (anything greater than O0).
728         match self.opts.optimize {
729             config::OptLevel::No => config::Lto::No,
730             _ => config::Lto::ThinLocal,
731         }
732     }
733
734     /// Returns the panic strategy for this compile session. If the user explicitly selected one
735     /// using '-C panic', use that, otherwise use the panic strategy defined by the target.
736     pub fn panic_strategy(&self) -> PanicStrategy {
737         self.opts.cg.panic.unwrap_or(self.target.panic_strategy)
738     }
739     pub fn fewer_names(&self) -> bool {
740         if let Some(fewer_names) = self.opts.debugging_opts.fewer_names {
741             fewer_names
742         } else {
743             let more_names = self.opts.output_types.contains_key(&OutputType::LlvmAssembly)
744                 || self.opts.output_types.contains_key(&OutputType::Bitcode)
745                 // AddressSanitizer and MemorySanitizer use alloca name when reporting an issue.
746                 || self.opts.debugging_opts.sanitizer.intersects(SanitizerSet::ADDRESS | SanitizerSet::MEMORY);
747             !more_names
748         }
749     }
750
751     pub fn unstable_options(&self) -> bool {
752         self.opts.debugging_opts.unstable_options
753     }
754     pub fn is_nightly_build(&self) -> bool {
755         self.opts.unstable_features.is_nightly_build()
756     }
757     pub fn overflow_checks(&self) -> bool {
758         self.opts
759             .cg
760             .overflow_checks
761             .or(self.opts.debugging_opts.force_overflow_checks)
762             .unwrap_or(self.opts.debug_assertions)
763     }
764
765     /// Check whether this compile session and crate type use static crt.
766     pub fn crt_static(&self, crate_type: Option<CrateType>) -> bool {
767         if !self.target.crt_static_respected {
768             // If the target does not opt in to crt-static support, use its default.
769             return self.target.crt_static_default;
770         }
771
772         let requested_features = self.opts.cg.target_feature.split(',');
773         let found_negative = requested_features.clone().any(|r| r == "-crt-static");
774         let found_positive = requested_features.clone().any(|r| r == "+crt-static");
775
776         if found_positive || found_negative {
777             found_positive
778         } else if crate_type == Some(CrateType::ProcMacro)
779             || crate_type == None && self.opts.crate_types.contains(&CrateType::ProcMacro)
780         {
781             // FIXME: When crate_type is not available,
782             // we use compiler options to determine the crate_type.
783             // We can't check `#![crate_type = "proc-macro"]` here.
784             false
785         } else {
786             self.target.crt_static_default
787         }
788     }
789
790     pub fn relocation_model(&self) -> RelocModel {
791         self.opts.cg.relocation_model.unwrap_or(self.target.relocation_model)
792     }
793
794     pub fn code_model(&self) -> Option<CodeModel> {
795         self.opts.cg.code_model.or(self.target.code_model)
796     }
797
798     pub fn tls_model(&self) -> TlsModel {
799         self.opts.debugging_opts.tls_model.unwrap_or(self.target.tls_model)
800     }
801
802     pub fn is_wasi_reactor(&self) -> bool {
803         self.target.options.os == "wasi"
804             && matches!(
805                 self.opts.debugging_opts.wasi_exec_model,
806                 Some(config::WasiExecModel::Reactor)
807             )
808     }
809
810     pub fn split_debuginfo(&self) -> SplitDebuginfo {
811         self.opts.cg.split_debuginfo.unwrap_or(self.target.split_debuginfo)
812     }
813
814     pub fn target_can_use_split_dwarf(&self) -> bool {
815         !self.target.is_like_windows && !self.target.is_like_osx
816     }
817
818     pub fn must_not_eliminate_frame_pointers(&self) -> bool {
819         // "mcount" function relies on stack pointer.
820         // See <https://sourceware.org/binutils/docs/gprof/Implementation.html>.
821         if self.instrument_mcount() {
822             true
823         } else if let Some(x) = self.opts.cg.force_frame_pointers {
824             x
825         } else {
826             !self.target.eliminate_frame_pointer
827         }
828     }
829
830     pub fn must_emit_unwind_tables(&self) -> bool {
831         // This is used to control the emission of the `uwtable` attribute on
832         // LLVM functions.
833         //
834         // At the very least, unwind tables are needed when compiling with
835         // `-C panic=unwind`.
836         //
837         // On some targets (including windows), however, exceptions include
838         // other events such as illegal instructions, segfaults, etc. This means
839         // that on Windows we end up still needing unwind tables even if the `-C
840         // panic=abort` flag is passed.
841         //
842         // You can also find more info on why Windows needs unwind tables in:
843         //      https://bugzilla.mozilla.org/show_bug.cgi?id=1302078
844         //
845         // If a target requires unwind tables, then they must be emitted.
846         // Otherwise, we can defer to the `-C force-unwind-tables=<yes/no>`
847         // value, if it is provided, or disable them, if not.
848         if self.panic_strategy() == PanicStrategy::Unwind {
849             true
850         } else if self.target.requires_uwtable {
851             true
852         } else {
853             self.opts.cg.force_unwind_tables.unwrap_or(false)
854         }
855     }
856
857     /// Returns the symbol name for the registrar function,
858     /// given the crate `Svh` and the function `DefIndex`.
859     pub fn generate_plugin_registrar_symbol(&self, disambiguator: CrateDisambiguator) -> String {
860         format!("__rustc_plugin_registrar_{}__", disambiguator.to_fingerprint().to_hex())
861     }
862
863     pub fn generate_proc_macro_decls_symbol(&self, disambiguator: CrateDisambiguator) -> String {
864         format!("__rustc_proc_macro_decls_{}__", disambiguator.to_fingerprint().to_hex())
865     }
866
867     pub fn target_filesearch(&self, kind: PathKind) -> filesearch::FileSearch<'_> {
868         filesearch::FileSearch::new(
869             &self.sysroot,
870             self.opts.target_triple.triple(),
871             &self.opts.search_paths,
872             // `target_tlib_path == None` means it's the same as `host_tlib_path`.
873             self.target_tlib_path.as_ref().unwrap_or(&self.host_tlib_path),
874             kind,
875         )
876     }
877     pub fn host_filesearch(&self, kind: PathKind) -> filesearch::FileSearch<'_> {
878         filesearch::FileSearch::new(
879             &self.sysroot,
880             config::host_triple(),
881             &self.opts.search_paths,
882             &self.host_tlib_path,
883             kind,
884         )
885     }
886
887     pub fn set_incr_session_load_dep_graph(&self, load: bool) {
888         let mut incr_comp_session = self.incr_comp_session.borrow_mut();
889
890         if let IncrCompSession::Active { ref mut load_dep_graph, .. } = *incr_comp_session {
891             *load_dep_graph = load;
892         }
893     }
894
895     pub fn incr_session_load_dep_graph(&self) -> bool {
896         let incr_comp_session = self.incr_comp_session.borrow();
897         match *incr_comp_session {
898             IncrCompSession::Active { load_dep_graph, .. } => load_dep_graph,
899             _ => false,
900         }
901     }
902
903     pub fn init_incr_comp_session(
904         &self,
905         session_dir: PathBuf,
906         lock_file: flock::Lock,
907         load_dep_graph: bool,
908     ) {
909         let mut incr_comp_session = self.incr_comp_session.borrow_mut();
910
911         if let IncrCompSession::NotInitialized = *incr_comp_session {
912         } else {
913             panic!("Trying to initialize IncrCompSession `{:?}`", *incr_comp_session)
914         }
915
916         *incr_comp_session =
917             IncrCompSession::Active { session_directory: session_dir, lock_file, load_dep_graph };
918     }
919
920     pub fn finalize_incr_comp_session(&self, new_directory_path: PathBuf) {
921         let mut incr_comp_session = self.incr_comp_session.borrow_mut();
922
923         if let IncrCompSession::Active { .. } = *incr_comp_session {
924         } else {
925             panic!("trying to finalize `IncrCompSession` `{:?}`", *incr_comp_session);
926         }
927
928         // Note: this will also drop the lock file, thus unlocking the directory.
929         *incr_comp_session = IncrCompSession::Finalized { session_directory: new_directory_path };
930     }
931
932     pub fn mark_incr_comp_session_as_invalid(&self) {
933         let mut incr_comp_session = self.incr_comp_session.borrow_mut();
934
935         let session_directory = match *incr_comp_session {
936             IncrCompSession::Active { ref session_directory, .. } => session_directory.clone(),
937             IncrCompSession::InvalidBecauseOfErrors { .. } => return,
938             _ => panic!("trying to invalidate `IncrCompSession` `{:?}`", *incr_comp_session),
939         };
940
941         // Note: this will also drop the lock file, thus unlocking the directory.
942         *incr_comp_session = IncrCompSession::InvalidBecauseOfErrors { session_directory };
943     }
944
945     pub fn incr_comp_session_dir(&self) -> cell::Ref<'_, PathBuf> {
946         let incr_comp_session = self.incr_comp_session.borrow();
947         cell::Ref::map(incr_comp_session, |incr_comp_session| match *incr_comp_session {
948             IncrCompSession::NotInitialized => panic!(
949                 "trying to get session directory from `IncrCompSession`: {:?}",
950                 *incr_comp_session,
951             ),
952             IncrCompSession::Active { ref session_directory, .. }
953             | IncrCompSession::Finalized { ref session_directory }
954             | IncrCompSession::InvalidBecauseOfErrors { ref session_directory } => {
955                 session_directory
956             }
957         })
958     }
959
960     pub fn incr_comp_session_dir_opt(&self) -> Option<cell::Ref<'_, PathBuf>> {
961         self.opts.incremental.as_ref().map(|_| self.incr_comp_session_dir())
962     }
963
964     pub fn print_perf_stats(&self) {
965         eprintln!(
966             "Total time spent computing symbol hashes:      {}",
967             duration_to_secs_str(*self.perf_stats.symbol_hash_time.lock())
968         );
969         eprintln!(
970             "Total queries canonicalized:                   {}",
971             self.perf_stats.queries_canonicalized.load(Ordering::Relaxed)
972         );
973         eprintln!(
974             "normalize_generic_arg_after_erasing_regions:   {}",
975             self.perf_stats.normalize_generic_arg_after_erasing_regions.load(Ordering::Relaxed)
976         );
977         eprintln!(
978             "normalize_projection_ty:                       {}",
979             self.perf_stats.normalize_projection_ty.load(Ordering::Relaxed)
980         );
981     }
982
983     /// We want to know if we're allowed to do an optimization for crate foo from -z fuel=foo=n.
984     /// This expends fuel if applicable, and records fuel if applicable.
985     pub fn consider_optimizing<T: Fn() -> String>(&self, crate_name: &str, msg: T) -> bool {
986         let mut ret = true;
987         if let Some(ref c) = self.optimization_fuel_crate {
988             if c == crate_name {
989                 assert_eq!(self.threads(), 1);
990                 let mut fuel = self.optimization_fuel.lock();
991                 ret = fuel.remaining != 0;
992                 if fuel.remaining == 0 && !fuel.out_of_fuel {
993                     self.warn(&format!("optimization-fuel-exhausted: {}", msg()));
994                     fuel.out_of_fuel = true;
995                 } else if fuel.remaining > 0 {
996                     fuel.remaining -= 1;
997                 }
998             }
999         }
1000         if let Some(ref c) = self.print_fuel_crate {
1001             if c == crate_name {
1002                 assert_eq!(self.threads(), 1);
1003                 self.print_fuel.fetch_add(1, SeqCst);
1004             }
1005         }
1006         ret
1007     }
1008
1009     /// Returns the number of query threads that should be used for this
1010     /// compilation
1011     pub fn threads(&self) -> usize {
1012         self.opts.debugging_opts.threads
1013     }
1014
1015     /// Returns the number of codegen units that should be used for this
1016     /// compilation
1017     pub fn codegen_units(&self) -> usize {
1018         if let Some(n) = self.opts.cli_forced_codegen_units {
1019             return n;
1020         }
1021         if let Some(n) = self.target.default_codegen_units {
1022             return n as usize;
1023         }
1024
1025         // If incremental compilation is turned on, we default to a high number
1026         // codegen units in order to reduce the "collateral damage" small
1027         // changes cause.
1028         if self.opts.incremental.is_some() {
1029             return 256;
1030         }
1031
1032         // Why is 16 codegen units the default all the time?
1033         //
1034         // The main reason for enabling multiple codegen units by default is to
1035         // leverage the ability for the codegen backend to do codegen and
1036         // optimization in parallel. This allows us, especially for large crates, to
1037         // make good use of all available resources on the machine once we've
1038         // hit that stage of compilation. Large crates especially then often
1039         // take a long time in codegen/optimization and this helps us amortize that
1040         // cost.
1041         //
1042         // Note that a high number here doesn't mean that we'll be spawning a
1043         // large number of threads in parallel. The backend of rustc contains
1044         // global rate limiting through the `jobserver` crate so we'll never
1045         // overload the system with too much work, but rather we'll only be
1046         // optimizing when we're otherwise cooperating with other instances of
1047         // rustc.
1048         //
1049         // Rather a high number here means that we should be able to keep a lot
1050         // of idle cpus busy. By ensuring that no codegen unit takes *too* long
1051         // to build we'll be guaranteed that all cpus will finish pretty closely
1052         // to one another and we should make relatively optimal use of system
1053         // resources
1054         //
1055         // Note that the main cost of codegen units is that it prevents LLVM
1056         // from inlining across codegen units. Users in general don't have a lot
1057         // of control over how codegen units are split up so it's our job in the
1058         // compiler to ensure that undue performance isn't lost when using
1059         // codegen units (aka we can't require everyone to slap `#[inline]` on
1060         // everything).
1061         //
1062         // If we're compiling at `-O0` then the number doesn't really matter too
1063         // much because performance doesn't matter and inlining is ok to lose.
1064         // In debug mode we just want to try to guarantee that no cpu is stuck
1065         // doing work that could otherwise be farmed to others.
1066         //
1067         // In release mode, however (O1 and above) performance does indeed
1068         // matter! To recover the loss in performance due to inlining we'll be
1069         // enabling ThinLTO by default (the function for which is just below).
1070         // This will ensure that we recover any inlining wins we otherwise lost
1071         // through codegen unit partitioning.
1072         //
1073         // ---
1074         //
1075         // Ok that's a lot of words but the basic tl;dr; is that we want a high
1076         // number here -- but not too high. Additionally we're "safe" to have it
1077         // always at the same number at all optimization levels.
1078         //
1079         // As a result 16 was chosen here! Mostly because it was a power of 2
1080         // and most benchmarks agreed it was roughly a local optimum. Not very
1081         // scientific.
1082         16
1083     }
1084
1085     pub fn teach(&self, code: &DiagnosticId) -> bool {
1086         self.opts.debugging_opts.teach && self.diagnostic().must_teach(code)
1087     }
1088
1089     pub fn rust_2015(&self) -> bool {
1090         self.opts.edition == Edition::Edition2015
1091     }
1092
1093     /// Are we allowed to use features from the Rust 2018 edition?
1094     pub fn rust_2018(&self) -> bool {
1095         self.opts.edition >= Edition::Edition2018
1096     }
1097
1098     /// Are we allowed to use features from the Rust 2021 edition?
1099     pub fn rust_2021(&self) -> bool {
1100         self.opts.edition >= Edition::Edition2021
1101     }
1102
1103     pub fn edition(&self) -> Edition {
1104         self.opts.edition
1105     }
1106
1107     /// Returns `true` if we cannot skip the PLT for shared library calls.
1108     pub fn needs_plt(&self) -> bool {
1109         // Check if the current target usually needs PLT to be enabled.
1110         // The user can use the command line flag to override it.
1111         let needs_plt = self.target.needs_plt;
1112
1113         let dbg_opts = &self.opts.debugging_opts;
1114
1115         let relro_level = dbg_opts.relro_level.unwrap_or(self.target.relro_level);
1116
1117         // Only enable this optimization by default if full relro is also enabled.
1118         // In this case, lazy binding was already unavailable, so nothing is lost.
1119         // This also ensures `-Wl,-z,now` is supported by the linker.
1120         let full_relro = RelroLevel::Full == relro_level;
1121
1122         // If user didn't explicitly forced us to use / skip the PLT,
1123         // then try to skip it where possible.
1124         dbg_opts.plt.unwrap_or(needs_plt || !full_relro)
1125     }
1126
1127     /// Checks if LLVM lifetime markers should be emitted.
1128     pub fn emit_lifetime_markers(&self) -> bool {
1129         self.opts.optimize != config::OptLevel::No
1130         // AddressSanitizer uses lifetimes to detect use after scope bugs.
1131         // MemorySanitizer uses lifetimes to detect use of uninitialized stack variables.
1132         // HWAddressSanitizer will use lifetimes to detect use after scope bugs in the future.
1133         || self.opts.debugging_opts.sanitizer.intersects(SanitizerSet::ADDRESS | SanitizerSet::MEMORY | SanitizerSet::HWADDRESS)
1134     }
1135
1136     pub fn link_dead_code(&self) -> bool {
1137         self.opts.cg.link_dead_code.unwrap_or(false)
1138     }
1139
1140     pub fn mark_attr_known(&self, attr: &Attribute) {
1141         self.known_attrs.lock().mark(attr)
1142     }
1143
1144     pub fn is_attr_known(&self, attr: &Attribute) -> bool {
1145         self.known_attrs.lock().is_marked(attr)
1146     }
1147
1148     pub fn mark_attr_used(&self, attr: &Attribute) {
1149         self.used_attrs.lock().mark(attr)
1150     }
1151
1152     pub fn is_attr_used(&self, attr: &Attribute) -> bool {
1153         self.used_attrs.lock().is_marked(attr)
1154     }
1155
1156     /// Returns `true` if the attribute's path matches the argument. If it
1157     /// matches, then the attribute is marked as used.
1158     ///
1159     /// This method should only be used by rustc, other tools can use
1160     /// `Attribute::has_name` instead, because only rustc is supposed to report
1161     /// the `unused_attributes` lint. (`MetaItem` and `NestedMetaItem` are
1162     /// produced by lowering an `Attribute` and don't have identity, so they
1163     /// only have the `has_name` method, and you need to mark the original
1164     /// `Attribute` as used when necessary.)
1165     pub fn check_name(&self, attr: &Attribute, name: Symbol) -> bool {
1166         let matches = attr.has_name(name);
1167         if matches {
1168             self.mark_attr_used(attr);
1169         }
1170         matches
1171     }
1172
1173     pub fn is_proc_macro_attr(&self, attr: &Attribute) -> bool {
1174         [sym::proc_macro, sym::proc_macro_attribute, sym::proc_macro_derive]
1175             .iter()
1176             .any(|kind| self.check_name(attr, *kind))
1177     }
1178
1179     pub fn contains_name(&self, attrs: &[Attribute], name: Symbol) -> bool {
1180         attrs.iter().any(|item| self.check_name(item, name))
1181     }
1182
1183     pub fn find_by_name<'a>(
1184         &'a self,
1185         attrs: &'a [Attribute],
1186         name: Symbol,
1187     ) -> Option<&'a Attribute> {
1188         attrs.iter().find(|attr| self.check_name(attr, name))
1189     }
1190
1191     pub fn filter_by_name<'a>(
1192         &'a self,
1193         attrs: &'a [Attribute],
1194         name: Symbol,
1195     ) -> impl Iterator<Item = &'a Attribute> {
1196         attrs.iter().filter(move |attr| self.check_name(attr, name))
1197     }
1198
1199     pub fn first_attr_value_str_by_name(
1200         &self,
1201         attrs: &[Attribute],
1202         name: Symbol,
1203     ) -> Option<Symbol> {
1204         attrs.iter().find(|at| self.check_name(at, name)).and_then(|at| at.value_str())
1205     }
1206 }
1207
1208 fn default_emitter(
1209     sopts: &config::Options,
1210     registry: rustc_errors::registry::Registry,
1211     source_map: Lrc<SourceMap>,
1212     emitter_dest: Option<Box<dyn Write + Send>>,
1213 ) -> Box<dyn Emitter + sync::Send> {
1214     let macro_backtrace = sopts.debugging_opts.macro_backtrace;
1215     match (sopts.error_format, emitter_dest) {
1216         (config::ErrorOutputType::HumanReadable(kind), dst) => {
1217             let (short, color_config) = kind.unzip();
1218
1219             if let HumanReadableErrorType::AnnotateSnippet(_) = kind {
1220                 let emitter =
1221                     AnnotateSnippetEmitterWriter::new(Some(source_map), short, macro_backtrace);
1222                 Box::new(emitter.ui_testing(sopts.debugging_opts.ui_testing))
1223             } else {
1224                 let emitter = match dst {
1225                     None => EmitterWriter::stderr(
1226                         color_config,
1227                         Some(source_map),
1228                         short,
1229                         sopts.debugging_opts.teach,
1230                         sopts.debugging_opts.terminal_width,
1231                         macro_backtrace,
1232                     ),
1233                     Some(dst) => EmitterWriter::new(
1234                         dst,
1235                         Some(source_map),
1236                         short,
1237                         false, // no teach messages when writing to a buffer
1238                         false, // no colors when writing to a buffer
1239                         None,  // no terminal width
1240                         macro_backtrace,
1241                     ),
1242                 };
1243                 Box::new(emitter.ui_testing(sopts.debugging_opts.ui_testing))
1244             }
1245         }
1246         (config::ErrorOutputType::Json { pretty, json_rendered }, None) => Box::new(
1247             JsonEmitter::stderr(
1248                 Some(registry),
1249                 source_map,
1250                 pretty,
1251                 json_rendered,
1252                 sopts.debugging_opts.terminal_width,
1253                 macro_backtrace,
1254             )
1255             .ui_testing(sopts.debugging_opts.ui_testing),
1256         ),
1257         (config::ErrorOutputType::Json { pretty, json_rendered }, Some(dst)) => Box::new(
1258             JsonEmitter::new(
1259                 dst,
1260                 Some(registry),
1261                 source_map,
1262                 pretty,
1263                 json_rendered,
1264                 sopts.debugging_opts.terminal_width,
1265                 macro_backtrace,
1266             )
1267             .ui_testing(sopts.debugging_opts.ui_testing),
1268         ),
1269     }
1270 }
1271
1272 pub enum DiagnosticOutput {
1273     Default,
1274     Raw(Box<dyn Write + Send>),
1275 }
1276
1277 pub fn build_session(
1278     sopts: config::Options,
1279     local_crate_source_file: Option<PathBuf>,
1280     registry: rustc_errors::registry::Registry,
1281     diagnostics_output: DiagnosticOutput,
1282     driver_lint_caps: FxHashMap<lint::LintId, lint::Level>,
1283     file_loader: Option<Box<dyn FileLoader + Send + Sync + 'static>>,
1284     target_override: Option<Target>,
1285 ) -> Session {
1286     // FIXME: This is not general enough to make the warning lint completely override
1287     // normal diagnostic warnings, since the warning lint can also be denied and changed
1288     // later via the source code.
1289     let warnings_allow = sopts
1290         .lint_opts
1291         .iter()
1292         .filter(|&&(ref key, _)| *key == "warnings")
1293         .map(|&(_, ref level)| *level == lint::Allow)
1294         .last()
1295         .unwrap_or(false);
1296     let cap_lints_allow = sopts.lint_cap.map_or(false, |cap| cap == lint::Allow);
1297     let can_emit_warnings = !(warnings_allow || cap_lints_allow);
1298
1299     let write_dest = match diagnostics_output {
1300         DiagnosticOutput::Default => None,
1301         DiagnosticOutput::Raw(write) => Some(write),
1302     };
1303
1304     let target_cfg = config::build_target_config(&sopts, target_override);
1305     let host_triple = TargetTriple::from_triple(config::host_triple());
1306     let host = Target::search(&host_triple).unwrap_or_else(|e| {
1307         early_error(sopts.error_format, &format!("Error loading host specification: {}", e))
1308     });
1309
1310     let loader = file_loader.unwrap_or_else(|| Box::new(RealFileLoader));
1311     let hash_kind = sopts.debugging_opts.src_hash_algorithm.unwrap_or_else(|| {
1312         if target_cfg.is_like_msvc {
1313             SourceFileHashAlgorithm::Sha1
1314         } else {
1315             SourceFileHashAlgorithm::Md5
1316         }
1317     });
1318     let source_map = Lrc::new(SourceMap::with_file_loader_and_hash_kind(
1319         loader,
1320         sopts.file_path_mapping(),
1321         hash_kind,
1322     ));
1323     let emitter = default_emitter(&sopts, registry, source_map.clone(), write_dest);
1324
1325     let span_diagnostic = rustc_errors::Handler::with_emitter_and_flags(
1326         emitter,
1327         sopts.debugging_opts.diagnostic_handler_flags(can_emit_warnings),
1328     );
1329
1330     let self_profiler = if let SwitchWithOptPath::Enabled(ref d) = sopts.debugging_opts.self_profile
1331     {
1332         let directory =
1333             if let Some(ref directory) = d { directory } else { std::path::Path::new(".") };
1334
1335         let profiler = SelfProfiler::new(
1336             directory,
1337             sopts.crate_name.as_deref(),
1338             &sopts.debugging_opts.self_profile_events,
1339         );
1340         match profiler {
1341             Ok(profiler) => Some(Arc::new(profiler)),
1342             Err(e) => {
1343                 early_warn(sopts.error_format, &format!("failed to create profiler: {}", e));
1344                 None
1345             }
1346         }
1347     } else {
1348         None
1349     };
1350
1351     let mut parse_sess = ParseSess::with_span_handler(span_diagnostic, source_map);
1352     parse_sess.assume_incomplete_release = sopts.debugging_opts.assume_incomplete_release;
1353     let sysroot = match &sopts.maybe_sysroot {
1354         Some(sysroot) => sysroot.clone(),
1355         None => filesearch::get_or_default_sysroot(),
1356     };
1357
1358     let host_triple = config::host_triple();
1359     let target_triple = sopts.target_triple.triple();
1360     let host_tlib_path = SearchPath::from_sysroot_and_triple(&sysroot, host_triple);
1361     let target_tlib_path = if host_triple == target_triple {
1362         None
1363     } else {
1364         Some(SearchPath::from_sysroot_and_triple(&sysroot, target_triple))
1365     };
1366
1367     let file_path_mapping = sopts.file_path_mapping();
1368
1369     let local_crate_source_file =
1370         local_crate_source_file.map(|path| file_path_mapping.map_prefix(path).0);
1371
1372     let optimization_fuel_crate = sopts.debugging_opts.fuel.as_ref().map(|i| i.0.clone());
1373     let optimization_fuel = Lock::new(OptimizationFuel {
1374         remaining: sopts.debugging_opts.fuel.as_ref().map_or(0, |i| i.1),
1375         out_of_fuel: false,
1376     });
1377     let print_fuel_crate = sopts.debugging_opts.print_fuel.clone();
1378     let print_fuel = AtomicU64::new(0);
1379
1380     let working_dir = env::current_dir().unwrap_or_else(|e| {
1381         parse_sess.span_diagnostic.fatal(&format!("Current directory is invalid: {}", e)).raise()
1382     });
1383     let working_dir = file_path_mapping.map_prefix(working_dir);
1384
1385     let cgu_reuse_tracker = if sopts.debugging_opts.query_dep_graph {
1386         CguReuseTracker::new()
1387     } else {
1388         CguReuseTracker::new_disabled()
1389     };
1390
1391     let prof = SelfProfilerRef::new(
1392         self_profiler,
1393         sopts.debugging_opts.time_passes || sopts.debugging_opts.time,
1394         sopts.debugging_opts.time_passes,
1395     );
1396
1397     let ctfe_backtrace = Lock::new(match env::var("RUSTC_CTFE_BACKTRACE") {
1398         Ok(ref val) if val == "immediate" => CtfeBacktrace::Immediate,
1399         Ok(ref val) if val != "0" => CtfeBacktrace::Capture,
1400         _ => CtfeBacktrace::Disabled,
1401     });
1402
1403     // Try to find a directory containing the Rust `src`, for more details see
1404     // the doc comment on the `real_rust_source_base_dir` field.
1405     let real_rust_source_base_dir = {
1406         // This is the location used by the `rust-src` `rustup` component.
1407         let mut candidate = sysroot.join("lib/rustlib/src/rust");
1408         if let Ok(metadata) = candidate.symlink_metadata() {
1409             // Replace the symlink rustbuild creates, with its destination.
1410             // We could try to use `fs::canonicalize` instead, but that might
1411             // produce unnecessarily verbose path.
1412             if metadata.file_type().is_symlink() {
1413                 if let Ok(symlink_dest) = std::fs::read_link(&candidate) {
1414                     candidate = symlink_dest;
1415                 }
1416             }
1417         }
1418
1419         // Only use this directory if it has a file we can expect to always find.
1420         if candidate.join("library/std/src/lib.rs").is_file() { Some(candidate) } else { None }
1421     };
1422
1423     let asm_arch =
1424         if target_cfg.allow_asm { InlineAsmArch::from_str(&target_cfg.arch).ok() } else { None };
1425
1426     let sess = Session {
1427         target: target_cfg,
1428         host,
1429         opts: sopts,
1430         host_tlib_path,
1431         target_tlib_path,
1432         parse_sess,
1433         sysroot,
1434         local_crate_source_file,
1435         working_dir,
1436         one_time_diagnostics: Default::default(),
1437         crate_types: OnceCell::new(),
1438         crate_disambiguator: OnceCell::new(),
1439         features: OnceCell::new(),
1440         lint_store: OnceCell::new(),
1441         recursion_limit: OnceCell::new(),
1442         type_length_limit: OnceCell::new(),
1443         const_eval_limit: OnceCell::new(),
1444         incr_comp_session: OneThread::new(RefCell::new(IncrCompSession::NotInitialized)),
1445         cgu_reuse_tracker,
1446         prof,
1447         perf_stats: PerfStats {
1448             symbol_hash_time: Lock::new(Duration::from_secs(0)),
1449             queries_canonicalized: AtomicUsize::new(0),
1450             normalize_generic_arg_after_erasing_regions: AtomicUsize::new(0),
1451             normalize_projection_ty: AtomicUsize::new(0),
1452         },
1453         code_stats: Default::default(),
1454         optimization_fuel_crate,
1455         optimization_fuel,
1456         print_fuel_crate,
1457         print_fuel,
1458         jobserver: jobserver::client(),
1459         driver_lint_caps,
1460         trait_methods_not_found: Lock::new(Default::default()),
1461         confused_type_with_std_module: Lock::new(Default::default()),
1462         system_library_path: OneThread::new(RefCell::new(Default::default())),
1463         ctfe_backtrace,
1464         miri_unleashed_features: Lock::new(Default::default()),
1465         real_rust_source_base_dir,
1466         asm_arch,
1467         target_features: FxHashSet::default(),
1468         known_attrs: Lock::new(MarkedAttrs::new()),
1469         used_attrs: Lock::new(MarkedAttrs::new()),
1470         if_let_suggestions: Default::default(),
1471     };
1472
1473     validate_commandline_args_with_session_available(&sess);
1474
1475     sess
1476 }
1477
1478 // If it is useful to have a Session available already for validating a
1479 // commandline argument, you can do so here.
1480 fn validate_commandline_args_with_session_available(sess: &Session) {
1481     // Since we don't know if code in an rlib will be linked to statically or
1482     // dynamically downstream, rustc generates `__imp_` symbols that help linkers
1483     // on Windows deal with this lack of knowledge (#27438). Unfortunately,
1484     // these manually generated symbols confuse LLD when it tries to merge
1485     // bitcode during ThinLTO. Therefore we disallow dynamic linking on Windows
1486     // when compiling for LLD ThinLTO. This way we can validly just not generate
1487     // the `dllimport` attributes and `__imp_` symbols in that case.
1488     if sess.opts.cg.linker_plugin_lto.enabled()
1489         && sess.opts.cg.prefer_dynamic
1490         && sess.target.is_like_windows
1491     {
1492         sess.err(
1493             "Linker plugin based LTO is not supported together with \
1494                   `-C prefer-dynamic` when targeting Windows-like targets",
1495         );
1496     }
1497
1498     // Make sure that any given profiling data actually exists so LLVM can't
1499     // decide to silently skip PGO.
1500     if let Some(ref path) = sess.opts.cg.profile_use {
1501         if !path.exists() {
1502             sess.err(&format!(
1503                 "File `{}` passed to `-C profile-use` does not exist.",
1504                 path.display()
1505             ));
1506         }
1507     }
1508
1509     // Unwind tables cannot be disabled if the target requires them.
1510     if let Some(include_uwtables) = sess.opts.cg.force_unwind_tables {
1511         if sess.panic_strategy() == PanicStrategy::Unwind && !include_uwtables {
1512             sess.err(
1513                 "panic=unwind requires unwind tables, they cannot be disabled \
1514                      with `-C force-unwind-tables=no`.",
1515             );
1516         }
1517
1518         if sess.target.requires_uwtable && !include_uwtables {
1519             sess.err(
1520                 "target requires unwind tables, they cannot be disabled with \
1521                      `-C force-unwind-tables=no`.",
1522             );
1523         }
1524     }
1525
1526     // PGO does not work reliably with panic=unwind on Windows. Let's make it
1527     // an error to combine the two for now. It always runs into an assertions
1528     // if LLVM is built with assertions, but without assertions it sometimes
1529     // does not crash and will probably generate a corrupted binary.
1530     // We should only display this error if we're actually going to run PGO.
1531     // If we're just supposed to print out some data, don't show the error (#61002).
1532     if sess.opts.cg.profile_generate.enabled()
1533         && sess.target.is_like_msvc
1534         && sess.panic_strategy() == PanicStrategy::Unwind
1535         && sess.opts.prints.iter().all(|&p| p == PrintRequest::NativeStaticLibs)
1536     {
1537         sess.err(
1538             "Profile-guided optimization does not yet work in conjunction \
1539                   with `-Cpanic=unwind` on Windows when targeting MSVC. \
1540                   See issue #61002 <https://github.com/rust-lang/rust/issues/61002> \
1541                   for more information.",
1542         );
1543     }
1544
1545     const ASAN_SUPPORTED_TARGETS: &[&str] = &[
1546         "aarch64-apple-darwin",
1547         "aarch64-fuchsia",
1548         "aarch64-unknown-linux-gnu",
1549         "x86_64-apple-darwin",
1550         "x86_64-fuchsia",
1551         "x86_64-unknown-freebsd",
1552         "x86_64-unknown-linux-gnu",
1553     ];
1554     const LSAN_SUPPORTED_TARGETS: &[&str] = &[
1555         "aarch64-apple-darwin",
1556         "aarch64-unknown-linux-gnu",
1557         "x86_64-apple-darwin",
1558         "x86_64-unknown-linux-gnu",
1559     ];
1560     const MSAN_SUPPORTED_TARGETS: &[&str] =
1561         &["aarch64-unknown-linux-gnu", "x86_64-unknown-freebsd", "x86_64-unknown-linux-gnu"];
1562     const TSAN_SUPPORTED_TARGETS: &[&str] = &[
1563         "aarch64-apple-darwin",
1564         "aarch64-unknown-linux-gnu",
1565         "x86_64-apple-darwin",
1566         "x86_64-unknown-freebsd",
1567         "x86_64-unknown-linux-gnu",
1568     ];
1569     const HWASAN_SUPPORTED_TARGETS: &[&str] =
1570         &["aarch64-linux-android", "aarch64-unknown-linux-gnu"];
1571
1572     // Sanitizers can only be used on some tested platforms.
1573     for s in sess.opts.debugging_opts.sanitizer {
1574         let supported_targets = match s {
1575             SanitizerSet::ADDRESS => ASAN_SUPPORTED_TARGETS,
1576             SanitizerSet::LEAK => LSAN_SUPPORTED_TARGETS,
1577             SanitizerSet::MEMORY => MSAN_SUPPORTED_TARGETS,
1578             SanitizerSet::THREAD => TSAN_SUPPORTED_TARGETS,
1579             SanitizerSet::HWADDRESS => HWASAN_SUPPORTED_TARGETS,
1580             _ => panic!("unrecognized sanitizer {}", s),
1581         };
1582         if !supported_targets.contains(&&*sess.opts.target_triple.triple()) {
1583             sess.err(&format!(
1584                 "`-Zsanitizer={}` only works with targets: {}",
1585                 s,
1586                 supported_targets.join(", ")
1587             ));
1588         }
1589         let conflicting = sess.opts.debugging_opts.sanitizer - s;
1590         if !conflicting.is_empty() {
1591             sess.err(&format!(
1592                 "`-Zsanitizer={}` is incompatible with `-Zsanitizer={}`",
1593                 s, conflicting,
1594             ));
1595             // Don't report additional errors.
1596             break;
1597         }
1598     }
1599 }
1600
1601 /// Holds data on the current incremental compilation session, if there is one.
1602 #[derive(Debug)]
1603 pub enum IncrCompSession {
1604     /// This is the state the session will be in until the incr. comp. dir is
1605     /// needed.
1606     NotInitialized,
1607     /// This is the state during which the session directory is private and can
1608     /// be modified.
1609     Active { session_directory: PathBuf, lock_file: flock::Lock, load_dep_graph: bool },
1610     /// This is the state after the session directory has been finalized. In this
1611     /// state, the contents of the directory must not be modified any more.
1612     Finalized { session_directory: PathBuf },
1613     /// This is an error state that is reached when some compilation error has
1614     /// occurred. It indicates that the contents of the session directory must
1615     /// not be used, since they might be invalid.
1616     InvalidBecauseOfErrors { session_directory: PathBuf },
1617 }
1618
1619 pub fn early_error(output: config::ErrorOutputType, msg: &str) -> ! {
1620     let emitter: Box<dyn Emitter + sync::Send> = match output {
1621         config::ErrorOutputType::HumanReadable(kind) => {
1622             let (short, color_config) = kind.unzip();
1623             Box::new(EmitterWriter::stderr(color_config, None, short, false, None, false))
1624         }
1625         config::ErrorOutputType::Json { pretty, json_rendered } => {
1626             Box::new(JsonEmitter::basic(pretty, json_rendered, None, false))
1627         }
1628     };
1629     let handler = rustc_errors::Handler::with_emitter(true, None, emitter);
1630     handler.struct_fatal(msg).emit();
1631     rustc_errors::FatalError.raise();
1632 }
1633
1634 pub fn early_warn(output: config::ErrorOutputType, msg: &str) {
1635     let emitter: Box<dyn Emitter + sync::Send> = match output {
1636         config::ErrorOutputType::HumanReadable(kind) => {
1637             let (short, color_config) = kind.unzip();
1638             Box::new(EmitterWriter::stderr(color_config, None, short, false, None, false))
1639         }
1640         config::ErrorOutputType::Json { pretty, json_rendered } => {
1641             Box::new(JsonEmitter::basic(pretty, json_rendered, None, false))
1642         }
1643     };
1644     let handler = rustc_errors::Handler::with_emitter(true, None, emitter);
1645     handler.struct_warn(msg).emit();
1646 }