From 614e5a2272a96eabe30acea2f7c797a305a4f2fd Mon Sep 17 00:00:00 2001 From: Jonas Schievink Date: Thu, 10 Dec 2020 17:50:56 +0100 Subject: Improve macro limit error and move to const --- crates/hir_expand/src/db.rs | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'crates/hir_expand/src/db.rs') diff --git a/crates/hir_expand/src/db.rs b/crates/hir_expand/src/db.rs index 842a177db..3c0ee284e 100644 --- a/crates/hir_expand/src/db.rs +++ b/crates/hir_expand/src/db.rs @@ -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, ¯o_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, )); } -- cgit v1.2.3