included #68 cpu binding for FreeBSD, thanks to devnexen

1 files changed, 15 insertions, 2 deletions

diff --git a/src/afl-gotcpu.c b/src/afl-gotcpu.c
index de41177c..c3a60c6c 100644
--- a/src/afl-gotcpu.c
+++ b/src/afl-gotcpu.c
@@ -52,9 +52,15 @@
 #include "types.h"
 #include "debug.h"
 
-#ifdef __linux__
+#if defined(__linux__) || defined(__FreeBSD__)
 #define HAVE_AFFINITY 1
-#endif                                                         /* __linux__ */
+# if defined(__FreeBSD__)
+#   include <pthread.h>
+#   include <pthread_np.h>
+#   include <sys/cpuset.h>
+#   define cpu_set_t cpuset_t
+# endif
+#endif                                      /* __linux__ || __FreeBSD__ */
 
 /* Get unix time in microseconds. */
 
@@ -160,8 +166,15 @@ int main(int argc, char** argv) {
       CPU_ZERO(&c);
       CPU_SET(i, &c);
 
+#if defined(__FreeBSD__)
+      if (pthread_setaffinity_np(pthread_self(), sizeof(c), &c))
+        PFATAL("pthread_setaffinity_np failed");
+#endif
+
+#if defined(__linux__)
       if (sched_setaffinity(0, sizeof(c), &c))
         PFATAL("sched_setaffinity failed");
+#endif
 
       util_perc = measure_preemption(CTEST_CORE_TRG_MS);