Set AFL_NO_COLOR when stdout is not a terminal

trailofbits · Feb 19, 2024 · 1713ba8 · 1713ba8
1 parent 9844f85
commit 1713ba8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cargo-test-fuzz/src/lib.rs b/cargo-test-fuzz/src/lib.rs
@@ -25,7 +25,7 @@ use std::{
     ffi::OsStr,
     fmt::{Debug, Formatter},
     fs::{create_dir_all, read, read_dir, remove_dir_all, File},
-    io::{BufRead, Read},
+    io::{stdout, BufRead, IsTerminal, Read},
     iter,
     path::{Path, PathBuf},
     process::{exit, Command},
@@ -908,6 +908,9 @@ fn fuzz(opts: &TestFuzz, executable: &Executable, target: &str) -> Result<()> {
     create_dir_all(&output_dir).unwrap_or_default();
 
     let mut envs = BASE_ENVS.to_vec();
+    if !stdout().is_terminal() {
+        envs.push(("AFL_NO_COLOR", "1"));
+    }
     if opts.no_ui {
         envs.push(("AFL_NO_UI", "1"));
     }