1 //! An "interner" is a data structure that associates values with usize tags and
2 //! allows bidirectional lookup; i.e., given a value, one can easily find the
3 //! type, and vice versa.
5 use arena::DroplessArena;
6 use rustc_data_structures::fx::FxHashMap;
7 use rustc_index::vec::Idx;
8 use rustc_macros::{symbols, HashStable_Generic};
9 use rustc_serialize::{Decodable, Decoder, Encodable, Encoder};
10 use rustc_serialize::{UseSpecializedDecodable, UseSpecializedEncodable};
11 use rustc_data_structures::stable_hasher::{HashStable, ToStableHashKey, StableHasher};
13 use std::cmp::{PartialEq, PartialOrd, Ord};
15 use std::hash::{Hash, Hasher};
18 use crate::{Span, DUMMY_SP, GLOBALS};
24 // After modifying this list adjust `is_special`, `is_used_keyword`/`is_unused_keyword`,
25 // this should be rarely necessary though if the keywords are kept in alphabetic order.
27 // Special reserved identifiers used internally for elided lifetimes,
28 // unnamed method parameters, crate root module, error recovery etc.
31 DollarCrate: "$crate",
34 // Keywords that are used in stable Rust.
71 // Keywords that are used in unstable Rust or reserved for future use.
85 // Edition-specific keywords that are used in stable Rust.
86 Async: "async", // >= 2018 Edition only
87 Await: "await", // >= 2018 Edition only
88 Dyn: "dyn", // >= 2018 Edition only
90 // Edition-specific keywords that are used in unstable Rust or reserved for future use.
91 Try: "try", // >= 2018 Edition only
93 // Special lifetime names
94 UnderscoreLifetime: "'_",
95 StaticLifetime: "'static",
97 // Weak keywords, have special meaning only in specific contexts.
105 // Symbols that can be referred to with syntax_pos::sym::*. The symbol is
106 // the stringified identifier unless otherwise specified (e.g.
107 // `proc_dash_macro` represents "proc-macro").
109 // As well as the symbols listed, there are symbols for the the strings
110 // "0", "1", ..., "9", which are accessible via `sym::integer`.
112 aarch64_target_feature,
116 abi_msp430_interrupt,
125 advanced_slice_patterns,
137 allow_internal_unsafe,
138 allow_internal_unstable,
139 allow_internal_unstable_backcompat_hack,
143 arbitrary_enum_discriminant,
144 arbitrary_self_types,
151 associated_type_bounds,
152 associated_type_defaults,
160 augmented_assignments,
161 automatically_derived,
162 avx512_target_feature,
167 bind_by_move_pattern_guards,
170 borrowck_graphviz_postflow,
171 borrowck_graphviz_preflow,
174 braced_empty_structs,
186 cfg_target_has_atomic,
187 cfg_target_thread_local,
195 closure_to_fn_coercion,
197 cmpxchg16b_target_feature,
204 conservative_impl_trait,
206 const_compare_raw_pointers,
214 const_in_array_repeat_expressions,
218 const_raw_ptr_to_usize_cast,
232 crate_visibility_modifier,
240 custom_inner_attributes,
241 custom_test_frameworks,
249 default_lib_allocator,
250 default_type_parameter_fallback,
252 delay_span_bug_from_inside_query,
267 document_private_items,
268 dotdoteq_in_patterns,
269 dotdot_in_tuple_patterns,
270 double_braced_crate: "{{crate}}",
271 double_braced_impl: "{{impl}}",
272 double_braced_misc: "{{misc}}",
273 double_braced_closure: "{{closure}}",
274 double_braced_constructor: "{{constructor}}",
275 double_braced_constant: "{{constant}}",
276 double_braced_opaque: "{{opaque}}",
278 dropck_parametricity,
294 exclusive_range_pattern,
295 exhaustive_integer_patterns,
301 extern_absolute_paths,
303 extern_crate_item_prelude,
314 field_init_shorthand,
337 generic_associated_types,
346 hexagon_target_feature,
348 homogeneous_aggregate,
362 if_while_or_patterns,
364 impl_header_lifetime_elision,
366 impl_trait_in_bindings,
374 inclusive_range_syntax,
375 infer_outlives_requirements,
376 infer_static_outlives_requirements,
383 irrefutable_let_patterns,
386 issue_5723_bootstrap,
387 issue_tracker_base_url,
389 item_context: "ItemContext",
408 link_llvm_intrinsics,
418 macro_at_most_once_rep,
421 macro_lifetime_matcher,
422 macro_literal_matcher,
433 match_beginning_vert,
434 match_default_bindings,
452 movbe_target_feature,
488 object_safe_for_dispatch,
490 omit_gdb_pretty_printer_section,
497 optin_builtin_traits,
507 overlapping_marker_traits,
512 panic_implementation,
532 poll_with_tls_context,
533 powerpc_target_feature,
534 precise_pointer_size_matching,
539 proc_dash_macro: "proc-macro",
541 proc_macro_attribute,
547 proc_macro_internals,
549 proc_macro_non_items,
550 proc_macro_path_invoc,
555 quad_precision_float,
570 reexport_test_harness_main,
582 re_rebalance_coherence,
600 rustc_allocator_nounwind,
601 rustc_allow_const_fn_ptr,
602 rustc_args_required_const,
606 rustc_const_unstable,
607 rustc_conversion_suggestion,
610 rustc_diagnostic_item,
611 rustc_diagnostic_macros,
614 rustc_dump_env_program_clauses,
615 rustc_dump_program_clauses,
616 rustc_dump_user_substs,
618 rustc_expected_cgu_reuse,
619 rustc_if_this_changed,
620 rustc_inherit_overflow_checks,
622 rustc_layout_scalar_valid_range_end,
623 rustc_layout_scalar_valid_range_start,
624 rustc_macro_transparency,
626 rustc_nonnull_optimization_guaranteed,
627 rustc_object_lifetime_default,
628 rustc_on_unimplemented,
631 rustc_partition_codegened,
632 rustc_partition_reused,
634 rustc_peek_definite_init,
635 rustc_peek_maybe_init,
636 rustc_peek_maybe_uninit,
637 rustc_peek_indirectly_mutable,
639 rustc_proc_macro_decls,
643 rustc_std_internal_symbol,
646 rustc_reservation_impl,
648 rustc_then_this_would_need,
652 rust_eh_unwind_resume,
654 rvalue_static_promotion,
677 sse4a_target_feature,
690 stmt_expr_attributes,
692 struct_field_attributes,
701 target_has_atomic_load_store,
706 termination_trait_test,
709 test_accepted_feature,
711 test_removed_feature,
732 type_alias_impl_trait,
737 type_alias_enum_variants,
749 underscore_const_names,
751 underscore_lifetimes,
753 universal_impl_trait,
756 unrestricted_attribute_tokens,
759 unsized_tuple_coercion,
774 visible_private_types,
789 #[derive(Copy, Clone, Eq, HashStable_Generic)]
797 /// Constructs a new identifier from a symbol and a span.
798 pub const fn new(name: Symbol, span: Span) -> Ident {
802 /// Constructs a new identifier with a dummy span.
804 pub const fn with_dummy_span(name: Symbol) -> Ident {
805 Ident::new(name, DUMMY_SP)
809 pub fn invalid() -> Ident {
810 Ident::with_dummy_span(kw::Invalid)
813 /// Maps a string to an identifier with a dummy span.
814 pub fn from_str(string: &str) -> Ident {
815 Ident::with_dummy_span(Symbol::intern(string))
818 /// Maps a string and a span to an identifier.
819 pub fn from_str_and_span(string: &str, span: Span) -> Ident {
820 Ident::new(Symbol::intern(string), span)
823 /// Replaces `lo` and `hi` with those from `span`, but keep hygiene context.
824 pub fn with_span_pos(self, span: Span) -> Ident {
825 Ident::new(self.name, span.with_ctxt(self.span.ctxt()))
828 pub fn without_first_quote(self) -> Ident {
829 Ident::new(Symbol::intern(self.as_str().trim_start_matches('\'')), self.span)
832 /// "Normalize" ident for use in comparisons using "item hygiene".
833 /// Identifiers with same string value become same if they came from the same "modern" macro
834 /// (e.g., `macro` item, but not `macro_rules` item) and stay different if they came from
835 /// different "modern" macros.
836 /// Technically, this operation strips all non-opaque marks from ident's syntactic context.
837 pub fn modern(self) -> Ident {
838 Ident::new(self.name, self.span.modern())
841 /// "Normalize" ident for use in comparisons using "local variable hygiene".
842 /// Identifiers with same string value become same if they came from the same non-transparent
843 /// macro (e.g., `macro` or `macro_rules!` items) and stay different if they came from different
844 /// non-transparent macros.
845 /// Technically, this operation strips all transparent marks from ident's syntactic context.
846 pub fn modern_and_legacy(self) -> Ident {
847 Ident::new(self.name, self.span.modern_and_legacy())
850 /// Convert the name to a `SymbolStr`. This is a slowish operation because
851 /// it requires locking the symbol interner.
852 pub fn as_str(self) -> SymbolStr {
857 impl PartialEq for Ident {
858 fn eq(&self, rhs: &Self) -> bool {
859 self.name == rhs.name && self.span.ctxt() == rhs.span.ctxt()
863 impl Hash for Ident {
864 fn hash<H: Hasher>(&self, state: &mut H) {
865 self.name.hash(state);
866 self.span.ctxt().hash(state);
870 impl fmt::Debug for Ident {
871 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
872 if self.is_raw_guess() {
875 write!(f, "{}{:?}", self.name, self.span.ctxt())
879 impl fmt::Display for Ident {
880 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
881 if self.is_raw_guess() {
884 fmt::Display::fmt(&self.name, f)
888 impl UseSpecializedEncodable for Ident {
889 fn default_encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
890 s.emit_struct("Ident", 2, |s| {
891 s.emit_struct_field("name", 0, |s| {
894 s.emit_struct_field("span", 1, |s| {
901 impl UseSpecializedDecodable for Ident {
902 fn default_decode<D: Decoder>(d: &mut D) -> Result<Self, D::Error> {
903 d.read_struct("Ident", 2, |d| {
905 name: d.read_struct_field("name", 0, Decodable::decode)?,
906 span: d.read_struct_field("span", 1, Decodable::decode)?,
912 /// An interned string.
914 /// Internally, a `Symbol` is implemented as an index, and all operations
915 /// (including hashing, equality, and ordering) operate on that index. The use
916 /// of `rustc_index::newtype_index!` means that `Option<Symbol>` only takes up 4 bytes,
917 /// because `rustc_index::newtype_index!` reserves the last 256 values for tagging purposes.
919 /// Note that `Symbol` cannot directly be a `rustc_index::newtype_index!` because it
920 /// implements `fmt::Debug`, `Encodable`, and `Decodable` in special ways.
921 #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
922 pub struct Symbol(SymbolIndex);
924 rustc_index::newtype_index! {
925 pub struct SymbolIndex { .. }
929 const fn new(n: u32) -> Self {
930 Symbol(SymbolIndex::from_u32_const(n))
933 /// Maps a string to its interned representation.
934 pub fn intern(string: &str) -> Self {
935 with_interner(|interner| interner.intern(string))
938 /// Access the symbol's chars. This is a slowish operation because it
939 /// requires locking the symbol interner.
940 pub fn with<F: FnOnce(&str) -> R, R>(self, f: F) -> R {
941 with_interner(|interner| {
942 f(interner.get(self))
946 /// Convert to a `SymbolStr`. This is a slowish operation because it
947 /// requires locking the symbol interner.
948 pub fn as_str(self) -> SymbolStr {
949 with_interner(|interner| unsafe {
951 string: std::mem::transmute::<&str, &str>(interner.get(self))
956 pub fn as_u32(self) -> u32 {
961 impl fmt::Debug for Symbol {
962 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
963 self.with(|str| fmt::Debug::fmt(&str, f))
967 impl fmt::Display for Symbol {
968 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
969 self.with(|str| fmt::Display::fmt(&str, f))
973 impl Encodable for Symbol {
974 fn encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
975 self.with(|string| s.emit_str(string))
979 impl Decodable for Symbol {
980 fn decode<D: Decoder>(d: &mut D) -> Result<Symbol, D::Error> {
981 Ok(Symbol::intern(&d.read_str()?))
985 impl<CTX> HashStable<CTX> for Symbol {
987 fn hash_stable(&self, hcx: &mut CTX, hasher: &mut StableHasher) {
988 self.as_str().hash_stable(hcx, hasher);
992 impl<CTX> ToStableHashKey<CTX> for Symbol {
993 type KeyType = SymbolStr;
996 fn to_stable_hash_key(&self, _: &CTX) -> SymbolStr {
1001 // The `&'static str`s in this type actually point into the arena.
1003 pub struct Interner {
1004 arena: DroplessArena,
1005 names: FxHashMap<&'static str, Symbol>,
1006 strings: Vec<&'static str>,
1010 fn prefill(init: &[&'static str]) -> Self {
1012 strings: init.into(),
1013 names: init.iter().copied().zip((0..).map(Symbol::new)).collect(),
1014 ..Default::default()
1018 pub fn intern(&mut self, string: &str) -> Symbol {
1019 if let Some(&name) = self.names.get(string) {
1023 let name = Symbol::new(self.strings.len() as u32);
1025 // `from_utf8_unchecked` is safe since we just allocated a `&str` which is known to be
1027 let string: &str = unsafe {
1028 str::from_utf8_unchecked(self.arena.alloc_slice(string.as_bytes()))
1030 // It is safe to extend the arena allocation to `'static` because we only access
1031 // these while the arena is still alive.
1032 let string: &'static str = unsafe {
1033 &*(string as *const str)
1035 self.strings.push(string);
1036 self.names.insert(string, name);
1040 // Get the symbol as a string. `Symbol::as_str()` should be used in
1041 // preference to this function.
1042 pub fn get(&self, symbol: Symbol) -> &str {
1043 self.strings[symbol.0.as_usize()]
1047 // This module has a very short name because it's used a lot.
1053 // This module has a very short name because it's used a lot.
1055 use std::convert::TryInto;
1060 // Get the symbol for an integer. The first few non-negative integers each
1061 // have a static symbol and therefore are fast.
1062 pub fn integer<N: TryInto<usize> + Copy + ToString>(n: N) -> Symbol {
1063 if let Result::Ok(idx) = n.try_into() {
1064 if let Option::Some(&sym) = digits_array.get(idx) {
1068 Symbol::intern(&n.to_string())
1073 fn is_used_keyword_2018(self) -> bool {
1074 self >= kw::Async && self <= kw::Dyn
1077 fn is_unused_keyword_2018(self) -> bool {
1081 /// Used for sanity checking rustdoc keyword sections.
1082 pub fn is_doc_keyword(self) -> bool {
1086 /// A keyword or reserved identifier that can be used as a path segment.
1087 pub fn is_path_segment_keyword(self) -> bool {
1088 self == kw::Super ||
1089 self == kw::SelfLower ||
1090 self == kw::SelfUpper ||
1091 self == kw::Crate ||
1092 self == kw::PathRoot ||
1093 self == kw::DollarCrate
1096 /// Returns `true` if the symbol is `true` or `false`.
1097 pub fn is_bool_lit(self) -> bool {
1098 self == kw::True || self == kw::False
1101 /// This symbol can be a raw identifier.
1102 pub fn can_be_raw(self) -> bool {
1103 self != kw::Invalid && self != kw::Underscore && !self.is_path_segment_keyword()
1108 // Returns `true` for reserved identifiers used internally for elided lifetimes,
1109 // unnamed method parameters, crate root module, error recovery etc.
1110 pub fn is_special(self) -> bool {
1111 self.name <= kw::Underscore
1114 /// Returns `true` if the token is a keyword used in the language.
1115 pub fn is_used_keyword(self) -> bool {
1116 // Note: `span.edition()` is relatively expensive, don't call it unless necessary.
1117 self.name >= kw::As && self.name <= kw::While ||
1118 self.name.is_used_keyword_2018() && self.span.rust_2018()
1121 /// Returns `true` if the token is a keyword reserved for possible future use.
1122 pub fn is_unused_keyword(self) -> bool {
1123 // Note: `span.edition()` is relatively expensive, don't call it unless necessary.
1124 self.name >= kw::Abstract && self.name <= kw::Yield ||
1125 self.name.is_unused_keyword_2018() && self.span.rust_2018()
1128 /// Returns `true` if the token is either a special identifier or a keyword.
1129 pub fn is_reserved(self) -> bool {
1130 self.is_special() || self.is_used_keyword() || self.is_unused_keyword()
1133 /// A keyword or reserved identifier that can be used as a path segment.
1134 pub fn is_path_segment_keyword(self) -> bool {
1135 self.name.is_path_segment_keyword()
1138 /// We see this identifier in a normal identifier position, like variable name or a type.
1139 /// How was it written originally? Did it use the raw form? Let's try to guess.
1140 pub fn is_raw_guess(self) -> bool {
1141 self.name.can_be_raw() && self.is_reserved()
1146 fn with_interner<T, F: FnOnce(&mut Interner) -> T>(f: F) -> T {
1147 GLOBALS.with(|globals| f(&mut *globals.symbol_interner.lock()))
1150 /// An alternative to `Symbol`, useful when the chars within the symbol need to
1151 /// be accessed. It deliberately has limited functionality and should only be
1152 /// used for temporary values.
1154 /// Because the interner outlives any thread which uses this type, we can
1155 /// safely treat `string` which points to interner data, as an immortal string,
1156 /// as long as this type never crosses between threads.
1158 // FIXME: ensure that the interner outlives any thread which uses `SymbolStr`,
1159 // by creating a new thread right after constructing the interner.
1160 #[derive(Clone, Eq, PartialOrd, Ord)]
1161 pub struct SymbolStr {
1162 string: &'static str,
1165 // This impl allows a `SymbolStr` to be directly equated with a `String` or
1167 impl<T: std::ops::Deref<Target = str>> std::cmp::PartialEq<T> for SymbolStr {
1168 fn eq(&self, other: &T) -> bool {
1169 self.string == other.deref()
1173 impl !Send for SymbolStr {}
1174 impl !Sync for SymbolStr {}
1176 /// This impl means that if `ss` is a `SymbolStr`:
1177 /// - `*ss` is a `str`;
1178 /// - `&*ss` is a `&str`;
1179 /// - `&ss as &str` is a `&str`, which means that `&ss` can be passed to a
1180 /// function expecting a `&str`.
1181 impl std::ops::Deref for SymbolStr {
1184 fn deref(&self) -> &str { self.string }
1187 impl fmt::Debug for SymbolStr {
1188 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1189 fmt::Debug::fmt(self.string, f)
1193 impl fmt::Display for SymbolStr {
1194 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1195 fmt::Display::fmt(self.string, f)
1199 impl<CTX> HashStable<CTX> for SymbolStr {
1201 fn hash_stable(&self, hcx: &mut CTX, hasher: &mut StableHasher) {
1202 self.string.hash_stable(hcx, hasher)
1206 impl<CTX> ToStableHashKey<CTX> for SymbolStr {
1207 type KeyType = SymbolStr;
1210 fn to_stable_hash_key(&self, _: &CTX) -> SymbolStr {