1 use rustc_data_structures::sync::Lock;
3 use rustc_hir::intravisit;
4 use rustc_hir::{HirId, ItemLocalId};
5 use rustc_index::bit_set::GrowableBitSet;
6 use rustc_middle::hir::map::Map;
7 use rustc_middle::hir::nested_filter;
8 use rustc_middle::ty::TyCtxt;
10 pub fn check_crate(tcx: TyCtxt<'_>) {
11 tcx.dep_graph.assert_ignored();
13 if tcx.sess.opts.unstable_opts.hir_stats {
14 crate::hir_stats::print_hir_stats(tcx);
17 #[cfg(debug_assertions)]
19 let errors = Lock::new(Vec::new());
20 let hir_map = tcx.hir();
22 hir_map.par_for_each_module(|module_id| {
23 let mut v = HirIdValidator {
26 hir_ids_seen: Default::default(),
30 tcx.hir().visit_item_likes_in_module(module_id, &mut v);
33 let errors = errors.into_inner();
35 if !errors.is_empty() {
36 let message = errors.iter().fold(String::new(), |s1, s2| s1 + "\n" + s2);
37 tcx.sess.delay_span_bug(rustc_span::DUMMY_SP, &message);
42 struct HirIdValidator<'a, 'hir> {
44 owner: Option<hir::OwnerId>,
45 hir_ids_seen: GrowableBitSet<ItemLocalId>,
46 errors: &'a Lock<Vec<String>>,
49 impl<'a, 'hir> HirIdValidator<'a, 'hir> {
50 fn new_visitor(&self, hir_map: Map<'hir>) -> HirIdValidator<'a, 'hir> {
54 hir_ids_seen: Default::default(),
61 fn error(&self, f: impl FnOnce() -> String) {
62 self.errors.lock().push(f());
65 fn check<F: FnOnce(&mut HirIdValidator<'a, 'hir>)>(&mut self, owner: hir::OwnerId, walk: F) {
66 assert!(self.owner.is_none());
67 self.owner = Some(owner);
70 if owner == hir::CRATE_OWNER_ID {
74 // There's always at least one entry for the owning item itself
78 .map(|local_id| local_id.as_usize())
80 .expect("owning item has no entry");
82 if max != self.hir_ids_seen.len() - 1 {
83 // Collect the missing ItemLocalIds
84 let missing: Vec<_> = (0..=max as u32)
85 .filter(|&i| !self.hir_ids_seen.contains(ItemLocalId::from_u32(i)))
88 // Try to map those to something more useful
89 let mut missing_items = Vec::with_capacity(missing.len());
91 for local_id in missing {
92 let hir_id = HirId { owner, local_id: ItemLocalId::from_u32(local_id) };
94 trace!("missing hir id {:#?}", hir_id);
96 missing_items.push(format!(
97 "[local_id: {}, owner: {}]",
99 self.hir_map.def_path(owner.def_id).to_string_no_crate_verbose()
104 "ItemLocalIds not assigned densely in {}. \
105 Max ItemLocalId = {}, missing IDs = {:#?}; seens IDs = {:#?}",
106 self.hir_map.def_path(owner.def_id).to_string_no_crate_verbose(),
111 .map(|local_id| HirId { owner, local_id })
112 .map(|h| format!("({:?} {})", h, self.hir_map.node_to_string(h)))
120 impl<'a, 'hir> intravisit::Visitor<'hir> for HirIdValidator<'a, 'hir> {
121 type NestedFilter = nested_filter::OnlyBodies;
123 fn nested_visit_map(&mut self) -> Self::Map {
127 fn visit_item(&mut self, i: &'hir hir::Item<'hir>) {
128 let mut inner_visitor = self.new_visitor(self.hir_map);
129 inner_visitor.check(i.owner_id, |this| intravisit::walk_item(this, i));
132 fn visit_id(&mut self, hir_id: HirId) {
133 let owner = self.owner.expect("no owner");
135 if owner != hir_id.owner {
138 "HirIdValidator: The recorded owner of {} is {} instead of {}",
139 self.hir_map.node_to_string(hir_id),
140 self.hir_map.def_path(hir_id.owner.def_id).to_string_no_crate_verbose(),
141 self.hir_map.def_path(owner.def_id).to_string_no_crate_verbose()
146 self.hir_ids_seen.insert(hir_id.local_id);
149 fn visit_foreign_item(&mut self, i: &'hir hir::ForeignItem<'hir>) {
150 let mut inner_visitor = self.new_visitor(self.hir_map);
151 inner_visitor.check(i.owner_id, |this| intravisit::walk_foreign_item(this, i));
154 fn visit_trait_item(&mut self, i: &'hir hir::TraitItem<'hir>) {
155 let mut inner_visitor = self.new_visitor(self.hir_map);
156 inner_visitor.check(i.owner_id, |this| intravisit::walk_trait_item(this, i));
159 fn visit_impl_item(&mut self, i: &'hir hir::ImplItem<'hir>) {
160 let mut inner_visitor = self.new_visitor(self.hir_map);
161 inner_visitor.check(i.owner_id, |this| intravisit::walk_impl_item(this, i));