compiler/rustc_hir/src/definitions.rs

   1 //! For each definition, we track the following data. A definition
   2 //! here is defined somewhat circularly as "something with a `DefId`",
   3 //! but it generally corresponds to things like structs, enums, etc.
   4 //! There are also some rather random cases (like const initializer
   5 //! expressions) that are mostly just leftovers.
   6
   7 pub use crate::def_id::DefPathHash;
   8 use crate::def_id::{CrateNum, DefIndex, LocalDefId, StableCrateId, CRATE_DEF_INDEX, LOCAL_CRATE};
   9 use crate::def_path_hash_map::DefPathHashMap;
  10
  11 use rustc_data_structures::fx::FxHashMap;
  12 use rustc_data_structures::stable_hasher::StableHasher;
  13 use rustc_index::vec::IndexVec;
  14 use rustc_span::hygiene::ExpnId;
  15 use rustc_span::symbol::{kw, sym, Symbol};
  16 use rustc_span::Span;
  17
  18 use std::fmt::{self, Write};
  19 use std::hash::Hash;
  20 use tracing::debug;
  21
  22 /// The `DefPathTable` maps `DefIndex`es to `DefKey`s and vice versa.
  23 /// Internally the `DefPathTable` holds a tree of `DefKey`s, where each `DefKey`
  24 /// stores the `DefIndex` of its parent.
  25 /// There is one `DefPathTable` for each crate.
  26 #[derive(Clone, Default, Debug)]
  27 pub struct DefPathTable {
  28     index_to_key: IndexVec<DefIndex, DefKey>,
  29     def_path_hashes: IndexVec<DefIndex, DefPathHash>,
  30     def_path_hash_to_index: DefPathHashMap,
  31 }
  32
  33 impl DefPathTable {
  34     fn allocate(&mut self, key: DefKey, def_path_hash: DefPathHash) -> DefIndex {
  35         let index = {
  36             let index = DefIndex::from(self.index_to_key.len());
  37             debug!("DefPathTable::insert() - {:?} <-> {:?}", key, index);
  38             self.index_to_key.push(key);
  39             index
  40         };
  41         self.def_path_hashes.push(def_path_hash);
  42         debug_assert!(self.def_path_hashes.len() == self.index_to_key.len());
  43
  44         // Check for hash collisions of DefPathHashes. These should be
  45         // exceedingly rare.
  46         if let Some(existing) = self.def_path_hash_to_index.insert(&def_path_hash, &index) {
  47             let def_path1 = DefPath::make(LOCAL_CRATE, existing, |idx| self.def_key(idx));
  48             let def_path2 = DefPath::make(LOCAL_CRATE, index, |idx| self.def_key(idx));
  49
  50             // Continuing with colliding DefPathHashes can lead to correctness
  51             // issues. We must abort compilation.
  52             //
  53             // The likelyhood of such a collision is very small, so actually
  54             // running into one could be indicative of a poor hash function
  55             // being used.
  56             //
  57             // See the documentation for DefPathHash for more information.
  58             panic!(
  59                 "found DefPathHash collsion between {:?} and {:?}. \
  60                     Compilation cannot continue.",
  61                 def_path1, def_path2
  62             );
  63         }
  64
  65         // Assert that all DefPathHashes correctly contain the local crate's
  66         // StableCrateId
  67         #[cfg(debug_assertions)]
  68         if let Some(root) = self.def_path_hashes.get(CRATE_DEF_INDEX) {
  69             assert!(def_path_hash.stable_crate_id() == root.stable_crate_id());
  70         }
  71
  72         index
  73     }
  74
  75     #[inline(always)]
  76     pub fn def_key(&self, index: DefIndex) -> DefKey {
  77         self.index_to_key[index]
  78     }
  79
  80     #[inline(always)]
  81     pub fn def_path_hash(&self, index: DefIndex) -> DefPathHash {
  82         let hash = self.def_path_hashes[index];
  83         debug!("def_path_hash({:?}) = {:?}", index, hash);
  84         hash
  85     }
  86
  87     pub fn enumerated_keys_and_path_hashes(
  88         &self,
  89     ) -> impl Iterator<Item = (DefIndex, &DefKey, &DefPathHash)> + ExactSizeIterator + '_ {
  90         self.index_to_key
  91             .iter_enumerated()
  92             .map(move |(index, key)| (index, key, &self.def_path_hashes[index]))
  93     }
  94 }
  95
  96 /// The definition table containing node definitions.
  97 /// It holds the `DefPathTable` for `LocalDefId`s/`DefPath`s.
  98 /// It also stores mappings to convert `LocalDefId`s to/from `HirId`s.
  99 #[derive(Clone, Debug)]
 100 pub struct Definitions {
 101     table: DefPathTable,
 102
 103     /// Item with a given `LocalDefId` was defined during macro expansion with ID `ExpnId`.
 104     expansions_that_defined: FxHashMap<LocalDefId, ExpnId>,
 105
 106     def_id_to_span: IndexVec<LocalDefId, Span>,
 107
 108     /// The [StableCrateId] of the local crate.
 109     stable_crate_id: StableCrateId,
 110 }
 111
 112 /// A unique identifier that we can use to lookup a definition
 113 /// precisely. It combines the index of the definition's parent (if
 114 /// any) with a `DisambiguatedDefPathData`.
 115 #[derive(Copy, Clone, PartialEq, Debug, Encodable, Decodable)]
 116 pub struct DefKey {
 117     /// The parent path.
 118     pub parent: Option<DefIndex>,
 119
 120     /// The identifier of this node.
 121     pub disambiguated_data: DisambiguatedDefPathData,
 122 }
 123
 124 impl DefKey {
 125     pub(crate) fn compute_stable_hash(&self, parent: DefPathHash) -> DefPathHash {
 126         let mut hasher = StableHasher::new();
 127
 128         parent.hash(&mut hasher);
 129
 130         let DisambiguatedDefPathData { ref data, disambiguator } = self.disambiguated_data;
 131
 132         std::mem::discriminant(data).hash(&mut hasher);
 133         if let Some(name) = data.get_opt_name() {
 134             // Get a stable hash by considering the symbol chars rather than
 135             // the symbol index.
 136             name.as_str().hash(&mut hasher);
 137         }
 138
 139         disambiguator.hash(&mut hasher);
 140
 141         let local_hash: u64 = hasher.finish();
 142
 143         // Construct the new DefPathHash, making sure that the `crate_id`
 144         // portion of the hash is properly copied from the parent. This way the
 145         // `crate_id` part will be recursively propagated from the root to all
 146         // DefPathHashes in this DefPathTable.
 147         DefPathHash::new(parent.stable_crate_id(), local_hash)
 148     }
 149 }
 150
 151 /// A pair of `DefPathData` and an integer disambiguator. The integer is
 152 /// normally `0`, but in the event that there are multiple defs with the
 153 /// same `parent` and `data`, we use this field to disambiguate
 154 /// between them. This introduces some artificial ordering dependency
 155 /// but means that if you have, e.g., two impls for the same type in
 156 /// the same module, they do get distinct `DefId`s.
 157 #[derive(Copy, Clone, PartialEq, Debug, Encodable, Decodable)]
 158 pub struct DisambiguatedDefPathData {
 159     pub data: DefPathData,
 160     pub disambiguator: u32,
 161 }
 162
 163 impl DisambiguatedDefPathData {
 164     pub fn fmt_maybe_verbose(&self, writer: &mut impl Write, verbose: bool) -> fmt::Result {
 165         match self.data.name() {
 166             DefPathDataName::Named(name) => {
 167                 if verbose && self.disambiguator != 0 {
 168                     write!(writer, "{}#{}", name, self.disambiguator)
 169                 } else {
 170                     writer.write_str(name.as_str())
 171                 }
 172             }
 173             DefPathDataName::Anon { namespace } => {
 174                 write!(writer, "{{{}#{}}}", namespace, self.disambiguator)
 175             }
 176         }
 177     }
 178 }
 179
 180 impl fmt::Display for DisambiguatedDefPathData {
 181     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 182         self.fmt_maybe_verbose(f, true)
 183     }
 184 }
 185
 186 #[derive(Clone, Debug, Encodable, Decodable)]
 187 pub struct DefPath {
 188     /// The path leading from the crate root to the item.
 189     pub data: Vec<DisambiguatedDefPathData>,
 190
 191     /// The crate root this path is relative to.
 192     pub krate: CrateNum,
 193 }
 194
 195 impl DefPath {
 196     pub fn make<FN>(krate: CrateNum, start_index: DefIndex, mut get_key: FN) -> DefPath
 197     where
 198         FN: FnMut(DefIndex) -> DefKey,
 199     {
 200         let mut data = vec![];
 201         let mut index = Some(start_index);
 202         loop {
 203             debug!("DefPath::make: krate={:?} index={:?}", krate, index);
 204             let p = index.unwrap();
 205             let key = get_key(p);
 206             debug!("DefPath::make: key={:?}", key);
 207             match key.disambiguated_data.data {
 208                 DefPathData::CrateRoot => {
 209                     assert!(key.parent.is_none());
 210                     break;
 211                 }
 212                 _ => {
 213                     data.push(key.disambiguated_data);
 214                     index = key.parent;
 215                 }
 216             }
 217         }
 218         data.reverse();
 219         DefPath { data, krate }
 220     }
 221
 222     /// Returns a string representation of the `DefPath` without
 223     /// the crate-prefix. This method is useful if you don't have
 224     /// a `TyCtxt` available.
 225     pub fn to_string_no_crate_verbose(&self) -> String {
 226         let mut s = String::with_capacity(self.data.len() * 16);
 227
 228         for component in &self.data {
 229             write!(s, "::{}", component).unwrap();
 230         }
 231
 232         s
 233     }
 234
 235     /// Returns a filename-friendly string of the `DefPath`, without
 236     /// the crate-prefix. This method is useful if you don't have
 237     /// a `TyCtxt` available.
 238     pub fn to_filename_friendly_no_crate(&self) -> String {
 239         let mut s = String::with_capacity(self.data.len() * 16);
 240
 241         let mut opt_delimiter = None;
 242         for component in &self.data {
 243             s.extend(opt_delimiter);
 244             opt_delimiter = Some('-');
 245             write!(s, "{}", component).unwrap();
 246         }
 247
 248         s
 249     }
 250 }
 251
 252 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, Encodable, Decodable)]
 253 pub enum DefPathData {
 254     // Root: these should only be used for the root nodes, because
 255     // they are treated specially by the `def_path` function.
 256     /// The crate root (marker).
 257     CrateRoot,
 258     // Catch-all for random `DefId` things like `DUMMY_NODE_ID`.
 259     Misc,
 260
 261     // Different kinds of items and item-like things:
 262     /// An impl.
 263     Impl,
 264     /// An `extern` block.
 265     ForeignMod,
 266     /// Something in the type namespace.
 267     TypeNs(Symbol),
 268     /// Something in the value namespace.
 269     ValueNs(Symbol),
 270     /// Something in the macro namespace.
 271     MacroNs(Symbol),
 272     /// Something in the lifetime namespace.
 273     LifetimeNs(Symbol),
 274     /// A closure expression.
 275     ClosureExpr,
 276
 277     // Subportions of items:
 278     /// Implicit constructor for a unit or tuple-like struct or enum variant.
 279     Ctor,
 280     /// A constant expression (see `{ast,hir}::AnonConst`).
 281     AnonConst,
 282     /// An `impl Trait` type node.
 283     ImplTrait,
 284 }
 285
 286 impl Definitions {
 287     pub fn def_path_table(&self) -> &DefPathTable {
 288         &self.table
 289     }
 290
 291     /// Gets the number of definitions.
 292     pub fn def_index_count(&self) -> usize {
 293         self.table.index_to_key.len()
 294     }
 295
 296     #[inline]
 297     pub fn def_key(&self, id: LocalDefId) -> DefKey {
 298         self.table.def_key(id.local_def_index)
 299     }
 300
 301     #[inline(always)]
 302     pub fn def_path_hash(&self, id: LocalDefId) -> DefPathHash {
 303         self.table.def_path_hash(id.local_def_index)
 304     }
 305
 306     /// Returns the path from the crate root to `index`. The root
 307     /// nodes are not included in the path (i.e., this will be an
 308     /// empty vector for the crate root). For an inlined item, this
 309     /// will be the path of the item in the external crate (but the
 310     /// path will begin with the path to the external crate).
 311     pub fn def_path(&self, id: LocalDefId) -> DefPath {
 312         DefPath::make(LOCAL_CRATE, id.local_def_index, |index| {
 313             self.def_key(LocalDefId { local_def_index: index })
 314         })
 315     }
 316
 317     /// Adds a root definition (no parent) and a few other reserved definitions.
 318     pub fn new(stable_crate_id: StableCrateId, crate_span: Span) -> Definitions {
 319         let key = DefKey {
 320             parent: None,
 321             disambiguated_data: DisambiguatedDefPathData {
 322                 data: DefPathData::CrateRoot,
 323                 disambiguator: 0,
 324             },
 325         };
 326
 327         let parent_hash = DefPathHash::new(stable_crate_id, 0);
 328         let def_path_hash = key.compute_stable_hash(parent_hash);
 329
 330         // Create the root definition.
 331         let mut table = DefPathTable::default();
 332         let root = LocalDefId { local_def_index: table.allocate(key, def_path_hash) };
 333         assert_eq!(root.local_def_index, CRATE_DEF_INDEX);
 334
 335         let mut def_id_to_span = IndexVec::new();
 336         // A relative span's parent must be an absolute span.
 337         debug_assert_eq!(crate_span.data_untracked().parent, None);
 338         let _root = def_id_to_span.push(crate_span);
 339         debug_assert_eq!(_root, root);
 340
 341         Definitions {
 342             table,
 343             expansions_that_defined: Default::default(),
 344             def_id_to_span,
 345             stable_crate_id,
 346         }
 347     }
 348
 349     /// Retrieves the root definition.
 350     pub fn get_root_def(&self) -> LocalDefId {
 351         LocalDefId { local_def_index: CRATE_DEF_INDEX }
 352     }
 353
 354     /// Adds a definition with a parent definition.
 355     pub fn create_def(
 356         &mut self,
 357         parent: LocalDefId,
 358         data: DefPathData,
 359         expn_id: ExpnId,
 360         mut next_disambiguator: impl FnMut(LocalDefId, DefPathData) -> u32,
 361         span: Span,
 362     ) -> LocalDefId {
 363         debug!("create_def(parent={:?}, data={:?}, expn_id={:?})", parent, data, expn_id);
 364
 365         // The root node must be created with `create_root_def()`.
 366         assert!(data != DefPathData::CrateRoot);
 367
 368         let disambiguator = next_disambiguator(parent, data);
 369         let key = DefKey {
 370             parent: Some(parent.local_def_index),
 371             disambiguated_data: DisambiguatedDefPathData { data, disambiguator },
 372         };
 373
 374         let parent_hash = self.table.def_path_hash(parent.local_def_index);
 375         let def_path_hash = key.compute_stable_hash(parent_hash);
 376
 377         debug!("create_def: after disambiguation, key = {:?}", key);
 378
 379         // Create the definition.
 380         let def_id = LocalDefId { local_def_index: self.table.allocate(key, def_path_hash) };
 381
 382         if expn_id != ExpnId::root() {
 383             self.expansions_that_defined.insert(def_id, expn_id);
 384         }
 385
 386         // A relative span's parent must be an absolute span.
 387         debug_assert_eq!(span.data_untracked().parent, None);
 388         let _id = self.def_id_to_span.push(span);
 389         debug_assert_eq!(_id, def_id);
 390
 391         def_id
 392     }
 393
 394     pub fn expansion_that_defined(&self, id: LocalDefId) -> ExpnId {
 395         self.expansions_that_defined.get(&id).copied().unwrap_or_else(ExpnId::root)
 396     }
 397
 398     /// Retrieves the span of the given `DefId` if `DefId` is in the local crate.
 399     #[inline]
 400     pub fn def_span(&self, def_id: LocalDefId) -> Span {
 401         self.def_id_to_span[def_id]
 402     }
 403
 404     pub fn iter_local_def_id(&self) -> impl Iterator<Item = LocalDefId> + '_ {
 405         self.table.def_path_hashes.indices().map(|local_def_index| LocalDefId { local_def_index })
 406     }
 407
 408     #[inline(always)]
 409     pub fn local_def_path_hash_to_def_id(
 410         &self,
 411         hash: DefPathHash,
 412         err: &mut dyn FnMut() -> !,
 413     ) -> LocalDefId {
 414         debug_assert!(hash.stable_crate_id() == self.stable_crate_id);
 415         self.table
 416             .def_path_hash_to_index
 417             .get(&hash)
 418             .map(|local_def_index| LocalDefId { local_def_index })
 419             .unwrap_or_else(|| err())
 420     }
 421
 422     pub fn def_path_hash_to_def_index_map(&self) -> &DefPathHashMap {
 423         &self.table.def_path_hash_to_index
 424     }
 425 }
 426
 427 #[derive(Copy, Clone, PartialEq, Debug)]
 428 pub enum DefPathDataName {
 429     Named(Symbol),
 430     Anon { namespace: Symbol },
 431 }
 432
 433 impl DefPathData {
 434     pub fn get_opt_name(&self) -> Option<Symbol> {
 435         use self::DefPathData::*;
 436         match *self {
 437             TypeNs(name) | ValueNs(name) | MacroNs(name) | LifetimeNs(name) => Some(name),
 438
 439             Impl | ForeignMod | CrateRoot | Misc | ClosureExpr | Ctor | AnonConst | ImplTrait => {
 440                 None
 441             }
 442         }
 443     }
 444
 445     pub fn name(&self) -> DefPathDataName {
 446         use self::DefPathData::*;
 447         match *self {
 448             TypeNs(name) | ValueNs(name) | MacroNs(name) | LifetimeNs(name) => {
 449                 DefPathDataName::Named(name)
 450             }
 451             // Note that this does not show up in user print-outs.
 452             CrateRoot => DefPathDataName::Anon { namespace: kw::Crate },
 453             Impl => DefPathDataName::Anon { namespace: kw::Impl },
 454             ForeignMod => DefPathDataName::Anon { namespace: kw::Extern },
 455             Misc => DefPathDataName::Anon { namespace: sym::misc },
 456             ClosureExpr => DefPathDataName::Anon { namespace: sym::closure },
 457             Ctor => DefPathDataName::Anon { namespace: sym::constructor },
 458             AnonConst => DefPathDataName::Anon { namespace: sym::constant },
 459             ImplTrait => DefPathDataName::Anon { namespace: sym::opaque },
 460         }
 461     }
 462 }
 463
 464 impl fmt::Display for DefPathData {
 465     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 466         match self.name() {
 467             DefPathDataName::Named(name) => f.write_str(name.as_str()),
 468             // FIXME(#70334): this will generate legacy {{closure}}, {{impl}}, etc
 469             DefPathDataName::Anon { namespace } => write!(f, "{{{{{}}}}}", namespace),
 470         }
 471     }
 472 }