Make llvm pass build feature dependent (#2357)

* push

* everybody is default

* fix

* fix

libafl_cc/Cargo.toml

@@ -14,7 +14,18 @@ categories = ["development-tools::testing", "emulators", "embedded", "os", "no-s
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [features]
-default = []
+default = ["ddg-instr", "function-logging", "cmplog-routines", "autotokens", "coverage-accounting", "cmplog-instructions", "ctx", "dump-cfg", "profiling"]
+
+# llvm passes
+ddg-instr = []
+function-logging = []
+cmplog-routines = []
+autotokens = []
+coverage-accounting = []
+cmplog-instructions = []
+ctx = []
+dump-cfg = []
+profiling = []
 
 [build-dependencies]
 cc = { version = "1.0", features = ["parallel"] }

libafl_cc/build.rs

@@ -144,6 +144,7 @@ fn find_llvm_version() -> Option<i32> {
 }
 
 #[allow(clippy::too_many_arguments)]
+#[allow(unused)]
 fn build_pass(
     bindir_path: &Path,
     out_dir: &Path,
@@ -199,7 +200,8 @@ fn build_pass(
     };
 
     match command_result {
-        Some(res) => match res {
+        Some(res) => {
+            match res {
                 Ok(s) => {
                     if !s.success() {
                         if required {
@@ -211,12 +213,13 @@ fn build_pass(
                 }
                 Err(err) => {
                     if required {
-                    panic!("Failed to compile required compiler pass src/{src_file} - {err}");
+                        panic!("Failed to compile required compiler pass src/{src_file} - Error: {err}");
                     } else {
-                    println!("cargo:warning=Skipping non-required compiler pass src/{src_file} - Reason: {err}");
+                        println!("cargo:warning=Skipping non-required compiler pass src/{src_file} - Error: {err}");
+                    }
+                }
             }
         }
-        },
         None => {
             println!("cargo:warning=Skipping compiler pass src/{src_file} - Only supported on Windows or *nix.");
         }
@@ -416,6 +419,7 @@ pub const LIBAFL_CC_LLVM_VERSION: Option<usize> = None;
         ldflags.push(&sdk_path);
     };
 
+    #[cfg(feature = "ddg-instr")]
     build_pass(
         bindir_path,
         out_dir,
@@ -424,42 +428,104 @@ pub const LIBAFL_CC_LLVM_VERSION: Option<usize> = None;
         src_dir,
         "ddg-instr.cc",
         Some(&vec!["ddg-utils.cc"]),
-        false,
+        true,
     );
 
-    for pass in [
-        "function-logging.cc",
-        "cmplog-routines-pass.cc",
-        "autotokens-pass.cc",
-        "coverage-accounting-pass.cc",
-        "cmplog-instructions-pass.cc",
-        "ctx-pass.cc",
-    ] {
+    #[cfg(feature = "function-logging")]
     build_pass(
         bindir_path,
         out_dir,
         &cxxflags,
         &ldflags,
         src_dir,
-            pass,
+        "function-logging.cc",
         None,
         true,
     );
-    }
 
-    // Optional pass
-    for pass in ["dump-cfg-pass.cc", "profiling.cc"] {
+    #[cfg(feature = "cmplog-routines")]
     build_pass(
         bindir_path,
         out_dir,
         &cxxflags,
         &ldflags,
         src_dir,
-            pass,
+        "cmplog-routines-pass.cc",
+        None,
+        true,
+    );
+
+    #[cfg(feature = "autotokens")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "autotokens-pass.cc",
+        None,
+        true,
+    );
+
+    #[cfg(feature = "coverage-accounting")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "coverage-accounting-pass.cc",
+        None,
+        true,
+    );
+
+    #[cfg(feature = "cmplog-instructions")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "cmplog-instructions-pass.cc",
+        None,
+        true,
+    );
+
+    #[cfg(feature = "ctx")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "ctx-pass.cc",
+        None,
+        true,
+    );
+
+    #[cfg(feature = "dump-cfg")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "dump-cfg-pass.cc",
+        None,
+        false,
+    );
+
+    #[cfg(feature = "profiling")]
+    build_pass(
+        bindir_path,
+        out_dir,
+        &cxxflags,
+        &ldflags,
+        src_dir,
+        "profiling-pass.cc",
         None,
         false,
     );
-    }
 
     cc::Build::new()
         .file(src_dir.join("no-link-rt.c"))