diff options
Diffstat (limited to 'include/afl-fuzz.h')
| -rw-r--r-- | include/afl-fuzz.h | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/include/afl-fuzz.h b/include/afl-fuzz.h index 4912b3f0..e45a9d84 100644 --- a/include/afl-fuzz.h +++ b/include/afl-fuzz.h @@ -78,8 +78,15 @@ /* For systems that have sched_setaffinity; right now just Linux, but one can hope... */ -#ifdef __linux__ +#if defined(__linux__) || defined(__FreeBSD__) #define HAVE_AFFINITY 1 +#if defined(__FreeBSD__) +#include <sys/cpuset.h> +#include <sys/user.h> +#include <pthread.h> +#include <pthread_np.h> +#define cpu_set_t cpuset_t +#endif #endif /* __linux__ */ #ifndef SIMPLE_FILES @@ -683,8 +690,8 @@ static u64 get_cur_time_us(void) { } #ifdef _AFL_DOCUMENT_MUTATIONS - extern u8 do_document; - extern u32 document_counter; +extern u8 do_document; +extern u32 document_counter; #endif #endif |