simpleos 0.6.3

A Simple OS use for MCU, that implements single-threaded asynchronous runtime.
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
27
28
29
30
31
32
33
34
35
36
37
38

use core::alloc::{GlobalAlloc, Layout};
use core::ffi::c_void;
use core::ffi::c_uint;

unsafe extern "C" {
    fn malloc(size: c_uint) -> *mut c_void;
    fn free(ptr: *mut c_void);
    fn aligned_alloc(alignment: c_uint, size: c_uint) -> *mut c_void;
}

pub struct CAllocator;

impl CAllocator {
    pub const fn new() -> Self {
        CAllocator
    }
}

unsafe impl GlobalAlloc for CAllocator {
    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
        if layout.align() <= 8 {
            // 对于小对齐要求，直接使用 malloc
            unsafe { malloc(layout.size() as _) as *mut u8 }
        } else {
            // 对于较大的对齐要求，使用 aligned_alloc
            unsafe { aligned_alloc(layout.align() as _, layout.size() as _) as *mut u8 }
        }
    }

    unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
        unsafe {
            free(ptr as *mut ::core::ffi::c_void);
        }
    }
}

#[global_allocator]
pub static ALLOCATOR: CAllocator = CAllocator::new();