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
50
51
52
53
54
55
56
57
58

use hal::{
    device::{AllocationError, MapError, OutOfMemory},
    memory::Segment,
    MemoryTypeId,
};
use std::cell::UnsafeCell;
use std::convert::TryInto;

/// A memory allocation.
#[derive(Debug)]
pub struct Memory {
    /// The type of memory this block was allocated from.
    memory_type: MemoryTypeId,
    /// Size of the allocated memory block.
    size: u64,
    /// The backing memory allocation.
    data: UnsafeCell<Box<[u8]>>,
}

impl Memory {
    pub fn allocate(memory_type: MemoryTypeId, size: u64) -> Result<Self, AllocationError> {
        assert_eq!(memory_type.0, 0, "We only support one memory type");

        // Allocate a buffer in RAM
        let data = {
            let size = size
                .try_into()
                // If we're on 32-bit and the given size is greater than 2^32,
                // we certainly can't allocate it.
                .map_err(|_| AllocationError::OutOfMemory(OutOfMemory::Host))?;

            vec![0u8; size].into_boxed_slice()
        };

        let memory = Memory {
            memory_type,
            size,
            data: UnsafeCell::new(data),
        };
        Ok(memory)
    }

    pub fn map(&self, segment: Segment) -> Result<*mut u8, MapError> {
        if segment.offset >= self.size {
            return Err(MapError::OutOfBounds);
        }
        if let Some(size) = segment.size {
            if segment.offset + size > self.size {
                return Err(MapError::OutOfBounds);
            }
        }

        let data = unsafe { &mut *self.data.get() };
        Ok(data.as_mut_ptr())
    }
}

unsafe impl Sync for Memory {}