SYS-OSS/FRET-LibAFL
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow multiple source file in libafl_cc (#1296)

Browse Source 
* add

* clp

* fi

* fmt
This commit is contained in:
Dongjia "toka" Zhang 2023-05-26 11:35:04 +02:00 committed by GitHub
parent 0336eae908
commit 8445ae54b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
libafl_cc/build.rs
View File

@ -1,6 +1,11 @@
#[cfg(target_vendor = "apple")] use std::{
use std::path::PathBuf; env,
use std::{env, fs::File, io::Write, path::Path, process::Command, str}; fs::File,
io::Write,
path::{Path, PathBuf},
process::Command,
str,
};
#[cfg(target_vendor = "apple")] #[cfg(target_vendor = "apple")]
use glob::glob; use glob::glob;
@ -125,6 +130,7 @@ fn find_llvm_version() -> Option<i32> {
None None
} }
#[allow(clippy::too_many_arguments)]
fn build_pass( fn build_pass(
bindir_path: &Path, bindir_path: &Path,
out_dir: &Path, out_dir: &Path,
@ -132,17 +138,25 @@ fn build_pass(
ldflags: &Vec<&str>, ldflags: &Vec<&str>,
src_dir: &Path, src_dir: &Path,
src_file: &str, src_file: &str,
additional_srcfiles: Option<&Vec<&str>>,
optional: bool, optional: bool,
) { ) {
let dot_offset = src_file.rfind('.').unwrap(); let dot_offset = src_file.rfind('.').unwrap();
let src_stub = &src_file[..dot_offset]; let src_stub = &src_file[..dot_offset];
let additionals = if let Some(x) = additional_srcfiles {
x.iter().map(|f| src_dir.join(f)).collect::<Vec<PathBuf>>()
} else {
Vec::new()
};
println!("cargo:rerun-if-changed=src/{src_file}"); println!("cargo:rerun-if-changed=src/{src_file}");
let r = if cfg!(unix) { let r = if cfg!(unix) {
let r = Command::new(bindir_path.join("clang++")) let r = Command::new(bindir_path.join("clang++"))
.arg("-v") .arg("-v")
.args(cxxflags) .args(cxxflags)
.arg(src_dir.join(src_file)) .arg(src_dir.join(src_file))
.args(additionals)
.args(ldflags) .args(ldflags)
.arg("-o") .arg("-o")
.arg(out_dir.join(format!("{src_stub}.{}", dll_extension()))) .arg(out_dir.join(format!("{src_stub}.{}", dll_extension())))
@ -154,6 +168,7 @@ fn build_pass(
.arg("-v") .arg("-v")
.args(cxxflags) .args(cxxflags)
.arg(src_dir.join(src_file)) .arg(src_dir.join(src_file))
.args(additionals)
.arg("/link") .arg("/link")
.args(ldflags) .args(ldflags)
.arg(format!( .arg(format!(
@ -359,6 +374,7 @@ pub const LIBAFL_CC_LLVM_VERSION: Option<usize> = None;
&ldflags, &ldflags,
src_dir, src_dir,
pass, pass,
None,
false, false,
); );
} }
@ -372,6 +388,7 @@ pub const LIBAFL_CC_LLVM_VERSION: Option<usize> = None;
&ldflags, &ldflags,
src_dir, src_dir,
pass, pass,
None,
true, true,
); );
} }