Fix memory leak

Author: hasumikin

mrbgems/picoruby-mruby/src/task.c

@@ -224,13 +224,27 @@ mrb_tcb_new(mrb_state *mrb, enum MrbTaskState task_state, int priority)
   return tcb;
 }
 
+static void
+cleanup_context(mrb_state *mrb, struct mrb_context *c)
+{
+  if (c->stbase) {
+    mrb_free(mrb, c->stbase);
+    c->stbase = NULL;
+  }
+  if (c->cibase) {
+    mrb_free(mrb, c->cibase);
+    c->cibase = NULL;
+  }
+}
+
 
 #define TASK_STACK_INIT_SIZE 64
 #define TASK_CI_INIT_SIZE 8
 
 void
 mrb_tcb_init_context(mrb_state *mrb, struct mrb_context *c, struct RProc *proc)
 {
+  cleanup_context(mrb, c);
   size_t slen = TASK_STACK_INIT_SIZE;
   if (proc->body.irep->nregs > slen) {
     slen += proc->body.irep->nregs;

mrbgems/picoruby-sandbox/src/mruby/sandbox.c

@@ -220,6 +220,7 @@ mrb_sandbox_exec_vm_code(mrb_state *mrb, mrb_value self)
   mrb_value vm_code;
   mrb_get_args(mrb, "S", &vm_code);
   const uint8_t *code = (const uint8_t *)RSTRING_PTR(vm_code);
+  if (ss->irep) mrc_irep_free(ss->cc, ss->irep);
   ss->irep = mrb_read_irep(mrb, code);
   if (sandbox_exec_vm_code_sub(mrb, ss)) {
     return mrb_true_value();
@@ -234,6 +235,7 @@ mrb_sandbox_exec_vm_code_from_memory(mrb_state *mrb, mrb_value self)
   SS();
   mrb_int address;
   mrb_get_args(mrb, "i", &address);
+  if (ss->irep) mrc_irep_free(ss->cc, ss->irep);
   ss->irep = mrb_read_irep(mrb, (const uint8_t *)(uintptr_t)address);
   if (sandbox_exec_vm_code_sub(mrb, ss)) {
     return mrb_true_value();