types: mark further GC roots

Mark reachable native function and ucode function objects during incremental
GC steps in order to avoid freeing them prematurely.

Signed-off-by: Jo-Philipp Wich <jo@mein.io>

1 files changed, 9 insertions, 1 deletions

diff --git a/types.c b/types.c
index 9c75560..2648da9 100644
--- a/types.c
+++ b/types.c
@@ -2064,10 +2064,18 @@ ucv_gc_common(uc_vm_t *vm, bool final)
 	if (!final) {
 		/* mark reachable objects */
 		ucv_gc_mark(vm->globals);
+		ucv_gc_mark(vm->exception.stacktrace);
 
-		for (i = 0; i < vm->callframes.count; i++)
+		for (i = 0; i < vm->callframes.count; i++) {
 			ucv_gc_mark(vm->callframes.entries[i].ctx);
 
+			if (vm->callframes.entries[i].closure)
+				ucv_gc_mark(&vm->callframes.entries[i].closure->header);
+
+			if (vm->callframes.entries[i].cfunction)
+				ucv_gc_mark(&vm->callframes.entries[i].cfunction->header);
+		}
+
 		for (i = 0; i < vm->stack.count; i++)
 			ucv_gc_mark(vm->stack.entries[i]);
 	}