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
// From this PR https://github.com/kajacx/wasm-bridge/pull/3 by zimond

use js_sys::{Reflect, Uint8Array, WebAssembly};

use crate::{helpers::map_js_error, AsContextMut, Error};

#[derive(Clone)]
pub struct Memory {
    memory: WebAssembly::Memory,
}

impl Memory {
    pub(crate) fn new(memory: WebAssembly::Memory) -> Self {
        Self { memory }
    }

    pub fn write(&self, _: impl AsContextMut, offset: usize, buffer: &[u8]) -> Result<(), Error> {
        let memory = Reflect::get(&self.memory, &"buffer".into())
            .map_err(map_js_error("Memory has no buffer field"))?;
        let mem = Uint8Array::new_with_byte_offset_and_length(
            &memory,
            offset as u32,
            buffer.len() as u32,
        );
        mem.copy_from(buffer);
        Ok(())
    }

    pub fn read(
        &self,
        _: impl AsContextMut,
        offset: usize,
        buffer: &mut [u8],
    ) -> Result<(), Error> {
        let memory = Reflect::get(&self.memory, &"buffer".into())
            .map_err(map_js_error("Memory has no buffer field"))?;
        let mem = Uint8Array::new_with_byte_offset_and_length(
            &memory,
            offset as u32,
            buffer.len() as u32,
        );
        mem.copy_to(buffer);
        Ok(())
    }
}