Skip to content

Commit

Permalink
[RFC-3086] Restrict the parsing of count
Browse files Browse the repository at this point in the history
  • Loading branch information
c410-f3r committed Aug 17, 2023
1 parent d4a881e commit 6395dc2
Show file tree
Hide file tree
Showing 3 changed files with 44 additions and 1 deletion.
12 changes: 11 additions & 1 deletion compiler/rustc_expand/src/mbe/metavar_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,17 @@ fn parse_count<'sess>(
span: Span,
) -> PResult<'sess, MetaVarExpr> {
let ident = parse_ident(iter, sess, span)?;
let depth = if try_eat_comma(iter) { Some(parse_depth(iter, sess, span)?) } else { None };
let depth = if try_eat_comma(iter) {
if iter.look_ahead(0).is_none() {
return Err(sess.span_diagnostic.struct_span_err(
span,
"`count` followed by a comma must have an associated index indicating its depth",
));
}
Some(parse_depth(iter, sess, span)?)
} else {
None
};
Ok(MetaVarExpr::Count(ident, depth))
}

Expand Down
14 changes: 14 additions & 0 deletions tests/ui/macros/rfc-3086-metavar-expr/issue-111904.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
#![feature(macro_metavar_expr)]

macro_rules! foo {
( $( $($t:ident),* );* ) => { ${count(t,)} }
//~^ ERROR `count` followed by a comma must have an associated
//~| ERROR expected expression, found `$`
}

fn test() {
foo!(a, a; b, b);
}

fn main() {
}
19 changes: 19 additions & 0 deletions tests/ui/macros/rfc-3086-metavar-expr/issue-111904.stderr
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
error: `count` followed by a comma must have an associated index indicating its depth
--> $DIR/issue-111904.rs:4:37
|
LL | ( $( $($t:ident),* );* ) => { ${count(t,)} }
| ^^^^^

error: expected expression, found `$`
--> $DIR/issue-111904.rs:4:35
|
LL | ( $( $($t:ident),* );* ) => { ${count(t,)} }
| ^ expected expression
...
LL | foo!(a, a; b, b);
| ---------------- in this macro invocation
|
= note: this error originates in the macro `foo` (in Nightly builds, run with -Z macro-backtrace for more info)

error: aborting due to 2 previous errors

0 comments on commit 6395dc2

Please sign in to comment.