src/librustc_hir/weak_lang_items.rs

   1 //! Validity checking for weak lang items
   2
   3 use crate::def_id::DefId;
   4 use crate::{lang_items, LangItem, LanguageItems};
   5
   6 use rustc_ast as ast;
   7 use rustc_data_structures::fx::FxHashMap;
   8 use rustc_span::symbol::{sym, Symbol};
   9
  10 use lazy_static::lazy_static;
  11
  12 macro_rules! weak_lang_items {
  13     ($($name:ident, $item:ident, $sym:ident;)*) => (
  14
  15 lazy_static! {
  16     pub static ref WEAK_ITEMS_REFS: FxHashMap<Symbol, LangItem> = {
  17         let mut map = FxHashMap::default();
  18         $(map.insert(sym::$name, LangItem::$item);)*
  19         map
  20     };
  21 }
  22
  23 /// The `check_name` argument avoids the need for `librustc_hir` to depend on
  24 /// `librustc_session`.
  25 pub fn link_name<'a, F>(check_name: F, attrs: &'a [ast::Attribute]) -> Option<Symbol>
  26 where
  27     F: Fn(&'a ast::Attribute, Symbol) -> bool
  28 {
  29     lang_items::extract(check_name, attrs).and_then(|(name, _)| {
  30         $(if name == sym::$name {
  31             Some(sym::$sym)
  32         } else)* {
  33             None
  34         }
  35     })
  36 }
  37
  38 impl LanguageItems {
  39     pub fn is_weak_lang_item(&self, item_def_id: DefId) -> bool {
  40         let did = Some(item_def_id);
  41
  42         $(self.$name() == did)||*
  43     }
  44 }
  45
  46 ) }
  47
  48 weak_lang_items! {
  49     panic_impl,         PanicImpl,          rust_begin_unwind;
  50     eh_personality,     EhPersonality,      rust_eh_personality;
  51     oom,                Oom,                rust_oom;
  52 }