src/librustc_metadata/link_args.rs

   1 use rustc::hir;
   2 use rustc::hir::itemlikevisit::ItemLikeVisitor;
   3 use rustc::ty::TyCtxt;
   4 use rustc_span::symbol::sym;
   5 use rustc_target::spec::abi::Abi;
   6
   7 crate fn collect(tcx: TyCtxt<'_>) -> Vec<String> {
   8     let mut collector = Collector { args: Vec::new() };
   9     tcx.hir().krate().visit_all_item_likes(&mut collector);
  10
  11     for attr in tcx.hir().krate().attrs.iter() {
  12         if attr.has_name(sym::link_args) {
  13             if let Some(linkarg) = attr.value_str() {
  14                 collector.add_link_args(&linkarg.as_str());
  15             }
  16         }
  17     }
  18
  19     return collector.args;
  20 }
  21
  22 struct Collector {
  23     args: Vec<String>,
  24 }
  25
  26 impl<'tcx> ItemLikeVisitor<'tcx> for Collector {
  27     fn visit_item(&mut self, it: &'tcx hir::Item<'tcx>) {
  28         let fm = match it.kind {
  29             hir::ItemKind::ForeignMod(ref fm) => fm,
  30             _ => return,
  31         };
  32         if fm.abi == Abi::Rust || fm.abi == Abi::RustIntrinsic || fm.abi == Abi::PlatformIntrinsic {
  33             return;
  34         }
  35
  36         // First, add all of the custom #[link_args] attributes
  37         for m in it.attrs.iter().filter(|a| a.check_name(sym::link_args)) {
  38             if let Some(linkarg) = m.value_str() {
  39                 self.add_link_args(&linkarg.as_str());
  40             }
  41         }
  42     }
  43
  44     fn visit_trait_item(&mut self, _it: &'tcx hir::TraitItem<'tcx>) {}
  45     fn visit_impl_item(&mut self, _it: &'tcx hir::ImplItem<'tcx>) {}
  46 }
  47
  48 impl Collector {
  49     fn add_link_args(&mut self, args: &str) {
  50         self.args.extend(args.split(' ').filter(|s| !s.is_empty()).map(|s| s.to_string()))
  51     }
  52 }