From: Kristian Høgsberg <krh@redhat.com>
Date: Sat, 21 Jun 2008 00:57:51 +0000 (-0400)
Subject: Fix package queries for empty sets.
X-Git-Tag: 0.1~120
X-Git-Url: http://project.juiblex.co.uk/git/?a=commitdiff_plain;h=b8ddfec6c21dffca5a867aeb3bc9d873aeeb4f56;p=razor2.git%2F.git

Fix package queries for empty sets.
---

diff --git a/librazor/iterator.c b/librazor/iterator.c
index 909b954..addbe68 100644
--- a/librazor/iterator.c
+++ b/librazor/iterator.c
@@ -240,13 +240,15 @@ razor_package_query_finish(struct razor_package_query *pq)
 	struct razor_package_iterator *pi;
 	struct razor_set *set;
 	struct list *index;
-	int i, j, count;
+	int i, j;
 
 	set = pq->set;
-	count = set->packages.size / sizeof(struct razor_package);
-	index = zalloc(pq->count * sizeof *index);
+	if (pq->count > 0)
+		index = zalloc(pq->count * sizeof *index);
+	else
+		index = NULL;
 
-	for (i = 0, j = 0; i < count; i++) {
+	for (i = 0, j = 0; i < pq->count; i++) {
 		if (!pq->vector[i])
 			continue;
 
@@ -256,6 +258,7 @@ razor_package_query_finish(struct razor_package_query *pq)
 		j++;
 	}
 
+	free(pq->vector);
 	free(pq);
 
 	pi = razor_package_iterator_create_with_index(set, index);