2022-01-07 22:09:52 -05:00
// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
2019-03-26 11:56:34 -04:00
2020-01-06 14:07:35 -05:00
use crate ::bindings ;
2020-10-14 08:04:09 -04:00
use crate ::error ::generic_error ;
2022-04-26 09:28:42 -04:00
use crate ::error ::to_v8_type_error ;
2020-09-14 12:48:57 -04:00
use crate ::error ::JsError ;
2022-03-15 18:43:17 -04:00
use crate ::extensions ::OpDecl ;
2022-03-08 09:40:34 -05:00
use crate ::extensions ::OpEventLoopFn ;
2021-05-26 15:07:12 -04:00
use crate ::inspector ::JsRuntimeInspector ;
2020-09-06 10:50:49 -04:00
use crate ::module_specifier ::ModuleSpecifier ;
2022-04-26 09:28:42 -04:00
use crate ::modules ::ModuleError ;
2020-09-06 10:50:49 -04:00
use crate ::modules ::ModuleId ;
use crate ::modules ::ModuleLoadId ;
use crate ::modules ::ModuleLoader ;
2021-02-23 09:22:55 -05:00
use crate ::modules ::ModuleMap ;
2020-09-06 10:50:49 -04:00
use crate ::modules ::NoopModuleLoader ;
2022-03-22 11:39:58 -04:00
use crate ::op_void_async ;
use crate ::op_void_sync ;
2019-09-30 14:59:44 -04:00
use crate ::ops ::* ;
2022-06-20 08:42:20 -04:00
use crate ::source_map ::SourceMapCache ;
2022-04-15 10:08:09 -04:00
use crate ::source_map ::SourceMapGetter ;
2021-04-28 12:41:50 -04:00
use crate ::Extension ;
use crate ::OpMiddlewareFn ;
2021-04-30 10:51:54 -04:00
use crate ::OpResult ;
2020-09-10 09:57:45 -04:00
use crate ::OpState ;
2021-03-31 10:37:38 -04:00
use crate ::PromiseId ;
2021-11-16 09:02:28 -05:00
use anyhow ::Error ;
2021-07-30 07:36:43 -04:00
use futures ::channel ::oneshot ;
2020-10-11 07:20:40 -04:00
use futures ::future ::poll_fn ;
2022-03-14 13:44:15 -04:00
use futures ::future ::Future ;
2021-05-26 15:07:12 -04:00
use futures ::future ::FutureExt ;
2019-08-07 12:55:39 -04:00
use futures ::stream ::FuturesUnordered ;
2019-11-16 19:17:47 -05:00
use futures ::stream ::StreamExt ;
use futures ::task ::AtomicWaker ;
2022-11-10 06:56:02 -05:00
use smallvec ::SmallVec ;
2020-08-11 21:07:14 -04:00
use std ::any ::Any ;
2020-04-21 09:48:44 -04:00
use std ::cell ::RefCell ;
2020-01-06 14:07:35 -05:00
use std ::collections ::HashMap ;
2021-11-25 13:49:09 -05:00
use std ::collections ::HashSet ;
2022-09-02 06:43:39 -04:00
use std ::collections ::VecDeque ;
2020-08-11 21:07:14 -04:00
use std ::ffi ::c_void ;
2020-01-06 14:07:35 -05:00
use std ::option ::Option ;
2020-04-21 09:48:44 -04:00
use std ::rc ::Rc ;
2021-07-06 13:42:52 -04:00
use std ::sync ::Arc ;
use std ::sync ::Mutex ;
2020-05-29 17:41:39 -04:00
use std ::sync ::Once ;
2019-11-16 19:17:47 -05:00
use std ::task ::Context ;
use std ::task ::Poll ;
2022-10-15 10:01:01 -04:00
use v8 ::OwnedIsolate ;
2020-01-24 15:10:49 -05:00
2022-10-20 22:54:22 -04:00
type PendingOpFuture = OpCall < ( PromiseId , OpId , OpResult ) > ;
2020-04-18 20:05:13 -04:00
2020-04-22 14:24:49 -04:00
pub enum Snapshot {
Static ( & 'static [ u8 ] ) ,
JustCreated ( v8 ::StartupData ) ,
2020-05-09 21:00:40 -04:00
Boxed ( Box < [ u8 ] > ) ,
2020-04-22 14:24:49 -04:00
}
2021-11-16 09:02:28 -05:00
pub type JsErrorCreateFn = dyn Fn ( JsError ) -> Error ;
2019-07-10 18:53:48 -04:00
2021-11-16 09:02:28 -05:00
pub type GetErrorClassFn = & 'static dyn for < ' e > Fn ( & ' e Error ) -> & 'static str ;
2020-08-25 18:22:15 -04:00
2020-08-11 21:07:14 -04:00
/// Objects that need to live as long as the isolate
#[ derive(Default) ]
struct IsolateAllocations {
near_heap_limit_callback_data :
Option < ( Box < RefCell < dyn Any > > , v8 ::NearHeapLimitCallback ) > ,
}
2019-03-12 18:47:54 -04:00
/// A single execution context of JavaScript. Corresponds roughly to the "Web
2020-09-06 15:44:29 -04:00
/// Worker" concept in the DOM. A JsRuntime is a Future that can be used with
/// an event loop (Tokio, async_std).
////
/// The JsRuntime future completes when there is an error or when all
2019-03-12 18:47:54 -04:00
/// pending ops have completed.
///
2021-04-23 11:50:45 -04:00
/// Pending ops are created in JavaScript by calling Deno.core.opAsync(), and in Rust
/// by implementing an async function that takes a serde::Deserialize "control argument"
/// and an optional zero copy buffer, each async Op is tied to a Promise in JavaScript.
2020-09-06 15:44:29 -04:00
pub struct JsRuntime {
2022-10-21 10:13:42 -04:00
state : Rc < RefCell < JsRuntimeState > > ,
2022-10-25 08:27:33 -04:00
module_map : Option < Rc < RefCell < ModuleMap > > > ,
2020-05-29 17:41:39 -04:00
// This is an Option<OwnedIsolate> instead of just OwnedIsolate to workaround
2021-06-21 13:37:51 -04:00
// a safety issue with SnapshotCreator. See JsRuntime::drop.
2020-05-29 17:41:39 -04:00
v8_isolate : Option < v8 ::OwnedIsolate > ,
2022-10-28 15:31:01 -04:00
snapshot_options : SnapshotOptions ,
2022-04-17 07:53:08 -04:00
built_from_snapshot : bool ,
2020-08-11 21:07:14 -04:00
allocations : IsolateAllocations ,
2021-04-28 12:41:50 -04:00
extensions : Vec < Extension > ,
2022-03-08 09:40:34 -05:00
event_loop_middlewares : Vec < Box < OpEventLoopFn > > ,
2022-11-26 17:09:48 -05:00
// Marks if this is considered the top-level runtime. Used only be inspector.
is_main : bool ,
2020-05-29 17:41:39 -04:00
}
2022-06-28 10:49:30 -04:00
pub ( crate ) struct DynImportModEvaluate {
2021-04-28 12:28:46 -04:00
load_id : ModuleLoadId ,
2020-10-14 08:04:09 -04:00
module_id : ModuleId ,
promise : v8 ::Global < v8 ::Promise > ,
module : v8 ::Global < v8 ::Module > ,
}
2022-06-28 10:49:30 -04:00
pub ( crate ) struct ModEvaluate {
2022-07-18 15:20:38 -04:00
pub ( crate ) promise : Option < v8 ::Global < v8 ::Promise > > ,
pub ( crate ) has_evaluated : bool ,
pub ( crate ) handled_promise_rejections : Vec < v8 ::Global < v8 ::Promise > > ,
2021-11-16 09:02:28 -05:00
sender : oneshot ::Sender < Result < ( ) , Error > > ,
2020-10-14 08:04:09 -04:00
}
2021-09-29 04:47:24 -04:00
pub struct CrossIsolateStore < T > ( Arc < Mutex < CrossIsolateStoreInner < T > > > ) ;
2021-07-06 13:42:52 -04:00
2021-09-29 04:47:24 -04:00
struct CrossIsolateStoreInner < T > {
map : HashMap < u32 , T > ,
2021-07-06 13:42:52 -04:00
last_id : u32 ,
}
2021-09-29 04:47:24 -04:00
impl < T > CrossIsolateStore < T > {
pub ( crate ) fn insert ( & self , value : T ) -> u32 {
let mut store = self . 0. lock ( ) . unwrap ( ) ;
let last_id = store . last_id ;
store . map . insert ( last_id , value ) ;
store . last_id + = 1 ;
2021-07-06 13:42:52 -04:00
last_id
}
2021-09-29 04:47:24 -04:00
pub ( crate ) fn take ( & self , id : u32 ) -> Option < T > {
let mut store = self . 0. lock ( ) . unwrap ( ) ;
store . map . remove ( & id )
}
}
impl < T > Default for CrossIsolateStore < T > {
fn default ( ) -> Self {
CrossIsolateStore ( Arc ::new ( Mutex ::new ( CrossIsolateStoreInner {
map : Default ::default ( ) ,
last_id : 0 ,
} ) ) )
2021-07-06 13:42:52 -04:00
}
}
2021-09-29 04:47:24 -04:00
impl < T > Clone for CrossIsolateStore < T > {
fn clone ( & self ) -> Self {
Self ( self . 0. clone ( ) )
}
}
pub type SharedArrayBufferStore =
CrossIsolateStore < v8 ::SharedRef < v8 ::BackingStore > > ;
pub type CompiledWasmModuleStore = CrossIsolateStore < v8 ::CompiledWasmModule > ;
2020-09-06 15:44:29 -04:00
/// Internal state for JsRuntime which is stored in one of v8::Isolate's
2020-05-29 17:41:39 -04:00
/// embedder slots.
2022-10-21 10:13:42 -04:00
pub struct JsRuntimeState {
2022-04-17 07:53:08 -04:00
global_realm : Option < JsRealm > ,
2022-10-20 22:54:22 -04:00
pub ( crate ) js_recv_cb : Option < v8 ::Global < v8 ::Function > > ,
2021-11-16 14:23:12 -05:00
pub ( crate ) js_macrotask_cbs : Vec < v8 ::Global < v8 ::Function > > ,
pub ( crate ) js_nexttick_cbs : Vec < v8 ::Global < v8 ::Function > > ,
2022-10-20 22:54:22 -04:00
pub ( crate ) js_promise_reject_cb : Option < v8 ::Global < v8 ::Function > > ,
pub ( crate ) js_format_exception_cb : Option < v8 ::Global < v8 ::Function > > ,
pub ( crate ) js_build_custom_error_cb : Option < v8 ::Global < v8 ::Function > > ,
2021-11-16 14:23:12 -05:00
pub ( crate ) has_tick_scheduled : bool ,
2022-10-20 22:54:22 -04:00
pub ( crate ) js_wasm_streaming_cb : Option < v8 ::Global < v8 ::Function > > ,
2020-11-11 17:11:40 -05:00
pub ( crate ) pending_promise_exceptions :
HashMap < v8 ::Global < v8 ::Promise > , v8 ::Global < v8 ::Value > > ,
2022-06-28 10:49:30 -04:00
pub ( crate ) pending_dyn_mod_evaluate : Vec < DynImportModEvaluate > ,
pub ( crate ) pending_mod_evaluate : Option < ModEvaluate > ,
2021-07-05 12:59:49 -04:00
/// A counter used to delay our dynamic import deadlock detection by one spin
/// of the event loop.
dyn_module_evaluate_idle_counter : u32 ,
2022-04-15 10:08:09 -04:00
pub ( crate ) source_map_getter : Option < Box < dyn SourceMapGetter > > ,
2022-06-20 08:42:20 -04:00
pub ( crate ) source_map_cache : SourceMapCache ,
2020-09-05 20:34:02 -04:00
pub ( crate ) pending_ops : FuturesUnordered < PendingOpFuture > ,
2022-10-20 22:54:22 -04:00
pub ( crate ) unrefed_ops : HashSet < i32 > ,
2021-02-23 07:08:50 -05:00
pub ( crate ) have_unpolled_ops : bool ,
2020-09-10 09:57:45 -04:00
pub ( crate ) op_state : Rc < RefCell < OpState > > ,
2022-04-08 04:32:48 -04:00
#[ allow(dead_code) ]
// We don't explicitly re-read this prop but need the slice to live alongside the isolate
pub ( crate ) op_ctxs : Box < [ OpCtx ] > ,
2021-07-06 13:42:52 -04:00
pub ( crate ) shared_array_buffer_store : Option < SharedArrayBufferStore > ,
2021-09-29 04:47:24 -04:00
pub ( crate ) compiled_wasm_module_store : Option < CompiledWasmModuleStore > ,
2022-09-02 06:43:39 -04:00
/// The error that was passed to an `op_dispatch_exception` call.
2022-04-13 05:50:57 -04:00
/// It will be retrieved by `exception_to_err_result` and used as an error
/// instead of any other exceptions.
2022-09-02 06:43:39 -04:00
// TODO(nayeemrmn): This is polled in `exception_to_err_result()` which is
// flimsy. Try to poll it similarly to `pending_promise_exceptions`.
pub ( crate ) dispatched_exceptions : VecDeque < v8 ::Global < v8 ::Value > > ,
2022-10-26 07:37:50 -04:00
pub ( crate ) inspector : Option < Rc < RefCell < JsRuntimeInspector > > > ,
2019-11-16 19:17:47 -05:00
waker : AtomicWaker ,
2020-05-29 17:41:39 -04:00
}
2022-05-17 11:19:55 -04:00
fn v8_init (
v8_platform : Option < v8 ::SharedRef < v8 ::Platform > > ,
predictable : bool ,
) {
2021-04-12 06:15:04 -04:00
// Include 10MB ICU data file.
#[ repr(C, align(16)) ]
2022-05-26 07:13:01 -04:00
struct IcuData ( [ u8 ; 10454784 ] ) ;
2021-04-12 06:15:04 -04:00
static ICU_DATA : IcuData = IcuData ( * include_bytes! ( " icudtl.dat " ) ) ;
2022-05-26 07:13:01 -04:00
v8 ::icu ::set_common_data_71 ( & ICU_DATA . 0 ) . unwrap ( ) ;
2021-04-12 06:15:04 -04:00
2021-04-12 06:31:20 -04:00
let flags = concat! (
2021-07-03 17:33:36 -04:00
" --wasm-test-streaming " ,
2021-04-12 06:31:20 -04:00
" --harmony-import-assertions " ,
" --no-validate-asm " ,
2022-07-08 13:49:09 -04:00
" --turbo_fast_api_calls " ,
2022-10-26 10:54:22 -04:00
" --harmony-change-array-by-copy " ,
2021-04-12 06:31:20 -04:00
) ;
2022-05-17 11:19:55 -04:00
if predictable {
v8 ::V8 ::set_flags_from_string ( & format! (
" {}{} " ,
flags , " --predictable --random-seed=42 "
) ) ;
} else {
v8 ::V8 ::set_flags_from_string ( flags ) ;
}
2022-08-15 08:12:11 -04:00
let v8_platform = v8_platform
. unwrap_or_else ( | | v8 ::new_default_platform ( 0 , false ) . make_shared ( ) ) ;
v8 ::V8 ::initialize_platform ( v8_platform ) ;
v8 ::V8 ::initialize ( ) ;
2020-01-06 14:07:35 -05:00
}
2022-08-18 08:05:02 -04:00
pub const V8_WRAPPER_TYPE_INDEX : i32 = 0 ;
pub const V8_WRAPPER_OBJECT_INDEX : i32 = 1 ;
2020-09-11 09:18:49 -04:00
#[ derive(Default) ]
pub struct RuntimeOptions {
2022-04-15 10:08:09 -04:00
/// Source map reference for errors.
pub source_map_getter : Option < Box < dyn SourceMapGetter > > ,
2020-11-21 09:56:14 -05:00
/// Allows to map error type to a string "class" used to represent
/// error in JavaScript.
pub get_error_class_fn : Option < GetErrorClassFn > ,
2020-09-11 09:18:49 -04:00
/// Implementation of `ModuleLoader` which will be
/// called when V8 requests to load ES modules.
///
/// If not provided runtime will error if code being
/// executed tries to load modules.
pub module_loader : Option < Rc < dyn ModuleLoader > > ,
2020-09-06 10:50:49 -04:00
2021-04-28 12:41:50 -04:00
/// JsRuntime extensions, not to be confused with ES modules
/// these are sets of ops and other JS code to be initialized.
pub extensions : Vec < Extension > ,
2020-09-11 09:18:49 -04:00
/// V8 snapshot that should be loaded on startup.
///
/// Currently can't be used with `will_snapshot`.
pub startup_snapshot : Option < Snapshot > ,
2020-08-11 21:07:14 -04:00
2020-09-11 09:18:49 -04:00
/// Prepare runtime to take snapshot of loaded code.
2022-05-17 11:19:55 -04:00
/// The snapshot is determinstic and uses predictable random numbers.
2020-09-11 09:18:49 -04:00
///
/// Currently can't be used with `startup_snapshot`.
pub will_snapshot : bool ,
2020-08-11 21:07:14 -04:00
2020-10-17 05:56:15 -04:00
/// Isolate creation parameters.
pub create_params : Option < v8 ::CreateParams > ,
2021-04-12 06:15:04 -04:00
/// V8 platform instance to use. Used when Deno initializes V8
/// (which it only does once), otherwise it's silenty dropped.
2021-07-02 03:32:48 -04:00
pub v8_platform : Option < v8 ::SharedRef < v8 ::Platform > > ,
2021-07-06 13:42:52 -04:00
2021-09-29 04:47:24 -04:00
/// The store to use for transferring SharedArrayBuffers between isolates.
2021-07-06 13:42:52 -04:00
/// If multiple isolates should have the possibility of sharing
2021-09-29 04:47:24 -04:00
/// SharedArrayBuffers, they should use the same [SharedArrayBufferStore]. If
/// no [SharedArrayBufferStore] is specified, SharedArrayBuffer can not be
/// serialized.
2021-07-06 13:42:52 -04:00
pub shared_array_buffer_store : Option < SharedArrayBufferStore > ,
2021-09-29 04:47:24 -04:00
/// The store to use for transferring `WebAssembly.Module` objects between
/// isolates.
/// If multiple isolates should have the possibility of sharing
/// `WebAssembly.Module` objects, they should use the same
/// [CompiledWasmModuleStore]. If no [CompiledWasmModuleStore] is specified,
/// `WebAssembly.Module` objects cannot be serialized.
pub compiled_wasm_module_store : Option < CompiledWasmModuleStore > ,
2022-11-26 17:09:48 -05:00
/// Start inspector instance to allow debuggers to connect.
2022-10-26 07:37:50 -04:00
pub inspector : bool ,
2022-11-26 17:09:48 -05:00
/// Describe if this is the main runtime instance, used by debuggers in some
/// situation - like disconnecting when program finishes running.
pub is_main : bool ,
2020-09-11 09:18:49 -04:00
}
2020-08-11 21:07:14 -04:00
2022-10-28 15:31:01 -04:00
#[ derive(Copy, Clone, PartialEq, Eq) ]
pub enum SnapshotOptions {
Load ,
CreateFromExisting ,
Create ,
None ,
}
impl SnapshotOptions {
pub fn loaded ( & self ) -> bool {
matches! (
self ,
SnapshotOptions ::Load | SnapshotOptions ::CreateFromExisting
)
}
pub fn will_snapshot ( & self ) -> bool {
matches! (
self ,
SnapshotOptions ::Create | SnapshotOptions ::CreateFromExisting
)
}
fn from_bools ( snapshot_loaded : bool , will_snapshot : bool ) -> Self {
match ( snapshot_loaded , will_snapshot ) {
( true , true ) = > SnapshotOptions ::CreateFromExisting ,
( false , true ) = > SnapshotOptions ::Create ,
( true , false ) = > SnapshotOptions ::Load ,
( false , false ) = > SnapshotOptions ::None ,
}
}
}
2022-10-15 10:01:01 -04:00
impl Drop for JsRuntime {
fn drop ( & mut self ) {
if let Some ( v8_isolate ) = self . v8_isolate . as_mut ( ) {
Self ::drop_state_and_module_map ( v8_isolate ) ;
}
}
}
2020-09-11 09:18:49 -04:00
impl JsRuntime {
2022-10-15 10:01:01 -04:00
const STATE_DATA_OFFSET : u32 = 0 ;
const MODULE_MAP_DATA_OFFSET : u32 = 1 ;
2020-11-05 20:26:14 -05:00
/// Only constructor, configuration is done through `options`.
2020-10-17 05:56:15 -04:00
pub fn new ( mut options : RuntimeOptions ) -> Self {
2021-04-12 06:15:04 -04:00
let v8_platform = options . v8_platform . take ( ) ;
2020-05-29 17:41:39 -04:00
static DENO_INIT : Once = Once ::new ( ) ;
2022-05-17 11:19:55 -04:00
DENO_INIT . call_once ( move | | v8_init ( v8_platform , options . will_snapshot ) ) ;
2019-03-11 17:57:36 -04:00
2021-01-05 16:10:50 -05:00
let has_startup_snapshot = options . startup_snapshot . is_some ( ) ;
2022-03-14 13:44:15 -04:00
// Add builtins extension
options
. extensions
. insert ( 0 , crate ::ops_builtin ::init_builtins ( ) ) ;
let ops = Self ::collect_ops ( & mut options . extensions ) ;
let mut op_state = OpState ::new ( ops . len ( ) ) ;
if let Some ( get_error_class_fn ) = options . get_error_class_fn {
op_state . get_error_class_fn = get_error_class_fn ;
}
let op_state = Rc ::new ( RefCell ::new ( op_state ) ) ;
2022-10-21 10:13:42 -04:00
let align = std ::mem ::align_of ::< usize > ( ) ;
let layout = std ::alloc ::Layout ::from_size_align (
std ::mem ::size_of ::< * mut v8 ::OwnedIsolate > ( ) ,
align ,
)
. unwrap ( ) ;
assert! ( layout . size ( ) > 0 ) ;
let isolate_ptr : * mut v8 ::OwnedIsolate =
// SAFETY: we just asserted that layout has non-0 size.
unsafe { std ::alloc ::alloc ( layout ) as * mut _ } ;
let state_rc = Rc ::new ( RefCell ::new ( JsRuntimeState {
pending_promise_exceptions : HashMap ::new ( ) ,
pending_dyn_mod_evaluate : vec ! [ ] ,
pending_mod_evaluate : None ,
dyn_module_evaluate_idle_counter : 0 ,
js_recv_cb : None ,
js_macrotask_cbs : vec ! [ ] ,
js_nexttick_cbs : vec ! [ ] ,
js_promise_reject_cb : None ,
js_format_exception_cb : None ,
js_build_custom_error_cb : None ,
has_tick_scheduled : false ,
js_wasm_streaming_cb : None ,
source_map_getter : options . source_map_getter ,
source_map_cache : Default ::default ( ) ,
pending_ops : FuturesUnordered ::new ( ) ,
unrefed_ops : HashSet ::new ( ) ,
shared_array_buffer_store : options . shared_array_buffer_store ,
compiled_wasm_module_store : options . compiled_wasm_module_store ,
op_state : op_state . clone ( ) ,
waker : AtomicWaker ::new ( ) ,
have_unpolled_ops : false ,
dispatched_exceptions : Default ::default ( ) ,
// Some fields are initialized later after isolate is created
inspector : None ,
op_ctxs : vec ! [ ] . into_boxed_slice ( ) ,
global_realm : None ,
} ) ) ;
let weak = Rc ::downgrade ( & state_rc ) ;
2022-04-08 04:32:48 -04:00
let op_ctxs = ops
. into_iter ( )
. enumerate ( )
. map ( | ( id , decl ) | OpCtx {
id ,
state : op_state . clone ( ) ,
2022-10-21 10:13:42 -04:00
runtime_state : weak . clone ( ) ,
2022-04-08 04:32:48 -04:00
decl ,
} )
. collect ::< Vec < _ > > ( )
. into_boxed_slice ( ) ;
2022-03-14 13:44:15 -04:00
2022-08-21 08:07:53 -04:00
let refs = bindings ::external_references ( & op_ctxs , ! options . will_snapshot ) ;
// V8 takes ownership of external_references.
let refs : & 'static v8 ::ExternalReferences = Box ::leak ( Box ::new ( refs ) ) ;
2020-07-18 16:32:11 -04:00
let global_context ;
2022-10-05 10:06:44 -04:00
2022-10-28 15:31:01 -04:00
let ( mut isolate , snapshot_options ) = if options . will_snapshot {
let ( snapshot_creator , snapshot_loaded ) =
if let Some ( snapshot ) = options . startup_snapshot {
(
match snapshot {
Snapshot ::Static ( data ) = > {
v8 ::Isolate ::snapshot_creator_from_existing_snapshot (
data ,
Some ( refs ) ,
)
}
Snapshot ::JustCreated ( data ) = > {
v8 ::Isolate ::snapshot_creator_from_existing_snapshot (
data ,
Some ( refs ) ,
)
}
Snapshot ::Boxed ( data ) = > {
v8 ::Isolate ::snapshot_creator_from_existing_snapshot (
data ,
Some ( refs ) ,
)
}
} ,
true ,
)
} else {
( v8 ::Isolate ::snapshot_creator ( Some ( refs ) ) , false )
} ;
let snapshot_options =
SnapshotOptions ::from_bools ( snapshot_loaded , options . will_snapshot ) ;
2022-10-14 19:53:13 -04:00
let mut isolate = JsRuntime ::setup_isolate ( snapshot_creator ) ;
2020-06-20 07:18:08 -04:00
{
2022-10-05 10:06:44 -04:00
// SAFETY: this is first use of `isolate_ptr` so we are sure we're
// not overwriting an existing pointer.
isolate = unsafe {
isolate_ptr . write ( isolate ) ;
isolate_ptr . read ( )
} ;
2020-06-20 07:18:08 -04:00
let scope = & mut v8 ::HandleScope ::new ( & mut isolate ) ;
2022-10-28 07:20:17 -04:00
let context =
2022-10-28 15:31:01 -04:00
bindings ::initialize_context ( scope , & op_ctxs , snapshot_options ) ;
2020-07-18 16:32:11 -04:00
global_context = v8 ::Global ::new ( scope , context ) ;
2022-10-14 19:53:13 -04:00
scope . set_default_context ( context ) ;
2020-06-20 07:18:08 -04:00
}
2022-10-28 15:31:01 -04:00
( isolate , snapshot_options )
2020-01-06 10:24:44 -05:00
} else {
2020-10-17 05:56:15 -04:00
let mut params = options
. create_params
. take ( )
2022-08-18 08:05:02 -04:00
. unwrap_or_else ( | | {
v8 ::CreateParams ::default ( ) . embedder_wrapper_type_info_offsets (
V8_WRAPPER_TYPE_INDEX ,
V8_WRAPPER_OBJECT_INDEX ,
)
} )
2022-08-21 08:07:53 -04:00
. external_references ( & * * refs ) ;
2020-08-11 21:07:14 -04:00
let snapshot_loaded = if let Some ( snapshot ) = options . startup_snapshot {
2020-04-22 14:24:49 -04:00
params = match snapshot {
Snapshot ::Static ( data ) = > params . snapshot_blob ( data ) ,
Snapshot ::JustCreated ( data ) = > params . snapshot_blob ( data ) ,
2020-05-09 21:00:40 -04:00
Snapshot ::Boxed ( data ) = > params . snapshot_blob ( data ) ,
2020-04-22 14:24:49 -04:00
} ;
true
} else {
false
} ;
2020-01-06 10:24:44 -05:00
2022-10-28 15:31:01 -04:00
let snapshot_options =
SnapshotOptions ::from_bools ( snapshot_loaded , options . will_snapshot ) ;
2020-01-06 10:24:44 -05:00
let isolate = v8 ::Isolate ::new ( params ) ;
2020-09-06 15:44:29 -04:00
let mut isolate = JsRuntime ::setup_isolate ( isolate ) ;
2020-06-20 07:18:08 -04:00
{
2022-10-05 10:06:44 -04:00
// SAFETY: this is first use of `isolate_ptr` so we are sure we're
// not overwriting an existing pointer.
isolate = unsafe {
isolate_ptr . write ( isolate ) ;
isolate_ptr . read ( )
} ;
2020-06-20 07:18:08 -04:00
let scope = & mut v8 ::HandleScope ::new ( & mut isolate ) ;
2022-04-08 04:32:48 -04:00
let context =
2022-10-28 15:31:01 -04:00
bindings ::initialize_context ( scope , & op_ctxs , snapshot_options ) ;
2022-03-14 13:44:15 -04:00
2020-07-18 16:32:11 -04:00
global_context = v8 ::Global ::new ( scope , context ) ;
2020-06-20 07:18:08 -04:00
}
2022-10-05 10:06:44 -04:00
2022-10-28 15:31:01 -04:00
( isolate , snapshot_options )
2020-01-06 10:24:44 -05:00
} ;
2022-10-05 10:06:44 -04:00
op_state . borrow_mut ( ) . put ( isolate_ptr ) ;
2022-10-26 07:37:50 -04:00
let inspector = if options . inspector {
Some ( JsRuntimeInspector ::new (
& mut isolate ,
global_context . clone ( ) ,
2022-11-26 17:09:48 -05:00
options . is_main ,
2022-10-26 07:37:50 -04:00
) )
} else {
None
} ;
2021-05-26 15:07:12 -04:00
2020-09-11 09:18:49 -04:00
let loader = options
. module_loader
. unwrap_or_else ( | | Rc ::new ( NoopModuleLoader ) ) ;
2022-10-21 10:13:42 -04:00
{
let mut state = state_rc . borrow_mut ( ) ;
state . global_realm = Some ( JsRealm ( global_context ) ) ;
state . op_ctxs = op_ctxs ;
2022-10-26 07:37:50 -04:00
state . inspector = inspector ;
2022-10-21 10:13:42 -04:00
}
2022-10-15 10:01:01 -04:00
isolate . set_data (
Self ::STATE_DATA_OFFSET ,
2022-10-21 10:13:42 -04:00
Rc ::into_raw ( state_rc . clone ( ) ) as * mut c_void ,
2022-10-15 10:01:01 -04:00
) ;
2020-01-06 10:24:44 -05:00
2022-10-15 10:01:01 -04:00
let module_map_rc = Rc ::new ( RefCell ::new ( ModuleMap ::new ( loader , op_state ) ) ) ;
isolate . set_data (
Self ::MODULE_MAP_DATA_OFFSET ,
2022-10-25 08:27:33 -04:00
Rc ::into_raw ( module_map_rc . clone ( ) ) as * mut c_void ,
2022-10-15 10:01:01 -04:00
) ;
2021-05-19 14:53:43 -04:00
2021-01-05 16:10:50 -05:00
let mut js_runtime = Self {
2020-05-29 17:41:39 -04:00
v8_isolate : Some ( isolate ) ,
2022-04-17 07:53:08 -04:00
built_from_snapshot : has_startup_snapshot ,
2022-10-28 15:31:01 -04:00
snapshot_options ,
2020-08-11 21:07:14 -04:00
allocations : IsolateAllocations ::default ( ) ,
2022-03-08 09:40:34 -05:00
event_loop_middlewares : Vec ::with_capacity ( options . extensions . len ( ) ) ,
2021-04-28 12:41:50 -04:00
extensions : options . extensions ,
2022-10-21 10:13:42 -04:00
state : state_rc ,
2022-10-25 08:27:33 -04:00
module_map : Some ( module_map_rc ) ,
2022-11-26 17:09:48 -05:00
is_main : options . is_main ,
2021-01-05 16:10:50 -05:00
} ;
2022-10-07 08:38:06 -04:00
// Init resources and ops before extensions to make sure they are
// available during the initialization process.
js_runtime . init_extension_ops ( ) . unwrap ( ) ;
2021-04-30 10:38:35 -04:00
// TODO(@AaronO): diff extensions inited in snapshot and those provided
// for now we assume that snapshot and extensions always match
2021-01-05 16:10:50 -05:00
if ! has_startup_snapshot {
2022-04-17 07:53:08 -04:00
let realm = js_runtime . global_realm ( ) ;
js_runtime . init_extension_js ( & realm ) . unwrap ( ) ;
2020-01-06 10:24:44 -05:00
}
2022-03-14 13:44:15 -04:00
// Init callbacks (opresolve)
2022-10-20 22:54:22 -04:00
js_runtime . init_cbs ( ) ;
2021-01-05 16:10:50 -05:00
js_runtime
2020-01-06 10:24:44 -05:00
}
2022-10-15 10:01:01 -04:00
fn drop_state_and_module_map ( v8_isolate : & mut OwnedIsolate ) {
let state_ptr = v8_isolate . get_data ( Self ::STATE_DATA_OFFSET ) ;
let state_rc =
// SAFETY: We are sure that it's a valid pointer for whole lifetime of
// the runtime.
unsafe { Rc ::from_raw ( state_ptr as * const RefCell < JsRuntimeState > ) } ;
drop ( state_rc ) ;
let module_map_ptr = v8_isolate . get_data ( Self ::MODULE_MAP_DATA_OFFSET ) ;
let module_map_rc =
// SAFETY: We are sure that it's a valid pointer for whole lifetime of
// the runtime.
unsafe { Rc ::from_raw ( module_map_ptr as * const RefCell < ModuleMap > ) } ;
drop ( module_map_rc ) ;
}
2022-10-25 08:27:33 -04:00
#[ inline ]
fn get_module_map ( & mut self ) -> & Rc < RefCell < ModuleMap > > {
self . module_map . as_ref ( ) . unwrap ( )
}
2022-10-21 10:13:42 -04:00
#[ inline ]
2020-10-07 09:56:52 -04:00
pub fn global_context ( & mut self ) -> v8 ::Global < v8 ::Context > {
2022-04-17 07:53:08 -04:00
self . global_realm ( ) . 0
2020-09-14 23:49:12 -04:00
}
2022-10-21 10:13:42 -04:00
#[ inline ]
2020-10-07 09:56:52 -04:00
pub fn v8_isolate ( & mut self ) -> & mut v8 ::OwnedIsolate {
2020-10-05 05:08:19 -04:00
self . v8_isolate . as_mut ( ) . unwrap ( )
}
2022-10-21 10:13:42 -04:00
#[ inline ]
2022-09-02 06:43:39 -04:00
pub fn inspector ( & mut self ) -> Rc < RefCell < JsRuntimeInspector > > {
2022-10-21 10:13:42 -04:00
self . state . borrow ( ) . inspector ( )
2021-05-26 15:07:12 -04:00
}
2022-10-21 10:13:42 -04:00
#[ inline ]
2022-04-17 07:53:08 -04:00
pub fn global_realm ( & mut self ) -> JsRealm {
2022-10-21 10:13:42 -04:00
let state = self . state . borrow ( ) ;
2022-04-17 07:53:08 -04:00
state . global_realm . clone ( ) . unwrap ( )
}
2022-10-15 16:44:51 -04:00
/// Creates a new realm (V8 context) in this JS execution context,
/// pre-initialized with all of the extensions that were passed in
/// [`RuntimeOptions::extensions`] when the [`JsRuntime`] was constructed.
///
/// If the [`JsRuntime`] was not built from a snapshot (see
/// [`RuntimeOptions::startup_snapshot`]), the JS code for the extensions will
/// be run in the call to this method. In contrast, if there is a snapshot,
/// that will be used instead, and the extensions' initialization will come
/// "for free".
2022-04-17 07:53:08 -04:00
pub fn create_realm ( & mut self ) -> Result < JsRealm , Error > {
let realm = {
// SAFETY: Having the scope tied to self's lifetime makes it impossible to
// reference self.ops while the scope is alive. Here we turn it into an
// unbound lifetime, which is sound because 1. it only lives until the end
// of this block, and 2. the HandleScope only has access to the isolate,
// and nothing else we're accessing from self does.
let scope = & mut v8 ::HandleScope ::new ( unsafe {
& mut * ( self . v8_isolate ( ) as * mut v8 ::OwnedIsolate )
} ) ;
let context = bindings ::initialize_context (
scope ,
2022-10-21 10:13:42 -04:00
& self . state . borrow ( ) . op_ctxs ,
2022-10-28 15:31:01 -04:00
SnapshotOptions ::from_bools (
self . built_from_snapshot ,
self . snapshot_options . will_snapshot ( ) ,
) ,
2022-04-17 07:53:08 -04:00
) ;
JsRealm ::new ( v8 ::Global ::new ( scope , context ) )
} ;
if ! self . built_from_snapshot {
self . init_extension_js ( & realm ) ? ;
}
Ok ( realm )
}
2022-10-21 10:13:42 -04:00
#[ inline ]
2021-04-28 12:28:46 -04:00
pub fn handle_scope ( & mut self ) -> v8 ::HandleScope {
2022-07-05 18:45:10 -04:00
self . global_realm ( ) . handle_scope ( self . v8_isolate ( ) )
2021-04-28 12:28:46 -04:00
}
2020-04-23 05:51:07 -04:00
fn setup_isolate ( mut isolate : v8 ::OwnedIsolate ) -> v8 ::OwnedIsolate {
2020-01-06 10:24:44 -05:00
isolate . set_capture_stack_trace_for_uncaught_exceptions ( true , 10 ) ;
isolate . set_promise_reject_callback ( bindings ::promise_reject_callback ) ;
2020-09-06 10:50:49 -04:00
isolate . set_host_initialize_import_meta_object_callback (
bindings ::host_initialize_import_meta_object_callback ,
) ;
isolate . set_host_import_module_dynamically_callback (
bindings ::host_import_module_dynamically_callback ,
) ;
2022-08-15 08:12:11 -04:00
isolate . set_wasm_async_resolve_promise_callback (
bindings ::wasm_async_resolve_promise_callback ,
) ;
2020-01-06 10:24:44 -05:00
isolate
}
2020-09-14 23:49:12 -04:00
pub ( crate ) fn state ( isolate : & v8 ::Isolate ) -> Rc < RefCell < JsRuntimeState > > {
2022-10-15 10:01:01 -04:00
let state_ptr = isolate . get_data ( Self ::STATE_DATA_OFFSET ) ;
let state_rc =
// SAFETY: We are sure that it's a valid pointer for whole lifetime of
// the runtime.
unsafe { Rc ::from_raw ( state_ptr as * const RefCell < JsRuntimeState > ) } ;
let state = state_rc . clone ( ) ;
Rc ::into_raw ( state_rc ) ;
state
2019-03-21 09:48:19 -04:00
}
2021-05-19 14:53:43 -04:00
pub ( crate ) fn module_map ( isolate : & v8 ::Isolate ) -> Rc < RefCell < ModuleMap > > {
2022-10-15 10:01:01 -04:00
let module_map_ptr = isolate . get_data ( Self ::MODULE_MAP_DATA_OFFSET ) ;
let module_map_rc =
// SAFETY: We are sure that it's a valid pointer for whole lifetime of
// the runtime.
unsafe { Rc ::from_raw ( module_map_ptr as * const RefCell < ModuleMap > ) } ;
let module_map = module_map_rc . clone ( ) ;
Rc ::into_raw ( module_map_rc ) ;
module_map
2021-05-19 14:53:43 -04:00
}
2022-04-17 07:53:08 -04:00
/// Initializes JS of provided Extensions in the given realm
fn init_extension_js ( & mut self , realm : & JsRealm ) -> Result < ( ) , Error > {
2021-04-28 12:41:50 -04:00
// Take extensions to avoid double-borrow
let mut extensions : Vec < Extension > = std ::mem ::take ( & mut self . extensions ) ;
for m in extensions . iter_mut ( ) {
let js_files = m . init_js ( ) ;
for ( filename , source ) in js_files {
// TODO(@AaronO): use JsRuntime::execute_static() here to move src off heap
2022-07-05 18:45:10 -04:00
realm . execute_script ( self . v8_isolate ( ) , filename , source ) ? ;
2021-04-28 12:41:50 -04:00
}
}
// Restore extensions
self . extensions = extensions ;
Ok ( ( ) )
}
2022-03-14 13:44:15 -04:00
/// Collects ops from extensions & applies middleware
2022-03-15 18:43:17 -04:00
fn collect_ops ( extensions : & mut [ Extension ] ) -> Vec < OpDecl > {
2021-04-28 12:41:50 -04:00
// Middleware
let middleware : Vec < Box < OpMiddlewareFn > > = extensions
. iter_mut ( )
. filter_map ( | e | e . init_middleware ( ) )
. collect ( ) ;
2022-03-14 13:44:15 -04:00
2021-04-28 12:41:50 -04:00
// macroware wraps an opfn in all the middleware
2022-03-15 18:43:17 -04:00
let macroware = move | d | middleware . iter ( ) . fold ( d , | d , m | m ( d ) ) ;
2021-04-28 12:41:50 -04:00
2022-03-22 11:39:58 -04:00
// Flatten ops, apply middlware & override disabled ops
2022-03-14 13:44:15 -04:00
extensions
. iter_mut ( )
. filter_map ( | e | e . init_ops ( ) )
. flatten ( )
2022-03-15 18:43:17 -04:00
. map ( | d | OpDecl {
name : d . name ,
.. macroware ( d )
} )
2022-03-22 11:39:58 -04:00
. map ( | op | match op . enabled {
true = > op ,
false = > OpDecl {
v8_fn_ptr : match op . is_async {
true = > op_void_async ::v8_fn_ptr ( ) ,
false = > op_void_sync ::v8_fn_ptr ( ) ,
} ,
.. op
} ,
} )
2022-03-14 13:44:15 -04:00
. collect ( )
}
/// Initializes ops of provided Extensions
fn init_extension_ops ( & mut self ) -> Result < ( ) , Error > {
let op_state = self . op_state ( ) ;
// Take extensions to avoid double-borrow
let mut extensions : Vec < Extension > = std ::mem ::take ( & mut self . extensions ) ;
// Setup state
2021-04-28 12:41:50 -04:00
for e in extensions . iter_mut ( ) {
2022-03-14 13:44:15 -04:00
// ops are already registered during in bindings::initialize_context();
2021-04-28 12:41:50 -04:00
e . init_state ( & mut op_state . borrow_mut ( ) ) ? ;
2022-03-08 09:40:34 -05:00
2022-03-14 13:44:15 -04:00
// Setup event-loop middleware
2022-03-08 09:40:34 -05:00
if let Some ( middleware ) = e . init_event_loop_middleware ( ) {
self . event_loop_middlewares . push ( middleware ) ;
}
2021-04-28 12:41:50 -04:00
}
2022-03-14 13:44:15 -04:00
2021-04-28 12:41:50 -04:00
// Restore extensions
self . extensions = extensions ;
Ok ( ( ) )
}
2022-11-07 20:39:48 -05:00
pub fn eval < ' s , T > (
2022-03-16 04:04:38 -04:00
scope : & mut v8 ::HandleScope < ' s > ,
2022-11-07 20:39:48 -05:00
code : & str ,
2022-03-16 04:04:38 -04:00
) -> Option < v8 ::Local < ' s , T > >
where
v8 ::Local < ' s , T > : TryFrom < v8 ::Local < ' s , v8 ::Value > , Error = v8 ::DataError > ,
{
2022-11-07 20:39:48 -05:00
let scope = & mut v8 ::EscapableHandleScope ::new ( scope ) ;
let source = v8 ::String ::new ( scope , code ) . unwrap ( ) ;
let script = v8 ::Script ::compile ( scope , source , None ) . unwrap ( ) ;
let v = script . run ( scope ) ? ;
scope . escape ( v ) . try_into ( ) . ok ( )
2022-03-15 17:50:17 -04:00
}
2021-10-04 05:45:41 -04:00
/// Grabs a reference to core.js' opresolve & syncOpsCache()
2022-10-20 22:54:22 -04:00
fn init_cbs ( & mut self ) {
let scope = & mut self . handle_scope ( ) ;
let recv_cb =
2022-11-07 20:39:48 -05:00
Self ::eval ::< v8 ::Function > ( scope , " Deno.core.opresolve " ) . unwrap ( ) ;
2022-10-20 22:54:22 -04:00
let recv_cb = v8 ::Global ::new ( scope , recv_cb ) ;
let build_custom_error_cb =
2022-11-07 20:39:48 -05:00
Self ::eval ::< v8 ::Function > ( scope , " Deno.core.buildCustomError " )
2022-10-20 22:54:22 -04:00
. expect ( " Deno.core.buildCustomError is undefined in the realm " ) ;
let build_custom_error_cb = v8 ::Global ::new ( scope , build_custom_error_cb ) ;
// Put global handles in state
let state_rc = JsRuntime ::state ( scope ) ;
let mut state = state_rc . borrow_mut ( ) ;
state . js_recv_cb . replace ( recv_cb ) ;
2022-08-11 05:57:20 -04:00
state
. js_build_custom_error_cb
. replace ( build_custom_error_cb ) ;
2021-04-25 16:00:05 -04:00
}
2020-11-05 20:26:14 -05:00
/// Returns the runtime's op state, which can be used to maintain ops
/// and access resources between op calls.
2020-09-10 09:57:45 -04:00
pub fn op_state ( & mut self ) -> Rc < RefCell < OpState > > {
2022-10-21 10:13:42 -04:00
let state = self . state . borrow ( ) ;
2020-09-10 09:57:45 -04:00
state . op_state . clone ( )
}
2021-06-21 19:45:41 -04:00
/// Executes traditional JavaScript code (traditional = not ES modules).
2020-01-07 06:45:44 -05:00
///
2020-11-05 20:26:14 -05:00
/// The execution takes place on the current global context, so it is possible
/// to maintain local JS state and invoke this method multiple times.
///
2021-06-21 19:45:41 -04:00
/// `name` can be a filepath or any other string, eg.
///
/// - "/some/file/path.js"
/// - "<anon>"
/// - "[native code]"
///
/// The same `name` value can be used for multiple executions.
///
2022-04-26 09:28:42 -04:00
/// `Error` can usually be downcast to `JsError`.
2021-06-21 19:45:41 -04:00
pub fn execute_script (
2020-01-06 10:24:44 -05:00
& mut self ,
2021-06-21 19:45:41 -04:00
name : & str ,
source_code : & str ,
2021-11-16 09:02:28 -05:00
) -> Result < v8 ::Global < v8 ::Value > , Error > {
2022-07-05 18:45:10 -04:00
self
. global_realm ( )
. execute_script ( self . v8_isolate ( ) , name , source_code )
2019-03-11 17:57:36 -04:00
}
2019-07-10 18:53:48 -04:00
/// Takes a snapshot. The isolate should have been created with will_snapshot
/// set to true.
///
2022-04-26 09:28:42 -04:00
/// `Error` can usually be downcast to `JsError`.
2022-10-14 19:53:13 -04:00
pub fn snapshot ( mut self ) -> v8 ::StartupData {
2022-03-15 17:50:17 -04:00
// Nuke Deno.core.ops.* to avoid ExternalReference snapshotting issues
// TODO(@AaronO): make ops stable across snapshots
{
let scope = & mut self . handle_scope ( ) ;
2022-11-07 20:39:48 -05:00
let o = Self ::eval ::< v8 ::Object > ( scope , " Deno.core.ops " ) . unwrap ( ) ;
2022-09-17 05:55:19 -04:00
let names = o . get_own_property_names ( scope , Default ::default ( ) ) . unwrap ( ) ;
2022-03-15 17:50:17 -04:00
for i in 0 .. names . length ( ) {
let key = names . get_index ( scope , i ) . unwrap ( ) ;
2022-03-16 04:04:38 -04:00
o . delete ( scope , key ) ;
2022-03-15 17:50:17 -04:00
}
}
2022-10-21 10:13:42 -04:00
self . state . borrow_mut ( ) . global_realm . take ( ) ;
self . state . borrow_mut ( ) . inspector . take ( ) ;
2021-06-21 13:37:51 -04:00
2022-10-15 10:01:01 -04:00
// Drop existing ModuleMap to drop v8::Global handles
{
2022-10-25 08:27:33 -04:00
self . module_map . take ( ) ;
2022-10-15 10:01:01 -04:00
let v8_isolate = self . v8_isolate ( ) ;
Self ::drop_state_and_module_map ( v8_isolate ) ;
}
2021-05-19 14:53:43 -04:00
// Drop other v8::Global handles before snapshotting
feat(core): Add support for async ops in realms (#14734)
Pull request #14019 enabled initial support for realms, but it did not
include support for async ops anywhere other than the main realm. The
main issue was that the `js_recv_cb` callback, which resolves promises
corresponding to async ops, was only set for the main realm, so async
ops in other realms would never resolve. Furthermore, promise ID's are
specific to each realm, which meant that async ops from other realms
would result in a wrong promise from the main realm being resolved.
This change creates a `ContextState` struct, similar to
`JsRuntimeState` but stored in a slot of each `v8::Context`, which
contains a `js_recv_cb` callback for each realm. Combined with a new
list of known realms, which stores them as `v8::Weak<v8::Context>`,
and a change in the `#[op]` macro to pass the current context to
`queue_async_op`, this makes it possible to send the results of
promises for different realms to their realm, and prevent the ID's
from getting mixed up.
Additionally, since promise ID's are no longer unique to the isolate,
having a single set of unrefed ops doesn't work. This change therefore
also moves `unrefed_ops` from `JsRuntimeState` to `ContextState`, and
adds the lengths of the unrefed op sets for all known realms to get
the total number of unrefed ops to compare in the event loop.
Co-authored-by: Luis Malheiro <luismalheiro@gmail.com>
2022-08-10 14:04:20 -04:00
{
2022-10-21 10:13:42 -04:00
let mut state = self . state . borrow_mut ( ) ;
2022-10-20 22:54:22 -04:00
std ::mem ::take ( & mut state . js_recv_cb ) ;
std ::mem ::take ( & mut state . js_promise_reject_cb ) ;
std ::mem ::take ( & mut state . js_format_exception_cb ) ;
std ::mem ::take ( & mut state . js_wasm_streaming_cb ) ;
std ::mem ::take ( & mut state . js_build_custom_error_cb ) ;
2022-08-21 15:03:56 -04:00
state . js_macrotask_cbs . clear ( ) ;
state . js_nexttick_cbs . clear ( ) ;
feat(core): Add support for async ops in realms (#14734)
Pull request #14019 enabled initial support for realms, but it did not
include support for async ops anywhere other than the main realm. The
main issue was that the `js_recv_cb` callback, which resolves promises
corresponding to async ops, was only set for the main realm, so async
ops in other realms would never resolve. Furthermore, promise ID's are
specific to each realm, which meant that async ops from other realms
would result in a wrong promise from the main realm being resolved.
This change creates a `ContextState` struct, similar to
`JsRuntimeState` but stored in a slot of each `v8::Context`, which
contains a `js_recv_cb` callback for each realm. Combined with a new
list of known realms, which stores them as `v8::Weak<v8::Context>`,
and a change in the `#[op]` macro to pass the current context to
`queue_async_op`, this makes it possible to send the results of
promises for different realms to their realm, and prevent the ID's
from getting mixed up.
Additionally, since promise ID's are no longer unique to the isolate,
having a single set of unrefed ops doesn't work. This change therefore
also moves `unrefed_ops` from `JsRuntimeState` to `ContextState`, and
adds the lengths of the unrefed op sets for all known realms to get
the total number of unrefed ops to compare in the event loop.
Co-authored-by: Luis Malheiro <luismalheiro@gmail.com>
2022-08-10 14:04:20 -04:00
}
2020-09-06 10:50:49 -04:00
2022-10-15 10:01:01 -04:00
let snapshot_creator = self . v8_isolate . take ( ) . unwrap ( ) ;
2022-10-14 19:53:13 -04:00
snapshot_creator
2020-01-06 10:24:44 -05:00
. create_blob ( v8 ::FunctionCodeHandling ::Keep )
2022-10-14 19:53:13 -04:00
. unwrap ( )
2019-04-08 10:12:43 -04:00
}
2020-05-29 17:41:39 -04:00
2022-03-22 09:32:32 -04:00
/// Returns the namespace object of a module.
///
/// This is only available after module evaluation has completed.
/// This function panics if module has not been instantiated.
pub fn get_module_namespace (
& mut self ,
module_id : ModuleId ,
) -> Result < v8 ::Global < v8 ::Object > , Error > {
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
let module_handle = module_map_rc
. borrow ( )
. get_handle ( module_id )
. expect ( " ModuleInfo not found " ) ;
let scope = & mut self . handle_scope ( ) ;
let module = module_handle . open ( scope ) ;
if module . get_status ( ) = = v8 ::ModuleStatus ::Errored {
let exception = module . get_exception ( ) ;
2022-04-26 09:28:42 -04:00
return exception_to_err_result ( scope , exception , false ) ;
2022-03-22 09:32:32 -04:00
}
assert! ( matches! (
module . get_status ( ) ,
v8 ::ModuleStatus ::Instantiated | v8 ::ModuleStatus ::Evaluated
) ) ;
let module_namespace : v8 ::Local < v8 ::Object > =
v8 ::Local ::try_from ( module . get_module_namespace ( ) )
. map_err ( | err : v8 ::DataError | generic_error ( err . to_string ( ) ) ) ? ;
Ok ( v8 ::Global ::new ( scope , module_namespace ) )
}
2020-08-11 21:07:14 -04:00
/// Registers a callback on the isolate when the memory limits are approached.
/// Use this to prevent V8 from crashing the process when reaching the limit.
///
/// Calls the closure with the current heap limit and the initial heap limit.
/// The return value of the closure is set as the new limit.
pub fn add_near_heap_limit_callback < C > ( & mut self , cb : C )
where
C : FnMut ( usize , usize ) -> usize + 'static ,
{
let boxed_cb = Box ::new ( RefCell ::new ( cb ) ) ;
let data = boxed_cb . as_ptr ( ) as * mut c_void ;
2020-08-12 00:08:50 -04:00
let prev = self
. allocations
. near_heap_limit_callback_data
. replace ( ( boxed_cb , near_heap_limit_callback ::< C > ) ) ;
if let Some ( ( _ , prev_cb ) ) = prev {
self
2020-10-05 05:08:19 -04:00
. v8_isolate ( )
2020-08-12 00:08:50 -04:00
. remove_near_heap_limit_callback ( prev_cb , 0 ) ;
}
2020-08-11 21:07:14 -04:00
self
2020-10-05 05:08:19 -04:00
. v8_isolate ( )
2020-08-11 21:07:14 -04:00
. add_near_heap_limit_callback ( near_heap_limit_callback ::< C > , data ) ;
}
pub fn remove_near_heap_limit_callback ( & mut self , heap_limit : usize ) {
if let Some ( ( _ , cb ) ) = self . allocations . near_heap_limit_callback_data . take ( )
{
self
2020-10-05 05:08:19 -04:00
. v8_isolate ( )
2020-08-11 21:07:14 -04:00
. remove_near_heap_limit_callback ( cb , heap_limit ) ;
}
}
2019-03-11 17:57:36 -04:00
2022-07-01 03:51:29 -04:00
fn pump_v8_message_loop ( & mut self ) -> Result < ( ) , Error > {
2021-07-02 04:46:37 -04:00
let scope = & mut self . handle_scope ( ) ;
while v8 ::Platform ::pump_message_loop (
& v8 ::V8 ::get_current_platform ( ) ,
scope ,
false , // don't block if there are no tasks
) {
// do nothing
}
2021-07-03 17:33:36 -04:00
2022-07-01 03:51:29 -04:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
tc_scope . perform_microtask_checkpoint ( ) ;
match tc_scope . exception ( ) {
None = > Ok ( ( ) ) ,
Some ( exception ) = > exception_to_err_result ( tc_scope , exception , false ) ,
}
2021-07-02 04:46:37 -04:00
}
2022-10-26 07:37:50 -04:00
pub fn maybe_init_inspector ( & mut self ) {
if self . state . borrow ( ) . inspector . is_some ( ) {
return ;
}
let mut state = self . state . borrow_mut ( ) ;
state . inspector = Some ( JsRuntimeInspector ::new (
self . v8_isolate . as_mut ( ) . unwrap ( ) ,
state . global_realm . clone ( ) . unwrap ( ) . 0 ,
2022-11-26 17:09:48 -05:00
self . is_main ,
2022-10-26 07:37:50 -04:00
) ) ;
}
2022-01-24 11:59:41 -05:00
pub fn poll_value (
& mut self ,
global : & v8 ::Global < v8 ::Value > ,
cx : & mut Context ,
) -> Poll < Result < v8 ::Global < v8 ::Value > , Error > > {
let state = self . poll_event_loop ( cx , false ) ;
let mut scope = self . handle_scope ( ) ;
let local = v8 ::Local ::< v8 ::Value > ::new ( & mut scope , global ) ;
if let Ok ( promise ) = v8 ::Local ::< v8 ::Promise > ::try_from ( local ) {
match promise . state ( ) {
v8 ::PromiseState ::Pending = > match state {
Poll ::Ready ( Ok ( _ ) ) = > {
let msg = " Promise resolution is still pending but the event loop has already resolved. " ;
Poll ::Ready ( Err ( generic_error ( msg ) ) )
}
Poll ::Ready ( Err ( e ) ) = > Poll ::Ready ( Err ( e ) ) ,
Poll ::Pending = > Poll ::Pending ,
} ,
v8 ::PromiseState ::Fulfilled = > {
let value = promise . result ( & mut scope ) ;
let value_handle = v8 ::Global ::new ( & mut scope , value ) ;
Poll ::Ready ( Ok ( value_handle ) )
}
v8 ::PromiseState ::Rejected = > {
let exception = promise . result ( & mut scope ) ;
Poll ::Ready ( exception_to_err_result ( & mut scope , exception , false ) )
}
}
} else {
let value_handle = v8 ::Global ::new ( & mut scope , local ) ;
Poll ::Ready ( Ok ( value_handle ) )
}
}
2021-09-04 14:19:26 -04:00
/// Waits for the given value to resolve while polling the event loop.
///
/// This future resolves when either the value is resolved or the event loop runs to
/// completion.
pub async fn resolve_value (
& mut self ,
global : v8 ::Global < v8 ::Value > ,
2021-11-16 09:02:28 -05:00
) -> Result < v8 ::Global < v8 ::Value > , Error > {
2022-01-24 11:59:41 -05:00
poll_fn ( | cx | self . poll_value ( & global , cx ) ) . await
2021-09-04 14:19:26 -04:00
}
2020-10-11 07:20:40 -04:00
/// Runs event loop to completion
///
/// This future resolves when:
/// - there are no more pending dynamic imports
/// - there are no more pending ops
2021-05-26 15:07:12 -04:00
/// - there are no more active inspector sessions (only if `wait_for_inspector` is set to true)
pub async fn run_event_loop (
& mut self ,
wait_for_inspector : bool ,
2021-11-16 09:02:28 -05:00
) -> Result < ( ) , Error > {
2021-05-26 15:07:12 -04:00
poll_fn ( | cx | self . poll_event_loop ( cx , wait_for_inspector ) ) . await
2020-10-11 07:20:40 -04:00
}
2019-06-12 13:53:24 -04:00
2020-10-11 07:20:40 -04:00
/// Runs a single tick of event loop
2021-05-26 15:07:12 -04:00
///
/// If `wait_for_inspector` is set to true event loop
/// will return `Poll::Pending` if there are active inspector sessions.
2020-10-11 07:20:40 -04:00
pub fn poll_event_loop (
& mut self ,
cx : & mut Context ,
2021-05-26 15:07:12 -04:00
wait_for_inspector : bool ,
2021-11-16 09:02:28 -05:00
) -> Poll < Result < ( ) , Error > > {
2022-10-26 07:37:50 -04:00
let has_inspector : bool ;
2020-05-29 17:41:39 -04:00
{
2022-10-21 10:13:42 -04:00
let state = self . state . borrow ( ) ;
2022-10-26 07:37:50 -04:00
has_inspector = state . inspector . is_some ( ) ;
2020-05-29 17:41:39 -04:00
state . waker . register ( cx . waker ( ) ) ;
}
2020-02-24 18:53:29 -05:00
2022-10-26 07:37:50 -04:00
if has_inspector {
// We poll the inspector first.
let _ = self . inspector ( ) . borrow_mut ( ) . poll_unpin ( cx ) ;
}
2022-07-01 03:51:29 -04:00
self . pump_v8_message_loop ( ) ? ;
2021-07-03 17:33:36 -04:00
2020-10-14 08:04:09 -04:00
// Ops
{
2021-10-24 15:41:57 -04:00
self . resolve_async_ops ( cx ) ? ;
2021-11-16 14:23:12 -05:00
self . drain_nexttick ( ) ? ;
2020-10-14 08:04:09 -04:00
self . drain_macrotasks ( ) ? ;
self . check_promise_exceptions ( ) ? ;
}
2020-10-05 05:08:19 -04:00
// Dynamic module loading - ie. modules loaded using "import()"
{
2022-06-25 14:56:29 -04:00
// Run in a loop so that dynamic imports that only depend on another
// dynamic import can be resolved in this event loop iteration.
//
// For example, a dynamically imported module like the following can be
// immediately resolved after `dependency.ts` is fully evaluated, but it
// wouldn't if not for this loop.
//
// await delay(1000);
// await import("./dependency.ts");
// console.log("test")
//
loop {
let poll_imports = self . prepare_dyn_imports ( cx ) ? ;
assert! ( poll_imports . is_ready ( ) ) ;
let poll_imports = self . poll_dyn_imports ( cx ) ? ;
assert! ( poll_imports . is_ready ( ) ) ;
if ! self . evaluate_dyn_imports ( ) {
break ;
}
}
2019-03-14 19:17:52 -04:00
2020-10-11 07:20:40 -04:00
self . check_promise_exceptions ( ) ? ;
2019-03-11 17:57:36 -04:00
}
2022-03-08 09:40:34 -05:00
// Event loop middlewares
let mut maybe_scheduling = false ;
{
2022-10-21 10:13:42 -04:00
let op_state = self . state . borrow ( ) . op_state . clone ( ) ;
2022-03-08 09:40:34 -05:00
for f in & self . event_loop_middlewares {
2022-06-28 05:23:36 -04:00
if f ( op_state . clone ( ) , cx ) {
2022-03-08 09:40:34 -05:00
maybe_scheduling = true ;
}
}
}
2020-10-14 08:04:09 -04:00
// Top level module
2020-11-27 14:47:35 -05:00
self . evaluate_pending_module ( ) ;
2020-10-14 08:04:09 -04:00
2022-10-21 10:13:42 -04:00
let pending_state = self . event_loop_pending_state ( ) ;
2022-07-11 06:08:37 -04:00
if ! pending_state . is_pending ( ) & & ! maybe_scheduling {
2022-10-26 07:37:50 -04:00
if has_inspector {
let inspector_has_active_sessions =
self . inspector ( ) . borrow_mut ( ) . has_active_sessions ( ) ;
if wait_for_inspector & & inspector_has_active_sessions {
return Poll ::Pending ;
}
2021-05-26 15:07:12 -04:00
}
2020-10-05 05:08:19 -04:00
return Poll ::Ready ( Ok ( ( ) ) ) ;
2020-05-29 17:41:39 -04:00
}
2019-03-11 17:57:36 -04:00
2022-10-21 10:13:42 -04:00
let mut state = self . state . borrow_mut ( ) ;
2022-07-11 06:08:37 -04:00
2020-10-05 05:08:19 -04:00
// Check if more async ops have been dispatched
// during this turn of event loop.
2021-07-03 17:33:36 -04:00
// If there are any pending background tasks, we also wake the runtime to
// make sure we don't miss them.
// TODO(andreubotella) The event loop will spin as long as there are pending
// background tasks. We should look into having V8 notify us when a
// background task is done.
2021-11-29 18:31:12 -05:00
if state . have_unpolled_ops
2022-07-11 06:08:37 -04:00
| | pending_state . has_pending_background_tasks
| | pending_state . has_tick_scheduled
2022-03-08 09:40:34 -05:00
| | maybe_scheduling
2021-11-29 18:31:12 -05:00
{
2020-10-05 05:08:19 -04:00
state . waker . wake ( ) ;
2019-03-11 17:57:36 -04:00
}
2020-10-05 05:08:19 -04:00
2022-07-11 06:08:37 -04:00
if pending_state . has_pending_module_evaluation {
if pending_state . has_pending_refed_ops
| | pending_state . has_pending_dyn_imports
| | pending_state . has_pending_dyn_module_evaluation
| | pending_state . has_pending_background_tasks
| | pending_state . has_tick_scheduled
2022-03-08 09:40:34 -05:00
| | maybe_scheduling
2020-10-14 08:04:09 -04:00
{
// pass, will be polled again
} else {
2021-12-01 09:22:11 -05:00
let msg = " Module evaluation is still pending but there are no pending ops or dynamic imports. This situation is often caused by unresolved promises. " ;
2020-10-14 08:04:09 -04:00
return Poll ::Ready ( Err ( generic_error ( msg ) ) ) ;
}
}
2022-07-11 06:08:37 -04:00
if pending_state . has_pending_dyn_module_evaluation {
if pending_state . has_pending_refed_ops
| | pending_state . has_pending_dyn_imports
| | pending_state . has_pending_background_tasks
| | pending_state . has_tick_scheduled
2021-07-03 17:33:36 -04:00
{
2020-10-14 08:04:09 -04:00
// pass, will be polled again
2021-07-05 12:59:49 -04:00
} else if state . dyn_module_evaluate_idle_counter > = 1 {
2021-12-01 09:22:11 -05:00
let mut msg = " Dynamically imported module evaluation is still pending but there are no pending ops. This situation is often caused by unresolved promises.
2021-04-28 12:28:46 -04:00
Pending dynamic modules :\ n " .to_string();
2022-10-25 08:27:33 -04:00
let module_map = self . module_map . as_mut ( ) . unwrap ( ) . borrow_mut ( ) ;
2021-04-28 12:28:46 -04:00
for pending_evaluate in & state . pending_dyn_mod_evaluate {
2021-05-19 14:53:43 -04:00
let module_info = module_map
2021-04-28 12:28:46 -04:00
. get_info_by_id ( & pending_evaluate . module_id )
. unwrap ( ) ;
2022-07-01 09:28:06 -04:00
msg . push_str ( " - " ) ;
msg . push_str ( module_info . name . as_str ( ) ) ;
2021-04-28 12:28:46 -04:00
}
2020-10-14 08:04:09 -04:00
return Poll ::Ready ( Err ( generic_error ( msg ) ) ) ;
2021-07-05 12:59:49 -04:00
} else {
// Delay the above error by one spin of the event loop. A dynamic import
// evaluation may complete during this, in which case the counter will
// reset.
state . dyn_module_evaluate_idle_counter + = 1 ;
state . waker . wake ( ) ;
2020-10-14 08:04:09 -04:00
}
}
2020-10-05 05:08:19 -04:00
Poll ::Pending
2019-03-11 17:57:36 -04:00
}
2022-06-28 10:49:30 -04:00
2022-10-21 10:13:42 -04:00
fn event_loop_pending_state ( & mut self ) -> EventLoopPendingState {
let isolate = self . v8_isolate . as_mut ( ) . unwrap ( ) ;
let state = self . state . borrow_mut ( ) ;
2022-10-25 08:27:33 -04:00
let module_map = self . module_map . as_mut ( ) . unwrap ( ) . borrow ( ) ;
2022-10-21 10:13:42 -04:00
EventLoopPendingState {
has_pending_refed_ops : state . pending_ops . len ( ) > state . unrefed_ops . len ( ) ,
has_pending_dyn_imports : module_map . has_pending_dynamic_imports ( ) ,
has_pending_dyn_module_evaluation : ! state
. pending_dyn_mod_evaluate
. is_empty ( ) ,
has_pending_module_evaluation : state . pending_mod_evaluate . is_some ( ) ,
has_pending_background_tasks : isolate . has_pending_background_tasks ( ) ,
has_tick_scheduled : state . has_tick_scheduled ,
}
}
pub ( crate ) fn event_loop_pending_state_from_isolate (
2022-07-11 06:08:37 -04:00
isolate : & mut v8 ::Isolate ,
) -> EventLoopPendingState {
let state_rc = Self ::state ( isolate ) ;
let module_map_rc = Self ::module_map ( isolate ) ;
2022-06-28 10:49:30 -04:00
let state = state_rc . borrow_mut ( ) ;
let module_map = module_map_rc . borrow ( ) ;
2022-07-11 06:08:37 -04:00
EventLoopPendingState {
2022-10-20 22:54:22 -04:00
has_pending_refed_ops : state . pending_ops . len ( ) > state . unrefed_ops . len ( ) ,
2022-07-11 06:08:37 -04:00
has_pending_dyn_imports : module_map . has_pending_dynamic_imports ( ) ,
has_pending_dyn_module_evaluation : ! state
. pending_dyn_mod_evaluate
. is_empty ( ) ,
has_pending_module_evaluation : state . pending_mod_evaluate . is_some ( ) ,
has_pending_background_tasks : isolate . has_pending_background_tasks ( ) ,
has_tick_scheduled : state . has_tick_scheduled ,
}
}
}
#[ derive(Clone, Copy, PartialEq, Eq, Debug) ]
pub ( crate ) struct EventLoopPendingState {
has_pending_refed_ops : bool ,
has_pending_dyn_imports : bool ,
has_pending_dyn_module_evaluation : bool ,
has_pending_module_evaluation : bool ,
has_pending_background_tasks : bool ,
has_tick_scheduled : bool ,
}
impl EventLoopPendingState {
pub fn is_pending ( & self ) -> bool {
self . has_pending_refed_ops
| | self . has_pending_dyn_imports
| | self . has_pending_dyn_module_evaluation
| | self . has_pending_module_evaluation
| | self . has_pending_background_tasks
| | self . has_tick_scheduled
2022-06-28 10:49:30 -04:00
}
2019-03-11 17:57:36 -04:00
}
2020-10-11 07:20:40 -04:00
extern " C " fn near_heap_limit_callback < F > (
data : * mut c_void ,
current_heap_limit : usize ,
initial_heap_limit : usize ,
) -> usize
where
F : FnMut ( usize , usize ) -> usize ,
{
2022-06-25 18:13:24 -04:00
// SAFETY: The data is a pointer to the Rust callback function. It is stored
// in `JsRuntime::allocations` and thus is guaranteed to outlive the isolate.
2020-10-11 07:20:40 -04:00
let callback = unsafe { & mut * ( data as * mut F ) } ;
callback ( current_heap_limit , initial_heap_limit )
}
2020-09-06 15:44:29 -04:00
impl JsRuntimeState {
2022-09-02 06:43:39 -04:00
pub ( crate ) fn inspector ( & self ) -> Rc < RefCell < JsRuntimeInspector > > {
self . inspector . as_ref ( ) . unwrap ( ) . clone ( )
}
2021-05-19 14:53:43 -04:00
/// Called by `bindings::host_import_module_dynamically_callback`
/// after initiating new dynamic import load.
pub fn notify_new_dynamic_import ( & mut self ) {
// Notify event loop to poll again soon.
2020-09-06 10:50:49 -04:00
self . waker . wake ( ) ;
}
2020-05-29 17:41:39 -04:00
}
2020-03-02 17:20:16 -05:00
pub ( crate ) fn exception_to_err_result < ' s , T > (
2020-06-20 07:18:08 -04:00
scope : & mut v8 ::HandleScope < ' s > ,
2020-02-24 18:53:29 -05:00
exception : v8 ::Local < v8 ::Value > ,
2020-10-25 23:34:00 -04:00
in_promise : bool ,
2021-11-16 09:02:28 -05:00
) -> Result < T , Error > {
2022-04-13 05:50:57 -04:00
let state_rc = JsRuntime ::state ( scope ) ;
2022-08-21 07:57:10 -04:00
let was_terminating_execution = scope . is_execution_terminating ( ) ;
// If TerminateExecution was called, cancel isolate termination so that the
// exception can be created. Note that `scope.is_execution_terminating()` may
// have returned false if TerminateExecution was indeed called but there was
// no JS to execute after the call.
scope . cancel_terminate_execution ( ) ;
2020-02-24 18:53:29 -05:00
let mut exception = exception ;
2022-08-21 07:57:10 -04:00
{
2022-09-02 06:43:39 -04:00
// If termination is the result of a `op_dispatch_exception` call, we want
2022-04-13 05:50:57 -04:00
// to use the exception that was passed to it rather than the exception that
// was passed to this function.
2022-09-06 07:18:23 -04:00
let state = state_rc . borrow ( ) ;
2022-04-13 05:50:57 -04:00
exception = state
2022-09-02 06:43:39 -04:00
. dispatched_exceptions
2022-09-06 07:18:23 -04:00
. back ( )
. map ( | exception | v8 ::Local ::new ( scope , exception . clone ( ) ) )
2022-04-13 05:50:57 -04:00
. unwrap_or_else ( | | {
// Maybe make a new exception object.
2022-09-02 06:43:39 -04:00
if was_terminating_execution & & exception . is_null_or_undefined ( ) {
2022-04-13 05:50:57 -04:00
let message = v8 ::String ::new ( scope , " execution terminated " ) . unwrap ( ) ;
v8 ::Exception ::error ( scope , message )
} else {
exception
}
} ) ;
2019-03-21 09:48:19 -04:00
}
2020-02-24 18:53:29 -05:00
2020-10-25 23:34:00 -04:00
let mut js_error = JsError ::from_v8_exception ( scope , exception ) ;
if in_promise {
2022-04-13 10:41:39 -04:00
js_error . exception_message = format! (
2020-10-25 23:34:00 -04:00
" Uncaught (in promise) {} " ,
2022-04-13 10:41:39 -04:00
js_error . exception_message . trim_start_matches ( " Uncaught " )
2020-10-25 23:34:00 -04:00
) ;
}
2020-02-24 18:53:29 -05:00
2022-08-21 07:57:10 -04:00
if was_terminating_execution {
// Resume exception termination.
2020-12-28 10:36:44 -05:00
scope . terminate_execution ( ) ;
2020-02-24 18:53:29 -05:00
}
2022-04-26 19:06:10 -04:00
Err ( js_error . into ( ) )
2020-02-24 18:53:29 -05:00
}
2020-09-06 10:50:49 -04:00
// Related to module loading
2020-09-06 15:44:29 -04:00
impl JsRuntime {
2021-05-19 14:53:43 -04:00
pub ( crate ) fn instantiate_module (
2020-09-06 10:50:49 -04:00
& mut self ,
2021-05-19 14:53:43 -04:00
id : ModuleId ,
2022-04-26 09:28:42 -04:00
) -> Result < ( ) , v8 ::Global < v8 ::Value > > {
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2021-02-23 09:22:55 -05:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
2020-09-06 10:50:49 -04:00
2021-05-19 14:53:43 -04:00
let module = module_map_rc
2021-02-23 09:22:55 -05:00
. borrow ( )
. get_handle ( id )
. map ( | handle | v8 ::Local ::new ( tc_scope , handle ) )
. expect ( " ModuleInfo not found " ) ;
2020-09-06 10:50:49 -04:00
2021-02-23 09:22:55 -05:00
if module . get_status ( ) = = v8 ::ModuleStatus ::Errored {
2022-04-26 09:28:42 -04:00
return Err ( v8 ::Global ::new ( tc_scope , module . get_exception ( ) ) ) ;
2020-09-06 10:50:49 -04:00
}
2021-05-19 14:53:43 -04:00
// IMPORTANT: No borrows to `ModuleMap` can be held at this point because
// `module_resolve_callback` will be calling into `ModuleMap` from within
// the isolate.
let instantiate_result =
module . instantiate_module ( tc_scope , bindings ::module_resolve_callback ) ;
if instantiate_result . is_none ( ) {
let exception = tc_scope . exception ( ) . unwrap ( ) ;
2022-04-26 09:28:42 -04:00
return Err ( v8 ::Global ::new ( tc_scope , exception ) ) ;
2021-05-19 14:53:43 -04:00
}
Ok ( ( ) )
2020-09-06 10:50:49 -04:00
}
2021-05-19 14:53:43 -04:00
fn dynamic_import_module_evaluate (
2020-10-14 08:04:09 -04:00
& mut self ,
load_id : ModuleLoadId ,
id : ModuleId ,
2021-11-16 09:02:28 -05:00
) -> Result < ( ) , Error > {
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2020-10-14 08:04:09 -04:00
2021-05-19 14:53:43 -04:00
let module_handle = module_map_rc
2020-10-14 08:04:09 -04:00
. borrow ( )
2020-11-21 10:23:35 -05:00
. get_handle ( id )
. expect ( " ModuleInfo not found " ) ;
2020-10-14 08:04:09 -04:00
let status = {
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2021-10-27 17:26:15 -04:00
let module = module_handle . open ( scope ) ;
2020-10-14 08:04:09 -04:00
module . get_status ( )
} ;
2021-04-04 07:26:00 -04:00
match status {
v8 ::ModuleStatus ::Instantiated | v8 ::ModuleStatus ::Evaluated = > { }
_ = > return Ok ( ( ) ) ,
2021-02-23 09:22:55 -05:00
}
2020-10-14 08:04:09 -04:00
2021-02-23 09:22:55 -05:00
// IMPORTANT: Top-level-await is enabled, which means that return value
// of module evaluation is a promise.
//
// This promise is internal, and not the same one that gets returned to
// the user. We add an empty `.catch()` handler so that it does not result
// in an exception if it rejects. That will instead happen for the other
// promise if not handled by the user.
//
// For more details see:
// https://github.com/denoland/deno/issues/4908
// https://v8.dev/features/top-level-await#module-execution-order
2022-10-21 10:13:42 -04:00
let global_realm = self . state . borrow_mut ( ) . global_realm . clone ( ) . unwrap ( ) ;
let scope =
& mut global_realm . handle_scope ( self . v8_isolate . as_mut ( ) . unwrap ( ) ) ;
2021-11-22 07:51:20 -05:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let module = v8 ::Local ::new ( tc_scope , & module_handle ) ;
let maybe_value = module . evaluate ( tc_scope ) ;
2021-02-23 09:22:55 -05:00
// Update status after evaluating.
let status = module . get_status ( ) ;
if let Some ( value ) = maybe_value {
assert! (
status = = v8 ::ModuleStatus ::Evaluated
| | status = = v8 ::ModuleStatus ::Errored
) ;
let promise = v8 ::Local ::< v8 ::Promise > ::try_from ( value )
. expect ( " Expected to get promise as module evaluation result " ) ;
2022-09-06 08:35:04 -04:00
let empty_fn = bindings ::create_empty_fn ( tc_scope ) . unwrap ( ) ;
2021-11-22 07:51:20 -05:00
promise . catch ( tc_scope , empty_fn ) ;
let promise_global = v8 ::Global ::new ( tc_scope , promise ) ;
let module_global = v8 ::Global ::new ( tc_scope , module ) ;
2020-10-14 08:04:09 -04:00
2021-02-23 09:22:55 -05:00
let dyn_import_mod_evaluate = DynImportModEvaluate {
2021-04-28 12:28:46 -04:00
load_id ,
2021-02-23 09:22:55 -05:00
module_id : id ,
promise : promise_global ,
module : module_global ,
} ;
2020-10-14 08:04:09 -04:00
2022-10-21 10:13:42 -04:00
self
. state
. borrow_mut ( )
. pending_dyn_mod_evaluate
. push ( dyn_import_mod_evaluate ) ;
2021-11-22 07:51:20 -05:00
} else if tc_scope . has_terminated ( ) | | tc_scope . is_execution_terminating ( ) {
return Err (
generic_error ( " Cannot evaluate dynamically imported module, because JavaScript execution has been terminated. " )
) ;
2021-02-23 09:22:55 -05:00
} else {
assert! ( status = = v8 ::ModuleStatus ::Errored ) ;
2020-10-14 08:04:09 -04:00
}
Ok ( ( ) )
}
2021-03-04 07:19:47 -05:00
// TODO(bartlomieju): make it return `ModuleEvaluationFuture`?
2020-10-14 08:04:09 -04:00
/// Evaluates an already instantiated ES module.
///
2021-03-04 07:19:47 -05:00
/// Returns a receiver handle that resolves when module promise resolves.
2022-06-20 08:40:57 -04:00
/// Implementors must manually call [`JsRuntime::run_event_loop`] to drive
/// module evaluation future.
2021-03-04 07:19:47 -05:00
///
2022-06-20 08:40:57 -04:00
/// `Error` can usually be downcast to `JsError` and should be awaited and
/// checked after [`JsRuntime::run_event_loop`] completion.
2021-02-23 09:22:55 -05:00
///
/// This function panics if module has not been instantiated.
2021-03-04 07:19:47 -05:00
pub fn mod_evaluate (
2020-10-14 08:04:09 -04:00
& mut self ,
id : ModuleId ,
2021-11-16 09:02:28 -05:00
) -> oneshot ::Receiver < Result < ( ) , Error > > {
2022-10-21 10:13:42 -04:00
let state_rc = self . state . clone ( ) ;
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2021-11-22 07:51:20 -05:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
2020-09-06 10:50:49 -04:00
2021-05-19 14:53:43 -04:00
let module = module_map_rc
2020-09-06 10:50:49 -04:00
. borrow ( )
2020-11-21 10:23:35 -05:00
. get_handle ( id )
2021-11-22 07:51:20 -05:00
. map ( | handle | v8 ::Local ::new ( tc_scope , handle ) )
2020-09-06 10:50:49 -04:00
. expect ( " ModuleInfo not found " ) ;
let mut status = module . get_status ( ) ;
2021-02-23 09:22:55 -05:00
assert_eq! ( status , v8 ::ModuleStatus ::Instantiated ) ;
2020-09-06 10:50:49 -04:00
2021-07-30 07:36:43 -04:00
let ( sender , receiver ) = oneshot ::channel ( ) ;
2020-10-14 08:04:09 -04:00
2021-02-23 09:22:55 -05:00
// IMPORTANT: Top-level-await is enabled, which means that return value
// of module evaluation is a promise.
//
// Because that promise is created internally by V8, when error occurs during
// module evaluation the promise is rejected, and since the promise has no rejection
// handler it will result in call to `bindings::promise_reject_callback` adding
// the promise to pending promise rejection table - meaning JsRuntime will return
// error on next poll().
//
// This situation is not desirable as we want to manually return error at the
// end of this function to handle it further. It means we need to manually
// remove this promise from pending promise rejection table.
//
// For more details see:
// https://github.com/denoland/deno/issues/4908
// https://v8.dev/features/top-level-await#module-execution-order
2022-07-18 15:20:38 -04:00
{
let mut state = state_rc . borrow_mut ( ) ;
assert! (
state . pending_mod_evaluate . is_none ( ) ,
" There is already pending top level module evaluation "
) ;
state . pending_mod_evaluate = Some ( ModEvaluate {
promise : None ,
has_evaluated : false ,
handled_promise_rejections : vec ! [ ] ,
sender ,
} ) ;
}
2021-11-22 07:51:20 -05:00
let maybe_value = module . evaluate ( tc_scope ) ;
2022-07-18 15:20:38 -04:00
{
let mut state = state_rc . borrow_mut ( ) ;
let pending_mod_evaluate = state . pending_mod_evaluate . as_mut ( ) . unwrap ( ) ;
pending_mod_evaluate . has_evaluated = true ;
}
2021-02-23 09:22:55 -05:00
// Update status after evaluating.
status = module . get_status ( ) ;
2022-09-02 06:43:39 -04:00
let has_dispatched_exception =
! state_rc . borrow_mut ( ) . dispatched_exceptions . is_empty ( ) ;
if has_dispatched_exception {
// This will be overrided in `exception_to_err_result()`.
let exception = v8 ::undefined ( tc_scope ) . into ( ) ;
2022-07-18 15:20:38 -04:00
let pending_mod_evaluate = {
let mut state = state_rc . borrow_mut ( ) ;
state . pending_mod_evaluate . take ( ) . unwrap ( )
} ;
pending_mod_evaluate
. sender
2022-04-13 05:50:57 -04:00
. send ( exception_to_err_result ( tc_scope , exception , false ) )
. expect ( " Failed to send module evaluation error. " ) ;
} else if let Some ( value ) = maybe_value {
2021-02-23 09:22:55 -05:00
assert! (
status = = v8 ::ModuleStatus ::Evaluated
| | status = = v8 ::ModuleStatus ::Errored
) ;
let promise = v8 ::Local ::< v8 ::Promise > ::try_from ( value )
. expect ( " Expected to get promise as module evaluation result " ) ;
2021-11-22 07:51:20 -05:00
let promise_global = v8 ::Global ::new ( tc_scope , promise ) ;
2021-02-23 09:22:55 -05:00
let mut state = state_rc . borrow_mut ( ) ;
2022-07-22 18:40:42 -04:00
{
let pending_mod_evaluate = state . pending_mod_evaluate . as_ref ( ) . unwrap ( ) ;
let pending_rejection_was_already_handled = pending_mod_evaluate
. handled_promise_rejections
. contains ( & promise_global ) ;
if ! pending_rejection_was_already_handled {
state . pending_promise_exceptions . remove ( & promise_global ) ;
}
}
2021-11-22 07:51:20 -05:00
let promise_global = v8 ::Global ::new ( tc_scope , promise ) ;
2022-07-18 15:20:38 -04:00
state . pending_mod_evaluate . as_mut ( ) . unwrap ( ) . promise =
Some ( promise_global ) ;
2021-11-22 07:51:20 -05:00
tc_scope . perform_microtask_checkpoint ( ) ;
} else if tc_scope . has_terminated ( ) | | tc_scope . is_execution_terminating ( ) {
2022-07-18 15:20:38 -04:00
let pending_mod_evaluate = {
let mut state = state_rc . borrow_mut ( ) ;
state . pending_mod_evaluate . take ( ) . unwrap ( )
} ;
pending_mod_evaluate . sender . send ( Err (
2021-11-22 07:51:20 -05:00
generic_error ( " Cannot evaluate module, because JavaScript execution has been terminated. " )
) ) . expect ( " Failed to send module evaluation error. " ) ;
2021-02-23 09:22:55 -05:00
} else {
assert! ( status = = v8 ::ModuleStatus ::Errored ) ;
2020-09-06 10:50:49 -04:00
}
2020-11-27 14:47:35 -05:00
receiver
2020-10-14 08:04:09 -04:00
}
2022-04-26 09:28:42 -04:00
fn dynamic_import_reject (
& mut self ,
id : ModuleLoadId ,
exception : v8 ::Global < v8 ::Value > ,
) {
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2020-09-06 10:50:49 -04:00
2021-05-19 14:53:43 -04:00
let resolver_handle = module_map_rc
2020-09-06 10:50:49 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. dynamic_import_map
2020-09-06 10:50:49 -04:00
. remove ( & id )
2021-05-19 14:53:43 -04:00
. expect ( " Invalid dynamic import id " ) ;
2021-10-27 17:26:15 -04:00
let resolver = resolver_handle . open ( scope ) ;
2020-09-06 10:50:49 -04:00
2021-05-19 14:53:43 -04:00
// IMPORTANT: No borrows to `ModuleMap` can be held at this point because
// rejecting the promise might initiate another `import()` which will
// in turn call `bindings::host_import_module_dynamically_callback` which
// will reach into `ModuleMap` from within the isolate.
2022-04-26 09:28:42 -04:00
let exception = v8 ::Local ::new ( scope , exception ) ;
2020-09-06 10:50:49 -04:00
resolver . reject ( scope , exception ) . unwrap ( ) ;
scope . perform_microtask_checkpoint ( ) ;
}
2021-05-19 14:53:43 -04:00
fn dynamic_import_resolve ( & mut self , id : ModuleLoadId , mod_id : ModuleId ) {
2022-10-21 10:13:42 -04:00
let state_rc = self . state . clone ( ) ;
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2020-09-06 10:50:49 -04:00
2021-05-19 14:53:43 -04:00
let resolver_handle = module_map_rc
2020-09-06 10:50:49 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. dynamic_import_map
2020-09-06 10:50:49 -04:00
. remove ( & id )
2021-05-19 14:53:43 -04:00
. expect ( " Invalid dynamic import id " ) ;
2021-10-27 17:26:15 -04:00
let resolver = resolver_handle . open ( scope ) ;
2020-09-06 10:50:49 -04:00
let module = {
2021-05-19 14:53:43 -04:00
module_map_rc
. borrow ( )
2020-11-21 10:23:35 -05:00
. get_handle ( mod_id )
. map ( | handle | v8 ::Local ::new ( scope , handle ) )
2020-09-06 10:50:49 -04:00
. expect ( " Dyn import module info not found " )
} ;
// Resolution success
assert_eq! ( module . get_status ( ) , v8 ::ModuleStatus ::Evaluated ) ;
2021-05-19 14:53:43 -04:00
// IMPORTANT: No borrows to `ModuleMap` can be held at this point because
// resolving the promise might initiate another `import()` which will
// in turn call `bindings::host_import_module_dynamically_callback` which
// will reach into `ModuleMap` from within the isolate.
2020-09-06 10:50:49 -04:00
let module_namespace = module . get_module_namespace ( ) ;
resolver . resolve ( scope , module_namespace ) . unwrap ( ) ;
2021-07-05 12:59:49 -04:00
state_rc . borrow_mut ( ) . dyn_module_evaluate_idle_counter = 0 ;
2020-09-06 10:50:49 -04:00
scope . perform_microtask_checkpoint ( ) ;
}
fn prepare_dyn_imports (
& mut self ,
cx : & mut Context ,
2021-11-16 09:02:28 -05:00
) -> Poll < Result < ( ) , Error > > {
2022-10-25 08:27:33 -04:00
if self
. get_module_map ( )
. borrow ( )
. preparing_dynamic_imports
. is_empty ( )
{
2020-10-05 05:08:19 -04:00
return Poll ::Ready ( Ok ( ( ) ) ) ;
}
2022-10-25 08:27:33 -04:00
let module_map_rc = self . get_module_map ( ) . clone ( ) ;
2020-09-06 10:50:49 -04:00
loop {
2021-05-19 14:53:43 -04:00
let poll_result = module_map_rc
2021-04-28 12:28:46 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. preparing_dynamic_imports
2021-04-28 12:28:46 -04:00
. poll_next_unpin ( cx ) ;
if let Poll ::Ready ( Some ( prepare_poll ) ) = poll_result {
let dyn_import_id = prepare_poll . 0 ;
let prepare_result = prepare_poll . 1 ;
match prepare_result {
Ok ( load ) = > {
2021-05-19 14:53:43 -04:00
module_map_rc
2021-04-28 12:28:46 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. pending_dynamic_imports
2021-04-28 12:28:46 -04:00
. push ( load . into_future ( ) ) ;
}
Err ( err ) = > {
2022-04-26 09:28:42 -04:00
let exception = to_v8_type_error ( & mut self . handle_scope ( ) , err ) ;
self . dynamic_import_reject ( dyn_import_id , exception ) ;
2020-09-06 10:50:49 -04:00
}
}
2021-04-28 12:28:46 -04:00
// Continue polling for more prepared dynamic imports.
continue ;
2020-09-06 10:50:49 -04:00
}
2021-04-28 12:28:46 -04:00
// There are no active dynamic import loads, or none are ready.
return Poll ::Ready ( Ok ( ( ) ) ) ;
2020-09-06 10:50:49 -04:00
}
}
2021-11-16 09:02:28 -05:00
fn poll_dyn_imports ( & mut self , cx : & mut Context ) -> Poll < Result < ( ) , Error > > {
2022-10-25 08:27:33 -04:00
if self
. get_module_map ( )
. borrow ( )
. pending_dynamic_imports
. is_empty ( )
{
2020-10-05 05:08:19 -04:00
return Poll ::Ready ( Ok ( ( ) ) ) ;
}
2022-10-25 08:27:33 -04:00
let module_map_rc = self . get_module_map ( ) . clone ( ) ;
2020-09-06 10:50:49 -04:00
loop {
2021-05-19 14:53:43 -04:00
let poll_result = module_map_rc
2021-04-28 12:28:46 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. pending_dynamic_imports
2021-04-28 12:28:46 -04:00
. poll_next_unpin ( cx ) ;
if let Poll ::Ready ( Some ( load_stream_poll ) ) = poll_result {
let maybe_result = load_stream_poll . 0 ;
let mut load = load_stream_poll . 1 ;
let dyn_import_id = load . id ;
if let Some ( load_stream_result ) = maybe_result {
match load_stream_result {
2021-12-15 13:22:36 -05:00
Ok ( ( request , info ) ) = > {
2021-04-28 12:28:46 -04:00
// A module (not necessarily the one dynamically imported) has been
// fetched. Create and register it, and if successful, poll for the
// next recursive-load event related to this dynamic import.
2021-12-15 13:22:36 -05:00
let register_result = load . register_and_recurse (
& mut self . handle_scope ( ) ,
& request ,
& info ,
) ;
2021-05-19 14:53:43 -04:00
match register_result {
2021-04-28 12:28:46 -04:00
Ok ( ( ) ) = > {
// Keep importing until it's fully drained
2021-05-19 14:53:43 -04:00
module_map_rc
2021-04-28 12:28:46 -04:00
. borrow_mut ( )
2021-05-19 14:53:43 -04:00
. pending_dynamic_imports
2021-04-28 12:28:46 -04:00
. push ( load . into_future ( ) ) ;
2020-09-06 10:50:49 -04:00
}
2022-04-26 09:28:42 -04:00
Err ( err ) = > {
let exception = match err {
ModuleError ::Exception ( e ) = > e ,
ModuleError ::Other ( e ) = > {
to_v8_type_error ( & mut self . handle_scope ( ) , e )
}
} ;
self . dynamic_import_reject ( dyn_import_id , exception )
}
2020-09-06 10:50:49 -04:00
}
}
2021-04-28 12:28:46 -04:00
Err ( err ) = > {
// A non-javascript error occurred; this could be due to a an invalid
// module specifier, or a problem with the source map, or a failure
// to fetch the module source code.
2022-04-26 09:28:42 -04:00
let exception = to_v8_type_error ( & mut self . handle_scope ( ) , err ) ;
self . dynamic_import_reject ( dyn_import_id , exception ) ;
2021-02-23 09:22:55 -05:00
}
2020-10-06 04:18:22 -04:00
}
2021-04-28 12:28:46 -04:00
} else {
// The top-level module from a dynamic import has been instantiated.
// Load is done.
2021-06-28 21:03:02 -04:00
let module_id =
load . root_module_id . expect ( " Root module should be loaded " ) ;
2021-05-19 14:53:43 -04:00
let result = self . instantiate_module ( module_id ) ;
2022-04-26 09:28:42 -04:00
if let Err ( exception ) = result {
self . dynamic_import_reject ( dyn_import_id , exception ) ;
2021-04-28 12:28:46 -04:00
}
2021-05-19 14:53:43 -04:00
self . dynamic_import_module_evaluate ( dyn_import_id , module_id ) ? ;
2020-10-06 04:18:22 -04:00
}
2021-04-28 12:28:46 -04:00
// Continue polling for more ready dynamic imports.
continue ;
2020-10-06 04:18:22 -04:00
}
2021-04-28 12:28:46 -04:00
// There are no active dynamic import loads, or none are ready.
return Poll ::Ready ( Ok ( ( ) ) ) ;
2020-10-06 04:18:22 -04:00
}
}
2021-06-22 11:47:09 -04:00
/// "deno_core" runs V8 with Top Level Await enabled. It means that each
/// module evaluation returns a promise from V8.
/// Feature docs: https://v8.dev/features/top-level-await
2020-10-14 08:04:09 -04:00
///
/// This promise resolves after all dependent modules have also
/// resolved. Each dependent module may perform calls to "import()" and APIs
/// using async ops will add futures to the runtime's event loop.
/// It means that the promise returned from module evaluation will
/// resolve only after all futures in the event loop are done.
///
/// Thus during turn of event loop we need to check if V8 has
/// resolved or rejected the promise. If the promise is still pending
/// then another turn of event loop must be performed.
2020-11-27 14:47:35 -05:00
fn evaluate_pending_module ( & mut self ) {
2021-04-28 12:28:46 -04:00
let maybe_module_evaluation =
2022-10-25 08:27:33 -04:00
self . state . borrow_mut ( ) . pending_mod_evaluate . take ( ) ;
2020-10-14 08:04:09 -04:00
2021-04-28 12:28:46 -04:00
if maybe_module_evaluation . is_none ( ) {
return ;
}
2020-10-14 08:04:09 -04:00
2022-07-18 15:20:38 -04:00
let mut module_evaluation = maybe_module_evaluation . unwrap ( ) ;
2022-10-25 08:27:33 -04:00
let state_rc = self . state . clone ( ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2022-07-18 15:20:38 -04:00
let promise_global = module_evaluation . promise . clone ( ) . unwrap ( ) ;
let promise = promise_global . open ( scope ) ;
2021-04-28 12:28:46 -04:00
let promise_state = promise . state ( ) ;
match promise_state {
v8 ::PromiseState ::Pending = > {
// NOTE: `poll_event_loop` will decide if
// runtime would be woken soon
state_rc . borrow_mut ( ) . pending_mod_evaluate = Some ( module_evaluation ) ;
}
v8 ::PromiseState ::Fulfilled = > {
scope . perform_microtask_checkpoint ( ) ;
// Receiver end might have been already dropped, ignore the result
2021-07-30 07:36:43 -04:00
let _ = module_evaluation . sender . send ( Ok ( ( ) ) ) ;
2022-07-18 15:20:38 -04:00
module_evaluation . handled_promise_rejections . clear ( ) ;
2021-04-28 12:28:46 -04:00
}
v8 ::PromiseState ::Rejected = > {
let exception = promise . result ( scope ) ;
scope . perform_microtask_checkpoint ( ) ;
2022-07-18 15:20:38 -04:00
2021-04-28 12:28:46 -04:00
// Receiver end might have been already dropped, ignore the result
2022-07-18 15:20:38 -04:00
if module_evaluation
. handled_promise_rejections
. contains ( & promise_global )
{
let _ = module_evaluation . sender . send ( Ok ( ( ) ) ) ;
module_evaluation . handled_promise_rejections . clear ( ) ;
} else {
let _ = module_evaluation
. sender
. send ( exception_to_err_result ( scope , exception , false ) ) ;
}
2021-04-28 12:28:46 -04:00
}
}
}
2022-06-25 14:56:29 -04:00
// Returns true if some dynamic import was resolved.
fn evaluate_dyn_imports ( & mut self ) -> bool {
2021-07-05 12:59:49 -04:00
let pending =
2022-10-21 10:13:42 -04:00
std ::mem ::take ( & mut self . state . borrow_mut ( ) . pending_dyn_mod_evaluate ) ;
2022-10-25 08:27:33 -04:00
if pending . is_empty ( ) {
return false ;
}
let mut resolved_any = false ;
let mut still_pending = vec! [ ] ;
2021-07-05 12:59:49 -04:00
for pending_dyn_evaluate in pending {
2021-04-28 12:28:46 -04:00
let maybe_result = {
let scope = & mut self . handle_scope ( ) ;
let module_id = pending_dyn_evaluate . module_id ;
2021-10-27 17:26:15 -04:00
let promise = pending_dyn_evaluate . promise . open ( scope ) ;
let _module = pending_dyn_evaluate . module . open ( scope ) ;
2020-10-14 08:04:09 -04:00
let promise_state = promise . state ( ) ;
match promise_state {
v8 ::PromiseState ::Pending = > {
2021-07-05 12:59:49 -04:00
still_pending . push ( pending_dyn_evaluate ) ;
2021-04-28 12:28:46 -04:00
None
2020-10-14 08:04:09 -04:00
}
v8 ::PromiseState ::Fulfilled = > {
2021-04-28 12:28:46 -04:00
Some ( Ok ( ( pending_dyn_evaluate . load_id , module_id ) ) )
2020-10-14 08:04:09 -04:00
}
v8 ::PromiseState ::Rejected = > {
let exception = promise . result ( scope ) ;
2022-04-26 09:28:42 -04:00
let exception = v8 ::Global ::new ( scope , exception ) ;
Some ( Err ( ( pending_dyn_evaluate . load_id , exception ) ) )
2020-10-14 08:04:09 -04:00
}
}
} ;
if let Some ( result ) = maybe_result {
2022-06-25 14:56:29 -04:00
resolved_any = true ;
2020-10-14 08:04:09 -04:00
match result {
Ok ( ( dyn_import_id , module_id ) ) = > {
2021-05-19 14:53:43 -04:00
self . dynamic_import_resolve ( dyn_import_id , module_id ) ;
2020-10-14 08:04:09 -04:00
}
2022-04-26 09:28:42 -04:00
Err ( ( dyn_import_id , exception ) ) = > {
self . dynamic_import_reject ( dyn_import_id , exception ) ;
2020-10-14 08:04:09 -04:00
}
}
}
}
2022-10-21 10:13:42 -04:00
self . state . borrow_mut ( ) . pending_dyn_mod_evaluate = still_pending ;
2022-06-25 14:56:29 -04:00
resolved_any
2020-10-14 08:04:09 -04:00
}
2021-09-17 21:44:53 -04:00
/// Asynchronously load specified module and all of its dependencies.
///
/// The module will be marked as "main", and because of that
/// "import.meta.main" will return true when checked inside that module.
2020-09-06 10:50:49 -04:00
///
2022-06-20 08:40:57 -04:00
/// User must call [`JsRuntime::mod_evaluate`] with returned `ModuleId`
2020-09-06 10:50:49 -04:00
/// manually after load is finished.
2021-09-17 21:44:53 -04:00
pub async fn load_main_module (
2020-09-06 10:50:49 -04:00
& mut self ,
specifier : & ModuleSpecifier ,
code : Option < String > ,
2021-11-16 09:02:28 -05:00
) -> Result < ModuleId , Error > {
2021-05-19 14:53:43 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
2021-06-28 21:03:02 -04:00
if let Some ( code ) = code {
2022-04-26 09:28:42 -04:00
let scope = & mut self . handle_scope ( ) ;
module_map_rc
. borrow_mut ( )
. new_es_module (
scope ,
// main module
true ,
specifier . as_str ( ) ,
2022-05-05 07:16:25 -04:00
code . as_bytes ( ) ,
2022-04-26 09:28:42 -04:00
)
. map_err ( | e | match e {
ModuleError ::Exception ( exception ) = > {
let exception = v8 ::Local ::new ( scope , exception ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
}
ModuleError ::Other ( error ) = > error ,
} ) ? ;
2021-06-28 21:03:02 -04:00
}
2021-05-19 14:53:43 -04:00
2021-06-28 21:03:02 -04:00
let mut load =
ModuleMap ::load_main ( module_map_rc . clone ( ) , specifier . as_str ( ) ) . await ? ;
2020-09-06 10:50:49 -04:00
2021-12-15 13:22:36 -05:00
while let Some ( load_result ) = load . next ( ) . await {
let ( request , info ) = load_result ? ;
2021-05-19 14:53:43 -04:00
let scope = & mut self . handle_scope ( ) ;
2022-04-26 09:28:42 -04:00
load . register_and_recurse ( scope , & request , & info ) . map_err (
| e | match e {
ModuleError ::Exception ( exception ) = > {
let exception = v8 ::Local ::new ( scope , exception ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
}
ModuleError ::Other ( error ) = > error ,
} ,
) ? ;
2020-09-06 10:50:49 -04:00
}
2021-06-28 21:03:02 -04:00
let root_id = load . root_module_id . expect ( " Root module should be loaded " ) ;
2022-04-26 09:28:42 -04:00
self . instantiate_module ( root_id ) . map_err ( | e | {
let scope = & mut self . handle_scope ( ) ;
let exception = v8 ::Local ::new ( scope , e ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
} ) ? ;
2021-06-19 10:14:43 -04:00
Ok ( root_id )
2020-09-06 10:50:49 -04:00
}
2020-10-05 05:08:19 -04:00
2021-09-17 21:44:53 -04:00
/// Asynchronously load specified ES module and all of its dependencies.
///
/// This method is meant to be used when loading some utility code that
/// might be later imported by the main module (ie. an entry point module).
///
2022-06-20 08:40:57 -04:00
/// User must call [`JsRuntime::mod_evaluate`] with returned `ModuleId`
2021-09-17 21:44:53 -04:00
/// manually after load is finished.
pub async fn load_side_module (
& mut self ,
specifier : & ModuleSpecifier ,
code : Option < String > ,
2021-11-16 09:02:28 -05:00
) -> Result < ModuleId , Error > {
2021-09-17 21:44:53 -04:00
let module_map_rc = Self ::module_map ( self . v8_isolate ( ) ) ;
if let Some ( code ) = code {
2022-04-26 09:28:42 -04:00
let scope = & mut self . handle_scope ( ) ;
module_map_rc
. borrow_mut ( )
. new_es_module (
scope ,
// not main module
false ,
specifier . as_str ( ) ,
2022-05-05 07:16:25 -04:00
code . as_bytes ( ) ,
2022-04-26 09:28:42 -04:00
)
. map_err ( | e | match e {
ModuleError ::Exception ( exception ) = > {
let exception = v8 ::Local ::new ( scope , exception ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
}
ModuleError ::Other ( error ) = > error ,
} ) ? ;
2021-09-17 21:44:53 -04:00
}
let mut load =
ModuleMap ::load_side ( module_map_rc . clone ( ) , specifier . as_str ( ) ) . await ? ;
2021-12-15 13:22:36 -05:00
while let Some ( load_result ) = load . next ( ) . await {
let ( request , info ) = load_result ? ;
2021-09-17 21:44:53 -04:00
let scope = & mut self . handle_scope ( ) ;
2022-04-26 09:28:42 -04:00
load . register_and_recurse ( scope , & request , & info ) . map_err (
| e | match e {
ModuleError ::Exception ( exception ) = > {
let exception = v8 ::Local ::new ( scope , exception ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
}
ModuleError ::Other ( error ) = > error ,
} ,
) ? ;
2021-09-17 21:44:53 -04:00
}
let root_id = load . root_module_id . expect ( " Root module should be loaded " ) ;
2022-04-26 09:28:42 -04:00
self . instantiate_module ( root_id ) . map_err ( | e | {
let scope = & mut self . handle_scope ( ) ;
let exception = v8 ::Local ::new ( scope , e ) ;
exception_to_err_result ::< ( ) > ( scope , exception , false ) . unwrap_err ( )
} ) ? ;
2021-09-17 21:44:53 -04:00
Ok ( root_id )
}
2021-11-16 09:02:28 -05:00
fn check_promise_exceptions ( & mut self ) -> Result < ( ) , Error > {
2022-10-21 10:13:42 -04:00
let mut state = self . state . borrow_mut ( ) ;
2020-10-05 05:08:19 -04:00
if state . pending_promise_exceptions . is_empty ( ) {
return Ok ( ( ) ) ;
}
2020-11-11 17:11:40 -05:00
let key = {
state
. pending_promise_exceptions
. keys ( )
. next ( )
. unwrap ( )
. clone ( )
} ;
2020-10-05 05:08:19 -04:00
let handle = state . pending_promise_exceptions . remove ( & key ) . unwrap ( ) ;
drop ( state ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2020-10-05 05:08:19 -04:00
let exception = v8 ::Local ::new ( scope , handle ) ;
2020-10-25 23:34:00 -04:00
exception_to_err_result ( scope , exception , true )
2020-10-05 05:08:19 -04:00
}
2021-03-31 10:37:38 -04:00
// Send finished responses to JS
2021-11-16 09:02:28 -05:00
fn resolve_async_ops ( & mut self , cx : & mut Context ) -> Result < ( ) , Error > {
2022-10-21 10:13:42 -04:00
let isolate = self . v8_isolate . as_mut ( ) . unwrap ( ) ;
2022-11-10 06:56:02 -05:00
let scope = & mut self
. state
. borrow ( )
. global_realm
. as_ref ( )
. unwrap ( )
. handle_scope ( isolate ) ;
2020-10-05 05:08:19 -04:00
2022-10-20 22:54:22 -04:00
// We return async responses to JS in unbounded batches (may change),
// each batch is a flat vector of tuples:
// `[promise_id1, op_result1, promise_id2, op_result2, ...]`
// promise_id is a simple integer, op_result is an ops::OpResult
// which contains a value OR an error, encoded as a tuple.
// This batch is received in JS via the special `arguments` variable
// and then each tuple is used to resolve or reject promises
2022-11-10 06:56:02 -05:00
//
// This can handle 16 promises (32 / 2) futures in a single batch without heap
// allocations.
let mut args : SmallVec < [ v8 ::Local < v8 ::Value > ; 32 ] > = SmallVec ::new ( ) ;
2021-10-24 15:41:57 -04:00
// Now handle actual ops.
{
2022-10-21 10:13:42 -04:00
let mut state = self . state . borrow_mut ( ) ;
2021-10-24 15:41:57 -04:00
state . have_unpolled_ops = false ;
2021-11-25 13:49:09 -05:00
while let Poll ::Ready ( Some ( item ) ) = state . pending_ops . poll_next_unpin ( cx )
{
2022-10-20 22:54:22 -04:00
let ( promise_id , op_id , mut resp ) = item ;
state . unrefed_ops . remove ( & promise_id ) ;
2022-03-14 13:44:15 -04:00
state . op_state . borrow ( ) . tracker . track_async_completed ( op_id ) ;
2022-10-20 22:54:22 -04:00
args . push ( v8 ::Integer ::new ( scope , promise_id as i32 ) . into ( ) ) ;
2022-09-01 16:20:11 -04:00
args . push ( match resp . to_v8 ( scope ) {
Ok ( v ) = > v ,
Err ( e ) = > OpResult ::Err ( OpError ::new ( & | _ | " TypeError " , e . into ( ) ) )
. to_v8 ( scope )
. unwrap ( ) ,
} ) ;
feat(core): Add support for async ops in realms (#14734)
Pull request #14019 enabled initial support for realms, but it did not
include support for async ops anywhere other than the main realm. The
main issue was that the `js_recv_cb` callback, which resolves promises
corresponding to async ops, was only set for the main realm, so async
ops in other realms would never resolve. Furthermore, promise ID's are
specific to each realm, which meant that async ops from other realms
would result in a wrong promise from the main realm being resolved.
This change creates a `ContextState` struct, similar to
`JsRuntimeState` but stored in a slot of each `v8::Context`, which
contains a `js_recv_cb` callback for each realm. Combined with a new
list of known realms, which stores them as `v8::Weak<v8::Context>`,
and a change in the `#[op]` macro to pass the current context to
`queue_async_op`, this makes it possible to send the results of
promises for different realms to their realm, and prevent the ID's
from getting mixed up.
Additionally, since promise ID's are no longer unique to the isolate,
having a single set of unrefed ops doesn't work. This change therefore
also moves `unrefed_ops` from `JsRuntimeState` to `ContextState`, and
adds the lengths of the unrefed op sets for all known realms to get
the total number of unrefed ops to compare in the event loop.
Co-authored-by: Luis Malheiro <luismalheiro@gmail.com>
2022-08-10 14:04:20 -04:00
}
2022-10-20 22:54:22 -04:00
}
2020-10-05 05:08:19 -04:00
2022-10-20 22:54:22 -04:00
if args . is_empty ( ) {
return Ok ( ( ) ) ;
2020-10-05 05:08:19 -04:00
}
feat(core): Add support for async ops in realms (#14734)
Pull request #14019 enabled initial support for realms, but it did not
include support for async ops anywhere other than the main realm. The
main issue was that the `js_recv_cb` callback, which resolves promises
corresponding to async ops, was only set for the main realm, so async
ops in other realms would never resolve. Furthermore, promise ID's are
specific to each realm, which meant that async ops from other realms
would result in a wrong promise from the main realm being resolved.
This change creates a `ContextState` struct, similar to
`JsRuntimeState` but stored in a slot of each `v8::Context`, which
contains a `js_recv_cb` callback for each realm. Combined with a new
list of known realms, which stores them as `v8::Weak<v8::Context>`,
and a change in the `#[op]` macro to pass the current context to
`queue_async_op`, this makes it possible to send the results of
promises for different realms to their realm, and prevent the ID's
from getting mixed up.
Additionally, since promise ID's are no longer unique to the isolate,
having a single set of unrefed ops doesn't work. This change therefore
also moves `unrefed_ops` from `JsRuntimeState` to `ContextState`, and
adds the lengths of the unrefed op sets for all known realms to get
the total number of unrefed ops to compare in the event loop.
Co-authored-by: Luis Malheiro <luismalheiro@gmail.com>
2022-08-10 14:04:20 -04:00
2022-11-10 06:56:02 -05:00
let js_recv_cb_handle = self . state . borrow ( ) . js_recv_cb . clone ( ) . unwrap ( ) ;
2022-10-20 22:54:22 -04:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let js_recv_cb = js_recv_cb_handle . open ( tc_scope ) ;
let this = v8 ::undefined ( tc_scope ) . into ( ) ;
js_recv_cb . call ( tc_scope , this , args . as_slice ( ) ) ;
match tc_scope . exception ( ) {
None = > Ok ( ( ) ) ,
Some ( exception ) = > exception_to_err_result ( tc_scope , exception , false ) ,
}
2020-10-05 05:08:19 -04:00
}
2021-11-16 09:02:28 -05:00
fn drain_macrotasks ( & mut self ) -> Result < ( ) , Error > {
2022-10-21 10:13:42 -04:00
if self . state . borrow ( ) . js_macrotask_cbs . is_empty ( ) {
2021-11-16 14:23:12 -05:00
return Ok ( ( ) ) ;
}
2020-10-05 05:08:19 -04:00
2022-10-21 10:13:42 -04:00
let js_macrotask_cb_handles = self . state . borrow ( ) . js_macrotask_cbs . clone ( ) ;
2021-04-28 12:28:46 -04:00
let scope = & mut self . handle_scope ( ) ;
2020-10-05 05:08:19 -04:00
2021-11-16 14:23:12 -05:00
for js_macrotask_cb_handle in js_macrotask_cb_handles {
let js_macrotask_cb = js_macrotask_cb_handle . open ( scope ) ;
2020-10-05 05:08:19 -04:00
2021-11-16 14:23:12 -05:00
// Repeatedly invoke macrotask callback until it returns true (done),
// such that ready microtasks would be automatically run before
// next macrotask is processed.
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let this = v8 ::undefined ( tc_scope ) . into ( ) ;
loop {
let is_done = js_macrotask_cb . call ( tc_scope , this , & [ ] ) ;
if let Some ( exception ) = tc_scope . exception ( ) {
return exception_to_err_result ( tc_scope , exception , false ) ;
}
if tc_scope . has_terminated ( ) | | tc_scope . is_execution_terminating ( ) {
return Ok ( ( ) ) ;
}
2020-10-05 05:08:19 -04:00
2021-11-16 14:23:12 -05:00
let is_done = is_done . unwrap ( ) ;
if is_done . is_true ( ) {
break ;
}
2021-09-30 13:52:58 -04:00
}
2021-11-16 14:23:12 -05:00
}
Ok ( ( ) )
}
2021-09-30 13:52:58 -04:00
2021-11-16 14:23:12 -05:00
fn drain_nexttick ( & mut self ) -> Result < ( ) , Error > {
2022-10-21 10:13:42 -04:00
if self . state . borrow ( ) . js_nexttick_cbs . is_empty ( ) {
2021-11-16 14:23:12 -05:00
return Ok ( ( ) ) ;
}
2022-10-21 10:13:42 -04:00
let state = self . state . clone ( ) ;
2021-11-16 14:23:12 -05:00
if ! state . borrow ( ) . has_tick_scheduled {
let scope = & mut self . handle_scope ( ) ;
scope . perform_microtask_checkpoint ( ) ;
}
// TODO(bartlomieju): Node also checks for absence of "rejection_to_warn"
if ! state . borrow ( ) . has_tick_scheduled {
return Ok ( ( ) ) ;
}
let js_nexttick_cb_handles = state . borrow ( ) . js_nexttick_cbs . clone ( ) ;
let scope = & mut self . handle_scope ( ) ;
for js_nexttick_cb_handle in js_nexttick_cb_handles {
let js_nexttick_cb = js_nexttick_cb_handle . open ( scope ) ;
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let this = v8 ::undefined ( tc_scope ) . into ( ) ;
js_nexttick_cb . call ( tc_scope , this , & [ ] ) ;
if let Some ( exception ) = tc_scope . exception ( ) {
return exception_to_err_result ( tc_scope , exception , false ) ;
2020-10-05 05:08:19 -04:00
}
}
Ok ( ( ) )
}
2020-09-06 10:50:49 -04:00
}
2022-04-17 07:53:08 -04:00
/// A representation of a JavaScript realm tied to a [`JsRuntime`], that allows
/// execution in the realm's context.
///
2022-10-15 16:44:51 -04:00
/// A [`JsRealm`] instance is a reference to an already existing realm, which
/// does not hold ownership of it, so instances can be created and dropped as
/// needed. As such, calling [`JsRealm::new`] doesn't create a new realm, and
/// cloning a [`JsRealm`] only creates a new reference. See
/// [`JsRuntime::create_realm`] to create new realms instead.
///
/// Despite [`JsRealm`] instances being references, multiple instances that
/// point to the same realm won't overlap because every operation requires
/// passing a mutable reference to the [`v8::Isolate`]. Therefore, no operation
/// on two [`JsRealm`] instances tied to the same isolate can be run at the same
/// time, regardless of whether they point to the same realm.
2022-04-17 07:53:08 -04:00
///
/// # Panics
///
/// Every method of [`JsRealm`] will panic if you call if with a reference to a
2022-07-05 18:45:10 -04:00
/// [`v8::Isolate`] other than the one that corresponds to the current context.
2022-04-17 07:53:08 -04:00
///
/// # Lifetime of the realm
///
2022-10-15 16:44:51 -04:00
/// As long as the corresponding isolate is alive, a [`JsRealm`] instance will
/// keep the underlying V8 context alive even if it would have otherwise been
/// garbage collected.
2022-04-17 07:53:08 -04:00
#[ derive(Clone) ]
pub struct JsRealm ( v8 ::Global < v8 ::Context > ) ;
impl JsRealm {
pub fn new ( context : v8 ::Global < v8 ::Context > ) -> Self {
JsRealm ( context )
}
pub fn context ( & self ) -> & v8 ::Global < v8 ::Context > {
& self . 0
}
pub fn handle_scope < ' s > (
& self ,
2022-07-05 18:45:10 -04:00
isolate : & ' s mut v8 ::Isolate ,
2022-04-17 07:53:08 -04:00
) -> v8 ::HandleScope < ' s > {
2022-07-05 18:45:10 -04:00
v8 ::HandleScope ::with_context ( isolate , & self . 0 )
2022-04-17 07:53:08 -04:00
}
pub fn global_object < ' s > (
& self ,
2022-07-05 18:45:10 -04:00
isolate : & ' s mut v8 ::Isolate ,
2022-04-17 07:53:08 -04:00
) -> v8 ::Local < ' s , v8 ::Object > {
2022-07-05 18:45:10 -04:00
let scope = & mut self . handle_scope ( isolate ) ;
2022-04-17 07:53:08 -04:00
self . 0. open ( scope ) . global ( scope )
}
/// Executes traditional JavaScript code (traditional = not ES modules) in the
/// realm's context.
///
/// `name` can be a filepath or any other string, eg.
///
/// - "/some/file/path.js"
/// - "<anon>"
/// - "[native code]"
///
/// The same `name` value can be used for multiple executions.
///
2022-04-26 09:28:42 -04:00
/// `Error` can usually be downcast to `JsError`.
2022-04-17 07:53:08 -04:00
pub fn execute_script (
& self ,
2022-07-05 18:45:10 -04:00
isolate : & mut v8 ::Isolate ,
2022-04-17 07:53:08 -04:00
name : & str ,
source_code : & str ,
) -> Result < v8 ::Global < v8 ::Value > , Error > {
2022-07-05 18:45:10 -04:00
let scope = & mut self . handle_scope ( isolate ) ;
2022-04-17 07:53:08 -04:00
let source = v8 ::String ::new ( scope , source_code ) . unwrap ( ) ;
let name = v8 ::String ::new ( scope , name ) . unwrap ( ) ;
let origin = bindings ::script_origin ( scope , name ) ;
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let script = match v8 ::Script ::compile ( tc_scope , source , Some ( & origin ) ) {
Some ( script ) = > script ,
None = > {
let exception = tc_scope . exception ( ) . unwrap ( ) ;
return exception_to_err_result ( tc_scope , exception , false ) ;
}
} ;
match script . run ( tc_scope ) {
Some ( value ) = > {
let value_handle = v8 ::Global ::new ( tc_scope , value ) ;
Ok ( value_handle )
}
None = > {
assert! ( tc_scope . has_caught ( ) ) ;
let exception = tc_scope . exception ( ) . unwrap ( ) ;
exception_to_err_result ( tc_scope , exception , false )
}
}
}
// TODO(andreubotella): `mod_evaluate`, `load_main_module`, `load_side_module`
}
2022-11-11 08:44:53 -05:00
#[ inline ]
pub fn queue_fast_async_op (
ctx : & OpCtx ,
op : impl Future < Output = ( PromiseId , OpId , OpResult ) > + 'static ,
) {
let runtime_state = match ctx . runtime_state . upgrade ( ) {
Some ( rc_state ) = > rc_state ,
// atleast 1 Rc is held by the JsRuntime.
None = > unreachable! ( ) ,
} ;
let mut state = runtime_state . borrow_mut ( ) ;
state . pending_ops . push ( OpCall ::lazy ( op ) ) ;
state . have_unpolled_ops = true ;
}
2022-03-14 13:44:15 -04:00
#[ inline ]
pub fn queue_async_op (
2022-10-21 10:13:42 -04:00
ctx : & OpCtx ,
2022-09-06 13:38:37 -04:00
scope : & mut v8 ::HandleScope ,
deferred : bool ,
2022-10-20 22:54:22 -04:00
op : impl Future < Output = ( PromiseId , OpId , OpResult ) > + 'static ,
2022-03-14 13:44:15 -04:00
) {
2022-10-21 10:13:42 -04:00
let runtime_state = match ctx . runtime_state . upgrade ( ) {
Some ( rc_state ) = > rc_state ,
// atleast 1 Rc is held by the JsRuntime.
None = > unreachable! ( ) ,
} ;
2022-09-06 13:38:37 -04:00
match OpCall ::eager ( op ) {
// This calls promise.resolve() before the control goes back to userland JS. It works something
// along the lines of:
//
// function opresolve(promiseId, ...) {
// getPromise(promiseId).resolve(...);
// }
// const p = setPromise();
// op.op_async(promiseId, ...); // Calls `opresolve`
// return p;
2022-10-20 22:54:22 -04:00
EagerPollResult ::Ready ( ( promise_id , op_id , mut resp ) ) if ! deferred = > {
2022-09-06 13:38:37 -04:00
let args = & [
v8 ::Integer ::new ( scope , promise_id ) . into ( ) ,
resp . to_v8 ( scope ) . unwrap ( ) ,
] ;
let js_recv_cb_handle =
2022-10-21 10:13:42 -04:00
runtime_state . borrow ( ) . js_recv_cb . clone ( ) . unwrap ( ) ;
ctx . state . borrow_mut ( ) . tracker . track_async_completed ( op_id ) ;
2022-09-06 13:38:37 -04:00
let tc_scope = & mut v8 ::TryCatch ::new ( scope ) ;
let js_recv_cb = js_recv_cb_handle . open ( tc_scope ) ;
let this = v8 ::undefined ( tc_scope ) . into ( ) ;
js_recv_cb . call ( tc_scope , this , args ) ;
}
EagerPollResult ::Ready ( op ) = > {
let ready = OpCall ::ready ( op ) ;
2022-10-21 10:13:42 -04:00
let mut state = runtime_state . borrow_mut ( ) ;
2022-09-06 13:38:37 -04:00
state . pending_ops . push ( ready ) ;
state . have_unpolled_ops = true ;
}
EagerPollResult ::Pending ( op ) = > {
2022-10-21 10:13:42 -04:00
let mut state = runtime_state . borrow_mut ( ) ;
2022-09-06 13:38:37 -04:00
state . pending_ops . push ( op ) ;
state . have_unpolled_ops = true ;
}
}
2022-03-14 13:44:15 -04:00
}
2019-03-11 17:57:36 -04:00
#[ cfg(test) ]
2019-03-26 11:56:34 -04:00
pub mod tests {
2019-03-11 17:57:36 -04:00
use super ::* ;
2021-05-03 11:30:41 -04:00
use crate ::error ::custom_error ;
2022-03-14 13:44:15 -04:00
use crate ::error ::AnyError ;
2021-11-22 07:51:20 -05:00
use crate ::modules ::ModuleSource ;
2020-09-06 10:50:49 -04:00
use crate ::modules ::ModuleSourceFuture ;
2021-12-15 13:22:36 -05:00
use crate ::modules ::ModuleType ;
2021-05-06 13:32:03 -04:00
use crate ::ZeroCopyBuf ;
2022-03-14 13:44:15 -04:00
use deno_ops ::op ;
2019-04-14 21:58:27 -04:00
use futures ::future ::lazy ;
use std ::ops ::FnOnce ;
2021-10-09 16:37:19 -04:00
use std ::pin ::Pin ;
2020-09-05 20:34:02 -04:00
use std ::rc ::Rc ;
2019-03-25 17:43:31 -04:00
use std ::sync ::atomic ::{ AtomicUsize , Ordering } ;
2020-05-29 17:41:39 -04:00
use std ::sync ::Arc ;
2022-03-14 13:44:15 -04:00
// deno_ops macros generate code assuming deno_core in scope.
mod deno_core {
pub use crate ::* ;
}
2019-03-14 19:17:52 -04:00
2019-11-16 19:17:47 -05:00
pub fn run_in_task < F > ( f : F )
2019-04-14 21:58:27 -04:00
where
2019-11-16 19:17:47 -05:00
F : FnOnce ( & mut Context ) + Send + 'static ,
2019-04-14 21:58:27 -04:00
{
2019-12-07 15:04:17 -05:00
futures ::executor ::block_on ( lazy ( move | cx | f ( cx ) ) ) ;
2019-04-14 21:58:27 -04:00
}
2021-12-29 09:21:42 -05:00
#[ derive(Copy, Clone) ]
2020-09-05 20:34:02 -04:00
enum Mode {
2019-12-07 15:04:17 -05:00
Async ,
2022-10-28 07:20:17 -04:00
AsyncDeferred ,
2021-03-31 10:37:38 -04:00
AsyncZeroCopy ( bool ) ,
2019-03-14 19:17:52 -04:00
}
2020-09-10 09:57:45 -04:00
struct TestState {
2020-09-05 20:34:02 -04:00
mode : Mode ,
dispatch_count : Arc < AtomicUsize > ,
}
2019-10-02 13:05:48 -04:00
2022-10-28 07:20:17 -04:00
#[ op ]
2022-03-14 13:44:15 -04:00
async fn op_test (
rc_op_state : Rc < RefCell < OpState > > ,
control : u8 ,
buf : Option < ZeroCopyBuf > ,
) -> Result < u8 , AnyError > {
2022-10-28 07:20:17 -04:00
#![ allow(clippy::await_holding_refcell_ref) ] // False positive.
2022-03-14 13:44:15 -04:00
let op_state_ = rc_op_state . borrow ( ) ;
2020-09-10 09:57:45 -04:00
let test_state = op_state_ . borrow ::< TestState > ( ) ;
test_state . dispatch_count . fetch_add ( 1 , Ordering ::Relaxed ) ;
2022-10-28 07:20:17 -04:00
let mode = test_state . mode ;
drop ( op_state_ ) ;
match mode {
2020-09-10 09:57:45 -04:00
Mode ::Async = > {
2021-03-31 10:37:38 -04:00
assert_eq! ( control , 42 ) ;
2022-03-14 13:44:15 -04:00
Ok ( 43 )
2020-09-05 20:34:02 -04:00
}
2022-10-28 07:20:17 -04:00
Mode ::AsyncDeferred = > {
tokio ::task ::yield_now ( ) . await ;
assert_eq! ( control , 42 ) ;
Ok ( 43 )
}
2021-03-31 10:37:38 -04:00
Mode ::AsyncZeroCopy ( has_buffer ) = > {
assert_eq! ( buf . is_some ( ) , has_buffer ) ;
if let Some ( buf ) = buf {
2020-09-10 09:57:45 -04:00
assert_eq! ( buf . len ( ) , 1 ) ;
2021-03-31 10:37:38 -04:00
}
2022-03-14 13:44:15 -04:00
Ok ( 43 )
2020-04-19 23:54:46 -04:00
}
2020-09-05 20:34:02 -04:00
}
}
2019-10-02 13:05:48 -04:00
2020-09-06 15:44:29 -04:00
fn setup ( mode : Mode ) -> ( JsRuntime , Arc < AtomicUsize > ) {
2020-09-05 20:34:02 -04:00
let dispatch_count = Arc ::new ( AtomicUsize ::new ( 0 ) ) ;
2021-12-29 09:21:42 -05:00
let dispatch_count2 = dispatch_count . clone ( ) ;
let ext = Extension ::builder ( )
2022-03-14 13:44:15 -04:00
. ops ( vec! [ op_test ::decl ( ) ] )
2021-12-29 09:21:42 -05:00
. state ( move | state | {
state . put ( TestState {
mode ,
dispatch_count : dispatch_count2 . clone ( ) ,
} ) ;
Ok ( ( ) )
} )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
2022-06-07 05:25:10 -04:00
get_error_class_fn : Some ( & | error | {
crate ::error ::get_custom_error_class ( error ) . unwrap ( )
} ) ,
2021-12-29 09:21:42 -05:00
.. Default ::default ( )
2020-09-05 20:34:02 -04:00
} ) ;
2020-09-10 09:57:45 -04:00
2020-09-22 17:30:03 -04:00
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-09-22 17:30:03 -04:00
" setup.js " ,
r #"
2019-04-23 18:58:00 -04:00
function assert ( cond ) {
if ( ! cond ) {
throw Error ( " assert " ) ;
}
2019-03-14 19:17:52 -04:00
}
2019-04-23 18:58:00 -04:00
" #,
2020-09-22 17:30:03 -04:00
)
. unwrap ( ) ;
2019-04-23 18:58:00 -04:00
assert_eq! ( dispatch_count . load ( Ordering ::Relaxed ) , 0 ) ;
2020-09-06 15:44:29 -04:00
( runtime , dispatch_count )
2019-03-14 19:17:52 -04:00
}
2019-03-11 17:57:36 -04:00
2021-11-25 13:49:09 -05:00
#[ test ]
fn test_ref_unref_ops ( ) {
2022-10-28 07:20:17 -04:00
let ( mut runtime , _dispatch_count ) = setup ( Mode ::AsyncDeferred ) ;
2021-11-25 13:49:09 -05:00
runtime
. execute_script (
" filename.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2021-11-25 13:49:09 -05:00
var promiseIdSymbol = Symbol . for ( " Deno.core.internalPromiseId " ) ;
2022-10-28 07:20:17 -04:00
var p1 = Deno . core . ops . op_test ( 42 ) ;
var p2 = Deno . core . ops . op_test ( 42 ) ;
2021-11-25 13:49:09 -05:00
" #,
)
. unwrap ( ) ;
{
let isolate = runtime . v8_isolate ( ) ;
let state_rc = JsRuntime ::state ( isolate ) ;
2022-10-20 22:54:22 -04:00
let state = state_rc . borrow ( ) ;
assert_eq! ( state . pending_ops . len ( ) , 2 ) ;
assert_eq! ( state . unrefed_ops . len ( ) , 0 ) ;
2021-11-25 13:49:09 -05:00
}
runtime
. execute_script (
" filename.js " ,
r #"
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_unref_op ( p1 [ promiseIdSymbol ] ) ;
Deno . core . ops . op_unref_op ( p2 [ promiseIdSymbol ] ) ;
2021-11-25 13:49:09 -05:00
" #,
)
. unwrap ( ) ;
{
let isolate = runtime . v8_isolate ( ) ;
let state_rc = JsRuntime ::state ( isolate ) ;
2022-10-20 22:54:22 -04:00
let state = state_rc . borrow ( ) ;
assert_eq! ( state . pending_ops . len ( ) , 2 ) ;
assert_eq! ( state . unrefed_ops . len ( ) , 2 ) ;
2021-11-25 13:49:09 -05:00
}
runtime
. execute_script (
" filename.js " ,
r #"
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_ref_op ( p1 [ promiseIdSymbol ] ) ;
Deno . core . ops . op_ref_op ( p2 [ promiseIdSymbol ] ) ;
2021-11-25 13:49:09 -05:00
" #,
)
. unwrap ( ) ;
{
let isolate = runtime . v8_isolate ( ) ;
let state_rc = JsRuntime ::state ( isolate ) ;
2022-10-20 22:54:22 -04:00
let state = state_rc . borrow ( ) ;
assert_eq! ( state . pending_ops . len ( ) , 2 ) ;
assert_eq! ( state . unrefed_ops . len ( ) , 0 ) ;
2021-11-25 13:49:09 -05:00
}
}
2022-09-06 13:38:37 -04:00
#[ test ]
fn test_dispatch ( ) {
let ( mut runtime , dispatch_count ) = setup ( Mode ::Async ) ;
runtime
. execute_script (
" filename.js " ,
r #"
let control = 42 ;
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2022-09-06 13:38:37 -04:00
Deno . core . opAsync ( " op_test " , control ) ;
async function main ( ) {
Deno . core . opAsync ( " op_test " , control ) ;
}
main ( ) ;
" #,
)
. unwrap ( ) ;
assert_eq! ( dispatch_count . load ( Ordering ::Relaxed ) , 2 ) ;
}
#[ test ]
fn test_op_async_promise_id ( ) {
let ( mut runtime , _dispatch_count ) = setup ( Mode ::Async ) ;
runtime
. execute_script (
" filename.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2022-09-06 13:38:37 -04:00
const p = Deno . core . opAsync ( " op_test " , 42 ) ;
if ( p [ Symbol . for ( " Deno.core.internalPromiseId " ) ] = = undefined ) {
throw new Error ( " missing id on returned promise " ) ;
}
" #,
)
. unwrap ( ) ;
}
2020-06-01 14:20:47 -04:00
#[ test ]
fn test_dispatch_no_zero_copy_buf ( ) {
2021-03-31 10:37:38 -04:00
let ( mut runtime , dispatch_count ) = setup ( Mode ::AsyncZeroCopy ( false ) ) ;
2020-09-22 17:30:03 -04:00
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-09-22 17:30:03 -04:00
" filename.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2021-04-30 21:08:29 -04:00
Deno . core . opAsync ( " op_test " ) ;
2020-06-01 14:20:47 -04:00
" #,
2020-09-22 17:30:03 -04:00
)
. unwrap ( ) ;
2020-06-01 14:20:47 -04:00
assert_eq! ( dispatch_count . load ( Ordering ::Relaxed ) , 1 ) ;
}
#[ test ]
2020-07-08 11:23:50 -04:00
fn test_dispatch_stack_zero_copy_bufs ( ) {
2021-03-31 10:37:38 -04:00
let ( mut runtime , dispatch_count ) = setup ( Mode ::AsyncZeroCopy ( true ) ) ;
2020-09-22 17:30:03 -04:00
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-09-22 17:30:03 -04:00
" filename.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2020-07-08 11:23:50 -04:00
let zero_copy_a = new Uint8Array ( [ 0 ] ) ;
2021-04-30 21:08:29 -04:00
Deno . core . opAsync ( " op_test " , null , zero_copy_a ) ;
2020-06-01 14:20:47 -04:00
" #,
2020-09-22 17:30:03 -04:00
)
. unwrap ( ) ;
2020-06-01 14:20:47 -04:00
assert_eq! ( dispatch_count . load ( Ordering ::Relaxed ) , 1 ) ;
}
2021-07-08 12:56:53 -04:00
#[ test ]
fn test_execute_script_return_value ( ) {
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
let value_global = runtime . execute_script ( " a.js " , " a = 1 + 2 " ) . unwrap ( ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
2021-10-27 17:26:15 -04:00
let value = value_global . open ( scope ) ;
2021-07-08 12:56:53 -04:00
assert_eq! ( value . integer_value ( scope ) . unwrap ( ) , 3 ) ;
}
let value_global = runtime . execute_script ( " b.js " , " b = 'foobar' " ) . unwrap ( ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
2021-10-27 17:26:15 -04:00
let value = value_global . open ( scope ) ;
2021-07-08 12:56:53 -04:00
assert! ( value . is_string ( ) ) ;
assert_eq! (
value . to_string ( scope ) . unwrap ( ) . to_rust_string_lossy ( scope ) ,
" foobar "
) ;
}
}
2022-01-24 11:59:41 -05:00
#[ tokio::test ]
async fn test_poll_value ( ) {
run_in_task ( | cx | {
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
let value_global = runtime
. execute_script ( " a.js " , " Promise.resolve(1 + 2) " )
. unwrap ( ) ;
let v = runtime . poll_value ( & value_global , cx ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
assert! (
matches! ( v , Poll ::Ready ( Ok ( v ) ) if v . open ( scope ) . integer_value ( scope ) . unwrap ( ) = = 3 )
) ;
}
let value_global = runtime
. execute_script (
" a.js " ,
" Promise.resolve(new Promise(resolve => resolve(2 + 2))) " ,
)
. unwrap ( ) ;
let v = runtime . poll_value ( & value_global , cx ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
assert! (
matches! ( v , Poll ::Ready ( Ok ( v ) ) if v . open ( scope ) . integer_value ( scope ) . unwrap ( ) = = 4 )
) ;
}
let value_global = runtime
. execute_script ( " a.js " , " Promise.reject(new Error('fail')) " )
. unwrap ( ) ;
let v = runtime . poll_value ( & value_global , cx ) ;
assert! (
2022-04-13 10:41:39 -04:00
matches! ( v , Poll ::Ready ( Err ( e ) ) if e . downcast_ref ::< JsError > ( ) . unwrap ( ) . exception_message = = " Uncaught Error: fail " )
2022-01-24 11:59:41 -05:00
) ;
let value_global = runtime
. execute_script ( " a.js " , " new Promise(resolve => {}) " )
. unwrap ( ) ;
let v = runtime . poll_value ( & value_global , cx ) ;
matches! ( v , Poll ::Ready ( Err ( e ) ) if e . to_string ( ) = = " Promise resolution is still pending but the event loop has already resolved. " ) ;
} ) ;
}
2021-09-04 14:19:26 -04:00
#[ tokio::test ]
async fn test_resolve_value ( ) {
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
let value_global = runtime
. execute_script ( " a.js " , " Promise.resolve(1 + 2) " )
. unwrap ( ) ;
let result_global = runtime . resolve_value ( value_global ) . await . unwrap ( ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
2021-10-27 17:26:15 -04:00
let value = result_global . open ( scope ) ;
2021-09-04 14:19:26 -04:00
assert_eq! ( value . integer_value ( scope ) . unwrap ( ) , 3 ) ;
}
let value_global = runtime
. execute_script (
" a.js " ,
" Promise.resolve(new Promise(resolve => resolve(2 + 2))) " ,
)
. unwrap ( ) ;
let result_global = runtime . resolve_value ( value_global ) . await . unwrap ( ) ;
{
let scope = & mut runtime . handle_scope ( ) ;
2021-10-27 17:26:15 -04:00
let value = result_global . open ( scope ) ;
2021-09-04 14:19:26 -04:00
assert_eq! ( value . integer_value ( scope ) . unwrap ( ) , 4 ) ;
}
let value_global = runtime
. execute_script ( " a.js " , " Promise.reject(new Error('fail')) " )
. unwrap ( ) ;
let err = runtime . resolve_value ( value_global ) . await . unwrap_err ( ) ;
assert_eq! (
" Uncaught Error: fail " ,
2022-04-13 10:41:39 -04:00
err . downcast ::< JsError > ( ) . unwrap ( ) . exception_message
2021-09-04 14:19:26 -04:00
) ;
let value_global = runtime
. execute_script ( " a.js " , " new Promise(resolve => {}) " )
. unwrap ( ) ;
let error_string = runtime
. resolve_value ( value_global )
. await
. unwrap_err ( )
. to_string ( ) ;
assert_eq! (
" Promise resolution is still pending but the event loop has already resolved. " ,
error_string ,
) ;
}
2022-07-01 03:51:29 -04:00
#[ test ]
fn terminate_execution_webassembly ( ) {
2022-07-07 10:28:29 -04:00
let ( mut runtime , _dispatch_count ) = setup ( Mode ::Async ) ;
let v8_isolate_handle = runtime . v8_isolate ( ) . thread_safe_handle ( ) ;
2022-07-01 03:51:29 -04:00
// Run an infinite loop in Webassemby code, which should be terminated.
2022-07-07 10:28:29 -04:00
let promise = runtime . execute_script ( " infinite_wasm_loop.js " ,
2022-07-01 03:51:29 -04:00
r #"
( async ( ) = > {
const wasmCode = new Uint8Array ( [
0 , 97 , 115 , 109 , 1 , 0 , 0 , 0 , 1 , 4 , 1 ,
96 , 0 , 0 , 3 , 2 , 1 , 0 , 7 , 17 , 1 , 13 ,
105 , 110 , 102 , 105 , 110 , 105 , 116 , 101 , 95 , 108 , 111 ,
111 , 112 , 0 , 0 , 10 , 9 , 1 , 7 , 0 , 3 , 64 ,
12 , 0 , 11 , 11 ,
] ) ;
const wasmModule = await WebAssembly . compile ( wasmCode ) ;
2022-07-07 10:28:29 -04:00
globalThis . wasmInstance = new WebAssembly . Instance ( wasmModule ) ;
} ) ( )
" #).unwrap();
futures ::executor ::block_on ( runtime . resolve_value ( promise ) ) . unwrap ( ) ;
let terminator_thread = std ::thread ::spawn ( move | | {
std ::thread ::sleep ( std ::time ::Duration ::from_millis ( 1000 ) ) ;
// terminate execution
let ok = v8_isolate_handle . terminate_execution ( ) ;
assert! ( ok ) ;
} ) ;
let err = runtime
. execute_script (
" infinite_wasm_loop2.js " ,
" globalThis.wasmInstance.exports.infinite_loop(); " ,
)
. unwrap_err ( ) ;
assert_eq! ( err . to_string ( ) , " Uncaught Error: execution terminated " ) ;
2022-07-01 03:51:29 -04:00
// Cancel the execution-terminating exception in order to allow script
// execution again.
2022-07-07 10:28:29 -04:00
let ok = runtime . v8_isolate ( ) . cancel_terminate_execution ( ) ;
2022-07-01 03:51:29 -04:00
assert! ( ok ) ;
// Verify that the isolate usable again.
2022-07-07 10:28:29 -04:00
runtime
2022-07-01 03:51:29 -04:00
. execute_script ( " simple.js " , " 1 + 1 " )
. expect ( " execution should be possible again " ) ;
terminator_thread . join ( ) . unwrap ( ) ;
}
2019-03-21 09:48:19 -04:00
#[ test ]
fn terminate_execution ( ) {
2019-12-07 15:04:17 -05:00
let ( mut isolate , _dispatch_count ) = setup ( Mode ::Async ) ;
2020-10-05 05:08:19 -04:00
let v8_isolate_handle = isolate . v8_isolate ( ) . thread_safe_handle ( ) ;
2019-03-21 09:48:19 -04:00
2020-02-24 18:53:29 -05:00
let terminator_thread = std ::thread ::spawn ( move | | {
2019-03-21 09:48:19 -04:00
// allow deno to boot and run
std ::thread ::sleep ( std ::time ::Duration ::from_millis ( 100 ) ) ;
// terminate execution
2020-02-24 18:53:29 -05:00
let ok = v8_isolate_handle . terminate_execution ( ) ;
2020-02-27 19:27:24 -05:00
assert! ( ok ) ;
2019-03-21 09:48:19 -04:00
} ) ;
2020-02-24 18:53:29 -05:00
// Rn an infinite loop, which should be terminated.
2021-06-21 19:45:41 -04:00
match isolate . execute_script ( " infinite_loop.js " , " for(;;) {} " ) {
2020-02-24 18:53:29 -05:00
Ok ( _ ) = > panic! ( " execution should be terminated " ) ,
Err ( e ) = > {
2020-10-31 14:57:19 -04:00
assert_eq! ( e . to_string ( ) , " Uncaught Error: execution terminated " )
2020-02-24 18:53:29 -05:00
}
} ;
2019-03-21 09:48:19 -04:00
2020-02-24 18:53:29 -05:00
// Cancel the execution-terminating exception in order to allow script
// execution again.
2020-12-28 10:36:44 -05:00
let ok = isolate . v8_isolate ( ) . cancel_terminate_execution ( ) ;
2020-02-24 18:53:29 -05:00
assert! ( ok ) ;
// Verify that the isolate usable again.
isolate
2021-06-21 19:45:41 -04:00
. execute_script ( " simple.js " , " 1 + 1 " )
2020-02-24 18:53:29 -05:00
. expect ( " execution should be possible again " ) ;
2019-03-21 09:48:19 -04:00
2020-02-24 18:53:29 -05:00
terminator_thread . join ( ) . unwrap ( ) ;
2019-03-21 09:48:19 -04:00
}
#[ test ]
fn dangling_shared_isolate ( ) {
2020-02-24 18:53:29 -05:00
let v8_isolate_handle = {
2019-03-21 09:48:19 -04:00
// isolate is dropped at the end of this block
2020-09-06 15:44:29 -04:00
let ( mut runtime , _dispatch_count ) = setup ( Mode ::Async ) ;
2020-10-05 05:08:19 -04:00
runtime . v8_isolate ( ) . thread_safe_handle ( )
2019-03-21 09:48:19 -04:00
} ;
// this should not SEGFAULT
2020-02-24 18:53:29 -05:00
v8_isolate_handle . terminate_execution ( ) ;
2019-03-21 09:48:19 -04:00
}
2020-04-16 06:58:17 -04:00
#[ test ]
fn syntax_error ( ) {
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
2020-04-16 06:58:17 -04:00
let src = " hocuspocus( " ;
2021-06-21 19:45:41 -04:00
let r = runtime . execute_script ( " i.js " , src ) ;
2020-04-16 06:58:17 -04:00
let e = r . unwrap_err ( ) ;
2020-09-06 15:44:29 -04:00
let js_error = e . downcast ::< JsError > ( ) . unwrap ( ) ;
2022-04-15 10:08:09 -04:00
let frame = js_error . frames . first ( ) . unwrap ( ) ;
assert_eq! ( frame . column_number , Some ( 12 ) ) ;
2020-04-16 06:58:17 -04:00
}
2020-03-15 10:31:55 -04:00
#[ test ]
fn test_encode_decode ( ) {
2021-12-04 08:19:06 -05:00
run_in_task ( | cx | {
2020-09-06 15:44:29 -04:00
let ( mut runtime , _dispatch_count ) = setup ( Mode ::Async ) ;
2020-09-22 17:30:03 -04:00
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-09-22 17:30:03 -04:00
" encode_decode_test.js " ,
include_str! ( " encode_decode_test.js " ) ,
)
. unwrap ( ) ;
2021-12-04 08:19:06 -05:00
if let Poll ::Ready ( Err ( _ ) ) = runtime . poll_event_loop ( cx , false ) {
2020-03-15 10:31:55 -04:00
unreachable! ( ) ;
}
} ) ;
}
2021-02-16 08:20:21 -05:00
#[ test ]
fn test_serialize_deserialize ( ) {
2021-12-04 08:19:06 -05:00
run_in_task ( | cx | {
2021-02-16 08:20:21 -05:00
let ( mut runtime , _dispatch_count ) = setup ( Mode ::Async ) ;
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2021-02-16 08:20:21 -05:00
" serialize_deserialize_test.js " ,
include_str! ( " serialize_deserialize_test.js " ) ,
)
. unwrap ( ) ;
2021-12-04 08:19:06 -05:00
if let Poll ::Ready ( Err ( _ ) ) = runtime . poll_event_loop ( cx , false ) {
2021-02-16 08:20:21 -05:00
unreachable! ( ) ;
}
} ) ;
}
2021-05-03 11:30:41 -04:00
#[ test ]
fn test_error_builder ( ) {
2022-03-14 13:44:15 -04:00
#[ op ]
2022-03-15 19:33:46 -04:00
fn op_err ( ) -> Result < ( ) , Error > {
2021-05-03 11:30:41 -04:00
Err ( custom_error ( " DOMExceptionOperationError " , " abc " ) )
}
2021-11-16 09:02:28 -05:00
pub fn get_error_class_name ( _ : & Error ) -> & 'static str {
2021-05-03 11:30:41 -04:00
" DOMExceptionOperationError "
}
2021-12-04 08:19:06 -05:00
run_in_task ( | cx | {
2022-03-14 13:44:15 -04:00
let ext = Extension ::builder ( ) . ops ( vec! [ op_err ::decl ( ) ] ) . build ( ) ;
2021-05-03 11:30:41 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
2021-12-29 09:21:42 -05:00
extensions : vec ! [ ext ] ,
2021-05-03 11:30:41 -04:00
get_error_class_fn : Some ( & get_error_class_name ) ,
.. Default ::default ( )
} ) ;
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2021-05-03 11:30:41 -04:00
" error_builder_test.js " ,
include_str! ( " error_builder_test.js " ) ,
)
. unwrap ( ) ;
2021-12-04 08:19:06 -05:00
if let Poll ::Ready ( Err ( _ ) ) = runtime . poll_event_loop ( cx , false ) {
2021-05-03 11:30:41 -04:00
unreachable! ( ) ;
}
} ) ;
}
2019-04-24 21:43:06 -04:00
#[ test ]
fn will_snapshot ( ) {
let snapshot = {
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
will_snapshot : true ,
.. Default ::default ( )
} ) ;
2021-06-21 19:45:41 -04:00
runtime . execute_script ( " a.js " , " a = 1 + 2 " ) . unwrap ( ) ;
2020-09-06 15:44:29 -04:00
runtime . snapshot ( )
2019-04-24 21:43:06 -04:00
} ;
2020-09-11 09:18:49 -04:00
let snapshot = Snapshot ::JustCreated ( snapshot ) ;
let mut runtime2 = JsRuntime ::new ( RuntimeOptions {
startup_snapshot : Some ( snapshot ) ,
.. Default ::default ( )
} ) ;
2020-09-22 17:30:03 -04:00
runtime2
2021-06-21 19:45:41 -04:00
. execute_script ( " check.js " , " if (a != 3) throw Error('x') " )
2020-09-22 17:30:03 -04:00
. unwrap ( ) ;
2019-04-24 21:43:06 -04:00
}
2022-10-28 15:31:01 -04:00
#[ test ]
fn will_snapshot2 ( ) {
let startup_data = {
let mut runtime = JsRuntime ::new ( RuntimeOptions {
will_snapshot : true ,
.. Default ::default ( )
} ) ;
runtime . execute_script ( " a.js " , " let a = 1 + 2 " ) . unwrap ( ) ;
runtime . snapshot ( )
} ;
let snapshot = Snapshot ::JustCreated ( startup_data ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
will_snapshot : true ,
startup_snapshot : Some ( snapshot ) ,
.. Default ::default ( )
} ) ;
let startup_data = {
runtime
. execute_script ( " check_a.js " , " if (a != 3) throw Error('x') " )
. unwrap ( ) ;
runtime . execute_script ( " b.js " , " b = 2 + 3 " ) . unwrap ( ) ;
runtime . snapshot ( )
} ;
let snapshot = Snapshot ::JustCreated ( startup_data ) ;
{
let mut runtime = JsRuntime ::new ( RuntimeOptions {
startup_snapshot : Some ( snapshot ) ,
.. Default ::default ( )
} ) ;
runtime
. execute_script ( " check_b.js " , " if (b != 5) throw Error('x') " )
. unwrap ( ) ;
runtime
. execute_script ( " check2.js " , " if (!Deno.core) throw Error('x') " )
. unwrap ( ) ;
}
}
2022-08-21 15:03:56 -04:00
#[ test ]
fn test_snapshot_callbacks ( ) {
let snapshot = {
let mut runtime = JsRuntime ::new ( RuntimeOptions {
will_snapshot : true ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" a.js " ,
r #"
Deno . core . ops . op_set_macrotask_callback ( ( ) = > {
return true ;
} ) ;
Deno . core . ops . op_set_format_exception_callback ( ( ) = > {
return null ;
} )
Deno . core . setPromiseRejectCallback ( ( ) = > {
return false ;
} ) ;
2022-09-02 06:43:39 -04:00
a = 1 + 2 ;
2022-08-21 15:03:56 -04:00
" #,
)
. unwrap ( ) ;
runtime . snapshot ( )
} ;
let snapshot = Snapshot ::JustCreated ( snapshot ) ;
let mut runtime2 = JsRuntime ::new ( RuntimeOptions {
startup_snapshot : Some ( snapshot ) ,
.. Default ::default ( )
} ) ;
runtime2
. execute_script ( " check.js " , " if (a != 3) throw Error('x') " )
. unwrap ( ) ;
}
2020-05-09 21:00:40 -04:00
#[ test ]
fn test_from_boxed_snapshot ( ) {
let snapshot = {
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
will_snapshot : true ,
.. Default ::default ( )
} ) ;
2021-06-21 19:45:41 -04:00
runtime . execute_script ( " a.js " , " a = 1 + 2 " ) . unwrap ( ) ;
2022-11-17 20:59:10 -05:00
let snap : & [ u8 ] = & runtime . snapshot ( ) ;
2020-05-09 21:00:40 -04:00
Vec ::from ( snap ) . into_boxed_slice ( )
} ;
2020-09-11 09:18:49 -04:00
let snapshot = Snapshot ::Boxed ( snapshot ) ;
let mut runtime2 = JsRuntime ::new ( RuntimeOptions {
startup_snapshot : Some ( snapshot ) ,
.. Default ::default ( )
} ) ;
2020-09-22 17:30:03 -04:00
runtime2
2021-06-21 19:45:41 -04:00
. execute_script ( " check.js " , " if (a != 3) throw Error('x') " )
2020-09-22 17:30:03 -04:00
. unwrap ( ) ;
2020-05-09 21:00:40 -04:00
}
2020-08-11 21:07:14 -04:00
2022-03-22 09:32:32 -04:00
#[ test ]
fn test_get_module_namespace ( ) {
#[ derive(Default) ]
struct ModsLoader ;
impl ModuleLoader for ModsLoader {
fn resolve (
& self ,
specifier : & str ,
referrer : & str ,
_is_main : bool ,
) -> Result < ModuleSpecifier , Error > {
assert_eq! ( specifier , " file:///main.js " ) ;
assert_eq! ( referrer , " . " ) ;
let s = crate ::resolve_import ( specifier , referrer ) . unwrap ( ) ;
Ok ( s )
}
fn load (
& self ,
_module_specifier : & ModuleSpecifier ,
_maybe_referrer : Option < ModuleSpecifier > ,
_is_dyn_import : bool ,
) -> Pin < Box < ModuleSourceFuture > > {
async { Err ( generic_error ( " Module loading is not supported " ) ) }
. boxed_local ( )
}
}
let loader = std ::rc ::Rc ::new ( ModsLoader ::default ( ) ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
module_loader : Some ( loader ) ,
.. Default ::default ( )
} ) ;
let specifier = crate ::resolve_url ( " file:///main.js " ) . unwrap ( ) ;
let source_code = r #"
export const a = " b " ;
export default 1 + 2 ;
" #
. to_string ( ) ;
let module_id = futures ::executor ::block_on (
runtime . load_main_module ( & specifier , Some ( source_code ) ) ,
)
. unwrap ( ) ;
let _ = runtime . mod_evaluate ( module_id ) ;
let module_namespace = runtime . get_module_namespace ( module_id ) . unwrap ( ) ;
let scope = & mut runtime . handle_scope ( ) ;
let module_namespace =
v8 ::Local ::< v8 ::Object > ::new ( scope , module_namespace ) ;
assert! ( module_namespace . is_module_namespace_object ( ) ) ;
let unknown_export_name = v8 ::String ::new ( scope , " none " ) . unwrap ( ) ;
let binding = module_namespace . get ( scope , unknown_export_name . into ( ) ) ;
assert! ( binding . is_some ( ) ) ;
assert! ( binding . unwrap ( ) . is_undefined ( ) ) ;
let empty_export_name = v8 ::String ::new ( scope , " " ) . unwrap ( ) ;
let binding = module_namespace . get ( scope , empty_export_name . into ( ) ) ;
assert! ( binding . is_some ( ) ) ;
assert! ( binding . unwrap ( ) . is_undefined ( ) ) ;
let a_export_name = v8 ::String ::new ( scope , " a " ) . unwrap ( ) ;
let binding = module_namespace . get ( scope , a_export_name . into ( ) ) ;
assert! ( binding . unwrap ( ) . is_string ( ) ) ;
assert_eq! ( binding . unwrap ( ) , v8 ::String ::new ( scope , " b " ) . unwrap ( ) ) ;
let default_export_name = v8 ::String ::new ( scope , " default " ) . unwrap ( ) ;
let binding = module_namespace . get ( scope , default_export_name . into ( ) ) ;
assert! ( binding . unwrap ( ) . is_number ( ) ) ;
assert_eq! ( binding . unwrap ( ) , v8 ::Number ::new ( scope , 3_ f64 ) ) ;
}
2020-08-11 21:07:14 -04:00
#[ test ]
fn test_heap_limits ( ) {
2021-07-02 06:18:30 -04:00
let create_params =
v8 ::Isolate ::create_params ( ) . heap_limits ( 0 , 3 * 1024 * 1024 ) ;
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
2020-10-17 05:56:15 -04:00
create_params : Some ( create_params ) ,
2020-09-11 09:18:49 -04:00
.. Default ::default ( )
} ) ;
2020-10-07 09:56:52 -04:00
let cb_handle = runtime . v8_isolate ( ) . thread_safe_handle ( ) ;
2020-08-11 21:07:14 -04:00
2022-03-14 13:44:15 -04:00
let callback_invoke_count = Rc ::new ( AtomicUsize ::new ( 0 ) ) ;
2020-08-11 21:07:14 -04:00
let inner_invoke_count = Rc ::clone ( & callback_invoke_count ) ;
2020-09-06 15:44:29 -04:00
runtime . add_near_heap_limit_callback (
2020-08-11 21:07:14 -04:00
move | current_limit , _initial_limit | {
inner_invoke_count . fetch_add ( 1 , Ordering ::SeqCst ) ;
cb_handle . terminate_execution ( ) ;
current_limit * 2
} ,
) ;
2020-09-06 15:44:29 -04:00
let err = runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-08-11 21:07:14 -04:00
" script name " ,
r # "let s = ""; while(true) { s += "Hello"; }"# ,
)
. expect_err ( " script should fail " ) ;
assert_eq! (
" Uncaught Error: execution terminated " ,
2022-04-13 10:41:39 -04:00
err . downcast ::< JsError > ( ) . unwrap ( ) . exception_message
2020-08-11 21:07:14 -04:00
) ;
assert! ( callback_invoke_count . load ( Ordering ::SeqCst ) > 0 )
}
#[ test ]
fn test_heap_limit_cb_remove ( ) {
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
2020-08-11 21:07:14 -04:00
2020-09-06 15:44:29 -04:00
runtime . add_near_heap_limit_callback ( | current_limit , _initial_limit | {
2020-08-11 21:07:14 -04:00
current_limit * 2
} ) ;
2021-07-02 06:18:30 -04:00
runtime . remove_near_heap_limit_callback ( 3 * 1024 * 1024 ) ;
2020-09-06 15:44:29 -04:00
assert! ( runtime . allocations . near_heap_limit_callback_data . is_none ( ) ) ;
2020-08-11 21:07:14 -04:00
}
2020-08-12 00:08:50 -04:00
#[ test ]
fn test_heap_limit_cb_multiple ( ) {
2021-07-02 06:18:30 -04:00
let create_params =
v8 ::Isolate ::create_params ( ) . heap_limits ( 0 , 3 * 1024 * 1024 ) ;
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
2020-10-17 05:56:15 -04:00
create_params : Some ( create_params ) ,
2020-09-11 09:18:49 -04:00
.. Default ::default ( )
} ) ;
2020-10-07 09:56:52 -04:00
let cb_handle = runtime . v8_isolate ( ) . thread_safe_handle ( ) ;
2020-08-12 00:08:50 -04:00
2022-03-14 13:44:15 -04:00
let callback_invoke_count_first = Rc ::new ( AtomicUsize ::new ( 0 ) ) ;
2020-08-12 00:08:50 -04:00
let inner_invoke_count_first = Rc ::clone ( & callback_invoke_count_first ) ;
2020-09-06 15:44:29 -04:00
runtime . add_near_heap_limit_callback (
2020-08-12 00:08:50 -04:00
move | current_limit , _initial_limit | {
inner_invoke_count_first . fetch_add ( 1 , Ordering ::SeqCst ) ;
current_limit * 2
} ,
) ;
2022-03-14 13:44:15 -04:00
let callback_invoke_count_second = Rc ::new ( AtomicUsize ::new ( 0 ) ) ;
2020-08-12 00:08:50 -04:00
let inner_invoke_count_second = Rc ::clone ( & callback_invoke_count_second ) ;
2020-09-06 15:44:29 -04:00
runtime . add_near_heap_limit_callback (
2020-08-12 00:08:50 -04:00
move | current_limit , _initial_limit | {
inner_invoke_count_second . fetch_add ( 1 , Ordering ::SeqCst ) ;
cb_handle . terminate_execution ( ) ;
current_limit * 2
} ,
) ;
2020-09-06 15:44:29 -04:00
let err = runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-08-12 00:08:50 -04:00
" script name " ,
r # "let s = ""; while(true) { s += "Hello"; }"# ,
)
. expect_err ( " script should fail " ) ;
assert_eq! (
" Uncaught Error: execution terminated " ,
2022-04-13 10:41:39 -04:00
err . downcast ::< JsError > ( ) . unwrap ( ) . exception_message
2020-08-12 00:08:50 -04:00
) ;
assert_eq! ( 0 , callback_invoke_count_first . load ( Ordering ::SeqCst ) ) ;
assert! ( callback_invoke_count_second . load ( Ordering ::SeqCst ) > 0 ) ;
}
2020-09-06 10:50:49 -04:00
#[ test ]
fn es_snapshot ( ) {
#[ derive(Default) ]
struct ModsLoader ;
impl ModuleLoader for ModsLoader {
fn resolve (
& self ,
specifier : & str ,
referrer : & str ,
_is_main : bool ,
2021-11-16 09:02:28 -05:00
) -> Result < ModuleSpecifier , Error > {
2020-09-06 10:50:49 -04:00
assert_eq! ( specifier , " file:///main.js " ) ;
assert_eq! ( referrer , " . " ) ;
2021-02-17 13:47:18 -05:00
let s = crate ::resolve_import ( specifier , referrer ) . unwrap ( ) ;
2020-09-06 10:50:49 -04:00
Ok ( s )
}
fn load (
& self ,
_module_specifier : & ModuleSpecifier ,
_maybe_referrer : Option < ModuleSpecifier > ,
_is_dyn_import : bool ,
) -> Pin < Box < ModuleSourceFuture > > {
unreachable! ( )
}
}
let loader = std ::rc ::Rc ::new ( ModsLoader ::default ( ) ) ;
2020-09-11 09:18:49 -04:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
module_loader : Some ( loader ) ,
will_snapshot : true ,
.. Default ::default ( )
} ) ;
2020-09-06 10:50:49 -04:00
2021-02-17 13:47:18 -05:00
let specifier = crate ::resolve_url ( " file:///main.js " ) . unwrap ( ) ;
2020-09-06 10:50:49 -04:00
let source_code = " Deno.core.print('hello \\ n') " . to_string ( ) ;
let module_id = futures ::executor ::block_on (
2021-09-17 21:44:53 -04:00
runtime . load_main_module ( & specifier , Some ( source_code ) ) ,
2020-09-06 10:50:49 -04:00
)
. unwrap ( ) ;
2021-07-30 07:36:43 -04:00
let _ = runtime . mod_evaluate ( module_id ) ;
2021-05-26 15:07:12 -04:00
futures ::executor ::block_on ( runtime . run_event_loop ( false ) ) . unwrap ( ) ;
2020-09-06 10:50:49 -04:00
2020-09-06 15:44:29 -04:00
let _snapshot = runtime . snapshot ( ) ;
2020-09-06 10:50:49 -04:00
}
2020-09-22 17:30:03 -04:00
#[ test ]
fn test_error_without_stack ( ) {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
// SyntaxError
2021-06-21 19:45:41 -04:00
let result = runtime . execute_script (
2020-09-22 17:30:03 -04:00
" error_without_stack.js " ,
r #"
function main ( ) {
console . log ( " asdf);
}
main ( ) ;
" #,
) ;
let expected_error = r #" Uncaught SyntaxError: Invalid or unexpected token
2022-04-15 10:08:09 -04:00
at error_without_stack . js :3 :15 " #;
2020-09-22 17:30:03 -04:00
assert_eq! ( result . unwrap_err ( ) . to_string ( ) , expected_error ) ;
}
#[ test ]
fn test_error_stack ( ) {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
2021-06-21 19:45:41 -04:00
let result = runtime . execute_script (
2020-09-22 17:30:03 -04:00
" error_stack.js " ,
r #"
function assert ( cond ) {
if ( ! cond ) {
throw Error ( " assert " ) ;
}
}
function main ( ) {
assert ( false ) ;
}
main ( ) ;
" #,
) ;
let expected_error = r #" Error: assert
at assert ( error_stack . js :4 :11 )
2022-10-28 07:20:17 -04:00
at main ( error_stack . js :8 :3 )
at error_stack . js :10 :1 " #;
2020-09-22 17:30:03 -04:00
assert_eq! ( result . unwrap_err ( ) . to_string ( ) , expected_error ) ;
}
#[ test ]
fn test_error_async_stack ( ) {
run_in_task ( | cx | {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-09-22 17:30:03 -04:00
" error_async_stack.js " ,
r #"
( async ( ) = > {
const p = ( async ( ) = > {
await Promise . resolve ( ) . then ( ( ) = > {
throw new Error ( " async " ) ;
} ) ;
} ) ( ) ;
try {
await p ;
} catch ( error ) {
console . log ( error . stack ) ;
throw error ;
}
} ) ( ) ; " #,
)
. unwrap ( ) ;
let expected_error = r #" Error: async
at error_async_stack . js :5 :13
at async error_async_stack . js :4 :5
2022-10-28 07:20:17 -04:00
at async error_async_stack . js :9 :5 " #;
2020-09-22 17:30:03 -04:00
2021-05-26 15:07:12 -04:00
match runtime . poll_event_loop ( cx , false ) {
2020-09-22 17:30:03 -04:00
Poll ::Ready ( Err ( e ) ) = > {
assert_eq! ( e . to_string ( ) , expected_error ) ;
}
_ = > panic! ( ) ,
} ;
} )
}
2020-12-07 18:36:15 -05:00
2022-10-26 11:30:44 -04:00
#[ test ]
fn test_error_context ( ) {
use anyhow ::anyhow ;
#[ op ]
fn op_err_sync ( ) -> Result < ( ) , Error > {
Err ( anyhow! ( " original sync error " ) . context ( " higher-level sync error " ) )
}
#[ op ]
async fn op_err_async ( ) -> Result < ( ) , Error > {
Err ( anyhow! ( " original async error " ) . context ( " higher-level async error " ) )
}
run_in_task ( | cx | {
let ext = Extension ::builder ( )
. ops ( vec! [ op_err_sync ::decl ( ) , op_err_async ::decl ( ) ] )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" test_error_context_sync.js " ,
r #"
let errMessage ;
try {
Deno . core . ops . op_err_sync ( ) ;
} catch ( err ) {
errMessage = err . message ;
}
if ( errMessage ! = = " higher-level sync error: original sync error " ) {
throw new Error ( " unexpected error message from op_err_sync: " + errMessage ) ;
}
" #,
)
. unwrap ( ) ;
let promise = runtime
. execute_script (
" test_error_context_async.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2022-10-26 11:30:44 -04:00
( async ( ) = > {
let errMessage ;
try {
await Deno . core . opAsync ( " op_err_async " ) ;
} catch ( err ) {
errMessage = err . message ;
}
if ( errMessage ! = = " higher-level async error: original async error " ) {
throw new Error ( " unexpected error message from op_err_async: " + errMessage ) ;
}
} ) ( )
" #,
)
. unwrap ( ) ;
match runtime . poll_value ( & promise , cx ) {
Poll ::Ready ( Ok ( _ ) ) = > { }
Poll ::Ready ( Err ( err ) ) = > panic! ( " {:?} " , err ) ,
_ = > panic! ( ) ,
}
} )
}
2021-07-02 04:46:37 -04:00
#[ test ]
fn test_pump_message_loop ( ) {
run_in_task ( | cx | {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
runtime
. execute_script (
" pump_message_loop.js " ,
r #"
function assertEquals ( a , b ) {
if ( a = = = b ) return ;
throw a + " does not equal " + b ;
}
const sab = new SharedArrayBuffer ( 16 ) ;
const i32a = new Int32Array ( sab ) ;
globalThis . resolved = false ;
( function ( ) {
const result = Atomics . waitAsync ( i32a , 0 , 0 ) ;
result . value . then (
( value ) = > { assertEquals ( " ok " , value ) ; globalThis . resolved = true ; } ,
( ) = > { assertUnreachable ( ) ;
} ) ;
} ) ( ) ;
const notify_return_value = Atomics . notify ( i32a , 0 , 1 ) ;
assertEquals ( 1 , notify_return_value ) ;
" #,
)
. unwrap ( ) ;
match runtime . poll_event_loop ( cx , false ) {
Poll ::Ready ( Ok ( ( ) ) ) = > { }
_ = > panic! ( ) ,
} ;
// noop script, will resolve promise from first script
runtime
. execute_script ( " pump_message_loop2.js " , r # "assertEquals(1, 1);"# )
. unwrap ( ) ;
// check that promise from `Atomics.waitAsync` has been resolved
runtime
. execute_script (
" pump_message_loop3.js " ,
r # "assertEquals(globalThis.resolved, true);"# ,
)
. unwrap ( ) ;
} )
}
2020-12-07 18:36:15 -05:00
#[ test ]
fn test_core_js_stack_frame ( ) {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
// Call non-existent op so we get error from `core.js`
let error = runtime
2021-06-21 19:45:41 -04:00
. execute_script (
2020-12-07 18:36:15 -05:00
" core_js_stack_frame.js " ,
2022-08-11 09:56:56 -04:00
" Deno.core.opAsync('non_existent'); " ,
2020-12-07 18:36:15 -05:00
)
. unwrap_err ( ) ;
let error_string = error . to_string ( ) ;
// Test that the script specifier is a URL: `deno:<repo-relative path>`.
2021-07-02 06:18:30 -04:00
assert! ( error_string . contains ( " deno:core/01_core.js " ) ) ;
2020-12-07 18:36:15 -05:00
}
2021-04-12 06:15:04 -04:00
#[ test ]
fn test_v8_platform ( ) {
let options = RuntimeOptions {
2021-07-02 03:32:48 -04:00
v8_platform : Some ( v8 ::new_default_platform ( 0 , false ) . make_shared ( ) ) ,
2021-04-12 06:15:04 -04:00
.. Default ::default ( )
} ;
let mut runtime = JsRuntime ::new ( options ) ;
2021-06-21 19:45:41 -04:00
runtime . execute_script ( " <none> " , " " ) . unwrap ( ) ;
2021-04-12 06:15:04 -04:00
}
2021-10-01 14:25:33 -04:00
2022-08-30 05:01:36 -04:00
#[ ignore ] // TODO(@littledivy): Fast API ops when snapshot is not loaded.
2021-10-01 14:25:33 -04:00
#[ test ]
fn test_is_proxy ( ) {
let mut runtime = JsRuntime ::new ( RuntimeOptions ::default ( ) ) ;
let all_true : v8 ::Global < v8 ::Value > = runtime
. execute_script (
" is_proxy.js " ,
r #"
( function ( ) {
const o = { a : 1 , b : 2 } ;
const p = new Proxy ( o , { } ) ;
2022-08-11 09:56:56 -04:00
return Deno . core . ops . op_is_proxy ( p ) & & ! Deno . core . ops . op_is_proxy ( o ) & & ! Deno . core . ops . op_is_proxy ( 42 ) ;
2021-10-01 14:25:33 -04:00
} ) ( )
" #,
)
. unwrap ( ) ;
let mut scope = runtime . handle_scope ( ) ;
let all_true = v8 ::Local ::< v8 ::Value > ::new ( & mut scope , & all_true ) ;
assert! ( all_true . is_true ( ) ) ;
}
2021-10-05 16:55:51 -04:00
2021-10-24 13:30:55 -04:00
#[ tokio::test ]
async fn test_async_opstate_borrow ( ) {
struct InnerState ( u64 ) ;
2022-03-14 13:44:15 -04:00
#[ op ]
2021-10-24 13:30:55 -04:00
async fn op_async_borrow (
op_state : Rc < RefCell < OpState > > ,
2021-11-16 09:02:28 -05:00
) -> Result < ( ) , Error > {
2021-11-03 09:27:36 -04:00
let n = {
let op_state = op_state . borrow ( ) ;
let inner_state = op_state . borrow ::< InnerState > ( ) ;
inner_state . 0
} ;
2021-10-24 13:30:55 -04:00
// Future must be Poll::Pending on first call
tokio ::time ::sleep ( std ::time ::Duration ::from_millis ( 1 ) ) . await ;
2021-11-03 09:27:36 -04:00
if n ! = 42 {
2021-10-24 13:30:55 -04:00
unreachable! ( ) ;
}
Ok ( ( ) )
}
let extension = Extension ::builder ( )
2022-03-14 13:44:15 -04:00
. ops ( vec! [ op_async_borrow ::decl ( ) ] )
2021-10-24 13:30:55 -04:00
. state ( | state | {
state . put ( InnerState ( 42 ) ) ;
Ok ( ( ) )
} )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" op_async_borrow.js " ,
2022-10-28 07:20:17 -04:00
" Deno.core.initializeAsyncOps(); Deno.core.ops.op_async_borrow() " ,
2021-10-24 13:30:55 -04:00
)
. unwrap ( ) ;
runtime . run_event_loop ( false ) . await . unwrap ( ) ;
}
2021-11-16 14:23:12 -05:00
2022-10-02 16:18:31 -04:00
#[ tokio::test ]
async fn test_sync_op_serialize_object_with_numbers_as_keys ( ) {
#[ op ]
fn op_sync_serialize_object_with_numbers_as_keys (
value : serde_json ::Value ,
) -> Result < ( ) , Error > {
assert_eq! (
value . to_string ( ) ,
r # "{"lines":{"100":{"unit":"m"},"200":{"unit":"cm"}}}"#
) ;
Ok ( ( ) )
}
let extension = Extension ::builder ( )
. ops ( vec! [ op_sync_serialize_object_with_numbers_as_keys ::decl ( ) ] )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" op_sync_serialize_object_with_numbers_as_keys.js " ,
r #"
Deno . core . ops . op_sync_serialize_object_with_numbers_as_keys ( {
lines : {
100 : {
unit : " m "
} ,
200 : {
unit : " cm "
}
}
} )
" #,
)
. unwrap ( ) ;
runtime . run_event_loop ( false ) . await . unwrap ( ) ;
}
#[ tokio::test ]
async fn test_async_op_serialize_object_with_numbers_as_keys ( ) {
#[ op ]
async fn op_async_serialize_object_with_numbers_as_keys (
value : serde_json ::Value ,
) -> Result < ( ) , Error > {
assert_eq! (
value . to_string ( ) ,
r # "{"lines":{"100":{"unit":"m"},"200":{"unit":"cm"}}}"#
) ;
Ok ( ( ) )
}
let extension = Extension ::builder ( )
. ops ( vec! [ op_async_serialize_object_with_numbers_as_keys ::decl ( ) ] )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" op_async_serialize_object_with_numbers_as_keys.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
Deno . core . ops . op_async_serialize_object_with_numbers_as_keys ( {
2022-10-02 16:18:31 -04:00
lines : {
100 : {
unit : " m "
} ,
200 : {
unit : " cm "
}
}
} )
" #,
)
. unwrap ( ) ;
runtime . run_event_loop ( false ) . await . unwrap ( ) ;
}
2021-11-16 14:23:12 -05:00
#[ tokio::test ]
async fn test_set_macrotask_callback_set_next_tick_callback ( ) {
2022-03-14 13:44:15 -04:00
#[ op ]
2022-03-15 19:33:46 -04:00
async fn op_async_sleep ( ) -> Result < ( ) , Error > {
2021-11-16 14:23:12 -05:00
// Future must be Poll::Pending on first call
tokio ::time ::sleep ( std ::time ::Duration ::from_millis ( 1 ) ) . await ;
Ok ( ( ) )
}
let extension = Extension ::builder ( )
2022-03-14 13:44:15 -04:00
. ops ( vec! [ op_async_sleep ::decl ( ) ] )
2021-11-16 14:23:12 -05:00
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" macrotasks_and_nextticks.js " ,
r #"
2022-10-28 07:20:17 -04:00
Deno . core . initializeAsyncOps ( ) ;
2021-11-16 14:23:12 -05:00
( async function ( ) {
const results = [ ] ;
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_macrotask_callback ( ( ) = > {
2021-11-16 14:23:12 -05:00
results . push ( " macrotask " ) ;
return true ;
} ) ;
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_next_tick_callback ( ( ) = > {
2021-11-16 14:23:12 -05:00
results . push ( " nextTick " ) ;
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_has_tick_scheduled ( false ) ;
2021-11-16 14:23:12 -05:00
} ) ;
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_has_tick_scheduled ( true ) ;
2021-11-16 14:23:12 -05:00
await Deno . core . opAsync ( ' op_async_sleep ' ) ;
if ( results [ 0 ] ! = " nextTick " ) {
throw new Error ( ` expected nextTick , got : $ { results [ 0 ] } ` ) ;
}
if ( results [ 1 ] ! = " macrotask " ) {
throw new Error ( ` expected macrotask , got : $ { results [ 1 ] } ` ) ;
}
} ) ( ) ;
" #,
)
. unwrap ( ) ;
runtime . run_event_loop ( false ) . await . unwrap ( ) ;
}
#[ tokio::test ]
async fn test_set_macrotask_callback_set_next_tick_callback_multiple ( ) {
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
runtime
. execute_script (
" multiple_macrotasks_and_nextticks.js " ,
r #"
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_macrotask_callback ( ( ) = > { return true ; } ) ;
Deno . core . ops . op_set_macrotask_callback ( ( ) = > { return true ; } ) ;
Deno . core . ops . op_set_next_tick_callback ( ( ) = > { } ) ;
Deno . core . ops . op_set_next_tick_callback ( ( ) = > { } ) ;
2021-11-16 14:23:12 -05:00
" #,
)
. unwrap ( ) ;
let isolate = runtime . v8_isolate ( ) ;
let state_rc = JsRuntime ::state ( isolate ) ;
let state = state_rc . borrow ( ) ;
assert_eq! ( state . js_macrotask_cbs . len ( ) , 2 ) ;
assert_eq! ( state . js_nexttick_cbs . len ( ) , 2 ) ;
}
2021-11-19 08:01:30 -05:00
2021-11-29 18:31:12 -05:00
#[ test ]
fn test_has_tick_scheduled ( ) {
use futures ::task ::ArcWake ;
2021-11-19 08:01:30 -05:00
2022-03-14 13:44:15 -04:00
static MACROTASK : AtomicUsize = AtomicUsize ::new ( 0 ) ;
static NEXT_TICK : AtomicUsize = AtomicUsize ::new ( 0 ) ;
2021-11-19 08:01:30 -05:00
2022-03-14 13:44:15 -04:00
#[ op ]
2022-03-15 19:33:46 -04:00
fn op_macrotask ( ) -> Result < ( ) , AnyError > {
2022-03-14 13:44:15 -04:00
MACROTASK . fetch_add ( 1 , Ordering ::Relaxed ) ;
2021-11-29 18:31:12 -05:00
Ok ( ( ) )
2022-03-14 13:44:15 -04:00
}
2021-11-19 08:01:30 -05:00
2022-03-14 13:44:15 -04:00
#[ op ]
2022-03-15 19:33:46 -04:00
fn op_next_tick ( ) -> Result < ( ) , AnyError > {
2022-03-14 13:44:15 -04:00
NEXT_TICK . fetch_add ( 1 , Ordering ::Relaxed ) ;
2021-11-29 18:31:12 -05:00
Ok ( ( ) )
2022-03-14 13:44:15 -04:00
}
2021-11-19 08:01:30 -05:00
2021-11-29 18:31:12 -05:00
let extension = Extension ::builder ( )
2022-03-14 13:44:15 -04:00
. ops ( vec! [ op_macrotask ::decl ( ) , op_next_tick ::decl ( ) ] )
2021-11-29 18:31:12 -05:00
. build ( ) ;
2021-11-19 08:01:30 -05:00
2021-11-29 18:31:12 -05:00
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" has_tick_scheduled.js " ,
r #"
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_macrotask_callback ( ( ) = > {
Deno . core . ops . op_macrotask ( ) ;
2021-11-19 08:01:30 -05:00
return true ; // We're done.
} ) ;
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_next_tick_callback ( ( ) = > Deno . core . ops . op_next_tick ( ) ) ;
Deno . core . ops . op_set_has_tick_scheduled ( true ) ;
2021-11-19 08:01:30 -05:00
" #,
2021-11-29 18:31:12 -05:00
)
. unwrap ( ) ;
struct ArcWakeImpl ( Arc < AtomicUsize > ) ;
impl ArcWake for ArcWakeImpl {
fn wake_by_ref ( arc_self : & Arc < Self > ) {
arc_self . 0. fetch_add ( 1 , Ordering ::Relaxed ) ;
}
}
let awoken_times = Arc ::new ( AtomicUsize ::new ( 0 ) ) ;
let waker =
futures ::task ::waker ( Arc ::new ( ArcWakeImpl ( awoken_times . clone ( ) ) ) ) ;
let cx = & mut Context ::from_waker ( & waker ) ;
assert! ( matches! ( runtime . poll_event_loop ( cx , false ) , Poll ::Pending ) ) ;
2022-03-14 13:44:15 -04:00
assert_eq! ( 1 , MACROTASK . load ( Ordering ::Relaxed ) ) ;
assert_eq! ( 1 , NEXT_TICK . load ( Ordering ::Relaxed ) ) ;
2021-11-29 18:31:12 -05:00
assert_eq! ( awoken_times . swap ( 0 , Ordering ::Relaxed ) , 1 ) ;
assert! ( matches! ( runtime . poll_event_loop ( cx , false ) , Poll ::Pending ) ) ;
assert_eq! ( awoken_times . swap ( 0 , Ordering ::Relaxed ) , 1 ) ;
assert! ( matches! ( runtime . poll_event_loop ( cx , false ) , Poll ::Pending ) ) ;
assert_eq! ( awoken_times . swap ( 0 , Ordering ::Relaxed ) , 1 ) ;
assert! ( matches! ( runtime . poll_event_loop ( cx , false ) , Poll ::Pending ) ) ;
assert_eq! ( awoken_times . swap ( 0 , Ordering ::Relaxed ) , 1 ) ;
let state_rc = JsRuntime ::state ( runtime . v8_isolate ( ) ) ;
state_rc . borrow_mut ( ) . has_tick_scheduled = false ;
assert! ( matches! (
runtime . poll_event_loop ( cx , false ) ,
Poll ::Ready ( Ok ( ( ) ) )
) ) ;
assert_eq! ( awoken_times . load ( Ordering ::Relaxed ) , 0 ) ;
assert! ( matches! (
runtime . poll_event_loop ( cx , false ) ,
Poll ::Ready ( Ok ( ( ) ) )
) ) ;
assert_eq! ( awoken_times . load ( Ordering ::Relaxed ) , 0 ) ;
2021-11-19 08:01:30 -05:00
}
2021-11-22 07:51:20 -05:00
#[ test ]
fn terminate_during_module_eval ( ) {
#[ derive(Default) ]
struct ModsLoader ;
impl ModuleLoader for ModsLoader {
fn resolve (
& self ,
specifier : & str ,
referrer : & str ,
_is_main : bool ,
) -> Result < ModuleSpecifier , Error > {
assert_eq! ( specifier , " file:///main.js " ) ;
assert_eq! ( referrer , " . " ) ;
let s = crate ::resolve_import ( specifier , referrer ) . unwrap ( ) ;
Ok ( s )
}
fn load (
& self ,
_module_specifier : & ModuleSpecifier ,
_maybe_referrer : Option < ModuleSpecifier > ,
_is_dyn_import : bool ,
) -> Pin < Box < ModuleSourceFuture > > {
async move {
Ok ( ModuleSource {
2022-05-05 07:16:25 -04:00
code : b " console.log('hello world'); " . to_vec ( ) . into_boxed_slice ( ) ,
2021-11-22 07:51:20 -05:00
module_url_specified : " file:///main.js " . to_string ( ) ,
module_url_found : " file:///main.js " . to_string ( ) ,
2021-12-15 13:22:36 -05:00
module_type : ModuleType ::JavaScript ,
2021-11-22 07:51:20 -05:00
} )
}
. boxed_local ( )
}
}
let loader = std ::rc ::Rc ::new ( ModsLoader ::default ( ) ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
module_loader : Some ( loader ) ,
.. Default ::default ( )
} ) ;
let specifier = crate ::resolve_url ( " file:///main.js " ) . unwrap ( ) ;
let source_code = " Deno.core.print('hello \\ n') " . to_string ( ) ;
let module_id = futures ::executor ::block_on (
runtime . load_main_module ( & specifier , Some ( source_code ) ) ,
)
. unwrap ( ) ;
runtime . v8_isolate ( ) . terminate_execution ( ) ;
let mod_result =
futures ::executor ::block_on ( runtime . mod_evaluate ( module_id ) ) . unwrap ( ) ;
assert! ( mod_result
. unwrap_err ( )
. to_string ( )
. contains ( " JavaScript execution has been terminated " ) ) ;
}
2021-11-27 18:46:12 -05:00
#[ tokio::test ]
async fn test_set_promise_reject_callback ( ) {
2022-03-14 13:44:15 -04:00
static PROMISE_REJECT : AtomicUsize = AtomicUsize ::new ( 0 ) ;
2021-11-27 18:46:12 -05:00
2022-03-14 13:44:15 -04:00
#[ op ]
2022-03-15 19:33:46 -04:00
fn op_promise_reject ( ) -> Result < ( ) , AnyError > {
2022-03-14 13:44:15 -04:00
PROMISE_REJECT . fetch_add ( 1 , Ordering ::Relaxed ) ;
2021-11-27 18:46:12 -05:00
Ok ( ( ) )
2022-03-14 13:44:15 -04:00
}
2021-11-27 18:46:12 -05:00
let extension = Extension ::builder ( )
2022-07-24 19:10:56 -04:00
. ops ( vec! [ op_promise_reject ::decl ( ) ] )
2021-11-27 18:46:12 -05:00
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" promise_reject_callback.js " ,
r #"
// Note: |promise| is not the promise created below, it's a child.
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_promise_reject_callback ( ( type , promise , reason ) = > {
2021-11-27 18:46:12 -05:00
if ( type ! = = /* PromiseRejectWithNoHandler */ 0 ) {
throw Error ( " unexpected type: " + type ) ;
}
if ( reason . message ! = = " reject " ) {
throw Error ( " unexpected reason: " + reason ) ;
}
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_store_pending_promise_exception ( promise ) ;
Deno . core . ops . op_promise_reject ( ) ;
2021-11-27 18:46:12 -05:00
} ) ;
new Promise ( ( _ , reject ) = > reject ( Error ( " reject " ) ) ) ;
" #,
)
. unwrap ( ) ;
2022-07-24 19:10:56 -04:00
runtime . run_event_loop ( false ) . await . unwrap_err ( ) ;
2021-11-27 18:46:12 -05:00
2022-03-14 13:44:15 -04:00
assert_eq! ( 1 , PROMISE_REJECT . load ( Ordering ::Relaxed ) ) ;
2021-11-27 18:46:12 -05:00
runtime
. execute_script (
" promise_reject_callback.js " ,
r #"
{
2022-08-11 09:56:56 -04:00
const prev = Deno . core . ops . op_set_promise_reject_callback ( ( .. . args ) = > {
2021-11-27 18:46:12 -05:00
prev ( .. . args ) ;
} ) ;
}
new Promise ( ( _ , reject ) = > reject ( Error ( " reject " ) ) ) ;
" #,
)
. unwrap ( ) ;
2022-07-24 19:10:56 -04:00
runtime . run_event_loop ( false ) . await . unwrap_err ( ) ;
2021-11-27 18:46:12 -05:00
2022-03-14 13:44:15 -04:00
assert_eq! ( 2 , PROMISE_REJECT . load ( Ordering ::Relaxed ) ) ;
2021-11-27 18:46:12 -05:00
}
2022-03-19 10:59:44 -04:00
2022-07-18 15:20:38 -04:00
#[ tokio::test ]
async fn test_set_promise_reject_callback_top_level_await ( ) {
static PROMISE_REJECT : AtomicUsize = AtomicUsize ::new ( 0 ) ;
#[ op ]
fn op_promise_reject ( ) -> Result < ( ) , AnyError > {
PROMISE_REJECT . fetch_add ( 1 , Ordering ::Relaxed ) ;
Ok ( ( ) )
}
let extension = Extension ::builder ( )
2022-07-24 19:10:56 -04:00
. ops ( vec! [ op_promise_reject ::decl ( ) ] )
2022-07-18 15:20:38 -04:00
. build ( ) ;
#[ derive(Default) ]
struct ModsLoader ;
impl ModuleLoader for ModsLoader {
fn resolve (
& self ,
specifier : & str ,
referrer : & str ,
_is_main : bool ,
) -> Result < ModuleSpecifier , Error > {
assert_eq! ( specifier , " file:///main.js " ) ;
assert_eq! ( referrer , " . " ) ;
let s = crate ::resolve_import ( specifier , referrer ) . unwrap ( ) ;
Ok ( s )
}
fn load (
& self ,
_module_specifier : & ModuleSpecifier ,
_maybe_referrer : Option < ModuleSpecifier > ,
_is_dyn_import : bool ,
) -> Pin < Box < ModuleSourceFuture > > {
let source = r #"
2022-08-11 09:56:56 -04:00
Deno . core . ops . op_set_promise_reject_callback ( ( type , promise , reason ) = > {
Deno . core . ops . op_promise_reject ( ) ;
2022-07-18 15:20:38 -04:00
} ) ;
throw new Error ( ' top level throw ' ) ;
" #;
async move {
Ok ( ModuleSource {
code : source . as_bytes ( ) . to_vec ( ) . into_boxed_slice ( ) ,
module_url_specified : " file:///main.js " . to_string ( ) ,
module_url_found : " file:///main.js " . to_string ( ) ,
module_type : ModuleType ::JavaScript ,
} )
}
. boxed_local ( )
}
}
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ extension ] ,
module_loader : Some ( Rc ::new ( ModsLoader ) ) ,
.. Default ::default ( )
} ) ;
let id = runtime
. load_main_module ( & crate ::resolve_url ( " file:///main.js " ) . unwrap ( ) , None )
. await
. unwrap ( ) ;
let receiver = runtime . mod_evaluate ( id ) ;
runtime . run_event_loop ( false ) . await . unwrap ( ) ;
2022-07-24 19:10:56 -04:00
receiver . await . unwrap ( ) . unwrap_err ( ) ;
2022-07-18 15:20:38 -04:00
assert_eq! ( 1 , PROMISE_REJECT . load ( Ordering ::Relaxed ) ) ;
}
2022-03-19 10:59:44 -04:00
#[ test ]
fn test_op_return_serde_v8_error ( ) {
#[ op ]
fn op_err ( ) -> Result < std ::collections ::BTreeMap < u64 , u64 > , anyhow ::Error > {
Ok ( [ ( 1 , 2 ) , ( 3 , 4 ) ] . into_iter ( ) . collect ( ) ) // Maps can't have non-string keys in serde_v8
}
let ext = Extension ::builder ( ) . ops ( vec! [ op_err ::decl ( ) ] ) . build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
assert! ( runtime
. execute_script (
" test_op_return_serde_v8_error.js " ,
2022-08-11 09:56:56 -04:00
" Deno.core.ops.op_err() "
2022-03-19 10:59:44 -04:00
)
. is_err ( ) ) ;
}
2022-03-21 13:04:57 -04:00
#[ test ]
fn test_op_high_arity ( ) {
#[ op ]
fn op_add_4 (
x1 : i64 ,
x2 : i64 ,
x3 : i64 ,
x4 : i64 ,
) -> Result < i64 , anyhow ::Error > {
Ok ( x1 + x2 + x3 + x4 )
}
let ext = Extension ::builder ( ) . ops ( vec! [ op_add_4 ::decl ( ) ] ) . build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
let r = runtime
2022-08-11 09:56:56 -04:00
. execute_script ( " test.js " , " Deno.core.ops.op_add_4(1, 2, 3, 4) " )
2022-03-21 13:04:57 -04:00
. unwrap ( ) ;
let scope = & mut runtime . handle_scope ( ) ;
assert_eq! ( r . open ( scope ) . integer_value ( scope ) , Some ( 10 ) ) ;
}
2022-03-22 11:39:58 -04:00
#[ test ]
fn test_op_disabled ( ) {
#[ op ]
fn op_foo ( ) -> Result < i64 , anyhow ::Error > {
Ok ( 42 )
}
let ext = Extension ::builder ( )
. ops ( vec! [ op_foo ::decl ( ) . disable ( ) ] )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
let r = runtime
2022-08-11 09:56:56 -04:00
. execute_script ( " test.js " , " Deno.core.ops.op_foo() " )
2022-03-22 11:39:58 -04:00
. unwrap ( ) ;
let scope = & mut runtime . handle_scope ( ) ;
assert! ( r . open ( scope ) . is_undefined ( ) ) ;
}
2022-04-01 18:09:21 -04:00
2022-04-02 07:35:57 -04:00
#[ test ]
fn test_op_detached_buffer ( ) {
use serde_v8 ::DetachedBuffer ;
#[ op ]
fn op_sum_take ( b : DetachedBuffer ) -> Result < u64 , anyhow ::Error > {
Ok ( b . as_ref ( ) . iter ( ) . clone ( ) . map ( | x | * x as u64 ) . sum ( ) )
}
#[ op ]
fn op_boomerang (
b : DetachedBuffer ,
) -> Result < DetachedBuffer , anyhow ::Error > {
Ok ( b )
}
let ext = Extension ::builder ( )
. ops ( vec! [ op_sum_take ::decl ( ) , op_boomerang ::decl ( ) ] )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" test.js " ,
r #"
const a1 = new Uint8Array ( [ 1 , 2 , 3 ] ) ;
const a1b = a1 . subarray ( 0 , 3 ) ;
const a2 = new Uint8Array ( [ 5 , 10 , 15 ] ) ;
const a2b = a2 . subarray ( 0 , 3 ) ;
if ( ! ( a1 . length > 0 & & a1b . length > 0 ) ) {
throw new Error ( " a1 & a1b should have a length " ) ;
}
2022-08-11 09:56:56 -04:00
let sum = Deno . core . ops . op_sum_take ( a1b ) ;
2022-04-02 07:35:57 -04:00
if ( sum ! = = 6 ) {
throw new Error ( ` Bad sum : $ { sum } ` ) ;
}
if ( a1 . length > 0 | | a1b . length > 0 ) {
throw new Error ( " expecting a1 & a1b to be detached " ) ;
}
2022-08-11 09:56:56 -04:00
const a3 = Deno . core . ops . op_boomerang ( a2b ) ;
2022-04-02 07:35:57 -04:00
if ( a3 . byteLength ! = 3 ) {
throw new Error ( ` Expected a3 . byteLength = = = 3 , got $ { a3 . byteLength } ` ) ;
}
if ( a3 [ 0 ] ! = = 5 | | a3 [ 1 ] ! = = 10 ) {
throw new Error ( ` Invalid a3 : $ { a3 [ 0 ] } , $ { a3 [ 1 ] } ` ) ;
}
if ( a2 . byteLength > 0 | | a2b . byteLength > 0 ) {
throw new Error ( " expecting a2 & a2b to be detached, a3 re-attached " ) ;
}
const wmem = new WebAssembly . Memory ( { initial : 1 , maximum : 2 } ) ;
const w32 = new Uint32Array ( wmem . buffer ) ;
w32 [ 0 ] = 1 ; w32 [ 1 ] = 2 ; w32 [ 2 ] = 3 ;
const assertWasmThrow = ( ( ) = > {
try {
2022-08-11 09:56:56 -04:00
let sum = Deno . core . ops . op_sum_take ( w32 . subarray ( 0 , 2 ) ) ;
2022-04-02 07:35:57 -04:00
return false ;
} catch ( e ) {
return e . message . includes ( ' ExpectedDetachable ' ) ;
}
} ) ;
if ( ! assertWasmThrow ( ) ) {
throw new Error ( " expected wasm mem to not be detachable " ) ;
}
" #,
)
. unwrap ( ) ;
}
2022-04-01 18:09:21 -04:00
#[ test ]
fn test_op_unstable_disabling ( ) {
#[ op ]
fn op_foo ( ) -> Result < i64 , anyhow ::Error > {
Ok ( 42 )
}
#[ op(unstable) ]
fn op_bar ( ) -> Result < i64 , anyhow ::Error > {
Ok ( 42 )
}
let ext = Extension ::builder ( )
. ops ( vec! [ op_foo ::decl ( ) , op_bar ::decl ( ) ] )
. middleware ( | op | if op . is_unstable { op . disable ( ) } else { op } )
. build ( ) ;
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ ext ] ,
.. Default ::default ( )
} ) ;
runtime
. execute_script (
" test.js " ,
r #"
2022-08-11 09:56:56 -04:00
if ( Deno . core . ops . op_foo ( ) ! = = 42 ) {
2022-04-01 18:09:21 -04:00
throw new Error ( " Exptected op_foo() === 42 " ) ;
}
2022-08-11 09:56:56 -04:00
if ( Deno . core . ops . op_bar ( ) ! = = undefined ) {
2022-04-01 18:09:21 -04:00
throw new Error ( " Expected op_bar to be disabled " )
}
" #,
)
. unwrap ( ) ;
}
2022-04-17 07:53:08 -04:00
#[ test ]
fn js_realm_simple ( ) {
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
let main_context = runtime . global_context ( ) ;
let main_global = {
let scope = & mut runtime . handle_scope ( ) ;
let local_global = main_context . open ( scope ) . global ( scope ) ;
v8 ::Global ::new ( scope , local_global )
} ;
let realm = runtime . create_realm ( ) . unwrap ( ) ;
assert_ne! ( realm . context ( ) , & main_context ) ;
2022-07-05 18:45:10 -04:00
assert_ne! ( realm . global_object ( runtime . v8_isolate ( ) ) , main_global ) ;
2022-04-17 07:53:08 -04:00
let main_object = runtime . execute_script ( " " , " Object " ) . unwrap ( ) ;
2022-07-05 18:45:10 -04:00
let realm_object = realm
. execute_script ( runtime . v8_isolate ( ) , " " , " Object " )
. unwrap ( ) ;
2022-04-17 07:53:08 -04:00
assert_ne! ( main_object , realm_object ) ;
}
#[ test ]
fn js_realm_init ( ) {
#[ op ]
fn op_test ( ) -> Result < String , Error > {
Ok ( String ::from ( " Test " ) )
}
let mut runtime = JsRuntime ::new ( RuntimeOptions {
extensions : vec ! [ Extension ::builder ( ) . ops ( vec! [ op_test ::decl ( ) ] ) . build ( ) ] ,
.. Default ::default ( )
} ) ;
let realm = runtime . create_realm ( ) . unwrap ( ) ;
let ret = realm
2022-08-11 09:56:56 -04:00
. execute_script ( runtime . v8_isolate ( ) , " " , " Deno.core.ops.op_test() " )
2022-04-17 07:53:08 -04:00
. unwrap ( ) ;
2022-07-05 18:45:10 -04:00
let scope = & mut realm . handle_scope ( runtime . v8_isolate ( ) ) ;
2022-04-17 07:53:08 -04:00
assert_eq! ( ret , serde_v8 ::to_v8 ( scope , " Test " ) . unwrap ( ) ) ;
}
#[ test ]
fn js_realm_init_snapshot ( ) {
let snapshot = {
2022-10-14 19:53:13 -04:00
let runtime = JsRuntime ::new ( RuntimeOptions {
2022-04-17 07:53:08 -04:00
will_snapshot : true ,
.. Default ::default ( )
} ) ;
2022-11-17 20:59:10 -05:00
let snap : & [ u8 ] = & runtime . snapshot ( ) ;
2022-04-17 07:53:08 -04:00
Vec ::from ( snap ) . into_boxed_slice ( )
} ;
#[ op ]
fn op_test ( ) -> Result < String , Error > {
Ok ( String ::from ( " Test " ) )
}
let mut runtime = JsRuntime ::new ( RuntimeOptions {
startup_snapshot : Some ( Snapshot ::Boxed ( snapshot ) ) ,
extensions : vec ! [ Extension ::builder ( ) . ops ( vec! [ op_test ::decl ( ) ] ) . build ( ) ] ,
.. Default ::default ( )
} ) ;
let realm = runtime . create_realm ( ) . unwrap ( ) ;
let ret = realm
2022-08-11 09:56:56 -04:00
. execute_script ( runtime . v8_isolate ( ) , " " , " Deno.core.ops.op_test() " )
2022-04-17 07:53:08 -04:00
. unwrap ( ) ;
2022-07-05 18:45:10 -04:00
let scope = & mut realm . handle_scope ( runtime . v8_isolate ( ) ) ;
2022-04-17 07:53:08 -04:00
assert_eq! ( ret , serde_v8 ::to_v8 ( scope , " Test " ) . unwrap ( ) ) ;
}
2022-10-26 10:54:22 -04:00
#[ test ]
fn test_array_by_copy ( ) {
// Verify that "array by copy" proposal is enabled (https://github.com/tc39/proposal-change-array-by-copy)
let mut runtime = JsRuntime ::new ( Default ::default ( ) ) ;
assert! ( runtime
. execute_script (
" test_array_by_copy.js " ,
" const a = [1, 2, 3];
const b = a . toReversed ( ) ;
if ( ! ( a [ 0 ] = = = 1 & & a [ 1 ] = = = 2 & & a [ 2 ] = = = 3 ) ) {
throw new Error ( ' Expected a to be intact ' ) ;
}
if ( ! ( b [ 0 ] = = = 3 & & b [ 1 ] = = = 2 & & b [ 2 ] = = = 1 ) ) {
throw new Error ( ' Expected b to be reversed ' ) ;
} " ,
)
. is_ok ( ) ) ;
}
2019-03-11 17:57:36 -04:00
}