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
use std::mem; use std::ops::{Deref, DerefMut}; use std::ptr; use super::Value; #[cfg(feature = "serde-json")] use super::ValueType; use crate::check_status; use crate::{sys, JsUnknown, NapiValue, Ref, Result}; pub struct JsBuffer(pub(crate) Value); pub struct JsBufferValue { pub(crate) value: JsBuffer, data: mem::ManuallyDrop<Vec<u8>>, } impl JsBuffer { #[inline] pub fn into_value(self) -> Result<JsBufferValue> { let mut data = ptr::null_mut(); let mut len: usize = 0; check_status!(unsafe { sys::napi_get_buffer_info(self.0.env, self.0.value, &mut data, &mut len) })?; Ok(JsBufferValue { data: mem::ManuallyDrop::new(unsafe { Vec::from_raw_parts(data as *mut _, len, len) }), value: self, }) } #[inline] pub fn into_ref(self) -> Result<Ref<JsBufferValue>> { Ref::new(self.0, 1, self.into_value()?) } } impl JsBufferValue { #[cfg(feature = "serde-json")] #[inline] pub(crate) fn from_raw(env: sys::napi_env, value: sys::napi_value) -> Result<Self> { let mut data = ptr::null_mut(); let mut len = 0usize; check_status!(unsafe { sys::napi_get_buffer_info(env, value, &mut data, &mut len as *mut usize as *mut _) })?; Ok(Self { value: JsBuffer(Value { env, value, value_type: ValueType::Object, }), data: mem::ManuallyDrop::new(unsafe { Vec::from_raw_parts(data as *mut _, len, len) }), }) } #[inline] pub fn new(value: JsBuffer, data: mem::ManuallyDrop<Vec<u8>>) -> Self { JsBufferValue { value, data } } #[inline] pub fn into_raw(self) -> JsBuffer { self.value } #[inline] pub fn into_unknown(self) -> JsUnknown { unsafe { JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value) } } } impl AsRef<[u8]> for JsBufferValue { fn as_ref(&self) -> &[u8] { self.data.as_slice() } } impl AsMut<[u8]> for JsBufferValue { fn as_mut(&mut self) -> &mut [u8] { self.data.as_mut_slice() } } impl Deref for JsBufferValue { type Target = [u8]; fn deref(&self) -> &[u8] { self.data.as_slice() } } impl DerefMut for JsBufferValue { fn deref_mut(&mut self) -> &mut Self::Target { self.data.as_mut_slice() } }