From 8a4bbe3d0b87cccfd5cef4d47764a0dbda43c242 Mon Sep 17 00:00:00 2001
From: Dominik Maier <dmnk@google.com>
Date: Thu, 6 Oct 2022 17:43:59 +0200
Subject: [PATCH] Fix FreeBSD CI (#820)

* Fix FreeBSD CI

* Fixed tests for freebsd
---
 libafl/src/bolts/core_affinity.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/libafl/src/bolts/core_affinity.rs b/libafl/src/bolts/core_affinity.rs
index 5a43c04a40..bb66bd5b88 100644
--- a/libafl/src/bolts/core_affinity.rs
+++ b/libafl/src/bolts/core_affinity.rs
@@ -618,7 +618,7 @@ mod freebsd {
     use alloc::vec::Vec;
     use std::{mem, thread::available_parallelism};
 
-    use libc::{cpuset_getaffinity, cpuset_setaffinity, cpuset_t, CPU_ISSET, CPU_SET};
+    use libc::{cpuset_getaffinity, cpuset_setaffinity, cpuset_t, CPU_SET};
 
     use super::CoreId;
     use crate::Error;
@@ -658,6 +658,7 @@ mod freebsd {
         }
     }
 
+    #[cfg(test)]
     fn get_affinity_mask() -> Result<cpuset_t, Error> {
         let mut set = new_cpuset();
 
@@ -688,6 +689,7 @@ mod freebsd {
     #[cfg(test)]
     mod tests {
         use super::*;
+        use libc::CPU_ISSET;
 
         #[test]
         fn test_freebsd_get_affinity_mask() {
@@ -711,7 +713,7 @@ mod freebsd {
 
             let mut is_equal = true;
 
-            for i in 0..256 as usize {
+            for i in 0..256 {
                 let is_set1 = unsafe { CPU_ISSET(i, &core_mask) };
                 let is_set2 = unsafe { CPU_ISSET(i, &new_mask) };