Protect stack region before calling mrb_convert_type(); fix #3504

mruby · Mar 16, 2017 · b64f087 · b64f087
1 parent f8b31a0
commit b64f087
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/vm.c b/src/vm.c
@@ -1207,6 +1207,7 @@ mrb_vm_exec(mrb_state *mrb, struct RProc *proc, mrb_code *pc)
       else {
         mrb_value blk = regs[bidx];
         if (!mrb_nil_p(blk) && mrb_type(blk) != MRB_TT_PROC) {
+          ci->nregs = bidx+1;
           regs[bidx] = mrb_convert_type(mrb, blk, MRB_TT_PROC, "Proc", "to_proc");
         }
       }
@@ -1438,6 +1439,7 @@ mrb_vm_exec(mrb_state *mrb, struct RProc *proc, mrb_code *pc)
         }
         blk = regs[bidx];
         if (!mrb_nil_p(blk) && mrb_type(blk) != MRB_TT_PROC) {
+          ci->nregs = bidx+1;
           regs[bidx] = mrb_convert_type(mrb, blk, MRB_TT_PROC, "Proc", "to_proc");
           ci = mrb->c->ci;
         }