From e14e5db15a93dfb52b796d7715ccdf577cf0855a Mon Sep 17 00:00:00 2001
From: xamidev <121681048+xamidev@users.noreply.github.com>
Date: Tue, 10 Sep 2024 09:11:08 +0200
Subject: [PATCH] Fix: memory leak in naval

---
 src/programs/navalbattle.c | 16 +++++++++++++++-
 src/programs/navalbattle.h |  1 +
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/programs/navalbattle.c b/src/programs/navalbattle.c
index ad6c898..8f1bfb5 100644
--- a/src/programs/navalbattle.c
+++ b/src/programs/navalbattle.c
@@ -42,6 +42,9 @@ void program_navalbattle()
 		show_game_stats(grid, enemyGrid);
 	} while (check_victory(grid, enemyGrid) == 0);
 
+	free_grid(grid);
+	free_grid(enemyGrid);
+
 	return;	
 }
 
@@ -52,7 +55,6 @@ void init_battlefield(grid_t* grid[SIZE][SIZE])
 		for (size_t j=0; j<SIZE; j++)
 		{
 			grid[i][j] = (grid_t*)malloc(sizeof(grid_t));
-			serial_printf(3, "malloc'd x=%d y=%d at 0x%x", i, j, grid[i][j]);
 			grid[i][j]->x = i;
 			grid[i][j]->y = j;
 			grid[i][j]->role = 0;
@@ -61,6 +63,18 @@ void init_battlefield(grid_t* grid[SIZE][SIZE])
 	}
 }
 
+// To avoid memory leaks..
+void free_grid(grid_t* grid[SIZE][SIZE])
+{
+	for (size_t i=0; i<SIZE; i++)
+	{
+		for (size_t j=0; j<SIZE; j++)
+		{
+			free(grid[i][j]);
+		}
+	}
+}
+
 void show_ally_battlefield(grid_t* grid[SIZE][SIZE])
 {	
 	puts("\n*** Ally grid ***\n");
diff --git a/src/programs/navalbattle.h b/src/programs/navalbattle.h
index d75db9a..0314c63 100644
--- a/src/programs/navalbattle.h
+++ b/src/programs/navalbattle.h
@@ -22,6 +22,7 @@ typedef struct
 #define BUFFER_SIZE   16
 
 void init_battlefield(grid_t* grid[SIZE][SIZE]);
+void free_grid(grid_t* grid[SIZE][SIZE]);
 void show_ally_battlefield(grid_t* grid[SIZE][SIZE]);
 void placing_ally_ships(grid_t* grid[SIZE][SIZE]);
 void placing_enemy_ships(grid_t* grid[SIZE][SIZE]);