1 // SPDX-License-Identifier: GPL-2.0
2
3 //! Allocator support.
4
5 use super::{flags::*, Flags};
6 use core::alloc::{GlobalAlloc, Layout};
7 use core::ptr;
8
9 struct KernelAllocator;
10
11 /// Calls `krealloc` with a proper size to all
12 ///
13 /// # Safety
14 ///
15 /// - `ptr` can be either null or a pointer wh
16 /// - `new_layout` must have a non-zero size.
17 pub(crate) unsafe fn krealloc_aligned(ptr: *mu
18 // Customized layouts from `Layout::from_s
19 let layout = new_layout.pad_to_align();
20
21 // Note that `layout.size()` (after paddin
22 // which together with the slab guarantees
23 // object (see comments in `kmalloc()` for
24 let size = layout.size();
25
26 // SAFETY:
27 // - `ptr` is either null or a pointer ret
28 // function safety requirement.
29 // - `size` is greater than 0 since it's f
30 // to the function safety requirement)
31 unsafe { bindings::krealloc(ptr as *const
32 }
33
34 unsafe impl GlobalAlloc for KernelAllocator {
35 unsafe fn alloc(&self, layout: Layout) ->
36 // SAFETY: `ptr::null_mut()` is null a
37 // requirement.
38 unsafe { krealloc_aligned(ptr::null_mu
39 }
40
41 unsafe fn dealloc(&self, ptr: *mut u8, _la
42 unsafe {
43 bindings::kfree(ptr as *const core
44 }
45 }
46
47 unsafe fn realloc(&self, ptr: *mut u8, lay
48 // SAFETY:
49 // - `new_size`, when rounded up to th
50 // overflow `isize` by the function
51 // - `layout.align()` is a proper alig
52 let layout = unsafe { Layout::from_siz
53
54 // SAFETY:
55 // - `ptr` is either null or a pointer
56 // requirement.
57 // - the size of `layout` is not zero
58 // requirement.
59 unsafe { krealloc_aligned(ptr, layout,
60 }
61
62 unsafe fn alloc_zeroed(&self, layout: Layo
63 // SAFETY: `ptr::null_mut()` is null a
64 // requirement.
65 unsafe { krealloc_aligned(ptr::null_mu
66 }
67 }
68
69 #[global_allocator]
70 static ALLOCATOR: KernelAllocator = KernelAllo
71
72 // See <https://github.com/rust-lang/rust/pull
73 #[no_mangle]
74 static __rust_no_alloc_shim_is_unstable: u8 =
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.