rustc_infer/infer/
mod.rs

1use std::cell::{Cell, RefCell};
2use std::fmt;
3
4pub use at::DefineOpaqueTypes;
5use free_regions::RegionRelations;
6pub use freshen::TypeFreshener;
7use lexical_region_resolve::LexicalRegionResolutions;
8pub use lexical_region_resolve::RegionResolutionError;
9pub use opaque_types::{OpaqueTypeStorage, OpaqueTypeStorageEntries, OpaqueTypeTable};
10use region_constraints::{
11    GenericKind, RegionConstraintCollector, RegionConstraintStorage, VarInfos, VerifyBound,
12};
13pub use relate::StructurallyRelateAliases;
14pub use relate::combine::PredicateEmittingRelation;
15use rustc_data_structures::fx::{FxHashSet, FxIndexMap};
16use rustc_data_structures::undo_log::{Rollback, UndoLogs};
17use rustc_data_structures::unify as ut;
18use rustc_errors::{DiagCtxtHandle, ErrorGuaranteed};
19use rustc_hir as hir;
20use rustc_hir::def_id::{DefId, LocalDefId};
21use rustc_macros::extension;
22pub use rustc_macros::{TypeFoldable, TypeVisitable};
23use rustc_middle::bug;
24use rustc_middle::infer::canonical::{CanonicalQueryInput, CanonicalVarValues};
25use rustc_middle::mir::ConstraintCategory;
26use rustc_middle::traits::select;
27use rustc_middle::traits::solve::Goal;
28use rustc_middle::ty::error::{ExpectedFound, TypeError};
29use rustc_middle::ty::{
30    self, BoundVarReplacerDelegate, ConstVid, FloatVid, GenericArg, GenericArgKind, GenericArgs,
31    GenericArgsRef, GenericParamDefKind, InferConst, IntVid, OpaqueHiddenType, OpaqueTypeKey,
32    PseudoCanonicalInput, Term, TermKind, Ty, TyCtxt, TyVid, TypeFoldable, TypeFolder,
33    TypeSuperFoldable, TypeVisitable, TypeVisitableExt, TypingEnv, TypingMode, fold_regions,
34};
35use rustc_span::{DUMMY_SP, Span, Symbol};
36use snapshot::undo_log::InferCtxtUndoLogs;
37use tracing::{debug, instrument};
38use type_variable::TypeVariableOrigin;
39
40use crate::infer::snapshot::undo_log::UndoLog;
41use crate::infer::unify_key::{ConstVariableOrigin, ConstVariableValue, ConstVidKey};
42use crate::traits::{
43    self, ObligationCause, ObligationInspector, PredicateObligation, PredicateObligations,
44    TraitEngine,
45};
46
47pub mod at;
48pub mod canonical;
49mod context;
50mod free_regions;
51mod freshen;
52mod lexical_region_resolve;
53mod opaque_types;
54pub mod outlives;
55mod projection;
56pub mod region_constraints;
57pub mod relate;
58pub mod resolve;
59pub(crate) mod snapshot;
60mod type_variable;
61mod unify_key;
62
63/// `InferOk<'tcx, ()>` is used a lot. It may seem like a useless wrapper
64/// around `PredicateObligations<'tcx>`, but it has one important property:
65/// because `InferOk` is marked with `#[must_use]`, if you have a method
66/// `InferCtxt::f` that returns `InferResult<'tcx, ()>` and you call it with
67/// `infcx.f()?;` you'll get a warning about the obligations being discarded
68/// without use, which is probably unintentional and has been a source of bugs
69/// in the past.
70#[must_use]
71#[derive(Debug)]
72pub struct InferOk<'tcx, T> {
73    pub value: T,
74    pub obligations: PredicateObligations<'tcx>,
75}
76pub type InferResult<'tcx, T> = Result<InferOk<'tcx, T>, TypeError<'tcx>>;
77
78pub(crate) type FixupResult<T> = Result<T, FixupError>; // "fixup result"
79
80pub(crate) type UnificationTable<'a, 'tcx, T> = ut::UnificationTable<
81    ut::InPlace<T, &'a mut ut::UnificationStorage<T>, &'a mut InferCtxtUndoLogs<'tcx>>,
82>;
83
84/// This type contains all the things within `InferCtxt` that sit within a
85/// `RefCell` and are involved with taking/rolling back snapshots. Snapshot
86/// operations are hot enough that we want only one call to `borrow_mut` per
87/// call to `start_snapshot` and `rollback_to`.
88#[derive(Clone)]
89pub struct InferCtxtInner<'tcx> {
90    undo_log: InferCtxtUndoLogs<'tcx>,
91
92    /// Cache for projections.
93    ///
94    /// This cache is snapshotted along with the infcx.
95    projection_cache: traits::ProjectionCacheStorage<'tcx>,
96
97    /// We instantiate `UnificationTable` with `bounds<Ty>` because the types
98    /// that might instantiate a general type variable have an order,
99    /// represented by its upper and lower bounds.
100    type_variable_storage: type_variable::TypeVariableStorage<'tcx>,
101
102    /// Map from const parameter variable to the kind of const it represents.
103    const_unification_storage: ut::UnificationTableStorage<ConstVidKey<'tcx>>,
104
105    /// Map from integral variable to the kind of integer it represents.
106    int_unification_storage: ut::UnificationTableStorage<ty::IntVid>,
107
108    /// Map from floating variable to the kind of float it represents.
109    float_unification_storage: ut::UnificationTableStorage<ty::FloatVid>,
110
111    /// Tracks the set of region variables and the constraints between them.
112    ///
113    /// This is initially `Some(_)` but when
114    /// `resolve_regions_and_report_errors` is invoked, this gets set to `None`
115    /// -- further attempts to perform unification, etc., may fail if new
116    /// region constraints would've been added.
117    region_constraint_storage: Option<RegionConstraintStorage<'tcx>>,
118
119    /// A set of constraints that regionck must validate.
120    ///
121    /// Each constraint has the form `T:'a`, meaning "some type `T` must
122    /// outlive the lifetime 'a". These constraints derive from
123    /// instantiated type parameters. So if you had a struct defined
124    /// like the following:
125    /// ```ignore (illustrative)
126    /// struct Foo<T: 'static> { ... }
127    /// ```
128    /// In some expression `let x = Foo { ... }`, it will
129    /// instantiate the type parameter `T` with a fresh type `$0`. At
130    /// the same time, it will record a region obligation of
131    /// `$0: 'static`. This will get checked later by regionck. (We
132    /// can't generally check these things right away because we have
133    /// to wait until types are resolved.)
134    ///
135    /// These are stored in a map keyed to the id of the innermost
136    /// enclosing fn body / static initializer expression. This is
137    /// because the location where the obligation was incurred can be
138    /// relevant with respect to which sublifetime assumptions are in
139    /// place. The reason that we store under the fn-id, and not
140    /// something more fine-grained, is so that it is easier for
141    /// regionck to be sure that it has found *all* the region
142    /// obligations (otherwise, it's easy to fail to walk to a
143    /// particular node-id).
144    ///
145    /// Before running `resolve_regions_and_report_errors`, the creator
146    /// of the inference context is expected to invoke
147    /// [`InferCtxt::process_registered_region_obligations`]
148    /// for each body-id in this map, which will process the
149    /// obligations within. This is expected to be done 'late enough'
150    /// that all type inference variables have been bound and so forth.
151    region_obligations: Vec<TypeOutlivesConstraint<'tcx>>,
152
153    /// The outlives bounds that we assume must hold about placeholders that
154    /// come from instantiating the binder of coroutine-witnesses. These bounds
155    /// are deduced from the well-formedness of the witness's types, and are
156    /// necessary because of the way we anonymize the regions in a coroutine,
157    /// which may cause types to no longer be considered well-formed.
158    region_assumptions: Vec<ty::ArgOutlivesPredicate<'tcx>>,
159
160    /// `-Znext-solver`: Successfully proven goals during HIR typeck which
161    /// reference inference variables and get reproven in case MIR type check
162    /// fails to prove something.
163    ///
164    /// See the documentation of `InferCtxt::in_hir_typeck` for more details.
165    hir_typeck_potentially_region_dependent_goals: Vec<PredicateObligation<'tcx>>,
166
167    /// Caches for opaque type inference.
168    opaque_type_storage: OpaqueTypeStorage<'tcx>,
169}
170
171impl<'tcx> InferCtxtInner<'tcx> {
172    fn new() -> InferCtxtInner<'tcx> {
173        InferCtxtInner {
174            undo_log: InferCtxtUndoLogs::default(),
175
176            projection_cache: Default::default(),
177            type_variable_storage: Default::default(),
178            const_unification_storage: Default::default(),
179            int_unification_storage: Default::default(),
180            float_unification_storage: Default::default(),
181            region_constraint_storage: Some(Default::default()),
182            region_obligations: Default::default(),
183            region_assumptions: Default::default(),
184            hir_typeck_potentially_region_dependent_goals: Default::default(),
185            opaque_type_storage: Default::default(),
186        }
187    }
188
189    #[inline]
190    pub fn region_obligations(&self) -> &[TypeOutlivesConstraint<'tcx>] {
191        &self.region_obligations
192    }
193
194    #[inline]
195    pub fn region_assumptions(&self) -> &[ty::ArgOutlivesPredicate<'tcx>] {
196        &self.region_assumptions
197    }
198
199    #[inline]
200    pub fn projection_cache(&mut self) -> traits::ProjectionCache<'_, 'tcx> {
201        self.projection_cache.with_log(&mut self.undo_log)
202    }
203
204    #[inline]
205    fn try_type_variables_probe_ref(
206        &self,
207        vid: ty::TyVid,
208    ) -> Option<&type_variable::TypeVariableValue<'tcx>> {
209        // Uses a read-only view of the unification table, this way we don't
210        // need an undo log.
211        self.type_variable_storage.eq_relations_ref().try_probe_value(vid)
212    }
213
214    #[inline]
215    fn type_variables(&mut self) -> type_variable::TypeVariableTable<'_, 'tcx> {
216        self.type_variable_storage.with_log(&mut self.undo_log)
217    }
218
219    #[inline]
220    pub fn opaque_types(&mut self) -> opaque_types::OpaqueTypeTable<'_, 'tcx> {
221        self.opaque_type_storage.with_log(&mut self.undo_log)
222    }
223
224    #[inline]
225    fn int_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ty::IntVid> {
226        self.int_unification_storage.with_log(&mut self.undo_log)
227    }
228
229    #[inline]
230    fn float_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ty::FloatVid> {
231        self.float_unification_storage.with_log(&mut self.undo_log)
232    }
233
234    #[inline]
235    fn const_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ConstVidKey<'tcx>> {
236        self.const_unification_storage.with_log(&mut self.undo_log)
237    }
238
239    #[inline]
240    pub fn unwrap_region_constraints(&mut self) -> RegionConstraintCollector<'_, 'tcx> {
241        self.region_constraint_storage
242            .as_mut()
243            .expect("region constraints already solved")
244            .with_log(&mut self.undo_log)
245    }
246}
247
248pub struct InferCtxt<'tcx> {
249    pub tcx: TyCtxt<'tcx>,
250
251    /// The mode of this inference context, see the struct documentation
252    /// for more details.
253    typing_mode: TypingMode<'tcx>,
254
255    /// Whether this inference context should care about region obligations in
256    /// the root universe. Most notably, this is used during HIR typeck as region
257    /// solving is left to borrowck instead.
258    pub considering_regions: bool,
259    /// `-Znext-solver`: Whether this inference context is used by HIR typeck. If so, we
260    /// need to make sure we don't rely on region identity in the trait solver or when
261    /// relating types. This is necessary as borrowck starts by replacing each occurrence of a
262    /// free region with a unique inference variable. If HIR typeck ends up depending on two
263    /// regions being equal we'd get unexpected mismatches between HIR typeck and MIR typeck,
264    /// resulting in an ICE.
265    ///
266    /// The trait solver sometimes depends on regions being identical. As a concrete example
267    /// the trait solver ignores other candidates if one candidate exists without any constraints.
268    /// The goal `&'a u32: Equals<&'a u32>` has no constraints right now. If we replace each
269    /// occurrence of `'a` with a unique region the goal now equates these regions. See
270    /// the tests in trait-system-refactor-initiative#27 for concrete examples.
271    ///
272    /// We handle this by *uniquifying* region when canonicalizing root goals during HIR typeck.
273    /// This is still insufficient as inference variables may *hide* region variables, so e.g.
274    /// `dyn TwoSuper<?x, ?x>: Super<?x>` may hold but MIR typeck could end up having to prove
275    /// `dyn TwoSuper<&'0 (), &'1 ()>: Super<&'2 ()>` which is now ambiguous. Because of this we
276    /// stash all successfully proven goals which reference inference variables and then reprove
277    /// them after writeback.
278    pub in_hir_typeck: bool,
279
280    /// If set, this flag causes us to skip the 'leak check' during
281    /// higher-ranked subtyping operations. This flag is a temporary one used
282    /// to manage the removal of the leak-check: for the time being, we still run the
283    /// leak-check, but we issue warnings.
284    skip_leak_check: bool,
285
286    pub inner: RefCell<InferCtxtInner<'tcx>>,
287
288    /// Once region inference is done, the values for each variable.
289    lexical_region_resolutions: RefCell<Option<LexicalRegionResolutions<'tcx>>>,
290
291    /// Caches the results of trait selection. This cache is used
292    /// for things that depends on inference variables or placeholders.
293    pub selection_cache: select::SelectionCache<'tcx, ty::ParamEnv<'tcx>>,
294
295    /// Caches the results of trait evaluation. This cache is used
296    /// for things that depends on inference variables or placeholders.
297    pub evaluation_cache: select::EvaluationCache<'tcx, ty::ParamEnv<'tcx>>,
298
299    /// The set of predicates on which errors have been reported, to
300    /// avoid reporting the same error twice.
301    pub reported_trait_errors:
302        RefCell<FxIndexMap<Span, (Vec<Goal<'tcx, ty::Predicate<'tcx>>>, ErrorGuaranteed)>>,
303
304    pub reported_signature_mismatch: RefCell<FxHashSet<(Span, Option<Span>)>>,
305
306    /// When an error occurs, we want to avoid reporting "derived"
307    /// errors that are due to this original failure. We have this
308    /// flag that one can set whenever one creates a type-error that
309    /// is due to an error in a prior pass.
310    ///
311    /// Don't read this flag directly, call `is_tainted_by_errors()`
312    /// and `set_tainted_by_errors()`.
313    tainted_by_errors: Cell<Option<ErrorGuaranteed>>,
314
315    /// What is the innermost universe we have created? Starts out as
316    /// `UniverseIndex::root()` but grows from there as we enter
317    /// universal quantifiers.
318    ///
319    /// N.B., at present, we exclude the universal quantifiers on the
320    /// item we are type-checking, and just consider those names as
321    /// part of the root universe. So this would only get incremented
322    /// when we enter into a higher-ranked (`for<..>`) type or trait
323    /// bound.
324    universe: Cell<ty::UniverseIndex>,
325
326    next_trait_solver: bool,
327
328    pub obligation_inspector: Cell<Option<ObligationInspector<'tcx>>>,
329}
330
331/// See the `error_reporting` module for more details.
332#[derive(Clone, Copy, Debug, PartialEq, Eq, TypeFoldable, TypeVisitable)]
333pub enum ValuePairs<'tcx> {
334    Regions(ExpectedFound<ty::Region<'tcx>>),
335    Terms(ExpectedFound<ty::Term<'tcx>>),
336    Aliases(ExpectedFound<ty::AliasTerm<'tcx>>),
337    TraitRefs(ExpectedFound<ty::TraitRef<'tcx>>),
338    PolySigs(ExpectedFound<ty::PolyFnSig<'tcx>>),
339    ExistentialTraitRef(ExpectedFound<ty::PolyExistentialTraitRef<'tcx>>),
340    ExistentialProjection(ExpectedFound<ty::PolyExistentialProjection<'tcx>>),
341}
342
343impl<'tcx> ValuePairs<'tcx> {
344    pub fn ty(&self) -> Option<(Ty<'tcx>, Ty<'tcx>)> {
345        if let ValuePairs::Terms(ExpectedFound { expected, found }) = self
346            && let Some(expected) = expected.as_type()
347            && let Some(found) = found.as_type()
348        {
349            Some((expected, found))
350        } else {
351            None
352        }
353    }
354}
355
356/// The trace designates the path through inference that we took to
357/// encounter an error or subtyping constraint.
358///
359/// See the `error_reporting` module for more details.
360#[derive(Clone, Debug)]
361pub struct TypeTrace<'tcx> {
362    pub cause: ObligationCause<'tcx>,
363    pub values: ValuePairs<'tcx>,
364}
365
366/// The origin of a `r1 <= r2` constraint.
367///
368/// See `error_reporting` module for more details
369#[derive(Clone, Debug)]
370pub enum SubregionOrigin<'tcx> {
371    /// Arose from a subtyping relation
372    Subtype(Box<TypeTrace<'tcx>>),
373
374    /// When casting `&'a T` to an `&'b Trait` object,
375    /// relating `'a` to `'b`.
376    RelateObjectBound(Span),
377
378    /// Some type parameter was instantiated with the given type,
379    /// and that type must outlive some region.
380    RelateParamBound(Span, Ty<'tcx>, Option<Span>),
381
382    /// The given region parameter was instantiated with a region
383    /// that must outlive some other region.
384    RelateRegionParamBound(Span, Option<Ty<'tcx>>),
385
386    /// Creating a pointer `b` to contents of another reference.
387    Reborrow(Span),
388
389    /// (&'a &'b T) where a >= b
390    ReferenceOutlivesReferent(Ty<'tcx>, Span),
391
392    /// Comparing the signature and requirements of an impl method against
393    /// the containing trait.
394    CompareImplItemObligation {
395        span: Span,
396        impl_item_def_id: LocalDefId,
397        trait_item_def_id: DefId,
398    },
399
400    /// Checking that the bounds of a trait's associated type hold for a given impl.
401    CheckAssociatedTypeBounds {
402        parent: Box<SubregionOrigin<'tcx>>,
403        impl_item_def_id: LocalDefId,
404        trait_item_def_id: DefId,
405    },
406
407    AscribeUserTypeProvePredicate(Span),
408}
409
410// `SubregionOrigin` is used a lot. Make sure it doesn't unintentionally get bigger.
411#[cfg(target_pointer_width = "64")]
412rustc_data_structures::static_assert_size!(SubregionOrigin<'_>, 32);
413
414impl<'tcx> SubregionOrigin<'tcx> {
415    pub fn to_constraint_category(&self) -> ConstraintCategory<'tcx> {
416        match self {
417            Self::Subtype(type_trace) => type_trace.cause.to_constraint_category(),
418            Self::AscribeUserTypeProvePredicate(span) => ConstraintCategory::Predicate(*span),
419            _ => ConstraintCategory::BoringNoLocation,
420        }
421    }
422}
423
424/// Times when we replace bound regions with existentials:
425#[derive(Clone, Copy, Debug)]
426pub enum BoundRegionConversionTime {
427    /// when a fn is called
428    FnCall,
429
430    /// when two higher-ranked types are compared
431    HigherRankedType,
432
433    /// when projecting an associated type
434    AssocTypeProjection(DefId),
435}
436
437/// Reasons to create a region inference variable.
438///
439/// See `error_reporting` module for more details.
440#[derive(Copy, Clone, Debug)]
441pub enum RegionVariableOrigin {
442    /// Region variables created for ill-categorized reasons.
443    ///
444    /// They mostly indicate places in need of refactoring.
445    Misc(Span),
446
447    /// Regions created by a `&P` or `[...]` pattern.
448    PatternRegion(Span),
449
450    /// Regions created by `&` operator.
451    BorrowRegion(Span),
452
453    /// Regions created as part of an autoref of a method receiver.
454    Autoref(Span),
455
456    /// Regions created as part of an automatic coercion.
457    Coercion(Span),
458
459    /// Region variables created as the values for early-bound regions.
460    ///
461    /// FIXME(@lcnr): This should also store a `DefId`, similar to
462    /// `TypeVariableOrigin`.
463    RegionParameterDefinition(Span, Symbol),
464
465    /// Region variables created when instantiating a binder with
466    /// existential variables, e.g. when calling a function or method.
467    BoundRegion(Span, ty::BoundRegionKind, BoundRegionConversionTime),
468
469    UpvarRegion(ty::UpvarId, Span),
470
471    /// This origin is used for the inference variables that we create
472    /// during NLL region processing.
473    Nll(NllRegionVariableOrigin),
474}
475
476#[derive(Copy, Clone, Debug)]
477pub enum NllRegionVariableOrigin {
478    /// During NLL region processing, we create variables for free
479    /// regions that we encounter in the function signature and
480    /// elsewhere. This origin indices we've got one of those.
481    FreeRegion,
482
483    /// "Universal" instantiation of a higher-ranked region (e.g.,
484    /// from a `for<'a> T` binder). Meant to represent "any region".
485    Placeholder(ty::PlaceholderRegion),
486
487    Existential {
488        name: Option<Symbol>,
489    },
490}
491
492#[derive(Copy, Clone, Debug)]
493pub struct FixupError {
494    unresolved: TyOrConstInferVar,
495}
496
497impl fmt::Display for FixupError {
498    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
499        match self.unresolved {
500            TyOrConstInferVar::TyInt(_) => write!(
501                f,
502                "cannot determine the type of this integer; \
503                 add a suffix to specify the type explicitly"
504            ),
505            TyOrConstInferVar::TyFloat(_) => write!(
506                f,
507                "cannot determine the type of this number; \
508                 add a suffix to specify the type explicitly"
509            ),
510            TyOrConstInferVar::Ty(_) => write!(f, "unconstrained type"),
511            TyOrConstInferVar::Const(_) => write!(f, "unconstrained const value"),
512        }
513    }
514}
515
516/// See the `region_obligations` field for more information.
517#[derive(Clone, Debug)]
518pub struct TypeOutlivesConstraint<'tcx> {
519    pub sub_region: ty::Region<'tcx>,
520    pub sup_type: Ty<'tcx>,
521    pub origin: SubregionOrigin<'tcx>,
522}
523
524/// Used to configure inference contexts before their creation.
525pub struct InferCtxtBuilder<'tcx> {
526    tcx: TyCtxt<'tcx>,
527    considering_regions: bool,
528    in_hir_typeck: bool,
529    skip_leak_check: bool,
530    /// Whether we should use the new trait solver in the local inference context,
531    /// which affects things like which solver is used in `predicate_may_hold`.
532    next_trait_solver: bool,
533}
534
535#[extension(pub trait TyCtxtInferExt<'tcx>)]
536impl<'tcx> TyCtxt<'tcx> {
537    fn infer_ctxt(self) -> InferCtxtBuilder<'tcx> {
538        InferCtxtBuilder {
539            tcx: self,
540            considering_regions: true,
541            in_hir_typeck: false,
542            skip_leak_check: false,
543            next_trait_solver: self.next_trait_solver_globally(),
544        }
545    }
546}
547
548impl<'tcx> InferCtxtBuilder<'tcx> {
549    pub fn with_next_trait_solver(mut self, next_trait_solver: bool) -> Self {
550        self.next_trait_solver = next_trait_solver;
551        self
552    }
553
554    pub fn ignoring_regions(mut self) -> Self {
555        self.considering_regions = false;
556        self
557    }
558
559    pub fn in_hir_typeck(mut self) -> Self {
560        self.in_hir_typeck = true;
561        self
562    }
563
564    pub fn skip_leak_check(mut self, skip_leak_check: bool) -> Self {
565        self.skip_leak_check = skip_leak_check;
566        self
567    }
568
569    /// Given a canonical value `C` as a starting point, create an
570    /// inference context that contains each of the bound values
571    /// within instantiated as a fresh variable. The `f` closure is
572    /// invoked with the new infcx, along with the instantiated value
573    /// `V` and a instantiation `S`. This instantiation `S` maps from
574    /// the bound values in `C` to their instantiated values in `V`
575    /// (in other words, `S(C) = V`).
576    pub fn build_with_canonical<T>(
577        mut self,
578        span: Span,
579        input: &CanonicalQueryInput<'tcx, T>,
580    ) -> (InferCtxt<'tcx>, T, CanonicalVarValues<'tcx>)
581    where
582        T: TypeFoldable<TyCtxt<'tcx>>,
583    {
584        let infcx = self.build(input.typing_mode);
585        let (value, args) = infcx.instantiate_canonical(span, &input.canonical);
586        (infcx, value, args)
587    }
588
589    pub fn build_with_typing_env(
590        mut self,
591        TypingEnv { typing_mode, param_env }: TypingEnv<'tcx>,
592    ) -> (InferCtxt<'tcx>, ty::ParamEnv<'tcx>) {
593        (self.build(typing_mode), param_env)
594    }
595
596    pub fn build(&mut self, typing_mode: TypingMode<'tcx>) -> InferCtxt<'tcx> {
597        let InferCtxtBuilder {
598            tcx,
599            considering_regions,
600            in_hir_typeck,
601            skip_leak_check,
602            next_trait_solver,
603        } = *self;
604        InferCtxt {
605            tcx,
606            typing_mode,
607            considering_regions,
608            in_hir_typeck,
609            skip_leak_check,
610            inner: RefCell::new(InferCtxtInner::new()),
611            lexical_region_resolutions: RefCell::new(None),
612            selection_cache: Default::default(),
613            evaluation_cache: Default::default(),
614            reported_trait_errors: Default::default(),
615            reported_signature_mismatch: Default::default(),
616            tainted_by_errors: Cell::new(None),
617            universe: Cell::new(ty::UniverseIndex::ROOT),
618            next_trait_solver,
619            obligation_inspector: Cell::new(None),
620        }
621    }
622}
623
624impl<'tcx, T> InferOk<'tcx, T> {
625    /// Extracts `value`, registering any obligations into `fulfill_cx`.
626    pub fn into_value_registering_obligations<E: 'tcx>(
627        self,
628        infcx: &InferCtxt<'tcx>,
629        fulfill_cx: &mut dyn TraitEngine<'tcx, E>,
630    ) -> T {
631        let InferOk { value, obligations } = self;
632        fulfill_cx.register_predicate_obligations(infcx, obligations);
633        value
634    }
635}
636
637impl<'tcx> InferOk<'tcx, ()> {
638    pub fn into_obligations(self) -> PredicateObligations<'tcx> {
639        self.obligations
640    }
641}
642
643impl<'tcx> InferCtxt<'tcx> {
644    pub fn dcx(&self) -> DiagCtxtHandle<'_> {
645        self.tcx.dcx().taintable_handle(&self.tainted_by_errors)
646    }
647
648    pub fn next_trait_solver(&self) -> bool {
649        self.next_trait_solver
650    }
651
652    #[inline(always)]
653    pub fn typing_mode(&self) -> TypingMode<'tcx> {
654        self.typing_mode
655    }
656
657    pub fn freshen<T: TypeFoldable<TyCtxt<'tcx>>>(&self, t: T) -> T {
658        t.fold_with(&mut self.freshener())
659    }
660
661    /// Returns the origin of the type variable identified by `vid`.
662    ///
663    /// No attempt is made to resolve `vid` to its root variable.
664    pub fn type_var_origin(&self, vid: TyVid) -> TypeVariableOrigin {
665        self.inner.borrow_mut().type_variables().var_origin(vid)
666    }
667
668    /// Returns the origin of the const variable identified by `vid`
669    // FIXME: We should store origins separately from the unification table
670    // so this doesn't need to be optional.
671    pub fn const_var_origin(&self, vid: ConstVid) -> Option<ConstVariableOrigin> {
672        match self.inner.borrow_mut().const_unification_table().probe_value(vid) {
673            ConstVariableValue::Known { .. } => None,
674            ConstVariableValue::Unknown { origin, .. } => Some(origin),
675        }
676    }
677
678    pub fn freshener<'b>(&'b self) -> TypeFreshener<'b, 'tcx> {
679        freshen::TypeFreshener::new(self)
680    }
681
682    pub fn unresolved_variables(&self) -> Vec<Ty<'tcx>> {
683        let mut inner = self.inner.borrow_mut();
684        let mut vars: Vec<Ty<'_>> = inner
685            .type_variables()
686            .unresolved_variables()
687            .into_iter()
688            .map(|t| Ty::new_var(self.tcx, t))
689            .collect();
690        vars.extend(
691            (0..inner.int_unification_table().len())
692                .map(|i| ty::IntVid::from_usize(i))
693                .filter(|&vid| inner.int_unification_table().probe_value(vid).is_unknown())
694                .map(|v| Ty::new_int_var(self.tcx, v)),
695        );
696        vars.extend(
697            (0..inner.float_unification_table().len())
698                .map(|i| ty::FloatVid::from_usize(i))
699                .filter(|&vid| inner.float_unification_table().probe_value(vid).is_unknown())
700                .map(|v| Ty::new_float_var(self.tcx, v)),
701        );
702        vars
703    }
704
705    #[instrument(skip(self), level = "debug")]
706    pub fn sub_regions(
707        &self,
708        origin: SubregionOrigin<'tcx>,
709        a: ty::Region<'tcx>,
710        b: ty::Region<'tcx>,
711    ) {
712        self.inner.borrow_mut().unwrap_region_constraints().make_subregion(origin, a, b);
713    }
714
715    /// Processes a `Coerce` predicate from the fulfillment context.
716    /// This is NOT the preferred way to handle coercion, which is to
717    /// invoke `FnCtxt::coerce` or a similar method (see `coercion.rs`).
718    ///
719    /// This method here is actually a fallback that winds up being
720    /// invoked when `FnCtxt::coerce` encounters unresolved type variables
721    /// and records a coercion predicate. Presently, this method is equivalent
722    /// to `subtype_predicate` -- that is, "coercing" `a` to `b` winds up
723    /// actually requiring `a <: b`. This is of course a valid coercion,
724    /// but it's not as flexible as `FnCtxt::coerce` would be.
725    ///
726    /// (We may refactor this in the future, but there are a number of
727    /// practical obstacles. Among other things, `FnCtxt::coerce` presently
728    /// records adjustments that are required on the HIR in order to perform
729    /// the coercion, and we don't currently have a way to manage that.)
730    pub fn coerce_predicate(
731        &self,
732        cause: &ObligationCause<'tcx>,
733        param_env: ty::ParamEnv<'tcx>,
734        predicate: ty::PolyCoercePredicate<'tcx>,
735    ) -> Result<InferResult<'tcx, ()>, (TyVid, TyVid)> {
736        let subtype_predicate = predicate.map_bound(|p| ty::SubtypePredicate {
737            a_is_expected: false, // when coercing from `a` to `b`, `b` is expected
738            a: p.a,
739            b: p.b,
740        });
741        self.subtype_predicate(cause, param_env, subtype_predicate)
742    }
743
744    pub fn subtype_predicate(
745        &self,
746        cause: &ObligationCause<'tcx>,
747        param_env: ty::ParamEnv<'tcx>,
748        predicate: ty::PolySubtypePredicate<'tcx>,
749    ) -> Result<InferResult<'tcx, ()>, (TyVid, TyVid)> {
750        // Check for two unresolved inference variables, in which case we can
751        // make no progress. This is partly a micro-optimization, but it's
752        // also an opportunity to "sub-unify" the variables. This isn't
753        // *necessary* to prevent cycles, because they would eventually be sub-unified
754        // anyhow during generalization, but it helps with diagnostics (we can detect
755        // earlier that they are sub-unified).
756        //
757        // Note that we can just skip the binders here because
758        // type variables can't (at present, at
759        // least) capture any of the things bound by this binder.
760        //
761        // Note that this sub here is not just for diagnostics - it has semantic
762        // effects as well.
763        let r_a = self.shallow_resolve(predicate.skip_binder().a);
764        let r_b = self.shallow_resolve(predicate.skip_binder().b);
765        match (r_a.kind(), r_b.kind()) {
766            (&ty::Infer(ty::TyVar(a_vid)), &ty::Infer(ty::TyVar(b_vid))) => {
767                self.sub_unify_ty_vids_raw(a_vid, b_vid);
768                return Err((a_vid, b_vid));
769            }
770            _ => {}
771        }
772
773        self.enter_forall(predicate, |ty::SubtypePredicate { a_is_expected, a, b }| {
774            if a_is_expected {
775                Ok(self.at(cause, param_env).sub(DefineOpaqueTypes::Yes, a, b))
776            } else {
777                Ok(self.at(cause, param_env).sup(DefineOpaqueTypes::Yes, b, a))
778            }
779        })
780    }
781
782    /// Number of type variables created so far.
783    pub fn num_ty_vars(&self) -> usize {
784        self.inner.borrow_mut().type_variables().num_vars()
785    }
786
787    pub fn next_ty_vid(&self, span: Span) -> TyVid {
788        self.next_ty_vid_with_origin(TypeVariableOrigin { span, param_def_id: None })
789    }
790
791    pub fn next_ty_vid_with_origin(&self, origin: TypeVariableOrigin) -> TyVid {
792        self.inner.borrow_mut().type_variables().new_var(self.universe(), origin)
793    }
794
795    pub fn next_ty_vid_in_universe(&self, span: Span, universe: ty::UniverseIndex) -> TyVid {
796        let origin = TypeVariableOrigin { span, param_def_id: None };
797        self.inner.borrow_mut().type_variables().new_var(universe, origin)
798    }
799
800    pub fn next_ty_var(&self, span: Span) -> Ty<'tcx> {
801        self.next_ty_var_with_origin(TypeVariableOrigin { span, param_def_id: None })
802    }
803
804    pub fn next_ty_var_with_origin(&self, origin: TypeVariableOrigin) -> Ty<'tcx> {
805        let vid = self.next_ty_vid_with_origin(origin);
806        Ty::new_var(self.tcx, vid)
807    }
808
809    pub fn next_ty_var_in_universe(&self, span: Span, universe: ty::UniverseIndex) -> Ty<'tcx> {
810        let vid = self.next_ty_vid_in_universe(span, universe);
811        Ty::new_var(self.tcx, vid)
812    }
813
814    pub fn next_const_var(&self, span: Span) -> ty::Const<'tcx> {
815        self.next_const_var_with_origin(ConstVariableOrigin { span, param_def_id: None })
816    }
817
818    pub fn next_const_var_with_origin(&self, origin: ConstVariableOrigin) -> ty::Const<'tcx> {
819        let vid = self
820            .inner
821            .borrow_mut()
822            .const_unification_table()
823            .new_key(ConstVariableValue::Unknown { origin, universe: self.universe() })
824            .vid;
825        ty::Const::new_var(self.tcx, vid)
826    }
827
828    pub fn next_const_var_in_universe(
829        &self,
830        span: Span,
831        universe: ty::UniverseIndex,
832    ) -> ty::Const<'tcx> {
833        let origin = ConstVariableOrigin { span, param_def_id: None };
834        let vid = self
835            .inner
836            .borrow_mut()
837            .const_unification_table()
838            .new_key(ConstVariableValue::Unknown { origin, universe })
839            .vid;
840        ty::Const::new_var(self.tcx, vid)
841    }
842
843    pub fn next_int_var(&self) -> Ty<'tcx> {
844        let next_int_var_id =
845            self.inner.borrow_mut().int_unification_table().new_key(ty::IntVarValue::Unknown);
846        Ty::new_int_var(self.tcx, next_int_var_id)
847    }
848
849    pub fn next_float_var(&self) -> Ty<'tcx> {
850        let next_float_var_id =
851            self.inner.borrow_mut().float_unification_table().new_key(ty::FloatVarValue::Unknown);
852        Ty::new_float_var(self.tcx, next_float_var_id)
853    }
854
855    /// Creates a fresh region variable with the next available index.
856    /// The variable will be created in the maximum universe created
857    /// thus far, allowing it to name any region created thus far.
858    pub fn next_region_var(&self, origin: RegionVariableOrigin) -> ty::Region<'tcx> {
859        self.next_region_var_in_universe(origin, self.universe())
860    }
861
862    /// Creates a fresh region variable with the next available index
863    /// in the given universe; typically, you can use
864    /// `next_region_var` and just use the maximal universe.
865    pub fn next_region_var_in_universe(
866        &self,
867        origin: RegionVariableOrigin,
868        universe: ty::UniverseIndex,
869    ) -> ty::Region<'tcx> {
870        let region_var =
871            self.inner.borrow_mut().unwrap_region_constraints().new_region_var(universe, origin);
872        ty::Region::new_var(self.tcx, region_var)
873    }
874
875    pub fn next_term_var_of_kind(&self, term: ty::Term<'tcx>, span: Span) -> ty::Term<'tcx> {
876        match term.kind() {
877            ty::TermKind::Ty(_) => self.next_ty_var(span).into(),
878            ty::TermKind::Const(_) => self.next_const_var(span).into(),
879        }
880    }
881
882    /// Return the universe that the region `r` was created in. For
883    /// most regions (e.g., `'static`, named regions from the user,
884    /// etc) this is the root universe U0. For inference variables or
885    /// placeholders, however, it will return the universe which they
886    /// are associated.
887    pub fn universe_of_region(&self, r: ty::Region<'tcx>) -> ty::UniverseIndex {
888        self.inner.borrow_mut().unwrap_region_constraints().universe(r)
889    }
890
891    /// Number of region variables created so far.
892    pub fn num_region_vars(&self) -> usize {
893        self.inner.borrow_mut().unwrap_region_constraints().num_region_vars()
894    }
895
896    /// Just a convenient wrapper of `next_region_var` for using during NLL.
897    #[instrument(skip(self), level = "debug")]
898    pub fn next_nll_region_var(&self, origin: NllRegionVariableOrigin) -> ty::Region<'tcx> {
899        self.next_region_var(RegionVariableOrigin::Nll(origin))
900    }
901
902    /// Just a convenient wrapper of `next_region_var` for using during NLL.
903    #[instrument(skip(self), level = "debug")]
904    pub fn next_nll_region_var_in_universe(
905        &self,
906        origin: NllRegionVariableOrigin,
907        universe: ty::UniverseIndex,
908    ) -> ty::Region<'tcx> {
909        self.next_region_var_in_universe(RegionVariableOrigin::Nll(origin), universe)
910    }
911
912    pub fn var_for_def(&self, span: Span, param: &ty::GenericParamDef) -> GenericArg<'tcx> {
913        match param.kind {
914            GenericParamDefKind::Lifetime => {
915                // Create a region inference variable for the given
916                // region parameter definition.
917                self.next_region_var(RegionVariableOrigin::RegionParameterDefinition(
918                    span, param.name,
919                ))
920                .into()
921            }
922            GenericParamDefKind::Type { .. } => {
923                // Create a type inference variable for the given
924                // type parameter definition. The generic parameters are
925                // for actual parameters that may be referred to by
926                // the default of this type parameter, if it exists.
927                // e.g., `struct Foo<A, B, C = (A, B)>(...);` when
928                // used in a path such as `Foo::<T, U>::new()` will
929                // use an inference variable for `C` with `[T, U]`
930                // as the generic parameters for the default, `(T, U)`.
931                let ty_var_id = self.inner.borrow_mut().type_variables().new_var(
932                    self.universe(),
933                    TypeVariableOrigin { param_def_id: Some(param.def_id), span },
934                );
935
936                Ty::new_var(self.tcx, ty_var_id).into()
937            }
938            GenericParamDefKind::Const { .. } => {
939                let origin = ConstVariableOrigin { param_def_id: Some(param.def_id), span };
940                let const_var_id = self
941                    .inner
942                    .borrow_mut()
943                    .const_unification_table()
944                    .new_key(ConstVariableValue::Unknown { origin, universe: self.universe() })
945                    .vid;
946                ty::Const::new_var(self.tcx, const_var_id).into()
947            }
948        }
949    }
950
951    /// Given a set of generics defined on a type or impl, returns the generic parameters mapping
952    /// each type/region parameter to a fresh inference variable.
953    pub fn fresh_args_for_item(&self, span: Span, def_id: DefId) -> GenericArgsRef<'tcx> {
954        GenericArgs::for_item(self.tcx, def_id, |param, _| self.var_for_def(span, param))
955    }
956
957    /// Returns `true` if errors have been reported since this infcx was
958    /// created. This is sometimes used as a heuristic to skip
959    /// reporting errors that often occur as a result of earlier
960    /// errors, but where it's hard to be 100% sure (e.g., unresolved
961    /// inference variables, regionck errors).
962    #[must_use = "this method does not have any side effects"]
963    pub fn tainted_by_errors(&self) -> Option<ErrorGuaranteed> {
964        self.tainted_by_errors.get()
965    }
966
967    /// Set the "tainted by errors" flag to true. We call this when we
968    /// observe an error from a prior pass.
969    pub fn set_tainted_by_errors(&self, e: ErrorGuaranteed) {
970        debug!("set_tainted_by_errors(ErrorGuaranteed)");
971        self.tainted_by_errors.set(Some(e));
972    }
973
974    pub fn region_var_origin(&self, vid: ty::RegionVid) -> RegionVariableOrigin {
975        let mut inner = self.inner.borrow_mut();
976        let inner = &mut *inner;
977        inner.unwrap_region_constraints().var_origin(vid)
978    }
979
980    /// Clone the list of variable regions. This is used only during NLL processing
981    /// to put the set of region variables into the NLL region context.
982    pub fn get_region_var_infos(&self) -> VarInfos {
983        let inner = self.inner.borrow();
984        assert!(!UndoLogs::<UndoLog<'_>>::in_snapshot(&inner.undo_log));
985        let storage = inner.region_constraint_storage.as_ref().expect("regions already resolved");
986        assert!(storage.data.is_empty(), "{:#?}", storage.data);
987        // We clone instead of taking because borrowck still wants to use the
988        // inference context after calling this for diagnostics and the new
989        // trait solver.
990        storage.var_infos.clone()
991    }
992
993    pub fn has_opaque_types_in_storage(&self) -> bool {
994        !self.inner.borrow().opaque_type_storage.is_empty()
995    }
996
997    #[instrument(level = "debug", skip(self), ret)]
998    pub fn take_opaque_types(&self) -> Vec<(OpaqueTypeKey<'tcx>, OpaqueHiddenType<'tcx>)> {
999        self.inner.borrow_mut().opaque_type_storage.take_opaque_types().collect()
1000    }
1001
1002    #[instrument(level = "debug", skip(self), ret)]
1003    pub fn clone_opaque_types(&self) -> Vec<(OpaqueTypeKey<'tcx>, OpaqueHiddenType<'tcx>)> {
1004        self.inner.borrow_mut().opaque_type_storage.iter_opaque_types().collect()
1005    }
1006
1007    #[inline(always)]
1008    pub fn can_define_opaque_ty(&self, id: impl Into<DefId>) -> bool {
1009        debug_assert!(!self.next_trait_solver());
1010        match self.typing_mode() {
1011            TypingMode::Analysis {
1012                defining_opaque_types_and_generators: defining_opaque_types,
1013            }
1014            | TypingMode::Borrowck { defining_opaque_types } => {
1015                id.into().as_local().is_some_and(|def_id| defining_opaque_types.contains(&def_id))
1016            }
1017            // FIXME(#132279): This function is quite weird in post-analysis
1018            // and post-borrowck analysis mode. We may need to modify its uses
1019            // to support PostBorrowckAnalysis in the old solver as well.
1020            TypingMode::Coherence
1021            | TypingMode::PostBorrowckAnalysis { .. }
1022            | TypingMode::PostAnalysis => false,
1023        }
1024    }
1025
1026    pub fn push_hir_typeck_potentially_region_dependent_goal(
1027        &self,
1028        goal: PredicateObligation<'tcx>,
1029    ) {
1030        let mut inner = self.inner.borrow_mut();
1031        inner.undo_log.push(UndoLog::PushHirTypeckPotentiallyRegionDependentGoal);
1032        inner.hir_typeck_potentially_region_dependent_goals.push(goal);
1033    }
1034
1035    pub fn take_hir_typeck_potentially_region_dependent_goals(
1036        &self,
1037    ) -> Vec<PredicateObligation<'tcx>> {
1038        assert!(!self.in_snapshot(), "cannot take goals in a snapshot");
1039        std::mem::take(&mut self.inner.borrow_mut().hir_typeck_potentially_region_dependent_goals)
1040    }
1041
1042    pub fn ty_to_string(&self, t: Ty<'tcx>) -> String {
1043        self.resolve_vars_if_possible(t).to_string()
1044    }
1045
1046    /// If `TyVar(vid)` resolves to a type, return that type. Else, return the
1047    /// universe index of `TyVar(vid)`.
1048    pub fn probe_ty_var(&self, vid: TyVid) -> Result<Ty<'tcx>, ty::UniverseIndex> {
1049        use self::type_variable::TypeVariableValue;
1050
1051        match self.inner.borrow_mut().type_variables().probe(vid) {
1052            TypeVariableValue::Known { value } => Ok(value),
1053            TypeVariableValue::Unknown { universe } => Err(universe),
1054        }
1055    }
1056
1057    pub fn shallow_resolve(&self, ty: Ty<'tcx>) -> Ty<'tcx> {
1058        if let ty::Infer(v) = *ty.kind() {
1059            match v {
1060                ty::TyVar(v) => {
1061                    // Not entirely obvious: if `typ` is a type variable,
1062                    // it can be resolved to an int/float variable, which
1063                    // can then be recursively resolved, hence the
1064                    // recursion. Note though that we prevent type
1065                    // variables from unifying to other type variables
1066                    // directly (though they may be embedded
1067                    // structurally), and we prevent cycles in any case,
1068                    // so this recursion should always be of very limited
1069                    // depth.
1070                    //
1071                    // Note: if these two lines are combined into one we get
1072                    // dynamic borrow errors on `self.inner`.
1073                    let known = self.inner.borrow_mut().type_variables().probe(v).known();
1074                    known.map_or(ty, |t| self.shallow_resolve(t))
1075                }
1076
1077                ty::IntVar(v) => {
1078                    match self.inner.borrow_mut().int_unification_table().probe_value(v) {
1079                        ty::IntVarValue::IntType(ty) => Ty::new_int(self.tcx, ty),
1080                        ty::IntVarValue::UintType(ty) => Ty::new_uint(self.tcx, ty),
1081                        ty::IntVarValue::Unknown => ty,
1082                    }
1083                }
1084
1085                ty::FloatVar(v) => {
1086                    match self.inner.borrow_mut().float_unification_table().probe_value(v) {
1087                        ty::FloatVarValue::Known(ty) => Ty::new_float(self.tcx, ty),
1088                        ty::FloatVarValue::Unknown => ty,
1089                    }
1090                }
1091
1092                ty::FreshTy(_) | ty::FreshIntTy(_) | ty::FreshFloatTy(_) => ty,
1093            }
1094        } else {
1095            ty
1096        }
1097    }
1098
1099    pub fn shallow_resolve_const(&self, ct: ty::Const<'tcx>) -> ty::Const<'tcx> {
1100        match ct.kind() {
1101            ty::ConstKind::Infer(infer_ct) => match infer_ct {
1102                InferConst::Var(vid) => self
1103                    .inner
1104                    .borrow_mut()
1105                    .const_unification_table()
1106                    .probe_value(vid)
1107                    .known()
1108                    .unwrap_or(ct),
1109                InferConst::Fresh(_) => ct,
1110            },
1111            ty::ConstKind::Param(_)
1112            | ty::ConstKind::Bound(_, _)
1113            | ty::ConstKind::Placeholder(_)
1114            | ty::ConstKind::Unevaluated(_)
1115            | ty::ConstKind::Value(_)
1116            | ty::ConstKind::Error(_)
1117            | ty::ConstKind::Expr(_) => ct,
1118        }
1119    }
1120
1121    pub fn shallow_resolve_term(&self, term: ty::Term<'tcx>) -> ty::Term<'tcx> {
1122        match term.kind() {
1123            ty::TermKind::Ty(ty) => self.shallow_resolve(ty).into(),
1124            ty::TermKind::Const(ct) => self.shallow_resolve_const(ct).into(),
1125        }
1126    }
1127
1128    pub fn root_var(&self, var: ty::TyVid) -> ty::TyVid {
1129        self.inner.borrow_mut().type_variables().root_var(var)
1130    }
1131
1132    pub fn sub_unify_ty_vids_raw(&self, a: ty::TyVid, b: ty::TyVid) {
1133        self.inner.borrow_mut().type_variables().sub_unify(a, b);
1134    }
1135
1136    pub fn sub_unification_table_root_var(&self, var: ty::TyVid) -> ty::TyVid {
1137        self.inner.borrow_mut().type_variables().sub_unification_table_root_var(var)
1138    }
1139
1140    pub fn root_const_var(&self, var: ty::ConstVid) -> ty::ConstVid {
1141        self.inner.borrow_mut().const_unification_table().find(var).vid
1142    }
1143
1144    /// Resolves an int var to a rigid int type, if it was constrained to one,
1145    /// or else the root int var in the unification table.
1146    pub fn opportunistic_resolve_int_var(&self, vid: ty::IntVid) -> Ty<'tcx> {
1147        let mut inner = self.inner.borrow_mut();
1148        let value = inner.int_unification_table().probe_value(vid);
1149        match value {
1150            ty::IntVarValue::IntType(ty) => Ty::new_int(self.tcx, ty),
1151            ty::IntVarValue::UintType(ty) => Ty::new_uint(self.tcx, ty),
1152            ty::IntVarValue::Unknown => {
1153                Ty::new_int_var(self.tcx, inner.int_unification_table().find(vid))
1154            }
1155        }
1156    }
1157
1158    /// Resolves a float var to a rigid int type, if it was constrained to one,
1159    /// or else the root float var in the unification table.
1160    pub fn opportunistic_resolve_float_var(&self, vid: ty::FloatVid) -> Ty<'tcx> {
1161        let mut inner = self.inner.borrow_mut();
1162        let value = inner.float_unification_table().probe_value(vid);
1163        match value {
1164            ty::FloatVarValue::Known(ty) => Ty::new_float(self.tcx, ty),
1165            ty::FloatVarValue::Unknown => {
1166                Ty::new_float_var(self.tcx, inner.float_unification_table().find(vid))
1167            }
1168        }
1169    }
1170
1171    /// Where possible, replaces type/const variables in
1172    /// `value` with their final value. Note that region variables
1173    /// are unaffected. If a type/const variable has not been unified, it
1174    /// is left as is. This is an idempotent operation that does
1175    /// not affect inference state in any way and so you can do it
1176    /// at will.
1177    pub fn resolve_vars_if_possible<T>(&self, value: T) -> T
1178    where
1179        T: TypeFoldable<TyCtxt<'tcx>>,
1180    {
1181        if let Err(guar) = value.error_reported() {
1182            self.set_tainted_by_errors(guar);
1183        }
1184        if !value.has_non_region_infer() {
1185            return value;
1186        }
1187        let mut r = resolve::OpportunisticVarResolver::new(self);
1188        value.fold_with(&mut r)
1189    }
1190
1191    pub fn resolve_numeric_literals_with_default<T>(&self, value: T) -> T
1192    where
1193        T: TypeFoldable<TyCtxt<'tcx>>,
1194    {
1195        if !value.has_infer() {
1196            return value; // Avoid duplicated type-folding.
1197        }
1198        let mut r = InferenceLiteralEraser { tcx: self.tcx };
1199        value.fold_with(&mut r)
1200    }
1201
1202    pub fn probe_const_var(&self, vid: ty::ConstVid) -> Result<ty::Const<'tcx>, ty::UniverseIndex> {
1203        match self.inner.borrow_mut().const_unification_table().probe_value(vid) {
1204            ConstVariableValue::Known { value } => Ok(value),
1205            ConstVariableValue::Unknown { origin: _, universe } => Err(universe),
1206        }
1207    }
1208
1209    /// Attempts to resolve all type/region/const variables in
1210    /// `value`. Region inference must have been run already (e.g.,
1211    /// by calling `resolve_regions_and_report_errors`). If some
1212    /// variable was never unified, an `Err` results.
1213    ///
1214    /// This method is idempotent, but it not typically not invoked
1215    /// except during the writeback phase.
1216    pub fn fully_resolve<T: TypeFoldable<TyCtxt<'tcx>>>(&self, value: T) -> FixupResult<T> {
1217        match resolve::fully_resolve(self, value) {
1218            Ok(value) => {
1219                if value.has_non_region_infer() {
1220                    bug!("`{value:?}` is not fully resolved");
1221                }
1222                if value.has_infer_regions() {
1223                    let guar = self.dcx().delayed_bug(format!("`{value:?}` is not fully resolved"));
1224                    Ok(fold_regions(self.tcx, value, |re, _| {
1225                        if re.is_var() { ty::Region::new_error(self.tcx, guar) } else { re }
1226                    }))
1227                } else {
1228                    Ok(value)
1229                }
1230            }
1231            Err(e) => Err(e),
1232        }
1233    }
1234
1235    // Instantiates the bound variables in a given binder with fresh inference
1236    // variables in the current universe.
1237    //
1238    // Use this method if you'd like to find some generic parameters of the binder's
1239    // variables (e.g. during a method call). If there isn't a [`BoundRegionConversionTime`]
1240    // that corresponds to your use case, consider whether or not you should
1241    // use [`InferCtxt::enter_forall`] instead.
1242    pub fn instantiate_binder_with_fresh_vars<T>(
1243        &self,
1244        span: Span,
1245        lbrct: BoundRegionConversionTime,
1246        value: ty::Binder<'tcx, T>,
1247    ) -> T
1248    where
1249        T: TypeFoldable<TyCtxt<'tcx>> + Copy,
1250    {
1251        if let Some(inner) = value.no_bound_vars() {
1252            return inner;
1253        }
1254
1255        let bound_vars = value.bound_vars();
1256        let mut args = Vec::with_capacity(bound_vars.len());
1257
1258        for bound_var_kind in bound_vars {
1259            let arg: ty::GenericArg<'_> = match bound_var_kind {
1260                ty::BoundVariableKind::Ty(_) => self.next_ty_var(span).into(),
1261                ty::BoundVariableKind::Region(br) => {
1262                    self.next_region_var(RegionVariableOrigin::BoundRegion(span, br, lbrct)).into()
1263                }
1264                ty::BoundVariableKind::Const => self.next_const_var(span).into(),
1265            };
1266            args.push(arg);
1267        }
1268
1269        struct ToFreshVars<'tcx> {
1270            args: Vec<ty::GenericArg<'tcx>>,
1271        }
1272
1273        impl<'tcx> BoundVarReplacerDelegate<'tcx> for ToFreshVars<'tcx> {
1274            fn replace_region(&mut self, br: ty::BoundRegion) -> ty::Region<'tcx> {
1275                self.args[br.var.index()].expect_region()
1276            }
1277            fn replace_ty(&mut self, bt: ty::BoundTy) -> Ty<'tcx> {
1278                self.args[bt.var.index()].expect_ty()
1279            }
1280            fn replace_const(&mut self, bc: ty::BoundConst) -> ty::Const<'tcx> {
1281                self.args[bc.var.index()].expect_const()
1282            }
1283        }
1284        let delegate = ToFreshVars { args };
1285        self.tcx.replace_bound_vars_uncached(value, delegate)
1286    }
1287
1288    /// See the [`region_constraints::RegionConstraintCollector::verify_generic_bound`] method.
1289    pub(crate) fn verify_generic_bound(
1290        &self,
1291        origin: SubregionOrigin<'tcx>,
1292        kind: GenericKind<'tcx>,
1293        a: ty::Region<'tcx>,
1294        bound: VerifyBound<'tcx>,
1295    ) {
1296        debug!("verify_generic_bound({:?}, {:?} <: {:?})", kind, a, bound);
1297
1298        self.inner
1299            .borrow_mut()
1300            .unwrap_region_constraints()
1301            .verify_generic_bound(origin, kind, a, bound);
1302    }
1303
1304    /// Obtains the latest type of the given closure; this may be a
1305    /// closure in the current function, in which case its
1306    /// `ClosureKind` may not yet be known.
1307    pub fn closure_kind(&self, closure_ty: Ty<'tcx>) -> Option<ty::ClosureKind> {
1308        let unresolved_kind_ty = match *closure_ty.kind() {
1309            ty::Closure(_, args) => args.as_closure().kind_ty(),
1310            ty::CoroutineClosure(_, args) => args.as_coroutine_closure().kind_ty(),
1311            _ => bug!("unexpected type {closure_ty}"),
1312        };
1313        let closure_kind_ty = self.shallow_resolve(unresolved_kind_ty);
1314        closure_kind_ty.to_opt_closure_kind()
1315    }
1316
1317    pub fn universe(&self) -> ty::UniverseIndex {
1318        self.universe.get()
1319    }
1320
1321    /// Creates and return a fresh universe that extends all previous
1322    /// universes. Updates `self.universe` to that new universe.
1323    pub fn create_next_universe(&self) -> ty::UniverseIndex {
1324        let u = self.universe.get().next_universe();
1325        debug!("create_next_universe {u:?}");
1326        self.universe.set(u);
1327        u
1328    }
1329
1330    /// Extract [`ty::TypingMode`] of this inference context to get a `TypingEnv`
1331    /// which contains the necessary information to use the trait system without
1332    /// using canonicalization or carrying this inference context around.
1333    pub fn typing_env(&self, param_env: ty::ParamEnv<'tcx>) -> ty::TypingEnv<'tcx> {
1334        let typing_mode = match self.typing_mode() {
1335            // FIXME(#132279): This erases the `defining_opaque_types` as it isn't possible
1336            // to handle them without proper canonicalization. This means we may cause cycle
1337            // errors and fail to reveal opaques while inside of bodies. We should rename this
1338            // function and require explicit comments on all use-sites in the future.
1339            ty::TypingMode::Analysis { defining_opaque_types_and_generators: _ }
1340            | ty::TypingMode::Borrowck { defining_opaque_types: _ } => {
1341                TypingMode::non_body_analysis()
1342            }
1343            mode @ (ty::TypingMode::Coherence
1344            | ty::TypingMode::PostBorrowckAnalysis { .. }
1345            | ty::TypingMode::PostAnalysis) => mode,
1346        };
1347        ty::TypingEnv { typing_mode, param_env }
1348    }
1349
1350    /// Similar to [`Self::canonicalize_query`], except that it returns
1351    /// a [`PseudoCanonicalInput`] and requires both the `value` and the
1352    /// `param_env` to not contain any inference variables or placeholders.
1353    pub fn pseudo_canonicalize_query<V>(
1354        &self,
1355        param_env: ty::ParamEnv<'tcx>,
1356        value: V,
1357    ) -> PseudoCanonicalInput<'tcx, V>
1358    where
1359        V: TypeVisitable<TyCtxt<'tcx>>,
1360    {
1361        debug_assert!(!value.has_infer());
1362        debug_assert!(!value.has_placeholders());
1363        debug_assert!(!param_env.has_infer());
1364        debug_assert!(!param_env.has_placeholders());
1365        self.typing_env(param_env).as_query_input(value)
1366    }
1367
1368    /// The returned function is used in a fast path. If it returns `true` the variable is
1369    /// unchanged, `false` indicates that the status is unknown.
1370    #[inline]
1371    pub fn is_ty_infer_var_definitely_unchanged(&self) -> impl Fn(TyOrConstInferVar) -> bool {
1372        // This hoists the borrow/release out of the loop body.
1373        let inner = self.inner.try_borrow();
1374
1375        move |infer_var: TyOrConstInferVar| match (infer_var, &inner) {
1376            (TyOrConstInferVar::Ty(ty_var), Ok(inner)) => {
1377                use self::type_variable::TypeVariableValue;
1378
1379                matches!(
1380                    inner.try_type_variables_probe_ref(ty_var),
1381                    Some(TypeVariableValue::Unknown { .. })
1382                )
1383            }
1384            _ => false,
1385        }
1386    }
1387
1388    /// `ty_or_const_infer_var_changed` is equivalent to one of these two:
1389    ///   * `shallow_resolve(ty) != ty` (where `ty.kind = ty::Infer(_)`)
1390    ///   * `shallow_resolve(ct) != ct` (where `ct.kind = ty::ConstKind::Infer(_)`)
1391    ///
1392    /// However, `ty_or_const_infer_var_changed` is more efficient. It's always
1393    /// inlined, despite being large, because it has only two call sites that
1394    /// are extremely hot (both in `traits::fulfill`'s checking of `stalled_on`
1395    /// inference variables), and it handles both `Ty` and `ty::Const` without
1396    /// having to resort to storing full `GenericArg`s in `stalled_on`.
1397    #[inline(always)]
1398    pub fn ty_or_const_infer_var_changed(&self, infer_var: TyOrConstInferVar) -> bool {
1399        match infer_var {
1400            TyOrConstInferVar::Ty(v) => {
1401                use self::type_variable::TypeVariableValue;
1402
1403                // If `inlined_probe` returns a `Known` value, it never equals
1404                // `ty::Infer(ty::TyVar(v))`.
1405                match self.inner.borrow_mut().type_variables().inlined_probe(v) {
1406                    TypeVariableValue::Unknown { .. } => false,
1407                    TypeVariableValue::Known { .. } => true,
1408                }
1409            }
1410
1411            TyOrConstInferVar::TyInt(v) => {
1412                // If `inlined_probe_value` returns a value it's always a
1413                // `ty::Int(_)` or `ty::UInt(_)`, which never matches a
1414                // `ty::Infer(_)`.
1415                self.inner.borrow_mut().int_unification_table().inlined_probe_value(v).is_known()
1416            }
1417
1418            TyOrConstInferVar::TyFloat(v) => {
1419                // If `probe_value` returns a value it's always a
1420                // `ty::Float(_)`, which never matches a `ty::Infer(_)`.
1421                //
1422                // Not `inlined_probe_value(v)` because this call site is colder.
1423                self.inner.borrow_mut().float_unification_table().probe_value(v).is_known()
1424            }
1425
1426            TyOrConstInferVar::Const(v) => {
1427                // If `probe_value` returns a `Known` value, it never equals
1428                // `ty::ConstKind::Infer(ty::InferConst::Var(v))`.
1429                //
1430                // Not `inlined_probe_value(v)` because this call site is colder.
1431                match self.inner.borrow_mut().const_unification_table().probe_value(v) {
1432                    ConstVariableValue::Unknown { .. } => false,
1433                    ConstVariableValue::Known { .. } => true,
1434                }
1435            }
1436        }
1437    }
1438
1439    /// Attach a callback to be invoked on each root obligation evaluated in the new trait solver.
1440    pub fn attach_obligation_inspector(&self, inspector: ObligationInspector<'tcx>) {
1441        debug_assert!(
1442            self.obligation_inspector.get().is_none(),
1443            "shouldn't override a set obligation inspector"
1444        );
1445        self.obligation_inspector.set(Some(inspector));
1446    }
1447}
1448
1449/// Helper for [InferCtxt::ty_or_const_infer_var_changed] (see comment on that), currently
1450/// used only for `traits::fulfill`'s list of `stalled_on` inference variables.
1451#[derive(Copy, Clone, Debug)]
1452pub enum TyOrConstInferVar {
1453    /// Equivalent to `ty::Infer(ty::TyVar(_))`.
1454    Ty(TyVid),
1455    /// Equivalent to `ty::Infer(ty::IntVar(_))`.
1456    TyInt(IntVid),
1457    /// Equivalent to `ty::Infer(ty::FloatVar(_))`.
1458    TyFloat(FloatVid),
1459
1460    /// Equivalent to `ty::ConstKind::Infer(ty::InferConst::Var(_))`.
1461    Const(ConstVid),
1462}
1463
1464impl<'tcx> TyOrConstInferVar {
1465    /// Tries to extract an inference variable from a type or a constant, returns `None`
1466    /// for types other than `ty::Infer(_)` (or `InferTy::Fresh*`) and
1467    /// for constants other than `ty::ConstKind::Infer(_)` (or `InferConst::Fresh`).
1468    pub fn maybe_from_generic_arg(arg: GenericArg<'tcx>) -> Option<Self> {
1469        match arg.kind() {
1470            GenericArgKind::Type(ty) => Self::maybe_from_ty(ty),
1471            GenericArgKind::Const(ct) => Self::maybe_from_const(ct),
1472            GenericArgKind::Lifetime(_) => None,
1473        }
1474    }
1475
1476    /// Tries to extract an inference variable from a type or a constant, returns `None`
1477    /// for types other than `ty::Infer(_)` (or `InferTy::Fresh*`) and
1478    /// for constants other than `ty::ConstKind::Infer(_)` (or `InferConst::Fresh`).
1479    pub fn maybe_from_term(term: Term<'tcx>) -> Option<Self> {
1480        match term.kind() {
1481            TermKind::Ty(ty) => Self::maybe_from_ty(ty),
1482            TermKind::Const(ct) => Self::maybe_from_const(ct),
1483        }
1484    }
1485
1486    /// Tries to extract an inference variable from a type, returns `None`
1487    /// for types other than `ty::Infer(_)` (or `InferTy::Fresh*`).
1488    fn maybe_from_ty(ty: Ty<'tcx>) -> Option<Self> {
1489        match *ty.kind() {
1490            ty::Infer(ty::TyVar(v)) => Some(TyOrConstInferVar::Ty(v)),
1491            ty::Infer(ty::IntVar(v)) => Some(TyOrConstInferVar::TyInt(v)),
1492            ty::Infer(ty::FloatVar(v)) => Some(TyOrConstInferVar::TyFloat(v)),
1493            _ => None,
1494        }
1495    }
1496
1497    /// Tries to extract an inference variable from a constant, returns `None`
1498    /// for constants other than `ty::ConstKind::Infer(_)` (or `InferConst::Fresh`).
1499    fn maybe_from_const(ct: ty::Const<'tcx>) -> Option<Self> {
1500        match ct.kind() {
1501            ty::ConstKind::Infer(InferConst::Var(v)) => Some(TyOrConstInferVar::Const(v)),
1502            _ => None,
1503        }
1504    }
1505}
1506
1507/// Replace `{integer}` with `i32` and `{float}` with `f64`.
1508/// Used only for diagnostics.
1509struct InferenceLiteralEraser<'tcx> {
1510    tcx: TyCtxt<'tcx>,
1511}
1512
1513impl<'tcx> TypeFolder<TyCtxt<'tcx>> for InferenceLiteralEraser<'tcx> {
1514    fn cx(&self) -> TyCtxt<'tcx> {
1515        self.tcx
1516    }
1517
1518    fn fold_ty(&mut self, ty: Ty<'tcx>) -> Ty<'tcx> {
1519        match ty.kind() {
1520            ty::Infer(ty::IntVar(_) | ty::FreshIntTy(_)) => self.tcx.types.i32,
1521            ty::Infer(ty::FloatVar(_) | ty::FreshFloatTy(_)) => self.tcx.types.f64,
1522            _ => ty.super_fold_with(self),
1523        }
1524    }
1525}
1526
1527impl<'tcx> TypeTrace<'tcx> {
1528    pub fn span(&self) -> Span {
1529        self.cause.span
1530    }
1531
1532    pub fn types(cause: &ObligationCause<'tcx>, a: Ty<'tcx>, b: Ty<'tcx>) -> TypeTrace<'tcx> {
1533        TypeTrace {
1534            cause: cause.clone(),
1535            values: ValuePairs::Terms(ExpectedFound::new(a.into(), b.into())),
1536        }
1537    }
1538
1539    pub fn trait_refs(
1540        cause: &ObligationCause<'tcx>,
1541        a: ty::TraitRef<'tcx>,
1542        b: ty::TraitRef<'tcx>,
1543    ) -> TypeTrace<'tcx> {
1544        TypeTrace { cause: cause.clone(), values: ValuePairs::TraitRefs(ExpectedFound::new(a, b)) }
1545    }
1546
1547    pub fn consts(
1548        cause: &ObligationCause<'tcx>,
1549        a: ty::Const<'tcx>,
1550        b: ty::Const<'tcx>,
1551    ) -> TypeTrace<'tcx> {
1552        TypeTrace {
1553            cause: cause.clone(),
1554            values: ValuePairs::Terms(ExpectedFound::new(a.into(), b.into())),
1555        }
1556    }
1557}
1558
1559impl<'tcx> SubregionOrigin<'tcx> {
1560    pub fn span(&self) -> Span {
1561        match *self {
1562            SubregionOrigin::Subtype(ref a) => a.span(),
1563            SubregionOrigin::RelateObjectBound(a) => a,
1564            SubregionOrigin::RelateParamBound(a, ..) => a,
1565            SubregionOrigin::RelateRegionParamBound(a, _) => a,
1566            SubregionOrigin::Reborrow(a) => a,
1567            SubregionOrigin::ReferenceOutlivesReferent(_, a) => a,
1568            SubregionOrigin::CompareImplItemObligation { span, .. } => span,
1569            SubregionOrigin::AscribeUserTypeProvePredicate(span) => span,
1570            SubregionOrigin::CheckAssociatedTypeBounds { ref parent, .. } => parent.span(),
1571        }
1572    }
1573
1574    pub fn from_obligation_cause<F>(cause: &traits::ObligationCause<'tcx>, default: F) -> Self
1575    where
1576        F: FnOnce() -> Self,
1577    {
1578        match *cause.code() {
1579            traits::ObligationCauseCode::ReferenceOutlivesReferent(ref_type) => {
1580                SubregionOrigin::ReferenceOutlivesReferent(ref_type, cause.span)
1581            }
1582
1583            traits::ObligationCauseCode::CompareImplItem {
1584                impl_item_def_id,
1585                trait_item_def_id,
1586                kind: _,
1587            } => SubregionOrigin::CompareImplItemObligation {
1588                span: cause.span,
1589                impl_item_def_id,
1590                trait_item_def_id,
1591            },
1592
1593            traits::ObligationCauseCode::CheckAssociatedTypeBounds {
1594                impl_item_def_id,
1595                trait_item_def_id,
1596            } => SubregionOrigin::CheckAssociatedTypeBounds {
1597                impl_item_def_id,
1598                trait_item_def_id,
1599                parent: Box::new(default()),
1600            },
1601
1602            traits::ObligationCauseCode::AscribeUserTypeProvePredicate(span) => {
1603                SubregionOrigin::AscribeUserTypeProvePredicate(span)
1604            }
1605
1606            traits::ObligationCauseCode::ObjectTypeBound(ty, _reg) => {
1607                SubregionOrigin::RelateRegionParamBound(cause.span, Some(ty))
1608            }
1609
1610            _ => default(),
1611        }
1612    }
1613}
1614
1615impl RegionVariableOrigin {
1616    pub fn span(&self) -> Span {
1617        match *self {
1618            RegionVariableOrigin::Misc(a)
1619            | RegionVariableOrigin::PatternRegion(a)
1620            | RegionVariableOrigin::BorrowRegion(a)
1621            | RegionVariableOrigin::Autoref(a)
1622            | RegionVariableOrigin::Coercion(a)
1623            | RegionVariableOrigin::RegionParameterDefinition(a, ..)
1624            | RegionVariableOrigin::BoundRegion(a, ..)
1625            | RegionVariableOrigin::UpvarRegion(_, a) => a,
1626            RegionVariableOrigin::Nll(..) => bug!("NLL variable used with `span`"),
1627        }
1628    }
1629}
1630
1631impl<'tcx> InferCtxt<'tcx> {
1632    /// Given a [`hir::Block`], get the span of its last expression or
1633    /// statement, peeling off any inner blocks.
1634    pub fn find_block_span(&self, block: &'tcx hir::Block<'tcx>) -> Span {
1635        let block = block.innermost_block();
1636        if let Some(expr) = &block.expr {
1637            expr.span
1638        } else if let Some(stmt) = block.stmts.last() {
1639            // possibly incorrect trailing `;` in the else arm
1640            stmt.span
1641        } else {
1642            // empty block; point at its entirety
1643            block.span
1644        }
1645    }
1646
1647    /// Given a [`hir::HirId`] for a block (or an expr of a block), get the span
1648    /// of its last expression or statement, peeling off any inner blocks.
1649    pub fn find_block_span_from_hir_id(&self, hir_id: hir::HirId) -> Span {
1650        match self.tcx.hir_node(hir_id) {
1651            hir::Node::Block(blk)
1652            | hir::Node::Expr(&hir::Expr { kind: hir::ExprKind::Block(blk, _), .. }) => {
1653                self.find_block_span(blk)
1654            }
1655            hir::Node::Expr(e) => e.span,
1656            _ => DUMMY_SP,
1657        }
1658    }
1659}