about summary refs log tree commit diff
path: root/src/afl-gotcpu.c
diff options
context:
space:
mode:
authorhexcoder <hexcoder-@users.noreply.github.com>2019-10-26 10:45:20 +0200
committerGitHub <noreply@github.com>2019-10-26 10:45:20 +0200
commit17729ce856058845314d1a3cb08e7a28d943db29 (patch)
tree372c3292fb52180e4395cae9cbb9f90180093e3c /src/afl-gotcpu.c
parenta1f7de2bc36d23c758bee88a9ce0a90f5d8f2057 (diff)
parentcfd7b906cbc961892646aee9f307f6ffd4eb2253 (diff)
downloadafl++-17729ce856058845314d1a3cb08e7a28d943db29.tar.gz
Merge pull request #89 from devnexen/dfbsd_porting
Porting cpu affinity to DragonFly.
Diffstat (limited to 'src/afl-gotcpu.c')
-rw-r--r--src/afl-gotcpu.c10
1 files changed, 6 insertions, 4 deletions
diff --git a/src/afl-gotcpu.c b/src/afl-gotcpu.c
index 850943dd..fe5d035f 100644
--- a/src/afl-gotcpu.c
+++ b/src/afl-gotcpu.c
@@ -52,12 +52,14 @@
 #include "types.h"
 #include "debug.h"
 
-#if defined(__linux__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__APPLE__)
+#if defined(__linux__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__APPLE__) || defined(__DragonFly__)
 #define HAVE_AFFINITY 1
-#if defined(__FreeBSD__)
+#if defined(__FreeBSD__) || defined(__DragonFly__)
 #include <pthread.h>
 #include <pthread_np.h>
+#if defined(__FreeBSD__)
 #include <sys/cpuset.h>
+#endif
 #define cpu_set_t cpuset_t
 #elif defined(__NetBSD__)
 #include <pthread.h>
@@ -168,7 +170,7 @@ int main(int argc, char** argv) {
     if (!fr) {
 
       u32 util_perc;
-#if defined(__linux__) || defined(__FreeBSD__)
+#if defined(__linux__) || defined(__FreeBSD__) || defined(__DragonFly__)
       cpu_set_t c;
 
       CPU_ZERO(&c);
@@ -188,7 +190,7 @@ int main(int argc, char** argv) {
 	PFATAL("thread_policy_set failed");
 #endif
 
-#if defined(__FreeBSD__)
+#if defined(__FreeBSD__) || defined(__DragonFly__)
       if (pthread_setaffinity_np(pthread_self(), sizeof(c), &c))
         PFATAL("pthread_setaffinity_np failed");
 #endif