blob: 27647be92b5134e3df9b499d4d84cf14981b2f47 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
// SPDX-License-Identifier: GPL-2.0
use core::alloc::{GlobalAlloc, Layout};
use core::ptr;
use crate::bindings;
use crate::c_types;
pub struct KernelAllocator;
unsafe impl GlobalAlloc for KernelAllocator {
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
// krealloc is used instead of kmalloc because kmalloc is an inline function and can't be
// bound to as a result
bindings::krealloc(ptr::null(), layout.size(), bindings::GFP_KERNEL) as *mut u8
}
unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
bindings::kfree(ptr as *const c_types::c_void);
}
}
#[alloc_error_handler]
fn oom(_layout: Layout) -> ! {
panic!("Out of memory!");
}
|