void Fiber::start_on_stack() {
#ifdef FIBER_ENABLED
VM* state = VM::current();
Fiber* fib = Fiber::current(state);
// Affix this fiber to this thread now.
fib->state_ = state;
fib->starter()->send(state, NULL, G(sym_call));
// GC has run! Don't use stack vars!
fib = Fiber::current(state);
fib->status_ = Fiber::eDead;
Fiber* dest = fib->prev();
assert(!dest->nil_p());
dest->run();
dest->value(state, Qnil);
state->set_current_fiber(dest);
if(setcontext(dest->ucontext()) != 0)
assert(0 && "fatal swapcontext() error");
assert(0 && "fatal start_on_stack error");
#else
abort();
#endif
}
Fiber* Fiber::current(STATE) {
#ifdef RBX_FIBER_ENABLED
Fiber* fib = state->vm()->current_fiber.get();
// Lazily allocate a root fiber.
if(fib->nil_p()) {
fib = state->new_object<Fiber>(G(fiber));
fib->prev(state, nil<Fiber>());
fib->locals(state, nil<LookupTable>());
fib->root_ = true;
fib->status_ = Fiber::eRunning;
fib->data_ = state->vm()->new_fiber_data(true);
state->memory()->needs_finalization(fib, (FinalizerFunction)&Fiber::finalize);
state->vm()->current_fiber.set(fib);
state->vm()->root_fiber.set(fib);
}
return fib;
#else
return nil<Fiber>();
#endif
}
Object* Fiber::s_yield(STATE, Arguments& args, CallFrame* calling_environment) {
#ifdef FIBER_ENABLED
Fiber* cur = Fiber::current(state);
Fiber* dest_fib = cur->prev();
assert(cur != dest_fib);
if(cur->root_) {
Exception::fiber_error(state, "can't yield from root fiber");
}
cur->prev(state, (Fiber*)Qnil);
Array* val = args.as_array(state);
dest_fib->value(state, val);
cur->sleep(calling_environment);
dest_fib->run();
state->set_current_fiber(dest_fib);
if(swapcontext(cur->ucontext(), dest_fib->ucontext()) != 0)
assert(0 && "fatal swapcontext() error");
// Back here when someone yields back to us!
// Beware here, because the GC has probably run so GC pointers on the C++ stack
// can't be accessed.
cur = Fiber::current(state);
Array* ret = cur->value();
if(ret->nil_p()) return Qnil;
switch(ret->size()) {
case 0: return Qnil;
case 1: return ret->get(state, 0);
default: return ret;
}
#else
return Primitives::failure();
#endif
}
void Fiber::start_on_stack() {
#ifdef RBX_FIBER_ENABLED
VM* vm = VM::current();
State state_obj(vm), *state = &state_obj;
Fiber* fib = Fiber::current(state);
// Reset the current fiber again to reset the stack limits so
// we can properly detect stack overflows
vm->set_current_fiber(fib);
Array* result = nil<Array>();
Object* obj = fib->starter()->send(state, G(sym_call), fib->value(), cNil, false);
// GC has run! Don't use stack vars!
fib = Fiber::current(state);
fib->status(Fiber::eDead);
fib->dead(cTrue);
fib->set_call_frame(state, 0);
Fiber* dest = fib->prev();
// If this fiber has already been cleaned up, just ignore this
if(!dest->data()) return;
assert(!dest->nil_p());
// Box this up so it's in a standard format at the point
// of returning, so we can deal with it in the same way
// as *args from #yield, #resume, and #transfer
if(obj) {
result = Array::create(state, 1);
result->set(state, 0, obj);
} else {
if(state->vm()->thread_state()->raise_reason() == cException) {
dest->exception(state, state->vm()->thread_state()->current_exception());
}
}
vm->metrics().system.fibers_destroyed++;
dest->run(state);
dest->value(state, result);
dest->data()->switch_and_orphan(state, fib->data());
// TODO: CallFrame: return from this function
rubinius::bug("returning from Fiber::start_on_stack");
#else
rubinius::bug("Fibers not supported on this platform");
#endif
}
Object* Fiber::s_yield(STATE, Arguments& args) {
#ifdef RBX_FIBER_ENABLED
Fiber* cur = Fiber::current(state);
Fiber* dest_fib = cur->prev();
assert(cur != dest_fib);
if(cur->root()) {
Exception::raise_fiber_error(state, "can't yield from root fiber");
}
cur->prev(state, nil<Fiber>());
Array* val = args.as_array(state);
dest_fib->value(state, val);
cur->sleep(state);
dest_fib->run(state);
dest_fib->data()->switch_to(state, cur->data());
// Back here when someone yields back to us!
// Beware here, because the GC has probably run so GC pointers on the C++ stack
// can't be accessed.
cur = Fiber::current(state);
Array* ret = cur->value();
if(ret->nil_p()) return cNil;
switch(ret->size()) {
case 0: return cNil;
case 1: return ret->get(state, 0);
default: return ret;
}
#else
Exception::raise_not_implemented_error(state, "Fibers not supported on this platform");
#endif
}
void Fiber::start_on_stack() {
#ifdef RBX_FIBER_ENABLED
VM* vm = VM::current();
State state(vm);
Fiber* fib = Fiber::current(&state);
// Reset the current fiber again to reset the stack limits so
// we can properly detect stack overflows
vm->set_current_fiber(fib);
Array* result = nil<Array>();
Object* obj = fib->starter()->send(&state, NULL, state.globals().sym_call.get(), fib->value(), cNil, false);
// GC has run! Don't use stack vars!
fib = Fiber::current(&state);
fib->status_ = Fiber::eDead;
fib->dead_ = cTrue;
Fiber* dest = fib->prev();
// If this fiber has already been cleaned up, just ignore this
if(!dest->data()) return;
assert(!dest->nil_p());
// Box this up so it's in a standard format at the point
// of returning, so we can deal with it in the same way
// as *args from #yield, #resume, and #transfer
if(obj) {
result = Array::create(&state, 1);
result->set(&state, 0, obj);
} else {
if(state.vm()->thread_state()->raise_reason() == cException) {
dest->exception(&state, state.vm()->thread_state()->current_exception());
}
}
dest->run();
dest->value(&state, result);
state.vm()->set_current_fiber(dest);
dest->data_->switch_and_orphan(&state, fib->data_);
assert(0 && "fatal start_on_stack error");
#else
rubinius::bug("Fibers not supported on this platform");
#endif
}
Object* Fiber::s_yield(STATE, Arguments& args, CallFrame* calling_environment) {
#ifdef FIBER_ENABLED
Fiber* cur = Fiber::current(state);
Fiber* dest_fib = cur->prev();
assert(cur != dest_fib);
cur->prev(state, (Fiber*)Qnil);
Object* val = Qnil;
if(args.total() == 1) {
val = args.get_argument(0);
} else if(args.total() > 1) {
val = args.as_array(state);
}
dest_fib->value(state, val);
cur->sleep(calling_environment);
dest_fib->run();
state->set_current_fiber(dest_fib);
if(swapcontext(cur->ucontext(), dest_fib->ucontext()) != 0)
assert(0 && "fatal swapcontext() error");
// Back here when someone yields back to us!
// Beware here, because the GC has probably run so GC pointers on the C++ stack
// can't be accessed.
cur = Fiber::current(state);
return cur->value();
#else
return Primitives::failure();
#endif
}
Fiber* Fiber::create(STATE, Object* self, Object* callable) {
#ifdef RBX_FIBER_ENABLED
Fiber* fib = state->new_object<Fiber>(as<Class>(self));
fib->starter(state, callable);
fib->prev(state, nil<Fiber>());
fib->locals(state, nil<LookupTable>());
fib->root_ = false;
fib->status_ = Fiber::eSleeping;
fib->data_ = 0;
state->memory()->needs_finalization(fib, (FinalizerFunction)&Fiber::finalize);
return fib;
#else
return static_cast<Fiber*>(Primitives::failure());
#endif
}
void Fiber::start_on_stack() {
#ifdef FIBER_ENABLED
VM* state = VM::current();
Fiber* fib = Fiber::current(state);
// Affix this fiber to this thread now.
fib->state_ = state;
Array* result = (Array*)Qnil;
Object* obj = fib->starter()->send(state, NULL, G(sym_call), fib->value(), Qnil, false);
// GC has run! Don't use stack vars!
fib = Fiber::current(state);
fib->status_ = Fiber::eDead;
fib->set_ivar(state, state->symbol("@dead"), Qtrue);
Fiber* dest = fib->prev();
assert(!dest->nil_p());
// Box this up so it's in a standard format at the point
// of returning, so we can deal with it in the same way
// as *args from #yield, #resume, and #transfer
if(obj) {
result = Array::create(state, 1);
result->set(state, 0, obj);
} else {
if(state->thread_state()->raise_reason() == cException) {
dest->exception(state, state->thread_state()->current_exception());
}
}
dest->run();
dest->value(state, result);
state->set_current_fiber(dest);
if(setcontext(dest->ucontext()) != 0)
assert(0 && "fatal swapcontext() error");
assert(0 && "fatal start_on_stack error");
#else
abort();
#endif
}
Fiber* Fiber::create(STATE, Integer* i_stack_size, Object* callable) {
#ifdef FIBER_ENABLED
int stack_size = i_stack_size->to_native();
if(stack_size < 64 * 1024) {
stack_size = 64 * 1024;
}
Fiber* fib = state->new_object<Fiber>(G(fiber));
fib->starter(state, callable);
fib->prev(state, nil<Fiber>());
fib->top_ = 0;
fib->root_ = false;
fib->state_ = 0;
fib->status_ = Fiber::eSleeping;
fib->stack_size_ = stack_size;
fib->stack_ = malloc(stack_size);
fib->context_ = new ucontext_t;
state->om->needs_finalization(fib, (FinalizerFunction)&Fiber::finalize);
ucontext_t* ctx = fib->ucontext();
if(getcontext(ctx) != 0) assert(0 && "fatal getcontext() error");
ctx->uc_link = 0;
ctx->uc_stack.ss_sp = (char *) fib->stack_;
ctx->uc_stack.ss_size = stack_size;
ctx->uc_stack.ss_flags = 0;
makecontext(ctx, start_on_stack, 0);
return fib;
#else
return reinterpret_cast<Fiber*>(Primitives::failure());
#endif
}