rustc_hir_typeck/
callee.rs

1use std::iter;
2
3use rustc_abi::{CanonAbi, ExternAbi};
4use rustc_ast::util::parser::ExprPrecedence;
5use rustc_errors::{Applicability, Diag, ErrorGuaranteed, StashKey};
6use rustc_hir::def::{self, CtorKind, Namespace, Res};
7use rustc_hir::def_id::DefId;
8use rustc_hir::{self as hir, HirId, LangItem};
9use rustc_hir_analysis::autoderef::Autoderef;
10use rustc_infer::infer::BoundRegionConversionTime;
11use rustc_infer::traits::{Obligation, ObligationCause, ObligationCauseCode};
12use rustc_middle::ty::adjustment::{
13    Adjust, Adjustment, AllowTwoPhase, AutoBorrow, AutoBorrowMutability,
14};
15use rustc_middle::ty::{self, GenericArgsRef, Ty, TyCtxt, TypeVisitableExt};
16use rustc_middle::{bug, span_bug};
17use rustc_span::def_id::LocalDefId;
18use rustc_span::{Span, sym};
19use rustc_target::spec::{AbiMap, AbiMapping};
20use rustc_trait_selection::error_reporting::traits::DefIdOrName;
21use rustc_trait_selection::infer::InferCtxtExt as _;
22use rustc_trait_selection::traits::query::evaluate_obligation::InferCtxtExt as _;
23use tracing::{debug, instrument};
24
25use super::method::MethodCallee;
26use super::method::probe::ProbeScope;
27use super::{Expectation, FnCtxt, TupleArgumentsFlag};
28use crate::{errors, fluent_generated};
29
30/// Checks that it is legal to call methods of the trait corresponding
31/// to `trait_id` (this only cares about the trait, not the specific
32/// method that is called).
33pub(crate) fn check_legal_trait_for_method_call(
34    tcx: TyCtxt<'_>,
35    span: Span,
36    receiver: Option<Span>,
37    expr_span: Span,
38    trait_id: DefId,
39    _body_id: DefId,
40) -> Result<(), ErrorGuaranteed> {
41    if tcx.is_lang_item(trait_id, LangItem::Drop) {
42        let sugg = if let Some(receiver) = receiver.filter(|s| !s.is_empty()) {
43            errors::ExplicitDestructorCallSugg::Snippet {
44                lo: expr_span.shrink_to_lo(),
45                hi: receiver.shrink_to_hi().to(expr_span.shrink_to_hi()),
46            }
47        } else {
48            errors::ExplicitDestructorCallSugg::Empty(span)
49        };
50        return Err(tcx.dcx().emit_err(errors::ExplicitDestructorCall { span, sugg }));
51    }
52    tcx.ensure_ok().coherent_trait(trait_id)
53}
54
55#[derive(Debug)]
56enum CallStep<'tcx> {
57    Builtin(Ty<'tcx>),
58    DeferredClosure(LocalDefId, ty::FnSig<'tcx>),
59    /// Call overloading when callee implements one of the Fn* traits.
60    Overloaded(MethodCallee<'tcx>),
61}
62
63impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
64    pub(crate) fn check_expr_call(
65        &self,
66        call_expr: &'tcx hir::Expr<'tcx>,
67        callee_expr: &'tcx hir::Expr<'tcx>,
68        arg_exprs: &'tcx [hir::Expr<'tcx>],
69        expected: Expectation<'tcx>,
70    ) -> Ty<'tcx> {
71        let original_callee_ty = match &callee_expr.kind {
72            hir::ExprKind::Path(hir::QPath::Resolved(..) | hir::QPath::TypeRelative(..)) => self
73                .check_expr_with_expectation_and_args(
74                    callee_expr,
75                    Expectation::NoExpectation,
76                    Some((call_expr, arg_exprs)),
77                ),
78            _ => self.check_expr(callee_expr),
79        };
80
81        let expr_ty = self.structurally_resolve_type(call_expr.span, original_callee_ty);
82
83        let mut autoderef = self.autoderef(callee_expr.span, expr_ty);
84        let mut result = None;
85        while result.is_none() && autoderef.next().is_some() {
86            result = self.try_overloaded_call_step(call_expr, callee_expr, arg_exprs, &autoderef);
87        }
88
89        match autoderef.final_ty(false).kind() {
90            ty::FnDef(def_id, _) => {
91                let abi = self.tcx.fn_sig(def_id).skip_binder().skip_binder().abi;
92                self.check_call_abi(abi, call_expr.span);
93            }
94            ty::FnPtr(_, header) => {
95                self.check_call_abi(header.abi, call_expr.span);
96            }
97            _ => { /* cannot have a non-rust abi */ }
98        }
99
100        self.register_predicates(autoderef.into_obligations());
101
102        let output = match result {
103            None => {
104                // Check all of the arg expressions, but with no expectations
105                // since we don't have a signature to compare them to.
106                for arg in arg_exprs {
107                    self.check_expr(arg);
108                }
109
110                if let hir::ExprKind::Path(hir::QPath::Resolved(_, path)) = &callee_expr.kind
111                    && let [segment] = path.segments
112                {
113                    self.dcx().try_steal_modify_and_emit_err(
114                        segment.ident.span,
115                        StashKey::CallIntoMethod,
116                        |err| {
117                            // Try suggesting `foo(a)` -> `a.foo()` if possible.
118                            self.suggest_call_as_method(
119                                err, segment, arg_exprs, call_expr, expected,
120                            );
121                        },
122                    );
123                }
124
125                let guar = self.report_invalid_callee(call_expr, callee_expr, expr_ty, arg_exprs);
126                Ty::new_error(self.tcx, guar)
127            }
128
129            Some(CallStep::Builtin(callee_ty)) => {
130                self.confirm_builtin_call(call_expr, callee_expr, callee_ty, arg_exprs, expected)
131            }
132
133            Some(CallStep::DeferredClosure(def_id, fn_sig)) => {
134                self.confirm_deferred_closure_call(call_expr, arg_exprs, expected, def_id, fn_sig)
135            }
136
137            Some(CallStep::Overloaded(method_callee)) => {
138                self.confirm_overloaded_call(call_expr, arg_exprs, expected, method_callee)
139            }
140        };
141
142        // we must check that return type of called functions is WF:
143        self.register_wf_obligation(
144            output.into(),
145            call_expr.span,
146            ObligationCauseCode::WellFormed(None),
147        );
148
149        output
150    }
151
152    /// Can a function with this ABI be called with a rust call expression?
153    ///
154    /// Some ABIs cannot be called from rust, either because rust does not know how to generate
155    /// code for the call, or because a call does not semantically make sense.
156    pub(crate) fn check_call_abi(&self, abi: ExternAbi, span: Span) {
157        let canon_abi = match AbiMap::from_target(&self.sess().target).canonize_abi(abi, false) {
158            AbiMapping::Direct(canon_abi) | AbiMapping::Deprecated(canon_abi) => canon_abi,
159            AbiMapping::Invalid => {
160                // This should be reported elsewhere, but we want to taint this body
161                // so that we don't try to evaluate calls to ABIs that are invalid.
162                let guar = self.dcx().span_delayed_bug(
163                    span,
164                    format!("invalid abi for platform should have reported an error: {abi}"),
165                );
166                self.set_tainted_by_errors(guar);
167                return;
168            }
169        };
170
171        let valid = match canon_abi {
172            // Rust doesn't know how to call functions with this ABI.
173            CanonAbi::Custom => false,
174
175            // These is an entry point for the host, and cannot be called on the GPU.
176            CanonAbi::GpuKernel => false,
177
178            // The interrupt ABIs should only be called by the CPU. They have complex
179            // pre- and postconditions, and can use non-standard instructions like `iret` on x86.
180            CanonAbi::Interrupt(_) => false,
181
182            CanonAbi::C
183            | CanonAbi::Rust
184            | CanonAbi::RustCold
185            | CanonAbi::Arm(_)
186            | CanonAbi::X86(_) => true,
187        };
188
189        if !valid {
190            let err = crate::errors::AbiCannotBeCalled { span, abi };
191            self.tcx.dcx().emit_err(err);
192        }
193    }
194
195    #[instrument(level = "debug", skip(self, call_expr, callee_expr, arg_exprs, autoderef), ret)]
196    fn try_overloaded_call_step(
197        &self,
198        call_expr: &'tcx hir::Expr<'tcx>,
199        callee_expr: &'tcx hir::Expr<'tcx>,
200        arg_exprs: &'tcx [hir::Expr<'tcx>],
201        autoderef: &Autoderef<'a, 'tcx>,
202    ) -> Option<CallStep<'tcx>> {
203        let adjusted_ty =
204            self.structurally_resolve_type(autoderef.span(), autoderef.final_ty(false));
205
206        // If the callee is a function pointer or a closure, then we're all set.
207        match *adjusted_ty.kind() {
208            ty::FnDef(..) | ty::FnPtr(..) => {
209                let adjustments = self.adjust_steps(autoderef);
210                self.apply_adjustments(callee_expr, adjustments);
211                return Some(CallStep::Builtin(adjusted_ty));
212            }
213
214            // Check whether this is a call to a closure where we
215            // haven't yet decided on whether the closure is fn vs
216            // fnmut vs fnonce. If so, we have to defer further processing.
217            ty::Closure(def_id, args) if self.closure_kind(adjusted_ty).is_none() => {
218                let def_id = def_id.expect_local();
219                let closure_sig = args.as_closure().sig();
220                let closure_sig = self.instantiate_binder_with_fresh_vars(
221                    call_expr.span,
222                    BoundRegionConversionTime::FnCall,
223                    closure_sig,
224                );
225                let adjustments = self.adjust_steps(autoderef);
226                self.record_deferred_call_resolution(
227                    def_id,
228                    DeferredCallResolution {
229                        call_expr,
230                        callee_expr,
231                        closure_ty: adjusted_ty,
232                        adjustments,
233                        fn_sig: closure_sig,
234                    },
235                );
236                return Some(CallStep::DeferredClosure(def_id, closure_sig));
237            }
238
239            // When calling a `CoroutineClosure` that is local to the body, we will
240            // not know what its `closure_kind` is yet. Instead, just fill in the
241            // signature with an infer var for the `tupled_upvars_ty` of the coroutine,
242            // and record a deferred call resolution which will constrain that var
243            // as part of `AsyncFn*` trait confirmation.
244            ty::CoroutineClosure(def_id, args) if self.closure_kind(adjusted_ty).is_none() => {
245                let def_id = def_id.expect_local();
246                let closure_args = args.as_coroutine_closure();
247                let coroutine_closure_sig = self.instantiate_binder_with_fresh_vars(
248                    call_expr.span,
249                    BoundRegionConversionTime::FnCall,
250                    closure_args.coroutine_closure_sig(),
251                );
252                let tupled_upvars_ty = self.next_ty_var(callee_expr.span);
253                // We may actually receive a coroutine back whose kind is different
254                // from the closure that this dispatched from. This is because when
255                // we have no captures, we automatically implement `FnOnce`. This
256                // impl forces the closure kind to `FnOnce` i.e. `u8`.
257                let kind_ty = self.next_ty_var(callee_expr.span);
258                let call_sig = self.tcx.mk_fn_sig(
259                    [coroutine_closure_sig.tupled_inputs_ty],
260                    coroutine_closure_sig.to_coroutine(
261                        self.tcx,
262                        closure_args.parent_args(),
263                        kind_ty,
264                        self.tcx.coroutine_for_closure(def_id),
265                        tupled_upvars_ty,
266                    ),
267                    coroutine_closure_sig.c_variadic,
268                    coroutine_closure_sig.safety,
269                    coroutine_closure_sig.abi,
270                );
271                let adjustments = self.adjust_steps(autoderef);
272                self.record_deferred_call_resolution(
273                    def_id,
274                    DeferredCallResolution {
275                        call_expr,
276                        callee_expr,
277                        closure_ty: adjusted_ty,
278                        adjustments,
279                        fn_sig: call_sig,
280                    },
281                );
282                return Some(CallStep::DeferredClosure(def_id, call_sig));
283            }
284
285            // Hack: we know that there are traits implementing Fn for &F
286            // where F:Fn and so forth. In the particular case of types
287            // like `f: &mut FnMut()`, if there is a call `f()`, we would
288            // normally translate to `FnMut::call_mut(&mut f, ())`, but
289            // that winds up potentially requiring the user to mark their
290            // variable as `mut` which feels unnecessary and unexpected.
291            //
292            //     fn foo(f: &mut impl FnMut()) { f() }
293            //            ^ without this hack `f` would have to be declared as mutable
294            //
295            // The simplest fix by far is to just ignore this case and deref again,
296            // so we wind up with `FnMut::call_mut(&mut *f, ())`.
297            ty::Ref(..) if autoderef.step_count() == 0 => {
298                return None;
299            }
300
301            ty::Error(_) => {
302                return None;
303            }
304
305            _ => {}
306        }
307
308        // Now, we look for the implementation of a Fn trait on the object's type.
309        // We first do it with the explicit instruction to look for an impl of
310        // `Fn<Tuple>`, with the tuple `Tuple` having an arity corresponding
311        // to the number of call parameters.
312        // If that fails (or_else branch), we try again without specifying the
313        // shape of the tuple (hence the None). This allows to detect an Fn trait
314        // is implemented, and use this information for diagnostic.
315        self.try_overloaded_call_traits(call_expr, adjusted_ty, Some(arg_exprs))
316            .or_else(|| self.try_overloaded_call_traits(call_expr, adjusted_ty, None))
317            .map(|(autoref, method)| {
318                let mut adjustments = self.adjust_steps(autoderef);
319                adjustments.extend(autoref);
320                self.apply_adjustments(callee_expr, adjustments);
321                CallStep::Overloaded(method)
322            })
323    }
324
325    fn try_overloaded_call_traits(
326        &self,
327        call_expr: &hir::Expr<'_>,
328        adjusted_ty: Ty<'tcx>,
329        opt_arg_exprs: Option<&'tcx [hir::Expr<'tcx>]>,
330    ) -> Option<(Option<Adjustment<'tcx>>, MethodCallee<'tcx>)> {
331        // HACK(async_closures): For async closures, prefer `AsyncFn*`
332        // over `Fn*`, since all async closures implement `FnOnce`, but
333        // choosing that over `AsyncFn`/`AsyncFnMut` would be more restrictive.
334        // For other callables, just prefer `Fn*` for perf reasons.
335        //
336        // The order of trait choices here is not that big of a deal,
337        // since it just guides inference (and our choice of autoref).
338        // Though in the future, I'd like typeck to choose:
339        // `Fn > AsyncFn > FnMut > AsyncFnMut > FnOnce > AsyncFnOnce`
340        // ...or *ideally*, we just have `LendingFn`/`LendingFnMut`, which
341        // would naturally unify these two trait hierarchies in the most
342        // general way.
343        let call_trait_choices = if self.shallow_resolve(adjusted_ty).is_coroutine_closure() {
344            [
345                (self.tcx.lang_items().async_fn_trait(), sym::async_call, true),
346                (self.tcx.lang_items().async_fn_mut_trait(), sym::async_call_mut, true),
347                (self.tcx.lang_items().async_fn_once_trait(), sym::async_call_once, false),
348                (self.tcx.lang_items().fn_trait(), sym::call, true),
349                (self.tcx.lang_items().fn_mut_trait(), sym::call_mut, true),
350                (self.tcx.lang_items().fn_once_trait(), sym::call_once, false),
351            ]
352        } else {
353            [
354                (self.tcx.lang_items().fn_trait(), sym::call, true),
355                (self.tcx.lang_items().fn_mut_trait(), sym::call_mut, true),
356                (self.tcx.lang_items().fn_once_trait(), sym::call_once, false),
357                (self.tcx.lang_items().async_fn_trait(), sym::async_call, true),
358                (self.tcx.lang_items().async_fn_mut_trait(), sym::async_call_mut, true),
359                (self.tcx.lang_items().async_fn_once_trait(), sym::async_call_once, false),
360            ]
361        };
362
363        // Try the options that are least restrictive on the caller first.
364        for (opt_trait_def_id, method_name, borrow) in call_trait_choices {
365            let Some(trait_def_id) = opt_trait_def_id else { continue };
366
367            let opt_input_type = opt_arg_exprs.map(|arg_exprs| {
368                Ty::new_tup_from_iter(self.tcx, arg_exprs.iter().map(|e| self.next_ty_var(e.span)))
369            });
370
371            if let Some(ok) = self.lookup_method_for_operator(
372                self.misc(call_expr.span),
373                method_name,
374                trait_def_id,
375                adjusted_ty,
376                opt_input_type,
377            ) {
378                let method = self.register_infer_ok_obligations(ok);
379                let mut autoref = None;
380                if borrow {
381                    // Check for &self vs &mut self in the method signature. Since this is either
382                    // the Fn or FnMut trait, it should be one of those.
383                    let ty::Ref(_, _, mutbl) = method.sig.inputs()[0].kind() else {
384                        bug!("Expected `FnMut`/`Fn` to take receiver by-ref/by-mut")
385                    };
386
387                    // For initial two-phase borrow
388                    // deployment, conservatively omit
389                    // overloaded function call ops.
390                    let mutbl = AutoBorrowMutability::new(*mutbl, AllowTwoPhase::No);
391
392                    autoref = Some(Adjustment {
393                        kind: Adjust::Borrow(AutoBorrow::Ref(mutbl)),
394                        target: method.sig.inputs()[0],
395                    });
396                }
397
398                return Some((autoref, method));
399            }
400        }
401
402        None
403    }
404
405    /// Give appropriate suggestion when encountering `||{/* not callable */}()`, where the
406    /// likely intention is to call the closure, suggest `(||{})()`. (#55851)
407    fn identify_bad_closure_def_and_call(
408        &self,
409        err: &mut Diag<'_>,
410        hir_id: hir::HirId,
411        callee_node: &hir::ExprKind<'_>,
412        callee_span: Span,
413    ) {
414        let hir::ExprKind::Block(..) = callee_node else {
415            // Only calls on blocks suggested here.
416            return;
417        };
418
419        let fn_decl_span = if let hir::Node::Expr(&hir::Expr {
420            kind: hir::ExprKind::Closure(&hir::Closure { fn_decl_span, .. }),
421            ..
422        }) = self.tcx.parent_hir_node(hir_id)
423        {
424            fn_decl_span
425        } else if let Some((
426            _,
427            hir::Node::Expr(&hir::Expr {
428                hir_id: parent_hir_id,
429                kind:
430                    hir::ExprKind::Closure(&hir::Closure {
431                        kind:
432                            hir::ClosureKind::Coroutine(hir::CoroutineKind::Desugared(
433                                hir::CoroutineDesugaring::Async,
434                                hir::CoroutineSource::Closure,
435                            )),
436                        ..
437                    }),
438                ..
439            }),
440        )) = self.tcx.hir_parent_iter(hir_id).nth(3)
441        {
442            // Actually need to unwrap one more layer of HIR to get to
443            // the _real_ closure...
444            if let hir::Node::Expr(&hir::Expr {
445                kind: hir::ExprKind::Closure(&hir::Closure { fn_decl_span, .. }),
446                ..
447            }) = self.tcx.parent_hir_node(parent_hir_id)
448            {
449                fn_decl_span
450            } else {
451                return;
452            }
453        } else {
454            return;
455        };
456
457        let start = fn_decl_span.shrink_to_lo();
458        let end = callee_span.shrink_to_hi();
459        err.multipart_suggestion(
460            "if you meant to create this closure and immediately call it, surround the \
461                closure with parentheses",
462            vec![(start, "(".to_string()), (end, ")".to_string())],
463            Applicability::MaybeIncorrect,
464        );
465    }
466
467    /// Give appropriate suggestion when encountering `[("a", 0) ("b", 1)]`, where the
468    /// likely intention is to create an array containing tuples.
469    fn maybe_suggest_bad_array_definition(
470        &self,
471        err: &mut Diag<'_>,
472        call_expr: &'tcx hir::Expr<'tcx>,
473        callee_expr: &'tcx hir::Expr<'tcx>,
474    ) -> bool {
475        let parent_node = self.tcx.parent_hir_node(call_expr.hir_id);
476        if let (
477            hir::Node::Expr(hir::Expr { kind: hir::ExprKind::Array(_), .. }),
478            hir::ExprKind::Tup(exp),
479            hir::ExprKind::Call(_, args),
480        ) = (parent_node, &callee_expr.kind, &call_expr.kind)
481            && args.len() == exp.len()
482        {
483            let start = callee_expr.span.shrink_to_hi();
484            err.span_suggestion(
485                start,
486                "consider separating array elements with a comma",
487                ",",
488                Applicability::MaybeIncorrect,
489            );
490            return true;
491        }
492        false
493    }
494
495    fn confirm_builtin_call(
496        &self,
497        call_expr: &'tcx hir::Expr<'tcx>,
498        callee_expr: &'tcx hir::Expr<'tcx>,
499        callee_ty: Ty<'tcx>,
500        arg_exprs: &'tcx [hir::Expr<'tcx>],
501        expected: Expectation<'tcx>,
502    ) -> Ty<'tcx> {
503        let (fn_sig, def_id) = match *callee_ty.kind() {
504            ty::FnDef(def_id, args) => {
505                self.enforce_context_effects(Some(call_expr.hir_id), call_expr.span, def_id, args);
506                let fn_sig = self.tcx.fn_sig(def_id).instantiate(self.tcx, args);
507
508                // Unit testing: function items annotated with
509                // `#[rustc_evaluate_where_clauses]` trigger special output
510                // to let us test the trait evaluation system.
511                // Untranslatable diagnostics are okay for rustc internals
512                #[allow(rustc::untranslatable_diagnostic)]
513                #[allow(rustc::diagnostic_outside_of_impl)]
514                if self.tcx.has_attr(def_id, sym::rustc_evaluate_where_clauses) {
515                    let predicates = self.tcx.predicates_of(def_id);
516                    let predicates = predicates.instantiate(self.tcx, args);
517                    for (predicate, predicate_span) in predicates {
518                        let obligation = Obligation::new(
519                            self.tcx,
520                            ObligationCause::dummy_with_span(callee_expr.span),
521                            self.param_env,
522                            predicate,
523                        );
524                        let result = self.evaluate_obligation(&obligation);
525                        self.dcx()
526                            .struct_span_err(
527                                callee_expr.span,
528                                format!("evaluate({predicate:?}) = {result:?}"),
529                            )
530                            .with_span_label(predicate_span, "predicate")
531                            .emit();
532                    }
533                }
534                (fn_sig, Some(def_id))
535            }
536
537            // FIXME(const_trait_impl): these arms should error because we can't enforce them
538            ty::FnPtr(sig_tys, hdr) => (sig_tys.with(hdr), None),
539
540            _ => unreachable!(),
541        };
542
543        // Replace any late-bound regions that appear in the function
544        // signature with region variables. We also have to
545        // renormalize the associated types at this point, since they
546        // previously appeared within a `Binder<>` and hence would not
547        // have been normalized before.
548        let fn_sig = self.instantiate_binder_with_fresh_vars(
549            call_expr.span,
550            BoundRegionConversionTime::FnCall,
551            fn_sig,
552        );
553        let fn_sig = self.normalize(call_expr.span, fn_sig);
554
555        self.check_argument_types(
556            call_expr.span,
557            call_expr,
558            fn_sig.inputs(),
559            fn_sig.output(),
560            expected,
561            arg_exprs,
562            fn_sig.c_variadic,
563            TupleArgumentsFlag::DontTupleArguments,
564            def_id,
565        );
566
567        if fn_sig.abi == rustc_abi::ExternAbi::RustCall {
568            let sp = arg_exprs.last().map_or(call_expr.span, |expr| expr.span);
569            if let Some(ty) = fn_sig.inputs().last().copied() {
570                self.register_bound(
571                    ty,
572                    self.tcx.require_lang_item(hir::LangItem::Tuple, sp),
573                    self.cause(sp, ObligationCauseCode::RustCall),
574                );
575                self.require_type_is_sized(ty, sp, ObligationCauseCode::RustCall);
576            } else {
577                self.dcx().emit_err(errors::RustCallIncorrectArgs { span: sp });
578            }
579        }
580
581        fn_sig.output()
582    }
583
584    /// Attempts to reinterpret `method(rcvr, args...)` as `rcvr.method(args...)`
585    /// and suggesting the fix if the method probe is successful.
586    fn suggest_call_as_method(
587        &self,
588        diag: &mut Diag<'_>,
589        segment: &'tcx hir::PathSegment<'tcx>,
590        arg_exprs: &'tcx [hir::Expr<'tcx>],
591        call_expr: &'tcx hir::Expr<'tcx>,
592        expected: Expectation<'tcx>,
593    ) {
594        if let [callee_expr, rest @ ..] = arg_exprs {
595            let Some(callee_ty) = self.typeck_results.borrow().expr_ty_adjusted_opt(callee_expr)
596            else {
597                return;
598            };
599
600            // First, do a probe with `IsSuggestion(true)` to avoid emitting
601            // any strange errors. If it's successful, then we'll do a true
602            // method lookup.
603            let Ok(pick) = self.lookup_probe_for_diagnostic(
604                segment.ident,
605                callee_ty,
606                call_expr,
607                // We didn't record the in scope traits during late resolution
608                // so we need to probe AllTraits unfortunately
609                ProbeScope::AllTraits,
610                expected.only_has_type(self),
611            ) else {
612                return;
613            };
614
615            let pick = self.confirm_method_for_diagnostic(
616                call_expr.span,
617                callee_expr,
618                call_expr,
619                callee_ty,
620                &pick,
621                segment,
622            );
623            if pick.illegal_sized_bound.is_some() {
624                return;
625            }
626
627            let Some(callee_expr_span) = callee_expr.span.find_ancestor_inside(call_expr.span)
628            else {
629                return;
630            };
631            let up_to_rcvr_span = segment.ident.span.until(callee_expr_span);
632            let rest_span = callee_expr_span.shrink_to_hi().to(call_expr.span.shrink_to_hi());
633            let rest_snippet = if let Some(first) = rest.first() {
634                self.tcx
635                    .sess
636                    .source_map()
637                    .span_to_snippet(first.span.to(call_expr.span.shrink_to_hi()))
638            } else {
639                Ok(")".to_string())
640            };
641
642            if let Ok(rest_snippet) = rest_snippet {
643                let sugg = if self.precedence(callee_expr) >= ExprPrecedence::Unambiguous {
644                    vec![
645                        (up_to_rcvr_span, "".to_string()),
646                        (rest_span, format!(".{}({rest_snippet}", segment.ident)),
647                    ]
648                } else {
649                    vec![
650                        (up_to_rcvr_span, "(".to_string()),
651                        (rest_span, format!(").{}({rest_snippet}", segment.ident)),
652                    ]
653                };
654                let self_ty = self.resolve_vars_if_possible(pick.callee.sig.inputs()[0]);
655                diag.multipart_suggestion(
656                    format!(
657                        "use the `.` operator to call the method `{}{}` on `{self_ty}`",
658                        self.tcx
659                            .associated_item(pick.callee.def_id)
660                            .trait_container(self.tcx)
661                            .map_or_else(
662                                || String::new(),
663                                |trait_def_id| self.tcx.def_path_str(trait_def_id) + "::"
664                            ),
665                        segment.ident
666                    ),
667                    sugg,
668                    Applicability::MaybeIncorrect,
669                );
670            }
671        }
672    }
673
674    fn report_invalid_callee(
675        &self,
676        call_expr: &'tcx hir::Expr<'tcx>,
677        callee_expr: &'tcx hir::Expr<'tcx>,
678        callee_ty: Ty<'tcx>,
679        arg_exprs: &'tcx [hir::Expr<'tcx>],
680    ) -> ErrorGuaranteed {
681        // Callee probe fails when APIT references errors, so suppress those
682        // errors here.
683        if let Some((_, _, args)) = self.extract_callable_info(callee_ty)
684            && let Err(err) = args.error_reported()
685        {
686            return err;
687        }
688
689        let mut unit_variant = None;
690        if let hir::ExprKind::Path(qpath) = &callee_expr.kind
691            && let Res::Def(def::DefKind::Ctor(kind, CtorKind::Const), _)
692                = self.typeck_results.borrow().qpath_res(qpath, callee_expr.hir_id)
693            // Only suggest removing parens if there are no arguments
694            && arg_exprs.is_empty()
695            && call_expr.span.contains(callee_expr.span)
696        {
697            let descr = match kind {
698                def::CtorOf::Struct => "struct",
699                def::CtorOf::Variant => "enum variant",
700            };
701            let removal_span = callee_expr.span.shrink_to_hi().to(call_expr.span.shrink_to_hi());
702            unit_variant =
703                Some((removal_span, descr, rustc_hir_pretty::qpath_to_string(&self.tcx, qpath)));
704        }
705
706        let callee_ty = self.resolve_vars_if_possible(callee_ty);
707        let mut path = None;
708        let mut err = self.dcx().create_err(errors::InvalidCallee {
709            span: callee_expr.span,
710            ty: callee_ty,
711            found: match &unit_variant {
712                Some((_, kind, path)) => format!("{kind} `{path}`"),
713                None => format!("`{}`", self.tcx.short_string(callee_ty, &mut path)),
714            },
715        });
716        *err.long_ty_path() = path;
717        if callee_ty.references_error() {
718            err.downgrade_to_delayed_bug();
719        }
720
721        self.identify_bad_closure_def_and_call(
722            &mut err,
723            call_expr.hir_id,
724            &callee_expr.kind,
725            callee_expr.span,
726        );
727
728        if let Some((removal_span, kind, path)) = &unit_variant {
729            err.span_suggestion_verbose(
730                *removal_span,
731                format!(
732                    "`{path}` is a unit {kind}, and does not take parentheses to be constructed",
733                ),
734                "",
735                Applicability::MachineApplicable,
736            );
737        }
738
739        if let hir::ExprKind::Path(hir::QPath::Resolved(None, path)) = callee_expr.kind
740            && let Res::Local(_) = path.res
741            && let [segment] = &path.segments
742        {
743            for id in self.tcx.hir_free_items() {
744                if let Some(node) = self.tcx.hir_get_if_local(id.owner_id.into())
745                    && let hir::Node::Item(item) = node
746                    && let hir::ItemKind::Fn { ident, .. } = item.kind
747                    && ident.name == segment.ident.name
748                {
749                    err.span_label(
750                        self.tcx.def_span(id.owner_id),
751                        "this function of the same name is available here, but it's shadowed by \
752                         the local binding",
753                    );
754                }
755            }
756        }
757
758        let mut inner_callee_path = None;
759        let def = match callee_expr.kind {
760            hir::ExprKind::Path(ref qpath) => {
761                self.typeck_results.borrow().qpath_res(qpath, callee_expr.hir_id)
762            }
763            hir::ExprKind::Call(inner_callee, _) => {
764                if let hir::ExprKind::Path(ref inner_qpath) = inner_callee.kind {
765                    inner_callee_path = Some(inner_qpath);
766                    self.typeck_results.borrow().qpath_res(inner_qpath, inner_callee.hir_id)
767                } else {
768                    Res::Err
769                }
770            }
771            _ => Res::Err,
772        };
773
774        if !self.maybe_suggest_bad_array_definition(&mut err, call_expr, callee_expr) {
775            // If the call spans more than one line and the callee kind is
776            // itself another `ExprCall`, that's a clue that we might just be
777            // missing a semicolon (#51055, #106515).
778            let call_is_multiline = self
779                .tcx
780                .sess
781                .source_map()
782                .is_multiline(call_expr.span.with_lo(callee_expr.span.hi()))
783                && call_expr.span.eq_ctxt(callee_expr.span);
784            if call_is_multiline {
785                err.span_suggestion(
786                    callee_expr.span.shrink_to_hi(),
787                    "consider using a semicolon here to finish the statement",
788                    ";",
789                    Applicability::MaybeIncorrect,
790                );
791            }
792            if let Some((maybe_def, output_ty, _)) = self.extract_callable_info(callee_ty)
793                && !self.type_is_sized_modulo_regions(self.param_env, output_ty)
794            {
795                let descr = match maybe_def {
796                    DefIdOrName::DefId(def_id) => self.tcx.def_descr(def_id),
797                    DefIdOrName::Name(name) => name,
798                };
799                err.span_label(
800                    callee_expr.span,
801                    format!("this {descr} returns an unsized value `{output_ty}`, so it cannot be called")
802                );
803                if let DefIdOrName::DefId(def_id) = maybe_def
804                    && let Some(def_span) = self.tcx.hir_span_if_local(def_id)
805                {
806                    err.span_label(def_span, "the callable type is defined here");
807                }
808            } else {
809                err.span_label(call_expr.span, "call expression requires function");
810            }
811        }
812
813        if let Some(span) = self.tcx.hir_res_span(def) {
814            let callee_ty = callee_ty.to_string();
815            let label = match (unit_variant, inner_callee_path) {
816                (Some((_, kind, path)), _) => {
817                    err.arg("kind", kind);
818                    err.arg("path", path);
819                    Some(fluent_generated::hir_typeck_invalid_defined_kind)
820                }
821                (_, Some(hir::QPath::Resolved(_, path))) => {
822                    self.tcx.sess.source_map().span_to_snippet(path.span).ok().map(|p| {
823                        err.arg("func", p);
824                        fluent_generated::hir_typeck_invalid_fn_defined
825                    })
826                }
827                _ => {
828                    match def {
829                        // Emit a different diagnostic for local variables, as they are not
830                        // type definitions themselves, but rather variables *of* that type.
831                        Res::Local(hir_id) => {
832                            err.arg("local_name", self.tcx.hir_name(hir_id));
833                            Some(fluent_generated::hir_typeck_invalid_local)
834                        }
835                        Res::Def(kind, def_id) if kind.ns() == Some(Namespace::ValueNS) => {
836                            err.arg("path", self.tcx.def_path_str(def_id));
837                            Some(fluent_generated::hir_typeck_invalid_defined)
838                        }
839                        _ => {
840                            err.arg("path", callee_ty);
841                            Some(fluent_generated::hir_typeck_invalid_defined)
842                        }
843                    }
844                }
845            };
846            if let Some(label) = label {
847                err.span_label(span, label);
848            }
849        }
850        err.emit()
851    }
852
853    fn confirm_deferred_closure_call(
854        &self,
855        call_expr: &'tcx hir::Expr<'tcx>,
856        arg_exprs: &'tcx [hir::Expr<'tcx>],
857        expected: Expectation<'tcx>,
858        closure_def_id: LocalDefId,
859        fn_sig: ty::FnSig<'tcx>,
860    ) -> Ty<'tcx> {
861        // `fn_sig` is the *signature* of the closure being called. We
862        // don't know the full details yet (`Fn` vs `FnMut` etc), but we
863        // do know the types expected for each argument and the return
864        // type.
865        self.check_argument_types(
866            call_expr.span,
867            call_expr,
868            fn_sig.inputs(),
869            fn_sig.output(),
870            expected,
871            arg_exprs,
872            fn_sig.c_variadic,
873            TupleArgumentsFlag::TupleArguments,
874            Some(closure_def_id.to_def_id()),
875        );
876
877        fn_sig.output()
878    }
879
880    #[tracing::instrument(level = "debug", skip(self, span))]
881    pub(super) fn enforce_context_effects(
882        &self,
883        call_hir_id: Option<HirId>,
884        span: Span,
885        callee_did: DefId,
886        callee_args: GenericArgsRef<'tcx>,
887    ) {
888        // FIXME(const_trait_impl): We should be enforcing these effects unconditionally.
889        // This can be done as soon as we convert the standard library back to
890        // using const traits, since if we were to enforce these conditions now,
891        // we'd fail on basically every builtin trait call (i.e. `1 + 2`).
892        if !self.tcx.features().const_trait_impl() {
893            return;
894        }
895
896        // If we have `rustc_do_not_const_check`, do not check `[const]` bounds.
897        if self.tcx.has_attr(self.body_id, sym::rustc_do_not_const_check) {
898            return;
899        }
900
901        let host = match self.tcx.hir_body_const_context(self.body_id) {
902            Some(hir::ConstContext::Const { .. } | hir::ConstContext::Static(_)) => {
903                ty::BoundConstness::Const
904            }
905            Some(hir::ConstContext::ConstFn) => ty::BoundConstness::Maybe,
906            None => return,
907        };
908
909        // FIXME(const_trait_impl): Should this be `is_const_fn_raw`? It depends on if we move
910        // const stability checking here too, I guess.
911        if self.tcx.is_conditionally_const(callee_did) {
912            let q = self.tcx.const_conditions(callee_did);
913            // FIXME(const_trait_impl): Use this span with a better cause code.
914            for (idx, (cond, pred_span)) in
915                q.instantiate(self.tcx, callee_args).into_iter().enumerate()
916            {
917                let cause = self.cause(
918                    span,
919                    if let Some(hir_id) = call_hir_id {
920                        ObligationCauseCode::HostEffectInExpr(callee_did, pred_span, hir_id, idx)
921                    } else {
922                        ObligationCauseCode::WhereClause(callee_did, pred_span)
923                    },
924                );
925                self.register_predicate(Obligation::new(
926                    self.tcx,
927                    cause,
928                    self.param_env,
929                    cond.to_host_effect_clause(self.tcx, host),
930                ));
931            }
932        } else {
933            // FIXME(const_trait_impl): This should eventually be caught here.
934            // For now, though, we defer some const checking to MIR.
935        }
936    }
937
938    fn confirm_overloaded_call(
939        &self,
940        call_expr: &'tcx hir::Expr<'tcx>,
941        arg_exprs: &'tcx [hir::Expr<'tcx>],
942        expected: Expectation<'tcx>,
943        method: MethodCallee<'tcx>,
944    ) -> Ty<'tcx> {
945        self.check_argument_types(
946            call_expr.span,
947            call_expr,
948            &method.sig.inputs()[1..],
949            method.sig.output(),
950            expected,
951            arg_exprs,
952            method.sig.c_variadic,
953            TupleArgumentsFlag::TupleArguments,
954            Some(method.def_id),
955        );
956
957        self.write_method_call_and_enforce_effects(call_expr.hir_id, call_expr.span, method);
958
959        method.sig.output()
960    }
961}
962
963#[derive(Debug)]
964pub(crate) struct DeferredCallResolution<'tcx> {
965    call_expr: &'tcx hir::Expr<'tcx>,
966    callee_expr: &'tcx hir::Expr<'tcx>,
967    closure_ty: Ty<'tcx>,
968    adjustments: Vec<Adjustment<'tcx>>,
969    fn_sig: ty::FnSig<'tcx>,
970}
971
972impl<'a, 'tcx> DeferredCallResolution<'tcx> {
973    pub(crate) fn resolve(self, fcx: &FnCtxt<'a, 'tcx>) {
974        debug!("DeferredCallResolution::resolve() {:?}", self);
975
976        // we should not be invoked until the closure kind has been
977        // determined by upvar inference
978        assert!(fcx.closure_kind(self.closure_ty).is_some());
979
980        // We may now know enough to figure out fn vs fnmut etc.
981        match fcx.try_overloaded_call_traits(self.call_expr, self.closure_ty, None) {
982            Some((autoref, method_callee)) => {
983                // One problem is that when we get here, we are going
984                // to have a newly instantiated function signature
985                // from the call trait. This has to be reconciled with
986                // the older function signature we had before. In
987                // principle we *should* be able to fn_sigs(), but we
988                // can't because of the annoying need for a TypeTrace.
989                // (This always bites me, should find a way to
990                // refactor it.)
991                let method_sig = method_callee.sig;
992
993                debug!("attempt_resolution: method_callee={:?}", method_callee);
994
995                for (method_arg_ty, self_arg_ty) in
996                    iter::zip(method_sig.inputs().iter().skip(1), self.fn_sig.inputs())
997                {
998                    fcx.demand_eqtype(self.call_expr.span, *self_arg_ty, *method_arg_ty);
999                }
1000
1001                fcx.demand_eqtype(self.call_expr.span, method_sig.output(), self.fn_sig.output());
1002
1003                let mut adjustments = self.adjustments;
1004                adjustments.extend(autoref);
1005                fcx.apply_adjustments(self.callee_expr, adjustments);
1006
1007                fcx.write_method_call_and_enforce_effects(
1008                    self.call_expr.hir_id,
1009                    self.call_expr.span,
1010                    method_callee,
1011                );
1012            }
1013            None => {
1014                span_bug!(
1015                    self.call_expr.span,
1016                    "Expected to find a suitable `Fn`/`FnMut`/`FnOnce` implementation for `{}`",
1017                    self.closure_ty
1018                )
1019            }
1020        }
1021    }
1022}