refactor(fuzz): Reject invalid UTF-8 data

toml-rs · Jul 15, 2024 · c0a772c · c0a772c
1 parent f30de96
commit c0a772c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/crates/toml_edit_fuzz/parse_document.rs b/crates/toml_edit_fuzz/parse_document.rs
@@ -2,17 +2,17 @@
 
 use toml_edit::DocumentMut;
 
-libfuzzer_sys::fuzz_target!(|data| {
+libfuzzer_sys::fuzz_target!(|data: &[u8]| -> libfuzzer_sys::Corpus {
     let Ok(data) = std::str::from_utf8(data) else {
-        return;
+        return libfuzzer_sys::Corpus::Reject;
     };
 
     println!("parsing: {data:?}");
     let doc = match data.parse::<DocumentMut>() {
         Ok(doc) => doc,
         Err(err) => {
             println!("{err}");
-            return;
+            return libfuzzer_sys::Corpus::Keep;
         }
     };
     let toml = doc.to_string();
@@ -26,4 +26,5 @@ libfuzzer_sys::fuzz_target!(|data| {
     );
     let doc = doc.unwrap();
     assert_eq!(doc.to_string(), toml);
+    libfuzzer_sys::Corpus::Keep
 });