From 5cf346885fda2546cba512fef0dfbb68c27d1e99 Mon Sep 17 00:00:00 2001
From: Paul Loyd <pavelko95@gmail.com>
Date: Sun, 13 Feb 2022 17:24:47 +0300
Subject: [PATCH 1/2] fix: `clear()` must clear, not toggle

---
 src/cpuset.rs | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/src/cpuset.rs b/src/cpuset.rs
index 26ab1d5..de0f089 100644
--- a/src/cpuset.rs
+++ b/src/cpuset.rs
@@ -1,9 +1,9 @@
 //! A CPU bitmask implementation to be used with the sched_[gs]etaffinity functions.
 
 use libc::{c_void, cpu_set_t, sched_getaffinity, sched_setaffinity};
+use std::cmp;
 use std::mem;
 use std::ptr;
-use std::cmp;
 
 type Mask = u64;
 const MASK_BITS: usize = 64;
@@ -59,7 +59,7 @@ impl CpuSet {
         let elem = cpu / MASK_BITS;
         let bit = cpu % MASK_BITS;
         if elem < self.mask.len() {
-            self.mask[elem] ^= 1 << bit;
+            self.mask[elem] &= !(1 << bit);
         }
     }
 
@@ -104,9 +104,11 @@ impl CpuSet {
         } else {
             let mut mask: u64 = 0;
             unsafe {
-                ptr::copy(self.mask_ptr(),
-                          (&mut mask) as *mut _ as *mut c_void,
-                          src_size)
+                ptr::copy(
+                    self.mask_ptr(),
+                    (&mut mask) as *mut _ as *mut c_void,
+                    src_size,
+                )
             }
             Ok(mask)
         }
@@ -134,8 +136,8 @@ impl CpuSet {
 
 #[cfg(test)]
 mod tests {
-    use std::ops::BitXor;
     use super::{CpuSet, MASK_BITS};
+    use std::ops::BitXor;
 
     #[test]
     fn test_new_one_byte() {
@@ -203,6 +205,16 @@ mod tests {
         }
     }
 
+    #[test]
+    fn test_clear() {
+        let mut cpuset = CpuSet::from_mask(0b01010101);
+        cpuset.clear(0);
+        cpuset.clear(1);
+        cpuset.clear(2);
+        cpuset.clear(3);
+        assert_eq!(cpuset.as_u64().unwrap(), 0b01010000);
+    }
+
     #[test]
     fn test_clear_and_is_set() {
         let mut cpuset = CpuSet::from_mask(::std::u64::MAX);

From 2e77da8d543b0703e2bf187ba0aeed9e0e37141b Mon Sep 17 00:00:00 2001
From: Ilya Titkov <ilya@titkov.me>
Date: Wed, 1 Feb 2023 17:21:54 +0600
Subject: [PATCH 2/2] fix: off-by-one error in set

---
 src/cpuset.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cpuset.rs b/src/cpuset.rs
index de0f089..95dbee0 100644
--- a/src/cpuset.rs
+++ b/src/cpuset.rs
@@ -46,7 +46,7 @@ impl CpuSet {
     pub fn set(&mut self, cpu: usize) {
         let elem = cpu / MASK_BITS;
         let bit = cpu % MASK_BITS;
-        while elem > self.mask.len() {
+        while elem >= self.mask.len() {
             self.mask.push(0);
         }
         self.mask[elem] |= 1 << bit;