1use std::ops::Deref;
2
3use rustc_hir as hir;
4use rustc_hir::GenericArg;
5use rustc_hir::def_id::DefId;
6use rustc_hir_analysis::hir_ty_lowering::generics::{
7 check_generic_arg_count_for_call, lower_generic_args,
8};
9use rustc_hir_analysis::hir_ty_lowering::{
10 FeedConstTy, GenericArgsLowerer, HirTyLowerer, IsMethodCall, RegionInferReason,
11};
12use rustc_infer::infer::{self, DefineOpaqueTypes, InferOk};
13use rustc_lint::builtin::SUPERTRAIT_ITEM_SHADOWING_USAGE;
14use rustc_middle::traits::ObligationCauseCode;
15use rustc_middle::ty::adjustment::{
16 Adjust, Adjustment, AllowTwoPhase, AutoBorrow, AutoBorrowMutability, PointerCoercion,
17};
18use rustc_middle::ty::{
19 self, GenericArgs, GenericArgsRef, GenericParamDefKind, Ty, TyCtxt, TypeFoldable,
20 TypeVisitableExt, UserArgs,
21};
22use rustc_middle::{bug, span_bug};
23use rustc_span::{DUMMY_SP, Span};
24use rustc_trait_selection::traits;
25use tracing::debug;
26
27use super::{MethodCallee, probe};
28use crate::errors::{SupertraitItemShadowee, SupertraitItemShadower, SupertraitItemShadowing};
29use crate::{FnCtxt, callee};
30
31struct ConfirmContext<'a, 'tcx> {
32 fcx: &'a FnCtxt<'a, 'tcx>,
33 span: Span,
34 self_expr: &'tcx hir::Expr<'tcx>,
35 call_expr: &'tcx hir::Expr<'tcx>,
36 skip_record_for_diagnostics: bool,
37}
38
39impl<'a, 'tcx> Deref for ConfirmContext<'a, 'tcx> {
40 type Target = FnCtxt<'a, 'tcx>;
41 fn deref(&self) -> &Self::Target {
42 self.fcx
43 }
44}
45
46#[derive(Debug)]
47pub(crate) struct ConfirmResult<'tcx> {
48 pub callee: MethodCallee<'tcx>,
49 pub illegal_sized_bound: Option<Span>,
50}
51
52impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
53 pub(crate) fn confirm_method(
54 &self,
55 span: Span,
56 self_expr: &'tcx hir::Expr<'tcx>,
57 call_expr: &'tcx hir::Expr<'tcx>,
58 unadjusted_self_ty: Ty<'tcx>,
59 pick: &probe::Pick<'tcx>,
60 segment: &'tcx hir::PathSegment<'tcx>,
61 ) -> ConfirmResult<'tcx> {
62 debug!(
63 "confirm(unadjusted_self_ty={:?}, pick={:?}, generic_args={:?})",
64 unadjusted_self_ty, pick, segment.args,
65 );
66
67 let mut confirm_cx = ConfirmContext::new(self, span, self_expr, call_expr);
68 confirm_cx.confirm(unadjusted_self_ty, pick, segment)
69 }
70
71 pub(crate) fn confirm_method_for_diagnostic(
72 &self,
73 span: Span,
74 self_expr: &'tcx hir::Expr<'tcx>,
75 call_expr: &'tcx hir::Expr<'tcx>,
76 unadjusted_self_ty: Ty<'tcx>,
77 pick: &probe::Pick<'tcx>,
78 segment: &hir::PathSegment<'tcx>,
79 ) -> ConfirmResult<'tcx> {
80 let mut confirm_cx = ConfirmContext::new(self, span, self_expr, call_expr);
81 confirm_cx.skip_record_for_diagnostics = true;
82 confirm_cx.confirm(unadjusted_self_ty, pick, segment)
83 }
84}
85
86impl<'a, 'tcx> ConfirmContext<'a, 'tcx> {
87 fn new(
88 fcx: &'a FnCtxt<'a, 'tcx>,
89 span: Span,
90 self_expr: &'tcx hir::Expr<'tcx>,
91 call_expr: &'tcx hir::Expr<'tcx>,
92 ) -> ConfirmContext<'a, 'tcx> {
93 ConfirmContext { fcx, span, self_expr, call_expr, skip_record_for_diagnostics: false }
94 }
95
96 fn confirm(
97 &mut self,
98 unadjusted_self_ty: Ty<'tcx>,
99 pick: &probe::Pick<'tcx>,
100 segment: &hir::PathSegment<'tcx>,
101 ) -> ConfirmResult<'tcx> {
102 let self_ty = self.adjust_self_ty(unadjusted_self_ty, pick);
104
105 let rcvr_args = self.fresh_receiver_args(self_ty, pick);
107 let all_args = self.instantiate_method_args(pick, segment, rcvr_args);
108
109 debug!("rcvr_args={rcvr_args:?}, all_args={all_args:?}");
110
111 let (method_sig, method_predicates) = self.instantiate_method_sig(pick, all_args);
113
114 let filler_args = rcvr_args
123 .extend_to(self.tcx, pick.item.def_id, |def, _| self.tcx.mk_param_from_def(def));
124 let illegal_sized_bound = self.predicates_require_illegal_sized_bound(
125 self.tcx.predicates_of(pick.item.def_id).instantiate(self.tcx, filler_args),
126 );
127
128 let method_sig_rcvr = self.normalize(self.span, method_sig.inputs()[0]);
135 debug!(
136 "confirm: self_ty={:?} method_sig_rcvr={:?} method_sig={:?} method_predicates={:?}",
137 self_ty, method_sig_rcvr, method_sig, method_predicates
138 );
139 self.unify_receivers(self_ty, method_sig_rcvr, pick);
140
141 let (method_sig, method_predicates) =
142 self.normalize(self.span, (method_sig, method_predicates));
143 let method_sig = ty::Binder::dummy(method_sig);
144
145 self.check_for_illegal_method_calls(pick);
147
148 self.lint_shadowed_supertrait_items(pick, segment);
150
151 if illegal_sized_bound.is_none() {
155 self.add_obligations(
156 Ty::new_fn_ptr(self.tcx, method_sig),
157 all_args,
158 method_predicates,
159 pick.item.def_id,
160 );
161 }
162
163 let callee = MethodCallee {
165 def_id: pick.item.def_id,
166 args: all_args,
167 sig: method_sig.skip_binder(),
168 };
169 ConfirmResult { callee, illegal_sized_bound }
170 }
171
172 fn adjust_self_ty(
176 &mut self,
177 unadjusted_self_ty: Ty<'tcx>,
178 pick: &probe::Pick<'tcx>,
179 ) -> Ty<'tcx> {
180 let mut autoderef = self.autoderef(self.call_expr.span, unadjusted_self_ty);
183 let Some((ty, n)) = autoderef.nth(pick.autoderefs) else {
184 return Ty::new_error_with_message(
185 self.tcx,
186 DUMMY_SP,
187 format!("failed autoderef {}", pick.autoderefs),
188 );
189 };
190 assert_eq!(n, pick.autoderefs);
191
192 let mut adjustments = self.adjust_steps(&autoderef);
193 let mut target = self.structurally_resolve_type(autoderef.span(), ty);
194
195 match pick.autoref_or_ptr_adjustment {
196 Some(probe::AutorefOrPtrAdjustment::Autoref { mutbl, unsize }) => {
197 let region = self.next_region_var(infer::Autoref(self.span));
198 let base_ty = target;
200
201 target = Ty::new_ref(self.tcx, region, target, mutbl);
202
203 let mutbl = AutoBorrowMutability::new(mutbl, AllowTwoPhase::Yes);
206
207 adjustments
208 .push(Adjustment { kind: Adjust::Borrow(AutoBorrow::Ref(mutbl)), target });
209
210 if unsize {
211 let unsized_ty = if let ty::Array(elem_ty, _) = base_ty.kind() {
212 Ty::new_slice(self.tcx, *elem_ty)
213 } else {
214 bug!(
215 "AutorefOrPtrAdjustment's unsize flag should only be set for array ty, found {}",
216 base_ty
217 )
218 };
219 target = Ty::new_ref(self.tcx, region, unsized_ty, mutbl.into());
220 adjustments.push(Adjustment {
221 kind: Adjust::Pointer(PointerCoercion::Unsize),
222 target,
223 });
224 }
225 }
226 Some(probe::AutorefOrPtrAdjustment::ToConstPtr) => {
227 target = match target.kind() {
228 &ty::RawPtr(ty, mutbl) => {
229 assert!(mutbl.is_mut());
230 Ty::new_imm_ptr(self.tcx, ty)
231 }
232 other => panic!("Cannot adjust receiver type {other:?} to const ptr"),
233 };
234
235 adjustments.push(Adjustment {
236 kind: Adjust::Pointer(PointerCoercion::MutToConstPointer),
237 target,
238 });
239 }
240
241 Some(probe::AutorefOrPtrAdjustment::ReborrowPin(mutbl)) => {
242 let region = self.next_region_var(infer::Autoref(self.span));
243
244 target = match target.kind() {
245 ty::Adt(pin, args) if self.tcx.is_lang_item(pin.did(), hir::LangItem::Pin) => {
246 let inner_ty = match args[0].expect_ty().kind() {
247 ty::Ref(_, ty, _) => *ty,
248 _ => bug!("Expected a reference type for argument to Pin"),
249 };
250 Ty::new_pinned_ref(self.tcx, region, inner_ty, mutbl)
251 }
252 _ => bug!("Cannot adjust receiver type for reborrowing pin of {target:?}"),
253 };
254
255 adjustments.push(Adjustment { kind: Adjust::ReborrowPin(mutbl), target });
256 }
257 None => {}
258 }
259
260 self.register_predicates(autoderef.into_obligations());
261
262 if !self.skip_record_for_diagnostics {
264 self.apply_adjustments(self.self_expr, adjustments);
265 }
266
267 target
268 }
269
270 fn fresh_receiver_args(
277 &mut self,
278 self_ty: Ty<'tcx>,
279 pick: &probe::Pick<'tcx>,
280 ) -> GenericArgsRef<'tcx> {
281 match pick.kind {
282 probe::InherentImplPick => {
283 let impl_def_id = pick.item.container_id(self.tcx);
284 assert!(
285 self.tcx.impl_trait_ref(impl_def_id).is_none(),
286 "impl {impl_def_id:?} is not an inherent impl"
287 );
288 self.fresh_args_for_item(self.span, impl_def_id)
289 }
290
291 probe::ObjectPick => {
292 let trait_def_id = pick.item.container_id(self.tcx);
293
294 if self_ty.references_error() {
301 return ty::GenericArgs::extend_with_error(self.tcx, trait_def_id, &[]);
302 }
303
304 self.extract_existential_trait_ref(self_ty, |this, object_ty, principal| {
305 let original_poly_trait_ref = principal.with_self_ty(this.tcx, object_ty);
316 let upcast_poly_trait_ref = this.upcast(original_poly_trait_ref, trait_def_id);
317 let upcast_trait_ref =
318 this.instantiate_binder_with_fresh_vars(upcast_poly_trait_ref);
319 debug!(
320 "original_poly_trait_ref={:?} upcast_trait_ref={:?} target_trait={:?}",
321 original_poly_trait_ref, upcast_trait_ref, trait_def_id
322 );
323 upcast_trait_ref.args
324 })
325 }
326
327 probe::TraitPick => {
328 let trait_def_id = pick.item.container_id(self.tcx);
329
330 self.fresh_args_for_item(self.span, trait_def_id)
336 }
337
338 probe::WhereClausePick(poly_trait_ref) => {
339 self.instantiate_binder_with_fresh_vars(poly_trait_ref).args
342 }
343 }
344 }
345
346 fn extract_existential_trait_ref<R, F>(&mut self, self_ty: Ty<'tcx>, mut closure: F) -> R
347 where
348 F: FnMut(&mut ConfirmContext<'a, 'tcx>, Ty<'tcx>, ty::PolyExistentialTraitRef<'tcx>) -> R,
349 {
350 let mut autoderef = self.fcx.autoderef(self.span, self_ty);
356
357 if self.tcx.features().arbitrary_self_types()
360 || self.tcx.features().arbitrary_self_types_pointers()
361 {
362 autoderef = autoderef.use_receiver_trait();
363 }
364
365 autoderef
366 .include_raw_pointers()
367 .find_map(|(ty, _)| match ty.kind() {
368 ty::Dynamic(data, ..) => Some(closure(
369 self,
370 ty,
371 data.principal().unwrap_or_else(|| {
372 span_bug!(self.span, "calling trait method on empty object?")
373 }),
374 )),
375 _ => None,
376 })
377 .unwrap_or_else(|| {
378 span_bug!(
379 self.span,
380 "self-type `{}` for ObjectPick never dereferenced to an object",
381 self_ty
382 )
383 })
384 }
385
386 fn instantiate_method_args(
387 &mut self,
388 pick: &probe::Pick<'tcx>,
389 seg: &hir::PathSegment<'tcx>,
390 parent_args: GenericArgsRef<'tcx>,
391 ) -> GenericArgsRef<'tcx> {
392 let generics = self.tcx.generics_of(pick.item.def_id);
396
397 let arg_count_correct = check_generic_arg_count_for_call(
398 self.fcx,
399 pick.item.def_id,
400 generics,
401 seg,
402 IsMethodCall::Yes,
403 );
404
405 assert_eq!(generics.parent_count, parent_args.len());
408
409 struct GenericArgsCtxt<'a, 'tcx> {
410 cfcx: &'a ConfirmContext<'a, 'tcx>,
411 pick: &'a probe::Pick<'tcx>,
412 seg: &'a hir::PathSegment<'tcx>,
413 }
414 impl<'a, 'tcx> GenericArgsLowerer<'a, 'tcx> for GenericArgsCtxt<'a, 'tcx> {
415 fn args_for_def_id(
416 &mut self,
417 def_id: DefId,
418 ) -> (Option<&'a hir::GenericArgs<'tcx>>, bool) {
419 if def_id == self.pick.item.def_id {
420 if let Some(data) = self.seg.args {
421 return (Some(data), false);
422 }
423 }
424 (None, false)
425 }
426
427 fn provided_kind(
428 &mut self,
429 preceding_args: &[ty::GenericArg<'tcx>],
430 param: &ty::GenericParamDef,
431 arg: &GenericArg<'tcx>,
432 ) -> ty::GenericArg<'tcx> {
433 match (¶m.kind, arg) {
434 (GenericParamDefKind::Lifetime, GenericArg::Lifetime(lt)) => self
435 .cfcx
436 .fcx
437 .lowerer()
438 .lower_lifetime(lt, RegionInferReason::Param(param))
439 .into(),
440 (GenericParamDefKind::Type { .. }, GenericArg::Type(ty)) => {
441 self.cfcx.lower_ty(ty.as_unambig_ty()).raw.into()
443 }
444 (GenericParamDefKind::Type { .. }, GenericArg::Infer(inf)) => {
445 self.cfcx.lower_ty(&inf.to_ty()).raw.into()
446 }
447 (GenericParamDefKind::Const { .. }, GenericArg::Const(ct)) => self
448 .cfcx
449 .lower_const_arg(
451 ct.as_unambig_ct(),
452 FeedConstTy::Param(param.def_id, preceding_args),
453 )
454 .into(),
455 (GenericParamDefKind::Const { .. }, GenericArg::Infer(inf)) => {
456 self.cfcx.ct_infer(Some(param), inf.span).into()
457 }
458 (kind, arg) => {
459 bug!("mismatched method arg kind {kind:?} in turbofish: {arg:?}")
460 }
461 }
462 }
463
464 fn inferred_kind(
465 &mut self,
466 _preceding_args: &[ty::GenericArg<'tcx>],
467 param: &ty::GenericParamDef,
468 _infer_args: bool,
469 ) -> ty::GenericArg<'tcx> {
470 self.cfcx.var_for_def(self.cfcx.span, param)
471 }
472 }
473
474 let args = lower_generic_args(
475 self.fcx,
476 pick.item.def_id,
477 parent_args,
478 false,
479 None,
480 &arg_count_correct,
481 &mut GenericArgsCtxt { cfcx: self, pick, seg },
482 );
483
484 if !args.is_empty() && !generics.is_own_empty() {
499 let user_type_annotation = self.probe(|_| {
500 let user_args = UserArgs {
501 args: GenericArgs::for_item(self.tcx, pick.item.def_id, |param, _| {
502 let i = param.index as usize;
503 if i < generics.parent_count {
504 self.fcx.var_for_def(DUMMY_SP, param)
505 } else {
506 args[i]
507 }
508 }),
509 user_self_ty: None, };
511
512 self.fcx.canonicalize_user_type_annotation(ty::UserType::new(
513 ty::UserTypeKind::TypeOf(pick.item.def_id, user_args),
514 ))
515 });
516
517 debug!("instantiate_method_args: user_type_annotation={:?}", user_type_annotation);
518
519 if !self.skip_record_for_diagnostics {
520 self.fcx.write_user_type_annotation(self.call_expr.hir_id, user_type_annotation);
521 }
522 }
523
524 self.normalize(self.span, args)
525 }
526
527 fn unify_receivers(
528 &mut self,
529 self_ty: Ty<'tcx>,
530 method_self_ty: Ty<'tcx>,
531 pick: &probe::Pick<'tcx>,
532 ) {
533 debug!(
534 "unify_receivers: self_ty={:?} method_self_ty={:?} span={:?} pick={:?}",
535 self_ty, method_self_ty, self.span, pick
536 );
537 let cause = self.cause(self.self_expr.span, ObligationCauseCode::Misc);
538 match self.at(&cause, self.param_env).sup(DefineOpaqueTypes::Yes, method_self_ty, self_ty) {
539 Ok(InferOk { obligations, value: () }) => {
540 self.register_predicates(obligations);
541 }
542 Err(terr) => {
543 if self.tcx.features().arbitrary_self_types() {
544 self.err_ctxt()
545 .report_mismatched_types(
546 &cause,
547 self.param_env,
548 method_self_ty,
549 self_ty,
550 terr,
551 )
552 .emit();
553 } else {
554 self.dcx().span_delayed_bug(
557 cause.span,
558 format!("{self_ty} was a subtype of {method_self_ty} but now is not?"),
559 );
560 }
561 }
562 }
563 }
564
565 fn instantiate_method_sig(
569 &mut self,
570 pick: &probe::Pick<'tcx>,
571 all_args: GenericArgsRef<'tcx>,
572 ) -> (ty::FnSig<'tcx>, ty::InstantiatedPredicates<'tcx>) {
573 debug!("instantiate_method_sig(pick={:?}, all_args={:?})", pick, all_args);
574
575 let def_id = pick.item.def_id;
579 let method_predicates = self.tcx.predicates_of(def_id).instantiate(self.tcx, all_args);
580
581 debug!("method_predicates after instantitation = {:?}", method_predicates);
582
583 let sig = self.tcx.fn_sig(def_id).instantiate(self.tcx, all_args);
584 debug!("type scheme instantiated, sig={:?}", sig);
585
586 let sig = self.instantiate_binder_with_fresh_vars(sig);
587 debug!("late-bound lifetimes from method instantiated, sig={:?}", sig);
588
589 (sig, method_predicates)
590 }
591
592 fn add_obligations(
593 &mut self,
594 fty: Ty<'tcx>,
595 all_args: GenericArgsRef<'tcx>,
596 method_predicates: ty::InstantiatedPredicates<'tcx>,
597 def_id: DefId,
598 ) {
599 debug!(
600 "add_obligations: fty={:?} all_args={:?} method_predicates={:?} def_id={:?}",
601 fty, all_args, method_predicates, def_id
602 );
603
604 for obligation in traits::predicates_for_generics(
608 |idx, span| {
609 let code = ObligationCauseCode::WhereClauseInExpr(
610 def_id,
611 span,
612 self.call_expr.hir_id,
613 idx,
614 );
615 self.cause(self.span, code)
616 },
617 self.param_env,
618 method_predicates,
619 ) {
620 self.register_predicate(obligation);
621 }
622
623 self.add_wf_bounds(all_args, self.call_expr.span);
626
627 self.register_wf_obligation(fty.into(), self.span, ObligationCauseCode::WellFormed(None));
631 }
632
633 fn predicates_require_illegal_sized_bound(
637 &self,
638 predicates: ty::InstantiatedPredicates<'tcx>,
639 ) -> Option<Span> {
640 let sized_def_id = self.tcx.lang_items().sized_trait()?;
641
642 traits::elaborate(self.tcx, predicates.predicates.iter().copied())
643 .filter_map(|pred| match pred.kind().skip_binder() {
645 ty::ClauseKind::Trait(trait_pred) if trait_pred.def_id() == sized_def_id => {
646 let span = predicates
647 .iter()
648 .find_map(|(p, span)| if p == pred { Some(span) } else { None })
649 .unwrap_or(DUMMY_SP);
650 Some((trait_pred, span))
651 }
652 _ => None,
653 })
654 .find_map(|(trait_pred, span)| match trait_pred.self_ty().kind() {
655 ty::Dynamic(..) => Some(span),
656 _ => None,
657 })
658 }
659
660 fn check_for_illegal_method_calls(&self, pick: &probe::Pick<'_>) {
661 if let Some(trait_def_id) = pick.item.trait_container(self.tcx) {
663 if let Err(e) = callee::check_legal_trait_for_method_call(
664 self.tcx,
665 self.span,
666 Some(self.self_expr.span),
667 self.call_expr.span,
668 trait_def_id,
669 self.body_id.to_def_id(),
670 ) {
671 self.set_tainted_by_errors(e);
672 }
673 }
674 }
675
676 fn lint_shadowed_supertrait_items(
677 &self,
678 pick: &probe::Pick<'_>,
679 segment: &hir::PathSegment<'tcx>,
680 ) {
681 if pick.shadowed_candidates.is_empty() {
682 return;
683 }
684
685 let shadower_span = self.tcx.def_span(pick.item.def_id);
686 let subtrait = self.tcx.item_name(pick.item.trait_container(self.tcx).unwrap());
687 let shadower = SupertraitItemShadower { span: shadower_span, subtrait };
688
689 let shadowee = if let [shadowee] = &pick.shadowed_candidates[..] {
690 let shadowee_span = self.tcx.def_span(shadowee.def_id);
691 let supertrait = self.tcx.item_name(shadowee.trait_container(self.tcx).unwrap());
692 SupertraitItemShadowee::Labeled { span: shadowee_span, supertrait }
693 } else {
694 let (traits, spans): (Vec<_>, Vec<_>) = pick
695 .shadowed_candidates
696 .iter()
697 .map(|item| {
698 (
699 self.tcx.item_name(item.trait_container(self.tcx).unwrap()),
700 self.tcx.def_span(item.def_id),
701 )
702 })
703 .unzip();
704 SupertraitItemShadowee::Several { traits: traits.into(), spans: spans.into() }
705 };
706
707 self.tcx.emit_node_span_lint(
708 SUPERTRAIT_ITEM_SHADOWING_USAGE,
709 segment.hir_id,
710 segment.ident.span,
711 SupertraitItemShadowing { shadower, shadowee, item: segment.ident.name, subtrait },
712 );
713 }
714
715 fn upcast(
716 &mut self,
717 source_trait_ref: ty::PolyTraitRef<'tcx>,
718 target_trait_def_id: DefId,
719 ) -> ty::PolyTraitRef<'tcx> {
720 let upcast_trait_refs =
721 traits::upcast_choices(self.tcx, source_trait_ref, target_trait_def_id);
722
723 if let &[upcast_trait_ref] = upcast_trait_refs.as_slice() {
725 upcast_trait_ref
726 } else {
727 self.dcx().span_delayed_bug(
728 self.span,
729 format!(
730 "cannot uniquely upcast `{:?}` to `{:?}`: `{:?}`",
731 source_trait_ref, target_trait_def_id, upcast_trait_refs
732 ),
733 );
734
735 ty::Binder::dummy(ty::TraitRef::new_from_args(
736 self.tcx,
737 target_trait_def_id,
738 ty::GenericArgs::extend_with_error(self.tcx, target_trait_def_id, &[]),
739 ))
740 }
741 }
742
743 fn instantiate_binder_with_fresh_vars<T>(&self, value: ty::Binder<'tcx, T>) -> T
744 where
745 T: TypeFoldable<TyCtxt<'tcx>> + Copy,
746 {
747 self.fcx.instantiate_binder_with_fresh_vars(self.span, infer::FnCall, value)
748 }
749}