compiler/rustc_mir_dataflow/src/lib.rs

   1 #![feature(associated_type_defaults)]
   2 #![feature(box_patterns)]
   3 #![feature(exact_size_is_empty)]
   4 #![feature(let_else)]
   5 #![feature(min_specialization)]
   6 #![feature(once_cell)]
   7 #![feature(stmt_expr_attributes)]
   8 #![feature(trusted_step)]
   9 #![recursion_limit = "256"]
  10
  11 #[macro_use]
  12 extern crate tracing;
  13 #[macro_use]
  14 extern crate rustc_middle;
  15
  16 use rustc_ast::MetaItem;
  17 use rustc_hir::def_id::DefId;
  18 use rustc_middle::ty::{self, TyCtxt};
  19 use rustc_span::symbol::{sym, Symbol};
  20
  21 pub use self::drop_flag_effects::{
  22     drop_flag_effects_for_function_entry, drop_flag_effects_for_location,
  23     move_path_children_matching, on_all_children_bits, on_all_drop_children_bits,
  24     on_lookup_result_bits,
  25 };
  26 pub use self::framework::{
  27     fmt, graphviz, lattice, visit_results, Analysis, AnalysisDomain, Backward, CallReturnPlaces,
  28     Direction, Engine, Forward, GenKill, GenKillAnalysis, JoinSemiLattice, Results, ResultsCursor,
  29     ResultsRefCursor, ResultsVisitable, ResultsVisitor, SwitchIntEdgeEffects,
  30 };
  31
  32 use self::move_paths::MoveData;
  33
  34 pub mod drop_flag_effects;
  35 pub mod elaborate_drops;
  36 mod framework;
  37 pub mod impls;
  38 pub mod move_paths;
  39 pub mod rustc_peek;
  40 pub mod storage;
  41
  42 pub(crate) mod indexes {
  43     pub(crate) use super::move_paths::MovePathIndex;
  44 }
  45
  46 pub struct MoveDataParamEnv<'tcx> {
  47     pub move_data: MoveData<'tcx>,
  48     pub param_env: ty::ParamEnv<'tcx>,
  49 }
  50
  51 pub fn has_rustc_mir_with(tcx: TyCtxt<'_>, def_id: DefId, name: Symbol) -> Option<MetaItem> {
  52     for attr in tcx.get_attrs(def_id, sym::rustc_mir) {
  53         let items = attr.meta_item_list();
  54         for item in items.iter().flat_map(|l| l.iter()) {
  55             match item.meta_item() {
  56                 Some(mi) if mi.has_name(name) => return Some(mi.clone()),
  57                 _ => continue,
  58             }
  59         }
  60     }
  61     None
  62 }