script: implement ReadableByteStreamController (#35410)

* script: implement ReadableByteStreamController

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* implement can_copy_data_block_bytes and copy_data_block_bytes

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Remove BufferSource::Default

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* implement StartAlgorithmFulfillmentHandler, StartAlgorithmRejectionHandler, PullAlgorithmFulfillmentHandler, PullAlgorithmRejectionHandler for ReadableByteStreamController

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* implement perform_pull_into

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix build

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix clippy

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix build

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Remove RefCell from PullIntoDescriptor and QueueEntry

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Remove commented code

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* add perform_cancel_steps, perform_release_steps and perform_pull_steps

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix clippy

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix crown

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* unskip readable-byte-streams

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix CRASH

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix clippy

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix more CRASHS

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix more crashes

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix bad-buffers-and-views.any.js test

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Update test expectations

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix BorrowMutError crashes

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix view_byte_length  test

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix non-transferable-buffers test

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Pass contexts as much as possible by reference

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Make respond_internal Fallible

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix crwon

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* fix process pull into descriptors using queue logic and resulting double-borrow

Signed-off-by: gterzian <2792687+gterzian@users.noreply.github.com>

* Fix clippy

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* FIx more crashes

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix timeout tests

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix all tests

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Remove all error! logs

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Remove  #[allow(unsafe_code)]

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix lint
Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix tidy

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

* Fix test expectation

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>

---------

Signed-off-by: Taym Haddadi <haddadi.taym@gmail.com>
Signed-off-by: gterzian <2792687+gterzian@users.noreply.github.com>
Co-authored-by: gterzian <2792687+gterzian@users.noreply.github.com>
This commit is contained in:
Taym Haddadi 2025-03-10 12:43:04 -07:00 committed by GitHub
parent 459aee27b6
commit a5cf04c479
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
30 changed files with 3410 additions and 324 deletions

View file

@ -16,15 +16,31 @@ use std::sync::Arc;
#[cfg(feature = "webgpu")]
use js::jsapi::NewExternalArrayBuffer;
use js::jsapi::{
GetArrayBufferByteLength, Heap, IsDetachedArrayBufferObject, JS_GetArrayBufferViewBuffer,
JS_GetArrayBufferViewByteLength, JS_IsArrayBufferViewObject, JS_IsTypedArrayObject, JSObject,
ArrayBufferClone, ArrayBufferCopyData, GetArrayBufferByteLength,
HasDefinedArrayBufferDetachKey, Heap, IsArrayBufferObject, IsDetachedArrayBufferObject,
JS_ClearPendingException, JS_GetArrayBufferViewBuffer, JS_GetArrayBufferViewByteLength,
JS_GetArrayBufferViewByteOffset, JS_GetArrayBufferViewType, JS_GetPendingException,
JS_GetTypedArrayLength, JS_IsArrayBufferViewObject, JS_IsTypedArrayObject,
JS_NewBigInt64ArrayWithBuffer, JS_NewBigUint64ArrayWithBuffer, JS_NewDataView,
JS_NewFloat16ArrayWithBuffer, JS_NewFloat32ArrayWithBuffer, JS_NewFloat64ArrayWithBuffer,
JS_NewInt8ArrayWithBuffer, JS_NewInt16ArrayWithBuffer, JS_NewInt32ArrayWithBuffer,
JS_NewUint8ArrayWithBuffer, JS_NewUint8ClampedArrayWithBuffer, JS_NewUint16ArrayWithBuffer,
JS_NewUint32ArrayWithBuffer, JSObject, NewArrayBuffer, NewArrayBufferWithContents,
StealArrayBufferContents, Type,
};
use js::jsval::{ObjectValue, UndefinedValue};
use js::rust::wrappers::DetachArrayBuffer;
use js::rust::{CustomAutoRooterGuard, Handle, MutableHandleObject};
#[cfg(feature = "webgpu")]
use js::typedarray::{ArrayBuffer, HeapArrayBuffer};
use js::typedarray::{CreateWith, TypedArray, TypedArrayElement, TypedArrayElementCreator};
use js::rust::{
CustomAutoRooterGuard, Handle, MutableHandleObject,
MutableHandleValue as SafeMutableHandleValue,
};
use js::typedarray::{
ArrayBuffer, ArrayBufferU8, ArrayBufferView, ArrayBufferViewU8, CreateWith, HeapArrayBuffer,
TypedArray, TypedArrayElement, TypedArrayElementCreator,
};
use crate::dom::bindings::error::Error;
use crate::dom::bindings::import::module::Fallible;
#[cfg(feature = "webgpu")]
use crate::dom::globalscope::GlobalScope;
use crate::script_runtime::{CanGc, JSContext};
@ -35,6 +51,7 @@ use crate::script_runtime::{CanGc, JSContext};
/// provides a view onto an `ArrayBuffer`.
///
/// See: <https://webidl.spec.whatwg.org/#BufferSource>
#[derive(PartialEq)]
pub(crate) enum BufferSource {
/// Represents an `ArrayBufferView` (e.g., `Uint8Array`, `DataView`).
/// See: <https://webidl.spec.whatwg.org/#ArrayBufferView>
@ -42,11 +59,7 @@ pub(crate) enum BufferSource {
/// Represents an `ArrayBuffer`, a fixed-length binary data buffer.
/// See: <https://webidl.spec.whatwg.org/#idl-ArrayBuffer>
#[allow(dead_code)]
ArrayBuffer(Box<Heap<*mut JSObject>>),
/// Default variant, used as a placeholder in initialization.
Default(Box<Heap<*mut JSObject>>),
}
pub(crate) fn new_initialized_heap_buffer_source<T>(
@ -69,16 +82,8 @@ where
if typed_array_result.is_err() {
return Err(());
}
let heap_buffer_source = HeapBufferSource::<T>::default();
match &heap_buffer_source.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
buffer.set(*array);
},
}
heap_buffer_source
HeapBufferSource::<T>::new(BufferSource::ArrayBufferView(Heap::boxed(*array.handle())))
},
};
Ok(heap_buffer_source)
@ -94,6 +99,25 @@ pub(crate) struct HeapBufferSource<T> {
phantom: PhantomData<T>,
}
impl<T> Eq for HeapBufferSource<T> where T: TypedArrayElement {}
impl<T> PartialEq for HeapBufferSource<T>
where
T: TypedArrayElement,
{
fn eq(&self, other: &Self) -> bool {
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => match &other
.buffer_source
{
BufferSource::ArrayBufferView(from_heap) | BufferSource::ArrayBuffer(from_heap) => unsafe {
heap.handle() == from_heap.handle()
},
},
}
}
}
impl<T> HeapBufferSource<T>
where
T: TypedArrayElement,
@ -105,34 +129,78 @@ where
}
}
pub(crate) fn default() -> HeapBufferSource<T> {
pub(crate) fn from_view(
chunk: CustomAutoRooterGuard<ArrayBufferView>,
) -> HeapBufferSource<ArrayBufferViewU8> {
HeapBufferSource::<ArrayBufferViewU8>::new(BufferSource::ArrayBufferView(Heap::boxed(
unsafe { *chunk.underlying_object() },
)))
}
pub(crate) fn default() -> Self {
HeapBufferSource {
buffer_source: BufferSource::Default(Box::default()),
buffer_source: BufferSource::ArrayBufferView(Heap::boxed(std::ptr::null_mut())),
phantom: PhantomData,
}
}
pub(crate) fn is_initialized(&self) -> bool {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => !buffer.get().is_null(),
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => {
!buffer.get().is_null()
},
}
}
pub(crate) fn get_buffer(&self) -> Result<TypedArray<T, *mut JSObject>, ()> {
pub(crate) fn get_typed_array(&self) -> Result<TypedArray<T, *mut JSObject>, ()> {
TypedArray::from(match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => buffer.get(),
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => {
buffer.get()
},
})
}
pub(crate) fn get_buffer_view_value(
&self,
cx: JSContext,
mut handle_mut: SafeMutableHandleValue,
) {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) => {
rooted!(in(*cx) let value = ObjectValue(buffer.get()));
handle_mut.set(*value);
},
BufferSource::ArrayBuffer(_) => {
unreachable!("BufferSource::ArrayBuffer does not have a view buffer.")
},
}
}
pub(crate) fn get_array_buffer_view_buffer(
&self,
cx: JSContext,
) -> HeapBufferSource<ArrayBufferU8> {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) => unsafe {
let mut is_shared = false;
rooted!(in (*cx) let view_buffer =
JS_GetArrayBufferViewBuffer(*cx, buffer.handle(), &mut is_shared));
HeapBufferSource::<ArrayBufferU8>::new(BufferSource::ArrayBuffer(Heap::boxed(
*view_buffer.handle(),
)))
},
BufferSource::ArrayBuffer(_) => {
unreachable!("BufferSource::ArrayBuffer does not have a view buffer.")
},
}
}
/// <https://tc39.es/ecma262/#sec-detacharraybuffer>
pub(crate) fn detach_buffer(&self, cx: JSContext) -> bool {
assert!(self.is_initialized());
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) => {
let mut is_shared = false;
unsafe {
// assert buffer is an ArrayBuffer view
@ -151,9 +219,9 @@ where
}
}
pub(crate) fn buffer_to_option(&self) -> Option<TypedArray<T, *mut JSObject>> {
pub(crate) fn typed_array_to_option(&self) -> Option<TypedArray<T, *mut JSObject>> {
if self.is_initialized() {
self.get_buffer().ok()
self.get_typed_array().ok()
} else {
warn!("Buffer not initialized.");
None
@ -163,7 +231,7 @@ where
pub(crate) fn is_detached_buffer(&self, cx: JSContext) -> bool {
assert!(self.is_initialized());
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) => {
let mut is_shared = false;
unsafe {
assert!(JS_IsArrayBufferViewObject(*buffer.handle()));
@ -182,7 +250,7 @@ where
pub(crate) fn viewed_buffer_array_byte_length(&self, cx: JSContext) -> usize {
assert!(self.is_initialized());
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) => {
let mut is_shared = false;
unsafe {
assert!(JS_IsArrayBufferViewObject(*buffer.handle()));
@ -200,7 +268,7 @@ where
pub(crate) fn byte_length(&self) -> usize {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::Default(buffer) => unsafe {
BufferSource::ArrayBufferView(buffer) => unsafe {
JS_GetArrayBufferViewByteLength(*buffer.handle())
},
BufferSource::ArrayBuffer(buffer) => unsafe {
@ -209,19 +277,54 @@ where
}
}
pub(crate) fn array_length(&self) -> usize {
self.get_buffer().unwrap().len()
pub(crate) fn get_byte_offset(&self) -> usize {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) => unsafe {
JS_GetArrayBufferViewByteOffset(*buffer.handle())
},
BufferSource::ArrayBuffer(_) => {
unreachable!("BufferSource::ArrayBuffer does not have a byte offset.")
},
}
}
pub(crate) fn get_typed_array_length(&self) -> usize {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) => unsafe {
JS_GetTypedArrayLength(*buffer.handle())
},
BufferSource::ArrayBuffer(_) => {
unreachable!("BufferSource::ArrayBuffer does not have a length.")
},
}
}
/// <https://tc39.es/ecma262/#typedarray>
pub(crate) fn has_typed_array_name(&self) -> bool {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::Default(buffer) => unsafe {
BufferSource::ArrayBufferView(buffer) => unsafe {
JS_IsTypedArrayObject(*buffer.handle())
},
BufferSource::ArrayBuffer(_) => false,
}
}
pub(crate) fn get_array_buffer_view_type(&self) -> Type {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) => unsafe {
JS_GetArrayBufferViewType(*buffer.handle())
},
BufferSource::ArrayBuffer(_) => unreachable!("ArrayBuffer does not have a view type."),
}
}
pub(crate) fn is_array_buffer_object(&self) -> bool {
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => unsafe {
IsArrayBufferObject(*heap.handle())
},
}
}
}
impl<T> HeapBufferSource<T>
@ -233,9 +336,8 @@ where
assert!(self.is_initialized());
typedarray!(in(*cx) let array: TypedArray = match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer)
=> {
buffer.get()
},
});
@ -250,9 +352,7 @@ where
};
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => {
buffer.set(ptr::null_mut());
},
}
@ -268,9 +368,8 @@ where
) -> Result<(), ()> {
assert!(self.is_initialized());
typedarray!(in(*cx) let array: TypedArray = match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer)
=> {
buffer.get()
},
});
@ -295,9 +394,8 @@ where
) -> Result<(), ()> {
assert!(self.is_initialized());
typedarray!(in(*cx) let mut array: TypedArray = match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer)
=> {
buffer.get()
},
});
@ -325,23 +423,194 @@ where
create_buffer_source(cx, data, array.handle_mut(), can_gc)?;
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => {
buffer.set(*array);
},
}
Ok(())
}
/// <https://tc39.es/ecma262/#sec-clonearraybuffer>
pub(crate) fn clone_array_buffer(
&self,
cx: JSContext,
byte_offset: usize,
byte_length: usize,
) -> Option<HeapBufferSource<ArrayBufferU8>> {
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => {
let result =
unsafe { ArrayBufferClone(*cx, heap.handle(), byte_offset, byte_length) };
if result.is_null() {
None
} else {
Some(HeapBufferSource::<ArrayBufferU8>::new(
BufferSource::ArrayBuffer(Heap::boxed(result)),
))
}
},
}
}
/// <https://streams.spec.whatwg.org/#abstract-opdef-cancopydatablockbytes>
// CanCopyDataBlockBytes(descriptorBuffer, destStart, queueBuffer, queueByteOffset, bytesToCopy)
pub(crate) fn can_copy_data_block_bytes(
&self,
cx: JSContext,
to_index: usize,
from_buffer: &HeapBufferSource<ArrayBufferU8>,
from_index: usize,
bytes_to_copy: usize,
) -> bool {
// Assert: toBuffer is an Object.
// Assert: toBuffer has an [[ArrayBufferData]] internal slot.
assert!(self.is_array_buffer_object());
// Assert: fromBuffer is an Object.
// Assert: fromBuffer has an [[ArrayBufferData]] internal slot.
assert!(from_buffer.is_array_buffer_object());
// If toBuffer is fromBuffer, return false.
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => {
match &from_buffer.buffer_source {
BufferSource::ArrayBufferView(from_heap) |
BufferSource::ArrayBuffer(from_heap) => {
unsafe {
if heap.handle() == from_heap.handle() {
return false;
}
};
},
}
},
}
// If ! IsDetachedBuffer(toBuffer) is true, return false.
if self.is_detached_buffer(cx) {
return false;
}
// If ! IsDetachedBuffer(fromBuffer) is true, return false.
if from_buffer.is_detached_buffer(cx) {
return false;
}
// If toIndex + count > toBuffer.[[ArrayBufferByteLength]], return false.
if to_index + bytes_to_copy > self.byte_length() {
return false;
}
// If fromIndex + count > fromBuffer.[[ArrayBufferByteLength]], return false.
if from_index + bytes_to_copy > from_buffer.byte_length() {
return false;
}
// Return true.
true
}
pub(crate) fn copy_data_block_bytes(
&self,
cx: JSContext,
dest_start: usize,
from_buffer: &HeapBufferSource<ArrayBufferU8>,
from_byte_offset: usize,
bytes_to_copy: usize,
) -> bool {
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => unsafe {
match &from_buffer.buffer_source {
BufferSource::ArrayBufferView(from_heap) |
BufferSource::ArrayBuffer(from_heap) => ArrayBufferCopyData(
*cx,
heap.handle(),
dest_start,
from_heap.handle(),
from_byte_offset,
bytes_to_copy,
),
}
},
}
}
/// <https://streams.spec.whatwg.org/#can-transfer-array-buffer>
pub(crate) fn can_transfer_array_buffer(&self, cx: JSContext) -> bool {
// Assert: O is an Object.
// Assert: O has an [[ArrayBufferData]] internal slot.
assert!(self.is_array_buffer_object());
// If ! IsDetachedBuffer(O) is true, return false.
if self.is_detached_buffer(cx) {
return false;
}
// If SameValue(O.[[ArrayBufferDetachKey]], undefined) is false, return false.
// Return true.
let mut is_defined = false;
match &self.buffer_source {
BufferSource::ArrayBufferView(heap) | BufferSource::ArrayBuffer(heap) => unsafe {
if !HasDefinedArrayBufferDetachKey(*cx, heap.handle(), &mut is_defined) {
return false;
}
},
}
!is_defined
}
/// <https://streams.spec.whatwg.org/#transfer-array-buffer>
pub(crate) fn transfer_array_buffer(
&self,
cx: JSContext,
) -> Fallible<HeapBufferSource<ArrayBufferU8>> {
assert!(self.is_array_buffer_object());
// Assert: ! IsDetachedBuffer(O) is false.
assert!(!self.is_detached_buffer(cx));
// Let arrayBufferByteLength be O.[[ArrayBufferByteLength]].
// Step 3 (Reordered)
let buffer_length = self.byte_length();
// Let arrayBufferData be O.[[ArrayBufferData]].
// Step 2 (Reordered)
let buffer_data = match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => unsafe {
StealArrayBufferContents(*cx, buffer.handle())
},
};
// Perform ? DetachArrayBuffer(O).
// This will throw an exception if O has an [[ArrayBufferDetachKey]] that is not undefined,
// such as a WebAssembly.Memorys buffer. [WASM-JS-API-1]
if !self.detach_buffer(cx) {
rooted!(in(*cx) let mut rval = UndefinedValue());
unsafe {
assert!(JS_GetPendingException(*cx, rval.handle_mut().into()));
JS_ClearPendingException(*cx)
};
Err(Error::Type("can't transfer array buffer".to_owned()))
} else {
// Return a new ArrayBuffer object, created in the current Realm,
// whose [[ArrayBufferData]] internal slot value is arrayBufferData and
// whose [[ArrayBufferByteLength]] internal slot value is arrayBufferByteLength.
Ok(HeapBufferSource::<ArrayBufferU8>::new(
BufferSource::ArrayBuffer(Heap::boxed(unsafe {
NewArrayBufferWithContents(*cx, buffer_length, buffer_data)
})),
))
}
}
}
unsafe impl<T> crate::dom::bindings::trace::JSTraceable for HeapBufferSource<T> {
#[inline]
unsafe fn trace(&self, tracer: *mut js::jsapi::JSTracer) {
match &self.buffer_source {
BufferSource::ArrayBufferView(buffer) |
BufferSource::ArrayBuffer(buffer) |
BufferSource::Default(buffer) => {
BufferSource::ArrayBufferView(buffer) | BufferSource::ArrayBuffer(buffer) => {
buffer.trace(tracer);
},
}
@ -385,6 +654,134 @@ where
}
}
pub(crate) fn byte_size(byte_type: Type) -> u64 {
match byte_type {
Type::Int8 | Type::Uint8 | Type::Uint8Clamped => 1,
Type::Int16 | Type::Uint16 | Type::Float16 => 2,
Type::Int32 | Type::Uint32 | Type::Float32 => 4,
Type::Int64 | Type::Float64 | Type::BigInt64 | Type::BigUint64 => 8,
Type::Simd128 => 16,
_ => unreachable!("invalid scalar type"),
}
}
#[derive(Clone, Eq, JSTraceable, MallocSizeOf, PartialEq)]
pub(crate) enum Constructor {
DataView,
Name(
#[ignore_malloc_size_of = "mozjs"]
#[no_trace]
Type,
),
}
pub(crate) fn create_buffer_source_with_constructor(
cx: JSContext,
constructor: &Constructor,
buffer_source: &HeapBufferSource<ArrayBufferU8>,
byte_offset: usize,
byte_length: usize,
) -> Fallible<HeapBufferSource<ArrayBufferViewU8>> {
let buffer = unsafe {
Heap::boxed(
*buffer_source
.get_typed_array()
.expect("Failed to get typed array")
.underlying_object(),
)
.handle()
};
match constructor {
Constructor::DataView => Ok(HeapBufferSource::new(BufferSource::ArrayBufferView(
Heap::boxed(unsafe { JS_NewDataView(*cx, buffer, byte_offset, byte_length) }),
))),
Constructor::Name(name_type) => construct_typed_array(
cx,
name_type,
buffer_source,
byte_offset,
byte_length as i64,
),
}
}
/// Helper function to construct different TypedArray views
fn construct_typed_array(
cx: JSContext,
name_type: &Type,
buffer_source: &HeapBufferSource<ArrayBufferU8>,
byte_offset: usize,
byte_length: i64,
) -> Fallible<HeapBufferSource<ArrayBufferViewU8>> {
let buffer = unsafe {
Heap::boxed(
*buffer_source
.get_typed_array()
.expect("Failed to get typed array")
.underlying_object(),
)
.handle()
};
let array_view = match name_type {
Type::Int8 => unsafe { JS_NewInt8ArrayWithBuffer(*cx, buffer, byte_offset, byte_length) },
Type::Uint8 => unsafe { JS_NewUint8ArrayWithBuffer(*cx, buffer, byte_offset, byte_length) },
Type::Uint16 => unsafe {
JS_NewUint16ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Int16 => unsafe { JS_NewInt16ArrayWithBuffer(*cx, buffer, byte_offset, byte_length) },
Type::Int32 => unsafe { JS_NewInt32ArrayWithBuffer(*cx, buffer, byte_offset, byte_length) },
Type::Uint32 => unsafe {
JS_NewUint32ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Float32 => unsafe {
JS_NewFloat32ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Float64 => unsafe {
JS_NewFloat64ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Uint8Clamped => unsafe {
JS_NewUint8ClampedArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::BigInt64 => unsafe {
JS_NewBigInt64ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::BigUint64 => unsafe {
JS_NewBigUint64ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Float16 => unsafe {
JS_NewFloat16ArrayWithBuffer(*cx, buffer, byte_offset, byte_length)
},
Type::Int64 | Type::Simd128 | Type::MaxTypedArrayViewType => {
unreachable!("Invalid TypedArray type")
},
};
Ok(HeapBufferSource::new(BufferSource::ArrayBufferView(
Heap::boxed(array_view),
)))
}
pub(crate) fn create_array_buffer_with_size(
cx: JSContext,
size: usize,
) -> Fallible<HeapBufferSource<ArrayBufferU8>> {
let result = unsafe { NewArrayBuffer(*cx, size) };
if result.is_null() {
rooted!(in(*cx) let mut rval = UndefinedValue());
unsafe {
assert!(JS_GetPendingException(*cx, rval.handle_mut().into()));
JS_ClearPendingException(*cx)
};
Err(Error::Type("can't create array buffer".to_owned()))
} else {
Ok(HeapBufferSource::<ArrayBufferU8>::new(
BufferSource::ArrayBuffer(Heap::boxed(result)),
))
}
}
#[cfg(feature = "webgpu")]
#[derive(JSTraceable, MallocSizeOf)]
pub(crate) struct DataBlock {