FreeBSD, implementing binding to free cpu.

Makefile

@@ -36,6 +36,10 @@ ifneq "$(filter Linux GNU%,$(shell uname))" ""
   LDFLAGS  += -ldl
 endif
 
+ifneq "$(filter FreeBSD GNU%,$(shell uname))" ""
+  LDFLAGS  += -pthread
+endif
+
 ifeq "$(findstring clang, $(shell $(CC) --version 2>/dev/null))" ""
   TEST_CC   = afl-gcc
 else

afl-fuzz.c

@@ -68,12 +68,18 @@
 
 #if defined(__APPLE__) || defined(__FreeBSD__) || defined (__OpenBSD__)
 #  include <sys/sysctl.h>
+#  ifdef __FreeBSD__
+#    include <sys/user.h>
+#    include <sys/cpuset.h>
+#    include <pthread.h>
+#    include <pthread_np.h>
+#  endif
 #endif /* __APPLE__ || __FreeBSD__ || __OpenBSD__ */
 
 /* 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
 #endif /* __linux__ */
 
@@ -405,14 +411,13 @@ static void shuffle_ptrs(void** ptrs, u32 cnt) {
    can be found. Assumes an upper bound of 4k CPUs. */
 
 static void bind_to_free_cpu(void) {
-
+  u8 cpu_used[4096] = { 0 };
+  u32 i;
+#ifdef __linux__
   DIR* d;
   struct dirent* de;
   cpu_set_t c;
 
-  u8 cpu_used[4096] = { 0 };
-  u32 i;
-
   if (cpu_core_count < 2) return;
 
   if (getenv("AFL_NO_AFFINITY")) {
@@ -485,6 +490,26 @@ static void bind_to_free_cpu(void) {
   }
 
   closedir(d);
+#else
+  struct kinfo_proc *procs;
+  size_t nprocs;
+  size_t proccount;
+  cpuset_t c;
+  int s_name[3] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL};
+  if (sysctl(s_name, 3, NULL, &nprocs, NULL, 0) < 0) return;
+  proccount = nprocs / sizeof(*procs);
+  procs = ck_alloc(nprocs);
+  if (sysctl(s_name, 3, NULL, &nprocs, NULL, 0) < 0) goto procs_free;
+
+  for (i = 0; i < proccount; i ++) {
+    if (procs[i].ki_oncpu < sizeof(cpu_used))
+      cpu_used[procs[i].ki_oncpu] = 1;
+  }
+
+procs_free:
+  ck_free(procs);
+
+#endif
 
   for (i = 0; i < cpu_core_count; i++) if (!cpu_used[i]) break;
 
@@ -508,8 +533,13 @@ static void bind_to_free_cpu(void) {
   CPU_ZERO(&c);
   CPU_SET(i, &c);
 
+#ifdef __linux__
   if (sched_setaffinity(0, sizeof(c), &c))
     PFATAL("sched_setaffinity failed");
+#else
+  if (pthread_setaffinity_np(pthread_self(), sizeof(c), &c))
+    PFATAL("pthread_setaffinity failed");
+#endif
 
 }