Reduce the size of VM stack clear window; ref #3676

mruby · May 30, 2017 · acd3ac6 · acd3ac6
1 parent 2599c19
commit acd3ac6
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/vm.c b/src/vm.c
@@ -851,17 +851,15 @@ mrb_vm_run(mrb_state *mrb, struct RProc *proc, mrb_value self, unsigned int stac
   mrb_value result;
   struct mrb_context *c = mrb->c;
   int cioff = c->ci - c->cibase;
-  int nregs = irep->nregs;
-  mrb_value *stk;
+  unsigned int nregs = irep->nregs;
 
   if (!c->stack) {
     stack_init(mrb);
   }
   if (stack_keep > nregs)
     nregs = stack_keep;
-  stk = c->stack + stack_keep;
-  if (stk < c->stend) {
-    stack_clear(stk, c->stend - stk);
+  if (nregs > stack_keep) {
+    stack_clear(c->stack + stack_keep, nregs - stack_keep);
   }
   stack_extend(mrb, nregs);
   c->stack[0] = self;