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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

use crate::types::RawHandle;

use crate::vm_hooks::VMHooksHandlerSource;

/// Provides VM hook implementations for methods that deal managed buffers.
pub trait VMHooksManagedBuffer: VMHooksHandlerSource {
    fn mb_new_empty(&self) -> RawHandle {
        self.m_types_lock().mb_new(Vec::new())
    }

    fn mb_new_from_bytes(&self, bytes: &[u8]) -> RawHandle {
        self.m_types_lock().mb_new(Vec::from(bytes))
    }

    fn mb_len(&self, handle: RawHandle) -> usize {
        self.m_types_lock().mb_len(handle)
    }

    fn mb_set(&self, handle: RawHandle, value: &[u8]) {
        self.m_types_lock().mb_set(handle, value.to_vec());
    }

    /// Copies bytes from a stored managed buffer to the given pointer.
    ///
    /// # Safety
    ///
    /// Argument `dest_ptr` should point to a valid location in memory, that has been pre-allocated with the appropriate size.
    unsafe fn mb_copy_bytes(&self, handle: RawHandle, dest_ptr: *mut u8) -> usize {
        let bytes = self.m_types_lock().mb_get(handle).to_vec();
        std::ptr::copy(bytes.as_ptr(), dest_ptr, bytes.len());
        bytes.len()
    }

    fn mb_load_slice(
        &self,
        source_handle: RawHandle,
        starting_position: usize,
        dest_slice: &mut [u8],
    ) -> i32 {
        let result =
            self.m_types_lock()
                .mb_load_slice(source_handle, starting_position, dest_slice);
        if result.is_ok() {
            0
        } else {
            1
        }
    }

    fn mb_copy_slice(
        &self,
        source_handle: RawHandle,
        starting_position: usize,
        slice_len: usize,
        dest_handle: RawHandle,
    ) -> i32 {
        let result = self
            .m_types_lock()
            .mb_get_slice(source_handle, starting_position, slice_len);
        if let Ok(slice) = result {
            self.m_types_lock().mb_set(dest_handle, slice);
            0
        } else {
            1
        }
    }

    fn mb_set_slice(
        &self,
        dest_handle: RawHandle,
        starting_position: usize,
        source_slice: &[u8],
    ) -> i32 {
        let result = self
            .m_types_lock()
            .mb_set_slice(dest_handle, starting_position, source_slice);
        if result.is_ok() {
            0
        } else {
            1
        }
    }

    fn mb_append(&self, accumulator_handle: RawHandle, data_handle: RawHandle) {
        let mut data = self.m_types_lock().mb_get(data_handle).to_vec();
        self.m_types_lock()
            .mb_update(accumulator_handle, |accumulator| {
                accumulator.append(&mut data);
            });
    }

    fn mb_append_bytes(&self, accumulator_handle: RawHandle, bytes: &[u8]) {
        self.m_types_lock()
            .mb_append_bytes(accumulator_handle, bytes);
    }

    fn mb_eq(&self, handle1: RawHandle, handle2: RawHandle) -> i32 {
        let managed_types = self.m_types_lock();
        let bytes1 = managed_types.mb_get(handle1);
        let bytes2 = managed_types.mb_get(handle2);
        if bytes1 == bytes2 {
            1
        } else {
            0
        }
    }

    fn mb_to_hex(&self, source_handle: RawHandle, dest_handle: RawHandle) {
        let encoded = hex::encode(self.m_types_lock().mb_get(source_handle));
        self.m_types_lock()
            .mb_set(dest_handle, encoded.into_bytes());
    }
}