From c1d34f66f77c950f586d3975e280e9cd6e7b391b Mon Sep 17 00:00:00 2001
From: Renata Hodovan <reni@inf.u-szeged.hu>
Date: Wed, 25 Sep 2024 00:32:47 +0200
Subject: [PATCH] Support report directory functionality for grimoire

Report funtionality is need to make Grimoire compatible with FuzzBench.
---
 grimoire/src/lib.rs | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/grimoire/src/lib.rs b/grimoire/src/lib.rs
index 951b280..72fa97c 100644
--- a/grimoire/src/lib.rs
+++ b/grimoire/src/lib.rs
@@ -74,6 +74,12 @@ pub fn libafl_main() {
                 .long("output")
                 .help("The directory to place finds in ('corpus')"),
         )
+        .arg(
+            Arg::new("report")
+                .short('r')
+                .long("report")
+                .help("The directory to place dumped testcases ('corpus')"),
+        )
         .arg(
             Arg::new("tokens")
                 .short('x')
@@ -139,6 +145,19 @@ pub fn libafl_main() {
     crashes.push("crashes");
     out_dir.push("queue");
 
+    let report_dir = PathBuf::from(
+        res.get_one::<String>("report")
+            .expect("The --report parameter is missing")
+            .to_string(),
+    );
+    if fs::create_dir(&report_dir).is_err() {
+        println!("Report dir at {:?} already exists.", &report_dir);
+        if !report_dir.is_dir() {
+            println!("Report dir at {:?} is not a valid directory!", &report_dir);
+            return;
+        }
+    }
+
     let in_dir = PathBuf::from(
         res.get_one::<String>("in")
             .expect("The --input parameter is missing")
@@ -159,7 +178,7 @@ pub fn libafl_main() {
             .expect("Could not parse timeout in milliseconds"),
     );
 
-    fuzz(in_dir, out_dir, crashes, tokens, timeout).expect("An error occurred while fuzzing");
+    fuzz(in_dir, out_dir, crashes, report_dir, tokens, timeout).expect("An error occurred while fuzzing");
 }
 
 fn run_testcases(filenames: &[&str]) {
@@ -196,6 +215,7 @@ fn fuzz(
     in_dir: PathBuf,
     corpus_dir: PathBuf,
     objective_dir: PathBuf,
+    report_dir: PathBuf,
     tokenfile: Option<PathBuf>,
     timeout: Duration,
 ) -> Result<(), Error> {
@@ -381,7 +401,19 @@ fn fuzz(
         3,
     );
 
+    let fuzzbench = libafl::stages::DumpToDiskStage::new(
+        |input: &BytesInput, state: &StdState<_, _, _, _>| {
+            let target_bytes = input.target_bytes();
+            let bytes = target_bytes.as_slice().to_vec();
+            bytes
+        },
+        &report_dir.join("queue"),
+        &report_dir.join("crashes"),
+    )
+    .unwrap();
+
     let mut stages = tuple_list!(
+        fuzzbench,
         generalization,
         tracing,
         i2s,