]> git.lizzy.rs Git - rust.git/blob - src/librustc_trait_selection/traits/auto_trait.rs
introduce PredicateAtom
[rust.git] / src / librustc_trait_selection / traits / auto_trait.rs
1 //! Support code for rustdoc and external tools.
2 //! You really don't want to be using this unless you need to.
3
4 use super::*;
5
6 use crate::infer::region_constraints::{Constraint, RegionConstraintData};
7 use crate::infer::InferCtxt;
8 use rustc_middle::ty::fold::TypeFolder;
9 use rustc_middle::ty::{Region, RegionVid};
10
11 use rustc_data_structures::fx::{FxHashMap, FxHashSet};
12
13 use std::collections::hash_map::Entry;
14 use std::collections::VecDeque;
15
16 // FIXME(twk): this is obviously not nice to duplicate like that
17 #[derive(Eq, PartialEq, Hash, Copy, Clone, Debug)]
18 pub enum RegionTarget<'tcx> {
19     Region(Region<'tcx>),
20     RegionVid(RegionVid),
21 }
22
23 #[derive(Default, Debug, Clone)]
24 pub struct RegionDeps<'tcx> {
25     larger: FxHashSet<RegionTarget<'tcx>>,
26     smaller: FxHashSet<RegionTarget<'tcx>>,
27 }
28
29 pub enum AutoTraitResult<A> {
30     ExplicitImpl,
31     PositiveImpl(A),
32     NegativeImpl,
33 }
34
35 impl<A> AutoTraitResult<A> {
36     fn is_auto(&self) -> bool {
37         match *self {
38             AutoTraitResult::PositiveImpl(_) | AutoTraitResult::NegativeImpl => true,
39             _ => false,
40         }
41     }
42 }
43
44 pub struct AutoTraitInfo<'cx> {
45     pub full_user_env: ty::ParamEnv<'cx>,
46     pub region_data: RegionConstraintData<'cx>,
47     pub vid_to_region: FxHashMap<ty::RegionVid, ty::Region<'cx>>,
48 }
49
50 pub struct AutoTraitFinder<'tcx> {
51     tcx: TyCtxt<'tcx>,
52 }
53
54 impl<'tcx> AutoTraitFinder<'tcx> {
55     pub fn new(tcx: TyCtxt<'tcx>) -> Self {
56         AutoTraitFinder { tcx }
57     }
58
59     /// Makes a best effort to determine whether and under which conditions an auto trait is
60     /// implemented for a type. For example, if you have
61     ///
62     /// ```
63     /// struct Foo<T> { data: Box<T> }
64     /// ```
65     ///
66     /// then this might return that Foo<T>: Send if T: Send (encoded in the AutoTraitResult type).
67     /// The analysis attempts to account for custom impls as well as other complex cases. This
68     /// result is intended for use by rustdoc and other such consumers.
69     ///
70     /// (Note that due to the coinductive nature of Send, the full and correct result is actually
71     /// quite simple to generate. That is, when a type has no custom impl, it is Send iff its field
72     /// types are all Send. So, in our example, we might have that Foo<T>: Send if Box<T>: Send.
73     /// But this is often not the best way to present to the user.)
74     ///
75     /// Warning: The API should be considered highly unstable, and it may be refactored or removed
76     /// in the future.
77     pub fn find_auto_trait_generics<A>(
78         &self,
79         ty: Ty<'tcx>,
80         orig_env: ty::ParamEnv<'tcx>,
81         trait_did: DefId,
82         auto_trait_callback: impl Fn(&InferCtxt<'_, 'tcx>, AutoTraitInfo<'tcx>) -> A,
83     ) -> AutoTraitResult<A> {
84         let tcx = self.tcx;
85
86         let trait_ref = ty::TraitRef { def_id: trait_did, substs: tcx.mk_substs_trait(ty, &[]) };
87
88         let trait_pred = ty::Binder::bind(trait_ref);
89
90         let bail_out = tcx.infer_ctxt().enter(|infcx| {
91             let mut selcx = SelectionContext::with_negative(&infcx, true);
92             let result = selcx.select(&Obligation::new(
93                 ObligationCause::dummy(),
94                 orig_env,
95                 trait_pred.to_poly_trait_predicate(),
96             ));
97
98             match result {
99                 Ok(Some(ImplSource::ImplSourceUserDefined(_))) => {
100                     debug!(
101                         "find_auto_trait_generics({:?}): \
102                          manual impl found, bailing out",
103                         trait_ref
104                     );
105                     true
106                 }
107                 _ => false,
108             }
109         });
110
111         // If an explicit impl exists, it always takes priority over an auto impl
112         if bail_out {
113             return AutoTraitResult::ExplicitImpl;
114         }
115
116         tcx.infer_ctxt().enter(|infcx| {
117             let mut fresh_preds = FxHashSet::default();
118
119             // Due to the way projections are handled by SelectionContext, we need to run
120             // evaluate_predicates twice: once on the original param env, and once on the result of
121             // the first evaluate_predicates call.
122             //
123             // The problem is this: most of rustc, including SelectionContext and traits::project,
124             // are designed to work with a concrete usage of a type (e.g., Vec<u8>
125             // fn<T>() { Vec<T> }. This information will generally never change - given
126             // the 'T' in fn<T>() { ... }, we'll never know anything else about 'T'.
127             // If we're unable to prove that 'T' implements a particular trait, we're done -
128             // there's nothing left to do but error out.
129             //
130             // However, synthesizing an auto trait impl works differently. Here, we start out with
131             // a set of initial conditions - the ParamEnv of the struct/enum/union we're dealing
132             // with - and progressively discover the conditions we need to fulfill for it to
133             // implement a certain auto trait. This ends up breaking two assumptions made by trait
134             // selection and projection:
135             //
136             // * We can always cache the result of a particular trait selection for the lifetime of
137             // an InfCtxt
138             // * Given a projection bound such as '<T as SomeTrait>::SomeItem = K', if 'T:
139             // SomeTrait' doesn't hold, then we don't need to care about the 'SomeItem = K'
140             //
141             // We fix the first assumption by manually clearing out all of the InferCtxt's caches
142             // in between calls to SelectionContext.select. This allows us to keep all of the
143             // intermediate types we create bound to the 'tcx lifetime, rather than needing to lift
144             // them between calls.
145             //
146             // We fix the second assumption by reprocessing the result of our first call to
147             // evaluate_predicates. Using the example of '<T as SomeTrait>::SomeItem = K', our first
148             // pass will pick up 'T: SomeTrait', but not 'SomeItem = K'. On our second pass,
149             // traits::project will see that 'T: SomeTrait' is in our ParamEnv, allowing
150             // SelectionContext to return it back to us.
151
152             let (new_env, user_env) = match self.evaluate_predicates(
153                 &infcx,
154                 trait_did,
155                 ty,
156                 orig_env,
157                 orig_env,
158                 &mut fresh_preds,
159                 false,
160             ) {
161                 Some(e) => e,
162                 None => return AutoTraitResult::NegativeImpl,
163             };
164
165             let (full_env, full_user_env) = self
166                 .evaluate_predicates(
167                     &infcx,
168                     trait_did,
169                     ty,
170                     new_env,
171                     user_env,
172                     &mut fresh_preds,
173                     true,
174                 )
175                 .unwrap_or_else(|| {
176                     panic!("Failed to fully process: {:?} {:?} {:?}", ty, trait_did, orig_env)
177                 });
178
179             debug!(
180                 "find_auto_trait_generics({:?}): fulfilling \
181                  with {:?}",
182                 trait_ref, full_env
183             );
184             infcx.clear_caches();
185
186             // At this point, we already have all of the bounds we need. FulfillmentContext is used
187             // to store all of the necessary region/lifetime bounds in the InferContext, as well as
188             // an additional sanity check.
189             let mut fulfill = FulfillmentContext::new();
190             fulfill.register_bound(&infcx, full_env, ty, trait_did, ObligationCause::dummy());
191             fulfill.select_all_or_error(&infcx).unwrap_or_else(|e| {
192                 panic!("Unable to fulfill trait {:?} for '{:?}': {:?}", trait_did, ty, e)
193             });
194
195             let body_id_map: FxHashMap<_, _> = infcx
196                 .inner
197                 .borrow()
198                 .region_obligations()
199                 .iter()
200                 .map(|&(id, _)| (id, vec![]))
201                 .collect();
202
203             infcx.process_registered_region_obligations(&body_id_map, None, full_env);
204
205             let region_data = infcx
206                 .inner
207                 .borrow_mut()
208                 .unwrap_region_constraints()
209                 .region_constraint_data()
210                 .clone();
211
212             let vid_to_region = self.map_vid_to_region(&region_data);
213
214             let info = AutoTraitInfo { full_user_env, region_data, vid_to_region };
215
216             AutoTraitResult::PositiveImpl(auto_trait_callback(&infcx, info))
217         })
218     }
219 }
220
221 impl AutoTraitFinder<'tcx> {
222     /// The core logic responsible for computing the bounds for our synthesized impl.
223     ///
224     /// To calculate the bounds, we call `SelectionContext.select` in a loop. Like
225     /// `FulfillmentContext`, we recursively select the nested obligations of predicates we
226     /// encounter. However, whenever we encounter an `UnimplementedError` involving a type
227     /// parameter, we add it to our `ParamEnv`. Since our goal is to determine when a particular
228     /// type implements an auto trait, Unimplemented errors tell us what conditions need to be met.
229     ///
230     /// This method ends up working somewhat similarly to `FulfillmentContext`, but with a few key
231     /// differences. `FulfillmentContext` works under the assumption that it's dealing with concrete
232     /// user code. According, it considers all possible ways that a `Predicate` could be met, which
233     /// isn't always what we want for a synthesized impl. For example, given the predicate `T:
234     /// Iterator`, `FulfillmentContext` can end up reporting an Unimplemented error for `T:
235     /// IntoIterator` -- since there's an implementation of `Iterator` where `T: IntoIterator`,
236     /// `FulfillmentContext` will drive `SelectionContext` to consider that impl before giving up.
237     /// If we were to rely on `FulfillmentContext`s decision, we might end up synthesizing an impl
238     /// like this:
239     ///
240     ///     impl<T> Send for Foo<T> where T: IntoIterator
241     ///
242     /// While it might be technically true that Foo implements Send where `T: IntoIterator`,
243     /// the bound is overly restrictive - it's really only necessary that `T: Iterator`.
244     ///
245     /// For this reason, `evaluate_predicates` handles predicates with type variables specially.
246     /// When we encounter an `Unimplemented` error for a bound such as `T: Iterator`, we immediately
247     /// add it to our `ParamEnv`, and add it to our stack for recursive evaluation. When we later
248     /// select it, we'll pick up any nested bounds, without ever inferring that `T: IntoIterator`
249     /// needs to hold.
250     ///
251     /// One additional consideration is supertrait bounds. Normally, a `ParamEnv` is only ever
252     /// constructed once for a given type. As part of the construction process, the `ParamEnv` will
253     /// have any supertrait bounds normalized -- e.g., if we have a type `struct Foo<T: Copy>`, the
254     /// `ParamEnv` will contain `T: Copy` and `T: Clone`, since `Copy: Clone`. When we construct our
255     /// own `ParamEnv`, we need to do this ourselves, through `traits::elaborate_predicates`, or
256     /// else `SelectionContext` will choke on the missing predicates. However, this should never
257     /// show up in the final synthesized generics: we don't want our generated docs page to contain
258     /// something like `T: Copy + Clone`, as that's redundant. Therefore, we keep track of a
259     /// separate `user_env`, which only holds the predicates that will actually be displayed to the
260     /// user.
261     fn evaluate_predicates(
262         &self,
263         infcx: &InferCtxt<'_, 'tcx>,
264         trait_did: DefId,
265         ty: Ty<'tcx>,
266         param_env: ty::ParamEnv<'tcx>,
267         user_env: ty::ParamEnv<'tcx>,
268         fresh_preds: &mut FxHashSet<ty::Predicate<'tcx>>,
269         only_projections: bool,
270     ) -> Option<(ty::ParamEnv<'tcx>, ty::ParamEnv<'tcx>)> {
271         let tcx = infcx.tcx;
272
273         let mut select = SelectionContext::with_negative(&infcx, true);
274
275         let mut already_visited = FxHashSet::default();
276         let mut predicates = VecDeque::new();
277         predicates.push_back(ty::Binder::bind(ty::TraitPredicate {
278             trait_ref: ty::TraitRef {
279                 def_id: trait_did,
280                 substs: infcx.tcx.mk_substs_trait(ty, &[]),
281             },
282         }));
283
284         let computed_preds = param_env.caller_bounds().iter();
285         let mut user_computed_preds: FxHashSet<_> = user_env.caller_bounds().iter().collect();
286
287         let mut new_env = param_env;
288         let dummy_cause = ObligationCause::dummy();
289
290         while let Some(pred) = predicates.pop_front() {
291             infcx.clear_caches();
292
293             if !already_visited.insert(pred) {
294                 continue;
295             }
296
297             // Call `infcx.resolve_vars_if_possible` to see if we can
298             // get rid of any inference variables.
299             let obligation = infcx.resolve_vars_if_possible(&Obligation::new(
300                 dummy_cause.clone(),
301                 new_env,
302                 pred,
303             ));
304             let result = select.select(&obligation);
305
306             match &result {
307                 &Ok(Some(ref impl_source)) => {
308                     // If we see an explicit negative impl (e.g., `impl !Send for MyStruct`),
309                     // we immediately bail out, since it's impossible for us to continue.
310
311                     if let ImplSource::ImplSourceUserDefined(ImplSourceUserDefinedData {
312                         impl_def_id,
313                         ..
314                     }) = impl_source
315                     {
316                         // Blame 'tidy' for the weird bracket placement.
317                         if infcx.tcx.impl_polarity(*impl_def_id) == ty::ImplPolarity::Negative {
318                             debug!(
319                                 "evaluate_nested_obligations: found explicit negative impl\
320                                         {:?}, bailing out",
321                                 impl_def_id
322                             );
323                             return None;
324                         }
325                     }
326
327                     let obligations = impl_source.clone().nested_obligations().into_iter();
328
329                     if !self.evaluate_nested_obligations(
330                         ty,
331                         obligations,
332                         &mut user_computed_preds,
333                         fresh_preds,
334                         &mut predicates,
335                         &mut select,
336                         only_projections,
337                     ) {
338                         return None;
339                     }
340                 }
341                 &Ok(None) => {}
342                 &Err(SelectionError::Unimplemented) => {
343                     if self.is_param_no_infer(pred.skip_binder().trait_ref.substs) {
344                         already_visited.remove(&pred);
345                         self.add_user_pred(
346                             &mut user_computed_preds,
347                             pred.without_const().to_predicate(self.tcx),
348                         );
349                         predicates.push_back(pred);
350                     } else {
351                         debug!(
352                             "evaluate_nested_obligations: `Unimplemented` found, bailing: \
353                              {:?} {:?} {:?}",
354                             ty,
355                             pred,
356                             pred.skip_binder().trait_ref.substs
357                         );
358                         return None;
359                     }
360                 }
361                 _ => panic!("Unexpected error for '{:?}': {:?}", ty, result),
362             };
363
364             let normalized_preds = elaborate_predicates(
365                 tcx,
366                 computed_preds.clone().chain(user_computed_preds.iter().cloned()),
367             )
368             .map(|o| o.predicate);
369             new_env =
370                 ty::ParamEnv::new(tcx.mk_predicates(normalized_preds), param_env.reveal(), None);
371         }
372
373         let final_user_env = ty::ParamEnv::new(
374             tcx.mk_predicates(user_computed_preds.into_iter()),
375             user_env.reveal(),
376             None,
377         );
378         debug!(
379             "evaluate_nested_obligations(ty={:?}, trait_did={:?}): succeeded with '{:?}' \
380              '{:?}'",
381             ty, trait_did, new_env, final_user_env
382         );
383
384         Some((new_env, final_user_env))
385     }
386
387     /// This method is designed to work around the following issue:
388     /// When we compute auto trait bounds, we repeatedly call `SelectionContext.select`,
389     /// progressively building a `ParamEnv` based on the results we get.
390     /// However, our usage of `SelectionContext` differs from its normal use within the compiler,
391     /// in that we capture and re-reprocess predicates from `Unimplemented` errors.
392     ///
393     /// This can lead to a corner case when dealing with region parameters.
394     /// During our selection loop in `evaluate_predicates`, we might end up with
395     /// two trait predicates that differ only in their region parameters:
396     /// one containing a HRTB lifetime parameter, and one containing a 'normal'
397     /// lifetime parameter. For example:
398     ///
399     ///     T as MyTrait<'a>
400     ///     T as MyTrait<'static>
401     ///
402     /// If we put both of these predicates in our computed `ParamEnv`, we'll
403     /// confuse `SelectionContext`, since it will (correctly) view both as being applicable.
404     ///
405     /// To solve this, we pick the 'more strict' lifetime bound -- i.e., the HRTB
406     /// Our end goal is to generate a user-visible description of the conditions
407     /// under which a type implements an auto trait. A trait predicate involving
408     /// a HRTB means that the type needs to work with any choice of lifetime,
409     /// not just one specific lifetime (e.g., `'static`).
410     fn add_user_pred(
411         &self,
412         user_computed_preds: &mut FxHashSet<ty::Predicate<'tcx>>,
413         new_pred: ty::Predicate<'tcx>,
414     ) {
415         let mut should_add_new = true;
416         user_computed_preds.retain(|&old_pred| {
417             if let (
418                 ty::PredicateAtom::Trait(new_trait, _),
419                 ty::PredicateAtom::Trait(old_trait, _),
420             ) = (new_pred.skip_binders(), old_pred.skip_binders())
421             {
422                 if new_trait.def_id() == old_trait.def_id() {
423                     let new_substs = new_trait.trait_ref.substs;
424                     let old_substs = old_trait.trait_ref.substs;
425
426                     if !new_substs.types().eq(old_substs.types()) {
427                         // We can't compare lifetimes if the types are different,
428                         // so skip checking `old_pred`.
429                         return true;
430                     }
431
432                     for (new_region, old_region) in new_substs.regions().zip(old_substs.regions()) {
433                         match (new_region, old_region) {
434                             // If both predicates have an `ReLateBound` (a HRTB) in the
435                             // same spot, we do nothing.
436                             (
437                                 ty::RegionKind::ReLateBound(_, _),
438                                 ty::RegionKind::ReLateBound(_, _),
439                             ) => {}
440
441                             (ty::RegionKind::ReLateBound(_, _), _)
442                             | (_, ty::RegionKind::ReVar(_)) => {
443                                 // One of these is true:
444                                 // The new predicate has a HRTB in a spot where the old
445                                 // predicate does not (if they both had a HRTB, the previous
446                                 // match arm would have executed). A HRBT is a 'stricter'
447                                 // bound than anything else, so we want to keep the newer
448                                 // predicate (with the HRBT) in place of the old predicate.
449                                 //
450                                 // OR
451                                 //
452                                 // The old predicate has a region variable where the new
453                                 // predicate has some other kind of region. An region
454                                 // variable isn't something we can actually display to a user,
455                                 // so we choose their new predicate (which doesn't have a region
456                                 // variable).
457                                 //
458                                 // In both cases, we want to remove the old predicate,
459                                 // from `user_computed_preds`, and replace it with the new
460                                 // one. Having both the old and the new
461                                 // predicate in a `ParamEnv` would confuse `SelectionContext`.
462                                 //
463                                 // We're currently in the predicate passed to 'retain',
464                                 // so we return `false` to remove the old predicate from
465                                 // `user_computed_preds`.
466                                 return false;
467                             }
468                             (_, ty::RegionKind::ReLateBound(_, _))
469                             | (ty::RegionKind::ReVar(_), _) => {
470                                 // This is the opposite situation as the previous arm.
471                                 // One of these is true:
472                                 //
473                                 // The old predicate has a HRTB lifetime in a place where the
474                                 // new predicate does not.
475                                 //
476                                 // OR
477                                 //
478                                 // The new predicate has a region variable where the old
479                                 // predicate has some other type of region.
480                                 //
481                                 // We want to leave the old
482                                 // predicate in `user_computed_preds`, and skip adding
483                                 // new_pred to `user_computed_params`.
484                                 should_add_new = false
485                             }
486                             _ => {}
487                         }
488                     }
489                 }
490             }
491             true
492         });
493
494         if should_add_new {
495             user_computed_preds.insert(new_pred);
496         }
497     }
498
499     /// This is very similar to `handle_lifetimes`. However, instead of matching `ty::Region`s
500     /// to each other, we match `ty::RegionVid`s to `ty::Region`s.
501     fn map_vid_to_region<'cx>(
502         &self,
503         regions: &RegionConstraintData<'cx>,
504     ) -> FxHashMap<ty::RegionVid, ty::Region<'cx>> {
505         let mut vid_map: FxHashMap<RegionTarget<'cx>, RegionDeps<'cx>> = FxHashMap::default();
506         let mut finished_map = FxHashMap::default();
507
508         for constraint in regions.constraints.keys() {
509             match constraint {
510                 &Constraint::VarSubVar(r1, r2) => {
511                     {
512                         let deps1 = vid_map.entry(RegionTarget::RegionVid(r1)).or_default();
513                         deps1.larger.insert(RegionTarget::RegionVid(r2));
514                     }
515
516                     let deps2 = vid_map.entry(RegionTarget::RegionVid(r2)).or_default();
517                     deps2.smaller.insert(RegionTarget::RegionVid(r1));
518                 }
519                 &Constraint::RegSubVar(region, vid) => {
520                     {
521                         let deps1 = vid_map.entry(RegionTarget::Region(region)).or_default();
522                         deps1.larger.insert(RegionTarget::RegionVid(vid));
523                     }
524
525                     let deps2 = vid_map.entry(RegionTarget::RegionVid(vid)).or_default();
526                     deps2.smaller.insert(RegionTarget::Region(region));
527                 }
528                 &Constraint::VarSubReg(vid, region) => {
529                     finished_map.insert(vid, region);
530                 }
531                 &Constraint::RegSubReg(r1, r2) => {
532                     {
533                         let deps1 = vid_map.entry(RegionTarget::Region(r1)).or_default();
534                         deps1.larger.insert(RegionTarget::Region(r2));
535                     }
536
537                     let deps2 = vid_map.entry(RegionTarget::Region(r2)).or_default();
538                     deps2.smaller.insert(RegionTarget::Region(r1));
539                 }
540             }
541         }
542
543         while !vid_map.is_empty() {
544             let target = *vid_map.keys().next().expect("Keys somehow empty");
545             let deps = vid_map.remove(&target).expect("Entry somehow missing");
546
547             for smaller in deps.smaller.iter() {
548                 for larger in deps.larger.iter() {
549                     match (smaller, larger) {
550                         (&RegionTarget::Region(_), &RegionTarget::Region(_)) => {
551                             if let Entry::Occupied(v) = vid_map.entry(*smaller) {
552                                 let smaller_deps = v.into_mut();
553                                 smaller_deps.larger.insert(*larger);
554                                 smaller_deps.larger.remove(&target);
555                             }
556
557                             if let Entry::Occupied(v) = vid_map.entry(*larger) {
558                                 let larger_deps = v.into_mut();
559                                 larger_deps.smaller.insert(*smaller);
560                                 larger_deps.smaller.remove(&target);
561                             }
562                         }
563                         (&RegionTarget::RegionVid(v1), &RegionTarget::Region(r1)) => {
564                             finished_map.insert(v1, r1);
565                         }
566                         (&RegionTarget::Region(_), &RegionTarget::RegionVid(_)) => {
567                             // Do nothing; we don't care about regions that are smaller than vids.
568                         }
569                         (&RegionTarget::RegionVid(_), &RegionTarget::RegionVid(_)) => {
570                             if let Entry::Occupied(v) = vid_map.entry(*smaller) {
571                                 let smaller_deps = v.into_mut();
572                                 smaller_deps.larger.insert(*larger);
573                                 smaller_deps.larger.remove(&target);
574                             }
575
576                             if let Entry::Occupied(v) = vid_map.entry(*larger) {
577                                 let larger_deps = v.into_mut();
578                                 larger_deps.smaller.insert(*smaller);
579                                 larger_deps.smaller.remove(&target);
580                             }
581                         }
582                     }
583                 }
584             }
585         }
586         finished_map
587     }
588
589     fn is_param_no_infer(&self, substs: SubstsRef<'_>) -> bool {
590         self.is_of_param(substs.type_at(0)) && !substs.types().any(|t| t.has_infer_types())
591     }
592
593     pub fn is_of_param(&self, ty: Ty<'_>) -> bool {
594         match ty.kind {
595             ty::Param(_) => true,
596             ty::Projection(p) => self.is_of_param(p.self_ty()),
597             _ => false,
598         }
599     }
600
601     fn is_self_referential_projection(&self, p: ty::PolyProjectionPredicate<'_>) -> bool {
602         match p.ty().skip_binder().kind {
603             ty::Projection(proj) if proj == p.skip_binder().projection_ty => true,
604             _ => false,
605         }
606     }
607
608     fn evaluate_nested_obligations(
609         &self,
610         ty: Ty<'_>,
611         nested: impl Iterator<Item = Obligation<'tcx, ty::Predicate<'tcx>>>,
612         computed_preds: &mut FxHashSet<ty::Predicate<'tcx>>,
613         fresh_preds: &mut FxHashSet<ty::Predicate<'tcx>>,
614         predicates: &mut VecDeque<ty::PolyTraitPredicate<'tcx>>,
615         select: &mut SelectionContext<'_, 'tcx>,
616         only_projections: bool,
617     ) -> bool {
618         let dummy_cause = ObligationCause::dummy();
619
620         for obligation in nested {
621             let is_new_pred =
622                 fresh_preds.insert(self.clean_pred(select.infcx(), obligation.predicate));
623
624             // Resolve any inference variables that we can, to help selection succeed
625             let predicate = select.infcx().resolve_vars_if_possible(&obligation.predicate);
626
627             // We only add a predicate as a user-displayable bound if
628             // it involves a generic parameter, and doesn't contain
629             // any inference variables.
630             //
631             // Displaying a bound involving a concrete type (instead of a generic
632             // parameter) would be pointless, since it's always true
633             // (e.g. u8: Copy)
634             // Displaying an inference variable is impossible, since they're
635             // an internal compiler detail without a defined visual representation
636             //
637             // We check this by calling is_of_param on the relevant types
638             // from the various possible predicates
639
640             match predicate.skip_binders() {
641                 ty::PredicateAtom::Trait(p, _) => {
642                     if self.is_param_no_infer(p.trait_ref.substs)
643                         && !only_projections
644                         && is_new_pred
645                     {
646                         self.add_user_pred(computed_preds, predicate);
647                     }
648                     predicates.push_back(ty::Binder::bind(p));
649                 }
650                 ty::PredicateAtom::Projection(p) => {
651                     let p = ty::Binder::bind(p);
652                     debug!(
653                         "evaluate_nested_obligations: examining projection predicate {:?}",
654                         predicate
655                     );
656
657                     // As described above, we only want to display
658                     // bounds which include a generic parameter but don't include
659                     // an inference variable.
660                     // Additionally, we check if we've seen this predicate before,
661                     // to avoid rendering duplicate bounds to the user.
662                     if self.is_param_no_infer(p.skip_binder().projection_ty.substs)
663                         && !p.ty().skip_binder().has_infer_types()
664                         && is_new_pred
665                     {
666                         debug!(
667                             "evaluate_nested_obligations: adding projection predicate\
668                             to computed_preds: {:?}",
669                             predicate
670                         );
671
672                         // Under unusual circumstances, we can end up with a self-refeential
673                         // projection predicate. For example:
674                         // <T as MyType>::Value == <T as MyType>::Value
675                         // Not only is displaying this to the user pointless,
676                         // having it in the ParamEnv will cause an issue if we try to call
677                         // poly_project_and_unify_type on the predicate, since this kind of
678                         // predicate will normally never end up in a ParamEnv.
679                         //
680                         // For these reasons, we ignore these weird predicates,
681                         // ensuring that we're able to properly synthesize an auto trait impl
682                         if self.is_self_referential_projection(p) {
683                             debug!(
684                                 "evaluate_nested_obligations: encountered a projection
685                                  predicate equating a type with itself! Skipping"
686                             );
687                         } else {
688                             self.add_user_pred(computed_preds, predicate);
689                         }
690                     }
691
692                     // There are three possible cases when we project a predicate:
693                     //
694                     // 1. We encounter an error. This means that it's impossible for
695                     // our current type to implement the auto trait - there's bound
696                     // that we could add to our ParamEnv that would 'fix' this kind
697                     // of error, as it's not caused by an unimplemented type.
698                     //
699                     // 2. We successfully project the predicate (Ok(Some(_))), generating
700                     //  some subobligations. We then process these subobligations
701                     //  like any other generated sub-obligations.
702                     //
703                     // 3. We receive an 'ambiguous' result (Ok(None))
704                     // If we were actually trying to compile a crate,
705                     // we would need to re-process this obligation later.
706                     // However, all we care about is finding out what bounds
707                     // are needed for our type to implement a particular auto trait.
708                     // We've already added this obligation to our computed ParamEnv
709                     // above (if it was necessary). Therefore, we don't need
710                     // to do any further processing of the obligation.
711                     //
712                     // Note that we *must* try to project *all* projection predicates
713                     // we encounter, even ones without inference variable.
714                     // This ensures that we detect any projection errors,
715                     // which indicate that our type can *never* implement the given
716                     // auto trait. In that case, we will generate an explicit negative
717                     // impl (e.g. 'impl !Send for MyType'). However, we don't
718                     // try to process any of the generated subobligations -
719                     // they contain no new information, since we already know
720                     // that our type implements the projected-through trait,
721                     // and can lead to weird region issues.
722                     //
723                     // Normally, we'll generate a negative impl as a result of encountering
724                     // a type with an explicit negative impl of an auto trait
725                     // (for example, raw pointers have !Send and !Sync impls)
726                     // However, through some **interesting** manipulations of the type
727                     // system, it's actually possible to write a type that never
728                     // implements an auto trait due to a projection error, not a normal
729                     // negative impl error. To properly handle this case, we need
730                     // to ensure that we catch any potential projection errors,
731                     // and turn them into an explicit negative impl for our type.
732                     debug!("Projecting and unifying projection predicate {:?}", predicate);
733
734                     match poly_project_and_unify_type(select, &obligation.with(p)) {
735                         Err(e) => {
736                             debug!(
737                                 "evaluate_nested_obligations: Unable to unify predicate \
738                                  '{:?}' '{:?}', bailing out",
739                                 ty, e
740                             );
741                             return false;
742                         }
743                         Ok(Some(v)) => {
744                             // We only care about sub-obligations
745                             // when we started out trying to unify
746                             // some inference variables. See the comment above
747                             // for more infomration
748                             if p.ty().skip_binder().has_infer_types() {
749                                 if !self.evaluate_nested_obligations(
750                                     ty,
751                                     v.into_iter(),
752                                     computed_preds,
753                                     fresh_preds,
754                                     predicates,
755                                     select,
756                                     only_projections,
757                                 ) {
758                                     return false;
759                                 }
760                             }
761                         }
762                         Ok(None) => {
763                             // It's ok not to make progress when have no inference variables -
764                             // in that case, we were only performing unifcation to check if an
765                             // error occurred (which would indicate that it's impossible for our
766                             // type to implement the auto trait).
767                             // However, we should always make progress (either by generating
768                             // subobligations or getting an error) when we started off with
769                             // inference variables
770                             if p.ty().skip_binder().has_infer_types() {
771                                 panic!("Unexpected result when selecting {:?} {:?}", ty, obligation)
772                             }
773                         }
774                     }
775                 }
776                 ty::PredicateAtom::RegionOutlives(binder) => {
777                     let binder = ty::Binder::bind(binder);
778                     if select.infcx().region_outlives_predicate(&dummy_cause, binder).is_err() {
779                         return false;
780                     }
781                 }
782                 ty::PredicateAtom::TypeOutlives(binder) => {
783                     let binder = ty::Binder::bind(binder);
784                     match (
785                         binder.no_bound_vars(),
786                         binder.map_bound_ref(|pred| pred.0).no_bound_vars(),
787                     ) {
788                         (None, Some(t_a)) => {
789                             select.infcx().register_region_obligation_with_cause(
790                                 t_a,
791                                 select.infcx().tcx.lifetimes.re_static,
792                                 &dummy_cause,
793                             );
794                         }
795                         (Some(ty::OutlivesPredicate(t_a, r_b)), _) => {
796                             select.infcx().register_region_obligation_with_cause(
797                                 t_a,
798                                 r_b,
799                                 &dummy_cause,
800                             );
801                         }
802                         _ => {}
803                     };
804                 }
805                 _ => panic!("Unexpected predicate {:?} {:?}", ty, predicate),
806             };
807         }
808         true
809     }
810
811     pub fn clean_pred(
812         &self,
813         infcx: &InferCtxt<'_, 'tcx>,
814         p: ty::Predicate<'tcx>,
815     ) -> ty::Predicate<'tcx> {
816         infcx.freshen(p)
817     }
818 }
819
820 // Replaces all ReVars in a type with ty::Region's, using the provided map
821 pub struct RegionReplacer<'a, 'tcx> {
822     vid_to_region: &'a FxHashMap<ty::RegionVid, ty::Region<'tcx>>,
823     tcx: TyCtxt<'tcx>,
824 }
825
826 impl<'a, 'tcx> TypeFolder<'tcx> for RegionReplacer<'a, 'tcx> {
827     fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
828         self.tcx
829     }
830
831     fn fold_region(&mut self, r: ty::Region<'tcx>) -> ty::Region<'tcx> {
832         (match r {
833             &ty::ReVar(vid) => self.vid_to_region.get(&vid).cloned(),
834             _ => None,
835         })
836         .unwrap_or_else(|| r.super_fold_with(self))
837     }
838 }