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