use std::ptr::{self, NonNull};
use mozjs::{
capture_stack,
jsapi::{self, JSContext, OnNewGlobalHookOption, Value},
jsval::UndefinedValue,
realm::AutoRealm,
rooted,
rust::{
evaluate_script, wrappers2, CompileOptionsWrapper, JSEngine, RealmOptions, Runtime,
SIMPLE_GLOBAL_CLASS,
},
};
#[test]
fn iterate_stack_frames() {
unsafe extern "C" fn assert_stack_state(
context: *mut JSContext,
_argc: u32,
_vp: *mut Value,
) -> bool {
let mut context = mozjs::context::JSContext::from_ptr(NonNull::new(context).unwrap());
let mut function_names = vec![];
capture_stack!(&in(context) let stack);
stack.unwrap().for_each_stack_frame(|frame| {
rooted!(&in(context) let mut result: *mut jsapi::JSString = ptr::null_mut());
unsafe {
wrappers2::GetSavedFrameFunctionDisplayName(
&mut context,
ptr::null_mut(),
frame.into(),
result.handle_mut().into(),
jsapi::SavedFrameSelfHosted::Include,
);
}
let buffer = if !result.is_null() {
let mut buffer = vec![0; 3];
wrappers2::JS_EncodeStringToBuffer(&mut context, *result, buffer.as_mut_ptr(), 3);
Some(buffer.into_iter().map(|c| c as u8).collect())
} else {
None
};
function_names.push(buffer);
});
assert_eq!(function_names.len(), 4);
assert_eq!(function_names[0], Some(b"baz".to_vec()));
assert_eq!(function_names[1], Some(b"bar".to_vec()));
assert_eq!(function_names[2], Some(b"foo".to_vec()));
assert_eq!(function_names[3], None);
true
}
let engine = JSEngine::init().unwrap();
let mut runtime = Runtime::new(engine.handle());
let context = runtime.cx();
#[cfg(feature = "debugmozjs")]
unsafe {
mozjs::jsapi::SetGCZeal(context.raw_cx(), 2, 1);
}
let h_option = OnNewGlobalHookOption::FireOnNewGlobalHook;
let c_option = RealmOptions::default();
unsafe {
rooted!(&in(context) let global = wrappers2::JS_NewGlobalObject(
context,
&SIMPLE_GLOBAL_CLASS,
ptr::null_mut(),
h_option,
&*c_option,
));
let mut realm = AutoRealm::new_from_handle(context, global.handle());
let context = &mut *realm;
let function = wrappers2::JS_DefineFunction(
context,
global.handle().into(),
c"assert_stack_state".as_ptr(),
Some(assert_stack_state),
0,
0,
);
assert!(!function.is_null());
let javascript = "
function foo() {
function bar() {
function baz() {
assert_stack_state();
}
baz();
}
bar();
}
foo();
";
rooted!(&in(context) let mut rval = UndefinedValue());
let options = CompileOptionsWrapper::new(&context, c"test.js".to_owned(), 0);
assert!(evaluate_script(
context,
global.handle(),
javascript,
rval.handle_mut(),
options
)
.is_ok());
}
}