87 lines
2.8 KiB
C
Vendored
87 lines
2.8 KiB
C
Vendored
#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)
|
|
{
|
|
ptrdiff_t 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)
|
|
{
|
|
ptrdiff_t 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);
|
|
}
|