rustc_middle/middle/
exported_symbols.rs1use rustc_hir::def_id::{DefId, LOCAL_CRATE};
2use rustc_macros::{Decodable, Encodable, HashStable, TyDecodable, TyEncodable};
3
4use crate::ty::{self, GenericArgsRef, Ty, TyCtxt};
5
6#[derive(Eq, PartialEq, Debug, Copy, Clone, TyEncodable, TyDecodable, HashStable)]
12pub enum SymbolExportLevel {
13 C,
14 Rust,
15}
16
17impl SymbolExportLevel {
18 pub fn is_below_threshold(self, threshold: SymbolExportLevel) -> bool {
19 threshold == SymbolExportLevel::Rust || self == SymbolExportLevel::C
21 }
22}
23
24#[derive(Eq, PartialEq, Debug, Copy, Clone, Encodable, Decodable, HashStable, Hash)]
26pub enum SymbolExportKind {
27 Text,
28 Data,
29 Tls,
30}
31
32#[derive(Eq, PartialEq, Debug, Copy, Clone, TyEncodable, TyDecodable, HashStable)]
35pub struct SymbolExportInfo {
36 pub level: SymbolExportLevel,
37 pub kind: SymbolExportKind,
38 pub used: bool,
39}
40
41#[derive(Eq, PartialEq, Debug, Copy, Clone, TyEncodable, TyDecodable, HashStable)]
42pub enum ExportedSymbol<'tcx> {
43 NonGeneric(DefId),
44 Generic(DefId, GenericArgsRef<'tcx>),
45 DropGlue(Ty<'tcx>),
46 AsyncDropGlueCtorShim(Ty<'tcx>),
47 AsyncDropGlue(DefId, Ty<'tcx>),
48 ThreadLocalShim(DefId),
49 NoDefId(ty::SymbolName<'tcx>),
50}
51
52impl<'tcx> ExportedSymbol<'tcx> {
53 pub fn symbol_name_for_local_instance(&self, tcx: TyCtxt<'tcx>) -> ty::SymbolName<'tcx> {
56 match *self {
57 ExportedSymbol::NonGeneric(def_id) => tcx.symbol_name(ty::Instance::mono(tcx, def_id)),
58 ExportedSymbol::Generic(def_id, args) => {
59 tcx.symbol_name(ty::Instance::new_raw(def_id, args))
60 }
61 ExportedSymbol::DropGlue(ty) => {
62 tcx.symbol_name(ty::Instance::resolve_drop_in_place(tcx, ty))
63 }
64 ExportedSymbol::AsyncDropGlueCtorShim(ty) => {
65 tcx.symbol_name(ty::Instance::resolve_async_drop_in_place(tcx, ty))
66 }
67 ExportedSymbol::AsyncDropGlue(def_id, ty) => {
68 tcx.symbol_name(ty::Instance::resolve_async_drop_in_place_poll(tcx, def_id, ty))
69 }
70 ExportedSymbol::ThreadLocalShim(def_id) => tcx.symbol_name(ty::Instance {
71 def: ty::InstanceKind::ThreadLocalShim(def_id),
72 args: ty::GenericArgs::empty(),
73 }),
74 ExportedSymbol::NoDefId(symbol_name) => symbol_name,
75 }
76 }
77}
78
79pub fn metadata_symbol_name(tcx: TyCtxt<'_>) -> String {
80 format!(
81 "rust_metadata_{}_{:08x}",
82 tcx.crate_name(LOCAL_CRATE),
83 tcx.stable_crate_id(LOCAL_CRATE),
84 )
85}