use crate::{qjs, Ctx, Error, FromJs, IntoJs, Object, Outlive, Result, Value};
use core::fmt;
use std::{
mem::{self, size_of, ManuallyDrop, MaybeUninit},
ops::Deref,
os::raw::c_void,
ptr::NonNull,
result::Result as StdResult,
slice,
};
use super::typed_array::TypedArrayItem;
pub struct RawArrayBuffer {
pub len: usize,
pub ptr: NonNull<u8>,
}
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum AsSliceError {
BufferUsed,
InvalidAlignment,
}
impl fmt::Display for AsSliceError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
AsSliceError::BufferUsed => write!(f, "Buffer was already used"),
AsSliceError::InvalidAlignment => {
write!(f, "Buffer had a different alignment than was requested")
}
}
}
}
#[cfg_attr(feature = "doc-cfg", doc(cfg(feature = "array-buffer")))]
#[derive(Debug, PartialEq, Clone, Eq, Hash)]
#[repr(transparent)]
pub struct ArrayBuffer<'js>(pub(crate) Object<'js>);
unsafe impl<'js> Outlive<'js> for ArrayBuffer<'js> {
type Target<'to> = ArrayBuffer<'to>;
}
impl<'js> ArrayBuffer<'js> {
pub fn new<T: Copy>(ctx: Ctx<'js>, src: impl Into<Vec<T>>) -> Result<Self> {
let mut src = ManuallyDrop::new(src.into());
let ptr = src.as_mut_ptr();
let capacity = src.capacity();
let size = src.len() * size_of::<T>();
extern "C" fn drop_raw<T>(_rt: *mut qjs::JSRuntime, opaque: *mut c_void, ptr: *mut c_void) {
let ptr = ptr as *mut T;
let capacity = opaque as usize;
unsafe { Vec::from_raw_parts(ptr, capacity, capacity) };
}
Ok(Self(Object(unsafe {
let val = qjs::JS_NewArrayBuffer(
ctx.as_ptr(),
ptr as _,
size as _,
Some(drop_raw::<T>),
capacity as _,
0,
);
ctx.handle_exception(val).map_err(|error| {
Vec::from_raw_parts(ptr, capacity, capacity);
error
})?;
Value::from_js_value(ctx, val)
})))
}
pub fn new_copy<T: Copy>(ctx: Ctx<'js>, src: impl AsRef<[T]>) -> Result<Self> {
let src = src.as_ref();
let ptr = src.as_ptr();
let size = std::mem::size_of_val(src);
Ok(Self(Object(unsafe {
let val = qjs::JS_NewArrayBufferCopy(ctx.as_ptr(), ptr as _, size as _);
ctx.handle_exception(val)?;
Value::from_js_value(ctx.clone(), val)
})))
}
pub fn len(&self) -> usize {
Self::get_raw(&self.0).expect("Not an ArrayBuffer").len
}
pub fn is_empty(&self) -> bool {
self.len() == 0
}
pub fn as_bytes(&self) -> Option<&[u8]> {
let raw = Self::get_raw(self.as_value())?;
Some(unsafe { slice::from_raw_parts_mut(raw.ptr.as_ptr(), raw.len) })
}
pub fn as_slice<T: TypedArrayItem>(&self) -> StdResult<&[T], AsSliceError> {
let raw = Self::get_raw(&self.0).ok_or(AsSliceError::BufferUsed)?;
if raw.ptr.as_ptr().align_offset(mem::align_of::<T>()) != 0 {
return Err(AsSliceError::InvalidAlignment);
}
let len = raw.len / size_of::<T>();
Ok(unsafe { slice::from_raw_parts(raw.ptr.as_ptr().cast(), len) })
}
pub fn detach(&mut self) {
unsafe { qjs::JS_DetachArrayBuffer(self.0.ctx.as_ptr(), self.0.as_js_value()) }
}
#[inline]
pub fn as_value(&self) -> &Value<'js> {
self.0.as_value()
}
#[inline]
pub fn into_value(self) -> Value<'js> {
self.0.into_value()
}
pub fn from_value(value: Value<'js>) -> Option<Self> {
Self::from_object(Object::from_value(value).ok()?)
}
#[inline]
pub fn as_object(&self) -> &Object<'js> {
&self.0
}
#[inline]
pub fn into_object(self) -> Object<'js> {
self.0
}
pub fn from_object(object: Object<'js>) -> Option<Self> {
if Self::get_raw(&object.0).is_some() {
Some(Self(object))
} else {
None
}
}
pub fn as_raw(&self) -> Option<RawArrayBuffer> {
Self::get_raw(self.as_value())
}
pub(crate) fn get_raw(val: &Value<'js>) -> Option<RawArrayBuffer> {
let ctx = val.ctx();
let val = val.as_js_value();
let mut size = MaybeUninit::<qjs::size_t>::uninit();
let ptr = unsafe { qjs::JS_GetArrayBuffer(ctx.as_ptr(), size.as_mut_ptr(), val) };
if let Some(ptr) = NonNull::new(ptr) {
let len = unsafe { size.assume_init() }
.try_into()
.expect(qjs::SIZE_T_ERROR);
Some(RawArrayBuffer { len, ptr })
} else {
None
}
}
}
impl<'js, T: TypedArrayItem> AsRef<[T]> for ArrayBuffer<'js> {
fn as_ref(&self) -> &[T] {
self.as_slice().expect("ArrayBuffer was detached")
}
}
impl<'js> Deref for ArrayBuffer<'js> {
type Target = Object<'js>;
fn deref(&self) -> &Self::Target {
self.as_object()
}
}
impl<'js> AsRef<Object<'js>> for ArrayBuffer<'js> {
fn as_ref(&self) -> &Object<'js> {
self.as_object()
}
}
impl<'js> AsRef<Value<'js>> for ArrayBuffer<'js> {
fn as_ref(&self) -> &Value<'js> {
self.as_value()
}
}
impl<'js> FromJs<'js> for ArrayBuffer<'js> {
fn from_js(_: &Ctx<'js>, value: Value<'js>) -> Result<Self> {
let ty_name = value.type_name();
if let Some(v) = Self::from_value(value) {
Ok(v)
} else {
Err(Error::new_from_js(ty_name, "ArrayBuffer"))
}
}
}
impl<'js> IntoJs<'js> for ArrayBuffer<'js> {
fn into_js(self, _: &Ctx<'js>) -> Result<Value<'js>> {
Ok(self.into_value())
}
}
impl<'js> Object<'js> {
pub fn is_array_buffer(&self) -> bool {
ArrayBuffer::get_raw(&self.0).is_some()
}
pub unsafe fn ref_array_buffer(&self) -> &ArrayBuffer {
mem::transmute(self)
}
pub fn as_array_buffer(&self) -> Option<&ArrayBuffer> {
self.is_array_buffer()
.then_some(unsafe { self.ref_array_buffer() })
}
}
#[cfg(test)]
mod test {
use crate::*;
#[test]
fn from_javascript_i8() {
test_with(|ctx| {
let val: ArrayBuffer = ctx
.eval(
r#"
new Int8Array([0, -5, 1, 11]).buffer
"#,
)
.unwrap();
assert_eq!(val.len(), 4);
assert_eq!(val.as_ref() as &[i8], &[0i8, -5, 1, 11]);
});
}
#[test]
fn into_javascript_i8() {
test_with(|ctx| {
let val = ArrayBuffer::new(ctx.clone(), [-1i8, 0, 22, 5]).unwrap();
ctx.globals().set("a", val).unwrap();
let res: i8 = ctx
.eval(
r#"
let v = new Int8Array(a);
v.length != 4 ? 1 :
v[0] != -1 ? 2 :
v[1] != 0 ? 3 :
v[2] != 22 ? 4 :
v[3] != 5 ? 5 :
0
"#,
)
.unwrap();
assert_eq!(res, 0);
})
}
#[test]
fn from_javascript_f32() {
test_with(|ctx| {
let val: ArrayBuffer = ctx
.eval(
r#"
new Float32Array([0.5, -5.25, 123.125]).buffer
"#,
)
.unwrap();
assert_eq!(val.len(), 12);
assert_eq!(val.as_ref() as &[f32], &[0.5f32, -5.25, 123.125]);
});
}
#[test]
fn into_javascript_f32() {
test_with(|ctx| {
let val = ArrayBuffer::new(ctx.clone(), [-1.5f32, 0.0, 2.25]).unwrap();
ctx.globals().set("a", val).unwrap();
let res: i8 = ctx
.eval(
r#"
let v = new Float32Array(a);
a.byteLength != 12 ? 1 :
v.length != 3 ? 2 :
v[0] != -1.5 ? 3 :
v[1] != 0 ? 4 :
v[2] != 2.25 ? 5 :
0
"#,
)
.unwrap();
assert_eq!(res, 0);
})
}
#[test]
fn as_bytes() {
test_with(|ctx| {
let val: ArrayBuffer = ctx
.eval(
r#"
new Uint32Array([0xCAFEDEAD,0xFEEDBEAD]).buffer
"#,
)
.unwrap();
let mut res = [0; 8];
let bytes_0 = 0xCAFEDEADu32.to_ne_bytes();
res[..4].copy_from_slice(&bytes_0);
let bytes_1 = 0xFEEDBEADu32.to_ne_bytes();
res[4..].copy_from_slice(&bytes_1);
assert_eq!(val.as_bytes().unwrap(), &res)
});
}
}