Method visit_terminator

src/analysis/callgraph/mod.rs:201–222

Resolve direct call. Inspired by rustc_mir/src/transform/inline.rs#get_valid_function_call.

(&mut self, terminator: &Terminator<'tcx>, location: Location)

impl<'tcx> Visitor<'tcx> for CallSiteCollector<'_, 'tcx> {
    /// Resolve direct call.
    /// Inspired by rustc_mir/src/transform/inline.rs#get_valid_function_call.
    fn visit_terminator(&mut self, terminator: &Terminator<'tcx>, location: Location) {
        if let TerminatorKind::Call { ref func, .. } = terminator.kind {
            let func_ty = func.ty(self.body, self.tcx);
            // Only after monomorphizing can Instance::try_resolve work
            let func_ty = self.caller.instantiate_mir_and_normalize_erasing_regions(
                self.tcx,
                self.typing_env,
                EarlyBinder::bind(func_ty),
            );
            if let ty::FnDef(def_id, substs) = *func_ty.kind() {
                if let Some(callee) =
                    Instance::try_resolve(self.tcx, self.typing_env, def_id, substs)
                        .ok()
                        .flatten()
                {
                    self.callsites
                        .push((callee, CallSiteLocation::Direct(location)));
                }
            }
        }
        self.super_terminator(terminator, location);
    }