flatty-base 0.1.0-rc.4

Flatty traits and primitives
Documentation 
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
39
40
41
42
43
44
45
46
47
48
49

use alloc::alloc::{alloc, dealloc, Layout};
use core::{
    ops::{Deref, DerefMut},
    slice,
};

pub struct AlignedBytes {
    data: *mut u8,
    layout: Layout,
}

impl AlignedBytes {
    pub fn new(size: usize, align: usize) -> Self {
        let layout = Layout::from_size_align(size, align).unwrap();
        let data = unsafe { alloc(layout) };
        assert!(!data.is_null());
        Self { data, layout }
    }
    pub fn from_slice(src: &[u8], align: usize) -> Self {
        let mut this = Self::new(src.len(), align);
        this.copy_from_slice(src);
        this
    }
    pub fn layout(&self) -> Layout {
        self.layout
    }
}

impl Drop for AlignedBytes {
    fn drop(&mut self) {
        unsafe { dealloc(self.data, self.layout) };
    }
}

impl Deref for AlignedBytes {
    type Target = [u8];
    fn deref(&self) -> &[u8] {
        unsafe { slice::from_raw_parts(self.data, self.layout.size()) }
    }
}

impl DerefMut for AlignedBytes {
    fn deref_mut(&mut self) -> &mut [u8] {
        unsafe { slice::from_raw_parts_mut(self.data, self.layout.size()) }
    }
}

unsafe impl Send for AlignedBytes {}
unsafe impl Sync for AlignedBytes {}