Add progress tracking and operations limit.

tests/operations.rs

@@ -0,0 +1,93 @@
+#![cfg(not(feature = "unchecked"))]
+use rhai::{Engine, EvalAltResult};
+
+#[test]
+fn test_max_operations() -> Result<(), Box<EvalAltResult>> {
+    let mut engine = Engine::new();
+    engine.set_max_operations(500);
+
+    engine.on_progress(|count| {
+        if count % 100 == 0 {
+            println!("{}", count);
+        }
+        true
+    });
+
+    engine.eval::<()>("let x = 0; while x < 20 { x += 1; }")?;
+
+    assert!(matches!(
+        *engine
+            .eval::<()>("for x in range(0, 500) {}")
+            .expect_err("should error"),
+        EvalAltResult::ErrorTooManyOperations(_)
+    ));
+
+    engine.set_max_operations(0);
+
+    engine.eval::<()>("for x in range(0, 10000) {}")?;
+
+    Ok(())
+}
+
+#[test]
+fn test_max_operations_functions() -> Result<(), Box<EvalAltResult>> {
+    let mut engine = Engine::new();
+    engine.set_max_operations(500);
+
+    engine.on_progress(|count| {
+        if count % 100 == 0 {
+            println!("{}", count);
+        }
+        true
+    });
+
+    engine.eval::<()>(
+        r#"
+                fn inc(x) { x + 1 }
+                let x = 0;
+                while x < 20 { x = inc(x); }
+            "#,
+    )?;
+
+    assert!(matches!(
+        *engine
+            .eval::<()>(
+                r#"
+                    fn inc(x) { x + 1 }
+                    let x = 0;
+                    while x < 1000 { x = inc(x); }
+        "#
+            )
+            .expect_err("should error"),
+        EvalAltResult::ErrorTooManyOperations(_)
+    ));
+
+    Ok(())
+}
+
+#[test]
+fn test_max_operations_eval() -> Result<(), Box<EvalAltResult>> {
+    let mut engine = Engine::new();
+    engine.set_max_operations(500);
+
+    engine.on_progress(|count| {
+        if count % 100 == 0 {
+            println!("{}", count);
+        }
+        true
+    });
+
+    assert!(matches!(
+        *engine
+            .eval::<()>(
+                r#"
+                    let script = "for x in range(0, 500) {}";
+                    eval(script);
+            "#
+            )
+            .expect_err("should error"),
+        EvalAltResult::ErrorTooManyOperations(_)
+    ));
+
+    Ok(())
+}

tests/stack.rs

@@ -15,6 +15,7 @@ fn test_stack_overflow() -> Result<(), Box<EvalAltResult>> {
         325
     );
 
+    #[cfg(not(feature = "unchecked"))]
     match engine.eval::<()>(
         r"
             fn foo(n) { if n == 0 { 0 } else { n + foo(n-1) } }