MappedFile: fix the insert range path not updating the root node

Author: jacobly0

src/link/MappedFile.zig

@@ -146,7 +146,7 @@ pub const Node = extern struct {
 
         pub fn hasMoved(ni: Node.Index, mf: *const MappedFile) bool {
             var parent_ni = ni;
-            while (parent_ni != .none) {
+            while (parent_ni != Node.Index.root) {
                 const parent = parent_ni.get(mf);
                 if (parent.flags.moved) return true;
                 parent_ni = parent.parent;
@@ -164,7 +164,7 @@ pub const Node = extern struct {
         }
         fn movedAssumeCapacity(ni: Node.Index, mf: *MappedFile) void {
             var parent_ni = ni;
-            while (parent_ni != .none) {
+            while (parent_ni != Node.Index.root) {
                 const parent_node = parent_ni.get(mf);
                 if (parent_node.flags.moved) return;
                 parent_ni = parent_node.parent;
@@ -585,56 +585,55 @@ fn resizeNode(mf: *MappedFile, gpa: std.mem.Allocator, ni: Node.Index, requested
                 continue;
             }
             const range_file_offset = ni.fileLocation(mf, false).offset + old_size;
-            retry: while (true) {
-                switch (linux.E.init(linux.fallocate(
-                    mf.file.handle,
-                    linux.FALLOC.FL_INSERT_RANGE,
-                    @intCast(range_file_offset),
-                    @intCast(range_size),
-                ))) {
-                    .SUCCESS => {
-                        var enclosing_ni = ni;
-                        while (enclosing_ni != .none) {
+            while (true) switch (linux.E.init(linux.fallocate(
+                mf.file.handle,
+                linux.FALLOC.FL_INSERT_RANGE,
+                @intCast(range_file_offset),
+                @intCast(range_size),
+            ))) {
+                .SUCCESS => {
+                    var enclosing_ni = ni;
+                    while (true) {
+                        try mf.ensureCapacityForSetLocation(gpa);
+                        const enclosing = enclosing_ni.get(mf);
+                        const enclosing_offset, const enclosing_size =
+                            enclosing.location().resolve(mf);
+                        enclosing_ni.setLocationAssumeCapacity(
+                            mf,
+                            enclosing_offset,
+                            enclosing_size + range_size,
+                        );
+                        if (enclosing_ni == Node.Index.root) break;
+                        var after_ni = enclosing.next;
+                        while (after_ni != .none) {
                             try mf.ensureCapacityForSetLocation(gpa);
-                            const enclosing = enclosing_ni.get(mf);
-                            const enclosing_offset, const enclosing_size =
-                                enclosing.location().resolve(mf);
-                            enclosing_ni.setLocationAssumeCapacity(
+                            const after = after_ni.get(mf);
+                            const after_offset, const after_size = after.location().resolve(mf);
+                            after_ni.setLocationAssumeCapacity(
                                 mf,
-                                enclosing_offset,
-                                enclosing_size + range_size,
+                                range_size + after_offset,
+                                after_size,
                             );
-                            var after_ni = enclosing.next;
-                            while (after_ni != .none) {
-                                try mf.ensureCapacityForSetLocation(gpa);
-                                const after = after_ni.get(mf);
-                                const after_offset, const after_size = after.location().resolve(mf);
-                                after_ni.setLocationAssumeCapacity(
-                                    mf,
-                                    range_size + after_offset,
-                                    after_size,
-                                );
-                                after_ni = after.next;
-                            }
-                            enclosing_ni = enclosing.parent;
+                            after_ni = after.next;
                         }
-                        return;
-                    },
-                    .INTR => continue :retry,
-                    .BADF, .FBIG, .INVAL => unreachable,
-                    .IO => return error.InputOutput,
-                    .NODEV => return error.NotFile,
-                    .NOSPC => return error.NoSpaceLeft,
-                    .NOSYS, .OPNOTSUPP => {
-                        mf.flags.fallocate_insert_range_unsupported = true;
-                        break :insert_range;
-                    },
-                    .PERM => return error.PermissionDenied,
-                    .SPIPE => return error.Unseekable,
-                    .TXTBSY => return error.FileBusy,
-                    else => |e| return std.posix.unexpectedErrno(e),
-                }
-            }
+                        enclosing_ni = enclosing.parent;
+                    }
+                    return;
+                },
+                .INTR => continue,
+                .BADF, .FBIG, .INVAL => unreachable,
+                .IO => return error.InputOutput,
+                .NODEV => return error.NotFile,
+                .NOSPC => return error.NoSpaceLeft,
+                .NOSYS, .OPNOTSUPP => {
+                    mf.flags.fallocate_insert_range_unsupported = true;
+                    break :insert_range;
+                },
+                .PERM => return error.PermissionDenied,
+                .SPIPE => return error.Unseekable,
+                .TXTBSY => return error.FileBusy,
+                else => |e| return std.posix.unexpectedErrno(e),
+            };
         }
         switch (node.next) {
             .none => {
@@ -728,7 +727,6 @@ fn moveRange(mf: *MappedFile, old_file_offset: u64, new_file_offset: u64, size:
     // delete the copy of this node at the old location
     if (is_linux and !mf.flags.fallocate_punch_hole_unsupported and
         size >= mf.flags.block_size.toByteUnits() * 2 - 1) while (true)
-    {
         switch (linux.E.init(linux.fallocate(
             mf.file.handle,
             linux.FALLOC.FL_PUNCH_HOLE | linux.FALLOC.FL_KEEP_SIZE,
@@ -749,8 +747,7 @@ fn moveRange(mf: *MappedFile, old_file_offset: u64, new_file_offset: u64, size:
             .SPIPE => return error.Unseekable,
             .TXTBSY => return error.FileBusy,
             else => |e| return std.posix.unexpectedErrno(e),
-        }
-    };
+        };
     @memset(mf.contents[@intCast(old_file_offset)..][0..@intCast(size)], 0);
 }