chore: cleanup heap mess before close clox

Author: aymaneelmaini

Makefile

@@ -3,7 +3,7 @@ CFLAGS = -std=c99 -Wall -Wextra -g
 CTEST_FLAGS = -std=c99 -g
 LDFLAGS = -lcriterion
 
-SOURCES = src/scanner.c src/chunk.c src/compiler.c src/debug.c src/memory.c src/value.c src/vm.c
+SOURCES = src/scanner.c src/chunk.c src/compiler.c src/debug.c src/memory.c src/value.c src/vm.c src/object.c
 TEST_SOURCES = tests/scanner_test.c tests/compiler_test.c
 
 all: clox

src/memory.c

@@ -1,6 +1,9 @@
 #include <stdlib.h>
 
 #include "memory.h"
+#include "object.h"
+#include "value.h"
+#include "vm.h"
 
 void* reallocate(void* pointer, size_t oldSize, size_t newSize)
 {
@@ -15,3 +18,27 @@ void* reallocate(void* pointer, size_t oldSize, size_t newSize)
         exit(1);
     return result;
 }
+
+static void free_object(Obj* object)
+{
+    switch (object->type)
+    {
+    case OBJ_STRING:
+    {
+        ObjString* string = (ObjString*)object;
+        FREE_ARRAY(char, string->chars, string->length + 1);
+        FREE(ObjString, object);
+        break;
+    }
+    }
+}
+void free_objects()
+{
+    Obj* object = vm.objects;
+    while (object != NULL)
+    {
+        Obj* next = object->next;
+        free_object(object);
+        object = next;
+    }
+}

src/memory.h

@@ -7,6 +7,8 @@
 
 #define GROW_CAPACITY(capacity) ((capacity) < 8 ? 8 : (capacity) * 2)
 
+#define FREE(type, pointer) reallocate(pointer, sizeof(type), 0)
+
 #define GROW_ARRAY(type, pointer, oldCount, newCount)                          \
     (type*)reallocate(pointer, sizeof(type) * (oldCount),                      \
                       sizeof(type) * (newCount))
@@ -21,5 +23,6 @@
 // size = smaller than old size => shrink existing allocation old size =
 // non-zero, new size = larger than old size => grow existing allocation
 void* reallocate(void* pointer, size_t oldSize, size_t newSize);
+void  free_objects();
 
 #endif

src/object.c

@@ -13,17 +13,19 @@ static Obj* allocate_object(size_t size, ObjType type)
 {
     Obj* object = (Obj*)reallocate(NULL, 0, size);
     object->type = type;
+    object->next = vm.objects;
+    vm.objects = object;
 
     return object;
 }
 
-static ObjString* allocate_string(char* chars, int length)
+static ObjString* allocate_string(const char* chars, int length)
 {
     ObjString* string = ALLOCATE_OBJ(ObjString, OBJ_STRING);
     string->length = length;
     string->chars = chars;
-
     return string;
+
 }
 
 ObjString* take_string(const char* chars, int length)

src/object.h

@@ -16,7 +16,8 @@ typedef enum
 
 struct Obj
 {
-    ObjType type;
+    ObjType     type;
+    struct Obj* next;
 };
 
 struct ObjString

src/vm.c

@@ -36,11 +36,12 @@ static void runtime_error(const char* format, ...)
 void init_VM()
 {
     reset_stack();
+    vm.objects = NULL;
 }
 
 void free_VM()
 {
-    printf("close vm \n");
+    free_objects();
 }
 
 void push(Value value)

src/vm.h

@@ -12,6 +12,7 @@ typedef struct
     uint8_t* ip;  // aka of Instrction Pointer
     Value    stack[STACK_MAX];
     Value*   stack_top;
+    Obj*     objects;
 } VM;
 
 typedef enum
@@ -21,6 +22,8 @@ typedef enum
     INTERPRET_RUNTIME_ERROR,
 } InterpretResult;
 
+extern VM vm;
+
 void            init_VM();
 void            free_VM();
 InterpretResult interpret(char* source);