]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - libbcachefs/fsck.c
Update bcachefs sources to 70fa0c1ff4 fixup! bcachefs: Btree key cache improvements
[bcachefs-tools-debian] / libbcachefs / fsck.c
index 9519ced976f2eb8f62382de4188e8c9d53d8e7fe..f4f0e0cec85d7c536fa8c242fed641dcf801fb5e 100644 (file)
@@ -3,6 +3,7 @@
 #include "bcachefs.h"
 #include "bkey_buf.h"
 #include "btree_update.h"
+#include "darray.h"
 #include "dirent.h"
 #include "error.h"
 #include "fs-common.h"
 
 #define QSTR(n) { { { .len = strlen(n) } }, .name = n }
 
+/*
+ * XXX: this is handling transaction restarts without returning
+ * -BCH_ERR_transaction_restart_nested, this is not how we do things anymore:
+ */
 static s64 bch2_count_inode_sectors(struct btree_trans *trans, u64 inum,
                                    u32 snapshot)
 {
@@ -133,11 +138,11 @@ static int lookup_first_inode(struct btree_trans *trans, u64 inode_nr,
                goto err;
        }
 
-       ret = bch2_inode_unpack(bkey_s_c_to_inode(k), inode);
+       ret = bch2_inode_unpack(k, inode);
 err:
-       if (ret && ret != -EINTR)
-               bch_err(trans->c, "error %i fetching inode %llu",
-                       ret, inode_nr);
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(trans->c, "error fetching inode %llu: %s",
+                       inode_nr, bch2_err_str(ret));
        bch2_trans_iter_exit(trans, &iter);
        return ret;
 }
@@ -157,15 +162,15 @@ static int __lookup_inode(struct btree_trans *trans, u64 inode_nr,
        if (ret)
                goto err;
 
-       ret = k.k->type == KEY_TYPE_inode
-               ? bch2_inode_unpack(bkey_s_c_to_inode(k), inode)
+       ret = bkey_is_inode(k.k)
+               ? bch2_inode_unpack(k, inode)
                : -ENOENT;
        if (!ret)
                *snapshot = iter.pos.snapshot;
 err:
-       if (ret && ret != -EINTR)
-               bch_err(trans->c, "error %i fetching inode %llu:%u",
-                       ret, inode_nr, *snapshot);
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(trans->c, "error fetching inode %llu:%u: %s",
+                       inode_nr, *snapshot, bch2_err_str(ret));
        bch2_trans_iter_exit(trans, &iter);
        return ret;
 }
@@ -219,35 +224,39 @@ static int write_inode(struct btree_trans *trans,
                       struct bch_inode_unpacked *inode,
                       u32 snapshot)
 {
-       int ret = __bch2_trans_do(trans, NULL, NULL,
+       int ret = commit_do(trans, NULL, NULL,
                                  BTREE_INSERT_NOFAIL|
                                  BTREE_INSERT_LAZY_RW,
                                  __write_inode(trans, inode, snapshot));
        if (ret)
-               bch_err(trans->c, "error in fsck: error %i updating inode", ret);
+               bch_err(trans->c, "error in fsck: error updating inode: %s",
+                       bch2_err_str(ret));
        return ret;
 }
 
 static int fsck_inode_rm(struct btree_trans *trans, u64 inum, u32 snapshot)
 {
+       struct bch_fs *c = trans->c;
        struct btree_iter iter = { NULL };
        struct bkey_i_inode_generation delete;
        struct bch_inode_unpacked inode_u;
        struct bkey_s_c k;
        int ret;
 
-       ret   = bch2_btree_delete_range_trans(trans, BTREE_ID_extents,
-                                             SPOS(inum, 0, snapshot),
-                                             SPOS(inum, U64_MAX, snapshot),
-                                             0, NULL) ?:
-               bch2_btree_delete_range_trans(trans, BTREE_ID_dirents,
-                                             SPOS(inum, 0, snapshot),
-                                             SPOS(inum, U64_MAX, snapshot),
-                                             0, NULL) ?:
-               bch2_btree_delete_range_trans(trans, BTREE_ID_xattrs,
-                                             SPOS(inum, 0, snapshot),
-                                             SPOS(inum, U64_MAX, snapshot),
-                                             0, NULL);
+       do {
+               ret   = bch2_btree_delete_range_trans(trans, BTREE_ID_extents,
+                                                     SPOS(inum, 0, snapshot),
+                                                     SPOS(inum, U64_MAX, snapshot),
+                                                     0, NULL) ?:
+                       bch2_btree_delete_range_trans(trans, BTREE_ID_dirents,
+                                                     SPOS(inum, 0, snapshot),
+                                                     SPOS(inum, U64_MAX, snapshot),
+                                                     0, NULL) ?:
+                       bch2_btree_delete_range_trans(trans, BTREE_ID_xattrs,
+                                                     SPOS(inum, 0, snapshot),
+                                                     SPOS(inum, U64_MAX, snapshot),
+                                                     0, NULL);
+       } while (ret == -BCH_ERR_transaction_restart_nested);
        if (ret)
                goto err;
 retry:
@@ -261,22 +270,19 @@ retry:
        if (ret)
                goto err;
 
-       if (k.k->type != KEY_TYPE_inode) {
-               bch2_fs_inconsistent(trans->c,
+       if (!bkey_is_inode(k.k)) {
+               bch2_fs_inconsistent(c,
                                     "inode %llu:%u not found when deleting",
                                     inum, snapshot);
                ret = -EIO;
                goto err;
        }
 
-       bch2_inode_unpack(bkey_s_c_to_inode(k), &inode_u);
+       bch2_inode_unpack(k, &inode_u);
 
        /* Subvolume root? */
-       if (inode_u.bi_subvol) {
-               ret = bch2_subvolume_delete(trans, inode_u.bi_subvol);
-               if (ret)
-                       goto err;
-       }
+       if (inode_u.bi_subvol)
+               bch_warn(c, "deleting inode %llu marked as unlinked, but also a subvolume root!?", inode_u.bi_inum);
 
        bkey_inode_generation_init(&delete.k_i);
        delete.k.p = iter.pos;
@@ -287,10 +293,10 @@ retry:
                                BTREE_INSERT_NOFAIL);
 err:
        bch2_trans_iter_exit(trans, &iter);
-       if (ret == -EINTR)
+       if (bch2_err_matches(ret, BCH_ERR_transaction_restart))
                goto retry;
 
-       return ret;
+       return ret ?: -BCH_ERR_transaction_restart_nested;
 }
 
 static int __remove_dirent(struct btree_trans *trans, struct bpos pos)
@@ -303,15 +309,19 @@ static int __remove_dirent(struct btree_trans *trans, struct bpos pos)
 
        ret = lookup_first_inode(trans, pos.inode, &dir_inode);
        if (ret)
-               return ret;
+               goto err;
 
        dir_hash_info = bch2_hash_info_init(c, &dir_inode);
 
        bch2_trans_iter_init(trans, &iter, BTREE_ID_dirents, pos, BTREE_ITER_INTENT);
 
        ret = bch2_hash_delete_at(trans, bch2_dirent_hash_desc,
-                                 &dir_hash_info, &iter, 0);
+                                 &dir_hash_info, &iter,
+                                 BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE);
        bch2_trans_iter_exit(trans, &iter);
+err:
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -346,8 +356,8 @@ static int lookup_lostfound(struct btree_trans *trans, u32 subvol,
                goto create_lostfound;
        }
 
-       if (ret && ret != -EINTR)
-               bch_err(c, "error looking up lost+found: %i", ret);
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "error looking up lost+found: %s", bch2_err_str(ret));
        if (ret)
                return ret;
 
@@ -369,8 +379,8 @@ create_lostfound:
                                lostfound, &lostfound_str,
                                0, 0, S_IFDIR|0700, 0, NULL, NULL,
                                (subvol_inum) { }, 0);
-       if (ret && ret != -EINTR)
-               bch_err(c, "error creating lost+found: %i", ret);
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "error creating lost+found: %s", bch2_err_str(ret));
        return ret;
 }
 
@@ -429,13 +439,13 @@ static int reattach_inode(struct btree_trans *trans,
                          struct bch_inode_unpacked *inode,
                          u32 inode_snapshot)
 {
-       int ret = __bch2_trans_do(trans, NULL, NULL,
+       int ret = commit_do(trans, NULL, NULL,
                                  BTREE_INSERT_LAZY_RW|
                                  BTREE_INSERT_NOFAIL,
                        __reattach_inode(trans, inode, inode_snapshot));
        if (ret) {
-               bch_err(trans->c, "error %i reattaching inode %llu",
-                       ret, inode->bi_inum);
+               bch_err(trans->c, "error reattaching inode %llu: %s",
+                       inode->bi_inum, bch2_err_str(ret));
                return ret;
        }
 
@@ -466,19 +476,82 @@ out:
        return ret;
 }
 
-static int snapshots_seen_update(struct bch_fs *c, struct snapshots_seen *s, struct bpos pos)
+struct snapshots_seen_entry {
+       u32                             id;
+       u32                             equiv;
+};
+
+struct snapshots_seen {
+       struct bpos                     pos;
+       DARRAY(struct snapshots_seen_entry) ids;
+};
+
+static inline void snapshots_seen_exit(struct snapshots_seen *s)
+{
+       darray_exit(&s->ids);
+}
+
+static inline void snapshots_seen_init(struct snapshots_seen *s)
 {
-       pos.snapshot = snapshot_t(c, pos.snapshot)->equiv;
+       memset(s, 0, sizeof(*s));
+}
+
+static int snapshots_seen_add(struct bch_fs *c, struct snapshots_seen *s, u32 id)
+{
+       struct snapshots_seen_entry *i, n = { id, id };
+       int ret;
+
+       darray_for_each(s->ids, i) {
+               if (n.equiv < i->equiv)
+                       break;
+
+               if (i->equiv == n.equiv) {
+                       bch_err(c, "%s(): adding duplicate snapshot", __func__);
+                       return -EINVAL;
+               }
+       }
+
+       ret = darray_insert_item(&s->ids, i - s->ids.data, n);
+       if (ret)
+               bch_err(c, "error reallocating snapshots_seen table (size %zu)",
+                       s->ids.size);
+       return ret;
+}
+
+static int snapshots_seen_update(struct bch_fs *c, struct snapshots_seen *s,
+                                enum btree_id btree_id, struct bpos pos)
+{
+       struct snapshots_seen_entry *i, n = {
+               .id     = pos.snapshot,
+               .equiv  = bch2_snapshot_equiv(c, pos.snapshot),
+       };
+       int ret = 0;
 
        if (bkey_cmp(s->pos, pos))
-               s->nr = 0;
+               s->ids.nr = 0;
+
+       pos.snapshot = n.equiv;
        s->pos = pos;
 
-       /* Might get called multiple times due to lock restarts */
-       if (s->nr && s->d[s->nr - 1] == pos.snapshot)
-               return 0;
+       darray_for_each(s->ids, i)
+               if (i->equiv == n.equiv) {
+                       if (fsck_err_on(i->id != n.id, c,
+                                       "snapshot deletion did not run correctly:\n"
+                                       "  duplicate keys in btree %s at %llu:%llu snapshots %u, %u (equiv %u)\n",
+                                       bch2_btree_ids[btree_id],
+                                       pos.inode, pos.offset,
+                                       i->id, n.id, n.equiv))
+                               return -BCH_ERR_need_snapshot_cleanup;
 
-       return snapshots_seen_add(c, s, pos.snapshot);
+                       return 0;
+               }
+
+       ret = darray_push(&s->ids, n);
+       if (ret)
+               bch_err(c, "error reallocating snapshots_seen table (size %zu)",
+                       s->ids.size);
+fsck_err:
+       return ret;
 }
 
 /**
@@ -491,15 +564,15 @@ static bool key_visible_in_snapshot(struct bch_fs *c, struct snapshots_seen *see
                                    u32 id, u32 ancestor)
 {
        ssize_t i;
+       u32 top = seen->ids.nr ? seen->ids.data[seen->ids.nr - 1].equiv : 0;
 
        BUG_ON(id > ancestor);
-
-       id              = snapshot_t(c, id)->equiv;
-       ancestor        = snapshot_t(c, ancestor)->equiv;
+       BUG_ON(!bch2_snapshot_is_equiv(c, id));
+       BUG_ON(!bch2_snapshot_is_equiv(c, ancestor));
 
        /* @ancestor should be the snapshot most recently added to @seen */
-       BUG_ON(!seen->nr || seen->d[seen->nr - 1] != ancestor);
-       BUG_ON(seen->pos.snapshot != ancestor);
+       BUG_ON(ancestor != seen->pos.snapshot);
+       BUG_ON(ancestor != top);
 
        if (id == ancestor)
                return true;
@@ -507,11 +580,11 @@ static bool key_visible_in_snapshot(struct bch_fs *c, struct snapshots_seen *see
        if (!bch2_snapshot_is_ancestor(c, id, ancestor))
                return false;
 
-       for (i = seen->nr - 2;
-            i >= 0 && seen->d[i] >= id;
+       for (i = seen->ids.nr - 2;
+            i >= 0 && seen->ids.data[i].equiv >= id;
             --i)
-               if (bch2_snapshot_is_ancestor(c, id, seen->d[i]) &&
-                   bch2_snapshot_is_ancestor(c, seen->d[i], ancestor))
+               if (bch2_snapshot_is_ancestor(c, id, seen->ids.data[i].equiv) &&
+                   bch2_snapshot_is_ancestor(c, seen->ids.data[i].equiv, ancestor))
                        return false;
 
        return true;
@@ -536,27 +609,27 @@ static int ref_visible(struct bch_fs *c, struct snapshots_seen *s,
                : bch2_snapshot_is_ancestor(c, src, dst);
 }
 
-#define for_each_visible_inode(_c, _s, _w, _snapshot, _i)      \
-       for (_i = (_w)->d; _i < (_w)->d + (_w)->nr && (_i)->snapshot <= (_snapshot); _i++)\
+#define for_each_visible_inode(_c, _s, _w, _snapshot, _i)                              \
+       for (_i = (_w)->inodes.data; _i < (_w)->inodes.data + (_w)->inodes.nr &&        \
+            (_i)->snapshot <= (_snapshot); _i++)                                       \
                if (key_visible_in_snapshot(_c, _s, _i->snapshot, _snapshot))
 
+struct inode_walker_entry {
+       struct bch_inode_unpacked inode;
+       u32                     snapshot;
+       u64                     count;
+};
+
 struct inode_walker {
        bool                            first_this_inode;
        u64                             cur_inum;
 
-       size_t                          nr;
-       size_t                          size;
-       struct inode_walker_entry {
-               struct bch_inode_unpacked inode;
-               u32                     snapshot;
-               u64                     count;
-       } *d;
+       DARRAY(struct inode_walker_entry) inodes;
 };
 
 static void inode_walker_exit(struct inode_walker *w)
 {
-       kfree(w->d);
-       w->d = NULL;
+       darray_exit(&w->inodes);
 }
 
 static struct inode_walker inode_walker_init(void)
@@ -564,40 +637,17 @@ static struct inode_walker inode_walker_init(void)
        return (struct inode_walker) { 0, };
 }
 
-static int inode_walker_realloc(struct inode_walker *w)
-{
-       if (w->nr == w->size) {
-               size_t new_size = max_t(size_t, 8UL, w->size * 2);
-               void *d = krealloc(w->d, new_size * sizeof(w->d[0]),
-                                  GFP_KERNEL);
-               if (!d)
-                       return -ENOMEM;
-
-               w->d = d;
-               w->size = new_size;
-       }
-
-       return 0;
-}
-
 static int add_inode(struct bch_fs *c, struct inode_walker *w,
-                    struct bkey_s_c_inode inode)
+                    struct bkey_s_c inode)
 {
        struct bch_inode_unpacked u;
-       int ret;
-
-       ret = inode_walker_realloc(w);
-       if (ret)
-               return ret;
 
        BUG_ON(bch2_inode_unpack(inode, &u));
 
-       w->d[w->nr++] = (struct inode_walker_entry) {
+       return darray_push(&w->inodes, ((struct inode_walker_entry) {
                .inode          = u,
-               .snapshot       = snapshot_t(c, inode.k->p.snapshot)->equiv,
-       };
-
-       return 0;
+               .snapshot       = bch2_snapshot_equiv(c, inode.k->p.snapshot),
+       }));
 }
 
 static int __walk_inode(struct btree_trans *trans,
@@ -606,25 +656,26 @@ static int __walk_inode(struct btree_trans *trans,
        struct bch_fs *c = trans->c;
        struct btree_iter iter;
        struct bkey_s_c k;
-       unsigned i, ancestor_pos;
+       u32 restart_count = trans->restart_count;
+       unsigned i;
        int ret;
 
-       pos.snapshot = snapshot_t(c, pos.snapshot)->equiv;
+       pos.snapshot = bch2_snapshot_equiv(c, pos.snapshot);
 
        if (pos.inode == w->cur_inum) {
                w->first_this_inode = false;
                goto lookup_snapshot;
        }
 
-       w->nr = 0;
+       w->inodes.nr = 0;
 
        for_each_btree_key(trans, iter, BTREE_ID_inodes, POS(0, pos.inode),
                           BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
                if (k.k->p.offset != pos.inode)
                        break;
 
-               if (k.k->type == KEY_TYPE_inode)
-                       add_inode(c, w, bkey_s_c_to_inode(k));
+               if (bkey_is_inode(k.k))
+                       add_inode(c, w, k);
        }
        bch2_trans_iter_exit(trans, &iter);
 
@@ -633,27 +684,33 @@ static int __walk_inode(struct btree_trans *trans,
 
        w->cur_inum             = pos.inode;
        w->first_this_inode     = true;
+
+       if (trans_was_restarted(trans, restart_count))
+               return -BCH_ERR_transaction_restart_nested;
+
 lookup_snapshot:
-       for (i = 0; i < w->nr; i++)
-               if (bch2_snapshot_is_ancestor(c, pos.snapshot, w->d[i].snapshot))
+       for (i = 0; i < w->inodes.nr; i++)
+               if (bch2_snapshot_is_ancestor(c, pos.snapshot, w->inodes.data[i].snapshot))
                        goto found;
        return INT_MAX;
 found:
-       BUG_ON(pos.snapshot > w->d[i].snapshot);
+       BUG_ON(pos.snapshot > w->inodes.data[i].snapshot);
 
-       if (pos.snapshot != w->d[i].snapshot) {
-               ancestor_pos = i;
+       if (pos.snapshot != w->inodes.data[i].snapshot) {
+               struct inode_walker_entry e = w->inodes.data[i];
 
-               while (i && w->d[i - 1].snapshot > pos.snapshot)
+               e.snapshot = pos.snapshot;
+               e.count = 0;
+
+               bch_info(c, "have key for inode %llu:%u but have inode in ancestor snapshot %u",
+                        pos.inode, pos.snapshot, w->inodes.data[i].snapshot);
+
+               while (i && w->inodes.data[i - 1].snapshot > pos.snapshot)
                        --i;
 
-               ret = inode_walker_realloc(w);
+               ret = darray_insert_item(&w->inodes, i, e);
                if (ret)
                        return ret;
-
-               array_insert_item(w->d, w->nr, i, w->d[ancestor_pos]);
-               w->d[i].snapshot = pos.snapshot;
-               w->d[i].count   = 0;
        }
 
        return i;
@@ -669,21 +726,23 @@ static int __get_visible_inodes(struct btree_trans *trans,
        struct bkey_s_c k;
        int ret;
 
-       w->nr = 0;
+       w->inodes.nr = 0;
 
-       for_each_btree_key(trans, iter, BTREE_ID_inodes, POS(0, inum),
+       for_each_btree_key_norestart(trans, iter, BTREE_ID_inodes, POS(0, inum),
                           BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
+               u32 equiv = bch2_snapshot_equiv(c, k.k->p.snapshot);
+
                if (k.k->p.offset != inum)
                        break;
 
-               if (k.k->type != KEY_TYPE_inode)
+               if (!ref_visible(c, s, s->pos.snapshot, equiv))
                        continue;
 
-               if (ref_visible(c, s, s->pos.snapshot, k.k->p.snapshot)) {
-                       add_inode(c, w, bkey_s_c_to_inode(k));
-                       if (k.k->p.snapshot >= s->pos.snapshot)
-                               break;
-               }
+               if (bkey_is_inode(k.k))
+                       add_inode(c, w, k);
+
+               if (equiv >= s->pos.snapshot)
+                       break;
        }
        bch2_trans_iter_exit(trans, &iter);
 
@@ -695,15 +754,16 @@ static int check_key_has_snapshot(struct btree_trans *trans,
                                  struct bkey_s_c k)
 {
        struct bch_fs *c = trans->c;
-       char buf[200];
+       struct printbuf buf = PRINTBUF;
        int ret = 0;
 
-       if (mustfix_fsck_err_on(!snapshot_t(c, k.k->p.snapshot)->equiv, c,
+       if (mustfix_fsck_err_on(!bch2_snapshot_equiv(c, k.k->p.snapshot), c,
                        "key in missing snapshot: %s",
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, k), buf)))
-               return bch2_btree_delete_at(trans, iter,
+                       (bch2_bkey_val_to_text(&buf, c, k), buf.buf)))
+               ret = bch2_btree_delete_at(trans, iter,
                                            BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE) ?: 1;
 fsck_err:
+       printbuf_exit(&buf);
        return ret;
 }
 
@@ -712,9 +772,6 @@ static int hash_redo_key(struct btree_trans *trans,
                         struct bch_hash_info *hash_info,
                         struct btree_iter *k_iter, struct bkey_s_c k)
 {
-       bch_err(trans->c, "hash_redo_key() not implemented yet");
-       return -EINVAL;
-#if 0
        struct bkey_i *delete;
        struct bkey_i *tmp;
 
@@ -732,8 +789,14 @@ static int hash_redo_key(struct btree_trans *trans,
        delete->k.p = k_iter->pos;
        return  bch2_btree_iter_traverse(k_iter) ?:
                bch2_trans_update(trans, k_iter, delete, 0) ?:
-               bch2_hash_set(trans, desc, hash_info, k_iter->pos.inode, tmp, 0);
-#endif
+               bch2_hash_set_snapshot(trans, desc, hash_info,
+                                      (subvol_inum) { 0, k.k->p.inode },
+                                      k.k->p.snapshot, tmp,
+                                      BCH_HASH_SET_MUST_CREATE,
+                                      BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE) ?:
+               bch2_trans_commit(trans, NULL, NULL,
+                                 BTREE_INSERT_NOFAIL|
+                                 BTREE_INSERT_LAZY_RW);
 }
 
 static int hash_check_key(struct btree_trans *trans,
@@ -743,7 +806,7 @@ static int hash_check_key(struct btree_trans *trans,
 {
        struct bch_fs *c = trans->c;
        struct btree_iter iter = { NULL };
-       char buf[200];
+       struct printbuf buf = PRINTBUF;
        struct bkey_s_c k;
        u64 hash;
        int ret = 0;
@@ -759,16 +822,18 @@ static int hash_check_key(struct btree_trans *trans,
        if (hash_k.k->p.offset < hash)
                goto bad_hash;
 
-       for_each_btree_key(trans, iter, desc.btree_id, POS(hash_k.k->p.inode, hash),
-                          BTREE_ITER_SLOTS, k, ret) {
+       for_each_btree_key_norestart(trans, iter, desc.btree_id,
+                                    POS(hash_k.k->p.inode, hash),
+                                    BTREE_ITER_SLOTS, k, ret) {
                if (!bkey_cmp(k.k->p, hash_k.k->p))
                        break;
 
                if (fsck_err_on(k.k->type == desc.key_type &&
                                !desc.cmp_bkey(k, hash_k), c,
                                "duplicate hash table keys:\n%s",
-                               (bch2_bkey_val_to_text(&PBUF(buf), c,
-                                                      hash_k), buf))) {
+                               (printbuf_reset(&buf),
+                                bch2_bkey_val_to_text(&buf, c, hash_k),
+                                buf.buf))) {
                        ret = bch2_hash_delete_at(trans, desc, hash_info, k_iter, 0) ?: 1;
                        break;
                }
@@ -777,50 +842,48 @@ static int hash_check_key(struct btree_trans *trans,
                        bch2_trans_iter_exit(trans, &iter);
                        goto bad_hash;
                }
-
        }
+out:
        bch2_trans_iter_exit(trans, &iter);
+       printbuf_exit(&buf);
        return ret;
 bad_hash:
-       if (fsck_err(c, "hash table key at wrong offset: btree %u inode %llu offset %llu, "
-                    "hashed to %llu\n%s",
-                    desc.btree_id, hash_k.k->p.inode, hash_k.k->p.offset, hash,
-                    (bch2_bkey_val_to_text(&PBUF(buf), c, hash_k), buf)) == FSCK_ERR_IGNORE)
-               return 0;
-
-       ret = hash_redo_key(trans, desc, hash_info, k_iter, hash_k);
-       if (ret) {
-               bch_err(c, "hash_redo_key err %i", ret);
-               return ret;
+       if (fsck_err(c, "hash table key at wrong offset: btree %s inode %llu offset %llu, hashed to %llu\n%s",
+                    bch2_btree_ids[desc.btree_id], hash_k.k->p.inode, hash_k.k->p.offset, hash,
+                    (printbuf_reset(&buf),
+                     bch2_bkey_val_to_text(&buf, c, hash_k), buf.buf))) {
+               ret = hash_redo_key(trans, desc, hash_info, k_iter, hash_k);
+               if (ret) {
+                       bch_err(c, "hash_redo_key err %s", bch2_err_str(ret));
+                       return ret;
+               }
+               ret = -BCH_ERR_transaction_restart_nested;
        }
-       return -EINTR;
 fsck_err:
-       return ret;
+       goto out;
 }
 
 static int check_inode(struct btree_trans *trans,
                       struct btree_iter *iter,
+                      struct bkey_s_c k,
                       struct bch_inode_unpacked *prev,
+                      struct snapshots_seen *s,
                       bool full)
 {
        struct bch_fs *c = trans->c;
-       struct bkey_s_c k;
-       struct bkey_s_c_inode inode;
        struct bch_inode_unpacked u;
        bool do_update = false;
        int ret;
 
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
+       ret = check_key_has_snapshot(trans, iter, k);
+       if (ret < 0)
+               goto err;
        if (ret)
-               return ret;
+               return 0;
 
-       ret = check_key_has_snapshot(trans, iter, k);
+       ret = snapshots_seen_update(c, s, iter->btree_id, k.k->p);
        if (ret)
-               return ret < 0 ? ret : 0;
+               goto err;
 
        /*
         * if snapshot id isn't a leaf node, skip it - deletion in
@@ -830,19 +893,17 @@ static int check_inode(struct btree_trans *trans,
        if (bch2_snapshot_internal_node(c, k.k->p.snapshot))
                return 0;
 
-       if (k.k->type != KEY_TYPE_inode)
+       if (!bkey_is_inode(k.k))
                return 0;
 
-       inode = bkey_s_c_to_inode(k);
+       BUG_ON(bch2_inode_unpack(k, &u));
 
        if (!full &&
-           !(inode.v->bi_flags & (BCH_INODE_I_SIZE_DIRTY|
-                                  BCH_INODE_I_SECTORS_DIRTY|
-                                  BCH_INODE_UNLINKED)))
+           !(u.bi_flags & (BCH_INODE_I_SIZE_DIRTY|
+                           BCH_INODE_I_SECTORS_DIRTY|
+                           BCH_INODE_UNLINKED)))
                return 0;
 
-       BUG_ON(bch2_inode_unpack(inode, &u));
-
        if (prev->bi_inum != u.bi_inum)
                *prev = u;
 
@@ -861,8 +922,9 @@ static int check_inode(struct btree_trans *trans,
                bch2_fs_lazy_rw(c);
 
                ret = fsck_inode_rm(trans, u.bi_inum, iter->pos.snapshot);
-               if (ret)
-                       bch_err(c, "error in fsck: error %i while deleting inode", ret);
+               if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+                       bch_err(c, "error in fsck: error while deleting inode: %s",
+                               bch2_err_str(ret));
                return ret;
        }
 
@@ -885,7 +947,8 @@ static int check_inode(struct btree_trans *trans,
                                POS(u.bi_inum, U64_MAX),
                                0, NULL);
                if (ret) {
-                       bch_err(c, "error in fsck: error %i truncating inode", ret);
+                       bch_err(c, "error in fsck: error truncating inode: %s",
+                               bch2_err_str(ret));
                        return ret;
                }
 
@@ -910,8 +973,8 @@ static int check_inode(struct btree_trans *trans,
 
                sectors = bch2_count_inode_sectors(trans, u.bi_inum, iter->pos.snapshot);
                if (sectors < 0) {
-                       bch_err(c, "error in fsck: error %i recounting inode sectors",
-                               (int) sectors);
+                       bch_err(c, "error in fsck: error recounting inode sectors: %s",
+                               bch2_err_str(sectors));
                        return sectors;
                }
 
@@ -928,12 +991,15 @@ static int check_inode(struct btree_trans *trans,
        }
 
        if (do_update) {
-               ret = write_inode(trans, &u, iter->pos.snapshot);
+               ret = __write_inode(trans, &u, iter->pos.snapshot);
                if (ret)
-                       bch_err(c, "error in fsck: error %i "
-                               "updating inode", ret);
+                       bch_err(c, "error in fsck: error updating inode: %s",
+                               bch2_err_str(ret));
        }
+err:
 fsck_err:
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -943,86 +1009,23 @@ static int check_inodes(struct bch_fs *c, bool full)
        struct btree_trans trans;
        struct btree_iter iter;
        struct bch_inode_unpacked prev = { 0 };
-       int ret;
-
-       bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
-
-       bch2_trans_iter_init(&trans, &iter, BTREE_ID_inodes, POS_MIN,
-                            BTREE_ITER_INTENT|
-                            BTREE_ITER_PREFETCH|
-                            BTREE_ITER_ALL_SNAPSHOTS);
-
-       do {
-               ret = __bch2_trans_do(&trans, NULL, NULL,
-                                     BTREE_INSERT_LAZY_RW|
-                                     BTREE_INSERT_NOFAIL,
-                       check_inode(&trans, &iter, &prev, full));
-               if (ret)
-                       break;
-       } while (bch2_btree_iter_advance(&iter));
-       bch2_trans_iter_exit(&trans, &iter);
-
-       bch2_trans_exit(&trans);
-       return ret;
-}
-
-static int check_subvol(struct btree_trans *trans,
-                       struct btree_iter *iter)
-{
+       struct snapshots_seen s;
        struct bkey_s_c k;
-       struct bkey_s_c_subvolume subvol;
-       int ret;
-
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
-       if (ret)
-               return ret;
-
-       if (k.k->type != KEY_TYPE_subvolume)
-               return 0;
-
-       subvol = bkey_s_c_to_subvolume(k);
-
-       if (BCH_SUBVOLUME_UNLINKED(subvol.v)) {
-               ret = bch2_subvolume_delete(trans, iter->pos.offset);
-               if (ret && ret != -EINTR)
-                       bch_err(trans->c, "error deleting subvolume %llu: %i",
-                               iter->pos.offset, ret);
-               if (ret)
-                       return ret;
-       }
-
-       return 0;
-}
-
-noinline_for_stack
-static int check_subvols(struct bch_fs *c)
-{
-       struct btree_trans trans;
-       struct btree_iter iter;
        int ret;
 
+       snapshots_seen_init(&s);
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
 
-       bch2_trans_iter_init(&trans, &iter, BTREE_ID_subvolumes,
-                            POS_MIN,
-                            BTREE_ITER_INTENT|
-                            BTREE_ITER_PREFETCH);
-
-       do {
-               ret = __bch2_trans_do(&trans, NULL, NULL,
-                                     BTREE_INSERT_LAZY_RW|
-                                     BTREE_INSERT_NOFAIL,
-                                     check_subvol(&trans, &iter));
-               if (ret)
-                       break;
-       } while (bch2_btree_iter_advance(&iter));
-       bch2_trans_iter_exit(&trans, &iter);
+       ret = for_each_btree_key_commit(&trans, iter, BTREE_ID_inodes,
+                       POS_MIN,
+                       BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS, k,
+                       NULL, NULL, BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
+               check_inode(&trans, &iter, k, &prev, &s, full));
 
        bch2_trans_exit(&trans);
+       snapshots_seen_exit(&s);
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1114,7 +1117,7 @@ static int inode_backpointer_exists(struct btree_trans *trans,
                        SPOS(inode->bi_dir, inode->bi_dir_offset, snapshot));
        ret = bkey_err(d.s_c);
        if (ret)
-               return ret;
+               return ret == -ENOENT ? 0 : ret;
 
        ret = dirent_points_to_inode(d, inode);
        bch2_trans_iter_exit(trans, &iter);
@@ -1125,15 +1128,15 @@ static int check_i_sectors(struct btree_trans *trans, struct inode_walker *w)
 {
        struct bch_fs *c = trans->c;
        struct inode_walker_entry *i;
-       int ret = 0, ret2 = 0;
+       u32 restart_count = trans->restart_count;
+       int ret = 0;
        s64 count2;
 
-       for (i = w->d; i < w->d + w->nr; i++) {
+       darray_for_each(w->inodes, i) {
                if (i->inode.bi_sectors == i->count)
                        continue;
 
-               count2 = lockrestart_do(trans,
-                       bch2_count_inode_sectors(trans, w->cur_inum, i->snapshot));
+               count2 = bch2_count_inode_sectors(trans, w->cur_inum, i->snapshot);
 
                if (i->count != count2) {
                        bch_err(c, "fsck counted i_sectors wrong: got %llu should be %llu",
@@ -1146,53 +1149,55 @@ static int check_i_sectors(struct btree_trans *trans, struct inode_walker *w)
                if (fsck_err_on(!(i->inode.bi_flags & BCH_INODE_I_SECTORS_DIRTY), c,
                            "inode %llu:%u has incorrect i_sectors: got %llu, should be %llu",
                            w->cur_inum, i->snapshot,
-                           i->inode.bi_sectors, i->count) == FSCK_ERR_IGNORE)
-                       continue;
-
-               i->inode.bi_sectors = i->count;
-               ret = write_inode(trans, &i->inode, i->snapshot);
-               if (ret)
-                       break;
-               ret2 = -EINTR;
+                           i->inode.bi_sectors, i->count)) {
+                       i->inode.bi_sectors = i->count;
+                       ret = write_inode(trans, &i->inode, i->snapshot);
+                       if (ret)
+                               break;
+               }
        }
 fsck_err:
-       return ret ?: ret2;
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
+       if (!ret && trans_was_restarted(trans, restart_count))
+               ret = -BCH_ERR_transaction_restart_nested;
+       return ret;
 }
 
 static int check_extent(struct btree_trans *trans, struct btree_iter *iter,
+                       struct bkey_s_c k,
                        struct inode_walker *inode,
                        struct snapshots_seen *s)
 {
        struct bch_fs *c = trans->c;
-       struct bkey_s_c k;
        struct inode_walker_entry *i;
-       char buf[200];
+       struct printbuf buf = PRINTBUF;
+       struct bpos equiv;
        int ret = 0;
 
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
-       if (ret)
-               return ret;
-
        ret = check_key_has_snapshot(trans, iter, k);
-       if (ret)
-               return ret < 0 ? ret : 0;
+       if (ret) {
+               ret = ret < 0 ? ret : 0;
+               goto out;
+       }
+
+       equiv = k.k->p;
+       equiv.snapshot = bch2_snapshot_equiv(c, k.k->p.snapshot);
 
-       ret = snapshots_seen_update(c, s, k.k->p);
+       ret = snapshots_seen_update(c, s, iter->btree_id, k.k->p);
        if (ret)
-               return ret;
+               goto err;
 
        if (k.k->type == KEY_TYPE_whiteout)
-               return 0;
+               goto out;
 
        if (inode->cur_inum != k.k->p.inode) {
                ret = check_i_sectors(trans, inode);
                if (ret)
-                       return ret;
+                       goto err;
        }
+
+       BUG_ON(!iter->path->should_be_locked);
 #if 0
        if (bkey_cmp(prev.k->k.p, bkey_start_pos(k.k)) > 0) {
                char buf1[200];
@@ -1201,59 +1206,95 @@ static int check_extent(struct btree_trans *trans, struct btree_iter *iter,
                bch2_bkey_val_to_text(&PBUF(buf1), c, bkey_i_to_s_c(prev.k));
                bch2_bkey_val_to_text(&PBUF(buf2), c, k);
 
-               if (fsck_err(c, "overlapping extents:\n%s\n%s", buf1, buf2))
-                       return fix_overlapping_extent(trans, k, prev.k->k.p) ?: -EINTR;
+               if (fsck_err(c, "overlapping extents:\n%s\n%s", buf1, buf2)) {
+                       ret = fix_overlapping_extent(trans, k, prev.k->k.p)
+                               ?: -BCH_ERR_transaction_restart_nested;
+                       goto out;
+               }
        }
 #endif
-       ret = __walk_inode(trans, inode, k.k->p);
+       ret = __walk_inode(trans, inode, equiv);
        if (ret < 0)
-               return ret;
+               goto err;
 
        if (fsck_err_on(ret == INT_MAX, c,
                        "extent in missing inode:\n  %s",
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, k), buf)))
-               return bch2_btree_delete_at(trans, iter,
+                       (printbuf_reset(&buf),
+                        bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
+               ret = bch2_btree_delete_at(trans, iter,
                                            BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE);
+               goto out;
+       }
 
-       if (ret == INT_MAX)
-               return 0;
+       if (ret == INT_MAX) {
+               ret = 0;
+               goto out;
+       }
 
-       i = inode->d + ret;
+       i = inode->inodes.data + ret;
        ret = 0;
 
        if (fsck_err_on(!S_ISREG(i->inode.bi_mode) &&
                        !S_ISLNK(i->inode.bi_mode), c,
                        "extent in non regular inode mode %o:\n  %s",
                        i->inode.bi_mode,
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, k), buf)))
-               return bch2_btree_delete_at(trans, iter,
+                       (printbuf_reset(&buf),
+                        bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
+               ret = bch2_btree_delete_at(trans, iter,
                                            BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE);
+               goto out;
+       }
+
+       /*
+        * Check inodes in reverse order, from oldest snapshots to newest, so
+        * that we emit the fewest number of whiteouts necessary:
+        */
+       for (i = inode->inodes.data + inode->inodes.nr - 1;
+            i >= inode->inodes.data;
+            --i) {
+               if (i->snapshot > equiv.snapshot ||
+                   !key_visible_in_snapshot(c, s, i->snapshot, equiv.snapshot))
+                       continue;
+
+               if (fsck_err_on(!(i->inode.bi_flags & BCH_INODE_I_SIZE_DIRTY) &&
+                               k.k->type != KEY_TYPE_reservation &&
+                               k.k->p.offset > round_up(i->inode.bi_size, block_bytes(c)) >> 9, c,
+                               "extent type past end of inode %llu:%u, i_size %llu\n  %s",
+                               i->inode.bi_inum, i->snapshot, i->inode.bi_size,
+                               (bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
+                       struct btree_iter iter2;
+
+                       bch2_trans_copy_iter(&iter2, iter);
+                       bch2_btree_iter_set_snapshot(&iter2, i->snapshot);
+                       ret =   bch2_btree_iter_traverse(&iter2) ?:
+                               bch2_btree_delete_at(trans, &iter2,
+                                       BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE);
+                       bch2_trans_iter_exit(trans, &iter2);
+                       if (ret)
+                               goto err;
 
-       if (!bch2_snapshot_internal_node(c, k.k->p.snapshot)) {
-               for_each_visible_inode(c, s, inode, k.k->p.snapshot, i) {
-                       if (fsck_err_on(!(i->inode.bi_flags & BCH_INODE_I_SIZE_DIRTY) &&
-                                       k.k->type != KEY_TYPE_reservation &&
-                                       k.k->p.offset > round_up(i->inode.bi_size, block_bytes(c)) >> 9, c,
-                                       "extent type %u offset %llu past end of inode %llu, i_size %llu",
-                                       k.k->type, k.k->p.offset, k.k->p.inode, i->inode.bi_size)) {
-                               bch2_fs_lazy_rw(c);
-                               return bch2_btree_delete_range_trans(trans, BTREE_ID_extents,
-                                               SPOS(k.k->p.inode, round_up(i->inode.bi_size, block_bytes(c)) >> 9,
-                                                    k.k->p.snapshot),
-                                               POS(k.k->p.inode, U64_MAX),
-                                               0, NULL) ?: -EINTR;
+                       if (i->snapshot != equiv.snapshot) {
+                               ret = snapshots_seen_add(c, s, i->snapshot);
+                               if (ret)
+                                       goto err;
                        }
                }
        }
 
        if (bkey_extent_is_allocation(k.k))
-               for_each_visible_inode(c, s, inode, k.k->p.snapshot, i)
+               for_each_visible_inode(c, s, inode, equiv.snapshot, i)
                        i->count += k.k->size;
 #if 0
        bch2_bkey_buf_reassemble(&prev, c, k);
 #endif
 
+out:
+err:
 fsck_err:
+       printbuf_exit(&buf);
+
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1268,6 +1309,7 @@ static int check_extents(struct bch_fs *c)
        struct snapshots_seen s;
        struct btree_trans trans;
        struct btree_iter iter;
+       struct bkey_s_c k;
        int ret = 0;
 
 #if 0
@@ -1280,21 +1322,12 @@ static int check_extents(struct bch_fs *c)
 
        bch_verbose(c, "checking extents");
 
-       bch2_trans_iter_init(&trans, &iter, BTREE_ID_extents,
-                            POS(BCACHEFS_ROOT_INO, 0),
-                            BTREE_ITER_INTENT|
-                            BTREE_ITER_PREFETCH|
-                            BTREE_ITER_ALL_SNAPSHOTS);
-
-       do {
-               ret = __bch2_trans_do(&trans, NULL, NULL,
-                                     BTREE_INSERT_LAZY_RW|
-                                     BTREE_INSERT_NOFAIL,
-                       check_extent(&trans, &iter, &w, &s));
-               if (ret)
-                       break;
-       } while (bch2_btree_iter_advance(&iter));
-       bch2_trans_iter_exit(&trans, &iter);
+       ret = for_each_btree_key_commit(&trans, iter, BTREE_ID_extents,
+                       POS(BCACHEFS_ROOT_INO, 0),
+                       BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS, k,
+                       NULL, NULL,
+                       BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
+               check_extent(&trans, &iter, k, &w, &s));
 #if 0
        bch2_bkey_buf_exit(&prev, c);
 #endif
@@ -1302,6 +1335,8 @@ static int check_extents(struct bch_fs *c)
        bch2_trans_exit(&trans);
        snapshots_seen_exit(&s);
 
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1309,15 +1344,17 @@ static int check_subdir_count(struct btree_trans *trans, struct inode_walker *w)
 {
        struct bch_fs *c = trans->c;
        struct inode_walker_entry *i;
-       int ret = 0, ret2 = 0;
+       u32 restart_count = trans->restart_count;
+       int ret = 0;
        s64 count2;
 
-       for (i = w->d; i < w->d + w->nr; i++) {
+       darray_for_each(w->inodes, i) {
                if (i->inode.bi_nlink == i->count)
                        continue;
 
-               count2 = lockrestart_do(trans,
-                               bch2_count_subdirs(trans, w->cur_inum, i->snapshot));
+               count2 = bch2_count_subdirs(trans, w->cur_inum, i->snapshot);
+               if (count2 < 0)
+                       return count2;
 
                if (i->count != count2) {
                        bch_err(c, "fsck counted subdirectories wrong: got %llu should be %llu",
@@ -1334,11 +1371,14 @@ static int check_subdir_count(struct btree_trans *trans, struct inode_walker *w)
                        ret = write_inode(trans, &i->inode, i->snapshot);
                        if (ret)
                                break;
-                       ret2 = -EINTR;
                }
        }
 fsck_err:
-       return ret ?: ret2;
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
+       if (!ret && trans_was_restarted(trans, restart_count))
+               ret = -BCH_ERR_transaction_restart_nested;
+       return ret;
 }
 
 static int check_dirent_target(struct btree_trans *trans,
@@ -1350,7 +1390,7 @@ static int check_dirent_target(struct btree_trans *trans,
        struct bch_fs *c = trans->c;
        struct bkey_i_dirent *n;
        bool backpointer_exists = true;
-       char buf[200];
+       struct printbuf buf = PRINTBUF;
        int ret = 0;
 
        if (!target->bi_dir &&
@@ -1376,15 +1416,13 @@ static int check_dirent_target(struct btree_trans *trans,
                                "directory %llu with multiple links",
                                target->bi_inum)) {
                        ret = __remove_dirent(trans, d.k->p);
-                       if (ret)
-                               goto err;
-                       return 0;
+                       goto out;
                }
 
                if (fsck_err_on(backpointer_exists &&
                                !target->bi_nlink, c,
-                               "inode %llu has multiple links but i_nlink 0",
-                               target->bi_inum)) {
+                               "inode %llu type %s has multiple links but i_nlink 0",
+                               target->bi_inum, bch2_d_types[d.v->d_type])) {
                        target->bi_nlink++;
                        target->bi_flags &= ~BCH_INODE_UNLINKED;
 
@@ -1415,18 +1453,19 @@ static int check_dirent_target(struct btree_trans *trans,
                        "incorrect d_type: got %s, should be %s:\n%s",
                        bch2_d_type_str(d.v->d_type),
                        bch2_d_type_str(inode_d_type(target)),
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, d.s_c), buf))) {
+                       (printbuf_reset(&buf),
+                        bch2_bkey_val_to_text(&buf, c, d.s_c), buf.buf))) {
                n = bch2_trans_kmalloc(trans, bkey_bytes(d.k));
                ret = PTR_ERR_OR_ZERO(n);
                if (ret)
-                       return ret;
+                       goto err;
 
                bkey_reassemble(&n->k_i, d.s_c);
                n->v.d_type = inode_d_type(target);
 
                ret = bch2_trans_update(trans, iter, &n->k_i, 0);
                if (ret)
-                       return ret;
+                       goto err;
 
                d = dirent_i_to_s_c(n);
        }
@@ -1440,94 +1479,110 @@ static int check_dirent_target(struct btree_trans *trans,
                n = bch2_trans_kmalloc(trans, bkey_bytes(d.k));
                ret = PTR_ERR_OR_ZERO(n);
                if (ret)
-                       return ret;
+                       goto err;
 
                bkey_reassemble(&n->k_i, d.s_c);
                n->v.d_parent_subvol = cpu_to_le32(target->bi_parent_subvol);
 
                ret = bch2_trans_update(trans, iter, &n->k_i, 0);
                if (ret)
-                       return ret;
+                       goto err;
 
                d = dirent_i_to_s_c(n);
        }
+out:
 err:
 fsck_err:
+       printbuf_exit(&buf);
+
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
 static int check_dirent(struct btree_trans *trans, struct btree_iter *iter,
+                       struct bkey_s_c k,
                        struct bch_hash_info *hash_info,
                        struct inode_walker *dir,
                        struct inode_walker *target,
                        struct snapshots_seen *s)
 {
        struct bch_fs *c = trans->c;
-       struct bkey_s_c k;
        struct bkey_s_c_dirent d;
        struct inode_walker_entry *i;
-       char buf[200];
-       int ret;
-
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
-       if (ret)
-               return ret;
+       struct printbuf buf = PRINTBUF;
+       struct bpos equiv;
+       int ret = 0;
 
        ret = check_key_has_snapshot(trans, iter, k);
-       if (ret)
-               return ret < 0 ? ret : 0;
+       if (ret) {
+               ret = ret < 0 ? ret : 0;
+               goto out;
+       }
 
-       ret = snapshots_seen_update(c, s, k.k->p);
+       equiv = k.k->p;
+       equiv.snapshot = bch2_snapshot_equiv(c, k.k->p.snapshot);
+
+       ret = snapshots_seen_update(c, s, iter->btree_id, k.k->p);
        if (ret)
-               return ret;
+               goto err;
 
        if (k.k->type == KEY_TYPE_whiteout)
-               return 0;
+               goto out;
 
        if (dir->cur_inum != k.k->p.inode) {
                ret = check_subdir_count(trans, dir);
                if (ret)
-                       return ret;
+                       goto err;
        }
 
-       ret = __walk_inode(trans, dir, k.k->p);
+       BUG_ON(!iter->path->should_be_locked);
+
+       ret = __walk_inode(trans, dir, equiv);
        if (ret < 0)
-               return ret;
+               goto err;
 
        if (fsck_err_on(ret == INT_MAX, c,
                        "dirent in nonexisting directory:\n%s",
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, k), buf)))
-               return bch2_btree_delete_at(trans, iter,
+                       (printbuf_reset(&buf),
+                        bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
+               ret = bch2_btree_delete_at(trans, iter,
                                BTREE_UPDATE_INTERNAL_SNAPSHOT_NODE);
+               goto out;
+       }
 
-       if (ret == INT_MAX)
-               return 0;
+       if (ret == INT_MAX) {
+               ret = 0;
+               goto out;
+       }
 
-       i = dir->d + ret;
+       i = dir->inodes.data + ret;
        ret = 0;
 
        if (fsck_err_on(!S_ISDIR(i->inode.bi_mode), c,
                        "dirent in non directory inode type %s:\n%s",
                        bch2_d_type_str(inode_d_type(&i->inode)),
-                       (bch2_bkey_val_to_text(&PBUF(buf), c, k), buf)))
-               return bch2_btree_delete_at(trans, iter, 0);
+                       (printbuf_reset(&buf),
+                        bch2_bkey_val_to_text(&buf, c, k), buf.buf))) {
+               ret = bch2_btree_delete_at(trans, iter, 0);
+               goto out;
+       }
 
        if (dir->first_this_inode)
-               *hash_info = bch2_hash_info_init(c, &dir->d[0].inode);
+               *hash_info = bch2_hash_info_init(c, &dir->inodes.data[0].inode);
 
        ret = hash_check_key(trans, bch2_dirent_hash_desc,
                             hash_info, iter, k);
        if (ret < 0)
-               return ret;
-       if (ret) /* dirent has been deleted */
-               return 0;
+               goto err;
+       if (ret) {
+               /* dirent has been deleted */
+               ret = 0;
+               goto out;
+       }
 
        if (k.k->type != KEY_TYPE_dirent)
-               return 0;
+               goto out;
 
        d = bkey_s_c_to_dirent(k);
 
@@ -1540,24 +1595,27 @@ static int check_dirent(struct btree_trans *trans, struct btree_iter *iter,
                ret = __subvol_lookup(trans, target_subvol,
                                      &target_snapshot, &target_inum);
                if (ret && ret != -ENOENT)
-                       return ret;
+                       goto err;
 
                if (fsck_err_on(ret, c,
                                "dirent points to missing subvolume %llu",
-                               le64_to_cpu(d.v->d_child_subvol)))
-                       return __remove_dirent(trans, d.k->p);
+                               le64_to_cpu(d.v->d_child_subvol))) {
+                       ret = __remove_dirent(trans, d.k->p);
+                       goto err;
+               }
 
                ret = __lookup_inode(trans, target_inum,
                                   &subvol_root, &target_snapshot);
                if (ret && ret != -ENOENT)
-                       return ret;
+                       goto err;
 
                if (fsck_err_on(ret, c,
                                "subvolume %u points to missing subvolume root %llu",
                                target_subvol,
                                target_inum)) {
                        bch_err(c, "repair not implemented yet");
-                       return -EINVAL;
+                       ret = -EINVAL;
+                       goto err;
                }
 
                if (fsck_err_on(subvol_root.bi_subvol != target_subvol, c,
@@ -1567,40 +1625,48 @@ static int check_dirent(struct btree_trans *trans, struct btree_iter *iter,
                        subvol_root.bi_subvol = target_subvol;
                        ret = __write_inode(trans, &subvol_root, target_snapshot);
                        if (ret)
-                               return ret;
+                               goto err;
                }
 
                ret = check_dirent_target(trans, iter, d, &subvol_root,
                                          target_snapshot);
                if (ret)
-                       return ret;
+                       goto err;
        } else {
                ret = __get_visible_inodes(trans, target, s, le64_to_cpu(d.v->d_inum));
                if (ret)
-                       return ret;
+                       goto err;
 
-               if (fsck_err_on(!target->nr, c,
-                               "dirent points to missing inode:\n%s",
-                               (bch2_bkey_val_to_text(&PBUF(buf), c,
-                                                      k), buf))) {
+               if (fsck_err_on(!target->inodes.nr, c,
+                               "dirent points to missing inode: (equiv %u)\n%s",
+                               equiv.snapshot,
+                               (printbuf_reset(&buf),
+                                bch2_bkey_val_to_text(&buf, c, k),
+                                buf.buf))) {
                        ret = __remove_dirent(trans, d.k->p);
                        if (ret)
-                               return ret;
+                               goto err;
                }
 
-               for (i = target->d; i < target->d + target->nr; i++) {
+               darray_for_each(target->inodes, i) {
                        ret = check_dirent_target(trans, iter, d,
                                                  &i->inode, i->snapshot);
                        if (ret)
-                               return ret;
+                               goto err;
                }
        }
 
        if (d.v->d_type == DT_DIR)
-               for_each_visible_inode(c, s, dir, d.k->p.snapshot, i)
+               for_each_visible_inode(c, s, dir, equiv.snapshot, i)
                        i->count++;
 
+out:
+err:
 fsck_err:
+       printbuf_exit(&buf);
+
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1617,6 +1683,7 @@ static int check_dirents(struct bch_fs *c)
        struct bch_hash_info hash_info;
        struct btree_trans trans;
        struct btree_iter iter;
+       struct bkey_s_c k;
        int ret = 0;
 
        bch_verbose(c, "checking dirents");
@@ -1624,46 +1691,32 @@ static int check_dirents(struct bch_fs *c)
        snapshots_seen_init(&s);
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
 
-       bch2_trans_iter_init(&trans, &iter, BTREE_ID_dirents,
-                            POS(BCACHEFS_ROOT_INO, 0),
-                            BTREE_ITER_INTENT|
-                            BTREE_ITER_PREFETCH|
-                            BTREE_ITER_ALL_SNAPSHOTS);
-
-       do {
-               ret = __bch2_trans_do(&trans, NULL, NULL,
-                                     BTREE_INSERT_LAZY_RW|
-                                     BTREE_INSERT_NOFAIL,
-                       check_dirent(&trans, &iter, &hash_info,
-                                    &dir, &target, &s));
-               if (ret)
-                       break;
-       } while (bch2_btree_iter_advance(&iter));
-       bch2_trans_iter_exit(&trans, &iter);
+       ret = for_each_btree_key_commit(&trans, iter, BTREE_ID_dirents,
+                       POS(BCACHEFS_ROOT_INO, 0),
+                       BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS,
+                       k,
+                       NULL, NULL,
+                       BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
+               check_dirent(&trans, &iter, k, &hash_info, &dir, &target, &s));
 
        bch2_trans_exit(&trans);
        snapshots_seen_exit(&s);
        inode_walker_exit(&dir);
        inode_walker_exit(&target);
+
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
 static int check_xattr(struct btree_trans *trans, struct btree_iter *iter,
+                      struct bkey_s_c k,
                       struct bch_hash_info *hash_info,
                       struct inode_walker *inode)
 {
        struct bch_fs *c = trans->c;
-       struct bkey_s_c k;
        int ret;
 
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
-       if (ret)
-               return ret;
-
        ret = check_key_has_snapshot(trans, iter, k);
        if (ret)
                return ret;
@@ -1683,10 +1736,12 @@ static int check_xattr(struct btree_trans *trans, struct btree_iter *iter,
        ret = 0;
 
        if (inode->first_this_inode)
-               *hash_info = bch2_hash_info_init(c, &inode->d[0].inode);
+               *hash_info = bch2_hash_info_init(c, &inode->inodes.data[0].inode);
 
        ret = hash_check_key(trans, bch2_xattr_hash_desc, hash_info, iter, k);
 fsck_err:
+       if (ret && !bch2_err_matches(ret, BCH_ERR_transaction_restart))
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1700,30 +1755,25 @@ static int check_xattrs(struct bch_fs *c)
        struct bch_hash_info hash_info;
        struct btree_trans trans;
        struct btree_iter iter;
+       struct bkey_s_c k;
        int ret = 0;
 
        bch_verbose(c, "checking xattrs");
 
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
 
-       bch2_trans_iter_init(&trans, &iter, BTREE_ID_xattrs,
-                            POS(BCACHEFS_ROOT_INO, 0),
-                            BTREE_ITER_INTENT|
-                            BTREE_ITER_PREFETCH|
-                            BTREE_ITER_ALL_SNAPSHOTS);
-
-       do {
-               ret = __bch2_trans_do(&trans, NULL, NULL,
-                                     BTREE_INSERT_LAZY_RW|
-                                     BTREE_INSERT_NOFAIL,
-                                     check_xattr(&trans, &iter, &hash_info,
-                                                 &inode));
-               if (ret)
-                       break;
-       } while (bch2_btree_iter_advance(&iter));
-       bch2_trans_iter_exit(&trans, &iter);
+       ret = for_each_btree_key_commit(&trans, iter, BTREE_ID_xattrs,
+                       POS(BCACHEFS_ROOT_INO, 0),
+                       BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS,
+                       k,
+                       NULL, NULL,
+                       BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
+               check_xattr(&trans, &iter, k, &hash_info, &inode));
 
        bch2_trans_exit(&trans);
+
+       if (ret)
+               bch_err(c, "%s(): error %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1750,12 +1800,12 @@ static int check_root_trans(struct btree_trans *trans)
                root_subvol.v.flags     = 0;
                root_subvol.v.snapshot  = cpu_to_le32(snapshot);
                root_subvol.v.inode     = cpu_to_le64(inum);
-               ret = __bch2_trans_do(trans, NULL, NULL,
+               ret = commit_do(trans, NULL, NULL,
                                      BTREE_INSERT_NOFAIL|
                                      BTREE_INSERT_LAZY_RW,
                        __bch2_btree_insert(trans, BTREE_ID_subvolumes, &root_subvol.k_i));
                if (ret) {
-                       bch_err(c, "error writing root subvol: %i", ret);
+                       bch_err(c, "error writing root subvol: %s", bch2_err_str(ret));
                        goto err;
                }
 
@@ -1774,7 +1824,7 @@ static int check_root_trans(struct btree_trans *trans)
 
                ret = __write_inode(trans, &root_inode, snapshot);
                if (ret)
-                       bch_err(c, "error writing root inode: %i", ret);
+                       bch_err(c, "error writing root inode: %s", bch2_err_str(ret));
        }
 err:
 fsck_err:
@@ -1793,21 +1843,18 @@ static int check_root(struct bch_fs *c)
                check_root_trans(&trans));
 }
 
-struct pathbuf {
-       size_t          nr;
-       size_t          size;
-
-       struct pathbuf_entry {
-               u64     inum;
-               u32     snapshot;
-       }               *entries;
+struct pathbuf_entry {
+       u64     inum;
+       u32     snapshot;
 };
 
-static bool path_is_dup(struct pathbuf *p, u64 inum, u32 snapshot)
+typedef DARRAY(struct pathbuf_entry) pathbuf;
+
+static bool path_is_dup(pathbuf *p, u64 inum, u32 snapshot)
 {
        struct pathbuf_entry *i;
 
-       for (i = p->entries; i < p->entries + p->nr; i++)
+       darray_for_each(*p, i)
                if (i->inum     == inum &&
                    i->snapshot == snapshot)
                        return true;
@@ -1815,26 +1862,18 @@ static bool path_is_dup(struct pathbuf *p, u64 inum, u32 snapshot)
        return false;
 }
 
-static int path_down(struct pathbuf *p, u64 inum, u32 snapshot)
+static int path_down(struct bch_fs *c, pathbuf *p,
+                    u64 inum, u32 snapshot)
 {
-       if (p->nr == p->size) {
-               size_t new_size = max_t(size_t, 256UL, p->size * 2);
-               void *n = krealloc(p->entries,
-                                  new_size * sizeof(p->entries[0]),
-                                  GFP_KERNEL);
-               if (!n) {
-                       return -ENOMEM;
-               }
-
-               p->entries = n;
-               p->size = new_size;
-       };
-
-       p->entries[p->nr++] = (struct pathbuf_entry) {
+       int ret = darray_push(p, ((struct pathbuf_entry) {
                .inum           = inum,
                .snapshot       = snapshot,
-       };
-       return 0;
+       }));
+
+       if (ret)
+               bch_err(c, "fsck: error allocating memory for pathbuf, size %zu",
+                       p->size);
+       return ret;
 }
 
 /*
@@ -1843,14 +1882,14 @@ static int path_down(struct pathbuf *p, u64 inum, u32 snapshot)
  * XXX: we should also be verifying that inodes are in the right subvolumes
  */
 static int check_path(struct btree_trans *trans,
-                     struct pathbuf *p,
+                     pathbuf *p,
                      struct bch_inode_unpacked *inode,
                      u32 snapshot)
 {
        struct bch_fs *c = trans->c;
        int ret = 0;
 
-       snapshot = snapshot_t(c, snapshot)->equiv;
+       snapshot = bch2_snapshot_equiv(c, snapshot);
        p->nr = 0;
 
        while (!(inode->bi_inum == BCACHEFS_ROOT_INO &&
@@ -1896,7 +1935,7 @@ static int check_path(struct btree_trans *trans,
                if (!S_ISDIR(inode->bi_mode))
                        break;
 
-               ret = path_down(p, inode->bi_inum, snapshot);
+               ret = path_down(c, p, inode->bi_inum, snapshot);
                if (ret) {
                        bch_err(c, "memory allocation failure");
                        return ret;
@@ -1917,14 +1956,14 @@ static int check_path(struct btree_trans *trans,
                        /* XXX print path */
                        bch_err(c, "directory structure loop");
 
-                       for (i = p->entries; i < p->entries + p->nr; i++)
+                       darray_for_each(*p, i)
                                pr_err("%llu:%u", i->inum, i->snapshot);
                        pr_err("%llu:%u", inode->bi_inum, snapshot);
 
                        if (!fsck_err(c, "directory structure loop"))
                                return 0;
 
-                       ret = __bch2_trans_do(trans, NULL, NULL,
+                       ret = commit_do(trans, NULL, NULL,
                                              BTREE_INSERT_NOFAIL|
                                              BTREE_INSERT_LAZY_RW,
                                        remove_backpointer(trans, inode));
@@ -1938,7 +1977,7 @@ static int check_path(struct btree_trans *trans,
        }
 fsck_err:
        if (ret)
-               bch_err(c, "%s: err %i", __func__, ret);
+               bch_err(c, "%s: err %s", __func__, bch2_err_str(ret));
        return ret;
 }
 
@@ -1954,7 +1993,7 @@ static int check_directory_structure(struct bch_fs *c)
        struct btree_iter iter;
        struct bkey_s_c k;
        struct bch_inode_unpacked u;
-       struct pathbuf path = { 0, 0, NULL };
+       pathbuf path = { 0, };
        int ret;
 
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
@@ -1963,10 +2002,10 @@ static int check_directory_structure(struct bch_fs *c)
                           BTREE_ITER_INTENT|
                           BTREE_ITER_PREFETCH|
                           BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
-               if (k.k->type != KEY_TYPE_inode)
+               if (!bkey_is_inode(k.k))
                        continue;
 
-               ret = bch2_inode_unpack(bkey_s_c_to_inode(k), &u);
+               ret = bch2_inode_unpack(k, &u);
                if (ret) {
                        /* Should have been caught earlier in fsck: */
                        bch_err(c, "error unpacking inode %llu: %i", k.k->p.offset, ret);
@@ -1982,9 +2021,7 @@ static int check_directory_structure(struct bch_fs *c)
        }
        bch2_trans_iter_exit(&trans, &iter);
 
-       BUG_ON(ret == -EINTR);
-
-       kfree(path.entries);
+       darray_exit(&path);
 
        bch2_trans_exit(&trans);
        return ret;
@@ -2001,12 +2038,16 @@ struct nlink_table {
        }               *d;
 };
 
-static int add_nlink(struct nlink_table *t, u64 inum, u32 snapshot)
+static int add_nlink(struct bch_fs *c, struct nlink_table *t,
+                    u64 inum, u32 snapshot)
 {
        if (t->nr == t->size) {
                size_t new_size = max_t(size_t, 128UL, t->size * 2);
-               void *d = kvmalloc(new_size * sizeof(t->d[0]), GFP_KERNEL);
+               void *d = kvmalloc_array(new_size, sizeof(t->d[0]), GFP_KERNEL);
+
                if (!d) {
+                       bch_err(c, "fsck: error allocating memory for nlink_table, size %zu",
+                               new_size);
                        return -ENOMEM;
                }
 
@@ -2070,7 +2111,6 @@ static int check_nlinks_find_hardlinks(struct bch_fs *c,
        struct btree_trans trans;
        struct btree_iter iter;
        struct bkey_s_c k;
-       struct bkey_s_c_inode inode;
        struct bch_inode_unpacked u;
        int ret = 0;
 
@@ -2081,25 +2121,23 @@ static int check_nlinks_find_hardlinks(struct bch_fs *c,
                           BTREE_ITER_INTENT|
                           BTREE_ITER_PREFETCH|
                           BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
-               if (k.k->type != KEY_TYPE_inode)
+               if (!bkey_is_inode(k.k))
                        continue;
 
-               inode = bkey_s_c_to_inode(k);
+               /* Should never fail, checked by bch2_inode_invalid: */
+               BUG_ON(bch2_inode_unpack(k, &u));
 
                /*
                 * Backpointer and directory structure checks are sufficient for
                 * directories, since they can't have hardlinks:
                 */
-               if (S_ISDIR(le16_to_cpu(inode.v->bi_mode)))
+               if (S_ISDIR(le16_to_cpu(u.bi_mode)))
                        continue;
 
-               /* Should never fail, checked by bch2_inode_invalid: */
-               BUG_ON(bch2_inode_unpack(inode, &u));
-
                if (!u.bi_nlink)
                        continue;
 
-               ret = add_nlink(t, k.k->p.offset, k.k->p.snapshot);
+               ret = add_nlink(c, t, k.k->p.offset, k.k->p.snapshot);
                if (ret) {
                        *end = k.k->p.offset;
                        ret = 0;
@@ -2135,7 +2173,7 @@ static int check_nlinks_walk_dirents(struct bch_fs *c, struct nlink_table *links
                           BTREE_ITER_INTENT|
                           BTREE_ITER_PREFETCH|
                           BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
-               ret = snapshots_seen_update(c, &s, k.k->p);
+               ret = snapshots_seen_update(c, &s, iter.btree_id, k.k->p);
                if (ret)
                        break;
 
@@ -2147,7 +2185,7 @@ static int check_nlinks_walk_dirents(struct bch_fs *c, struct nlink_table *links
                            d.v->d_type != DT_SUBVOL)
                                inc_link(c, &s, links, range_start, range_end,
                                         le64_to_cpu(d.v->d_inum),
-                                        d.k->p.snapshot);
+                                        bch2_snapshot_equiv(c, d.k->p.snapshot));
                        break;
                }
        }
@@ -2161,6 +2199,47 @@ static int check_nlinks_walk_dirents(struct bch_fs *c, struct nlink_table *links
        return ret;
 }
 
+static int check_nlinks_update_inode(struct btree_trans *trans, struct btree_iter *iter,
+                                    struct bkey_s_c k,
+                                    struct nlink_table *links,
+                                    size_t *idx, u64 range_end)
+{
+       struct bch_fs *c = trans->c;
+       struct bch_inode_unpacked u;
+       struct nlink *link = &links->d[*idx];
+       int ret = 0;
+
+       if (k.k->p.offset >= range_end)
+               return 1;
+
+       if (!bkey_is_inode(k.k))
+               return 0;
+
+       BUG_ON(bch2_inode_unpack(k, &u));
+
+       if (S_ISDIR(le16_to_cpu(u.bi_mode)))
+               return 0;
+
+       if (!u.bi_nlink)
+               return 0;
+
+       while ((cmp_int(link->inum, k.k->p.offset) ?:
+               cmp_int(link->snapshot, k.k->p.snapshot)) < 0) {
+               BUG_ON(*idx == links->nr);
+               link = &links->d[++*idx];
+       }
+
+       if (fsck_err_on(bch2_inode_nlink_get(&u) != link->count, c,
+                       "inode %llu type %s has wrong i_nlink (%u, should be %u)",
+                       u.bi_inum, bch2_d_types[mode_to_type(u.bi_mode)],
+                       bch2_inode_nlink_get(&u), link->count)) {
+               bch2_inode_nlink_set(&u, link->count);
+               ret = __write_inode(trans, &u, k.k->p.snapshot);
+       }
+fsck_err:
+       return ret;
+}
+
 noinline_for_stack
 static int check_nlinks_update_hardlinks(struct bch_fs *c,
                               struct nlink_table *links,
@@ -2169,58 +2248,25 @@ static int check_nlinks_update_hardlinks(struct bch_fs *c,
        struct btree_trans trans;
        struct btree_iter iter;
        struct bkey_s_c k;
-       struct bkey_s_c_inode inode;
-       struct bch_inode_unpacked u;
-       struct nlink *link = links->d;
+       size_t idx = 0;
        int ret = 0;
 
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
 
-       for_each_btree_key(&trans, iter, BTREE_ID_inodes,
-                          POS(0, range_start),
-                          BTREE_ITER_INTENT|
-                          BTREE_ITER_PREFETCH|
-                          BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
-               if (k.k->p.offset >= range_end)
-                       break;
+       ret = for_each_btree_key_commit(&trans, iter, BTREE_ID_inodes,
+                       POS(0, range_start),
+                       BTREE_ITER_INTENT|BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS, k,
+                       NULL, NULL, BTREE_INSERT_LAZY_RW|BTREE_INSERT_NOFAIL,
+               check_nlinks_update_inode(&trans, &iter, k, links, &idx, range_end));
 
-               if (k.k->type != KEY_TYPE_inode)
-                       continue;
-
-               inode = bkey_s_c_to_inode(k);
-               if (S_ISDIR(le16_to_cpu(inode.v->bi_mode)))
-                       continue;
-
-               BUG_ON(bch2_inode_unpack(inode, &u));
-
-               if (!u.bi_nlink)
-                       continue;
-
-               while ((cmp_int(link->inum, k.k->p.offset) ?:
-                       cmp_int(link->snapshot, k.k->p.snapshot)) < 0) {
-                       link++;
-                       BUG_ON(link >= links->d + links->nr);
-               }
-
-               if (fsck_err_on(bch2_inode_nlink_get(&u) != link->count, c,
-                               "inode %llu has wrong i_nlink (type %u i_nlink %u, should be %u)",
-                               u.bi_inum, mode_to_type(u.bi_mode),
-                               bch2_inode_nlink_get(&u), link->count)) {
-                       bch2_inode_nlink_set(&u, link->count);
-
-                       ret = write_inode(&trans, &u, k.k->p.snapshot);
-                       if (ret)
-                               bch_err(c, "error in fsck: error %i updating inode", ret);
-               }
-       }
-fsck_err:
-       bch2_trans_iter_exit(&trans, &iter);
        bch2_trans_exit(&trans);
 
-       if (ret)
+       if (ret < 0) {
                bch_err(c, "error in fsck: btree error %i while walking inodes", ret);
+               return ret;
+       }
 
-       return ret;
+       return 0;
 }
 
 noinline_for_stack
@@ -2260,21 +2306,13 @@ static int check_nlinks(struct bch_fs *c)
        return ret;
 }
 
-static int fix_reflink_p_key(struct btree_trans *trans, struct btree_iter *iter)
+static int fix_reflink_p_key(struct btree_trans *trans, struct btree_iter *iter,
+                            struct bkey_s_c k)
 {
-       struct bkey_s_c k;
        struct bkey_s_c_reflink_p p;
        struct bkey_i_reflink_p *u;
        int ret;
 
-       k = bch2_btree_iter_peek(iter);
-       if (!k.k)
-               return 0;
-
-       ret = bkey_err(k);
-       if (ret)
-               return ret;
-
        if (k.k->type != KEY_TYPE_reflink_p)
                return 0;
 
@@ -2310,20 +2348,11 @@ static int fix_reflink_p(struct bch_fs *c)
 
        bch2_trans_init(&trans, c, BTREE_ITER_MAX, 0);
 
-       for_each_btree_key(&trans, iter, BTREE_ID_extents, POS_MIN,
-                          BTREE_ITER_INTENT|
-                          BTREE_ITER_PREFETCH|
-                          BTREE_ITER_ALL_SNAPSHOTS, k, ret) {
-               if (k.k->type == KEY_TYPE_reflink_p) {
-                       ret = __bch2_trans_do(&trans, NULL, NULL,
-                                             BTREE_INSERT_NOFAIL|
-                                             BTREE_INSERT_LAZY_RW,
-                                             fix_reflink_p_key(&trans, &iter));
-                       if (ret)
-                               break;
-               }
-       }
-       bch2_trans_iter_exit(&trans, &iter);
+       ret = for_each_btree_key_commit(&trans, iter,
+                       BTREE_ID_extents, POS_MIN,
+                       BTREE_ITER_INTENT|BTREE_ITER_PREFETCH|BTREE_ITER_ALL_SNAPSHOTS, k,
+                       NULL, NULL, BTREE_INSERT_NOFAIL|BTREE_INSERT_LAZY_RW,
+               fix_reflink_p_key(&trans, &iter, k));
 
        bch2_trans_exit(&trans);
        return ret;
@@ -2335,9 +2364,12 @@ static int fix_reflink_p(struct bch_fs *c)
  */
 int bch2_fsck_full(struct bch_fs *c)
 {
-       return  bch2_fs_snapshots_check(c) ?:
+       int ret;
+again:
+       ret =   bch2_fs_check_snapshots(c) ?:
+               bch2_fs_check_subvols(c) ?:
+               bch2_delete_dead_snapshots(c) ?:
                check_inodes(c, true) ?:
-               check_subvols(c) ?:
                check_extents(c) ?:
                check_dirents(c) ?:
                check_xattrs(c) ?:
@@ -2345,9 +2377,19 @@ int bch2_fsck_full(struct bch_fs *c)
                check_directory_structure(c) ?:
                check_nlinks(c) ?:
                fix_reflink_p(c);
+
+       if (bch2_err_matches(ret, BCH_ERR_need_snapshot_cleanup)) {
+               set_bit(BCH_FS_HAVE_DELETED_SNAPSHOTS, &c->flags);
+               goto again;
+       }
+
+       return ret;
 }
 
 int bch2_fsck_walk_inodes_only(struct bch_fs *c)
 {
-       return check_inodes(c, false);
+       return  bch2_fs_check_snapshots(c) ?:
+               bch2_fs_check_subvols(c) ?:
+               bch2_delete_dead_snapshots(c) ?:
+               check_inodes(c, false);
 }