Fix memory leaks

1 files changed, 13 insertions, 0 deletions

diff --git a/priority_queue.c b/priority_queue.c
index d2517a6..d4be6e2 100644
--- a/priority_queue.c
+++ b/priority_queue.c
@@ -24,6 +24,7 @@ int ppq_insert(PositionPQ **ppq, Position pos, size_t priority) {
     }
     
     PositionPQ *n = ppq_new(pos, priority);
+
     if (start->priority > priority) {
         n->next = start;
         start = n;
@@ -34,11 +35,13 @@ int ppq_insert(PositionPQ **ppq, Position pos, size_t priority) {
 
     while(temp->next != NULL && temp->next->priority <= priority) {
         if (temp->pos.x == pos.x && temp->pos.y == pos.y && temp->priority <= priority) {
+            free(n);
             return 3; /* pos is already in ppq with a fine priority */
         }
         temp = temp->next;
     }
     if (temp->pos.x == pos.x && temp->pos.y == pos.y && temp->priority <= priority) {
+        free(n);
         return 3; /* pos is already in ppq with a fine priority */
     }
 
@@ -82,3 +85,13 @@ void ppq_print(PositionPQ *ppq) {
     }
     printf("%i - %li: %li, %li\n", i++, ppq->priority, ppq->pos.x, ppq->pos.y);
 }
+
+void ppq_free(PositionPQ *ppq) {
+    if (ppq == NULL) return;
+    while(ppq->next != NULL) {
+        PositionPQ *t = ppq;
+        ppq = ppq->next;
+        free(t);
+    }
+    if (ppq != NULL) free(ppq);
+}