Merge pull request #120 from Servostar/119-fix-ast-memory-leak

119 fix ast memory leak

src/ast/ast.c

@@ -1,15 +1,15 @@
 
 #include <ast/ast.h>
 #include <stdio.h>
-#include <stdlib.h>
 #include <sys/log.h>
 #include <assert.h>
+#include <mem/cache.h>
 
 struct AST_Node_t *AST_new_node(TokenLocation location, enum AST_SyntaxElement_t kind, const char* value) {
     DEBUG("creating new AST node: %d \"%s\"", kind, value);
     assert(kind < AST_ELEMENT_COUNT);
 
-  struct AST_Node_t *node = malloc(sizeof(struct AST_Node_t));
+    struct AST_Node_t *node = mem_alloc(MemoryNamespaceAst, sizeof(struct AST_Node_t));
 
     if (node == NULL) {
         PANIC("failed to allocate AST node");
@@ -134,12 +134,12 @@ void AST_push_node(struct AST_Node_t *owner, struct AST_Node_t *child) {
     // if there are no children for now
     if (owner->child_count == 0) {
         DEBUG("Allocating new children array");
-    owner->children = malloc(sizeof(struct AST_Node_t *));
+        owner->children = mem_alloc(MemoryNamespaceAst, sizeof(struct AST_Node_t *));
 
     } else {
         DEBUG("Rellocating old children array");
         const size_t size = sizeof(struct AST_Node_t *) * (owner->child_count + 1);
-    owner->children = realloc(owner->children, size);
+        owner->children = mem_realloc(MemoryNamespaceAst, owner->children, size);
     }
 
     if (owner->children == NULL) {
@@ -214,24 +214,22 @@ void AST_delete_node(struct AST_Node_t *node) {
 
     DEBUG("Deleting AST node: %p", node);
 
-  if (node->children == NULL) {
-    return;
-  }
-
     if (node->parent != NULL) {
         [[maybe_unused]]
         const struct AST_Node_t* child = AST_detach_child(node->parent, node);
         assert(child == node);
     }
 
+    if (node->children != NULL) {
         for (size_t i = 0; i < node->child_count; i++) {
             // prevent detach of children node
             node->children[i]->parent = NULL;
             AST_delete_node(node->children[i]);
         }
+        mem_free(node->children);
+    }
 
-  free(node->children);
+    mem_free(node);
-  free(node);
 }
 
 static void AST_visit_nodes_recurse2(struct AST_Node_t *root,

src/cfg/opt.c

@@ -8,6 +8,7 @@
 #include <io/files.h>
 #include <assert.h>
 #include <toml.h>
+#include <mem/cache.h>
 
 static GHashTable* args = NULL;
 
@@ -17,8 +18,8 @@ static void clean(void) {
     g_hash_table_iter_init(&iter, args);
 
     while (g_hash_table_iter_next(&iter, &key, &value)) {
-        free(value);
+        mem_free(value);
-        free(key);
+        mem_free(key);
     }
 
     g_hash_table_destroy(args);
@@ -30,9 +31,9 @@ void parse_options(int argc, char* argv[]) {
     atexit(clean);
 
     for (int i = 0; i < argc; i++) {
-        Option* option = malloc(sizeof(Option));
+        Option* option = mem_alloc(MemoryNamespaceOpt, sizeof(Option));
         option->is_opt = g_str_has_prefix(argv[i], "--");
-        option->string = strdup(argv[i] + (option->is_opt ? 2 : 0));
+        option->string = mem_strdup(MemoryNamespaceOpt, argv[i] + (option->is_opt ? 2 : 0));
         option->index = i;
         option->value = NULL;
 
@@ -91,15 +92,15 @@ GArray* get_non_options_after(const char* command) {
 TargetConfig* default_target_config() {
     DEBUG("generating default target config...");
 
-    TargetConfig* config = malloc(sizeof(TargetConfig));
+    TargetConfig* config = mem_alloc(MemoryNamespaceOpt, sizeof(TargetConfig));
 
-    config->name = strdup("out");
+    config->name = mem_strdup(MemoryNamespaceOpt, "out");
     config->print_ast = false;
     config->print_asm = false;
     config->print_ir = false;
     config->mode = Application;
-    config->archive_directory = strdup("archive");
+    config->archive_directory = mem_strdup(MemoryNamespaceOpt, "archive");
-    config->output_directory = strdup("bin");
+    config->output_directory = mem_strdup(MemoryNamespaceOpt, "bin");
     config->optimization_level = 1;
     config->root_module = NULL;
 
@@ -141,7 +142,7 @@ TargetConfig* default_target_config_from_args() {
         const Option* opt = get_option("output");
 
         if (opt->value != NULL) {
-            config->name = strdup(opt->value);
+            config->name = mem_strdup(MemoryNamespaceOpt, (char*) opt->value);
         }
     }
 
@@ -155,7 +156,7 @@ TargetConfig* default_target_config_from_args() {
             print_message(Warning, "Got more than one file to compile, using first, ignoring others.");
         }
 
-        config->root_module = strdup( ((char**) files->data) [0]);
+        config->root_module = mem_strdup(MemoryNamespaceOpt, ((char**) files->data) [0]);
 
         g_array_free(files, TRUE);
     }
@@ -182,6 +183,7 @@ void print_help(void) {
         "    --debug              print debug logs (if not disabled at compile time)",
         "    --version            print the version",
         "    --help               print this hel dialog",
+        "    --print-memory-stats print statistics of the garbage collector"
     };
 
     for (unsigned int i = 0; i < sizeof(lines) / sizeof(const char *); i++) {
@@ -361,32 +363,32 @@ int load_project_config(ProjectConfig *config) {
 
 void delete_target_config(TargetConfig* config) {
     if (config->root_module != NULL) {
-        free(config->root_module);
+        mem_free(config->root_module);
     }
     if (config->archive_directory != NULL) {
-        free(config->archive_directory);
+        mem_free(config->archive_directory);
     }
     if (config->name != NULL) {
-        free(config->name);
+        mem_free(config->name);
     }
     if (config->output_directory != NULL) {
-        free(config->output_directory);
+        mem_free(config->output_directory);
     }
-    free(config);
+    mem_free(config);
 }
 
 void delete_project_config(ProjectConfig* config) {
     if (config->name != NULL) {
-        free(config->name);
+        mem_free(config->name);
     }
     if (config->authors != NULL) {
         g_array_free(config->authors, TRUE);
     }
     if (config->desc != NULL) {
-        free(config->desc);
+        mem_free(config->desc);
     }
     if (config->license != NULL) {
-        free(config->license);
+        mem_free(config->license);
     }
     if (config->targets != NULL) {
         GHashTableIter iter;
@@ -402,11 +404,11 @@ void delete_project_config(ProjectConfig* config) {
         g_hash_table_destroy(config->targets);
     }
 
-    free(config);
+    mem_free_from(MemoryNamespaceOpt, config);
 }
 
 ProjectConfig* default_project_config() {
-    ProjectConfig* config = malloc(sizeof(ProjectConfig));
+    ProjectConfig* config = mem_alloc(MemoryNamespaceOpt, sizeof(ProjectConfig));
 
     config->authors = NULL;
     config->name = NULL;

src/compiler.c

@@ -11,6 +11,7 @@
 #include <io/files.h>
 #include <assert.h>
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 extern void yyrestart(FILE *);
 
@@ -148,6 +149,9 @@ static void build_target(ModuleFileStack *unit, const TargetConfig *target) {
 
     AST_delete_node(ast);
 
+    mem_purge_namespace(MemoryNamespaceLex);
+    mem_purge_namespace(MemoryNamespaceAst);
+
     print_file_statistics(file);
 }
 

src/io/files.c

@@ -6,6 +6,7 @@
 #include <sys/log.h>
 #include <assert.h>
 #include <sys/col.h>
+#include <mem/cache.h>
 
 #ifdef __unix__
 
@@ -122,7 +123,7 @@ void print_diagnostic(ModuleFile *file, TokenLocation *location, Message kind, c
     printf("%s%s:%ld:%s %s%s:%s %s\n", BOLD, absolute_path, location->line_start, RESET, accent_color, kind_text, RESET,
            message);
 
-    free((void *) absolute_path);
+    mem_free((void *) absolute_path);
 
     const unsigned long int lines = location->line_end - location->line_start + 1;
 
@@ -307,7 +308,7 @@ int create_directory(const char *path) {
 }
 
 const char *get_last_error() {
-    return strdup(strerror(errno));
+    return mem_strdup(MemoryNamespaceIo, strerror(errno));
 }
 
 const char *get_absolute_path(const char *path) {
@@ -325,7 +326,7 @@ const char *get_absolute_path(const char *path) {
     _fullpath((char*) path, absolute_path, _MAX_PATH);
 #endif
 
-    return strdup(absolute_path);
+    return mem_strdup(MemoryNamespaceIo, absolute_path);
 }
 
 const char* make_file_path(const char* name, const char* ext, int count, ...) {

src/lex/lexer.l

@@ -3,6 +3,7 @@
     #include <yacc/parser.tab.h>
     #include <sys/log.h>
     #include <lex/util.h>
+    #include <mem/cache.h>
 
     int yyLineNumber = 1;
   
@@ -89,20 +90,20 @@
 "lineno" {DEBUG("\"%s\" tokenized with \'FunLineno\'", yytext); return(FunLineno);};
 "extsupport" {DEBUG("\"%s\" tokenized with \'FunExtsupport\'", yytext); return(FunExtsupport);};
 
-[0-9]+    {DEBUG("\"%s\" tokenized with \'ValInt\'", yytext); yylval.string = strdup(yytext); return(ValInt); };
+[0-9]+    {DEBUG("\"%s\" tokenized with \'ValInt\'", yytext); yylval.string = mem_strdup(MemoryNamespaceLex, yytext); return(ValInt); };
-[0-9]*\.[0-9]+ {DEBUG("\"%s\" tokenized with \'ValFloat\'", yytext); yylval.string = strdup(yytext); return(ValFloat);};
+[0-9]*\.[0-9]+ {DEBUG("\"%s\" tokenized with \'ValFloat\'", yytext); yylval.string = mem_strdup(MemoryNamespaceLex, yytext); return(ValFloat);};
-[a-zA-Z_0-9]+ {DEBUG("\"%s\" tokenized with \'Ident\'", yytext); yylval.string = strdup(yytext); return(Ident); };
+[a-zA-Z_0-9]+ {DEBUG("\"%s\" tokenized with \'Ident\'", yytext); yylval.string = mem_strdup(MemoryNamespaceLex, yytext); return(Ident); };
 
 \"([^\"\n])*\" {
     yytext = yytext +1;
     yytext[yyleng - 2] = 0;
     
-    DEBUG("\"%s\" tokenized with \'ValStr\'", yytext); yylval.string = strdup(yytext); return(ValStr);};
+    DEBUG("\"%s\" tokenized with \'ValStr\'", yytext); yylval.string = mem_strdup(MemoryNamespaceLex, yytext); return(ValStr);};
 \"\"\"[^\"]*\"\"\" {
     yytext = yytext +3;
     yytext[yyleng - 4] = 0;
 
-    DEBUG("\"%s\" tokenized with \'ValMultistr\'", yytext); yylval.string = strdup(yytext); return(ValMultistr);};
+    DEBUG("\"%s\" tokenized with \'ValMultistr\'", yytext); yylval.string = mem_strdup(MemoryNamespaceLex, yytext); return(ValMultistr);};
 [ \r\t] { /* ignore whitespace */ };
 . { return yytext[0]; /* passthrough unknown token, let parser handle the error */ };
 %%

src/lex/util.c

@@ -2,6 +2,8 @@
 #include <lex/util.h>
 #include <string.h>
 #include <stdlib.h>
+#include <glib.h>
+#include <mem/cache.h>
 
 // implementation based on:
 // https://github.com/sunxfancy/flex-bison-examples/blob/master/error-handling/ccalc.c
@@ -16,13 +18,8 @@ static int nTokenStart = 0;
 static int nTokenLength = 0;
 static int nTokenNextStart = 0;
 
-static void lex_deinit(void) {
-  free(buffer);
-}
-
 void lex_init(void) {
-  buffer = malloc(MAX_READ_BUFFER_SIZE);
+  buffer = mem_alloc(MemoryNamespaceStatic, MAX_READ_BUFFER_SIZE);
-  atexit(lex_deinit);
 }
 
 void lex_reset(void) {

src/main.c

@@ -5,6 +5,7 @@
 #include <lex/util.h>
 #include <cfg/opt.h>
 #include <compiler.h>
+#include <mem/cache.h>
 
 /**
  * @brief Log a debug message to inform about beginning exit procedures
@@ -17,6 +18,8 @@ void notify_exit(void) { DEBUG("Exiting gemstone..."); }
  *
  */
 void setup(int argc, char *argv[]) {
+    mem_init();
+
     // setup preample
     parse_options(argc, argv);
 
@@ -57,5 +60,9 @@ int main(int argc, char *argv[]) {
 
     run_compiler();
 
+    if (is_option_set("print-memory-stats")) {
+        print_memory_statistics();
+    }
+
     return 0;
 }

src/mem/cache.c

@@ -0,0 +1,271 @@
+//
+// Created by servostar on 6/5/24.
+//
+
+#include <mem/cache.h>
+#include <sys/log.h>
+#include <glib.h>
+#include <string.h>
+#include <assert.h>
+
+static GHashTable* namespaces = NULL;
+
+typedef struct MemoryNamespaceStatistic_t {
+    size_t bytes_allocated;
+    size_t allocation_count;
+    size_t reallocation_count;
+    size_t manual_free_count;
+    size_t faulty_reallocations;
+    size_t faulty_allocations;
+    size_t purged_free_count;
+} MemoryNamespaceStatistic;
+
+typedef struct MemoryNamespace_t {
+    MemoryNamespaceStatistic statistic;
+    GArray* blocks;
+} MemoryNamespace;
+
+typedef MemoryNamespace* MemoryNamespaceRef;
+
+static void namespace_statistics_print(MemoryNamespaceStatistic* memoryNamespaceStatistic, char* name) {
+    printf("Memory namespace statistics: `%s`\n", name);
+    printf("------------------------------\n");
+    printf("  allocated bytes:      %ld\n", memoryNamespaceStatistic->bytes_allocated);
+    printf("  allocations:          %ld\n", memoryNamespaceStatistic->allocation_count);
+    printf("  reallocations:        %ld\n", memoryNamespaceStatistic->reallocation_count);
+    printf("  frees:                %ld\n", memoryNamespaceStatistic->manual_free_count);
+    printf("  faulty allocations:   %ld\n", memoryNamespaceStatistic->faulty_allocations);
+    printf("  faulty reallocations: %ld\n", memoryNamespaceStatistic->faulty_reallocations);
+    printf("  purged allocations:   %ld\n", memoryNamespaceStatistic->purged_free_count);
+    printf("\n");
+}
+
+static void* namespace_malloc(MemoryNamespaceRef memoryNamespace, size_t size) {
+    assert(memoryNamespace != NULL);
+    assert(size != 0);
+
+    void* block = malloc(size);
+
+    if (block == NULL) {
+        memoryNamespace->statistic.faulty_allocations ++;
+    } else {
+        g_array_append_val(memoryNamespace->blocks, block);
+
+        memoryNamespace->statistic.allocation_count ++;
+        memoryNamespace->statistic.bytes_allocated += size;
+    }
+
+    return block;
+}
+
+static gboolean namespace_free(MemoryNamespaceRef memoryNamespace, void* block) {
+    for (guint i = 0; i < memoryNamespace->blocks->len; i++) {
+        void* current_block = g_array_index(memoryNamespace->blocks, void*, i);
+
+        if (current_block == block) {
+            assert(block != NULL);
+
+            free(block);
+            g_array_remove_index(memoryNamespace->blocks, i);
+
+            memoryNamespace->statistic.manual_free_count++;
+            return TRUE;
+        }
+    }
+    return FALSE;
+}
+
+static void* namespace_realloc(MemoryNamespaceRef memoryNamespace, void* block, size_t size) {
+    void* reallocated_block = NULL;
+
+    for (guint i = 0; i < memoryNamespace->blocks->len; i++) {
+        void* current_block = g_array_index(memoryNamespace->blocks, void*, i);
+
+        if (current_block == block) {
+            reallocated_block = realloc(block, size);
+
+            if (reallocated_block != NULL) {
+                g_array_index(memoryNamespace->blocks, void*, i) = reallocated_block;
+                memoryNamespace->statistic.bytes_allocated += size;
+                memoryNamespace->statistic.reallocation_count ++;
+            } else {
+                memoryNamespace->statistic.faulty_reallocations++;
+            }
+
+            break;
+        }
+    }
+
+    return reallocated_block;
+}
+
+static void namespace_delete(MemoryNamespaceRef memoryNamespace) {
+    g_array_free(memoryNamespace->blocks, TRUE);
+    free(memoryNamespace);
+}
+
+static void namespace_purge(MemoryNamespaceRef memoryNamespace) {
+
+    for (guint i = 0; i < memoryNamespace->blocks->len; i++) {
+        void* current_block = g_array_index(memoryNamespace->blocks, void*, i);
+
+        free(current_block);
+
+        memoryNamespace->statistic.purged_free_count ++;
+    }
+
+    g_array_remove_range(memoryNamespace->blocks, 0, memoryNamespace->blocks->len);
+}
+
+static MemoryNamespaceRef namespace_new() {
+    MemoryNamespaceRef memoryNamespace = malloc(sizeof(MemoryNamespace));
+
+    memoryNamespace->blocks = g_array_new(FALSE, FALSE, sizeof(void*));
+    memoryNamespace->statistic.bytes_allocated = 0;
+    memoryNamespace->statistic.allocation_count = 0;
+    memoryNamespace->statistic.manual_free_count = 0;
+    memoryNamespace->statistic.faulty_reallocations = 0;
+    memoryNamespace->statistic.faulty_allocations = 0;
+    memoryNamespace->statistic.purged_free_count = 0;
+    memoryNamespace->statistic.reallocation_count = 0;
+
+    return memoryNamespace;
+}
+
+static void cleanup() {
+    if (namespaces == NULL) {
+        printf("==> Memory cache was unused <==\n");
+        return;
+    }
+
+    GHashTableIter iter;
+    char* name = NULL;
+    MemoryNamespaceRef memoryNamespace = NULL;
+
+    g_hash_table_iter_init(&iter, namespaces);
+
+    while (g_hash_table_iter_next(&iter, (gpointer) &name, (gpointer) &memoryNamespace)) {
+        assert(name != NULL);
+        assert(memoryNamespace != NULL);
+
+        namespace_purge(memoryNamespace);
+        namespace_delete(memoryNamespace);
+    }
+}
+
+void mem_init() {
+    atexit(cleanup);
+}
+
+static MemoryNamespaceRef check_namespace(MemoryNamespaceName name) {
+    if (namespaces == NULL) {
+        namespaces = g_hash_table_new(g_str_hash, g_str_equal);
+    }
+
+    if (g_hash_table_contains(namespaces, name)) {
+
+        return g_hash_table_lookup(namespaces, name);
+
+    } else {
+        MemoryNamespaceRef namespace = namespace_new();
+
+        g_hash_table_insert(namespaces, name, namespace);
+
+        return namespace;
+    }
+}
+
+void *mem_alloc(MemoryNamespaceName name, size_t size) {
+    MemoryNamespaceRef cache = check_namespace(name);
+
+    if (cache == NULL) {
+        PANIC("memory namespace not created");
+    }
+
+    return namespace_malloc(cache, size);
+}
+
+void *mem_realloc(MemoryNamespaceName name, void *ptr, size_t size) {
+    MemoryNamespaceRef cache = check_namespace(name);
+
+    if (cache == NULL) {
+        PANIC("memory namespace not created");
+    }
+
+    return namespace_realloc(cache, ptr, size);
+}
+
+void mem_free_from(MemoryNamespaceName name, void *memory) {
+    MemoryNamespaceRef cache = check_namespace(name);
+
+    namespace_free(cache, memory);
+}
+
+void mem_free(void* memory) {
+    GHashTableIter iter;
+    char* name;
+    MemoryNamespaceRef memoryNamespace;
+
+    g_hash_table_iter_init(&iter, namespaces);
+    while (g_hash_table_iter_next(&iter, (gpointer) &name, (gpointer) &memoryNamespace)) {
+
+        if (namespace_free(memoryNamespace, memory)) {
+            break;
+        }
+    }
+}
+
+void mem_purge_namespace(MemoryNamespaceName name) {
+    if (g_hash_table_contains(namespaces, name)) {
+        MemoryNamespaceRef cache = g_hash_table_lookup(namespaces, name);
+
+        namespace_purge(cache);
+    } else {
+        PANIC("purging invalid namespace: %s", name);
+    }
+}
+
+char* mem_strdup(MemoryNamespaceName name, char* string) {
+    return mem_clone(name, string, strlen(string) + 1);
+}
+
+void* mem_clone(MemoryNamespaceName name, void* data, size_t size) {
+    void *clone = mem_alloc(name, size);
+
+    memcpy(clone, data, size);
+
+    return clone;
+}
+
+void print_memory_statistics() {
+    GHashTableIter iter;
+    char* name;
+    MemoryNamespaceRef memoryNamespace;
+
+    MemoryNamespaceStatistic total;
+    total.bytes_allocated = 0;
+    total.faulty_reallocations = 0;
+    total.faulty_allocations = 0;
+    total.manual_free_count = 0;
+    total.allocation_count = 0;
+    total.purged_free_count = 0;
+    total.reallocation_count = 0;
+
+    g_hash_table_iter_init(&iter, namespaces);
+    while (g_hash_table_iter_next(&iter, (gpointer) &name, (gpointer) &memoryNamespace)) {
+
+        namespace_statistics_print(&memoryNamespace->statistic, name);
+
+        total.bytes_allocated += memoryNamespace->statistic.bytes_allocated;
+        total.faulty_reallocations += memoryNamespace->statistic.faulty_reallocations;
+        total.faulty_allocations += memoryNamespace->statistic.faulty_allocations;
+        total.manual_free_count += memoryNamespace->statistic.manual_free_count;
+        total.allocation_count += memoryNamespace->statistic.allocation_count;
+        total.purged_free_count += memoryNamespace->statistic.purged_free_count;
+        total.reallocation_count += memoryNamespace->statistic.reallocation_count;
+    }
+
+    namespace_statistics_print(&total, "summary");
+
+    printf("Note: untracked are memory allocations from external libraries.\n");
+}

src/mem/cache.h

@@ -0,0 +1,89 @@
+//
+// Created by servostar on 6/5/24.
+//
+
+#ifndef GEMSTONE_CACHE_H
+#define GEMSTONE_CACHE_H
+
+#include <mem/cache.h>
+#include <stddef.h>
+
+typedef char* MemoryNamespaceName;
+
+#define MemoryNamespaceAst "AST"
+#define MemoryNamespaceLex "Lexer"
+#define MemoryNamespaceLog "Logging"
+#define MemoryNamespaceOpt "Options"
+#define MemoryNamespaceSet "SET"
+#define MemoryNamespaceLlvm "LLVM"
+#define MemoryNamespaceIo "I/O"
+#define MemoryNamespaceStatic "Static"
+
+/**
+ * @brief Initialize the garbage collector.
+ *        Must be done to ensure cleanup of memory.
+ */
+void mem_init();
+
+/**
+ * @brief Allocate a block of memory in the specified namespace.
+ * @attention Must only be freed with mem_free() or mem_free_from()
+ * @param name
+ * @param size
+ * @return pointer to the block
+ */
+void* mem_alloc(MemoryNamespaceName name, size_t size);
+
+/**
+ * @brief Reallocate a block of memory in the specified namespace.
+ * @attention Must only be freed with mem_free() or mem_free_from()
+ * @param name
+ * @param size
+ * @return pointer to the block
+ */
+void* mem_realloc(MemoryNamespaceName name, void *ptr, size_t size);
+
+/**
+ * @brief Free a block of memory from a specified namespace.
+ *        Invoking multiple times on the same pointer will do nothing.
+ * @param name
+ * @param memory
+ */
+void mem_free_from(MemoryNamespaceName name, void* memory);
+
+/**
+ * @brief Free a block of memory.
+ *        Invoking multiple times on the same pointer will do nothing.
+ * @attention In case the namespace of the block is known, consider using mem_free_from()
+ *            to avoid unnecessary overhead.
+ * @param name
+ * @param memory
+ */
+void mem_free(void* memory);
+
+/**
+ * @brief Delete all memory from the given namespace.
+ * @param name
+ */
+void mem_purge_namespace(MemoryNamespaceName name);
+
+/**
+ * @brief Duplicate the given string with memory in the given namespace.
+ * @param name
+ * @param string
+ * @return
+ */
+char* mem_strdup(MemoryNamespaceName name, char* string);
+
+/**
+ * @brief Duplicate the given block of data with memory in the given namespace.
+ * @param name
+ * @param data
+ * @param size
+ * @return
+ */
+void* mem_clone(MemoryNamespaceName name, void* data, size_t size);
+
+void print_memory_statistics();
+
+#endif //GEMSTONE_CACHE_H

src/sys/log.c

@@ -6,6 +6,7 @@
 #include <assert.h>
 #include <string.h>
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 static struct Logger_t {
     FILE** streams;
@@ -39,7 +40,7 @@ void log_register_stream(FILE* restrict stream)
 
     if (GlobalLogger.stream_count == 0)
     {
-        GlobalLogger.streams = (FILE**) malloc(sizeof(FILE*));
+        GlobalLogger.streams = (FILE**) mem_alloc(MemoryNamespaceLog, sizeof(FILE*));
         GlobalLogger.stream_count = 1;
 
         if (GlobalLogger.streams == NULL)
@@ -51,7 +52,7 @@ void log_register_stream(FILE* restrict stream)
     {
         GlobalLogger.stream_count++;
         size_t bytes = GlobalLogger.stream_count * sizeof(FILE*);
-        GlobalLogger.streams = (FILE**) realloc(GlobalLogger.streams, bytes);
+        GlobalLogger.streams = (FILE**) mem_realloc(MemoryNamespaceLog, GlobalLogger.streams, bytes);
 
         if (GlobalLogger.streams == NULL)
         {

tests/CMakeLists.txt

@@ -11,3 +11,4 @@ add_subdirectory(input_file)
 add_subdirectory(ast)
 add_subdirectory(glib)
 add_subdirectory(project)
+add_subdirectory(cache)

tests/ast/CMakeLists.txt

@@ -26,6 +26,7 @@ add_executable(ast_build_tree
         ${PROJECT_SOURCE_DIR}/src/io/files.c
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         ${PROJECT_SOURCE_DIR}/dep/tomlc99/toml.c
         build_tree.c)
 set_target_properties(ast_build_tree
@@ -47,6 +48,7 @@ add_executable(ast_print_node
         ${PROJECT_SOURCE_DIR}/src/io/files.c
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         ${PROJECT_SOURCE_DIR}/dep/tomlc99/toml.c
         print_node.c)
 set_target_properties(ast_print_node
@@ -68,6 +70,7 @@ add_executable(ast_graphviz
         ${PROJECT_SOURCE_DIR}/src/io/files.c
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         ${PROJECT_SOURCE_DIR}/dep/tomlc99/toml.c
         print_graphviz.c)
 set_target_properties(ast_graphviz

tests/ast/build_tree.c

@@ -4,6 +4,7 @@
 
 #include <ast/ast.h>
 #include <sys/log.h>
+#include <mem/cache.h>
 
 void generate_statement(const AST_NODE_PTR stmt) {
     const AST_NODE_PTR add = AST_new_node(empty_location(), AST_Add, NULL);
@@ -29,6 +30,7 @@ void generate_branch(const AST_NODE_PTR stmt) {
 }
 
 int main(void) {
+    mem_init();
 
     const AST_NODE_PTR root = AST_new_node(empty_location(), AST_Stmt, NULL);
 

tests/ast/gen_graph.c

@@ -4,8 +4,10 @@
 
 #include <ast/ast.h>
 #include <sys/log.h>
+#include <mem/cache.h>
 
 int main(void) {
+    mem_init();
 
     struct AST_Node_t* node = AST_new_node(empty_location(), AST_If, NULL);
 

tests/ast/print_graphviz.c

@@ -4,6 +4,7 @@
 
 #include <ast/ast.h>
 #include <sys/log.h>
+#include <mem/cache.h>
 
 void generate_statement(const AST_NODE_PTR stmt) {
     const AST_NODE_PTR add = AST_new_node(empty_location(), AST_Add, NULL);
@@ -29,7 +30,7 @@ void generate_branch(const AST_NODE_PTR stmt) {
 }
 
 int main(void) {
-
+    mem_init();
     AST_init();
 
     const AST_NODE_PTR root = AST_new_node(empty_location(), AST_Stmt, NULL);

tests/ast/print_node.c

@@ -3,9 +3,10 @@
 //
 
 #include <ast/ast.h>
+#include <mem/cache.h>
 
 int main(void) {
-
+    mem_init();
     AST_init();
 
     const AST_NODE_PTR node = AST_new_node(empty_location(), 0, "value");

tests/cache/CMakeLists.txt

@@ -0,0 +1,38 @@
+include(CTest)
+
+include_directories(${PROJECT_SOURCE_DIR}/src)
+
+# ------------------------------------------------ #
+# Setup Glib 2.0                                   #
+# ------------------------------------------------ #
+
+find_package(PkgConfig REQUIRED)
+pkg_search_module(GLIB REQUIRED IMPORTED_TARGET glib-2.0)
+include_directories(PRIVATE ${GLIB_INCLUDE_DIRS})
+
+# ------------------------------------------------ #
+# Setup TOML-C99                                   #
+# ------------------------------------------------ #
+
+include_directories(${PROJECT_SOURCE_DIR}/dep/tomlc99)
+
+# ------------------------------------------------------- #
+# CTEST 1
+# test the memory cache
+
+add_executable(cache
+        ${PROJECT_SOURCE_DIR}/src/sys/log.c
+        ${PROJECT_SOURCE_DIR}/src/sys/col.c
+        ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
+        ${PROJECT_SOURCE_DIR}/src/io/files.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
+        cache_test.c)
+set_target_properties(cache
+        PROPERTIES
+        OUTPUT_NAME "cache"
+        RUNTIME_OUTPUT_DIRECTORY ${GEMSTONE_BINARY_DIR}/tests/cache)
+target_link_libraries(cache PkgConfig::GLIB)
+target_link_libraries(cache tomlc99)
+add_test(NAME cache
+        WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
+        COMMAND ${GEMSTONE_BINARY_DIR}/tests/cache/cache)

tests/cache/cache_test.c

@@ -0,0 +1,28 @@
+//
+// Created by servostar on 6/5/24.
+//
+
+#include <sys/log.h>
+#include <sys/col.h>
+#include <cfg/opt.h>
+#include <mem/cache.h>
+
+int main(int argc, char* argv[]) {
+    mem_init();
+    parse_options(argc, argv);
+    log_init();
+    set_log_level(LOG_LEVEL_DEBUG);
+    col_init();
+
+    for (int i = 0; i < 3; i++) {
+        void* data = mem_alloc(MemoryNamespaceAst, 457);
+        mem_realloc(MemoryNamespaceAst, data, 200);
+        mem_free(data);
+    }
+
+    mem_purge_namespace(MemoryNamespaceOpt);
+
+    print_memory_statistics();
+
+    return 0;
+}

tests/logging/CMakeLists.txt

@@ -25,6 +25,7 @@ add_executable(logging_output
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
         ${PROJECT_SOURCE_DIR}/src/io/files.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         output.c)
 set_target_properties(logging_output
         PROPERTIES
@@ -45,6 +46,7 @@ add_executable(logging_panic
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
         ${PROJECT_SOURCE_DIR}/src/io/files.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         panic.c)
 set_target_properties(logging_panic
         PROPERTIES
@@ -65,6 +67,7 @@ add_executable(logging_streams
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
         ${PROJECT_SOURCE_DIR}/src/io/files.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         streams.c)
 set_target_properties(logging_streams
         PROPERTIES
@@ -85,6 +88,7 @@ add_executable(logging_level
         ${PROJECT_SOURCE_DIR}/src/sys/col.c
         ${PROJECT_SOURCE_DIR}/src/cfg/opt.c
         ${PROJECT_SOURCE_DIR}/src/io/files.c
+        ${PROJECT_SOURCE_DIR}/src/mem/cache.c
         level.c)
 set_target_properties(logging_level
         PROPERTIES

tests/logging/level.c

@@ -5,10 +5,12 @@
 #include "sys/log.h"
 #include <sys/col.h>
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 #define LOG_LEVEL LOG_LEVEL_WARNING
 
 int main(int argc, char* argv[]) {
+    mem_init();
     parse_options(argc, argv);
     log_init();
     set_log_level(LOG_LEVEL_DEBUG);

tests/logging/output.c

@@ -5,8 +5,10 @@
 #include "sys/log.h"
 #include <sys/col.h>
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 int main(int argc, char* argv[]) {
+    mem_init();
     parse_options(argc, argv);
     log_init();
     set_log_level(LOG_LEVEL_DEBUG);

tests/logging/panic.c

@@ -4,8 +4,10 @@
 
 #include "sys/log.h"
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 int main(int argc, char* argv[]) {
+    mem_init();
     parse_options(argc, argv);
     log_init();
     set_log_level(LOG_LEVEL_DEBUG);

tests/logging/streams.c

@@ -5,6 +5,7 @@
 #include "sys/log.h"
 #include <stdlib.h>
 #include <cfg/opt.h>
+#include <mem/cache.h>
 
 static FILE* file;
 
@@ -15,6 +16,7 @@ void close_file(void) {
 }
 
 int main(int argc, char* argv[]) {
+    mem_init();
     parse_options(argc, argv);
     log_init();
     set_log_level(LOG_LEVEL_DEBUG);