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
|
#include <mruby.h>
#include <mruby/string.h>
#include <mruby/variable.h>
#include <stdio.h>
#include <stdlib.h>
static mrb_value
fiber_s_yield_by_c_func(mrb_state *mrb, mrb_value self)
{
mrb_value a = mrb_get_arg1(mrb);
return mrb_fiber_yield(mrb, 1, &a);
}
static mrb_value
fiber_s_yield_by_c_method(mrb_state *mrb, mrb_value self)
{
mrb_value a = mrb_get_arg1(mrb);
return mrb_funcall_argv(mrb, self, mrb_intern_lit(mrb, "yield"), 1, &a);
}
static mrb_value
fiber_resume_by_c_func(mrb_state *mrb, mrb_value self)
{
int ci_index = mrb->c->ci - mrb->c->cibase;
mrb_value ret = mrb_fiber_resume(mrb, self, 0, NULL);
if (ci_index != mrb->c->ci - mrb->c->cibase) {
mrb_raisef(mrb, E_EXCEPTION,
"[BUG] INVALID CI POSITION (expected %d, but actual %d) [BUG]",
(int)ci_index, (int)(mrb->c->ci - mrb->c->cibase));
}
return ret;
}
static mrb_value
fiber_resume_by_c_method(mrb_state *mrb, mrb_value self)
{
int ci_index = mrb->c->ci - mrb->c->cibase;
mrb_value ret = mrb_funcall_argv(mrb, self, mrb_intern_lit(mrb, "resume"), 0, NULL);
if (ci_index != mrb->c->ci - mrb->c->cibase) {
mrb_raisef(mrb, E_EXCEPTION,
"[BUG] INVALID CI POSITION (expected %d, but actual %d) [BUG]",
(int)ci_index, (int)(mrb->c->ci - mrb->c->cibase));
}
return ret;
}
static mrb_value
fiber_transfer_by_c(mrb_state *mrb, mrb_value self)
{
return mrb_funcall_argv(mrb, self, mrb_intern_lit(mrb, "transfer"), 0, NULL);
}
static mrb_value
proc_s_c_tunnel(mrb_state *mrb, mrb_value self)
{
mrb_value b;
mrb_get_args(mrb, "&!", &b);
return mrb_yield_argv(mrb, b, 0, NULL);
}
static void
check_activity(mrb_state *mrb)
{
mrb_value act = mrb_gv_get(mrb, mrb_intern_lit(mrb, "$fiber_test_activity"));
if (mrb_test(act)) {
act = mrb_obj_as_string(mrb, act);
fprintf(stderr, "\n\t<<<%s%.*s>>>\n",
"mruby VM has an unexpected outage in ", (int)RSTRING_LEN(act), RSTRING_PTR(act));
abort();
}
}
void
mrb_mruby_fiber_gem_test(mrb_state *mrb)
{
struct RClass *fiber_class = mrb_class_get(mrb, "Fiber");
mrb_define_class_method(mrb, fiber_class, "yield_by_c_func", fiber_s_yield_by_c_func, MRB_ARGS_ANY());
mrb_define_class_method(mrb, fiber_class, "yield_by_c_method", fiber_s_yield_by_c_method, MRB_ARGS_ANY());
mrb_define_method(mrb, fiber_class, "resume_by_c_func", fiber_resume_by_c_func, MRB_ARGS_NONE());
mrb_define_method(mrb, fiber_class, "resume_by_c_method", fiber_resume_by_c_method, MRB_ARGS_NONE());
mrb_define_method(mrb, fiber_class, "transfer_by_c", fiber_transfer_by_c, MRB_ARGS_NONE());
mrb_define_class_method(mrb, mrb->proc_class, "c_tunnel", proc_s_c_tunnel, MRB_ARGS_NONE() | MRB_ARGS_BLOCK());
mrb_gv_set(mrb, mrb_intern_lit(mrb, "$fiber_test_activity"), mrb_nil_value());
mrb_state_atexit(mrb, check_activity);
}
|
