compiler/rustc_monomorphize/src/collector.rs RUST 1,964 lines View on github.com → Search inside
1//! Mono Item Collection2//! ====================3//!4//! This module is responsible for discovering all items that will contribute5//! to code generation of the crate. The important part here is that it not only6//! needs to find syntax-level items (functions, structs, etc) but also all7//! their monomorphized instantiations. Every non-generic, non-const function8//! maps to one LLVM artifact. Every generic function can produce9//! from zero to N artifacts, depending on the sets of type arguments it10//! is instantiated with.11//! This also applies to generic items from other crates: A generic definition12//! in crate X might produce monomorphizations that are compiled into crate Y.13//! We also have to collect these here.14//!15//! The following kinds of "mono items" are handled here:16//!17//! - Functions18//! - Methods19//! - Closures20//! - Statics21//! - Drop glue22//!23//! The following things also result in LLVM artifacts, but are not collected24//! here, since we instantiate them locally on demand when needed in a given25//! codegen unit:26//!27//! - Constants28//! - VTables29//! - Object Shims30//!31//! The main entry point is `collect_crate_mono_items`, at the bottom of this file.32//!33//! General Algorithm34//! -----------------35//! Let's define some terms first:36//!37//! - A "mono item" is something that results in a function or global in38//!   the LLVM IR of a codegen unit. Mono items do not stand on their39//!   own, they can use other mono items. For example, if function40//!   `foo()` calls function `bar()` then the mono item for `foo()`41//!   uses the mono item for function `bar()`. In general, the42//!   definition for mono item A using a mono item B is that43//!   the LLVM artifact produced for A uses the LLVM artifact produced44//!   for B.45//!46//! - Mono items and the uses between them form a directed graph,47//!   where the mono items are the nodes and uses form the edges.48//!   Let's call this graph the "mono item graph".49//!50//! - The mono item graph for a program contains all mono items51//!   that are needed in order to produce the complete LLVM IR of the program.52//!53//! The purpose of the algorithm implemented in this module is to build the54//! mono item graph for the current crate. It runs in two phases:55//!56//! 1. Discover the roots of the graph by traversing the HIR of the crate.57//! 2. Starting from the roots, find uses by inspecting the MIR58//!    representation of the item corresponding to a given node, until no more59//!    new nodes are found.60//!61//! ### Discovering roots62//! The roots of the mono item graph correspond to the public non-generic63//! syntactic items in the source code. We find them by walking the HIR of the64//! crate, and whenever we hit upon a public function, method, or static item,65//! we create a mono item consisting of the items DefId and, since we only66//! consider non-generic items, an empty type-parameters set. (In eager67//! collection mode, during incremental compilation, all non-generic functions68//! are considered as roots, as well as when the `-Clink-dead-code` option is69//! specified. Functions marked `#[no_mangle]` and functions called by inlinable70//! functions also always act as roots.)71//!72//! ### Finding uses73//! Given a mono item node, we can discover uses by inspecting its MIR. We walk74//! the MIR to find other mono items used by each mono item. Since the mono75//! item we are currently at is always monomorphic, we also know the concrete76//! type arguments of its used mono items. The specific forms a use can take in77//! MIR are quite diverse. Here is an overview:78//!79//! #### Calling Functions/Methods80//! The most obvious way for one mono item to use another is a81//! function or method call (represented by a CALL terminator in MIR). But82//! calls are not the only thing that might introduce a use between two83//! function mono items, and as we will see below, they are just a84//! specialization of the form described next, and consequently will not get any85//! special treatment in the algorithm.86//!87//! #### Taking a reference to a function or method88//! A function does not need to actually be called in order to be used by89//! another function. It suffices to just take a reference in order to introduce90//! an edge. Consider the following example:91//!92//! ```93//! # use core::fmt::Display;94//! fn print_val<T: Display>(x: T) {95//!     println!("{}", x);96//! }97//!98//! fn call_fn(f: &dyn Fn(i32), x: i32) {99//!     f(x);100//! }101//!102//! fn main() {103//!     let print_i32 = print_val::<i32>;104//!     call_fn(&print_i32, 0);105//! }106//! ```107//! The MIR of none of these functions will contain an explicit call to108//! `print_val::<i32>`. Nonetheless, in order to mono this program, we need109//! an instance of this function. Thus, whenever we encounter a function or110//! method in operand position, we treat it as a use of the current111//! mono item. Calls are just a special case of that.112//!113//! #### Drop glue114//! Drop glue mono items are introduced by MIR drop-statements. The115//! generated mono item will have additional drop-glue item uses if the116//! type to be dropped contains nested values that also need to be dropped. It117//! might also have a function item use for the explicit `Drop::drop`118//! implementation of its type.119//!120//! #### Unsizing Casts121//! A subtle way of introducing use edges is by casting to a trait object.122//! Since the resulting wide-pointer contains a reference to a vtable, we need to123//! instantiate all dyn-compatible methods of the trait, as we need to store124//! pointers to these functions even if they never get called anywhere. This can125//! be seen as a special case of taking a function reference.126//!127//!128//! Interaction with Cross-Crate Inlining129//! -------------------------------------130//! The binary of a crate will not only contain machine code for the items131//! defined in the source code of that crate. It will also contain monomorphic132//! instantiations of any extern generic functions and of functions marked with133//! `#[inline]`.134//! The collection algorithm handles this more or less mono. If it is135//! about to create a mono item for something with an external `DefId`,136//! it will take a look if the MIR for that item is available, and if so just137//! proceed normally. If the MIR is not available, it assumes that the item is138//! just linked to and no node is created; which is exactly what we want, since139//! no machine code should be generated in the current crate for such an item.140//!141//! Eager and Lazy Collection Strategy142//! ----------------------------------143//! Mono item collection can be performed with one of two strategies:144//!145//! - Lazy strategy means that items will only be instantiated when actually146//!   used. The goal is to produce the least amount of machine code147//!   possible.148//!149//! - Eager strategy is meant to be used in conjunction with incremental compilation150//!   where a stable set of mono items is more important than a minimal151//!   one. Thus, eager strategy will instantiate drop-glue for every drop-able type152//!   in the crate, even if no drop call for that type exists (yet). It will153//!   also instantiate default implementations of trait methods, something that154//!   otherwise is only done on demand.155//!156//! Collection-time const evaluation and "mentioned" items157//! ------------------------------------------------------158//!159//! One important role of collection is to evaluate all constants that are used by all the items160//! which are being collected. Codegen can then rely on only encountering constants that evaluate161//! successfully, and if a constant fails to evaluate, the collector has much better context to be162//! able to show where this constant comes up.163//!164//! However, the exact set of "used" items (collected as described above), and therefore the exact165//! set of used constants, can depend on optimizations. Optimizing away dead code may optimize away166//! a function call that uses a failing constant, so an unoptimized build may fail where an167//! optimized build succeeds. This is undesirable.168//!169//! To avoid this, the collector has the concept of "mentioned" items. Some time during the MIR170//! pipeline, before any optimization-level-dependent optimizations, we compute a list of all items171//! that syntactically appear in the code. These are considered "mentioned", and even if they are in172//! dead code and get optimized away (which makes them no longer "used"), they are still173//! "mentioned". For every used item, the collector ensures that all mentioned items, recursively,174//! do not use a failing constant. This is reflected via the [`CollectionMode`], which determines175//! whether we are visiting a used item or merely a mentioned item.176//!177//! The collector and "mentioned items" gathering (which lives in `rustc_mir_transform::mentioned_items`)178//! need to stay in sync in the following sense:179//!180//! - For every item that the collector gather that could eventually lead to build failure (most181//!   likely due to containing a constant that fails to evaluate), a corresponding mentioned item182//!   must be added. This should use the exact same strategy as the ecollector to make sure they are183//!   in sync. However, while the collector works on monomorphized types, mentioned items are184//!   collected on generic MIR -- so any time the collector checks for a particular type (such as185//!   `ty::FnDef`), we have to just onconditionally add this as a mentioned item.186//! - In `visit_mentioned_item`, we then do with that mentioned item exactly what the collector187//!   would have done during regular MIR visiting. Basically you can think of the collector having188//!   two stages, a pre-monomorphization stage and a post-monomorphization stage (usually quite189//!   literally separated by a call to `self.monomorphize`); the pre-monomorphizationn stage is190//!   duplicated in mentioned items gathering and the post-monomorphization stage is duplicated in191//!   `visit_mentioned_item`.192//! - Finally, as a performance optimization, the collector should fill `used_mentioned_item` during193//!   its MIR traversal with exactly what mentioned item gathering would have added in the same194//!   situation. This detects mentioned items that have *not* been optimized away and hence don't195//!   need a dedicated traversal.196//!197//! Open Issues198//! -----------199//! Some things are not yet fully implemented in the current version of this200//! module.201//!202//! ### Const Fns203//! Ideally, no mono item should be generated for const fns unless there204//! is a call to them that cannot be evaluated at compile time. At the moment205//! this is not implemented however: a mono item will be produced206//! regardless of whether it is actually needed or not.207208use std::cell::OnceCell;209use std::ops::ControlFlow;210211use rustc_data_structures::fx::FxIndexMap;212use rustc_data_structures::sync::{Lock, par_for_each_in};213use rustc_data_structures::unord::{UnordMap, UnordSet};214use rustc_hir as hir;215use rustc_hir::attrs::InlineAttr;216use rustc_hir::attrs::lang_items::LangItem;217use rustc_hir::def::DefKind;218use rustc_hir::def_id::{DefId, DefIdMap, LocalDefId};219use rustc_middle::middle::codegen_fn_attrs::CodegenFnAttrFlags;220use rustc_middle::mir::interpret::{AllocId, ErrorHandled, GlobalAlloc, Scalar};221use rustc_middle::mir::visit::Visitor as MirVisitor;222use rustc_middle::mir::{self, Body, Location, MentionedItem, traversal};223use rustc_middle::mono::{CollectionMode, InstantiationMode, MonoItem, NormalizationErrorInMono};224use rustc_middle::query::TyCtxtAt;225use rustc_middle::ty::adjustment::{CustomCoerceUnsized, PointerCoercion};226use rustc_middle::ty::layout::ValidityRequirement;227use rustc_middle::ty::{228    self, GenericArgs, GenericParamDefKind, Instance, InstanceKind, ShimKind, Ty, TyCtxt,229    TypeFoldable, TypeVisitable, TypeVisitableExt, TypeVisitor, Unnormalized, VtblEntry,230};231use rustc_middle::util::Providers;232use rustc_middle::{bug, span_bug};233use rustc_session::config::{DebugInfo, EntryFnType, Offload};234use rustc_span::{DUMMY_SP, Span, Spanned, Symbol, dummy_spanned, respan};235use rustc_structures::Limit;236use tracing::{debug, instrument, trace};237238use crate::diagnostics::{239    self, EncounteredErrorWhileInstantiating, EncounteredErrorWhileInstantiatingGlobalAsm,240    NoOptimizedMir, RecursionLimit,241};242243#[derive(PartialEq)]244pub(crate) enum MonoItemCollectionStrategy {245    Eager,246    Lazy,247}248249/// The state that is shared across the concurrent threads that are doing collection.250struct SharedState<'tcx> {251    /// Items that have been or are currently being recursively collected.252    visited: Lock<UnordSet<MonoItem<'tcx>>>,253    /// Items that have been or are currently being recursively treated as "mentioned", i.e., their254    /// consts are evaluated but nothing is added to the collection.255    mentioned: Lock<UnordSet<MonoItem<'tcx>>>,256    /// Which items are being used where, for better errors.257    usage_map: Lock<UsageMap<'tcx>>,258}259260pub(crate) struct UsageMap<'tcx> {261    // Maps every mono item to the mono items used by it.262    pub used_map: UnordMap<MonoItem<'tcx>, Vec<MonoItem<'tcx>>>,263264    // Maps each mono item with users to the mono items that use it.265    // Be careful: subsets `used_map`, so unused items are vacant.266    user_map: UnordMap<MonoItem<'tcx>, Vec<MonoItem<'tcx>>>,267}268269impl<'tcx> UsageMap<'tcx> {270    fn new() -> UsageMap<'tcx> {271        UsageMap { used_map: Default::default(), user_map: Default::default() }272    }273274    fn record_used<'a>(&mut self, user_item: MonoItem<'tcx>, used_items: &'a MonoItems<'tcx>)275    where276        'tcx: 'a,277    {278        for used_item in used_items.items() {279            self.user_map.entry(used_item).or_default().push(user_item);280        }281282        assert!(self.used_map.insert(user_item, used_items.items().collect()).is_none());283    }284285    pub(crate) fn get_user_items(&self, item: MonoItem<'tcx>) -> &[MonoItem<'tcx>] {286        self.user_map.get(&item).map(|items| items.as_slice()).unwrap_or(&[])287    }288289    /// Internally iterate over all inlined items used by `item`.290    pub(crate) fn for_each_inlined_used_item<F>(291        &self,292        tcx: TyCtxt<'tcx>,293        item: MonoItem<'tcx>,294        mut f: F,295    ) where296        F: FnMut(MonoItem<'tcx>),297    {298        let used_items = self.used_map.get(&item).unwrap();299        for used_item in used_items.iter() {300            let is_inlined = used_item.instantiation_mode(tcx) == InstantiationMode::LocalCopy;301            if is_inlined {302                f(*used_item);303            }304        }305    }306}307308struct MonoItems<'tcx> {309    // We want a set of MonoItem + Span where trying to re-insert a MonoItem with a different Span310    // is ignored. Map does that, but it looks odd.311    items: FxIndexMap<MonoItem<'tcx>, Span>,312}313314impl<'tcx> MonoItems<'tcx> {315    fn new() -> Self {316        Self { items: FxIndexMap::default() }317    }318319    fn is_empty(&self) -> bool {320        self.items.is_empty()321    }322323    fn push(&mut self, item: Spanned<MonoItem<'tcx>>) {324        // Insert only if the entry does not exist. A normal insert would stomp the first span that325        // got inserted.326        self.items.entry(item.node).or_insert(item.span);327    }328329    fn items(&self) -> impl Iterator<Item = MonoItem<'tcx>> {330        self.items.keys().cloned()331    }332}333334impl<'tcx> IntoIterator for MonoItems<'tcx> {335    type Item = Spanned<MonoItem<'tcx>>;336    type IntoIter = impl Iterator<Item = Spanned<MonoItem<'tcx>>>;337338    fn into_iter(self) -> Self::IntoIter {339        self.items.into_iter().map(|(item, span)| respan(span, item))340    }341}342343impl<'tcx> Extend<Spanned<MonoItem<'tcx>>> for MonoItems<'tcx> {344    fn extend<I>(&mut self, iter: I)345    where346        I: IntoIterator<Item = Spanned<MonoItem<'tcx>>>,347    {348        for item in iter {349            self.push(item)350        }351    }352}353354fn collect_items_root<'tcx>(355    tcx: TyCtxt<'tcx>,356    starting_item: Spanned<MonoItem<'tcx>>,357    state: &SharedState<'tcx>,358    recursion_limit: Limit,359) {360    if !state.visited.lock().insert(starting_item.node) {361        // We've been here already, no need to search again.362        return;363    }364    let mut recursion_depths = DefIdMap::default();365    collect_items_rec(366        tcx,367        starting_item,368        state,369        &mut recursion_depths,370        recursion_limit,371        CollectionMode::UsedItems,372    );373}374375/// Collect all monomorphized items reachable from `starting_point`, and emit a note diagnostic if a376/// post-monomorphization error is encountered during a collection step.377///378/// `mode` determined whether we are scanning for [used items][CollectionMode::UsedItems]379/// or [mentioned items][CollectionMode::MentionedItems].380#[instrument(skip(tcx, state, recursion_depths, recursion_limit), level = "debug")]381fn collect_items_rec<'tcx>(382    tcx: TyCtxt<'tcx>,383    starting_item: Spanned<MonoItem<'tcx>>,384    state: &SharedState<'tcx>,385    recursion_depths: &mut DefIdMap<usize>,386    recursion_limit: Limit,387    mode: CollectionMode,388) {389    let mut used_items = MonoItems::new();390    let mut mentioned_items = MonoItems::new();391    let recursion_depth_reset;392393    // Post-monomorphization errors MVP394    //395    // We can encounter errors while monomorphizing an item, but we don't have a good way of396    // showing a complete stack of spans ultimately leading to collecting the erroneous one yet.397    // (It's also currently unclear exactly which diagnostics and information would be interesting398    // to report in such cases)399    //400    // This leads to suboptimal error reporting: a post-monomorphization error (PME) will be401    // shown with just a spanned piece of code causing the error, without information on where402    // it was called from. This is especially obscure if the erroneous mono item is in a403    // dependency. See for example issue #85155, where, before minimization, a PME happened two404    // crates downstream from libcore's stdarch, without a way to know which dependency was the405    // cause.406    //407    // If such an error occurs in the current crate, its span will be enough to locate the408    // source. If the cause is in another crate, the goal here is to quickly locate which mono409    // item in the current crate is ultimately responsible for causing the error.410    //411    // To give at least _some_ context to the user: while collecting mono items, we check the412    // error count. If it has changed, a PME occurred, and we trigger some diagnostics about the413    // current step of mono items collection.414    //415    // FIXME: don't rely on global state, instead bubble up errors. Note: this is very hard to do.416    let error_count = tcx.dcx().err_count_on_current_thread();417418    // In `mentioned_items` we collect items that were mentioned in this MIR but possibly do not419    // need to be monomorphized. This is done to ensure that optimizing away function calls does not420    // hide const-eval errors that those calls would otherwise have triggered.421    match starting_item.node {422        MonoItem::Static(def_id) => {423            recursion_depth_reset = None;424425            // Statics always get evaluated (which is possible because they can't be generic), so for426            // `MentionedItems` collection there's nothing to do here.427            if mode == CollectionMode::UsedItems {428                let instance = Instance::mono(tcx, def_id);429430                // Sanity check whether this ended up being collected accidentally431                debug_assert!(tcx.should_codegen_locally(instance));432433                let DefKind::Static { nested, .. } = tcx.def_kind(def_id) else { bug!() };434                // Nested statics have no type.435                if !nested {436                    let ty = instance.ty(tcx, ty::TypingEnv::fully_monomorphized());437                    visit_drop_use(tcx, ty, true, starting_item.span, &mut used_items);438                }439440                if let Ok(alloc) = tcx.eval_static_initializer(def_id) {441                    for &prov in alloc.inner().provenance().ptrs().values() {442                        collect_alloc(tcx, prov.alloc_id(), &mut used_items);443                    }444                }445446                if tcx.needs_thread_local_shim(def_id) {447                    used_items.push(respan(448                        starting_item.span,449                        MonoItem::Fn(Instance {450                            def: InstanceKind::Shim(ShimKind::ThreadLocal(def_id)),451                            args: GenericArgs::empty(),452                        }),453                    ));454                }455            }456457            // mentioned_items stays empty since there's no codegen for statics. statics don't get458            // optimized, and if they did then the const-eval interpreter would have to worry about459            // mentioned_items.460        }461        MonoItem::Fn(instance) => {462            // Sanity check whether this ended up being collected accidentally463            debug_assert!(tcx.should_codegen_locally(instance));464465            // Keep track of the monomorphization recursion depth466            recursion_depth_reset = Some(check_recursion_limit(467                tcx,468                instance,469                starting_item.span,470                recursion_depths,471                recursion_limit,472            ));473474            let Ok((used, mentioned)) = tcx.items_of_instance((instance, mode)) else {475                // Normalization errors here are usually due to trait solving overflow.476                // FIXME: I assume that there are few type errors at post-analysis stage, but not477                // entirely sure.478                // We have to emit the error outside of `items_of_instance` to access the479                // span of the `starting_item`.480                let def_id = instance.def_id();481                let def_span = tcx.def_span(def_id);482                let def_path_str = tcx.def_path_str(def_id);483                tcx.dcx().emit_fatal(RecursionLimit {484                    span: starting_item.span,485                    instance,486                    def_span,487                    def_path_str,488                });489            };490            used_items.extend(used.into_iter().copied());491            mentioned_items.extend(mentioned.into_iter().copied());492        }493        MonoItem::GlobalAsm(item_id) => {494            assert!(495                mode == CollectionMode::UsedItems,496                "should never encounter global_asm when collecting mentioned items"497            );498            recursion_depth_reset = None;499500            let item = tcx.hir_item(item_id);501            if let hir::ItemKind::GlobalAsm { asm, .. } = item.kind {502                for (op, op_sp) in asm.operands {503                    match *op {504                        hir::InlineAsmOperand::Const { anon_const } => {505                            match tcx.const_eval_poly(anon_const.def_id.to_def_id()) {506                                Ok(val) => {507                                    collect_const_value(tcx, val, &mut used_items);508                                }509                                Err(ErrorHandled::TooGeneric(..)) => {510                                    span_bug!(*op_sp, "asm const cannot be resolved; too generic")511                                }512                                Err(ErrorHandled::Reported(..)) => {513                                    continue;514                                }515                            }516                        }517                        hir::InlineAsmOperand::SymFn { expr } => {518                            let fn_ty = tcx.typeck(item_id.owner_id).expr_ty(expr);519                            visit_fn_use(tcx, fn_ty, false, *op_sp, &mut used_items);520                        }521                        hir::InlineAsmOperand::SymStatic { path: _, def_id } => {522                            let instance = Instance::mono(tcx, def_id);523                            if tcx.should_codegen_locally(instance) {524                                trace!("collecting static {:?}", def_id);525                                used_items.push(dummy_spanned(MonoItem::Static(def_id)));526                            }527                        }528                        hir::InlineAsmOperand::In { .. }529                        | hir::InlineAsmOperand::Out { .. }530                        | hir::InlineAsmOperand::InOut { .. }531                        | hir::InlineAsmOperand::SplitInOut { .. }532                        | hir::InlineAsmOperand::Label { .. } => {533                            span_bug!(*op_sp, "invalid operand type for global_asm!")534                        }535                    }536                }537            } else {538                span_bug!(item.span, "Mismatch between hir::Item type and MonoItem type")539            }540541            // mention_items stays empty as nothing gets optimized here.542        }543    };544545    // Check for PMEs and emit a diagnostic if one happened. To try to show relevant edges of the546    // mono item graph.547    if tcx.dcx().err_count_on_current_thread() > error_count548        && starting_item.node.is_generic_fn()549        && starting_item.node.is_user_defined()550    {551        match starting_item.node {552            MonoItem::Fn(instance) => tcx.dcx().emit_note(EncounteredErrorWhileInstantiating {553                span: starting_item.span,554                kind: "fn",555                instance,556            }),557            MonoItem::Static(def_id) => tcx.dcx().emit_note(EncounteredErrorWhileInstantiating {558                span: starting_item.span,559                kind: "static",560                instance: Instance::new_raw(def_id, GenericArgs::empty()),561            }),562            MonoItem::GlobalAsm(_) => {563                tcx.dcx().emit_note(EncounteredErrorWhileInstantiatingGlobalAsm {564                    span: starting_item.span,565                })566            }567        }568    }569    // Only updating `usage_map` for used items as otherwise we may be inserting the same item570    // multiple times (if it is first 'mentioned' and then later actually used), and the usage map571    // logic does not like that.572    // This is part of the output of collection and hence only relevant for "used" items.573    // ("Mentioned" items are only considered internally during collection.)574    if mode == CollectionMode::UsedItems {575        state.usage_map.lock().record_used(starting_item.node, &used_items);576    }577578    {579        let mut visited = OnceCell::default();580        if mode == CollectionMode::UsedItems {581            used_items582                .items583                .retain(|k, _| visited.get_mut_or_init(|| state.visited.lock()).insert(*k));584        }585586        let mut mentioned = OnceCell::default();587        mentioned_items.items.retain(|k, _| {588            !visited.get_or_init(|| state.visited.lock()).contains(k)589                && mentioned.get_mut_or_init(|| state.mentioned.lock()).insert(*k)590        });591    }592    if mode == CollectionMode::MentionedItems {593        assert!(used_items.is_empty(), "'mentioned' collection should never encounter used items");594    } else {595        for used_item in used_items {596            collect_items_rec(597                tcx,598                used_item,599                state,600                recursion_depths,601                recursion_limit,602                CollectionMode::UsedItems,603            );604        }605    }606607    // Walk over mentioned items *after* used items, so that if an item is both mentioned and used then608    // the loop above has fully collected it, so this loop will skip it.609    for mentioned_item in mentioned_items {610        collect_items_rec(611            tcx,612            mentioned_item,613            state,614            recursion_depths,615            recursion_limit,616            CollectionMode::MentionedItems,617        );618    }619620    if let Some((def_id, depth)) = recursion_depth_reset {621        recursion_depths.insert(def_id, depth);622    }623}624625// Check whether we can normalize every type in the instantiated MIR body.626fn check_normalization_error<'tcx>(627    tcx: TyCtxt<'tcx>,628    instance: Instance<'tcx>,629    body: &Body<'tcx>,630) -> Result<(), NormalizationErrorInMono> {631    struct NormalizationChecker<'tcx> {632        tcx: TyCtxt<'tcx>,633        instance: Instance<'tcx>,634    }635    impl<'tcx> TypeVisitor<TyCtxt<'tcx>> for NormalizationChecker<'tcx> {636        type Result = ControlFlow<()>;637638        fn visit_ty(&mut self, t: Ty<'tcx>) -> Self::Result {639            match self.instance.try_instantiate_mir_and_normalize_erasing_regions(640                self.tcx,641                ty::TypingEnv::fully_monomorphized(),642                ty::EarlyBinder::bind(self.tcx, t),643            ) {644                Ok(_) => ControlFlow::Continue(()),645                Err(_) => ControlFlow::Break(()),646            }647        }648    }649650    let mut checker = NormalizationChecker { tcx, instance };651    if body.visit_with(&mut checker).is_break() { Err(NormalizationErrorInMono) } else { Ok(()) }652}653654fn check_recursion_limit<'tcx>(655    tcx: TyCtxt<'tcx>,656    instance: Instance<'tcx>,657    span: Span,658    recursion_depths: &mut DefIdMap<usize>,659    recursion_limit: Limit,660) -> (DefId, usize) {661    let def_id = instance.def_id();662    let recursion_depth = recursion_depths.get(&def_id).cloned().unwrap_or(0);663    debug!(" => recursion depth={}", recursion_depth);664665    let adjusted_recursion_depth = if tcx.is_lang_item(def_id, LangItem::DropGlue) {666        // HACK: `drop_glue` creates tight monomorphization loops. Give667        // it more margin.668        recursion_depth / 4669    } else {670        recursion_depth671    };672673    // Code that needs to instantiate the same function recursively674    // more than the recursion limit is assumed to be causing an675    // infinite expansion.676    if !recursion_limit.value_within_limit(adjusted_recursion_depth) {677        let def_span = tcx.def_span(def_id);678        let def_path_str = tcx.def_path_str(def_id);679        tcx.dcx().emit_fatal(RecursionLimit { span, instance, def_span, def_path_str });680    }681682    recursion_depths.insert(def_id, recursion_depth + 1);683684    (def_id, recursion_depth)685}686687struct MirUsedCollector<'a, 'tcx> {688    tcx: TyCtxt<'tcx>,689    body: &'a mir::Body<'tcx>,690    used_items: &'a mut MonoItems<'tcx>,691    /// See the comment in `collect_items_of_instance` for the purpose of this set.692    /// Note that this contains *not-monomorphized* items!693    used_mentioned_items: &'a mut UnordSet<MentionedItem<'tcx>>,694    instance: Instance<'tcx>,695}696697impl<'a, 'tcx> MirUsedCollector<'a, 'tcx> {698    fn monomorphize<T>(&self, value: T) -> T699    where700        T: TypeFoldable<TyCtxt<'tcx>>,701    {702        trace!("monomorphize: self.instance={:?}", self.instance);703        self.instance.instantiate_mir_and_normalize_erasing_regions(704            self.tcx,705            ty::TypingEnv::fully_monomorphized(),706            ty::EarlyBinder::bind(self.tcx, value),707        )708    }709710    /// Evaluates a *not yet monomorphized* constant.711    fn eval_constant(&mut self, constant: &mir::ConstOperand<'tcx>) -> Option<mir::ConstValue> {712        let const_ = self.monomorphize(constant.const_);713        // Evaluate the constant. This makes const eval failure a collection-time error (rather than714        // a codegen-time error). rustc stops after collection if there was an error, so this715        // ensures codegen never has to worry about failing consts.716        // (codegen relies on this and ICEs will happen if this is violated.)717        match const_.eval(self.tcx, ty::TypingEnv::fully_monomorphized(), constant.span) {718            Ok(v) => Some(v),719            Err(ErrorHandled::TooGeneric(..)) => span_bug!(720                constant.span,721                "collection encountered polymorphic constant: {:?}",722                const_723            ),724            Err(err @ ErrorHandled::Reported(..)) => {725                err.emit_note(self.tcx);726                return None;727            }728        }729    }730}731732impl<'a, 'tcx> MirVisitor<'tcx> for MirUsedCollector<'a, 'tcx> {733    fn visit_rvalue(&mut self, rvalue: &mir::Rvalue<'tcx>, location: Location) {734        debug!("visiting rvalue {:?}", *rvalue);735736        let span = self.body.source_info(location).span;737738        match *rvalue {739            // When doing an cast from a regular pointer to a wide pointer, we740            // have to instantiate all methods of the trait being cast to, so we741            // can build the appropriate vtable.742            mir::Rvalue::Cast(743                mir::CastKind::PointerCoercion(PointerCoercion::Unsize, _),744                ref operand,745                target_ty,746            ) => {747                let source_ty = operand.ty(self.body, self.tcx);748                // *Before* monomorphizing, record that we already handled this mention.749                self.used_mentioned_items750                    .insert(MentionedItem::UnsizeCast { source_ty, target_ty });751                let target_ty = self.monomorphize(target_ty);752                let source_ty = self.monomorphize(source_ty);753                let (source_ty, target_ty) =754                    find_tails_for_unsizing(self.tcx.at(span), source_ty, target_ty);755                // This could also be a different Unsize instruction, like756                // from a fixed sized array to a slice. But we are only757                // interested in things that produce a vtable.758                if target_ty.is_trait() && !source_ty.is_trait() {759                    create_mono_items_for_vtable_methods(760                        self.tcx,761                        target_ty,762                        source_ty,763                        span,764                        self.used_items,765                    );766                }767            }768            mir::Rvalue::Cast(769                mir::CastKind::PointerCoercion(PointerCoercion::ReifyFnPointer(_), _),770                ref operand,771                _,772            ) => {773                let fn_ty = operand.ty(self.body, self.tcx);774                // *Before* monomorphizing, record that we already handled this mention.775                self.used_mentioned_items.insert(MentionedItem::Fn(fn_ty));776                let fn_ty = self.monomorphize(fn_ty);777                visit_fn_use(self.tcx, fn_ty, false, span, self.used_items);778            }779            mir::Rvalue::Cast(780                mir::CastKind::PointerCoercion(PointerCoercion::ClosureFnPointer(_), _),781                ref operand,782                _,783            ) => {784                let source_ty = operand.ty(self.body, self.tcx);785                // *Before* monomorphizing, record that we already handled this mention.786                self.used_mentioned_items.insert(MentionedItem::Closure(source_ty));787                let source_ty = self.monomorphize(source_ty);788                if let ty::Closure(def_id, args) = *source_ty.kind() {789                    let instance =790                        Instance::resolve_closure(self.tcx, def_id, args, ty::ClosureKind::FnOnce);791                    if self.tcx.should_codegen_locally(instance) {792                        self.used_items.push(create_fn_mono_item(self.tcx, instance, span));793                    }794                } else {795                    bug!()796                }797            }798            mir::Rvalue::ThreadLocalRef(def_id) => {799                assert!(self.tcx.is_thread_local_static(def_id));800                let instance = Instance::mono(self.tcx, def_id);801                if self.tcx.should_codegen_locally(instance) {802                    trace!("collecting thread-local static {:?}", def_id);803                    self.used_items.push(respan(span, MonoItem::Static(def_id)));804                }805            }806            _ => { /* not interesting */ }807        }808809        self.super_rvalue(rvalue, location);810    }811812    /// This does not walk the MIR of the constant as that is not needed for codegen, all we need is813    /// to ensure that the constant evaluates successfully and walk the result.814    #[instrument(skip(self), level = "debug")]815    fn visit_const_operand(&mut self, constant: &mir::ConstOperand<'tcx>, _location: Location) {816        // No `super_constant` as we don't care about `visit_ty`/`visit_ty_const`.817        let Some(val) = self.eval_constant(constant) else { return };818        collect_const_value(self.tcx, val, self.used_items);819    }820821    fn visit_terminator(&mut self, terminator: &mir::Terminator<'tcx>, location: Location) {822        debug!("visiting terminator {:?} @ {:?}", terminator, location);823        let source = self.body.source_info(location).span;824825        let tcx = self.tcx;826        let push_mono_lang_item = |this: &mut Self, lang_item: LangItem| {827            let instance = Instance::mono(tcx, tcx.require_lang_item(lang_item, source));828            if tcx.should_codegen_locally(instance) {829                this.used_items.push(create_fn_mono_item(tcx, instance, source));830            }831        };832833        match terminator.kind {834            mir::TerminatorKind::Call { ref func, ref args, .. }835            | mir::TerminatorKind::TailCall { ref func, ref args, .. } => {836                let callee_ty = func.ty(self.body, tcx);837                // *Before* monomorphizing, record that we already handled this mention.838                self.used_mentioned_items.insert(MentionedItem::Fn(callee_ty));839                let callee_ty = self.monomorphize(callee_ty);840841                // HACK(explicit_tail_calls): collect tail calls to `#[track_caller]` functions as indirect,842                // because we later call them as such, to prevent issues with ABI incompatibility.843                // Ideally we'd replace such tail calls with normal call + return, but this requires844                // post-mono MIR optimizations, which we don't yet have.845                let force_indirect_call =846                    if matches!(terminator.kind, mir::TerminatorKind::TailCall { .. })847                        && let &ty::FnDef(def_id, args) = callee_ty.kind()848                        && let instance = ty::Instance::expect_resolve(849                            self.tcx,850                            ty::TypingEnv::fully_monomorphized(),851                            def_id,852                            args.no_bound_vars().unwrap(),853                            source,854                        )855                        && instance.def.requires_caller_location(self.tcx)856                    {857                        true858                    } else {859                        false860                    };861862                visit_fn_use(863                    self.tcx,864                    callee_ty,865                    !force_indirect_call,866                    source,867                    &mut self.used_items,868                );869870                if let ty::FnDef(def_id, _) = *callee_ty.kind()871                    && self.tcx.is_intrinsic(def_id, rustc_span::sym::offload)872                    && let Some(kernel) = args.first()873                {874                    let kernel_ty = kernel.node.ty(self.body, self.tcx);875                    let kernel_ty = self.monomorphize(kernel_ty);876                    visit_fn_use(self.tcx, kernel_ty, false, source, &mut self.used_items);877                }878            }879            mir::TerminatorKind::Drop { ref place, .. } => {880                let ty = place.ty(self.body, self.tcx).ty;881                // *Before* monomorphizing, record that we already handled this mention.882                self.used_mentioned_items.insert(MentionedItem::Drop(ty));883                let ty = self.monomorphize(ty);884                visit_drop_use(self.tcx, ty, true, source, self.used_items);885            }886            mir::TerminatorKind::InlineAsm { ref operands, .. } => {887                for op in operands {888                    match *op {889                        mir::InlineAsmOperand::SymFn { ref value } => {890                            let fn_ty = value.const_.ty();891                            // *Before* monomorphizing, record that we already handled this mention.892                            self.used_mentioned_items.insert(MentionedItem::Fn(fn_ty));893                            let fn_ty = self.monomorphize(fn_ty);894                            visit_fn_use(self.tcx, fn_ty, false, source, self.used_items);895                        }896                        mir::InlineAsmOperand::SymStatic { def_id } => {897                            let instance = Instance::mono(self.tcx, def_id);898                            if self.tcx.should_codegen_locally(instance) {899                                trace!("collecting asm sym static {:?}", def_id);900                                self.used_items.push(respan(source, MonoItem::Static(def_id)));901                            }902                        }903                        _ => {}904                    }905                }906            }907            mir::TerminatorKind::Assert { ref msg, .. } => match &**msg {908                mir::AssertKind::BoundsCheck { .. } => {909                    push_mono_lang_item(self, LangItem::PanicBoundsCheck);910                }911                mir::AssertKind::MisalignedPointerDereference { .. } => {912                    push_mono_lang_item(self, LangItem::PanicMisalignedPointerDereference);913                }914                mir::AssertKind::NullPointerDereference => {915                    push_mono_lang_item(self, LangItem::PanicNullPointerDereference);916                }917                mir::AssertKind::NullReferenceConstructed => {918                    push_mono_lang_item(self, LangItem::PanicNullReferenceConstructed);919                }920                mir::AssertKind::InvalidEnumConstruction(_) => {921                    push_mono_lang_item(self, LangItem::PanicInvalidEnumConstruction);922                }923                _ => {924                    push_mono_lang_item(self, msg.panic_function());925                }926            },927            mir::TerminatorKind::UnwindTerminate(reason) => {928                push_mono_lang_item(self, reason.lang_item());929            }930            mir::TerminatorKind::Goto { .. }931            | mir::TerminatorKind::SwitchInt { .. }932            | mir::TerminatorKind::UnwindResume933            | mir::TerminatorKind::Return934            | mir::TerminatorKind::Unreachable => {}935            mir::TerminatorKind::CoroutineDrop936            | mir::TerminatorKind::Yield { .. }937            | mir::TerminatorKind::FalseEdge { .. }938            | mir::TerminatorKind::FalseUnwind { .. } => bug!(),939        }940941        if let Some(mir::UnwindAction::Terminate(reason)) = terminator.unwind() {942            push_mono_lang_item(self, reason.lang_item());943        }944945        self.super_terminator(terminator, location);946    }947}948949fn visit_drop_use<'tcx>(950    tcx: TyCtxt<'tcx>,951    ty: Ty<'tcx>,952    is_direct_call: bool,953    source: Span,954    output: &mut MonoItems<'tcx>,955) {956    let instance = Instance::resolve_drop_glue(tcx, ty);957    visit_instance_use(tcx, instance, is_direct_call, source, output);958}959960/// For every call of this function in the visitor, make sure there is a matching call in the961/// `mentioned_items` pass!962fn visit_fn_use<'tcx>(963    tcx: TyCtxt<'tcx>,964    ty: Ty<'tcx>,965    is_direct_call: bool,966    source: Span,967    output: &mut MonoItems<'tcx>,968) {969    if let ty::FnDef(def_id, args) = *ty.kind() {970        let args = args.no_bound_vars().unwrap();971        let instance = if is_direct_call {972            ty::Instance::expect_resolve(973                tcx,974                ty::TypingEnv::fully_monomorphized(),975                def_id,976                args,977                source,978            )979        } else {980            match ty::Instance::resolve_for_fn_ptr(981                tcx,982                ty::TypingEnv::fully_monomorphized(),983                def_id,984                args,985            ) {986                Some(instance) => instance,987                _ => bug!("failed to resolve instance for {ty}"),988            }989        };990        visit_instance_use(tcx, instance, is_direct_call, source, output);991    }992}993994fn visit_instance_use<'tcx>(995    tcx: TyCtxt<'tcx>,996    instance: ty::Instance<'tcx>,997    is_direct_call: bool,998    source: Span,999    output: &mut MonoItems<'tcx>,1000) {1001    debug!("visit_item_use({:?}, is_direct_call={:?})", instance, is_direct_call);1002    if !tcx.should_codegen_locally(instance) {1003        return;1004    }1005    if let Some(intrinsic) = tcx.intrinsic(instance.def_id()) {1006        if let Some(_requirement) = ValidityRequirement::from_intrinsic(intrinsic.name) {1007            // The intrinsics assert_inhabited, assert_zero_valid, and assert_mem_uninitialized_valid will1008            // be lowered in codegen to nothing or a call to panic_nounwind. So if we encounter any1009            // of those intrinsics, we need to include a mono item for panic_nounwind, else we may try to1010            // codegen a call to that function without generating code for the function itself.1011            let def_id = tcx.require_lang_item(LangItem::PanicNounwind, source);1012            let panic_instance = Instance::mono(tcx, def_id);1013            if tcx.should_codegen_locally(panic_instance) {1014                output.push(create_fn_mono_item(tcx, panic_instance, source));1015            }1016        } else if !intrinsic.must_be_overridden1017            && (tcx.sess.opts.unstable_opts.force_intrinsic_fallback1018                || !tcx.sess.replaced_intrinsics.contains(&intrinsic.name))1019        {1020            // Codegen the fallback body of intrinsics with fallback bodies.1021            // We have to skip this otherwise as there's no body to codegen.1022            //1023            // We also skip `replaced_intrinsics` which are always replaced by the backend and hence1024            // monomorphizing the fallback body would be pointless.1025            //1026            // However, when -Zforce-intrinsic-fallback is set (e.g. to test the fallback1027            // implementations) we ignore the optimization hint and do monomorphize1028            // the fallback body.1029            let instance = ty::Instance::new_raw(instance.def_id(), instance.args);1030            if tcx.should_codegen_locally(instance) {1031                output.push(create_fn_mono_item(tcx, instance, source));1032            }1033        }1034    }10351036    match instance.def {1037        ty::InstanceKind::Virtual(..)1038        | ty::InstanceKind::Intrinsic(_)1039        | ty::InstanceKind::LlvmIntrinsic(_) => {1040            if !is_direct_call {1041                bug!("{:?} being reified", instance);1042            }1043        }1044        ty::InstanceKind::Shim(ty::ShimKind::ThreadLocal(..)) => {1045            bug!("{:?} being reified", instance);1046        }1047        ty::InstanceKind::Shim(ty::ShimKind::DropGlue(_, None)) => {1048            // Don't need to emit noop drop glue if we are calling directly.1049            //1050            // Note that we also optimize away the call to visit_instance_use in vtable construction1051            // (see create_mono_items_for_vtable_methods).1052            if !is_direct_call {1053                output.push(create_fn_mono_item(tcx, instance, source));1054            }1055        }1056        ty::InstanceKind::Item(..)1057        | ty::InstanceKind::Shim(ty::ShimKind::DropGlue(_, Some(_)))1058        | ty::InstanceKind::Shim(ty::ShimKind::FutureDropPoll(..))1059        | ty::InstanceKind::Shim(ty::ShimKind::AsyncDropGlue(_, _))1060        | ty::InstanceKind::Shim(ty::ShimKind::AsyncDropGlueCtor(_, _))1061        | ty::InstanceKind::Shim(ty::ShimKind::VTable(..))1062        | ty::InstanceKind::Shim(ty::ShimKind::Reify(..))1063        | ty::InstanceKind::Shim(ty::ShimKind::ClosureOnce { .. })1064        | ty::InstanceKind::Shim(ty::ShimKind::ConstructCoroutineInClosure { .. })1065        | ty::InstanceKind::Shim(ty::ShimKind::FnPtr(..))1066        | ty::InstanceKind::Shim(ty::ShimKind::Clone(..))1067        | ty::InstanceKind::Shim(ty::ShimKind::FnPtrAsPtr(..))1068        | ty::InstanceKind::Shim(ty::ShimKind::FnPtrFromPtr(..)) => {1069            output.push(create_fn_mono_item(tcx, instance, source));1070        }1071    }1072}10731074/// Returns `true` if we should codegen an instance in the local crate, or returns `false` if we1075/// can just link to the upstream crate and therefore don't need a mono item.1076fn should_codegen_locally<'tcx>(tcx: TyCtxt<'tcx>, instance: Instance<'tcx>) -> bool {1077    let Some(def_id) = instance.def.def_id_if_not_guaranteed_local_codegen() else {1078        return true;1079    };10801081    if tcx.is_foreign_item(def_id) {1082        // Foreign items are always linked against, there's no way of instantiating them.1083        return false;1084    }10851086    if tcx.def_kind(def_id).has_codegen_attrs()1087        && matches!(tcx.codegen_fn_attrs(def_id).inline, InlineAttr::Force { .. })1088    {1089        // `#[rustc_force_inline]` items should never be codegened. This should be caught by1090        // the MIR validator.1091        tcx.dcx().delayed_bug("attempt to codegen `#[rustc_force_inline]` item");1092    }10931094    if def_id.is_local() {1095        // Local items cannot be referred to locally without monomorphizing them locally.1096        return true;1097    }10981099    if tcx.is_reachable_non_generic(def_id) || instance.upstream_monomorphization(tcx).is_some() {1100        // We can link to the item in question, no instance needed in this crate.1101        return false;1102    }11031104    if let DefKind::Static { .. } = tcx.def_kind(def_id) {1105        // We cannot monomorphize statics from upstream crates.1106        return false;1107    }11081109    // See comment in should_encode_mir in rustc_metadata for why we don't report1110    // an error for constructors.1111    if !tcx.is_mir_available(def_id) && !matches!(tcx.def_kind(def_id), DefKind::Ctor(..)) {1112        tcx.dcx().emit_fatal(NoOptimizedMir {1113            span: tcx.def_span(def_id),1114            crate_name: tcx.crate_name(def_id.krate),1115            instance: instance.to_string(),1116        });1117    }11181119    true1120}11211122/// For a given pair of source and target type that occur in an unsizing coercion,1123/// this function finds the pair of types that determines the vtable linking1124/// them.1125///1126/// For example, the source type might be `&SomeStruct` and the target type1127/// might be `&dyn SomeTrait` in a cast like:1128///1129/// ```rust,ignore (not real code)1130/// let src: &SomeStruct = ...;1131/// let target = src as &dyn SomeTrait;1132/// ```1133///1134/// Then the output of this function would be (SomeStruct, SomeTrait) since for1135/// constructing the `target` wide-pointer we need the vtable for that pair.1136///1137/// Things can get more complicated though because there's also the case where1138/// the unsized type occurs as a field:1139///1140/// ```rust1141/// struct ComplexStruct<T: ?Sized> {1142///    a: u32,1143///    b: f64,1144///    c: T1145/// }1146/// ```1147///1148/// In this case, if `T` is sized, `&ComplexStruct<T>` is a thin pointer. If `T`1149/// is unsized, `&SomeStruct` is a wide pointer, and the vtable it points to is1150/// for the pair of `T` (which is a trait) and the concrete type that `T` was1151/// originally coerced from:1152///1153/// ```rust,ignore (not real code)1154/// let src: &ComplexStruct<SomeStruct> = ...;1155/// let target = src as &ComplexStruct<dyn SomeTrait>;1156/// ```1157///1158/// Again, we want this `find_vtable_types_for_unsizing()` to provide the pair1159/// `(SomeStruct, SomeTrait)`.1160///1161/// Finally, there is also the case of custom unsizing coercions, e.g., for1162/// smart pointers such as `Rc` and `Arc`.1163fn find_tails_for_unsizing<'tcx>(1164    tcx: TyCtxtAt<'tcx>,1165    source_ty: Ty<'tcx>,1166    target_ty: Ty<'tcx>,1167) -> (Ty<'tcx>, Ty<'tcx>) {1168    let typing_env = ty::TypingEnv::fully_monomorphized();1169    debug_assert!(!source_ty.has_param(), "{source_ty} should be fully monomorphic");1170    debug_assert!(!target_ty.has_param(), "{target_ty} should be fully monomorphic");11711172    match (source_ty.kind(), target_ty.kind()) {1173        (&ty::Pat(source, _), &ty::Pat(target, _)) => find_tails_for_unsizing(tcx, source, target),1174        (1175            &ty::Ref(_, source_pointee, _),1176            &ty::Ref(_, target_pointee, _) | &ty::RawPtr(target_pointee, _),1177        )1178        | (&ty::RawPtr(source_pointee, _), &ty::RawPtr(target_pointee, _)) => {1179            tcx.struct_lockstep_tails_for_codegen(source_pointee, target_pointee, typing_env)1180        }11811182        // `Box<T>` could go through the ADT code below, b/c it'll unpeel to `Unique<T>`,1183        // and eventually bottom out in a raw ref, but we can micro-optimize it here.1184        (_, _)1185            if let Some(source_boxed) = source_ty.boxed_ty()1186                && let Some(target_boxed) = target_ty.boxed_ty() =>1187        {1188            tcx.struct_lockstep_tails_for_codegen(source_boxed, target_boxed, typing_env)1189        }11901191        (&ty::Adt(source_adt_def, source_args), &ty::Adt(target_adt_def, target_args)) => {1192            assert_eq!(source_adt_def, target_adt_def);1193            let CustomCoerceUnsized::Struct(coerce_index) =1194                match crate::custom_coerce_unsize_info(tcx, source_ty, target_ty) {1195                    Ok(ccu) => ccu,1196                    Err(e) => {1197                        let e = Ty::new_error(tcx.tcx, e);1198                        return (e, e);1199                    }1200                };1201            let coerce_field = &source_adt_def.non_enum_variant().fields[coerce_index];1202            // We're getting a possibly unnormalized type, so normalize it.1203            let source_field =1204                tcx.normalize_erasing_regions(typing_env, coerce_field.ty(*tcx, source_args));1205            let target_field =1206                tcx.normalize_erasing_regions(typing_env, coerce_field.ty(*tcx, target_args));1207            find_tails_for_unsizing(tcx, source_field, target_field)1208        }12091210        _ => bug!(1211            "find_vtable_types_for_unsizing: invalid coercion {:?} -> {:?}",1212            source_ty,1213            target_ty1214        ),1215    }1216}12171218#[instrument(skip(tcx), level = "debug", ret)]1219fn create_fn_mono_item<'tcx>(1220    tcx: TyCtxt<'tcx>,1221    instance: Instance<'tcx>,1222    source: Span,1223) -> Spanned<MonoItem<'tcx>> {1224    let def_id = instance.def_id();1225    if tcx.sess.opts.unstable_opts.profile_closures1226        && def_id.is_local()1227        && tcx.is_closure_like(def_id)1228    {1229        crate::util::dump_closure_profile(tcx, instance);1230    }12311232    respan(source, MonoItem::Fn(instance))1233}12341235/// Creates a `MonoItem` for each method that is referenced by the vtable for1236/// the given trait/impl pair.1237fn create_mono_items_for_vtable_methods<'tcx>(1238    tcx: TyCtxt<'tcx>,1239    trait_ty: Ty<'tcx>,1240    impl_ty: Ty<'tcx>,1241    source: Span,1242    output: &mut MonoItems<'tcx>,1243) {1244    assert!(!trait_ty.has_escaping_bound_vars() && !impl_ty.has_escaping_bound_vars());12451246    let ty::Dynamic(trait_ty, ..) = trait_ty.kind() else {1247        bug!("create_mono_items_for_vtable_methods: {trait_ty:?} not a trait type");1248    };1249    if let Some(principal) = trait_ty.principal() {1250        let trait_ref =1251            tcx.instantiate_bound_regions_with_erased(principal.with_self_ty(tcx, impl_ty));1252        assert!(!trait_ref.has_escaping_bound_vars());12531254        // Walk all methods of the trait, including those of its supertraits1255        let entries = tcx.vtable_entries(trait_ref);1256        debug!(?entries);1257        let methods = entries1258            .iter()1259            .filter_map(|entry| match entry {1260                VtblEntry::MetadataDropInPlace1261                | VtblEntry::MetadataSize1262                | VtblEntry::MetadataAlign1263                | VtblEntry::Vacant => None,1264                VtblEntry::TraitVPtr(_) => {1265                    // all super trait items already covered, so skip them.1266                    None1267                }1268                VtblEntry::Method(instance) => {1269                    Some(*instance).filter(|instance| tcx.should_codegen_locally(*instance))1270                }1271            })1272            .map(|item| create_fn_mono_item(tcx, item, source));1273        output.extend(methods);1274    }12751276    // Also add the destructor, if it's necessary.1277    //1278    // This matches the check in vtable_allocation_provider in middle/ty/vtable.rs,1279    // if we don't need drop we're not adding an actual pointer to the vtable.1280    if impl_ty.needs_drop(tcx, ty::TypingEnv::fully_monomorphized()) {1281        visit_drop_use(tcx, impl_ty, false, source, output);1282    }1283}12841285/// Scans the CTFE alloc in order to find function pointers and statics that must be monomorphized.1286fn collect_alloc<'tcx>(tcx: TyCtxt<'tcx>, alloc_id: AllocId, output: &mut MonoItems<'tcx>) {1287    match tcx.global_alloc(alloc_id) {1288        GlobalAlloc::Static(def_id) => {1289            assert!(!tcx.is_thread_local_static(def_id));1290            let instance = Instance::mono(tcx, def_id);1291            if tcx.should_codegen_locally(instance) {1292                trace!("collecting static {:?}", def_id);1293                output.push(dummy_spanned(MonoItem::Static(def_id)));1294            }1295        }1296        GlobalAlloc::Memory(alloc) => {1297            trace!("collecting {:?} with {:#?}", alloc_id, alloc);1298            let ptrs = alloc.inner().provenance().ptrs();1299            if !ptrs.is_empty() {1300                for &prov in ptrs.values() {1301                    collect_alloc(tcx, prov.alloc_id(), output);1302                }1303            }1304        }1305        GlobalAlloc::Function { instance, .. } => {1306            if tcx.should_codegen_locally(instance) {1307                trace!("collecting {:?} with {:#?}", alloc_id, instance);1308                output.push(create_fn_mono_item(tcx, instance, DUMMY_SP));1309            }1310        }1311        GlobalAlloc::VTable(ty, dyn_ty) => {1312            let alloc_id = tcx.vtable_allocation((1313                ty,1314                dyn_ty1315                    .principal()1316                    .map(|principal| tcx.instantiate_bound_regions_with_erased(principal)),1317            ));1318            collect_alloc(tcx, alloc_id, output)1319        }1320        GlobalAlloc::TypeId { .. } => {}1321    }1322}13231324/// Scans the MIR in order to find function calls, closures, and drop-glue.1325///1326/// Anything that's found is added to `output`. Furthermore the "mentioned items" of the MIR are returned.1327#[instrument(skip(tcx), level = "debug")]1328fn collect_items_of_instance<'tcx>(1329    tcx: TyCtxt<'tcx>,1330    instance: Instance<'tcx>,1331    mode: CollectionMode,1332) -> Result<(MonoItems<'tcx>, MonoItems<'tcx>), NormalizationErrorInMono> {1333    // This item is getting monomorphized, do mono-time checks.1334    let body = tcx.instance_mir(instance.def);1335    // Plenty of code paths later assume that everything can be normalized. So we have to check1336    // normalization first.1337    // We choose to emit the error outside to provide helpful diagnostics.1338    check_normalization_error(tcx, instance, body)?;1339    tcx.ensure_ok().check_mono_item(instance);13401341    // Naively, in "used" collection mode, all functions get added to *both* `used_items` and1342    // `mentioned_items`. Mentioned items processing will then notice that they have already been1343    // visited, but at that point each mentioned item has been monomorphized, added to the1344    // `mentioned_items` worklist, and checked in the global set of visited items. To remove that1345    // overhead, we have a special optimization that avoids adding items to `mentioned_items` when1346    // they are already added in `used_items`. We could just scan `used_items`, but that's a linear1347    // scan and not very efficient. Furthermore we can only do that *after* monomorphizing the1348    // mentioned item. So instead we collect all pre-monomorphized `MentionedItem` that were already1349    // added to `used_items` in a hash set, which can efficiently query in the1350    // `body.mentioned_items` loop below without even having to monomorphize the item.1351    let mut used_items = MonoItems::new();1352    let mut mentioned_items = MonoItems::new();1353    let mut used_mentioned_items = Default::default();1354    let mut collector = MirUsedCollector {1355        tcx,1356        body,1357        used_items: &mut used_items,1358        used_mentioned_items: &mut used_mentioned_items,1359        instance,1360    };13611362    if mode == CollectionMode::UsedItems {1363        if tcx.sess.opts.debuginfo == DebugInfo::Full {1364            for var_debug_info in &body.var_debug_info {1365                collector.visit_var_debug_info(var_debug_info);1366            }1367        }1368        for (bb, data) in traversal::mono_reachable(body, tcx, instance) {1369            collector.visit_basic_block_data(bb, data)1370        }1371    }13721373    // Always visit all `required_consts`, so that we evaluate them and abort compilation if any of1374    // them errors.1375    for const_op in body.required_consts() {1376        if let Some(val) = collector.eval_constant(const_op) {1377            collect_const_value(tcx, val, &mut mentioned_items);1378        }1379    }13801381    // Always gather mentioned items. We try to avoid processing items that we have already added to1382    // `used_items` above.1383    for item in body.mentioned_items() {1384        if !collector.used_mentioned_items.contains(&item.node) {1385            let item_mono = collector.monomorphize(item.node);1386            visit_mentioned_item(tcx, &item_mono, item.span, &mut mentioned_items);1387        }1388    }13891390    Ok((used_items, mentioned_items))1391}13921393fn items_of_instance<'tcx>(1394    tcx: TyCtxt<'tcx>,1395    (instance, mode): (Instance<'tcx>, CollectionMode),1396) -> Result<1397    (&'tcx [Spanned<MonoItem<'tcx>>], &'tcx [Spanned<MonoItem<'tcx>>]),1398    NormalizationErrorInMono,1399> {1400    let (used_items, mentioned_items) = collect_items_of_instance(tcx, instance, mode)?;14011402    let used_items = tcx.arena.alloc_from_iter(used_items);1403    let mentioned_items = tcx.arena.alloc_from_iter(mentioned_items);14041405    Ok((used_items, mentioned_items))1406}14071408/// `item` must be already monomorphized.1409#[instrument(skip(tcx, span, output), level = "debug")]1410fn visit_mentioned_item<'tcx>(1411    tcx: TyCtxt<'tcx>,1412    item: &MentionedItem<'tcx>,1413    span: Span,1414    output: &mut MonoItems<'tcx>,1415) {1416    match *item {1417        MentionedItem::Fn(ty) => {1418            if let ty::FnDef(def_id, args) = *ty.kind() {1419                let args = args.no_bound_vars().unwrap();1420                let instance = Instance::expect_resolve(1421                    tcx,1422                    ty::TypingEnv::fully_monomorphized(),1423                    def_id,1424                    args,1425                    span,1426                );1427                // `visit_instance_use` was written for "used" item collection but works just as well1428                // for "mentioned" item collection.1429                // We can set `is_direct_call`; that just means we'll skip a bunch of shims that anyway1430                // can't have their own failing constants.1431                visit_instance_use(tcx, instance, /*is_direct_call*/ true, span, output);1432            }1433        }1434        MentionedItem::Drop(ty) => {1435            visit_drop_use(tcx, ty, /*is_direct_call*/ true, span, output);1436        }1437        MentionedItem::UnsizeCast { source_ty, target_ty } => {1438            let (source_ty, target_ty) =1439                find_tails_for_unsizing(tcx.at(span), source_ty, target_ty);1440            // This could also be a different Unsize instruction, like1441            // from a fixed sized array to a slice. But we are only1442            // interested in things that produce a vtable.1443            if target_ty.is_trait() && !source_ty.is_trait() {1444                create_mono_items_for_vtable_methods(tcx, target_ty, source_ty, span, output);1445            }1446        }1447        MentionedItem::Closure(source_ty) => {1448            if let ty::Closure(def_id, args) = *source_ty.kind() {1449                let instance =1450                    Instance::resolve_closure(tcx, def_id, args, ty::ClosureKind::FnOnce);1451                if tcx.should_codegen_locally(instance) {1452                    output.push(create_fn_mono_item(tcx, instance, span));1453                }1454            } else {1455                bug!()1456            }1457        }1458    }1459}14601461#[instrument(skip(tcx, output), level = "debug")]1462fn collect_const_value<'tcx>(1463    tcx: TyCtxt<'tcx>,1464    value: mir::ConstValue,1465    output: &mut MonoItems<'tcx>,1466) {1467    match value {1468        mir::ConstValue::Scalar(Scalar::Ptr(ptr, _size)) => {1469            collect_alloc(tcx, ptr.provenance.alloc_id(), output)1470        }1471        mir::ConstValue::Indirect { alloc_id, .. }1472        | mir::ConstValue::Slice { alloc_id, meta: _ } => collect_alloc(tcx, alloc_id, output),1473        _ => {}1474    }1475}14761477//=-----------------------------------------------------------------------------1478// Root Collection1479//=-----------------------------------------------------------------------------14801481// Find all non-generic items by walking the HIR. These items serve as roots to1482// start monomorphizing from.1483#[instrument(skip(tcx, mode), level = "debug")]1484fn collect_roots(tcx: TyCtxt<'_>, mode: MonoItemCollectionStrategy) -> Vec<MonoItem<'_>> {1485    debug!("collecting roots");1486    let mut roots = MonoItems::new();14871488    // Read the manifest and add the recorded kernel instantiations as roots so they are codegened.1489    if let Some(manifest_path) = tcx.sess.opts.unstable_opts.offload.iter().find_map(|o| {1490        if let rustc_session::config::Offload::Device(p) = o1491            && !p.is_empty()1492        {1493            Some(p)1494        } else {1495            None1496        }1497    }) {1498        tcx.sess.file_depinfo.borrow_mut().insert(Symbol::intern(manifest_path));1499        match crate::offload::manifest::read_manifest(std::path::Path::new(manifest_path), tcx) {1500            Ok(instances) => {1501                for instance in instances {1502                    if instance.def_id().is_local() {1503                        roots.push(dummy_spanned(MonoItem::Fn(instance)));1504                    }1505                }1506            }1507            Err(e) => {1508                tcx.dcx().emit_err(crate::diagnostics::OffloadManifestReadError {1509                    path: manifest_path.clone(),1510                    err: e.to_string(),1511                });1512            }1513        }1514    }15151516    {1517        let entry_fn = tcx.entry_fn(());15181519        debug!("collect_roots: entry_fn = {:?}", entry_fn);15201521        let mut collector = RootCollector { tcx, strategy: mode, entry_fn, output: &mut roots };15221523        let crate_items = tcx.hir_crate_items(());15241525        for id in crate_items.free_items() {1526            collector.process_item(id);1527        }15281529        for id in crate_items.impl_items() {1530            collector.process_impl_item(id);1531        }15321533        for id in crate_items.nested_bodies() {1534            collector.process_nested_body(id);1535        }15361537        collector.push_extra_entry_roots();1538    }15391540    let is_host_metadata = tcx1541        .sess1542        .opts1543        .unstable_opts1544        .offload1545        .iter()1546        .any(|o| matches!(o, rustc_session::config::Offload::HostMetadata(_)));1547    if is_host_metadata {1548        let crate_items = tcx.hir_crate_items(());1549        for id in crate_items.free_items() {1550            if !matches!(tcx.def_kind(id.owner_id), DefKind::Fn | DefKind::AssocFn) {1551                continue;1552            }1553            let def_id = id.owner_id.to_def_id();1554            if !tcx.generics_of(def_id).requires_monomorphization(tcx)1555                && tcx.codegen_fn_attrs(def_id).flags.intersects(CodegenFnAttrFlags::OFFLOAD_KERNEL)1556            {1557                roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));1558            }1559        }1560        for id in crate_items.impl_items() {1561            if !matches!(tcx.def_kind(id.owner_id), DefKind::Fn | DefKind::AssocFn) {1562                continue;1563            }1564            let def_id = id.owner_id.to_def_id();1565            if !tcx.generics_of(def_id).requires_monomorphization(tcx)1566                && tcx.codegen_fn_attrs(def_id).flags.intersects(CodegenFnAttrFlags::OFFLOAD_KERNEL)1567            {1568                roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));1569            }1570        }1571        for id in crate_items.trait_items() {1572            if !matches!(tcx.def_kind(id.owner_id), DefKind::Fn | DefKind::AssocFn) {1573                continue;1574            }1575            let def_id = id.owner_id.to_def_id();1576            if !tcx.generics_of(def_id).requires_monomorphization(tcx)1577                && tcx.codegen_fn_attrs(def_id).flags.intersects(CodegenFnAttrFlags::OFFLOAD_KERNEL)1578            {1579                roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));1580            }1581        }1582    }15831584    // We can only codegen items that are instantiable - items all of1585    // whose predicates hold. Luckily, items that aren't instantiable1586    // can't actually be used, so we can just skip codegenning them.1587    roots1588        .into_iter()1589        .filter_map(|Spanned { node: mono_item, .. }| {1590            mono_item.is_instantiable(tcx).then_some(mono_item)1591        })1592        .collect()1593}15941595struct RootCollector<'a, 'tcx> {1596    tcx: TyCtxt<'tcx>,1597    strategy: MonoItemCollectionStrategy,1598    output: &'a mut MonoItems<'tcx>,1599    entry_fn: Option<(DefId, EntryFnType)>,1600}16011602impl<'v> RootCollector<'_, 'v> {1603    fn process_item(&mut self, id: hir::ItemId) {1604        match self.tcx.def_kind(id.owner_id) {1605            DefKind::Enum | DefKind::Struct | DefKind::Union => {1606                if self.strategy == MonoItemCollectionStrategy::Eager1607                    && !self.tcx.generics_of(id.owner_id).requires_monomorphization(self.tcx)1608                {1609                    debug!("RootCollector: ADT drop-glue for `{id:?}`",);1610                    let id_args =1611                        ty::GenericArgs::for_item(self.tcx, id.owner_id.to_def_id(), |param, _| {1612                            match param.kind {1613                                GenericParamDefKind::Lifetime => {1614                                    self.tcx.lifetimes.re_erased.into()1615                                }1616                                GenericParamDefKind::Type { .. }1617                                | GenericParamDefKind::Const { .. } => {1618                                    unreachable!(1619                                        "`own_requires_monomorphization` check means that \1620                                we should have no type/const params"1621                                    )1622                                }1623                            }1624                        });16251626                    // This type is impossible to instantiate, so we should not try to1627                    // generate a `drop_glue` instance for it.1628                    if self.tcx.instantiate_and_check_impossible_clauses((1629                        id.owner_id.to_def_id(),1630                        id_args,1631                    )) {1632                        return;1633                    }16341635                    let ty = self1636                        .tcx1637                        .type_of(id.owner_id.to_def_id())1638                        .instantiate(self.tcx, id_args)1639                        .skip_norm_wip();1640                    assert!(!ty.has_non_region_param());1641                    visit_drop_use(self.tcx, ty, true, DUMMY_SP, self.output);1642                }1643            }1644            DefKind::GlobalAsm => {1645                debug!(1646                    "RootCollector: ItemKind::GlobalAsm({})",1647                    self.tcx.def_path_str(id.owner_id)1648                );1649                self.output.push(dummy_spanned(MonoItem::GlobalAsm(id)));1650            }1651            DefKind::Static { .. } => {1652                let def_id = id.owner_id.to_def_id();1653                debug!("RootCollector: ItemKind::Static({})", self.tcx.def_path_str(def_id));1654                self.output.push(dummy_spanned(MonoItem::Static(def_id)));1655            }1656            DefKind::Const { .. } => {1657                // Const items only generate mono items if they are actually used somewhere.1658                // Just declaring them is insufficient.16591660                // If we're collecting items eagerly, then recurse into all constants.1661                // Otherwise the value is only collected when explicitly mentioned in other items.1662                if self.strategy == MonoItemCollectionStrategy::Eager {1663                    let def_id = id.owner_id.to_def_id();1664                    // Type Consts don't have bodies to evaluate1665                    // nor do they make sense as a static.1666                    if self.tcx.is_type_const(def_id) {1667                        // FIXME(mgca): Is this actually what we want? We may want to1668                        // normalize to a ValTree then convert to a const allocation and1669                        // collect that?1670                        return;1671                    }1672                    if self.tcx.generics_of(id.owner_id).own_requires_monomorphization() {1673                        return;1674                    }1675                    let Ok(val) = self.tcx.const_eval_poly(def_id) else {1676                        return;1677                    };1678                    collect_const_value(self.tcx, val, self.output);1679                }1680            }1681            DefKind::Impl { of_trait: true } => {1682                if self.strategy == MonoItemCollectionStrategy::Eager {1683                    create_mono_items_for_default_impls(self.tcx, id, self.output);1684                }1685            }1686            DefKind::Fn => {1687                self.push_if_root(id.owner_id.def_id);1688            }1689            _ => {}1690        }1691    }16921693    fn process_impl_item(&mut self, id: hir::ImplItemId) {1694        if self.tcx.def_kind(id.owner_id) == DefKind::AssocFn {1695            self.push_if_root(id.owner_id.def_id);1696        }1697    }16981699    fn process_nested_body(&mut self, def_id: LocalDefId) {1700        match self.tcx.def_kind(def_id) {1701            DefKind::Closure => {1702                // for 'pub async fn foo(..)' also trying to monomorphize foo::{closure}1703                let is_pub_fn_coroutine =1704                    match *self.tcx.type_of(def_id).instantiate_identity().skip_norm_wip().kind() {1705                        ty::Coroutine(cor_id, _args) => {1706                            let tcx = self.tcx;1707                            let parent_id = tcx.parent(cor_id);1708                            tcx.def_kind(parent_id) == DefKind::Fn1709                                && tcx.asyncness(parent_id).is_async()1710                                && tcx.visibility(parent_id).is_public()1711                        }1712                        ty::Closure(..) | ty::CoroutineClosure(..) => false,1713                        _ => unreachable!(),1714                    };1715                if (self.strategy == MonoItemCollectionStrategy::Eager || is_pub_fn_coroutine)1716                    && !self1717                        .tcx1718                        .generics_of(self.tcx.typeck_root_def_id_local(def_id))1719                        .requires_monomorphization(self.tcx)1720                {1721                    let instance = match *self1722                        .tcx1723                        .type_of(def_id)1724                        .instantiate_identity()1725                        .skip_norm_wip()1726                        .kind()1727                    {1728                        ty::Closure(def_id, args)1729                        | ty::Coroutine(def_id, args)1730                        | ty::CoroutineClosure(def_id, args) => {1731                            Instance::new_raw(def_id, self.tcx.erase_and_anonymize_regions(args))1732                        }1733                        _ => unreachable!(),1734                    };1735                    let Ok(instance) = self.tcx.try_normalize_erasing_regions(1736                        ty::TypingEnv::fully_monomorphized(),1737                        Unnormalized::new_wip(instance),1738                    ) else {1739                        // Don't ICE on an impossible-to-normalize closure.1740                        return;1741                    };1742                    let mono_item = create_fn_mono_item(self.tcx, instance, DUMMY_SP);1743                    if mono_item.node.is_instantiable(self.tcx) {1744                        self.output.push(mono_item);1745                    }1746                }1747            }1748            _ => {}1749        }1750    }17511752    fn is_root(&self, def_id: LocalDefId) -> bool {1753        !self.tcx.generics_of(def_id).requires_monomorphization(self.tcx)1754            && match self.strategy {1755                MonoItemCollectionStrategy::Eager => {1756                    !matches!(self.tcx.codegen_fn_attrs(def_id).inline, InlineAttr::Force { .. })1757                    // comptime fns can't be codegenned, so we need to prevent collecting them even1758                    // with link-dead-code. Lazy mode prevents them by them not showing up in1759                    // `is_reachable_non_generic` (and `entry_fn` can't be comptime).1760                    && match self.tcx.def_kind(def_id) {1761                        DefKind::Fn | DefKind::AssocFn => {1762                            self.tcx.constness(def_id) != hir::Constness::Const { always: true }1763                        }1764                        _ => true,1765                    }1766                }1767                MonoItemCollectionStrategy::Lazy => {1768                    self.entry_fn.and_then(|(id, _)| id.as_local()) == Some(def_id)1769                        || self.tcx.is_reachable_non_generic(def_id)1770                        || {1771                            let flags = self.tcx.codegen_fn_attrs(def_id).flags;1772                            flags.intersects(1773                                CodegenFnAttrFlags::RUSTC_STD_INTERNAL_SYMBOL1774                                    | CodegenFnAttrFlags::EXTERNALLY_IMPLEMENTABLE_ITEM,1775                            )1776                        }1777                }1778            }1779    }17801781    /// If `def_id` represents a root, pushes it onto the list of1782    /// outputs. (Note that all roots must be monomorphic.)1783    #[instrument(skip(self), level = "debug")]1784    fn push_if_root(&mut self, def_id: LocalDefId) {1785        if self.is_root(def_id) {1786            debug!("found root");17871788            let instance = Instance::mono(self.tcx, def_id.to_def_id());1789            self.output.push(create_fn_mono_item(self.tcx, instance, DUMMY_SP));1790        }1791    }17921793    /// As a special case, when/if we encounter the1794    /// `main()` function, we also have to generate a1795    /// monomorphized copy of the start lang item based on1796    /// the return type of `main`. This is not needed when1797    /// the user writes their own `start` manually.1798    fn push_extra_entry_roots(&mut self) {1799        let Some((main_def_id, EntryFnType::Main { .. })) = self.entry_fn else {1800            return;1801        };18021803        let main_instance = Instance::mono(self.tcx, main_def_id);1804        if self.tcx.should_codegen_locally(main_instance) {1805            self.output.push(create_fn_mono_item(1806                self.tcx,1807                main_instance,1808                self.tcx.def_span(main_def_id),1809            ));1810        }18111812        let Some(start_def_id) = self.tcx.lang_items().start_fn() else {1813            self.tcx.dcx().emit_fatal(diagnostics::StartNotFound);1814        };1815        let main_ret_ty = self.tcx.fn_sig(main_def_id).no_bound_vars().unwrap().output();18161817        // Given that `main()` has no arguments,1818        // then its return type cannot have1819        // late-bound regions, since late-bound1820        // regions must appear in the argument1821        // listing.1822        let main_ret_ty = self.tcx.normalize_erasing_regions(1823            ty::TypingEnv::fully_monomorphized(),1824            Unnormalized::new_wip(main_ret_ty.no_bound_vars().unwrap()),1825        );18261827        let start_instance = Instance::expect_resolve(1828            self.tcx,1829            ty::TypingEnv::fully_monomorphized(),1830            start_def_id,1831            self.tcx.mk_args(&[main_ret_ty.into()]),1832            DUMMY_SP,1833        );18341835        self.output.push(create_fn_mono_item(self.tcx, start_instance, DUMMY_SP));1836    }1837}18381839#[instrument(level = "debug", skip(tcx, output))]1840fn create_mono_items_for_default_impls<'tcx>(1841    tcx: TyCtxt<'tcx>,1842    item: hir::ItemId,1843    output: &mut MonoItems<'tcx>,1844) {1845    let impl_ = tcx.impl_trait_header(item.owner_id);18461847    if impl_.polarity == ty::ImplPolarity::Negative {1848        return;1849    }18501851    if tcx.generics_of(item.owner_id).own_requires_monomorphization() {1852        return;1853    }18541855    // Lifetimes never affect trait selection, so we are allowed to eagerly1856    // instantiate an instance of an impl method if the impl (and method,1857    // which we check below) is only parameterized over lifetime. In that case,1858    // we use the ReErased, which has no lifetime information associated with1859    // it, to validate whether or not the impl is legal to instantiate at all.1860    let only_region_params = |param: &ty::GenericParamDef, _: &_| match param.kind {1861        GenericParamDefKind::Lifetime => tcx.lifetimes.re_erased.into(),1862        GenericParamDefKind::Type { .. } | GenericParamDefKind::Const { .. } => {1863            unreachable!(1864                "`own_requires_monomorphization` check means that \1865                we should have no type/const params"1866            )1867        }1868    };1869    let impl_args = GenericArgs::for_item(tcx, item.owner_id.to_def_id(), only_region_params);1870    let trait_ref = impl_.trait_ref.instantiate(tcx, impl_args).skip_norm_wip();18711872    // Unlike 'lazy' monomorphization that begins by collecting items transitively1873    // called by `main` or other global items, when eagerly monomorphizing impl1874    // items, we never actually check that the predicates of this impl are satisfied1875    // in a empty param env (i.e. with no assumptions).1876    //1877    // Even though this impl has no type or const generic parameters, because we don't1878    // consider higher-ranked predicates such as `for<'a> &'a mut [u8]: Copy` to1879    // be trivially false. We must now check that the impl has no impossible-to-satisfy1880    // clauses.1881    if tcx.instantiate_and_check_impossible_clauses((item.owner_id.to_def_id(), impl_args)) {1882        return;1883    }18841885    let typing_env = ty::TypingEnv::fully_monomorphized();1886    let trait_ref = tcx.normalize_erasing_regions(typing_env, Unnormalized::new_wip(trait_ref));1887    let overridden_methods = tcx.impl_item_implementor_ids(item.owner_id);1888    for method in tcx.provided_trait_methods(trait_ref.def_id) {1889        if overridden_methods.contains_key(&method.def_id) {1890            continue;1891        }18921893        if tcx.generics_of(method.def_id).own_requires_monomorphization() {1894            continue;1895        }18961897        // As mentioned above, the method is legal to eagerly instantiate if it1898        // only has lifetime generic parameters. This is validated by calling1899        // `own_requires_monomorphization` on both the impl and method.1900        let args = trait_ref.args.extend_to(tcx, method.def_id, only_region_params);1901        let instance = ty::Instance::expect_resolve(tcx, typing_env, method.def_id, args, DUMMY_SP);19021903        let mono_item = create_fn_mono_item(tcx, instance, DUMMY_SP);1904        if mono_item.node.is_instantiable(tcx) && tcx.should_codegen_locally(instance) {1905            output.push(mono_item);1906        }1907    }1908}19091910//=-----------------------------------------------------------------------------1911// Top-level entry point, tying it all together1912//=-----------------------------------------------------------------------------19131914#[instrument(skip(tcx, strategy), level = "debug")]1915pub(crate) fn collect_crate_mono_items<'tcx>(1916    tcx: TyCtxt<'tcx>,1917    strategy: MonoItemCollectionStrategy,1918) -> (Vec<MonoItem<'tcx>>, UsageMap<'tcx>) {1919    let _prof_timer = tcx.prof.generic_activity("monomorphization_collector");19201921    let roots = tcx1922        .sess1923        .time("monomorphization_collector_root_collections", || collect_roots(tcx, strategy));19241925    debug!("building mono item graph, beginning at roots");19261927    let state = SharedState {1928        visited: Lock::new(UnordSet::default()),1929        mentioned: Lock::new(UnordSet::default()),1930        usage_map: Lock::new(UsageMap::new()),1931    };1932    let recursion_limit = tcx.recursion_limit();19331934    tcx.sess.time("monomorphization_collector_graph_walk", || {1935        par_for_each_in(roots, |root| {1936            collect_items_root(tcx, dummy_spanned(*root), &state, recursion_limit);1937        });1938    });19391940    // The set of MonoItems was created in an inherently indeterministic order because1941    // of parallelism. We sort it here to ensure that the output is deterministic.1942    let mono_items = tcx.with_stable_hashing_context(move |mut hcx| {1943        state.visited.into_inner().into_sorted(&mut hcx, true)1944    });19451946    if tcx1947        .sess1948        .opts1949        .unstable_opts1950        .offload1951        .iter()1952        .any(|o| matches!(o, Offload::Device(p) if p.is_empty()))1953    {1954        crate::offload::check_offload_kernels_instantiated(tcx, &mono_items);1955    }19561957    (mono_items, state.usage_map.into_inner())1958}19591960pub(crate) fn provide(providers: &mut Providers) {1961    providers.hooks.should_codegen_locally = should_codegen_locally;1962    providers.queries.items_of_instance = items_of_instance;1963}

Code quality findings 29

Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
//! do not use a failing constant. This is reflected via the [`CollectionMode`], which determines
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
let used_items = self.used_map.get(&item).unwrap();
Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
/// `mode` determined whether we are scanning for [used items][CollectionMode::UsedItems]
Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
/// or [mentioned items][CollectionMode::MentionedItems].
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
args.no_bound_vars().unwrap(),
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
let args = args.no_bound_vars().unwrap();
Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
let coerce_field = &source_adt_def.non_enum_variant().fields[coerce_index];
Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
(&'tcx [Spanned<MonoItem<'tcx>>], &'tcx [Spanned<MonoItem<'tcx>>]),
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
let args = args.no_bound_vars().unwrap();
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
let main_ret_ty = self.tcx.fn_sig(main_def_id).no_bound_vars().unwrap().output();
Warning: '.unwrap()' will panic on None/Err variants. Prefer using pattern matching (match, if let), combinators (map, and_then), or the '?' operator for robust error handling.
warning correctness unwrap-usage
Unnormalized::new_wip(main_ret_ty.no_bound_vars().unwrap()),
Warning: Direct indexing (e.g., `vec[i]`, `slice[i]`) panics on out-of-bounds access. Prefer using `.get(index)` or `.get_mut(index)` which return Option<&T>/Option<&mut T>.
warning correctness unchecked-indexing
// consider higher-ranked predicates such as `for<'a> &'a mut [u8]: Copy` to
Info: Direct printing to stdout/stderr. For application logging, prefer using a logging facade like `log` or `tracing` for better control over levels, formatting, and output destinations.
info maintainability println-macro
//! println!("{}", x);
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
self.user_map.entry(used_item).or_default().push(user_item);
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
self.push(item)
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
used_items.push(respan(
Info: Ensure 'match' statements are exhaustive. If matching on enums, consider adding a wildcard arm `_ => {}` only if necessary and intentional, as it suppresses warnings about unhandled variants.
info correctness match-wildcard
match ty::Instance::resolve_for_fn_ptr(
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
output.push(dummy_spanned(MonoItem::Static(def_id)));
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
output.push(create_fn_mono_item(tcx, instance, DUMMY_SP));
Info: Ensure 'match' statements are exhaustive. If matching on enums, consider adding a wildcard arm `_ => {}` only if necessary and intentional, as it suppresses warnings about unhandled variants.
info correctness match-wildcard
match value {
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
roots.push(dummy_spanned(MonoItem::Fn(instance)));
Performance Info: Frequent cloning, especially of Strings, Vecs, or other heap-allocated types inside loops, can be expensive. Consider using references/borrowing where possible.
info performance clone-in-loop
path: manifest_path.clone(),
Performance Info: Calling .to_string() (especially on &str) allocates a new String. If done repeatedly in loops, consider alternatives like working with &str or using crates like `itoa`/`ryu` for number-to-string conversion.
info performance to-string-in-loop
err: e.to_string(),
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));
Performance Info: Calling .push() repeatedly inside a loop without prior capacity reservation can lead to multiple reallocations. Consider using `Vec::with_capacity(n)` or `vec.reserve(n)` if the approximate number of elements is known.
info performance push-without-reserve
roots.push(dummy_spanned(MonoItem::Fn(Instance::mono(tcx, def_id))));
Info: Ensure 'match' statements are exhaustive. If matching on enums, consider adding a wildcard arm `_ => {}` only if necessary and intentional, as it suppresses warnings about unhandled variants.
info correctness match-wildcard
match *self.tcx.type_of(def_id).instantiate_identity().skip_norm_wip().kind() {
Info: Ensure 'match' statements are exhaustive. If matching on enums, consider adding a wildcard arm `_ => {}` only if necessary and intentional, as it suppresses warnings about unhandled variants.
info correctness match-wildcard
&& match self.strategy {
Info: Ensure 'match' statements are exhaustive. If matching on enums, consider adding a wildcard arm `_ => {}` only if necessary and intentional, as it suppresses warnings about unhandled variants.
info correctness match-wildcard
&& match self.tcx.def_kind(def_id) {

Get this view in your editor

Same data, no extra tab — call code_get_file + code_get_findings over MCP from Claude/Cursor/Copilot.