9aa40c0734
* prep for publishing libafl_libfuzzer * learn to use linkers * document-features * special handling for fuzzbench builds * Update cmplog.c * drop dep for llvm-tools; add testcase for memcmp sanity --------- Co-authored-by: Dongjia "toka" Zhang <tokazerkje@outlook.com>
76 lines
2.3 KiB
Rust
76 lines
2.3 KiB
Rust
use std::{path::PathBuf, process::Command};
|
|
|
|
fn main() {
|
|
if cfg!(any(feature = "cargo-clippy", docsrs)) {
|
|
return; // skip when clippy or docs is running
|
|
}
|
|
if cfg!(not(target_os = "linux")) {
|
|
println!(
|
|
"cargo:error=The libafl_libfuzzer runtime may only be built for linux; failing fast."
|
|
);
|
|
return;
|
|
}
|
|
println!("cargo:rerun-if-changed=libafl_libfuzzer_runtime/src");
|
|
println!("cargo:rerun-if-changed=libafl_libfuzzer_runtime/Cargo.toml");
|
|
println!("cargo:rerun-if-changed=libafl_libfuzzer_runtime/build.rs");
|
|
|
|
let custom_lib_dir =
|
|
PathBuf::from(std::env::var_os("OUT_DIR").unwrap()).join("libafl_libfuzzer");
|
|
std::fs::create_dir_all(&custom_lib_dir)
|
|
.expect("Couldn't create the output directory for the fuzzer runtime build");
|
|
|
|
let mut lib_src = PathBuf::from(std::env::var_os("CARGO_MANIFEST_DIR").unwrap());
|
|
lib_src.push("libafl_libfuzzer_runtime");
|
|
|
|
let mut command = Command::new(std::env::var_os("CARGO").unwrap());
|
|
command
|
|
.env_remove("RUSTFLAGS")
|
|
.env_remove("CARGO_ENCODED_RUSTFLAGS");
|
|
|
|
for (var, _) in std::env::vars() {
|
|
if var.starts_with("CARGO_PKG_") || var.starts_with("CARGO_FEATURE_") {
|
|
command.env_remove(var);
|
|
}
|
|
}
|
|
|
|
command
|
|
.env("PATH", std::env::var_os("PATH").unwrap())
|
|
.current_dir(&lib_src);
|
|
|
|
command.arg("build");
|
|
|
|
let mut features = vec!["serdeany_autoreg"];
|
|
|
|
if cfg!(any(feature = "fork")) {
|
|
features.push("fork");
|
|
}
|
|
if cfg!(any(feature = "introspection")) {
|
|
features.push("libafl/introspection");
|
|
}
|
|
|
|
command
|
|
.arg("--release")
|
|
.arg("--no-default-features")
|
|
.arg("--target-dir")
|
|
.arg(&custom_lib_dir)
|
|
.arg("--features")
|
|
.arg(features.join(","))
|
|
.arg("--target")
|
|
.arg(std::env::var_os("TARGET").unwrap());
|
|
|
|
assert!(
|
|
!command.status().map(|s| !s.success()).unwrap_or(true),
|
|
"Couldn't build runtime crate! Did you remember to use nightly?"
|
|
);
|
|
|
|
let mut lib_path = custom_lib_dir.join(std::env::var_os("TARGET").unwrap());
|
|
lib_path.push("release");
|
|
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
lib_path.to_str().unwrap()
|
|
);
|
|
println!("cargo:rustc-link-lib=static=afl_libfuzzer_runtime");
|
|
println!("cargo:rustc-link-lib=stdc++");
|
|
}
|