authorgravatar for git@vexu.euVeikka Tuominen <git@vexu.eu> 2023-06-10 05:09:34+03:00
committergravatar for noreply@github.comGitHub <noreply@github.com> 2023-06-10 05:09:34+03:00
log2094d98694b0336e2f37131abe34d0573b6b0633
tree4c1ca27d92f7c705b594267d6b31e4b6f67e8746
parentf04e65bc09ab77eafc317b2d2981970509aae6af
parentaf835111fa1397578392a4774454ac0d71eca77d
signaturebadge-question-mark Signed by PGP key 4AEE18F83AFDEB23

Merge pull request #15980 from squeek502/walker-fixes

A few `IterableDir.Walker`/`Iterator` fixes

1 files changed, 13 insertions(+), 2 deletions(-)

lib/std/fs.zig+13-2
......@@ -662,6 +662,7 @@ pub const IterableDir = struct {
662662 .NOTDIR => unreachable,
663663 .NOENT => return error.DirNotFound, // The directory being iterated was deleted during iteration.
664664 .INVAL => return error.Unexpected, // Linux may in some cases return EINVAL when reading /proc/$PID/net.
665 .ACCES => return error.AccessDenied, // Do not have permission to iterate this directory.
665666 else => |err| return os.unexpectedErrno(err),
666667 }
667668 if (rc == 0) return null;
......@@ -957,7 +958,17 @@ pub const IterableDir = struct {
957958 var top = &self.stack.items[self.stack.items.len - 1];
958959 var containing = top;
959960 var dirname_len = top.dirname_len;
960 if (try top.iter.next()) |base| {
961 if (top.iter.next() catch |err| {
962 // If we get an error, then we want the user to be able to continue
963 // walking if they want, which means that we need to pop the directory
964 // that errored from the stack. Otherwise, all future `next` calls would
965 // likely just fail with the same error.
966 var item = self.stack.pop();
967 if (self.stack.items.len != 0) {
968 item.iter.dir.close();
969 }
970 return err;
971 }) |base| {
961972 self.name_buffer.shrinkRetainingCapacity(dirname_len);
962973 if (self.name_buffer.items.len != 0) {
963974 try self.name_buffer.append(path.sep);
......@@ -972,7 +983,7 @@ pub const IterableDir = struct {
972983 {
973984 errdefer new_dir.close();
974985 try self.stack.append(StackItem{
975 .iter = new_dir.iterate(),
986 .iter = new_dir.iterateAssumeFirstIteration(),
976987 .dirname_len = self.name_buffer.items.len,
977988 });
978989 top = &self.stack.items[self.stack.items.len - 1];