Fixed some use-after-frees

1 files changed, 12 insertions, 0 deletions

diff --git a/src/lib/pqueue.c b/src/lib/pqueue.c
index cf0611f..6162cbc 100644
--- a/src/lib/pqueue.c
+++ b/src/lib/pqueue.c
@@ -160,6 +160,18 @@ void pqueue_change_priority(struct pqueue *q, pqueue_pri_t new_pri, void *d)
 		percolate_down(q, posn);
 }
 
+int pqueue_remove(struct pqueue *q, void *d)
+{
+	size_t posn = q->getpos(d);
+	q->d[posn] = q->d[--q->size];
+	if (q->cmppri(q->getpri(d), q->getpri(q->d[posn])))
+		bubble_up(q, posn);
+	else
+		percolate_down(q, posn);
+
+	return 0;
+}
+
 void *pqueue_pop(struct pqueue *q)
 {
 	void *head;