]> git.lizzy.rs Git - rust.git/blob - src/linkage.rs
Add custom driver
[rust.git] / src / linkage.rs
1 use rustc_middle::mir::mono::{Linkage as RLinkage, MonoItem, Visibility};
2
3 use crate::prelude::*;
4
5 pub(crate) fn get_clif_linkage(
6     mono_item: MonoItem<'_>,
7     linkage: RLinkage,
8     visibility: Visibility,
9 ) -> Linkage {
10     match (linkage, visibility) {
11         (RLinkage::External, Visibility::Default) => Linkage::Export,
12         (RLinkage::Internal, Visibility::Default) => Linkage::Local,
13         (RLinkage::External, Visibility::Hidden) => Linkage::Hidden,
14         _ => panic!("{:?} = {:?} {:?}", mono_item, linkage, visibility),
15     }
16 }
17
18 pub(crate) fn get_static_linkage(tcx: TyCtxt<'_>, def_id: DefId) -> Linkage {
19     let fn_attrs = tcx.codegen_fn_attrs(def_id);
20
21     if let Some(linkage) = fn_attrs.linkage {
22         match linkage {
23             RLinkage::External => Linkage::Export,
24             RLinkage::Internal => Linkage::Local,
25             RLinkage::ExternalWeak | RLinkage::WeakAny => Linkage::Preemptible,
26             _ => panic!("{:?}", linkage),
27         }
28     } else {
29         if tcx.is_reachable_non_generic(def_id) {
30             Linkage::Export
31         } else {
32             Linkage::Hidden
33         }
34     }
35 }