Add support for multiple AFL flag arguments

srlabs · Jan 31, 2024 · 0fdca67 · 0fdca67
1 parent 1793f05
commit 0fdca67
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/bin/cargo-ziggy/fuzz.rs b/src/bin/cargo-ziggy/fuzz.rs
@@ -365,12 +365,12 @@ impl Fuzz {
                                 mutation_option,
                                 &timeout_option_afl,
                                 &dictionary_option,
-                                &self.afl_flags.clone().unwrap_or_default(),
-                                &format!("./target/afl/debug/{}", self.target),
                             ]
                             .iter()
                             .filter(|a| a != &&""),
                         )
+                        .args(self.afl_flags.clone())
+                        .arg(format!("./target/afl/debug/{}", self.target))
                         .env("AFL_AUTORESUME", "1")
                         .env("AFL_TESTCACHE_SIZE", "100")
                         .env("AFL_FAST_CAL", "1")

diff --git a/src/bin/cargo-ziggy/main.rs b/src/bin/cargo-ziggy/main.rs
@@ -151,7 +151,7 @@ pub struct Fuzz {
 
     /// Pass flags to AFL++ directly
     #[clap(short, long)]
-    afl_flags: Option<String>,
+    afl_flags: Vec<String>,
 }
 
 #[derive(Args)]