fix: fix the error for talc allocator

subsystems/mem/src/lib.rs

@@ -11,6 +11,8 @@ mod frame;
 #[cfg(feature = "buddy")]
 mod heap;
 mod manager;
+#[cfg(feature = "talloc")]
+mod talc_wrapper;
 mod vmm;
 
 pub use frame::{alloc_frame_trackers, alloc_frames, free_frames, FrameTracker, VmmPageAllocator};
@@ -23,12 +25,7 @@ static HEAP_ALLOCATOR: heap::HeapAllocator = heap::HeapAllocator::new();
 
 #[cfg(feature = "talc")]
 #[global_allocator]
-static HEAP_ALLOCATOR: talc::Talck<ksync::Mutex<()>, talc::ClaimOnOom> = talc::Talc::new(unsafe {
-    talc::ClaimOnOom::new(talc::Span::from_const_array(core::ptr::addr_of!(
-        KERNEL_HEAP
-    )))
-})
-.lock();
+static HEAP_ALLOCATOR: talc_wrapper::TalcAllocator = talc_wrapper::TalcAllocator;
 
 #[cfg(any(feature = "talloc", feature = "buddy"))]
 static mut KERNEL_HEAP: [u8; HEAP_SIZE] = [0; HEAP_SIZE];

subsystems/mem/src/talc_wrapper.rs

@@ -0,0 +1,37 @@
+use core::alloc::GlobalAlloc;
+
+use config::FRAME_SIZE;
+use log::trace;
+use talc::{ClaimOnOom, Talc, Talck};
+
+use crate::{alloc_frames, free_frames, KERNEL_HEAP};
+
+static HEAP_ALLOCATOR: Talck<ksync::Mutex<()>, ClaimOnOom> = Talc::new(unsafe {
+    ClaimOnOom::new(talc::Span::from_const_array(core::ptr::addr_of!(
+        KERNEL_HEAP
+    )))
+})
+.lock();
+
+pub struct TalcAllocator;
+
+unsafe impl GlobalAlloc for TalcAllocator {
+    unsafe fn alloc(&self, layout: core::alloc::Layout) -> *mut u8 {
+        if layout.size() >= 5 * 1024 * 1024 {
+            let need_page = (layout.size() + FRAME_SIZE - 1) / FRAME_SIZE;
+            trace!("alloc big page: {:#x}", layout.size());
+            alloc_frames(need_page)
+        } else {
+            HEAP_ALLOCATOR.alloc(layout)
+        }
+    }
+    unsafe fn dealloc(&self, ptr: *mut u8, layout: core::alloc::Layout) {
+        if layout.size() >= 5 * 1024 * 1024 {
+            let need_page = (layout.size() + FRAME_SIZE - 1) / FRAME_SIZE;
+            trace!("free big page: {:#x}", layout.size());
+            free_frames(ptr, need_page);
+        } else {
+            HEAP_ALLOCATOR.dealloc(ptr, layout);
+        }
+    }
+}