From b8869498b7b1458af1cf2a5ccc7644849f826230 Mon Sep 17 00:00:00 2001 From: Takeshi Watanabe Date: Fri, 27 Apr 2018 10:45:59 +0900 Subject: Fix heap use after free on mruby-aws-sigv4. --- src/vm.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src') diff --git a/src/vm.c b/src/vm.c index 9e43d0a4e..429cc2648 100644 --- 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) { -- cgit v1.2.3