[fix] improve dealloc_pages

arceos-org · Dec 12, 2024 · acd8c18 · acd8c18
1 parent 444d7ee
commit acd8c18
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/bitmap.rs b/src/bitmap.rs
@@ -135,11 +135,13 @@ impl<const PAGE_SIZE: usize> PageAllocator for BitmapPageAllocator<PAGE_SIZE> {
             crate::is_aligned(pos, Self::PAGE_SIZE),
             "pos must be aligned to PAGE_SIZE"
         );
-
-        if self
-            .inner
-            .dealloc_contiguous((pos - self.base) / PAGE_SIZE, num_pages)
-        {
+        if match num_pages.cmp(&1) {
+            core::cmp::Ordering::Equal => self.inner.dealloc((pos - self.base) / PAGE_SIZE),
+            core::cmp::Ordering::Greater => self
+                .inner
+                .dealloc_contiguous((pos - self.base) / PAGE_SIZE, num_pages),
+            _ => false,
+        } {
             self.used_pages -= num_pages;
         }
     }