fix: put a call limit on the toml fuzzer (#692)

pest-parser · Aug 14, 2022 · 8d1f7b8 · 8d1f7b8
1 parent 40e3752
commit 8d1f7b8
Show file tree

Hide file tree

Showing 5 changed files with 8,681 additions and 2 deletions.
diff --git a/grammars/fuzz/Cargo.toml b/grammars/fuzz/Cargo.toml
@@ -10,6 +10,8 @@ cargo-fuzz = true
 
 [dependencies.pest_grammars]
 path = ".."
+[dependencies.pest]
+path = "../../pest"
 [dependencies.libfuzzer-sys]
 git = "https://github.com/rust-fuzz/libfuzzer-sys.git"
 

diff --git a/grammars/fuzz/fuzz_targets/json.rs b/grammars/fuzz/fuzz_targets/json.rs
@@ -1,5 +1,6 @@
 #![no_main]
-#[macro_use] extern crate libfuzzer_sys;
+#[macro_use]
+extern crate libfuzzer_sys;
 extern crate pest_grammars;
 
 fuzz_target!(|data: &[u8]| {

diff --git a/grammars/fuzz/fuzz_targets/toml.rs b/grammars/fuzz/fuzz_targets/toml.rs
@@ -1,12 +1,17 @@
 #![no_main]
-#[macro_use] extern crate libfuzzer_sys;
+#[macro_use]
+extern crate libfuzzer_sys;
+extern crate pest;
 extern crate pest_grammars;
 
+use std::convert::TryInto;
+
 fuzz_target!(|data: &[u8]| {
     use pest_grammars::toml;
     use pest_grammars::Parser;
 
     if let Ok(s) = std::str::from_utf8(data) {
+        pest::set_call_limit(Some(100_000usize.try_into().unwrap()));
         let _ = toml::TomlParser::parse(toml::Rule::toml, s);
     }
 });