1 | #include <mruby.h>
|
---|
2 | #include <mruby/throw.h>
|
---|
3 | #include <mruby/error.h>
|
---|
4 |
|
---|
5 | MRB_API mrb_value
|
---|
6 | mrb_protect(mrb_state *mrb, mrb_func_t body, mrb_value data, mrb_bool *state)
|
---|
7 | {
|
---|
8 | struct mrb_jmpbuf *prev_jmp = mrb->jmp;
|
---|
9 | struct mrb_jmpbuf c_jmp;
|
---|
10 | mrb_value result = mrb_nil_value();
|
---|
11 | int ai = mrb_gc_arena_save(mrb);
|
---|
12 |
|
---|
13 | if (state) { *state = FALSE; }
|
---|
14 |
|
---|
15 | MRB_TRY(&c_jmp) {
|
---|
16 | mrb->jmp = &c_jmp;
|
---|
17 | result = body(mrb, data);
|
---|
18 | mrb->jmp = prev_jmp;
|
---|
19 | } MRB_CATCH(&c_jmp) {
|
---|
20 | mrb->jmp = prev_jmp;
|
---|
21 | result = mrb_obj_value(mrb->exc);
|
---|
22 | mrb->exc = NULL;
|
---|
23 | if (state) { *state = TRUE; }
|
---|
24 | } MRB_END_EXC(&c_jmp);
|
---|
25 |
|
---|
26 | mrb_gc_arena_restore(mrb, ai);
|
---|
27 | mrb_gc_protect(mrb, result);
|
---|
28 | return result;
|
---|
29 | }
|
---|
30 |
|
---|
31 | MRB_API mrb_value
|
---|
32 | mrb_ensure(mrb_state *mrb, mrb_func_t body, mrb_value b_data, mrb_func_t ensure, mrb_value e_data)
|
---|
33 | {
|
---|
34 | struct mrb_jmpbuf *prev_jmp = mrb->jmp;
|
---|
35 | struct mrb_jmpbuf c_jmp;
|
---|
36 | mrb_value result;
|
---|
37 | int ai = mrb_gc_arena_save(mrb);
|
---|
38 |
|
---|
39 | MRB_TRY(&c_jmp) {
|
---|
40 | mrb->jmp = &c_jmp;
|
---|
41 | result = body(mrb, b_data);
|
---|
42 | mrb->jmp = prev_jmp;
|
---|
43 | } MRB_CATCH(&c_jmp) {
|
---|
44 | mrb->jmp = prev_jmp;
|
---|
45 | mrb_gc_arena_restore(mrb, ai);
|
---|
46 | ensure(mrb, e_data);
|
---|
47 | MRB_THROW(mrb->jmp); /* rethrow catched exceptions */
|
---|
48 | } MRB_END_EXC(&c_jmp);
|
---|
49 |
|
---|
50 | mrb_gc_arena_restore(mrb, ai);
|
---|
51 | mrb_gc_protect(mrb, result);
|
---|
52 | ensure(mrb, e_data);
|
---|
53 | mrb_gc_arena_restore(mrb, ai);
|
---|
54 | mrb_gc_protect(mrb, result);
|
---|
55 | return result;
|
---|
56 | }
|
---|
57 |
|
---|
58 | MRB_API mrb_value
|
---|
59 | mrb_rescue(mrb_state *mrb, mrb_func_t body, mrb_value b_data,
|
---|
60 | mrb_func_t rescue, mrb_value r_data)
|
---|
61 | {
|
---|
62 | return mrb_rescue_exceptions(mrb, body, b_data, rescue, r_data, 1, &mrb->eStandardError_class);
|
---|
63 | }
|
---|
64 |
|
---|
65 | MRB_API mrb_value
|
---|
66 | mrb_rescue_exceptions(mrb_state *mrb, mrb_func_t body, mrb_value b_data, mrb_func_t rescue, mrb_value r_data,
|
---|
67 | mrb_int len, struct RClass **classes)
|
---|
68 | {
|
---|
69 | struct mrb_jmpbuf *prev_jmp = mrb->jmp;
|
---|
70 | struct mrb_jmpbuf c_jmp;
|
---|
71 | mrb_value result;
|
---|
72 | mrb_bool error_matched = FALSE;
|
---|
73 | mrb_int i;
|
---|
74 | int ai = mrb_gc_arena_save(mrb);
|
---|
75 |
|
---|
76 | MRB_TRY(&c_jmp) {
|
---|
77 | mrb->jmp = &c_jmp;
|
---|
78 | result = body(mrb, b_data);
|
---|
79 | mrb->jmp = prev_jmp;
|
---|
80 | } MRB_CATCH(&c_jmp) {
|
---|
81 | mrb->jmp = prev_jmp;
|
---|
82 |
|
---|
83 | for (i = 0; i < len; ++i) {
|
---|
84 | if (mrb_obj_is_kind_of(mrb, mrb_obj_value(mrb->exc), classes[i])) {
|
---|
85 | error_matched = TRUE;
|
---|
86 | break;
|
---|
87 | }
|
---|
88 | }
|
---|
89 |
|
---|
90 | if (!error_matched) { MRB_THROW(mrb->jmp); }
|
---|
91 |
|
---|
92 | mrb->exc = NULL;
|
---|
93 | mrb_gc_arena_restore(mrb, ai);
|
---|
94 | result = rescue(mrb, r_data);
|
---|
95 | } MRB_END_EXC(&c_jmp);
|
---|
96 |
|
---|
97 | mrb_gc_arena_restore(mrb, ai);
|
---|
98 | mrb_gc_protect(mrb, result);
|
---|
99 | return result;
|
---|
100 | }
|
---|
101 |
|
---|
102 | void
|
---|
103 | mrb_mruby_error_gem_init(mrb_state *mrb)
|
---|
104 | {
|
---|
105 | }
|
---|
106 |
|
---|
107 | void
|
---|
108 | mrb_mruby_error_gem_final(mrb_state *mrb)
|
---|
109 | {
|
---|
110 | }
|
---|