1 files changed, 23 insertions, 49 deletions

diff --git a/fs/bcachefs/btree_iter.c b/fs/bcachefs/btree_iter.c
index 8962c481e310..a67babf69d39 100644
--- a/fs/bcachefs/btree_iter.c
+++ b/fs/bcachefs/btree_iter.c
@@ -2450,27 +2450,10 @@ struct bkey_s_c bch2_btree_iter_peek_max(struct btree_iter *iter, struct bpos en
 				continue;
 			}
 
-			if (!(iter->flags & BTREE_ITER_nofilter_whiteouts)) {
-				/*
-				 * KEY_TYPE_extent_whiteout indicates that there
-				 * are no extents that overlap with this
-				 * whiteout - meaning bkey_start_pos() is
-				 * monotonically increasing when including
-				 * KEY_TYPE_extent_whiteout (not
-				 * KEY_TYPE_whiteout).
-				 *
-				 * Without this @end wouldn't be able to
-				 * terminate searches and we'd have to scan
-				 * through tons of whiteouts:
-				 */
-				if (k.k->type == KEY_TYPE_extent_whiteout &&
-				    bkey_ge(k.k->p, end))
-					goto end;
-
-				if (bkey_extent_whiteout(k.k)) {
-					search_key = bkey_successor(iter, k.k->p);
-					continue;
-				}
+			if (bkey_whiteout(k.k) &&
+			    !(iter->flags & BTREE_ITER_nofilter_whiteouts)) {
+				search_key = bkey_successor(iter, k.k->p);
+				continue;
 			}
 		}
 
@@ -2728,7 +2711,7 @@ struct bkey_s_c bch2_btree_iter_peek_prev_min(struct btree_iter *iter, struct bp
 					saved_path = 0;
 				}
 
-				if (!bkey_extent_whiteout(k.k)) {
+				if (!bkey_whiteout(k.k)) {
 					saved_path = btree_path_clone(trans, iter->path,
 								iter->flags & BTREE_ITER_intent,
 								_THIS_IP_);
@@ -2741,7 +2724,7 @@ struct bkey_s_c bch2_btree_iter_peek_prev_min(struct btree_iter *iter, struct bp
 				continue;
 			}
 
-			if (bkey_extent_whiteout(k.k)) {
+			if (bkey_whiteout(k.k)) {
 				search_key = bkey_predecessor(iter, k.k->p);
 				search_key.snapshot = U32_MAX;
 				continue;
@@ -2882,7 +2865,7 @@ struct bkey_s_c bch2_btree_iter_peek_slot(struct btree_iter *iter)
 			iter->k = *k.k;
 		}
 
-		if (unlikely(bkey_extent_whiteout(k.k) &&
+		if (unlikely(k.k->type == KEY_TYPE_whiteout &&
 			     (iter->flags & BTREE_ITER_filter_snapshots) &&
 			     !(iter->flags & BTREE_ITER_nofilter_whiteouts)))
 			iter->k.type = KEY_TYPE_deleted;
@@ -2895,40 +2878,31 @@ struct bkey_s_c bch2_btree_iter_peek_slot(struct btree_iter *iter)
 
 		EBUG_ON(btree_iter_path(trans, iter)->level);
 
-		struct btree_iter iter2;
-
-		bch2_trans_copy_iter(&iter2, iter);
-		iter2.flags |= BTREE_ITER_nofilter_whiteouts;
+		if (iter->flags & BTREE_ITER_intent) {
+			struct btree_iter iter2;
 
-		while (1) {
+			bch2_trans_copy_iter(&iter2, iter);
 			k = bch2_btree_iter_peek_max(&iter2, end);
-			if ((iter2.flags & BTREE_ITER_is_extents) &&
-			    k.k &&
-			    !bkey_err(k) &&
-			    k.k->type == KEY_TYPE_whiteout) {
-				bch2_btree_iter_set_pos(&iter2, k.k->p);
-				continue;
-			}
 
-			break;
-		}
+			if (k.k && !bkey_err(k)) {
+				swap(iter->key_cache_path, iter2.key_cache_path);
+				iter->k = iter2.k;
+				k.k = &iter->k;
+			}
+			bch2_trans_iter_exit(&iter2);
+		} else {
+			struct bpos pos = iter->pos;
 
-		if (k.k && !bkey_err(k)) {
-			swap(iter->key_cache_path, iter2.key_cache_path);
-			iter->k = iter2.k;
-			k.k = &iter->k;
+			k = bch2_btree_iter_peek_max(iter, end);
+			if (unlikely(bkey_err(k)))
+				bch2_btree_iter_set_pos(iter, pos);
+			else
+				iter->pos = pos;
 		}
-		bch2_trans_iter_exit(&iter2);
 
 		if (unlikely(bkey_err(k)))
 			goto out;
 
-		if (unlikely(k.k &&
-			     bkey_extent_whiteout(k.k) &&
-			     (iter->flags & BTREE_ITER_filter_snapshots) &&
-			     !(iter->flags & BTREE_ITER_nofilter_whiteouts)))
-			iter->k.type = KEY_TYPE_deleted;
-
 		next = k.k ? bkey_start_pos(k.k) : POS_MAX;
 
 		if (bkey_lt(iter->pos, next)) {