* cont.c: support Fiber. Check test/ruby/test_fiber.rb for detail.
Fiber is known as "Micro Thread", "Coroutine", and other terms. At this time, only Fiber#pass is supported to change context. I want to know more suitable method name/API for Fiber (... do you know more suitable class name instead of Fiber?) as "suspend/resume", "call", "yield", "start/kick/stop/restart", .... * eval.c, eval_intern.h, thread.c, yarvcore.c, yarvcore.h: ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@12395 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
b69b1af650
commit
57062d91b9
11
ChangeLog
11
ChangeLog
@ -1,3 +1,14 @@
|
|||||||
|
Mon May 28 03:56:44 2007 Koichi Sasada <ko1@atdot.net>
|
||||||
|
|
||||||
|
* cont.c: support Fiber. Check test/ruby/test_fiber.rb for detail.
|
||||||
|
Fiber is known as "Micro Thread", "Coroutine", and other terms.
|
||||||
|
At this time, only Fiber#pass is supported to change context.
|
||||||
|
I want to know more suitable method name/API for Fiber (... do you
|
||||||
|
know more suitable class name instead of Fiber?) as "suspend/resume",
|
||||||
|
"call", "yield", "start/kick/stop/restart", ....
|
||||||
|
|
||||||
|
* eval.c, eval_intern.h, thread.c, yarvcore.c, yarvcore.h: ditto.
|
||||||
|
|
||||||
Sat May 26 00:38:21 2007 Nobuyoshi Nakada <nobu@ruby-lang.org>
|
Sat May 26 00:38:21 2007 Nobuyoshi Nakada <nobu@ruby-lang.org>
|
||||||
|
|
||||||
* eval.c (ruby_exec_internal): do nothing if no code.
|
* eval.c (ruby_exec_internal): do nothing if no code.
|
||||||
|
387
cont.c
387
cont.c
@ -15,18 +15,24 @@
|
|||||||
#include "gc.h"
|
#include "gc.h"
|
||||||
#include "eval_intern.h"
|
#include "eval_intern.h"
|
||||||
|
|
||||||
typedef struct rb_continuation_struct {
|
typedef struct rb_context_struct {
|
||||||
rb_thread_t saved_thread;
|
VALUE self;
|
||||||
rb_jmpbuf_t jmpbuf;
|
|
||||||
VALUE retval;
|
VALUE retval;
|
||||||
|
VALUE prev; /* for fiber */
|
||||||
VALUE *vm_stack;
|
VALUE *vm_stack;
|
||||||
VALUE *machine_stack;
|
VALUE *machine_stack;
|
||||||
VALUE *machine_stack_src;
|
VALUE *machine_stack_src;
|
||||||
|
rb_thread_t saved_thread;
|
||||||
|
rb_jmpbuf_t jmpbuf;
|
||||||
int machine_stack_size;
|
int machine_stack_size;
|
||||||
} rb_continuation_t;
|
int alive;
|
||||||
|
} rb_context_t;
|
||||||
|
|
||||||
|
VALUE rb_cCont;
|
||||||
|
VALUE rb_cFiber;
|
||||||
|
|
||||||
#define GetContPtr(obj, ptr) \
|
#define GetContPtr(obj, ptr) \
|
||||||
Data_Get_Struct(obj, rb_continuation_t, ptr)
|
Data_Get_Struct(obj, rb_context_t, ptr)
|
||||||
|
|
||||||
NOINLINE(static VALUE cont_capture(volatile int *stat));
|
NOINLINE(static VALUE cont_capture(volatile int *stat));
|
||||||
|
|
||||||
@ -37,8 +43,10 @@ cont_mark(void *ptr)
|
|||||||
{
|
{
|
||||||
MARK_REPORT_ENTER("cont");
|
MARK_REPORT_ENTER("cont");
|
||||||
if (ptr) {
|
if (ptr) {
|
||||||
rb_continuation_t *cont = ptr;
|
rb_context_t *cont = ptr;
|
||||||
rb_gc_mark(cont->retval);
|
rb_gc_mark(cont->retval);
|
||||||
|
rb_gc_mark(cont->prev);
|
||||||
|
|
||||||
rb_thread_mark(&cont->saved_thread);
|
rb_thread_mark(&cont->saved_thread);
|
||||||
|
|
||||||
if (cont->vm_stack) {
|
if (cont->vm_stack) {
|
||||||
@ -59,7 +67,8 @@ cont_free(void *ptr)
|
|||||||
{
|
{
|
||||||
FREE_REPORT_ENTER("cont");
|
FREE_REPORT_ENTER("cont");
|
||||||
if (ptr) {
|
if (ptr) {
|
||||||
rb_continuation_t *cont = ptr;
|
rb_context_t *cont = ptr;
|
||||||
|
FREE_UNLESS_NULL(cont->saved_thread.stack);
|
||||||
FREE_UNLESS_NULL(cont->machine_stack);
|
FREE_UNLESS_NULL(cont->machine_stack);
|
||||||
FREE_UNLESS_NULL(cont->vm_stack);
|
FREE_UNLESS_NULL(cont->vm_stack);
|
||||||
ruby_xfree(ptr);
|
ruby_xfree(ptr);
|
||||||
@ -67,24 +76,11 @@ cont_free(void *ptr)
|
|||||||
FREE_REPORT_LEAVE("cont");
|
FREE_REPORT_LEAVE("cont");
|
||||||
}
|
}
|
||||||
|
|
||||||
static VALUE
|
static void
|
||||||
cont_capture(volatile int *stat)
|
cont_save_machine_stack(rb_thread_t *th, rb_context_t *cont)
|
||||||
{
|
{
|
||||||
rb_continuation_t *cont;
|
|
||||||
VALUE contval;
|
|
||||||
rb_thread_t *th = GET_THREAD(), *sth;
|
|
||||||
int size;
|
int size;
|
||||||
|
|
||||||
contval = Data_Make_Struct(rb_cCont, rb_continuation_t,
|
|
||||||
cont_mark, cont_free, cont);
|
|
||||||
|
|
||||||
/* save context */
|
|
||||||
cont->saved_thread = *th;
|
|
||||||
sth = &cont->saved_thread;
|
|
||||||
sth->stack = 0; /* clear to skip GC marking */
|
|
||||||
cont->vm_stack = ALLOC_N(VALUE, sth->stack_size);
|
|
||||||
MEMCPY(cont->vm_stack, th->stack, VALUE, sth->stack_size);
|
|
||||||
|
|
||||||
rb_gc_set_stack_end(&th->machine_stack_end);
|
rb_gc_set_stack_end(&th->machine_stack_end);
|
||||||
if (th->machine_stack_start > th->machine_stack_end) {
|
if (th->machine_stack_start > th->machine_stack_end) {
|
||||||
size = cont->machine_stack_size = th->machine_stack_start - th->machine_stack_end;
|
size = cont->machine_stack_size = th->machine_stack_start - th->machine_stack_end;
|
||||||
@ -95,8 +91,45 @@ cont_capture(volatile int *stat)
|
|||||||
cont->machine_stack_src = th->machine_stack_start;
|
cont->machine_stack_src = th->machine_stack_start;
|
||||||
}
|
}
|
||||||
|
|
||||||
cont->machine_stack = ALLOC_N(VALUE, size);
|
if (cont->machine_stack) {
|
||||||
|
REALLOC_N(cont->machine_stack, VALUE, size);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
cont->machine_stack = ALLOC_N(VALUE, size);
|
||||||
|
}
|
||||||
|
|
||||||
MEMCPY(cont->machine_stack, cont->machine_stack_src, VALUE, size);
|
MEMCPY(cont->machine_stack, cont->machine_stack_src, VALUE, size);
|
||||||
|
}
|
||||||
|
|
||||||
|
static rb_context_t *
|
||||||
|
cont_new(VALUE klass)
|
||||||
|
{
|
||||||
|
rb_context_t *cont;
|
||||||
|
volatile VALUE contval;
|
||||||
|
rb_thread_t *th = GET_THREAD(), *sth;
|
||||||
|
|
||||||
|
contval = Data_Make_Struct(klass, rb_context_t,
|
||||||
|
cont_mark, cont_free, cont);
|
||||||
|
cont->self = contval;
|
||||||
|
cont->alive = Qtrue;
|
||||||
|
|
||||||
|
/* save context */
|
||||||
|
cont->saved_thread = *th;
|
||||||
|
sth = &cont->saved_thread;
|
||||||
|
|
||||||
|
return cont;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
cont_capture(volatile int *stat)
|
||||||
|
{
|
||||||
|
rb_context_t *cont = cont_new(rb_cCont);
|
||||||
|
rb_thread_t *th = &cont->saved_thread;
|
||||||
|
|
||||||
|
cont->vm_stack = ALLOC_N(VALUE, th->stack_size);
|
||||||
|
MEMCPY(cont->vm_stack, th->stack, VALUE, th->stack_size);
|
||||||
|
|
||||||
|
cont_save_machine_stack(th, cont);
|
||||||
|
|
||||||
if (ruby_setjmp(cont->jmpbuf)) {
|
if (ruby_setjmp(cont->jmpbuf)) {
|
||||||
VALUE retval;
|
VALUE retval;
|
||||||
@ -108,17 +141,25 @@ cont_capture(volatile int *stat)
|
|||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
*stat = 0;
|
*stat = 0;
|
||||||
return contval;
|
return cont->self;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
cont_restore_context_1(rb_continuation_t *cont)
|
cont_restore_1(rb_context_t *cont)
|
||||||
{
|
{
|
||||||
rb_thread_t *th = GET_THREAD(), *sth = &cont->saved_thread;
|
rb_thread_t *th = GET_THREAD(), *sth = &cont->saved_thread;
|
||||||
|
|
||||||
/* restore thread context */
|
/* restore thread context */
|
||||||
MEMCPY(th->stack, cont->vm_stack, VALUE, sth->stack_size);
|
if (sth->stack) {
|
||||||
|
/* fiber */
|
||||||
|
th->stack = sth->stack;
|
||||||
|
th->stack_size = sth->stack_size;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
/* continuation */
|
||||||
|
MEMCPY(th->stack, cont->vm_stack, VALUE, sth->stack_size);
|
||||||
|
}
|
||||||
|
|
||||||
th->cfp = sth->cfp;
|
th->cfp = sth->cfp;
|
||||||
th->safe_level = sth->safe_level;
|
th->safe_level = sth->safe_level;
|
||||||
@ -126,45 +167,53 @@ cont_restore_context_1(rb_continuation_t *cont)
|
|||||||
th->state = sth->state;
|
th->state = sth->state;
|
||||||
th->status = sth->status;
|
th->status = sth->status;
|
||||||
th->tag = sth->tag;
|
th->tag = sth->tag;
|
||||||
|
th->errinfo = sth->errinfo;
|
||||||
|
th->first_proc = sth->first_proc;
|
||||||
|
|
||||||
|
th->fiber = cont->self;
|
||||||
|
|
||||||
/* restore machine stack */
|
/* restore machine stack */
|
||||||
MEMCPY(cont->machine_stack_src, cont->machine_stack,
|
if (cont->machine_stack_src) {
|
||||||
VALUE, cont->machine_stack_size);
|
MEMCPY(cont->machine_stack_src, cont->machine_stack,
|
||||||
|
VALUE, cont->machine_stack_size);
|
||||||
|
}
|
||||||
|
|
||||||
ruby_longjmp(cont->jmpbuf, 1);
|
ruby_longjmp(cont->jmpbuf, 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
NORETURN(NOINLINE(static void restore_context_0(rb_continuation_t *, VALUE *)));
|
NORETURN(NOINLINE(static void restore_context_0(rb_context_t *, VALUE *)));
|
||||||
|
|
||||||
static void
|
static void
|
||||||
cont_restore_context_0(rb_continuation_t *cont, VALUE *addr_in_prev_frame)
|
cont_restore_0(rb_context_t *cont, VALUE *addr_in_prev_frame)
|
||||||
{
|
{
|
||||||
|
if (cont->machine_stack_src) {
|
||||||
#define STACK_PAD_SIZE 1024
|
#define STACK_PAD_SIZE 1024
|
||||||
VALUE space[STACK_PAD_SIZE];
|
VALUE space[STACK_PAD_SIZE];
|
||||||
|
|
||||||
#if STACK_GROW_DIRECTION < 0 /* downward */
|
#if STACK_GROW_DIRECTION < 0 /* downward */
|
||||||
if (addr_in_prev_frame > cont->machine_stack_src) {
|
if (addr_in_prev_frame > cont->machine_stack_src) {
|
||||||
cont_restore_context_0(cont, &space[0]);
|
cont_restore_0(cont, &space[0]);
|
||||||
}
|
}
|
||||||
#elif STACK_GROW_DIRECTION > 0 /* upward */
|
#elif STACK_GROW_DIRECTION > 0 /* upward */
|
||||||
if (addr_in_prev_frame < cont->machine_stack_src + cont->machine_stack_size) {
|
|
||||||
cont_restore_context_0(cont, &space[STACK_PAD_SIZE-1]);
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
if (addr_in_prev_frame > &space[0]) {
|
|
||||||
/* Stack grows downward */
|
|
||||||
if (addr_in_prev_frame > cont->saved_thread.machine_stack_src) {
|
|
||||||
cont_restore_context_0(cont, &space[0]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
/* Stack grows upward */
|
|
||||||
if (addr_in_prev_frame < cont->machine_stack_src + cont->machine_stack_size) {
|
if (addr_in_prev_frame < cont->machine_stack_src + cont->machine_stack_size) {
|
||||||
cont_restore_context_0(cont, &space[STACK_PAD_SIZE-1]);
|
cont_restore_0(cont, &space[STACK_PAD_SIZE-1]);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
if (addr_in_prev_frame > &space[0]) {
|
||||||
|
/* Stack grows downward */
|
||||||
|
if (addr_in_prev_frame > cont->saved_thread.machine_stack_src) {
|
||||||
|
cont_restore_0(cont, &space[0]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
/* Stack grows upward */
|
||||||
|
if (addr_in_prev_frame < cont->machine_stack_src + cont->machine_stack_size) {
|
||||||
|
cont_restore_0(cont, &space[STACK_PAD_SIZE-1]);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
#endif
|
#endif
|
||||||
cont_restore_context_1(cont);
|
}
|
||||||
|
cont_restore_1(cont);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@ -214,8 +263,6 @@ cont_restore_context_0(rb_continuation_t *cont, VALUE *addr_in_prev_frame)
|
|||||||
* 3: 15 16
|
* 3: 15 16
|
||||||
*/
|
*/
|
||||||
|
|
||||||
VALUE rb_cCont;
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* call-seq:
|
* call-seq:
|
||||||
* callcc {|cont| block } => obj
|
* callcc {|cont| block } => obj
|
||||||
@ -244,6 +291,19 @@ rb_callcc(VALUE self)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
make_passing_arg(int argc, VALUE *argv)
|
||||||
|
{
|
||||||
|
switch(argc) {
|
||||||
|
case 0:
|
||||||
|
return Qnil;
|
||||||
|
case 1:
|
||||||
|
return argv[0];
|
||||||
|
default:
|
||||||
|
return rb_ary_new4(argc, argv);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* call-seq:
|
* call-seq:
|
||||||
* cont.call(args, ...)
|
* cont.call(args, ...)
|
||||||
@ -263,31 +323,216 @@ rb_callcc(VALUE self)
|
|||||||
static VALUE
|
static VALUE
|
||||||
rb_cont_call(int argc, VALUE *argv, VALUE contval)
|
rb_cont_call(int argc, VALUE *argv, VALUE contval)
|
||||||
{
|
{
|
||||||
rb_continuation_t *cont;
|
rb_context_t *cont;
|
||||||
rb_thread_t *th = GET_THREAD();
|
rb_thread_t *th = GET_THREAD();
|
||||||
GetContPtr(contval, cont);
|
GetContPtr(contval, cont);
|
||||||
|
|
||||||
if (cont->saved_thread.value != th->value) {
|
if (cont->saved_thread.self != th->self) {
|
||||||
rb_raise(rb_eRuntimeError, "continuation called across threads");
|
rb_raise(rb_eRuntimeError, "continuation called across threads");
|
||||||
}
|
}
|
||||||
if (cont->saved_thread.trap_tag != th->trap_tag) {
|
if (cont->saved_thread.trap_tag != th->trap_tag) {
|
||||||
rb_raise(rb_eRuntimeError, "continuation called across trap");
|
rb_raise(rb_eRuntimeError, "continuation called across trap");
|
||||||
}
|
}
|
||||||
|
|
||||||
switch(argc) {
|
cont->retval = make_passing_arg(argc, argv);
|
||||||
case 0:
|
|
||||||
cont->retval = Qnil;
|
cont_restore_0(cont, (VALUE *)&cont);
|
||||||
break;
|
return Qnil; /* unreachable */
|
||||||
case 1:
|
}
|
||||||
cont->retval = argv[0];
|
|
||||||
break;
|
/*********/
|
||||||
default:
|
/* fiber */
|
||||||
cont->retval = rb_ary_new4(argc, argv);
|
/*********/
|
||||||
break;
|
|
||||||
|
#define FIBER_STACK_SIZE (4 * 1024)
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_s_new(int argc, VALUE *argv, VALUE self)
|
||||||
|
{
|
||||||
|
rb_context_t *cont = cont_new(rb_cFiber);
|
||||||
|
rb_thread_t *th = &cont->saved_thread;
|
||||||
|
volatile VALUE fval = cont->self;
|
||||||
|
|
||||||
|
/* initialize */
|
||||||
|
cont->prev = Qnil;
|
||||||
|
cont->vm_stack = 0;
|
||||||
|
|
||||||
|
th->stack = 0;
|
||||||
|
th->stack_size = FIBER_STACK_SIZE;
|
||||||
|
th->stack = ALLOC_N(VALUE, th->stack_size);
|
||||||
|
th->cfp = (void *)(th->stack + th->stack_size);
|
||||||
|
th->cfp--;
|
||||||
|
th->cfp->pc = 0;
|
||||||
|
th->cfp->sp = th->stack + 1;
|
||||||
|
th->cfp->bp = 0;
|
||||||
|
th->cfp->lfp = th->stack;
|
||||||
|
*th->cfp->lfp = 0;
|
||||||
|
th->cfp->dfp = th->stack;
|
||||||
|
th->cfp->self = Qnil;
|
||||||
|
th->cfp->magic = 0;
|
||||||
|
th->cfp->iseq = 0;
|
||||||
|
th->cfp->proc = 0;
|
||||||
|
th->cfp->block_iseq = 0;
|
||||||
|
|
||||||
|
th->first_proc = rb_block_proc();
|
||||||
|
th->first_args = rb_ary_new4(argc, argv);
|
||||||
|
|
||||||
|
MEMCPY(&cont->jmpbuf, &th->root_jmpbuf, rb_jmpbuf_t, 1);
|
||||||
|
|
||||||
|
return cont->self;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE rb_fiber_pass(int argc, VALUE *args, VALUE fval);
|
||||||
|
|
||||||
|
static void
|
||||||
|
rb_fiber_terminate(rb_context_t *cont)
|
||||||
|
{
|
||||||
|
rb_context_t *prev_cont;
|
||||||
|
VALUE retval = cont->retval;
|
||||||
|
|
||||||
|
GetContPtr(cont->prev, prev_cont);
|
||||||
|
|
||||||
|
cont->alive = Qfalse;
|
||||||
|
|
||||||
|
|
||||||
|
if (prev_cont->alive == Qfalse) {
|
||||||
|
rb_fiber_pass(1, &retval, GET_THREAD()->root_fiber);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
rb_fiber_pass(1, &retval, cont->prev);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
rb_fiber_start(void)
|
||||||
|
{
|
||||||
|
rb_thread_t *th = GET_THREAD();
|
||||||
|
rb_context_t *cont;
|
||||||
|
rb_proc_t *proc;
|
||||||
|
VALUE args;
|
||||||
|
int state;
|
||||||
|
|
||||||
|
TH_PUSH_TAG(th);
|
||||||
|
if ((state = EXEC_TAG()) == 0) {
|
||||||
|
GetContPtr(th->fiber, cont);
|
||||||
|
GetProcPtr(cont->saved_thread.first_proc, proc);
|
||||||
|
args = cont->saved_thread.first_args;
|
||||||
|
th->errinfo = Qnil;
|
||||||
|
th->local_lfp = proc->block.lfp;
|
||||||
|
th->local_svar = Qnil;
|
||||||
|
cont->retval = th_invoke_proc(th, proc, proc->block.self,
|
||||||
|
RARRAY_LEN(args), RARRAY_PTR(args));
|
||||||
|
}
|
||||||
|
TH_POP_TAG();
|
||||||
|
|
||||||
|
if (state) {
|
||||||
|
th->thrown_errinfo = th->errinfo;
|
||||||
|
th->interrupt_flag = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
cont_restore_context_0(cont, (VALUE *)&cont);
|
rb_fiber_terminate(cont);
|
||||||
return Qnil; /* unreachable */
|
rb_bug("rb_fiber_start: unreachable");
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_current(rb_thread_t *th)
|
||||||
|
{
|
||||||
|
if (th->fiber == 0) {
|
||||||
|
/* save root */
|
||||||
|
th->root_fiber = th->fiber = cont_new(rb_cFiber)->self;
|
||||||
|
}
|
||||||
|
return th->fiber;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
cont_store(rb_context_t *next_cont)
|
||||||
|
{
|
||||||
|
rb_thread_t *th = GET_THREAD();
|
||||||
|
rb_context_t *cont;
|
||||||
|
|
||||||
|
if (th->fiber) {
|
||||||
|
GetContPtr(th->fiber, cont);
|
||||||
|
cont->saved_thread = *th;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
/* create current fiber */
|
||||||
|
cont = cont_new(rb_cFiber); /* no need to allocate vm stack */
|
||||||
|
th->root_fiber = th->fiber = cont->self;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cont->alive) {
|
||||||
|
next_cont->prev = cont->self;
|
||||||
|
}
|
||||||
|
cont_save_machine_stack(th, cont);
|
||||||
|
|
||||||
|
if (ruby_setjmp(cont->jmpbuf)) {
|
||||||
|
/* restored */
|
||||||
|
GetContPtr(th->fiber, cont);
|
||||||
|
return cont->retval;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
return Qundef;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_pass(int argc, VALUE *argv, VALUE fval)
|
||||||
|
{
|
||||||
|
VALUE retval;
|
||||||
|
rb_context_t *cont;
|
||||||
|
rb_thread_t *th = GET_THREAD();
|
||||||
|
|
||||||
|
GetContPtr(fval, cont);
|
||||||
|
|
||||||
|
if (cont->saved_thread.self != th->self) {
|
||||||
|
rb_raise(rb_eRuntimeError, "fiber called across threads");
|
||||||
|
}
|
||||||
|
if (cont->saved_thread.trap_tag != th->trap_tag) {
|
||||||
|
rb_raise(rb_eRuntimeError, "fiber called across trap");
|
||||||
|
}
|
||||||
|
|
||||||
|
cont->retval = make_passing_arg(argc, argv);
|
||||||
|
|
||||||
|
if ((retval = cont_store(cont)) == Qundef) {
|
||||||
|
cont_restore_0(cont, (VALUE *)&cont);
|
||||||
|
rb_bug("rb_fiber_pass: unreachable");
|
||||||
|
}
|
||||||
|
|
||||||
|
return retval;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_prev(VALUE fval)
|
||||||
|
{
|
||||||
|
rb_context_t *cont;
|
||||||
|
GetContPtr(fval, cont);
|
||||||
|
return cont->prev;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_alive_p(VALUE fval)
|
||||||
|
{
|
||||||
|
rb_context_t *cont;
|
||||||
|
GetContPtr(fval, cont);
|
||||||
|
return cont->alive;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_s_current(VALUE klass)
|
||||||
|
{
|
||||||
|
return rb_fiber_current(GET_THREAD());
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_s_prev(VALUE klass)
|
||||||
|
{
|
||||||
|
return rb_fiber_prev(rb_fiber_s_current(Qnil));
|
||||||
|
}
|
||||||
|
|
||||||
|
static VALUE
|
||||||
|
rb_fiber_s_pass(int argc, VALUE *argv, VALUE fval)
|
||||||
|
{
|
||||||
|
return rb_fiber_pass(argc, argv, rb_fiber_s_prev(Qnil));
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
void
|
||||||
@ -299,5 +544,17 @@ Init_Cont(void)
|
|||||||
rb_define_method(rb_cCont, "call", rb_cont_call, -1);
|
rb_define_method(rb_cCont, "call", rb_cont_call, -1);
|
||||||
rb_define_method(rb_cCont, "[]", rb_cont_call, -1);
|
rb_define_method(rb_cCont, "[]", rb_cont_call, -1);
|
||||||
rb_define_global_function("callcc", rb_callcc, 0);
|
rb_define_global_function("callcc", rb_callcc, 0);
|
||||||
|
|
||||||
|
rb_cFiber = rb_define_class("Fiber", rb_cObject);
|
||||||
|
rb_undef_alloc_func(rb_cFiber);
|
||||||
|
rb_define_method(rb_cFiber, "pass", rb_fiber_pass, -1);
|
||||||
|
rb_define_method(rb_cFiber, "prev", rb_fiber_prev, 0);
|
||||||
|
rb_define_method(rb_cFiber, "alive?", rb_fiber_alive_p, 0);
|
||||||
|
|
||||||
|
rb_define_singleton_method(rb_cFiber, "current", rb_fiber_s_current, 0);
|
||||||
|
rb_define_singleton_method(rb_cFiber, "prev", rb_fiber_s_prev, 0);
|
||||||
|
rb_define_singleton_method(rb_cFiber, "pass", rb_fiber_s_pass, -1);
|
||||||
|
rb_define_singleton_method(rb_cFiber, "new", rb_fiber_s_new, -1);
|
||||||
|
rb_define_singleton_method(rb_cFiber, "start", rb_fiber_s_new, -1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
16
eval.c
16
eval.c
@ -45,7 +45,6 @@ static VALUE rb_call(VALUE, VALUE, ID, int, const VALUE *, int);
|
|||||||
#include "eval_safe.h"
|
#include "eval_safe.h"
|
||||||
#include "eval_jump.h"
|
#include "eval_jump.h"
|
||||||
|
|
||||||
|
|
||||||
/* initialize ruby */
|
/* initialize ruby */
|
||||||
|
|
||||||
#if defined(__APPLE__)
|
#if defined(__APPLE__)
|
||||||
@ -114,7 +113,7 @@ ruby_options(int argc, char **argv)
|
|||||||
Init_stack((void *)&state);
|
Init_stack((void *)&state);
|
||||||
PUSH_TAG();
|
PUSH_TAG();
|
||||||
if ((state = EXEC_TAG()) == 0) {
|
if ((state = EXEC_TAG()) == 0) {
|
||||||
ruby_process_options(argc, argv);
|
SAVE_ROOT_JMPBUF(GET_THREAD(), ruby_process_options(argc, argv));
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
rb_clear_trace_func();
|
rb_clear_trace_func();
|
||||||
@ -166,7 +165,7 @@ ruby_cleanup(int ex)
|
|||||||
errs[0] = th->errinfo;
|
errs[0] = th->errinfo;
|
||||||
PUSH_TAG();
|
PUSH_TAG();
|
||||||
if ((state = EXEC_TAG()) == 0) {
|
if ((state = EXEC_TAG()) == 0) {
|
||||||
rb_thread_terminate_all();
|
SAVE_ROOT_JMPBUF(th, rb_thread_terminate_all());
|
||||||
}
|
}
|
||||||
else if (ex == 0) {
|
else if (ex == 0) {
|
||||||
ex = state;
|
ex = state;
|
||||||
@ -204,13 +203,16 @@ ruby_exec_internal(void)
|
|||||||
{
|
{
|
||||||
int state;
|
int state;
|
||||||
VALUE val;
|
VALUE val;
|
||||||
|
rb_thread_t *th = GET_THREAD();
|
||||||
|
|
||||||
if (!ruby_eval_tree) return 0;
|
if (!ruby_eval_tree) return 0;
|
||||||
|
|
||||||
PUSH_TAG();
|
PUSH_TAG();
|
||||||
if ((state = EXEC_TAG()) == 0) {
|
if ((state = EXEC_TAG()) == 0) {
|
||||||
GET_THREAD()->base_block = 0;
|
SAVE_ROOT_JMPBUF(th, {
|
||||||
val = yarvcore_eval_parsed(ruby_eval_tree,
|
th->base_block = 0;
|
||||||
rb_str_new2(ruby_sourcefile));
|
val = yarvcore_eval_parsed(ruby_eval_tree, rb_str_new2(ruby_sourcefile));
|
||||||
|
});
|
||||||
}
|
}
|
||||||
POP_TAG();
|
POP_TAG();
|
||||||
return state;
|
return state;
|
||||||
@ -1171,7 +1173,7 @@ rb_protect(VALUE (*proc) (VALUE), VALUE data, int *state)
|
|||||||
PUSH_TAG();
|
PUSH_TAG();
|
||||||
th->trap_tag = &trap_tag;
|
th->trap_tag = &trap_tag;
|
||||||
if ((status = EXEC_TAG()) == 0) {
|
if ((status = EXEC_TAG()) == 0) {
|
||||||
result = (*proc) (data);
|
SAVE_ROOT_JMPBUF(th, result = (*proc) (data));
|
||||||
}
|
}
|
||||||
th->trap_tag = trap_tag.prev;
|
th->trap_tag = trap_tag.prev;
|
||||||
POP_TAG();
|
POP_TAG();
|
||||||
|
@ -117,6 +117,14 @@ char *strrchr _((const char *, const char));
|
|||||||
|
|
||||||
#include <sys/stat.h>
|
#include <sys/stat.h>
|
||||||
|
|
||||||
|
#define SAVE_ROOT_JMPBUF(th, stmt) do \
|
||||||
|
if (ruby_setjmp((th)->root_jmpbuf) == 0) { \
|
||||||
|
stmt; \
|
||||||
|
} \
|
||||||
|
else { \
|
||||||
|
rb_fiber_start(th); \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
#define TH_PUSH_TAG(th) do { \
|
#define TH_PUSH_TAG(th) do { \
|
||||||
rb_thread_t * const _th = th; \
|
rb_thread_t * const _th = th; \
|
||||||
struct rb_vm_tag _tag; \
|
struct rb_vm_tag _tag; \
|
||||||
|
66
test/ruby/test_fiber.rb
Normal file
66
test/ruby/test_fiber.rb
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
require 'test/unit'
|
||||||
|
|
||||||
|
class TestFiber < Test::Unit::TestCase
|
||||||
|
def test_normal
|
||||||
|
f = Fiber.current
|
||||||
|
assert_equal(:ok2,
|
||||||
|
Fiber.new(:ok1){|e|
|
||||||
|
assert_equal(:ok1, e)
|
||||||
|
assert_equal(f, Fiber.prev)
|
||||||
|
Fiber.pass :ok2
|
||||||
|
}.pass)
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_term
|
||||||
|
assert_equal(:ok, Fiber.new{:ok}.pass)
|
||||||
|
assert_equal([:a, :b, :c, :d, :e],
|
||||||
|
Fiber.new{
|
||||||
|
Fiber.new{
|
||||||
|
Fiber.new{
|
||||||
|
Fiber.new{
|
||||||
|
[:a]
|
||||||
|
}.pass + [:b]
|
||||||
|
}.pass + [:c]
|
||||||
|
}.pass + [:d]
|
||||||
|
}.pass + [:e])
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_many_fibers
|
||||||
|
max = 10000
|
||||||
|
assert_equal(max, max.times{
|
||||||
|
Fiber.new{}
|
||||||
|
})
|
||||||
|
assert_equal(max,
|
||||||
|
max.times{|i|
|
||||||
|
Fiber.new{
|
||||||
|
}.pass
|
||||||
|
}
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_error
|
||||||
|
assert_raise(ArgumentError){
|
||||||
|
Fiber.new # Fiber without block
|
||||||
|
}
|
||||||
|
assert_raise(RuntimeError){
|
||||||
|
f = Fiber.new{}
|
||||||
|
Thread.new{f.pass}.join # Fiber passing across thread
|
||||||
|
}
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_loop
|
||||||
|
ary = []
|
||||||
|
f2 = nil
|
||||||
|
f1 = Fiber.new{
|
||||||
|
ary << f2.pass(:foo)
|
||||||
|
:bar
|
||||||
|
}
|
||||||
|
f2 = Fiber.new{
|
||||||
|
ary << f1.pass(:baz)
|
||||||
|
:ok
|
||||||
|
}
|
||||||
|
assert_equal(:ok, f1.pass)
|
||||||
|
assert_equal([:baz, :bar], ary)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
24
thread.c
24
thread.c
@ -298,17 +298,19 @@ thread_start_func_2(rb_thread_t *th, VALUE *stack_start)
|
|||||||
|
|
||||||
TH_PUSH_TAG(th);
|
TH_PUSH_TAG(th);
|
||||||
if ((state = EXEC_TAG()) == 0) {
|
if ((state = EXEC_TAG()) == 0) {
|
||||||
if (th->first_proc) {
|
SAVE_ROOT_JMPBUF(th, {
|
||||||
GetProcPtr(th->first_proc, proc);
|
if (th->first_proc) {
|
||||||
th->errinfo = Qnil;
|
GetProcPtr(th->first_proc, proc);
|
||||||
th->local_lfp = proc->block.lfp;
|
th->errinfo = Qnil;
|
||||||
th->local_svar = Qnil;
|
th->local_lfp = proc->block.lfp;
|
||||||
th->value = th_invoke_proc(th, proc, proc->block.self,
|
th->local_svar = Qnil;
|
||||||
RARRAY_LEN(args), RARRAY_PTR(args));
|
th->value = th_invoke_proc(th, proc, proc->block.self,
|
||||||
}
|
RARRAY_LEN(args), RARRAY_PTR(args));
|
||||||
else {
|
}
|
||||||
th->value = (*th->first_func)(th->first_func_arg);
|
else {
|
||||||
}
|
th->value = (*th->first_func)(th->first_func_arg);
|
||||||
|
}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
th->value = Qnil;
|
th->value = Qnil;
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
#define RUBY_VERSION "1.9.0"
|
#define RUBY_VERSION "1.9.0"
|
||||||
#define RUBY_RELEASE_DATE "2007-05-26"
|
#define RUBY_RELEASE_DATE "2007-05-28"
|
||||||
#define RUBY_VERSION_CODE 190
|
#define RUBY_VERSION_CODE 190
|
||||||
#define RUBY_RELEASE_CODE 20070526
|
#define RUBY_RELEASE_CODE 20070528
|
||||||
#define RUBY_PATCHLEVEL 0
|
#define RUBY_PATCHLEVEL 0
|
||||||
|
|
||||||
#define RUBY_VERSION_MAJOR 1
|
#define RUBY_VERSION_MAJOR 1
|
||||||
@ -9,7 +9,7 @@
|
|||||||
#define RUBY_VERSION_TEENY 0
|
#define RUBY_VERSION_TEENY 0
|
||||||
#define RUBY_RELEASE_YEAR 2007
|
#define RUBY_RELEASE_YEAR 2007
|
||||||
#define RUBY_RELEASE_MONTH 5
|
#define RUBY_RELEASE_MONTH 5
|
||||||
#define RUBY_RELEASE_DAY 26
|
#define RUBY_RELEASE_DAY 28
|
||||||
|
|
||||||
#ifdef RUBY_EXTERN
|
#ifdef RUBY_EXTERN
|
||||||
RUBY_EXTERN const char ruby_version[];
|
RUBY_EXTERN const char ruby_version[];
|
||||||
|
@ -268,7 +268,7 @@ thread_mark(void *ptr)
|
|||||||
VALUE *sp = th->cfp->sp + th->mark_stack_len;
|
VALUE *sp = th->cfp->sp + th->mark_stack_len;
|
||||||
rb_control_frame_t *cfp = th->cfp;
|
rb_control_frame_t *cfp = th->cfp;
|
||||||
rb_control_frame_t *limit_cfp =
|
rb_control_frame_t *limit_cfp =
|
||||||
(void *)(th->stack + th->stack_size);
|
(void *)(th->stack + th->stack_size);
|
||||||
|
|
||||||
while (p < sp) {
|
while (p < sp) {
|
||||||
rb_gc_mark(*p++);
|
rb_gc_mark(*p++);
|
||||||
@ -289,11 +289,12 @@ thread_mark(void *ptr)
|
|||||||
MARK_UNLESS_NULL(th->local_svar);
|
MARK_UNLESS_NULL(th->local_svar);
|
||||||
MARK_UNLESS_NULL(th->top_self);
|
MARK_UNLESS_NULL(th->top_self);
|
||||||
MARK_UNLESS_NULL(th->top_wrapper);
|
MARK_UNLESS_NULL(th->top_wrapper);
|
||||||
|
MARK_UNLESS_NULL(th->fiber);
|
||||||
|
MARK_UNLESS_NULL(th->root_fiber);
|
||||||
|
|
||||||
rb_mark_tbl(th->local_storage);
|
rb_mark_tbl(th->local_storage);
|
||||||
|
|
||||||
if (GET_THREAD() != th &&
|
if (GET_THREAD() != th && th->machine_stack_start && th->machine_stack_end) {
|
||||||
th->machine_stack_start && th->machine_stack_end) {
|
|
||||||
yarv_machine_stack_mark(th);
|
yarv_machine_stack_mark(th);
|
||||||
rb_gc_mark_locations((VALUE *)&th->machine_regs,
|
rb_gc_mark_locations((VALUE *)&th->machine_regs,
|
||||||
(VALUE *)(&th->machine_regs) +
|
(VALUE *)(&th->machine_regs) +
|
||||||
|
@ -497,6 +497,11 @@ struct rb_thread_struct
|
|||||||
rb_event_flag_t event_flags;
|
rb_event_flag_t event_flags;
|
||||||
int tracing;
|
int tracing;
|
||||||
|
|
||||||
|
/* fiber */
|
||||||
|
VALUE fiber;
|
||||||
|
VALUE root_fiber;
|
||||||
|
rb_jmpbuf_t root_jmpbuf;
|
||||||
|
|
||||||
/* misc */
|
/* misc */
|
||||||
int method_missing_reason;
|
int method_missing_reason;
|
||||||
int abort_on_exception;
|
int abort_on_exception;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user