compiler/rustc_middle/src/traits/mod.rs

   1 //! Trait Resolution. See the [rustc dev guide] for more information on how this works.
   2 //!
   3 //! [rustc dev guide]: https://rustc-dev-guide.rust-lang.org/traits/resolution.html
   4
   5 mod chalk;
   6 pub mod query;
   7 pub mod select;
   8 pub mod specialization_graph;
   9 mod structural_impls;
  10 pub mod util;
  11
  12 use crate::infer::canonical::Canonical;
  13 use crate::thir::abstract_const::NotConstEvaluatable;
  14 use crate::ty::subst::SubstsRef;
  15 use crate::ty::{self, AdtKind, Ty, TyCtxt};
  16
  17 use rustc_data_structures::sync::Lrc;
  18 use rustc_errors::{Applicability, DiagnosticBuilder};
  19 use rustc_hir as hir;
  20 use rustc_hir::def_id::{DefId, LocalDefId};
  21 use rustc_span::symbol::Symbol;
  22 use rustc_span::{Span, DUMMY_SP};
  23 use smallvec::SmallVec;
  24
  25 use std::borrow::Cow;
  26 use std::hash::{Hash, Hasher};
  27
  28 pub use self::select::{EvaluationCache, EvaluationResult, OverflowError, SelectionCache};
  29
  30 pub type CanonicalChalkEnvironmentAndGoal<'tcx> = Canonical<'tcx, ChalkEnvironmentAndGoal<'tcx>>;
  31
  32 pub use self::ObligationCauseCode::*;
  33
  34 pub use self::chalk::{ChalkEnvironmentAndGoal, RustInterner as ChalkRustInterner};
  35
  36 /// Depending on the stage of compilation, we want projection to be
  37 /// more or less conservative.
  38 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, HashStable)]
  39 pub enum Reveal {
  40     /// At type-checking time, we refuse to project any associated
  41     /// type that is marked `default`. Non-`default` ("final") types
  42     /// are always projected. This is necessary in general for
  43     /// soundness of specialization. However, we *could* allow
  44     /// projections in fully-monomorphic cases. We choose not to,
  45     /// because we prefer for `default type` to force the type
  46     /// definition to be treated abstractly by any consumers of the
  47     /// impl. Concretely, that means that the following example will
  48     /// fail to compile:
  49     ///
  50     /// ```
  51     /// trait Assoc {
  52     ///     type Output;
  53     /// }
  54     ///
  55     /// impl<T> Assoc for T {
  56     ///     default type Output = bool;
  57     /// }
  58     ///
  59     /// fn main() {
  60     ///     let <() as Assoc>::Output = true;
  61     /// }
  62     /// ```
  63     UserFacing,
  64
  65     /// At codegen time, all monomorphic projections will succeed.
  66     /// Also, `impl Trait` is normalized to the concrete type,
  67     /// which has to be already collected by type-checking.
  68     ///
  69     /// NOTE: as `impl Trait`'s concrete type should *never*
  70     /// be observable directly by the user, `Reveal::All`
  71     /// should not be used by checks which may expose
  72     /// type equality or type contents to the user.
  73     /// There are some exceptions, e.g., around auto traits and
  74     /// transmute-checking, which expose some details, but
  75     /// not the whole concrete type of the `impl Trait`.
  76     All,
  77 }
  78
  79 /// The reason why we incurred this obligation; used for error reporting.
  80 ///
  81 /// Non-misc `ObligationCauseCode`s are stored on the heap. This gives the
  82 /// best trade-off between keeping the type small (which makes copies cheaper)
  83 /// while not doing too many heap allocations.
  84 ///
  85 /// We do not want to intern this as there are a lot of obligation causes which
  86 /// only live for a short period of time.
  87 #[derive(Clone, Debug, PartialEq, Eq, Lift)]
  88 pub struct ObligationCause<'tcx> {
  89     pub span: Span,
  90
  91     /// The ID of the fn body that triggered this obligation. This is
  92     /// used for region obligations to determine the precise
  93     /// environment in which the region obligation should be evaluated
  94     /// (in particular, closures can add new assumptions). See the
  95     /// field `region_obligations` of the `FulfillmentContext` for more
  96     /// information.
  97     pub body_id: hir::HirId,
  98
  99     /// `None` for `MISC_OBLIGATION_CAUSE_CODE` (a common case, occurs ~60% of
 100     /// the time). `Some` otherwise.
 101     code: Option<Lrc<ObligationCauseCode<'tcx>>>,
 102 }
 103
 104 // This custom hash function speeds up hashing for `Obligation` deduplication
 105 // greatly by skipping the `code` field, which can be large and complex. That
 106 // shouldn't affect hash quality much since there are several other fields in
 107 // `Obligation` which should be unique enough, especially the predicate itself
 108 // which is hashed as an interned pointer. See #90996.
 109 impl Hash for ObligationCause<'_> {
 110     fn hash<H: Hasher>(&self, state: &mut H) {
 111         self.body_id.hash(state);
 112         self.span.hash(state);
 113     }
 114 }
 115
 116 const MISC_OBLIGATION_CAUSE_CODE: ObligationCauseCode<'static> = MiscObligation;
 117
 118 impl<'tcx> ObligationCause<'tcx> {
 119     #[inline]
 120     pub fn new(
 121         span: Span,
 122         body_id: hir::HirId,
 123         code: ObligationCauseCode<'tcx>,
 124     ) -> ObligationCause<'tcx> {
 125         ObligationCause {
 126             span,
 127             body_id,
 128             code: if code == MISC_OBLIGATION_CAUSE_CODE { None } else { Some(Lrc::new(code)) },
 129         }
 130     }
 131
 132     pub fn misc(span: Span, body_id: hir::HirId) -> ObligationCause<'tcx> {
 133         ObligationCause::new(span, body_id, MiscObligation)
 134     }
 135
 136     #[inline(always)]
 137     pub fn dummy() -> ObligationCause<'tcx> {
 138         ObligationCause { span: DUMMY_SP, body_id: hir::CRATE_HIR_ID, code: None }
 139     }
 140
 141     pub fn dummy_with_span(span: Span) -> ObligationCause<'tcx> {
 142         ObligationCause { span, body_id: hir::CRATE_HIR_ID, code: None }
 143     }
 144
 145     pub fn make_mut_code(&mut self) -> &mut ObligationCauseCode<'tcx> {
 146         Lrc::make_mut(self.code.get_or_insert_with(|| Lrc::new(MISC_OBLIGATION_CAUSE_CODE)))
 147     }
 148
 149     pub fn span(&self, tcx: TyCtxt<'tcx>) -> Span {
 150         match *self.code() {
 151             ObligationCauseCode::CompareImplMethodObligation { .. }
 152             | ObligationCauseCode::MainFunctionType
 153             | ObligationCauseCode::StartFunctionType => {
 154                 tcx.sess.source_map().guess_head_span(self.span)
 155             }
 156             ObligationCauseCode::MatchExpressionArm(box MatchExpressionArmCause {
 157                 arm_span,
 158                 ..
 159             }) => arm_span,
 160             _ => self.span,
 161         }
 162     }
 163
 164     #[inline]
 165     pub fn code(&self) -> &ObligationCauseCode<'tcx> {
 166         self.code.as_deref().unwrap_or(&MISC_OBLIGATION_CAUSE_CODE)
 167     }
 168
 169     pub fn clone_code(&self) -> Lrc<ObligationCauseCode<'tcx>> {
 170         match &self.code {
 171             Some(code) => code.clone(),
 172             None => Lrc::new(MISC_OBLIGATION_CAUSE_CODE),
 173         }
 174     }
 175 }
 176
 177 #[derive(Clone, Debug, PartialEq, Eq, Hash, Lift)]
 178 pub struct UnifyReceiverContext<'tcx> {
 179     pub assoc_item: ty::AssocItem,
 180     pub param_env: ty::ParamEnv<'tcx>,
 181     pub substs: SubstsRef<'tcx>,
 182 }
 183
 184 #[derive(Clone, Debug, PartialEq, Eq, Hash, Lift)]
 185 pub enum ObligationCauseCode<'tcx> {
 186     /// Not well classified or should be obvious from the span.
 187     MiscObligation,
 188
 189     /// A slice or array is WF only if `T: Sized`.
 190     SliceOrArrayElem,
 191
 192     /// A tuple is WF only if its middle elements are `Sized`.
 193     TupleElem,
 194
 195     /// This is the trait reference from the given projection.
 196     ProjectionWf(ty::ProjectionTy<'tcx>),
 197
 198     /// In an impl of trait `X` for type `Y`, type `Y` must
 199     /// also implement all supertraits of `X`.
 200     ItemObligation(DefId),
 201
 202     /// Like `ItemObligation`, but with extra detail on the source of the obligation.
 203     BindingObligation(DefId, Span),
 204
 205     /// A type like `&'a T` is WF only if `T: 'a`.
 206     ReferenceOutlivesReferent(Ty<'tcx>),
 207
 208     /// A type like `Box<Foo<'a> + 'b>` is WF only if `'b: 'a`.
 209     ObjectTypeBound(Ty<'tcx>, ty::Region<'tcx>),
 210
 211     /// Obligation incurred due to an object cast.
 212     ObjectCastObligation(/* Object type */ Ty<'tcx>),
 213
 214     /// Obligation incurred due to a coercion.
 215     Coercion {
 216         source: Ty<'tcx>,
 217         target: Ty<'tcx>,
 218     },
 219
 220     /// Various cases where expressions must be `Sized` / `Copy` / etc.
 221     /// `L = X` implies that `L` is `Sized`.
 222     AssignmentLhsSized,
 223     /// `(x1, .., xn)` must be `Sized`.
 224     TupleInitializerSized,
 225     /// `S { ... }` must be `Sized`.
 226     StructInitializerSized,
 227     /// Type of each variable must be `Sized`.
 228     VariableType(hir::HirId),
 229     /// Argument type must be `Sized`.
 230     SizedArgumentType(Option<Span>),
 231     /// Return type must be `Sized`.
 232     SizedReturnType,
 233     /// Yield type must be `Sized`.
 234     SizedYieldType,
 235     /// Box expression result type must be `Sized`.
 236     SizedBoxType,
 237     /// Inline asm operand type must be `Sized`.
 238     InlineAsmSized,
 239     /// `[T, ..n]` implies that `T` must be `Copy`.
 240     /// If the function in the array repeat expression is a `const fn`,
 241     /// display a help message suggesting to move the function call to a
 242     /// new `const` item while saying that `T` doesn't implement `Copy`.
 243     RepeatVec(bool),
 244
 245     /// Types of fields (other than the last, except for packed structs) in a struct must be sized.
 246     FieldSized {
 247         adt_kind: AdtKind,
 248         span: Span,
 249         last: bool,
 250     },
 251
 252     /// Constant expressions must be sized.
 253     ConstSized,
 254
 255     /// `static` items must have `Sync` type.
 256     SharedStatic,
 257
 258     BuiltinDerivedObligation(DerivedObligationCause<'tcx>),
 259
 260     ImplDerivedObligation(DerivedObligationCause<'tcx>),
 261
 262     DerivedObligation(DerivedObligationCause<'tcx>),
 263
 264     FunctionArgumentObligation {
 265         /// The node of the relevant argument in the function call.
 266         arg_hir_id: hir::HirId,
 267         /// The node of the function call.
 268         call_hir_id: hir::HirId,
 269         /// The obligation introduced by this argument.
 270         parent_code: Lrc<ObligationCauseCode<'tcx>>,
 271     },
 272
 273     /// Error derived when matching traits/impls; see ObligationCause for more details
 274     CompareImplConstObligation,
 275
 276     /// Error derived when matching traits/impls; see ObligationCause for more details
 277     CompareImplMethodObligation {
 278         impl_item_def_id: DefId,
 279         trait_item_def_id: DefId,
 280     },
 281
 282     /// Error derived when matching traits/impls; see ObligationCause for more details
 283     CompareImplTypeObligation {
 284         impl_item_def_id: DefId,
 285         trait_item_def_id: DefId,
 286     },
 287
 288     /// Checking that this expression can be assigned where it needs to be
 289     // FIXME(eddyb) #11161 is the original Expr required?
 290     ExprAssignable,
 291
 292     /// Computing common supertype in the arms of a match expression
 293     MatchExpressionArm(Box<MatchExpressionArmCause<'tcx>>),
 294
 295     /// Type error arising from type checking a pattern against an expected type.
 296     Pattern {
 297         /// The span of the scrutinee or type expression which caused the `root_ty` type.
 298         span: Option<Span>,
 299         /// The root expected type induced by a scrutinee or type expression.
 300         root_ty: Ty<'tcx>,
 301         /// Whether the `Span` came from an expression or a type expression.
 302         origin_expr: bool,
 303     },
 304
 305     /// Constants in patterns must have `Structural` type.
 306     ConstPatternStructural,
 307
 308     /// Computing common supertype in an if expression
 309     IfExpression(Box<IfExpressionCause>),
 310
 311     /// Computing common supertype of an if expression with no else counter-part
 312     IfExpressionWithNoElse,
 313
 314     /// `main` has wrong type
 315     MainFunctionType,
 316
 317     /// `start` has wrong type
 318     StartFunctionType,
 319
 320     /// Intrinsic has wrong type
 321     IntrinsicType,
 322
 323     /// A let else block does not diverge
 324     LetElse,
 325
 326     /// Method receiver
 327     MethodReceiver,
 328
 329     UnifyReceiver(Box<UnifyReceiverContext<'tcx>>),
 330
 331     /// `return` with no expression
 332     ReturnNoExpression,
 333
 334     /// `return` with an expression
 335     ReturnValue(hir::HirId),
 336
 337     /// Return type of this function
 338     ReturnType,
 339
 340     /// Block implicit return
 341     BlockTailExpression(hir::HirId),
 342
 343     /// #[feature(trivial_bounds)] is not enabled
 344     TrivialBound,
 345
 346     /// If `X` is the concrete type of an opaque type `impl Y`, then `X` must implement `Y`
 347     OpaqueType,
 348
 349     AwaitableExpr(Option<hir::HirId>),
 350
 351     ForLoopIterator,
 352
 353     QuestionMark,
 354
 355     /// Well-formed checking. If a `WellFormedLoc` is provided,
 356     /// then it will be used to eprform HIR-based wf checking
 357     /// after an error occurs, in order to generate a more precise error span.
 358     /// This is purely for diagnostic purposes - it is always
 359     /// correct to use `MiscObligation` instead, or to specify
 360     /// `WellFormed(None)`
 361     WellFormed(Option<WellFormedLoc>),
 362
 363     /// From `match_impl`. The cause for us having to match an impl, and the DefId we are matching against.
 364     MatchImpl(ObligationCause<'tcx>, DefId),
 365 }
 366
 367 /// The 'location' at which we try to perform HIR-based wf checking.
 368 /// This information is used to obtain an `hir::Ty`, which
 369 /// we can walk in order to obtain precise spans for any
 370 /// 'nested' types (e.g. `Foo` in `Option<Foo>`).
 371 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, HashStable)]
 372 pub enum WellFormedLoc {
 373     /// Use the type of the provided definition.
 374     Ty(LocalDefId),
 375     /// Use the type of the parameter of the provided function.
 376     /// We cannot use `hir::Param`, since the function may
 377     /// not have a body (e.g. a trait method definition)
 378     Param {
 379         /// The function to lookup the parameter in
 380         function: LocalDefId,
 381         /// The index of the parameter to use.
 382         /// Parameters are indexed from 0, with the return type
 383         /// being the last 'parameter'
 384         param_idx: u16,
 385     },
 386 }
 387
 388 impl ObligationCauseCode<'_> {
 389     // Return the base obligation, ignoring derived obligations.
 390     pub fn peel_derives(&self) -> &Self {
 391         let mut base_cause = self;
 392         while let BuiltinDerivedObligation(DerivedObligationCause { parent_code, .. })
 393         | ImplDerivedObligation(DerivedObligationCause { parent_code, .. })
 394         | DerivedObligation(DerivedObligationCause { parent_code, .. })
 395         | FunctionArgumentObligation { parent_code, .. } = base_cause
 396         {
 397             base_cause = &parent_code;
 398         }
 399         base_cause
 400     }
 401 }
 402
 403 // `ObligationCauseCode` is used a lot. Make sure it doesn't unintentionally get bigger.
 404 #[cfg(all(target_arch = "x86_64", target_pointer_width = "64"))]
 405 static_assert_size!(ObligationCauseCode<'_>, 48);
 406
 407 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
 408 pub enum StatementAsExpression {
 409     CorrectType,
 410     NeedsBoxing,
 411 }
 412
 413 impl<'tcx> ty::Lift<'tcx> for StatementAsExpression {
 414     type Lifted = StatementAsExpression;
 415     fn lift_to_tcx(self, _tcx: TyCtxt<'tcx>) -> Option<StatementAsExpression> {
 416         Some(self)
 417     }
 418 }
 419
 420 #[derive(Clone, Debug, PartialEq, Eq, Hash, Lift)]
 421 pub struct MatchExpressionArmCause<'tcx> {
 422     pub arm_span: Span,
 423     pub scrut_span: Span,
 424     pub semi_span: Option<(Span, StatementAsExpression)>,
 425     pub source: hir::MatchSource,
 426     pub prior_arms: Vec<Span>,
 427     pub last_ty: Ty<'tcx>,
 428     pub scrut_hir_id: hir::HirId,
 429     pub opt_suggest_box_span: Option<Span>,
 430 }
 431
 432 #[derive(Clone, Debug, PartialEq, Eq, Hash)]
 433 pub struct IfExpressionCause {
 434     pub then: Span,
 435     pub else_sp: Span,
 436     pub outer: Option<Span>,
 437     pub semicolon: Option<(Span, StatementAsExpression)>,
 438     pub opt_suggest_box_span: Option<Span>,
 439 }
 440
 441 #[derive(Clone, Debug, PartialEq, Eq, Hash, Lift)]
 442 pub struct DerivedObligationCause<'tcx> {
 443     /// The trait predicate of the parent obligation that led to the
 444     /// current obligation. Note that only trait obligations lead to
 445     /// derived obligations, so we just store the trait predicate here
 446     /// directly.
 447     pub parent_trait_pred: ty::PolyTraitPredicate<'tcx>,
 448
 449     /// The parent trait had this cause.
 450     pub parent_code: Lrc<ObligationCauseCode<'tcx>>,
 451 }
 452
 453 #[derive(Clone, Debug, TypeFoldable, Lift)]
 454 pub enum SelectionError<'tcx> {
 455     /// The trait is not implemented.
 456     Unimplemented,
 457     /// After a closure impl has selected, its "outputs" were evaluated
 458     /// (which for closures includes the "input" type params) and they
 459     /// didn't resolve. See `confirm_poly_trait_refs` for more.
 460     OutputTypeParameterMismatch(
 461         ty::PolyTraitRef<'tcx>,
 462         ty::PolyTraitRef<'tcx>,
 463         ty::error::TypeError<'tcx>,
 464     ),
 465     /// The trait pointed by `DefId` is not object safe.
 466     TraitNotObjectSafe(DefId),
 467     /// A given constant couldn't be evaluated.
 468     NotConstEvaluatable(NotConstEvaluatable),
 469     /// Exceeded the recursion depth during type projection.
 470     Overflow,
 471     /// Signaling that an error has already been emitted, to avoid
 472     /// multiple errors being shown.
 473     ErrorReporting,
 474     /// Multiple applicable `impl`s where found. The `DefId`s correspond to
 475     /// all the `impl`s' Items.
 476     Ambiguous(Vec<DefId>),
 477 }
 478
 479 /// When performing resolution, it is typically the case that there
 480 /// can be one of three outcomes:
 481 ///
 482 /// - `Ok(Some(r))`: success occurred with result `r`
 483 /// - `Ok(None)`: could not definitely determine anything, usually due
 484 ///   to inconclusive type inference.
 485 /// - `Err(e)`: error `e` occurred
 486 pub type SelectionResult<'tcx, T> = Result<Option<T>, SelectionError<'tcx>>;
 487
 488 /// Given the successful resolution of an obligation, the `ImplSource`
 489 /// indicates where the impl comes from.
 490 ///
 491 /// For example, the obligation may be satisfied by a specific impl (case A),
 492 /// or it may be relative to some bound that is in scope (case B).
 493 ///
 494 /// ```
 495 /// impl<T:Clone> Clone<T> for Option<T> { ... } // Impl_1
 496 /// impl<T:Clone> Clone<T> for Box<T> { ... }    // Impl_2
 497 /// impl Clone for i32 { ... }                   // Impl_3
 498 ///
 499 /// fn foo<T: Clone>(concrete: Option<Box<i32>>, param: T, mixed: Option<T>) {
 500 ///     // Case A: ImplSource points at a specific impl. Only possible when
 501 ///     // type is concretely known. If the impl itself has bounded
 502 ///     // type parameters, ImplSource will carry resolutions for those as well:
 503 ///     concrete.clone(); // ImpleSource(Impl_1, [ImplSource(Impl_2, [ImplSource(Impl_3)])])
 504 ///
 505 ///     // Case A: ImplSource points at a specific impl. Only possible when
 506 ///     // type is concretely known. If the impl itself has bounded
 507 ///     // type parameters, ImplSource will carry resolutions for those as well:
 508 ///     concrete.clone(); // ImplSource(Impl_1, [ImplSource(Impl_2, [ImplSource(Impl_3)])])
 509 ///
 510 ///     // Case B: ImplSource must be provided by caller. This applies when
 511 ///     // type is a type parameter.
 512 ///     param.clone();    // ImplSource::Param
 513 ///
 514 ///     // Case C: A mix of cases A and B.
 515 ///     mixed.clone();    // ImplSource(Impl_1, [ImplSource::Param])
 516 /// }
 517 /// ```
 518 ///
 519 /// ### The type parameter `N`
 520 ///
 521 /// See explanation on `ImplSourceUserDefinedData`.
 522 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 523 pub enum ImplSource<'tcx, N> {
 524     /// ImplSource identifying a particular impl.
 525     UserDefined(ImplSourceUserDefinedData<'tcx, N>),
 526
 527     /// ImplSource for auto trait implementations.
 528     /// This carries the information and nested obligations with regards
 529     /// to an auto implementation for a trait `Trait`. The nested obligations
 530     /// ensure the trait implementation holds for all the constituent types.
 531     AutoImpl(ImplSourceAutoImplData<N>),
 532
 533     /// Successful resolution to an obligation provided by the caller
 534     /// for some type parameter. The `Vec<N>` represents the
 535     /// obligations incurred from normalizing the where-clause (if
 536     /// any).
 537     Param(Vec<N>, ty::BoundConstness),
 538
 539     /// Virtual calls through an object.
 540     Object(ImplSourceObjectData<'tcx, N>),
 541
 542     /// Successful resolution for a builtin trait.
 543     Builtin(ImplSourceBuiltinData<N>),
 544
 545     /// ImplSource for trait upcasting coercion
 546     TraitUpcasting(ImplSourceTraitUpcastingData<'tcx, N>),
 547
 548     /// ImplSource automatically generated for a closure. The `DefId` is the ID
 549     /// of the closure expression. This is an `ImplSource::UserDefined` in spirit, but the
 550     /// impl is generated by the compiler and does not appear in the source.
 551     Closure(ImplSourceClosureData<'tcx, N>),
 552
 553     /// Same as above, but for a function pointer type with the given signature.
 554     FnPointer(ImplSourceFnPointerData<'tcx, N>),
 555
 556     /// ImplSource for a builtin `DeterminantKind` trait implementation.
 557     DiscriminantKind(ImplSourceDiscriminantKindData),
 558
 559     /// ImplSource for a builtin `Pointee` trait implementation.
 560     Pointee(ImplSourcePointeeData),
 561
 562     /// ImplSource automatically generated for a generator.
 563     Generator(ImplSourceGeneratorData<'tcx, N>),
 564
 565     /// ImplSource for a trait alias.
 566     TraitAlias(ImplSourceTraitAliasData<'tcx, N>),
 567
 568     /// ImplSource for a `const Drop` implementation.
 569     ConstDrop(ImplSourceConstDropData<N>),
 570 }
 571
 572 impl<'tcx, N> ImplSource<'tcx, N> {
 573     pub fn nested_obligations(self) -> Vec<N> {
 574         match self {
 575             ImplSource::UserDefined(i) => i.nested,
 576             ImplSource::Param(n, _) => n,
 577             ImplSource::Builtin(i) => i.nested,
 578             ImplSource::AutoImpl(d) => d.nested,
 579             ImplSource::Closure(c) => c.nested,
 580             ImplSource::Generator(c) => c.nested,
 581             ImplSource::Object(d) => d.nested,
 582             ImplSource::FnPointer(d) => d.nested,
 583             ImplSource::DiscriminantKind(ImplSourceDiscriminantKindData)
 584             | ImplSource::Pointee(ImplSourcePointeeData) => Vec::new(),
 585             ImplSource::TraitAlias(d) => d.nested,
 586             ImplSource::TraitUpcasting(d) => d.nested,
 587             ImplSource::ConstDrop(i) => i.nested,
 588         }
 589     }
 590
 591     pub fn borrow_nested_obligations(&self) -> &[N] {
 592         match &self {
 593             ImplSource::UserDefined(i) => &i.nested[..],
 594             ImplSource::Param(n, _) => &n,
 595             ImplSource::Builtin(i) => &i.nested,
 596             ImplSource::AutoImpl(d) => &d.nested,
 597             ImplSource::Closure(c) => &c.nested,
 598             ImplSource::Generator(c) => &c.nested,
 599             ImplSource::Object(d) => &d.nested,
 600             ImplSource::FnPointer(d) => &d.nested,
 601             ImplSource::DiscriminantKind(ImplSourceDiscriminantKindData)
 602             | ImplSource::Pointee(ImplSourcePointeeData) => &[],
 603             ImplSource::TraitAlias(d) => &d.nested,
 604             ImplSource::TraitUpcasting(d) => &d.nested,
 605             ImplSource::ConstDrop(i) => &i.nested,
 606         }
 607     }
 608
 609     pub fn map<M, F>(self, f: F) -> ImplSource<'tcx, M>
 610     where
 611         F: FnMut(N) -> M,
 612     {
 613         match self {
 614             ImplSource::UserDefined(i) => ImplSource::UserDefined(ImplSourceUserDefinedData {
 615                 impl_def_id: i.impl_def_id,
 616                 substs: i.substs,
 617                 nested: i.nested.into_iter().map(f).collect(),
 618             }),
 619             ImplSource::Param(n, ct) => ImplSource::Param(n.into_iter().map(f).collect(), ct),
 620             ImplSource::Builtin(i) => ImplSource::Builtin(ImplSourceBuiltinData {
 621                 nested: i.nested.into_iter().map(f).collect(),
 622             }),
 623             ImplSource::Object(o) => ImplSource::Object(ImplSourceObjectData {
 624                 upcast_trait_ref: o.upcast_trait_ref,
 625                 vtable_base: o.vtable_base,
 626                 nested: o.nested.into_iter().map(f).collect(),
 627             }),
 628             ImplSource::AutoImpl(d) => ImplSource::AutoImpl(ImplSourceAutoImplData {
 629                 trait_def_id: d.trait_def_id,
 630                 nested: d.nested.into_iter().map(f).collect(),
 631             }),
 632             ImplSource::Closure(c) => ImplSource::Closure(ImplSourceClosureData {
 633                 closure_def_id: c.closure_def_id,
 634                 substs: c.substs,
 635                 nested: c.nested.into_iter().map(f).collect(),
 636             }),
 637             ImplSource::Generator(c) => ImplSource::Generator(ImplSourceGeneratorData {
 638                 generator_def_id: c.generator_def_id,
 639                 substs: c.substs,
 640                 nested: c.nested.into_iter().map(f).collect(),
 641             }),
 642             ImplSource::FnPointer(p) => ImplSource::FnPointer(ImplSourceFnPointerData {
 643                 fn_ty: p.fn_ty,
 644                 nested: p.nested.into_iter().map(f).collect(),
 645             }),
 646             ImplSource::DiscriminantKind(ImplSourceDiscriminantKindData) => {
 647                 ImplSource::DiscriminantKind(ImplSourceDiscriminantKindData)
 648             }
 649             ImplSource::Pointee(ImplSourcePointeeData) => {
 650                 ImplSource::Pointee(ImplSourcePointeeData)
 651             }
 652             ImplSource::TraitAlias(d) => ImplSource::TraitAlias(ImplSourceTraitAliasData {
 653                 alias_def_id: d.alias_def_id,
 654                 substs: d.substs,
 655                 nested: d.nested.into_iter().map(f).collect(),
 656             }),
 657             ImplSource::TraitUpcasting(d) => {
 658                 ImplSource::TraitUpcasting(ImplSourceTraitUpcastingData {
 659                     upcast_trait_ref: d.upcast_trait_ref,
 660                     vtable_vptr_slot: d.vtable_vptr_slot,
 661                     nested: d.nested.into_iter().map(f).collect(),
 662                 })
 663             }
 664             ImplSource::ConstDrop(i) => ImplSource::ConstDrop(ImplSourceConstDropData {
 665                 nested: i.nested.into_iter().map(f).collect(),
 666             }),
 667         }
 668     }
 669 }
 670
 671 /// Identifies a particular impl in the source, along with a set of
 672 /// substitutions from the impl's type/lifetime parameters. The
 673 /// `nested` vector corresponds to the nested obligations attached to
 674 /// the impl's type parameters.
 675 ///
 676 /// The type parameter `N` indicates the type used for "nested
 677 /// obligations" that are required by the impl. During type-check, this
 678 /// is `Obligation`, as one might expect. During codegen, however, this
 679 /// is `()`, because codegen only requires a shallow resolution of an
 680 /// impl, and nested obligations are satisfied later.
 681 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 682 pub struct ImplSourceUserDefinedData<'tcx, N> {
 683     pub impl_def_id: DefId,
 684     pub substs: SubstsRef<'tcx>,
 685     pub nested: Vec<N>,
 686 }
 687
 688 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 689 pub struct ImplSourceGeneratorData<'tcx, N> {
 690     pub generator_def_id: DefId,
 691     pub substs: SubstsRef<'tcx>,
 692     /// Nested obligations. This can be non-empty if the generator
 693     /// signature contains associated types.
 694     pub nested: Vec<N>,
 695 }
 696
 697 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 698 pub struct ImplSourceClosureData<'tcx, N> {
 699     pub closure_def_id: DefId,
 700     pub substs: SubstsRef<'tcx>,
 701     /// Nested obligations. This can be non-empty if the closure
 702     /// signature contains associated types.
 703     pub nested: Vec<N>,
 704 }
 705
 706 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 707 pub struct ImplSourceAutoImplData<N> {
 708     pub trait_def_id: DefId,
 709     pub nested: Vec<N>,
 710 }
 711
 712 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 713 pub struct ImplSourceTraitUpcastingData<'tcx, N> {
 714     /// `Foo` upcast to the obligation trait. This will be some supertrait of `Foo`.
 715     pub upcast_trait_ref: ty::PolyTraitRef<'tcx>,
 716
 717     /// The vtable is formed by concatenating together the method lists of
 718     /// the base object trait and all supertraits, pointers to supertrait vtable will
 719     /// be provided when necessary; this is the position of `upcast_trait_ref`'s vtable
 720     /// within that vtable.
 721     pub vtable_vptr_slot: Option<usize>,
 722
 723     pub nested: Vec<N>,
 724 }
 725
 726 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 727 pub struct ImplSourceBuiltinData<N> {
 728     pub nested: Vec<N>,
 729 }
 730
 731 #[derive(PartialEq, Eq, Clone, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 732 pub struct ImplSourceObjectData<'tcx, N> {
 733     /// `Foo` upcast to the obligation trait. This will be some supertrait of `Foo`.
 734     pub upcast_trait_ref: ty::PolyTraitRef<'tcx>,
 735
 736     /// The vtable is formed by concatenating together the method lists of
 737     /// the base object trait and all supertraits, pointers to supertrait vtable will
 738     /// be provided when necessary; this is the start of `upcast_trait_ref`'s methods
 739     /// in that vtable.
 740     pub vtable_base: usize,
 741
 742     pub nested: Vec<N>,
 743 }
 744
 745 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 746 pub struct ImplSourceFnPointerData<'tcx, N> {
 747     pub fn_ty: Ty<'tcx>,
 748     pub nested: Vec<N>,
 749 }
 750
 751 // FIXME(@lcnr): This should be  refactored and merged with other builtin vtables.
 752 #[derive(Clone, Debug, PartialEq, Eq, TyEncodable, TyDecodable, HashStable)]
 753 pub struct ImplSourceDiscriminantKindData;
 754
 755 #[derive(Clone, Debug, PartialEq, Eq, TyEncodable, TyDecodable, HashStable)]
 756 pub struct ImplSourcePointeeData;
 757
 758 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 759 pub struct ImplSourceConstDropData<N> {
 760     pub nested: Vec<N>,
 761 }
 762
 763 #[derive(Clone, PartialEq, Eq, TyEncodable, TyDecodable, HashStable, TypeFoldable, Lift)]
 764 pub struct ImplSourceTraitAliasData<'tcx, N> {
 765     pub alias_def_id: DefId,
 766     pub substs: SubstsRef<'tcx>,
 767     pub nested: Vec<N>,
 768 }
 769
 770 #[derive(Clone, Debug, PartialEq, Eq, Hash, HashStable, PartialOrd, Ord)]
 771 pub enum ObjectSafetyViolation {
 772     /// `Self: Sized` declared on the trait.
 773     SizedSelf(SmallVec<[Span; 1]>),
 774
 775     /// Supertrait reference references `Self` an in illegal location
 776     /// (e.g., `trait Foo : Bar<Self>`).
 777     SupertraitSelf(SmallVec<[Span; 1]>),
 778
 779     /// Method has something illegal.
 780     Method(Symbol, MethodViolationCode, Span),
 781
 782     /// Associated const.
 783     AssocConst(Symbol, Span),
 784
 785     /// GAT
 786     GAT(Symbol, Span),
 787 }
 788
 789 impl ObjectSafetyViolation {
 790     pub fn error_msg(&self) -> Cow<'static, str> {
 791         match *self {
 792             ObjectSafetyViolation::SizedSelf(_) => "it requires `Self: Sized`".into(),
 793             ObjectSafetyViolation::SupertraitSelf(ref spans) => {
 794                 if spans.iter().any(|sp| *sp != DUMMY_SP) {
 795                     "it uses `Self` as a type parameter".into()
 796                 } else {
 797                     "it cannot use `Self` as a type parameter in a supertrait or `where`-clause"
 798                         .into()
 799                 }
 800             }
 801             ObjectSafetyViolation::Method(name, MethodViolationCode::StaticMethod(_, _, _), _) => {
 802                 format!("associated function `{}` has no `self` parameter", name).into()
 803             }
 804             ObjectSafetyViolation::Method(
 805                 name,
 806                 MethodViolationCode::ReferencesSelfInput(_),
 807                 DUMMY_SP,
 808             ) => format!("method `{}` references the `Self` type in its parameters", name).into(),
 809             ObjectSafetyViolation::Method(name, MethodViolationCode::ReferencesSelfInput(_), _) => {
 810                 format!("method `{}` references the `Self` type in this parameter", name).into()
 811             }
 812             ObjectSafetyViolation::Method(name, MethodViolationCode::ReferencesSelfOutput, _) => {
 813                 format!("method `{}` references the `Self` type in its return type", name).into()
 814             }
 815             ObjectSafetyViolation::Method(
 816                 name,
 817                 MethodViolationCode::WhereClauseReferencesSelf,
 818                 _,
 819             ) => {
 820                 format!("method `{}` references the `Self` type in its `where` clause", name).into()
 821             }
 822             ObjectSafetyViolation::Method(name, MethodViolationCode::Generic, _) => {
 823                 format!("method `{}` has generic type parameters", name).into()
 824             }
 825             ObjectSafetyViolation::Method(name, MethodViolationCode::UndispatchableReceiver, _) => {
 826                 format!("method `{}`'s `self` parameter cannot be dispatched on", name).into()
 827             }
 828             ObjectSafetyViolation::AssocConst(name, DUMMY_SP) => {
 829                 format!("it contains associated `const` `{}`", name).into()
 830             }
 831             ObjectSafetyViolation::AssocConst(..) => "it contains this associated `const`".into(),
 832             ObjectSafetyViolation::GAT(name, _) => {
 833                 format!("it contains the generic associated type `{}`", name).into()
 834             }
 835         }
 836     }
 837
 838     pub fn solution(&self, err: &mut DiagnosticBuilder<'_>) {
 839         match *self {
 840             ObjectSafetyViolation::SizedSelf(_) | ObjectSafetyViolation::SupertraitSelf(_) => {}
 841             ObjectSafetyViolation::Method(
 842                 name,
 843                 MethodViolationCode::StaticMethod(sugg, self_span, has_args),
 844                 _,
 845             ) => {
 846                 err.span_suggestion(
 847                     self_span,
 848                     &format!(
 849                         "consider turning `{}` into a method by giving it a `&self` argument",
 850                         name
 851                     ),
 852                     format!("&self{}", if has_args { ", " } else { "" }),
 853                     Applicability::MaybeIncorrect,
 854                 );
 855                 match sugg {
 856                     Some((sugg, span)) => {
 857                         err.span_suggestion(
 858                             span,
 859                             &format!(
 860                                 "alternatively, consider constraining `{}` so it does not apply to \
 861                                  trait objects",
 862                                 name
 863                             ),
 864                             sugg.to_string(),
 865                             Applicability::MaybeIncorrect,
 866                         );
 867                     }
 868                     None => {
 869                         err.help(&format!(
 870                             "consider turning `{}` into a method by giving it a `&self` \
 871                              argument or constraining it so it does not apply to trait objects",
 872                             name
 873                         ));
 874                     }
 875                 }
 876             }
 877             ObjectSafetyViolation::Method(
 878                 name,
 879                 MethodViolationCode::UndispatchableReceiver,
 880                 span,
 881             ) => {
 882                 err.span_suggestion(
 883                     span,
 884                     &format!(
 885                         "consider changing method `{}`'s `self` parameter to be `&self`",
 886                         name
 887                     ),
 888                     "&Self".to_string(),
 889                     Applicability::MachineApplicable,
 890                 );
 891             }
 892             ObjectSafetyViolation::AssocConst(name, _)
 893             | ObjectSafetyViolation::GAT(name, _)
 894             | ObjectSafetyViolation::Method(name, ..) => {
 895                 err.help(&format!("consider moving `{}` to another trait", name));
 896             }
 897         }
 898     }
 899
 900     pub fn spans(&self) -> SmallVec<[Span; 1]> {
 901         // When `span` comes from a separate crate, it'll be `DUMMY_SP`. Treat it as `None` so
 902         // diagnostics use a `note` instead of a `span_label`.
 903         match self {
 904             ObjectSafetyViolation::SupertraitSelf(spans)
 905             | ObjectSafetyViolation::SizedSelf(spans) => spans.clone(),
 906             ObjectSafetyViolation::AssocConst(_, span)
 907             | ObjectSafetyViolation::GAT(_, span)
 908             | ObjectSafetyViolation::Method(_, _, span)
 909                 if *span != DUMMY_SP =>
 910             {
 911                 smallvec![*span]
 912             }
 913             _ => smallvec![],
 914         }
 915     }
 916 }
 917
 918 /// Reasons a method might not be object-safe.
 919 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, HashStable, PartialOrd, Ord)]
 920 pub enum MethodViolationCode {
 921     /// e.g., `fn foo()`
 922     StaticMethod(Option<(&'static str, Span)>, Span, bool /* has args */),
 923
 924     /// e.g., `fn foo(&self, x: Self)`
 925     ReferencesSelfInput(usize),
 926
 927     /// e.g., `fn foo(&self) -> Self`
 928     ReferencesSelfOutput,
 929
 930     /// e.g., `fn foo(&self) where Self: Clone`
 931     WhereClauseReferencesSelf,
 932
 933     /// e.g., `fn foo<A>()`
 934     Generic,
 935
 936     /// the method's receiver (`self` argument) can't be dispatched on
 937     UndispatchableReceiver,
 938 }