Save block argument position in e->cioff; fix #3593

parent d9fb8b69
...@@ -157,8 +157,14 @@ mrb_f_block_given_p_m(mrb_state *mrb, mrb_value self) ...@@ -157,8 +157,14 @@ mrb_f_block_given_p_m(mrb_state *mrb, mrb_value self)
/* top-level does not have block slot (alway false) */ /* top-level does not have block slot (alway false) */
if (sp == mrb->c->stbase) if (sp == mrb->c->stbase)
return mrb_false_value(); return mrb_false_value();
ci = mrb->c->cibase + e->cioff; if (e->cioff < 0) {
bp = ci[1].stackent + 1; /* use saved block arg position */
bp = &e->stack[-e->cioff];
}
else {
ci = mrb->c->cibase + e->cioff;
bp = ci[1].stackent + 1;
}
} }
} }
if (ci->argc > 0) { if (ci->argc > 0) {
......
...@@ -257,8 +257,13 @@ mrb_env_unshare(mrb_state *mrb, struct REnv *e) ...@@ -257,8 +257,13 @@ mrb_env_unshare(mrb_state *mrb, struct REnv *e)
{ {
size_t len = (size_t)MRB_ENV_STACK_LEN(e); size_t len = (size_t)MRB_ENV_STACK_LEN(e);
mrb_value *p = (mrb_value *)mrb_malloc(mrb, sizeof(mrb_value)*len); mrb_value *p = (mrb_value *)mrb_malloc(mrb, sizeof(mrb_value)*len);
ptrdiff_t cioff = e->cioff;
MRB_ENV_UNSHARE_STACK(e); MRB_ENV_UNSHARE_STACK(e);
if (!e->c) {
/* save block argument position (negated) */
e->cioff = -mrb->c->cibase[cioff].argc-1;
}
if (len > 0) { if (len > 0) {
stack_copy(p, e->stack, len); stack_copy(p, e->stack, len);
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment