Skip to main content

rustc_expand/mbe/
macro_parser.rs

1//! This is an NFA-based parser, which calls out to the main Rust parser for named non-terminals
2//! (which it commits to fully when it hits one in a grammar). There's a set of current NFA threads
3//! and a set of next ones. Instead of NTs, we have a special case for Kleene star. The big-O, in
4//! pathological cases, is worse than traditional use of NFA or Earley parsing, but it's an easier
5//! fit for Macro-by-Example-style rules.
6//!
7//! (In order to prevent the pathological case, we'd need to lazily construct the resulting
8//! `NamedMatch`es at the very end. It'd be a pain, and require more memory to keep around old
9//! matcher positions, but it would also save overhead)
10//!
11//! We don't say this parser uses the Earley algorithm, because it's unnecessarily inaccurate.
12//! The macro parser restricts itself to the features of finite state automata. Earley parsers
13//! can be described as an extension of NFAs with completion rules, prediction rules, and recursion.
14//!
15//! Quick intro to how the parser works:
16//!
17//! A "matcher position" (a.k.a. "position" or "mp") is a dot in the middle of a matcher, usually
18//! written as a `·`. For example `· a $( a )* a b` is one, as is `a $( · a )* a b`.
19//!
20//! The parser walks through the input a token at a time, maintaining a list
21//! of threads consistent with the current position in the input string: `cur_mps`.
22//!
23//! As it processes them, it fills up `eof_mps` with threads that would be valid if
24//! the macro invocation is now over, `bb_mps` with threads that are waiting on
25//! a Rust non-terminal like `$e:expr`, and `next_mps` with threads that are waiting
26//! on a particular token. Most of the logic concerns moving the · through the
27//! repetitions indicated by Kleene stars. The rules for moving the · without
28//! consuming any input are called epsilon transitions. It only advances or calls
29//! out to the real Rust parser when no `cur_mps` threads remain.
30//!
31//! Example:
32//!
33//! ```text, ignore
34//! Start parsing a a a a b against [· a $( a )* a b].
35//!
36//! Remaining input: a a a a b
37//! next: [· a $( a )* a b]
38//!
39//! - - - Advance over an a. - - -
40//!
41//! Remaining input: a a a b
42//! cur: [a · $( a )* a b]
43//! Descend/Skip (first position).
44//! next: [a $( · a )* a b]  [a $( a )* · a b].
45//!
46//! - - - Advance over an a. - - -
47//!
48//! Remaining input: a a b
49//! cur: [a $( a · )* a b]  [a $( a )* a · b]
50//! Follow epsilon transition: Finish/Repeat (first position)
51//! next: [a $( a )* · a b]  [a $( · a )* a b]  [a $( a )* a · b]
52//!
53//! - - - Advance over an a. - - - (this looks exactly like the last step)
54//!
55//! Remaining input: a b
56//! cur: [a $( a · )* a b]  [a $( a )* a · b]
57//! Follow epsilon transition: Finish/Repeat (first position)
58//! next: [a $( a )* · a b]  [a $( · a )* a b]  [a $( a )* a · b]
59//!
60//! - - - Advance over an a. - - - (this looks exactly like the last step)
61//!
62//! Remaining input: b
63//! cur: [a $( a · )* a b]  [a $( a )* a · b]
64//! Follow epsilon transition: Finish/Repeat (first position)
65//! next: [a $( a )* · a b]  [a $( · a )* a b]  [a $( a )* a · b]
66//!
67//! - - - Advance over a b. - - -
68//!
69//! Remaining input: ''
70//! eof: [a $( a )* a b ·]
71//! ```
72
73use std::borrow::Cow;
74use std::fmt::Display;
75use std::rc::Rc;
76
77pub(crate) use NamedMatch::*;
78pub(crate) use ParseResult::*;
79use rustc_ast::token::{self, DocComment, NonterminalKind, Token, TokenKind};
80use rustc_data_structures::fx::FxHashMap;
81use rustc_errors::{Diag, ErrorGuaranteed};
82use rustc_middle::span_bug;
83use rustc_parse::parser::{ParseNtResult, Parser, token_descr};
84use rustc_span::{Ident, MacroRulesNormalizedIdent, Span};
85use smallvec::SmallVec;
86
87use crate::mbe::macro_rules::Tracker;
88use crate::mbe::{KleeneOp, TokenTree};
89
90/// A unit within a matcher that a `MatcherPos` can refer to. Similar to (and derived from)
91/// `mbe::TokenTree`, but designed specifically for fast and easy traversal during matching.
92/// Notable differences to `mbe::TokenTree`:
93/// - It is non-recursive, i.e. there is no nesting.
94/// - The end pieces of each sequence (the separator, if present, and the Kleene op) are
95///   represented explicitly, as is the very end of the matcher.
96///
97/// This means a matcher can be represented by `&[MatcherLoc]`, and traversal mostly involves
98/// simply incrementing the current matcher position index by one.
99#[derive(#[automatically_derived]
impl ::core::fmt::Debug for MatcherLoc {
    #[inline]
    fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
        match self {
            MatcherLoc::Token { token: __self_0 } =>
                ::core::fmt::Formatter::debug_struct_field1_finish(f, "Token",
                    "token", &__self_0),
            MatcherLoc::Delimited =>
                ::core::fmt::Formatter::write_str(f, "Delimited"),
            MatcherLoc::Sequence {
                op: __self_0,
                num_metavar_decls: __self_1,
                idx_first_after: __self_2,
                next_metavar: __self_3,
                seq_depth: __self_4 } =>
                ::core::fmt::Formatter::debug_struct_field5_finish(f,
                    "Sequence", "op", __self_0, "num_metavar_decls", __self_1,
                    "idx_first_after", __self_2, "next_metavar", __self_3,
                    "seq_depth", &__self_4),
            MatcherLoc::SequenceKleeneOpNoSep {
                op: __self_0, idx_first: __self_1 } =>
                ::core::fmt::Formatter::debug_struct_field2_finish(f,
                    "SequenceKleeneOpNoSep", "op", __self_0, "idx_first",
                    &__self_1),
            MatcherLoc::SequenceSep { separator: __self_0 } =>
                ::core::fmt::Formatter::debug_struct_field1_finish(f,
                    "SequenceSep", "separator", &__self_0),
            MatcherLoc::SequenceKleeneOpAfterSep { idx_first: __self_0 } =>
                ::core::fmt::Formatter::debug_struct_field1_finish(f,
                    "SequenceKleeneOpAfterSep", "idx_first", &__self_0),
            MatcherLoc::MetaVarDecl {
                span: __self_0,
                bind: __self_1,
                kind: __self_2,
                next_metavar: __self_3,
                seq_depth: __self_4 } =>
                ::core::fmt::Formatter::debug_struct_field5_finish(f,
                    "MetaVarDecl", "span", __self_0, "bind", __self_1, "kind",
                    __self_2, "next_metavar", __self_3, "seq_depth", &__self_4),
            MatcherLoc::Eof => ::core::fmt::Formatter::write_str(f, "Eof"),
        }
    }
}Debug, #[automatically_derived]
impl ::core::cmp::PartialEq for MatcherLoc {
    #[inline]
    fn eq(&self, other: &MatcherLoc) -> bool {
        let __self_discr = ::core::intrinsics::discriminant_value(self);
        let __arg1_discr = ::core::intrinsics::discriminant_value(other);
        __self_discr == __arg1_discr &&
            match (self, other) {
                (MatcherLoc::Token { token: __self_0 }, MatcherLoc::Token {
                    token: __arg1_0 }) => __self_0 == __arg1_0,
                (MatcherLoc::Sequence {
                    op: __self_0,
                    num_metavar_decls: __self_1,
                    idx_first_after: __self_2,
                    next_metavar: __self_3,
                    seq_depth: __self_4 }, MatcherLoc::Sequence {
                    op: __arg1_0,
                    num_metavar_decls: __arg1_1,
                    idx_first_after: __arg1_2,
                    next_metavar: __arg1_3,
                    seq_depth: __arg1_4 }) =>
                    __self_0 == __arg1_0 && __self_1 == __arg1_1 &&
                                __self_2 == __arg1_2 && __self_3 == __arg1_3 &&
                        __self_4 == __arg1_4,
                (MatcherLoc::SequenceKleeneOpNoSep {
                    op: __self_0, idx_first: __self_1 },
                    MatcherLoc::SequenceKleeneOpNoSep {
                    op: __arg1_0, idx_first: __arg1_1 }) =>
                    __self_0 == __arg1_0 && __self_1 == __arg1_1,
                (MatcherLoc::SequenceSep { separator: __self_0 },
                    MatcherLoc::SequenceSep { separator: __arg1_0 }) =>
                    __self_0 == __arg1_0,
                (MatcherLoc::SequenceKleeneOpAfterSep { idx_first: __self_0 },
                    MatcherLoc::SequenceKleeneOpAfterSep { idx_first: __arg1_0
                    }) => __self_0 == __arg1_0,
                (MatcherLoc::MetaVarDecl {
                    span: __self_0,
                    bind: __self_1,
                    kind: __self_2,
                    next_metavar: __self_3,
                    seq_depth: __self_4 }, MatcherLoc::MetaVarDecl {
                    span: __arg1_0,
                    bind: __arg1_1,
                    kind: __arg1_2,
                    next_metavar: __arg1_3,
                    seq_depth: __arg1_4 }) =>
                    __self_0 == __arg1_0 && __self_1 == __arg1_1 &&
                                __self_2 == __arg1_2 && __self_3 == __arg1_3 &&
                        __self_4 == __arg1_4,
                _ => true,
            }
    }
}PartialEq, #[automatically_derived]
impl ::core::clone::Clone for MatcherLoc {
    #[inline]
    fn clone(&self) -> MatcherLoc {
        match self {
            MatcherLoc::Token { token: __self_0 } =>
                MatcherLoc::Token {
                    token: ::core::clone::Clone::clone(__self_0),
                },
            MatcherLoc::Delimited => MatcherLoc::Delimited,
            MatcherLoc::Sequence {
                op: __self_0,
                num_metavar_decls: __self_1,
                idx_first_after: __self_2,
                next_metavar: __self_3,
                seq_depth: __self_4 } =>
                MatcherLoc::Sequence {
                    op: ::core::clone::Clone::clone(__self_0),
                    num_metavar_decls: ::core::clone::Clone::clone(__self_1),
                    idx_first_after: ::core::clone::Clone::clone(__self_2),
                    next_metavar: ::core::clone::Clone::clone(__self_3),
                    seq_depth: ::core::clone::Clone::clone(__self_4),
                },
            MatcherLoc::SequenceKleeneOpNoSep {
                op: __self_0, idx_first: __self_1 } =>
                MatcherLoc::SequenceKleeneOpNoSep {
                    op: ::core::clone::Clone::clone(__self_0),
                    idx_first: ::core::clone::Clone::clone(__self_1),
                },
            MatcherLoc::SequenceSep { separator: __self_0 } =>
                MatcherLoc::SequenceSep {
                    separator: ::core::clone::Clone::clone(__self_0),
                },
            MatcherLoc::SequenceKleeneOpAfterSep { idx_first: __self_0 } =>
                MatcherLoc::SequenceKleeneOpAfterSep {
                    idx_first: ::core::clone::Clone::clone(__self_0),
                },
            MatcherLoc::MetaVarDecl {
                span: __self_0,
                bind: __self_1,
                kind: __self_2,
                next_metavar: __self_3,
                seq_depth: __self_4 } =>
                MatcherLoc::MetaVarDecl {
                    span: ::core::clone::Clone::clone(__self_0),
                    bind: ::core::clone::Clone::clone(__self_1),
                    kind: ::core::clone::Clone::clone(__self_2),
                    next_metavar: ::core::clone::Clone::clone(__self_3),
                    seq_depth: ::core::clone::Clone::clone(__self_4),
                },
            MatcherLoc::Eof => MatcherLoc::Eof,
        }
    }
}Clone)]
100pub(crate) enum MatcherLoc {
101    Token {
102        token: Token,
103    },
104    Delimited,
105    Sequence {
106        op: KleeneOp,
107        num_metavar_decls: usize,
108        idx_first_after: usize,
109        next_metavar: usize,
110        seq_depth: usize,
111    },
112    SequenceKleeneOpNoSep {
113        op: KleeneOp,
114        idx_first: usize,
115    },
116    SequenceSep {
117        separator: Token,
118    },
119    SequenceKleeneOpAfterSep {
120        idx_first: usize,
121    },
122    MetaVarDecl {
123        span: Span,
124        bind: Ident,
125        kind: NonterminalKind,
126        next_metavar: usize,
127        seq_depth: usize,
128    },
129    Eof,
130}
131
132impl MatcherLoc {
133    pub(super) fn span(&self) -> Option<Span> {
134        match self {
135            MatcherLoc::Token { token } => Some(token.span),
136            MatcherLoc::Delimited => None,
137            MatcherLoc::Sequence { .. } => None,
138            MatcherLoc::SequenceKleeneOpNoSep { .. } => None,
139            MatcherLoc::SequenceSep { .. } => None,
140            MatcherLoc::SequenceKleeneOpAfterSep { .. } => None,
141            MatcherLoc::MetaVarDecl { span, .. } => Some(*span),
142            MatcherLoc::Eof => None,
143        }
144    }
145}
146
147impl Display for MatcherLoc {
148    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
149        match self {
150            MatcherLoc::Token { token } | MatcherLoc::SequenceSep { separator: token } => {
151                f.write_fmt(format_args!("{0}", token_descr(token)))write!(f, "{}", token_descr(token))
152            }
153            MatcherLoc::MetaVarDecl { bind, kind, .. } => {
154                f.write_fmt(format_args!("meta-variable `${0}:{1}`", bind, kind))write!(f, "meta-variable `${bind}:{kind}`")
155            }
156            MatcherLoc::Eof => f.write_str("end of macro"),
157
158            // These are not printed in the diagnostic
159            MatcherLoc::Delimited => f.write_str("delimiter"),
160            MatcherLoc::Sequence { .. } => f.write_str("sequence start"),
161            MatcherLoc::SequenceKleeneOpNoSep { .. } => f.write_str("sequence end"),
162            MatcherLoc::SequenceKleeneOpAfterSep { .. } => f.write_str("sequence end"),
163        }
164    }
165}
166
167pub(super) fn compute_locs(matcher: &[TokenTree]) -> Vec<MatcherLoc> {
168    fn inner(
169        tts: &[TokenTree],
170        locs: &mut Vec<MatcherLoc>,
171        next_metavar: &mut usize,
172        seq_depth: usize,
173    ) {
174        for tt in tts {
175            match tt {
176                TokenTree::Token(token) => {
177                    locs.push(MatcherLoc::Token { token: *token });
178                }
179                TokenTree::Delimited(span, _, delimited) => {
180                    let open_token = Token::new(delimited.delim.as_open_token_kind(), span.open);
181                    let close_token = Token::new(delimited.delim.as_close_token_kind(), span.close);
182
183                    locs.push(MatcherLoc::Delimited);
184                    locs.push(MatcherLoc::Token { token: open_token });
185                    inner(&delimited.tts, locs, next_metavar, seq_depth);
186                    locs.push(MatcherLoc::Token { token: close_token });
187                }
188                TokenTree::Sequence(_, seq) => {
189                    // We can't determine `idx_first_after` and construct the final
190                    // `MatcherLoc::Sequence` until after `inner()` is called and the sequence end
191                    // pieces are processed. So we push a dummy value (`Eof` is cheapest to
192                    // construct) now, and overwrite it with the proper value below.
193                    let dummy = MatcherLoc::Eof;
194                    locs.push(dummy);
195
196                    let next_metavar_orig = *next_metavar;
197                    let op = seq.kleene.op;
198                    let idx_first = locs.len();
199                    let idx_seq = idx_first - 1;
200                    inner(&seq.tts, locs, next_metavar, seq_depth + 1);
201
202                    if let Some(separator) = seq.separator {
203                        locs.push(MatcherLoc::SequenceSep { separator });
204                        locs.push(MatcherLoc::SequenceKleeneOpAfterSep { idx_first });
205                    } else {
206                        locs.push(MatcherLoc::SequenceKleeneOpNoSep { op, idx_first });
207                    }
208
209                    // Overwrite the dummy value pushed above with the proper value.
210                    locs[idx_seq] = MatcherLoc::Sequence {
211                        op,
212                        num_metavar_decls: seq.num_captures,
213                        idx_first_after: locs.len(),
214                        next_metavar: next_metavar_orig,
215                        seq_depth,
216                    };
217                }
218                &TokenTree::MetaVarDecl { span, name: bind, kind } => {
219                    locs.push(MatcherLoc::MetaVarDecl {
220                        span,
221                        bind,
222                        kind,
223                        next_metavar: *next_metavar,
224                        seq_depth,
225                    });
226                    *next_metavar += 1;
227                }
228                TokenTree::MetaVar(..) | TokenTree::MetaVarExpr(..) => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
229            }
230        }
231    }
232
233    let mut locs = ::alloc::vec::Vec::new()vec![];
234    let mut next_metavar = 0;
235    inner(matcher, &mut locs, &mut next_metavar, /* seq_depth */ 0);
236
237    // A final entry is needed for eof.
238    locs.push(MatcherLoc::Eof);
239
240    locs
241}
242
243/// A single matcher position, representing the state of matching.
244#[derive(#[automatically_derived]
impl ::core::fmt::Debug for MatcherPos {
    #[inline]
    fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
        ::core::fmt::Formatter::debug_struct_field2_finish(f, "MatcherPos",
            "idx", &self.idx, "matches", &&self.matches)
    }
}Debug)]
245struct MatcherPos {
246    /// The index into `TtParser::locs`, which represents the "dot".
247    idx: usize,
248
249    /// The matches made against metavar decls so far. On a successful match, this vector ends up
250    /// with one element per metavar decl in the matcher. Each element records token trees matched
251    /// against the relevant metavar by the black box parser. An element will be a `MatchedSeq` if
252    /// the corresponding metavar decl is within a sequence.
253    ///
254    /// It is critical to performance that this is an `Rc`, because it gets cloned frequently when
255    /// processing sequences. Mostly for sequence-ending possibilities that must be tried but end
256    /// up failing.
257    matches: Rc<Vec<NamedMatch>>,
258}
259
260// This type is used a lot. Make sure it doesn't unintentionally get bigger.
261#[cfg(target_pointer_width = "64")]
262const _: [(); 16] = [(); ::std::mem::size_of::<MatcherPos>()];rustc_data_structures::static_assert_size!(MatcherPos, 16);
263
264impl MatcherPos {
265    /// Adds `m` as a named match for the `metavar_idx`-th metavar. There are only two call sites,
266    /// and both are hot enough to be always worth inlining.
267    #[inline(always)]
268    fn push_match(&mut self, metavar_idx: usize, seq_depth: usize, m: NamedMatch) {
269        let matches = Rc::make_mut(&mut self.matches);
270        match seq_depth {
271            0 => {
272                // We are not within a sequence. Just append `m`.
273                {
    match (&metavar_idx, &matches.len()) {
        (left_val, right_val) => {
            if !(*left_val == *right_val) {
                let kind = ::core::panicking::AssertKind::Eq;
                ::core::panicking::assert_failed(kind, &*left_val,
                    &*right_val, ::core::option::Option::None);
            }
        }
    }
};assert_eq!(metavar_idx, matches.len());
274                matches.push(m);
275            }
276            _ => {
277                // We are within a sequence. Find the final `MatchedSeq` at the appropriate depth
278                // and append `m` to its vector.
279                let mut curr = &mut matches[metavar_idx];
280                for _ in 0..seq_depth - 1 {
281                    match curr {
282                        MatchedSeq(seq) => curr = seq.last_mut().unwrap(),
283                        _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
284                    }
285                }
286                match curr {
287                    MatchedSeq(seq) => seq.push(m),
288                    _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
289                }
290            }
291        }
292    }
293}
294
295/// Represents the possible results of an attempted parse.
296#[derive(#[automatically_derived]
impl<T: ::core::fmt::Debug> ::core::fmt::Debug for ParseResult<T> {
    #[inline]
    fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
        match self {
            ParseResult::Success(__self_0) =>
                ::core::fmt::Formatter::debug_tuple_field1_finish(f,
                    "Success", &__self_0),
            ParseResult::Failure =>
                ::core::fmt::Formatter::write_str(f, "Failure"),
            ParseResult::Ambiguity =>
                ::core::fmt::Formatter::write_str(f, "Ambiguity"),
            ParseResult::ErrorReported(__self_0) =>
                ::core::fmt::Formatter::debug_tuple_field1_finish(f,
                    "ErrorReported", &__self_0),
        }
    }
}Debug)]
297pub(crate) enum ParseResult<T> {
298    /// Parsed successfully.
299    Success(T),
300    /// Arm failed to match.
301    ///
302    /// [`Tracker::failure()`] will be called beforehand.
303    Failure,
304    /// The input could be parsed in multiple distinct ways.
305    ///
306    /// [`Tracker::ambiguity()`] will be called beforehand.
307    Ambiguity,
308    ErrorReported(ErrorGuaranteed),
309}
310
311/// A `ParseResult` where the `Success` variant contains a mapping of
312/// `MacroRulesNormalizedIdent`s to `NamedMatch`es. This represents the mapping
313/// of metavars to the token trees they bind to.
314pub(crate) type NamedParseResult = ParseResult<NamedMatches>;
315
316/// Contains a mapping of `MacroRulesNormalizedIdent`s to `NamedMatch`es.
317/// This represents the mapping of metavars to the token trees they bind to.
318pub(crate) type NamedMatches = FxHashMap<MacroRulesNormalizedIdent, NamedMatch>;
319
320/// Count how many metavars declarations are in `matcher`.
321pub(super) fn count_metavar_decls(matcher: &[TokenTree]) -> usize {
322    matcher
323        .iter()
324        .map(|tt| match tt {
325            TokenTree::MetaVarDecl { .. } => 1,
326            TokenTree::Sequence(_, seq) => seq.num_captures,
327            TokenTree::Delimited(.., delim) => count_metavar_decls(&delim.tts),
328            TokenTree::Token(..) => 0,
329            TokenTree::MetaVar(..) | TokenTree::MetaVarExpr(..) => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
330        })
331        .sum()
332}
333
334/// `NamedMatch` is a pattern-match result for a single metavar. All
335/// `MatchedNonterminal`s in the `NamedMatch` have the same non-terminal type
336/// (expr, item, etc).
337///
338/// The in-memory structure of a particular `NamedMatch` represents the match
339/// that occurred when a particular subset of a matcher was applied to a
340/// particular token tree.
341///
342/// The width of each `MatchedSeq` in the `NamedMatch`, and the identity of
343/// the `MatchedNtNonTts`s, will depend on the token tree it was applied
344/// to: each `MatchedSeq` corresponds to a single repetition in the originating
345/// token tree. The depth of the `NamedMatch` structure will therefore depend
346/// only on the nesting depth of repetitions in the originating token tree it
347/// was derived from.
348///
349/// In layperson's terms: `NamedMatch` will form a tree representing nested matches of a particular
350/// meta variable. For example, if we are matching the following macro against the following
351/// invocation...
352///
353/// ```rust
354/// macro_rules! foo {
355///   ($($($x:ident),+);+) => {}
356/// }
357///
358/// foo!(a, b, c, d; a, b, c, d, e);
359/// ```
360///
361/// Then, the tree will have the following shape:
362///
363/// ```ignore (private-internal)
364/// # use NamedMatch::*;
365/// MatchedSeq([
366///   MatchedSeq([
367///     MatchedNonterminal(a),
368///     MatchedNonterminal(b),
369///     MatchedNonterminal(c),
370///     MatchedNonterminal(d),
371///   ]),
372///   MatchedSeq([
373///     MatchedNonterminal(a),
374///     MatchedNonterminal(b),
375///     MatchedNonterminal(c),
376///     MatchedNonterminal(d),
377///     MatchedNonterminal(e),
378///   ])
379/// ])
380/// ```
381#[derive(#[automatically_derived]
impl ::core::fmt::Debug for NamedMatch {
    #[inline]
    fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
        match self {
            NamedMatch::MatchedSeq(__self_0) =>
                ::core::fmt::Formatter::debug_tuple_field1_finish(f,
                    "MatchedSeq", &__self_0),
            NamedMatch::MatchedSingle(__self_0) =>
                ::core::fmt::Formatter::debug_tuple_field1_finish(f,
                    "MatchedSingle", &__self_0),
        }
    }
}Debug, #[automatically_derived]
impl ::core::clone::Clone for NamedMatch {
    #[inline]
    fn clone(&self) -> NamedMatch {
        match self {
            NamedMatch::MatchedSeq(__self_0) =>
                NamedMatch::MatchedSeq(::core::clone::Clone::clone(__self_0)),
            NamedMatch::MatchedSingle(__self_0) =>
                NamedMatch::MatchedSingle(::core::clone::Clone::clone(__self_0)),
        }
    }
}Clone)]
382pub(crate) enum NamedMatch {
383    MatchedSeq(Vec<NamedMatch>),
384    MatchedSingle(ParseNtResult),
385}
386
387impl NamedMatch {
388    pub(super) fn is_repeatable(&self) -> bool {
389        match self {
390            NamedMatch::MatchedSeq(_) => true,
391            NamedMatch::MatchedSingle(_) => false,
392        }
393    }
394}
395
396/// Performs a token equality check, ignoring syntax context (that is, an unhygienic comparison)
397fn token_name_eq(t1: &Token, t2: &Token) -> bool {
398    if let (Some((ident1, is_raw1)), Some((ident2, is_raw2))) = (t1.ident(), t2.ident()) {
399        ident1.name == ident2.name && is_raw1 == is_raw2
400    } else if let (Some((ident1, is_raw1)), Some((ident2, is_raw2))) =
401        (t1.lifetime(), t2.lifetime())
402    {
403        ident1.name == ident2.name && is_raw1 == is_raw2
404    } else {
405        // Note: we SHOULD NOT use `t1.kind == t2.kind` here, and we should instead compare the
406        // tokens using the special comparison logic below.
407        // It makes sure that variants containing `InvisibleOrigin` will
408        // never compare equal to one another.
409        //
410        // When we had AST-based nonterminals we couldn't compare them, and the
411        // old `Nonterminal` type had an `eq` that always returned false,
412        // resulting in this restriction:
413        // <https://doc.rust-lang.org/nightly/reference/macros-by-example.html#forwarding-a-matched-fragment>
414        // This comparison logic emulates that behaviour. We could consider lifting this
415        // restriction now but there are still cases involving invisible
416        // delimiters that make it harder than it first appears.
417        match (t1.kind, t2.kind) {
418            (TokenKind::OpenInvisible(_) | TokenKind::CloseInvisible(_), _)
419            | (_, TokenKind::OpenInvisible(_) | TokenKind::CloseInvisible(_)) => false,
420            (a, b) => a == b,
421        }
422    }
423}
424
425// Note: the vectors could be created and dropped within `parse_tt`, but to avoid excess
426// allocations we have a single vector for each kind that is cleared and reused repeatedly.
427pub(crate) struct TtParser {
428    /// The set of current mps to be processed. This should be empty by the end of a successful
429    /// execution of `parse_tt_inner`.
430    cur_mps: Vec<MatcherPos>,
431
432    /// The set of newly generated mps. These are used to replenish `cur_mps` in the function
433    /// `parse_tt`.
434    next_mps: Vec<MatcherPos>,
435
436    /// The set of mps that are waiting for the black-box parser.
437    bb_mps: Vec<MatcherPos>,
438
439    /// Pre-allocate an empty match array, so it can be cloned cheaply for macros with many rules
440    /// that have no metavars.
441    empty_matches: Rc<Vec<NamedMatch>>,
442}
443
444impl TtParser {
445    pub(super) fn new() -> TtParser {
446        TtParser {
447            cur_mps: ::alloc::vec::Vec::new()vec![],
448            next_mps: ::alloc::vec::Vec::new()vec![],
449            bb_mps: ::alloc::vec::Vec::new()vec![],
450            empty_matches: Rc::new(::alloc::vec::Vec::new()vec![]),
451        }
452    }
453
454    pub(super) fn has_no_remaining_items_for_step(&self) -> bool {
455        self.cur_mps.is_empty()
456    }
457
458    /// Process the matcher positions of `cur_mps` until it is empty. In the process, this will
459    /// produce more mps in `next_mps` and `bb_mps`.
460    ///
461    /// # Returns
462    ///
463    /// `Some(result)` if everything is finished, `None` otherwise. Note that matches are kept
464    /// track of through the mps generated.
465    fn parse_tt_inner<'matcher, T: Tracker<'matcher>>(
466        &mut self,
467        parser: &Parser<'_>,
468        matcher: &'matcher [MatcherLoc],
469        track: &mut T,
470    ) -> Option<NamedParseResult> {
471        // Matcher positions that would be valid if the macro invocation was over now. Only
472        // modified if `token == Eof`.
473        let mut eof_mps = SmallVec::<[MatcherPos; 1]>::new();
474
475        while let Some(mp) = self.cur_mps.pop() {
476            self.match_one(parser, matcher, mp, track, &mut eof_mps);
477        }
478
479        // If we reached the end of input, check that there is EXACTLY ONE possible matcher.
480        // Otherwise, either the parse is ambiguous (which is an error) or there is a syntax error.
481        let token = &parser.token;
482        if *token == token::Eof {
483            if !self.next_mps.is_empty() {
    ::core::panicking::panic("assertion failed: self.next_mps.is_empty()")
};assert!(self.next_mps.is_empty());
484            if !self.bb_mps.is_empty() {
    ::core::panicking::panic("assertion failed: self.bb_mps.is_empty()")
};assert!(self.bb_mps.is_empty());
485
486            Some(match *eof_mps {
487                [_] => {
488                    let eof_mp = eof_mps.pop().unwrap();
489                    let matches = Rc::unwrap_or_clone(eof_mp.matches).into_iter();
490                    Success(self.nameize(matcher, matches))
491                }
492                [] => {
493                    track.failure(parser);
494                    Failure
495                }
496                _ => self.ambiguity_error(parser, matcher, track),
497            })
498        } else {
499            None
500        }
501    }
502
503    /// Match a single [`MatcherPos`].
504    #[inline(always)] // must be inlined in `parse_tt_inner()`
505    fn match_one<'matcher, T: Tracker<'matcher>>(
506        &mut self,
507        parser: &Parser<'_>,
508        matcher: &'matcher [MatcherLoc],
509        mut mp: MatcherPos,
510        track: &mut T,
511        eof_mps: &mut SmallVec<[MatcherPos; 1]>,
512    ) {
513        let matcher_loc = &matcher[mp.idx];
514        track.before_match_loc(self, matcher_loc);
515        let token = &parser.token;
516
517        match matcher_loc {
518            MatcherLoc::Token { token: t } => {
519                // If it's a doc comment, we just ignore it and move on to the next tt in the
520                // matcher. This is a bug, but #95267 showed that existing programs rely on this
521                // behaviour, and changing it would require some care and a transition period.
522                //
523                // If the token matches, we can just advance the parser.
524                //
525                // Otherwise, this match has failed, there is nothing to do, and hopefully another
526                // mp in `cur_mps` will match.
527                if #[allow(non_exhaustive_omitted_patterns)] match t {
    Token { kind: DocComment(..), .. } => true,
    _ => false,
}matches!(t, Token { kind: DocComment(..), .. }) {
528                    mp.idx += 1;
529                    self.cur_mps.push(mp);
530                } else if token_name_eq(t, token) {
531                    mp.idx += 1;
532                    self.next_mps.push(mp);
533                }
534            }
535            MatcherLoc::Delimited => {
536                // Entering the delimiter is trivial.
537                mp.idx += 1;
538                self.cur_mps.push(mp);
539            }
540            &MatcherLoc::Sequence {
541                op,
542                num_metavar_decls,
543                idx_first_after,
544                next_metavar,
545                seq_depth,
546            } => {
547                // Install an empty vec for each metavar within the sequence.
548                for metavar_idx in next_metavar..next_metavar + num_metavar_decls {
549                    mp.push_match(metavar_idx, seq_depth, MatchedSeq(::alloc::vec::Vec::new()vec![]));
550                }
551
552                if #[allow(non_exhaustive_omitted_patterns)] match op {
    KleeneOp::ZeroOrMore | KleeneOp::ZeroOrOne => true,
    _ => false,
}matches!(op, KleeneOp::ZeroOrMore | KleeneOp::ZeroOrOne) {
553                    // Try zero matches of this sequence, by skipping over it.
554                    self.cur_mps
555                        .push(MatcherPos { idx: idx_first_after, matches: Rc::clone(&mp.matches) });
556                }
557
558                // Try one or more matches of this sequence, by entering it.
559                mp.idx += 1;
560                self.cur_mps.push(mp);
561            }
562            &MatcherLoc::SequenceKleeneOpNoSep { op, idx_first } => {
563                // We are past the end of a sequence with no separator. Try ending the sequence. If
564                // that's not possible, `ending_mp` will fail quietly when it is processed next time
565                // around the loop.
566                let ending_mp = MatcherPos {
567                    idx: mp.idx + 1, // +1 skips the Kleene op
568                    matches: Rc::clone(&mp.matches),
569                };
570                self.cur_mps.push(ending_mp);
571
572                if op != KleeneOp::ZeroOrOne {
573                    // Try another repetition.
574                    mp.idx = idx_first;
575                    self.cur_mps.push(mp);
576                }
577            }
578            MatcherLoc::SequenceSep { separator } => {
579                // We are past the end of a sequence with a separator but we haven't seen the
580                // separator yet. Try ending the sequence. If that's not possible, `ending_mp` will
581                // fail quietly when it is processed next time around the loop.
582                let ending_mp = MatcherPos {
583                    idx: mp.idx + 2, // +2 skips the separator and the Kleene op
584                    matches: Rc::clone(&mp.matches),
585                };
586                self.cur_mps.push(ending_mp);
587
588                if token_name_eq(token, separator) {
589                    // The separator matches the current token. Advance past it.
590                    mp.idx += 1;
591                    self.next_mps.push(mp);
592                }
593            }
594            &MatcherLoc::SequenceKleeneOpAfterSep { idx_first } => {
595                // We are past the sequence separator. This can't be a `?` Kleene op, because they
596                // don't permit separators. Try another repetition.
597                mp.idx = idx_first;
598                self.cur_mps.push(mp);
599            }
600            &MatcherLoc::MetaVarDecl { kind, .. } => {
601                // Built-in nonterminals never start with these tokens, so we can eliminate them
602                // from consideration. We use the span of the metavariable declaration to determine
603                // any edition-specific matching behavior for non-terminals.
604                if Parser::nonterminal_may_begin_with(kind, token) {
605                    self.bb_mps.push(mp);
606                }
607            }
608            MatcherLoc::Eof => {
609                // We are past the matcher's end, and not in a sequence. Try to end things.
610                if true {
    {
        match (&mp.idx, &(matcher.len() - 1)) {
            (left_val, right_val) => {
                if !(*left_val == *right_val) {
                    let kind = ::core::panicking::AssertKind::Eq;
                    ::core::panicking::assert_failed(kind, &*left_val,
                        &*right_val, ::core::option::Option::None);
                }
            }
        }
    };
};debug_assert_eq!(mp.idx, matcher.len() - 1);
611                if *token == token::Eof {
612                    eof_mps.push(mp);
613                }
614            }
615        }
616    }
617
618    /// Match the token stream from `parser` against `matcher`.
619    pub(super) fn parse_tt<'matcher, T: Tracker<'matcher>>(
620        &mut self,
621        parser: &mut Cow<'_, Parser<'_>>,
622        matcher: &'matcher [MatcherLoc],
623        track: &mut T,
624    ) -> NamedParseResult {
625        // A queue of possible matcher positions. We initialize it with the matcher position in
626        // which the "dot" is before the first token of the first token tree in `matcher`.
627        // `parse_tt_inner` then processes all of these possible matcher positions and produces
628        // possible next positions into `next_mps`. After some post-processing, the contents of
629        // `next_mps` replenish `cur_mps` and we start over again.
630        self.cur_mps.clear();
631        self.cur_mps.push(MatcherPos { idx: 0, matches: Rc::clone(&self.empty_matches) });
632
633        loop {
634            self.next_mps.clear();
635            self.bb_mps.clear();
636
637            // Process `cur_mps` until either we have finished the input or we need to get some
638            // parsing from the black-box parser done.
639            let res = self.parse_tt_inner(parser, matcher, track);
640
641            if let Some(res) = res {
642                return res;
643            }
644
645            // `parse_tt_inner` handled all of `cur_mps`, so it's empty.
646            if !self.cur_mps.is_empty() {
    ::core::panicking::panic("assertion failed: self.cur_mps.is_empty()")
};assert!(self.cur_mps.is_empty());
647
648            // Error messages here could be improved with links to original rules.
649            match (self.next_mps.len(), self.bb_mps.len()) {
650                (0, 0) => {
651                    // There are no possible next positions AND we aren't waiting for the black-box
652                    // parser: syntax error.
653                    track.failure(parser);
654                    return Failure;
655                }
656
657                (_, 0) => {
658                    // Dump all possible `next_mps` into `cur_mps` for the next iteration. Then
659                    // process the next token.
660                    self.cur_mps.append(&mut self.next_mps);
661                    parser.to_mut().bump();
662                }
663
664                (0, 1) => {
665                    // We need to call the black-box parser to get some nonterminal.
666                    let mut mp = self.bb_mps.pop().unwrap();
667                    let loc = &matcher[mp.idx];
668                    let MatcherLoc::MetaVarDecl { kind, next_metavar, seq_depth, .. } = *loc else {
669                        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
670                    };
671
672                    // We use the span of the metavariable declaration to determine any
673                    // edition-specific matching behavior for non-terminals.
674                    let nt = match parser.to_mut().parse_nonterminal(kind) {
675                        Err(err) => return self.nt_parsing_error(loc, err),
676                        Ok(nt) => nt,
677                    };
678                    mp.push_match(next_metavar, seq_depth, MatchedSingle(nt));
679
680                    mp.idx += 1;
681                    self.cur_mps.push(mp);
682                }
683
684                (_, _) => {
685                    // Too many possibilities!
686                    return self.ambiguity_error(parser, matcher, track);
687                }
688            }
689
690            if !!self.cur_mps.is_empty() {
    ::core::panicking::panic("assertion failed: !self.cur_mps.is_empty()")
};assert!(!self.cur_mps.is_empty());
691        }
692    }
693
694    fn nt_parsing_error(&self, loc: &MatcherLoc, err: Diag<'_>) -> NamedParseResult {
695        let &MatcherLoc::MetaVarDecl { span, kind, .. } = loc else { ::core::panicking::panic("internal error: entered unreachable code")unreachable!() };
696        let guarantee = err
697            .with_span_label(
698                span,
699                ::alloc::__export::must_use({
        ::alloc::fmt::format(format_args!("while parsing argument for this `{0}` macro fragment",
                kind))
    })format!("while parsing argument for this `{kind}` macro fragment"),
700            )
701            .emit();
702        ErrorReported(guarantee)
703    }
704
705    fn ambiguity_error<'matcher, T: Tracker<'matcher>>(
706        &mut self,
707        parser: &Parser<'_>,
708        matcher: &'matcher [MatcherLoc],
709        track: &mut T,
710    ) -> NamedParseResult {
711        // Use a reasonable and deterministic ordering for data in the error message.
712        self.bb_mps.sort_unstable_by_key(|mp| mp.idx);
713        self.next_mps.sort_unstable_by_key(|mp| mp.idx);
714
715        let bb_locs = self.bb_mps.iter().map(|mp| &matcher[mp.idx]);
716        let next_locs = self.next_mps.iter().map(|mp| &matcher[mp.idx]);
717        track.ambiguity(parser, bb_locs, next_locs);
718        Ambiguity
719    }
720
721    fn nameize<I: Iterator<Item = NamedMatch>>(
722        &self,
723        matcher: &[MatcherLoc],
724        mut res: I,
725    ) -> NamedMatches {
726        // Make that each metavar has _exactly one_ binding. If so, insert the binding into the
727        // `NamedParseResult`. Otherwise, it's an error.
728        let mut ret_val = FxHashMap::default();
729        for loc in matcher {
730            if let &MatcherLoc::MetaVarDecl { span, bind, .. } = loc
731                && ret_val
732                    .insert(MacroRulesNormalizedIdent::new(bind), res.next().unwrap())
733                    .is_some()
734            {
735                // Duplicate binds are checked for when the macro definition is processed,
736                // and should have prevented the definition from ever being used.
737                ::rustc_middle::util::bug::span_bug_fmt(span,
    format_args!("duplicate meta-variable binding went undetected at macro definition"))span_bug!(
738                    span,
739                    "duplicate meta-variable binding went undetected at macro definition"
740                )
741            }
742        }
743        ret_val
744    }
745}