]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - libbcachefs/btree_iter.c
New upstream snapshot
[bcachefs-tools-debian] / libbcachefs / btree_iter.c
index fe710d19ca1994d5149f2496dc500704b4c10a42..ae63ecbc19548a66f8fd9396eca41e650dc02ec5 100644 (file)
 #include "error.h"
 #include "extents.h"
 #include "journal.h"
+#include "recovery.h"
 #include "replicas.h"
+#include "subvolume.h"
 
 #include <linux/prefetch.h>
 #include <trace/events/bcachefs.h>
 
-static void btree_iter_set_search_pos(struct btree_iter *, struct bpos);
-static void btree_trans_sort_iters(struct btree_trans *);
-static void btree_iter_check_sort(struct btree_trans *, struct btree_iter *);
-static struct btree_iter *btree_iter_child_alloc(struct btree_iter *, unsigned long);
-static struct btree_iter *btree_trans_iter_alloc(struct btree_trans *,
-                                                struct btree_iter *);
-static void btree_iter_copy(struct btree_iter *, struct btree_iter *);
+static void btree_trans_verify_sorted(struct btree_trans *);
+static void btree_path_check_sort(struct btree_trans *, struct btree_path *, int);
 
-static inline int btree_iter_cmp(const struct btree_iter *l,
-                                const struct btree_iter *r)
+static inline void btree_path_list_remove(struct btree_trans *, struct btree_path *);
+static inline void btree_path_list_add(struct btree_trans *, struct btree_path *,
+                                      struct btree_path *);
+
+static inline unsigned long btree_iter_ip_allocated(struct btree_iter *iter)
 {
-       return   cmp_int(l->btree_id, r->btree_id) ?:
-               -cmp_int(btree_iter_is_cached(l), btree_iter_is_cached(r)) ?:
-                bkey_cmp(l->real_pos, r->real_pos);
+#ifdef CONFIG_BCACHEFS_DEBUG
+       return iter->ip_allocated;
+#else
+       return 0;
+#endif
 }
 
-static inline struct bpos bkey_successor(struct btree_iter *iter, struct bpos p)
+static struct btree_path *btree_path_alloc(struct btree_trans *, struct btree_path *);
+
+/*
+ * Unlocks before scheduling
+ * Note: does not revalidate iterator
+ */
+static inline int bch2_trans_cond_resched(struct btree_trans *trans)
+{
+       if (need_resched() || race_fault()) {
+               bch2_trans_unlock(trans);
+               schedule();
+               return bch2_trans_relock(trans) ? 0 : -EINTR;
+       } else {
+               return 0;
+       }
+}
+
+static inline int __btree_path_cmp(const struct btree_path *l,
+                                  enum btree_id        r_btree_id,
+                                  bool                 r_cached,
+                                  struct bpos          r_pos,
+                                  unsigned             r_level)
+{
+       /*
+        * Must match lock ordering as defined by __bch2_btree_node_lock:
+        */
+       return   cmp_int(l->btree_id,   r_btree_id) ?:
+                cmp_int((int) l->cached,       (int) r_cached) ?:
+                bpos_cmp(l->pos,       r_pos) ?:
+               -cmp_int(l->level,      r_level);
+}
+
+static inline int btree_path_cmp(const struct btree_path *l,
+                                const struct btree_path *r)
 {
-       EBUG_ON(btree_iter_type(iter) == BTREE_ITER_NODES);
+       return __btree_path_cmp(l, r->btree_id, r->cached, r->pos, r->level);
+}
 
+static inline struct bpos bkey_successor(struct btree_iter *iter, struct bpos p)
+{
        /* Are we iterating over keys in all snapshots? */
        if (iter->flags & BTREE_ITER_ALL_SNAPSHOTS) {
                p = bpos_successor(p);
@@ -50,8 +88,6 @@ static inline struct bpos bkey_successor(struct btree_iter *iter, struct bpos p)
 
 static inline struct bpos bkey_predecessor(struct btree_iter *iter, struct bpos p)
 {
-       EBUG_ON(btree_iter_type(iter) == BTREE_ITER_NODES);
-
        /* Are we iterating over keys in all snapshots? */
        if (iter->flags & BTREE_ITER_ALL_SNAPSHOTS) {
                p = bpos_predecessor(p);
@@ -63,10 +99,10 @@ static inline struct bpos bkey_predecessor(struct btree_iter *iter, struct bpos
        return p;
 }
 
-static inline bool is_btree_node(struct btree_iter *iter, unsigned l)
+static inline bool is_btree_node(struct btree_path *path, unsigned l)
 {
        return l < BTREE_MAX_DEPTH &&
-               (unsigned long) iter->l[l].b >= 128;
+               (unsigned long) path->l[l].b >= 128;
 }
 
 static inline struct bpos btree_iter_search_key(struct btree_iter *iter)
@@ -79,41 +115,40 @@ static inline struct bpos btree_iter_search_key(struct btree_iter *iter)
        return pos;
 }
 
-static inline bool btree_iter_pos_before_node(struct btree_iter *iter,
+static inline bool btree_path_pos_before_node(struct btree_path *path,
                                              struct btree *b)
 {
-       return bpos_cmp(iter->real_pos, b->data->min_key) < 0;
+       return bpos_cmp(path->pos, b->data->min_key) < 0;
 }
 
-static inline bool btree_iter_pos_after_node(struct btree_iter *iter,
+static inline bool btree_path_pos_after_node(struct btree_path *path,
                                             struct btree *b)
 {
-       return bpos_cmp(b->key.k.p, iter->real_pos) < 0;
+       return bpos_cmp(b->key.k.p, path->pos) < 0;
 }
 
-static inline bool btree_iter_pos_in_node(struct btree_iter *iter,
+static inline bool btree_path_pos_in_node(struct btree_path *path,
                                          struct btree *b)
 {
-       return iter->btree_id == b->c.btree_id &&
-               !btree_iter_pos_before_node(iter, b) &&
-               !btree_iter_pos_after_node(iter, b);
+       return path->btree_id == b->c.btree_id &&
+               !btree_path_pos_before_node(path, b) &&
+               !btree_path_pos_after_node(path, b);
 }
 
 /* Btree node locking: */
 
-void bch2_btree_node_unlock_write(struct btree *b, struct btree_iter *iter)
+void bch2_btree_node_unlock_write(struct btree_trans *trans,
+                       struct btree_path *path, struct btree *b)
 {
-       bch2_btree_node_unlock_write_inlined(b, iter);
+       bch2_btree_node_unlock_write_inlined(trans, path, b);
 }
 
-void __bch2_btree_node_lock_write(struct btree *b, struct btree_iter *iter)
+void __bch2_btree_node_lock_write(struct btree_trans *trans, struct btree *b)
 {
-       struct btree_iter *linked;
+       struct btree_path *linked;
        unsigned readers = 0;
 
-       EBUG_ON(!btree_node_intent_locked(iter, b->c.level));
-
-       trans_for_each_iter(iter->trans, linked)
+       trans_for_each_path(trans, linked)
                if (linked->l[b->c.level].b == b &&
                    btree_node_read_locked(linked, b->c.level))
                        readers++;
@@ -124,140 +159,155 @@ void __bch2_btree_node_lock_write(struct btree *b, struct btree_iter *iter)
         * goes to 0, and it's safe because we have the node intent
         * locked:
         */
-       atomic64_sub(__SIX_VAL(read_lock, readers),
-                    &b->c.lock.state.counter);
-       btree_node_lock_type(iter->trans->c, b, SIX_LOCK_write);
-       atomic64_add(__SIX_VAL(read_lock, readers),
-                    &b->c.lock.state.counter);
+       if (!b->c.lock.readers)
+               atomic64_sub(__SIX_VAL(read_lock, readers),
+                            &b->c.lock.state.counter);
+       else
+               this_cpu_sub(*b->c.lock.readers, readers);
+
+       six_lock_write(&b->c.lock, NULL, NULL);
+
+       if (!b->c.lock.readers)
+               atomic64_add(__SIX_VAL(read_lock, readers),
+                            &b->c.lock.state.counter);
+       else
+               this_cpu_add(*b->c.lock.readers, readers);
 }
 
-bool __bch2_btree_node_relock(struct btree_iter *iter, unsigned level)
+bool __bch2_btree_node_relock(struct btree_trans *trans,
+                             struct btree_path *path, unsigned level)
 {
-       struct btree *b = btree_iter_node(iter, level);
-       int want = __btree_lock_want(iter, level);
+       struct btree *b = btree_path_node(path, level);
+       int want = __btree_lock_want(path, level);
 
-       if (!is_btree_node(iter, level))
-               return false;
+       if (!is_btree_node(path, level))
+               goto fail;
 
        if (race_fault())
-               return false;
+               goto fail;
 
-       if (six_relock_type(&b->c.lock, want, iter->l[level].lock_seq) ||
-           (btree_node_lock_seq_matches(iter, b, level) &&
-            btree_node_lock_increment(iter->trans, b, level, want))) {
-               mark_btree_node_locked(iter, level, want);
+       if (six_relock_type(&b->c.lock, want, path->l[level].lock_seq) ||
+           (btree_node_lock_seq_matches(path, b, level) &&
+            btree_node_lock_increment(trans, b, level, want))) {
+               mark_btree_node_locked(path, level, want);
                return true;
-       } else {
-               return false;
        }
+fail:
+       trace_btree_node_relock_fail(trans->fn, _RET_IP_,
+                                    path->btree_id,
+                                    &path->pos,
+                                    (unsigned long) b,
+                                    path->l[level].lock_seq,
+                                    is_btree_node(path, level) ? b->c.lock.state.seq : 0);
+       return false;
 }
 
-static bool bch2_btree_node_upgrade(struct btree_iter *iter, unsigned level)
+bool bch2_btree_node_upgrade(struct btree_trans *trans,
+                            struct btree_path *path, unsigned level)
 {
-       struct btree *b = iter->l[level].b;
+       struct btree *b = path->l[level].b;
 
-       EBUG_ON(btree_lock_want(iter, level) != BTREE_NODE_INTENT_LOCKED);
-
-       if (!is_btree_node(iter, level))
+       if (!is_btree_node(path, level))
                return false;
 
-       if (btree_node_intent_locked(iter, level))
+       switch (btree_lock_want(path, level)) {
+       case BTREE_NODE_UNLOCKED:
+               BUG_ON(btree_node_locked(path, level));
+               return true;
+       case BTREE_NODE_READ_LOCKED:
+               BUG_ON(btree_node_intent_locked(path, level));
+               return bch2_btree_node_relock(trans, path, level);
+       case BTREE_NODE_INTENT_LOCKED:
+               break;
+       }
+
+       if (btree_node_intent_locked(path, level))
                return true;
 
        if (race_fault())
                return false;
 
-       if (btree_node_locked(iter, level)
+       if (btree_node_locked(path, level)
            ? six_lock_tryupgrade(&b->c.lock)
-           : six_relock_type(&b->c.lock, SIX_LOCK_intent, iter->l[level].lock_seq))
+           : six_relock_type(&b->c.lock, SIX_LOCK_intent, path->l[level].lock_seq))
                goto success;
 
-       if (btree_node_lock_seq_matches(iter, b, level) &&
-           btree_node_lock_increment(iter->trans, b, level, BTREE_NODE_INTENT_LOCKED)) {
-               btree_node_unlock(iter, level);
+       if (btree_node_lock_seq_matches(path, b, level) &&
+           btree_node_lock_increment(trans, b, level, BTREE_NODE_INTENT_LOCKED)) {
+               btree_node_unlock(path, level);
                goto success;
        }
 
        return false;
 success:
-       mark_btree_node_intent_locked(iter, level);
+       mark_btree_node_intent_locked(path, level);
        return true;
 }
 
-static inline bool btree_iter_get_locks(struct btree_iter *iter, bool upgrade,
-                                       unsigned long trace_ip)
+static inline bool btree_path_get_locks(struct btree_trans *trans,
+                                       struct btree_path *path,
+                                       bool upgrade)
 {
-       unsigned l = iter->level;
+       unsigned l = path->level;
        int fail_idx = -1;
 
        do {
-               if (!btree_iter_node(iter, l))
+               if (!btree_path_node(path, l))
                        break;
 
                if (!(upgrade
-                     ? bch2_btree_node_upgrade(iter, l)
-                     : bch2_btree_node_relock(iter, l))) {
-                       (upgrade
-                        ? trace_node_upgrade_fail
-                        : trace_node_relock_fail)(iter->trans->ip, trace_ip,
-                                       btree_iter_type(iter) == BTREE_ITER_CACHED,
-                                       iter->btree_id, &iter->real_pos,
-                                       l, iter->l[l].lock_seq,
-                                       is_btree_node(iter, l)
-                                       ? 0
-                                       : (unsigned long) iter->l[l].b,
-                                       is_btree_node(iter, l)
-                                       ? iter->l[l].b->c.lock.state.seq
-                                       : 0);
+                     ? bch2_btree_node_upgrade(trans, path, l)
+                     : bch2_btree_node_relock(trans, path, l)))
                        fail_idx = l;
-                       btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
-               }
 
                l++;
-       } while (l < iter->locks_want);
+       } while (l < path->locks_want);
 
        /*
         * When we fail to get a lock, we have to ensure that any child nodes
-        * can't be relocked so bch2_btree_iter_traverse has to walk back up to
+        * can't be relocked so bch2_btree_path_traverse has to walk back up to
         * the node that we failed to relock:
         */
-       while (fail_idx >= 0) {
-               btree_node_unlock(iter, fail_idx);
-               iter->l[fail_idx].b = BTREE_ITER_NO_NODE_GET_LOCKS;
-               --fail_idx;
+       if (fail_idx >= 0) {
+               __bch2_btree_path_unlock(path);
+               btree_path_set_dirty(path, BTREE_ITER_NEED_TRAVERSE);
+
+               do {
+                       path->l[fail_idx].b = BTREE_ITER_NO_NODE_GET_LOCKS;
+                       --fail_idx;
+               } while (fail_idx >= 0);
        }
 
-       if (iter->uptodate == BTREE_ITER_NEED_RELOCK)
-               iter->uptodate = BTREE_ITER_NEED_PEEK;
+       if (path->uptodate == BTREE_ITER_NEED_RELOCK)
+               path->uptodate = BTREE_ITER_UPTODATE;
 
-       bch2_btree_trans_verify_locks(iter->trans);
+       bch2_trans_verify_locks(trans);
 
-       return iter->uptodate < BTREE_ITER_NEED_RELOCK;
+       return path->uptodate < BTREE_ITER_NEED_RELOCK;
 }
 
 static struct bpos btree_node_pos(struct btree_bkey_cached_common *_b,
-                                 enum btree_iter_type type)
+                                 bool cached)
 {
-       return  type != BTREE_ITER_CACHED
+       return !cached
                ? container_of(_b, struct btree, c)->key.k.p
                : container_of(_b, struct bkey_cached, c)->key.pos;
 }
 
 /* Slowpath: */
-bool __bch2_btree_node_lock(struct btree *b, struct bpos pos,
-                           unsigned level, struct btree_iter *iter,
+bool __bch2_btree_node_lock(struct btree_trans *trans,
+                           struct btree_path *path,
+                           struct btree *b,
+                           struct bpos pos, unsigned level,
                            enum six_lock_type type,
                            six_lock_should_sleep_fn should_sleep_fn, void *p,
                            unsigned long ip)
 {
-       struct btree_trans *trans = iter->trans;
-       struct btree_iter *linked, *deadlock_iter = NULL;
-       u64 start_time = local_clock();
-       unsigned reason = 9;
-       bool ret;
+       struct btree_path *linked;
+       unsigned reason;
 
        /* Check if it's safe to block: */
-       trans_for_each_iter(trans, linked) {
+       trans_for_each_path(trans, linked) {
                if (!linked->nodes_locked)
                        continue;
 
@@ -275,141 +325,114 @@ bool __bch2_btree_node_lock(struct btree *b, struct bpos pos,
                 */
                if (type == SIX_LOCK_intent &&
                    linked->nodes_locked != linked->nodes_intent_locked) {
-                       deadlock_iter = linked;
                        reason = 1;
+                       goto deadlock;
                }
 
-               if (linked->btree_id != iter->btree_id) {
-                       if (linked->btree_id > iter->btree_id) {
-                               deadlock_iter = linked;
-                               reason = 3;
-                       }
-                       continue;
+               if (linked->btree_id != path->btree_id) {
+                       if (linked->btree_id < path->btree_id)
+                               continue;
+
+                       reason = 3;
+                       goto deadlock;
                }
 
                /*
-                * Within the same btree, cached iterators come before non
-                * cached iterators:
+                * Within the same btree, non-cached paths come before cached
+                * paths:
                 */
-               if (btree_iter_is_cached(linked) != btree_iter_is_cached(iter)) {
-                       if (btree_iter_is_cached(iter)) {
-                               deadlock_iter = linked;
-                               reason = 4;
-                       }
-                       continue;
+               if (linked->cached != path->cached) {
+                       if (!linked->cached)
+                               continue;
+
+                       reason = 4;
+                       goto deadlock;
                }
 
                /*
                 * Interior nodes must be locked before their descendants: if
-                * another iterator has possible descendants locked of the node
+                * another path has possible descendants locked of the node
                 * we're about to lock, it must have the ancestors locked too:
                 */
                if (level > __fls(linked->nodes_locked)) {
-                       deadlock_iter = linked;
                        reason = 5;
+                       goto deadlock;
                }
 
                /* Must lock btree nodes in key order: */
                if (btree_node_locked(linked, level) &&
                    bpos_cmp(pos, btree_node_pos((void *) linked->l[level].b,
-                                                btree_iter_type(linked))) <= 0) {
-                       deadlock_iter = linked;
-                       reason = 7;
+                                                linked->cached)) <= 0) {
                        BUG_ON(trans->in_traverse_all);
+                       reason = 7;
+                       goto deadlock;
                }
        }
 
-       if (unlikely(deadlock_iter)) {
-               trace_trans_restart_would_deadlock(trans->ip, ip,
-                               trans->in_traverse_all, reason,
-                               deadlock_iter->btree_id,
-                               btree_iter_type(deadlock_iter),
-                               &deadlock_iter->real_pos,
-                               iter->btree_id,
-                               btree_iter_type(iter),
-                               &pos);
-               btree_trans_restart(trans);
-               return false;
-       }
-
-       if (six_trylock_type(&b->c.lock, type))
-               return true;
-
-#ifdef CONFIG_BCACHEFS_DEBUG
-       trans->locking_iter_idx = iter->idx;
-       trans->locking_pos      = pos;
-       trans->locking_btree_id = iter->btree_id;
-       trans->locking_level    = level;
-       trans->locking          = b;
-#endif
-
-       ret = six_lock_type(&b->c.lock, type, should_sleep_fn, p) == 0;
-
-#ifdef CONFIG_BCACHEFS_DEBUG
-       trans->locking = NULL;
-#endif
-       if (ret)
-               bch2_time_stats_update(&trans->c->times[lock_to_time_stat(type)],
-                                      start_time);
-       return ret;
+       return btree_node_lock_type(trans, path, b, pos, level,
+                                   type, should_sleep_fn, p);
+deadlock:
+       trace_trans_restart_would_deadlock(trans->fn, ip,
+                       trans->in_traverse_all, reason,
+                       linked->btree_id,
+                       linked->cached,
+                       &linked->pos,
+                       path->btree_id,
+                       path->cached,
+                       &pos);
+       btree_trans_restart(trans);
+       return false;
 }
 
 /* Btree iterator locking: */
 
 #ifdef CONFIG_BCACHEFS_DEBUG
-static void bch2_btree_iter_verify_locks(struct btree_iter *iter)
+
+static void bch2_btree_path_verify_locks(struct btree_path *path)
 {
        unsigned l;
 
-       if (!(iter->trans->iters_linked & (1ULL << iter->idx))) {
-               BUG_ON(iter->nodes_locked);
+       if (!path->nodes_locked) {
+               BUG_ON(path->uptodate == BTREE_ITER_UPTODATE &&
+                      btree_path_node(path, path->level));
                return;
        }
 
-       for (l = 0; btree_iter_node(iter, l); l++) {
-               if (iter->uptodate >= BTREE_ITER_NEED_RELOCK &&
-                   !btree_node_locked(iter, l))
-                       continue;
-
-               BUG_ON(btree_lock_want(iter, l) !=
-                      btree_node_locked_type(iter, l));
-       }
+       for (l = 0; btree_path_node(path, l); l++)
+               BUG_ON(btree_lock_want(path, l) !=
+                      btree_node_locked_type(path, l));
 }
 
-void bch2_btree_trans_verify_locks(struct btree_trans *trans)
+void bch2_trans_verify_locks(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
+       struct btree_path *path;
 
-       trans_for_each_iter(trans, iter)
-               bch2_btree_iter_verify_locks(iter);
+       trans_for_each_path(trans, path)
+               bch2_btree_path_verify_locks(path);
 }
 #else
-static inline void bch2_btree_iter_verify_locks(struct btree_iter *iter) {}
+static inline void bch2_btree_path_verify_locks(struct btree_path *path) {}
 #endif
 
+/* Btree path locking: */
+
 /*
  * Only for btree_cache.c - only relocks intent locks
  */
-bool bch2_btree_iter_relock_intent(struct btree_iter *iter)
+bool bch2_btree_path_relock_intent(struct btree_trans *trans,
+                                  struct btree_path *path)
 {
        unsigned l;
 
-       for (l = iter->level;
-            l < iter->locks_want && btree_iter_node(iter, l);
+       for (l = path->level;
+            l < path->locks_want && btree_path_node(path, l);
             l++) {
-               if (!bch2_btree_node_relock(iter, l)) {
-                       trace_node_relock_fail(iter->trans->ip, _RET_IP_,
-                                       btree_iter_type(iter) == BTREE_ITER_CACHED,
-                                       iter->btree_id, &iter->real_pos,
-                                       l, iter->l[l].lock_seq,
-                                       is_btree_node(iter, l)
-                                       ? 0
-                                       : (unsigned long) iter->l[l].b,
-                                       is_btree_node(iter, l)
-                                       ? iter->l[l].b->c.lock.state.seq
-                                       : 0);
-                       btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
-                       btree_trans_restart(iter->trans);
+               if (!bch2_btree_node_relock(trans, path, l)) {
+                       __bch2_btree_path_unlock(path);
+                       btree_path_set_dirty(path, BTREE_ITER_NEED_TRAVERSE);
+                       trace_trans_restart_relock_path_intent(trans->fn, _RET_IP_,
+                                                  path->btree_id, &path->pos);
+                       btree_trans_restart(trans);
                        return false;
                }
        }
@@ -418,25 +441,30 @@ bool bch2_btree_iter_relock_intent(struct btree_iter *iter)
 }
 
 __flatten
-bool bch2_btree_iter_relock(struct btree_iter *iter, unsigned long trace_ip)
+static bool bch2_btree_path_relock(struct btree_trans *trans,
+                       struct btree_path *path, unsigned long trace_ip)
 {
-       bool ret = btree_iter_get_locks(iter, false, trace_ip);
+       bool ret = btree_path_get_locks(trans, path, false);
 
-       if (!ret)
-               btree_trans_restart(iter->trans);
+       if (!ret) {
+               trace_trans_restart_relock_path(trans->fn, trace_ip,
+                                               path->btree_id, &path->pos);
+               btree_trans_restart(trans);
+       }
        return ret;
 }
 
-bool __bch2_btree_iter_upgrade(struct btree_iter *iter,
+bool __bch2_btree_path_upgrade(struct btree_trans *trans,
+                              struct btree_path *path,
                               unsigned new_locks_want)
 {
-       struct btree_iter *linked;
+       struct btree_path *linked;
 
-       EBUG_ON(iter->locks_want >= new_locks_want);
+       EBUG_ON(path->locks_want >= new_locks_want);
 
-       iter->locks_want = new_locks_want;
+       path->locks_want = new_locks_want;
 
-       if (btree_iter_get_locks(iter, true, _THIS_IP_))
+       if (btree_path_get_locks(trans, path, true))
                return true;
 
        /*
@@ -444,7 +472,7 @@ bool __bch2_btree_iter_upgrade(struct btree_iter *iter,
         * iterators in the btree_trans here.
         *
         * On failure to upgrade the iterator, setting iter->locks_want and
-        * calling get_locks() is sufficient to make bch2_btree_iter_traverse()
+        * calling get_locks() is sufficient to make bch2_btree_path_traverse()
         * get the locks we want on transaction restart.
         *
         * But if this iterator was a clone, on transaction restart what we did
@@ -456,75 +484,67 @@ bool __bch2_btree_iter_upgrade(struct btree_iter *iter,
         *
         * The code below used to be needed to ensure ancestor nodes get locked
         * before interior nodes - now that's handled by
-        * bch2_btree_iter_traverse_all().
+        * bch2_btree_path_traverse_all().
         */
-       trans_for_each_iter(iter->trans, linked)
-               if (linked != iter &&
-                   btree_iter_type(linked) == btree_iter_type(iter) &&
-                   linked->btree_id == iter->btree_id &&
+       trans_for_each_path(trans, linked)
+               if (linked != path &&
+                   linked->cached == path->cached &&
+                   linked->btree_id == path->btree_id &&
                    linked->locks_want < new_locks_want) {
                        linked->locks_want = new_locks_want;
-                       btree_iter_get_locks(linked, true, _THIS_IP_);
+                       btree_path_get_locks(trans, linked, true);
                }
 
-       if (iter->should_be_locked)
-               btree_trans_restart(iter->trans);
        return false;
 }
 
-void __bch2_btree_iter_downgrade(struct btree_iter *iter,
+void __bch2_btree_path_downgrade(struct btree_path *path,
                                 unsigned new_locks_want)
 {
        unsigned l;
 
-       EBUG_ON(iter->locks_want < new_locks_want);
+       EBUG_ON(path->locks_want < new_locks_want);
 
-       iter->locks_want = new_locks_want;
+       path->locks_want = new_locks_want;
 
-       while (iter->nodes_locked &&
-              (l = __fls(iter->nodes_locked)) >= iter->locks_want) {
-               if (l > iter->level) {
-                       btree_node_unlock(iter, l);
+       while (path->nodes_locked &&
+              (l = __fls(path->nodes_locked)) >= path->locks_want) {
+               if (l > path->level) {
+                       btree_node_unlock(path, l);
                } else {
-                       if (btree_node_intent_locked(iter, l)) {
-                               six_lock_downgrade(&iter->l[l].b->c.lock);
-                               iter->nodes_intent_locked ^= 1 << l;
+                       if (btree_node_intent_locked(path, l)) {
+                               six_lock_downgrade(&path->l[l].b->c.lock);
+                               path->nodes_intent_locked ^= 1 << l;
                        }
                        break;
                }
        }
 
-       bch2_btree_trans_verify_locks(iter->trans);
+       bch2_btree_path_verify_locks(path);
 }
 
 void bch2_trans_downgrade(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
+       struct btree_path *path;
 
-       trans_for_each_iter(trans, iter)
-               bch2_btree_iter_downgrade(iter);
+       trans_for_each_path(trans, path)
+               bch2_btree_path_downgrade(path);
 }
 
 /* Btree transaction locking: */
 
-static inline bool btree_iter_should_be_locked(struct btree_iter *iter)
-{
-       return (iter->flags & BTREE_ITER_KEEP_UNTIL_COMMIT) ||
-               iter->should_be_locked;
-}
-
 bool bch2_trans_relock(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
+       struct btree_path *path;
 
        if (unlikely(trans->restarted))
                return false;
 
-       trans_for_each_iter(trans, iter)
-               if (btree_iter_should_be_locked(iter) &&
-                   !bch2_btree_iter_relock(iter, _RET_IP_)) {
-                       trace_trans_restart_relock(trans->ip, _RET_IP_,
-                                       iter->btree_id, &iter->real_pos);
+       trans_for_each_path(trans, path)
+               if (path->should_be_locked &&
+                   !bch2_btree_path_relock(trans, path, _RET_IP_)) {
+                       trace_trans_restart_relock(trans->fn, _RET_IP_,
+                                       path->btree_id, &path->pos);
                        BUG_ON(!trans->restarted);
                        return false;
                }
@@ -533,10 +553,10 @@ bool bch2_trans_relock(struct btree_trans *trans)
 
 void bch2_trans_unlock(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
+       struct btree_path *path;
 
-       trans_for_each_iter(trans, iter)
-               __bch2_btree_iter_unlock(iter);
+       trans_for_each_path(trans, path)
+               __bch2_btree_path_unlock(path);
 
        BUG_ON(lock_class_is_held(&bch2_btree_node_lock_key));
 }
@@ -545,26 +565,27 @@ void bch2_trans_unlock(struct btree_trans *trans)
 
 #ifdef CONFIG_BCACHEFS_DEBUG
 
-static void bch2_btree_iter_verify_cached(struct btree_iter *iter)
+static void bch2_btree_path_verify_cached(struct btree_trans *trans,
+                                         struct btree_path *path)
 {
        struct bkey_cached *ck;
-       bool locked = btree_node_locked(iter, 0);
+       bool locked = btree_node_locked(path, 0);
 
-       if (!bch2_btree_node_relock(iter, 0))
+       if (!bch2_btree_node_relock(trans, path, 0))
                return;
 
-       ck = (void *) iter->l[0].b;
-       BUG_ON(ck->key.btree_id != iter->btree_id ||
-              bkey_cmp(ck->key.pos, iter->pos));
+       ck = (void *) path->l[0].b;
+       BUG_ON(ck->key.btree_id != path->btree_id ||
+              bkey_cmp(ck->key.pos, path->pos));
 
        if (!locked)
-               btree_node_unlock(iter, 0);
+               btree_node_unlock(path, 0);
 }
 
-static void bch2_btree_iter_verify_level(struct btree_iter *iter,
-                                        unsigned level)
+static void bch2_btree_path_verify_level(struct btree_trans *trans,
+                               struct btree_path *path, unsigned level)
 {
-       struct btree_iter_level *l;
+       struct btree_path_level *l;
        struct btree_node_iter tmp;
        bool locked;
        struct bkey_packed *p, *k;
@@ -574,65 +595,52 @@ static void bch2_btree_iter_verify_level(struct btree_iter *iter,
        if (!bch2_debug_check_iterators)
                return;
 
-       l       = &iter->l[level];
+       l       = &path->l[level];
        tmp     = l->iter;
-       locked  = btree_node_locked(iter, level);
+       locked  = btree_node_locked(path, level);
 
-       if (btree_iter_type(iter) == BTREE_ITER_CACHED) {
+       if (path->cached) {
                if (!level)
-                       bch2_btree_iter_verify_cached(iter);
+                       bch2_btree_path_verify_cached(trans, path);
                return;
        }
 
-       BUG_ON(iter->level < iter->min_depth);
-
-       if (!btree_iter_node(iter, level))
+       if (!btree_path_node(path, level))
                return;
 
-       if (!bch2_btree_node_relock(iter, level))
+       if (!bch2_btree_node_relock(trans, path, level))
                return;
 
-       BUG_ON(!btree_iter_pos_in_node(iter, l->b));
-
-       /*
-        * node iterators don't use leaf node iterator:
-        */
-       if (btree_iter_type(iter) == BTREE_ITER_NODES &&
-           level <= iter->min_depth)
-               goto unlock;
+       BUG_ON(!btree_path_pos_in_node(path, l->b));
 
        bch2_btree_node_iter_verify(&l->iter, l->b);
 
        /*
-        * For interior nodes, the iterator will have skipped past
-        * deleted keys:
-        *
-        * For extents, the iterator may have skipped past deleted keys (but not
-        * whiteouts)
+        * For interior nodes, the iterator will have skipped past deleted keys:
         */
-       p = level || btree_node_type_is_extents(iter->btree_id)
+       p = level
                ? bch2_btree_node_iter_prev(&tmp, l->b)
                : bch2_btree_node_iter_prev_all(&tmp, l->b);
        k = bch2_btree_node_iter_peek_all(&l->iter, l->b);
 
-       if (p && bkey_iter_pos_cmp(l->b, p, &iter->real_pos) >= 0) {
+       if (p && bkey_iter_pos_cmp(l->b, p, &path->pos) >= 0) {
                msg = "before";
                goto err;
        }
 
-       if (k && bkey_iter_pos_cmp(l->b, k, &iter->real_pos) < 0) {
+       if (k && bkey_iter_pos_cmp(l->b, k, &path->pos) < 0) {
                msg = "after";
                goto err;
        }
-unlock:
+
        if (!locked)
-               btree_node_unlock(iter, level);
+               btree_node_unlock(path, level);
        return;
 err:
        strcpy(buf2, "(none)");
        strcpy(buf3, "(none)");
 
-       bch2_bpos_to_text(&PBUF(buf1), iter->real_pos);
+       bch2_bpos_to_text(&PBUF(buf1), path->pos);
 
        if (p) {
                struct bkey uk = bkey_unpack_key(l->b, p);
@@ -644,79 +652,175 @@ err:
                bch2_bkey_to_text(&PBUF(buf3), &uk);
        }
 
-       panic("iterator should be %s key at level %u:\n"
-             "iter pos %s\n"
+       panic("path should be %s key at level %u:\n"
+             "path pos %s\n"
              "prev key %s\n"
              "cur  key %s\n",
              msg, level, buf1, buf2, buf3);
 }
 
-static void bch2_btree_iter_verify(struct btree_iter *iter)
+static void bch2_btree_path_verify(struct btree_trans *trans,
+                                  struct btree_path *path)
 {
-       struct btree_trans *trans = iter->trans;
        struct bch_fs *c = trans->c;
-       enum btree_iter_type type = btree_iter_type(iter);
        unsigned i;
 
-       EBUG_ON(iter->btree_id >= BTREE_ID_NR);
+       EBUG_ON(path->btree_id >= BTREE_ID_NR);
 
-       BUG_ON(!(iter->flags & BTREE_ITER_ALL_SNAPSHOTS) &&
-              iter->pos.snapshot != iter->snapshot);
+       for (i = 0; i < (!path->cached ? BTREE_MAX_DEPTH : 1); i++) {
+               if (!path->l[i].b) {
+                       BUG_ON(!path->cached &&
+                              c->btree_roots[path->btree_id].b->c.level > i);
+                       break;
+               }
+
+               bch2_btree_path_verify_level(trans, path, i);
+       }
+
+       bch2_btree_path_verify_locks(path);
+}
+
+void bch2_trans_verify_paths(struct btree_trans *trans)
+{
+       struct btree_path *path;
+
+       trans_for_each_path(trans, path)
+               bch2_btree_path_verify(trans, path);
+}
+
+static void bch2_btree_iter_verify(struct btree_iter *iter)
+{
+       struct btree_trans *trans = iter->trans;
+
+       BUG_ON(iter->btree_id >= BTREE_ID_NR);
+
+       BUG_ON(!!(iter->flags & BTREE_ITER_CACHED) != iter->path->cached);
 
        BUG_ON((iter->flags & BTREE_ITER_IS_EXTENTS) &&
               (iter->flags & BTREE_ITER_ALL_SNAPSHOTS));
 
-       BUG_ON(type == BTREE_ITER_NODES &&
-              !(iter->flags & BTREE_ITER_ALL_SNAPSHOTS));
-
-       BUG_ON(type != BTREE_ITER_NODES &&
+       BUG_ON(!(iter->flags & __BTREE_ITER_ALL_SNAPSHOTS) &&
               (iter->flags & BTREE_ITER_ALL_SNAPSHOTS) &&
               !btree_type_has_snapshots(iter->btree_id));
 
-       for (i = 0; i < (type != BTREE_ITER_CACHED ? BTREE_MAX_DEPTH : 1); i++) {
-               if (!iter->l[i].b) {
-                       BUG_ON(c->btree_roots[iter->btree_id].b->c.level > i);
-                       break;
-               }
-
-               bch2_btree_iter_verify_level(iter, i);
-       }
-
-       bch2_btree_iter_verify_locks(iter);
+       if (iter->update_path)
+               bch2_btree_path_verify(trans, iter->update_path);
+       bch2_btree_path_verify(trans, iter->path);
 }
 
 static void bch2_btree_iter_verify_entry_exit(struct btree_iter *iter)
 {
-       enum btree_iter_type type = btree_iter_type(iter);
+       BUG_ON((iter->flags & BTREE_ITER_FILTER_SNAPSHOTS) &&
+              !iter->pos.snapshot);
 
        BUG_ON(!(iter->flags & BTREE_ITER_ALL_SNAPSHOTS) &&
               iter->pos.snapshot != iter->snapshot);
 
-       BUG_ON((type == BTREE_ITER_KEYS ||
-               type == BTREE_ITER_CACHED) &&
-              (bkey_cmp(iter->pos, bkey_start_pos(&iter->k)) < 0 ||
-               bkey_cmp(iter->pos, iter->k.p) > 0));
+       BUG_ON(bkey_cmp(iter->pos, bkey_start_pos(&iter->k)) < 0 ||
+              bkey_cmp(iter->pos, iter->k.p) > 0);
 }
 
-void bch2_btree_trans_verify_iters(struct btree_trans *trans, struct btree *b)
+static int bch2_btree_iter_verify_ret(struct btree_iter *iter, struct bkey_s_c k)
 {
-       struct btree_iter *iter;
+       struct btree_trans *trans = iter->trans;
+       struct btree_iter copy;
+       struct bkey_s_c prev;
+       int ret = 0;
 
        if (!bch2_debug_check_iterators)
-               return;
+               return 0;
+
+       if (!(iter->flags & BTREE_ITER_FILTER_SNAPSHOTS))
+               return 0;
+
+       if (bkey_err(k) || !k.k)
+               return 0;
+
+       BUG_ON(!bch2_snapshot_is_ancestor(trans->c,
+                                         iter->snapshot,
+                                         k.k->p.snapshot));
+
+       bch2_trans_iter_init(trans, &copy, iter->btree_id, iter->pos,
+                            BTREE_ITER_NOPRESERVE|
+                            BTREE_ITER_ALL_SNAPSHOTS);
+       prev = bch2_btree_iter_prev(&copy);
+       if (!prev.k)
+               goto out;
+
+       ret = bkey_err(prev);
+       if (ret)
+               goto out;
+
+       if (!bkey_cmp(prev.k->p, k.k->p) &&
+           bch2_snapshot_is_ancestor(trans->c, iter->snapshot,
+                                     prev.k->p.snapshot) > 0) {
+               char buf1[100], buf2[200];
+
+               bch2_bkey_to_text(&PBUF(buf1), k.k);
+               bch2_bkey_to_text(&PBUF(buf2), prev.k);
+
+               panic("iter snap %u\n"
+                     "k    %s\n"
+                     "prev %s\n",
+                     iter->snapshot,
+                     buf1, buf2);
+       }
+out:
+       bch2_trans_iter_exit(trans, &copy);
+       return ret;
+}
+
+void bch2_assert_pos_locked(struct btree_trans *trans, enum btree_id id,
+                           struct bpos pos, bool key_cache)
+{
+       struct btree_path *path;
+       unsigned idx;
+       char buf[100];
+
+       trans_for_each_path_inorder(trans, path, idx) {
+               int cmp = cmp_int(path->btree_id, id) ?:
+                       cmp_int(path->cached, key_cache);
+
+               if (cmp > 0)
+                       break;
+               if (cmp < 0)
+                       continue;
+
+               if (!(path->nodes_locked & 1) ||
+                   !path->should_be_locked)
+                       continue;
+
+               if (!key_cache) {
+                       if (bkey_cmp(pos, path->l[0].b->data->min_key) >= 0 &&
+                           bkey_cmp(pos, path->l[0].b->key.k.p) <= 0)
+                               return;
+               } else {
+                       if (!bkey_cmp(pos, path->pos))
+                               return;
+               }
+       }
 
-       trans_for_each_iter_with_node(trans, b, iter)
-               bch2_btree_iter_verify_level(iter, b->c.level);
+       bch2_dump_trans_paths_updates(trans);
+       panic("not locked: %s %s%s\n",
+             bch2_btree_ids[id],
+             (bch2_bpos_to_text(&PBUF(buf), pos), buf),
+             key_cache ? " cached" : "");
 }
 
 #else
 
-static inline void bch2_btree_iter_verify_level(struct btree_iter *iter, unsigned l) {}
+static inline void bch2_btree_path_verify_level(struct btree_trans *trans,
+                                               struct btree_path *path, unsigned l) {}
+static inline void bch2_btree_path_verify(struct btree_trans *trans,
+                                         struct btree_path *path) {}
 static inline void bch2_btree_iter_verify(struct btree_iter *iter) {}
 static inline void bch2_btree_iter_verify_entry_exit(struct btree_iter *iter) {}
+static inline int bch2_btree_iter_verify_ret(struct btree_iter *iter, struct bkey_s_c k) { return 0; }
 
 #endif
 
+/* Btree path: fixups after btree updates */
+
 static void btree_node_iter_set_set_pos(struct btree_node_iter *iter,
                                        struct btree *b,
                                        struct bset_tree *t,
@@ -734,40 +838,38 @@ static void btree_node_iter_set_set_pos(struct btree_node_iter *iter,
        bch2_btree_node_iter_push(iter, b, k, btree_bkey_last(b, t));
 }
 
-static void __bch2_btree_iter_fix_key_modified(struct btree_iter *iter,
+static void __bch2_btree_path_fix_key_modified(struct btree_path *path,
                                               struct btree *b,
                                               struct bkey_packed *where)
 {
-       struct btree_iter_level *l = &iter->l[b->c.level];
+       struct btree_path_level *l = &path->l[b->c.level];
 
        if (where != bch2_btree_node_iter_peek_all(&l->iter, l->b))
                return;
 
-       if (bkey_iter_pos_cmp(l->b, where, &iter->real_pos) < 0)
+       if (bkey_iter_pos_cmp(l->b, where, &path->pos) < 0)
                bch2_btree_node_iter_advance(&l->iter, l->b);
-
-       btree_iter_set_dirty(iter, BTREE_ITER_NEED_PEEK);
 }
 
-void bch2_btree_iter_fix_key_modified(struct btree_iter *iter,
+void bch2_btree_path_fix_key_modified(struct btree_trans *trans,
                                      struct btree *b,
                                      struct bkey_packed *where)
 {
-       struct btree_iter *linked;
+       struct btree_path *path;
 
-       trans_for_each_iter_with_node(iter->trans, b, linked) {
-               __bch2_btree_iter_fix_key_modified(linked, b, where);
-               bch2_btree_iter_verify_level(linked, b->c.level);
+       trans_for_each_path_with_node(trans, b, path) {
+               __bch2_btree_path_fix_key_modified(path, b, where);
+               bch2_btree_path_verify_level(trans, path, b->c.level);
        }
 }
 
-static void __bch2_btree_node_iter_fix(struct btree_iter *iter,
-                                     struct btree *b,
-                                     struct btree_node_iter *node_iter,
-                                     struct bset_tree *t,
-                                     struct bkey_packed *where,
-                                     unsigned clobber_u64s,
-                                     unsigned new_u64s)
+static void __bch2_btree_node_iter_fix(struct btree_path *path,
+                                      struct btree *b,
+                                      struct btree_node_iter *node_iter,
+                                      struct bset_tree *t,
+                                      struct bkey_packed *where,
+                                      unsigned clobber_u64s,
+                                      unsigned new_u64s)
 {
        const struct bkey_packed *end = btree_bkey_last(b, t);
        struct btree_node_iter_set *set;
@@ -785,7 +887,7 @@ static void __bch2_btree_node_iter_fix(struct btree_iter *iter,
 
        /* didn't find the bset in the iterator - might have to readd it: */
        if (new_u64s &&
-           bkey_iter_pos_cmp(b, where, &iter->real_pos) >= 0) {
+           bkey_iter_pos_cmp(b, where, &path->pos) >= 0) {
                bch2_btree_node_iter_push(node_iter, b, where, end);
                goto fixup_done;
        } else {
@@ -800,7 +902,7 @@ found:
                return;
 
        if (new_u64s &&
-           bkey_iter_pos_cmp(b, where, &iter->real_pos) >= 0) {
+           bkey_iter_pos_cmp(b, where, &path->pos) >= 0) {
                set->k = offset;
        } else if (set->k < offset + clobber_u64s) {
                set->k = offset + new_u64s;
@@ -826,8 +928,7 @@ fixup_done:
         */
        if (!bch2_btree_node_iter_end(node_iter) &&
            iter_current_key_modified &&
-           (b->c.level ||
-            btree_node_type_is_extents(iter->btree_id))) {
+           b->c.level) {
                struct bset_tree *t;
                struct bkey_packed *k, *k2, *p;
 
@@ -852,14 +953,10 @@ fixup_done:
                                                            b, t, k2);
                }
        }
-
-       if (!b->c.level &&
-           node_iter == &iter->l[0].iter &&
-           iter_current_key_modified)
-               btree_iter_set_dirty(iter, BTREE_ITER_NEED_PEEK);
 }
 
-void bch2_btree_node_iter_fix(struct btree_iter *iter,
+void bch2_btree_node_iter_fix(struct btree_trans *trans,
+                             struct btree_path *path,
                              struct btree *b,
                              struct btree_node_iter *node_iter,
                              struct bkey_packed *where,
@@ -867,31 +964,31 @@ void bch2_btree_node_iter_fix(struct btree_iter *iter,
                              unsigned new_u64s)
 {
        struct bset_tree *t = bch2_bkey_to_bset(b, where);
-       struct btree_iter *linked;
+       struct btree_path *linked;
 
-       if (node_iter != &iter->l[b->c.level].iter) {
-               __bch2_btree_node_iter_fix(iter, b, node_iter, t,
+       if (node_iter != &path->l[b->c.level].iter) {
+               __bch2_btree_node_iter_fix(path, b, node_iter, t,
                                           where, clobber_u64s, new_u64s);
 
                if (bch2_debug_check_iterators)
                        bch2_btree_node_iter_verify(node_iter, b);
        }
 
-       trans_for_each_iter_with_node(iter->trans, b, linked) {
+       trans_for_each_path_with_node(trans, b, linked) {
                __bch2_btree_node_iter_fix(linked, b,
                                           &linked->l[b->c.level].iter, t,
                                           where, clobber_u64s, new_u64s);
-               bch2_btree_iter_verify_level(linked, b->c.level);
+               bch2_btree_path_verify_level(trans, linked, b->c.level);
        }
 }
 
-static inline struct bkey_s_c __btree_iter_unpack(struct btree_iter *iter,
-                                                 struct btree_iter_level *l,
+/* Btree path level: pointer to a particular btree node and node iter */
+
+static inline struct bkey_s_c __btree_iter_unpack(struct bch_fs *c,
+                                                 struct btree_path_level *l,
                                                  struct bkey *u,
                                                  struct bkey_packed *k)
 {
-       struct bkey_s_c ret;
-
        if (unlikely(!k)) {
                /*
                 * signal to bch2_btree_iter_peek_slot() that we're currently at
@@ -901,58 +998,50 @@ static inline struct bkey_s_c __btree_iter_unpack(struct btree_iter *iter,
                return bkey_s_c_null;
        }
 
-       ret = bkey_disassemble(l->b, k, u);
-
-       /*
-        * XXX: bch2_btree_bset_insert_key() generates invalid keys when we
-        * overwrite extents - it sets k->type = KEY_TYPE_deleted on the key
-        * being overwritten but doesn't change k->size. But this is ok, because
-        * those keys are never written out, we just have to avoid a spurious
-        * assertion here:
-        */
-       if (bch2_debug_check_bkeys && !bkey_deleted(ret.k))
-               bch2_bkey_debugcheck(iter->trans->c, l->b, ret);
-
-       return ret;
+       return bkey_disassemble(l->b, k, u);
 }
 
-/* peek_all() doesn't skip deleted keys */
-static inline struct bkey_s_c btree_iter_level_peek_all(struct btree_iter *iter,
-                                                       struct btree_iter_level *l)
+static inline struct bkey_s_c btree_path_level_peek_all(struct bch_fs *c,
+                                                       struct btree_path_level *l,
+                                                       struct bkey *u)
 {
-       return __btree_iter_unpack(iter, l, &iter->k,
+       return __btree_iter_unpack(c, l, u,
                        bch2_btree_node_iter_peek_all(&l->iter, l->b));
 }
 
-static inline struct bkey_s_c btree_iter_level_peek(struct btree_iter *iter,
-                                                   struct btree_iter_level *l)
+static inline struct bkey_s_c btree_path_level_peek(struct bch_fs *c,
+                                                   struct btree_path *path,
+                                                   struct btree_path_level *l,
+                                                   struct bkey *u)
 {
-       struct bkey_s_c k = __btree_iter_unpack(iter, l, &iter->k,
+       struct bkey_s_c k = __btree_iter_unpack(c, l, u,
                        bch2_btree_node_iter_peek(&l->iter, l->b));
 
-       iter->real_pos = k.k ? k.k->p : l->b->key.k.p;
+       path->pos = k.k ? k.k->p : l->b->key.k.p;
        return k;
 }
 
-static inline struct bkey_s_c btree_iter_level_prev(struct btree_iter *iter,
-                                                   struct btree_iter_level *l)
+static inline struct bkey_s_c btree_path_level_prev(struct bch_fs *c,
+                                                   struct btree_path *path,
+                                                   struct btree_path_level *l,
+                                                   struct bkey *u)
 {
-       struct bkey_s_c k = __btree_iter_unpack(iter, l, &iter->k,
+       struct bkey_s_c k = __btree_iter_unpack(c, l, u,
                        bch2_btree_node_iter_prev(&l->iter, l->b));
 
-       iter->real_pos = k.k ? k.k->p : l->b->data->min_key;
+       path->pos = k.k ? k.k->p : l->b->data->min_key;
        return k;
 }
 
-static inline bool btree_iter_advance_to_pos(struct btree_iter *iter,
-                                            struct btree_iter_level *l,
+static inline bool btree_path_advance_to_pos(struct btree_path *path,
+                                            struct btree_path_level *l,
                                             int max_advance)
 {
        struct bkey_packed *k;
        int nr_advanced = 0;
 
        while ((k = bch2_btree_node_iter_peek_all(&l->iter, l->b)) &&
-              bkey_iter_pos_cmp(l->b, k, &iter->real_pos) < 0) {
+              bkey_iter_pos_cmp(l->b, k, &path->pos) < 0) {
                if (max_advance > 0 && nr_advanced >= max_advance)
                        return false;
 
@@ -966,9 +1055,11 @@ static inline bool btree_iter_advance_to_pos(struct btree_iter *iter,
 /*
  * Verify that iterator for parent node points to child node:
  */
-static void btree_iter_verify_new_node(struct btree_iter *iter, struct btree *b)
+static void btree_path_verify_new_node(struct btree_trans *trans,
+                                      struct btree_path *path, struct btree *b)
 {
-       struct btree_iter_level *l;
+       struct bch_fs *c = trans->c;
+       struct btree_path_level *l;
        unsigned plevel;
        bool parent_locked;
        struct bkey_packed *k;
@@ -976,16 +1067,19 @@ static void btree_iter_verify_new_node(struct btree_iter *iter, struct btree *b)
        if (!IS_ENABLED(CONFIG_BCACHEFS_DEBUG))
                return;
 
+       if (!test_bit(JOURNAL_REPLAY_DONE, &c->journal.flags))
+               return;
+
        plevel = b->c.level + 1;
-       if (!btree_iter_node(iter, plevel))
+       if (!btree_path_node(path, plevel))
                return;
 
-       parent_locked = btree_node_locked(iter, plevel);
+       parent_locked = btree_node_locked(path, plevel);
 
-       if (!bch2_btree_node_relock(iter, plevel))
+       if (!bch2_btree_node_relock(trans, path, plevel))
                return;
 
-       l = &iter->l[plevel];
+       l = &path->l[plevel];
        k = bch2_btree_node_iter_peek_all(&l->iter, l->b);
        if (!k ||
            bkey_deleted(k) ||
@@ -996,8 +1090,8 @@ static void btree_iter_verify_new_node(struct btree_iter *iter, struct btree *b)
                char buf4[100];
                struct bkey uk = bkey_unpack_key(b, k);
 
-               bch2_dump_btree_node(iter->trans->c, l->b);
-               bch2_bpos_to_text(&PBUF(buf1), iter->real_pos);
+               bch2_dump_btree_node(c, l->b);
+               bch2_bpos_to_text(&PBUF(buf1), path->pos);
                bch2_bkey_to_text(&PBUF(buf2), &uk);
                bch2_bpos_to_text(&PBUF(buf3), b->data->min_key);
                bch2_bpos_to_text(&PBUF(buf3), b->data->max_key);
@@ -1005,20 +1099,20 @@ static void btree_iter_verify_new_node(struct btree_iter *iter, struct btree *b)
                      "iter pos %s %s\n"
                      "iter key %s\n"
                      "new node %s-%s\n",
-                     bch2_btree_ids[iter->btree_id], buf1,
+                     bch2_btree_ids[path->btree_id], buf1,
                      buf2, buf3, buf4);
        }
 
        if (!parent_locked)
-               btree_node_unlock(iter, b->c.level + 1);
+               btree_node_unlock(path, plevel);
 }
 
-static inline void __btree_iter_init(struct btree_iter *iter,
-                                    unsigned level)
+static inline void __btree_path_level_init(struct btree_path *path,
+                                          unsigned level)
 {
-       struct btree_iter_level *l = &iter->l[level];
+       struct btree_path_level *l = &path->l[level];
 
-       bch2_btree_node_iter_init(&l->iter, l->b, &iter->real_pos);
+       bch2_btree_node_iter_init(&l->iter, l->b, &path->pos);
 
        /*
         * Iterators to interior nodes should always be pointed at the first non
@@ -1026,63 +1120,48 @@ static inline void __btree_iter_init(struct btree_iter *iter,
         */
        if (level)
                bch2_btree_node_iter_peek(&l->iter, l->b);
-
-       btree_iter_set_dirty(iter, BTREE_ITER_NEED_PEEK);
 }
 
-static inline void btree_iter_node_set(struct btree_iter *iter,
-                                      struct btree *b)
+static inline void btree_path_level_init(struct btree_trans *trans,
+                                        struct btree_path *path,
+                                        struct btree *b)
 {
-       BUG_ON(btree_iter_type(iter) == BTREE_ITER_CACHED);
+       BUG_ON(path->cached);
 
-       btree_iter_verify_new_node(iter, b);
+       btree_path_verify_new_node(trans, path, b);
 
-       EBUG_ON(!btree_iter_pos_in_node(iter, b));
+       EBUG_ON(!btree_path_pos_in_node(path, b));
        EBUG_ON(b->c.lock.state.seq & 1);
 
-       iter->l[b->c.level].lock_seq = b->c.lock.state.seq;
-       iter->l[b->c.level].b = b;
-       __btree_iter_init(iter, b->c.level);
+       path->l[b->c.level].lock_seq = b->c.lock.state.seq;
+       path->l[b->c.level].b = b;
+       __btree_path_level_init(path, b->c.level);
 }
 
+/* Btree path: fixups after btree node updates: */
+
 /*
  * A btree node is being replaced - update the iterator to point to the new
  * node:
  */
-void bch2_btree_iter_node_replace(struct btree_iter *iter, struct btree *b)
+void bch2_trans_node_add(struct btree_trans *trans, struct btree *b)
 {
-       enum btree_node_locked_type t;
-       struct btree_iter *linked;
+       struct btree_path *path;
 
-       trans_for_each_iter(iter->trans, linked)
-               if (btree_iter_type(linked) != BTREE_ITER_CACHED &&
-                   btree_iter_pos_in_node(linked, b)) {
-                       /*
-                        * bch2_btree_iter_node_drop() has already been called -
-                        * the old node we're replacing has already been
-                        * unlocked and the pointer invalidated
-                        */
-                       BUG_ON(btree_node_locked(linked, b->c.level));
+       trans_for_each_path(trans, path)
+               if (!path->cached &&
+                   btree_path_pos_in_node(path, b)) {
+                       enum btree_node_locked_type t =
+                               btree_lock_want(path, b->c.level);
 
-                       t = btree_lock_want(linked, b->c.level);
-                       if (t != BTREE_NODE_UNLOCKED) {
+                       if (path->nodes_locked &&
+                           t != BTREE_NODE_UNLOCKED) {
+                               btree_node_unlock(path, b->c.level);
                                six_lock_increment(&b->c.lock, t);
-                               mark_btree_node_locked(linked, b->c.level, t);
+                               mark_btree_node_locked(path, b->c.level, t);
                        }
 
-                       btree_iter_node_set(linked, b);
-               }
-}
-
-void bch2_btree_iter_node_drop(struct btree_iter *iter, struct btree *b)
-{
-       struct btree_iter *linked;
-       unsigned level = b->c.level;
-
-       trans_for_each_iter(iter->trans, linked)
-               if (linked->l[level].b == b) {
-                       btree_node_unlock(linked, level);
-                       linked->l[level].b = BTREE_ITER_NO_NODE_DROP;
+                       btree_path_level_init(trans, path, b);
                }
 }
 
@@ -1090,14 +1169,16 @@ void bch2_btree_iter_node_drop(struct btree_iter *iter, struct btree *b)
  * A btree node has been modified in such a way as to invalidate iterators - fix
  * them:
  */
-void bch2_btree_iter_reinit_node(struct btree_iter *iter, struct btree *b)
+void bch2_trans_node_reinit_iter(struct btree_trans *trans, struct btree *b)
 {
-       struct btree_iter *linked;
+       struct btree_path *path;
 
-       trans_for_each_iter_with_node(iter->trans, b, linked)
-               __btree_iter_init(linked, b->c.level);
+       trans_for_each_path_with_node(trans, b, path)
+               __btree_path_level_init(path, b->c.level);
 }
 
+/* Btree path: traverse, set_pos: */
+
 static int lock_root_check_fn(struct six_lock *lock, void *p)
 {
        struct btree *b = container_of(lock, struct btree, c.lock);
@@ -1106,38 +1187,38 @@ static int lock_root_check_fn(struct six_lock *lock, void *p)
        return b == *rootp ? 0 : -1;
 }
 
-static inline int btree_iter_lock_root(struct btree_trans *trans,
-                                      struct btree_iter *iter,
+static inline int btree_path_lock_root(struct btree_trans *trans,
+                                      struct btree_path *path,
                                       unsigned depth_want,
                                       unsigned long trace_ip)
 {
        struct bch_fs *c = trans->c;
-       struct btree *b, **rootp = &c->btree_roots[iter->btree_id].b;
+       struct btree *b, **rootp = &c->btree_roots[path->btree_id].b;
        enum six_lock_type lock_type;
        unsigned i;
 
-       EBUG_ON(iter->nodes_locked);
+       EBUG_ON(path->nodes_locked);
 
        while (1) {
                b = READ_ONCE(*rootp);
-               iter->level = READ_ONCE(b->c.level);
+               path->level = READ_ONCE(b->c.level);
 
-               if (unlikely(iter->level < depth_want)) {
+               if (unlikely(path->level < depth_want)) {
                        /*
                         * the root is at a lower depth than the depth we want:
                         * got to the end of the btree, or we're walking nodes
                         * greater than some depth and there are no nodes >=
                         * that depth
                         */
-                       iter->level = depth_want;
-                       for (i = iter->level; i < BTREE_MAX_DEPTH; i++)
-                               iter->l[i].b = NULL;
+                       path->level = depth_want;
+                       for (i = path->level; i < BTREE_MAX_DEPTH; i++)
+                               path->l[i].b = NULL;
                        return 1;
                }
 
-               lock_type = __btree_lock_want(iter, iter->level);
-               if (unlikely(!btree_node_lock(b, SPOS_MAX, iter->level,
-                                             iter, lock_type,
+               lock_type = __btree_lock_want(path, path->level);
+               if (unlikely(!btree_node_lock(trans, path, b, SPOS_MAX,
+                                             path->level, lock_type,
                                              lock_root_check_fn, rootp,
                                              trace_ip))) {
                        if (trans->restarted)
@@ -1146,16 +1227,16 @@ static inline int btree_iter_lock_root(struct btree_trans *trans,
                }
 
                if (likely(b == READ_ONCE(*rootp) &&
-                          b->c.level == iter->level &&
+                          b->c.level == path->level &&
                           !race_fault())) {
-                       for (i = 0; i < iter->level; i++)
-                               iter->l[i].b = BTREE_ITER_NO_NODE_LOCK_ROOT;
-                       iter->l[iter->level].b = b;
-                       for (i = iter->level + 1; i < BTREE_MAX_DEPTH; i++)
-                               iter->l[i].b = NULL;
-
-                       mark_btree_node_locked(iter, iter->level, lock_type);
-                       btree_iter_node_set(iter, b);
+                       for (i = 0; i < path->level; i++)
+                               path->l[i].b = BTREE_ITER_NO_NODE_LOCK_ROOT;
+                       path->l[path->level].b = b;
+                       for (i = path->level + 1; i < BTREE_MAX_DEPTH; i++)
+                               path->l[i].b = NULL;
+
+                       mark_btree_node_locked(path, path->level, lock_type);
+                       btree_path_level_init(trans, path, b);
                        return 0;
                }
 
@@ -1164,23 +1245,23 @@ static inline int btree_iter_lock_root(struct btree_trans *trans,
 }
 
 noinline
-static int btree_iter_prefetch(struct btree_iter *iter)
+static int btree_path_prefetch(struct btree_trans *trans, struct btree_path *path)
 {
-       struct bch_fs *c = iter->trans->c;
-       struct btree_iter_level *l = &iter->l[iter->level];
+       struct bch_fs *c = trans->c;
+       struct btree_path_level *l = path_l(path);
        struct btree_node_iter node_iter = l->iter;
        struct bkey_packed *k;
        struct bkey_buf tmp;
        unsigned nr = test_bit(BCH_FS_STARTED, &c->flags)
-               ? (iter->level > 1 ? 0 :  2)
-               : (iter->level > 1 ? 1 : 16);
-       bool was_locked = btree_node_locked(iter, iter->level);
+               ? (path->level > 1 ? 0 :  2)
+               : (path->level > 1 ? 1 : 16);
+       bool was_locked = btree_node_locked(path, path->level);
        int ret = 0;
 
        bch2_bkey_buf_init(&tmp);
 
        while (nr && !ret) {
-               if (!bch2_btree_node_relock(iter, iter->level))
+               if (!bch2_btree_node_relock(trans, path, path->level))
                        break;
 
                bch2_btree_node_iter_advance(&node_iter, l->b);
@@ -1189,26 +1270,62 @@ static int btree_iter_prefetch(struct btree_iter *iter)
                        break;
 
                bch2_bkey_buf_unpack(&tmp, c, l->b, k);
-               ret = bch2_btree_node_prefetch(c, iter, tmp.k, iter->btree_id,
-                                              iter->level - 1);
+               ret = bch2_btree_node_prefetch(c, trans, path, tmp.k, path->btree_id,
+                                              path->level - 1);
+       }
+
+       if (!was_locked)
+               btree_node_unlock(path, path->level);
+
+       bch2_bkey_buf_exit(&tmp, c);
+       return ret;
+}
+
+static int btree_path_prefetch_j(struct btree_trans *trans, struct btree_path *path,
+                                struct btree_and_journal_iter *jiter)
+{
+       struct bch_fs *c = trans->c;
+       struct bkey_s_c k;
+       struct bkey_buf tmp;
+       unsigned nr = test_bit(BCH_FS_STARTED, &c->flags)
+               ? (path->level > 1 ? 0 :  2)
+               : (path->level > 1 ? 1 : 16);
+       bool was_locked = btree_node_locked(path, path->level);
+       int ret = 0;
+
+       bch2_bkey_buf_init(&tmp);
+
+       while (nr && !ret) {
+               if (!bch2_btree_node_relock(trans, path, path->level))
+                       break;
+
+               bch2_btree_and_journal_iter_advance(jiter);
+               k = bch2_btree_and_journal_iter_peek(jiter);
+               if (!k.k)
+                       break;
+
+               bch2_bkey_buf_reassemble(&tmp, c, k);
+               ret = bch2_btree_node_prefetch(c, trans, path, tmp.k, path->btree_id,
+                                              path->level - 1);
        }
 
        if (!was_locked)
-               btree_node_unlock(iter, iter->level);
+               btree_node_unlock(path, path->level);
 
        bch2_bkey_buf_exit(&tmp, c);
        return ret;
 }
 
-static noinline void btree_node_mem_ptr_set(struct btree_iter *iter,
+static noinline void btree_node_mem_ptr_set(struct btree_trans *trans,
+                                           struct btree_path *path,
                                            unsigned plevel, struct btree *b)
 {
-       struct btree_iter_level *l = &iter->l[plevel];
-       bool locked = btree_node_locked(iter, plevel);
+       struct btree_path_level *l = &path->l[plevel];
+       bool locked = btree_node_locked(path, plevel);
        struct bkey_packed *k;
        struct bch_btree_ptr_v2 *bp;
 
-       if (!bch2_btree_node_relock(iter, plevel))
+       if (!bch2_btree_node_relock(trans, path, plevel))
                return;
 
        k = bch2_btree_node_iter_peek_all(&l->iter, l->b);
@@ -1218,59 +1335,96 @@ static noinline void btree_node_mem_ptr_set(struct btree_iter *iter,
        bp->mem_ptr = (unsigned long)b;
 
        if (!locked)
-               btree_node_unlock(iter, plevel);
+               btree_node_unlock(path, plevel);
+}
+
+static noinline int btree_node_iter_and_journal_peek(struct btree_trans *trans,
+                                                    struct btree_path *path,
+                                                    unsigned flags,
+                                                    struct bkey_buf *out)
+{
+       struct bch_fs *c = trans->c;
+       struct btree_path_level *l = path_l(path);
+       struct btree_and_journal_iter jiter;
+       struct bkey_s_c k;
+       int ret = 0;
+
+       __bch2_btree_and_journal_iter_init_node_iter(&jiter, c, l->b, l->iter, path->pos);
+
+       k = bch2_btree_and_journal_iter_peek(&jiter);
+
+       bch2_bkey_buf_reassemble(out, c, k);
+
+       if (flags & BTREE_ITER_PREFETCH)
+               ret = btree_path_prefetch_j(trans, path, &jiter);
+
+       bch2_btree_and_journal_iter_exit(&jiter);
+       return ret;
 }
 
-static __always_inline int btree_iter_down(struct btree_trans *trans,
-                                          struct btree_iter *iter,
+static __always_inline int btree_path_down(struct btree_trans *trans,
+                                          struct btree_path *path,
+                                          unsigned flags,
                                           unsigned long trace_ip)
 {
        struct bch_fs *c = trans->c;
-       struct btree_iter_level *l = &iter->l[iter->level];
+       struct btree_path_level *l = path_l(path);
        struct btree *b;
-       unsigned level = iter->level - 1;
-       enum six_lock_type lock_type = __btree_lock_want(iter, level);
+       unsigned level = path->level - 1;
+       enum six_lock_type lock_type = __btree_lock_want(path, level);
+       bool replay_done = test_bit(JOURNAL_REPLAY_DONE, &c->journal.flags);
        struct bkey_buf tmp;
        int ret;
 
-       EBUG_ON(!btree_node_locked(iter, iter->level));
+       EBUG_ON(!btree_node_locked(path, path->level));
 
        bch2_bkey_buf_init(&tmp);
-       bch2_bkey_buf_unpack(&tmp, c, l->b,
-                        bch2_btree_node_iter_peek(&l->iter, l->b));
 
-       b = bch2_btree_node_get(trans, iter, tmp.k, level, lock_type, trace_ip);
+       if (unlikely(!replay_done)) {
+               ret = btree_node_iter_and_journal_peek(trans, path, flags, &tmp);
+               if (ret)
+                       goto err;
+       } else {
+               bch2_bkey_buf_unpack(&tmp, c, l->b,
+                                bch2_btree_node_iter_peek(&l->iter, l->b));
+
+               if (flags & BTREE_ITER_PREFETCH) {
+                       ret = btree_path_prefetch(trans, path);
+                       if (ret)
+                               goto err;
+               }
+       }
+
+       b = bch2_btree_node_get(trans, path, tmp.k, level, lock_type, trace_ip);
        ret = PTR_ERR_OR_ZERO(b);
        if (unlikely(ret))
                goto err;
 
-       mark_btree_node_locked(iter, level, lock_type);
-       btree_iter_node_set(iter, b);
+       mark_btree_node_locked(path, level, lock_type);
+       btree_path_level_init(trans, path, b);
 
-       if (tmp.k->k.type == KEY_TYPE_btree_ptr_v2 &&
+       if (likely(replay_done && tmp.k->k.type == KEY_TYPE_btree_ptr_v2) &&
            unlikely(b != btree_node_mem_ptr(tmp.k)))
-               btree_node_mem_ptr_set(iter, level + 1, b);
-
-       if (iter->flags & BTREE_ITER_PREFETCH)
-               ret = btree_iter_prefetch(iter);
+               btree_node_mem_ptr_set(trans, path, level + 1, b);
 
-       if (btree_node_read_locked(iter, level + 1))
-               btree_node_unlock(iter, level + 1);
-       iter->level = level;
+       if (btree_node_read_locked(path, level + 1))
+               btree_node_unlock(path, level + 1);
+       path->level = level;
 
-       bch2_btree_iter_verify_locks(iter);
+       bch2_btree_path_verify_locks(path);
 err:
        bch2_bkey_buf_exit(&tmp, c);
        return ret;
 }
 
-static int btree_iter_traverse_one(struct btree_iter *, unsigned long);
+static int btree_path_traverse_one(struct btree_trans *, struct btree_path *,
+                                  unsigned, unsigned long);
 
-static int __btree_iter_traverse_all(struct btree_trans *trans, int ret,
+static int __btree_path_traverse_all(struct btree_trans *trans, int ret,
                                     unsigned long trace_ip)
 {
        struct bch_fs *c = trans->c;
-       struct btree_iter *iter;
+       struct btree_path *path;
        int i;
 
        if (trans->in_traverse_all)
@@ -1280,20 +1434,20 @@ static int __btree_iter_traverse_all(struct btree_trans *trans, int ret,
 retry_all:
        trans->restarted = false;
 
-       trans_for_each_iter(trans, iter)
-               iter->should_be_locked = false;
+       trans_for_each_path(trans, path)
+               path->should_be_locked = false;
 
-       btree_trans_sort_iters(trans);
+       btree_trans_verify_sorted(trans);
 
        for (i = trans->nr_sorted - 2; i >= 0; --i) {
-               struct btree_iter *iter1 = trans->iters + trans->sorted[i];
-               struct btree_iter *iter2 = trans->iters + trans->sorted[i + 1];
-
-               if (iter1->btree_id == iter2->btree_id &&
-                   iter1->locks_want < iter2->locks_want)
-                       __bch2_btree_iter_upgrade(iter1, iter2->locks_want);
-               else if (!iter1->locks_want && iter2->locks_want)
-                       __bch2_btree_iter_upgrade(iter1, 1);
+               struct btree_path *path1 = trans->paths + trans->sorted[i];
+               struct btree_path *path2 = trans->paths + trans->sorted[i + 1];
+
+               if (path1->btree_id == path2->btree_id &&
+                   path1->locks_want < path2->locks_want)
+                       __bch2_btree_path_upgrade(trans, path1, path2->locks_want);
+               else if (!path1->locks_want && path2->locks_want)
+                       __bch2_btree_path_upgrade(trans, path1, 1);
        }
 
        bch2_trans_unlock(trans);
@@ -1310,66 +1464,89 @@ retry_all:
                } while (ret);
        }
 
-       if (unlikely(ret == -EIO)) {
-               trans->error = true;
+       if (unlikely(ret == -EIO))
                goto out;
-       }
 
        BUG_ON(ret && ret != -EINTR);
 
        /* Now, redo traversals in correct order: */
-       trans_for_each_iter_inorder(trans, iter) {
-               EBUG_ON(!(trans->iters_linked & (1ULL << iter->idx)));
-
-               ret = btree_iter_traverse_one(iter, _THIS_IP_);
-               if (ret)
-                       goto retry_all;
+       i = 0;
+       while (i < trans->nr_sorted) {
+               path = trans->paths + trans->sorted[i];
 
-               EBUG_ON(!(trans->iters_linked & (1ULL << iter->idx)));
+               /*
+                * Traversing a path can cause another path to be added at about
+                * the same position:
+                */
+               if (path->uptodate) {
+                       ret = btree_path_traverse_one(trans, path, 0, _THIS_IP_);
+                       if (ret)
+                               goto retry_all;
+               } else {
+                       i++;
+               }
        }
 
-       trans_for_each_iter(trans, iter)
-               BUG_ON(iter->uptodate > BTREE_ITER_NEED_PEEK);
+       /*
+        * BTREE_ITER_NEED_RELOCK is ok here - if we called bch2_trans_unlock()
+        * and relock(), relock() won't relock since path->should_be_locked
+        * isn't set yet, which is all fine
+        */
+       trans_for_each_path(trans, path)
+               BUG_ON(path->uptodate >= BTREE_ITER_NEED_TRAVERSE);
 out:
        bch2_btree_cache_cannibalize_unlock(c);
 
        trans->in_traverse_all = false;
 
-       trace_trans_traverse_all(trans->ip, trace_ip);
+       trace_trans_traverse_all(trans->fn, trace_ip);
        return ret;
 }
 
-static int bch2_btree_iter_traverse_all(struct btree_trans *trans)
+static int bch2_btree_path_traverse_all(struct btree_trans *trans)
 {
-       return __btree_iter_traverse_all(trans, 0, _RET_IP_);
+       return __btree_path_traverse_all(trans, 0, _RET_IP_);
 }
 
-static inline bool btree_iter_good_node(struct btree_iter *iter,
+static inline bool btree_path_good_node(struct btree_trans *trans,
+                                       struct btree_path *path,
                                        unsigned l, int check_pos)
 {
-       if (!is_btree_node(iter, l) ||
-           !bch2_btree_node_relock(iter, l))
+       if (!is_btree_node(path, l) ||
+           !bch2_btree_node_relock(trans, path, l))
                return false;
 
-       if (check_pos < 0 && btree_iter_pos_before_node(iter, iter->l[l].b))
+       if (check_pos < 0 && btree_path_pos_before_node(path, path->l[l].b))
                return false;
-       if (check_pos > 0 && btree_iter_pos_after_node(iter, iter->l[l].b))
+       if (check_pos > 0 && btree_path_pos_after_node(path, path->l[l].b))
                return false;
        return true;
 }
 
-static inline unsigned btree_iter_up_until_good_node(struct btree_iter *iter,
+static inline unsigned btree_path_up_until_good_node(struct btree_trans *trans,
+                                                    struct btree_path *path,
                                                     int check_pos)
 {
-       unsigned l = iter->level;
+       unsigned i, l = path->level;
 
-       while (btree_iter_node(iter, l) &&
-              !btree_iter_good_node(iter, l, check_pos)) {
-               btree_node_unlock(iter, l);
-               iter->l[l].b = BTREE_ITER_NO_NODE_UP;
+       while (btree_path_node(path, l) &&
+              !btree_path_good_node(trans, path, l, check_pos)) {
+               btree_node_unlock(path, l);
+               path->l[l].b = BTREE_ITER_NO_NODE_UP;
                l++;
        }
 
+       /* If we need intent locks, take them too: */
+       for (i = l + 1;
+            i < path->locks_want && btree_path_node(path, i);
+            i++)
+               if (!bch2_btree_node_relock(trans, path, i))
+                       while (l <= i) {
+                               btree_node_unlock(path, l);
+                               path->l[l].b = BTREE_ITER_NO_NODE_UP;
+                               l++;
+                       }
+
        return l;
 }
 
@@ -1382,131 +1559,459 @@ static inline unsigned btree_iter_up_until_good_node(struct btree_iter *iter,
  * On error, caller (peek_node()/peek_key()) must return NULL; the error is
  * stashed in the iterator and returned from bch2_trans_exit().
  */
-static int btree_iter_traverse_one(struct btree_iter *iter,
+static int btree_path_traverse_one(struct btree_trans *trans,
+                                  struct btree_path *path,
+                                  unsigned flags,
                                   unsigned long trace_ip)
 {
-       struct btree_trans *trans = iter->trans;
-       unsigned l, depth_want = iter->level;
+       unsigned depth_want = path->level;
        int ret = 0;
 
+       if (unlikely(trans->restarted)) {
+               ret = -EINTR;
+               goto out;
+       }
+
        /*
-        * Ensure we obey iter->should_be_locked: if it's set, we can't unlock
-        * and re-traverse the iterator without a transaction restart:
+        * Ensure we obey path->should_be_locked: if it's set, we can't unlock
+        * and re-traverse the path without a transaction restart:
         */
-       if (iter->should_be_locked) {
-               ret = bch2_btree_iter_relock(iter, trace_ip) ? 0 : -EINTR;
+       if (path->should_be_locked) {
+               ret = bch2_btree_path_relock(trans, path, trace_ip) ? 0 : -EINTR;
                goto out;
        }
 
-       if (btree_iter_type(iter) == BTREE_ITER_CACHED) {
-               ret = bch2_btree_iter_traverse_cached(iter);
+       if (path->cached) {
+               ret = bch2_btree_path_traverse_cached(trans, path, flags);
                goto out;
        }
 
-       if (unlikely(iter->level >= BTREE_MAX_DEPTH))
+       if (unlikely(path->level >= BTREE_MAX_DEPTH))
                goto out;
 
-       iter->level = btree_iter_up_until_good_node(iter, 0);
-
-       /* If we need intent locks, take them too: */
-       for (l = iter->level + 1;
-            l < iter->locks_want && btree_iter_node(iter, l);
-            l++)
-               if (!bch2_btree_node_relock(iter, l))
-                       while (iter->level <= l) {
-                               btree_node_unlock(iter, iter->level);
-                               iter->l[iter->level].b = BTREE_ITER_NO_NODE_UP;
-                               iter->level++;
-                       }
+       path->level = btree_path_up_until_good_node(trans, path, 0);
 
        /*
-        * Note: iter->nodes[iter->level] may be temporarily NULL here - that
+        * Note: path->nodes[path->level] may be temporarily NULL here - that
         * would indicate to other code that we got to the end of the btree,
         * here it indicates that relocking the root failed - it's critical that
-        * btree_iter_lock_root() comes next and that it can't fail
+        * btree_path_lock_root() comes next and that it can't fail
         */
-       while (iter->level > depth_want) {
-               ret = btree_iter_node(iter, iter->level)
-                       ? btree_iter_down(trans, iter, trace_ip)
-                       : btree_iter_lock_root(trans, iter, depth_want, trace_ip);
+       while (path->level > depth_want) {
+               ret = btree_path_node(path, path->level)
+                       ? btree_path_down(trans, path, flags, trace_ip)
+                       : btree_path_lock_root(trans, path, depth_want, trace_ip);
                if (unlikely(ret)) {
                        if (ret == 1) {
                                /*
-                                * Got to the end of the btree (in
-                                * BTREE_ITER_NODES mode)
+                                * No nodes at this level - got to the end of
+                                * the btree:
                                 */
                                ret = 0;
                                goto out;
                        }
 
-                       __bch2_btree_iter_unlock(iter);
-                       iter->level = depth_want;
+                       __bch2_btree_path_unlock(path);
+                       path->level = depth_want;
 
-                       if (ret == -EIO) {
-                               iter->flags |= BTREE_ITER_ERROR;
-                               iter->l[iter->level].b =
+                       if (ret == -EIO)
+                               path->l[path->level].b =
                                        BTREE_ITER_NO_NODE_ERROR;
-                       } else {
-                               iter->l[iter->level].b =
+                       else
+                               path->l[path->level].b =
                                        BTREE_ITER_NO_NODE_DOWN;
-                       }
                        goto out;
                }
        }
 
-       iter->uptodate = BTREE_ITER_NEED_PEEK;
+       path->uptodate = BTREE_ITER_UPTODATE;
 out:
        BUG_ON((ret == -EINTR) != !!trans->restarted);
-       trace_iter_traverse(trans->ip, trace_ip,
-                           btree_iter_type(iter) == BTREE_ITER_CACHED,
-                           iter->btree_id, &iter->real_pos, ret);
-       bch2_btree_iter_verify(iter);
+       bch2_btree_path_verify(trans, path);
        return ret;
 }
 
-static int __must_check __bch2_btree_iter_traverse(struct btree_iter *iter)
-{
-       struct btree_trans *trans = iter->trans;
-       int ret;
+static int __btree_path_traverse_all(struct btree_trans *, int, unsigned long);
 
-       ret =   bch2_trans_cond_resched(trans) ?:
-               btree_iter_traverse_one(iter, _RET_IP_);
-       if (unlikely(ret) && hweight64(trans->iters_linked) == 1) {
-               ret = __btree_iter_traverse_all(trans, ret, _RET_IP_);
-               BUG_ON(ret == -EINTR);
-       }
+int __must_check bch2_btree_path_traverse(struct btree_trans *trans,
+                                         struct btree_path *path, unsigned flags)
+{
+       if (path->uptodate < BTREE_ITER_NEED_RELOCK)
+               return 0;
 
-       return ret;
+       return  bch2_trans_cond_resched(trans) ?:
+               btree_path_traverse_one(trans, path, flags, _RET_IP_);
 }
 
-/*
- * Note:
- * bch2_btree_iter_traverse() is for external users, btree_iter_traverse() is
- * for internal btree iterator users
- *
- * bch2_btree_iter_traverse sets iter->real_pos to iter->pos,
- * btree_iter_traverse() does not:
- */
-static inline int __must_check
-btree_iter_traverse(struct btree_iter *iter)
+static void btree_path_copy(struct btree_trans *trans, struct btree_path *dst,
+                           struct btree_path *src)
 {
-       return iter->uptodate >= BTREE_ITER_NEED_RELOCK
-               ? __bch2_btree_iter_traverse(iter)
-               : 0;
+       unsigned i;
+
+       memcpy(&dst->pos, &src->pos,
+              sizeof(struct btree_path) - offsetof(struct btree_path, pos));
+
+       for (i = 0; i < BTREE_MAX_DEPTH; i++)
+               if (btree_node_locked(dst, i))
+                       six_lock_increment(&dst->l[i].b->c.lock,
+                                          __btree_lock_want(dst, i));
+
+       btree_path_check_sort(trans, dst, 0);
 }
 
-int __must_check
-bch2_btree_iter_traverse(struct btree_iter *iter)
+static struct btree_path *btree_path_clone(struct btree_trans *trans, struct btree_path *src,
+                                          bool intent)
 {
-       int ret;
+       struct btree_path *new = btree_path_alloc(trans, src);
 
-       btree_iter_set_search_pos(iter, btree_iter_search_key(iter));
+       btree_path_copy(trans, new, src);
+       __btree_path_get(new, intent);
+       return new;
+}
+
+inline struct btree_path * __must_check
+bch2_btree_path_make_mut(struct btree_trans *trans,
+                        struct btree_path *path, bool intent,
+                        unsigned long ip)
+{
+       if (path->ref > 1 || path->preserve) {
+               __btree_path_put(path, intent);
+               path = btree_path_clone(trans, path, intent);
+               path->preserve = false;
+#ifdef CONFIG_BCACHEFS_DEBUG
+               path->ip_allocated = ip;
+#endif
+               btree_trans_verify_sorted(trans);
+       }
+
+       return path;
+}
+
+struct btree_path * __must_check
+bch2_btree_path_set_pos(struct btree_trans *trans,
+                  struct btree_path *path, struct bpos new_pos,
+                  bool intent, unsigned long ip)
+{
+       int cmp = bpos_cmp(new_pos, path->pos);
+       unsigned l = path->level;
+
+       EBUG_ON(trans->restarted);
+       EBUG_ON(!path->ref);
+
+       if (!cmp)
+               return path;
+
+       path = bch2_btree_path_make_mut(trans, path, intent, ip);
+
+       path->pos               = new_pos;
+       path->should_be_locked  = false;
+
+       btree_path_check_sort(trans, path, cmp);
+
+       if (unlikely(path->cached)) {
+               btree_node_unlock(path, 0);
+               path->l[0].b = BTREE_ITER_NO_NODE_CACHED;
+               btree_path_set_dirty(path, BTREE_ITER_NEED_TRAVERSE);
+               goto out;
+       }
 
-       ret = btree_iter_traverse(iter);
+       l = btree_path_up_until_good_node(trans, path, cmp);
+
+       if (btree_path_node(path, l)) {
+               /*
+                * We might have to skip over many keys, or just a few: try
+                * advancing the node iterator, and if we have to skip over too
+                * many keys just reinit it (or if we're rewinding, since that
+                * is expensive).
+                */
+               if (cmp < 0 ||
+                   !btree_path_advance_to_pos(path, &path->l[l], 8))
+                       __btree_path_level_init(path, l);
+       }
+
+       if (l != path->level) {
+               btree_path_set_dirty(path, BTREE_ITER_NEED_TRAVERSE);
+               __bch2_btree_path_unlock(path);
+       }
+out:
+       bch2_btree_path_verify(trans, path);
+       return path;
+}
+
+/* Btree path: main interface: */
+
+static struct btree_path *have_path_at_pos(struct btree_trans *trans, struct btree_path *path)
+{
+       struct btree_path *next;
+
+       next = prev_btree_path(trans, path);
+       if (next && !btree_path_cmp(next, path))
+               return next;
+
+       next = next_btree_path(trans, path);
+       if (next && !btree_path_cmp(next, path))
+               return next;
+
+       return NULL;
+}
+
+static struct btree_path *have_node_at_pos(struct btree_trans *trans, struct btree_path *path)
+{
+       struct btree_path *next;
+
+       next = prev_btree_path(trans, path);
+       if (next && next->level == path->level && path_l(next)->b == path_l(path)->b)
+               return next;
+
+       next = next_btree_path(trans, path);
+       if (next && next->level == path->level && path_l(next)->b == path_l(path)->b)
+               return next;
+
+       return NULL;
+}
+
+static inline void __bch2_path_free(struct btree_trans *trans, struct btree_path *path)
+{
+       __bch2_btree_path_unlock(path);
+       btree_path_list_remove(trans, path);
+       trans->paths_allocated &= ~(1ULL << path->idx);
+}
+
+void bch2_path_put(struct btree_trans *trans, struct btree_path *path, bool intent)
+{
+       struct btree_path *dup;
+
+       EBUG_ON(trans->paths + path->idx != path);
+       EBUG_ON(!path->ref);
+
+       if (!__btree_path_put(path, intent))
+               return;
+
+       /*
+        * Perhaps instead we should check for duplicate paths in traverse_all:
+        */
+       if (path->preserve &&
+           (dup = have_path_at_pos(trans, path))) {
+               dup->preserve = true;
+               path->preserve = false;
+               goto free;
+       }
+
+       if (!path->preserve &&
+           (dup = have_node_at_pos(trans, path)))
+               goto free;
+       return;
+free:
+       if (path->should_be_locked &&
+           !btree_node_locked(dup, path->level))
+               return;
+
+       dup->should_be_locked |= path->should_be_locked;
+       __bch2_path_free(trans, path);
+}
+
+noinline __cold
+void bch2_dump_trans_paths_updates(struct btree_trans *trans)
+{
+       struct btree_path *path;
+       struct btree_insert_entry *i;
+       unsigned idx;
+       char buf1[300], buf2[300];
+
+       btree_trans_verify_sorted(trans);
+
+       trans_for_each_path_inorder(trans, path, idx)
+               printk(KERN_ERR "path: idx %u ref %u:%u%s%s btree %s pos %s locks %u %pS\n",
+                      path->idx, path->ref, path->intent_ref,
+                      path->should_be_locked ? " S" : "",
+                      path->preserve ? " P" : "",
+                      bch2_btree_ids[path->btree_id],
+                      (bch2_bpos_to_text(&PBUF(buf1), path->pos), buf1),
+                      path->nodes_locked,
+#ifdef CONFIG_BCACHEFS_DEBUG
+                      (void *) path->ip_allocated
+#else
+                      NULL
+#endif
+                      );
+
+       trans_for_each_update(trans, i) {
+               struct bkey u;
+               struct bkey_s_c old = bch2_btree_path_peek_slot(i->path, &u);
+
+               printk(KERN_ERR "update: btree %s %pS\n  old %s\n  new %s",
+                      bch2_btree_ids[i->btree_id],
+                      (void *) i->ip_allocated,
+                      (bch2_bkey_val_to_text(&PBUF(buf1), trans->c, old), buf1),
+                      (bch2_bkey_val_to_text(&PBUF(buf2), trans->c, bkey_i_to_s_c(i->k)), buf2));
+       }
+}
+
+static struct btree_path *btree_path_alloc(struct btree_trans *trans,
+                                          struct btree_path *pos)
+{
+       struct btree_path *path;
+       unsigned idx;
+
+       if (unlikely(trans->paths_allocated ==
+                    ~((~0ULL << 1) << (BTREE_ITER_MAX - 1)))) {
+               bch2_dump_trans_paths_updates(trans);
+               panic("trans path oveflow\n");
+       }
+
+       idx = __ffs64(~trans->paths_allocated);
+       trans->paths_allocated |= 1ULL << idx;
+
+       path = &trans->paths[idx];
+
+       path->idx               = idx;
+       path->ref               = 0;
+       path->intent_ref        = 0;
+       path->nodes_locked      = 0;
+       path->nodes_intent_locked = 0;
+
+       btree_path_list_add(trans, pos, path);
+       return path;
+}
+
+struct btree_path *bch2_path_get(struct btree_trans *trans,
+                                enum btree_id btree_id, struct bpos pos,
+                                unsigned locks_want, unsigned level,
+                                unsigned flags, unsigned long ip)
+{
+       struct btree_path *path, *path_pos = NULL;
+       bool cached = flags & BTREE_ITER_CACHED;
+       bool intent = flags & BTREE_ITER_INTENT;
+       int i;
+
+       BUG_ON(trans->restarted);
+
+       trans_for_each_path_inorder(trans, path, i) {
+               if (__btree_path_cmp(path,
+                                    btree_id,
+                                    cached,
+                                    pos,
+                                    level) > 0)
+                       break;
+
+               path_pos = path;
+       }
+
+       if (path_pos &&
+           path_pos->cached    == cached &&
+           path_pos->btree_id  == btree_id &&
+           path_pos->level     == level) {
+               __btree_path_get(path_pos, intent);
+               path = bch2_btree_path_set_pos(trans, path_pos, pos, intent, ip);
+       } else {
+               path = btree_path_alloc(trans, path_pos);
+               path_pos = NULL;
+
+               __btree_path_get(path, intent);
+               path->pos                       = pos;
+               path->btree_id                  = btree_id;
+               path->cached                    = cached;
+               path->uptodate                  = BTREE_ITER_NEED_TRAVERSE;
+               path->should_be_locked          = false;
+               path->level                     = level;
+               path->locks_want                = locks_want;
+               path->nodes_locked              = 0;
+               path->nodes_intent_locked       = 0;
+               for (i = 0; i < ARRAY_SIZE(path->l); i++)
+                       path->l[i].b            = BTREE_ITER_NO_NODE_INIT;
+#ifdef CONFIG_BCACHEFS_DEBUG
+               path->ip_allocated              = ip;
+#endif
+               btree_trans_verify_sorted(trans);
+       }
+
+       if (!(flags & BTREE_ITER_NOPRESERVE))
+               path->preserve = true;
+
+       if (path->intent_ref)
+               locks_want = max(locks_want, level + 1);
+
+       /*
+        * If the path has locks_want greater than requested, we don't downgrade
+        * it here - on transaction restart because btree node split needs to
+        * upgrade locks, we might be putting/getting the iterator again.
+        * Downgrading iterators only happens via bch2_trans_downgrade(), after
+        * a successful transaction commit.
+        */
+
+       locks_want = min(locks_want, BTREE_MAX_DEPTH);
+       if (locks_want > path->locks_want) {
+               path->locks_want = locks_want;
+               btree_path_get_locks(trans, path, true);
+       }
+
+       return path;
+}
+
+inline struct bkey_s_c bch2_btree_path_peek_slot(struct btree_path *path, struct bkey *u)
+{
+
+       struct bkey_s_c k;
+
+       if (!path->cached) {
+               struct btree_path_level *l = path_l(path);
+               struct bkey_packed *_k;
+
+               EBUG_ON(path->uptodate != BTREE_ITER_UPTODATE);
+
+               _k = bch2_btree_node_iter_peek_all(&l->iter, l->b);
+               k = _k ? bkey_disassemble(l->b, _k, u) : bkey_s_c_null;
+
+               EBUG_ON(k.k && bkey_deleted(k.k) && bpos_cmp(k.k->p, path->pos) == 0);
+
+               if (!k.k || bpos_cmp(path->pos, k.k->p))
+                       goto hole;
+       } else {
+               struct bkey_cached *ck = (void *) path->l[0].b;
+
+               EBUG_ON(ck &&
+                       (path->btree_id != ck->key.btree_id ||
+                        bkey_cmp(path->pos, ck->key.pos)));
+
+               /* BTREE_ITER_CACHED_NOFILL|BTREE_ITER_CACHED_NOCREATE? */
+               if (unlikely(!ck || !ck->valid))
+                       return bkey_s_c_null;
+
+               EBUG_ON(path->uptodate != BTREE_ITER_UPTODATE);
+
+               k = bkey_i_to_s_c(ck->k);
+       }
+
+       return k;
+hole:
+       bkey_init(u);
+       u->p = path->pos;
+       return (struct bkey_s_c) { u, NULL };
+}
+
+/* Btree iterators: */
+
+int __must_check
+__bch2_btree_iter_traverse(struct btree_iter *iter)
+{
+       return bch2_btree_path_traverse(iter->trans, iter->path, iter->flags);
+}
+
+int __must_check
+bch2_btree_iter_traverse(struct btree_iter *iter)
+{
+       int ret;
+
+       iter->path = bch2_btree_path_set_pos(iter->trans, iter->path,
+                                       btree_iter_search_key(iter),
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
+
+       ret = bch2_btree_path_traverse(iter->trans, iter->path, iter->flags);
        if (ret)
                return ret;
 
-       iter->should_be_locked = true;
+       iter->path->should_be_locked = true;
        return 0;
 }
 
@@ -1514,149 +2019,134 @@ bch2_btree_iter_traverse(struct btree_iter *iter)
 
 struct btree *bch2_btree_iter_peek_node(struct btree_iter *iter)
 {
-       struct btree *b;
+       struct btree_trans *trans = iter->trans;
+       struct btree *b = NULL;
        int ret;
 
-       EBUG_ON(btree_iter_type(iter) != BTREE_ITER_NODES);
+       EBUG_ON(iter->path->cached);
        bch2_btree_iter_verify(iter);
 
-       ret = btree_iter_traverse(iter);
+       ret = bch2_btree_path_traverse(trans, iter->path, iter->flags);
        if (ret)
-               return NULL;
+               goto err;
 
-       b = btree_iter_node(iter, iter->level);
+       b = btree_path_node(iter->path, iter->path->level);
        if (!b)
-               return NULL;
+               goto out;
 
        BUG_ON(bpos_cmp(b->key.k.p, iter->pos) < 0);
 
-       iter->pos = iter->real_pos = b->key.k.p;
+       bkey_init(&iter->k);
+       iter->k.p = iter->pos = b->key.k.p;
 
+       iter->path = bch2_btree_path_set_pos(trans, iter->path, b->key.k.p,
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
+       iter->path->should_be_locked = true;
+       BUG_ON(iter->path->uptodate);
+out:
+       bch2_btree_iter_verify_entry_exit(iter);
        bch2_btree_iter_verify(iter);
-       iter->should_be_locked = true;
 
        return b;
+err:
+       b = ERR_PTR(ret);
+       goto out;
 }
 
 struct btree *bch2_btree_iter_next_node(struct btree_iter *iter)
 {
-       struct btree *b;
+       struct btree_trans *trans = iter->trans;
+       struct btree_path *path = iter->path;
+       struct btree *b = NULL;
+       unsigned l;
        int ret;
 
-       EBUG_ON(btree_iter_type(iter) != BTREE_ITER_NODES);
+       BUG_ON(trans->restarted);
+       EBUG_ON(iter->path->cached);
        bch2_btree_iter_verify(iter);
 
-       /* already got to end? */
-       if (!btree_iter_node(iter, iter->level))
-               return NULL;
-
-       bch2_trans_cond_resched(iter->trans);
-
-       btree_node_unlock(iter, iter->level);
-       iter->l[iter->level].b = BTREE_ITER_NO_NODE_UP;
-       iter->level++;
-
-       btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
-       ret = btree_iter_traverse(iter);
-       if (ret)
+       /* already at end? */
+       if (!btree_path_node(path, path->level))
                return NULL;
 
        /* got to end? */
-       b = btree_iter_node(iter, iter->level);
-       if (!b)
+       if (!btree_path_node(path, path->level + 1)) {
+               btree_node_unlock(path, path->level);
+               path->l[path->level].b = BTREE_ITER_NO_NODE_UP;
+               path->level++;
                return NULL;
+       }
+
+       if (!bch2_btree_node_relock(trans, path, path->level + 1)) {
+               __bch2_btree_path_unlock(path);
+               path->l[path->level].b = BTREE_ITER_NO_NODE_GET_LOCKS;
+               path->l[path->level + 1].b = BTREE_ITER_NO_NODE_GET_LOCKS;
+               trace_trans_restart_relock_next_node(trans->fn, _THIS_IP_,
+                                          path->btree_id, &path->pos);
+               btree_trans_restart(trans);
+               ret = -EINTR;
+               goto err;
+       }
+
+       b = btree_path_node(path, path->level + 1);
 
-       if (bpos_cmp(iter->pos, b->key.k.p) < 0) {
+       if (!bpos_cmp(iter->pos, b->key.k.p)) {
+               btree_node_unlock(path, path->level);
+               path->l[path->level].b = BTREE_ITER_NO_NODE_UP;
+               path->level++;
+       } else {
                /*
                 * Haven't gotten to the end of the parent node: go back down to
                 * the next child node
                 */
-               btree_iter_set_search_pos(iter, bpos_successor(iter->pos));
+               path = iter->path =
+                       bch2_btree_path_set_pos(trans, path, bpos_successor(iter->pos),
+                                          iter->flags & BTREE_ITER_INTENT,
+                                          btree_iter_ip_allocated(iter));
+
+               path->level = iter->min_depth;
 
-               /* Unlock to avoid screwing up our lock invariants: */
-               btree_node_unlock(iter, iter->level);
+               for (l = path->level + 1; l < BTREE_MAX_DEPTH; l++)
+                       if (btree_lock_want(path, l) == BTREE_NODE_UNLOCKED)
+                               btree_node_unlock(path, l);
 
-               iter->level = iter->min_depth;
-               btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
+               btree_path_set_dirty(path, BTREE_ITER_NEED_TRAVERSE);
                bch2_btree_iter_verify(iter);
 
-               ret = btree_iter_traverse(iter);
+               ret = bch2_btree_path_traverse(trans, path, iter->flags);
                if (ret)
-                       return NULL;
+                       goto err;
 
-               b = iter->l[iter->level].b;
+               b = path->l[path->level].b;
        }
 
-       iter->pos = iter->real_pos = b->key.k.p;
+       bkey_init(&iter->k);
+       iter->k.p = iter->pos = b->key.k.p;
 
+       iter->path = bch2_btree_path_set_pos(trans, iter->path, b->key.k.p,
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
+       iter->path->should_be_locked = true;
+       BUG_ON(iter->path->uptodate);
+out:
+       bch2_btree_iter_verify_entry_exit(iter);
        bch2_btree_iter_verify(iter);
-       iter->should_be_locked = true;
 
        return b;
+err:
+       b = ERR_PTR(ret);
+       goto out;
 }
 
 /* Iterate across keys (in leaf nodes only) */
 
-static void btree_iter_set_search_pos(struct btree_iter *iter, struct bpos new_pos)
-{
-#ifdef CONFIG_BCACHEFS_DEBUG
-       struct bpos old_pos = iter->real_pos;
-#endif
-       int cmp = bpos_cmp(new_pos, iter->real_pos);
-       unsigned l = iter->level;
-
-       EBUG_ON(iter->trans->restarted);
-
-       if (!cmp)
-               goto out;
-
-       iter->real_pos = new_pos;
-       iter->should_be_locked = false;
-
-       btree_iter_check_sort(iter->trans, iter);
-
-       if (unlikely(btree_iter_type(iter) == BTREE_ITER_CACHED)) {
-               btree_node_unlock(iter, 0);
-               iter->l[0].b = BTREE_ITER_NO_NODE_CACHED;
-               btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
-               return;
-       }
-
-       l = btree_iter_up_until_good_node(iter, cmp);
-
-       if (btree_iter_node(iter, l)) {
-               /*
-                * We might have to skip over many keys, or just a few: try
-                * advancing the node iterator, and if we have to skip over too
-                * many keys just reinit it (or if we're rewinding, since that
-                * is expensive).
-                */
-               if (cmp < 0 ||
-                   !btree_iter_advance_to_pos(iter, &iter->l[l], 8))
-                       __btree_iter_init(iter, l);
-
-               /* Don't leave it locked if we're not supposed to: */
-               if (btree_lock_want(iter, l) == BTREE_NODE_UNLOCKED)
-                       btree_node_unlock(iter, l);
-       }
-out:
-       if (l != iter->level)
-               btree_iter_set_dirty(iter, BTREE_ITER_NEED_TRAVERSE);
-       else
-               btree_iter_set_dirty(iter, BTREE_ITER_NEED_PEEK);
-
-       bch2_btree_iter_verify(iter);
-#ifdef CONFIG_BCACHEFS_DEBUG
-       trace_iter_set_search_pos(iter->trans->ip, _RET_IP_,
-                                 iter->btree_id,
-                                 &old_pos, &new_pos, l);
-#endif
-}
-
 inline bool bch2_btree_iter_advance(struct btree_iter *iter)
 {
        struct bpos pos = iter->k.p;
-       bool ret = bpos_cmp(pos, SPOS_MAX) != 0;
+       bool ret = (iter->flags & BTREE_ITER_ALL_SNAPSHOTS
+                   ? bpos_cmp(pos, SPOS_MAX)
+                   : bkey_cmp(pos, SPOS_MAX)) != 0;
 
        if (ret && !(iter->flags & BTREE_ITER_IS_EXTENTS))
                pos = bkey_successor(iter, pos);
@@ -1677,54 +2167,177 @@ inline bool bch2_btree_iter_rewind(struct btree_iter *iter)
        return ret;
 }
 
-static inline bool btree_iter_set_pos_to_next_leaf(struct btree_iter *iter)
+static inline struct bkey_i *btree_trans_peek_updates(struct btree_trans *trans,
+                                                     enum btree_id btree_id,
+                                                     struct bpos pos)
 {
-       struct bpos next_pos = iter->l[0].b->key.k.p;
-       bool ret = bpos_cmp(next_pos, SPOS_MAX) != 0;
+       struct btree_insert_entry *i;
 
-       /*
-        * Typically, we don't want to modify iter->pos here, since that
-        * indicates where we searched from - unless we got to the end of the
-        * btree, in that case we want iter->pos to reflect that:
-        */
-       if (ret)
-               btree_iter_set_search_pos(iter, bpos_successor(next_pos));
-       else
-               bch2_btree_iter_set_pos(iter, SPOS_MAX);
+       trans_for_each_update(trans, i)
+               if ((cmp_int(btree_id,  i->btree_id) ?:
+                    bpos_cmp(pos,      i->k->k.p)) <= 0) {
+                       if (btree_id == i->btree_id)
+                               return i->k;
+                       break;
+               }
 
-       return ret;
+       return NULL;
 }
 
-static inline bool btree_iter_set_pos_to_prev_leaf(struct btree_iter *iter)
+static noinline
+struct bkey_i *__btree_trans_peek_journal(struct btree_trans *trans,
+                                         struct btree_path *path)
 {
-       struct bpos next_pos = iter->l[0].b->data->min_key;
-       bool ret = bpos_cmp(next_pos, POS_MIN) != 0;
+       struct journal_keys *keys = &trans->c->journal_keys;
+       size_t idx = bch2_journal_key_search(keys, path->btree_id,
+                                            path->level, path->pos);
 
-       if (ret)
-               btree_iter_set_search_pos(iter, bpos_predecessor(next_pos));
-       else
-               bch2_btree_iter_set_pos(iter, POS_MIN);
+       while (idx < keys->nr && keys->d[idx].overwritten)
+               idx++;
 
-       return ret;
+       return (idx < keys->nr &&
+               keys->d[idx].btree_id   == path->btree_id &&
+               keys->d[idx].level      == path->level)
+               ? keys->d[idx].k
+               : NULL;
 }
 
-static inline struct bkey_i *btree_trans_peek_updates(struct btree_iter *iter,
-                                                     struct bpos pos)
+static noinline
+struct bkey_s_c btree_trans_peek_journal(struct btree_trans *trans,
+                                        struct btree_iter *iter,
+                                        struct bkey_s_c k)
 {
-       struct btree_insert_entry *i;
+       struct bkey_i *next_journal =
+               __btree_trans_peek_journal(trans, iter->path);
 
-       if (!(iter->flags & BTREE_ITER_WITH_UPDATES))
-               return NULL;
+       if (next_journal &&
+           bpos_cmp(next_journal->k.p,
+                    k.k ? k.k->p : iter->path->l[0].b->key.k.p) <= 0) {
+               iter->k = next_journal->k;
+               k = bkey_i_to_s_c(next_journal);
+       }
 
-       trans_for_each_update(iter->trans, i)
-               if ((cmp_int(iter->btree_id,    i->iter->btree_id) ?:
-                    bkey_cmp(pos,              i->k->k.p)) <= 0) {
-                       if (iter->btree_id ==   i->iter->btree_id)
-                               return i->k;
+       return k;
+}
+
+/*
+ * Checks btree key cache for key at iter->pos and returns it if present, or
+ * bkey_s_c_null:
+ */
+static noinline
+struct bkey_s_c btree_trans_peek_key_cache(struct btree_iter *iter, struct bpos pos)
+{
+       struct btree_trans *trans = iter->trans;
+       struct bch_fs *c = trans->c;
+       struct bkey u;
+       int ret;
+
+       if (!bch2_btree_key_cache_find(c, iter->btree_id, pos))
+               return bkey_s_c_null;
+
+       if (!iter->key_cache_path)
+               iter->key_cache_path = bch2_path_get(trans, iter->btree_id, pos,
+                                                    iter->flags & BTREE_ITER_INTENT, 0,
+                                                    iter->flags|BTREE_ITER_CACHED,
+                                                    _THIS_IP_);
+
+       iter->key_cache_path = bch2_btree_path_set_pos(trans, iter->key_cache_path, pos,
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
+
+       ret = bch2_btree_path_traverse(trans, iter->key_cache_path, iter->flags|BTREE_ITER_CACHED);
+       if (unlikely(ret))
+               return bkey_s_c_err(ret);
+
+       iter->key_cache_path->should_be_locked = true;
+
+       return bch2_btree_path_peek_slot(iter->key_cache_path, &u);
+}
+
+static struct bkey_s_c __bch2_btree_iter_peek(struct btree_iter *iter, struct bpos search_key)
+{
+       struct btree_trans *trans = iter->trans;
+       struct bkey_i *next_update;
+       struct bkey_s_c k, k2;
+       int ret;
+
+       EBUG_ON(iter->path->cached || iter->path->level);
+       bch2_btree_iter_verify(iter);
+
+       while (1) {
+               iter->path = bch2_btree_path_set_pos(trans, iter->path, search_key,
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
+
+               ret = bch2_btree_path_traverse(trans, iter->path, iter->flags);
+               if (unlikely(ret)) {
+                       /* ensure that iter->k is consistent with iter->pos: */
+                       bch2_btree_iter_set_pos(iter, iter->pos);
+                       k = bkey_s_c_err(ret);
+                       goto out;
+               }
+
+               iter->path->should_be_locked = true;
+
+               k = btree_path_level_peek_all(trans->c, &iter->path->l[0], &iter->k);
+
+               if (unlikely(iter->flags & BTREE_ITER_WITH_KEY_CACHE) &&
+                   k.k &&
+                   (k2 = btree_trans_peek_key_cache(iter, k.k->p)).k) {
+                       ret = bkey_err(k2);
+                       if (ret) {
+                               k = k2;
+                               bch2_btree_iter_set_pos(iter, iter->pos);
+                               goto out;
+                       }
+
+                       k = k2;
+                       iter->k = *k.k;
+               }
+
+               if (unlikely(iter->flags & BTREE_ITER_WITH_JOURNAL))
+                       k = btree_trans_peek_journal(trans, iter, k);
+
+               next_update = iter->flags & BTREE_ITER_WITH_UPDATES
+                       ? btree_trans_peek_updates(trans, iter->btree_id, search_key)
+                       : NULL;
+               if (next_update &&
+                   bpos_cmp(next_update->k.p,
+                            k.k ? k.k->p : iter->path->l[0].b->key.k.p) <= 0) {
+                       iter->k = next_update->k;
+                       k = bkey_i_to_s_c(next_update);
+               }
+
+               if (k.k && bkey_deleted(k.k)) {
+                       /*
+                        * If we've got a whiteout, and it's after the search
+                        * key, advance the search key to the whiteout instead
+                        * of just after the whiteout - it might be a btree
+                        * whiteout, with a real key at the same position, since
+                        * in the btree deleted keys sort before non deleted.
+                        */
+                       search_key = bpos_cmp(search_key, k.k->p)
+                               ? k.k->p
+                               : bpos_successor(k.k->p);
+                       continue;
+               }
+
+               if (likely(k.k)) {
                        break;
+               } else if (likely(bpos_cmp(iter->path->l[0].b->key.k.p, SPOS_MAX))) {
+                       /* Advance to next leaf node: */
+                       search_key = bpos_successor(iter->path->l[0].b->key.k.p);
+               } else {
+                       /* End of btree: */
+                       bch2_btree_iter_set_pos(iter, SPOS_MAX);
+                       k = bkey_s_c_null;
+                       goto out;
                }
+       }
+out:
+       bch2_btree_iter_verify(iter);
 
-       return NULL;
+       return k;
 }
 
 /**
@@ -1733,42 +2346,79 @@ static inline struct bkey_i *btree_trans_peek_updates(struct btree_iter *iter,
  */
 struct bkey_s_c bch2_btree_iter_peek(struct btree_iter *iter)
 {
+       struct btree_trans *trans = iter->trans;
        struct bpos search_key = btree_iter_search_key(iter);
-       struct bkey_i *next_update;
        struct bkey_s_c k;
        int ret;
 
-       EBUG_ON(btree_iter_type(iter) != BTREE_ITER_KEYS);
-       bch2_btree_iter_verify(iter);
+       if (iter->update_path) {
+               bch2_path_put(trans, iter->update_path,
+                             iter->flags & BTREE_ITER_INTENT);
+               iter->update_path = NULL;
+       }
+
        bch2_btree_iter_verify_entry_exit(iter);
-start:
-       next_update = btree_trans_peek_updates(iter, search_key);
-       btree_iter_set_search_pos(iter, search_key);
 
        while (1) {
-               ret = btree_iter_traverse(iter);
-               if (unlikely(ret))
-                       return bkey_s_c_err(ret);
+               k = __bch2_btree_iter_peek(iter, search_key);
+               if (!k.k || bkey_err(k))
+                       goto out;
 
-               k = btree_iter_level_peek(iter, &iter->l[0]);
+               if (iter->update_path &&
+                   bkey_cmp(iter->update_path->pos, k.k->p)) {
+                       bch2_path_put(trans, iter->update_path,
+                                     iter->flags & BTREE_ITER_INTENT);
+                       iter->update_path = NULL;
+               }
 
-               if (next_update &&
-                   bpos_cmp(next_update->k.p, iter->real_pos) <= 0) {
-                       iter->k = next_update->k;
-                       k = bkey_i_to_s_c(next_update);
+               if ((iter->flags & BTREE_ITER_FILTER_SNAPSHOTS) &&
+                   (iter->flags & BTREE_ITER_INTENT) &&
+                   !(iter->flags & BTREE_ITER_IS_EXTENTS) &&
+                   !iter->update_path) {
+                       struct bpos pos = k.k->p;
+
+                       if (pos.snapshot < iter->snapshot) {
+                               search_key = bpos_successor(k.k->p);
+                               continue;
+                       }
+
+                       pos.snapshot = iter->snapshot;
+
+                       /*
+                        * advance, same as on exit for iter->path, but only up
+                        * to snapshot
+                        */
+                       __btree_path_get(iter->path, iter->flags & BTREE_ITER_INTENT);
+                       iter->update_path = iter->path;
+
+                       iter->update_path = bch2_btree_path_set_pos(trans,
+                                               iter->update_path, pos,
+                                               iter->flags & BTREE_ITER_INTENT,
+                                               btree_iter_ip_allocated(iter));
+
+                       BUG_ON(!(iter->update_path->nodes_locked & 1));
+                       iter->update_path->should_be_locked = true;
+               }
+
+               /*
+                * We can never have a key in a leaf node at POS_MAX, so
+                * we don't have to check these successor() calls:
+                */
+               if ((iter->flags & BTREE_ITER_FILTER_SNAPSHOTS) &&
+                   !bch2_snapshot_is_ancestor(trans->c,
+                                              iter->snapshot,
+                                              k.k->p.snapshot)) {
+                       search_key = bpos_successor(k.k->p);
+                       continue;
                }
 
-               if (likely(k.k)) {
-                       if (bkey_deleted(k.k)) {
-                               search_key = bkey_successor(iter, k.k->p);
-                               goto start;
-                       }
-
-                       break;
+               if (bkey_whiteout(k.k) &&
+                   !(iter->flags & BTREE_ITER_ALL_SNAPSHOTS)) {
+                       search_key = bkey_successor(iter, k.k->p);
+                       continue;
                }
 
-               if (!btree_iter_set_pos_to_next_leaf(iter))
-                       return bkey_s_c_null;
+               break;
        }
 
        /*
@@ -1780,9 +2430,28 @@ start:
        else if (bkey_cmp(bkey_start_pos(k.k), iter->pos) > 0)
                iter->pos = bkey_start_pos(k.k);
 
+       iter->path = bch2_btree_path_set_pos(trans, iter->path, k.k->p,
+                               iter->flags & BTREE_ITER_INTENT,
+                               btree_iter_ip_allocated(iter));
+       BUG_ON(!iter->path->nodes_locked);
+out:
+       if (iter->update_path) {
+               BUG_ON(!(iter->update_path->nodes_locked & 1));
+               iter->update_path->should_be_locked = true;
+       }
+       iter->path->should_be_locked = true;
+
+       if (!(iter->flags & BTREE_ITER_ALL_SNAPSHOTS))
+               iter->pos.snapshot = iter->snapshot;
+
+       ret = bch2_btree_iter_verify_ret(iter, k);
+       if (unlikely(ret)) {
+               bch2_btree_iter_set_pos(iter, iter->pos);
+               k = bkey_s_c_err(ret);
+       }
+
        bch2_btree_iter_verify_entry_exit(iter);
-       bch2_btree_iter_verify(iter);
-       iter->should_be_locked = true;
+
        return k;
 }
 
@@ -1804,37 +2473,103 @@ struct bkey_s_c bch2_btree_iter_next(struct btree_iter *iter)
  */
 struct bkey_s_c bch2_btree_iter_peek_prev(struct btree_iter *iter)
 {
-       struct btree_iter_level *l = &iter->l[0];
+       struct btree_trans *trans = iter->trans;
+       struct bpos search_key = iter->pos;
+       struct btree_path *saved_path = NULL;
        struct bkey_s_c k;
+       struct bkey saved_k;
+       const struct bch_val *saved_v;
        int ret;
 
-       EBUG_ON(btree_iter_type(iter) != BTREE_ITER_KEYS);
+       EBUG_ON(iter->path->cached || iter->path->level);
        EBUG_ON(iter->flags & BTREE_ITER_WITH_UPDATES);
+
+       if (iter->flags & BTREE_ITER_WITH_JOURNAL)
+               return bkey_s_c_err(-EIO);
+
        bch2_btree_iter_verify(iter);
        bch2_btree_iter_verify_entry_exit(iter);
 
-       btree_iter_set_search_pos(iter, iter->pos);
+       if (iter->flags & BTREE_ITER_FILTER_SNAPSHOTS)
+               search_key.snapshot = U32_MAX;
 
        while (1) {
-               ret = btree_iter_traverse(iter);
+               iter->path = bch2_btree_path_set_pos(trans, iter->path, search_key,
+                                               iter->flags & BTREE_ITER_INTENT,
+                                               btree_iter_ip_allocated(iter));
+
+               ret = bch2_btree_path_traverse(trans, iter->path, iter->flags);
                if (unlikely(ret)) {
+                       /* ensure that iter->k is consistent with iter->pos: */
+                       bch2_btree_iter_set_pos(iter, iter->pos);
                        k = bkey_s_c_err(ret);
-                       goto no_key;
+                       goto out;
                }
 
-               k = btree_iter_level_peek(iter, l);
+               k = btree_path_level_peek(trans->c, iter->path,
+                                         &iter->path->l[0], &iter->k);
                if (!k.k ||
                    ((iter->flags & BTREE_ITER_IS_EXTENTS)
-                    ? bkey_cmp(bkey_start_pos(k.k), iter->pos) >= 0
-                    : bkey_cmp(k.k->p, iter->pos) > 0))
-                       k = btree_iter_level_prev(iter, l);
+                    ? bpos_cmp(bkey_start_pos(k.k), search_key) >= 0
+                    : bpos_cmp(k.k->p, search_key) > 0))
+                       k = btree_path_level_prev(trans->c, iter->path,
+                                                 &iter->path->l[0], &iter->k);
 
-               if (likely(k.k))
-                       break;
+               btree_path_check_sort(trans, iter->path, 0);
+
+               if (likely(k.k)) {
+                       if (iter->flags & BTREE_ITER_FILTER_SNAPSHOTS) {
+                               if (k.k->p.snapshot == iter->snapshot)
+                                       goto got_key;
+
+                               /*
+                                * If we have a saved candidate, and we're no
+                                * longer at the same _key_ (not pos), return
+                                * that candidate
+                                */
+                               if (saved_path && bkey_cmp(k.k->p, saved_k.p)) {
+                                       bch2_path_put(trans, iter->path,
+                                                     iter->flags & BTREE_ITER_INTENT);
+                                       iter->path = saved_path;
+                                       saved_path = NULL;
+                                       iter->k = saved_k;
+                                       k.v     = saved_v;
+                                       goto got_key;
+                               }
+
+                               if (bch2_snapshot_is_ancestor(iter->trans->c,
+                                                             iter->snapshot,
+                                                             k.k->p.snapshot)) {
+                                       if (saved_path)
+                                               bch2_path_put(trans, saved_path,
+                                                     iter->flags & BTREE_ITER_INTENT);
+                                       saved_path = btree_path_clone(trans, iter->path,
+                                                               iter->flags & BTREE_ITER_INTENT);
+                                       saved_k = *k.k;
+                                       saved_v = k.v;
+                               }
+
+                               search_key = bpos_predecessor(k.k->p);
+                               continue;
+                       }
+got_key:
+                       if (bkey_whiteout(k.k) &&
+                           !(iter->flags & BTREE_ITER_ALL_SNAPSHOTS)) {
+                               search_key = bkey_predecessor(iter, k.k->p);
+                               if (iter->flags & BTREE_ITER_FILTER_SNAPSHOTS)
+                                       search_key.snapshot = U32_MAX;
+                               continue;
+                       }
 
-               if (!btree_iter_set_pos_to_prev_leaf(iter)) {
+                       break;
+               } else if (likely(bpos_cmp(iter->path->l[0].b->data->min_key, POS_MIN))) {
+                       /* Advance to previous leaf node: */
+                       search_key = bpos_predecessor(iter->path->l[0].b->data->min_key);
+               } else {
+                       /* Start of btree: */
+                       bch2_btree_iter_set_pos(iter, POS_MIN);
                        k = bkey_s_c_null;
-                       goto no_key;
+                       goto out;
                }
        }
 
@@ -1843,20 +2578,18 @@ struct bkey_s_c bch2_btree_iter_peek_prev(struct btree_iter *iter)
        /* Extents can straddle iter->pos: */
        if (bkey_cmp(k.k->p, iter->pos) < 0)
                iter->pos = k.k->p;
+
+       if (iter->flags & BTREE_ITER_FILTER_SNAPSHOTS)
+               iter->pos.snapshot = iter->snapshot;
 out:
+       if (saved_path)
+               bch2_path_put(trans, saved_path, iter->flags & BTREE_ITER_INTENT);
+       iter->path->should_be_locked = true;
+
        bch2_btree_iter_verify_entry_exit(iter);
        bch2_btree_iter_verify(iter);
-       iter->should_be_locked = true;
+
        return k;
-no_key:
-       /*
-        * btree_iter_level_peek() may have set iter->k to a key we didn't want, and
-        * then we errored going to the previous leaf - make sure it's
-        * consistent with iter->pos:
-        */
-       bkey_init(&iter->k);
-       iter->k.p = iter->pos;
-       goto out;
 }
 
 /**
@@ -1873,12 +2606,12 @@ struct bkey_s_c bch2_btree_iter_prev(struct btree_iter *iter)
 
 struct bkey_s_c bch2_btree_iter_peek_slot(struct btree_iter *iter)
 {
+       struct btree_trans *trans = iter->trans;
        struct bpos search_key;
        struct bkey_s_c k;
        int ret;
 
-       EBUG_ON(btree_iter_type(iter) != BTREE_ITER_KEYS &&
-               btree_iter_type(iter) != BTREE_ITER_CACHED);
+       EBUG_ON(iter->path->level);
        bch2_btree_iter_verify(iter);
        bch2_btree_iter_verify_entry_exit(iter);
 
@@ -1892,50 +2625,57 @@ struct bkey_s_c bch2_btree_iter_peek_slot(struct btree_iter *iter)
        }
 
        search_key = btree_iter_search_key(iter);
-       btree_iter_set_search_pos(iter, search_key);
+       iter->path = bch2_btree_path_set_pos(trans, iter->path, search_key,
+                                       iter->flags & BTREE_ITER_INTENT,
+                                       btree_iter_ip_allocated(iter));
 
-       ret = btree_iter_traverse(iter);
+       ret = bch2_btree_path_traverse(trans, iter->path, iter->flags);
        if (unlikely(ret))
                return bkey_s_c_err(ret);
 
-       if (btree_iter_type(iter) == BTREE_ITER_CACHED ||
-           !(iter->flags & BTREE_ITER_IS_EXTENTS)) {
+       if ((iter->flags & BTREE_ITER_CACHED) ||
+           !(iter->flags & (BTREE_ITER_IS_EXTENTS|BTREE_ITER_FILTER_SNAPSHOTS))) {
                struct bkey_i *next_update;
-               struct bkey_cached *ck;
 
-               switch (btree_iter_type(iter)) {
-               case BTREE_ITER_KEYS:
-                       k = btree_iter_level_peek_all(iter, &iter->l[0]);
-                       EBUG_ON(k.k && bkey_deleted(k.k) && bpos_cmp(k.k->p, iter->pos) == 0);
-                       break;
-               case BTREE_ITER_CACHED:
-                       ck = (void *) iter->l[0].b;
-                       EBUG_ON(iter->btree_id != ck->key.btree_id ||
-                               bkey_cmp(iter->pos, ck->key.pos));
-                       BUG_ON(!ck->valid);
-
-                       k = bkey_i_to_s_c(ck->k);
-                       break;
-               case BTREE_ITER_NODES:
-                       BUG();
+               if ((iter->flags & BTREE_ITER_WITH_UPDATES) &&
+                   (next_update = btree_trans_peek_updates(trans,
+                                               iter->btree_id, search_key)) &&
+                   !bpos_cmp(next_update->k.p, iter->pos)) {
+                       iter->k = next_update->k;
+                       k = bkey_i_to_s_c(next_update);
+                       goto out;
                }
 
-               next_update = btree_trans_peek_updates(iter, search_key);
-               if (next_update &&
-                   (!k.k || bpos_cmp(next_update->k.p, k.k->p) <= 0)) {
+               if (unlikely(iter->flags & BTREE_ITER_WITH_JOURNAL) &&
+                   (next_update = __btree_trans_peek_journal(trans, iter->path)) &&
+                   !bpos_cmp(next_update->k.p, iter->pos)) {
                        iter->k = next_update->k;
                        k = bkey_i_to_s_c(next_update);
+                       goto out;
                }
+
+               if (unlikely(iter->flags & BTREE_ITER_WITH_KEY_CACHE) &&
+                   (k = btree_trans_peek_key_cache(iter, iter->pos)).k) {
+                       if (!bkey_err(k))
+                               iter->k = *k.k;
+                       goto out;
+               }
+
+               k = bch2_btree_path_peek_slot(iter->path, &iter->k);
        } else {
-               if ((iter->flags & BTREE_ITER_INTENT)) {
-                       struct btree_iter *child =
-                               btree_iter_child_alloc(iter, _THIS_IP_);
+               struct bpos next;
 
-                       btree_iter_copy(child, iter);
-                       k = bch2_btree_iter_peek(child);
+               if (iter->flags & BTREE_ITER_INTENT) {
+                       struct btree_iter iter2;
 
-                       if (k.k && !bkey_err(k))
-                               iter->k = child->k;
+                       bch2_trans_copy_iter(&iter2, iter);
+                       k = bch2_btree_iter_peek(&iter2);
+
+                       if (k.k && !bkey_err(k)) {
+                               iter->k = iter2.k;
+                               k.k = &iter->k;
+                       }
+                       bch2_trans_iter_exit(trans, &iter2);
                } else {
                        struct bpos pos = iter->pos;
 
@@ -1945,38 +2685,34 @@ struct bkey_s_c bch2_btree_iter_peek_slot(struct btree_iter *iter)
 
                if (unlikely(bkey_err(k)))
                        return k;
-       }
 
-       if (!(iter->flags & BTREE_ITER_IS_EXTENTS)) {
-               if (!k.k ||
-                   ((iter->flags & BTREE_ITER_ALL_SNAPSHOTS)
-                    ? bpos_cmp(iter->pos, k.k->p)
-                    : bkey_cmp(iter->pos, k.k->p))) {
-                       bkey_init(&iter->k);
-                       iter->k.p = iter->pos;
-                       k = (struct bkey_s_c) { &iter->k, NULL };
-               }
-       } else {
-               struct bpos next = k.k ? bkey_start_pos(k.k) : POS_MAX;
+               next = k.k ? bkey_start_pos(k.k) : POS_MAX;
 
                if (bkey_cmp(iter->pos, next) < 0) {
                        bkey_init(&iter->k);
                        iter->k.p = iter->pos;
-                       bch2_key_resize(&iter->k,
-                                       min_t(u64, KEY_SIZE_MAX,
-                                             (next.inode == iter->pos.inode
-                                              ? next.offset
-                                              : KEY_OFFSET_MAX) -
-                                             iter->pos.offset));
+
+                       if (iter->flags & BTREE_ITER_IS_EXTENTS) {
+                               bch2_key_resize(&iter->k,
+                                               min_t(u64, KEY_SIZE_MAX,
+                                                     (next.inode == iter->pos.inode
+                                                      ? next.offset
+                                                      : KEY_OFFSET_MAX) -
+                                                     iter->pos.offset));
+                               EBUG_ON(!iter->k.size);
+                       }
 
                        k = (struct bkey_s_c) { &iter->k, NULL };
-                       EBUG_ON(!k.k->size);
                }
        }
+out:
+       iter->path->should_be_locked = true;
 
        bch2_btree_iter_verify_entry_exit(iter);
        bch2_btree_iter_verify(iter);
-       iter->should_be_locked = true;
+       ret = bch2_btree_iter_verify_ret(iter, k);
+       if (unlikely(ret))
+               return bkey_s_c_err(ret);
 
        return k;
 }
@@ -1997,35 +2733,14 @@ struct bkey_s_c bch2_btree_iter_prev_slot(struct btree_iter *iter)
        return bch2_btree_iter_peek_slot(iter);
 }
 
-static inline void bch2_btree_iter_init(struct btree_trans *trans,
-                       struct btree_iter *iter, enum btree_id btree_id)
-{
-       struct bch_fs *c = trans->c;
-       unsigned i;
-
-       iter->trans                     = trans;
-       iter->uptodate                  = BTREE_ITER_NEED_TRAVERSE;
-       iter->btree_id                  = btree_id;
-       iter->real_pos                  = POS_MIN;
-       iter->level                     = 0;
-       iter->min_depth                 = 0;
-       iter->locks_want                = 0;
-       iter->nodes_locked              = 0;
-       iter->nodes_intent_locked       = 0;
-       for (i = 0; i < ARRAY_SIZE(iter->l); i++)
-               iter->l[i].b            = BTREE_ITER_NO_NODE_INIT;
-
-       prefetch(c->btree_roots[btree_id].b);
-}
-
 /* new transactional stuff: */
 
-static inline void btree_iter_verify_sorted_ref(struct btree_trans *trans,
-                                               struct btree_iter *iter)
+static inline void btree_path_verify_sorted_ref(struct btree_trans *trans,
+                                               struct btree_path *path)
 {
-       EBUG_ON(iter->sorted_idx >= trans->nr_sorted);
-       EBUG_ON(trans->sorted[iter->sorted_idx] != iter->idx);
-       EBUG_ON(!(trans->iters_linked & (1ULL << iter->idx)));
+       EBUG_ON(path->sorted_idx >= trans->nr_sorted);
+       EBUG_ON(trans->sorted[path->sorted_idx] != path->idx);
+       EBUG_ON(!(trans->paths_allocated & (1ULL << path->idx)));
 }
 
 static inline void btree_trans_verify_sorted_refs(struct btree_trans *trans)
@@ -2034,432 +2749,201 @@ static inline void btree_trans_verify_sorted_refs(struct btree_trans *trans)
        unsigned i;
 
        for (i = 0; i < trans->nr_sorted; i++)
-               btree_iter_verify_sorted_ref(trans, trans->iters + trans->sorted[i]);
+               btree_path_verify_sorted_ref(trans, trans->paths + trans->sorted[i]);
 #endif
 }
 
-static inline void btree_trans_verify_sorted(struct btree_trans *trans)
+static void btree_trans_verify_sorted(struct btree_trans *trans)
 {
 #ifdef CONFIG_BCACHEFS_DEBUG
-       struct btree_iter *iter, *prev = NULL;
+       struct btree_path *path, *prev = NULL;
+       unsigned i;
 
-       trans_for_each_iter_inorder(trans, iter)
-               BUG_ON(prev && btree_iter_cmp(prev, iter) > 0);
+       trans_for_each_path_inorder(trans, path, i) {
+               BUG_ON(prev && btree_path_cmp(prev, path) > 0);
+               prev = path;
+       }
 #endif
 }
 
-static inline void btree_iter_swap(struct btree_trans *trans,
-                                  struct btree_iter *l, struct btree_iter *r)
+static inline void btree_path_swap(struct btree_trans *trans,
+                                  struct btree_path *l, struct btree_path *r)
 {
        swap(l->sorted_idx, r->sorted_idx);
        swap(trans->sorted[l->sorted_idx],
             trans->sorted[r->sorted_idx]);
 
-       btree_iter_verify_sorted_ref(trans, l);
-       btree_iter_verify_sorted_ref(trans, r);
+       btree_path_verify_sorted_ref(trans, l);
+       btree_path_verify_sorted_ref(trans, r);
 }
 
-static void btree_trans_sort_iters(struct btree_trans *trans)
+static void btree_path_check_sort(struct btree_trans *trans, struct btree_path *path,
+                                 int cmp)
 {
-       bool swapped = false;
-       int i, l = 0, r = trans->nr_sorted;
-
-       while (1) {
-               for (i = l; i + 1 < r; i++) {
-                       if (btree_iter_cmp(trans->iters + trans->sorted[i],
-                                          trans->iters + trans->sorted[i + 1]) > 0) {
-                               swap(trans->sorted[i], trans->sorted[i + 1]);
-                               trans->iters[trans->sorted[i]].sorted_idx = i;
-                               trans->iters[trans->sorted[i + 1]].sorted_idx = i + 1;
-                               swapped = true;
-                       }
-               }
+       struct btree_path *n;
 
-               if (!swapped)
-                       break;
+       if (cmp <= 0) {
+               n = prev_btree_path(trans, path);
+               if (n && btree_path_cmp(n, path) > 0) {
+                       do {
+                               btree_path_swap(trans, n, path);
+                               n = prev_btree_path(trans, path);
+                       } while (n && btree_path_cmp(n, path) > 0);
 
-               r--;
-               swapped = false;
-
-               for (i = r - 2; i >= l; --i) {
-                       if (btree_iter_cmp(trans->iters + trans->sorted[i],
-                                          trans->iters + trans->sorted[i + 1]) > 0) {
-                               swap(trans->sorted[i],
-                                    trans->sorted[i + 1]);
-                               trans->iters[trans->sorted[i]].sorted_idx = i;
-                               trans->iters[trans->sorted[i + 1]].sorted_idx = i + 1;
-                               swapped = true;
-                       }
+                       goto out;
                }
-
-               if (!swapped)
-                       break;
-
-               l++;
-               swapped = false;
-       }
-
-       btree_trans_verify_sorted_refs(trans);
-       btree_trans_verify_sorted(trans);
-}
-
-static void btree_iter_check_sort(struct btree_trans *trans, struct btree_iter *iter)
-{
-       struct btree_iter *n;
-
-       EBUG_ON(iter->sorted_idx == U8_MAX);
-
-       n = next_btree_iter(trans, iter);
-       if (n && btree_iter_cmp(iter, n) > 0) {
-               do {
-                       btree_iter_swap(trans, iter, n);
-                       n = next_btree_iter(trans, iter);
-               } while (n && btree_iter_cmp(iter, n) > 0);
-
-               return;
        }
 
-       n = prev_btree_iter(trans, iter);
-       if (n && btree_iter_cmp(n, iter) > 0) {
-               do {
-                       btree_iter_swap(trans, n, iter);
-                       n = prev_btree_iter(trans, iter);
-               } while (n && btree_iter_cmp(n, iter) > 0);
+       if (cmp >= 0) {
+               n = next_btree_path(trans, path);
+               if (n && btree_path_cmp(path, n) > 0) {
+                       do {
+                               btree_path_swap(trans, path, n);
+                               n = next_btree_path(trans, path);
+                       } while (n && btree_path_cmp(path, n) > 0);
+               }
        }
-
+out:
        btree_trans_verify_sorted(trans);
 }
 
-static inline void btree_iter_list_remove(struct btree_trans *trans,
-                                         struct btree_iter *iter)
+static inline void btree_path_list_remove(struct btree_trans *trans,
+                                         struct btree_path *path)
 {
        unsigned i;
 
-       EBUG_ON(iter->sorted_idx >= trans->nr_sorted);
+       EBUG_ON(path->sorted_idx >= trans->nr_sorted);
 
-       array_remove_item(trans->sorted, trans->nr_sorted, iter->sorted_idx);
+       array_remove_item(trans->sorted, trans->nr_sorted, path->sorted_idx);
 
-       for (i = iter->sorted_idx; i < trans->nr_sorted; i++)
-               trans->iters[trans->sorted[i]].sorted_idx = i;
+       for (i = path->sorted_idx; i < trans->nr_sorted; i++)
+               trans->paths[trans->sorted[i]].sorted_idx = i;
 
-       iter->sorted_idx = U8_MAX;
+       path->sorted_idx = U8_MAX;
 
        btree_trans_verify_sorted_refs(trans);
 }
 
-static inline void btree_iter_list_add(struct btree_trans *trans,
-                                      struct btree_iter *pos,
-                                      struct btree_iter *iter)
+static inline void btree_path_list_add(struct btree_trans *trans,
+                                      struct btree_path *pos,
+                                      struct btree_path *path)
 {
        unsigned i;
 
        btree_trans_verify_sorted_refs(trans);
 
-       iter->sorted_idx = pos ? pos->sorted_idx : trans->nr_sorted;
+       path->sorted_idx = pos ? pos->sorted_idx + 1 : 0;
 
-       array_insert_item(trans->sorted, trans->nr_sorted, iter->sorted_idx, iter->idx);
+       array_insert_item(trans->sorted, trans->nr_sorted, path->sorted_idx, path->idx);
 
-       for (i = iter->sorted_idx; i < trans->nr_sorted; i++)
-               trans->iters[trans->sorted[i]].sorted_idx = i;
+       for (i = path->sorted_idx; i < trans->nr_sorted; i++)
+               trans->paths[trans->sorted[i]].sorted_idx = i;
 
        btree_trans_verify_sorted_refs(trans);
 }
 
-static void btree_iter_child_free(struct btree_iter *iter)
-{
-       struct btree_iter *child = btree_iter_child(iter);
-
-       if (child) {
-               bch2_trans_iter_free(iter->trans, child);
-               iter->child_idx = U8_MAX;
-       }
-}
-
-static struct btree_iter *btree_iter_child_alloc(struct btree_iter *iter,
-                                                unsigned long ip)
-{
-       struct btree_trans *trans = iter->trans;
-       struct btree_iter *child = btree_iter_child(iter);
-
-       if (!child) {
-               child = btree_trans_iter_alloc(trans, iter);
-               child->ip_allocated     = ip;
-               iter->child_idx         = child->idx;
-
-               trans->iters_live       |= 1ULL << child->idx;
-               trans->iters_touched    |= 1ULL << child->idx;
-       }
-
-       return child;
-}
-
-static inline void __bch2_trans_iter_free(struct btree_trans *trans,
-                                         unsigned idx)
-{
-       btree_iter_child_free(&trans->iters[idx]);
-
-       btree_iter_list_remove(trans, &trans->iters[idx]);
-
-       __bch2_btree_iter_unlock(&trans->iters[idx]);
-       trans->iters_linked             &= ~(1ULL << idx);
-       trans->iters_live               &= ~(1ULL << idx);
-       trans->iters_touched            &= ~(1ULL << idx);
-}
-
-int bch2_trans_iter_put(struct btree_trans *trans,
-                       struct btree_iter *iter)
-{
-       int ret;
-
-       if (IS_ERR_OR_NULL(iter))
-               return 0;
-
-       BUG_ON(trans->iters + iter->idx != iter);
-       BUG_ON(!btree_iter_live(trans, iter));
-
-       ret = btree_iter_err(iter);
-
-       if (!(trans->iters_touched & (1ULL << iter->idx)) &&
-           !(iter->flags & BTREE_ITER_KEEP_UNTIL_COMMIT))
-               __bch2_trans_iter_free(trans, iter->idx);
-
-       trans->iters_live       &= ~(1ULL << iter->idx);
-       return ret;
-}
-
-int bch2_trans_iter_free(struct btree_trans *trans,
-                        struct btree_iter *iter)
-{
-       if (IS_ERR_OR_NULL(iter))
-               return 0;
-
-       set_btree_iter_dontneed(trans, iter);
-
-       return bch2_trans_iter_put(trans, iter);
-}
-
-noinline __cold
-static void btree_trans_iter_alloc_fail(struct btree_trans *trans)
-{
-
-       struct btree_iter *iter;
-       struct btree_insert_entry *i;
-       char buf[100];
-
-       btree_trans_sort_iters(trans);
-
-       trans_for_each_iter_inorder(trans, iter)
-               printk(KERN_ERR "iter: btree %s pos %s%s%s%s %pS\n",
-                      bch2_btree_ids[iter->btree_id],
-                      (bch2_bpos_to_text(&PBUF(buf), iter->real_pos), buf),
-                      btree_iter_live(trans, iter) ? " live" : "",
-                      (trans->iters_touched & (1ULL << iter->idx)) ? " touched" : "",
-                      iter->flags & BTREE_ITER_KEEP_UNTIL_COMMIT ? " keep" : "",
-                      (void *) iter->ip_allocated);
-
-       trans_for_each_update(trans, i) {
-               char buf[300];
-
-               bch2_bkey_val_to_text(&PBUF(buf), trans->c, bkey_i_to_s_c(i->k));
-               printk(KERN_ERR "update: btree %s %s\n",
-                      bch2_btree_ids[i->iter->btree_id], buf);
-       }
-       panic("trans iter oveflow\n");
-}
-
-static struct btree_iter *btree_trans_iter_alloc(struct btree_trans *trans,
-                                                struct btree_iter *pos)
-{
-       struct btree_iter *iter;
-       unsigned idx;
-
-       if (unlikely(trans->iters_linked ==
-                    ~((~0ULL << 1) << (BTREE_ITER_MAX - 1))))
-               btree_trans_iter_alloc_fail(trans);
-
-       idx = __ffs64(~trans->iters_linked);
-       iter = &trans->iters[idx];
-
-       iter->trans             = trans;
-       iter->idx               = idx;
-       iter->child_idx         = U8_MAX;
-       iter->sorted_idx        = U8_MAX;
-       iter->flags             = 0;
-       iter->nodes_locked      = 0;
-       iter->nodes_intent_locked = 0;
-       trans->iters_linked     |= 1ULL << idx;
-
-       btree_iter_list_add(trans, pos, iter);
-       return iter;
-}
-
-static void btree_iter_copy(struct btree_iter *dst, struct btree_iter *src)
+void bch2_trans_iter_exit(struct btree_trans *trans, struct btree_iter *iter)
 {
-       unsigned i;
-
-       __bch2_btree_iter_unlock(dst);
-       btree_iter_child_free(dst);
-
-       memcpy(&dst->flags, &src->flags,
-              sizeof(struct btree_iter) - offsetof(struct btree_iter, flags));
-
-       for (i = 0; i < BTREE_MAX_DEPTH; i++)
-               if (btree_node_locked(dst, i))
-                       six_lock_increment(&dst->l[i].b->c.lock,
-                                          __btree_lock_want(dst, i));
-
-       dst->flags &= ~BTREE_ITER_KEEP_UNTIL_COMMIT;
-       dst->flags &= ~BTREE_ITER_SET_POS_AFTER_COMMIT;
-
-       btree_iter_check_sort(dst->trans, dst);
+       if (iter->path)
+               bch2_path_put(trans, iter->path,
+                             iter->flags & BTREE_ITER_INTENT);
+       if (iter->update_path)
+               bch2_path_put(trans, iter->update_path,
+                             iter->flags & BTREE_ITER_INTENT);
+       if (iter->key_cache_path)
+               bch2_path_put(trans, iter->key_cache_path,
+                             iter->flags & BTREE_ITER_INTENT);
+       iter->path = NULL;
+       iter->update_path = NULL;
+       iter->key_cache_path = NULL;
 }
 
-struct btree_iter *__bch2_trans_get_iter(struct btree_trans *trans,
-                                        unsigned btree_id, struct bpos pos,
-                                        unsigned locks_want,
-                                        unsigned depth,
-                                        unsigned flags)
+static void __bch2_trans_iter_init(struct btree_trans *trans,
+                                  struct btree_iter *iter,
+                                  unsigned btree_id, struct bpos pos,
+                                  unsigned locks_want,
+                                  unsigned depth,
+                                  unsigned flags,
+                                  unsigned long ip)
 {
-       struct btree_iter *iter, *best = NULL;
-       struct bpos real_pos, pos_min = POS_MIN;
-
        EBUG_ON(trans->restarted);
 
-       if ((flags & BTREE_ITER_TYPE) != BTREE_ITER_NODES &&
-           btree_node_type_is_extents(btree_id) &&
-           !(flags & BTREE_ITER_NOT_EXTENTS) &&
-           !(flags & BTREE_ITER_ALL_SNAPSHOTS))
+       if (!(flags & (BTREE_ITER_ALL_SNAPSHOTS|BTREE_ITER_NOT_EXTENTS)) &&
+           btree_node_type_is_extents(btree_id))
                flags |= BTREE_ITER_IS_EXTENTS;
 
-       if ((flags & BTREE_ITER_TYPE) != BTREE_ITER_NODES &&
+       if (!(flags & __BTREE_ITER_ALL_SNAPSHOTS) &&
            !btree_type_has_snapshots(btree_id))
                flags &= ~BTREE_ITER_ALL_SNAPSHOTS;
 
-       if (!(flags & BTREE_ITER_ALL_SNAPSHOTS))
-               pos.snapshot = btree_type_has_snapshots(btree_id)
-                       ? U32_MAX : 0;
-
-       real_pos = pos;
-
-       if ((flags & BTREE_ITER_IS_EXTENTS) &&
-           bkey_cmp(pos, POS_MAX))
-               real_pos = bpos_nosnap_successor(pos);
-
-       trans_for_each_iter(trans, iter) {
-               if (btree_iter_type(iter) != (flags & BTREE_ITER_TYPE))
-                       continue;
-
-               if (iter->btree_id != btree_id)
-                       continue;
-
-               if (best) {
-                       int cmp = bkey_cmp(bpos_diff(best->real_pos, real_pos),
-                                          bpos_diff(iter->real_pos, real_pos));
-
-                       if (cmp < 0 ||
-                           ((cmp == 0 && btree_iter_keep(trans, iter))))
-                               continue;
-               }
-
-               best = iter;
-       }
-
-       if (!best) {
-               iter = btree_trans_iter_alloc(trans, NULL);
-               bch2_btree_iter_init(trans, iter, btree_id);
-       } else if (btree_iter_keep(trans, best)) {
-               iter = btree_trans_iter_alloc(trans, best);
-               btree_iter_copy(iter, best);
-       } else {
-               iter = best;
-       }
-
-       trans->iters_live       |= 1ULL << iter->idx;
-       trans->iters_touched    |= 1ULL << iter->idx;
-
-       iter->flags = flags;
-
-       iter->snapshot = pos.snapshot;
+       if (!(flags & BTREE_ITER_ALL_SNAPSHOTS) &&
+           btree_type_has_snapshots(btree_id))
+               flags |= BTREE_ITER_FILTER_SNAPSHOTS;
 
-       /*
-        * If the iterator has locks_want greater than requested, we explicitly
-        * do not downgrade it here - on transaction restart because btree node
-        * split needs to upgrade locks, we might be putting/getting the
-        * iterator again. Downgrading iterators only happens via an explicit
-        * bch2_trans_downgrade().
-        */
-
-       locks_want = min(locks_want, BTREE_MAX_DEPTH);
-       if (locks_want > iter->locks_want) {
-               iter->locks_want = locks_want;
-               btree_iter_get_locks(iter, true, _THIS_IP_);
-       }
+       if (!test_bit(JOURNAL_REPLAY_DONE, &trans->c->journal.flags))
+               flags |= BTREE_ITER_WITH_JOURNAL;
 
-       while (iter->level != depth) {
-               btree_node_unlock(iter, iter->level);
-               iter->l[iter->level].b = BTREE_ITER_NO_NODE_INIT;
-               iter->uptodate = BTREE_ITER_NEED_TRAVERSE;
-               if (iter->level < depth)
-                       iter->level++;
-               else
-                       iter->level--;
-       }
+       if (!btree_id_cached(trans->c, btree_id)) {
+               flags &= ~BTREE_ITER_CACHED;
+               flags &= ~BTREE_ITER_WITH_KEY_CACHE;
+       } else if (!(flags & BTREE_ITER_CACHED))
+               flags |= BTREE_ITER_WITH_KEY_CACHE;
 
+       iter->trans     = trans;
+       iter->path      = NULL;
+       iter->update_path = NULL;
+       iter->key_cache_path = NULL;
+       iter->btree_id  = btree_id;
        iter->min_depth = depth;
+       iter->flags     = flags;
+       iter->snapshot  = pos.snapshot;
+       iter->pos       = pos;
+       iter->k.type    = KEY_TYPE_deleted;
+       iter->k.p       = pos;
+       iter->k.size    = 0;
+#ifdef CONFIG_BCACHEFS_DEBUG
+       iter->ip_allocated = ip;
+#endif
 
-       bch2_btree_iter_set_pos(iter, pos);
-       btree_iter_set_search_pos(iter, real_pos);
-
-       trace_trans_get_iter(_RET_IP_, trans->ip,
-                            btree_id,
-                            &real_pos, locks_want, iter->uptodate,
-                            best ? &best->real_pos     : &pos_min,
-                            best ? best->locks_want    : U8_MAX,
-                            best ? best->uptodate      : U8_MAX);
-
-       return iter;
+       iter->path = bch2_path_get(trans, btree_id, iter->pos,
+                                  locks_want, depth, flags, ip);
 }
 
-struct btree_iter *bch2_trans_get_node_iter(struct btree_trans *trans,
-                                           enum btree_id btree_id,
-                                           struct bpos pos,
-                                           unsigned locks_want,
-                                           unsigned depth,
-                                           unsigned flags)
+void bch2_trans_iter_init(struct btree_trans *trans,
+                         struct btree_iter *iter,
+                         unsigned btree_id, struct bpos pos,
+                         unsigned flags)
 {
-       struct btree_iter *iter =
-               __bch2_trans_get_iter(trans, btree_id, pos,
-                                     locks_want, depth,
-                                     BTREE_ITER_NODES|
-                                     BTREE_ITER_NOT_EXTENTS|
-                                     BTREE_ITER_ALL_SNAPSHOTS|
-                                     flags);
-
-       BUG_ON(bkey_cmp(iter->pos, pos));
-       BUG_ON(iter->locks_want != min(locks_want, BTREE_MAX_DEPTH));
-       BUG_ON(iter->level      != depth);
-       BUG_ON(iter->min_depth  != depth);
-       iter->ip_allocated = _RET_IP_;
-
-       return iter;
+       __bch2_trans_iter_init(trans, iter, btree_id, pos,
+                              0, 0, flags, _RET_IP_);
 }
 
-struct btree_iter *__bch2_trans_copy_iter(struct btree_trans *trans,
-                                       struct btree_iter *src)
+void bch2_trans_node_iter_init(struct btree_trans *trans,
+                              struct btree_iter *iter,
+                              enum btree_id btree_id,
+                              struct bpos pos,
+                              unsigned locks_want,
+                              unsigned depth,
+                              unsigned flags)
 {
-       struct btree_iter *iter;
-
-       iter = btree_trans_iter_alloc(trans, src);
-       btree_iter_copy(iter, src);
-
-       trans->iters_live |= 1ULL << iter->idx;
-       /*
-        * We don't need to preserve this iter since it's cheap to copy it
-        * again - this will cause trans_iter_put() to free it right away:
-        */
-       set_btree_iter_dontneed(trans, iter);
+       __bch2_trans_iter_init(trans, iter, btree_id, pos, locks_want, depth,
+                              BTREE_ITER_NOT_EXTENTS|
+                              __BTREE_ITER_ALL_SNAPSHOTS|
+                              BTREE_ITER_ALL_SNAPSHOTS|
+                              flags, _RET_IP_);
+       BUG_ON(iter->path->locks_want    < min(locks_want, BTREE_MAX_DEPTH));
+       BUG_ON(iter->path->level        != depth);
+       BUG_ON(iter->min_depth          != depth);
+}
 
-       return iter;
+void bch2_trans_copy_iter(struct btree_iter *dst, struct btree_iter *src)
+{
+       *dst = *src;
+       if (src->path)
+               __btree_path_get(src->path, src->flags & BTREE_ITER_INTENT);
+       if (src->update_path)
+               __btree_path_get(src->update_path, src->flags & BTREE_ITER_INTENT);
+       dst->key_cache_path = NULL;
 }
 
 void *bch2_trans_kmalloc(struct btree_trans *trans, size_t size)
@@ -2488,7 +2972,7 @@ void *bch2_trans_kmalloc(struct btree_trans *trans, size_t size)
                trans->mem_bytes = new_bytes;
 
                if (old_bytes) {
-                       trace_trans_restart_mem_realloced(trans->ip, _RET_IP_, new_bytes);
+                       trace_trans_restart_mem_realloced(trans->fn, _RET_IP_, new_bytes);
                        btree_trans_restart(trans);
                        return ERR_PTR(-EINTR);
                }
@@ -2500,20 +2984,6 @@ void *bch2_trans_kmalloc(struct btree_trans *trans, size_t size)
        return p;
 }
 
-inline void bch2_trans_unlink_iters(struct btree_trans *trans)
-{
-       u64 iters = trans->iters_linked &
-               ~trans->iters_touched &
-               ~trans->iters_live;
-
-       while (iters) {
-               unsigned idx = __ffs64(iters);
-
-               iters &= ~(1ULL << idx);
-               __bch2_trans_iter_free(trans, idx);
-       }
-}
-
 /**
  * bch2_trans_begin() - reset a transaction after a interrupted attempt
  * @trans: transaction to reset
@@ -2524,19 +2994,13 @@ inline void bch2_trans_unlink_iters(struct btree_trans *trans)
  */
 void bch2_trans_begin(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
-
-       trans_for_each_iter(trans, iter)
-               iter->flags &= ~(BTREE_ITER_KEEP_UNTIL_COMMIT|
-                                BTREE_ITER_SET_POS_AFTER_COMMIT);
+       struct btree_insert_entry *i;
+       struct btree_path *path;
 
-       /*
-        * XXX: we shouldn't be doing this if the transaction was restarted, but
-        * currently we still overflow transaction iterators if we do that
-        * */
-       bch2_trans_unlink_iters(trans);
-       trans->iters_touched &= trans->iters_live;
+       trans_for_each_update(trans, i)
+               __btree_path_put(i->path, true);
 
+       memset(&trans->journal_res, 0, sizeof(trans->journal_res));
        trans->extra_journal_res        = 0;
        trans->nr_updates               = 0;
        trans->mem_top                  = 0;
@@ -2552,48 +3016,59 @@ void bch2_trans_begin(struct btree_trans *trans)
                       (void *) &trans->fs_usage_deltas->memset_start);
        }
 
+       trans_for_each_path(trans, path) {
+               path->should_be_locked = false;
+
+               /*
+                * XXX: we probably shouldn't be doing this if the transaction
+                * was restarted, but currently we still overflow transaction
+                * iterators if we do that
+                */
+               if (!path->ref && !path->preserve)
+                       __bch2_path_free(trans, path);
+               else if (!path->ref)
+                       path->preserve = false;
+       }
+
        bch2_trans_cond_resched(trans);
 
        if (trans->restarted)
-               bch2_btree_iter_traverse_all(trans);
+               bch2_btree_path_traverse_all(trans);
 
        trans->restarted = false;
 }
 
-static void bch2_trans_alloc_iters(struct btree_trans *trans, struct bch_fs *c)
+static void bch2_trans_alloc_paths(struct btree_trans *trans, struct bch_fs *c)
 {
-       size_t iters_bytes      = sizeof(struct btree_iter) * BTREE_ITER_MAX;
+       size_t paths_bytes      = sizeof(struct btree_path) * BTREE_ITER_MAX;
        size_t updates_bytes    = sizeof(struct btree_insert_entry) * BTREE_ITER_MAX;
-       size_t sorted_bytes     = sizeof(u8) * BTREE_ITER_MAX;
        void *p = NULL;
 
        BUG_ON(trans->used_mempool);
 
 #ifdef __KERNEL__
-       p = this_cpu_xchg(c->btree_iters_bufs->iter, NULL);
+       p = this_cpu_xchg(c->btree_paths_bufs->path , NULL);
 #endif
        if (!p)
-               p = mempool_alloc(&trans->c->btree_iters_pool, GFP_NOFS);
+               p = mempool_alloc(&trans->c->btree_paths_pool, GFP_NOFS);
 
-       trans->iters            = p; p += iters_bytes;
+       trans->paths            = p; p += paths_bytes;
        trans->updates          = p; p += updates_bytes;
-       trans->sorted           = p; p += sorted_bytes;
 }
 
-void bch2_trans_init(struct btree_trans *trans, struct bch_fs *c,
-                    unsigned expected_nr_iters,
-                    size_t expected_mem_bytes)
+void __bch2_trans_init(struct btree_trans *trans, struct bch_fs *c,
+                      unsigned expected_nr_iters,
+                      size_t expected_mem_bytes,
+                      const char *fn)
        __acquires(&c->btree_trans_barrier)
 {
+       BUG_ON(lock_class_is_held(&bch2_btree_node_lock_key));
+
        memset(trans, 0, sizeof(*trans));
        trans->c                = c;
-       trans->ip               = _RET_IP_;
+       trans->fn               = fn;
 
-       /*
-        * reallocating iterators currently completely breaks
-        * bch2_trans_iter_put(), we always allocate the max:
-        */
-       bch2_trans_alloc_iters(trans, c);
+       bch2_trans_alloc_paths(trans, c);
 
        if (expected_mem_bytes) {
                trans->mem_bytes = roundup_pow_of_two(expected_mem_bytes);
@@ -2607,62 +3082,67 @@ void bch2_trans_init(struct btree_trans *trans, struct bch_fs *c,
 
        trans->srcu_idx = srcu_read_lock(&c->btree_trans_barrier);
 
-#ifdef CONFIG_BCACHEFS_DEBUG
        trans->pid = current->pid;
        mutex_lock(&c->btree_trans_lock);
        list_add(&trans->list, &c->btree_trans_list);
        mutex_unlock(&c->btree_trans_lock);
+}
+
+static void check_btree_paths_leaked(struct btree_trans *trans)
+{
+#ifdef CONFIG_BCACHEFS_DEBUG
+       struct bch_fs *c = trans->c;
+       struct btree_path *path;
+
+       trans_for_each_path(trans, path)
+               if (path->ref)
+                       goto leaked;
+       return;
+leaked:
+       bch_err(c, "btree paths leaked from %s!", trans->fn);
+       trans_for_each_path(trans, path)
+               if (path->ref)
+                       printk(KERN_ERR "  btree %s %pS\n",
+                              bch2_btree_ids[path->btree_id],
+                              (void *) path->ip_allocated);
+       /* Be noisy about this: */
+       bch2_fatal_error(c);
 #endif
 }
 
-int bch2_trans_exit(struct btree_trans *trans)
+void bch2_trans_exit(struct btree_trans *trans)
        __releases(&c->btree_trans_barrier)
 {
+       struct btree_insert_entry *i;
        struct bch_fs *c = trans->c;
 
        bch2_trans_unlock(trans);
 
-#ifdef CONFIG_BCACHEFS_DEBUG
-       if (trans->iters_live) {
-               struct btree_iter *iter;
-
-               trans_for_each_iter(trans, iter)
-                       btree_iter_child_free(iter);
-       }
-
-       if (trans->iters_live) {
-               struct btree_iter *iter;
+       trans_for_each_update(trans, i)
+               __btree_path_put(i->path, true);
+       trans->nr_updates               = 0;
 
-               bch_err(c, "btree iterators leaked!");
-               trans_for_each_iter(trans, iter)
-                       if (btree_iter_live(trans, iter))
-                               printk(KERN_ERR "  btree %s allocated at %pS\n",
-                                      bch2_btree_ids[iter->btree_id],
-                                      (void *) iter->ip_allocated);
-               /* Be noisy about this: */
-               bch2_fatal_error(c);
-       }
+       check_btree_paths_leaked(trans);
 
-       mutex_lock(&trans->c->btree_trans_lock);
+       mutex_lock(&c->btree_trans_lock);
        list_del(&trans->list);
-       mutex_unlock(&trans->c->btree_trans_lock);
-#endif
+       mutex_unlock(&c->btree_trans_lock);
 
        srcu_read_unlock(&c->btree_trans_barrier, trans->srcu_idx);
 
-       bch2_journal_preres_put(&trans->c->journal, &trans->journal_preres);
+       bch2_journal_preres_put(&c->journal, &trans->journal_preres);
 
        if (trans->fs_usage_deltas) {
                if (trans->fs_usage_deltas->size + sizeof(trans->fs_usage_deltas) ==
                    REPLICAS_DELTA_LIST_MAX)
                        mempool_free(trans->fs_usage_deltas,
-                                    &trans->c->replicas_delta_pool);
+                                    &c->replicas_delta_pool);
                else
                        kfree(trans->fs_usage_deltas);
        }
 
        if (trans->mem_bytes == BTREE_TRANS_MEM_MAX)
-               mempool_free(trans->mem, &trans->c->btree_trans_mem_pool);
+               mempool_free(trans->mem, &c->btree_trans_mem_pool);
        else
                kfree(trans->mem);
 
@@ -2670,74 +3150,70 @@ int bch2_trans_exit(struct btree_trans *trans)
        /*
         * Userspace doesn't have a real percpu implementation:
         */
-       trans->iters = this_cpu_xchg(c->btree_iters_bufs->iter, trans->iters);
+       trans->paths = this_cpu_xchg(c->btree_paths_bufs->path, trans->paths);
 #endif
 
-       if (trans->iters)
-               mempool_free(trans->iters, &trans->c->btree_iters_pool);
+       if (trans->paths)
+               mempool_free(trans->paths, &c->btree_paths_pool);
 
        trans->mem      = (void *) 0x1;
-       trans->iters    = (void *) 0x1;
-
-       return trans->error ? -EIO : 0;
+       trans->paths    = (void *) 0x1;
 }
 
 static void __maybe_unused
-bch2_btree_iter_node_to_text(struct printbuf *out,
+bch2_btree_path_node_to_text(struct printbuf *out,
                             struct btree_bkey_cached_common *_b,
-                            enum btree_iter_type type)
+                            bool cached)
 {
        pr_buf(out, "    l=%u %s:",
               _b->level, bch2_btree_ids[_b->btree_id]);
-       bch2_bpos_to_text(out, btree_node_pos(_b, type));
+       bch2_bpos_to_text(out, btree_node_pos(_b, cached));
 }
 
-#ifdef CONFIG_BCACHEFS_DEBUG
-static bool trans_has_btree_nodes_locked(struct btree_trans *trans)
+static bool trans_has_locks(struct btree_trans *trans)
 {
-       struct btree_iter *iter;
+       struct btree_path *path;
 
-       trans_for_each_iter(trans, iter)
-               if (btree_iter_type(iter) != BTREE_ITER_CACHED &&
-                   iter->nodes_locked)
+       trans_for_each_path(trans, path)
+               if (path->nodes_locked)
                        return true;
        return false;
 }
-#endif
 
 void bch2_btree_trans_to_text(struct printbuf *out, struct bch_fs *c)
 {
-#ifdef CONFIG_BCACHEFS_DEBUG
        struct btree_trans *trans;
-       struct btree_iter *iter;
+       struct btree_path *path;
        struct btree *b;
+       static char lock_types[] = { 'r', 'i', 'w' };
        unsigned l;
 
        mutex_lock(&c->btree_trans_lock);
        list_for_each_entry(trans, &c->btree_trans_list, list) {
-               if (!trans_has_btree_nodes_locked(trans))
+               if (!trans_has_locks(trans))
                        continue;
 
-               pr_buf(out, "%i %ps\n", trans->pid, (void *) trans->ip);
+               pr_buf(out, "%i %s\n", trans->pid, trans->fn);
 
-               trans_for_each_iter(trans, iter) {
-                       if (!iter->nodes_locked)
+               trans_for_each_path(trans, path) {
+                       if (!path->nodes_locked)
                                continue;
 
-                       pr_buf(out, "  iter %u %c %s:",
-                              iter->idx,
-                              btree_iter_type(iter) == BTREE_ITER_CACHED ? 'c' : 'b',
-                              bch2_btree_ids[iter->btree_id]);
-                       bch2_bpos_to_text(out, iter->pos);
+                       pr_buf(out, "  path %u %c l=%u %s:",
+                              path->idx,
+                              path->cached ? 'c' : 'b',
+                              path->level,
+                              bch2_btree_ids[path->btree_id]);
+                       bch2_bpos_to_text(out, path->pos);
                        pr_buf(out, "\n");
 
                        for (l = 0; l < BTREE_MAX_DEPTH; l++) {
-                               if (btree_node_locked(iter, l)) {
+                               if (btree_node_locked(path, l)) {
                                        pr_buf(out, "    %s l=%u ",
-                                              btree_node_intent_locked(iter, l) ? "i" : "r", l);
-                                       bch2_btree_iter_node_to_text(out,
-                                                       (void *) iter->l[l].b,
-                                                       btree_iter_type(iter));
+                                              btree_node_intent_locked(path, l) ? "i" : "r", l);
+                                       bch2_btree_path_node_to_text(out,
+                                                       (void *) path->l[l].b,
+                                                       path->cached);
                                        pr_buf(out, "\n");
                                }
                        }
@@ -2745,44 +3221,47 @@ void bch2_btree_trans_to_text(struct printbuf *out, struct bch_fs *c)
 
                b = READ_ONCE(trans->locking);
                if (b) {
-                       iter = &trans->iters[trans->locking_iter_idx];
-                       pr_buf(out, "  locking iter %u %c l=%u %s:",
-                              trans->locking_iter_idx,
-                              btree_iter_type(iter) == BTREE_ITER_CACHED ? 'c' : 'b',
+                       path = &trans->paths[trans->locking_path_idx];
+                       pr_buf(out, "  locking path %u %c l=%u %c %s:",
+                              trans->locking_path_idx,
+                              path->cached ? 'c' : 'b',
                               trans->locking_level,
+                              lock_types[trans->locking_lock_type],
                               bch2_btree_ids[trans->locking_btree_id]);
                        bch2_bpos_to_text(out, trans->locking_pos);
 
                        pr_buf(out, " node ");
-                       bch2_btree_iter_node_to_text(out,
-                                       (void *) b,
-                                       btree_iter_type(iter));
+                       bch2_btree_path_node_to_text(out,
+                                       (void *) b, path->cached);
                        pr_buf(out, "\n");
                }
        }
        mutex_unlock(&c->btree_trans_lock);
-#endif
 }
 
 void bch2_fs_btree_iter_exit(struct bch_fs *c)
 {
+       if (c->btree_trans_barrier_initialized)
+               cleanup_srcu_struct(&c->btree_trans_barrier);
        mempool_exit(&c->btree_trans_mem_pool);
-       mempool_exit(&c->btree_iters_pool);
-       cleanup_srcu_struct(&c->btree_trans_barrier);
+       mempool_exit(&c->btree_paths_pool);
 }
 
 int bch2_fs_btree_iter_init(struct bch_fs *c)
 {
        unsigned nr = BTREE_ITER_MAX;
+       int ret;
 
        INIT_LIST_HEAD(&c->btree_trans_list);
        mutex_init(&c->btree_trans_lock);
 
-       return  init_srcu_struct(&c->btree_trans_barrier) ?:
-               mempool_init_kmalloc_pool(&c->btree_iters_pool, 1,
-                       sizeof(u8) * nr +
-                       sizeof(struct btree_iter) * nr +
+       ret   = mempool_init_kmalloc_pool(&c->btree_paths_pool, 1,
+                       sizeof(struct btree_path) * nr +
                        sizeof(struct btree_insert_entry) * nr) ?:
                mempool_init_kmalloc_pool(&c->btree_trans_mem_pool, 1,
-                                         BTREE_TRANS_MEM_MAX);
+                                         BTREE_TRANS_MEM_MAX) ?:
+               init_srcu_struct(&c->btree_trans_barrier);
+       if (!ret)
+               c->btree_trans_barrier_initialized = true;
+       return ret;
 }