From a049f578af3a428ecd9144fb4a871950394a515d Mon Sep 17 00:00:00 2001
From: augustin64 <augustin.lucas64@gmail.com>
Date: Sun, 19 Feb 2023 16:28:34 +0100
Subject: [PATCH] Add print_memory & fix gree

---
 src/include/memory_management.h | 11 +++++++++++
 src/memory_management.c         | 23 +++++++++++++++++++++--
 src/memory_management.cu        | 23 +++++++++++++++++++++--
 3 files changed, 53 insertions(+), 4 deletions(-)

diff --git a/src/include/memory_management.h b/src/include/memory_management.h
index 70b13c6..99538f2 100644
--- a/src/include/memory_management.h
+++ b/src/include/memory_management.h
@@ -15,6 +15,7 @@ typedef struct Memory {
    void* cursor; // Current cursor
    size_t size; // Taille de la mémoire allouée
    int nb_alloc; // Nombre d'allocations dans le bloc
+   unsigned int id; // Nombre aléatoire permettant d'identifier le bloc plus facilement lors du débogage
    struct Memory* next; // Élément suivant
 } Memory;
 
@@ -47,6 +48,16 @@ int get_memory_blocks_number();
 int get_length(Memory* mem);
 
 
+/*
+* Appels récursifs de la fonction print_memory
+*/
+void print_memory_rec(Memory* mem);
+
+/*
+* Affiche les blocs actuels de la mémoire ainsi que leur utilisation
+*/
+void print_memory();
+
 
 /*
 * Créer un bloc de mémoire de taille size
diff --git a/src/memory_management.c b/src/memory_management.c
index 4852722..8d2d52a 100644
--- a/src/memory_management.c
+++ b/src/memory_management.c
@@ -35,6 +35,22 @@ int get_memory_blocks_number() {
     return get_length(memory);
 }
 
+void print_memory_rec(Memory* mem) {
+    if (!mem) {
+        return;
+    }
+    printf("==== %u ====\n", mem->id);
+    printf("plage d'addresses: %p-%p\n", mem->start, (void*)((intptr_t)mem->start +mem->size));
+    printf("in-use: %ld/%ld\n", ((intptr_t)mem->cursor - (intptr_t)mem->start), mem->size);
+    printf("allocations: %d\n\n", mem->nb_alloc);
+    print_memory_rec(mem->next);
+}
+
+
+void print_memory() {
+    printf(BLUE "==== MEMORY ====\n" RESET);
+    print_memory_rec(memory);
+}
 
 Memory* create_memory_block(size_t size) {
     Memory* mem = (Memory*)malloc(sizeof(Memory));
@@ -47,6 +63,7 @@ Memory* create_memory_block(size_t size) {
     mem->size = size;
     mem->nb_alloc = 0;
     mem->next = NULL;
+    mem->id = rand() %100000;
     
     return mem;
 }
@@ -72,11 +89,13 @@ void* allocate_memory(size_t size, Memory* mem) {
 
 Memory* free_memory(void* ptr, Memory* mem) {
     if (!mem) {
-        printf("[ERREUR] Le pointeur %p a déjà été libéré ou n'a jamais été alloué\n", ptr);
+        printf_error((char*)"Le pointeur ");
+        printf("%p a déjà été libéré ou n'a jamais été alloué\n", ptr);
         return mem;
     }
-    if ((intptr_t)mem->start <= (intptr_t)ptr && (intptr_t)ptr <= (intptr_t)mem->start + (intptr_t)mem->size) {
+    if (mem->start <= ptr && ptr < (void*)((intptr_t)mem->start + mem->size)) {
         mem->nb_alloc--;
+        // printf(GREEN "%p <= %p < %p\n" RESET, mem->start, ptr, (void*)((intptr_t)mem->start + mem->size));
         if (mem->nb_alloc == 0) {
             Memory* mem_next = mem->next;
             #ifdef __CUDACC__
diff --git a/src/memory_management.cu b/src/memory_management.cu
index 4852722..8d2d52a 100644
--- a/src/memory_management.cu
+++ b/src/memory_management.cu
@@ -35,6 +35,22 @@ int get_memory_blocks_number() {
     return get_length(memory);
 }
 
+void print_memory_rec(Memory* mem) {
+    if (!mem) {
+        return;
+    }
+    printf("==== %u ====\n", mem->id);
+    printf("plage d'addresses: %p-%p\n", mem->start, (void*)((intptr_t)mem->start +mem->size));
+    printf("in-use: %ld/%ld\n", ((intptr_t)mem->cursor - (intptr_t)mem->start), mem->size);
+    printf("allocations: %d\n\n", mem->nb_alloc);
+    print_memory_rec(mem->next);
+}
+
+
+void print_memory() {
+    printf(BLUE "==== MEMORY ====\n" RESET);
+    print_memory_rec(memory);
+}
 
 Memory* create_memory_block(size_t size) {
     Memory* mem = (Memory*)malloc(sizeof(Memory));
@@ -47,6 +63,7 @@ Memory* create_memory_block(size_t size) {
     mem->size = size;
     mem->nb_alloc = 0;
     mem->next = NULL;
+    mem->id = rand() %100000;
     
     return mem;
 }
@@ -72,11 +89,13 @@ void* allocate_memory(size_t size, Memory* mem) {
 
 Memory* free_memory(void* ptr, Memory* mem) {
     if (!mem) {
-        printf("[ERREUR] Le pointeur %p a déjà été libéré ou n'a jamais été alloué\n", ptr);
+        printf_error((char*)"Le pointeur ");
+        printf("%p a déjà été libéré ou n'a jamais été alloué\n", ptr);
         return mem;
     }
-    if ((intptr_t)mem->start <= (intptr_t)ptr && (intptr_t)ptr <= (intptr_t)mem->start + (intptr_t)mem->size) {
+    if (mem->start <= ptr && ptr < (void*)((intptr_t)mem->start + mem->size)) {
         mem->nb_alloc--;
+        // printf(GREEN "%p <= %p < %p\n" RESET, mem->start, ptr, (void*)((intptr_t)mem->start + mem->size));
         if (mem->nb_alloc == 0) {
             Memory* mem_next = mem->next;
             #ifdef __CUDACC__