mirror of https://github.com/boa-dev/boa.git
Haled Odat
2 years ago
committed by
GitHub
37 changed files with 620 additions and 214 deletions
@ -0,0 +1,35 @@
|
||||
use boa_engine::{ |
||||
object::{FunctionObjectBuilder, ObjectInitializer}, |
||||
property::Attribute, |
||||
Context, JsArgs, JsObject, JsResult, JsValue, NativeFunction, |
||||
}; |
||||
|
||||
fn get_loop(_: &JsValue, _: &[JsValue], context: &mut Context<'_>) -> JsResult<JsValue> { |
||||
let max = context.runtime_limits().loop_iteration_limit(); |
||||
Ok(JsValue::from(max)) |
||||
} |
||||
|
||||
fn set_loop(_: &JsValue, args: &[JsValue], context: &mut Context<'_>) -> JsResult<JsValue> { |
||||
let value = args.get_or_undefined(0).to_length(context)?; |
||||
context.runtime_limits_mut().set_loop_iteration_limit(value); |
||||
Ok(JsValue::undefined()) |
||||
} |
||||
|
||||
pub(super) fn create_object(context: &mut Context<'_>) -> JsObject { |
||||
let get_loop = FunctionObjectBuilder::new(context, NativeFunction::from_fn_ptr(get_loop)) |
||||
.name("get loop") |
||||
.length(0) |
||||
.build(); |
||||
let set_loop = FunctionObjectBuilder::new(context, NativeFunction::from_fn_ptr(set_loop)) |
||||
.name("set loop") |
||||
.length(1) |
||||
.build(); |
||||
ObjectInitializer::new(context) |
||||
.accessor( |
||||
"loop", |
||||
Some(get_loop), |
||||
Some(set_loop), |
||||
Attribute::WRITABLE | Attribute::CONFIGURABLE | Attribute::NON_ENUMERABLE, |
||||
) |
||||
.build() |
||||
} |
||||
@ -0,0 +1,61 @@
|
||||
/// Represents the limits of different runtime operations.
|
||||
#[derive(Debug, Clone, Copy)] |
||||
pub struct RuntimeLimits { |
||||
/// Max stack size before an error is thrown.
|
||||
stack_size_limit: usize, |
||||
|
||||
/// Max loop iterations before an error is thrown.
|
||||
loop_iteration_limit: u64, |
||||
} |
||||
|
||||
impl Default for RuntimeLimits { |
||||
#[inline] |
||||
fn default() -> Self { |
||||
Self { |
||||
loop_iteration_limit: u64::MAX, |
||||
stack_size_limit: 1024, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl RuntimeLimits { |
||||
/// Return the loop iteration limit.
|
||||
///
|
||||
/// If the limit is exceeded in a loop it will throw and errror.
|
||||
///
|
||||
/// The limit value [`u64::MAX`] means that there is no limit.
|
||||
#[inline] |
||||
#[must_use] |
||||
pub const fn loop_iteration_limit(&self) -> u64 { |
||||
self.loop_iteration_limit |
||||
} |
||||
|
||||
/// Set the loop iteration limit.
|
||||
///
|
||||
/// If the limit is exceeded in a loop it will throw and errror.
|
||||
///
|
||||
/// Setting the limit to [`u64::MAX`] means that there is no limit.
|
||||
#[inline] |
||||
pub fn set_loop_iteration_limit(&mut self, value: u64) { |
||||
self.loop_iteration_limit = value; |
||||
} |
||||
|
||||
/// Disable loop iteration limit.
|
||||
#[inline] |
||||
pub fn disable_loop_iteration_limit(&mut self) { |
||||
self.loop_iteration_limit = u64::MAX; |
||||
} |
||||
|
||||
/// Get max stack size.
|
||||
#[inline] |
||||
#[must_use] |
||||
pub const fn stack_size_limit(&self) -> usize { |
||||
self.stack_size_limit |
||||
} |
||||
|
||||
/// Set max stack size before an error is thrown.
|
||||
#[inline] |
||||
pub fn set_stack_size_limit(&mut self, value: usize) { |
||||
self.stack_size_limit = value; |
||||
} |
||||
} |
||||
@ -0,0 +1,47 @@
|
||||
use boa_engine::{Context, Source}; |
||||
|
||||
fn main() { |
||||
// Create the JavaScript context.
|
||||
let mut context = Context::default(); |
||||
|
||||
// Set the context's runtime limit on loops to 10 iterations.
|
||||
context.runtime_limits_mut().set_loop_iteration_limit(10); |
||||
|
||||
// The code below iterates 5 times, so no error is thrown.
|
||||
let result = context.eval_script(Source::from_bytes( |
||||
r" |
||||
for (let i = 0; i < 5; ++i) { } |
||||
", |
||||
)); |
||||
result.expect("no error should be thrown"); |
||||
|
||||
// Here we exceed the limit by 1 iteration and a `RuntimeLimit` error is thrown.
|
||||
//
|
||||
// This error cannot be caught in JavaScript it propagates to rust caller.
|
||||
let result = context.eval_script(Source::from_bytes( |
||||
r" |
||||
try { |
||||
for (let i = 0; i < 11; ++i) { } |
||||
} catch (e) { |
||||
|
||||
} |
||||
", |
||||
)); |
||||
result.expect_err("should have throw an error"); |
||||
|
||||
// Preventing an infinity loops
|
||||
let result = context.eval_script(Source::from_bytes( |
||||
r" |
||||
while (true) { } |
||||
", |
||||
)); |
||||
result.expect_err("should have throw an error"); |
||||
|
||||
// The limit applies to all types of loops.
|
||||
let result = context.eval_script(Source::from_bytes( |
||||
r" |
||||
for (let e of [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) { } |
||||
", |
||||
)); |
||||
result.expect_err("should have throw an error"); |
||||
} |
||||
Loading…
Reference in new issue