cedcee01c0
* build fuzzers with shared cargo target dir * Make external build scripts aware of CARGO_TARGET_DIR * fix libmozjpeg fuzzer with shared target dir * fix cargo-make default value for CARGO_TARGET_DIR * avoid ./ in cargo-make for windows compat * CI: cargo-hack's --feature-powerset is too powerful * fuzzer_concolic: support CARGO_TARGET_DIR * ci: install z3 to avoid building from source * ci: update actions * ci: test nightly features with nightly rust * test_all_fuzzers: try pruning more compilation artifacts * ci: fix nightly feature check * ci: apply rust-cache action after checkout (d'oh) The rust-cache action populates the checkout directory, which is promply deleted by the checkout action during checkout.. whoops!
107 lines
2.9 KiB
Rust
107 lines
2.9 KiB
Rust
// build.rs
|
|
|
|
use std::{
|
|
env,
|
|
io::{stdout, Write},
|
|
path::{Path, PathBuf},
|
|
process::exit,
|
|
};
|
|
|
|
use which::which;
|
|
|
|
fn build_dep_check(tools: &[&str]) {
|
|
for tool in tools {
|
|
println!("Checking for build tool {}...", tool);
|
|
|
|
if let Ok(path) = which(tool) {
|
|
println!("Found build tool {}", path.to_str().unwrap());
|
|
} else {
|
|
println!("ERROR: missing build tool {}", tool);
|
|
exit(1);
|
|
};
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let out_path = PathBuf::from(&env::var_os("OUT_DIR").unwrap());
|
|
|
|
println!("cargo:rerun-if-changed=harness.c");
|
|
|
|
build_dep_check(&["clang", "clang++"]);
|
|
|
|
// Enforce clang for its -fsanitize-coverage support.
|
|
std::env::set_var("CC", "clang");
|
|
std::env::set_var("CXX", "clang++");
|
|
|
|
cc::Build::new()
|
|
// Use sanitizer coverage to track the edges in the PUT
|
|
.flag("-fsanitize-coverage=trace-pc-guard,trace-cmp")
|
|
// Take advantage of LTO (needs lld-link set in your cargo config)
|
|
//.flag("-flto=thin")
|
|
.flag("-Wno-sign-compare")
|
|
.file("./harness.c")
|
|
.compile("harness");
|
|
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
&out_path.to_string_lossy()
|
|
);
|
|
|
|
let symcc_dir = clone_and_build_symcc(&out_path);
|
|
|
|
let runtime_dir = std::env::var("CARGO_TARGET_DIR")
|
|
.map(PathBuf::from)
|
|
.unwrap_or_else(|_| {
|
|
std::env::current_dir()
|
|
.unwrap()
|
|
.join("..")
|
|
.join("runtime")
|
|
.join("target")
|
|
})
|
|
.join(std::env::var("PROFILE").unwrap());
|
|
|
|
if !runtime_dir.join("libSymRuntime.so").exists() {
|
|
println!("cargo:warning=Runtime not found. Build it first.");
|
|
exit(1);
|
|
}
|
|
|
|
// SymCC.
|
|
std::env::set_var("CC", symcc_dir.join("symcc"));
|
|
std::env::set_var("CXX", symcc_dir.join("sym++"));
|
|
std::env::set_var("SYMCC_RUNTIME_DIR", runtime_dir);
|
|
|
|
println!("cargo:rerun-if-changed=harness_symcc.c");
|
|
|
|
let output = cc::Build::new()
|
|
.flag("-Wno-sign-compare")
|
|
.cargo_metadata(false)
|
|
.get_compiler()
|
|
.to_command()
|
|
.arg("./harness_symcc.c")
|
|
.args(["-o", "target_symcc.out"])
|
|
.arg("-lm")
|
|
.output()
|
|
.expect("failed to execute symcc");
|
|
if !output.status.success() {
|
|
println!("cargo:warning=Building the target with SymCC failed");
|
|
let mut stdout = stdout();
|
|
stdout
|
|
.write_all(&output.stderr)
|
|
.expect("failed to write cc error message to stdout");
|
|
exit(1);
|
|
}
|
|
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=harness.c");
|
|
println!("cargo:rerun-if-changed=harness_symcc.c");
|
|
}
|
|
|
|
fn clone_and_build_symcc(out_path: &Path) -> PathBuf {
|
|
let repo_dir = out_path.join("libafl_symcc_src");
|
|
if !repo_dir.exists() {
|
|
symcc_libafl::clone_symcc(&repo_dir);
|
|
}
|
|
|
|
symcc_libafl::build_symcc(&repo_dir)
|
|
}
|