]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_trait_selection/src/traits/project.rs
Rollup merge of #105071 - nnethercote:nnethercote-mailmap, r=compiler-errors
[rust.git] / compiler / rustc_trait_selection / src / traits / project.rs
1 //! Code for projecting associated types out of trait references.
2
3 use super::specialization_graph;
4 use super::translate_substs;
5 use super::util;
6 use super::MismatchedProjectionTypes;
7 use super::Obligation;
8 use super::ObligationCause;
9 use super::PredicateObligation;
10 use super::Selection;
11 use super::SelectionContext;
12 use super::SelectionError;
13 use super::{
14     ImplSourceClosureData, ImplSourceFnPointerData, ImplSourceFutureData, ImplSourceGeneratorData,
15     ImplSourceUserDefinedData,
16 };
17 use super::{Normalized, NormalizedTy, ProjectionCacheEntry, ProjectionCacheKey};
18
19 use crate::infer::type_variable::{TypeVariableOrigin, TypeVariableOriginKind};
20 use crate::infer::{InferCtxt, InferOk, LateBoundRegionConversionTime};
21 use crate::traits::error_reporting::TypeErrCtxtExt as _;
22 use crate::traits::query::evaluate_obligation::InferCtxtExt as _;
23 use crate::traits::select::ProjectionMatchesProjection;
24 use rustc_data_structures::sso::SsoHashSet;
25 use rustc_data_structures::stack::ensure_sufficient_stack;
26 use rustc_errors::ErrorGuaranteed;
27 use rustc_hir::def::DefKind;
28 use rustc_hir::def_id::DefId;
29 use rustc_hir::lang_items::LangItem;
30 use rustc_infer::infer::resolve::OpportunisticRegionResolver;
31 use rustc_infer::traits::ImplSourceBuiltinData;
32 use rustc_middle::traits::select::OverflowError;
33 use rustc_middle::ty::fold::{TypeFoldable, TypeFolder, TypeSuperFoldable};
34 use rustc_middle::ty::visit::{MaxUniverse, TypeVisitable};
35 use rustc_middle::ty::DefIdTree;
36 use rustc_middle::ty::{self, Term, ToPredicate, Ty, TyCtxt};
37 use rustc_span::symbol::sym;
38
39 use std::collections::BTreeMap;
40
41 pub use rustc_middle::traits::Reveal;
42
43 pub type PolyProjectionObligation<'tcx> = Obligation<'tcx, ty::PolyProjectionPredicate<'tcx>>;
44
45 pub type ProjectionObligation<'tcx> = Obligation<'tcx, ty::ProjectionPredicate<'tcx>>;
46
47 pub type ProjectionTyObligation<'tcx> = Obligation<'tcx, ty::ProjectionTy<'tcx>>;
48
49 pub(super) struct InProgress;
50
51 /// When attempting to resolve `<T as TraitRef>::Name` ...
52 #[derive(Debug)]
53 pub enum ProjectionError<'tcx> {
54     /// ...we found multiple sources of information and couldn't resolve the ambiguity.
55     TooManyCandidates,
56
57     /// ...an error occurred matching `T : TraitRef`
58     TraitSelectionError(SelectionError<'tcx>),
59 }
60
61 #[derive(PartialEq, Eq, Debug)]
62 enum ProjectionCandidate<'tcx> {
63     /// From a where-clause in the env or object type
64     ParamEnv(ty::PolyProjectionPredicate<'tcx>),
65
66     /// From the definition of `Trait` when you have something like
67     /// `<<A as Trait>::B as Trait2>::C`.
68     TraitDef(ty::PolyProjectionPredicate<'tcx>),
69
70     /// Bounds specified on an object type
71     Object(ty::PolyProjectionPredicate<'tcx>),
72
73     /// From an "impl" (or a "pseudo-impl" returned by select)
74     Select(Selection<'tcx>),
75
76     ImplTraitInTrait(ImplTraitInTraitCandidate<'tcx>),
77 }
78
79 #[derive(PartialEq, Eq, Debug)]
80 enum ImplTraitInTraitCandidate<'tcx> {
81     // The `impl Trait` from a trait function's default body
82     Trait,
83     // A concrete type provided from a trait's `impl Trait` from an impl
84     Impl(ImplSourceUserDefinedData<'tcx, PredicateObligation<'tcx>>),
85 }
86
87 enum ProjectionCandidateSet<'tcx> {
88     None,
89     Single(ProjectionCandidate<'tcx>),
90     Ambiguous,
91     Error(SelectionError<'tcx>),
92 }
93
94 impl<'tcx> ProjectionCandidateSet<'tcx> {
95     fn mark_ambiguous(&mut self) {
96         *self = ProjectionCandidateSet::Ambiguous;
97     }
98
99     fn mark_error(&mut self, err: SelectionError<'tcx>) {
100         *self = ProjectionCandidateSet::Error(err);
101     }
102
103     // Returns true if the push was successful, or false if the candidate
104     // was discarded -- this could be because of ambiguity, or because
105     // a higher-priority candidate is already there.
106     fn push_candidate(&mut self, candidate: ProjectionCandidate<'tcx>) -> bool {
107         use self::ProjectionCandidate::*;
108         use self::ProjectionCandidateSet::*;
109
110         // This wacky variable is just used to try and
111         // make code readable and avoid confusing paths.
112         // It is assigned a "value" of `()` only on those
113         // paths in which we wish to convert `*self` to
114         // ambiguous (and return false, because the candidate
115         // was not used). On other paths, it is not assigned,
116         // and hence if those paths *could* reach the code that
117         // comes after the match, this fn would not compile.
118         let convert_to_ambiguous;
119
120         match self {
121             None => {
122                 *self = Single(candidate);
123                 return true;
124             }
125
126             Single(current) => {
127                 // Duplicates can happen inside ParamEnv. In the case, we
128                 // perform a lazy deduplication.
129                 if current == &candidate {
130                     return false;
131                 }
132
133                 // Prefer where-clauses. As in select, if there are multiple
134                 // candidates, we prefer where-clause candidates over impls.  This
135                 // may seem a bit surprising, since impls are the source of
136                 // "truth" in some sense, but in fact some of the impls that SEEM
137                 // applicable are not, because of nested obligations. Where
138                 // clauses are the safer choice. See the comment on
139                 // `select::SelectionCandidate` and #21974 for more details.
140                 match (current, candidate) {
141                     (ParamEnv(..), ParamEnv(..)) => convert_to_ambiguous = (),
142                     (ParamEnv(..), _) => return false,
143                     (_, ParamEnv(..)) => unreachable!(),
144                     (_, _) => convert_to_ambiguous = (),
145                 }
146             }
147
148             Ambiguous | Error(..) => {
149                 return false;
150             }
151         }
152
153         // We only ever get here when we moved from a single candidate
154         // to ambiguous.
155         let () = convert_to_ambiguous;
156         *self = Ambiguous;
157         false
158     }
159 }
160
161 /// States returned from `poly_project_and_unify_type`. Takes the place
162 /// of the old return type, which was:
163 /// ```ignore (not-rust)
164 /// Result<
165 ///     Result<Option<Vec<PredicateObligation<'tcx>>>, InProgress>,
166 ///     MismatchedProjectionTypes<'tcx>,
167 /// >
168 /// ```
169 pub(super) enum ProjectAndUnifyResult<'tcx> {
170     /// The projection bound holds subject to the given obligations. If the
171     /// projection cannot be normalized because the required trait bound does
172     /// not hold, this is returned, with `obligations` being a predicate that
173     /// cannot be proven.
174     Holds(Vec<PredicateObligation<'tcx>>),
175     /// The projection cannot be normalized due to ambiguity. Resolving some
176     /// inference variables in the projection may fix this.
177     FailedNormalization,
178     /// The project cannot be normalized because `poly_project_and_unify_type`
179     /// is called recursively while normalizing the same projection.
180     Recursive,
181     // the projection can be normalized, but is not equal to the expected type.
182     // Returns the type error that arose from the mismatch.
183     MismatchedProjectionTypes(MismatchedProjectionTypes<'tcx>),
184 }
185
186 /// Evaluates constraints of the form:
187 /// ```ignore (not-rust)
188 /// for<...> <T as Trait>::U == V
189 /// ```
190 /// If successful, this may result in additional obligations. Also returns
191 /// the projection cache key used to track these additional obligations.
192 #[instrument(level = "debug", skip(selcx))]
193 pub(super) fn poly_project_and_unify_type<'cx, 'tcx>(
194     selcx: &mut SelectionContext<'cx, 'tcx>,
195     obligation: &PolyProjectionObligation<'tcx>,
196 ) -> ProjectAndUnifyResult<'tcx> {
197     let infcx = selcx.infcx;
198     let r = infcx.commit_if_ok(|_snapshot| {
199         let old_universe = infcx.universe();
200         let placeholder_predicate =
201             infcx.replace_bound_vars_with_placeholders(obligation.predicate);
202         let new_universe = infcx.universe();
203
204         let placeholder_obligation = obligation.with(infcx.tcx, placeholder_predicate);
205         match project_and_unify_type(selcx, &placeholder_obligation) {
206             ProjectAndUnifyResult::MismatchedProjectionTypes(e) => Err(e),
207             ProjectAndUnifyResult::Holds(obligations)
208                 if old_universe != new_universe
209                     && selcx.tcx().features().generic_associated_types_extended =>
210             {
211                 // If the `generic_associated_types_extended` feature is active, then we ignore any
212                 // obligations references lifetimes from any universe greater than or equal to the
213                 // universe just created. Otherwise, we can end up with something like `for<'a> I: 'a`,
214                 // which isn't quite what we want. Ideally, we want either an implied
215                 // `for<'a where I: 'a> I: 'a` or we want to "lazily" check these hold when we
216                 // substitute concrete regions. There is design work to be done here; until then,
217                 // however, this allows experimenting potential GAT features without running into
218                 // well-formedness issues.
219                 let new_obligations = obligations
220                     .into_iter()
221                     .filter(|obligation| {
222                         let mut visitor = MaxUniverse::new();
223                         obligation.predicate.visit_with(&mut visitor);
224                         visitor.max_universe() < new_universe
225                     })
226                     .collect();
227                 Ok(ProjectAndUnifyResult::Holds(new_obligations))
228             }
229             other => Ok(other),
230         }
231     });
232
233     match r {
234         Ok(inner) => inner,
235         Err(err) => ProjectAndUnifyResult::MismatchedProjectionTypes(err),
236     }
237 }
238
239 /// Evaluates constraints of the form:
240 /// ```ignore (not-rust)
241 /// <T as Trait>::U == V
242 /// ```
243 /// If successful, this may result in additional obligations.
244 ///
245 /// See [poly_project_and_unify_type] for an explanation of the return value.
246 #[instrument(level = "debug", skip(selcx))]
247 fn project_and_unify_type<'cx, 'tcx>(
248     selcx: &mut SelectionContext<'cx, 'tcx>,
249     obligation: &ProjectionObligation<'tcx>,
250 ) -> ProjectAndUnifyResult<'tcx> {
251     let mut obligations = vec![];
252
253     let infcx = selcx.infcx;
254     let normalized = match opt_normalize_projection_type(
255         selcx,
256         obligation.param_env,
257         obligation.predicate.projection_ty,
258         obligation.cause.clone(),
259         obligation.recursion_depth,
260         &mut obligations,
261     ) {
262         Ok(Some(n)) => n,
263         Ok(None) => return ProjectAndUnifyResult::FailedNormalization,
264         Err(InProgress) => return ProjectAndUnifyResult::Recursive,
265     };
266     debug!(?normalized, ?obligations, "project_and_unify_type result");
267     let actual = obligation.predicate.term;
268     // For an example where this is necessary see src/test/ui/impl-trait/nested-return-type2.rs
269     // This allows users to omit re-mentioning all bounds on an associated type and just use an
270     // `impl Trait` for the assoc type to add more bounds.
271     let InferOk { value: actual, obligations: new } =
272         selcx.infcx.replace_opaque_types_with_inference_vars(
273             actual,
274             obligation.cause.body_id,
275             obligation.cause.span,
276             obligation.param_env,
277         );
278     obligations.extend(new);
279
280     match infcx.at(&obligation.cause, obligation.param_env).eq(normalized, actual) {
281         Ok(InferOk { obligations: inferred_obligations, value: () }) => {
282             obligations.extend(inferred_obligations);
283             ProjectAndUnifyResult::Holds(obligations)
284         }
285         Err(err) => {
286             debug!("equating types encountered error {:?}", err);
287             ProjectAndUnifyResult::MismatchedProjectionTypes(MismatchedProjectionTypes { err })
288         }
289     }
290 }
291
292 /// Normalizes any associated type projections in `value`, replacing
293 /// them with a fully resolved type where possible. The return value
294 /// combines the normalized result and any additional obligations that
295 /// were incurred as result.
296 pub fn normalize<'a, 'b, 'tcx, T>(
297     selcx: &'a mut SelectionContext<'b, 'tcx>,
298     param_env: ty::ParamEnv<'tcx>,
299     cause: ObligationCause<'tcx>,
300     value: T,
301 ) -> Normalized<'tcx, T>
302 where
303     T: TypeFoldable<'tcx>,
304 {
305     let mut obligations = Vec::new();
306     let value = normalize_to(selcx, param_env, cause, value, &mut obligations);
307     Normalized { value, obligations }
308 }
309
310 pub fn normalize_to<'a, 'b, 'tcx, T>(
311     selcx: &'a mut SelectionContext<'b, 'tcx>,
312     param_env: ty::ParamEnv<'tcx>,
313     cause: ObligationCause<'tcx>,
314     value: T,
315     obligations: &mut Vec<PredicateObligation<'tcx>>,
316 ) -> T
317 where
318     T: TypeFoldable<'tcx>,
319 {
320     normalize_with_depth_to(selcx, param_env, cause, 0, value, obligations)
321 }
322
323 /// As `normalize`, but with a custom depth.
324 pub fn normalize_with_depth<'a, 'b, 'tcx, T>(
325     selcx: &'a mut SelectionContext<'b, 'tcx>,
326     param_env: ty::ParamEnv<'tcx>,
327     cause: ObligationCause<'tcx>,
328     depth: usize,
329     value: T,
330 ) -> Normalized<'tcx, T>
331 where
332     T: TypeFoldable<'tcx>,
333 {
334     let mut obligations = Vec::new();
335     let value = normalize_with_depth_to(selcx, param_env, cause, depth, value, &mut obligations);
336     Normalized { value, obligations }
337 }
338
339 #[instrument(level = "info", skip(selcx, param_env, cause, obligations))]
340 pub fn normalize_with_depth_to<'a, 'b, 'tcx, T>(
341     selcx: &'a mut SelectionContext<'b, 'tcx>,
342     param_env: ty::ParamEnv<'tcx>,
343     cause: ObligationCause<'tcx>,
344     depth: usize,
345     value: T,
346     obligations: &mut Vec<PredicateObligation<'tcx>>,
347 ) -> T
348 where
349     T: TypeFoldable<'tcx>,
350 {
351     debug!(obligations.len = obligations.len());
352     let mut normalizer = AssocTypeNormalizer::new(selcx, param_env, cause, depth, obligations);
353     let result = ensure_sufficient_stack(|| normalizer.fold(value));
354     debug!(?result, obligations.len = normalizer.obligations.len());
355     debug!(?normalizer.obligations,);
356     result
357 }
358
359 #[instrument(level = "info", skip(selcx, param_env, cause, obligations))]
360 pub fn try_normalize_with_depth_to<'a, 'b, 'tcx, T>(
361     selcx: &'a mut SelectionContext<'b, 'tcx>,
362     param_env: ty::ParamEnv<'tcx>,
363     cause: ObligationCause<'tcx>,
364     depth: usize,
365     value: T,
366     obligations: &mut Vec<PredicateObligation<'tcx>>,
367 ) -> T
368 where
369     T: TypeFoldable<'tcx>,
370 {
371     debug!(obligations.len = obligations.len());
372     let mut normalizer = AssocTypeNormalizer::new_without_eager_inference_replacement(
373         selcx,
374         param_env,
375         cause,
376         depth,
377         obligations,
378     );
379     let result = ensure_sufficient_stack(|| normalizer.fold(value));
380     debug!(?result, obligations.len = normalizer.obligations.len());
381     debug!(?normalizer.obligations,);
382     result
383 }
384
385 pub(crate) fn needs_normalization<'tcx, T: TypeVisitable<'tcx>>(value: &T, reveal: Reveal) -> bool {
386     match reveal {
387         Reveal::UserFacing => value
388             .has_type_flags(ty::TypeFlags::HAS_TY_PROJECTION | ty::TypeFlags::HAS_CT_PROJECTION),
389         Reveal::All => value.has_type_flags(
390             ty::TypeFlags::HAS_TY_PROJECTION
391                 | ty::TypeFlags::HAS_TY_OPAQUE
392                 | ty::TypeFlags::HAS_CT_PROJECTION,
393         ),
394     }
395 }
396
397 struct AssocTypeNormalizer<'a, 'b, 'tcx> {
398     selcx: &'a mut SelectionContext<'b, 'tcx>,
399     param_env: ty::ParamEnv<'tcx>,
400     cause: ObligationCause<'tcx>,
401     obligations: &'a mut Vec<PredicateObligation<'tcx>>,
402     depth: usize,
403     universes: Vec<Option<ty::UniverseIndex>>,
404     /// If true, when a projection is unable to be completed, an inference
405     /// variable will be created and an obligation registered to project to that
406     /// inference variable. Also, constants will be eagerly evaluated.
407     eager_inference_replacement: bool,
408 }
409
410 impl<'a, 'b, 'tcx> AssocTypeNormalizer<'a, 'b, 'tcx> {
411     fn new(
412         selcx: &'a mut SelectionContext<'b, 'tcx>,
413         param_env: ty::ParamEnv<'tcx>,
414         cause: ObligationCause<'tcx>,
415         depth: usize,
416         obligations: &'a mut Vec<PredicateObligation<'tcx>>,
417     ) -> AssocTypeNormalizer<'a, 'b, 'tcx> {
418         AssocTypeNormalizer {
419             selcx,
420             param_env,
421             cause,
422             obligations,
423             depth,
424             universes: vec![],
425             eager_inference_replacement: true,
426         }
427     }
428
429     fn new_without_eager_inference_replacement(
430         selcx: &'a mut SelectionContext<'b, 'tcx>,
431         param_env: ty::ParamEnv<'tcx>,
432         cause: ObligationCause<'tcx>,
433         depth: usize,
434         obligations: &'a mut Vec<PredicateObligation<'tcx>>,
435     ) -> AssocTypeNormalizer<'a, 'b, 'tcx> {
436         AssocTypeNormalizer {
437             selcx,
438             param_env,
439             cause,
440             obligations,
441             depth,
442             universes: vec![],
443             eager_inference_replacement: false,
444         }
445     }
446
447     fn fold<T: TypeFoldable<'tcx>>(&mut self, value: T) -> T {
448         let value = self.selcx.infcx.resolve_vars_if_possible(value);
449         debug!(?value);
450
451         assert!(
452             !value.has_escaping_bound_vars(),
453             "Normalizing {:?} without wrapping in a `Binder`",
454             value
455         );
456
457         if !needs_normalization(&value, self.param_env.reveal()) {
458             value
459         } else {
460             value.fold_with(self)
461         }
462     }
463 }
464
465 impl<'a, 'b, 'tcx> TypeFolder<'tcx> for AssocTypeNormalizer<'a, 'b, 'tcx> {
466     fn tcx<'c>(&'c self) -> TyCtxt<'tcx> {
467         self.selcx.tcx()
468     }
469
470     fn fold_binder<T: TypeFoldable<'tcx>>(
471         &mut self,
472         t: ty::Binder<'tcx, T>,
473     ) -> ty::Binder<'tcx, T> {
474         self.universes.push(None);
475         let t = t.super_fold_with(self);
476         self.universes.pop();
477         t
478     }
479
480     fn fold_ty(&mut self, ty: Ty<'tcx>) -> Ty<'tcx> {
481         if !needs_normalization(&ty, self.param_env.reveal()) {
482             return ty;
483         }
484
485         // We try to be a little clever here as a performance optimization in
486         // cases where there are nested projections under binders.
487         // For example:
488         // ```
489         // for<'a> fn(<T as Foo>::One<'a, Box<dyn Bar<'a, Item=<T as Foo>::Two<'a>>>>)
490         // ```
491         // We normalize the substs on the projection before the projecting, but
492         // if we're naive, we'll
493         //   replace bound vars on inner, project inner, replace placeholders on inner,
494         //   replace bound vars on outer, project outer, replace placeholders on outer
495         //
496         // However, if we're a bit more clever, we can replace the bound vars
497         // on the entire type before normalizing nested projections, meaning we
498         //   replace bound vars on outer, project inner,
499         //   project outer, replace placeholders on outer
500         //
501         // This is possible because the inner `'a` will already be a placeholder
502         // when we need to normalize the inner projection
503         //
504         // On the other hand, this does add a bit of complexity, since we only
505         // replace bound vars if the current type is a `Projection` and we need
506         // to make sure we don't forget to fold the substs regardless.
507
508         match *ty.kind() {
509             // This is really important. While we *can* handle this, this has
510             // severe performance implications for large opaque types with
511             // late-bound regions. See `issue-88862` benchmark.
512             ty::Opaque(def_id, substs) if !substs.has_escaping_bound_vars() => {
513                 // Only normalize `impl Trait` outside of type inference, usually in codegen.
514                 match self.param_env.reveal() {
515                     Reveal::UserFacing => ty.super_fold_with(self),
516
517                     Reveal::All => {
518                         let recursion_limit = self.tcx().recursion_limit();
519                         if !recursion_limit.value_within_limit(self.depth) {
520                             let obligation = Obligation::with_depth(
521                                 self.tcx(),
522                                 self.cause.clone(),
523                                 recursion_limit.0,
524                                 self.param_env,
525                                 ty,
526                             );
527                             self.selcx.infcx.err_ctxt().report_overflow_error(&obligation, true);
528                         }
529
530                         let substs = substs.fold_with(self);
531                         let generic_ty = self.tcx().bound_type_of(def_id);
532                         let concrete_ty = generic_ty.subst(self.tcx(), substs);
533                         self.depth += 1;
534                         let folded_ty = self.fold_ty(concrete_ty);
535                         self.depth -= 1;
536                         folded_ty
537                     }
538                 }
539             }
540
541             ty::Projection(data) if !data.has_escaping_bound_vars() => {
542                 // This branch is *mostly* just an optimization: when we don't
543                 // have escaping bound vars, we don't need to replace them with
544                 // placeholders (see branch below). *Also*, we know that we can
545                 // register an obligation to *later* project, since we know
546                 // there won't be bound vars there.
547                 let data = data.fold_with(self);
548                 let normalized_ty = if self.eager_inference_replacement {
549                     normalize_projection_type(
550                         self.selcx,
551                         self.param_env,
552                         data,
553                         self.cause.clone(),
554                         self.depth,
555                         &mut self.obligations,
556                     )
557                 } else {
558                     opt_normalize_projection_type(
559                         self.selcx,
560                         self.param_env,
561                         data,
562                         self.cause.clone(),
563                         self.depth,
564                         &mut self.obligations,
565                     )
566                     .ok()
567                     .flatten()
568                     .unwrap_or_else(|| ty.super_fold_with(self).into())
569                 };
570                 debug!(
571                     ?self.depth,
572                     ?ty,
573                     ?normalized_ty,
574                     obligations.len = ?self.obligations.len(),
575                     "AssocTypeNormalizer: normalized type"
576                 );
577                 normalized_ty.ty().unwrap()
578             }
579
580             ty::Projection(data) => {
581                 // If there are escaping bound vars, we temporarily replace the
582                 // bound vars with placeholders. Note though, that in the case
583                 // that we still can't project for whatever reason (e.g. self
584                 // type isn't known enough), we *can't* register an obligation
585                 // and return an inference variable (since then that obligation
586                 // would have bound vars and that's a can of worms). Instead,
587                 // we just give up and fall back to pretending like we never tried!
588                 //
589                 // Note: this isn't necessarily the final approach here; we may
590                 // want to figure out how to register obligations with escaping vars
591                 // or handle this some other way.
592
593                 let infcx = self.selcx.infcx;
594                 let (data, mapped_regions, mapped_types, mapped_consts) =
595                     BoundVarReplacer::replace_bound_vars(infcx, &mut self.universes, data);
596                 let data = data.fold_with(self);
597                 let normalized_ty = opt_normalize_projection_type(
598                     self.selcx,
599                     self.param_env,
600                     data,
601                     self.cause.clone(),
602                     self.depth,
603                     &mut self.obligations,
604                 )
605                 .ok()
606                 .flatten()
607                 .map(|term| term.ty().unwrap())
608                 .map(|normalized_ty| {
609                     PlaceholderReplacer::replace_placeholders(
610                         infcx,
611                         mapped_regions,
612                         mapped_types,
613                         mapped_consts,
614                         &self.universes,
615                         normalized_ty,
616                     )
617                 })
618                 .unwrap_or_else(|| ty.super_fold_with(self));
619
620                 debug!(
621                     ?self.depth,
622                     ?ty,
623                     ?normalized_ty,
624                     obligations.len = ?self.obligations.len(),
625                     "AssocTypeNormalizer: normalized type"
626                 );
627                 normalized_ty
628             }
629
630             _ => ty.super_fold_with(self),
631         }
632     }
633
634     #[instrument(skip(self), level = "debug")]
635     fn fold_const(&mut self, constant: ty::Const<'tcx>) -> ty::Const<'tcx> {
636         let tcx = self.selcx.tcx();
637         if tcx.lazy_normalization() || !needs_normalization(&constant, self.param_env.reveal()) {
638             constant
639         } else {
640             let constant = constant.super_fold_with(self);
641             debug!(?constant, ?self.param_env);
642             with_replaced_escaping_bound_vars(
643                 self.selcx.infcx,
644                 &mut self.universes,
645                 constant,
646                 |constant| constant.eval(tcx, self.param_env),
647             )
648         }
649     }
650
651     #[inline]
652     fn fold_predicate(&mut self, p: ty::Predicate<'tcx>) -> ty::Predicate<'tcx> {
653         if p.allow_normalization() && needs_normalization(&p, self.param_env.reveal()) {
654             p.super_fold_with(self)
655         } else {
656             p
657         }
658     }
659 }
660
661 pub struct BoundVarReplacer<'me, 'tcx> {
662     infcx: &'me InferCtxt<'tcx>,
663     // These three maps track the bound variable that were replaced by placeholders. It might be
664     // nice to remove these since we already have the `kind` in the placeholder; we really just need
665     // the `var` (but we *could* bring that into scope if we were to track them as we pass them).
666     mapped_regions: BTreeMap<ty::PlaceholderRegion, ty::BoundRegion>,
667     mapped_types: BTreeMap<ty::PlaceholderType, ty::BoundTy>,
668     mapped_consts: BTreeMap<ty::PlaceholderConst<'tcx>, ty::BoundVar>,
669     // The current depth relative to *this* folding, *not* the entire normalization. In other words,
670     // the depth of binders we've passed here.
671     current_index: ty::DebruijnIndex,
672     // The `UniverseIndex` of the binding levels above us. These are optional, since we are lazy:
673     // we don't actually create a universe until we see a bound var we have to replace.
674     universe_indices: &'me mut Vec<Option<ty::UniverseIndex>>,
675 }
676
677 /// Executes `f` on `value` after replacing all escaping bound variables with placeholders
678 /// and then replaces these placeholders with the original bound variables in the result.
679 ///
680 /// In most places, bound variables should be replaced right when entering a binder, making
681 /// this function unnecessary. However, normalization currently does not do that, so we have
682 /// to do this lazily.
683 ///
684 /// You should not add any additional uses of this function, at least not without first
685 /// discussing it with t-types.
686 ///
687 /// FIXME(@lcnr): We may even consider experimenting with eagerly replacing bound vars during
688 /// normalization as well, at which point this function will be unnecessary and can be removed.
689 pub fn with_replaced_escaping_bound_vars<'a, 'tcx, T: TypeFoldable<'tcx>, R: TypeFoldable<'tcx>>(
690     infcx: &'a InferCtxt<'tcx>,
691     universe_indices: &'a mut Vec<Option<ty::UniverseIndex>>,
692     value: T,
693     f: impl FnOnce(T) -> R,
694 ) -> R {
695     if value.has_escaping_bound_vars() {
696         let (value, mapped_regions, mapped_types, mapped_consts) =
697             BoundVarReplacer::replace_bound_vars(infcx, universe_indices, value);
698         let result = f(value);
699         PlaceholderReplacer::replace_placeholders(
700             infcx,
701             mapped_regions,
702             mapped_types,
703             mapped_consts,
704             universe_indices,
705             result,
706         )
707     } else {
708         f(value)
709     }
710 }
711
712 impl<'me, 'tcx> BoundVarReplacer<'me, 'tcx> {
713     /// Returns `Some` if we *were* able to replace bound vars. If there are any bound vars that
714     /// use a binding level above `universe_indices.len()`, we fail.
715     pub fn replace_bound_vars<T: TypeFoldable<'tcx>>(
716         infcx: &'me InferCtxt<'tcx>,
717         universe_indices: &'me mut Vec<Option<ty::UniverseIndex>>,
718         value: T,
719     ) -> (
720         T,
721         BTreeMap<ty::PlaceholderRegion, ty::BoundRegion>,
722         BTreeMap<ty::PlaceholderType, ty::BoundTy>,
723         BTreeMap<ty::PlaceholderConst<'tcx>, ty::BoundVar>,
724     ) {
725         let mapped_regions: BTreeMap<ty::PlaceholderRegion, ty::BoundRegion> = BTreeMap::new();
726         let mapped_types: BTreeMap<ty::PlaceholderType, ty::BoundTy> = BTreeMap::new();
727         let mapped_consts: BTreeMap<ty::PlaceholderConst<'tcx>, ty::BoundVar> = BTreeMap::new();
728
729         let mut replacer = BoundVarReplacer {
730             infcx,
731             mapped_regions,
732             mapped_types,
733             mapped_consts,
734             current_index: ty::INNERMOST,
735             universe_indices,
736         };
737
738         let value = value.fold_with(&mut replacer);
739
740         (value, replacer.mapped_regions, replacer.mapped_types, replacer.mapped_consts)
741     }
742
743     fn universe_for(&mut self, debruijn: ty::DebruijnIndex) -> ty::UniverseIndex {
744         let infcx = self.infcx;
745         let index =
746             self.universe_indices.len() + self.current_index.as_usize() - debruijn.as_usize() - 1;
747         let universe = self.universe_indices[index].unwrap_or_else(|| {
748             for i in self.universe_indices.iter_mut().take(index + 1) {
749                 *i = i.or_else(|| Some(infcx.create_next_universe()))
750             }
751             self.universe_indices[index].unwrap()
752         });
753         universe
754     }
755 }
756
757 impl<'tcx> TypeFolder<'tcx> for BoundVarReplacer<'_, 'tcx> {
758     fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
759         self.infcx.tcx
760     }
761
762     fn fold_binder<T: TypeFoldable<'tcx>>(
763         &mut self,
764         t: ty::Binder<'tcx, T>,
765     ) -> ty::Binder<'tcx, T> {
766         self.current_index.shift_in(1);
767         let t = t.super_fold_with(self);
768         self.current_index.shift_out(1);
769         t
770     }
771
772     fn fold_region(&mut self, r: ty::Region<'tcx>) -> ty::Region<'tcx> {
773         match *r {
774             ty::ReLateBound(debruijn, _)
775                 if debruijn.as_usize() + 1
776                     > self.current_index.as_usize() + self.universe_indices.len() =>
777             {
778                 bug!("Bound vars outside of `self.universe_indices`");
779             }
780             ty::ReLateBound(debruijn, br) if debruijn >= self.current_index => {
781                 let universe = self.universe_for(debruijn);
782                 let p = ty::PlaceholderRegion { universe, name: br.kind };
783                 self.mapped_regions.insert(p, br);
784                 self.infcx.tcx.mk_region(ty::RePlaceholder(p))
785             }
786             _ => r,
787         }
788     }
789
790     fn fold_ty(&mut self, t: Ty<'tcx>) -> Ty<'tcx> {
791         match *t.kind() {
792             ty::Bound(debruijn, _)
793                 if debruijn.as_usize() + 1
794                     > self.current_index.as_usize() + self.universe_indices.len() =>
795             {
796                 bug!("Bound vars outside of `self.universe_indices`");
797             }
798             ty::Bound(debruijn, bound_ty) if debruijn >= self.current_index => {
799                 let universe = self.universe_for(debruijn);
800                 let p = ty::PlaceholderType { universe, name: bound_ty.var };
801                 self.mapped_types.insert(p, bound_ty);
802                 self.infcx.tcx.mk_ty(ty::Placeholder(p))
803             }
804             _ if t.has_vars_bound_at_or_above(self.current_index) => t.super_fold_with(self),
805             _ => t,
806         }
807     }
808
809     fn fold_const(&mut self, ct: ty::Const<'tcx>) -> ty::Const<'tcx> {
810         match ct.kind() {
811             ty::ConstKind::Bound(debruijn, _)
812                 if debruijn.as_usize() + 1
813                     > self.current_index.as_usize() + self.universe_indices.len() =>
814             {
815                 bug!("Bound vars outside of `self.universe_indices`");
816             }
817             ty::ConstKind::Bound(debruijn, bound_const) if debruijn >= self.current_index => {
818                 let universe = self.universe_for(debruijn);
819                 let p = ty::PlaceholderConst { universe, name: bound_const };
820                 self.mapped_consts.insert(p, bound_const);
821                 self.infcx.tcx.mk_const(p, ct.ty())
822             }
823             _ => ct.super_fold_with(self),
824         }
825     }
826
827     fn fold_predicate(&mut self, p: ty::Predicate<'tcx>) -> ty::Predicate<'tcx> {
828         if p.has_vars_bound_at_or_above(self.current_index) { p.super_fold_with(self) } else { p }
829     }
830 }
831
832 /// The inverse of [`BoundVarReplacer`]: replaces placeholders with the bound vars from which they came.
833 pub struct PlaceholderReplacer<'me, 'tcx> {
834     infcx: &'me InferCtxt<'tcx>,
835     mapped_regions: BTreeMap<ty::PlaceholderRegion, ty::BoundRegion>,
836     mapped_types: BTreeMap<ty::PlaceholderType, ty::BoundTy>,
837     mapped_consts: BTreeMap<ty::PlaceholderConst<'tcx>, ty::BoundVar>,
838     universe_indices: &'me [Option<ty::UniverseIndex>],
839     current_index: ty::DebruijnIndex,
840 }
841
842 impl<'me, 'tcx> PlaceholderReplacer<'me, 'tcx> {
843     pub fn replace_placeholders<T: TypeFoldable<'tcx>>(
844         infcx: &'me InferCtxt<'tcx>,
845         mapped_regions: BTreeMap<ty::PlaceholderRegion, ty::BoundRegion>,
846         mapped_types: BTreeMap<ty::PlaceholderType, ty::BoundTy>,
847         mapped_consts: BTreeMap<ty::PlaceholderConst<'tcx>, ty::BoundVar>,
848         universe_indices: &'me [Option<ty::UniverseIndex>],
849         value: T,
850     ) -> T {
851         let mut replacer = PlaceholderReplacer {
852             infcx,
853             mapped_regions,
854             mapped_types,
855             mapped_consts,
856             universe_indices,
857             current_index: ty::INNERMOST,
858         };
859         value.fold_with(&mut replacer)
860     }
861 }
862
863 impl<'tcx> TypeFolder<'tcx> for PlaceholderReplacer<'_, 'tcx> {
864     fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
865         self.infcx.tcx
866     }
867
868     fn fold_binder<T: TypeFoldable<'tcx>>(
869         &mut self,
870         t: ty::Binder<'tcx, T>,
871     ) -> ty::Binder<'tcx, T> {
872         if !t.has_placeholders() && !t.has_infer_regions() {
873             return t;
874         }
875         self.current_index.shift_in(1);
876         let t = t.super_fold_with(self);
877         self.current_index.shift_out(1);
878         t
879     }
880
881     fn fold_region(&mut self, r0: ty::Region<'tcx>) -> ty::Region<'tcx> {
882         let r1 = match *r0 {
883             ty::ReVar(_) => self
884                 .infcx
885                 .inner
886                 .borrow_mut()
887                 .unwrap_region_constraints()
888                 .opportunistic_resolve_region(self.infcx.tcx, r0),
889             _ => r0,
890         };
891
892         let r2 = match *r1 {
893             ty::RePlaceholder(p) => {
894                 let replace_var = self.mapped_regions.get(&p);
895                 match replace_var {
896                     Some(replace_var) => {
897                         let index = self
898                             .universe_indices
899                             .iter()
900                             .position(|u| matches!(u, Some(pu) if *pu == p.universe))
901                             .unwrap_or_else(|| bug!("Unexpected placeholder universe."));
902                         let db = ty::DebruijnIndex::from_usize(
903                             self.universe_indices.len() - index + self.current_index.as_usize() - 1,
904                         );
905                         self.tcx().mk_region(ty::ReLateBound(db, *replace_var))
906                     }
907                     None => r1,
908                 }
909             }
910             _ => r1,
911         };
912
913         debug!(?r0, ?r1, ?r2, "fold_region");
914
915         r2
916     }
917
918     fn fold_ty(&mut self, ty: Ty<'tcx>) -> Ty<'tcx> {
919         match *ty.kind() {
920             ty::Placeholder(p) => {
921                 let replace_var = self.mapped_types.get(&p);
922                 match replace_var {
923                     Some(replace_var) => {
924                         let index = self
925                             .universe_indices
926                             .iter()
927                             .position(|u| matches!(u, Some(pu) if *pu == p.universe))
928                             .unwrap_or_else(|| bug!("Unexpected placeholder universe."));
929                         let db = ty::DebruijnIndex::from_usize(
930                             self.universe_indices.len() - index + self.current_index.as_usize() - 1,
931                         );
932                         self.tcx().mk_ty(ty::Bound(db, *replace_var))
933                     }
934                     None => ty,
935                 }
936             }
937
938             _ if ty.has_placeholders() || ty.has_infer_regions() => ty.super_fold_with(self),
939             _ => ty,
940         }
941     }
942
943     fn fold_const(&mut self, ct: ty::Const<'tcx>) -> ty::Const<'tcx> {
944         if let ty::ConstKind::Placeholder(p) = ct.kind() {
945             let replace_var = self.mapped_consts.get(&p);
946             match replace_var {
947                 Some(replace_var) => {
948                     let index = self
949                         .universe_indices
950                         .iter()
951                         .position(|u| matches!(u, Some(pu) if *pu == p.universe))
952                         .unwrap_or_else(|| bug!("Unexpected placeholder universe."));
953                     let db = ty::DebruijnIndex::from_usize(
954                         self.universe_indices.len() - index + self.current_index.as_usize() - 1,
955                     );
956                     self.tcx().mk_const(ty::ConstKind::Bound(db, *replace_var), ct.ty())
957                 }
958                 None => ct,
959             }
960         } else {
961             ct.super_fold_with(self)
962         }
963     }
964 }
965
966 /// The guts of `normalize`: normalize a specific projection like `<T
967 /// as Trait>::Item`. The result is always a type (and possibly
968 /// additional obligations). If ambiguity arises, which implies that
969 /// there are unresolved type variables in the projection, we will
970 /// substitute a fresh type variable `$X` and generate a new
971 /// obligation `<T as Trait>::Item == $X` for later.
972 pub fn normalize_projection_type<'a, 'b, 'tcx>(
973     selcx: &'a mut SelectionContext<'b, 'tcx>,
974     param_env: ty::ParamEnv<'tcx>,
975     projection_ty: ty::ProjectionTy<'tcx>,
976     cause: ObligationCause<'tcx>,
977     depth: usize,
978     obligations: &mut Vec<PredicateObligation<'tcx>>,
979 ) -> Term<'tcx> {
980     opt_normalize_projection_type(
981         selcx,
982         param_env,
983         projection_ty,
984         cause.clone(),
985         depth,
986         obligations,
987     )
988     .ok()
989     .flatten()
990     .unwrap_or_else(move || {
991         // if we bottom out in ambiguity, create a type variable
992         // and a deferred predicate to resolve this when more type
993         // information is available.
994
995         selcx.infcx.infer_projection(param_env, projection_ty, cause, depth + 1, obligations).into()
996     })
997 }
998
999 /// The guts of `normalize`: normalize a specific projection like `<T
1000 /// as Trait>::Item`. The result is always a type (and possibly
1001 /// additional obligations). Returns `None` in the case of ambiguity,
1002 /// which indicates that there are unbound type variables.
1003 ///
1004 /// This function used to return `Option<NormalizedTy<'tcx>>`, which contains a
1005 /// `Ty<'tcx>` and an obligations vector. But that obligation vector was very
1006 /// often immediately appended to another obligations vector. So now this
1007 /// function takes an obligations vector and appends to it directly, which is
1008 /// slightly uglier but avoids the need for an extra short-lived allocation.
1009 #[instrument(level = "debug", skip(selcx, param_env, cause, obligations))]
1010 fn opt_normalize_projection_type<'a, 'b, 'tcx>(
1011     selcx: &'a mut SelectionContext<'b, 'tcx>,
1012     param_env: ty::ParamEnv<'tcx>,
1013     projection_ty: ty::ProjectionTy<'tcx>,
1014     cause: ObligationCause<'tcx>,
1015     depth: usize,
1016     obligations: &mut Vec<PredicateObligation<'tcx>>,
1017 ) -> Result<Option<Term<'tcx>>, InProgress> {
1018     let infcx = selcx.infcx;
1019     // Don't use the projection cache in intercrate mode -
1020     // the `infcx` may be re-used between intercrate in non-intercrate
1021     // mode, which could lead to using incorrect cache results.
1022     let use_cache = !selcx.is_intercrate();
1023
1024     let projection_ty = infcx.resolve_vars_if_possible(projection_ty);
1025     let cache_key = ProjectionCacheKey::new(projection_ty);
1026
1027     // FIXME(#20304) For now, I am caching here, which is good, but it
1028     // means we don't capture the type variables that are created in
1029     // the case of ambiguity. Which means we may create a large stream
1030     // of such variables. OTOH, if we move the caching up a level, we
1031     // would not benefit from caching when proving `T: Trait<U=Foo>`
1032     // bounds. It might be the case that we want two distinct caches,
1033     // or else another kind of cache entry.
1034
1035     let cache_result = if use_cache {
1036         infcx.inner.borrow_mut().projection_cache().try_start(cache_key)
1037     } else {
1038         Ok(())
1039     };
1040     match cache_result {
1041         Ok(()) => debug!("no cache"),
1042         Err(ProjectionCacheEntry::Ambiguous) => {
1043             // If we found ambiguity the last time, that means we will continue
1044             // to do so until some type in the key changes (and we know it
1045             // hasn't, because we just fully resolved it).
1046             debug!("found cache entry: ambiguous");
1047             return Ok(None);
1048         }
1049         Err(ProjectionCacheEntry::InProgress) => {
1050             // Under lazy normalization, this can arise when
1051             // bootstrapping.  That is, imagine an environment with a
1052             // where-clause like `A::B == u32`. Now, if we are asked
1053             // to normalize `A::B`, we will want to check the
1054             // where-clauses in scope. So we will try to unify `A::B`
1055             // with `A::B`, which can trigger a recursive
1056             // normalization.
1057
1058             debug!("found cache entry: in-progress");
1059
1060             // Cache that normalizing this projection resulted in a cycle. This
1061             // should ensure that, unless this happens within a snapshot that's
1062             // rolled back, fulfillment or evaluation will notice the cycle.
1063
1064             if use_cache {
1065                 infcx.inner.borrow_mut().projection_cache().recur(cache_key);
1066             }
1067             return Err(InProgress);
1068         }
1069         Err(ProjectionCacheEntry::Recur) => {
1070             debug!("recur cache");
1071             return Err(InProgress);
1072         }
1073         Err(ProjectionCacheEntry::NormalizedTy { ty, complete: _ }) => {
1074             // This is the hottest path in this function.
1075             //
1076             // If we find the value in the cache, then return it along
1077             // with the obligations that went along with it. Note
1078             // that, when using a fulfillment context, these
1079             // obligations could in principle be ignored: they have
1080             // already been registered when the cache entry was
1081             // created (and hence the new ones will quickly be
1082             // discarded as duplicated). But when doing trait
1083             // evaluation this is not the case, and dropping the trait
1084             // evaluations can causes ICEs (e.g., #43132).
1085             debug!(?ty, "found normalized ty");
1086             obligations.extend(ty.obligations);
1087             return Ok(Some(ty.value));
1088         }
1089         Err(ProjectionCacheEntry::Error) => {
1090             debug!("opt_normalize_projection_type: found error");
1091             let result = normalize_to_error(selcx, param_env, projection_ty, cause, depth);
1092             obligations.extend(result.obligations);
1093             return Ok(Some(result.value.into()));
1094         }
1095     }
1096
1097     let obligation =
1098         Obligation::with_depth(selcx.tcx(), cause.clone(), depth, param_env, projection_ty);
1099
1100     match project(selcx, &obligation) {
1101         Ok(Projected::Progress(Progress {
1102             term: projected_term,
1103             obligations: mut projected_obligations,
1104         })) => {
1105             // if projection succeeded, then what we get out of this
1106             // is also non-normalized (consider: it was derived from
1107             // an impl, where-clause etc) and hence we must
1108             // re-normalize it
1109
1110             let projected_term = selcx.infcx.resolve_vars_if_possible(projected_term);
1111
1112             let mut result = if projected_term.has_projections() {
1113                 let mut normalizer = AssocTypeNormalizer::new(
1114                     selcx,
1115                     param_env,
1116                     cause,
1117                     depth + 1,
1118                     &mut projected_obligations,
1119                 );
1120                 let normalized_ty = normalizer.fold(projected_term);
1121
1122                 Normalized { value: normalized_ty, obligations: projected_obligations }
1123             } else {
1124                 Normalized { value: projected_term, obligations: projected_obligations }
1125             };
1126
1127             let mut deduped: SsoHashSet<_> = Default::default();
1128             result.obligations.drain_filter(|projected_obligation| {
1129                 if !deduped.insert(projected_obligation.clone()) {
1130                     return true;
1131                 }
1132                 false
1133             });
1134
1135             if use_cache {
1136                 infcx.inner.borrow_mut().projection_cache().insert_term(cache_key, result.clone());
1137             }
1138             obligations.extend(result.obligations);
1139             Ok(Some(result.value))
1140         }
1141         Ok(Projected::NoProgress(projected_ty)) => {
1142             let result = Normalized { value: projected_ty, obligations: vec![] };
1143             if use_cache {
1144                 infcx.inner.borrow_mut().projection_cache().insert_term(cache_key, result.clone());
1145             }
1146             // No need to extend `obligations`.
1147             Ok(Some(result.value))
1148         }
1149         Err(ProjectionError::TooManyCandidates) => {
1150             debug!("opt_normalize_projection_type: too many candidates");
1151             if use_cache {
1152                 infcx.inner.borrow_mut().projection_cache().ambiguous(cache_key);
1153             }
1154             Ok(None)
1155         }
1156         Err(ProjectionError::TraitSelectionError(_)) => {
1157             debug!("opt_normalize_projection_type: ERROR");
1158             // if we got an error processing the `T as Trait` part,
1159             // just return `ty::err` but add the obligation `T :
1160             // Trait`, which when processed will cause the error to be
1161             // reported later
1162
1163             if use_cache {
1164                 infcx.inner.borrow_mut().projection_cache().error(cache_key);
1165             }
1166             let result = normalize_to_error(selcx, param_env, projection_ty, cause, depth);
1167             obligations.extend(result.obligations);
1168             Ok(Some(result.value.into()))
1169         }
1170     }
1171 }
1172
1173 /// If we are projecting `<T as Trait>::Item`, but `T: Trait` does not
1174 /// hold. In various error cases, we cannot generate a valid
1175 /// normalized projection. Therefore, we create an inference variable
1176 /// return an associated obligation that, when fulfilled, will lead to
1177 /// an error.
1178 ///
1179 /// Note that we used to return `Error` here, but that was quite
1180 /// dubious -- the premise was that an error would *eventually* be
1181 /// reported, when the obligation was processed. But in general once
1182 /// you see an `Error` you are supposed to be able to assume that an
1183 /// error *has been* reported, so that you can take whatever heuristic
1184 /// paths you want to take. To make things worse, it was possible for
1185 /// cycles to arise, where you basically had a setup like `<MyType<$0>
1186 /// as Trait>::Foo == $0`. Here, normalizing `<MyType<$0> as
1187 /// Trait>::Foo> to `[type error]` would lead to an obligation of
1188 /// `<MyType<[type error]> as Trait>::Foo`. We are supposed to report
1189 /// an error for this obligation, but we legitimately should not,
1190 /// because it contains `[type error]`. Yuck! (See issue #29857 for
1191 /// one case where this arose.)
1192 fn normalize_to_error<'a, 'tcx>(
1193     selcx: &mut SelectionContext<'a, 'tcx>,
1194     param_env: ty::ParamEnv<'tcx>,
1195     projection_ty: ty::ProjectionTy<'tcx>,
1196     cause: ObligationCause<'tcx>,
1197     depth: usize,
1198 ) -> NormalizedTy<'tcx> {
1199     let trait_ref = ty::Binder::dummy(projection_ty.trait_ref(selcx.tcx()));
1200     let trait_obligation = Obligation {
1201         cause,
1202         recursion_depth: depth,
1203         param_env,
1204         predicate: trait_ref.without_const().to_predicate(selcx.tcx()),
1205     };
1206     let tcx = selcx.infcx.tcx;
1207     let def_id = projection_ty.item_def_id;
1208     let new_value = selcx.infcx.next_ty_var(TypeVariableOrigin {
1209         kind: TypeVariableOriginKind::NormalizeProjectionType,
1210         span: tcx.def_span(def_id),
1211     });
1212     Normalized { value: new_value, obligations: vec![trait_obligation] }
1213 }
1214
1215 enum Projected<'tcx> {
1216     Progress(Progress<'tcx>),
1217     NoProgress(ty::Term<'tcx>),
1218 }
1219
1220 struct Progress<'tcx> {
1221     term: ty::Term<'tcx>,
1222     obligations: Vec<PredicateObligation<'tcx>>,
1223 }
1224
1225 impl<'tcx> Progress<'tcx> {
1226     fn error(tcx: TyCtxt<'tcx>) -> Self {
1227         Progress { term: tcx.ty_error().into(), obligations: vec![] }
1228     }
1229
1230     fn with_addl_obligations(mut self, mut obligations: Vec<PredicateObligation<'tcx>>) -> Self {
1231         self.obligations.append(&mut obligations);
1232         self
1233     }
1234 }
1235
1236 /// Computes the result of a projection type (if we can).
1237 ///
1238 /// IMPORTANT:
1239 /// - `obligation` must be fully normalized
1240 #[instrument(level = "info", skip(selcx))]
1241 fn project<'cx, 'tcx>(
1242     selcx: &mut SelectionContext<'cx, 'tcx>,
1243     obligation: &ProjectionTyObligation<'tcx>,
1244 ) -> Result<Projected<'tcx>, ProjectionError<'tcx>> {
1245     if !selcx.tcx().recursion_limit().value_within_limit(obligation.recursion_depth) {
1246         // This should really be an immediate error, but some existing code
1247         // relies on being able to recover from this.
1248         return Err(ProjectionError::TraitSelectionError(SelectionError::Overflow(
1249             OverflowError::Canonical,
1250         )));
1251     }
1252
1253     if obligation.predicate.references_error() {
1254         return Ok(Projected::Progress(Progress::error(selcx.tcx())));
1255     }
1256
1257     let mut candidates = ProjectionCandidateSet::None;
1258
1259     assemble_candidate_for_impl_trait_in_trait(selcx, obligation, &mut candidates);
1260
1261     // Make sure that the following procedures are kept in order. ParamEnv
1262     // needs to be first because it has highest priority, and Select checks
1263     // the return value of push_candidate which assumes it's ran at last.
1264     assemble_candidates_from_param_env(selcx, obligation, &mut candidates);
1265
1266     assemble_candidates_from_trait_def(selcx, obligation, &mut candidates);
1267
1268     assemble_candidates_from_object_ty(selcx, obligation, &mut candidates);
1269
1270     if let ProjectionCandidateSet::Single(ProjectionCandidate::Object(_)) = candidates {
1271         // Avoid normalization cycle from selection (see
1272         // `assemble_candidates_from_object_ty`).
1273         // FIXME(lazy_normalization): Lazy normalization should save us from
1274         // having to special case this.
1275     } else {
1276         assemble_candidates_from_impls(selcx, obligation, &mut candidates);
1277     };
1278
1279     match candidates {
1280         ProjectionCandidateSet::Single(candidate) => {
1281             Ok(Projected::Progress(confirm_candidate(selcx, obligation, candidate)))
1282         }
1283         ProjectionCandidateSet::None => Ok(Projected::NoProgress(
1284             // FIXME(associated_const_generics): this may need to change in the future?
1285             // need to investigate whether or not this is fine.
1286             selcx
1287                 .tcx()
1288                 .mk_projection(obligation.predicate.item_def_id, obligation.predicate.substs)
1289                 .into(),
1290         )),
1291         // Error occurred while trying to processing impls.
1292         ProjectionCandidateSet::Error(e) => Err(ProjectionError::TraitSelectionError(e)),
1293         // Inherent ambiguity that prevents us from even enumerating the
1294         // candidates.
1295         ProjectionCandidateSet::Ambiguous => Err(ProjectionError::TooManyCandidates),
1296     }
1297 }
1298
1299 /// If the predicate's item is an `ImplTraitPlaceholder`, we do a select on the
1300 /// corresponding trait ref. If this yields an `impl`, then we're able to project
1301 /// to a concrete type, since we have an `impl`'s method  to provide the RPITIT.
1302 fn assemble_candidate_for_impl_trait_in_trait<'cx, 'tcx>(
1303     selcx: &mut SelectionContext<'cx, 'tcx>,
1304     obligation: &ProjectionTyObligation<'tcx>,
1305     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1306 ) {
1307     let tcx = selcx.tcx();
1308     if tcx.def_kind(obligation.predicate.item_def_id) == DefKind::ImplTraitPlaceholder {
1309         let trait_fn_def_id = tcx.impl_trait_in_trait_parent(obligation.predicate.item_def_id);
1310         // If we are trying to project an RPITIT with trait's default `Self` parameter,
1311         // then we must be within a default trait body.
1312         if obligation.predicate.self_ty()
1313             == ty::InternalSubsts::identity_for_item(tcx, obligation.predicate.item_def_id)
1314                 .type_at(0)
1315             && tcx.associated_item(trait_fn_def_id).defaultness(tcx).has_value()
1316         {
1317             candidate_set.push_candidate(ProjectionCandidate::ImplTraitInTrait(
1318                 ImplTraitInTraitCandidate::Trait,
1319             ));
1320             return;
1321         }
1322
1323         let trait_def_id = tcx.parent(trait_fn_def_id);
1324         let trait_substs =
1325             obligation.predicate.substs.truncate_to(tcx, tcx.generics_of(trait_def_id));
1326         // FIXME(named-returns): Binders
1327         let trait_predicate =
1328             ty::Binder::dummy(ty::TraitRef { def_id: trait_def_id, substs: trait_substs });
1329
1330         let _ = selcx.infcx.commit_if_ok(|_| {
1331             match selcx.select(&obligation.with(tcx, trait_predicate)) {
1332                 Ok(Some(super::ImplSource::UserDefined(data))) => {
1333                     candidate_set.push_candidate(ProjectionCandidate::ImplTraitInTrait(
1334                         ImplTraitInTraitCandidate::Impl(data),
1335                     ));
1336                     Ok(())
1337                 }
1338                 Ok(None) => {
1339                     candidate_set.mark_ambiguous();
1340                     return Err(());
1341                 }
1342                 Ok(Some(_)) => {
1343                     // Don't know enough about the impl to provide a useful signature
1344                     return Err(());
1345                 }
1346                 Err(e) => {
1347                     debug!(error = ?e, "selection error");
1348                     candidate_set.mark_error(e);
1349                     return Err(());
1350                 }
1351             }
1352         });
1353     }
1354 }
1355
1356 /// The first thing we have to do is scan through the parameter
1357 /// environment to see whether there are any projection predicates
1358 /// there that can answer this question.
1359 fn assemble_candidates_from_param_env<'cx, 'tcx>(
1360     selcx: &mut SelectionContext<'cx, 'tcx>,
1361     obligation: &ProjectionTyObligation<'tcx>,
1362     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1363 ) {
1364     assemble_candidates_from_predicates(
1365         selcx,
1366         obligation,
1367         candidate_set,
1368         ProjectionCandidate::ParamEnv,
1369         obligation.param_env.caller_bounds().iter(),
1370         false,
1371     );
1372 }
1373
1374 /// In the case of a nested projection like `<<A as Foo>::FooT as Bar>::BarT`, we may find
1375 /// that the definition of `Foo` has some clues:
1376 ///
1377 /// ```ignore (illustrative)
1378 /// trait Foo {
1379 ///     type FooT : Bar<BarT=i32>
1380 /// }
1381 /// ```
1382 ///
1383 /// Here, for example, we could conclude that the result is `i32`.
1384 fn assemble_candidates_from_trait_def<'cx, 'tcx>(
1385     selcx: &mut SelectionContext<'cx, 'tcx>,
1386     obligation: &ProjectionTyObligation<'tcx>,
1387     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1388 ) {
1389     debug!("assemble_candidates_from_trait_def(..)");
1390
1391     let tcx = selcx.tcx();
1392     // Check whether the self-type is itself a projection.
1393     // If so, extract what we know from the trait and try to come up with a good answer.
1394     let bounds = match *obligation.predicate.self_ty().kind() {
1395         ty::Projection(ref data) => tcx.bound_item_bounds(data.item_def_id).subst(tcx, data.substs),
1396         ty::Opaque(def_id, substs) => tcx.bound_item_bounds(def_id).subst(tcx, substs),
1397         ty::Infer(ty::TyVar(_)) => {
1398             // If the self-type is an inference variable, then it MAY wind up
1399             // being a projected type, so induce an ambiguity.
1400             candidate_set.mark_ambiguous();
1401             return;
1402         }
1403         _ => return,
1404     };
1405
1406     assemble_candidates_from_predicates(
1407         selcx,
1408         obligation,
1409         candidate_set,
1410         ProjectionCandidate::TraitDef,
1411         bounds.iter(),
1412         true,
1413     );
1414 }
1415
1416 /// In the case of a trait object like
1417 /// `<dyn Iterator<Item = ()> as Iterator>::Item` we can use the existential
1418 /// predicate in the trait object.
1419 ///
1420 /// We don't go through the select candidate for these bounds to avoid cycles:
1421 /// In the above case, `dyn Iterator<Item = ()>: Iterator` would create a
1422 /// nested obligation of `<dyn Iterator<Item = ()> as Iterator>::Item: Sized`,
1423 /// this then has to be normalized without having to prove
1424 /// `dyn Iterator<Item = ()>: Iterator` again.
1425 fn assemble_candidates_from_object_ty<'cx, 'tcx>(
1426     selcx: &mut SelectionContext<'cx, 'tcx>,
1427     obligation: &ProjectionTyObligation<'tcx>,
1428     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1429 ) {
1430     debug!("assemble_candidates_from_object_ty(..)");
1431
1432     let tcx = selcx.tcx();
1433
1434     let self_ty = obligation.predicate.self_ty();
1435     let object_ty = selcx.infcx.shallow_resolve(self_ty);
1436     let data = match object_ty.kind() {
1437         ty::Dynamic(data, ..) => data,
1438         ty::Infer(ty::TyVar(_)) => {
1439             // If the self-type is an inference variable, then it MAY wind up
1440             // being an object type, so induce an ambiguity.
1441             candidate_set.mark_ambiguous();
1442             return;
1443         }
1444         _ => return,
1445     };
1446     let env_predicates = data
1447         .projection_bounds()
1448         .filter(|bound| bound.item_def_id() == obligation.predicate.item_def_id)
1449         .map(|p| p.with_self_ty(tcx, object_ty).to_predicate(tcx));
1450
1451     assemble_candidates_from_predicates(
1452         selcx,
1453         obligation,
1454         candidate_set,
1455         ProjectionCandidate::Object,
1456         env_predicates,
1457         false,
1458     );
1459 }
1460
1461 #[instrument(
1462     level = "debug",
1463     skip(selcx, candidate_set, ctor, env_predicates, potentially_unnormalized_candidates)
1464 )]
1465 fn assemble_candidates_from_predicates<'cx, 'tcx>(
1466     selcx: &mut SelectionContext<'cx, 'tcx>,
1467     obligation: &ProjectionTyObligation<'tcx>,
1468     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1469     ctor: fn(ty::PolyProjectionPredicate<'tcx>) -> ProjectionCandidate<'tcx>,
1470     env_predicates: impl Iterator<Item = ty::Predicate<'tcx>>,
1471     potentially_unnormalized_candidates: bool,
1472 ) {
1473     let infcx = selcx.infcx;
1474     for predicate in env_predicates {
1475         let bound_predicate = predicate.kind();
1476         if let ty::PredicateKind::Clause(ty::Clause::Projection(data)) =
1477             predicate.kind().skip_binder()
1478         {
1479             let data = bound_predicate.rebind(data);
1480             if data.projection_def_id() != obligation.predicate.item_def_id {
1481                 continue;
1482             }
1483
1484             let is_match = infcx.probe(|_| {
1485                 selcx.match_projection_projections(
1486                     obligation,
1487                     data,
1488                     potentially_unnormalized_candidates,
1489                 )
1490             });
1491
1492             match is_match {
1493                 ProjectionMatchesProjection::Yes => {
1494                     candidate_set.push_candidate(ctor(data));
1495
1496                     if potentially_unnormalized_candidates
1497                         && !obligation.predicate.has_non_region_infer()
1498                     {
1499                         // HACK: Pick the first trait def candidate for a fully
1500                         // inferred predicate. This is to allow duplicates that
1501                         // differ only in normalization.
1502                         return;
1503                     }
1504                 }
1505                 ProjectionMatchesProjection::Ambiguous => {
1506                     candidate_set.mark_ambiguous();
1507                 }
1508                 ProjectionMatchesProjection::No => {}
1509             }
1510         }
1511     }
1512 }
1513
1514 #[instrument(level = "debug", skip(selcx, obligation, candidate_set))]
1515 fn assemble_candidates_from_impls<'cx, 'tcx>(
1516     selcx: &mut SelectionContext<'cx, 'tcx>,
1517     obligation: &ProjectionTyObligation<'tcx>,
1518     candidate_set: &mut ProjectionCandidateSet<'tcx>,
1519 ) {
1520     // Can't assemble candidate from impl for RPITIT
1521     if selcx.tcx().def_kind(obligation.predicate.item_def_id) == DefKind::ImplTraitPlaceholder {
1522         return;
1523     }
1524
1525     // If we are resolving `<T as TraitRef<...>>::Item == Type`,
1526     // start out by selecting the predicate `T as TraitRef<...>`:
1527     let poly_trait_ref = ty::Binder::dummy(obligation.predicate.trait_ref(selcx.tcx()));
1528     let trait_obligation = obligation.with(selcx.tcx(), poly_trait_ref);
1529     let _ = selcx.infcx.commit_if_ok(|_| {
1530         let impl_source = match selcx.select(&trait_obligation) {
1531             Ok(Some(impl_source)) => impl_source,
1532             Ok(None) => {
1533                 candidate_set.mark_ambiguous();
1534                 return Err(());
1535             }
1536             Err(e) => {
1537                 debug!(error = ?e, "selection error");
1538                 candidate_set.mark_error(e);
1539                 return Err(());
1540             }
1541         };
1542
1543         let eligible = match &impl_source {
1544             super::ImplSource::Closure(_)
1545             | super::ImplSource::Generator(_)
1546             | super::ImplSource::Future(_)
1547             | super::ImplSource::FnPointer(_)
1548             | super::ImplSource::TraitAlias(_) => true,
1549             super::ImplSource::UserDefined(impl_data) => {
1550                 // We have to be careful when projecting out of an
1551                 // impl because of specialization. If we are not in
1552                 // codegen (i.e., projection mode is not "any"), and the
1553                 // impl's type is declared as default, then we disable
1554                 // projection (even if the trait ref is fully
1555                 // monomorphic). In the case where trait ref is not
1556                 // fully monomorphic (i.e., includes type parameters),
1557                 // this is because those type parameters may
1558                 // ultimately be bound to types from other crates that
1559                 // may have specialized impls we can't see. In the
1560                 // case where the trait ref IS fully monomorphic, this
1561                 // is a policy decision that we made in the RFC in
1562                 // order to preserve flexibility for the crate that
1563                 // defined the specializable impl to specialize later
1564                 // for existing types.
1565                 //
1566                 // In either case, we handle this by not adding a
1567                 // candidate for an impl if it contains a `default`
1568                 // type.
1569                 //
1570                 // NOTE: This should be kept in sync with the similar code in
1571                 // `rustc_ty_utils::instance::resolve_associated_item()`.
1572                 let node_item =
1573                     assoc_def(selcx, impl_data.impl_def_id, obligation.predicate.item_def_id)
1574                         .map_err(|ErrorGuaranteed { .. }| ())?;
1575
1576                 if node_item.is_final() {
1577                     // Non-specializable items are always projectable.
1578                     true
1579                 } else {
1580                     // Only reveal a specializable default if we're past type-checking
1581                     // and the obligation is monomorphic, otherwise passes such as
1582                     // transmute checking and polymorphic MIR optimizations could
1583                     // get a result which isn't correct for all monomorphizations.
1584                     if obligation.param_env.reveal() == Reveal::All {
1585                         // NOTE(eddyb) inference variables can resolve to parameters, so
1586                         // assume `poly_trait_ref` isn't monomorphic, if it contains any.
1587                         let poly_trait_ref = selcx.infcx.resolve_vars_if_possible(poly_trait_ref);
1588                         !poly_trait_ref.still_further_specializable()
1589                     } else {
1590                         debug!(
1591                             assoc_ty = ?selcx.tcx().def_path_str(node_item.item.def_id),
1592                             ?obligation.predicate,
1593                             "assemble_candidates_from_impls: not eligible due to default",
1594                         );
1595                         false
1596                     }
1597                 }
1598             }
1599             super::ImplSource::Builtin(..) => {
1600                 // While a builtin impl may be known to exist, the associated type may not yet
1601                 // be known. Any type with multiple potential associated types is therefore
1602                 // not eligible.
1603                 let self_ty = selcx.infcx.shallow_resolve(obligation.predicate.self_ty());
1604
1605                 let lang_items = selcx.tcx().lang_items();
1606                 if lang_items.discriminant_kind_trait() == Some(poly_trait_ref.def_id()) {
1607                     match self_ty.kind() {
1608                         ty::Bool
1609                         | ty::Char
1610                         | ty::Int(_)
1611                         | ty::Uint(_)
1612                         | ty::Float(_)
1613                         | ty::Adt(..)
1614                         | ty::Foreign(_)
1615                         | ty::Str
1616                         | ty::Array(..)
1617                         | ty::Slice(_)
1618                         | ty::RawPtr(..)
1619                         | ty::Ref(..)
1620                         | ty::FnDef(..)
1621                         | ty::FnPtr(..)
1622                         | ty::Dynamic(..)
1623                         | ty::Closure(..)
1624                         | ty::Generator(..)
1625                         | ty::GeneratorWitness(..)
1626                         | ty::Never
1627                         | ty::Tuple(..)
1628                         // Integers and floats always have `u8` as their discriminant.
1629                         | ty::Infer(ty::InferTy::IntVar(_) | ty::InferTy::FloatVar(..)) => true,
1630
1631                          // type parameters, opaques, and unnormalized projections have pointer
1632                         // metadata if they're known (e.g. by the param_env) to be sized
1633                         ty::Param(_)
1634                         | ty::Projection(..)
1635                         | ty::Opaque(..)
1636                         | ty::Bound(..)
1637                         | ty::Placeholder(..)
1638                         | ty::Infer(..)
1639                         | ty::Error(_) => false,
1640                     }
1641                 } else if lang_items.pointee_trait() == Some(poly_trait_ref.def_id()) {
1642                     let tail = selcx.tcx().struct_tail_with_normalize(
1643                         self_ty,
1644                         |ty| {
1645                             // We throw away any obligations we get from this, since we normalize
1646                             // and confirm these obligations once again during confirmation
1647                             normalize_with_depth(
1648                                 selcx,
1649                                 obligation.param_env,
1650                                 obligation.cause.clone(),
1651                                 obligation.recursion_depth + 1,
1652                                 ty,
1653                             )
1654                             .value
1655                         },
1656                         || {},
1657                     );
1658
1659                     match tail.kind() {
1660                         ty::Bool
1661                         | ty::Char
1662                         | ty::Int(_)
1663                         | ty::Uint(_)
1664                         | ty::Float(_)
1665                         | ty::Str
1666                         | ty::Array(..)
1667                         | ty::Slice(_)
1668                         | ty::RawPtr(..)
1669                         | ty::Ref(..)
1670                         | ty::FnDef(..)
1671                         | ty::FnPtr(..)
1672                         | ty::Dynamic(..)
1673                         | ty::Closure(..)
1674                         | ty::Generator(..)
1675                         | ty::GeneratorWitness(..)
1676                         | ty::Never
1677                         // Extern types have unit metadata, according to RFC 2850
1678                         | ty::Foreign(_)
1679                         // If returned by `struct_tail_without_normalization` this is a unit struct
1680                         // without any fields, or not a struct, and therefore is Sized.
1681                         | ty::Adt(..)
1682                         // If returned by `struct_tail_without_normalization` this is the empty tuple.
1683                         | ty::Tuple(..)
1684                         // Integers and floats are always Sized, and so have unit type metadata.
1685                         | ty::Infer(ty::InferTy::IntVar(_) | ty::InferTy::FloatVar(..)) => true,
1686
1687                         // type parameters, opaques, and unnormalized projections have pointer
1688                         // metadata if they're known (e.g. by the param_env) to be sized
1689                         ty::Param(_) | ty::Projection(..) | ty::Opaque(..)
1690                             if selcx.infcx.predicate_must_hold_modulo_regions(
1691                                 &obligation.with(
1692                                     selcx.tcx(),
1693                                     ty::Binder::dummy(
1694                                         selcx.tcx().at(obligation.cause.span()).mk_trait_ref(LangItem::Sized, [self_ty]),
1695                                     )
1696                                     .without_const(),
1697                                 ),
1698                             ) =>
1699                         {
1700                             true
1701                         }
1702
1703                         // FIXME(compiler-errors): are Bound and Placeholder types ever known sized?
1704                         ty::Param(_)
1705                         | ty::Projection(..)
1706                         | ty::Opaque(..)
1707                         | ty::Bound(..)
1708                         | ty::Placeholder(..)
1709                         | ty::Infer(..)
1710                         | ty::Error(_) => {
1711                             if tail.has_infer_types() {
1712                                 candidate_set.mark_ambiguous();
1713                             }
1714                             false
1715                         }
1716                     }
1717                 } else {
1718                     bug!("unexpected builtin trait with associated type: {poly_trait_ref:?}")
1719                 }
1720             }
1721             super::ImplSource::Param(..) => {
1722                 // This case tell us nothing about the value of an
1723                 // associated type. Consider:
1724                 //
1725                 // ```
1726                 // trait SomeTrait { type Foo; }
1727                 // fn foo<T:SomeTrait>(...) { }
1728                 // ```
1729                 //
1730                 // If the user writes `<T as SomeTrait>::Foo`, then the `T
1731                 // : SomeTrait` binding does not help us decide what the
1732                 // type `Foo` is (at least, not more specifically than
1733                 // what we already knew).
1734                 //
1735                 // But wait, you say! What about an example like this:
1736                 //
1737                 // ```
1738                 // fn bar<T:SomeTrait<Foo=usize>>(...) { ... }
1739                 // ```
1740                 //
1741                 // Doesn't the `T : SomeTrait<Foo=usize>` predicate help
1742                 // resolve `T::Foo`? And of course it does, but in fact
1743                 // that single predicate is desugared into two predicates
1744                 // in the compiler: a trait predicate (`T : SomeTrait`) and a
1745                 // projection. And the projection where clause is handled
1746                 // in `assemble_candidates_from_param_env`.
1747                 false
1748             }
1749             super::ImplSource::Object(_) => {
1750                 // Handled by the `Object` projection candidate. See
1751                 // `assemble_candidates_from_object_ty` for an explanation of
1752                 // why we special case object types.
1753                 false
1754             }
1755             super::ImplSource::AutoImpl(..)
1756             | super::ImplSource::TraitUpcasting(_)
1757             | super::ImplSource::ConstDestruct(_) => {
1758                 // These traits have no associated types.
1759                 selcx.tcx().sess.delay_span_bug(
1760                     obligation.cause.span,
1761                     &format!("Cannot project an associated type from `{:?}`", impl_source),
1762                 );
1763                 return Err(());
1764             }
1765         };
1766
1767         if eligible {
1768             if candidate_set.push_candidate(ProjectionCandidate::Select(impl_source)) {
1769                 Ok(())
1770             } else {
1771                 Err(())
1772             }
1773         } else {
1774             Err(())
1775         }
1776     });
1777 }
1778
1779 fn confirm_candidate<'cx, 'tcx>(
1780     selcx: &mut SelectionContext<'cx, 'tcx>,
1781     obligation: &ProjectionTyObligation<'tcx>,
1782     candidate: ProjectionCandidate<'tcx>,
1783 ) -> Progress<'tcx> {
1784     debug!(?obligation, ?candidate, "confirm_candidate");
1785     let mut progress = match candidate {
1786         ProjectionCandidate::ParamEnv(poly_projection)
1787         | ProjectionCandidate::Object(poly_projection) => {
1788             confirm_param_env_candidate(selcx, obligation, poly_projection, false)
1789         }
1790
1791         ProjectionCandidate::TraitDef(poly_projection) => {
1792             confirm_param_env_candidate(selcx, obligation, poly_projection, true)
1793         }
1794
1795         ProjectionCandidate::Select(impl_source) => {
1796             confirm_select_candidate(selcx, obligation, impl_source)
1797         }
1798         ProjectionCandidate::ImplTraitInTrait(ImplTraitInTraitCandidate::Impl(data)) => {
1799             confirm_impl_trait_in_trait_candidate(selcx, obligation, data)
1800         }
1801         // If we're projecting an RPITIT for a default trait body, that's just
1802         // the same def-id, but as an opaque type (with regular RPIT semantics).
1803         ProjectionCandidate::ImplTraitInTrait(ImplTraitInTraitCandidate::Trait) => Progress {
1804             term: selcx
1805                 .tcx()
1806                 .mk_opaque(obligation.predicate.item_def_id, obligation.predicate.substs)
1807                 .into(),
1808             obligations: vec![],
1809         },
1810     };
1811
1812     // When checking for cycle during evaluation, we compare predicates with
1813     // "syntactic" equality. Since normalization generally introduces a type
1814     // with new region variables, we need to resolve them to existing variables
1815     // when possible for this to work. See `auto-trait-projection-recursion.rs`
1816     // for a case where this matters.
1817     if progress.term.has_infer_regions() {
1818         progress.term = progress.term.fold_with(&mut OpportunisticRegionResolver::new(selcx.infcx));
1819     }
1820     progress
1821 }
1822
1823 fn confirm_select_candidate<'cx, 'tcx>(
1824     selcx: &mut SelectionContext<'cx, 'tcx>,
1825     obligation: &ProjectionTyObligation<'tcx>,
1826     impl_source: Selection<'tcx>,
1827 ) -> Progress<'tcx> {
1828     match impl_source {
1829         super::ImplSource::UserDefined(data) => confirm_impl_candidate(selcx, obligation, data),
1830         super::ImplSource::Generator(data) => confirm_generator_candidate(selcx, obligation, data),
1831         super::ImplSource::Future(data) => confirm_future_candidate(selcx, obligation, data),
1832         super::ImplSource::Closure(data) => confirm_closure_candidate(selcx, obligation, data),
1833         super::ImplSource::FnPointer(data) => confirm_fn_pointer_candidate(selcx, obligation, data),
1834         super::ImplSource::Builtin(data) => confirm_builtin_candidate(selcx, obligation, data),
1835         super::ImplSource::Object(_)
1836         | super::ImplSource::AutoImpl(..)
1837         | super::ImplSource::Param(..)
1838         | super::ImplSource::TraitUpcasting(_)
1839         | super::ImplSource::TraitAlias(..)
1840         | super::ImplSource::ConstDestruct(_) => {
1841             // we don't create Select candidates with this kind of resolution
1842             span_bug!(
1843                 obligation.cause.span,
1844                 "Cannot project an associated type from `{:?}`",
1845                 impl_source
1846             )
1847         }
1848     }
1849 }
1850
1851 fn confirm_generator_candidate<'cx, 'tcx>(
1852     selcx: &mut SelectionContext<'cx, 'tcx>,
1853     obligation: &ProjectionTyObligation<'tcx>,
1854     impl_source: ImplSourceGeneratorData<'tcx, PredicateObligation<'tcx>>,
1855 ) -> Progress<'tcx> {
1856     let gen_sig = impl_source.substs.as_generator().poly_sig();
1857     let Normalized { value: gen_sig, obligations } = normalize_with_depth(
1858         selcx,
1859         obligation.param_env,
1860         obligation.cause.clone(),
1861         obligation.recursion_depth + 1,
1862         gen_sig,
1863     );
1864
1865     debug!(?obligation, ?gen_sig, ?obligations, "confirm_generator_candidate");
1866
1867     let tcx = selcx.tcx();
1868
1869     let gen_def_id = tcx.require_lang_item(LangItem::Generator, None);
1870
1871     let predicate = super::util::generator_trait_ref_and_outputs(
1872         tcx,
1873         gen_def_id,
1874         obligation.predicate.self_ty(),
1875         gen_sig,
1876     )
1877     .map_bound(|(trait_ref, yield_ty, return_ty)| {
1878         let name = tcx.associated_item(obligation.predicate.item_def_id).name;
1879         let ty = if name == sym::Return {
1880             return_ty
1881         } else if name == sym::Yield {
1882             yield_ty
1883         } else {
1884             bug!()
1885         };
1886
1887         ty::ProjectionPredicate {
1888             projection_ty: ty::ProjectionTy {
1889                 substs: trait_ref.substs,
1890                 item_def_id: obligation.predicate.item_def_id,
1891             },
1892             term: ty.into(),
1893         }
1894     });
1895
1896     confirm_param_env_candidate(selcx, obligation, predicate, false)
1897         .with_addl_obligations(impl_source.nested)
1898         .with_addl_obligations(obligations)
1899 }
1900
1901 fn confirm_future_candidate<'cx, 'tcx>(
1902     selcx: &mut SelectionContext<'cx, 'tcx>,
1903     obligation: &ProjectionTyObligation<'tcx>,
1904     impl_source: ImplSourceFutureData<'tcx, PredicateObligation<'tcx>>,
1905 ) -> Progress<'tcx> {
1906     let gen_sig = impl_source.substs.as_generator().poly_sig();
1907     let Normalized { value: gen_sig, obligations } = normalize_with_depth(
1908         selcx,
1909         obligation.param_env,
1910         obligation.cause.clone(),
1911         obligation.recursion_depth + 1,
1912         gen_sig,
1913     );
1914
1915     debug!(?obligation, ?gen_sig, ?obligations, "confirm_future_candidate");
1916
1917     let tcx = selcx.tcx();
1918     let fut_def_id = tcx.require_lang_item(LangItem::Future, None);
1919
1920     let predicate = super::util::future_trait_ref_and_outputs(
1921         tcx,
1922         fut_def_id,
1923         obligation.predicate.self_ty(),
1924         gen_sig,
1925     )
1926     .map_bound(|(trait_ref, return_ty)| {
1927         debug_assert_eq!(tcx.associated_item(obligation.predicate.item_def_id).name, sym::Output);
1928
1929         ty::ProjectionPredicate {
1930             projection_ty: ty::ProjectionTy {
1931                 substs: trait_ref.substs,
1932                 item_def_id: obligation.predicate.item_def_id,
1933             },
1934             term: return_ty.into(),
1935         }
1936     });
1937
1938     confirm_param_env_candidate(selcx, obligation, predicate, false)
1939         .with_addl_obligations(impl_source.nested)
1940         .with_addl_obligations(obligations)
1941 }
1942
1943 fn confirm_builtin_candidate<'cx, 'tcx>(
1944     selcx: &mut SelectionContext<'cx, 'tcx>,
1945     obligation: &ProjectionTyObligation<'tcx>,
1946     data: ImplSourceBuiltinData<PredicateObligation<'tcx>>,
1947 ) -> Progress<'tcx> {
1948     let tcx = selcx.tcx();
1949     let self_ty = obligation.predicate.self_ty();
1950     let substs = tcx.mk_substs([self_ty.into()].iter());
1951     let lang_items = tcx.lang_items();
1952     let item_def_id = obligation.predicate.item_def_id;
1953     let trait_def_id = tcx.trait_of_item(item_def_id).unwrap();
1954     let (term, obligations) = if lang_items.discriminant_kind_trait() == Some(trait_def_id) {
1955         let discriminant_def_id = tcx.require_lang_item(LangItem::Discriminant, None);
1956         assert_eq!(discriminant_def_id, item_def_id);
1957
1958         (self_ty.discriminant_ty(tcx).into(), Vec::new())
1959     } else if lang_items.pointee_trait() == Some(trait_def_id) {
1960         let metadata_def_id = tcx.require_lang_item(LangItem::Metadata, None);
1961         assert_eq!(metadata_def_id, item_def_id);
1962
1963         let mut obligations = Vec::new();
1964         let (metadata_ty, check_is_sized) = self_ty.ptr_metadata_ty(tcx, |ty| {
1965             normalize_with_depth_to(
1966                 selcx,
1967                 obligation.param_env,
1968                 obligation.cause.clone(),
1969                 obligation.recursion_depth + 1,
1970                 ty,
1971                 &mut obligations,
1972             )
1973         });
1974         if check_is_sized {
1975             let sized_predicate = ty::Binder::dummy(
1976                 tcx.at(obligation.cause.span()).mk_trait_ref(LangItem::Sized, [self_ty]),
1977             )
1978             .without_const();
1979             obligations.push(obligation.with(tcx, sized_predicate));
1980         }
1981         (metadata_ty.into(), obligations)
1982     } else {
1983         bug!("unexpected builtin trait with associated type: {:?}", obligation.predicate);
1984     };
1985
1986     let predicate =
1987         ty::ProjectionPredicate { projection_ty: ty::ProjectionTy { substs, item_def_id }, term };
1988
1989     confirm_param_env_candidate(selcx, obligation, ty::Binder::dummy(predicate), false)
1990         .with_addl_obligations(obligations)
1991         .with_addl_obligations(data.nested)
1992 }
1993
1994 fn confirm_fn_pointer_candidate<'cx, 'tcx>(
1995     selcx: &mut SelectionContext<'cx, 'tcx>,
1996     obligation: &ProjectionTyObligation<'tcx>,
1997     fn_pointer_impl_source: ImplSourceFnPointerData<'tcx, PredicateObligation<'tcx>>,
1998 ) -> Progress<'tcx> {
1999     let fn_type = selcx.infcx.shallow_resolve(fn_pointer_impl_source.fn_ty);
2000     let sig = fn_type.fn_sig(selcx.tcx());
2001     let Normalized { value: sig, obligations } = normalize_with_depth(
2002         selcx,
2003         obligation.param_env,
2004         obligation.cause.clone(),
2005         obligation.recursion_depth + 1,
2006         sig,
2007     );
2008
2009     confirm_callable_candidate(selcx, obligation, sig, util::TupleArgumentsFlag::Yes)
2010         .with_addl_obligations(fn_pointer_impl_source.nested)
2011         .with_addl_obligations(obligations)
2012 }
2013
2014 fn confirm_closure_candidate<'cx, 'tcx>(
2015     selcx: &mut SelectionContext<'cx, 'tcx>,
2016     obligation: &ProjectionTyObligation<'tcx>,
2017     impl_source: ImplSourceClosureData<'tcx, PredicateObligation<'tcx>>,
2018 ) -> Progress<'tcx> {
2019     let closure_sig = impl_source.substs.as_closure().sig();
2020     let Normalized { value: closure_sig, obligations } = normalize_with_depth(
2021         selcx,
2022         obligation.param_env,
2023         obligation.cause.clone(),
2024         obligation.recursion_depth + 1,
2025         closure_sig,
2026     );
2027
2028     debug!(?obligation, ?closure_sig, ?obligations, "confirm_closure_candidate");
2029
2030     confirm_callable_candidate(selcx, obligation, closure_sig, util::TupleArgumentsFlag::No)
2031         .with_addl_obligations(impl_source.nested)
2032         .with_addl_obligations(obligations)
2033 }
2034
2035 fn confirm_callable_candidate<'cx, 'tcx>(
2036     selcx: &mut SelectionContext<'cx, 'tcx>,
2037     obligation: &ProjectionTyObligation<'tcx>,
2038     fn_sig: ty::PolyFnSig<'tcx>,
2039     flag: util::TupleArgumentsFlag,
2040 ) -> Progress<'tcx> {
2041     let tcx = selcx.tcx();
2042
2043     debug!(?obligation, ?fn_sig, "confirm_callable_candidate");
2044
2045     let fn_once_def_id = tcx.require_lang_item(LangItem::FnOnce, None);
2046     let fn_once_output_def_id = tcx.require_lang_item(LangItem::FnOnceOutput, None);
2047
2048     let predicate = super::util::closure_trait_ref_and_return_type(
2049         tcx,
2050         fn_once_def_id,
2051         obligation.predicate.self_ty(),
2052         fn_sig,
2053         flag,
2054     )
2055     .map_bound(|(trait_ref, ret_type)| ty::ProjectionPredicate {
2056         projection_ty: ty::ProjectionTy {
2057             substs: trait_ref.substs,
2058             item_def_id: fn_once_output_def_id,
2059         },
2060         term: ret_type.into(),
2061     });
2062
2063     confirm_param_env_candidate(selcx, obligation, predicate, true)
2064 }
2065
2066 fn confirm_param_env_candidate<'cx, 'tcx>(
2067     selcx: &mut SelectionContext<'cx, 'tcx>,
2068     obligation: &ProjectionTyObligation<'tcx>,
2069     poly_cache_entry: ty::PolyProjectionPredicate<'tcx>,
2070     potentially_unnormalized_candidate: bool,
2071 ) -> Progress<'tcx> {
2072     let infcx = selcx.infcx;
2073     let cause = &obligation.cause;
2074     let param_env = obligation.param_env;
2075
2076     let cache_entry = infcx.replace_bound_vars_with_fresh_vars(
2077         cause.span,
2078         LateBoundRegionConversionTime::HigherRankedType,
2079         poly_cache_entry,
2080     );
2081
2082     let cache_projection = cache_entry.projection_ty;
2083     let mut nested_obligations = Vec::new();
2084     let obligation_projection = obligation.predicate;
2085     let obligation_projection = ensure_sufficient_stack(|| {
2086         normalize_with_depth_to(
2087             selcx,
2088             obligation.param_env,
2089             obligation.cause.clone(),
2090             obligation.recursion_depth + 1,
2091             obligation_projection,
2092             &mut nested_obligations,
2093         )
2094     });
2095     let cache_projection = if potentially_unnormalized_candidate {
2096         ensure_sufficient_stack(|| {
2097             normalize_with_depth_to(
2098                 selcx,
2099                 obligation.param_env,
2100                 obligation.cause.clone(),
2101                 obligation.recursion_depth + 1,
2102                 cache_projection,
2103                 &mut nested_obligations,
2104             )
2105         })
2106     } else {
2107         cache_projection
2108     };
2109
2110     debug!(?cache_projection, ?obligation_projection);
2111
2112     match infcx.at(cause, param_env).eq(cache_projection, obligation_projection) {
2113         Ok(InferOk { value: _, obligations }) => {
2114             nested_obligations.extend(obligations);
2115             assoc_ty_own_obligations(selcx, obligation, &mut nested_obligations);
2116             // FIXME(associated_const_equality): Handle consts here as well? Maybe this progress type should just take
2117             // a term instead.
2118             Progress { term: cache_entry.term, obligations: nested_obligations }
2119         }
2120         Err(e) => {
2121             let msg = format!(
2122                 "Failed to unify obligation `{:?}` with poly_projection `{:?}`: {:?}",
2123                 obligation, poly_cache_entry, e,
2124             );
2125             debug!("confirm_param_env_candidate: {}", msg);
2126             let err = infcx.tcx.ty_error_with_message(obligation.cause.span, &msg);
2127             Progress { term: err.into(), obligations: vec![] }
2128         }
2129     }
2130 }
2131
2132 fn confirm_impl_candidate<'cx, 'tcx>(
2133     selcx: &mut SelectionContext<'cx, 'tcx>,
2134     obligation: &ProjectionTyObligation<'tcx>,
2135     impl_impl_source: ImplSourceUserDefinedData<'tcx, PredicateObligation<'tcx>>,
2136 ) -> Progress<'tcx> {
2137     let tcx = selcx.tcx();
2138
2139     let ImplSourceUserDefinedData { impl_def_id, substs, mut nested } = impl_impl_source;
2140     let assoc_item_id = obligation.predicate.item_def_id;
2141     let trait_def_id = tcx.trait_id_of_impl(impl_def_id).unwrap();
2142
2143     let param_env = obligation.param_env;
2144     let Ok(assoc_ty) = assoc_def(selcx, impl_def_id, assoc_item_id) else {
2145         return Progress { term: tcx.ty_error().into(), obligations: nested };
2146     };
2147
2148     if !assoc_ty.item.defaultness(tcx).has_value() {
2149         // This means that the impl is missing a definition for the
2150         // associated type. This error will be reported by the type
2151         // checker method `check_impl_items_against_trait`, so here we
2152         // just return Error.
2153         debug!(
2154             "confirm_impl_candidate: no associated type {:?} for {:?}",
2155             assoc_ty.item.name, obligation.predicate
2156         );
2157         return Progress { term: tcx.ty_error().into(), obligations: nested };
2158     }
2159     // If we're trying to normalize `<Vec<u32> as X>::A<S>` using
2160     //`impl<T> X for Vec<T> { type A<Y> = Box<Y>; }`, then:
2161     //
2162     // * `obligation.predicate.substs` is `[Vec<u32>, S]`
2163     // * `substs` is `[u32]`
2164     // * `substs` ends up as `[u32, S]`
2165     let substs = obligation.predicate.substs.rebase_onto(tcx, trait_def_id, substs);
2166     let substs =
2167         translate_substs(selcx.infcx, param_env, impl_def_id, substs, assoc_ty.defining_node);
2168     let ty = tcx.bound_type_of(assoc_ty.item.def_id);
2169     let is_const = matches!(tcx.def_kind(assoc_ty.item.def_id), DefKind::AssocConst);
2170     let term: ty::EarlyBinder<ty::Term<'tcx>> = if is_const {
2171         let identity_substs =
2172             crate::traits::InternalSubsts::identity_for_item(tcx, assoc_ty.item.def_id);
2173         let did = ty::WithOptConstParam::unknown(assoc_ty.item.def_id);
2174         let kind = ty::ConstKind::Unevaluated(ty::UnevaluatedConst::new(did, identity_substs));
2175         ty.map_bound(|ty| tcx.mk_const(kind, ty).into())
2176     } else {
2177         ty.map_bound(|ty| ty.into())
2178     };
2179     if !check_substs_compatible(tcx, &assoc_ty.item, substs) {
2180         let err = tcx.ty_error_with_message(
2181             obligation.cause.span,
2182             "impl item and trait item have different parameters",
2183         );
2184         Progress { term: err.into(), obligations: nested }
2185     } else {
2186         assoc_ty_own_obligations(selcx, obligation, &mut nested);
2187         Progress { term: term.subst(tcx, substs), obligations: nested }
2188     }
2189 }
2190
2191 // Verify that the trait item and its implementation have compatible substs lists
2192 fn check_substs_compatible<'tcx>(
2193     tcx: TyCtxt<'tcx>,
2194     assoc_item: &ty::AssocItem,
2195     substs: ty::SubstsRef<'tcx>,
2196 ) -> bool {
2197     fn check_substs_compatible_inner<'tcx>(
2198         tcx: TyCtxt<'tcx>,
2199         generics: &'tcx ty::Generics,
2200         args: &'tcx [ty::GenericArg<'tcx>],
2201     ) -> bool {
2202         if generics.count() != args.len() {
2203             return false;
2204         }
2205
2206         let (parent_args, own_args) = args.split_at(generics.parent_count);
2207
2208         if let Some(parent) = generics.parent
2209             && let parent_generics = tcx.generics_of(parent)
2210             && !check_substs_compatible_inner(tcx, parent_generics, parent_args) {
2211             return false;
2212         }
2213
2214         for (param, arg) in std::iter::zip(&generics.params, own_args) {
2215             match (&param.kind, arg.unpack()) {
2216                 (ty::GenericParamDefKind::Type { .. }, ty::GenericArgKind::Type(_))
2217                 | (ty::GenericParamDefKind::Lifetime, ty::GenericArgKind::Lifetime(_))
2218                 | (ty::GenericParamDefKind::Const { .. }, ty::GenericArgKind::Const(_)) => {}
2219                 _ => return false,
2220             }
2221         }
2222
2223         true
2224     }
2225
2226     let generics = tcx.generics_of(assoc_item.def_id);
2227     // Chop off any additional substs (RPITIT) substs
2228     let substs = &substs[0..generics.count().min(substs.len())];
2229     check_substs_compatible_inner(tcx, generics, substs)
2230 }
2231
2232 fn confirm_impl_trait_in_trait_candidate<'tcx>(
2233     selcx: &mut SelectionContext<'_, 'tcx>,
2234     obligation: &ProjectionTyObligation<'tcx>,
2235     data: ImplSourceUserDefinedData<'tcx, PredicateObligation<'tcx>>,
2236 ) -> Progress<'tcx> {
2237     let tcx = selcx.tcx();
2238     let mut obligations = data.nested;
2239
2240     let trait_fn_def_id = tcx.impl_trait_in_trait_parent(obligation.predicate.item_def_id);
2241     let Ok(leaf_def) = assoc_def(selcx, data.impl_def_id, trait_fn_def_id) else {
2242         return Progress { term: tcx.ty_error().into(), obligations };
2243     };
2244     if !leaf_def.item.defaultness(tcx).has_value() {
2245         return Progress { term: tcx.ty_error().into(), obligations };
2246     }
2247
2248     // Use the default `impl Trait` for the trait, e.g., for a default trait body
2249     if leaf_def.item.container == ty::AssocItemContainer::TraitContainer {
2250         return Progress {
2251             term: tcx
2252                 .mk_opaque(obligation.predicate.item_def_id, obligation.predicate.substs)
2253                 .into(),
2254             obligations,
2255         };
2256     }
2257
2258     // Rebase from {trait}::{fn}::{opaque} to {impl}::{fn}::{opaque},
2259     // since `data.substs` are the impl substs.
2260     let impl_fn_substs =
2261         obligation.predicate.substs.rebase_onto(tcx, tcx.parent(trait_fn_def_id), data.substs);
2262     let impl_fn_substs = translate_substs(
2263         selcx.infcx,
2264         obligation.param_env,
2265         data.impl_def_id,
2266         impl_fn_substs,
2267         leaf_def.defining_node,
2268     );
2269
2270     if !check_substs_compatible(tcx, &leaf_def.item, impl_fn_substs) {
2271         let err = tcx.ty_error_with_message(
2272             obligation.cause.span,
2273             "impl method and trait method have different parameters",
2274         );
2275         return Progress { term: err.into(), obligations };
2276     }
2277
2278     let impl_fn_def_id = leaf_def.item.def_id;
2279
2280     let cause = ObligationCause::new(
2281         obligation.cause.span,
2282         obligation.cause.body_id,
2283         super::ItemObligation(impl_fn_def_id),
2284     );
2285     let predicates = normalize_with_depth_to(
2286         selcx,
2287         obligation.param_env,
2288         cause.clone(),
2289         obligation.recursion_depth + 1,
2290         tcx.predicates_of(impl_fn_def_id).instantiate(tcx, impl_fn_substs),
2291         &mut obligations,
2292     );
2293     obligations.extend(std::iter::zip(predicates.predicates, predicates.spans).map(
2294         |(pred, span)| {
2295             Obligation::with_depth(
2296                 tcx,
2297                 ObligationCause::new(
2298                     obligation.cause.span,
2299                     obligation.cause.body_id,
2300                     if span.is_dummy() {
2301                         super::ItemObligation(impl_fn_def_id)
2302                     } else {
2303                         super::BindingObligation(impl_fn_def_id, span)
2304                     },
2305                 ),
2306                 obligation.recursion_depth + 1,
2307                 obligation.param_env,
2308                 pred,
2309             )
2310         },
2311     ));
2312
2313     let ty = super::normalize_to(
2314         selcx,
2315         obligation.param_env,
2316         cause.clone(),
2317         tcx.bound_trait_impl_trait_tys(impl_fn_def_id)
2318             .map_bound(|tys| {
2319                 tys.map_or_else(|_| tcx.ty_error(), |tys| tys[&obligation.predicate.item_def_id])
2320             })
2321             .subst(tcx, impl_fn_substs),
2322         &mut obligations,
2323     );
2324
2325     Progress { term: ty.into(), obligations }
2326 }
2327
2328 // Get obligations corresponding to the predicates from the where-clause of the
2329 // associated type itself.
2330 fn assoc_ty_own_obligations<'cx, 'tcx>(
2331     selcx: &mut SelectionContext<'cx, 'tcx>,
2332     obligation: &ProjectionTyObligation<'tcx>,
2333     nested: &mut Vec<PredicateObligation<'tcx>>,
2334 ) {
2335     let tcx = selcx.tcx();
2336     for predicate in tcx
2337         .predicates_of(obligation.predicate.item_def_id)
2338         .instantiate_own(tcx, obligation.predicate.substs)
2339         .predicates
2340     {
2341         let normalized = normalize_with_depth_to(
2342             selcx,
2343             obligation.param_env,
2344             obligation.cause.clone(),
2345             obligation.recursion_depth + 1,
2346             predicate,
2347             nested,
2348         );
2349         nested.push(Obligation::with_depth(
2350             tcx,
2351             obligation.cause.clone(),
2352             obligation.recursion_depth + 1,
2353             obligation.param_env,
2354             normalized,
2355         ));
2356     }
2357 }
2358
2359 /// Locate the definition of an associated type in the specialization hierarchy,
2360 /// starting from the given impl.
2361 ///
2362 /// Based on the "projection mode", this lookup may in fact only examine the
2363 /// topmost impl. See the comments for `Reveal` for more details.
2364 fn assoc_def(
2365     selcx: &SelectionContext<'_, '_>,
2366     impl_def_id: DefId,
2367     assoc_def_id: DefId,
2368 ) -> Result<specialization_graph::LeafDef, ErrorGuaranteed> {
2369     let tcx = selcx.tcx();
2370     let trait_def_id = tcx.impl_trait_ref(impl_def_id).unwrap().def_id;
2371     let trait_def = tcx.trait_def(trait_def_id);
2372
2373     // This function may be called while we are still building the
2374     // specialization graph that is queried below (via TraitDef::ancestors()),
2375     // so, in order to avoid unnecessary infinite recursion, we manually look
2376     // for the associated item at the given impl.
2377     // If there is no such item in that impl, this function will fail with a
2378     // cycle error if the specialization graph is currently being built.
2379     if let Some(&impl_item_id) = tcx.impl_item_implementor_ids(impl_def_id).get(&assoc_def_id) {
2380         let item = tcx.associated_item(impl_item_id);
2381         let impl_node = specialization_graph::Node::Impl(impl_def_id);
2382         return Ok(specialization_graph::LeafDef {
2383             item: *item,
2384             defining_node: impl_node,
2385             finalizing_node: if item.defaultness(tcx).is_default() {
2386                 None
2387             } else {
2388                 Some(impl_node)
2389             },
2390         });
2391     }
2392
2393     let ancestors = trait_def.ancestors(tcx, impl_def_id)?;
2394     if let Some(assoc_item) = ancestors.leaf_def(tcx, assoc_def_id) {
2395         Ok(assoc_item)
2396     } else {
2397         // This is saying that neither the trait nor
2398         // the impl contain a definition for this
2399         // associated type.  Normally this situation
2400         // could only arise through a compiler bug --
2401         // if the user wrote a bad item name, it
2402         // should have failed in astconv.
2403         bug!(
2404             "No associated type `{}` for {}",
2405             tcx.item_name(assoc_def_id),
2406             tcx.def_path_str(impl_def_id)
2407         )
2408     }
2409 }
2410
2411 pub(crate) trait ProjectionCacheKeyExt<'cx, 'tcx>: Sized {
2412     fn from_poly_projection_predicate(
2413         selcx: &mut SelectionContext<'cx, 'tcx>,
2414         predicate: ty::PolyProjectionPredicate<'tcx>,
2415     ) -> Option<Self>;
2416 }
2417
2418 impl<'cx, 'tcx> ProjectionCacheKeyExt<'cx, 'tcx> for ProjectionCacheKey<'tcx> {
2419     fn from_poly_projection_predicate(
2420         selcx: &mut SelectionContext<'cx, 'tcx>,
2421         predicate: ty::PolyProjectionPredicate<'tcx>,
2422     ) -> Option<Self> {
2423         let infcx = selcx.infcx;
2424         // We don't do cross-snapshot caching of obligations with escaping regions,
2425         // so there's no cache key to use
2426         predicate.no_bound_vars().map(|predicate| {
2427             ProjectionCacheKey::new(
2428                 // We don't attempt to match up with a specific type-variable state
2429                 // from a specific call to `opt_normalize_projection_type` - if
2430                 // there's no precise match, the original cache entry is "stranded"
2431                 // anyway.
2432                 infcx.resolve_vars_if_possible(predicate.projection_ty),
2433             )
2434         })
2435     }
2436 }