mirror of
https://github.com/servo/servo.git
synced 2025-07-22 23:03:42 +01:00
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:
parent
459aee27b6
commit
a5cf04c479
30 changed files with 3410 additions and 324 deletions
|
@ -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.Memory’s 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 {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue