From b7c705d135c87d161d336d061319cc170ea4eb81 Mon Sep 17 00:00:00 2001
From: Santiago Lo Coco <slococo@itba.edu.ar>
Date: Mon, 8 Nov 2021 18:55:38 -0300
Subject: [PATCH] Free resources

Co-authored-by: Ezequiel Bellver <ebellver@itba.edu.ar>
Co-authored-by: Juan Barmasch <jbarmasch@itba.edu.ar>
---
 Kernel/interruptions/systemCallsDispatcher.c |  7 +++-
 Kernel/utils/include/schedulerLib.h          |  2 +
 Kernel/utils/scheduler.c                     | 39 ++++++++++++++++++++
 3 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/Kernel/interruptions/systemCallsDispatcher.c b/Kernel/interruptions/systemCallsDispatcher.c
index 1287ee7..41a7af4 100644
--- a/Kernel/interruptions/systemCallsDispatcher.c
+++ b/Kernel/interruptions/systemCallsDispatcher.c
@@ -3,6 +3,8 @@
 #include "systemCallsDispatcher.h"
 
 uint64_t systemCallsDispatcher(uint64_t rdi, uint64_t rsi, uint64_t rdx, uint64_t rcx, uint64_t r8, uint64_t r9) {
+    void *aux;
+
     switch (rdi) {
         case 0:
             return write(rsi, rdx, rcx);
@@ -22,8 +24,11 @@ uint64_t systemCallsDispatcher(uint64_t rdi, uint64_t rsi, uint64_t rdx, uint64_
         case 7:
             return (uint64_t) getSems();
         case 8:
-            return (uint64_t) pvPortMalloc((size_t) rsi);
+            aux = pvPortMalloc((size_t) rsi);
+            processMallocs(aux);
+            return (uint64_t) aux;
         case 9:
+            processFrees((void *) rsi);
             vPortFree((void *) rsi);
             break;
         case 10:
diff --git a/Kernel/utils/include/schedulerLib.h b/Kernel/utils/include/schedulerLib.h
index 813c4b1..56fa7db 100644
--- a/Kernel/utils/include/schedulerLib.h
+++ b/Kernel/utils/include/schedulerLib.h
@@ -23,5 +23,7 @@ void unblockIO();
 char getState(int pid);
 char isForeground();
 void wait();
+void processFrees(void *ptr);
+void processMallocs(void *ptr);
 
 #endif
\ No newline at end of file
diff --git a/Kernel/utils/scheduler.c b/Kernel/utils/scheduler.c
index e47a401..b8dd2ee 100644
--- a/Kernel/utils/scheduler.c
+++ b/Kernel/utils/scheduler.c
@@ -2,6 +2,8 @@
 // PVS-Studio Static Code Analyzer for C, C++ and C#: http://www.viva64.com
 #include "scheduler.h"
 
+#define MAX_MALLOCS 40
+
 enum states {
     READY = 0, DEAD, BLOCKED, WAITING, BLOCKEDIO
 };
@@ -21,6 +23,8 @@ typedef struct processCDT {
     int children;
     char backWait;
     uint64_t bPointer;
+    void ** mmRegs;
+    int mmRegsQty;
 } processCDT;
 
 typedef struct sleepCDT {
@@ -150,6 +154,9 @@ int enqueueProcess(void (*fn)(int, char **), char foreground, int argc, char *ar
     process->children = 0;
     process->backWait = 0;
     process->bPointer = (uint64_t) auxi;
+    void * mmRegs[MAX_MALLOCS];
+    process->mmRegs = mmRegs;
+    process->mmRegsQty = 0;
 
     process->rsp = _initialize_stack_frame(fn, rbp, argc, argv);
 
@@ -342,6 +349,10 @@ char kill(int pid) {
 
     vPortFree(del->fd);
     vPortFree(del->name);
+    for (int i = 0; i < del->mmRegsQty; i++) {
+        vPortFree(del->mmRegs[i]);
+    }
+
     // vPortFree((void *) ((uint64_t) del->rbp - STACK_SIZE));
     vPortFree((void *) del->bPointer);
     del->state = DEAD;
@@ -353,6 +364,34 @@ char kill(int pid) {
     return EXIT_SUCCESS;
 }
 
+void processMallocs(void *ptr) {
+    if (currentProcess == NULL)
+        return;
+    
+    currentProcess->mmRegs[currentProcess->mmRegsQty++] = ptr;
+}
+
+void processFrees(void *ptr) {
+    if (currentProcess == NULL)
+        return;
+    
+    char flag = 0;
+    for (int i = 0; i < currentProcess->mmRegsQty; i++) {
+        if (currentProcess->mmRegs[i] == ptr) {
+            flag = 1;
+        }
+        if (flag) {
+            if (i != currentProcess->mmRegsQty - 1) {
+                currentProcess->mmRegs[i] = currentProcess->mmRegs[i + 1];
+            }
+            else currentProcess->mmRegs[i] = NULL;
+        }
+        
+    }
+    if (flag)
+        currentProcess->mmRegsQty--;
+}
+
 int getFdOut() {
     if (currentProcess == NULL)
         return EXIT_FAILURE;