rustc_codegen_ssa/
mono_item.rs

1use rustc_hir::attrs::Linkage;
2use rustc_middle::middle::codegen_fn_attrs::CodegenFnAttrFlags;
3use rustc_middle::mir::mono::{MonoItem, MonoItemData, Visibility};
4use rustc_middle::ty::layout::HasTyCtxt;
5use tracing::debug;
6
7use crate::base;
8use crate::mir::naked_asm;
9use crate::traits::*;
10
11pub trait MonoItemExt<'a, 'tcx> {
12    fn define<Bx: BuilderMethods<'a, 'tcx>>(
13        &self,
14        cx: &'a mut Bx::CodegenCx,
15        cgu_name: &str,
16        item_data: MonoItemData,
17    );
18    fn predefine<Bx: BuilderMethods<'a, 'tcx>>(
19        &self,
20        cx: &'a mut Bx::CodegenCx,
21        cgu_name: &str,
22        linkage: Linkage,
23        visibility: Visibility,
24    );
25    fn to_raw_string(&self) -> String;
26}
27
28impl<'a, 'tcx: 'a> MonoItemExt<'a, 'tcx> for MonoItem<'tcx> {
29    fn define<Bx: BuilderMethods<'a, 'tcx>>(
30        &self,
31        cx: &'a mut Bx::CodegenCx,
32        cgu_name: &str,
33        item_data: MonoItemData,
34    ) {
35        debug!("BEGIN IMPLEMENTING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
36
37        match *self {
38            MonoItem::Static(def_id) => {
39                cx.codegen_static(def_id);
40            }
41            MonoItem::GlobalAsm(item_id) => {
42                base::codegen_global_asm(cx, item_id);
43            }
44            MonoItem::Fn(instance) => {
45                let flags = cx.tcx().codegen_instance_attrs(instance.def).flags;
46                if flags.contains(CodegenFnAttrFlags::NAKED) {
47                    naked_asm::codegen_naked_asm::<Bx::CodegenCx>(cx, instance, item_data);
48                } else {
49                    base::codegen_instance::<Bx>(cx, instance);
50                }
51            }
52        }
53
54        debug!("END IMPLEMENTING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
55    }
56
57    fn predefine<Bx: BuilderMethods<'a, 'tcx>>(
58        &self,
59        cx: &'a mut Bx::CodegenCx,
60        cgu_name: &str,
61        linkage: Linkage,
62        visibility: Visibility,
63    ) {
64        debug!("BEGIN PREDEFINING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
65
66        let symbol_name = self.symbol_name(cx.tcx()).name;
67
68        debug!("symbol {symbol_name}");
69
70        match *self {
71            MonoItem::Static(def_id) => {
72                cx.predefine_static(def_id, linkage, visibility, symbol_name);
73            }
74            MonoItem::Fn(instance) => {
75                let attrs = cx.tcx().codegen_instance_attrs(instance.def);
76
77                if attrs.flags.contains(CodegenFnAttrFlags::NAKED) {
78                    // do not define this function; it will become a global assembly block
79                } else {
80                    cx.predefine_fn(instance, linkage, visibility, symbol_name);
81                };
82            }
83            MonoItem::GlobalAsm(..) => {}
84        }
85
86        debug!("END PREDEFINING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
87    }
88
89    fn to_raw_string(&self) -> String {
90        match *self {
91            MonoItem::Fn(instance) => {
92                format!("Fn({:?}, {})", instance.def, instance.args.as_ptr().addr())
93            }
94            MonoItem::Static(id) => format!("Static({id:?})"),
95            MonoItem::GlobalAsm(id) => format!("GlobalAsm({id:?})"),
96        }
97    }
98}
rustc_codegen_ssa/mono_item.rs

rustc_codegen_ssa/
mono_item.rs
