Fix heap use after free on mruby-aws-sigv4.

b8869498 · Takeshi Watanabe · 1b261382 · b8869498
Unverified Commit b8869498 authored Apr 27, 2018 by Takeshi Watanabe
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/vm.c src/vm.c +2 -0

No files found.
--- a/src/vm.c
+++ b/src/vm.c
@@ -527,6 +527,8 @@ mrb_exec_irep(mrb_state *mrb, mrb_value self, struct RProc *p)
    return MRB_PROC_CFUNC(p)(mrb, self);
  }
  ci->nregs = p->body.irep->nregs;
+  ci->env = MRB_PROC_ENV(p);
+  if (ci->env) ci->env->stack[0] = self;
  if (ci->argc < 0) keep = 3;
  else keep = ci->argc + 2;
  if (ci->nregs < keep) {