Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-11-10 15:14:32 +00:00
Code Issues Projects Releases Packages Wiki Activity

Improve macro limit error and move to const

Browse source
This commit is contained in:
Jonas Schievink 2020-12-10 17:50:56 +01:00
parent a6c8098113
commit 614e5a2272
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
crates/hir_expand/src/db.rs
View file

@ -13,6 +13,12 @@ use crate::{
MacroFile, ProcMacroExpander,
};
/// Total limit on the number of tokens produced by any macro invocation.
///
/// If an invocation produces more tokens than this limit, it will not be stored in the database and
/// an error will be emitted.
const TOKEN_LIMIT: usize = 262144;
#[derive(Debug, Clone, Eq, PartialEq)]
pub enum TokenExpander {
MacroRules(mbe::MacroRules),
@ -227,10 +233,10 @@ fn macro_expand_with_arg(
let ExpandResult { value: tt, err } = macro_rules.0.expand(db, lazy_id, &macro_arg.0);
// Set a hard limit for the expanded tt
let count = tt.count();
if count > 262144 {
if count > TOKEN_LIMIT {
return ExpandResult::str_err(format!(
"Total tokens count exceed limit : count = {}",
count
"macro invocation exceeds token limit: produced {} tokens, limit is {}",
count, TOKEN_LIMIT,
));
}