rustc_codegen_ssa/
mono_item.rs

1use 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                let flags = cx.tcx().codegen_instance_attrs(instance.def).flags;
45                if flags.contains(CodegenFnAttrFlags::NAKED) {
46                    naked_asm::codegen_naked_asm::<Bx::CodegenCx>(cx, instance, item_data);
47                } else {
48                    base::codegen_instance::<Bx>(cx, instance);
49                }
50            }
51        }
52
53        debug!("END IMPLEMENTING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
54    }
55
56    fn predefine<Bx: BuilderMethods<'a, 'tcx>>(
57        &self,
58        cx: &'a mut Bx::CodegenCx,
59        cgu_name: &str,
60        linkage: Linkage,
61        visibility: Visibility,
62    ) {
63        debug!("BEGIN PREDEFINING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
64
65        let symbol_name = self.symbol_name(cx.tcx()).name;
66
67        debug!("symbol {symbol_name}");
68
69        match *self {
70            MonoItem::Static(def_id) => {
71                cx.predefine_static(def_id, linkage, visibility, symbol_name);
72            }
73            MonoItem::Fn(instance) => {
74                let attrs = cx.tcx().codegen_instance_attrs(instance.def);
75
76                if attrs.flags.contains(CodegenFnAttrFlags::NAKED) {
77                    // do not define this function; it will become a global assembly block
78                } else {
79                    cx.predefine_fn(instance, linkage, visibility, symbol_name);
80                };
81            }
82            MonoItem::GlobalAsm(..) => {}
83        }
84
85        debug!("END PREDEFINING '{} ({})' in cgu {}", self, self.to_raw_string(), cgu_name);
86    }
87
88    fn to_raw_string(&self) -> String {
89        match *self {
90            MonoItem::Fn(instance) => {
91                format!("Fn({:?}, {})", instance.def, instance.args.as_ptr().addr())
92            }
93            MonoItem::Static(id) => format!("Static({id:?})"),
94            MonoItem::GlobalAsm(id) => format!("GlobalAsm({id:?})"),
95        }
96    }
97}
rustc_codegen_ssa/mono_item.rs

rustc_codegen_ssa/
mono_item.rs
