rustc_middle/ty/
mod.rs

1//! Defines how the compiler represents types internally.
2//!
3//! Two important entities in this module are:
4//!
5//! - [`rustc_middle::ty::Ty`], used to represent the semantics of a type.
6//! - [`rustc_middle::ty::TyCtxt`], the central data structure in the compiler.
7//!
8//! For more information, see ["The `ty` module: representing types"] in the rustc-dev-guide.
9//!
10//! ["The `ty` module: representing types"]: https://rustc-dev-guide.rust-lang.org/ty.html
11
12#![allow(rustc::usage_of_ty_tykind)]
13
14use std::assert_matches::assert_matches;
15use std::fmt::Debug;
16use std::hash::{Hash, Hasher};
17use std::marker::PhantomData;
18use std::num::NonZero;
19use std::ptr::NonNull;
20use std::{fmt, iter, str};
21
22pub use adt::*;
23pub use assoc::*;
24pub use generic_args::{GenericArgKind, TermKind, *};
25pub use generics::*;
26pub use intrinsic::IntrinsicDef;
27use rustc_abi::{Align, FieldIdx, Integer, IntegerType, ReprFlags, ReprOptions, VariantIdx};
28use rustc_ast::node_id::NodeMap;
29pub use rustc_ast_ir::{Movability, Mutability, try_visit};
30use rustc_data_structures::fx::{FxHashMap, FxHashSet, FxIndexMap, FxIndexSet};
31use rustc_data_structures::intern::Interned;
32use rustc_data_structures::stable_hasher::{HashStable, StableHasher};
33use rustc_data_structures::steal::Steal;
34use rustc_data_structures::unord::{UnordMap, UnordSet};
35use rustc_errors::{Diag, ErrorGuaranteed};
36use rustc_hir::attrs::{AttributeKind, StrippedCfgItem};
37use rustc_hir::def::{CtorKind, CtorOf, DefKind, DocLinkResMap, LifetimeRes, Res};
38use rustc_hir::def_id::{CrateNum, DefId, DefIdMap, LocalDefId, LocalDefIdMap};
39use rustc_hir::definitions::DisambiguatorState;
40use rustc_hir::{LangItem, attrs as attr, find_attr};
41use rustc_index::IndexVec;
42use rustc_index::bit_set::BitMatrix;
43use rustc_macros::{
44    Decodable, Encodable, HashStable, TyDecodable, TyEncodable, TypeFoldable, TypeVisitable,
45    extension,
46};
47use rustc_query_system::ich::StableHashingContext;
48use rustc_serialize::{Decodable, Encodable};
49use rustc_session::lint::LintBuffer;
50pub use rustc_session::lint::RegisteredTools;
51use rustc_span::hygiene::MacroKind;
52use rustc_span::{DUMMY_SP, ExpnId, ExpnKind, Ident, Span, Symbol, sym};
53pub use rustc_type_ir::data_structures::{DelayedMap, DelayedSet};
54pub use rustc_type_ir::fast_reject::DeepRejectCtxt;
55#[allow(
56    hidden_glob_reexports,
57    rustc::usage_of_type_ir_inherent,
58    rustc::non_glob_import_of_type_ir_inherent
59)]
60use rustc_type_ir::inherent;
61pub use rustc_type_ir::relate::VarianceDiagInfo;
62pub use rustc_type_ir::solve::SizedTraitKind;
63pub use rustc_type_ir::*;
64#[allow(hidden_glob_reexports, unused_imports)]
65use rustc_type_ir::{InferCtxtLike, Interner};
66use tracing::{debug, instrument};
67pub use vtable::*;
68use {rustc_ast as ast, rustc_hir as hir};
69
70pub use self::closure::{
71    BorrowKind, CAPTURE_STRUCT_LOCAL, CaptureInfo, CapturedPlace, ClosureTypeInfo,
72    MinCaptureInformationMap, MinCaptureList, RootVariableMinCaptureList, UpvarCapture, UpvarId,
73    UpvarPath, analyze_coroutine_closure_captures, is_ancestor_or_same_capture,
74    place_to_string_for_capture,
75};
76pub use self::consts::{
77    AnonConstKind, AtomicOrdering, Const, ConstInt, ConstKind, ConstToValTreeResult, Expr,
78    ExprKind, ScalarInt, UnevaluatedConst, ValTree, ValTreeKind, Value,
79};
80pub use self::context::{
81    CtxtInterners, CurrentGcx, DeducedParamAttrs, Feed, FreeRegionInfo, GlobalCtxt, Lift, TyCtxt,
82    TyCtxtFeed, tls,
83};
84pub use self::fold::*;
85pub use self::instance::{Instance, InstanceKind, ReifyReason, UnusedGenericParams};
86pub use self::list::{List, ListWithCachedTypeInfo};
87pub use self::opaque_types::OpaqueTypeKey;
88pub use self::pattern::{Pattern, PatternKind};
89pub use self::predicate::{
90    AliasTerm, ArgOutlivesPredicate, Clause, ClauseKind, CoercePredicate, ExistentialPredicate,
91    ExistentialPredicateStableCmpExt, ExistentialProjection, ExistentialTraitRef,
92    HostEffectPredicate, NormalizesTo, OutlivesPredicate, PolyCoercePredicate,
93    PolyExistentialPredicate, PolyExistentialProjection, PolyExistentialTraitRef,
94    PolyProjectionPredicate, PolyRegionOutlivesPredicate, PolySubtypePredicate, PolyTraitPredicate,
95    PolyTraitRef, PolyTypeOutlivesPredicate, Predicate, PredicateKind, ProjectionPredicate,
96    RegionOutlivesPredicate, SubtypePredicate, TraitPredicate, TraitRef, TypeOutlivesPredicate,
97};
98pub use self::region::{
99    BoundRegion, BoundRegionKind, EarlyParamRegion, LateParamRegion, LateParamRegionKind, Region,
100    RegionKind, RegionVid,
101};
102pub use self::rvalue_scopes::RvalueScopes;
103pub use self::sty::{
104    AliasTy, Article, Binder, BoundTy, BoundTyKind, BoundVariableKind, CanonicalPolyFnSig,
105    CoroutineArgsExt, EarlyBinder, FnSig, InlineConstArgs, InlineConstArgsParts, ParamConst,
106    ParamTy, PolyFnSig, TyKind, TypeAndMut, TypingMode, UpvarArgs,
107};
108pub use self::trait_def::TraitDef;
109pub use self::typeck_results::{
110    CanonicalUserType, CanonicalUserTypeAnnotation, CanonicalUserTypeAnnotations, IsIdentity,
111    Rust2024IncompatiblePatInfo, TypeckResults, UserType, UserTypeAnnotationIndex, UserTypeKind,
112};
113pub use self::visit::*;
114use crate::error::{OpaqueHiddenTypeMismatch, TypeMismatchReason};
115use crate::metadata::ModChild;
116use crate::middle::privacy::EffectiveVisibilities;
117use crate::mir::{Body, CoroutineLayout, CoroutineSavedLocal, SourceInfo};
118use crate::query::{IntoQueryParam, Providers};
119use crate::ty;
120use crate::ty::codec::{TyDecoder, TyEncoder};
121pub use crate::ty::diagnostics::*;
122use crate::ty::fast_reject::SimplifiedType;
123use crate::ty::layout::LayoutError;
124use crate::ty::util::Discr;
125use crate::ty::walk::TypeWalker;
126
127pub mod abstract_const;
128pub mod adjustment;
129pub mod cast;
130pub mod codec;
131pub mod error;
132pub mod fast_reject;
133pub mod inhabitedness;
134pub mod layout;
135pub mod normalize_erasing_regions;
136pub mod pattern;
137pub mod print;
138pub mod relate;
139pub mod significant_drop_order;
140pub mod trait_def;
141pub mod util;
142pub mod vtable;
143
144mod adt;
145mod assoc;
146mod closure;
147mod consts;
148mod context;
149mod diagnostics;
150mod elaborate_impl;
151mod erase_regions;
152mod fold;
153mod generic_args;
154mod generics;
155mod impls_ty;
156mod instance;
157mod intrinsic;
158mod list;
159mod opaque_types;
160mod predicate;
161mod region;
162mod rvalue_scopes;
163mod structural_impls;
164#[allow(hidden_glob_reexports)]
165mod sty;
166mod typeck_results;
167mod visit;
168
169// Data types
170
171#[derive(Debug, HashStable)]
172pub struct ResolverGlobalCtxt {
173    pub visibilities_for_hashing: Vec<(LocalDefId, Visibility)>,
174    /// Item with a given `LocalDefId` was defined during macro expansion with ID `ExpnId`.
175    pub expn_that_defined: UnordMap<LocalDefId, ExpnId>,
176    pub effective_visibilities: EffectiveVisibilities,
177    pub extern_crate_map: UnordMap<LocalDefId, CrateNum>,
178    pub maybe_unused_trait_imports: FxIndexSet<LocalDefId>,
179    pub module_children: LocalDefIdMap<Vec<ModChild>>,
180    pub glob_map: FxIndexMap<LocalDefId, FxIndexSet<Symbol>>,
181    pub main_def: Option<MainDefinition>,
182    pub trait_impls: FxIndexMap<DefId, Vec<LocalDefId>>,
183    /// A list of proc macro LocalDefIds, written out in the order in which
184    /// they are declared in the static array generated by proc_macro_harness.
185    pub proc_macros: Vec<LocalDefId>,
186    /// Mapping from ident span to path span for paths that don't exist as written, but that
187    /// exist under `std`. For example, wrote `str::from_utf8` instead of `std::str::from_utf8`.
188    pub confused_type_with_std_module: FxIndexMap<Span, Span>,
189    pub doc_link_resolutions: FxIndexMap<LocalDefId, DocLinkResMap>,
190    pub doc_link_traits_in_scope: FxIndexMap<LocalDefId, Vec<DefId>>,
191    pub all_macro_rules: UnordSet<Symbol>,
192    pub stripped_cfg_items: Vec<StrippedCfgItem>,
193}
194
195/// Resolutions that should only be used for lowering.
196/// This struct is meant to be consumed by lowering.
197#[derive(Debug)]
198pub struct ResolverAstLowering {
199    pub legacy_const_generic_args: FxHashMap<DefId, Option<Vec<usize>>>,
200
201    /// Resolutions for nodes that have a single resolution.
202    pub partial_res_map: NodeMap<hir::def::PartialRes>,
203    /// Resolutions for import nodes, which have multiple resolutions in different namespaces.
204    pub import_res_map: NodeMap<hir::def::PerNS<Option<Res<ast::NodeId>>>>,
205    /// Resolutions for labels (node IDs of their corresponding blocks or loops).
206    pub label_res_map: NodeMap<ast::NodeId>,
207    /// Resolutions for lifetimes.
208    pub lifetimes_res_map: NodeMap<LifetimeRes>,
209    /// Lifetime parameters that lowering will have to introduce.
210    pub extra_lifetime_params_map: NodeMap<Vec<(Ident, ast::NodeId, LifetimeRes)>>,
211
212    pub next_node_id: ast::NodeId,
213
214    pub node_id_to_def_id: NodeMap<LocalDefId>,
215
216    pub disambiguator: DisambiguatorState,
217
218    pub trait_map: NodeMap<Vec<hir::TraitCandidate>>,
219    /// List functions and methods for which lifetime elision was successful.
220    pub lifetime_elision_allowed: FxHashSet<ast::NodeId>,
221
222    /// Lints that were emitted by the resolver and early lints.
223    pub lint_buffer: Steal<LintBuffer>,
224
225    /// Information about functions signatures for delegation items expansion
226    pub delegation_fn_sigs: LocalDefIdMap<DelegationFnSig>,
227}
228
229#[derive(Debug)]
230pub struct DelegationFnSig {
231    pub header: ast::FnHeader,
232    pub param_count: usize,
233    pub has_self: bool,
234    pub c_variadic: bool,
235    pub target_feature: bool,
236}
237
238#[derive(Clone, Copy, Debug, HashStable)]
239pub struct MainDefinition {
240    pub res: Res<ast::NodeId>,
241    pub is_import: bool,
242    pub span: Span,
243}
244
245impl MainDefinition {
246    pub fn opt_fn_def_id(self) -> Option<DefId> {
247        if let Res::Def(DefKind::Fn, def_id) = self.res { Some(def_id) } else { None }
248    }
249}
250
251#[derive(Copy, Clone, Debug, TyEncodable, TyDecodable, HashStable)]
252pub struct ImplTraitHeader<'tcx> {
253    pub trait_ref: ty::EarlyBinder<'tcx, ty::TraitRef<'tcx>>,
254    pub polarity: ImplPolarity,
255    pub safety: hir::Safety,
256    pub constness: hir::Constness,
257}
258
259#[derive(Copy, Clone, PartialEq, Eq, Debug, TypeFoldable, TypeVisitable)]
260pub enum ImplSubject<'tcx> {
261    Trait(TraitRef<'tcx>),
262    Inherent(Ty<'tcx>),
263}
264
265#[derive(Copy, Clone, PartialEq, Eq, Hash, TyEncodable, TyDecodable, HashStable, Debug)]
266#[derive(TypeFoldable, TypeVisitable)]
267pub enum Asyncness {
268    Yes,
269    No,
270}
271
272impl Asyncness {
273    pub fn is_async(self) -> bool {
274        matches!(self, Asyncness::Yes)
275    }
276}
277
278#[derive(Clone, Debug, PartialEq, Eq, Copy, Hash, Encodable, Decodable, HashStable)]
279pub enum Visibility<Id = LocalDefId> {
280    /// Visible everywhere (including in other crates).
281    Public,
282    /// Visible only in the given crate-local module.
283    Restricted(Id),
284}
285
286impl Visibility {
287    pub fn to_string(self, def_id: LocalDefId, tcx: TyCtxt<'_>) -> String {
288        match self {
289            ty::Visibility::Restricted(restricted_id) => {
290                if restricted_id.is_top_level_module() {
291                    "pub(crate)".to_string()
292                } else if restricted_id == tcx.parent_module_from_def_id(def_id).to_local_def_id() {
293                    "pub(self)".to_string()
294                } else {
295                    format!(
296                        "pub(in crate{})",
297                        tcx.def_path(restricted_id.to_def_id()).to_string_no_crate_verbose()
298                    )
299                }
300            }
301            ty::Visibility::Public => "pub".to_string(),
302        }
303    }
304}
305
306#[derive(Clone, Debug, PartialEq, Eq, Copy, Hash, TyEncodable, TyDecodable, HashStable)]
307#[derive(TypeFoldable, TypeVisitable)]
308pub struct ClosureSizeProfileData<'tcx> {
309    /// Tuple containing the types of closure captures before the feature `capture_disjoint_fields`
310    pub before_feature_tys: Ty<'tcx>,
311    /// Tuple containing the types of closure captures after the feature `capture_disjoint_fields`
312    pub after_feature_tys: Ty<'tcx>,
313}
314
315impl TyCtxt<'_> {
316    #[inline]
317    pub fn opt_parent(self, id: DefId) -> Option<DefId> {
318        self.def_key(id).parent.map(|index| DefId { index, ..id })
319    }
320
321    #[inline]
322    #[track_caller]
323    pub fn parent(self, id: DefId) -> DefId {
324        match self.opt_parent(id) {
325            Some(id) => id,
326            // not `unwrap_or_else` to avoid breaking caller tracking
327            None => bug!("{id:?} doesn't have a parent"),
328        }
329    }
330
331    #[inline]
332    #[track_caller]
333    pub fn opt_local_parent(self, id: LocalDefId) -> Option<LocalDefId> {
334        self.opt_parent(id.to_def_id()).map(DefId::expect_local)
335    }
336
337    #[inline]
338    #[track_caller]
339    pub fn local_parent(self, id: impl Into<LocalDefId>) -> LocalDefId {
340        self.parent(id.into().to_def_id()).expect_local()
341    }
342
343    pub fn is_descendant_of(self, mut descendant: DefId, ancestor: DefId) -> bool {
344        if descendant.krate != ancestor.krate {
345            return false;
346        }
347
348        while descendant != ancestor {
349            match self.opt_parent(descendant) {
350                Some(parent) => descendant = parent,
351                None => return false,
352            }
353        }
354        true
355    }
356}
357
358impl<Id> Visibility<Id> {
359    pub fn is_public(self) -> bool {
360        matches!(self, Visibility::Public)
361    }
362
363    pub fn map_id<OutId>(self, f: impl FnOnce(Id) -> OutId) -> Visibility<OutId> {
364        match self {
365            Visibility::Public => Visibility::Public,
366            Visibility::Restricted(id) => Visibility::Restricted(f(id)),
367        }
368    }
369}
370
371impl<Id: Into<DefId>> Visibility<Id> {
372    pub fn to_def_id(self) -> Visibility<DefId> {
373        self.map_id(Into::into)
374    }
375
376    /// Returns `true` if an item with this visibility is accessible from the given module.
377    pub fn is_accessible_from(self, module: impl Into<DefId>, tcx: TyCtxt<'_>) -> bool {
378        match self {
379            // Public items are visible everywhere.
380            Visibility::Public => true,
381            Visibility::Restricted(id) => tcx.is_descendant_of(module.into(), id.into()),
382        }
383    }
384
385    /// Returns `true` if this visibility is at least as accessible as the given visibility
386    pub fn is_at_least(self, vis: Visibility<impl Into<DefId>>, tcx: TyCtxt<'_>) -> bool {
387        match vis {
388            Visibility::Public => self.is_public(),
389            Visibility::Restricted(id) => self.is_accessible_from(id, tcx),
390        }
391    }
392}
393
394impl Visibility<DefId> {
395    pub fn expect_local(self) -> Visibility {
396        self.map_id(|id| id.expect_local())
397    }
398
399    /// Returns `true` if this item is visible anywhere in the local crate.
400    pub fn is_visible_locally(self) -> bool {
401        match self {
402            Visibility::Public => true,
403            Visibility::Restricted(def_id) => def_id.is_local(),
404        }
405    }
406}
407
408/// The crate variances map is computed during typeck and contains the
409/// variance of every item in the local crate. You should not use it
410/// directly, because to do so will make your pass dependent on the
411/// HIR of every item in the local crate. Instead, use
412/// `tcx.variances_of()` to get the variance for a *particular*
413/// item.
414#[derive(HashStable, Debug)]
415pub struct CrateVariancesMap<'tcx> {
416    /// For each item with generics, maps to a vector of the variance
417    /// of its generics. If an item has no generics, it will have no
418    /// entry.
419    pub variances: DefIdMap<&'tcx [ty::Variance]>,
420}
421
422// Contains information needed to resolve types and (in the future) look up
423// the types of AST nodes.
424#[derive(Copy, Clone, PartialEq, Eq, Hash)]
425pub struct CReaderCacheKey {
426    pub cnum: Option<CrateNum>,
427    pub pos: usize,
428}
429
430/// Use this rather than `TyKind`, whenever possible.
431#[derive(Copy, Clone, PartialEq, Eq, Hash, HashStable)]
432#[rustc_diagnostic_item = "Ty"]
433#[rustc_pass_by_value]
434pub struct Ty<'tcx>(Interned<'tcx, WithCachedTypeInfo<TyKind<'tcx>>>);
435
436impl<'tcx> rustc_type_ir::inherent::IntoKind for Ty<'tcx> {
437    type Kind = TyKind<'tcx>;
438
439    fn kind(self) -> TyKind<'tcx> {
440        *self.kind()
441    }
442}
443
444impl<'tcx> rustc_type_ir::Flags for Ty<'tcx> {
445    fn flags(&self) -> TypeFlags {
446        self.0.flags
447    }
448
449    fn outer_exclusive_binder(&self) -> DebruijnIndex {
450        self.0.outer_exclusive_binder
451    }
452}
453
454/// The crate outlives map is computed during typeck and contains the
455/// outlives of every item in the local crate. You should not use it
456/// directly, because to do so will make your pass dependent on the
457/// HIR of every item in the local crate. Instead, use
458/// `tcx.inferred_outlives_of()` to get the outlives for a *particular*
459/// item.
460#[derive(HashStable, Debug)]
461pub struct CratePredicatesMap<'tcx> {
462    /// For each struct with outlive bounds, maps to a vector of the
463    /// predicate of its outlive bounds. If an item has no outlives
464    /// bounds, it will have no entry.
465    pub predicates: DefIdMap<&'tcx [(Clause<'tcx>, Span)]>,
466}
467
468#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
469pub struct Term<'tcx> {
470    ptr: NonNull<()>,
471    marker: PhantomData<(Ty<'tcx>, Const<'tcx>)>,
472}
473
474impl<'tcx> rustc_type_ir::inherent::Term<TyCtxt<'tcx>> for Term<'tcx> {}
475
476impl<'tcx> rustc_type_ir::inherent::IntoKind for Term<'tcx> {
477    type Kind = TermKind<'tcx>;
478
479    fn kind(self) -> Self::Kind {
480        self.kind()
481    }
482}
483
484unsafe impl<'tcx> rustc_data_structures::sync::DynSend for Term<'tcx> where
485    &'tcx (Ty<'tcx>, Const<'tcx>): rustc_data_structures::sync::DynSend
486{
487}
488unsafe impl<'tcx> rustc_data_structures::sync::DynSync for Term<'tcx> where
489    &'tcx (Ty<'tcx>, Const<'tcx>): rustc_data_structures::sync::DynSync
490{
491}
492unsafe impl<'tcx> Send for Term<'tcx> where &'tcx (Ty<'tcx>, Const<'tcx>): Send {}
493unsafe impl<'tcx> Sync for Term<'tcx> where &'tcx (Ty<'tcx>, Const<'tcx>): Sync {}
494
495impl Debug for Term<'_> {
496    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
497        match self.kind() {
498            TermKind::Ty(ty) => write!(f, "Term::Ty({ty:?})"),
499            TermKind::Const(ct) => write!(f, "Term::Const({ct:?})"),
500        }
501    }
502}
503
504impl<'tcx> From<Ty<'tcx>> for Term<'tcx> {
505    fn from(ty: Ty<'tcx>) -> Self {
506        TermKind::Ty(ty).pack()
507    }
508}
509
510impl<'tcx> From<Const<'tcx>> for Term<'tcx> {
511    fn from(c: Const<'tcx>) -> Self {
512        TermKind::Const(c).pack()
513    }
514}
515
516impl<'a, 'tcx> HashStable<StableHashingContext<'a>> for Term<'tcx> {
517    fn hash_stable(&self, hcx: &mut StableHashingContext<'a>, hasher: &mut StableHasher) {
518        self.kind().hash_stable(hcx, hasher);
519    }
520}
521
522impl<'tcx> TypeFoldable<TyCtxt<'tcx>> for Term<'tcx> {
523    fn try_fold_with<F: FallibleTypeFolder<TyCtxt<'tcx>>>(
524        self,
525        folder: &mut F,
526    ) -> Result<Self, F::Error> {
527        match self.kind() {
528            ty::TermKind::Ty(ty) => ty.try_fold_with(folder).map(Into::into),
529            ty::TermKind::Const(ct) => ct.try_fold_with(folder).map(Into::into),
530        }
531    }
532
533    fn fold_with<F: TypeFolder<TyCtxt<'tcx>>>(self, folder: &mut F) -> Self {
534        match self.kind() {
535            ty::TermKind::Ty(ty) => ty.fold_with(folder).into(),
536            ty::TermKind::Const(ct) => ct.fold_with(folder).into(),
537        }
538    }
539}
540
541impl<'tcx> TypeVisitable<TyCtxt<'tcx>> for Term<'tcx> {
542    fn visit_with<V: TypeVisitor<TyCtxt<'tcx>>>(&self, visitor: &mut V) -> V::Result {
543        match self.kind() {
544            ty::TermKind::Ty(ty) => ty.visit_with(visitor),
545            ty::TermKind::Const(ct) => ct.visit_with(visitor),
546        }
547    }
548}
549
550impl<'tcx, E: TyEncoder<'tcx>> Encodable<E> for Term<'tcx> {
551    fn encode(&self, e: &mut E) {
552        self.kind().encode(e)
553    }
554}
555
556impl<'tcx, D: TyDecoder<'tcx>> Decodable<D> for Term<'tcx> {
557    fn decode(d: &mut D) -> Self {
558        let res: TermKind<'tcx> = Decodable::decode(d);
559        res.pack()
560    }
561}
562
563impl<'tcx> Term<'tcx> {
564    #[inline]
565    pub fn kind(self) -> TermKind<'tcx> {
566        let ptr =
567            unsafe { self.ptr.map_addr(|addr| NonZero::new_unchecked(addr.get() & !TAG_MASK)) };
568        // SAFETY: use of `Interned::new_unchecked` here is ok because these
569        // pointers were originally created from `Interned` types in `pack()`,
570        // and this is just going in the other direction.
571        unsafe {
572            match self.ptr.addr().get() & TAG_MASK {
573                TYPE_TAG => TermKind::Ty(Ty(Interned::new_unchecked(
574                    ptr.cast::<WithCachedTypeInfo<ty::TyKind<'tcx>>>().as_ref(),
575                ))),
576                CONST_TAG => TermKind::Const(ty::Const(Interned::new_unchecked(
577                    ptr.cast::<WithCachedTypeInfo<ty::ConstKind<'tcx>>>().as_ref(),
578                ))),
579                _ => core::intrinsics::unreachable(),
580            }
581        }
582    }
583
584    pub fn as_type(&self) -> Option<Ty<'tcx>> {
585        if let TermKind::Ty(ty) = self.kind() { Some(ty) } else { None }
586    }
587
588    pub fn expect_type(&self) -> Ty<'tcx> {
589        self.as_type().expect("expected a type, but found a const")
590    }
591
592    pub fn as_const(&self) -> Option<Const<'tcx>> {
593        if let TermKind::Const(c) = self.kind() { Some(c) } else { None }
594    }
595
596    pub fn expect_const(&self) -> Const<'tcx> {
597        self.as_const().expect("expected a const, but found a type")
598    }
599
600    pub fn into_arg(self) -> GenericArg<'tcx> {
601        match self.kind() {
602            TermKind::Ty(ty) => ty.into(),
603            TermKind::Const(c) => c.into(),
604        }
605    }
606
607    pub fn to_alias_term(self) -> Option<AliasTerm<'tcx>> {
608        match self.kind() {
609            TermKind::Ty(ty) => match *ty.kind() {
610                ty::Alias(_kind, alias_ty) => Some(alias_ty.into()),
611                _ => None,
612            },
613            TermKind::Const(ct) => match ct.kind() {
614                ConstKind::Unevaluated(uv) => Some(uv.into()),
615                _ => None,
616            },
617        }
618    }
619
620    pub fn is_infer(&self) -> bool {
621        match self.kind() {
622            TermKind::Ty(ty) => ty.is_ty_var(),
623            TermKind::Const(ct) => ct.is_ct_infer(),
624        }
625    }
626
627    pub fn is_trivially_wf(&self, tcx: TyCtxt<'tcx>) -> bool {
628        match self.kind() {
629            TermKind::Ty(ty) => ty.is_trivially_wf(tcx),
630            TermKind::Const(ct) => ct.is_trivially_wf(),
631        }
632    }
633
634    /// Iterator that walks `self` and any types reachable from
635    /// `self`, in depth-first order. Note that just walks the types
636    /// that appear in `self`, it does not descend into the fields of
637    /// structs or variants. For example:
638    ///
639    /// ```text
640    /// isize => { isize }
641    /// Foo<Bar<isize>> => { Foo<Bar<isize>>, Bar<isize>, isize }
642    /// [isize] => { [isize], isize }
643    /// ```
644    pub fn walk(self) -> TypeWalker<TyCtxt<'tcx>> {
645        TypeWalker::new(self.into())
646    }
647}
648
649const TAG_MASK: usize = 0b11;
650const TYPE_TAG: usize = 0b00;
651const CONST_TAG: usize = 0b01;
652
653#[extension(pub trait TermKindPackExt<'tcx>)]
654impl<'tcx> TermKind<'tcx> {
655    #[inline]
656    fn pack(self) -> Term<'tcx> {
657        let (tag, ptr) = match self {
658            TermKind::Ty(ty) => {
659                // Ensure we can use the tag bits.
660                assert_eq!(align_of_val(&*ty.0.0) & TAG_MASK, 0);
661                (TYPE_TAG, NonNull::from(ty.0.0).cast())
662            }
663            TermKind::Const(ct) => {
664                // Ensure we can use the tag bits.
665                assert_eq!(align_of_val(&*ct.0.0) & TAG_MASK, 0);
666                (CONST_TAG, NonNull::from(ct.0.0).cast())
667            }
668        };
669
670        Term { ptr: ptr.map_addr(|addr| addr | tag), marker: PhantomData }
671    }
672}
673
674/// Represents the bounds declared on a particular set of type
675/// parameters. Should eventually be generalized into a flag list of
676/// where-clauses. You can obtain an `InstantiatedPredicates` list from a
677/// `GenericPredicates` by using the `instantiate` method. Note that this method
678/// reflects an important semantic invariant of `InstantiatedPredicates`: while
679/// the `GenericPredicates` are expressed in terms of the bound type
680/// parameters of the impl/trait/whatever, an `InstantiatedPredicates` instance
681/// represented a set of bounds for some particular instantiation,
682/// meaning that the generic parameters have been instantiated with
683/// their values.
684///
685/// Example:
686/// ```ignore (illustrative)
687/// struct Foo<T, U: Bar<T>> { ... }
688/// ```
689/// Here, the `GenericPredicates` for `Foo` would contain a list of bounds like
690/// `[[], [U:Bar<T>]]`. Now if there were some particular reference
691/// like `Foo<isize,usize>`, then the `InstantiatedPredicates` would be `[[],
692/// [usize:Bar<isize>]]`.
693#[derive(Clone, Debug, TypeFoldable, TypeVisitable)]
694pub struct InstantiatedPredicates<'tcx> {
695    pub predicates: Vec<Clause<'tcx>>,
696    pub spans: Vec<Span>,
697}
698
699impl<'tcx> InstantiatedPredicates<'tcx> {
700    pub fn empty() -> InstantiatedPredicates<'tcx> {
701        InstantiatedPredicates { predicates: vec![], spans: vec![] }
702    }
703
704    pub fn is_empty(&self) -> bool {
705        self.predicates.is_empty()
706    }
707
708    pub fn iter(&self) -> <&Self as IntoIterator>::IntoIter {
709        self.into_iter()
710    }
711}
712
713impl<'tcx> IntoIterator for InstantiatedPredicates<'tcx> {
714    type Item = (Clause<'tcx>, Span);
715
716    type IntoIter = std::iter::Zip<std::vec::IntoIter<Clause<'tcx>>, std::vec::IntoIter<Span>>;
717
718    fn into_iter(self) -> Self::IntoIter {
719        debug_assert_eq!(self.predicates.len(), self.spans.len());
720        std::iter::zip(self.predicates, self.spans)
721    }
722}
723
724impl<'a, 'tcx> IntoIterator for &'a InstantiatedPredicates<'tcx> {
725    type Item = (Clause<'tcx>, Span);
726
727    type IntoIter = std::iter::Zip<
728        std::iter::Copied<std::slice::Iter<'a, Clause<'tcx>>>,
729        std::iter::Copied<std::slice::Iter<'a, Span>>,
730    >;
731
732    fn into_iter(self) -> Self::IntoIter {
733        debug_assert_eq!(self.predicates.len(), self.spans.len());
734        std::iter::zip(self.predicates.iter().copied(), self.spans.iter().copied())
735    }
736}
737
738#[derive(Copy, Clone, Debug, TypeFoldable, TypeVisitable, HashStable, TyEncodable, TyDecodable)]
739pub struct OpaqueHiddenType<'tcx> {
740    /// The span of this particular definition of the opaque type. So
741    /// for example:
742    ///
743    /// ```ignore (incomplete snippet)
744    /// type Foo = impl Baz;
745    /// fn bar() -> Foo {
746    /// //          ^^^ This is the span we are looking for!
747    /// }
748    /// ```
749    ///
750    /// In cases where the fn returns `(impl Trait, impl Trait)` or
751    /// other such combinations, the result is currently
752    /// over-approximated, but better than nothing.
753    pub span: Span,
754
755    /// The type variable that represents the value of the opaque type
756    /// that we require. In other words, after we compile this function,
757    /// we will be created a constraint like:
758    /// ```ignore (pseudo-rust)
759    /// Foo<'a, T> = ?C
760    /// ```
761    /// where `?C` is the value of this type variable. =) It may
762    /// naturally refer to the type and lifetime parameters in scope
763    /// in this function, though ultimately it should only reference
764    /// those that are arguments to `Foo` in the constraint above. (In
765    /// other words, `?C` should not include `'b`, even though it's a
766    /// lifetime parameter on `foo`.)
767    pub ty: Ty<'tcx>,
768}
769
770/// Whether we're currently in HIR typeck or MIR borrowck.
771#[derive(Debug, Clone, Copy)]
772pub enum DefiningScopeKind {
773    /// During writeback in typeck, we don't care about regions and simply
774    /// erase them. This means we also don't check whether regions are
775    /// universal in the opaque type key. This will only be checked in
776    /// MIR borrowck.
777    HirTypeck,
778    MirBorrowck,
779}
780
781impl<'tcx> OpaqueHiddenType<'tcx> {
782    pub fn new_error(tcx: TyCtxt<'tcx>, guar: ErrorGuaranteed) -> OpaqueHiddenType<'tcx> {
783        OpaqueHiddenType { span: DUMMY_SP, ty: Ty::new_error(tcx, guar) }
784    }
785
786    pub fn build_mismatch_error(
787        &self,
788        other: &Self,
789        tcx: TyCtxt<'tcx>,
790    ) -> Result<Diag<'tcx>, ErrorGuaranteed> {
791        (self.ty, other.ty).error_reported()?;
792        // Found different concrete types for the opaque type.
793        let sub_diag = if self.span == other.span {
794            TypeMismatchReason::ConflictType { span: self.span }
795        } else {
796            TypeMismatchReason::PreviousUse { span: self.span }
797        };
798        Ok(tcx.dcx().create_err(OpaqueHiddenTypeMismatch {
799            self_ty: self.ty,
800            other_ty: other.ty,
801            other_span: other.span,
802            sub: sub_diag,
803        }))
804    }
805
806    #[instrument(level = "debug", skip(tcx), ret)]
807    pub fn remap_generic_params_to_declaration_params(
808        self,
809        opaque_type_key: OpaqueTypeKey<'tcx>,
810        tcx: TyCtxt<'tcx>,
811        defining_scope_kind: DefiningScopeKind,
812    ) -> Self {
813        let OpaqueTypeKey { def_id, args } = opaque_type_key;
814
815        // Use args to build up a reverse map from regions to their
816        // identity mappings. This is necessary because of `impl
817        // Trait` lifetimes are computed by replacing existing
818        // lifetimes with 'static and remapping only those used in the
819        // `impl Trait` return type, resulting in the parameters
820        // shifting.
821        let id_args = GenericArgs::identity_for_item(tcx, def_id);
822        debug!(?id_args);
823
824        // This zip may have several times the same lifetime in `args` paired with a different
825        // lifetime from `id_args`. Simply `collect`ing the iterator is the correct behaviour:
826        // it will pick the last one, which is the one we introduced in the impl-trait desugaring.
827        let map = args.iter().zip(id_args).collect();
828        debug!("map = {:#?}", map);
829
830        // Convert the type from the function into a type valid outside by mapping generic
831        // parameters to into the context of the opaque.
832        //
833        // We erase regions when doing this during HIR typeck.
834        let this = match defining_scope_kind {
835            DefiningScopeKind::HirTypeck => tcx.erase_regions(self),
836            DefiningScopeKind::MirBorrowck => self,
837        };
838        let result = this.fold_with(&mut opaque_types::ReverseMapper::new(tcx, map, self.span));
839        if cfg!(debug_assertions) && matches!(defining_scope_kind, DefiningScopeKind::HirTypeck) {
840            assert_eq!(result.ty, tcx.erase_regions(result.ty));
841        }
842        result
843    }
844}
845
846/// The "placeholder index" fully defines a placeholder region, type, or const. Placeholders are
847/// identified by both a universe, as well as a name residing within that universe. Distinct bound
848/// regions/types/consts within the same universe simply have an unknown relationship to one
849/// another.
850#[derive(Copy, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
851#[derive(HashStable, TyEncodable, TyDecodable)]
852pub struct Placeholder<T> {
853    pub universe: UniverseIndex,
854    pub bound: T,
855}
856
857pub type PlaceholderRegion = Placeholder<BoundRegion>;
858
859impl<'tcx> rustc_type_ir::inherent::PlaceholderLike<TyCtxt<'tcx>> for PlaceholderRegion {
860    type Bound = BoundRegion;
861
862    fn universe(self) -> UniverseIndex {
863        self.universe
864    }
865
866    fn var(self) -> BoundVar {
867        self.bound.var
868    }
869
870    fn with_updated_universe(self, ui: UniverseIndex) -> Self {
871        Placeholder { universe: ui, ..self }
872    }
873
874    fn new(ui: UniverseIndex, bound: BoundRegion) -> Self {
875        Placeholder { universe: ui, bound }
876    }
877
878    fn new_anon(ui: UniverseIndex, var: BoundVar) -> Self {
879        Placeholder { universe: ui, bound: BoundRegion { var, kind: BoundRegionKind::Anon } }
880    }
881}
882
883pub type PlaceholderType = Placeholder<BoundTy>;
884
885impl<'tcx> rustc_type_ir::inherent::PlaceholderLike<TyCtxt<'tcx>> for PlaceholderType {
886    type Bound = BoundTy;
887
888    fn universe(self) -> UniverseIndex {
889        self.universe
890    }
891
892    fn var(self) -> BoundVar {
893        self.bound.var
894    }
895
896    fn with_updated_universe(self, ui: UniverseIndex) -> Self {
897        Placeholder { universe: ui, ..self }
898    }
899
900    fn new(ui: UniverseIndex, bound: BoundTy) -> Self {
901        Placeholder { universe: ui, bound }
902    }
903
904    fn new_anon(ui: UniverseIndex, var: BoundVar) -> Self {
905        Placeholder { universe: ui, bound: BoundTy { var, kind: BoundTyKind::Anon } }
906    }
907}
908
909#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, HashStable)]
910#[derive(TyEncodable, TyDecodable)]
911pub struct BoundConst {
912    pub var: BoundVar,
913}
914
915impl<'tcx> rustc_type_ir::inherent::BoundVarLike<TyCtxt<'tcx>> for BoundConst {
916    fn var(self) -> BoundVar {
917        self.var
918    }
919
920    fn assert_eq(self, var: ty::BoundVariableKind) {
921        var.expect_const()
922    }
923}
924
925pub type PlaceholderConst = Placeholder<BoundConst>;
926
927impl<'tcx> rustc_type_ir::inherent::PlaceholderLike<TyCtxt<'tcx>> for PlaceholderConst {
928    type Bound = BoundConst;
929
930    fn universe(self) -> UniverseIndex {
931        self.universe
932    }
933
934    fn var(self) -> BoundVar {
935        self.bound.var
936    }
937
938    fn with_updated_universe(self, ui: UniverseIndex) -> Self {
939        Placeholder { universe: ui, ..self }
940    }
941
942    fn new(ui: UniverseIndex, bound: BoundConst) -> Self {
943        Placeholder { universe: ui, bound }
944    }
945
946    fn new_anon(ui: UniverseIndex, var: BoundVar) -> Self {
947        Placeholder { universe: ui, bound: BoundConst { var } }
948    }
949}
950
951pub type Clauses<'tcx> = &'tcx ListWithCachedTypeInfo<Clause<'tcx>>;
952
953impl<'tcx> rustc_type_ir::Flags for Clauses<'tcx> {
954    fn flags(&self) -> TypeFlags {
955        (**self).flags()
956    }
957
958    fn outer_exclusive_binder(&self) -> DebruijnIndex {
959        (**self).outer_exclusive_binder()
960    }
961}
962
963/// When interacting with the type system we must provide information about the
964/// environment. `ParamEnv` is the type that represents this information. See the
965/// [dev guide chapter][param_env_guide] for more information.
966///
967/// [param_env_guide]: https://rustc-dev-guide.rust-lang.org/typing_parameter_envs.html
968#[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
969#[derive(HashStable, TypeVisitable, TypeFoldable)]
970pub struct ParamEnv<'tcx> {
971    /// Caller bounds are `Obligation`s that the caller must satisfy. This is
972    /// basically the set of bounds on the in-scope type parameters, translated
973    /// into `Obligation`s, and elaborated and normalized.
974    ///
975    /// Use the `caller_bounds()` method to access.
976    caller_bounds: Clauses<'tcx>,
977}
978
979impl<'tcx> rustc_type_ir::inherent::ParamEnv<TyCtxt<'tcx>> for ParamEnv<'tcx> {
980    fn caller_bounds(self) -> impl inherent::SliceLike<Item = ty::Clause<'tcx>> {
981        self.caller_bounds()
982    }
983}
984
985impl<'tcx> ParamEnv<'tcx> {
986    /// Construct a trait environment suitable for contexts where there are
987    /// no where-clauses in scope. In the majority of cases it is incorrect
988    /// to use an empty environment. See the [dev guide section][param_env_guide]
989    /// for information on what a `ParamEnv` is and how to acquire one.
990    ///
991    /// [param_env_guide]: https://rustc-dev-guide.rust-lang.org/typing_parameter_envs.html
992    #[inline]
993    pub fn empty() -> Self {
994        Self::new(ListWithCachedTypeInfo::empty())
995    }
996
997    #[inline]
998    pub fn caller_bounds(self) -> Clauses<'tcx> {
999        self.caller_bounds
1000    }
1001
1002    /// Construct a trait environment with the given set of predicates.
1003    #[inline]
1004    pub fn new(caller_bounds: Clauses<'tcx>) -> Self {
1005        ParamEnv { caller_bounds }
1006    }
1007
1008    /// Creates a pair of param-env and value for use in queries.
1009    pub fn and<T: TypeVisitable<TyCtxt<'tcx>>>(self, value: T) -> ParamEnvAnd<'tcx, T> {
1010        ParamEnvAnd { param_env: self, value }
1011    }
1012}
1013
1014#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, TypeFoldable, TypeVisitable)]
1015#[derive(HashStable)]
1016pub struct ParamEnvAnd<'tcx, T> {
1017    pub param_env: ParamEnv<'tcx>,
1018    pub value: T,
1019}
1020
1021/// The environment in which to do trait solving.
1022///
1023/// Most of the time you only need to care about the `ParamEnv`
1024/// as the `TypingMode` is simply stored in the `InferCtxt`.
1025///
1026/// However, there are some places which rely on trait solving
1027/// without using an `InferCtxt` themselves. For these to be
1028/// able to use the trait system they have to be able to initialize
1029/// such an `InferCtxt` with the right `typing_mode`, so they need
1030/// to track both.
1031#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, HashStable)]
1032#[derive(TypeVisitable, TypeFoldable)]
1033pub struct TypingEnv<'tcx> {
1034    #[type_foldable(identity)]
1035    #[type_visitable(ignore)]
1036    pub typing_mode: TypingMode<'tcx>,
1037    pub param_env: ParamEnv<'tcx>,
1038}
1039
1040impl<'tcx> TypingEnv<'tcx> {
1041    /// Create a typing environment with no where-clauses in scope
1042    /// where all opaque types and default associated items are revealed.
1043    ///
1044    /// This is only suitable for monomorphized, post-typeck environments.
1045    /// Do not use this for MIR optimizations, as even though they also
1046    /// use `TypingMode::PostAnalysis`, they may still have where-clauses
1047    /// in scope.
1048    pub fn fully_monomorphized() -> TypingEnv<'tcx> {
1049        TypingEnv { typing_mode: TypingMode::PostAnalysis, param_env: ParamEnv::empty() }
1050    }
1051
1052    /// Create a typing environment for use during analysis outside of a body.
1053    ///
1054    /// Using a typing environment inside of bodies is not supported as the body
1055    /// may define opaque types. In this case the used functions have to be
1056    /// converted to use proper canonical inputs instead.
1057    pub fn non_body_analysis(
1058        tcx: TyCtxt<'tcx>,
1059        def_id: impl IntoQueryParam<DefId>,
1060    ) -> TypingEnv<'tcx> {
1061        TypingEnv { typing_mode: TypingMode::non_body_analysis(), param_env: tcx.param_env(def_id) }
1062    }
1063
1064    pub fn post_analysis(tcx: TyCtxt<'tcx>, def_id: impl IntoQueryParam<DefId>) -> TypingEnv<'tcx> {
1065        tcx.typing_env_normalized_for_post_analysis(def_id)
1066    }
1067
1068    /// Modify the `typing_mode` to `PostAnalysis` and eagerly reveal all
1069    /// opaque types in the `param_env`.
1070    pub fn with_post_analysis_normalized(self, tcx: TyCtxt<'tcx>) -> TypingEnv<'tcx> {
1071        let TypingEnv { typing_mode, param_env } = self;
1072        if let TypingMode::PostAnalysis = typing_mode {
1073            return self;
1074        }
1075
1076        // No need to reveal opaques with the new solver enabled,
1077        // since we have lazy norm.
1078        let param_env = if tcx.next_trait_solver_globally() {
1079            param_env
1080        } else {
1081            ParamEnv::new(tcx.reveal_opaque_types_in_bounds(param_env.caller_bounds()))
1082        };
1083        TypingEnv { typing_mode: TypingMode::PostAnalysis, param_env }
1084    }
1085
1086    /// Combine this typing environment with the given `value` to be used by
1087    /// not (yet) canonicalized queries. This only works if the value does not
1088    /// contain anything local to some `InferCtxt`, i.e. inference variables or
1089    /// placeholders.
1090    pub fn as_query_input<T>(self, value: T) -> PseudoCanonicalInput<'tcx, T>
1091    where
1092        T: TypeVisitable<TyCtxt<'tcx>>,
1093    {
1094        // FIXME(#132279): We should assert that the value does not contain any placeholders
1095        // as these placeholders are also local to the current inference context. However, we
1096        // currently use pseudo-canonical queries in the trait solver, which replaces params
1097        // with placeholders during canonicalization. We should also simply not use pseudo-
1098        // canonical queries in the trait solver, at which point we can readd this assert.
1099        //
1100        // As of writing this comment, this is only used when normalizing consts that mention
1101        // params.
1102        /* debug_assert!(
1103            !value.has_placeholders(),
1104            "{value:?} which has placeholder shouldn't be pseudo-canonicalized"
1105        ); */
1106        PseudoCanonicalInput { typing_env: self, value }
1107    }
1108}
1109
1110/// Similar to `CanonicalInput`, this carries the `typing_mode` and the environment
1111/// necessary to do any kind of trait solving inside of nested queries.
1112///
1113/// Unlike proper canonicalization, this requires the `param_env` and the `value` to not
1114/// contain anything local to the `infcx` of the caller, so we don't actually canonicalize
1115/// anything.
1116///
1117/// This should be created by using `infcx.pseudo_canonicalize_query(param_env, value)`
1118/// or by using `typing_env.as_query_input(value)`.
1119#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
1120#[derive(HashStable, TypeVisitable, TypeFoldable)]
1121pub struct PseudoCanonicalInput<'tcx, T> {
1122    pub typing_env: TypingEnv<'tcx>,
1123    pub value: T,
1124}
1125
1126#[derive(Copy, Clone, Debug, HashStable, Encodable, Decodable)]
1127pub struct Destructor {
1128    /// The `DefId` of the destructor method
1129    pub did: DefId,
1130}
1131
1132// FIXME: consider combining this definition with regular `Destructor`
1133#[derive(Copy, Clone, Debug, HashStable, Encodable, Decodable)]
1134pub struct AsyncDestructor {
1135    /// The `DefId` of the `impl AsyncDrop`
1136    pub impl_did: DefId,
1137}
1138
1139#[derive(Clone, Copy, PartialEq, Eq, HashStable, TyEncodable, TyDecodable)]
1140pub struct VariantFlags(u8);
1141bitflags::bitflags! {
1142    impl VariantFlags: u8 {
1143        const NO_VARIANT_FLAGS        = 0;
1144        /// Indicates whether the field list of this variant is `#[non_exhaustive]`.
1145        const IS_FIELD_LIST_NON_EXHAUSTIVE = 1 << 0;
1146    }
1147}
1148rustc_data_structures::external_bitflags_debug! { VariantFlags }
1149
1150/// Definition of a variant -- a struct's fields or an enum variant.
1151#[derive(Debug, HashStable, TyEncodable, TyDecodable)]
1152pub struct VariantDef {
1153    /// `DefId` that identifies the variant itself.
1154    /// If this variant belongs to a struct or union, then this is a copy of its `DefId`.
1155    pub def_id: DefId,
1156    /// `DefId` that identifies the variant's constructor.
1157    /// If this variant is a struct variant, then this is `None`.
1158    pub ctor: Option<(CtorKind, DefId)>,
1159    /// Variant or struct name.
1160    pub name: Symbol,
1161    /// Discriminant of this variant.
1162    pub discr: VariantDiscr,
1163    /// Fields of this variant.
1164    pub fields: IndexVec<FieldIdx, FieldDef>,
1165    /// The error guarantees from parser, if any.
1166    tainted: Option<ErrorGuaranteed>,
1167    /// Flags of the variant (e.g. is field list non-exhaustive)?
1168    flags: VariantFlags,
1169}
1170
1171impl VariantDef {
1172    /// Creates a new `VariantDef`.
1173    ///
1174    /// `variant_did` is the `DefId` that identifies the enum variant (if this `VariantDef`
1175    /// represents an enum variant).
1176    ///
1177    /// `ctor_did` is the `DefId` that identifies the constructor of unit or
1178    /// tuple-variants/structs. If this is a `struct`-variant then this should be `None`.
1179    ///
1180    /// `parent_did` is the `DefId` of the `AdtDef` representing the enum or struct that
1181    /// owns this variant. It is used for checking if a struct has `#[non_exhaustive]` w/out having
1182    /// to go through the redirect of checking the ctor's attributes - but compiling a small crate
1183    /// requires loading the `AdtDef`s for all the structs in the universe (e.g., coherence for any
1184    /// built-in trait), and we do not want to load attributes twice.
1185    ///
1186    /// If someone speeds up attribute loading to not be a performance concern, they can
1187    /// remove this hack and use the constructor `DefId` everywhere.
1188    #[instrument(level = "debug")]
1189    pub fn new(
1190        name: Symbol,
1191        variant_did: Option<DefId>,
1192        ctor: Option<(CtorKind, DefId)>,
1193        discr: VariantDiscr,
1194        fields: IndexVec<FieldIdx, FieldDef>,
1195        parent_did: DefId,
1196        recover_tainted: Option<ErrorGuaranteed>,
1197        is_field_list_non_exhaustive: bool,
1198    ) -> Self {
1199        let mut flags = VariantFlags::NO_VARIANT_FLAGS;
1200        if is_field_list_non_exhaustive {
1201            flags |= VariantFlags::IS_FIELD_LIST_NON_EXHAUSTIVE;
1202        }
1203
1204        VariantDef {
1205            def_id: variant_did.unwrap_or(parent_did),
1206            ctor,
1207            name,
1208            discr,
1209            fields,
1210            flags,
1211            tainted: recover_tainted,
1212        }
1213    }
1214
1215    /// Returns `true` if the field list of this variant is `#[non_exhaustive]`.
1216    ///
1217    /// Note that this function will return `true` even if the type has been
1218    /// defined in the crate currently being compiled. If that's not what you
1219    /// want, see [`Self::field_list_has_applicable_non_exhaustive`].
1220    #[inline]
1221    pub fn is_field_list_non_exhaustive(&self) -> bool {
1222        self.flags.intersects(VariantFlags::IS_FIELD_LIST_NON_EXHAUSTIVE)
1223    }
1224
1225    /// Returns `true` if the field list of this variant is `#[non_exhaustive]`
1226    /// and the type has been defined in another crate.
1227    #[inline]
1228    pub fn field_list_has_applicable_non_exhaustive(&self) -> bool {
1229        self.is_field_list_non_exhaustive() && !self.def_id.is_local()
1230    }
1231
1232    /// Computes the `Ident` of this variant by looking up the `Span`
1233    pub fn ident(&self, tcx: TyCtxt<'_>) -> Ident {
1234        Ident::new(self.name, tcx.def_ident_span(self.def_id).unwrap())
1235    }
1236
1237    /// Was this variant obtained as part of recovering from a syntactic error?
1238    #[inline]
1239    pub fn has_errors(&self) -> Result<(), ErrorGuaranteed> {
1240        self.tainted.map_or(Ok(()), Err)
1241    }
1242
1243    #[inline]
1244    pub fn ctor_kind(&self) -> Option<CtorKind> {
1245        self.ctor.map(|(kind, _)| kind)
1246    }
1247
1248    #[inline]
1249    pub fn ctor_def_id(&self) -> Option<DefId> {
1250        self.ctor.map(|(_, def_id)| def_id)
1251    }
1252
1253    /// Returns the one field in this variant.
1254    ///
1255    /// `panic!`s if there are no fields or multiple fields.
1256    #[inline]
1257    pub fn single_field(&self) -> &FieldDef {
1258        assert!(self.fields.len() == 1);
1259
1260        &self.fields[FieldIdx::ZERO]
1261    }
1262
1263    /// Returns the last field in this variant, if present.
1264    #[inline]
1265    pub fn tail_opt(&self) -> Option<&FieldDef> {
1266        self.fields.raw.last()
1267    }
1268
1269    /// Returns the last field in this variant.
1270    ///
1271    /// # Panics
1272    ///
1273    /// Panics, if the variant has no fields.
1274    #[inline]
1275    pub fn tail(&self) -> &FieldDef {
1276        self.tail_opt().expect("expected unsized ADT to have a tail field")
1277    }
1278
1279    /// Returns whether this variant has unsafe fields.
1280    pub fn has_unsafe_fields(&self) -> bool {
1281        self.fields.iter().any(|x| x.safety.is_unsafe())
1282    }
1283}
1284
1285impl PartialEq for VariantDef {
1286    #[inline]
1287    fn eq(&self, other: &Self) -> bool {
1288        // There should be only one `VariantDef` for each `def_id`, therefore
1289        // it is fine to implement `PartialEq` only based on `def_id`.
1290        //
1291        // Below, we exhaustively destructure `self` and `other` so that if the
1292        // definition of `VariantDef` changes, a compile-error will be produced,
1293        // reminding us to revisit this assumption.
1294
1295        let Self {
1296            def_id: lhs_def_id,
1297            ctor: _,
1298            name: _,
1299            discr: _,
1300            fields: _,
1301            flags: _,
1302            tainted: _,
1303        } = &self;
1304        let Self {
1305            def_id: rhs_def_id,
1306            ctor: _,
1307            name: _,
1308            discr: _,
1309            fields: _,
1310            flags: _,
1311            tainted: _,
1312        } = other;
1313
1314        let res = lhs_def_id == rhs_def_id;
1315
1316        // Double check that implicit assumption detailed above.
1317        if cfg!(debug_assertions) && res {
1318            let deep = self.ctor == other.ctor
1319                && self.name == other.name
1320                && self.discr == other.discr
1321                && self.fields == other.fields
1322                && self.flags == other.flags;
1323            assert!(deep, "VariantDef for the same def-id has differing data");
1324        }
1325
1326        res
1327    }
1328}
1329
1330impl Eq for VariantDef {}
1331
1332impl Hash for VariantDef {
1333    #[inline]
1334    fn hash<H: Hasher>(&self, s: &mut H) {
1335        // There should be only one `VariantDef` for each `def_id`, therefore
1336        // it is fine to implement `Hash` only based on `def_id`.
1337        //
1338        // Below, we exhaustively destructure `self` so that if the definition
1339        // of `VariantDef` changes, a compile-error will be produced, reminding
1340        // us to revisit this assumption.
1341
1342        let Self { def_id, ctor: _, name: _, discr: _, fields: _, flags: _, tainted: _ } = &self;
1343        def_id.hash(s)
1344    }
1345}
1346
1347#[derive(Copy, Clone, Debug, PartialEq, Eq, TyEncodable, TyDecodable, HashStable)]
1348pub enum VariantDiscr {
1349    /// Explicit value for this variant, i.e., `X = 123`.
1350    /// The `DefId` corresponds to the embedded constant.
1351    Explicit(DefId),
1352
1353    /// The previous variant's discriminant plus one.
1354    /// For efficiency reasons, the distance from the
1355    /// last `Explicit` discriminant is being stored,
1356    /// or `0` for the first variant, if it has none.
1357    Relative(u32),
1358}
1359
1360#[derive(Debug, HashStable, TyEncodable, TyDecodable)]
1361pub struct FieldDef {
1362    pub did: DefId,
1363    pub name: Symbol,
1364    pub vis: Visibility<DefId>,
1365    pub safety: hir::Safety,
1366    pub value: Option<DefId>,
1367}
1368
1369impl PartialEq for FieldDef {
1370    #[inline]
1371    fn eq(&self, other: &Self) -> bool {
1372        // There should be only one `FieldDef` for each `did`, therefore it is
1373        // fine to implement `PartialEq` only based on `did`.
1374        //
1375        // Below, we exhaustively destructure `self` so that if the definition
1376        // of `FieldDef` changes, a compile-error will be produced, reminding
1377        // us to revisit this assumption.
1378
1379        let Self { did: lhs_did, name: _, vis: _, safety: _, value: _ } = &self;
1380
1381        let Self { did: rhs_did, name: _, vis: _, safety: _, value: _ } = other;
1382
1383        let res = lhs_did == rhs_did;
1384
1385        // Double check that implicit assumption detailed above.
1386        if cfg!(debug_assertions) && res {
1387            let deep =
1388                self.name == other.name && self.vis == other.vis && self.safety == other.safety;
1389            assert!(deep, "FieldDef for the same def-id has differing data");
1390        }
1391
1392        res
1393    }
1394}
1395
1396impl Eq for FieldDef {}
1397
1398impl Hash for FieldDef {
1399    #[inline]
1400    fn hash<H: Hasher>(&self, s: &mut H) {
1401        // There should be only one `FieldDef` for each `did`, therefore it is
1402        // fine to implement `Hash` only based on `did`.
1403        //
1404        // Below, we exhaustively destructure `self` so that if the definition
1405        // of `FieldDef` changes, a compile-error will be produced, reminding
1406        // us to revisit this assumption.
1407
1408        let Self { did, name: _, vis: _, safety: _, value: _ } = &self;
1409
1410        did.hash(s)
1411    }
1412}
1413
1414impl<'tcx> FieldDef {
1415    /// Returns the type of this field. The resulting type is not normalized. The `arg` is
1416    /// typically obtained via the second field of [`TyKind::Adt`].
1417    pub fn ty(&self, tcx: TyCtxt<'tcx>, args: GenericArgsRef<'tcx>) -> Ty<'tcx> {
1418        tcx.type_of(self.did).instantiate(tcx, args)
1419    }
1420
1421    /// Computes the `Ident` of this variant by looking up the `Span`
1422    pub fn ident(&self, tcx: TyCtxt<'_>) -> Ident {
1423        Ident::new(self.name, tcx.def_ident_span(self.did).unwrap())
1424    }
1425}
1426
1427#[derive(Debug, PartialEq, Eq)]
1428pub enum ImplOverlapKind {
1429    /// These impls are always allowed to overlap.
1430    Permitted {
1431        /// Whether or not the impl is permitted due to the trait being a `#[marker]` trait
1432        marker: bool,
1433    },
1434}
1435
1436/// Useful source information about where a desugared associated type for an
1437/// RPITIT originated from.
1438#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, Encodable, Decodable, HashStable)]
1439pub enum ImplTraitInTraitData {
1440    Trait { fn_def_id: DefId, opaque_def_id: DefId },
1441    Impl { fn_def_id: DefId },
1442}
1443
1444impl<'tcx> TyCtxt<'tcx> {
1445    pub fn typeck_body(self, body: hir::BodyId) -> &'tcx TypeckResults<'tcx> {
1446        self.typeck(self.hir_body_owner_def_id(body))
1447    }
1448
1449    pub fn provided_trait_methods(self, id: DefId) -> impl 'tcx + Iterator<Item = &'tcx AssocItem> {
1450        self.associated_items(id)
1451            .in_definition_order()
1452            .filter(move |item| item.is_fn() && item.defaultness(self).has_value())
1453    }
1454
1455    pub fn repr_options_of_def(self, did: LocalDefId) -> ReprOptions {
1456        let mut flags = ReprFlags::empty();
1457        let mut size = None;
1458        let mut max_align: Option<Align> = None;
1459        let mut min_pack: Option<Align> = None;
1460
1461        // Generate a deterministically-derived seed from the item's path hash
1462        // to allow for cross-crate compilation to actually work
1463        let mut field_shuffle_seed = self.def_path_hash(did.to_def_id()).0.to_smaller_hash();
1464
1465        // If the user defined a custom seed for layout randomization, xor the item's
1466        // path hash with the user defined seed, this will allowing determinism while
1467        // still allowing users to further randomize layout generation for e.g. fuzzing
1468        if let Some(user_seed) = self.sess.opts.unstable_opts.layout_seed {
1469            field_shuffle_seed ^= user_seed;
1470        }
1471
1472        if let Some(reprs) =
1473            find_attr!(self.get_all_attrs(did), AttributeKind::Repr { reprs, .. } => reprs)
1474        {
1475            for (r, _) in reprs {
1476                flags.insert(match *r {
1477                    attr::ReprRust => ReprFlags::empty(),
1478                    attr::ReprC => ReprFlags::IS_C,
1479                    attr::ReprPacked(pack) => {
1480                        min_pack = Some(if let Some(min_pack) = min_pack {
1481                            min_pack.min(pack)
1482                        } else {
1483                            pack
1484                        });
1485                        ReprFlags::empty()
1486                    }
1487                    attr::ReprTransparent => ReprFlags::IS_TRANSPARENT,
1488                    attr::ReprSimd => ReprFlags::IS_SIMD,
1489                    attr::ReprInt(i) => {
1490                        size = Some(match i {
1491                            attr::IntType::SignedInt(x) => match x {
1492                                ast::IntTy::Isize => IntegerType::Pointer(true),
1493                                ast::IntTy::I8 => IntegerType::Fixed(Integer::I8, true),
1494                                ast::IntTy::I16 => IntegerType::Fixed(Integer::I16, true),
1495                                ast::IntTy::I32 => IntegerType::Fixed(Integer::I32, true),
1496                                ast::IntTy::I64 => IntegerType::Fixed(Integer::I64, true),
1497                                ast::IntTy::I128 => IntegerType::Fixed(Integer::I128, true),
1498                            },
1499                            attr::IntType::UnsignedInt(x) => match x {
1500                                ast::UintTy::Usize => IntegerType::Pointer(false),
1501                                ast::UintTy::U8 => IntegerType::Fixed(Integer::I8, false),
1502                                ast::UintTy::U16 => IntegerType::Fixed(Integer::I16, false),
1503                                ast::UintTy::U32 => IntegerType::Fixed(Integer::I32, false),
1504                                ast::UintTy::U64 => IntegerType::Fixed(Integer::I64, false),
1505                                ast::UintTy::U128 => IntegerType::Fixed(Integer::I128, false),
1506                            },
1507                        });
1508                        ReprFlags::empty()
1509                    }
1510                    attr::ReprAlign(align) => {
1511                        max_align = max_align.max(Some(align));
1512                        ReprFlags::empty()
1513                    }
1514                });
1515            }
1516        }
1517
1518        // If `-Z randomize-layout` was enabled for the type definition then we can
1519        // consider performing layout randomization
1520        if self.sess.opts.unstable_opts.randomize_layout {
1521            flags.insert(ReprFlags::RANDOMIZE_LAYOUT);
1522        }
1523
1524        // box is special, on the one hand the compiler assumes an ordered layout, with the pointer
1525        // always at offset zero. On the other hand we want scalar abi optimizations.
1526        let is_box = self.is_lang_item(did.to_def_id(), LangItem::OwnedBox);
1527
1528        // This is here instead of layout because the choice must make it into metadata.
1529        if is_box {
1530            flags.insert(ReprFlags::IS_LINEAR);
1531        }
1532
1533        ReprOptions { int: size, align: max_align, pack: min_pack, flags, field_shuffle_seed }
1534    }
1535
1536    /// Look up the name of a definition across crates. This does not look at HIR.
1537    pub fn opt_item_name(self, def_id: impl IntoQueryParam<DefId>) -> Option<Symbol> {
1538        let def_id = def_id.into_query_param();
1539        if let Some(cnum) = def_id.as_crate_root() {
1540            Some(self.crate_name(cnum))
1541        } else {
1542            let def_key = self.def_key(def_id);
1543            match def_key.disambiguated_data.data {
1544                // The name of a constructor is that of its parent.
1545                rustc_hir::definitions::DefPathData::Ctor => self
1546                    .opt_item_name(DefId { krate: def_id.krate, index: def_key.parent.unwrap() }),
1547                _ => def_key.get_opt_name(),
1548            }
1549        }
1550    }
1551
1552    /// Look up the name of a definition across crates. This does not look at HIR.
1553    ///
1554    /// This method will ICE if the corresponding item does not have a name. In these cases, use
1555    /// [`opt_item_name`] instead.
1556    ///
1557    /// [`opt_item_name`]: Self::opt_item_name
1558    pub fn item_name(self, id: impl IntoQueryParam<DefId>) -> Symbol {
1559        let id = id.into_query_param();
1560        self.opt_item_name(id).unwrap_or_else(|| {
1561            bug!("item_name: no name for {:?}", self.def_path(id));
1562        })
1563    }
1564
1565    /// Look up the name and span of a definition.
1566    ///
1567    /// See [`item_name`][Self::item_name] for more information.
1568    pub fn opt_item_ident(self, def_id: impl IntoQueryParam<DefId>) -> Option<Ident> {
1569        let def_id = def_id.into_query_param();
1570        let def = self.opt_item_name(def_id)?;
1571        let span = self
1572            .def_ident_span(def_id)
1573            .unwrap_or_else(|| bug!("missing ident span for {def_id:?}"));
1574        Some(Ident::new(def, span))
1575    }
1576
1577    /// Look up the name and span of a definition.
1578    ///
1579    /// See [`item_name`][Self::item_name] for more information.
1580    pub fn item_ident(self, def_id: impl IntoQueryParam<DefId>) -> Ident {
1581        let def_id = def_id.into_query_param();
1582        self.opt_item_ident(def_id).unwrap_or_else(|| {
1583            bug!("item_ident: no name for {:?}", self.def_path(def_id));
1584        })
1585    }
1586
1587    pub fn opt_associated_item(self, def_id: DefId) -> Option<AssocItem> {
1588        if let DefKind::AssocConst | DefKind::AssocFn | DefKind::AssocTy = self.def_kind(def_id) {
1589            Some(self.associated_item(def_id))
1590        } else {
1591            None
1592        }
1593    }
1594
1595    /// If the `def_id` is an associated type that was desugared from a
1596    /// return-position `impl Trait` from a trait, then provide the source info
1597    /// about where that RPITIT came from.
1598    pub fn opt_rpitit_info(self, def_id: DefId) -> Option<ImplTraitInTraitData> {
1599        if let DefKind::AssocTy = self.def_kind(def_id)
1600            && let AssocKind::Type { data: AssocTypeData::Rpitit(rpitit_info) } =
1601                self.associated_item(def_id).kind
1602        {
1603            Some(rpitit_info)
1604        } else {
1605            None
1606        }
1607    }
1608
1609    pub fn find_field_index(self, ident: Ident, variant: &VariantDef) -> Option<FieldIdx> {
1610        variant.fields.iter_enumerated().find_map(|(i, field)| {
1611            self.hygienic_eq(ident, field.ident(self), variant.def_id).then_some(i)
1612        })
1613    }
1614
1615    /// Returns `Some` if the impls are the same polarity and the trait either
1616    /// has no items or is annotated `#[marker]` and prevents item overrides.
1617    #[instrument(level = "debug", skip(self), ret)]
1618    pub fn impls_are_allowed_to_overlap(
1619        self,
1620        def_id1: DefId,
1621        def_id2: DefId,
1622    ) -> Option<ImplOverlapKind> {
1623        let impl1 = self.impl_trait_header(def_id1).unwrap();
1624        let impl2 = self.impl_trait_header(def_id2).unwrap();
1625
1626        let trait_ref1 = impl1.trait_ref.skip_binder();
1627        let trait_ref2 = impl2.trait_ref.skip_binder();
1628
1629        // If either trait impl references an error, they're allowed to overlap,
1630        // as one of them essentially doesn't exist.
1631        if trait_ref1.references_error() || trait_ref2.references_error() {
1632            return Some(ImplOverlapKind::Permitted { marker: false });
1633        }
1634
1635        match (impl1.polarity, impl2.polarity) {
1636            (ImplPolarity::Reservation, _) | (_, ImplPolarity::Reservation) => {
1637                // `#[rustc_reservation_impl]` impls don't overlap with anything
1638                return Some(ImplOverlapKind::Permitted { marker: false });
1639            }
1640            (ImplPolarity::Positive, ImplPolarity::Negative)
1641            | (ImplPolarity::Negative, ImplPolarity::Positive) => {
1642                // `impl AutoTrait for Type` + `impl !AutoTrait for Type`
1643                return None;
1644            }
1645            (ImplPolarity::Positive, ImplPolarity::Positive)
1646            | (ImplPolarity::Negative, ImplPolarity::Negative) => {}
1647        };
1648
1649        let is_marker_impl = |trait_ref: TraitRef<'_>| self.trait_def(trait_ref.def_id).is_marker;
1650        let is_marker_overlap = is_marker_impl(trait_ref1) && is_marker_impl(trait_ref2);
1651
1652        if is_marker_overlap {
1653            return Some(ImplOverlapKind::Permitted { marker: true });
1654        }
1655
1656        None
1657    }
1658
1659    /// Returns `ty::VariantDef` if `res` refers to a struct,
1660    /// or variant or their constructors, panics otherwise.
1661    pub fn expect_variant_res(self, res: Res) -> &'tcx VariantDef {
1662        match res {
1663            Res::Def(DefKind::Variant, did) => {
1664                let enum_did = self.parent(did);
1665                self.adt_def(enum_did).variant_with_id(did)
1666            }
1667            Res::Def(DefKind::Struct | DefKind::Union, did) => self.adt_def(did).non_enum_variant(),
1668            Res::Def(DefKind::Ctor(CtorOf::Variant, ..), variant_ctor_did) => {
1669                let variant_did = self.parent(variant_ctor_did);
1670                let enum_did = self.parent(variant_did);
1671                self.adt_def(enum_did).variant_with_ctor_id(variant_ctor_did)
1672            }
1673            Res::Def(DefKind::Ctor(CtorOf::Struct, ..), ctor_did) => {
1674                let struct_did = self.parent(ctor_did);
1675                self.adt_def(struct_did).non_enum_variant()
1676            }
1677            _ => bug!("expect_variant_res used with unexpected res {:?}", res),
1678        }
1679    }
1680
1681    /// Returns the possibly-auto-generated MIR of a [`ty::InstanceKind`].
1682    #[instrument(skip(self), level = "debug")]
1683    pub fn instance_mir(self, instance: ty::InstanceKind<'tcx>) -> &'tcx Body<'tcx> {
1684        match instance {
1685            ty::InstanceKind::Item(def) => {
1686                debug!("calling def_kind on def: {:?}", def);
1687                let def_kind = self.def_kind(def);
1688                debug!("returned from def_kind: {:?}", def_kind);
1689                match def_kind {
1690                    DefKind::Const
1691                    | DefKind::Static { .. }
1692                    | DefKind::AssocConst
1693                    | DefKind::Ctor(..)
1694                    | DefKind::AnonConst
1695                    | DefKind::InlineConst => self.mir_for_ctfe(def),
1696                    // If the caller wants `mir_for_ctfe` of a function they should not be using
1697                    // `instance_mir`, so we'll assume const fn also wants the optimized version.
1698                    _ => self.optimized_mir(def),
1699                }
1700            }
1701            ty::InstanceKind::VTableShim(..)
1702            | ty::InstanceKind::ReifyShim(..)
1703            | ty::InstanceKind::Intrinsic(..)
1704            | ty::InstanceKind::FnPtrShim(..)
1705            | ty::InstanceKind::Virtual(..)
1706            | ty::InstanceKind::ClosureOnceShim { .. }
1707            | ty::InstanceKind::ConstructCoroutineInClosureShim { .. }
1708            | ty::InstanceKind::FutureDropPollShim(..)
1709            | ty::InstanceKind::DropGlue(..)
1710            | ty::InstanceKind::CloneShim(..)
1711            | ty::InstanceKind::ThreadLocalShim(..)
1712            | ty::InstanceKind::FnPtrAddrShim(..)
1713            | ty::InstanceKind::AsyncDropGlueCtorShim(..)
1714            | ty::InstanceKind::AsyncDropGlue(..) => self.mir_shims(instance),
1715        }
1716    }
1717
1718    /// Gets all attributes with the given name.
1719    pub fn get_attrs(
1720        self,
1721        did: impl Into<DefId>,
1722        attr: Symbol,
1723    ) -> impl Iterator<Item = &'tcx hir::Attribute> {
1724        self.get_all_attrs(did).iter().filter(move |a: &&hir::Attribute| a.has_name(attr))
1725    }
1726
1727    /// Gets all attributes.
1728    ///
1729    /// To see if an item has a specific attribute, you should use
1730    /// [`rustc_hir::find_attr!`] so you can use matching.
1731    pub fn get_all_attrs(self, did: impl Into<DefId>) -> &'tcx [hir::Attribute] {
1732        let did: DefId = did.into();
1733        if let Some(did) = did.as_local() {
1734            self.hir_attrs(self.local_def_id_to_hir_id(did))
1735        } else {
1736            self.attrs_for_def(did)
1737        }
1738    }
1739
1740    /// Get an attribute from the diagnostic attribute namespace
1741    ///
1742    /// This function requests an attribute with the following structure:
1743    ///
1744    /// `#[diagnostic::$attr]`
1745    ///
1746    /// This function performs feature checking, so if an attribute is returned
1747    /// it can be used by the consumer
1748    pub fn get_diagnostic_attr(
1749        self,
1750        did: impl Into<DefId>,
1751        attr: Symbol,
1752    ) -> Option<&'tcx hir::Attribute> {
1753        let did: DefId = did.into();
1754        if did.as_local().is_some() {
1755            // it's a crate local item, we need to check feature flags
1756            if rustc_feature::is_stable_diagnostic_attribute(attr, self.features()) {
1757                self.get_attrs_by_path(did, &[sym::diagnostic, sym::do_not_recommend]).next()
1758            } else {
1759                None
1760            }
1761        } else {
1762            // we filter out unstable diagnostic attributes before
1763            // encoding attributes
1764            debug_assert!(rustc_feature::encode_cross_crate(attr));
1765            self.attrs_for_def(did)
1766                .iter()
1767                .find(|a| matches!(a.path().as_ref(), [sym::diagnostic, a] if *a == attr))
1768        }
1769    }
1770
1771    pub fn get_attrs_by_path(
1772        self,
1773        did: DefId,
1774        attr: &[Symbol],
1775    ) -> impl Iterator<Item = &'tcx hir::Attribute> {
1776        let filter_fn = move |a: &&hir::Attribute| a.path_matches(attr);
1777        if let Some(did) = did.as_local() {
1778            self.hir_attrs(self.local_def_id_to_hir_id(did)).iter().filter(filter_fn)
1779        } else {
1780            self.attrs_for_def(did).iter().filter(filter_fn)
1781        }
1782    }
1783
1784    pub fn get_attr(self, did: impl Into<DefId>, attr: Symbol) -> Option<&'tcx hir::Attribute> {
1785        if cfg!(debug_assertions) && !rustc_feature::is_valid_for_get_attr(attr) {
1786            let did: DefId = did.into();
1787            bug!("get_attr: unexpected called with DefId `{:?}`, attr `{:?}`", did, attr);
1788        } else {
1789            self.get_attrs(did, attr).next()
1790        }
1791    }
1792
1793    /// Determines whether an item is annotated with an attribute.
1794    pub fn has_attr(self, did: impl Into<DefId>, attr: Symbol) -> bool {
1795        self.get_attrs(did, attr).next().is_some()
1796    }
1797
1798    /// Determines whether an item is annotated with a multi-segment attribute
1799    pub fn has_attrs_with_path(self, did: impl Into<DefId>, attrs: &[Symbol]) -> bool {
1800        self.get_attrs_by_path(did.into(), attrs).next().is_some()
1801    }
1802
1803    /// Returns `true` if this is an `auto trait`.
1804    pub fn trait_is_auto(self, trait_def_id: DefId) -> bool {
1805        self.trait_def(trait_def_id).has_auto_impl
1806    }
1807
1808    /// Returns `true` if this is coinductive, either because it is
1809    /// an auto trait or because it has the `#[rustc_coinductive]` attribute.
1810    pub fn trait_is_coinductive(self, trait_def_id: DefId) -> bool {
1811        self.trait_def(trait_def_id).is_coinductive
1812    }
1813
1814    /// Returns `true` if this is a trait alias.
1815    pub fn trait_is_alias(self, trait_def_id: DefId) -> bool {
1816        self.def_kind(trait_def_id) == DefKind::TraitAlias
1817    }
1818
1819    /// Arena-alloc of LayoutError for coroutine layout
1820    fn layout_error(self, err: LayoutError<'tcx>) -> &'tcx LayoutError<'tcx> {
1821        self.arena.alloc(err)
1822    }
1823
1824    /// Returns layout of a non-async-drop coroutine. Layout might be unavailable if the
1825    /// coroutine is tainted by errors.
1826    ///
1827    /// Takes `coroutine_kind` which can be acquired from the `CoroutineArgs::kind_ty`,
1828    /// e.g. `args.as_coroutine().kind_ty()`.
1829    fn ordinary_coroutine_layout(
1830        self,
1831        def_id: DefId,
1832        args: GenericArgsRef<'tcx>,
1833    ) -> Result<&'tcx CoroutineLayout<'tcx>, &'tcx LayoutError<'tcx>> {
1834        let coroutine_kind_ty = args.as_coroutine().kind_ty();
1835        let mir = self.optimized_mir(def_id);
1836        let ty = || Ty::new_coroutine(self, def_id, args);
1837        // Regular coroutine
1838        if coroutine_kind_ty.is_unit() {
1839            mir.coroutine_layout_raw().ok_or_else(|| self.layout_error(LayoutError::Unknown(ty())))
1840        } else {
1841            // If we have a `Coroutine` that comes from an coroutine-closure,
1842            // then it may be a by-move or by-ref body.
1843            let ty::Coroutine(_, identity_args) =
1844                *self.type_of(def_id).instantiate_identity().kind()
1845            else {
1846                unreachable!();
1847            };
1848            let identity_kind_ty = identity_args.as_coroutine().kind_ty();
1849            // If the types differ, then we must be getting the by-move body of
1850            // a by-ref coroutine.
1851            if identity_kind_ty == coroutine_kind_ty {
1852                mir.coroutine_layout_raw()
1853                    .ok_or_else(|| self.layout_error(LayoutError::Unknown(ty())))
1854            } else {
1855                assert_matches!(coroutine_kind_ty.to_opt_closure_kind(), Some(ClosureKind::FnOnce));
1856                assert_matches!(
1857                    identity_kind_ty.to_opt_closure_kind(),
1858                    Some(ClosureKind::Fn | ClosureKind::FnMut)
1859                );
1860                self.optimized_mir(self.coroutine_by_move_body_def_id(def_id))
1861                    .coroutine_layout_raw()
1862                    .ok_or_else(|| self.layout_error(LayoutError::Unknown(ty())))
1863            }
1864        }
1865    }
1866
1867    /// Returns layout of a `async_drop_in_place::{closure}` coroutine
1868    ///   (returned from `async fn async_drop_in_place<T>(..)`).
1869    /// Layout might be unavailable if the coroutine is tainted by errors.
1870    fn async_drop_coroutine_layout(
1871        self,
1872        def_id: DefId,
1873        args: GenericArgsRef<'tcx>,
1874    ) -> Result<&'tcx CoroutineLayout<'tcx>, &'tcx LayoutError<'tcx>> {
1875        let ty = || Ty::new_coroutine(self, def_id, args);
1876        if args[0].has_placeholders() || args[0].has_non_region_param() {
1877            return Err(self.layout_error(LayoutError::TooGeneric(ty())));
1878        }
1879        let instance = InstanceKind::AsyncDropGlue(def_id, Ty::new_coroutine(self, def_id, args));
1880        self.mir_shims(instance)
1881            .coroutine_layout_raw()
1882            .ok_or_else(|| self.layout_error(LayoutError::Unknown(ty())))
1883    }
1884
1885    /// Returns layout of a coroutine. Layout might be unavailable if the
1886    /// coroutine is tainted by errors.
1887    pub fn coroutine_layout(
1888        self,
1889        def_id: DefId,
1890        args: GenericArgsRef<'tcx>,
1891    ) -> Result<&'tcx CoroutineLayout<'tcx>, &'tcx LayoutError<'tcx>> {
1892        if self.is_async_drop_in_place_coroutine(def_id) {
1893            // layout of `async_drop_in_place<T>::{closure}` in case,
1894            // when T is a coroutine, contains this internal coroutine's ptr in upvars
1895            // and doesn't require any locals. Here is an `empty coroutine's layout`
1896            let arg_cor_ty = args.first().unwrap().expect_ty();
1897            if arg_cor_ty.is_coroutine() {
1898                let span = self.def_span(def_id);
1899                let source_info = SourceInfo::outermost(span);
1900                // Even minimal, empty coroutine has 3 states (RESERVED_VARIANTS),
1901                // so variant_fields and variant_source_info should have 3 elements.
1902                let variant_fields: IndexVec<VariantIdx, IndexVec<FieldIdx, CoroutineSavedLocal>> =
1903                    iter::repeat(IndexVec::new()).take(CoroutineArgs::RESERVED_VARIANTS).collect();
1904                let variant_source_info: IndexVec<VariantIdx, SourceInfo> =
1905                    iter::repeat(source_info).take(CoroutineArgs::RESERVED_VARIANTS).collect();
1906                let proxy_layout = CoroutineLayout {
1907                    field_tys: [].into(),
1908                    field_names: [].into(),
1909                    variant_fields,
1910                    variant_source_info,
1911                    storage_conflicts: BitMatrix::new(0, 0),
1912                };
1913                return Ok(self.arena.alloc(proxy_layout));
1914            } else {
1915                self.async_drop_coroutine_layout(def_id, args)
1916            }
1917        } else {
1918            self.ordinary_coroutine_layout(def_id, args)
1919        }
1920    }
1921
1922    /// Given the `DefId` of an impl, returns the `DefId` of the trait it implements.
1923    /// If it implements no trait, returns `None`.
1924    pub fn trait_id_of_impl(self, def_id: DefId) -> Option<DefId> {
1925        self.impl_trait_ref(def_id).map(|tr| tr.skip_binder().def_id)
1926    }
1927
1928    /// If the given `DefId` is an associated item, returns the `DefId` and `DefKind` of the parent trait or impl.
1929    pub fn assoc_parent(self, def_id: DefId) -> Option<(DefId, DefKind)> {
1930        if !self.def_kind(def_id).is_assoc() {
1931            return None;
1932        }
1933        let parent = self.parent(def_id);
1934        let def_kind = self.def_kind(parent);
1935        Some((parent, def_kind))
1936    }
1937
1938    /// If the given `DefId` is an associated item of a trait,
1939    /// returns the `DefId` of the trait; otherwise, returns `None`.
1940    pub fn trait_of_assoc(self, def_id: DefId) -> Option<DefId> {
1941        match self.assoc_parent(def_id) {
1942            Some((id, DefKind::Trait)) => Some(id),
1943            _ => None,
1944        }
1945    }
1946
1947    /// If the given `DefId` is an associated item of an impl,
1948    /// returns the `DefId` of the impl; otherwise returns `None`.
1949    pub fn impl_of_assoc(self, def_id: DefId) -> Option<DefId> {
1950        match self.assoc_parent(def_id) {
1951            Some((id, DefKind::Impl { .. })) => Some(id),
1952            _ => None,
1953        }
1954    }
1955
1956    /// If the given `DefId` is an associated item of an inherent impl,
1957    /// returns the `DefId` of the impl; otherwise, returns `None`.
1958    pub fn inherent_impl_of_assoc(self, def_id: DefId) -> Option<DefId> {
1959        match self.assoc_parent(def_id) {
1960            Some((id, DefKind::Impl { of_trait: false })) => Some(id),
1961            _ => None,
1962        }
1963    }
1964
1965    /// If the given `DefId` is an associated item of a trait impl,
1966    /// returns the `DefId` of the impl; otherwise, returns `None`.
1967    pub fn trait_impl_of_assoc(self, def_id: DefId) -> Option<DefId> {
1968        match self.assoc_parent(def_id) {
1969            Some((id, DefKind::Impl { of_trait: true })) => Some(id),
1970            _ => None,
1971        }
1972    }
1973
1974    pub fn is_exportable(self, def_id: DefId) -> bool {
1975        self.exportable_items(def_id.krate).contains(&def_id)
1976    }
1977
1978    /// Check if the given `DefId` is `#\[automatically_derived\]`, *and*
1979    /// whether it was produced by expanding a builtin derive macro.
1980    pub fn is_builtin_derived(self, def_id: DefId) -> bool {
1981        if self.is_automatically_derived(def_id)
1982            && let Some(def_id) = def_id.as_local()
1983            && let outer = self.def_span(def_id).ctxt().outer_expn_data()
1984            && matches!(outer.kind, ExpnKind::Macro(MacroKind::Derive, _))
1985            && find_attr!(
1986                self.get_all_attrs(outer.macro_def_id.unwrap()),
1987                AttributeKind::RustcBuiltinMacro { .. }
1988            )
1989        {
1990            true
1991        } else {
1992            false
1993        }
1994    }
1995
1996    /// Check if the given `DefId` is `#\[automatically_derived\]`.
1997    pub fn is_automatically_derived(self, def_id: DefId) -> bool {
1998        find_attr!(self.get_all_attrs(def_id), AttributeKind::AutomaticallyDerived(..))
1999    }
2000
2001    /// Looks up the span of `impl_did` if the impl is local; otherwise returns `Err`
2002    /// with the name of the crate containing the impl.
2003    pub fn span_of_impl(self, impl_def_id: DefId) -> Result<Span, Symbol> {
2004        if let Some(impl_def_id) = impl_def_id.as_local() {
2005            Ok(self.def_span(impl_def_id))
2006        } else {
2007            Err(self.crate_name(impl_def_id.krate))
2008        }
2009    }
2010
2011    /// Hygienically compares a use-site name (`use_name`) for a field or an associated item with
2012    /// its supposed definition name (`def_name`). The method also needs `DefId` of the supposed
2013    /// definition's parent/scope to perform comparison.
2014    pub fn hygienic_eq(self, use_ident: Ident, def_ident: Ident, def_parent_def_id: DefId) -> bool {
2015        // We could use `Ident::eq` here, but we deliberately don't. The identifier
2016        // comparison fails frequently, and we want to avoid the expensive
2017        // `normalize_to_macros_2_0()` calls required for the span comparison whenever possible.
2018        use_ident.name == def_ident.name
2019            && use_ident
2020                .span
2021                .ctxt()
2022                .hygienic_eq(def_ident.span.ctxt(), self.expn_that_defined(def_parent_def_id))
2023    }
2024
2025    pub fn adjust_ident(self, mut ident: Ident, scope: DefId) -> Ident {
2026        ident.span.normalize_to_macros_2_0_and_adjust(self.expn_that_defined(scope));
2027        ident
2028    }
2029
2030    // FIXME(vincenzopalazzo): move the HirId to a LocalDefId
2031    pub fn adjust_ident_and_get_scope(
2032        self,
2033        mut ident: Ident,
2034        scope: DefId,
2035        block: hir::HirId,
2036    ) -> (Ident, DefId) {
2037        let scope = ident
2038            .span
2039            .normalize_to_macros_2_0_and_adjust(self.expn_that_defined(scope))
2040            .and_then(|actual_expansion| actual_expansion.expn_data().parent_module)
2041            .unwrap_or_else(|| self.parent_module(block).to_def_id());
2042        (ident, scope)
2043    }
2044
2045    /// Checks whether this is a `const fn`. Returns `false` for non-functions.
2046    ///
2047    /// Even if this returns `true`, constness may still be unstable!
2048    #[inline]
2049    pub fn is_const_fn(self, def_id: DefId) -> bool {
2050        matches!(
2051            self.def_kind(def_id),
2052            DefKind::Fn | DefKind::AssocFn | DefKind::Ctor(_, CtorKind::Fn) | DefKind::Closure
2053        ) && self.constness(def_id) == hir::Constness::Const
2054    }
2055
2056    /// Whether this item is conditionally constant for the purposes of the
2057    /// effects implementation.
2058    ///
2059    /// This roughly corresponds to all const functions and other callable
2060    /// items, along with const impls and traits, and associated types within
2061    /// those impls and traits.
2062    pub fn is_conditionally_const(self, def_id: impl Into<DefId>) -> bool {
2063        let def_id: DefId = def_id.into();
2064        match self.def_kind(def_id) {
2065            DefKind::Impl { of_trait: true } => {
2066                let header = self.impl_trait_header(def_id).unwrap();
2067                header.constness == hir::Constness::Const
2068                    && self.is_const_trait(header.trait_ref.skip_binder().def_id)
2069            }
2070            DefKind::Fn | DefKind::Ctor(_, CtorKind::Fn) => {
2071                self.constness(def_id) == hir::Constness::Const
2072            }
2073            DefKind::Trait => self.is_const_trait(def_id),
2074            DefKind::AssocTy => {
2075                let parent_def_id = self.parent(def_id);
2076                match self.def_kind(parent_def_id) {
2077                    DefKind::Impl { of_trait: false } => false,
2078                    DefKind::Impl { of_trait: true } | DefKind::Trait => {
2079                        self.is_conditionally_const(parent_def_id)
2080                    }
2081                    _ => bug!("unexpected parent item of associated type: {parent_def_id:?}"),
2082                }
2083            }
2084            DefKind::AssocFn => {
2085                let parent_def_id = self.parent(def_id);
2086                match self.def_kind(parent_def_id) {
2087                    DefKind::Impl { of_trait: false } => {
2088                        self.constness(def_id) == hir::Constness::Const
2089                    }
2090                    DefKind::Impl { of_trait: true } | DefKind::Trait => {
2091                        self.is_conditionally_const(parent_def_id)
2092                    }
2093                    _ => bug!("unexpected parent item of associated fn: {parent_def_id:?}"),
2094                }
2095            }
2096            DefKind::OpaqueTy => match self.opaque_ty_origin(def_id) {
2097                hir::OpaqueTyOrigin::FnReturn { parent, .. } => self.is_conditionally_const(parent),
2098                hir::OpaqueTyOrigin::AsyncFn { .. } => false,
2099                // FIXME(const_trait_impl): ATPITs could be conditionally const?
2100                hir::OpaqueTyOrigin::TyAlias { .. } => false,
2101            },
2102            DefKind::Closure => {
2103                // Closures and RPITs will eventually have const conditions
2104                // for `[const]` bounds.
2105                false
2106            }
2107            DefKind::Ctor(_, CtorKind::Const)
2108            | DefKind::Impl { of_trait: false }
2109            | DefKind::Mod
2110            | DefKind::Struct
2111            | DefKind::Union
2112            | DefKind::Enum
2113            | DefKind::Variant
2114            | DefKind::TyAlias
2115            | DefKind::ForeignTy
2116            | DefKind::TraitAlias
2117            | DefKind::TyParam
2118            | DefKind::Const
2119            | DefKind::ConstParam
2120            | DefKind::Static { .. }
2121            | DefKind::AssocConst
2122            | DefKind::Macro(_)
2123            | DefKind::ExternCrate
2124            | DefKind::Use
2125            | DefKind::ForeignMod
2126            | DefKind::AnonConst
2127            | DefKind::InlineConst
2128            | DefKind::Field
2129            | DefKind::LifetimeParam
2130            | DefKind::GlobalAsm
2131            | DefKind::SyntheticCoroutineBody => false,
2132        }
2133    }
2134
2135    #[inline]
2136    pub fn is_const_trait(self, def_id: DefId) -> bool {
2137        self.trait_def(def_id).constness == hir::Constness::Const
2138    }
2139
2140    #[inline]
2141    pub fn is_const_default_method(self, def_id: DefId) -> bool {
2142        matches!(self.trait_of_assoc(def_id), Some(trait_id) if self.is_const_trait(trait_id))
2143    }
2144
2145    pub fn impl_method_has_trait_impl_trait_tys(self, def_id: DefId) -> bool {
2146        if self.def_kind(def_id) != DefKind::AssocFn {
2147            return false;
2148        }
2149
2150        let Some(item) = self.opt_associated_item(def_id) else {
2151            return false;
2152        };
2153        if item.container != ty::AssocItemContainer::Impl {
2154            return false;
2155        }
2156
2157        let Some(trait_item_def_id) = item.trait_item_def_id else {
2158            return false;
2159        };
2160
2161        return !self
2162            .associated_types_for_impl_traits_in_associated_fn(trait_item_def_id)
2163            .is_empty();
2164    }
2165}
2166
2167pub fn provide(providers: &mut Providers) {
2168    closure::provide(providers);
2169    context::provide(providers);
2170    erase_regions::provide(providers);
2171    inhabitedness::provide(providers);
2172    util::provide(providers);
2173    print::provide(providers);
2174    super::util::bug::provide(providers);
2175    *providers = Providers {
2176        trait_impls_of: trait_def::trait_impls_of_provider,
2177        incoherent_impls: trait_def::incoherent_impls_provider,
2178        trait_impls_in_crate: trait_def::trait_impls_in_crate_provider,
2179        traits: trait_def::traits_provider,
2180        vtable_allocation: vtable::vtable_allocation_provider,
2181        ..*providers
2182    };
2183}
2184
2185/// A map for the local crate mapping each type to a vector of its
2186/// inherent impls. This is not meant to be used outside of coherence;
2187/// rather, you should request the vector for a specific type via
2188/// `tcx.inherent_impls(def_id)` so as to minimize your dependencies
2189/// (constructing this map requires touching the entire crate).
2190#[derive(Clone, Debug, Default, HashStable)]
2191pub struct CrateInherentImpls {
2192    pub inherent_impls: FxIndexMap<LocalDefId, Vec<DefId>>,
2193    pub incoherent_impls: FxIndexMap<SimplifiedType, Vec<LocalDefId>>,
2194}
2195
2196#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, TyEncodable, HashStable)]
2197pub struct SymbolName<'tcx> {
2198    /// `&str` gives a consistent ordering, which ensures reproducible builds.
2199    pub name: &'tcx str,
2200}
2201
2202impl<'tcx> SymbolName<'tcx> {
2203    pub fn new(tcx: TyCtxt<'tcx>, name: &str) -> SymbolName<'tcx> {
2204        SymbolName { name: tcx.arena.alloc_str(name) }
2205    }
2206}
2207
2208impl<'tcx> fmt::Display for SymbolName<'tcx> {
2209    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
2210        fmt::Display::fmt(&self.name, fmt)
2211    }
2212}
2213
2214impl<'tcx> fmt::Debug for SymbolName<'tcx> {
2215    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
2216        fmt::Display::fmt(&self.name, fmt)
2217    }
2218}
2219
2220/// The constituent parts of a type level constant of kind ADT or array.
2221#[derive(Copy, Clone, Debug, HashStable)]
2222pub struct DestructuredConst<'tcx> {
2223    pub variant: Option<VariantIdx>,
2224    pub fields: &'tcx [ty::Const<'tcx>],
2225}