diff --git a/src/vm.c b/src/vm.c index 57df10ace..15b25ecb0 100644 --- a/src/vm.c +++ b/src/vm.c @@ -326,7 +326,10 @@ ecall(mrb_state *mrb) if (!p) return; mrb_assert(!MRB_PROC_CFUNC_P(p)); c->ensure[i] = NULL; - nregs = ci->proc->body.irep->nregs; + nregs = p->upper->body.irep->nregs; + if (ci->proc->body.irep->nregs > nregs) { + nregs = ci->proc->body.irep->nregs; + } cioff = ci - c->cibase; ci = cipush(mrb); ci->stackent = mrb->c->stack; @@ -336,9 +339,9 @@ ecall(mrb_state *mrb) ci->proc = p; ci->nregs = p->body.irep->nregs; ci->target_class = MRB_PROC_TARGET_CLASS(p); - c->stack += nregs; env = MRB_PROC_ENV(p); mrb_assert(env); + c->stack += nregs; exc = mrb->exc; mrb->exc = 0; if (exc) { mrb_gc_protect(mrb, mrb_obj_value(exc));