ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-13 18:48:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix cfg in macro_rules in rust custom mutator

Browse Source
This commit is contained in:
Andrea Fioraldi
2022-01-26 21:44:07 +01:00
parent f75535807a
commit 08c39c1552
1 changed files with 31 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
custom_mutators/rust/custom_mutator/src/lib.rs
View File

@ -358,6 +358,36 @@ pub mod wrappers {
} }
} }
/// An exported macro to defined afl_custom_init meant for insternal usage
#[cfg(feature = "afl_internals")]
#[macro_export]
macro_rules! _define_afl_custom_init {
($mutator_type:ty) => {
#[no_mangle]
pub extern "C" fn afl_custom_init(
afl: ::std::option::Option<&'static $crate::afl_state>,
seed: ::std::os::raw::c_uint,
) -> *const ::std::os::raw::c_void {
$crate::wrappers::afl_custom_init_::<$mutator_type>(afl, seed as u32)
}
}
}
/// An exported macro to defined afl_custom_init meant for insternal usage
#[cfg(not(feature = "afl_internals"))]
#[macro_export]
macro_rules! _define_afl_custom_init {
($mutator_type:ty) => {
#[no_mangle]
pub extern "C" fn afl_custom_init(
_afl: *const ::std::os::raw::c_void,
seed: ::std::os::raw::c_uint,
) -> *const ::std::os::raw::c_void {
$crate::wrappers::afl_custom_init_::<$mutator_type>(seed as u32)
}
}
}
/// exports the given Mutator as a custom mutator as the C interface that AFL++ expects. /// exports the given Mutator as a custom mutator as the C interface that AFL++ expects.
/// It is not possible to call this macro multiple times, because it would define the custom mutator symbols multiple times. /// It is not possible to call this macro multiple times, because it would define the custom mutator symbols multiple times.
/// # Example /// # Example
@ -381,23 +411,7 @@ pub mod wrappers {
#[macro_export] #[macro_export]
macro_rules! export_mutator { macro_rules! export_mutator {
($mutator_type:ty) => { ($mutator_type:ty) => {
#[cfg(feature = "afl_internals")] $crate::_define_afl_custom_init!($mutator_type);
#[no_mangle]
pub extern "C" fn afl_custom_init(
afl: ::std::option::Option<&'static $crate::afl_state>,
seed: ::std::os::raw::c_uint,
) -> *const ::std::os::raw::c_void {
$crate::wrappers::afl_custom_init_::<$mutator_type>(afl, seed as u32)
}
#[cfg(not(feature = "afl_internals"))]
#[no_mangle]
pub extern "C" fn afl_custom_init(
_afl: *const ::std::os::raw::c_void,
seed: ::std::os::raw::c_uint,
) -> *const ::std::os::raw::c_void {
$crate::wrappers::afl_custom_init_::<$mutator_type>(seed as u32)
}
#[no_mangle] #[no_mangle]
pub unsafe extern "C" fn afl_custom_fuzz_count( pub unsafe extern "C" fn afl_custom_fuzz_count(