Further memory leak fixes

1 files changed, 12 insertions, 2 deletions

diff --git a/main.c b/main.c
index 0f99ad3..f0803ca 100644
--- a/main.c
+++ b/main.c
@@ -59,7 +59,7 @@ int main(int argc, char *argv[])
 
   if(bitmap.error)
   {
-    printf("Error reading file");
+    printf("Error reading file\n");
     return 1;
   }
 
@@ -131,10 +131,20 @@ int main(int argc, char *argv[])
   //Cleanup
 
 
-  for(int i = 0; i < size_y; i++)
+  for(int i = 0; i < size_x; i++)
     free (ascii_buff[i]);
   free(ascii_buff);
 
+  for(int i = 0; i < bitmap.x; i++)
+  {
+    free(bitmap.R[i]);
+    free(bitmap.G[i]);
+    free(bitmap.B[i]);
+  }
+  free(bitmap.R);
+  free(bitmap.G);
+  free(bitmap.B);
+
   fclose(out);
 
   return 0;