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

#[derive(Clone)]
pub enum Memory {
    Empty,
    #[cfg(feature = "vm-wasmer")]
    Wasmer(wasmer::Memory),
    #[cfg(feature = "vm-wasmtime")]
    Wasmtime(wasmtime::Memory),
}

impl Memory {
    pub fn from<M: Into<Memory>>(memory: M) -> Self {
        memory.into()
    }

    pub fn as_mut_slice(&self) -> &mut [u8] {
        unsafe {
            match self {
                #[cfg(feature = "vm-wasmer")]
                Memory::Wasmer(mem) => mem.data_unchecked_mut(),
                #[cfg(feature = "vm-wasmtime")]
                Memory::Wasmtime(mem) => mem.data_unchecked_mut(),
                Memory::Empty => panic!("Called as_mut_slice() on uptown_funk::Memory::Empty"),
            }
        }
    }
}

#[cfg(feature = "vm-wasmer")]
impl Into<Memory> for wasmer::Memory {
    fn into(self) -> Memory {
        Memory::Wasmer(self)
    }
}

#[cfg(feature = "vm-wasmtime")]
impl Into<Memory> for wasmtime::Memory {
    fn into(self) -> Memory {
        Memory::Wasmtime(self)
    }
}