create ctrl-c channel only if fuzzin

srlabs · Jun 20, 2024 · 0ec6442 · 0ec6442
1 parent d34df09
commit 0ec6442
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/bin/cargo-ziggy/main.rs b/src/bin/cargo-ziggy/main.rs
@@ -313,12 +313,14 @@ fn main() -> Result<(), anyhow::Error> {
 
  env_logger::init();
  let Cargo::Ziggy(command) = Cargo::parse();
- let (tx, rx) = channel();
- ctrlc::set_handler(move || tx.send(()).expect("Could not send signal on channel."))
- .expect("Error setting Ctrl-C handler");
  match command {
  Ziggy::Build(args) => args.build().context("Failed to build the fuzzers"),
- Ziggy::Fuzz(mut args) => args.fuzz(rx).context("Failure running fuzzers"),
+ Ziggy::Fuzz(mut args) => {
+ let (tx, rx) = channel();
+ ctrlc::set_handler(move || tx.send(()).expect("Could not send signal on channel."))
+ .expect("Error setting Ctrl-C handler");
+ args.fuzz(rx).context("Failure running fuzzers")
+ }
  Ziggy::Run(mut args) => args.run().context("Failure running inputs"),
  Ziggy::Minimize(mut args) => args.minimize().context("Failure running minimization"),
  Ziggy::Cover(mut args) => args