From 862af7192c65034619a45cee8b3abddfc9a3f1f9 Mon Sep 17 00:00:00 2001
From: Isaac Shoebottom <ir.shoebottom@gmail.com>
Date: Mon, 4 Dec 2023 20:19:20 -0400
Subject: [PATCH] Minor code cleanup

---
 Assignment9/src/main.c | 1 +
 Assignment9/src/png.c  | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Assignment9/src/main.c b/Assignment9/src/main.c
index be53e1d..8292ead 100644
--- a/Assignment9/src/main.c
+++ b/Assignment9/src/main.c
@@ -3,6 +3,7 @@
 #include <limits.h>
 #include <string.h>
 #include <stdbool.h>
+#include <unistd.h>
 #include "png.h"
 
 int main(int argc, char *argv[]) {
diff --git a/Assignment9/src/png.c b/Assignment9/src/png.c
index 2ca3c62..1d998d9 100644
--- a/Assignment9/src/png.c
+++ b/Assignment9/src/png.c
@@ -139,7 +139,8 @@ void xor_data(png_chunk *chunk) {
 
 void destroy_chunks(png_chunk **chunks) {
 	unsigned int size = get_number_of_chunks(chunks);
-	for (int i = 0; i < size; i++) {
+	// +1 for IEND chunk
+	for (int i = 0; i < size + 1; i++) {
 		free(chunks[i]->data);
 		free(chunks[i]);
 	}