]> git.lizzy.rs Git - rust.git/blob - src/librustc_trait_selection/traits/object_safety.rs
8da5e7318c002268baeacfa72164377c49bf578b
[rust.git] / src / librustc_trait_selection / traits / object_safety.rs
1 //! "Object safety" refers to the ability for a trait to be converted
2 //! to an object. In general, traits may only be converted to an
3 //! object if all of their methods meet certain criteria. In particular,
4 //! they must:
5 //!
6 //!   - have a suitable receiver from which we can extract a vtable and coerce to a "thin" version
7 //!     that doesn't contain the vtable;
8 //!   - not reference the erased type `Self` except for in this receiver;
9 //!   - not have generic type parameters.
10
11 use super::elaborate_predicates;
12
13 use crate::infer::TyCtxtInferExt;
14 use crate::traits::query::evaluate_obligation::InferCtxtExt;
15 use crate::traits::{self, Obligation, ObligationCause};
16 use rustc_errors::{Applicability, FatalError};
17 use rustc_hir as hir;
18 use rustc_hir::def_id::DefId;
19 use rustc_middle::ty::subst::{GenericArg, InternalSubsts, Subst};
20 use rustc_middle::ty::{self, Ty, TyCtxt, TypeFoldable, TypeVisitor, WithConstness};
21 use rustc_middle::ty::{Predicate, ToPredicate};
22 use rustc_session::lint::builtin::WHERE_CLAUSES_OBJECT_SAFETY;
23 use rustc_span::symbol::Symbol;
24 use rustc_span::Span;
25 use smallvec::SmallVec;
26
27 use std::iter;
28
29 pub use crate::traits::{MethodViolationCode, ObjectSafetyViolation};
30
31 /// Returns the object safety violations that affect
32 /// astconv -- currently, `Self` in supertraits. This is needed
33 /// because `object_safety_violations` can't be used during
34 /// type collection.
35 pub fn astconv_object_safety_violations(
36     tcx: TyCtxt<'_>,
37     trait_def_id: DefId,
38 ) -> Vec<ObjectSafetyViolation> {
39     debug_assert!(tcx.generics_of(trait_def_id).has_self);
40     let violations = traits::supertrait_def_ids(tcx, trait_def_id)
41         .map(|def_id| predicates_reference_self(tcx, def_id, true))
42         .filter(|spans| !spans.is_empty())
43         .map(ObjectSafetyViolation::SupertraitSelf)
44         .collect();
45
46     debug!("astconv_object_safety_violations(trait_def_id={:?}) = {:?}", trait_def_id, violations);
47
48     violations
49 }
50
51 fn object_safety_violations(
52     tcx: TyCtxt<'tcx>,
53     trait_def_id: DefId,
54 ) -> &'tcx [ObjectSafetyViolation] {
55     debug_assert!(tcx.generics_of(trait_def_id).has_self);
56     debug!("object_safety_violations: {:?}", trait_def_id);
57
58     tcx.arena.alloc_from_iter(
59         traits::supertrait_def_ids(tcx, trait_def_id)
60             .flat_map(|def_id| object_safety_violations_for_trait(tcx, def_id)),
61     )
62 }
63
64 /// We say a method is *vtable safe* if it can be invoked on a trait
65 /// object. Note that object-safe traits can have some
66 /// non-vtable-safe methods, so long as they require `Self: Sized` or
67 /// otherwise ensure that they cannot be used when `Self = Trait`.
68 pub fn is_vtable_safe_method(tcx: TyCtxt<'_>, trait_def_id: DefId, method: &ty::AssocItem) -> bool {
69     debug_assert!(tcx.generics_of(trait_def_id).has_self);
70     debug!("is_vtable_safe_method({:?}, {:?})", trait_def_id, method);
71     // Any method that has a `Self: Sized` bound cannot be called.
72     if generics_require_sized_self(tcx, method.def_id) {
73         return false;
74     }
75
76     match virtual_call_violation_for_method(tcx, trait_def_id, method) {
77         None | Some(MethodViolationCode::WhereClauseReferencesSelf) => true,
78         Some(_) => false,
79     }
80 }
81
82 fn object_safety_violations_for_trait(
83     tcx: TyCtxt<'_>,
84     trait_def_id: DefId,
85 ) -> Vec<ObjectSafetyViolation> {
86     // Check methods for violations.
87     let mut violations: Vec<_> = tcx
88         .associated_items(trait_def_id)
89         .in_definition_order()
90         .filter(|item| item.kind == ty::AssocKind::Fn)
91         .filter_map(|item| {
92             object_safety_violation_for_method(tcx, trait_def_id, &item)
93                 .map(|(code, span)| ObjectSafetyViolation::Method(item.ident.name, code, span))
94         })
95         .filter(|violation| {
96             if let ObjectSafetyViolation::Method(
97                 _,
98                 MethodViolationCode::WhereClauseReferencesSelf,
99                 span,
100             ) = violation
101             {
102                 // Using `CRATE_NODE_ID` is wrong, but it's hard to get a more precise id.
103                 // It's also hard to get a use site span, so we use the method definition span.
104                 tcx.struct_span_lint_hir(
105                     WHERE_CLAUSES_OBJECT_SAFETY,
106                     hir::CRATE_HIR_ID,
107                     *span,
108                     |lint| {
109                         let mut err = lint.build(&format!(
110                             "the trait `{}` cannot be made into an object",
111                             tcx.def_path_str(trait_def_id)
112                         ));
113                         let node = tcx.hir().get_if_local(trait_def_id);
114                         let msg = if let Some(hir::Node::Item(item)) = node {
115                             err.span_label(
116                                 item.ident.span,
117                                 "this trait cannot be made into an object...",
118                             );
119                             format!("...because {}", violation.error_msg())
120                         } else {
121                             format!(
122                                 "the trait cannot be made into an object because {}",
123                                 violation.error_msg()
124                             )
125                         };
126                         err.span_label(*span, &msg);
127                         match (node, violation.solution()) {
128                             (Some(_), Some((note, None))) => {
129                                 err.help(&note);
130                             }
131                             (Some(_), Some((note, Some((sugg, span))))) => {
132                                 err.span_suggestion(
133                                     span,
134                                     &note,
135                                     sugg,
136                                     Applicability::MachineApplicable,
137                                 );
138                             }
139                             // Only provide the help if its a local trait, otherwise it's not actionable.
140                             _ => {}
141                         }
142                         err.emit();
143                     },
144                 );
145                 false
146             } else {
147                 true
148             }
149         })
150         .collect();
151
152     // Check the trait itself.
153     if trait_has_sized_self(tcx, trait_def_id) {
154         // We don't want to include the requirement from `Sized` itself to be `Sized` in the list.
155         let spans = get_sized_bounds(tcx, trait_def_id);
156         violations.push(ObjectSafetyViolation::SizedSelf(spans));
157     }
158     let spans = predicates_reference_self(tcx, trait_def_id, false);
159     if !spans.is_empty() {
160         violations.push(ObjectSafetyViolation::SupertraitSelf(spans));
161     }
162
163     violations.extend(
164         tcx.associated_items(trait_def_id)
165             .in_definition_order()
166             .filter(|item| item.kind == ty::AssocKind::Const)
167             .map(|item| ObjectSafetyViolation::AssocConst(item.ident.name, item.ident.span)),
168     );
169
170     debug!(
171         "object_safety_violations_for_trait(trait_def_id={:?}) = {:?}",
172         trait_def_id, violations
173     );
174
175     violations
176 }
177
178 fn sized_trait_bound_spans<'tcx>(
179     tcx: TyCtxt<'tcx>,
180     bounds: hir::GenericBounds<'tcx>,
181 ) -> impl 'tcx + Iterator<Item = Span> {
182     bounds.iter().filter_map(move |b| match b {
183         hir::GenericBound::Trait(trait_ref, hir::TraitBoundModifier::None)
184             if trait_has_sized_self(
185                 tcx,
186                 trait_ref.trait_ref.trait_def_id().unwrap_or_else(|| FatalError.raise()),
187             ) =>
188         {
189             // Fetch spans for supertraits that are `Sized`: `trait T: Super`
190             Some(trait_ref.span)
191         }
192         _ => None,
193     })
194 }
195
196 fn get_sized_bounds(tcx: TyCtxt<'_>, trait_def_id: DefId) -> SmallVec<[Span; 1]> {
197     tcx.hir()
198         .get_if_local(trait_def_id)
199         .and_then(|node| match node {
200             hir::Node::Item(hir::Item {
201                 kind: hir::ItemKind::Trait(.., generics, bounds, _),
202                 ..
203             }) => Some(
204                 generics
205                     .where_clause
206                     .predicates
207                     .iter()
208                     .filter_map(|pred| {
209                         match pred {
210                             hir::WherePredicate::BoundPredicate(pred)
211                                 if pred.bounded_ty.hir_id.owner.to_def_id() == trait_def_id =>
212                             {
213                                 // Fetch spans for trait bounds that are Sized:
214                                 // `trait T where Self: Pred`
215                                 Some(sized_trait_bound_spans(tcx, pred.bounds))
216                             }
217                             _ => None,
218                         }
219                     })
220                     .flatten()
221                     // Fetch spans for supertraits that are `Sized`: `trait T: Super`.
222                     .chain(sized_trait_bound_spans(tcx, bounds))
223                     .collect::<SmallVec<[Span; 1]>>(),
224             ),
225             _ => None,
226         })
227         .unwrap_or_else(SmallVec::new)
228 }
229
230 fn predicates_reference_self(
231     tcx: TyCtxt<'_>,
232     trait_def_id: DefId,
233     supertraits_only: bool,
234 ) -> SmallVec<[Span; 1]> {
235     let trait_ref = ty::Binder::dummy(ty::TraitRef::identity(tcx, trait_def_id));
236     let predicates = if supertraits_only {
237         tcx.super_predicates_of(trait_def_id)
238     } else {
239         tcx.predicates_of(trait_def_id)
240     };
241     let self_ty = tcx.types.self_param;
242     let has_self_ty = |arg: &GenericArg<'_>| arg.walk().any(|arg| arg == self_ty.into());
243     predicates
244         .predicates
245         .iter()
246         .map(|(predicate, sp)| (predicate.subst_supertrait(tcx, &trait_ref), sp))
247         .filter_map(|(predicate, &sp)| {
248             match predicate.ignore_quantifiers().skip_binder().kind() {
249                 ty::PredicateKind::Trait(ref data, _) => {
250                     // In the case of a trait predicate, we can skip the "self" type.
251                     if data.trait_ref.substs[1..].iter().any(has_self_ty) { Some(sp) } else { None }
252                 }
253                 ty::PredicateKind::Projection(ref data) => {
254                     // And similarly for projections. This should be redundant with
255                     // the previous check because any projection should have a
256                     // matching `Trait` predicate with the same inputs, but we do
257                     // the check to be safe.
258                     //
259                     // Note that we *do* allow projection *outputs* to contain
260                     // `self` (i.e., `trait Foo: Bar<Output=Self::Result> { type Result; }`),
261                     // we just require the user to specify *both* outputs
262                     // in the object type (i.e., `dyn Foo<Output=(), Result=()>`).
263                     //
264                     // This is ALT2 in issue #56288, see that for discussion of the
265                     // possible alternatives.
266                     if data.projection_ty.trait_ref(tcx).substs[1..].iter().any(has_self_ty) {
267                         Some(sp)
268                     } else {
269                         None
270                     }
271                 }
272                 ty::PredicateKind::WellFormed(..)
273                 | ty::PredicateKind::ObjectSafe(..)
274                 | ty::PredicateKind::TypeOutlives(..)
275                 | ty::PredicateKind::RegionOutlives(..)
276                 | ty::PredicateKind::ClosureKind(..)
277                 | ty::PredicateKind::Subtype(..)
278                 | ty::PredicateKind::ConstEvaluatable(..)
279                 | ty::PredicateKind::ConstEquate(..) => None,
280                 ty::PredicateKind::ForAll(..) => bug!("unexpected predicate: {:?}", predicate),
281             }
282         })
283         .collect()
284 }
285
286 fn trait_has_sized_self(tcx: TyCtxt<'_>, trait_def_id: DefId) -> bool {
287     generics_require_sized_self(tcx, trait_def_id)
288 }
289
290 fn generics_require_sized_self(tcx: TyCtxt<'_>, def_id: DefId) -> bool {
291     let sized_def_id = match tcx.lang_items().sized_trait() {
292         Some(def_id) => def_id,
293         None => {
294             return false; /* No Sized trait, can't require it! */
295         }
296     };
297
298     // Search for a predicate like `Self : Sized` amongst the trait bounds.
299     let predicates = tcx.predicates_of(def_id);
300     let predicates = predicates.instantiate_identity(tcx).predicates;
301     elaborate_predicates(tcx, predicates.into_iter()).any(|obligation| {
302         match obligation.predicate.ignore_quantifiers().skip_binder().kind() {
303             ty::PredicateKind::Trait(ref trait_pred, _) => {
304                 trait_pred.def_id() == sized_def_id && trait_pred.self_ty().is_param(0)
305             }
306             ty::PredicateKind::Projection(..)
307             | ty::PredicateKind::Subtype(..)
308             | ty::PredicateKind::RegionOutlives(..)
309             | ty::PredicateKind::WellFormed(..)
310             | ty::PredicateKind::ObjectSafe(..)
311             | ty::PredicateKind::ClosureKind(..)
312             | ty::PredicateKind::TypeOutlives(..)
313             | ty::PredicateKind::ConstEvaluatable(..)
314             | ty::PredicateKind::ConstEquate(..) => false,
315             ty::PredicateKind::ForAll(_) => {
316                 bug!("unexpected predicate: {:?}", obligation.predicate)
317             }
318         }
319     })
320 }
321
322 /// Returns `Some(_)` if this method makes the containing trait not object safe.
323 fn object_safety_violation_for_method(
324     tcx: TyCtxt<'_>,
325     trait_def_id: DefId,
326     method: &ty::AssocItem,
327 ) -> Option<(MethodViolationCode, Span)> {
328     debug!("object_safety_violation_for_method({:?}, {:?})", trait_def_id, method);
329     // Any method that has a `Self : Sized` requisite is otherwise
330     // exempt from the regulations.
331     if generics_require_sized_self(tcx, method.def_id) {
332         return None;
333     }
334
335     let violation = virtual_call_violation_for_method(tcx, trait_def_id, method);
336     // Get an accurate span depending on the violation.
337     violation.map(|v| {
338         let node = tcx.hir().get_if_local(method.def_id);
339         let span = match (v, node) {
340             (MethodViolationCode::ReferencesSelfInput(arg), Some(node)) => node
341                 .fn_decl()
342                 .and_then(|decl| decl.inputs.get(arg + 1))
343                 .map_or(method.ident.span, |arg| arg.span),
344             (MethodViolationCode::UndispatchableReceiver, Some(node)) => node
345                 .fn_decl()
346                 .and_then(|decl| decl.inputs.get(0))
347                 .map_or(method.ident.span, |arg| arg.span),
348             (MethodViolationCode::ReferencesSelfOutput, Some(node)) => {
349                 node.fn_decl().map_or(method.ident.span, |decl| decl.output.span())
350             }
351             _ => method.ident.span,
352         };
353         (v, span)
354     })
355 }
356
357 /// Returns `Some(_)` if this method cannot be called on a trait
358 /// object; this does not necessarily imply that the enclosing trait
359 /// is not object safe, because the method might have a where clause
360 /// `Self:Sized`.
361 fn virtual_call_violation_for_method<'tcx>(
362     tcx: TyCtxt<'tcx>,
363     trait_def_id: DefId,
364     method: &ty::AssocItem,
365 ) -> Option<MethodViolationCode> {
366     // The method's first parameter must be named `self`
367     if !method.fn_has_self_parameter {
368         // We'll attempt to provide a structured suggestion for `Self: Sized`.
369         let sugg =
370             tcx.hir().get_if_local(method.def_id).as_ref().and_then(|node| node.generics()).map(
371                 |generics| match generics.where_clause.predicates {
372                     [] => (" where Self: Sized", generics.where_clause.span),
373                     [.., pred] => (", Self: Sized", pred.span().shrink_to_hi()),
374                 },
375             );
376         return Some(MethodViolationCode::StaticMethod(sugg));
377     }
378
379     let sig = tcx.fn_sig(method.def_id);
380
381     for (i, input_ty) in sig.skip_binder().inputs()[1..].iter().enumerate() {
382         if contains_illegal_self_type_reference(tcx, trait_def_id, input_ty) {
383             return Some(MethodViolationCode::ReferencesSelfInput(i));
384         }
385     }
386     if contains_illegal_self_type_reference(tcx, trait_def_id, sig.output().skip_binder()) {
387         return Some(MethodViolationCode::ReferencesSelfOutput);
388     }
389
390     // We can't monomorphize things like `fn foo<A>(...)`.
391     let own_counts = tcx.generics_of(method.def_id).own_counts();
392     if own_counts.types + own_counts.consts != 0 {
393         return Some(MethodViolationCode::Generic);
394     }
395
396     if tcx
397         .predicates_of(method.def_id)
398         .predicates
399         .iter()
400         // A trait object can't claim to live more than the concrete type,
401         // so outlives predicates will always hold.
402         .cloned()
403         .filter(|(p, _)| p.to_opt_type_outlives().is_none())
404         .collect::<Vec<_>>()
405         // Do a shallow visit so that `contains_illegal_self_type_reference`
406         // may apply it's custom visiting.
407         .visit_tys_shallow(|t| contains_illegal_self_type_reference(tcx, trait_def_id, t))
408     {
409         return Some(MethodViolationCode::WhereClauseReferencesSelf);
410     }
411
412     let receiver_ty =
413         tcx.liberate_late_bound_regions(method.def_id, &sig.map_bound(|sig| sig.inputs()[0]));
414
415     // Until `unsized_locals` is fully implemented, `self: Self` can't be dispatched on.
416     // However, this is already considered object-safe. We allow it as a special case here.
417     // FIXME(mikeyhew) get rid of this `if` statement once `receiver_is_dispatchable` allows
418     // `Receiver: Unsize<Receiver[Self => dyn Trait]>`.
419     if receiver_ty != tcx.types.self_param {
420         if !receiver_is_dispatchable(tcx, method, receiver_ty) {
421             return Some(MethodViolationCode::UndispatchableReceiver);
422         } else {
423             // Do sanity check to make sure the receiver actually has the layout of a pointer.
424
425             use rustc_target::abi::Abi;
426
427             let param_env = tcx.param_env(method.def_id);
428
429             let abi_of_ty = |ty: Ty<'tcx>| -> &Abi {
430                 match tcx.layout_of(param_env.and(ty)) {
431                     Ok(layout) => &layout.abi,
432                     Err(err) => bug!("error: {}\n while computing layout for type {:?}", err, ty),
433                 }
434             };
435
436             // e.g., `Rc<()>`
437             let unit_receiver_ty =
438                 receiver_for_self_ty(tcx, receiver_ty, tcx.mk_unit(), method.def_id);
439
440             match abi_of_ty(unit_receiver_ty) {
441                 &Abi::Scalar(..) => (),
442                 abi => {
443                     tcx.sess.delay_span_bug(
444                         tcx.def_span(method.def_id),
445                         &format!(
446                             "receiver when `Self = ()` should have a Scalar ABI; found {:?}",
447                             abi
448                         ),
449                     );
450                 }
451             }
452
453             let trait_object_ty =
454                 object_ty_for_trait(tcx, trait_def_id, tcx.mk_region(ty::ReStatic));
455
456             // e.g., `Rc<dyn Trait>`
457             let trait_object_receiver =
458                 receiver_for_self_ty(tcx, receiver_ty, trait_object_ty, method.def_id);
459
460             match abi_of_ty(trait_object_receiver) {
461                 &Abi::ScalarPair(..) => (),
462                 abi => {
463                     tcx.sess.delay_span_bug(
464                         tcx.def_span(method.def_id),
465                         &format!(
466                             "receiver when `Self = {}` should have a ScalarPair ABI; \
467                                  found {:?}",
468                             trait_object_ty, abi
469                         ),
470                     );
471                 }
472             }
473         }
474     }
475
476     None
477 }
478
479 /// Performs a type substitution to produce the version of `receiver_ty` when `Self = self_ty`.
480 /// For example, for `receiver_ty = Rc<Self>` and `self_ty = Foo`, returns `Rc<Foo>`.
481 fn receiver_for_self_ty<'tcx>(
482     tcx: TyCtxt<'tcx>,
483     receiver_ty: Ty<'tcx>,
484     self_ty: Ty<'tcx>,
485     method_def_id: DefId,
486 ) -> Ty<'tcx> {
487     debug!("receiver_for_self_ty({:?}, {:?}, {:?})", receiver_ty, self_ty, method_def_id);
488     let substs = InternalSubsts::for_item(tcx, method_def_id, |param, _| {
489         if param.index == 0 { self_ty.into() } else { tcx.mk_param_from_def(param) }
490     });
491
492     let result = receiver_ty.subst(tcx, substs);
493     debug!(
494         "receiver_for_self_ty({:?}, {:?}, {:?}) = {:?}",
495         receiver_ty, self_ty, method_def_id, result
496     );
497     result
498 }
499
500 /// Creates the object type for the current trait. For example,
501 /// if the current trait is `Deref`, then this will be
502 /// `dyn Deref<Target = Self::Target> + 'static`.
503 fn object_ty_for_trait<'tcx>(
504     tcx: TyCtxt<'tcx>,
505     trait_def_id: DefId,
506     lifetime: ty::Region<'tcx>,
507 ) -> Ty<'tcx> {
508     debug!("object_ty_for_trait: trait_def_id={:?}", trait_def_id);
509
510     let trait_ref = ty::TraitRef::identity(tcx, trait_def_id);
511
512     let trait_predicate =
513         ty::ExistentialPredicate::Trait(ty::ExistentialTraitRef::erase_self_ty(tcx, trait_ref));
514
515     let mut associated_types = traits::supertraits(tcx, ty::Binder::dummy(trait_ref))
516         .flat_map(|super_trait_ref| {
517             tcx.associated_items(super_trait_ref.def_id())
518                 .in_definition_order()
519                 .map(move |item| (super_trait_ref, item))
520         })
521         .filter(|(_, item)| item.kind == ty::AssocKind::Type)
522         .collect::<Vec<_>>();
523
524     // existential predicates need to be in a specific order
525     associated_types.sort_by_cached_key(|(_, item)| tcx.def_path_hash(item.def_id));
526
527     let projection_predicates = associated_types.into_iter().map(|(super_trait_ref, item)| {
528         // We *can* get bound lifetimes here in cases like
529         // `trait MyTrait: for<'s> OtherTrait<&'s T, Output=bool>`.
530         //
531         // binder moved to (*)...
532         let super_trait_ref = super_trait_ref.skip_binder();
533         ty::ExistentialPredicate::Projection(ty::ExistentialProjection {
534             ty: tcx.mk_projection(item.def_id, super_trait_ref.substs),
535             item_def_id: item.def_id,
536             substs: super_trait_ref.substs,
537         })
538     });
539
540     let existential_predicates =
541         tcx.mk_existential_predicates(iter::once(trait_predicate).chain(projection_predicates));
542
543     let object_ty = tcx.mk_dynamic(
544         // (*) ... binder re-introduced here
545         ty::Binder::bind(existential_predicates),
546         lifetime,
547     );
548
549     debug!("object_ty_for_trait: object_ty=`{}`", object_ty);
550
551     object_ty
552 }
553
554 /// Checks the method's receiver (the `self` argument) can be dispatched on when `Self` is a
555 /// trait object. We require that `DispatchableFromDyn` be implemented for the receiver type
556 /// in the following way:
557 /// - let `Receiver` be the type of the `self` argument, i.e `Self`, `&Self`, `Rc<Self>`,
558 /// - require the following bound:
559 ///
560 ///   ```
561 ///   Receiver[Self => T]: DispatchFromDyn<Receiver[Self => dyn Trait]>
562 ///   ```
563 ///
564 ///   where `Foo[X => Y]` means "the same type as `Foo`, but with `X` replaced with `Y`"
565 ///   (substitution notation).
566 ///
567 /// Some examples of receiver types and their required obligation:
568 /// - `&'a mut self` requires `&'a mut Self: DispatchFromDyn<&'a mut dyn Trait>`,
569 /// - `self: Rc<Self>` requires `Rc<Self>: DispatchFromDyn<Rc<dyn Trait>>`,
570 /// - `self: Pin<Box<Self>>` requires `Pin<Box<Self>>: DispatchFromDyn<Pin<Box<dyn Trait>>>`.
571 ///
572 /// The only case where the receiver is not dispatchable, but is still a valid receiver
573 /// type (just not object-safe), is when there is more than one level of pointer indirection.
574 /// E.g., `self: &&Self`, `self: &Rc<Self>`, `self: Box<Box<Self>>`. In these cases, there
575 /// is no way, or at least no inexpensive way, to coerce the receiver from the version where
576 /// `Self = dyn Trait` to the version where `Self = T`, where `T` is the unknown erased type
577 /// contained by the trait object, because the object that needs to be coerced is behind
578 /// a pointer.
579 ///
580 /// In practice, we cannot use `dyn Trait` explicitly in the obligation because it would result
581 /// in a new check that `Trait` is object safe, creating a cycle (until object_safe_for_dispatch
582 /// is stabilized, see tracking issue https://github.com/rust-lang/rust/issues/43561).
583 /// Instead, we fudge a little by introducing a new type parameter `U` such that
584 /// `Self: Unsize<U>` and `U: Trait + ?Sized`, and use `U` in place of `dyn Trait`.
585 /// Written as a chalk-style query:
586 ///
587 ///     forall (U: Trait + ?Sized) {
588 ///         if (Self: Unsize<U>) {
589 ///             Receiver: DispatchFromDyn<Receiver[Self => U]>
590 ///         }
591 ///     }
592 ///
593 /// for `self: &'a mut Self`, this means `&'a mut Self: DispatchFromDyn<&'a mut U>`
594 /// for `self: Rc<Self>`, this means `Rc<Self>: DispatchFromDyn<Rc<U>>`
595 /// for `self: Pin<Box<Self>>`, this means `Pin<Box<Self>>: DispatchFromDyn<Pin<Box<U>>>`
596 //
597 // FIXME(mikeyhew) when unsized receivers are implemented as part of unsized rvalues, add this
598 // fallback query: `Receiver: Unsize<Receiver[Self => U]>` to support receivers like
599 // `self: Wrapper<Self>`.
600 #[allow(dead_code)]
601 fn receiver_is_dispatchable<'tcx>(
602     tcx: TyCtxt<'tcx>,
603     method: &ty::AssocItem,
604     receiver_ty: Ty<'tcx>,
605 ) -> bool {
606     debug!("receiver_is_dispatchable: method = {:?}, receiver_ty = {:?}", method, receiver_ty);
607
608     let traits = (tcx.lang_items().unsize_trait(), tcx.lang_items().dispatch_from_dyn_trait());
609     let (unsize_did, dispatch_from_dyn_did) = if let (Some(u), Some(cu)) = traits {
610         (u, cu)
611     } else {
612         debug!("receiver_is_dispatchable: Missing Unsize or DispatchFromDyn traits");
613         return false;
614     };
615
616     // the type `U` in the query
617     // use a bogus type parameter to mimic a forall(U) query using u32::MAX for now.
618     // FIXME(mikeyhew) this is a total hack. Once object_safe_for_dispatch is stabilized, we can
619     // replace this with `dyn Trait`
620     let unsized_self_ty: Ty<'tcx> =
621         tcx.mk_ty_param(u32::MAX, Symbol::intern("RustaceansAreAwesome"));
622
623     // `Receiver[Self => U]`
624     let unsized_receiver_ty =
625         receiver_for_self_ty(tcx, receiver_ty, unsized_self_ty, method.def_id);
626
627     // create a modified param env, with `Self: Unsize<U>` and `U: Trait` added to caller bounds
628     // `U: ?Sized` is already implied here
629     let param_env = {
630         let param_env = tcx.param_env(method.def_id);
631
632         // Self: Unsize<U>
633         let unsize_predicate = ty::TraitRef {
634             def_id: unsize_did,
635             substs: tcx.mk_substs_trait(tcx.types.self_param, &[unsized_self_ty.into()]),
636         }
637         .without_const()
638         .to_predicate(tcx);
639
640         // U: Trait<Arg1, ..., ArgN>
641         let trait_predicate = {
642             let substs =
643                 InternalSubsts::for_item(tcx, method.container.assert_trait(), |param, _| {
644                     if param.index == 0 {
645                         unsized_self_ty.into()
646                     } else {
647                         tcx.mk_param_from_def(param)
648                     }
649                 });
650
651             ty::TraitRef { def_id: unsize_did, substs }.without_const().to_predicate(tcx)
652         };
653
654         let caller_bounds: Vec<Predicate<'tcx>> = param_env
655             .caller_bounds()
656             .iter()
657             .chain(iter::once(unsize_predicate))
658             .chain(iter::once(trait_predicate))
659             .collect();
660
661         ty::ParamEnv::new(
662             tcx.intern_predicates(&caller_bounds),
663             param_env.reveal(),
664             param_env.def_id,
665         )
666     };
667
668     // Receiver: DispatchFromDyn<Receiver[Self => U]>
669     let obligation = {
670         let predicate = ty::TraitRef {
671             def_id: dispatch_from_dyn_did,
672             substs: tcx.mk_substs_trait(receiver_ty, &[unsized_receiver_ty.into()]),
673         }
674         .without_const()
675         .to_predicate(tcx);
676
677         Obligation::new(ObligationCause::dummy(), param_env, predicate)
678     };
679
680     tcx.infer_ctxt().enter(|ref infcx| {
681         // the receiver is dispatchable iff the obligation holds
682         infcx.predicate_must_hold_modulo_regions(&obligation)
683     })
684 }
685
686 fn contains_illegal_self_type_reference<'tcx>(
687     tcx: TyCtxt<'tcx>,
688     trait_def_id: DefId,
689     ty: Ty<'tcx>,
690 ) -> bool {
691     // This is somewhat subtle. In general, we want to forbid
692     // references to `Self` in the argument and return types,
693     // since the value of `Self` is erased. However, there is one
694     // exception: it is ok to reference `Self` in order to access
695     // an associated type of the current trait, since we retain
696     // the value of those associated types in the object type
697     // itself.
698     //
699     // ```rust
700     // trait SuperTrait {
701     //     type X;
702     // }
703     //
704     // trait Trait : SuperTrait {
705     //     type Y;
706     //     fn foo(&self, x: Self) // bad
707     //     fn foo(&self) -> Self // bad
708     //     fn foo(&self) -> Option<Self> // bad
709     //     fn foo(&self) -> Self::Y // OK, desugars to next example
710     //     fn foo(&self) -> <Self as Trait>::Y // OK
711     //     fn foo(&self) -> Self::X // OK, desugars to next example
712     //     fn foo(&self) -> <Self as SuperTrait>::X // OK
713     // }
714     // ```
715     //
716     // However, it is not as simple as allowing `Self` in a projected
717     // type, because there are illegal ways to use `Self` as well:
718     //
719     // ```rust
720     // trait Trait : SuperTrait {
721     //     ...
722     //     fn foo(&self) -> <Self as SomeOtherTrait>::X;
723     // }
724     // ```
725     //
726     // Here we will not have the type of `X` recorded in the
727     // object type, and we cannot resolve `Self as SomeOtherTrait`
728     // without knowing what `Self` is.
729
730     struct IllegalSelfTypeVisitor<'tcx> {
731         tcx: TyCtxt<'tcx>,
732         self_ty: Ty<'tcx>,
733         trait_def_id: DefId,
734         supertraits: Option<Vec<ty::PolyTraitRef<'tcx>>>,
735     }
736
737     impl<'tcx> TypeVisitor<'tcx> for IllegalSelfTypeVisitor<'tcx> {
738         fn visit_ty(&mut self, t: Ty<'tcx>) -> bool {
739             match t.kind {
740                 ty::Param(_) => t == self.self_ty,
741                 ty::Projection(ref data) => {
742                     // This is a projected type `<Foo as SomeTrait>::X`.
743
744                     // Compute supertraits of current trait lazily.
745                     if self.supertraits.is_none() {
746                         let trait_ref =
747                             ty::Binder::bind(ty::TraitRef::identity(self.tcx, self.trait_def_id));
748                         self.supertraits = Some(traits::supertraits(self.tcx, trait_ref).collect());
749                     }
750
751                     // Determine whether the trait reference `Foo as
752                     // SomeTrait` is in fact a supertrait of the
753                     // current trait. In that case, this type is
754                     // legal, because the type `X` will be specified
755                     // in the object type.  Note that we can just use
756                     // direct equality here because all of these types
757                     // are part of the formal parameter listing, and
758                     // hence there should be no inference variables.
759                     let projection_trait_ref = ty::Binder::bind(data.trait_ref(self.tcx));
760                     let is_supertrait_of_current_trait =
761                         self.supertraits.as_ref().unwrap().contains(&projection_trait_ref);
762
763                     if is_supertrait_of_current_trait {
764                         false // do not walk contained types, do not report error, do collect $200
765                     } else {
766                         t.super_visit_with(self) // DO walk contained types, POSSIBLY reporting an error
767                     }
768                 }
769                 _ => t.super_visit_with(self), // walk contained types, if any
770             }
771         }
772
773         fn visit_const(&mut self, _c: &ty::Const<'tcx>) -> bool {
774             // FIXME(#72219) Look into the unevaluated constants for object safety violations.
775             // Do not walk substitutions of unevaluated consts, as they contain `Self`, even
776             // though the const expression doesn't necessary use it. Currently type variables
777             // inside array length expressions are forbidden, so they can't break the above
778             // rules.
779             false
780         }
781     }
782
783     ty.visit_with(&mut IllegalSelfTypeVisitor {
784         tcx,
785         self_ty: tcx.types.self_param,
786         trait_def_id,
787         supertraits: None,
788     })
789 }
790
791 pub fn provide(providers: &mut ty::query::Providers) {
792     *providers = ty::query::Providers { object_safety_violations, ..*providers };
793 }