]>
Commit | Line | Data |
---|---|---|
60c5eb7d | 1 | use crate::abi::FnAbi; |
9fa01778 XL |
2 | use crate::attributes; |
3 | use crate::base; | |
4 | use crate::context::CodegenCx; | |
5 | use crate::llvm; | |
9fa01778 | 6 | use crate::type_of::LayoutLlvmExt; |
dfeec247 | 7 | use log::debug; |
a1dfa0c6 | 8 | use rustc_codegen_ssa::traits::*; |
dfeec247 | 9 | use rustc_hir::def_id::{DefId, LOCAL_CRATE}; |
ba9703b0 XL |
10 | pub use rustc_middle::mir::mono::MonoItem; |
11 | use rustc_middle::mir::mono::{Linkage, Visibility}; | |
12 | use rustc_middle::ty::layout::FnAbiExt; | |
13 | use rustc_middle::ty::{Instance, TypeFoldable}; | |
14 | use rustc_target::abi::LayoutOf; | |
a7813a04 | 15 | |
a1dfa0c6 | 16 | impl PreDefineMethods<'tcx> for CodegenCx<'ll, 'tcx> { |
dfeec247 XL |
17 | fn predefine_static( |
18 | &self, | |
19 | def_id: DefId, | |
20 | linkage: Linkage, | |
21 | visibility: Visibility, | |
22 | symbol_name: &str, | |
23 | ) { | |
a1dfa0c6 | 24 | let instance = Instance::mono(self.tcx, def_id); |
dfeec247 | 25 | let ty = instance.monomorphic_ty(self.tcx); |
a1dfa0c6 XL |
26 | let llty = self.layout_of(ty).llvm_type(self); |
27 | ||
28 | let g = self.define_global(symbol_name, llty).unwrap_or_else(|| { | |
dfeec247 XL |
29 | self.sess().span_fatal( |
30 | self.tcx.def_span(def_id), | |
31 | &format!("symbol `{}` is already defined", symbol_name), | |
32 | ) | |
a1dfa0c6 | 33 | }); |
5bcae85e | 34 | |
a1dfa0c6 XL |
35 | unsafe { |
36 | llvm::LLVMRustSetLinkage(g, base::linkage_to_llvm(linkage)); | |
37 | llvm::LLVMRustSetVisibility(g, base::visibility_to_llvm(visibility)); | |
5bcae85e SL |
38 | } |
39 | ||
a1dfa0c6 | 40 | self.instances.borrow_mut().insert(instance, g); |
5bcae85e SL |
41 | } |
42 | ||
dfeec247 XL |
43 | fn predefine_fn( |
44 | &self, | |
45 | instance: Instance<'tcx>, | |
46 | linkage: Linkage, | |
47 | visibility: Visibility, | |
48 | symbol_name: &str, | |
49 | ) { | |
ba9703b0 | 50 | assert!(!instance.substs.needs_infer() && !instance.substs.has_param_types_or_consts()); |
a1dfa0c6 | 51 | |
60c5eb7d XL |
52 | let fn_abi = FnAbi::of_instance(self, instance, &[]); |
53 | let lldecl = self.declare_fn(symbol_name, &fn_abi); | |
a1dfa0c6 | 54 | unsafe { llvm::LLVMRustSetLinkage(lldecl, base::linkage_to_llvm(linkage)) }; |
60c5eb7d | 55 | let attrs = self.tcx.codegen_fn_attrs(instance.def_id()); |
a1dfa0c6 | 56 | base::set_link_section(lldecl, &attrs); |
dfeec247 | 57 | if linkage == Linkage::LinkOnceODR || linkage == Linkage::WeakODR { |
a1dfa0c6 | 58 | llvm::SetUniqueComdat(self.llmod, lldecl); |
5bcae85e SL |
59 | } |
60 | ||
0731742a | 61 | // If we're compiling the compiler-builtins crate, e.g., the equivalent of |
a1dfa0c6 XL |
62 | // compiler-rt, then we want to implicitly compile everything with hidden |
63 | // visibility as we're going to link this object all over the place but | |
64 | // don't want the symbols to get exported. | |
dfeec247 XL |
65 | if linkage != Linkage::Internal |
66 | && linkage != Linkage::Private | |
67 | && self.tcx.is_compiler_builtins(LOCAL_CRATE) | |
68 | { | |
a1dfa0c6 XL |
69 | unsafe { |
70 | llvm::LLVMRustSetVisibility(lldecl, llvm::Visibility::Hidden); | |
5bcae85e | 71 | } |
a1dfa0c6 XL |
72 | } else { |
73 | unsafe { | |
74 | llvm::LLVMRustSetVisibility(lldecl, base::visibility_to_llvm(visibility)); | |
cc61c64b | 75 | } |
5bcae85e | 76 | } |
5bcae85e | 77 | |
60c5eb7d XL |
78 | debug!("predefine_fn: instance = {:?}", instance); |
79 | ||
ba9703b0 | 80 | attributes::from_fn_attrs(self, lldecl, instance); |
ea8adc8c | 81 | |
a1dfa0c6 | 82 | self.instances.borrow_mut().insert(instance, lldecl); |
ea8adc8c | 83 | } |
ea8adc8c | 84 | } |