mrubyedge 1.1.12

mruby/edge is yet another mruby that is specialized for running on WASM
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

#![cfg(feature = "insn-limit")]
extern crate mec_mrbc_sys;
extern crate mrubyedge;

mod helpers;
use helpers::*;

#[test]
fn insn_limit_basic_test() {
    let code = r#"
    def test_simple
      a = 1
      b = 2
      a + b
    end
    "#;
    let binary = mrbc_compile("insn_limit_basic", code);
    let mut rite = mrubyedge::rite::load(&binary).unwrap();
    let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);

    // Simple function should complete within limit
    let result = vm.run();
    assert!(result.is_ok());

    let args = vec![];
    let result: i32 = mrb_funcall(&mut vm, None, "test_simple", &args)
        .unwrap()
        .as_ref()
        .try_into()
        .unwrap();
    assert_eq!(result, 3);
}

#[test]
fn insn_limit_exceeded_test() {
    let code = r#"
    def test_infinite_loop
      i = 0
      loop do
        i += 1
      end
    end
    "#;
    let binary = mrbc_compile("insn_limit_exceeded", code);
    let mut rite = mrubyedge::rite::load(&binary).unwrap();
    let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);
    vm.run().unwrap();

    let args = vec![];
    let result = mrb_funcall(&mut vm, None, "test_infinite_loop", &args);

    // Should fail due to instruction limit
    assert!(result.is_err());
    let err_msg = format!("{:?}", result.unwrap_err());
    assert!(err_msg.contains("instruction limit exceeded"));
}

#[test]
fn insn_limit_reset_test() {
    let code = r#"
    def test_count
      sum = 0
      10.times do |i|
        sum += i
      end
      sum
    end
    "#;
    let binary = mrbc_compile("insn_limit_reset", code);
    let mut rite = mrubyedge::rite::load(&binary).unwrap();
    let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);
    vm.run().unwrap();

    // First call
    let args = vec![];
    let result: i32 = mrb_funcall(&mut vm, None, "test_count", &args)
        .unwrap()
        .as_ref()
        .try_into()
        .unwrap();
    assert_eq!(result, 45);

    let count_before_reset = vm.get_insn_count();
    assert!(count_before_reset > 0);

    vm.reset_insn_count();
    assert_eq!(vm.get_insn_count(), 0);

    // Second call should work after reset
    let result: i32 = mrb_funcall(&mut vm, None, "test_count", &args)
        .unwrap()
        .as_ref()
        .try_into()
        .unwrap();
    assert_eq!(result, 45);
}

#[test]
fn insn_limit_while_loop_test() {
    let code = r#"
    def test_while
      i = 0
      sum = 0
      while i < 100000
        i += 1
        sum += i
      end
      sum
    end
    "#;
    let binary = mrbc_compile("insn_limit_while", code);
    let mut rite = mrubyedge::rite::load(&binary).unwrap();
    let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);
    vm.run().unwrap();

    let args = vec![];
    let result = mrb_funcall(&mut vm, None, "test_while", &args);

    assert!(result.is_err());
    assert!(format!("{:?}", result.unwrap_err()).contains("instruction limit exceeded"));
}

#[test]
fn insn_limit_counter_increments_test() {
    let code = r#"
    def test_increment
      a = 1
      b = 2
      c = 3
      a + b + c
    end
    "#;
    let binary = mrbc_compile("insn_limit_increment", code);
    let mut rite = mrubyedge::rite::load(&binary).unwrap();
    let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);

    let initial_count = vm.get_insn_count();
    assert_eq!(initial_count, 0);

    vm.run().unwrap();

    let count_after_run = vm.get_insn_count();
    assert!(count_after_run > 0);

    let args = vec![];
    mrb_funcall(&mut vm, None, "test_increment", &args).unwrap();

    let count_after_call = vm.get_insn_count();
    assert!(count_after_call > count_after_run);
}