Commit 78baa16da0

Literally Void <literallyvoid@gmail.com>
2020-09-10 05:49:49
Fix issue #6303: iterating empty PriorityQueue crashes
1 parent 749417a
Changed files (1)
lib/std/priority_queue.zig
@@ -195,7 +195,7 @@ pub fn PriorityQueue(comptime T: type) type {
             count: usize,
 
             pub fn next(it: *Iterator) ?T {
-                if (it.count > it.queue.len - 1) return null;
+                if (it.count >= it.queue.len) return null;
                 const out = it.count;
                 it.count += 1;
                 return it.queue.items[out];
@@ -428,3 +428,12 @@ test "std.PriorityQueue: remove at index" {
     expectEqual(queue.remove(), 3);
     expectEqual(queue.removeOrNull(), null);
 }
+
+test "std.PriorityQueue: iterator while empty" {
+    var queue = PQ.init(testing.allocator, lessThan);
+    defer queue.deinit();
+
+    var it = queue.iterator();
+
+    expectEqual(it.next(), null);
+}