compiler/rustc_metadata/src/link_args.rs

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