xref: /openbmc/linux/fs/btrfs/ctree.c (revision 6a7a665d78c5dd8bc76a010648c4e7d84517ab5a)
16cbd5570SChris Mason /*
2d352ac68SChris Mason  * Copyright (C) 2007,2008 Oracle.  All rights reserved.
36cbd5570SChris Mason  *
46cbd5570SChris Mason  * This program is free software; you can redistribute it and/or
56cbd5570SChris Mason  * modify it under the terms of the GNU General Public
66cbd5570SChris Mason  * License v2 as published by the Free Software Foundation.
76cbd5570SChris Mason  *
86cbd5570SChris Mason  * This program is distributed in the hope that it will be useful,
96cbd5570SChris Mason  * but WITHOUT ANY WARRANTY; without even the implied warranty of
106cbd5570SChris Mason  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
116cbd5570SChris Mason  * General Public License for more details.
126cbd5570SChris Mason  *
136cbd5570SChris Mason  * You should have received a copy of the GNU General Public
146cbd5570SChris Mason  * License along with this program; if not, write to the
156cbd5570SChris Mason  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
166cbd5570SChris Mason  * Boston, MA 021110-1307, USA.
176cbd5570SChris Mason  */
186cbd5570SChris Mason 
19a6b6e75eSChris Mason #include <linux/sched.h>
205a0e3ad6STejun Heo #include <linux/slab.h>
21bd989ba3SJan Schmidt #include <linux/rbtree.h>
22eb60ceacSChris Mason #include "ctree.h"
23eb60ceacSChris Mason #include "disk-io.h"
247f5c1516SChris Mason #include "transaction.h"
255f39d397SChris Mason #include "print-tree.h"
26925baeddSChris Mason #include "locking.h"
279a8dd150SChris Mason 
28e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
29e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
30e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
31d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
32cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend);
335f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
345f39d397SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
35971a1f66SChris Mason 			  struct extent_buffer *src, int empty);
365f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
375f39d397SChris Mason 			      struct btrfs_root *root,
385f39d397SChris Mason 			      struct extent_buffer *dst_buf,
395f39d397SChris Mason 			      struct extent_buffer *src_buf);
40143bede5SJeff Mahoney static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
41f3ea38daSJan Schmidt 		    struct btrfs_path *path, int level, int slot,
42f3ea38daSJan Schmidt 		    int tree_mod_log);
43f230475eSJan Schmidt static void tree_mod_log_free_eb(struct btrfs_fs_info *fs_info,
44f230475eSJan Schmidt 				 struct extent_buffer *eb);
45f230475eSJan Schmidt struct extent_buffer *read_old_tree_block(struct btrfs_root *root, u64 bytenr,
46f230475eSJan Schmidt 					  u32 blocksize, u64 parent_transid,
47f230475eSJan Schmidt 					  u64 time_seq);
48f230475eSJan Schmidt struct extent_buffer *btrfs_find_old_tree_block(struct btrfs_root *root,
49f230475eSJan Schmidt 						u64 bytenr, u32 blocksize,
50f230475eSJan Schmidt 						u64 time_seq);
51d97e63b6SChris Mason 
522c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
532c90e5d6SChris Mason {
54df24a2b9SChris Mason 	struct btrfs_path *path;
55e00f7308SJeff Mahoney 	path = kmem_cache_zalloc(btrfs_path_cachep, GFP_NOFS);
56df24a2b9SChris Mason 	return path;
572c90e5d6SChris Mason }
582c90e5d6SChris Mason 
59b4ce94deSChris Mason /*
60b4ce94deSChris Mason  * set all locked nodes in the path to blocking locks.  This should
61b4ce94deSChris Mason  * be done before scheduling
62b4ce94deSChris Mason  */
63b4ce94deSChris Mason noinline void btrfs_set_path_blocking(struct btrfs_path *p)
64b4ce94deSChris Mason {
65b4ce94deSChris Mason 	int i;
66b4ce94deSChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
67bd681513SChris Mason 		if (!p->nodes[i] || !p->locks[i])
68bd681513SChris Mason 			continue;
69bd681513SChris Mason 		btrfs_set_lock_blocking_rw(p->nodes[i], p->locks[i]);
70bd681513SChris Mason 		if (p->locks[i] == BTRFS_READ_LOCK)
71bd681513SChris Mason 			p->locks[i] = BTRFS_READ_LOCK_BLOCKING;
72bd681513SChris Mason 		else if (p->locks[i] == BTRFS_WRITE_LOCK)
73bd681513SChris Mason 			p->locks[i] = BTRFS_WRITE_LOCK_BLOCKING;
74b4ce94deSChris Mason 	}
75b4ce94deSChris Mason }
76b4ce94deSChris Mason 
77b4ce94deSChris Mason /*
78b4ce94deSChris Mason  * reset all the locked nodes in the patch to spinning locks.
794008c04aSChris Mason  *
804008c04aSChris Mason  * held is used to keep lockdep happy, when lockdep is enabled
814008c04aSChris Mason  * we set held to a blocking lock before we go around and
824008c04aSChris Mason  * retake all the spinlocks in the path.  You can safely use NULL
834008c04aSChris Mason  * for held
84b4ce94deSChris Mason  */
854008c04aSChris Mason noinline void btrfs_clear_path_blocking(struct btrfs_path *p,
86bd681513SChris Mason 					struct extent_buffer *held, int held_rw)
87b4ce94deSChris Mason {
88b4ce94deSChris Mason 	int i;
894008c04aSChris Mason 
904008c04aSChris Mason #ifdef CONFIG_DEBUG_LOCK_ALLOC
914008c04aSChris Mason 	/* lockdep really cares that we take all of these spinlocks
924008c04aSChris Mason 	 * in the right order.  If any of the locks in the path are not
934008c04aSChris Mason 	 * currently blocking, it is going to complain.  So, make really
944008c04aSChris Mason 	 * really sure by forcing the path to blocking before we clear
954008c04aSChris Mason 	 * the path blocking.
964008c04aSChris Mason 	 */
97bd681513SChris Mason 	if (held) {
98bd681513SChris Mason 		btrfs_set_lock_blocking_rw(held, held_rw);
99bd681513SChris Mason 		if (held_rw == BTRFS_WRITE_LOCK)
100bd681513SChris Mason 			held_rw = BTRFS_WRITE_LOCK_BLOCKING;
101bd681513SChris Mason 		else if (held_rw == BTRFS_READ_LOCK)
102bd681513SChris Mason 			held_rw = BTRFS_READ_LOCK_BLOCKING;
103bd681513SChris Mason 	}
1044008c04aSChris Mason 	btrfs_set_path_blocking(p);
1054008c04aSChris Mason #endif
1064008c04aSChris Mason 
1074008c04aSChris Mason 	for (i = BTRFS_MAX_LEVEL - 1; i >= 0; i--) {
108bd681513SChris Mason 		if (p->nodes[i] && p->locks[i]) {
109bd681513SChris Mason 			btrfs_clear_lock_blocking_rw(p->nodes[i], p->locks[i]);
110bd681513SChris Mason 			if (p->locks[i] == BTRFS_WRITE_LOCK_BLOCKING)
111bd681513SChris Mason 				p->locks[i] = BTRFS_WRITE_LOCK;
112bd681513SChris Mason 			else if (p->locks[i] == BTRFS_READ_LOCK_BLOCKING)
113bd681513SChris Mason 				p->locks[i] = BTRFS_READ_LOCK;
114bd681513SChris Mason 		}
115b4ce94deSChris Mason 	}
1164008c04aSChris Mason 
1174008c04aSChris Mason #ifdef CONFIG_DEBUG_LOCK_ALLOC
1184008c04aSChris Mason 	if (held)
119bd681513SChris Mason 		btrfs_clear_lock_blocking_rw(held, held_rw);
1204008c04aSChris Mason #endif
121b4ce94deSChris Mason }
122b4ce94deSChris Mason 
123d352ac68SChris Mason /* this also releases the path */
1242c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
1252c90e5d6SChris Mason {
126ff175d57SJesper Juhl 	if (!p)
127ff175d57SJesper Juhl 		return;
128b3b4aa74SDavid Sterba 	btrfs_release_path(p);
1292c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
1302c90e5d6SChris Mason }
1312c90e5d6SChris Mason 
132d352ac68SChris Mason /*
133d352ac68SChris Mason  * path release drops references on the extent buffers in the path
134d352ac68SChris Mason  * and it drops any locks held by this path
135d352ac68SChris Mason  *
136d352ac68SChris Mason  * It is safe to call this on paths that no locks or extent buffers held.
137d352ac68SChris Mason  */
138b3b4aa74SDavid Sterba noinline void btrfs_release_path(struct btrfs_path *p)
139eb60ceacSChris Mason {
140eb60ceacSChris Mason 	int i;
141a2135011SChris Mason 
142234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
1433f157a2fSChris Mason 		p->slots[i] = 0;
144eb60ceacSChris Mason 		if (!p->nodes[i])
145925baeddSChris Mason 			continue;
146925baeddSChris Mason 		if (p->locks[i]) {
147bd681513SChris Mason 			btrfs_tree_unlock_rw(p->nodes[i], p->locks[i]);
148925baeddSChris Mason 			p->locks[i] = 0;
149925baeddSChris Mason 		}
1505f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
1513f157a2fSChris Mason 		p->nodes[i] = NULL;
152eb60ceacSChris Mason 	}
153eb60ceacSChris Mason }
154eb60ceacSChris Mason 
155d352ac68SChris Mason /*
156d352ac68SChris Mason  * safely gets a reference on the root node of a tree.  A lock
157d352ac68SChris Mason  * is not taken, so a concurrent writer may put a different node
158d352ac68SChris Mason  * at the root of the tree.  See btrfs_lock_root_node for the
159d352ac68SChris Mason  * looping required.
160d352ac68SChris Mason  *
161d352ac68SChris Mason  * The extent buffer returned by this has a reference taken, so
162d352ac68SChris Mason  * it won't disappear.  It may stop being the root of the tree
163d352ac68SChris Mason  * at any time because there are no locks held.
164d352ac68SChris Mason  */
165925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
166925baeddSChris Mason {
167925baeddSChris Mason 	struct extent_buffer *eb;
168240f62c8SChris Mason 
1693083ee2eSJosef Bacik 	while (1) {
170240f62c8SChris Mason 		rcu_read_lock();
171240f62c8SChris Mason 		eb = rcu_dereference(root->node);
1723083ee2eSJosef Bacik 
1733083ee2eSJosef Bacik 		/*
1743083ee2eSJosef Bacik 		 * RCU really hurts here, we could free up the root node because
1753083ee2eSJosef Bacik 		 * it was cow'ed but we may not get the new root node yet so do
1763083ee2eSJosef Bacik 		 * the inc_not_zero dance and if it doesn't work then
1773083ee2eSJosef Bacik 		 * synchronize_rcu and try again.
1783083ee2eSJosef Bacik 		 */
1793083ee2eSJosef Bacik 		if (atomic_inc_not_zero(&eb->refs)) {
180240f62c8SChris Mason 			rcu_read_unlock();
1813083ee2eSJosef Bacik 			break;
1823083ee2eSJosef Bacik 		}
1833083ee2eSJosef Bacik 		rcu_read_unlock();
1843083ee2eSJosef Bacik 		synchronize_rcu();
1853083ee2eSJosef Bacik 	}
186925baeddSChris Mason 	return eb;
187925baeddSChris Mason }
188925baeddSChris Mason 
189d352ac68SChris Mason /* loop around taking references on and locking the root node of the
190d352ac68SChris Mason  * tree until you end up with a lock on the root.  A locked buffer
191d352ac68SChris Mason  * is returned, with a reference held.
192d352ac68SChris Mason  */
193925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
194925baeddSChris Mason {
195925baeddSChris Mason 	struct extent_buffer *eb;
196925baeddSChris Mason 
197925baeddSChris Mason 	while (1) {
198925baeddSChris Mason 		eb = btrfs_root_node(root);
199925baeddSChris Mason 		btrfs_tree_lock(eb);
200240f62c8SChris Mason 		if (eb == root->node)
201925baeddSChris Mason 			break;
202925baeddSChris Mason 		btrfs_tree_unlock(eb);
203925baeddSChris Mason 		free_extent_buffer(eb);
204925baeddSChris Mason 	}
205925baeddSChris Mason 	return eb;
206925baeddSChris Mason }
207925baeddSChris Mason 
208bd681513SChris Mason /* loop around taking references on and locking the root node of the
209bd681513SChris Mason  * tree until you end up with a lock on the root.  A locked buffer
210bd681513SChris Mason  * is returned, with a reference held.
211bd681513SChris Mason  */
212bd681513SChris Mason struct extent_buffer *btrfs_read_lock_root_node(struct btrfs_root *root)
213bd681513SChris Mason {
214bd681513SChris Mason 	struct extent_buffer *eb;
215bd681513SChris Mason 
216bd681513SChris Mason 	while (1) {
217bd681513SChris Mason 		eb = btrfs_root_node(root);
218bd681513SChris Mason 		btrfs_tree_read_lock(eb);
219bd681513SChris Mason 		if (eb == root->node)
220bd681513SChris Mason 			break;
221bd681513SChris Mason 		btrfs_tree_read_unlock(eb);
222bd681513SChris Mason 		free_extent_buffer(eb);
223bd681513SChris Mason 	}
224bd681513SChris Mason 	return eb;
225bd681513SChris Mason }
226bd681513SChris Mason 
227d352ac68SChris Mason /* cowonly root (everything not a reference counted cow subvolume), just get
228d352ac68SChris Mason  * put onto a simple dirty list.  transaction.c walks this to make sure they
229d352ac68SChris Mason  * get properly updated on disk.
230d352ac68SChris Mason  */
2310b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root)
2320b86a832SChris Mason {
233e5846fc6SChris Mason 	spin_lock(&root->fs_info->trans_lock);
2340b86a832SChris Mason 	if (root->track_dirty && list_empty(&root->dirty_list)) {
2350b86a832SChris Mason 		list_add(&root->dirty_list,
2360b86a832SChris Mason 			 &root->fs_info->dirty_cowonly_roots);
2370b86a832SChris Mason 	}
238e5846fc6SChris Mason 	spin_unlock(&root->fs_info->trans_lock);
2390b86a832SChris Mason }
2400b86a832SChris Mason 
241d352ac68SChris Mason /*
242d352ac68SChris Mason  * used by snapshot creation to make a copy of a root for a tree with
243d352ac68SChris Mason  * a given objectid.  The buffer with the new root node is returned in
244d352ac68SChris Mason  * cow_ret, and this func returns zero on success or a negative error code.
245d352ac68SChris Mason  */
246be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
247be20aa9dSChris Mason 		      struct btrfs_root *root,
248be20aa9dSChris Mason 		      struct extent_buffer *buf,
249be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
250be20aa9dSChris Mason {
251be20aa9dSChris Mason 	struct extent_buffer *cow;
252be20aa9dSChris Mason 	int ret = 0;
253be20aa9dSChris Mason 	int level;
2545d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
255be20aa9dSChris Mason 
256be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
257be20aa9dSChris Mason 		root->fs_info->running_transaction->transid);
258be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
259be20aa9dSChris Mason 
260be20aa9dSChris Mason 	level = btrfs_header_level(buf);
2615d4f98a2SYan Zheng 	if (level == 0)
2625d4f98a2SYan Zheng 		btrfs_item_key(buf, &disk_key, 0);
2635d4f98a2SYan Zheng 	else
2645d4f98a2SYan Zheng 		btrfs_node_key(buf, &disk_key, 0);
26531840ae1SZheng Yan 
2665d4f98a2SYan Zheng 	cow = btrfs_alloc_free_block(trans, root, buf->len, 0,
2675d4f98a2SYan Zheng 				     new_root_objectid, &disk_key, level,
2685581a51aSJan Schmidt 				     buf->start, 0);
2695d4f98a2SYan Zheng 	if (IS_ERR(cow))
270be20aa9dSChris Mason 		return PTR_ERR(cow);
271be20aa9dSChris Mason 
272be20aa9dSChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
273be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
274be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
2755d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
2765d4f98a2SYan Zheng 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
2775d4f98a2SYan Zheng 				     BTRFS_HEADER_FLAG_RELOC);
2785d4f98a2SYan Zheng 	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
2795d4f98a2SYan Zheng 		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
2805d4f98a2SYan Zheng 	else
281be20aa9dSChris Mason 		btrfs_set_header_owner(cow, new_root_objectid);
282be20aa9dSChris Mason 
2832b82032cSYan Zheng 	write_extent_buffer(cow, root->fs_info->fsid,
2842b82032cSYan Zheng 			    (unsigned long)btrfs_header_fsid(cow),
2852b82032cSYan Zheng 			    BTRFS_FSID_SIZE);
2862b82032cSYan Zheng 
287be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
2885d4f98a2SYan Zheng 	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
28966d7e7f0SArne Jansen 		ret = btrfs_inc_ref(trans, root, cow, 1, 1);
2905d4f98a2SYan Zheng 	else
29166d7e7f0SArne Jansen 		ret = btrfs_inc_ref(trans, root, cow, 0, 1);
2924aec2b52SChris Mason 
293be20aa9dSChris Mason 	if (ret)
294be20aa9dSChris Mason 		return ret;
295be20aa9dSChris Mason 
296be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
297be20aa9dSChris Mason 	*cow_ret = cow;
298be20aa9dSChris Mason 	return 0;
299be20aa9dSChris Mason }
300be20aa9dSChris Mason 
301bd989ba3SJan Schmidt enum mod_log_op {
302bd989ba3SJan Schmidt 	MOD_LOG_KEY_REPLACE,
303bd989ba3SJan Schmidt 	MOD_LOG_KEY_ADD,
304bd989ba3SJan Schmidt 	MOD_LOG_KEY_REMOVE,
305bd989ba3SJan Schmidt 	MOD_LOG_KEY_REMOVE_WHILE_FREEING,
306bd989ba3SJan Schmidt 	MOD_LOG_KEY_REMOVE_WHILE_MOVING,
307bd989ba3SJan Schmidt 	MOD_LOG_MOVE_KEYS,
308bd989ba3SJan Schmidt 	MOD_LOG_ROOT_REPLACE,
309bd989ba3SJan Schmidt };
310bd989ba3SJan Schmidt 
311bd989ba3SJan Schmidt struct tree_mod_move {
312bd989ba3SJan Schmidt 	int dst_slot;
313bd989ba3SJan Schmidt 	int nr_items;
314bd989ba3SJan Schmidt };
315bd989ba3SJan Schmidt 
316bd989ba3SJan Schmidt struct tree_mod_root {
317bd989ba3SJan Schmidt 	u64 logical;
318bd989ba3SJan Schmidt 	u8 level;
319bd989ba3SJan Schmidt };
320bd989ba3SJan Schmidt 
321bd989ba3SJan Schmidt struct tree_mod_elem {
322bd989ba3SJan Schmidt 	struct rb_node node;
323bd989ba3SJan Schmidt 	u64 index;		/* shifted logical */
324097b8a7cSJan Schmidt 	u64 seq;
325bd989ba3SJan Schmidt 	enum mod_log_op op;
326bd989ba3SJan Schmidt 
327bd989ba3SJan Schmidt 	/* this is used for MOD_LOG_KEY_* and MOD_LOG_MOVE_KEYS operations */
328bd989ba3SJan Schmidt 	int slot;
329bd989ba3SJan Schmidt 
330bd989ba3SJan Schmidt 	/* this is used for MOD_LOG_KEY* and MOD_LOG_ROOT_REPLACE */
331bd989ba3SJan Schmidt 	u64 generation;
332bd989ba3SJan Schmidt 
333bd989ba3SJan Schmidt 	/* those are used for op == MOD_LOG_KEY_{REPLACE,REMOVE} */
334bd989ba3SJan Schmidt 	struct btrfs_disk_key key;
335bd989ba3SJan Schmidt 	u64 blockptr;
336bd989ba3SJan Schmidt 
337bd989ba3SJan Schmidt 	/* this is used for op == MOD_LOG_MOVE_KEYS */
338bd989ba3SJan Schmidt 	struct tree_mod_move move;
339bd989ba3SJan Schmidt 
340bd989ba3SJan Schmidt 	/* this is used for op == MOD_LOG_ROOT_REPLACE */
341bd989ba3SJan Schmidt 	struct tree_mod_root old_root;
342bd989ba3SJan Schmidt };
343bd989ba3SJan Schmidt 
344097b8a7cSJan Schmidt static inline void tree_mod_log_read_lock(struct btrfs_fs_info *fs_info)
345bd989ba3SJan Schmidt {
346097b8a7cSJan Schmidt 	read_lock(&fs_info->tree_mod_log_lock);
347bd989ba3SJan Schmidt }
348bd989ba3SJan Schmidt 
349097b8a7cSJan Schmidt static inline void tree_mod_log_read_unlock(struct btrfs_fs_info *fs_info)
350097b8a7cSJan Schmidt {
351097b8a7cSJan Schmidt 	read_unlock(&fs_info->tree_mod_log_lock);
352097b8a7cSJan Schmidt }
353097b8a7cSJan Schmidt 
354097b8a7cSJan Schmidt static inline void tree_mod_log_write_lock(struct btrfs_fs_info *fs_info)
355097b8a7cSJan Schmidt {
356097b8a7cSJan Schmidt 	write_lock(&fs_info->tree_mod_log_lock);
357097b8a7cSJan Schmidt }
358097b8a7cSJan Schmidt 
359097b8a7cSJan Schmidt static inline void tree_mod_log_write_unlock(struct btrfs_fs_info *fs_info)
360097b8a7cSJan Schmidt {
361097b8a7cSJan Schmidt 	write_unlock(&fs_info->tree_mod_log_lock);
362097b8a7cSJan Schmidt }
363097b8a7cSJan Schmidt 
364097b8a7cSJan Schmidt /*
365097b8a7cSJan Schmidt  * This adds a new blocker to the tree mod log's blocker list if the @elem
366097b8a7cSJan Schmidt  * passed does not already have a sequence number set. So when a caller expects
367097b8a7cSJan Schmidt  * to record tree modifications, it should ensure to set elem->seq to zero
368097b8a7cSJan Schmidt  * before calling btrfs_get_tree_mod_seq.
369097b8a7cSJan Schmidt  * Returns a fresh, unused tree log modification sequence number, even if no new
370097b8a7cSJan Schmidt  * blocker was added.
371097b8a7cSJan Schmidt  */
372097b8a7cSJan Schmidt u64 btrfs_get_tree_mod_seq(struct btrfs_fs_info *fs_info,
373bd989ba3SJan Schmidt 			   struct seq_list *elem)
374bd989ba3SJan Schmidt {
375097b8a7cSJan Schmidt 	u64 seq;
376097b8a7cSJan Schmidt 
377097b8a7cSJan Schmidt 	tree_mod_log_write_lock(fs_info);
378bd989ba3SJan Schmidt 	spin_lock(&fs_info->tree_mod_seq_lock);
379097b8a7cSJan Schmidt 	if (!elem->seq) {
380097b8a7cSJan Schmidt 		elem->seq = btrfs_inc_tree_mod_seq(fs_info);
381097b8a7cSJan Schmidt 		list_add_tail(&elem->list, &fs_info->tree_mod_seq_list);
382097b8a7cSJan Schmidt 	}
383097b8a7cSJan Schmidt 	seq = btrfs_inc_tree_mod_seq(fs_info);
384bd989ba3SJan Schmidt 	spin_unlock(&fs_info->tree_mod_seq_lock);
385097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
386097b8a7cSJan Schmidt 
387097b8a7cSJan Schmidt 	return seq;
388bd989ba3SJan Schmidt }
389bd989ba3SJan Schmidt 
390bd989ba3SJan Schmidt void btrfs_put_tree_mod_seq(struct btrfs_fs_info *fs_info,
391bd989ba3SJan Schmidt 			    struct seq_list *elem)
392bd989ba3SJan Schmidt {
393bd989ba3SJan Schmidt 	struct rb_root *tm_root;
394bd989ba3SJan Schmidt 	struct rb_node *node;
395bd989ba3SJan Schmidt 	struct rb_node *next;
396bd989ba3SJan Schmidt 	struct seq_list *cur_elem;
397bd989ba3SJan Schmidt 	struct tree_mod_elem *tm;
398bd989ba3SJan Schmidt 	u64 min_seq = (u64)-1;
399bd989ba3SJan Schmidt 	u64 seq_putting = elem->seq;
400bd989ba3SJan Schmidt 
401bd989ba3SJan Schmidt 	if (!seq_putting)
402bd989ba3SJan Schmidt 		return;
403bd989ba3SJan Schmidt 
404bd989ba3SJan Schmidt 	spin_lock(&fs_info->tree_mod_seq_lock);
405bd989ba3SJan Schmidt 	list_del(&elem->list);
406097b8a7cSJan Schmidt 	elem->seq = 0;
407bd989ba3SJan Schmidt 
408bd989ba3SJan Schmidt 	list_for_each_entry(cur_elem, &fs_info->tree_mod_seq_list, list) {
409097b8a7cSJan Schmidt 		if (cur_elem->seq < min_seq) {
410bd989ba3SJan Schmidt 			if (seq_putting > cur_elem->seq) {
411bd989ba3SJan Schmidt 				/*
412bd989ba3SJan Schmidt 				 * blocker with lower sequence number exists, we
413bd989ba3SJan Schmidt 				 * cannot remove anything from the log
414bd989ba3SJan Schmidt 				 */
415097b8a7cSJan Schmidt 				spin_unlock(&fs_info->tree_mod_seq_lock);
416097b8a7cSJan Schmidt 				return;
417bd989ba3SJan Schmidt 			}
418bd989ba3SJan Schmidt 			min_seq = cur_elem->seq;
419bd989ba3SJan Schmidt 		}
420bd989ba3SJan Schmidt 	}
421097b8a7cSJan Schmidt 	spin_unlock(&fs_info->tree_mod_seq_lock);
422097b8a7cSJan Schmidt 
423097b8a7cSJan Schmidt 	/*
424bd989ba3SJan Schmidt 	 * anything that's lower than the lowest existing (read: blocked)
425bd989ba3SJan Schmidt 	 * sequence number can be removed from the tree.
426bd989ba3SJan Schmidt 	 */
427097b8a7cSJan Schmidt 	tree_mod_log_write_lock(fs_info);
428bd989ba3SJan Schmidt 	tm_root = &fs_info->tree_mod_log;
429bd989ba3SJan Schmidt 	for (node = rb_first(tm_root); node; node = next) {
430bd989ba3SJan Schmidt 		next = rb_next(node);
431bd989ba3SJan Schmidt 		tm = container_of(node, struct tree_mod_elem, node);
432097b8a7cSJan Schmidt 		if (tm->seq > min_seq)
433bd989ba3SJan Schmidt 			continue;
434bd989ba3SJan Schmidt 		rb_erase(node, tm_root);
435bd989ba3SJan Schmidt 		kfree(tm);
436bd989ba3SJan Schmidt 	}
437097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
438bd989ba3SJan Schmidt }
439bd989ba3SJan Schmidt 
440bd989ba3SJan Schmidt /*
441bd989ba3SJan Schmidt  * key order of the log:
442bd989ba3SJan Schmidt  *       index -> sequence
443bd989ba3SJan Schmidt  *
444bd989ba3SJan Schmidt  * the index is the shifted logical of the *new* root node for root replace
445bd989ba3SJan Schmidt  * operations, or the shifted logical of the affected block for all other
446bd989ba3SJan Schmidt  * operations.
447bd989ba3SJan Schmidt  */
448bd989ba3SJan Schmidt static noinline int
449bd989ba3SJan Schmidt __tree_mod_log_insert(struct btrfs_fs_info *fs_info, struct tree_mod_elem *tm)
450bd989ba3SJan Schmidt {
451bd989ba3SJan Schmidt 	struct rb_root *tm_root;
452bd989ba3SJan Schmidt 	struct rb_node **new;
453bd989ba3SJan Schmidt 	struct rb_node *parent = NULL;
454bd989ba3SJan Schmidt 	struct tree_mod_elem *cur;
455bd989ba3SJan Schmidt 
456097b8a7cSJan Schmidt 	BUG_ON(!tm || !tm->seq);
457bd989ba3SJan Schmidt 
458bd989ba3SJan Schmidt 	tm_root = &fs_info->tree_mod_log;
459bd989ba3SJan Schmidt 	new = &tm_root->rb_node;
460bd989ba3SJan Schmidt 	while (*new) {
461bd989ba3SJan Schmidt 		cur = container_of(*new, struct tree_mod_elem, node);
462bd989ba3SJan Schmidt 		parent = *new;
463bd989ba3SJan Schmidt 		if (cur->index < tm->index)
464bd989ba3SJan Schmidt 			new = &((*new)->rb_left);
465bd989ba3SJan Schmidt 		else if (cur->index > tm->index)
466bd989ba3SJan Schmidt 			new = &((*new)->rb_right);
467097b8a7cSJan Schmidt 		else if (cur->seq < tm->seq)
468bd989ba3SJan Schmidt 			new = &((*new)->rb_left);
469097b8a7cSJan Schmidt 		else if (cur->seq > tm->seq)
470bd989ba3SJan Schmidt 			new = &((*new)->rb_right);
471bd989ba3SJan Schmidt 		else {
472bd989ba3SJan Schmidt 			kfree(tm);
473097b8a7cSJan Schmidt 			return -EEXIST;
474bd989ba3SJan Schmidt 		}
475bd989ba3SJan Schmidt 	}
476bd989ba3SJan Schmidt 
477bd989ba3SJan Schmidt 	rb_link_node(&tm->node, parent, new);
478bd989ba3SJan Schmidt 	rb_insert_color(&tm->node, tm_root);
479097b8a7cSJan Schmidt 	return 0;
480bd989ba3SJan Schmidt }
481bd989ba3SJan Schmidt 
482097b8a7cSJan Schmidt /*
483097b8a7cSJan Schmidt  * Determines if logging can be omitted. Returns 1 if it can. Otherwise, it
484097b8a7cSJan Schmidt  * returns zero with the tree_mod_log_lock acquired. The caller must hold
485097b8a7cSJan Schmidt  * this until all tree mod log insertions are recorded in the rb tree and then
486097b8a7cSJan Schmidt  * call tree_mod_log_write_unlock() to release.
487097b8a7cSJan Schmidt  */
488e9b7fd4dSJan Schmidt static inline int tree_mod_dont_log(struct btrfs_fs_info *fs_info,
489e9b7fd4dSJan Schmidt 				    struct extent_buffer *eb) {
490e9b7fd4dSJan Schmidt 	smp_mb();
491e9b7fd4dSJan Schmidt 	if (list_empty(&(fs_info)->tree_mod_seq_list))
492e9b7fd4dSJan Schmidt 		return 1;
493097b8a7cSJan Schmidt 	if (eb && btrfs_header_level(eb) == 0)
494e9b7fd4dSJan Schmidt 		return 1;
495097b8a7cSJan Schmidt 
496097b8a7cSJan Schmidt 	tree_mod_log_write_lock(fs_info);
497097b8a7cSJan Schmidt 	if (list_empty(&fs_info->tree_mod_seq_list)) {
498097b8a7cSJan Schmidt 		/*
499097b8a7cSJan Schmidt 		 * someone emptied the list while we were waiting for the lock.
500097b8a7cSJan Schmidt 		 * we must not add to the list when no blocker exists.
501097b8a7cSJan Schmidt 		 */
502097b8a7cSJan Schmidt 		tree_mod_log_write_unlock(fs_info);
503097b8a7cSJan Schmidt 		return 1;
504097b8a7cSJan Schmidt 	}
505097b8a7cSJan Schmidt 
506e9b7fd4dSJan Schmidt 	return 0;
507e9b7fd4dSJan Schmidt }
508e9b7fd4dSJan Schmidt 
5093310c36eSJan Schmidt /*
510097b8a7cSJan Schmidt  * This allocates memory and gets a tree modification sequence number.
5113310c36eSJan Schmidt  *
512097b8a7cSJan Schmidt  * Returns <0 on error.
513097b8a7cSJan Schmidt  * Returns >0 (the added sequence number) on success.
5143310c36eSJan Schmidt  */
515926dd8a6SJan Schmidt static inline int tree_mod_alloc(struct btrfs_fs_info *fs_info, gfp_t flags,
516bd989ba3SJan Schmidt 				 struct tree_mod_elem **tm_ret)
517bd989ba3SJan Schmidt {
518bd989ba3SJan Schmidt 	struct tree_mod_elem *tm;
519bd989ba3SJan Schmidt 
520097b8a7cSJan Schmidt 	/*
521097b8a7cSJan Schmidt 	 * once we switch from spin locks to something different, we should
522097b8a7cSJan Schmidt 	 * honor the flags parameter here.
523097b8a7cSJan Schmidt 	 */
524097b8a7cSJan Schmidt 	tm = *tm_ret = kzalloc(sizeof(*tm), GFP_ATOMIC);
525bd989ba3SJan Schmidt 	if (!tm)
526bd989ba3SJan Schmidt 		return -ENOMEM;
527bd989ba3SJan Schmidt 
528097b8a7cSJan Schmidt 	tm->seq = btrfs_inc_tree_mod_seq(fs_info);
529097b8a7cSJan Schmidt 	return tm->seq;
530926dd8a6SJan Schmidt }
531bd989ba3SJan Schmidt 
532097b8a7cSJan Schmidt static inline int
533097b8a7cSJan Schmidt __tree_mod_log_insert_key(struct btrfs_fs_info *fs_info,
534bd989ba3SJan Schmidt 			  struct extent_buffer *eb, int slot,
535bd989ba3SJan Schmidt 			  enum mod_log_op op, gfp_t flags)
536bd989ba3SJan Schmidt {
537bd989ba3SJan Schmidt 	int ret;
538097b8a7cSJan Schmidt 	struct tree_mod_elem *tm;
539bd989ba3SJan Schmidt 
540bd989ba3SJan Schmidt 	ret = tree_mod_alloc(fs_info, flags, &tm);
541097b8a7cSJan Schmidt 	if (ret < 0)
542bd989ba3SJan Schmidt 		return ret;
543bd989ba3SJan Schmidt 
544bd989ba3SJan Schmidt 	tm->index = eb->start >> PAGE_CACHE_SHIFT;
545bd989ba3SJan Schmidt 	if (op != MOD_LOG_KEY_ADD) {
546bd989ba3SJan Schmidt 		btrfs_node_key(eb, &tm->key, slot);
547bd989ba3SJan Schmidt 		tm->blockptr = btrfs_node_blockptr(eb, slot);
548bd989ba3SJan Schmidt 	}
549bd989ba3SJan Schmidt 	tm->op = op;
550bd989ba3SJan Schmidt 	tm->slot = slot;
551bd989ba3SJan Schmidt 	tm->generation = btrfs_node_ptr_generation(eb, slot);
552bd989ba3SJan Schmidt 
553097b8a7cSJan Schmidt 	return __tree_mod_log_insert(fs_info, tm);
554097b8a7cSJan Schmidt }
555097b8a7cSJan Schmidt 
556097b8a7cSJan Schmidt static noinline int
557097b8a7cSJan Schmidt tree_mod_log_insert_key_mask(struct btrfs_fs_info *fs_info,
558097b8a7cSJan Schmidt 			     struct extent_buffer *eb, int slot,
559097b8a7cSJan Schmidt 			     enum mod_log_op op, gfp_t flags)
560097b8a7cSJan Schmidt {
561097b8a7cSJan Schmidt 	int ret;
562097b8a7cSJan Schmidt 
563097b8a7cSJan Schmidt 	if (tree_mod_dont_log(fs_info, eb))
564097b8a7cSJan Schmidt 		return 0;
565097b8a7cSJan Schmidt 
566097b8a7cSJan Schmidt 	ret = __tree_mod_log_insert_key(fs_info, eb, slot, op, flags);
567097b8a7cSJan Schmidt 
568097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
5693310c36eSJan Schmidt 	return ret;
570bd989ba3SJan Schmidt }
571bd989ba3SJan Schmidt 
572bd989ba3SJan Schmidt static noinline int
573bd989ba3SJan Schmidt tree_mod_log_insert_key(struct btrfs_fs_info *fs_info, struct extent_buffer *eb,
574bd989ba3SJan Schmidt 			int slot, enum mod_log_op op)
575bd989ba3SJan Schmidt {
576bd989ba3SJan Schmidt 	return tree_mod_log_insert_key_mask(fs_info, eb, slot, op, GFP_NOFS);
577bd989ba3SJan Schmidt }
578bd989ba3SJan Schmidt 
579bd989ba3SJan Schmidt static noinline int
580097b8a7cSJan Schmidt tree_mod_log_insert_key_locked(struct btrfs_fs_info *fs_info,
581097b8a7cSJan Schmidt 			     struct extent_buffer *eb, int slot,
582097b8a7cSJan Schmidt 			     enum mod_log_op op)
583097b8a7cSJan Schmidt {
584097b8a7cSJan Schmidt 	return __tree_mod_log_insert_key(fs_info, eb, slot, op, GFP_NOFS);
585097b8a7cSJan Schmidt }
586097b8a7cSJan Schmidt 
587097b8a7cSJan Schmidt static noinline int
588bd989ba3SJan Schmidt tree_mod_log_insert_move(struct btrfs_fs_info *fs_info,
589bd989ba3SJan Schmidt 			 struct extent_buffer *eb, int dst_slot, int src_slot,
590bd989ba3SJan Schmidt 			 int nr_items, gfp_t flags)
591bd989ba3SJan Schmidt {
592bd989ba3SJan Schmidt 	struct tree_mod_elem *tm;
593bd989ba3SJan Schmidt 	int ret;
594bd989ba3SJan Schmidt 	int i;
595bd989ba3SJan Schmidt 
596f395694cSJan Schmidt 	if (tree_mod_dont_log(fs_info, eb))
597f395694cSJan Schmidt 		return 0;
598bd989ba3SJan Schmidt 
59901763a2eSJan Schmidt 	/*
60001763a2eSJan Schmidt 	 * When we override something during the move, we log these removals.
60101763a2eSJan Schmidt 	 * This can only happen when we move towards the beginning of the
60201763a2eSJan Schmidt 	 * buffer, i.e. dst_slot < src_slot.
60301763a2eSJan Schmidt 	 */
604bd989ba3SJan Schmidt 	for (i = 0; i + dst_slot < src_slot && i < nr_items; i++) {
605097b8a7cSJan Schmidt 		ret = tree_mod_log_insert_key_locked(fs_info, eb, i + dst_slot,
606bd989ba3SJan Schmidt 					      MOD_LOG_KEY_REMOVE_WHILE_MOVING);
607bd989ba3SJan Schmidt 		BUG_ON(ret < 0);
608bd989ba3SJan Schmidt 	}
609bd989ba3SJan Schmidt 
610f395694cSJan Schmidt 	ret = tree_mod_alloc(fs_info, flags, &tm);
611097b8a7cSJan Schmidt 	if (ret < 0)
612097b8a7cSJan Schmidt 		goto out;
613f395694cSJan Schmidt 
614bd989ba3SJan Schmidt 	tm->index = eb->start >> PAGE_CACHE_SHIFT;
615bd989ba3SJan Schmidt 	tm->slot = src_slot;
616bd989ba3SJan Schmidt 	tm->move.dst_slot = dst_slot;
617bd989ba3SJan Schmidt 	tm->move.nr_items = nr_items;
618bd989ba3SJan Schmidt 	tm->op = MOD_LOG_MOVE_KEYS;
619bd989ba3SJan Schmidt 
6203310c36eSJan Schmidt 	ret = __tree_mod_log_insert(fs_info, tm);
621097b8a7cSJan Schmidt out:
622097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
6233310c36eSJan Schmidt 	return ret;
624bd989ba3SJan Schmidt }
625bd989ba3SJan Schmidt 
626097b8a7cSJan Schmidt static inline void
627097b8a7cSJan Schmidt __tree_mod_log_free_eb(struct btrfs_fs_info *fs_info, struct extent_buffer *eb)
628097b8a7cSJan Schmidt {
629097b8a7cSJan Schmidt 	int i;
630097b8a7cSJan Schmidt 	u32 nritems;
631097b8a7cSJan Schmidt 	int ret;
632097b8a7cSJan Schmidt 
633b12a3b1eSChris Mason 	if (btrfs_header_level(eb) == 0)
634b12a3b1eSChris Mason 		return;
635b12a3b1eSChris Mason 
636097b8a7cSJan Schmidt 	nritems = btrfs_header_nritems(eb);
637097b8a7cSJan Schmidt 	for (i = nritems - 1; i >= 0; i--) {
638097b8a7cSJan Schmidt 		ret = tree_mod_log_insert_key_locked(fs_info, eb, i,
639097b8a7cSJan Schmidt 					      MOD_LOG_KEY_REMOVE_WHILE_FREEING);
640097b8a7cSJan Schmidt 		BUG_ON(ret < 0);
641097b8a7cSJan Schmidt 	}
642097b8a7cSJan Schmidt }
643097b8a7cSJan Schmidt 
644bd989ba3SJan Schmidt static noinline int
645bd989ba3SJan Schmidt tree_mod_log_insert_root(struct btrfs_fs_info *fs_info,
646bd989ba3SJan Schmidt 			 struct extent_buffer *old_root,
647bd989ba3SJan Schmidt 			 struct extent_buffer *new_root, gfp_t flags)
648bd989ba3SJan Schmidt {
649bd989ba3SJan Schmidt 	struct tree_mod_elem *tm;
650bd989ba3SJan Schmidt 	int ret;
651bd989ba3SJan Schmidt 
652097b8a7cSJan Schmidt 	if (tree_mod_dont_log(fs_info, NULL))
653097b8a7cSJan Schmidt 		return 0;
654097b8a7cSJan Schmidt 
655bd989ba3SJan Schmidt 	ret = tree_mod_alloc(fs_info, flags, &tm);
656097b8a7cSJan Schmidt 	if (ret < 0)
657097b8a7cSJan Schmidt 		goto out;
658bd989ba3SJan Schmidt 
659bd989ba3SJan Schmidt 	tm->index = new_root->start >> PAGE_CACHE_SHIFT;
660bd989ba3SJan Schmidt 	tm->old_root.logical = old_root->start;
661bd989ba3SJan Schmidt 	tm->old_root.level = btrfs_header_level(old_root);
662bd989ba3SJan Schmidt 	tm->generation = btrfs_header_generation(old_root);
663bd989ba3SJan Schmidt 	tm->op = MOD_LOG_ROOT_REPLACE;
664bd989ba3SJan Schmidt 
6653310c36eSJan Schmidt 	ret = __tree_mod_log_insert(fs_info, tm);
666097b8a7cSJan Schmidt out:
667097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
6683310c36eSJan Schmidt 	return ret;
669bd989ba3SJan Schmidt }
670bd989ba3SJan Schmidt 
671bd989ba3SJan Schmidt static struct tree_mod_elem *
672bd989ba3SJan Schmidt __tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq,
673bd989ba3SJan Schmidt 		      int smallest)
674bd989ba3SJan Schmidt {
675bd989ba3SJan Schmidt 	struct rb_root *tm_root;
676bd989ba3SJan Schmidt 	struct rb_node *node;
677bd989ba3SJan Schmidt 	struct tree_mod_elem *cur = NULL;
678bd989ba3SJan Schmidt 	struct tree_mod_elem *found = NULL;
679bd989ba3SJan Schmidt 	u64 index = start >> PAGE_CACHE_SHIFT;
680bd989ba3SJan Schmidt 
681097b8a7cSJan Schmidt 	tree_mod_log_read_lock(fs_info);
682bd989ba3SJan Schmidt 	tm_root = &fs_info->tree_mod_log;
683bd989ba3SJan Schmidt 	node = tm_root->rb_node;
684bd989ba3SJan Schmidt 	while (node) {
685bd989ba3SJan Schmidt 		cur = container_of(node, struct tree_mod_elem, node);
686bd989ba3SJan Schmidt 		if (cur->index < index) {
687bd989ba3SJan Schmidt 			node = node->rb_left;
688bd989ba3SJan Schmidt 		} else if (cur->index > index) {
689bd989ba3SJan Schmidt 			node = node->rb_right;
690097b8a7cSJan Schmidt 		} else if (cur->seq < min_seq) {
691bd989ba3SJan Schmidt 			node = node->rb_left;
692bd989ba3SJan Schmidt 		} else if (!smallest) {
693bd989ba3SJan Schmidt 			/* we want the node with the highest seq */
694bd989ba3SJan Schmidt 			if (found)
695097b8a7cSJan Schmidt 				BUG_ON(found->seq > cur->seq);
696bd989ba3SJan Schmidt 			found = cur;
697bd989ba3SJan Schmidt 			node = node->rb_left;
698097b8a7cSJan Schmidt 		} else if (cur->seq > min_seq) {
699bd989ba3SJan Schmidt 			/* we want the node with the smallest seq */
700bd989ba3SJan Schmidt 			if (found)
701097b8a7cSJan Schmidt 				BUG_ON(found->seq < cur->seq);
702bd989ba3SJan Schmidt 			found = cur;
703bd989ba3SJan Schmidt 			node = node->rb_right;
704bd989ba3SJan Schmidt 		} else {
705bd989ba3SJan Schmidt 			found = cur;
706bd989ba3SJan Schmidt 			break;
707bd989ba3SJan Schmidt 		}
708bd989ba3SJan Schmidt 	}
709097b8a7cSJan Schmidt 	tree_mod_log_read_unlock(fs_info);
710bd989ba3SJan Schmidt 
711bd989ba3SJan Schmidt 	return found;
712bd989ba3SJan Schmidt }
713bd989ba3SJan Schmidt 
714bd989ba3SJan Schmidt /*
715bd989ba3SJan Schmidt  * this returns the element from the log with the smallest time sequence
716bd989ba3SJan Schmidt  * value that's in the log (the oldest log item). any element with a time
717bd989ba3SJan Schmidt  * sequence lower than min_seq will be ignored.
718bd989ba3SJan Schmidt  */
719bd989ba3SJan Schmidt static struct tree_mod_elem *
720bd989ba3SJan Schmidt tree_mod_log_search_oldest(struct btrfs_fs_info *fs_info, u64 start,
721bd989ba3SJan Schmidt 			   u64 min_seq)
722bd989ba3SJan Schmidt {
723bd989ba3SJan Schmidt 	return __tree_mod_log_search(fs_info, start, min_seq, 1);
724bd989ba3SJan Schmidt }
725bd989ba3SJan Schmidt 
726bd989ba3SJan Schmidt /*
727bd989ba3SJan Schmidt  * this returns the element from the log with the largest time sequence
728bd989ba3SJan Schmidt  * value that's in the log (the most recent log item). any element with
729bd989ba3SJan Schmidt  * a time sequence lower than min_seq will be ignored.
730bd989ba3SJan Schmidt  */
731bd989ba3SJan Schmidt static struct tree_mod_elem *
732bd989ba3SJan Schmidt tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq)
733bd989ba3SJan Schmidt {
734bd989ba3SJan Schmidt 	return __tree_mod_log_search(fs_info, start, min_seq, 0);
735bd989ba3SJan Schmidt }
736bd989ba3SJan Schmidt 
737097b8a7cSJan Schmidt static noinline void
738bd989ba3SJan Schmidt tree_mod_log_eb_copy(struct btrfs_fs_info *fs_info, struct extent_buffer *dst,
739bd989ba3SJan Schmidt 		     struct extent_buffer *src, unsigned long dst_offset,
740bd989ba3SJan Schmidt 		     unsigned long src_offset, int nr_items)
741bd989ba3SJan Schmidt {
742bd989ba3SJan Schmidt 	int ret;
743bd989ba3SJan Schmidt 	int i;
744bd989ba3SJan Schmidt 
745e9b7fd4dSJan Schmidt 	if (tree_mod_dont_log(fs_info, NULL))
746bd989ba3SJan Schmidt 		return;
747bd989ba3SJan Schmidt 
748097b8a7cSJan Schmidt 	if (btrfs_header_level(dst) == 0 && btrfs_header_level(src) == 0) {
749097b8a7cSJan Schmidt 		tree_mod_log_write_unlock(fs_info);
750bd989ba3SJan Schmidt 		return;
751097b8a7cSJan Schmidt 	}
752bd989ba3SJan Schmidt 
753bd989ba3SJan Schmidt 	for (i = 0; i < nr_items; i++) {
754097b8a7cSJan Schmidt 		ret = tree_mod_log_insert_key_locked(fs_info, src,
755097b8a7cSJan Schmidt 						     i + src_offset,
756bd989ba3SJan Schmidt 						     MOD_LOG_KEY_REMOVE);
757bd989ba3SJan Schmidt 		BUG_ON(ret < 0);
758097b8a7cSJan Schmidt 		ret = tree_mod_log_insert_key_locked(fs_info, dst,
759097b8a7cSJan Schmidt 						     i + dst_offset,
760bd989ba3SJan Schmidt 						     MOD_LOG_KEY_ADD);
761bd989ba3SJan Schmidt 		BUG_ON(ret < 0);
762bd989ba3SJan Schmidt 	}
763097b8a7cSJan Schmidt 
764097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
765bd989ba3SJan Schmidt }
766bd989ba3SJan Schmidt 
767bd989ba3SJan Schmidt static inline void
768bd989ba3SJan Schmidt tree_mod_log_eb_move(struct btrfs_fs_info *fs_info, struct extent_buffer *dst,
769bd989ba3SJan Schmidt 		     int dst_offset, int src_offset, int nr_items)
770bd989ba3SJan Schmidt {
771bd989ba3SJan Schmidt 	int ret;
772bd989ba3SJan Schmidt 	ret = tree_mod_log_insert_move(fs_info, dst, dst_offset, src_offset,
773bd989ba3SJan Schmidt 				       nr_items, GFP_NOFS);
774bd989ba3SJan Schmidt 	BUG_ON(ret < 0);
775bd989ba3SJan Schmidt }
776bd989ba3SJan Schmidt 
777097b8a7cSJan Schmidt static noinline void
778bd989ba3SJan Schmidt tree_mod_log_set_node_key(struct btrfs_fs_info *fs_info,
779bd989ba3SJan Schmidt 			  struct extent_buffer *eb,
780bd989ba3SJan Schmidt 			  struct btrfs_disk_key *disk_key, int slot, int atomic)
781bd989ba3SJan Schmidt {
782bd989ba3SJan Schmidt 	int ret;
783bd989ba3SJan Schmidt 
784bd989ba3SJan Schmidt 	ret = tree_mod_log_insert_key_mask(fs_info, eb, slot,
785bd989ba3SJan Schmidt 					   MOD_LOG_KEY_REPLACE,
786bd989ba3SJan Schmidt 					   atomic ? GFP_ATOMIC : GFP_NOFS);
787bd989ba3SJan Schmidt 	BUG_ON(ret < 0);
788bd989ba3SJan Schmidt }
789bd989ba3SJan Schmidt 
790097b8a7cSJan Schmidt static noinline void
791097b8a7cSJan Schmidt tree_mod_log_free_eb(struct btrfs_fs_info *fs_info, struct extent_buffer *eb)
792bd989ba3SJan Schmidt {
793e9b7fd4dSJan Schmidt 	if (tree_mod_dont_log(fs_info, eb))
794bd989ba3SJan Schmidt 		return;
795bd989ba3SJan Schmidt 
796097b8a7cSJan Schmidt 	__tree_mod_log_free_eb(fs_info, eb);
797097b8a7cSJan Schmidt 
798097b8a7cSJan Schmidt 	tree_mod_log_write_unlock(fs_info);
799bd989ba3SJan Schmidt }
800bd989ba3SJan Schmidt 
801097b8a7cSJan Schmidt static noinline void
802bd989ba3SJan Schmidt tree_mod_log_set_root_pointer(struct btrfs_root *root,
803bd989ba3SJan Schmidt 			      struct extent_buffer *new_root_node)
804bd989ba3SJan Schmidt {
805bd989ba3SJan Schmidt 	int ret;
806bd989ba3SJan Schmidt 	ret = tree_mod_log_insert_root(root->fs_info, root->node,
807bd989ba3SJan Schmidt 				       new_root_node, GFP_NOFS);
808bd989ba3SJan Schmidt 	BUG_ON(ret < 0);
809bd989ba3SJan Schmidt }
810bd989ba3SJan Schmidt 
811d352ac68SChris Mason /*
8125d4f98a2SYan Zheng  * check if the tree block can be shared by multiple trees
8135d4f98a2SYan Zheng  */
8145d4f98a2SYan Zheng int btrfs_block_can_be_shared(struct btrfs_root *root,
8155d4f98a2SYan Zheng 			      struct extent_buffer *buf)
8165d4f98a2SYan Zheng {
8175d4f98a2SYan Zheng 	/*
8185d4f98a2SYan Zheng 	 * Tree blocks not in refernece counted trees and tree roots
8195d4f98a2SYan Zheng 	 * are never shared. If a block was allocated after the last
8205d4f98a2SYan Zheng 	 * snapshot and the block was not allocated by tree relocation,
8215d4f98a2SYan Zheng 	 * we know the block is not shared.
8225d4f98a2SYan Zheng 	 */
8235d4f98a2SYan Zheng 	if (root->ref_cows &&
8245d4f98a2SYan Zheng 	    buf != root->node && buf != root->commit_root &&
8255d4f98a2SYan Zheng 	    (btrfs_header_generation(buf) <=
8265d4f98a2SYan Zheng 	     btrfs_root_last_snapshot(&root->root_item) ||
8275d4f98a2SYan Zheng 	     btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)))
8285d4f98a2SYan Zheng 		return 1;
8295d4f98a2SYan Zheng #ifdef BTRFS_COMPAT_EXTENT_TREE_V0
8305d4f98a2SYan Zheng 	if (root->ref_cows &&
8315d4f98a2SYan Zheng 	    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
8325d4f98a2SYan Zheng 		return 1;
8335d4f98a2SYan Zheng #endif
8345d4f98a2SYan Zheng 	return 0;
8355d4f98a2SYan Zheng }
8365d4f98a2SYan Zheng 
8375d4f98a2SYan Zheng static noinline int update_ref_for_cow(struct btrfs_trans_handle *trans,
8385d4f98a2SYan Zheng 				       struct btrfs_root *root,
8395d4f98a2SYan Zheng 				       struct extent_buffer *buf,
840f0486c68SYan, Zheng 				       struct extent_buffer *cow,
841f0486c68SYan, Zheng 				       int *last_ref)
8425d4f98a2SYan Zheng {
8435d4f98a2SYan Zheng 	u64 refs;
8445d4f98a2SYan Zheng 	u64 owner;
8455d4f98a2SYan Zheng 	u64 flags;
8465d4f98a2SYan Zheng 	u64 new_flags = 0;
8475d4f98a2SYan Zheng 	int ret;
8485d4f98a2SYan Zheng 
8495d4f98a2SYan Zheng 	/*
8505d4f98a2SYan Zheng 	 * Backrefs update rules:
8515d4f98a2SYan Zheng 	 *
8525d4f98a2SYan Zheng 	 * Always use full backrefs for extent pointers in tree block
8535d4f98a2SYan Zheng 	 * allocated by tree relocation.
8545d4f98a2SYan Zheng 	 *
8555d4f98a2SYan Zheng 	 * If a shared tree block is no longer referenced by its owner
8565d4f98a2SYan Zheng 	 * tree (btrfs_header_owner(buf) == root->root_key.objectid),
8575d4f98a2SYan Zheng 	 * use full backrefs for extent pointers in tree block.
8585d4f98a2SYan Zheng 	 *
8595d4f98a2SYan Zheng 	 * If a tree block is been relocating
8605d4f98a2SYan Zheng 	 * (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID),
8615d4f98a2SYan Zheng 	 * use full backrefs for extent pointers in tree block.
8625d4f98a2SYan Zheng 	 * The reason for this is some operations (such as drop tree)
8635d4f98a2SYan Zheng 	 * are only allowed for blocks use full backrefs.
8645d4f98a2SYan Zheng 	 */
8655d4f98a2SYan Zheng 
8665d4f98a2SYan Zheng 	if (btrfs_block_can_be_shared(root, buf)) {
8675d4f98a2SYan Zheng 		ret = btrfs_lookup_extent_info(trans, root, buf->start,
8685d4f98a2SYan Zheng 					       buf->len, &refs, &flags);
869be1a5564SMark Fasheh 		if (ret)
870be1a5564SMark Fasheh 			return ret;
871e5df9573SMark Fasheh 		if (refs == 0) {
872e5df9573SMark Fasheh 			ret = -EROFS;
873e5df9573SMark Fasheh 			btrfs_std_error(root->fs_info, ret);
874e5df9573SMark Fasheh 			return ret;
875e5df9573SMark Fasheh 		}
8765d4f98a2SYan Zheng 	} else {
8775d4f98a2SYan Zheng 		refs = 1;
8785d4f98a2SYan Zheng 		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
8795d4f98a2SYan Zheng 		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
8805d4f98a2SYan Zheng 			flags = BTRFS_BLOCK_FLAG_FULL_BACKREF;
8815d4f98a2SYan Zheng 		else
8825d4f98a2SYan Zheng 			flags = 0;
8835d4f98a2SYan Zheng 	}
8845d4f98a2SYan Zheng 
8855d4f98a2SYan Zheng 	owner = btrfs_header_owner(buf);
8865d4f98a2SYan Zheng 	BUG_ON(owner == BTRFS_TREE_RELOC_OBJECTID &&
8875d4f98a2SYan Zheng 	       !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF));
8885d4f98a2SYan Zheng 
8895d4f98a2SYan Zheng 	if (refs > 1) {
8905d4f98a2SYan Zheng 		if ((owner == root->root_key.objectid ||
8915d4f98a2SYan Zheng 		     root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) &&
8925d4f98a2SYan Zheng 		    !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF)) {
89366d7e7f0SArne Jansen 			ret = btrfs_inc_ref(trans, root, buf, 1, 1);
89479787eaaSJeff Mahoney 			BUG_ON(ret); /* -ENOMEM */
8955d4f98a2SYan Zheng 
8965d4f98a2SYan Zheng 			if (root->root_key.objectid ==
8975d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID) {
89866d7e7f0SArne Jansen 				ret = btrfs_dec_ref(trans, root, buf, 0, 1);
89979787eaaSJeff Mahoney 				BUG_ON(ret); /* -ENOMEM */
90066d7e7f0SArne Jansen 				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
90179787eaaSJeff Mahoney 				BUG_ON(ret); /* -ENOMEM */
9025d4f98a2SYan Zheng 			}
9035d4f98a2SYan Zheng 			new_flags |= BTRFS_BLOCK_FLAG_FULL_BACKREF;
9045d4f98a2SYan Zheng 		} else {
9055d4f98a2SYan Zheng 
9065d4f98a2SYan Zheng 			if (root->root_key.objectid ==
9075d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID)
90866d7e7f0SArne Jansen 				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
9095d4f98a2SYan Zheng 			else
91066d7e7f0SArne Jansen 				ret = btrfs_inc_ref(trans, root, cow, 0, 1);
91179787eaaSJeff Mahoney 			BUG_ON(ret); /* -ENOMEM */
9125d4f98a2SYan Zheng 		}
9135d4f98a2SYan Zheng 		if (new_flags != 0) {
9145d4f98a2SYan Zheng 			ret = btrfs_set_disk_extent_flags(trans, root,
9155d4f98a2SYan Zheng 							  buf->start,
9165d4f98a2SYan Zheng 							  buf->len,
9175d4f98a2SYan Zheng 							  new_flags, 0);
918be1a5564SMark Fasheh 			if (ret)
919be1a5564SMark Fasheh 				return ret;
9205d4f98a2SYan Zheng 		}
9215d4f98a2SYan Zheng 	} else {
9225d4f98a2SYan Zheng 		if (flags & BTRFS_BLOCK_FLAG_FULL_BACKREF) {
9235d4f98a2SYan Zheng 			if (root->root_key.objectid ==
9245d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID)
92566d7e7f0SArne Jansen 				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
9265d4f98a2SYan Zheng 			else
92766d7e7f0SArne Jansen 				ret = btrfs_inc_ref(trans, root, cow, 0, 1);
92879787eaaSJeff Mahoney 			BUG_ON(ret); /* -ENOMEM */
92966d7e7f0SArne Jansen 			ret = btrfs_dec_ref(trans, root, buf, 1, 1);
93079787eaaSJeff Mahoney 			BUG_ON(ret); /* -ENOMEM */
9315d4f98a2SYan Zheng 		}
932f230475eSJan Schmidt 		tree_mod_log_free_eb(root->fs_info, buf);
9335d4f98a2SYan Zheng 		clean_tree_block(trans, root, buf);
934f0486c68SYan, Zheng 		*last_ref = 1;
9355d4f98a2SYan Zheng 	}
9365d4f98a2SYan Zheng 	return 0;
9375d4f98a2SYan Zheng }
9385d4f98a2SYan Zheng 
9395d4f98a2SYan Zheng /*
940d397712bSChris Mason  * does the dirty work in cow of a single block.  The parent block (if
941d397712bSChris Mason  * supplied) is updated to point to the new cow copy.  The new buffer is marked
942d397712bSChris Mason  * dirty and returned locked.  If you modify the block it needs to be marked
943d397712bSChris Mason  * dirty again.
944d352ac68SChris Mason  *
945d352ac68SChris Mason  * search_start -- an allocation hint for the new block
946d352ac68SChris Mason  *
947d397712bSChris Mason  * empty_size -- a hint that you plan on doing more cow.  This is the size in
948d397712bSChris Mason  * bytes the allocator should try to find free next to the block it returns.
949d397712bSChris Mason  * This is just a hint and may be ignored by the allocator.
950d352ac68SChris Mason  */
951d397712bSChris Mason static noinline int __btrfs_cow_block(struct btrfs_trans_handle *trans,
9525f39d397SChris Mason 			     struct btrfs_root *root,
9535f39d397SChris Mason 			     struct extent_buffer *buf,
9545f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
9555f39d397SChris Mason 			     struct extent_buffer **cow_ret,
9569fa8cfe7SChris Mason 			     u64 search_start, u64 empty_size)
9576702ed49SChris Mason {
9585d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
9595f39d397SChris Mason 	struct extent_buffer *cow;
960be1a5564SMark Fasheh 	int level, ret;
961f0486c68SYan, Zheng 	int last_ref = 0;
962925baeddSChris Mason 	int unlock_orig = 0;
9635d4f98a2SYan Zheng 	u64 parent_start;
9646702ed49SChris Mason 
965925baeddSChris Mason 	if (*cow_ret == buf)
966925baeddSChris Mason 		unlock_orig = 1;
967925baeddSChris Mason 
968b9447ef8SChris Mason 	btrfs_assert_tree_locked(buf);
969925baeddSChris Mason 
9707bb86316SChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
9717bb86316SChris Mason 		root->fs_info->running_transaction->transid);
9726702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
9735f39d397SChris Mason 
9747bb86316SChris Mason 	level = btrfs_header_level(buf);
97531840ae1SZheng Yan 
9765d4f98a2SYan Zheng 	if (level == 0)
9775d4f98a2SYan Zheng 		btrfs_item_key(buf, &disk_key, 0);
9785d4f98a2SYan Zheng 	else
9795d4f98a2SYan Zheng 		btrfs_node_key(buf, &disk_key, 0);
9805d4f98a2SYan Zheng 
9815d4f98a2SYan Zheng 	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) {
9825d4f98a2SYan Zheng 		if (parent)
9835d4f98a2SYan Zheng 			parent_start = parent->start;
9845d4f98a2SYan Zheng 		else
9855d4f98a2SYan Zheng 			parent_start = 0;
9865d4f98a2SYan Zheng 	} else
9875d4f98a2SYan Zheng 		parent_start = 0;
9885d4f98a2SYan Zheng 
9895d4f98a2SYan Zheng 	cow = btrfs_alloc_free_block(trans, root, buf->len, parent_start,
9905d4f98a2SYan Zheng 				     root->root_key.objectid, &disk_key,
9915581a51aSJan Schmidt 				     level, search_start, empty_size);
9926702ed49SChris Mason 	if (IS_ERR(cow))
9936702ed49SChris Mason 		return PTR_ERR(cow);
9946702ed49SChris Mason 
995b4ce94deSChris Mason 	/* cow is set to blocking by btrfs_init_new_buffer */
996b4ce94deSChris Mason 
9975f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
998db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
9995f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
10005d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
10015d4f98a2SYan Zheng 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
10025d4f98a2SYan Zheng 				     BTRFS_HEADER_FLAG_RELOC);
10035d4f98a2SYan Zheng 	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID)
10045d4f98a2SYan Zheng 		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
10055d4f98a2SYan Zheng 	else
10065f39d397SChris Mason 		btrfs_set_header_owner(cow, root->root_key.objectid);
10076702ed49SChris Mason 
10082b82032cSYan Zheng 	write_extent_buffer(cow, root->fs_info->fsid,
10092b82032cSYan Zheng 			    (unsigned long)btrfs_header_fsid(cow),
10102b82032cSYan Zheng 			    BTRFS_FSID_SIZE);
10112b82032cSYan Zheng 
1012be1a5564SMark Fasheh 	ret = update_ref_for_cow(trans, root, buf, cow, &last_ref);
1013b68dc2a9SMark Fasheh 	if (ret) {
101479787eaaSJeff Mahoney 		btrfs_abort_transaction(trans, root, ret);
1015b68dc2a9SMark Fasheh 		return ret;
1016b68dc2a9SMark Fasheh 	}
10171a40e23bSZheng Yan 
10183fd0a558SYan, Zheng 	if (root->ref_cows)
10193fd0a558SYan, Zheng 		btrfs_reloc_cow_block(trans, root, buf, cow);
10203fd0a558SYan, Zheng 
10216702ed49SChris Mason 	if (buf == root->node) {
1022925baeddSChris Mason 		WARN_ON(parent && parent != buf);
10235d4f98a2SYan Zheng 		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
10245d4f98a2SYan Zheng 		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
10255d4f98a2SYan Zheng 			parent_start = buf->start;
10265d4f98a2SYan Zheng 		else
10275d4f98a2SYan Zheng 			parent_start = 0;
1028925baeddSChris Mason 
10295f39d397SChris Mason 		extent_buffer_get(cow);
1030f230475eSJan Schmidt 		tree_mod_log_set_root_pointer(root, cow);
1031240f62c8SChris Mason 		rcu_assign_pointer(root->node, cow);
1032925baeddSChris Mason 
1033f0486c68SYan, Zheng 		btrfs_free_tree_block(trans, root, buf, parent_start,
10345581a51aSJan Schmidt 				      last_ref);
10355f39d397SChris Mason 		free_extent_buffer(buf);
10360b86a832SChris Mason 		add_root_to_dirty_list(root);
10376702ed49SChris Mason 	} else {
10385d4f98a2SYan Zheng 		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID)
10395d4f98a2SYan Zheng 			parent_start = parent->start;
10405d4f98a2SYan Zheng 		else
10415d4f98a2SYan Zheng 			parent_start = 0;
10425d4f98a2SYan Zheng 
10435d4f98a2SYan Zheng 		WARN_ON(trans->transid != btrfs_header_generation(parent));
1044f230475eSJan Schmidt 		tree_mod_log_insert_key(root->fs_info, parent, parent_slot,
1045f230475eSJan Schmidt 					MOD_LOG_KEY_REPLACE);
10465f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
1047db94535dSChris Mason 					cow->start);
104874493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
104974493f7aSChris Mason 					      trans->transid);
10506702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
1051f0486c68SYan, Zheng 		btrfs_free_tree_block(trans, root, buf, parent_start,
10525581a51aSJan Schmidt 				      last_ref);
10536702ed49SChris Mason 	}
1054925baeddSChris Mason 	if (unlock_orig)
1055925baeddSChris Mason 		btrfs_tree_unlock(buf);
10563083ee2eSJosef Bacik 	free_extent_buffer_stale(buf);
10576702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
10586702ed49SChris Mason 	*cow_ret = cow;
10596702ed49SChris Mason 	return 0;
10606702ed49SChris Mason }
10616702ed49SChris Mason 
10625d9e75c4SJan Schmidt /*
10635d9e75c4SJan Schmidt  * returns the logical address of the oldest predecessor of the given root.
10645d9e75c4SJan Schmidt  * entries older than time_seq are ignored.
10655d9e75c4SJan Schmidt  */
10665d9e75c4SJan Schmidt static struct tree_mod_elem *
10675d9e75c4SJan Schmidt __tree_mod_log_oldest_root(struct btrfs_fs_info *fs_info,
10685d9e75c4SJan Schmidt 			   struct btrfs_root *root, u64 time_seq)
10695d9e75c4SJan Schmidt {
10705d9e75c4SJan Schmidt 	struct tree_mod_elem *tm;
10715d9e75c4SJan Schmidt 	struct tree_mod_elem *found = NULL;
10725d9e75c4SJan Schmidt 	u64 root_logical = root->node->start;
10735d9e75c4SJan Schmidt 	int looped = 0;
10745d9e75c4SJan Schmidt 
10755d9e75c4SJan Schmidt 	if (!time_seq)
10765d9e75c4SJan Schmidt 		return 0;
10775d9e75c4SJan Schmidt 
10785d9e75c4SJan Schmidt 	/*
10795d9e75c4SJan Schmidt 	 * the very last operation that's logged for a root is the replacement
10805d9e75c4SJan Schmidt 	 * operation (if it is replaced at all). this has the index of the *new*
10815d9e75c4SJan Schmidt 	 * root, making it the very first operation that's logged for this root.
10825d9e75c4SJan Schmidt 	 */
10835d9e75c4SJan Schmidt 	while (1) {
10845d9e75c4SJan Schmidt 		tm = tree_mod_log_search_oldest(fs_info, root_logical,
10855d9e75c4SJan Schmidt 						time_seq);
10865d9e75c4SJan Schmidt 		if (!looped && !tm)
10875d9e75c4SJan Schmidt 			return 0;
10885d9e75c4SJan Schmidt 		/*
108928da9fb4SJan Schmidt 		 * if there are no tree operation for the oldest root, we simply
109028da9fb4SJan Schmidt 		 * return it. this should only happen if that (old) root is at
109128da9fb4SJan Schmidt 		 * level 0.
10925d9e75c4SJan Schmidt 		 */
109328da9fb4SJan Schmidt 		if (!tm)
109428da9fb4SJan Schmidt 			break;
10955d9e75c4SJan Schmidt 
109628da9fb4SJan Schmidt 		/*
109728da9fb4SJan Schmidt 		 * if there's an operation that's not a root replacement, we
109828da9fb4SJan Schmidt 		 * found the oldest version of our root. normally, we'll find a
109928da9fb4SJan Schmidt 		 * MOD_LOG_KEY_REMOVE_WHILE_FREEING operation here.
110028da9fb4SJan Schmidt 		 */
11015d9e75c4SJan Schmidt 		if (tm->op != MOD_LOG_ROOT_REPLACE)
11025d9e75c4SJan Schmidt 			break;
11035d9e75c4SJan Schmidt 
11045d9e75c4SJan Schmidt 		found = tm;
11055d9e75c4SJan Schmidt 		root_logical = tm->old_root.logical;
11065d9e75c4SJan Schmidt 		BUG_ON(root_logical == root->node->start);
11075d9e75c4SJan Schmidt 		looped = 1;
11085d9e75c4SJan Schmidt 	}
11095d9e75c4SJan Schmidt 
1110a95236d9SJan Schmidt 	/* if there's no old root to return, return what we found instead */
1111a95236d9SJan Schmidt 	if (!found)
1112a95236d9SJan Schmidt 		found = tm;
1113a95236d9SJan Schmidt 
11145d9e75c4SJan Schmidt 	return found;
11155d9e75c4SJan Schmidt }
11165d9e75c4SJan Schmidt 
11175d9e75c4SJan Schmidt /*
11185d9e75c4SJan Schmidt  * tm is a pointer to the first operation to rewind within eb. then, all
11195d9e75c4SJan Schmidt  * previous operations will be rewinded (until we reach something older than
11205d9e75c4SJan Schmidt  * time_seq).
11215d9e75c4SJan Schmidt  */
11225d9e75c4SJan Schmidt static void
11235d9e75c4SJan Schmidt __tree_mod_log_rewind(struct extent_buffer *eb, u64 time_seq,
11245d9e75c4SJan Schmidt 		      struct tree_mod_elem *first_tm)
11255d9e75c4SJan Schmidt {
11265d9e75c4SJan Schmidt 	u32 n;
11275d9e75c4SJan Schmidt 	struct rb_node *next;
11285d9e75c4SJan Schmidt 	struct tree_mod_elem *tm = first_tm;
11295d9e75c4SJan Schmidt 	unsigned long o_dst;
11305d9e75c4SJan Schmidt 	unsigned long o_src;
11315d9e75c4SJan Schmidt 	unsigned long p_size = sizeof(struct btrfs_key_ptr);
11325d9e75c4SJan Schmidt 
11335d9e75c4SJan Schmidt 	n = btrfs_header_nritems(eb);
1134097b8a7cSJan Schmidt 	while (tm && tm->seq >= time_seq) {
11355d9e75c4SJan Schmidt 		/*
11365d9e75c4SJan Schmidt 		 * all the operations are recorded with the operator used for
11375d9e75c4SJan Schmidt 		 * the modification. as we're going backwards, we do the
11385d9e75c4SJan Schmidt 		 * opposite of each operation here.
11395d9e75c4SJan Schmidt 		 */
11405d9e75c4SJan Schmidt 		switch (tm->op) {
11415d9e75c4SJan Schmidt 		case MOD_LOG_KEY_REMOVE_WHILE_FREEING:
11425d9e75c4SJan Schmidt 			BUG_ON(tm->slot < n);
11435d9e75c4SJan Schmidt 		case MOD_LOG_KEY_REMOVE:
114495c80bb1SLiu Bo 			n++;
114595c80bb1SLiu Bo 		case MOD_LOG_KEY_REMOVE_WHILE_MOVING:
11465d9e75c4SJan Schmidt 			btrfs_set_node_key(eb, &tm->key, tm->slot);
11475d9e75c4SJan Schmidt 			btrfs_set_node_blockptr(eb, tm->slot, tm->blockptr);
11485d9e75c4SJan Schmidt 			btrfs_set_node_ptr_generation(eb, tm->slot,
11495d9e75c4SJan Schmidt 						      tm->generation);
11505d9e75c4SJan Schmidt 			break;
11515d9e75c4SJan Schmidt 		case MOD_LOG_KEY_REPLACE:
11525d9e75c4SJan Schmidt 			BUG_ON(tm->slot >= n);
11535d9e75c4SJan Schmidt 			btrfs_set_node_key(eb, &tm->key, tm->slot);
11545d9e75c4SJan Schmidt 			btrfs_set_node_blockptr(eb, tm->slot, tm->blockptr);
11555d9e75c4SJan Schmidt 			btrfs_set_node_ptr_generation(eb, tm->slot,
11565d9e75c4SJan Schmidt 						      tm->generation);
11575d9e75c4SJan Schmidt 			break;
11585d9e75c4SJan Schmidt 		case MOD_LOG_KEY_ADD:
115919956c7eSJan Schmidt 			/* if a move operation is needed it's in the log */
11605d9e75c4SJan Schmidt 			n--;
11615d9e75c4SJan Schmidt 			break;
11625d9e75c4SJan Schmidt 		case MOD_LOG_MOVE_KEYS:
1163c3193108SJan Schmidt 			o_dst = btrfs_node_key_ptr_offset(tm->slot);
1164c3193108SJan Schmidt 			o_src = btrfs_node_key_ptr_offset(tm->move.dst_slot);
1165c3193108SJan Schmidt 			memmove_extent_buffer(eb, o_dst, o_src,
11665d9e75c4SJan Schmidt 					      tm->move.nr_items * p_size);
11675d9e75c4SJan Schmidt 			break;
11685d9e75c4SJan Schmidt 		case MOD_LOG_ROOT_REPLACE:
11695d9e75c4SJan Schmidt 			/*
11705d9e75c4SJan Schmidt 			 * this operation is special. for roots, this must be
11715d9e75c4SJan Schmidt 			 * handled explicitly before rewinding.
11725d9e75c4SJan Schmidt 			 * for non-roots, this operation may exist if the node
11735d9e75c4SJan Schmidt 			 * was a root: root A -> child B; then A gets empty and
11745d9e75c4SJan Schmidt 			 * B is promoted to the new root. in the mod log, we'll
11755d9e75c4SJan Schmidt 			 * have a root-replace operation for B, a tree block
11765d9e75c4SJan Schmidt 			 * that is no root. we simply ignore that operation.
11775d9e75c4SJan Schmidt 			 */
11785d9e75c4SJan Schmidt 			break;
11795d9e75c4SJan Schmidt 		}
11805d9e75c4SJan Schmidt 		next = rb_next(&tm->node);
11815d9e75c4SJan Schmidt 		if (!next)
11825d9e75c4SJan Schmidt 			break;
11835d9e75c4SJan Schmidt 		tm = container_of(next, struct tree_mod_elem, node);
11845d9e75c4SJan Schmidt 		if (tm->index != first_tm->index)
11855d9e75c4SJan Schmidt 			break;
11865d9e75c4SJan Schmidt 	}
11875d9e75c4SJan Schmidt 	btrfs_set_header_nritems(eb, n);
11885d9e75c4SJan Schmidt }
11895d9e75c4SJan Schmidt 
11905d9e75c4SJan Schmidt static struct extent_buffer *
11915d9e75c4SJan Schmidt tree_mod_log_rewind(struct btrfs_fs_info *fs_info, struct extent_buffer *eb,
11925d9e75c4SJan Schmidt 		    u64 time_seq)
11935d9e75c4SJan Schmidt {
11945d9e75c4SJan Schmidt 	struct extent_buffer *eb_rewin;
11955d9e75c4SJan Schmidt 	struct tree_mod_elem *tm;
11965d9e75c4SJan Schmidt 
11975d9e75c4SJan Schmidt 	if (!time_seq)
11985d9e75c4SJan Schmidt 		return eb;
11995d9e75c4SJan Schmidt 
12005d9e75c4SJan Schmidt 	if (btrfs_header_level(eb) == 0)
12015d9e75c4SJan Schmidt 		return eb;
12025d9e75c4SJan Schmidt 
12035d9e75c4SJan Schmidt 	tm = tree_mod_log_search(fs_info, eb->start, time_seq);
12045d9e75c4SJan Schmidt 	if (!tm)
12055d9e75c4SJan Schmidt 		return eb;
12065d9e75c4SJan Schmidt 
12075d9e75c4SJan Schmidt 	if (tm->op == MOD_LOG_KEY_REMOVE_WHILE_FREEING) {
12085d9e75c4SJan Schmidt 		BUG_ON(tm->slot != 0);
12095d9e75c4SJan Schmidt 		eb_rewin = alloc_dummy_extent_buffer(eb->start,
12105d9e75c4SJan Schmidt 						fs_info->tree_root->nodesize);
12115d9e75c4SJan Schmidt 		BUG_ON(!eb_rewin);
12125d9e75c4SJan Schmidt 		btrfs_set_header_bytenr(eb_rewin, eb->start);
12135d9e75c4SJan Schmidt 		btrfs_set_header_backref_rev(eb_rewin,
12145d9e75c4SJan Schmidt 					     btrfs_header_backref_rev(eb));
12155d9e75c4SJan Schmidt 		btrfs_set_header_owner(eb_rewin, btrfs_header_owner(eb));
1216c3193108SJan Schmidt 		btrfs_set_header_level(eb_rewin, btrfs_header_level(eb));
12175d9e75c4SJan Schmidt 	} else {
12185d9e75c4SJan Schmidt 		eb_rewin = btrfs_clone_extent_buffer(eb);
12195d9e75c4SJan Schmidt 		BUG_ON(!eb_rewin);
12205d9e75c4SJan Schmidt 	}
12215d9e75c4SJan Schmidt 
12225d9e75c4SJan Schmidt 	extent_buffer_get(eb_rewin);
12235d9e75c4SJan Schmidt 	free_extent_buffer(eb);
12245d9e75c4SJan Schmidt 
12255d9e75c4SJan Schmidt 	__tree_mod_log_rewind(eb_rewin, time_seq, tm);
122657911b8bSJan Schmidt 	WARN_ON(btrfs_header_nritems(eb_rewin) >
122757911b8bSJan Schmidt 		BTRFS_NODEPTRS_PER_BLOCK(fs_info->fs_root));
12285d9e75c4SJan Schmidt 
12295d9e75c4SJan Schmidt 	return eb_rewin;
12305d9e75c4SJan Schmidt }
12315d9e75c4SJan Schmidt 
12328ba97a15SJan Schmidt /*
12338ba97a15SJan Schmidt  * get_old_root() rewinds the state of @root's root node to the given @time_seq
12348ba97a15SJan Schmidt  * value. If there are no changes, the current root->root_node is returned. If
12358ba97a15SJan Schmidt  * anything changed in between, there's a fresh buffer allocated on which the
12368ba97a15SJan Schmidt  * rewind operations are done. In any case, the returned buffer is read locked.
12378ba97a15SJan Schmidt  * Returns NULL on error (with no locks held).
12388ba97a15SJan Schmidt  */
12395d9e75c4SJan Schmidt static inline struct extent_buffer *
12405d9e75c4SJan Schmidt get_old_root(struct btrfs_root *root, u64 time_seq)
12415d9e75c4SJan Schmidt {
12425d9e75c4SJan Schmidt 	struct tree_mod_elem *tm;
12435d9e75c4SJan Schmidt 	struct extent_buffer *eb;
12447bfdcf7fSLiu Bo 	struct extent_buffer *old;
1245a95236d9SJan Schmidt 	struct tree_mod_root *old_root = NULL;
12464325edd0SChris Mason 	u64 old_generation = 0;
1247a95236d9SJan Schmidt 	u64 logical;
1248834328a8SJan Schmidt 	u32 blocksize;
12495d9e75c4SJan Schmidt 
12508ba97a15SJan Schmidt 	eb = btrfs_read_lock_root_node(root);
12515d9e75c4SJan Schmidt 	tm = __tree_mod_log_oldest_root(root->fs_info, root, time_seq);
12525d9e75c4SJan Schmidt 	if (!tm)
12535d9e75c4SJan Schmidt 		return root->node;
12545d9e75c4SJan Schmidt 
1255a95236d9SJan Schmidt 	if (tm->op == MOD_LOG_ROOT_REPLACE) {
12565d9e75c4SJan Schmidt 		old_root = &tm->old_root;
12575d9e75c4SJan Schmidt 		old_generation = tm->generation;
1258a95236d9SJan Schmidt 		logical = old_root->logical;
1259a95236d9SJan Schmidt 	} else {
1260a95236d9SJan Schmidt 		logical = root->node->start;
1261a95236d9SJan Schmidt 	}
12625d9e75c4SJan Schmidt 
1263a95236d9SJan Schmidt 	tm = tree_mod_log_search(root->fs_info, logical, time_seq);
1264834328a8SJan Schmidt 	if (old_root && tm && tm->op != MOD_LOG_KEY_REMOVE_WHILE_FREEING) {
1265834328a8SJan Schmidt 		btrfs_tree_read_unlock(root->node);
1266834328a8SJan Schmidt 		free_extent_buffer(root->node);
1267834328a8SJan Schmidt 		blocksize = btrfs_level_size(root, old_root->level);
12687bfdcf7fSLiu Bo 		old = read_tree_block(root, logical, blocksize, 0);
12697bfdcf7fSLiu Bo 		if (!old) {
1270834328a8SJan Schmidt 			pr_warn("btrfs: failed to read tree block %llu from get_old_root\n",
1271834328a8SJan Schmidt 				logical);
1272834328a8SJan Schmidt 			WARN_ON(1);
1273834328a8SJan Schmidt 		} else {
12747bfdcf7fSLiu Bo 			eb = btrfs_clone_extent_buffer(old);
12757bfdcf7fSLiu Bo 			free_extent_buffer(old);
1276834328a8SJan Schmidt 		}
1277834328a8SJan Schmidt 	} else if (old_root) {
1278834328a8SJan Schmidt 		btrfs_tree_read_unlock(root->node);
1279834328a8SJan Schmidt 		free_extent_buffer(root->node);
128028da9fb4SJan Schmidt 		eb = alloc_dummy_extent_buffer(logical, root->nodesize);
1281834328a8SJan Schmidt 	} else {
1282a95236d9SJan Schmidt 		eb = btrfs_clone_extent_buffer(root->node);
12838ba97a15SJan Schmidt 		btrfs_tree_read_unlock(root->node);
12848ba97a15SJan Schmidt 		free_extent_buffer(root->node);
1285834328a8SJan Schmidt 	}
1286834328a8SJan Schmidt 
12878ba97a15SJan Schmidt 	if (!eb)
12888ba97a15SJan Schmidt 		return NULL;
1289d6381084SJan Schmidt 	extent_buffer_get(eb);
12908ba97a15SJan Schmidt 	btrfs_tree_read_lock(eb);
1291a95236d9SJan Schmidt 	if (old_root) {
12925d9e75c4SJan Schmidt 		btrfs_set_header_bytenr(eb, eb->start);
12935d9e75c4SJan Schmidt 		btrfs_set_header_backref_rev(eb, BTRFS_MIXED_BACKREF_REV);
12945d9e75c4SJan Schmidt 		btrfs_set_header_owner(eb, root->root_key.objectid);
12955d9e75c4SJan Schmidt 		btrfs_set_header_level(eb, old_root->level);
12965d9e75c4SJan Schmidt 		btrfs_set_header_generation(eb, old_generation);
1297a95236d9SJan Schmidt 	}
129828da9fb4SJan Schmidt 	if (tm)
12995d9e75c4SJan Schmidt 		__tree_mod_log_rewind(eb, time_seq, tm);
130028da9fb4SJan Schmidt 	else
130128da9fb4SJan Schmidt 		WARN_ON(btrfs_header_level(eb) != 0);
130257911b8bSJan Schmidt 	WARN_ON(btrfs_header_nritems(eb) > BTRFS_NODEPTRS_PER_BLOCK(root));
13035d9e75c4SJan Schmidt 
13045d9e75c4SJan Schmidt 	return eb;
13055d9e75c4SJan Schmidt }
13065d9e75c4SJan Schmidt 
13075b6602e7SJan Schmidt int btrfs_old_root_level(struct btrfs_root *root, u64 time_seq)
13085b6602e7SJan Schmidt {
13095b6602e7SJan Schmidt 	struct tree_mod_elem *tm;
13105b6602e7SJan Schmidt 	int level;
13115b6602e7SJan Schmidt 
13125b6602e7SJan Schmidt 	tm = __tree_mod_log_oldest_root(root->fs_info, root, time_seq);
13135b6602e7SJan Schmidt 	if (tm && tm->op == MOD_LOG_ROOT_REPLACE) {
13145b6602e7SJan Schmidt 		level = tm->old_root.level;
13155b6602e7SJan Schmidt 	} else {
13165b6602e7SJan Schmidt 		rcu_read_lock();
13175b6602e7SJan Schmidt 		level = btrfs_header_level(root->node);
13185b6602e7SJan Schmidt 		rcu_read_unlock();
13195b6602e7SJan Schmidt 	}
13205b6602e7SJan Schmidt 
13215b6602e7SJan Schmidt 	return level;
13225b6602e7SJan Schmidt }
13235b6602e7SJan Schmidt 
13245d4f98a2SYan Zheng static inline int should_cow_block(struct btrfs_trans_handle *trans,
13255d4f98a2SYan Zheng 				   struct btrfs_root *root,
13265d4f98a2SYan Zheng 				   struct extent_buffer *buf)
13275d4f98a2SYan Zheng {
1328f1ebcc74SLiu Bo 	/* ensure we can see the force_cow */
1329f1ebcc74SLiu Bo 	smp_rmb();
1330f1ebcc74SLiu Bo 
1331f1ebcc74SLiu Bo 	/*
1332f1ebcc74SLiu Bo 	 * We do not need to cow a block if
1333f1ebcc74SLiu Bo 	 * 1) this block is not created or changed in this transaction;
1334f1ebcc74SLiu Bo 	 * 2) this block does not belong to TREE_RELOC tree;
1335f1ebcc74SLiu Bo 	 * 3) the root is not forced COW.
1336f1ebcc74SLiu Bo 	 *
1337f1ebcc74SLiu Bo 	 * What is forced COW:
1338f1ebcc74SLiu Bo 	 *    when we create snapshot during commiting the transaction,
1339f1ebcc74SLiu Bo 	 *    after we've finished coping src root, we must COW the shared
1340f1ebcc74SLiu Bo 	 *    block to ensure the metadata consistency.
1341f1ebcc74SLiu Bo 	 */
13425d4f98a2SYan Zheng 	if (btrfs_header_generation(buf) == trans->transid &&
13435d4f98a2SYan Zheng 	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN) &&
13445d4f98a2SYan Zheng 	    !(root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID &&
1345f1ebcc74SLiu Bo 	      btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)) &&
1346f1ebcc74SLiu Bo 	    !root->force_cow)
13475d4f98a2SYan Zheng 		return 0;
13485d4f98a2SYan Zheng 	return 1;
13495d4f98a2SYan Zheng }
13505d4f98a2SYan Zheng 
1351d352ac68SChris Mason /*
1352d352ac68SChris Mason  * cows a single block, see __btrfs_cow_block for the real work.
1353d352ac68SChris Mason  * This version of it has extra checks so that a block isn't cow'd more than
1354d352ac68SChris Mason  * once per transaction, as long as it hasn't been written yet
1355d352ac68SChris Mason  */
1356d397712bSChris Mason noinline int btrfs_cow_block(struct btrfs_trans_handle *trans,
13575f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
13585f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
13599fa8cfe7SChris Mason 		    struct extent_buffer **cow_ret)
136002217ed2SChris Mason {
13616702ed49SChris Mason 	u64 search_start;
1362f510cfecSChris Mason 	int ret;
1363dc17ff8fSChris Mason 
1364ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
1365d397712bSChris Mason 		printk(KERN_CRIT "trans %llu running %llu\n",
1366d397712bSChris Mason 		       (unsigned long long)trans->transid,
1367d397712bSChris Mason 		       (unsigned long long)
1368ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
1369ccd467d6SChris Mason 		WARN_ON(1);
1370ccd467d6SChris Mason 	}
1371ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
1372d397712bSChris Mason 		printk(KERN_CRIT "trans %llu running %llu\n",
1373d397712bSChris Mason 		       (unsigned long long)trans->transid,
1374d397712bSChris Mason 		       (unsigned long long)root->fs_info->generation);
1375ccd467d6SChris Mason 		WARN_ON(1);
1376ccd467d6SChris Mason 	}
1377dc17ff8fSChris Mason 
13785d4f98a2SYan Zheng 	if (!should_cow_block(trans, root, buf)) {
137902217ed2SChris Mason 		*cow_ret = buf;
138002217ed2SChris Mason 		return 0;
138102217ed2SChris Mason 	}
1382c487685dSChris Mason 
13830b86a832SChris Mason 	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
1384b4ce94deSChris Mason 
1385b4ce94deSChris Mason 	if (parent)
1386b4ce94deSChris Mason 		btrfs_set_lock_blocking(parent);
1387b4ce94deSChris Mason 	btrfs_set_lock_blocking(buf);
1388b4ce94deSChris Mason 
1389f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
13909fa8cfe7SChris Mason 				 parent_slot, cow_ret, search_start, 0);
13911abe9b8aSliubo 
13921abe9b8aSliubo 	trace_btrfs_cow_block(root, buf, *cow_ret);
13931abe9b8aSliubo 
1394f510cfecSChris Mason 	return ret;
13952c90e5d6SChris Mason }
13966702ed49SChris Mason 
1397d352ac68SChris Mason /*
1398d352ac68SChris Mason  * helper function for defrag to decide if two blocks pointed to by a
1399d352ac68SChris Mason  * node are actually close by
1400d352ac68SChris Mason  */
14016b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
14026702ed49SChris Mason {
14036b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
14046702ed49SChris Mason 		return 1;
14056b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
14066702ed49SChris Mason 		return 1;
140702217ed2SChris Mason 	return 0;
140802217ed2SChris Mason }
140902217ed2SChris Mason 
1410081e9573SChris Mason /*
1411081e9573SChris Mason  * compare two keys in a memcmp fashion
1412081e9573SChris Mason  */
1413081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
1414081e9573SChris Mason {
1415081e9573SChris Mason 	struct btrfs_key k1;
1416081e9573SChris Mason 
1417081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
1418081e9573SChris Mason 
141920736abaSDiego Calleja 	return btrfs_comp_cpu_keys(&k1, k2);
1420081e9573SChris Mason }
1421081e9573SChris Mason 
1422f3465ca4SJosef Bacik /*
1423f3465ca4SJosef Bacik  * same as comp_keys only with two btrfs_key's
1424f3465ca4SJosef Bacik  */
14255d4f98a2SYan Zheng int btrfs_comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2)
1426f3465ca4SJosef Bacik {
1427f3465ca4SJosef Bacik 	if (k1->objectid > k2->objectid)
1428f3465ca4SJosef Bacik 		return 1;
1429f3465ca4SJosef Bacik 	if (k1->objectid < k2->objectid)
1430f3465ca4SJosef Bacik 		return -1;
1431f3465ca4SJosef Bacik 	if (k1->type > k2->type)
1432f3465ca4SJosef Bacik 		return 1;
1433f3465ca4SJosef Bacik 	if (k1->type < k2->type)
1434f3465ca4SJosef Bacik 		return -1;
1435f3465ca4SJosef Bacik 	if (k1->offset > k2->offset)
1436f3465ca4SJosef Bacik 		return 1;
1437f3465ca4SJosef Bacik 	if (k1->offset < k2->offset)
1438f3465ca4SJosef Bacik 		return -1;
1439f3465ca4SJosef Bacik 	return 0;
1440f3465ca4SJosef Bacik }
1441081e9573SChris Mason 
1442d352ac68SChris Mason /*
1443d352ac68SChris Mason  * this is used by the defrag code to go through all the
1444d352ac68SChris Mason  * leaves pointed to by a node and reallocate them so that
1445d352ac68SChris Mason  * disk order is close to key order
1446d352ac68SChris Mason  */
14476702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
14485f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
1449a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
1450a6b6e75eSChris Mason 		       struct btrfs_key *progress)
14516702ed49SChris Mason {
14526b80053dSChris Mason 	struct extent_buffer *cur;
14536702ed49SChris Mason 	u64 blocknr;
1454ca7a79adSChris Mason 	u64 gen;
1455e9d0b13bSChris Mason 	u64 search_start = *last_ret;
1456e9d0b13bSChris Mason 	u64 last_block = 0;
14576702ed49SChris Mason 	u64 other;
14586702ed49SChris Mason 	u32 parent_nritems;
14596702ed49SChris Mason 	int end_slot;
14606702ed49SChris Mason 	int i;
14616702ed49SChris Mason 	int err = 0;
1462f2183bdeSChris Mason 	int parent_level;
14636b80053dSChris Mason 	int uptodate;
14646b80053dSChris Mason 	u32 blocksize;
1465081e9573SChris Mason 	int progress_passed = 0;
1466081e9573SChris Mason 	struct btrfs_disk_key disk_key;
14676702ed49SChris Mason 
14685708b959SChris Mason 	parent_level = btrfs_header_level(parent);
14695708b959SChris Mason 	if (cache_only && parent_level != 1)
14705708b959SChris Mason 		return 0;
14715708b959SChris Mason 
1472d397712bSChris Mason 	if (trans->transaction != root->fs_info->running_transaction)
14736702ed49SChris Mason 		WARN_ON(1);
1474d397712bSChris Mason 	if (trans->transid != root->fs_info->generation)
14756702ed49SChris Mason 		WARN_ON(1);
147686479a04SChris Mason 
14776b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
14786b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
14796702ed49SChris Mason 	end_slot = parent_nritems;
14806702ed49SChris Mason 
14816702ed49SChris Mason 	if (parent_nritems == 1)
14826702ed49SChris Mason 		return 0;
14836702ed49SChris Mason 
1484b4ce94deSChris Mason 	btrfs_set_lock_blocking(parent);
1485b4ce94deSChris Mason 
14866702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
14876702ed49SChris Mason 		int close = 1;
1488a6b6e75eSChris Mason 
1489081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
1490081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
1491081e9573SChris Mason 			continue;
1492081e9573SChris Mason 
1493081e9573SChris Mason 		progress_passed = 1;
14946b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
1495ca7a79adSChris Mason 		gen = btrfs_node_ptr_generation(parent, i);
1496e9d0b13bSChris Mason 		if (last_block == 0)
1497e9d0b13bSChris Mason 			last_block = blocknr;
14985708b959SChris Mason 
14996702ed49SChris Mason 		if (i > 0) {
15006b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
15016b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
15026702ed49SChris Mason 		}
15030ef3e66bSChris Mason 		if (!close && i < end_slot - 2) {
15046b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
15056b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
15066702ed49SChris Mason 		}
1507e9d0b13bSChris Mason 		if (close) {
1508e9d0b13bSChris Mason 			last_block = blocknr;
15096702ed49SChris Mason 			continue;
1510e9d0b13bSChris Mason 		}
15116702ed49SChris Mason 
15126b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
15136b80053dSChris Mason 		if (cur)
1514b9fab919SChris Mason 			uptodate = btrfs_buffer_uptodate(cur, gen, 0);
15156b80053dSChris Mason 		else
15166b80053dSChris Mason 			uptodate = 0;
15175708b959SChris Mason 		if (!cur || !uptodate) {
15186702ed49SChris Mason 			if (cache_only) {
15196b80053dSChris Mason 				free_extent_buffer(cur);
15206702ed49SChris Mason 				continue;
15216702ed49SChris Mason 			}
15226b80053dSChris Mason 			if (!cur) {
15236b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
1524ca7a79adSChris Mason 							 blocksize, gen);
152597d9a8a4STsutomu Itoh 				if (!cur)
152697d9a8a4STsutomu Itoh 					return -EIO;
15276b80053dSChris Mason 			} else if (!uptodate) {
1528018642a1STsutomu Itoh 				err = btrfs_read_buffer(cur, gen);
1529018642a1STsutomu Itoh 				if (err) {
1530018642a1STsutomu Itoh 					free_extent_buffer(cur);
1531018642a1STsutomu Itoh 					return err;
1532018642a1STsutomu Itoh 				}
15336702ed49SChris Mason 			}
1534f2183bdeSChris Mason 		}
1535e9d0b13bSChris Mason 		if (search_start == 0)
15366b80053dSChris Mason 			search_start = last_block;
1537e9d0b13bSChris Mason 
1538e7a84565SChris Mason 		btrfs_tree_lock(cur);
1539b4ce94deSChris Mason 		btrfs_set_lock_blocking(cur);
15406b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
1541e7a84565SChris Mason 					&cur, search_start,
15426b80053dSChris Mason 					min(16 * blocksize,
15439fa8cfe7SChris Mason 					    (end_slot - i) * blocksize));
1544252c38f0SYan 		if (err) {
1545e7a84565SChris Mason 			btrfs_tree_unlock(cur);
15466b80053dSChris Mason 			free_extent_buffer(cur);
15476702ed49SChris Mason 			break;
1548252c38f0SYan 		}
1549e7a84565SChris Mason 		search_start = cur->start;
1550e7a84565SChris Mason 		last_block = cur->start;
1551f2183bdeSChris Mason 		*last_ret = search_start;
1552e7a84565SChris Mason 		btrfs_tree_unlock(cur);
1553e7a84565SChris Mason 		free_extent_buffer(cur);
15546702ed49SChris Mason 	}
15556702ed49SChris Mason 	return err;
15566702ed49SChris Mason }
15576702ed49SChris Mason 
155874123bd7SChris Mason /*
155974123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
156074123bd7SChris Mason  * this returns the address of the start of the last item,
156174123bd7SChris Mason  * which is the stop of the leaf data stack
156274123bd7SChris Mason  */
1563123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
15645f39d397SChris Mason 					 struct extent_buffer *leaf)
1565be0e5c09SChris Mason {
15665f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
1567be0e5c09SChris Mason 	if (nr == 0)
1568123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
15695f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
1570be0e5c09SChris Mason }
1571be0e5c09SChris Mason 
1572aa5d6bedSChris Mason 
157374123bd7SChris Mason /*
15745f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
15755f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
15765f39d397SChris Mason  *
157774123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
157874123bd7SChris Mason  * the place where you would insert key if it is not found in
157974123bd7SChris Mason  * the array.
158074123bd7SChris Mason  *
158174123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
158274123bd7SChris Mason  */
1583e02119d5SChris Mason static noinline int generic_bin_search(struct extent_buffer *eb,
1584e02119d5SChris Mason 				       unsigned long p,
15855f39d397SChris Mason 				       int item_size, struct btrfs_key *key,
1586be0e5c09SChris Mason 				       int max, int *slot)
1587be0e5c09SChris Mason {
1588be0e5c09SChris Mason 	int low = 0;
1589be0e5c09SChris Mason 	int high = max;
1590be0e5c09SChris Mason 	int mid;
1591be0e5c09SChris Mason 	int ret;
1592479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
15935f39d397SChris Mason 	struct btrfs_disk_key unaligned;
15945f39d397SChris Mason 	unsigned long offset;
15955f39d397SChris Mason 	char *kaddr = NULL;
15965f39d397SChris Mason 	unsigned long map_start = 0;
15975f39d397SChris Mason 	unsigned long map_len = 0;
1598479965d6SChris Mason 	int err;
1599be0e5c09SChris Mason 
1600be0e5c09SChris Mason 	while (low < high) {
1601be0e5c09SChris Mason 		mid = (low + high) / 2;
16025f39d397SChris Mason 		offset = p + mid * item_size;
16035f39d397SChris Mason 
1604a6591715SChris Mason 		if (!kaddr || offset < map_start ||
16055f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
16065f39d397SChris Mason 		    map_start + map_len) {
1607934d375bSChris Mason 
1608934d375bSChris Mason 			err = map_private_extent_buffer(eb, offset,
1609479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
1610a6591715SChris Mason 						&kaddr, &map_start, &map_len);
16115f39d397SChris Mason 
1612479965d6SChris Mason 			if (!err) {
1613479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
1614479965d6SChris Mason 							map_start);
1615479965d6SChris Mason 			} else {
16165f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
16175f39d397SChris Mason 						   offset, sizeof(unaligned));
16185f39d397SChris Mason 				tmp = &unaligned;
1619479965d6SChris Mason 			}
1620479965d6SChris Mason 
16215f39d397SChris Mason 		} else {
16225f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
16235f39d397SChris Mason 							map_start);
16245f39d397SChris Mason 		}
1625be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
1626be0e5c09SChris Mason 
1627be0e5c09SChris Mason 		if (ret < 0)
1628be0e5c09SChris Mason 			low = mid + 1;
1629be0e5c09SChris Mason 		else if (ret > 0)
1630be0e5c09SChris Mason 			high = mid;
1631be0e5c09SChris Mason 		else {
1632be0e5c09SChris Mason 			*slot = mid;
1633be0e5c09SChris Mason 			return 0;
1634be0e5c09SChris Mason 		}
1635be0e5c09SChris Mason 	}
1636be0e5c09SChris Mason 	*slot = low;
1637be0e5c09SChris Mason 	return 1;
1638be0e5c09SChris Mason }
1639be0e5c09SChris Mason 
164097571fd0SChris Mason /*
164197571fd0SChris Mason  * simple bin_search frontend that does the right thing for
164297571fd0SChris Mason  * leaves vs nodes
164397571fd0SChris Mason  */
16445f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
16455f39d397SChris Mason 		      int level, int *slot)
1646be0e5c09SChris Mason {
1647f775738fSWang Sheng-Hui 	if (level == 0)
16485f39d397SChris Mason 		return generic_bin_search(eb,
16495f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
16500783fcfcSChris Mason 					  sizeof(struct btrfs_item),
16515f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
16527518a238SChris Mason 					  slot);
1653f775738fSWang Sheng-Hui 	else
16545f39d397SChris Mason 		return generic_bin_search(eb,
16555f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
1656123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
16575f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
16587518a238SChris Mason 					  slot);
1659be0e5c09SChris Mason }
1660be0e5c09SChris Mason 
16615d4f98a2SYan Zheng int btrfs_bin_search(struct extent_buffer *eb, struct btrfs_key *key,
16625d4f98a2SYan Zheng 		     int level, int *slot)
16635d4f98a2SYan Zheng {
16645d4f98a2SYan Zheng 	return bin_search(eb, key, level, slot);
16655d4f98a2SYan Zheng }
16665d4f98a2SYan Zheng 
1667f0486c68SYan, Zheng static void root_add_used(struct btrfs_root *root, u32 size)
1668f0486c68SYan, Zheng {
1669f0486c68SYan, Zheng 	spin_lock(&root->accounting_lock);
1670f0486c68SYan, Zheng 	btrfs_set_root_used(&root->root_item,
1671f0486c68SYan, Zheng 			    btrfs_root_used(&root->root_item) + size);
1672f0486c68SYan, Zheng 	spin_unlock(&root->accounting_lock);
1673f0486c68SYan, Zheng }
1674f0486c68SYan, Zheng 
1675f0486c68SYan, Zheng static void root_sub_used(struct btrfs_root *root, u32 size)
1676f0486c68SYan, Zheng {
1677f0486c68SYan, Zheng 	spin_lock(&root->accounting_lock);
1678f0486c68SYan, Zheng 	btrfs_set_root_used(&root->root_item,
1679f0486c68SYan, Zheng 			    btrfs_root_used(&root->root_item) - size);
1680f0486c68SYan, Zheng 	spin_unlock(&root->accounting_lock);
1681f0486c68SYan, Zheng }
1682f0486c68SYan, Zheng 
1683d352ac68SChris Mason /* given a node and slot number, this reads the blocks it points to.  The
1684d352ac68SChris Mason  * extent buffer is returned with a reference taken (but unlocked).
1685d352ac68SChris Mason  * NULL is returned on error.
1686d352ac68SChris Mason  */
1687e02119d5SChris Mason static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root,
16885f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
1689bb803951SChris Mason {
1690ca7a79adSChris Mason 	int level = btrfs_header_level(parent);
1691bb803951SChris Mason 	if (slot < 0)
1692bb803951SChris Mason 		return NULL;
16935f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
1694bb803951SChris Mason 		return NULL;
1695ca7a79adSChris Mason 
1696ca7a79adSChris Mason 	BUG_ON(level == 0);
1697ca7a79adSChris Mason 
1698db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
1699ca7a79adSChris Mason 		       btrfs_level_size(root, level - 1),
1700ca7a79adSChris Mason 		       btrfs_node_ptr_generation(parent, slot));
1701bb803951SChris Mason }
1702bb803951SChris Mason 
1703d352ac68SChris Mason /*
1704d352ac68SChris Mason  * node level balancing, used to make sure nodes are in proper order for
1705d352ac68SChris Mason  * item deletion.  We balance from the top down, so we have to make sure
1706d352ac68SChris Mason  * that a deletion won't leave an node completely empty later on.
1707d352ac68SChris Mason  */
1708e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans,
170998ed5174SChris Mason 			 struct btrfs_root *root,
171098ed5174SChris Mason 			 struct btrfs_path *path, int level)
1711bb803951SChris Mason {
17125f39d397SChris Mason 	struct extent_buffer *right = NULL;
17135f39d397SChris Mason 	struct extent_buffer *mid;
17145f39d397SChris Mason 	struct extent_buffer *left = NULL;
17155f39d397SChris Mason 	struct extent_buffer *parent = NULL;
1716bb803951SChris Mason 	int ret = 0;
1717bb803951SChris Mason 	int wret;
1718bb803951SChris Mason 	int pslot;
1719bb803951SChris Mason 	int orig_slot = path->slots[level];
172079f95c82SChris Mason 	u64 orig_ptr;
1721bb803951SChris Mason 
1722bb803951SChris Mason 	if (level == 0)
1723bb803951SChris Mason 		return 0;
1724bb803951SChris Mason 
17255f39d397SChris Mason 	mid = path->nodes[level];
1726b4ce94deSChris Mason 
1727bd681513SChris Mason 	WARN_ON(path->locks[level] != BTRFS_WRITE_LOCK &&
1728bd681513SChris Mason 		path->locks[level] != BTRFS_WRITE_LOCK_BLOCKING);
17297bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
17307bb86316SChris Mason 
17311d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
173279f95c82SChris Mason 
1733a05a9bb1SLi Zefan 	if (level < BTRFS_MAX_LEVEL - 1) {
17345f39d397SChris Mason 		parent = path->nodes[level + 1];
1735bb803951SChris Mason 		pslot = path->slots[level + 1];
1736a05a9bb1SLi Zefan 	}
1737bb803951SChris Mason 
173840689478SChris Mason 	/*
173940689478SChris Mason 	 * deal with the case where there is only one pointer in the root
174040689478SChris Mason 	 * by promoting the node below to a root
174140689478SChris Mason 	 */
17425f39d397SChris Mason 	if (!parent) {
17435f39d397SChris Mason 		struct extent_buffer *child;
1744bb803951SChris Mason 
17455f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
1746bb803951SChris Mason 			return 0;
1747bb803951SChris Mason 
1748bb803951SChris Mason 		/* promote the child to a root */
17495f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
1750305a26afSMark Fasheh 		if (!child) {
1751305a26afSMark Fasheh 			ret = -EROFS;
1752305a26afSMark Fasheh 			btrfs_std_error(root->fs_info, ret);
1753305a26afSMark Fasheh 			goto enospc;
1754305a26afSMark Fasheh 		}
1755305a26afSMark Fasheh 
1756925baeddSChris Mason 		btrfs_tree_lock(child);
1757b4ce94deSChris Mason 		btrfs_set_lock_blocking(child);
17589fa8cfe7SChris Mason 		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
1759f0486c68SYan, Zheng 		if (ret) {
1760f0486c68SYan, Zheng 			btrfs_tree_unlock(child);
1761f0486c68SYan, Zheng 			free_extent_buffer(child);
1762f0486c68SYan, Zheng 			goto enospc;
1763f0486c68SYan, Zheng 		}
17642f375ab9SYan 
1765ba1bfbd5SJan Schmidt 		tree_mod_log_free_eb(root->fs_info, root->node);
1766f230475eSJan Schmidt 		tree_mod_log_set_root_pointer(root, child);
1767240f62c8SChris Mason 		rcu_assign_pointer(root->node, child);
1768925baeddSChris Mason 
17690b86a832SChris Mason 		add_root_to_dirty_list(root);
1770925baeddSChris Mason 		btrfs_tree_unlock(child);
1771b4ce94deSChris Mason 
1772925baeddSChris Mason 		path->locks[level] = 0;
1773bb803951SChris Mason 		path->nodes[level] = NULL;
17745f39d397SChris Mason 		clean_tree_block(trans, root, mid);
1775925baeddSChris Mason 		btrfs_tree_unlock(mid);
1776bb803951SChris Mason 		/* once for the path */
17775f39d397SChris Mason 		free_extent_buffer(mid);
1778f0486c68SYan, Zheng 
1779f0486c68SYan, Zheng 		root_sub_used(root, mid->len);
17805581a51aSJan Schmidt 		btrfs_free_tree_block(trans, root, mid, 0, 1);
1781bb803951SChris Mason 		/* once for the root ptr */
17823083ee2eSJosef Bacik 		free_extent_buffer_stale(mid);
1783f0486c68SYan, Zheng 		return 0;
1784bb803951SChris Mason 	}
17855f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
1786123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
1787bb803951SChris Mason 		return 0;
1788bb803951SChris Mason 
17895f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
17905f39d397SChris Mason 	if (left) {
1791925baeddSChris Mason 		btrfs_tree_lock(left);
1792b4ce94deSChris Mason 		btrfs_set_lock_blocking(left);
17935f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
17949fa8cfe7SChris Mason 				       parent, pslot - 1, &left);
179554aa1f4dSChris Mason 		if (wret) {
179654aa1f4dSChris Mason 			ret = wret;
179754aa1f4dSChris Mason 			goto enospc;
179854aa1f4dSChris Mason 		}
17992cc58cf2SChris Mason 	}
18005f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
18015f39d397SChris Mason 	if (right) {
1802925baeddSChris Mason 		btrfs_tree_lock(right);
1803b4ce94deSChris Mason 		btrfs_set_lock_blocking(right);
18045f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
18059fa8cfe7SChris Mason 				       parent, pslot + 1, &right);
18062cc58cf2SChris Mason 		if (wret) {
18072cc58cf2SChris Mason 			ret = wret;
18082cc58cf2SChris Mason 			goto enospc;
18092cc58cf2SChris Mason 		}
18102cc58cf2SChris Mason 	}
18112cc58cf2SChris Mason 
18122cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
18135f39d397SChris Mason 	if (left) {
18145f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
1815bce4eae9SChris Mason 		wret = push_node_left(trans, root, left, mid, 1);
181679f95c82SChris Mason 		if (wret < 0)
181779f95c82SChris Mason 			ret = wret;
1818bb803951SChris Mason 	}
181979f95c82SChris Mason 
182079f95c82SChris Mason 	/*
182179f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
182279f95c82SChris Mason 	 */
18235f39d397SChris Mason 	if (right) {
1824971a1f66SChris Mason 		wret = push_node_left(trans, root, mid, right, 1);
182554aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
182679f95c82SChris Mason 			ret = wret;
18275f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
18285f39d397SChris Mason 			clean_tree_block(trans, root, right);
1829925baeddSChris Mason 			btrfs_tree_unlock(right);
1830f3ea38daSJan Schmidt 			del_ptr(trans, root, path, level + 1, pslot + 1, 1);
1831f0486c68SYan, Zheng 			root_sub_used(root, right->len);
18325581a51aSJan Schmidt 			btrfs_free_tree_block(trans, root, right, 0, 1);
18333083ee2eSJosef Bacik 			free_extent_buffer_stale(right);
1834f0486c68SYan, Zheng 			right = NULL;
1835bb803951SChris Mason 		} else {
18365f39d397SChris Mason 			struct btrfs_disk_key right_key;
18375f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
1838f230475eSJan Schmidt 			tree_mod_log_set_node_key(root->fs_info, parent,
1839f230475eSJan Schmidt 						  &right_key, pslot + 1, 0);
18405f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
18415f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
1842bb803951SChris Mason 		}
1843bb803951SChris Mason 	}
18445f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
184579f95c82SChris Mason 		/*
184679f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
184779f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
184879f95c82SChris Mason 		 * could try to delete the only pointer in this node.
184979f95c82SChris Mason 		 * So, pull some keys from the left.
185079f95c82SChris Mason 		 * There has to be a left pointer at this point because
185179f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
185279f95c82SChris Mason 		 * right
185379f95c82SChris Mason 		 */
1854305a26afSMark Fasheh 		if (!left) {
1855305a26afSMark Fasheh 			ret = -EROFS;
1856305a26afSMark Fasheh 			btrfs_std_error(root->fs_info, ret);
1857305a26afSMark Fasheh 			goto enospc;
1858305a26afSMark Fasheh 		}
18595f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
186054aa1f4dSChris Mason 		if (wret < 0) {
186179f95c82SChris Mason 			ret = wret;
186254aa1f4dSChris Mason 			goto enospc;
186354aa1f4dSChris Mason 		}
1864bce4eae9SChris Mason 		if (wret == 1) {
1865bce4eae9SChris Mason 			wret = push_node_left(trans, root, left, mid, 1);
1866bce4eae9SChris Mason 			if (wret < 0)
1867bce4eae9SChris Mason 				ret = wret;
1868bce4eae9SChris Mason 		}
186979f95c82SChris Mason 		BUG_ON(wret == 1);
187079f95c82SChris Mason 	}
18715f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
18725f39d397SChris Mason 		clean_tree_block(trans, root, mid);
1873925baeddSChris Mason 		btrfs_tree_unlock(mid);
1874f3ea38daSJan Schmidt 		del_ptr(trans, root, path, level + 1, pslot, 1);
1875f0486c68SYan, Zheng 		root_sub_used(root, mid->len);
18765581a51aSJan Schmidt 		btrfs_free_tree_block(trans, root, mid, 0, 1);
18773083ee2eSJosef Bacik 		free_extent_buffer_stale(mid);
1878f0486c68SYan, Zheng 		mid = NULL;
187979f95c82SChris Mason 	} else {
188079f95c82SChris Mason 		/* update the parent key to reflect our changes */
18815f39d397SChris Mason 		struct btrfs_disk_key mid_key;
18825f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
1883f230475eSJan Schmidt 		tree_mod_log_set_node_key(root->fs_info, parent, &mid_key,
1884f230475eSJan Schmidt 					  pslot, 0);
18855f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
18865f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
188779f95c82SChris Mason 	}
1888bb803951SChris Mason 
188979f95c82SChris Mason 	/* update the path */
18905f39d397SChris Mason 	if (left) {
18915f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
18925f39d397SChris Mason 			extent_buffer_get(left);
1893925baeddSChris Mason 			/* left was locked after cow */
18945f39d397SChris Mason 			path->nodes[level] = left;
1895bb803951SChris Mason 			path->slots[level + 1] -= 1;
1896bb803951SChris Mason 			path->slots[level] = orig_slot;
1897925baeddSChris Mason 			if (mid) {
1898925baeddSChris Mason 				btrfs_tree_unlock(mid);
18995f39d397SChris Mason 				free_extent_buffer(mid);
1900925baeddSChris Mason 			}
1901bb803951SChris Mason 		} else {
19025f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
1903bb803951SChris Mason 			path->slots[level] = orig_slot;
1904bb803951SChris Mason 		}
1905bb803951SChris Mason 	}
190679f95c82SChris Mason 	/* double check we haven't messed things up */
1907e20d96d6SChris Mason 	if (orig_ptr !=
19085f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
190979f95c82SChris Mason 		BUG();
191054aa1f4dSChris Mason enospc:
1911925baeddSChris Mason 	if (right) {
1912925baeddSChris Mason 		btrfs_tree_unlock(right);
19135f39d397SChris Mason 		free_extent_buffer(right);
1914925baeddSChris Mason 	}
1915925baeddSChris Mason 	if (left) {
1916925baeddSChris Mason 		if (path->nodes[level] != left)
1917925baeddSChris Mason 			btrfs_tree_unlock(left);
19185f39d397SChris Mason 		free_extent_buffer(left);
1919925baeddSChris Mason 	}
1920bb803951SChris Mason 	return ret;
1921bb803951SChris Mason }
1922bb803951SChris Mason 
1923d352ac68SChris Mason /* Node balancing for insertion.  Here we only split or push nodes around
1924d352ac68SChris Mason  * when they are completely full.  This is also done top down, so we
1925d352ac68SChris Mason  * have to be pessimistic.
1926d352ac68SChris Mason  */
1927d397712bSChris Mason static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans,
1928e66f709bSChris Mason 					  struct btrfs_root *root,
1929e66f709bSChris Mason 					  struct btrfs_path *path, int level)
1930e66f709bSChris Mason {
19315f39d397SChris Mason 	struct extent_buffer *right = NULL;
19325f39d397SChris Mason 	struct extent_buffer *mid;
19335f39d397SChris Mason 	struct extent_buffer *left = NULL;
19345f39d397SChris Mason 	struct extent_buffer *parent = NULL;
1935e66f709bSChris Mason 	int ret = 0;
1936e66f709bSChris Mason 	int wret;
1937e66f709bSChris Mason 	int pslot;
1938e66f709bSChris Mason 	int orig_slot = path->slots[level];
1939e66f709bSChris Mason 
1940e66f709bSChris Mason 	if (level == 0)
1941e66f709bSChris Mason 		return 1;
1942e66f709bSChris Mason 
19435f39d397SChris Mason 	mid = path->nodes[level];
19447bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1945e66f709bSChris Mason 
1946a05a9bb1SLi Zefan 	if (level < BTRFS_MAX_LEVEL - 1) {
19475f39d397SChris Mason 		parent = path->nodes[level + 1];
1948e66f709bSChris Mason 		pslot = path->slots[level + 1];
1949a05a9bb1SLi Zefan 	}
1950e66f709bSChris Mason 
19515f39d397SChris Mason 	if (!parent)
1952e66f709bSChris Mason 		return 1;
1953e66f709bSChris Mason 
19545f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
1955e66f709bSChris Mason 
1956e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
19575f39d397SChris Mason 	if (left) {
1958e66f709bSChris Mason 		u32 left_nr;
1959925baeddSChris Mason 
1960925baeddSChris Mason 		btrfs_tree_lock(left);
1961b4ce94deSChris Mason 		btrfs_set_lock_blocking(left);
1962b4ce94deSChris Mason 
19635f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
196433ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
196533ade1f8SChris Mason 			wret = 1;
196633ade1f8SChris Mason 		} else {
19675f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
19689fa8cfe7SChris Mason 					      pslot - 1, &left);
196954aa1f4dSChris Mason 			if (ret)
197054aa1f4dSChris Mason 				wret = 1;
197154aa1f4dSChris Mason 			else {
197254aa1f4dSChris Mason 				wret = push_node_left(trans, root,
1973971a1f66SChris Mason 						      left, mid, 0);
197454aa1f4dSChris Mason 			}
197533ade1f8SChris Mason 		}
1976e66f709bSChris Mason 		if (wret < 0)
1977e66f709bSChris Mason 			ret = wret;
1978e66f709bSChris Mason 		if (wret == 0) {
19795f39d397SChris Mason 			struct btrfs_disk_key disk_key;
1980e66f709bSChris Mason 			orig_slot += left_nr;
19815f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
1982f230475eSJan Schmidt 			tree_mod_log_set_node_key(root->fs_info, parent,
1983f230475eSJan Schmidt 						  &disk_key, pslot, 0);
19845f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
19855f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
19865f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
19875f39d397SChris Mason 				path->nodes[level] = left;
1988e66f709bSChris Mason 				path->slots[level + 1] -= 1;
1989e66f709bSChris Mason 				path->slots[level] = orig_slot;
1990925baeddSChris Mason 				btrfs_tree_unlock(mid);
19915f39d397SChris Mason 				free_extent_buffer(mid);
1992e66f709bSChris Mason 			} else {
1993e66f709bSChris Mason 				orig_slot -=
19945f39d397SChris Mason 					btrfs_header_nritems(left);
1995e66f709bSChris Mason 				path->slots[level] = orig_slot;
1996925baeddSChris Mason 				btrfs_tree_unlock(left);
19975f39d397SChris Mason 				free_extent_buffer(left);
1998e66f709bSChris Mason 			}
1999e66f709bSChris Mason 			return 0;
2000e66f709bSChris Mason 		}
2001925baeddSChris Mason 		btrfs_tree_unlock(left);
20025f39d397SChris Mason 		free_extent_buffer(left);
2003e66f709bSChris Mason 	}
20045f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
2005e66f709bSChris Mason 
2006e66f709bSChris Mason 	/*
2007e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
2008e66f709bSChris Mason 	 */
20095f39d397SChris Mason 	if (right) {
201033ade1f8SChris Mason 		u32 right_nr;
2011b4ce94deSChris Mason 
2012925baeddSChris Mason 		btrfs_tree_lock(right);
2013b4ce94deSChris Mason 		btrfs_set_lock_blocking(right);
2014b4ce94deSChris Mason 
20155f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
201633ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
201733ade1f8SChris Mason 			wret = 1;
201833ade1f8SChris Mason 		} else {
20195f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
20205f39d397SChris Mason 					      parent, pslot + 1,
20219fa8cfe7SChris Mason 					      &right);
202254aa1f4dSChris Mason 			if (ret)
202354aa1f4dSChris Mason 				wret = 1;
202454aa1f4dSChris Mason 			else {
202533ade1f8SChris Mason 				wret = balance_node_right(trans, root,
20265f39d397SChris Mason 							  right, mid);
202733ade1f8SChris Mason 			}
202854aa1f4dSChris Mason 		}
2029e66f709bSChris Mason 		if (wret < 0)
2030e66f709bSChris Mason 			ret = wret;
2031e66f709bSChris Mason 		if (wret == 0) {
20325f39d397SChris Mason 			struct btrfs_disk_key disk_key;
20335f39d397SChris Mason 
20345f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
2035f230475eSJan Schmidt 			tree_mod_log_set_node_key(root->fs_info, parent,
2036f230475eSJan Schmidt 						  &disk_key, pslot + 1, 0);
20375f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
20385f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
20395f39d397SChris Mason 
20405f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
20415f39d397SChris Mason 				path->nodes[level] = right;
2042e66f709bSChris Mason 				path->slots[level + 1] += 1;
2043e66f709bSChris Mason 				path->slots[level] = orig_slot -
20445f39d397SChris Mason 					btrfs_header_nritems(mid);
2045925baeddSChris Mason 				btrfs_tree_unlock(mid);
20465f39d397SChris Mason 				free_extent_buffer(mid);
2047e66f709bSChris Mason 			} else {
2048925baeddSChris Mason 				btrfs_tree_unlock(right);
20495f39d397SChris Mason 				free_extent_buffer(right);
2050e66f709bSChris Mason 			}
2051e66f709bSChris Mason 			return 0;
2052e66f709bSChris Mason 		}
2053925baeddSChris Mason 		btrfs_tree_unlock(right);
20545f39d397SChris Mason 		free_extent_buffer(right);
2055e66f709bSChris Mason 	}
2056e66f709bSChris Mason 	return 1;
2057e66f709bSChris Mason }
2058e66f709bSChris Mason 
205974123bd7SChris Mason /*
2060d352ac68SChris Mason  * readahead one full node of leaves, finding things that are close
2061d352ac68SChris Mason  * to the block in 'slot', and triggering ra on them.
20623c69faecSChris Mason  */
2063c8c42864SChris Mason static void reada_for_search(struct btrfs_root *root,
2064e02119d5SChris Mason 			     struct btrfs_path *path,
206501f46658SChris Mason 			     int level, int slot, u64 objectid)
20663c69faecSChris Mason {
20675f39d397SChris Mason 	struct extent_buffer *node;
206801f46658SChris Mason 	struct btrfs_disk_key disk_key;
20693c69faecSChris Mason 	u32 nritems;
20703c69faecSChris Mason 	u64 search;
2071a7175319SChris Mason 	u64 target;
20726b80053dSChris Mason 	u64 nread = 0;
2073cb25c2eaSJosef Bacik 	u64 gen;
20743c69faecSChris Mason 	int direction = path->reada;
20755f39d397SChris Mason 	struct extent_buffer *eb;
20766b80053dSChris Mason 	u32 nr;
20776b80053dSChris Mason 	u32 blocksize;
20786b80053dSChris Mason 	u32 nscan = 0;
2079db94535dSChris Mason 
2080a6b6e75eSChris Mason 	if (level != 1)
20813c69faecSChris Mason 		return;
20823c69faecSChris Mason 
20836702ed49SChris Mason 	if (!path->nodes[level])
20846702ed49SChris Mason 		return;
20856702ed49SChris Mason 
20865f39d397SChris Mason 	node = path->nodes[level];
2087925baeddSChris Mason 
20883c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
20896b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
20906b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
20915f39d397SChris Mason 	if (eb) {
20925f39d397SChris Mason 		free_extent_buffer(eb);
20933c69faecSChris Mason 		return;
20943c69faecSChris Mason 	}
20953c69faecSChris Mason 
2096a7175319SChris Mason 	target = search;
20976b80053dSChris Mason 
20985f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
20996b80053dSChris Mason 	nr = slot;
210025b8b936SJosef Bacik 
21013c69faecSChris Mason 	while (1) {
21026b80053dSChris Mason 		if (direction < 0) {
21036b80053dSChris Mason 			if (nr == 0)
21043c69faecSChris Mason 				break;
21056b80053dSChris Mason 			nr--;
21066b80053dSChris Mason 		} else if (direction > 0) {
21076b80053dSChris Mason 			nr++;
21086b80053dSChris Mason 			if (nr >= nritems)
21096b80053dSChris Mason 				break;
21103c69faecSChris Mason 		}
211101f46658SChris Mason 		if (path->reada < 0 && objectid) {
211201f46658SChris Mason 			btrfs_node_key(node, &disk_key, nr);
211301f46658SChris Mason 			if (btrfs_disk_key_objectid(&disk_key) != objectid)
211401f46658SChris Mason 				break;
211501f46658SChris Mason 		}
21166b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
2117a7175319SChris Mason 		if ((search <= target && target - search <= 65536) ||
2118a7175319SChris Mason 		    (search > target && search - target <= 65536)) {
2119cb25c2eaSJosef Bacik 			gen = btrfs_node_ptr_generation(node, nr);
2120cb25c2eaSJosef Bacik 			readahead_tree_block(root, search, blocksize, gen);
21216b80053dSChris Mason 			nread += blocksize;
21223c69faecSChris Mason 		}
21236b80053dSChris Mason 		nscan++;
2124a7175319SChris Mason 		if ((nread > 65536 || nscan > 32))
21256b80053dSChris Mason 			break;
21263c69faecSChris Mason 	}
21273c69faecSChris Mason }
2128925baeddSChris Mason 
2129d352ac68SChris Mason /*
2130b4ce94deSChris Mason  * returns -EAGAIN if it had to drop the path, or zero if everything was in
2131b4ce94deSChris Mason  * cache
2132b4ce94deSChris Mason  */
2133b4ce94deSChris Mason static noinline int reada_for_balance(struct btrfs_root *root,
2134b4ce94deSChris Mason 				      struct btrfs_path *path, int level)
2135b4ce94deSChris Mason {
2136b4ce94deSChris Mason 	int slot;
2137b4ce94deSChris Mason 	int nritems;
2138b4ce94deSChris Mason 	struct extent_buffer *parent;
2139b4ce94deSChris Mason 	struct extent_buffer *eb;
2140b4ce94deSChris Mason 	u64 gen;
2141b4ce94deSChris Mason 	u64 block1 = 0;
2142b4ce94deSChris Mason 	u64 block2 = 0;
2143b4ce94deSChris Mason 	int ret = 0;
2144b4ce94deSChris Mason 	int blocksize;
2145b4ce94deSChris Mason 
21468c594ea8SChris Mason 	parent = path->nodes[level + 1];
2147b4ce94deSChris Mason 	if (!parent)
2148b4ce94deSChris Mason 		return 0;
2149b4ce94deSChris Mason 
2150b4ce94deSChris Mason 	nritems = btrfs_header_nritems(parent);
21518c594ea8SChris Mason 	slot = path->slots[level + 1];
2152b4ce94deSChris Mason 	blocksize = btrfs_level_size(root, level);
2153b4ce94deSChris Mason 
2154b4ce94deSChris Mason 	if (slot > 0) {
2155b4ce94deSChris Mason 		block1 = btrfs_node_blockptr(parent, slot - 1);
2156b4ce94deSChris Mason 		gen = btrfs_node_ptr_generation(parent, slot - 1);
2157b4ce94deSChris Mason 		eb = btrfs_find_tree_block(root, block1, blocksize);
2158b9fab919SChris Mason 		/*
2159b9fab919SChris Mason 		 * if we get -eagain from btrfs_buffer_uptodate, we
2160b9fab919SChris Mason 		 * don't want to return eagain here.  That will loop
2161b9fab919SChris Mason 		 * forever
2162b9fab919SChris Mason 		 */
2163b9fab919SChris Mason 		if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0)
2164b4ce94deSChris Mason 			block1 = 0;
2165b4ce94deSChris Mason 		free_extent_buffer(eb);
2166b4ce94deSChris Mason 	}
21678c594ea8SChris Mason 	if (slot + 1 < nritems) {
2168b4ce94deSChris Mason 		block2 = btrfs_node_blockptr(parent, slot + 1);
2169b4ce94deSChris Mason 		gen = btrfs_node_ptr_generation(parent, slot + 1);
2170b4ce94deSChris Mason 		eb = btrfs_find_tree_block(root, block2, blocksize);
2171b9fab919SChris Mason 		if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0)
2172b4ce94deSChris Mason 			block2 = 0;
2173b4ce94deSChris Mason 		free_extent_buffer(eb);
2174b4ce94deSChris Mason 	}
2175b4ce94deSChris Mason 	if (block1 || block2) {
2176b4ce94deSChris Mason 		ret = -EAGAIN;
21778c594ea8SChris Mason 
21788c594ea8SChris Mason 		/* release the whole path */
2179b3b4aa74SDavid Sterba 		btrfs_release_path(path);
21808c594ea8SChris Mason 
21818c594ea8SChris Mason 		/* read the blocks */
2182b4ce94deSChris Mason 		if (block1)
2183b4ce94deSChris Mason 			readahead_tree_block(root, block1, blocksize, 0);
2184b4ce94deSChris Mason 		if (block2)
2185b4ce94deSChris Mason 			readahead_tree_block(root, block2, blocksize, 0);
2186b4ce94deSChris Mason 
2187b4ce94deSChris Mason 		if (block1) {
2188b4ce94deSChris Mason 			eb = read_tree_block(root, block1, blocksize, 0);
2189b4ce94deSChris Mason 			free_extent_buffer(eb);
2190b4ce94deSChris Mason 		}
21918c594ea8SChris Mason 		if (block2) {
2192b4ce94deSChris Mason 			eb = read_tree_block(root, block2, blocksize, 0);
2193b4ce94deSChris Mason 			free_extent_buffer(eb);
2194b4ce94deSChris Mason 		}
2195b4ce94deSChris Mason 	}
2196b4ce94deSChris Mason 	return ret;
2197b4ce94deSChris Mason }
2198b4ce94deSChris Mason 
2199b4ce94deSChris Mason 
2200b4ce94deSChris Mason /*
2201d397712bSChris Mason  * when we walk down the tree, it is usually safe to unlock the higher layers
2202d397712bSChris Mason  * in the tree.  The exceptions are when our path goes through slot 0, because
2203d397712bSChris Mason  * operations on the tree might require changing key pointers higher up in the
2204d397712bSChris Mason  * tree.
2205d352ac68SChris Mason  *
2206d397712bSChris Mason  * callers might also have set path->keep_locks, which tells this code to keep
2207d397712bSChris Mason  * the lock if the path points to the last slot in the block.  This is part of
2208d397712bSChris Mason  * walking through the tree, and selecting the next slot in the higher block.
2209d352ac68SChris Mason  *
2210d397712bSChris Mason  * lowest_unlock sets the lowest level in the tree we're allowed to unlock.  so
2211d397712bSChris Mason  * if lowest_unlock is 1, level 0 won't be unlocked
2212d352ac68SChris Mason  */
2213e02119d5SChris Mason static noinline void unlock_up(struct btrfs_path *path, int level,
2214f7c79f30SChris Mason 			       int lowest_unlock, int min_write_lock_level,
2215f7c79f30SChris Mason 			       int *write_lock_level)
2216925baeddSChris Mason {
2217925baeddSChris Mason 	int i;
2218925baeddSChris Mason 	int skip_level = level;
2219051e1b9fSChris Mason 	int no_skips = 0;
2220925baeddSChris Mason 	struct extent_buffer *t;
2221925baeddSChris Mason 
2222925baeddSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2223925baeddSChris Mason 		if (!path->nodes[i])
2224925baeddSChris Mason 			break;
2225925baeddSChris Mason 		if (!path->locks[i])
2226925baeddSChris Mason 			break;
2227051e1b9fSChris Mason 		if (!no_skips && path->slots[i] == 0) {
2228925baeddSChris Mason 			skip_level = i + 1;
2229925baeddSChris Mason 			continue;
2230925baeddSChris Mason 		}
2231051e1b9fSChris Mason 		if (!no_skips && path->keep_locks) {
2232925baeddSChris Mason 			u32 nritems;
2233925baeddSChris Mason 			t = path->nodes[i];
2234925baeddSChris Mason 			nritems = btrfs_header_nritems(t);
2235051e1b9fSChris Mason 			if (nritems < 1 || path->slots[i] >= nritems - 1) {
2236925baeddSChris Mason 				skip_level = i + 1;
2237925baeddSChris Mason 				continue;
2238925baeddSChris Mason 			}
2239925baeddSChris Mason 		}
2240051e1b9fSChris Mason 		if (skip_level < i && i >= lowest_unlock)
2241051e1b9fSChris Mason 			no_skips = 1;
2242051e1b9fSChris Mason 
2243925baeddSChris Mason 		t = path->nodes[i];
2244925baeddSChris Mason 		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
2245bd681513SChris Mason 			btrfs_tree_unlock_rw(t, path->locks[i]);
2246925baeddSChris Mason 			path->locks[i] = 0;
2247f7c79f30SChris Mason 			if (write_lock_level &&
2248f7c79f30SChris Mason 			    i > min_write_lock_level &&
2249f7c79f30SChris Mason 			    i <= *write_lock_level) {
2250f7c79f30SChris Mason 				*write_lock_level = i - 1;
2251f7c79f30SChris Mason 			}
2252925baeddSChris Mason 		}
2253925baeddSChris Mason 	}
2254925baeddSChris Mason }
2255925baeddSChris Mason 
22563c69faecSChris Mason /*
2257b4ce94deSChris Mason  * This releases any locks held in the path starting at level and
2258b4ce94deSChris Mason  * going all the way up to the root.
2259b4ce94deSChris Mason  *
2260b4ce94deSChris Mason  * btrfs_search_slot will keep the lock held on higher nodes in a few
2261b4ce94deSChris Mason  * corner cases, such as COW of the block at slot zero in the node.  This
2262b4ce94deSChris Mason  * ignores those rules, and it should only be called when there are no
2263b4ce94deSChris Mason  * more updates to be done higher up in the tree.
2264b4ce94deSChris Mason  */
2265b4ce94deSChris Mason noinline void btrfs_unlock_up_safe(struct btrfs_path *path, int level)
2266b4ce94deSChris Mason {
2267b4ce94deSChris Mason 	int i;
2268b4ce94deSChris Mason 
22695d4f98a2SYan Zheng 	if (path->keep_locks)
2270b4ce94deSChris Mason 		return;
2271b4ce94deSChris Mason 
2272b4ce94deSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2273b4ce94deSChris Mason 		if (!path->nodes[i])
227412f4daccSChris Mason 			continue;
2275b4ce94deSChris Mason 		if (!path->locks[i])
227612f4daccSChris Mason 			continue;
2277bd681513SChris Mason 		btrfs_tree_unlock_rw(path->nodes[i], path->locks[i]);
2278b4ce94deSChris Mason 		path->locks[i] = 0;
2279b4ce94deSChris Mason 	}
2280b4ce94deSChris Mason }
2281b4ce94deSChris Mason 
2282b4ce94deSChris Mason /*
2283c8c42864SChris Mason  * helper function for btrfs_search_slot.  The goal is to find a block
2284c8c42864SChris Mason  * in cache without setting the path to blocking.  If we find the block
2285c8c42864SChris Mason  * we return zero and the path is unchanged.
2286c8c42864SChris Mason  *
2287c8c42864SChris Mason  * If we can't find the block, we set the path blocking and do some
2288c8c42864SChris Mason  * reada.  -EAGAIN is returned and the search must be repeated.
2289c8c42864SChris Mason  */
2290c8c42864SChris Mason static int
2291c8c42864SChris Mason read_block_for_search(struct btrfs_trans_handle *trans,
2292c8c42864SChris Mason 		       struct btrfs_root *root, struct btrfs_path *p,
2293c8c42864SChris Mason 		       struct extent_buffer **eb_ret, int level, int slot,
22945d9e75c4SJan Schmidt 		       struct btrfs_key *key, u64 time_seq)
2295c8c42864SChris Mason {
2296c8c42864SChris Mason 	u64 blocknr;
2297c8c42864SChris Mason 	u64 gen;
2298c8c42864SChris Mason 	u32 blocksize;
2299c8c42864SChris Mason 	struct extent_buffer *b = *eb_ret;
2300c8c42864SChris Mason 	struct extent_buffer *tmp;
230176a05b35SChris Mason 	int ret;
2302c8c42864SChris Mason 
2303c8c42864SChris Mason 	blocknr = btrfs_node_blockptr(b, slot);
2304c8c42864SChris Mason 	gen = btrfs_node_ptr_generation(b, slot);
2305c8c42864SChris Mason 	blocksize = btrfs_level_size(root, level - 1);
2306c8c42864SChris Mason 
2307c8c42864SChris Mason 	tmp = btrfs_find_tree_block(root, blocknr, blocksize);
2308cb44921aSChris Mason 	if (tmp) {
2309b9fab919SChris Mason 		/* first we do an atomic uptodate check */
2310b9fab919SChris Mason 		if (btrfs_buffer_uptodate(tmp, 0, 1) > 0) {
2311b9fab919SChris Mason 			if (btrfs_buffer_uptodate(tmp, gen, 1) > 0) {
231276a05b35SChris Mason 				/*
2313cb44921aSChris Mason 				 * we found an up to date block without
2314cb44921aSChris Mason 				 * sleeping, return
231576a05b35SChris Mason 				 * right away
231676a05b35SChris Mason 				 */
2317c8c42864SChris Mason 				*eb_ret = tmp;
2318c8c42864SChris Mason 				return 0;
2319c8c42864SChris Mason 			}
2320cb44921aSChris Mason 			/* the pages were up to date, but we failed
2321cb44921aSChris Mason 			 * the generation number check.  Do a full
2322cb44921aSChris Mason 			 * read for the generation number that is correct.
2323cb44921aSChris Mason 			 * We must do this without dropping locks so
2324cb44921aSChris Mason 			 * we can trust our generation number
2325cb44921aSChris Mason 			 */
2326cb44921aSChris Mason 			free_extent_buffer(tmp);
2327bd681513SChris Mason 			btrfs_set_path_blocking(p);
2328bd681513SChris Mason 
2329b9fab919SChris Mason 			/* now we're allowed to do a blocking uptodate check */
2330cb44921aSChris Mason 			tmp = read_tree_block(root, blocknr, blocksize, gen);
2331b9fab919SChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen, 0) > 0) {
2332cb44921aSChris Mason 				*eb_ret = tmp;
2333cb44921aSChris Mason 				return 0;
2334cb44921aSChris Mason 			}
2335cb44921aSChris Mason 			free_extent_buffer(tmp);
2336b3b4aa74SDavid Sterba 			btrfs_release_path(p);
2337cb44921aSChris Mason 			return -EIO;
2338cb44921aSChris Mason 		}
2339cb44921aSChris Mason 	}
2340c8c42864SChris Mason 
2341c8c42864SChris Mason 	/*
2342c8c42864SChris Mason 	 * reduce lock contention at high levels
2343c8c42864SChris Mason 	 * of the btree by dropping locks before
234476a05b35SChris Mason 	 * we read.  Don't release the lock on the current
234576a05b35SChris Mason 	 * level because we need to walk this node to figure
234676a05b35SChris Mason 	 * out which blocks to read.
2347c8c42864SChris Mason 	 */
23488c594ea8SChris Mason 	btrfs_unlock_up_safe(p, level + 1);
23498c594ea8SChris Mason 	btrfs_set_path_blocking(p);
23508c594ea8SChris Mason 
2351c8c42864SChris Mason 	free_extent_buffer(tmp);
2352c8c42864SChris Mason 	if (p->reada)
2353c8c42864SChris Mason 		reada_for_search(root, p, level, slot, key->objectid);
2354c8c42864SChris Mason 
2355b3b4aa74SDavid Sterba 	btrfs_release_path(p);
235676a05b35SChris Mason 
235776a05b35SChris Mason 	ret = -EAGAIN;
23585bdd3536SYan, Zheng 	tmp = read_tree_block(root, blocknr, blocksize, 0);
235976a05b35SChris Mason 	if (tmp) {
236076a05b35SChris Mason 		/*
236176a05b35SChris Mason 		 * If the read above didn't mark this buffer up to date,
236276a05b35SChris Mason 		 * it will never end up being up to date.  Set ret to EIO now
236376a05b35SChris Mason 		 * and give up so that our caller doesn't loop forever
236476a05b35SChris Mason 		 * on our EAGAINs.
236576a05b35SChris Mason 		 */
2366b9fab919SChris Mason 		if (!btrfs_buffer_uptodate(tmp, 0, 0))
236776a05b35SChris Mason 			ret = -EIO;
2368c8c42864SChris Mason 		free_extent_buffer(tmp);
236976a05b35SChris Mason 	}
237076a05b35SChris Mason 	return ret;
2371c8c42864SChris Mason }
2372c8c42864SChris Mason 
2373c8c42864SChris Mason /*
2374c8c42864SChris Mason  * helper function for btrfs_search_slot.  This does all of the checks
2375c8c42864SChris Mason  * for node-level blocks and does any balancing required based on
2376c8c42864SChris Mason  * the ins_len.
2377c8c42864SChris Mason  *
2378c8c42864SChris Mason  * If no extra work was required, zero is returned.  If we had to
2379c8c42864SChris Mason  * drop the path, -EAGAIN is returned and btrfs_search_slot must
2380c8c42864SChris Mason  * start over
2381c8c42864SChris Mason  */
2382c8c42864SChris Mason static int
2383c8c42864SChris Mason setup_nodes_for_search(struct btrfs_trans_handle *trans,
2384c8c42864SChris Mason 		       struct btrfs_root *root, struct btrfs_path *p,
2385bd681513SChris Mason 		       struct extent_buffer *b, int level, int ins_len,
2386bd681513SChris Mason 		       int *write_lock_level)
2387c8c42864SChris Mason {
2388c8c42864SChris Mason 	int ret;
2389c8c42864SChris Mason 	if ((p->search_for_split || ins_len > 0) && btrfs_header_nritems(b) >=
2390c8c42864SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
2391c8c42864SChris Mason 		int sret;
2392c8c42864SChris Mason 
2393bd681513SChris Mason 		if (*write_lock_level < level + 1) {
2394bd681513SChris Mason 			*write_lock_level = level + 1;
2395bd681513SChris Mason 			btrfs_release_path(p);
2396bd681513SChris Mason 			goto again;
2397bd681513SChris Mason 		}
2398bd681513SChris Mason 
2399c8c42864SChris Mason 		sret = reada_for_balance(root, p, level);
2400c8c42864SChris Mason 		if (sret)
2401c8c42864SChris Mason 			goto again;
2402c8c42864SChris Mason 
2403c8c42864SChris Mason 		btrfs_set_path_blocking(p);
2404c8c42864SChris Mason 		sret = split_node(trans, root, p, level);
2405bd681513SChris Mason 		btrfs_clear_path_blocking(p, NULL, 0);
2406c8c42864SChris Mason 
2407c8c42864SChris Mason 		BUG_ON(sret > 0);
2408c8c42864SChris Mason 		if (sret) {
2409c8c42864SChris Mason 			ret = sret;
2410c8c42864SChris Mason 			goto done;
2411c8c42864SChris Mason 		}
2412c8c42864SChris Mason 		b = p->nodes[level];
2413c8c42864SChris Mason 	} else if (ins_len < 0 && btrfs_header_nritems(b) <
2414cfbb9308SChris Mason 		   BTRFS_NODEPTRS_PER_BLOCK(root) / 2) {
2415c8c42864SChris Mason 		int sret;
2416c8c42864SChris Mason 
2417bd681513SChris Mason 		if (*write_lock_level < level + 1) {
2418bd681513SChris Mason 			*write_lock_level = level + 1;
2419bd681513SChris Mason 			btrfs_release_path(p);
2420bd681513SChris Mason 			goto again;
2421bd681513SChris Mason 		}
2422bd681513SChris Mason 
2423c8c42864SChris Mason 		sret = reada_for_balance(root, p, level);
2424c8c42864SChris Mason 		if (sret)
2425c8c42864SChris Mason 			goto again;
2426c8c42864SChris Mason 
2427c8c42864SChris Mason 		btrfs_set_path_blocking(p);
2428c8c42864SChris Mason 		sret = balance_level(trans, root, p, level);
2429bd681513SChris Mason 		btrfs_clear_path_blocking(p, NULL, 0);
2430c8c42864SChris Mason 
2431c8c42864SChris Mason 		if (sret) {
2432c8c42864SChris Mason 			ret = sret;
2433c8c42864SChris Mason 			goto done;
2434c8c42864SChris Mason 		}
2435c8c42864SChris Mason 		b = p->nodes[level];
2436c8c42864SChris Mason 		if (!b) {
2437b3b4aa74SDavid Sterba 			btrfs_release_path(p);
2438c8c42864SChris Mason 			goto again;
2439c8c42864SChris Mason 		}
2440c8c42864SChris Mason 		BUG_ON(btrfs_header_nritems(b) == 1);
2441c8c42864SChris Mason 	}
2442c8c42864SChris Mason 	return 0;
2443c8c42864SChris Mason 
2444c8c42864SChris Mason again:
2445c8c42864SChris Mason 	ret = -EAGAIN;
2446c8c42864SChris Mason done:
2447c8c42864SChris Mason 	return ret;
2448c8c42864SChris Mason }
2449c8c42864SChris Mason 
2450c8c42864SChris Mason /*
245174123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
245274123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
245374123bd7SChris Mason  * level of the path (level 0)
245474123bd7SChris Mason  *
245574123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
2456aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
2457aa5d6bedSChris Mason  * search a negative error number is returned.
245897571fd0SChris Mason  *
245997571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
246097571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
246197571fd0SChris Mason  * possible)
246274123bd7SChris Mason  */
2463e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
2464e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
2465e089f05cSChris Mason 		      ins_len, int cow)
2466be0e5c09SChris Mason {
24675f39d397SChris Mason 	struct extent_buffer *b;
2468be0e5c09SChris Mason 	int slot;
2469be0e5c09SChris Mason 	int ret;
247033c66f43SYan Zheng 	int err;
2471be0e5c09SChris Mason 	int level;
2472925baeddSChris Mason 	int lowest_unlock = 1;
2473bd681513SChris Mason 	int root_lock;
2474bd681513SChris Mason 	/* everything at write_lock_level or lower must be write locked */
2475bd681513SChris Mason 	int write_lock_level = 0;
24769f3a7427SChris Mason 	u8 lowest_level = 0;
2477f7c79f30SChris Mason 	int min_write_lock_level;
24789f3a7427SChris Mason 
24796702ed49SChris Mason 	lowest_level = p->lowest_level;
2480323ac95bSChris Mason 	WARN_ON(lowest_level && ins_len > 0);
248122b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
248225179201SJosef Bacik 
2483bd681513SChris Mason 	if (ins_len < 0) {
2484925baeddSChris Mason 		lowest_unlock = 2;
248565b51a00SChris Mason 
2486bd681513SChris Mason 		/* when we are removing items, we might have to go up to level
2487bd681513SChris Mason 		 * two as we update tree pointers  Make sure we keep write
2488bd681513SChris Mason 		 * for those levels as well
2489bd681513SChris Mason 		 */
2490bd681513SChris Mason 		write_lock_level = 2;
2491bd681513SChris Mason 	} else if (ins_len > 0) {
2492bd681513SChris Mason 		/*
2493bd681513SChris Mason 		 * for inserting items, make sure we have a write lock on
2494bd681513SChris Mason 		 * level 1 so we can update keys
2495bd681513SChris Mason 		 */
2496bd681513SChris Mason 		write_lock_level = 1;
2497bd681513SChris Mason 	}
2498bd681513SChris Mason 
2499bd681513SChris Mason 	if (!cow)
2500bd681513SChris Mason 		write_lock_level = -1;
2501bd681513SChris Mason 
2502bd681513SChris Mason 	if (cow && (p->keep_locks || p->lowest_level))
2503bd681513SChris Mason 		write_lock_level = BTRFS_MAX_LEVEL;
2504bd681513SChris Mason 
2505f7c79f30SChris Mason 	min_write_lock_level = write_lock_level;
2506f7c79f30SChris Mason 
2507bb803951SChris Mason again:
2508bd681513SChris Mason 	/*
2509bd681513SChris Mason 	 * we try very hard to do read locks on the root
2510bd681513SChris Mason 	 */
2511bd681513SChris Mason 	root_lock = BTRFS_READ_LOCK;
2512bd681513SChris Mason 	level = 0;
25135d4f98a2SYan Zheng 	if (p->search_commit_root) {
2514bd681513SChris Mason 		/*
2515bd681513SChris Mason 		 * the commit roots are read only
2516bd681513SChris Mason 		 * so we always do read locks
2517bd681513SChris Mason 		 */
25185d4f98a2SYan Zheng 		b = root->commit_root;
25195d4f98a2SYan Zheng 		extent_buffer_get(b);
2520bd681513SChris Mason 		level = btrfs_header_level(b);
25215d4f98a2SYan Zheng 		if (!p->skip_locking)
2522bd681513SChris Mason 			btrfs_tree_read_lock(b);
25235d4f98a2SYan Zheng 	} else {
2524bd681513SChris Mason 		if (p->skip_locking) {
25255cd57b2cSChris Mason 			b = btrfs_root_node(root);
2526bd681513SChris Mason 			level = btrfs_header_level(b);
2527bd681513SChris Mason 		} else {
2528bd681513SChris Mason 			/* we don't know the level of the root node
2529bd681513SChris Mason 			 * until we actually have it read locked
2530bd681513SChris Mason 			 */
2531bd681513SChris Mason 			b = btrfs_read_lock_root_node(root);
2532bd681513SChris Mason 			level = btrfs_header_level(b);
2533bd681513SChris Mason 			if (level <= write_lock_level) {
2534bd681513SChris Mason 				/* whoops, must trade for write lock */
2535bd681513SChris Mason 				btrfs_tree_read_unlock(b);
2536bd681513SChris Mason 				free_extent_buffer(b);
2537925baeddSChris Mason 				b = btrfs_lock_root_node(root);
2538bd681513SChris Mason 				root_lock = BTRFS_WRITE_LOCK;
2539bd681513SChris Mason 
2540bd681513SChris Mason 				/* the level might have changed, check again */
2541bd681513SChris Mason 				level = btrfs_header_level(b);
25425d4f98a2SYan Zheng 			}
2543bd681513SChris Mason 		}
2544bd681513SChris Mason 	}
2545bd681513SChris Mason 	p->nodes[level] = b;
2546bd681513SChris Mason 	if (!p->skip_locking)
2547bd681513SChris Mason 		p->locks[level] = root_lock;
2548925baeddSChris Mason 
2549eb60ceacSChris Mason 	while (b) {
25505f39d397SChris Mason 		level = btrfs_header_level(b);
255165b51a00SChris Mason 
255265b51a00SChris Mason 		/*
255365b51a00SChris Mason 		 * setup the path here so we can release it under lock
255465b51a00SChris Mason 		 * contention with the cow code
255565b51a00SChris Mason 		 */
255602217ed2SChris Mason 		if (cow) {
2557c8c42864SChris Mason 			/*
2558c8c42864SChris Mason 			 * if we don't really need to cow this block
2559c8c42864SChris Mason 			 * then we don't want to set the path blocking,
2560c8c42864SChris Mason 			 * so we test it here
2561c8c42864SChris Mason 			 */
25625d4f98a2SYan Zheng 			if (!should_cow_block(trans, root, b))
256365b51a00SChris Mason 				goto cow_done;
25645d4f98a2SYan Zheng 
2565b4ce94deSChris Mason 			btrfs_set_path_blocking(p);
2566b4ce94deSChris Mason 
2567bd681513SChris Mason 			/*
2568bd681513SChris Mason 			 * must have write locks on this node and the
2569bd681513SChris Mason 			 * parent
2570bd681513SChris Mason 			 */
2571bd681513SChris Mason 			if (level + 1 > write_lock_level) {
2572bd681513SChris Mason 				write_lock_level = level + 1;
2573bd681513SChris Mason 				btrfs_release_path(p);
2574bd681513SChris Mason 				goto again;
2575bd681513SChris Mason 			}
2576bd681513SChris Mason 
257733c66f43SYan Zheng 			err = btrfs_cow_block(trans, root, b,
2578e20d96d6SChris Mason 					      p->nodes[level + 1],
25799fa8cfe7SChris Mason 					      p->slots[level + 1], &b);
258033c66f43SYan Zheng 			if (err) {
258133c66f43SYan Zheng 				ret = err;
258265b51a00SChris Mason 				goto done;
258354aa1f4dSChris Mason 			}
258402217ed2SChris Mason 		}
258565b51a00SChris Mason cow_done:
258602217ed2SChris Mason 		BUG_ON(!cow && ins_len);
258765b51a00SChris Mason 
2588eb60ceacSChris Mason 		p->nodes[level] = b;
2589bd681513SChris Mason 		btrfs_clear_path_blocking(p, NULL, 0);
2590b4ce94deSChris Mason 
2591b4ce94deSChris Mason 		/*
2592b4ce94deSChris Mason 		 * we have a lock on b and as long as we aren't changing
2593b4ce94deSChris Mason 		 * the tree, there is no way to for the items in b to change.
2594b4ce94deSChris Mason 		 * It is safe to drop the lock on our parent before we
2595b4ce94deSChris Mason 		 * go through the expensive btree search on b.
2596b4ce94deSChris Mason 		 *
2597b4ce94deSChris Mason 		 * If cow is true, then we might be changing slot zero,
2598b4ce94deSChris Mason 		 * which may require changing the parent.  So, we can't
2599b4ce94deSChris Mason 		 * drop the lock until after we know which slot we're
2600b4ce94deSChris Mason 		 * operating on.
2601b4ce94deSChris Mason 		 */
2602b4ce94deSChris Mason 		if (!cow)
2603b4ce94deSChris Mason 			btrfs_unlock_up_safe(p, level + 1);
2604b4ce94deSChris Mason 
26055f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
2606b4ce94deSChris Mason 
26075f39d397SChris Mason 		if (level != 0) {
260833c66f43SYan Zheng 			int dec = 0;
260933c66f43SYan Zheng 			if (ret && slot > 0) {
261033c66f43SYan Zheng 				dec = 1;
2611be0e5c09SChris Mason 				slot -= 1;
261233c66f43SYan Zheng 			}
2613be0e5c09SChris Mason 			p->slots[level] = slot;
261433c66f43SYan Zheng 			err = setup_nodes_for_search(trans, root, p, b, level,
2615bd681513SChris Mason 					     ins_len, &write_lock_level);
261633c66f43SYan Zheng 			if (err == -EAGAIN)
2617b4ce94deSChris Mason 				goto again;
261833c66f43SYan Zheng 			if (err) {
261933c66f43SYan Zheng 				ret = err;
262065b51a00SChris Mason 				goto done;
262133c66f43SYan Zheng 			}
26225c680ed6SChris Mason 			b = p->nodes[level];
26235c680ed6SChris Mason 			slot = p->slots[level];
2624b4ce94deSChris Mason 
2625bd681513SChris Mason 			/*
2626bd681513SChris Mason 			 * slot 0 is special, if we change the key
2627bd681513SChris Mason 			 * we have to update the parent pointer
2628bd681513SChris Mason 			 * which means we must have a write lock
2629bd681513SChris Mason 			 * on the parent
2630bd681513SChris Mason 			 */
2631bd681513SChris Mason 			if (slot == 0 && cow &&
2632bd681513SChris Mason 			    write_lock_level < level + 1) {
2633bd681513SChris Mason 				write_lock_level = level + 1;
2634bd681513SChris Mason 				btrfs_release_path(p);
2635bd681513SChris Mason 				goto again;
2636bd681513SChris Mason 			}
2637bd681513SChris Mason 
2638f7c79f30SChris Mason 			unlock_up(p, level, lowest_unlock,
2639f7c79f30SChris Mason 				  min_write_lock_level, &write_lock_level);
2640f9efa9c7SChris Mason 
2641925baeddSChris Mason 			if (level == lowest_level) {
264233c66f43SYan Zheng 				if (dec)
264333c66f43SYan Zheng 					p->slots[level]++;
26445b21f2edSZheng Yan 				goto done;
2645925baeddSChris Mason 			}
2646ca7a79adSChris Mason 
264733c66f43SYan Zheng 			err = read_block_for_search(trans, root, p,
26485d9e75c4SJan Schmidt 						    &b, level, slot, key, 0);
264933c66f43SYan Zheng 			if (err == -EAGAIN)
2650051e1b9fSChris Mason 				goto again;
265133c66f43SYan Zheng 			if (err) {
265233c66f43SYan Zheng 				ret = err;
265376a05b35SChris Mason 				goto done;
265433c66f43SYan Zheng 			}
265576a05b35SChris Mason 
2656b4ce94deSChris Mason 			if (!p->skip_locking) {
2657bd681513SChris Mason 				level = btrfs_header_level(b);
2658bd681513SChris Mason 				if (level <= write_lock_level) {
2659bd681513SChris Mason 					err = btrfs_try_tree_write_lock(b);
266033c66f43SYan Zheng 					if (!err) {
2661b4ce94deSChris Mason 						btrfs_set_path_blocking(p);
2662925baeddSChris Mason 						btrfs_tree_lock(b);
2663bd681513SChris Mason 						btrfs_clear_path_blocking(p, b,
2664bd681513SChris Mason 								  BTRFS_WRITE_LOCK);
2665b4ce94deSChris Mason 					}
2666bd681513SChris Mason 					p->locks[level] = BTRFS_WRITE_LOCK;
2667bd681513SChris Mason 				} else {
2668bd681513SChris Mason 					err = btrfs_try_tree_read_lock(b);
2669bd681513SChris Mason 					if (!err) {
2670bd681513SChris Mason 						btrfs_set_path_blocking(p);
2671bd681513SChris Mason 						btrfs_tree_read_lock(b);
2672bd681513SChris Mason 						btrfs_clear_path_blocking(p, b,
2673bd681513SChris Mason 								  BTRFS_READ_LOCK);
2674bd681513SChris Mason 					}
2675bd681513SChris Mason 					p->locks[level] = BTRFS_READ_LOCK;
2676bd681513SChris Mason 				}
2677bd681513SChris Mason 				p->nodes[level] = b;
2678b4ce94deSChris Mason 			}
2679be0e5c09SChris Mason 		} else {
2680be0e5c09SChris Mason 			p->slots[level] = slot;
268187b29b20SYan Zheng 			if (ins_len > 0 &&
268287b29b20SYan Zheng 			    btrfs_leaf_free_space(root, b) < ins_len) {
2683bd681513SChris Mason 				if (write_lock_level < 1) {
2684bd681513SChris Mason 					write_lock_level = 1;
2685bd681513SChris Mason 					btrfs_release_path(p);
2686bd681513SChris Mason 					goto again;
2687bd681513SChris Mason 				}
2688bd681513SChris Mason 
2689b4ce94deSChris Mason 				btrfs_set_path_blocking(p);
269033c66f43SYan Zheng 				err = split_leaf(trans, root, key,
2691cc0c5538SChris Mason 						 p, ins_len, ret == 0);
2692bd681513SChris Mason 				btrfs_clear_path_blocking(p, NULL, 0);
2693b4ce94deSChris Mason 
269433c66f43SYan Zheng 				BUG_ON(err > 0);
269533c66f43SYan Zheng 				if (err) {
269633c66f43SYan Zheng 					ret = err;
269765b51a00SChris Mason 					goto done;
269865b51a00SChris Mason 				}
26995c680ed6SChris Mason 			}
2700459931ecSChris Mason 			if (!p->search_for_split)
2701f7c79f30SChris Mason 				unlock_up(p, level, lowest_unlock,
2702f7c79f30SChris Mason 					  min_write_lock_level, &write_lock_level);
270365b51a00SChris Mason 			goto done;
270465b51a00SChris Mason 		}
270565b51a00SChris Mason 	}
270665b51a00SChris Mason 	ret = 1;
270765b51a00SChris Mason done:
2708b4ce94deSChris Mason 	/*
2709b4ce94deSChris Mason 	 * we don't really know what they plan on doing with the path
2710b4ce94deSChris Mason 	 * from here on, so for now just mark it as blocking
2711b4ce94deSChris Mason 	 */
2712b9473439SChris Mason 	if (!p->leave_spinning)
2713b4ce94deSChris Mason 		btrfs_set_path_blocking(p);
271476a05b35SChris Mason 	if (ret < 0)
2715b3b4aa74SDavid Sterba 		btrfs_release_path(p);
2716be0e5c09SChris Mason 	return ret;
2717be0e5c09SChris Mason }
2718be0e5c09SChris Mason 
271974123bd7SChris Mason /*
27205d9e75c4SJan Schmidt  * Like btrfs_search_slot, this looks for a key in the given tree. It uses the
27215d9e75c4SJan Schmidt  * current state of the tree together with the operations recorded in the tree
27225d9e75c4SJan Schmidt  * modification log to search for the key in a previous version of this tree, as
27235d9e75c4SJan Schmidt  * denoted by the time_seq parameter.
27245d9e75c4SJan Schmidt  *
27255d9e75c4SJan Schmidt  * Naturally, there is no support for insert, delete or cow operations.
27265d9e75c4SJan Schmidt  *
27275d9e75c4SJan Schmidt  * The resulting path and return value will be set up as if we called
27285d9e75c4SJan Schmidt  * btrfs_search_slot at that point in time with ins_len and cow both set to 0.
27295d9e75c4SJan Schmidt  */
27305d9e75c4SJan Schmidt int btrfs_search_old_slot(struct btrfs_root *root, struct btrfs_key *key,
27315d9e75c4SJan Schmidt 			  struct btrfs_path *p, u64 time_seq)
27325d9e75c4SJan Schmidt {
27335d9e75c4SJan Schmidt 	struct extent_buffer *b;
27345d9e75c4SJan Schmidt 	int slot;
27355d9e75c4SJan Schmidt 	int ret;
27365d9e75c4SJan Schmidt 	int err;
27375d9e75c4SJan Schmidt 	int level;
27385d9e75c4SJan Schmidt 	int lowest_unlock = 1;
27395d9e75c4SJan Schmidt 	u8 lowest_level = 0;
27405d9e75c4SJan Schmidt 
27415d9e75c4SJan Schmidt 	lowest_level = p->lowest_level;
27425d9e75c4SJan Schmidt 	WARN_ON(p->nodes[0] != NULL);
27435d9e75c4SJan Schmidt 
27445d9e75c4SJan Schmidt 	if (p->search_commit_root) {
27455d9e75c4SJan Schmidt 		BUG_ON(time_seq);
27465d9e75c4SJan Schmidt 		return btrfs_search_slot(NULL, root, key, p, 0, 0);
27475d9e75c4SJan Schmidt 	}
27485d9e75c4SJan Schmidt 
27495d9e75c4SJan Schmidt again:
27505d9e75c4SJan Schmidt 	b = get_old_root(root, time_seq);
27515d9e75c4SJan Schmidt 	level = btrfs_header_level(b);
27525d9e75c4SJan Schmidt 	p->locks[level] = BTRFS_READ_LOCK;
27535d9e75c4SJan Schmidt 
27545d9e75c4SJan Schmidt 	while (b) {
27555d9e75c4SJan Schmidt 		level = btrfs_header_level(b);
27565d9e75c4SJan Schmidt 		p->nodes[level] = b;
27575d9e75c4SJan Schmidt 		btrfs_clear_path_blocking(p, NULL, 0);
27585d9e75c4SJan Schmidt 
27595d9e75c4SJan Schmidt 		/*
27605d9e75c4SJan Schmidt 		 * we have a lock on b and as long as we aren't changing
27615d9e75c4SJan Schmidt 		 * the tree, there is no way to for the items in b to change.
27625d9e75c4SJan Schmidt 		 * It is safe to drop the lock on our parent before we
27635d9e75c4SJan Schmidt 		 * go through the expensive btree search on b.
27645d9e75c4SJan Schmidt 		 */
27655d9e75c4SJan Schmidt 		btrfs_unlock_up_safe(p, level + 1);
27665d9e75c4SJan Schmidt 
27675d9e75c4SJan Schmidt 		ret = bin_search(b, key, level, &slot);
27685d9e75c4SJan Schmidt 
27695d9e75c4SJan Schmidt 		if (level != 0) {
27705d9e75c4SJan Schmidt 			int dec = 0;
27715d9e75c4SJan Schmidt 			if (ret && slot > 0) {
27725d9e75c4SJan Schmidt 				dec = 1;
27735d9e75c4SJan Schmidt 				slot -= 1;
27745d9e75c4SJan Schmidt 			}
27755d9e75c4SJan Schmidt 			p->slots[level] = slot;
27765d9e75c4SJan Schmidt 			unlock_up(p, level, lowest_unlock, 0, NULL);
27775d9e75c4SJan Schmidt 
27785d9e75c4SJan Schmidt 			if (level == lowest_level) {
27795d9e75c4SJan Schmidt 				if (dec)
27805d9e75c4SJan Schmidt 					p->slots[level]++;
27815d9e75c4SJan Schmidt 				goto done;
27825d9e75c4SJan Schmidt 			}
27835d9e75c4SJan Schmidt 
27845d9e75c4SJan Schmidt 			err = read_block_for_search(NULL, root, p, &b, level,
27855d9e75c4SJan Schmidt 						    slot, key, time_seq);
27865d9e75c4SJan Schmidt 			if (err == -EAGAIN)
27875d9e75c4SJan Schmidt 				goto again;
27885d9e75c4SJan Schmidt 			if (err) {
27895d9e75c4SJan Schmidt 				ret = err;
27905d9e75c4SJan Schmidt 				goto done;
27915d9e75c4SJan Schmidt 			}
27925d9e75c4SJan Schmidt 
27935d9e75c4SJan Schmidt 			level = btrfs_header_level(b);
27945d9e75c4SJan Schmidt 			err = btrfs_try_tree_read_lock(b);
27955d9e75c4SJan Schmidt 			if (!err) {
27965d9e75c4SJan Schmidt 				btrfs_set_path_blocking(p);
27975d9e75c4SJan Schmidt 				btrfs_tree_read_lock(b);
27985d9e75c4SJan Schmidt 				btrfs_clear_path_blocking(p, b,
27995d9e75c4SJan Schmidt 							  BTRFS_READ_LOCK);
28005d9e75c4SJan Schmidt 			}
28015d9e75c4SJan Schmidt 			p->locks[level] = BTRFS_READ_LOCK;
28025d9e75c4SJan Schmidt 			p->nodes[level] = b;
28035d9e75c4SJan Schmidt 			b = tree_mod_log_rewind(root->fs_info, b, time_seq);
28045d9e75c4SJan Schmidt 			if (b != p->nodes[level]) {
28055d9e75c4SJan Schmidt 				btrfs_tree_unlock_rw(p->nodes[level],
28065d9e75c4SJan Schmidt 						     p->locks[level]);
28075d9e75c4SJan Schmidt 				p->locks[level] = 0;
28085d9e75c4SJan Schmidt 				p->nodes[level] = b;
28095d9e75c4SJan Schmidt 			}
28105d9e75c4SJan Schmidt 		} else {
28115d9e75c4SJan Schmidt 			p->slots[level] = slot;
28125d9e75c4SJan Schmidt 			unlock_up(p, level, lowest_unlock, 0, NULL);
28135d9e75c4SJan Schmidt 			goto done;
28145d9e75c4SJan Schmidt 		}
28155d9e75c4SJan Schmidt 	}
28165d9e75c4SJan Schmidt 	ret = 1;
28175d9e75c4SJan Schmidt done:
28185d9e75c4SJan Schmidt 	if (!p->leave_spinning)
28195d9e75c4SJan Schmidt 		btrfs_set_path_blocking(p);
28205d9e75c4SJan Schmidt 	if (ret < 0)
28215d9e75c4SJan Schmidt 		btrfs_release_path(p);
28225d9e75c4SJan Schmidt 
28235d9e75c4SJan Schmidt 	return ret;
28245d9e75c4SJan Schmidt }
28255d9e75c4SJan Schmidt 
28265d9e75c4SJan Schmidt /*
28272f38b3e1SArne Jansen  * helper to use instead of search slot if no exact match is needed but
28282f38b3e1SArne Jansen  * instead the next or previous item should be returned.
28292f38b3e1SArne Jansen  * When find_higher is true, the next higher item is returned, the next lower
28302f38b3e1SArne Jansen  * otherwise.
28312f38b3e1SArne Jansen  * When return_any and find_higher are both true, and no higher item is found,
28322f38b3e1SArne Jansen  * return the next lower instead.
28332f38b3e1SArne Jansen  * When return_any is true and find_higher is false, and no lower item is found,
28342f38b3e1SArne Jansen  * return the next higher instead.
28352f38b3e1SArne Jansen  * It returns 0 if any item is found, 1 if none is found (tree empty), and
28362f38b3e1SArne Jansen  * < 0 on error
28372f38b3e1SArne Jansen  */
28382f38b3e1SArne Jansen int btrfs_search_slot_for_read(struct btrfs_root *root,
28392f38b3e1SArne Jansen 			       struct btrfs_key *key, struct btrfs_path *p,
28402f38b3e1SArne Jansen 			       int find_higher, int return_any)
28412f38b3e1SArne Jansen {
28422f38b3e1SArne Jansen 	int ret;
28432f38b3e1SArne Jansen 	struct extent_buffer *leaf;
28442f38b3e1SArne Jansen 
28452f38b3e1SArne Jansen again:
28462f38b3e1SArne Jansen 	ret = btrfs_search_slot(NULL, root, key, p, 0, 0);
28472f38b3e1SArne Jansen 	if (ret <= 0)
28482f38b3e1SArne Jansen 		return ret;
28492f38b3e1SArne Jansen 	/*
28502f38b3e1SArne Jansen 	 * a return value of 1 means the path is at the position where the
28512f38b3e1SArne Jansen 	 * item should be inserted. Normally this is the next bigger item,
28522f38b3e1SArne Jansen 	 * but in case the previous item is the last in a leaf, path points
28532f38b3e1SArne Jansen 	 * to the first free slot in the previous leaf, i.e. at an invalid
28542f38b3e1SArne Jansen 	 * item.
28552f38b3e1SArne Jansen 	 */
28562f38b3e1SArne Jansen 	leaf = p->nodes[0];
28572f38b3e1SArne Jansen 
28582f38b3e1SArne Jansen 	if (find_higher) {
28592f38b3e1SArne Jansen 		if (p->slots[0] >= btrfs_header_nritems(leaf)) {
28602f38b3e1SArne Jansen 			ret = btrfs_next_leaf(root, p);
28612f38b3e1SArne Jansen 			if (ret <= 0)
28622f38b3e1SArne Jansen 				return ret;
28632f38b3e1SArne Jansen 			if (!return_any)
28642f38b3e1SArne Jansen 				return 1;
28652f38b3e1SArne Jansen 			/*
28662f38b3e1SArne Jansen 			 * no higher item found, return the next
28672f38b3e1SArne Jansen 			 * lower instead
28682f38b3e1SArne Jansen 			 */
28692f38b3e1SArne Jansen 			return_any = 0;
28702f38b3e1SArne Jansen 			find_higher = 0;
28712f38b3e1SArne Jansen 			btrfs_release_path(p);
28722f38b3e1SArne Jansen 			goto again;
28732f38b3e1SArne Jansen 		}
28742f38b3e1SArne Jansen 	} else {
28752f38b3e1SArne Jansen 		if (p->slots[0] == 0) {
28762f38b3e1SArne Jansen 			ret = btrfs_prev_leaf(root, p);
2877e6793769SArne Jansen 			if (ret < 0)
28782f38b3e1SArne Jansen 				return ret;
2879e6793769SArne Jansen 			if (!ret) {
2880e6793769SArne Jansen 				p->slots[0] = btrfs_header_nritems(leaf) - 1;
2881e6793769SArne Jansen 				return 0;
2882e6793769SArne Jansen 			}
28832f38b3e1SArne Jansen 			if (!return_any)
28842f38b3e1SArne Jansen 				return 1;
28852f38b3e1SArne Jansen 			/*
28862f38b3e1SArne Jansen 			 * no lower item found, return the next
28872f38b3e1SArne Jansen 			 * higher instead
28882f38b3e1SArne Jansen 			 */
28892f38b3e1SArne Jansen 			return_any = 0;
28902f38b3e1SArne Jansen 			find_higher = 1;
28912f38b3e1SArne Jansen 			btrfs_release_path(p);
28922f38b3e1SArne Jansen 			goto again;
2893e6793769SArne Jansen 		} else {
28942f38b3e1SArne Jansen 			--p->slots[0];
28952f38b3e1SArne Jansen 		}
28962f38b3e1SArne Jansen 	}
28972f38b3e1SArne Jansen 	return 0;
28982f38b3e1SArne Jansen }
28992f38b3e1SArne Jansen 
29002f38b3e1SArne Jansen /*
290174123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
290274123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
290374123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
290474123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
290574123bd7SChris Mason  * higher levels
2906aa5d6bedSChris Mason  *
290774123bd7SChris Mason  */
2908143bede5SJeff Mahoney static void fixup_low_keys(struct btrfs_trans_handle *trans,
29095f39d397SChris Mason 			   struct btrfs_root *root, struct btrfs_path *path,
29105f39d397SChris Mason 			   struct btrfs_disk_key *key, int level)
2911be0e5c09SChris Mason {
2912be0e5c09SChris Mason 	int i;
29135f39d397SChris Mason 	struct extent_buffer *t;
29145f39d397SChris Mason 
2915234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2916be0e5c09SChris Mason 		int tslot = path->slots[i];
2917eb60ceacSChris Mason 		if (!path->nodes[i])
2918be0e5c09SChris Mason 			break;
29195f39d397SChris Mason 		t = path->nodes[i];
2920f230475eSJan Schmidt 		tree_mod_log_set_node_key(root->fs_info, t, key, tslot, 1);
29215f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
2922d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
2923be0e5c09SChris Mason 		if (tslot != 0)
2924be0e5c09SChris Mason 			break;
2925be0e5c09SChris Mason 	}
2926be0e5c09SChris Mason }
2927be0e5c09SChris Mason 
292874123bd7SChris Mason /*
292931840ae1SZheng Yan  * update item key.
293031840ae1SZheng Yan  *
293131840ae1SZheng Yan  * This function isn't completely safe. It's the caller's responsibility
293231840ae1SZheng Yan  * that the new key won't break the order
293331840ae1SZheng Yan  */
2934143bede5SJeff Mahoney void btrfs_set_item_key_safe(struct btrfs_trans_handle *trans,
293531840ae1SZheng Yan 			     struct btrfs_root *root, struct btrfs_path *path,
293631840ae1SZheng Yan 			     struct btrfs_key *new_key)
293731840ae1SZheng Yan {
293831840ae1SZheng Yan 	struct btrfs_disk_key disk_key;
293931840ae1SZheng Yan 	struct extent_buffer *eb;
294031840ae1SZheng Yan 	int slot;
294131840ae1SZheng Yan 
294231840ae1SZheng Yan 	eb = path->nodes[0];
294331840ae1SZheng Yan 	slot = path->slots[0];
294431840ae1SZheng Yan 	if (slot > 0) {
294531840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot - 1);
2946143bede5SJeff Mahoney 		BUG_ON(comp_keys(&disk_key, new_key) >= 0);
294731840ae1SZheng Yan 	}
294831840ae1SZheng Yan 	if (slot < btrfs_header_nritems(eb) - 1) {
294931840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot + 1);
2950143bede5SJeff Mahoney 		BUG_ON(comp_keys(&disk_key, new_key) <= 0);
295131840ae1SZheng Yan 	}
295231840ae1SZheng Yan 
295331840ae1SZheng Yan 	btrfs_cpu_key_to_disk(&disk_key, new_key);
295431840ae1SZheng Yan 	btrfs_set_item_key(eb, &disk_key, slot);
295531840ae1SZheng Yan 	btrfs_mark_buffer_dirty(eb);
295631840ae1SZheng Yan 	if (slot == 0)
295731840ae1SZheng Yan 		fixup_low_keys(trans, root, path, &disk_key, 1);
295831840ae1SZheng Yan }
295931840ae1SZheng Yan 
296031840ae1SZheng Yan /*
296174123bd7SChris Mason  * try to push data from one node into the next node left in the
296279f95c82SChris Mason  * tree.
2963aa5d6bedSChris Mason  *
2964aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
2965aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
296674123bd7SChris Mason  */
296798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
296898ed5174SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
2969971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
2970be0e5c09SChris Mason {
2971be0e5c09SChris Mason 	int push_items = 0;
2972bb803951SChris Mason 	int src_nritems;
2973bb803951SChris Mason 	int dst_nritems;
2974aa5d6bedSChris Mason 	int ret = 0;
2975be0e5c09SChris Mason 
29765f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
29775f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
2978123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
29797bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
29807bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
298154aa1f4dSChris Mason 
2982bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
2983971a1f66SChris Mason 		return 1;
2984971a1f66SChris Mason 
2985d397712bSChris Mason 	if (push_items <= 0)
2986be0e5c09SChris Mason 		return 1;
2987be0e5c09SChris Mason 
2988bce4eae9SChris Mason 	if (empty) {
2989971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
2990bce4eae9SChris Mason 		if (push_items < src_nritems) {
2991bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
2992bce4eae9SChris Mason 			 * we aren't going to empty it
2993bce4eae9SChris Mason 			 */
2994bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
2995bce4eae9SChris Mason 				if (push_items <= 8)
2996bce4eae9SChris Mason 					return 1;
2997bce4eae9SChris Mason 				push_items -= 8;
2998bce4eae9SChris Mason 			}
2999bce4eae9SChris Mason 		}
3000bce4eae9SChris Mason 	} else
3001bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
300279f95c82SChris Mason 
3003f230475eSJan Schmidt 	tree_mod_log_eb_copy(root->fs_info, dst, src, dst_nritems, 0,
3004f230475eSJan Schmidt 			     push_items);
30055f39d397SChris Mason 	copy_extent_buffer(dst, src,
30065f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
30075f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
3008123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
30095f39d397SChris Mason 
3010bb803951SChris Mason 	if (push_items < src_nritems) {
301157911b8bSJan Schmidt 		/*
301257911b8bSJan Schmidt 		 * don't call tree_mod_log_eb_move here, key removal was already
301357911b8bSJan Schmidt 		 * fully logged by tree_mod_log_eb_copy above.
301457911b8bSJan Schmidt 		 */
30155f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
30165f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
3017e2fa7227SChris Mason 				      (src_nritems - push_items) *
3018123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
3019bb803951SChris Mason 	}
30205f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
30215f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
30225f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
30235f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
302431840ae1SZheng Yan 
3025bb803951SChris Mason 	return ret;
3026be0e5c09SChris Mason }
3027be0e5c09SChris Mason 
302897571fd0SChris Mason /*
302979f95c82SChris Mason  * try to push data from one node into the next node right in the
303079f95c82SChris Mason  * tree.
303179f95c82SChris Mason  *
303279f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
303379f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
303479f95c82SChris Mason  *
303579f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
303679f95c82SChris Mason  */
30375f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
30385f39d397SChris Mason 			      struct btrfs_root *root,
30395f39d397SChris Mason 			      struct extent_buffer *dst,
30405f39d397SChris Mason 			      struct extent_buffer *src)
304179f95c82SChris Mason {
304279f95c82SChris Mason 	int push_items = 0;
304379f95c82SChris Mason 	int max_push;
304479f95c82SChris Mason 	int src_nritems;
304579f95c82SChris Mason 	int dst_nritems;
304679f95c82SChris Mason 	int ret = 0;
304779f95c82SChris Mason 
30487bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
30497bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
30507bb86316SChris Mason 
30515f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
30525f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
3053123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
3054d397712bSChris Mason 	if (push_items <= 0)
305579f95c82SChris Mason 		return 1;
3056bce4eae9SChris Mason 
3057d397712bSChris Mason 	if (src_nritems < 4)
3058bce4eae9SChris Mason 		return 1;
305979f95c82SChris Mason 
306079f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
306179f95c82SChris Mason 	/* don't try to empty the node */
3062d397712bSChris Mason 	if (max_push >= src_nritems)
306379f95c82SChris Mason 		return 1;
3064252c38f0SYan 
306579f95c82SChris Mason 	if (max_push < push_items)
306679f95c82SChris Mason 		push_items = max_push;
306779f95c82SChris Mason 
3068f230475eSJan Schmidt 	tree_mod_log_eb_move(root->fs_info, dst, push_items, 0, dst_nritems);
30695f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
30705f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
30715f39d397SChris Mason 				      (dst_nritems) *
30725f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
3073d6025579SChris Mason 
3074f230475eSJan Schmidt 	tree_mod_log_eb_copy(root->fs_info, dst, src, 0,
3075f230475eSJan Schmidt 			     src_nritems - push_items, push_items);
30765f39d397SChris Mason 	copy_extent_buffer(dst, src,
30775f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
30785f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
3079123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
308079f95c82SChris Mason 
30815f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
30825f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
308379f95c82SChris Mason 
30845f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
30855f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
308631840ae1SZheng Yan 
308779f95c82SChris Mason 	return ret;
308879f95c82SChris Mason }
308979f95c82SChris Mason 
309079f95c82SChris Mason /*
309197571fd0SChris Mason  * helper function to insert a new root level in the tree.
309297571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
309397571fd0SChris Mason  * point to the existing root
3094aa5d6bedSChris Mason  *
3095aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
309697571fd0SChris Mason  */
3097d397712bSChris Mason static noinline int insert_new_root(struct btrfs_trans_handle *trans,
30985f39d397SChris Mason 			   struct btrfs_root *root,
30995f39d397SChris Mason 			   struct btrfs_path *path, int level)
310074123bd7SChris Mason {
31017bb86316SChris Mason 	u64 lower_gen;
31025f39d397SChris Mason 	struct extent_buffer *lower;
31035f39d397SChris Mason 	struct extent_buffer *c;
3104925baeddSChris Mason 	struct extent_buffer *old;
31055f39d397SChris Mason 	struct btrfs_disk_key lower_key;
31065c680ed6SChris Mason 
31075c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
31085c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
31095c680ed6SChris Mason 
31107bb86316SChris Mason 	lower = path->nodes[level-1];
31117bb86316SChris Mason 	if (level == 1)
31127bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
31137bb86316SChris Mason 	else
31147bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
31157bb86316SChris Mason 
311631840ae1SZheng Yan 	c = btrfs_alloc_free_block(trans, root, root->nodesize, 0,
31175d4f98a2SYan Zheng 				   root->root_key.objectid, &lower_key,
31185581a51aSJan Schmidt 				   level, root->node->start, 0);
31195f39d397SChris Mason 	if (IS_ERR(c))
31205f39d397SChris Mason 		return PTR_ERR(c);
3121925baeddSChris Mason 
3122f0486c68SYan, Zheng 	root_add_used(root, root->nodesize);
3123f0486c68SYan, Zheng 
31245d4f98a2SYan Zheng 	memset_extent_buffer(c, 0, 0, sizeof(struct btrfs_header));
31255f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
31265f39d397SChris Mason 	btrfs_set_header_level(c, level);
3127db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
31285f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
31295d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(c, BTRFS_MIXED_BACKREF_REV);
31305f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
3131d5719762SChris Mason 
31325f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
31335f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
31345f39d397SChris Mason 			    BTRFS_FSID_SIZE);
3135e17cade2SChris Mason 
3136e17cade2SChris Mason 	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
3137e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
3138e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
3139e17cade2SChris Mason 
31405f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
3141db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
31427bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
314331840ae1SZheng Yan 	WARN_ON(lower_gen != trans->transid);
31447bb86316SChris Mason 
31457bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
31465f39d397SChris Mason 
31475f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
3148d5719762SChris Mason 
3149925baeddSChris Mason 	old = root->node;
3150f230475eSJan Schmidt 	tree_mod_log_set_root_pointer(root, c);
3151240f62c8SChris Mason 	rcu_assign_pointer(root->node, c);
3152925baeddSChris Mason 
3153925baeddSChris Mason 	/* the super has an extra ref to root->node */
3154925baeddSChris Mason 	free_extent_buffer(old);
3155925baeddSChris Mason 
31560b86a832SChris Mason 	add_root_to_dirty_list(root);
31575f39d397SChris Mason 	extent_buffer_get(c);
31585f39d397SChris Mason 	path->nodes[level] = c;
3159bd681513SChris Mason 	path->locks[level] = BTRFS_WRITE_LOCK;
316074123bd7SChris Mason 	path->slots[level] = 0;
316174123bd7SChris Mason 	return 0;
316274123bd7SChris Mason }
31635c680ed6SChris Mason 
31645c680ed6SChris Mason /*
31655c680ed6SChris Mason  * worker function to insert a single pointer in a node.
31665c680ed6SChris Mason  * the node should have enough room for the pointer already
316797571fd0SChris Mason  *
31685c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
31695c680ed6SChris Mason  * blocknr is the block the key points to.
31705c680ed6SChris Mason  */
3171143bede5SJeff Mahoney static void insert_ptr(struct btrfs_trans_handle *trans,
3172143bede5SJeff Mahoney 		       struct btrfs_root *root, struct btrfs_path *path,
3173143bede5SJeff Mahoney 		       struct btrfs_disk_key *key, u64 bytenr,
3174c3e06965SJan Schmidt 		       int slot, int level)
31755c680ed6SChris Mason {
31765f39d397SChris Mason 	struct extent_buffer *lower;
31775c680ed6SChris Mason 	int nritems;
3178f3ea38daSJan Schmidt 	int ret;
31795c680ed6SChris Mason 
31805c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
3181f0486c68SYan, Zheng 	btrfs_assert_tree_locked(path->nodes[level]);
31825f39d397SChris Mason 	lower = path->nodes[level];
31835f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
3184c293498bSStoyan Gaydarov 	BUG_ON(slot > nritems);
3185143bede5SJeff Mahoney 	BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(root));
318674123bd7SChris Mason 	if (slot != nritems) {
3187c3e06965SJan Schmidt 		if (level)
3188f3ea38daSJan Schmidt 			tree_mod_log_eb_move(root->fs_info, lower, slot + 1,
3189f3ea38daSJan Schmidt 					     slot, nritems - slot);
31905f39d397SChris Mason 		memmove_extent_buffer(lower,
31915f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
31925f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
3193123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
319474123bd7SChris Mason 	}
3195c3e06965SJan Schmidt 	if (level) {
3196f3ea38daSJan Schmidt 		ret = tree_mod_log_insert_key(root->fs_info, lower, slot,
3197f3ea38daSJan Schmidt 					      MOD_LOG_KEY_ADD);
3198f3ea38daSJan Schmidt 		BUG_ON(ret < 0);
3199f3ea38daSJan Schmidt 	}
32005f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
3201db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
320274493f7aSChris Mason 	WARN_ON(trans->transid == 0);
320374493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
32045f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
32055f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
320674123bd7SChris Mason }
320774123bd7SChris Mason 
320897571fd0SChris Mason /*
320997571fd0SChris Mason  * split the node at the specified level in path in two.
321097571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
321197571fd0SChris Mason  *
321297571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
321397571fd0SChris Mason  * left and right, if either one works, it returns right away.
3214aa5d6bedSChris Mason  *
3215aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
321697571fd0SChris Mason  */
3217e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans,
3218e02119d5SChris Mason 			       struct btrfs_root *root,
3219e02119d5SChris Mason 			       struct btrfs_path *path, int level)
3220be0e5c09SChris Mason {
32215f39d397SChris Mason 	struct extent_buffer *c;
32225f39d397SChris Mason 	struct extent_buffer *split;
32235f39d397SChris Mason 	struct btrfs_disk_key disk_key;
3224be0e5c09SChris Mason 	int mid;
32255c680ed6SChris Mason 	int ret;
32267518a238SChris Mason 	u32 c_nritems;
3227be0e5c09SChris Mason 
32285f39d397SChris Mason 	c = path->nodes[level];
32297bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
32305f39d397SChris Mason 	if (c == root->node) {
32315c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
3232e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
32335c680ed6SChris Mason 		if (ret)
32345c680ed6SChris Mason 			return ret;
3235b3612421SChris Mason 	} else {
3236e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
32375f39d397SChris Mason 		c = path->nodes[level];
32385f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
3239c448acf0SChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
3240e66f709bSChris Mason 			return 0;
324154aa1f4dSChris Mason 		if (ret < 0)
324254aa1f4dSChris Mason 			return ret;
32435c680ed6SChris Mason 	}
3244e66f709bSChris Mason 
32455f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
32465d4f98a2SYan Zheng 	mid = (c_nritems + 1) / 2;
32475d4f98a2SYan Zheng 	btrfs_node_key(c, &disk_key, mid);
32487bb86316SChris Mason 
32495d4f98a2SYan Zheng 	split = btrfs_alloc_free_block(trans, root, root->nodesize, 0,
32507bb86316SChris Mason 					root->root_key.objectid,
32515581a51aSJan Schmidt 					&disk_key, level, c->start, 0);
32525f39d397SChris Mason 	if (IS_ERR(split))
32535f39d397SChris Mason 		return PTR_ERR(split);
325454aa1f4dSChris Mason 
3255f0486c68SYan, Zheng 	root_add_used(root, root->nodesize);
3256f0486c68SYan, Zheng 
32575d4f98a2SYan Zheng 	memset_extent_buffer(split, 0, 0, sizeof(struct btrfs_header));
32585f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
3259db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
32605f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
32615d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(split, BTRFS_MIXED_BACKREF_REV);
32625f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
32635f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
32645f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
32655f39d397SChris Mason 			    BTRFS_FSID_SIZE);
3266e17cade2SChris Mason 	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
3267e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
3268e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
32695f39d397SChris Mason 
3270f230475eSJan Schmidt 	tree_mod_log_eb_copy(root->fs_info, split, c, 0, mid, c_nritems - mid);
32715f39d397SChris Mason 	copy_extent_buffer(split, c,
32725f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
32735f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
3274123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
32755f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
32765f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
3277aa5d6bedSChris Mason 	ret = 0;
3278aa5d6bedSChris Mason 
32795f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
32805f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
32815f39d397SChris Mason 
3282143bede5SJeff Mahoney 	insert_ptr(trans, root, path, &disk_key, split->start,
3283c3e06965SJan Schmidt 		   path->slots[level + 1] + 1, level + 1);
3284aa5d6bedSChris Mason 
32855de08d7dSChris Mason 	if (path->slots[level] >= mid) {
32865c680ed6SChris Mason 		path->slots[level] -= mid;
3287925baeddSChris Mason 		btrfs_tree_unlock(c);
32885f39d397SChris Mason 		free_extent_buffer(c);
32895f39d397SChris Mason 		path->nodes[level] = split;
32905c680ed6SChris Mason 		path->slots[level + 1] += 1;
3291eb60ceacSChris Mason 	} else {
3292925baeddSChris Mason 		btrfs_tree_unlock(split);
32935f39d397SChris Mason 		free_extent_buffer(split);
3294be0e5c09SChris Mason 	}
3295aa5d6bedSChris Mason 	return ret;
3296be0e5c09SChris Mason }
3297be0e5c09SChris Mason 
329874123bd7SChris Mason /*
329974123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
330074123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
330174123bd7SChris Mason  * space used both by the item structs and the item data
330274123bd7SChris Mason  */
33035f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
3304be0e5c09SChris Mason {
3305be0e5c09SChris Mason 	int data_len;
33065f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
3307d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
3308be0e5c09SChris Mason 
3309be0e5c09SChris Mason 	if (!nr)
3310be0e5c09SChris Mason 		return 0;
33115f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
33125f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
33130783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
3314d4dbff95SChris Mason 	WARN_ON(data_len < 0);
3315be0e5c09SChris Mason 	return data_len;
3316be0e5c09SChris Mason }
3317be0e5c09SChris Mason 
331874123bd7SChris Mason /*
3319d4dbff95SChris Mason  * The space between the end of the leaf items and
3320d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
3321d4dbff95SChris Mason  * the leaf has left for both items and data
3322d4dbff95SChris Mason  */
3323d397712bSChris Mason noinline int btrfs_leaf_free_space(struct btrfs_root *root,
3324e02119d5SChris Mason 				   struct extent_buffer *leaf)
3325d4dbff95SChris Mason {
33265f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
33275f39d397SChris Mason 	int ret;
33285f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
33295f39d397SChris Mason 	if (ret < 0) {
3330d397712bSChris Mason 		printk(KERN_CRIT "leaf free space ret %d, leaf data size %lu, "
3331d397712bSChris Mason 		       "used %d nritems %d\n",
3332ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
33335f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
33345f39d397SChris Mason 	}
33355f39d397SChris Mason 	return ret;
3336d4dbff95SChris Mason }
3337d4dbff95SChris Mason 
333899d8f83cSChris Mason /*
333999d8f83cSChris Mason  * min slot controls the lowest index we're willing to push to the
334099d8f83cSChris Mason  * right.  We'll push up to and including min_slot, but no lower
334199d8f83cSChris Mason  */
334244871b1bSChris Mason static noinline int __push_leaf_right(struct btrfs_trans_handle *trans,
334344871b1bSChris Mason 				      struct btrfs_root *root,
334444871b1bSChris Mason 				      struct btrfs_path *path,
334544871b1bSChris Mason 				      int data_size, int empty,
334644871b1bSChris Mason 				      struct extent_buffer *right,
334799d8f83cSChris Mason 				      int free_space, u32 left_nritems,
334899d8f83cSChris Mason 				      u32 min_slot)
334900ec4c51SChris Mason {
33505f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
335144871b1bSChris Mason 	struct extent_buffer *upper = path->nodes[1];
3352cfed81a0SChris Mason 	struct btrfs_map_token token;
33535f39d397SChris Mason 	struct btrfs_disk_key disk_key;
335400ec4c51SChris Mason 	int slot;
335534a38218SChris Mason 	u32 i;
335600ec4c51SChris Mason 	int push_space = 0;
335700ec4c51SChris Mason 	int push_items = 0;
33580783fcfcSChris Mason 	struct btrfs_item *item;
335934a38218SChris Mason 	u32 nr;
33607518a238SChris Mason 	u32 right_nritems;
33615f39d397SChris Mason 	u32 data_end;
3362db94535dSChris Mason 	u32 this_item_size;
336300ec4c51SChris Mason 
3364cfed81a0SChris Mason 	btrfs_init_map_token(&token);
3365cfed81a0SChris Mason 
336634a38218SChris Mason 	if (empty)
336734a38218SChris Mason 		nr = 0;
336834a38218SChris Mason 	else
336999d8f83cSChris Mason 		nr = max_t(u32, 1, min_slot);
337034a38218SChris Mason 
337131840ae1SZheng Yan 	if (path->slots[0] >= left_nritems)
337287b29b20SYan Zheng 		push_space += data_size;
337331840ae1SZheng Yan 
337444871b1bSChris Mason 	slot = path->slots[1];
337534a38218SChris Mason 	i = left_nritems - 1;
337634a38218SChris Mason 	while (i >= nr) {
33775f39d397SChris Mason 		item = btrfs_item_nr(left, i);
3378db94535dSChris Mason 
337931840ae1SZheng Yan 		if (!empty && push_items > 0) {
338031840ae1SZheng Yan 			if (path->slots[0] > i)
338131840ae1SZheng Yan 				break;
338231840ae1SZheng Yan 			if (path->slots[0] == i) {
338331840ae1SZheng Yan 				int space = btrfs_leaf_free_space(root, left);
338431840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
338531840ae1SZheng Yan 					break;
338631840ae1SZheng Yan 			}
338731840ae1SZheng Yan 		}
338831840ae1SZheng Yan 
338900ec4c51SChris Mason 		if (path->slots[0] == i)
339087b29b20SYan Zheng 			push_space += data_size;
3391db94535dSChris Mason 
3392db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
3393db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
339400ec4c51SChris Mason 			break;
339531840ae1SZheng Yan 
339600ec4c51SChris Mason 		push_items++;
3397db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
339834a38218SChris Mason 		if (i == 0)
339934a38218SChris Mason 			break;
340034a38218SChris Mason 		i--;
3401db94535dSChris Mason 	}
34025f39d397SChris Mason 
3403925baeddSChris Mason 	if (push_items == 0)
3404925baeddSChris Mason 		goto out_unlock;
34055f39d397SChris Mason 
340634a38218SChris Mason 	if (!empty && push_items == left_nritems)
3407a429e513SChris Mason 		WARN_ON(1);
34085f39d397SChris Mason 
340900ec4c51SChris Mason 	/* push left to right */
34105f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
341134a38218SChris Mason 
34125f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
3413123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
34145f39d397SChris Mason 
341500ec4c51SChris Mason 	/* make room in the right data area */
34165f39d397SChris Mason 	data_end = leaf_data_end(root, right);
34175f39d397SChris Mason 	memmove_extent_buffer(right,
34185f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
34195f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
34205f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
34215f39d397SChris Mason 
342200ec4c51SChris Mason 	/* copy from the left data area */
34235f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
3424d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
3425d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
3426d6025579SChris Mason 		     push_space);
34275f39d397SChris Mason 
34285f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
34295f39d397SChris Mason 			      btrfs_item_nr_offset(0),
34300783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
34315f39d397SChris Mason 
343200ec4c51SChris Mason 	/* copy the items from left to right */
34335f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
34345f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
34350783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
343600ec4c51SChris Mason 
343700ec4c51SChris Mason 	/* update the item pointers */
34387518a238SChris Mason 	right_nritems += push_items;
34395f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
3440123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
34417518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
34425f39d397SChris Mason 		item = btrfs_item_nr(right, i);
3443cfed81a0SChris Mason 		push_space -= btrfs_token_item_size(right, item, &token);
3444cfed81a0SChris Mason 		btrfs_set_token_item_offset(right, item, push_space, &token);
3445db94535dSChris Mason 	}
3446db94535dSChris Mason 
34477518a238SChris Mason 	left_nritems -= push_items;
34485f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
344900ec4c51SChris Mason 
345034a38218SChris Mason 	if (left_nritems)
34515f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
3452f0486c68SYan, Zheng 	else
3453f0486c68SYan, Zheng 		clean_tree_block(trans, root, left);
3454f0486c68SYan, Zheng 
34555f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
3456a429e513SChris Mason 
34575f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
34585f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
3459d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
346002217ed2SChris Mason 
346100ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
34627518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
34637518a238SChris Mason 		path->slots[0] -= left_nritems;
3464925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
3465925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
3466925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
34675f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
34685f39d397SChris Mason 		path->nodes[0] = right;
346900ec4c51SChris Mason 		path->slots[1] += 1;
347000ec4c51SChris Mason 	} else {
3471925baeddSChris Mason 		btrfs_tree_unlock(right);
34725f39d397SChris Mason 		free_extent_buffer(right);
347300ec4c51SChris Mason 	}
347400ec4c51SChris Mason 	return 0;
3475925baeddSChris Mason 
3476925baeddSChris Mason out_unlock:
3477925baeddSChris Mason 	btrfs_tree_unlock(right);
3478925baeddSChris Mason 	free_extent_buffer(right);
3479925baeddSChris Mason 	return 1;
348000ec4c51SChris Mason }
3481925baeddSChris Mason 
348200ec4c51SChris Mason /*
348344871b1bSChris Mason  * push some data in the path leaf to the right, trying to free up at
348474123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
348544871b1bSChris Mason  *
348644871b1bSChris Mason  * returns 1 if the push failed because the other node didn't have enough
348744871b1bSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
348899d8f83cSChris Mason  *
348999d8f83cSChris Mason  * this will push starting from min_slot to the end of the leaf.  It won't
349099d8f83cSChris Mason  * push any slot lower than min_slot
349174123bd7SChris Mason  */
349244871b1bSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
349399d8f83cSChris Mason 			   *root, struct btrfs_path *path,
349499d8f83cSChris Mason 			   int min_data_size, int data_size,
349599d8f83cSChris Mason 			   int empty, u32 min_slot)
3496be0e5c09SChris Mason {
349744871b1bSChris Mason 	struct extent_buffer *left = path->nodes[0];
349844871b1bSChris Mason 	struct extent_buffer *right;
349944871b1bSChris Mason 	struct extent_buffer *upper;
350044871b1bSChris Mason 	int slot;
350144871b1bSChris Mason 	int free_space;
350244871b1bSChris Mason 	u32 left_nritems;
350344871b1bSChris Mason 	int ret;
350444871b1bSChris Mason 
350544871b1bSChris Mason 	if (!path->nodes[1])
350644871b1bSChris Mason 		return 1;
350744871b1bSChris Mason 
350844871b1bSChris Mason 	slot = path->slots[1];
350944871b1bSChris Mason 	upper = path->nodes[1];
351044871b1bSChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
351144871b1bSChris Mason 		return 1;
351244871b1bSChris Mason 
351344871b1bSChris Mason 	btrfs_assert_tree_locked(path->nodes[1]);
351444871b1bSChris Mason 
351544871b1bSChris Mason 	right = read_node_slot(root, upper, slot + 1);
351691ca338dSTsutomu Itoh 	if (right == NULL)
351791ca338dSTsutomu Itoh 		return 1;
351891ca338dSTsutomu Itoh 
351944871b1bSChris Mason 	btrfs_tree_lock(right);
352044871b1bSChris Mason 	btrfs_set_lock_blocking(right);
352144871b1bSChris Mason 
352244871b1bSChris Mason 	free_space = btrfs_leaf_free_space(root, right);
352344871b1bSChris Mason 	if (free_space < data_size)
352444871b1bSChris Mason 		goto out_unlock;
352544871b1bSChris Mason 
352644871b1bSChris Mason 	/* cow and double check */
352744871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
352844871b1bSChris Mason 			      slot + 1, &right);
352944871b1bSChris Mason 	if (ret)
353044871b1bSChris Mason 		goto out_unlock;
353144871b1bSChris Mason 
353244871b1bSChris Mason 	free_space = btrfs_leaf_free_space(root, right);
353344871b1bSChris Mason 	if (free_space < data_size)
353444871b1bSChris Mason 		goto out_unlock;
353544871b1bSChris Mason 
353644871b1bSChris Mason 	left_nritems = btrfs_header_nritems(left);
353744871b1bSChris Mason 	if (left_nritems == 0)
353844871b1bSChris Mason 		goto out_unlock;
353944871b1bSChris Mason 
354099d8f83cSChris Mason 	return __push_leaf_right(trans, root, path, min_data_size, empty,
354199d8f83cSChris Mason 				right, free_space, left_nritems, min_slot);
354244871b1bSChris Mason out_unlock:
354344871b1bSChris Mason 	btrfs_tree_unlock(right);
354444871b1bSChris Mason 	free_extent_buffer(right);
354544871b1bSChris Mason 	return 1;
354644871b1bSChris Mason }
354744871b1bSChris Mason 
354844871b1bSChris Mason /*
354944871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
355044871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
355199d8f83cSChris Mason  *
355299d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
355399d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us do all the
355499d8f83cSChris Mason  * items
355544871b1bSChris Mason  */
355644871b1bSChris Mason static noinline int __push_leaf_left(struct btrfs_trans_handle *trans,
355744871b1bSChris Mason 				     struct btrfs_root *root,
355844871b1bSChris Mason 				     struct btrfs_path *path, int data_size,
355944871b1bSChris Mason 				     int empty, struct extent_buffer *left,
356099d8f83cSChris Mason 				     int free_space, u32 right_nritems,
356199d8f83cSChris Mason 				     u32 max_slot)
356244871b1bSChris Mason {
35635f39d397SChris Mason 	struct btrfs_disk_key disk_key;
35645f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
3565be0e5c09SChris Mason 	int i;
3566be0e5c09SChris Mason 	int push_space = 0;
3567be0e5c09SChris Mason 	int push_items = 0;
35680783fcfcSChris Mason 	struct btrfs_item *item;
35697518a238SChris Mason 	u32 old_left_nritems;
357034a38218SChris Mason 	u32 nr;
3571aa5d6bedSChris Mason 	int ret = 0;
3572db94535dSChris Mason 	u32 this_item_size;
3573db94535dSChris Mason 	u32 old_left_item_size;
3574cfed81a0SChris Mason 	struct btrfs_map_token token;
3575cfed81a0SChris Mason 
3576cfed81a0SChris Mason 	btrfs_init_map_token(&token);
3577be0e5c09SChris Mason 
357834a38218SChris Mason 	if (empty)
357999d8f83cSChris Mason 		nr = min(right_nritems, max_slot);
358034a38218SChris Mason 	else
358199d8f83cSChris Mason 		nr = min(right_nritems - 1, max_slot);
358234a38218SChris Mason 
358334a38218SChris Mason 	for (i = 0; i < nr; i++) {
35845f39d397SChris Mason 		item = btrfs_item_nr(right, i);
3585db94535dSChris Mason 
358631840ae1SZheng Yan 		if (!empty && push_items > 0) {
358731840ae1SZheng Yan 			if (path->slots[0] < i)
358831840ae1SZheng Yan 				break;
358931840ae1SZheng Yan 			if (path->slots[0] == i) {
359031840ae1SZheng Yan 				int space = btrfs_leaf_free_space(root, right);
359131840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
359231840ae1SZheng Yan 					break;
359331840ae1SZheng Yan 			}
359431840ae1SZheng Yan 		}
359531840ae1SZheng Yan 
3596be0e5c09SChris Mason 		if (path->slots[0] == i)
359787b29b20SYan Zheng 			push_space += data_size;
3598db94535dSChris Mason 
3599db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
3600db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
3601be0e5c09SChris Mason 			break;
3602db94535dSChris Mason 
3603be0e5c09SChris Mason 		push_items++;
3604db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
3605be0e5c09SChris Mason 	}
3606db94535dSChris Mason 
3607be0e5c09SChris Mason 	if (push_items == 0) {
3608925baeddSChris Mason 		ret = 1;
3609925baeddSChris Mason 		goto out;
3610be0e5c09SChris Mason 	}
361134a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
3612a429e513SChris Mason 		WARN_ON(1);
36135f39d397SChris Mason 
3614be0e5c09SChris Mason 	/* push data from right to left */
36155f39d397SChris Mason 	copy_extent_buffer(left, right,
36165f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
36175f39d397SChris Mason 			   btrfs_item_nr_offset(0),
36185f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
36195f39d397SChris Mason 
3620123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
36215f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1);
36225f39d397SChris Mason 
36235f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
3624d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
3625123abc88SChris Mason 		     btrfs_leaf_data(right) +
36265f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
3627be0e5c09SChris Mason 		     push_space);
36285f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
362987b29b20SYan Zheng 	BUG_ON(old_left_nritems <= 0);
3630eb60ceacSChris Mason 
3631db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
3632be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
36335f39d397SChris Mason 		u32 ioff;
3634db94535dSChris Mason 
36355f39d397SChris Mason 		item = btrfs_item_nr(left, i);
3636db94535dSChris Mason 
3637cfed81a0SChris Mason 		ioff = btrfs_token_item_offset(left, item, &token);
3638cfed81a0SChris Mason 		btrfs_set_token_item_offset(left, item,
3639cfed81a0SChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size),
3640cfed81a0SChris Mason 		      &token);
3641be0e5c09SChris Mason 	}
36425f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
3643be0e5c09SChris Mason 
3644be0e5c09SChris Mason 	/* fixup right node */
364534a38218SChris Mason 	if (push_items > right_nritems) {
3646d397712bSChris Mason 		printk(KERN_CRIT "push items %d nr %u\n", push_items,
3647d397712bSChris Mason 		       right_nritems);
364834a38218SChris Mason 		WARN_ON(1);
364934a38218SChris Mason 	}
365034a38218SChris Mason 
365134a38218SChris Mason 	if (push_items < right_nritems) {
36525f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
3653123abc88SChris Mason 						  leaf_data_end(root, right);
36545f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
3655d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
3656d6025579SChris Mason 				      btrfs_leaf_data(right) +
3657123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
36585f39d397SChris Mason 
36595f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
36605f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
36615f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
36620783fcfcSChris Mason 			     sizeof(struct btrfs_item));
366334a38218SChris Mason 	}
3664eef1c494SYan 	right_nritems -= push_items;
3665eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
3666123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
36675f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
36685f39d397SChris Mason 		item = btrfs_item_nr(right, i);
3669db94535dSChris Mason 
3670cfed81a0SChris Mason 		push_space = push_space - btrfs_token_item_size(right,
3671cfed81a0SChris Mason 								item, &token);
3672cfed81a0SChris Mason 		btrfs_set_token_item_offset(right, item, push_space, &token);
3673db94535dSChris Mason 	}
3674eb60ceacSChris Mason 
36755f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
367634a38218SChris Mason 	if (right_nritems)
36775f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
3678f0486c68SYan, Zheng 	else
3679f0486c68SYan, Zheng 		clean_tree_block(trans, root, right);
3680098f59c2SChris Mason 
36815f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
3682143bede5SJeff Mahoney 	fixup_low_keys(trans, root, path, &disk_key, 1);
3683be0e5c09SChris Mason 
3684be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
3685be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
3686be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
3687925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
36885f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
36895f39d397SChris Mason 		path->nodes[0] = left;
3690be0e5c09SChris Mason 		path->slots[1] -= 1;
3691be0e5c09SChris Mason 	} else {
3692925baeddSChris Mason 		btrfs_tree_unlock(left);
36935f39d397SChris Mason 		free_extent_buffer(left);
3694be0e5c09SChris Mason 		path->slots[0] -= push_items;
3695be0e5c09SChris Mason 	}
3696eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
3697aa5d6bedSChris Mason 	return ret;
3698925baeddSChris Mason out:
3699925baeddSChris Mason 	btrfs_tree_unlock(left);
3700925baeddSChris Mason 	free_extent_buffer(left);
3701925baeddSChris Mason 	return ret;
3702be0e5c09SChris Mason }
3703be0e5c09SChris Mason 
370474123bd7SChris Mason /*
370544871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
370644871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
370799d8f83cSChris Mason  *
370899d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
370999d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us push all the
371099d8f83cSChris Mason  * items
371144871b1bSChris Mason  */
371244871b1bSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
371399d8f83cSChris Mason 			  *root, struct btrfs_path *path, int min_data_size,
371499d8f83cSChris Mason 			  int data_size, int empty, u32 max_slot)
371544871b1bSChris Mason {
371644871b1bSChris Mason 	struct extent_buffer *right = path->nodes[0];
371744871b1bSChris Mason 	struct extent_buffer *left;
371844871b1bSChris Mason 	int slot;
371944871b1bSChris Mason 	int free_space;
372044871b1bSChris Mason 	u32 right_nritems;
372144871b1bSChris Mason 	int ret = 0;
372244871b1bSChris Mason 
372344871b1bSChris Mason 	slot = path->slots[1];
372444871b1bSChris Mason 	if (slot == 0)
372544871b1bSChris Mason 		return 1;
372644871b1bSChris Mason 	if (!path->nodes[1])
372744871b1bSChris Mason 		return 1;
372844871b1bSChris Mason 
372944871b1bSChris Mason 	right_nritems = btrfs_header_nritems(right);
373044871b1bSChris Mason 	if (right_nritems == 0)
373144871b1bSChris Mason 		return 1;
373244871b1bSChris Mason 
373344871b1bSChris Mason 	btrfs_assert_tree_locked(path->nodes[1]);
373444871b1bSChris Mason 
373544871b1bSChris Mason 	left = read_node_slot(root, path->nodes[1], slot - 1);
373691ca338dSTsutomu Itoh 	if (left == NULL)
373791ca338dSTsutomu Itoh 		return 1;
373891ca338dSTsutomu Itoh 
373944871b1bSChris Mason 	btrfs_tree_lock(left);
374044871b1bSChris Mason 	btrfs_set_lock_blocking(left);
374144871b1bSChris Mason 
374244871b1bSChris Mason 	free_space = btrfs_leaf_free_space(root, left);
374344871b1bSChris Mason 	if (free_space < data_size) {
374444871b1bSChris Mason 		ret = 1;
374544871b1bSChris Mason 		goto out;
374644871b1bSChris Mason 	}
374744871b1bSChris Mason 
374844871b1bSChris Mason 	/* cow and double check */
374944871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, left,
375044871b1bSChris Mason 			      path->nodes[1], slot - 1, &left);
375144871b1bSChris Mason 	if (ret) {
375244871b1bSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
375379787eaaSJeff Mahoney 		if (ret == -ENOSPC)
375444871b1bSChris Mason 			ret = 1;
375544871b1bSChris Mason 		goto out;
375644871b1bSChris Mason 	}
375744871b1bSChris Mason 
375844871b1bSChris Mason 	free_space = btrfs_leaf_free_space(root, left);
375944871b1bSChris Mason 	if (free_space < data_size) {
376044871b1bSChris Mason 		ret = 1;
376144871b1bSChris Mason 		goto out;
376244871b1bSChris Mason 	}
376344871b1bSChris Mason 
376499d8f83cSChris Mason 	return __push_leaf_left(trans, root, path, min_data_size,
376599d8f83cSChris Mason 			       empty, left, free_space, right_nritems,
376699d8f83cSChris Mason 			       max_slot);
376744871b1bSChris Mason out:
376844871b1bSChris Mason 	btrfs_tree_unlock(left);
376944871b1bSChris Mason 	free_extent_buffer(left);
377044871b1bSChris Mason 	return ret;
377144871b1bSChris Mason }
377244871b1bSChris Mason 
377344871b1bSChris Mason /*
377474123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
377574123bd7SChris Mason  * available for the resulting leaf level of the path.
377674123bd7SChris Mason  */
3777143bede5SJeff Mahoney static noinline void copy_for_split(struct btrfs_trans_handle *trans,
3778e02119d5SChris Mason 				    struct btrfs_root *root,
377944871b1bSChris Mason 				    struct btrfs_path *path,
378044871b1bSChris Mason 				    struct extent_buffer *l,
378144871b1bSChris Mason 				    struct extent_buffer *right,
378244871b1bSChris Mason 				    int slot, int mid, int nritems)
3783be0e5c09SChris Mason {
3784be0e5c09SChris Mason 	int data_copy_size;
3785be0e5c09SChris Mason 	int rt_data_off;
3786be0e5c09SChris Mason 	int i;
3787d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
3788cfed81a0SChris Mason 	struct btrfs_map_token token;
3789cfed81a0SChris Mason 
3790cfed81a0SChris Mason 	btrfs_init_map_token(&token);
3791be0e5c09SChris Mason 
37925f39d397SChris Mason 	nritems = nritems - mid;
37935f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
37945f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
37955f39d397SChris Mason 
37965f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
37975f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
37985f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
37995f39d397SChris Mason 
38005f39d397SChris Mason 	copy_extent_buffer(right, l,
3801d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
3802123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
3803123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
380474123bd7SChris Mason 
38055f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
38065f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
38075f39d397SChris Mason 
38085f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
38095f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
3810db94535dSChris Mason 		u32 ioff;
3811db94535dSChris Mason 
3812cfed81a0SChris Mason 		ioff = btrfs_token_item_offset(right, item, &token);
3813cfed81a0SChris Mason 		btrfs_set_token_item_offset(right, item,
3814cfed81a0SChris Mason 					    ioff + rt_data_off, &token);
38150783fcfcSChris Mason 	}
381674123bd7SChris Mason 
38175f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
38185f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
3819143bede5SJeff Mahoney 	insert_ptr(trans, root, path, &disk_key, right->start,
3820c3e06965SJan Schmidt 		   path->slots[1] + 1, 1);
38215f39d397SChris Mason 
38225f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
38235f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
3824eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
38255f39d397SChris Mason 
3826be0e5c09SChris Mason 	if (mid <= slot) {
3827925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
38285f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
38295f39d397SChris Mason 		path->nodes[0] = right;
3830be0e5c09SChris Mason 		path->slots[0] -= mid;
3831be0e5c09SChris Mason 		path->slots[1] += 1;
3832925baeddSChris Mason 	} else {
3833925baeddSChris Mason 		btrfs_tree_unlock(right);
38345f39d397SChris Mason 		free_extent_buffer(right);
3835925baeddSChris Mason 	}
38365f39d397SChris Mason 
3837eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
383844871b1bSChris Mason }
383944871b1bSChris Mason 
384044871b1bSChris Mason /*
384199d8f83cSChris Mason  * double splits happen when we need to insert a big item in the middle
384299d8f83cSChris Mason  * of a leaf.  A double split can leave us with 3 mostly empty leaves:
384399d8f83cSChris Mason  * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ]
384499d8f83cSChris Mason  *          A                 B                 C
384599d8f83cSChris Mason  *
384699d8f83cSChris Mason  * We avoid this by trying to push the items on either side of our target
384799d8f83cSChris Mason  * into the adjacent leaves.  If all goes well we can avoid the double split
384899d8f83cSChris Mason  * completely.
384999d8f83cSChris Mason  */
385099d8f83cSChris Mason static noinline int push_for_double_split(struct btrfs_trans_handle *trans,
385199d8f83cSChris Mason 					  struct btrfs_root *root,
385299d8f83cSChris Mason 					  struct btrfs_path *path,
385399d8f83cSChris Mason 					  int data_size)
385499d8f83cSChris Mason {
385599d8f83cSChris Mason 	int ret;
385699d8f83cSChris Mason 	int progress = 0;
385799d8f83cSChris Mason 	int slot;
385899d8f83cSChris Mason 	u32 nritems;
385999d8f83cSChris Mason 
386099d8f83cSChris Mason 	slot = path->slots[0];
386199d8f83cSChris Mason 
386299d8f83cSChris Mason 	/*
386399d8f83cSChris Mason 	 * try to push all the items after our slot into the
386499d8f83cSChris Mason 	 * right leaf
386599d8f83cSChris Mason 	 */
386699d8f83cSChris Mason 	ret = push_leaf_right(trans, root, path, 1, data_size, 0, slot);
386799d8f83cSChris Mason 	if (ret < 0)
386899d8f83cSChris Mason 		return ret;
386999d8f83cSChris Mason 
387099d8f83cSChris Mason 	if (ret == 0)
387199d8f83cSChris Mason 		progress++;
387299d8f83cSChris Mason 
387399d8f83cSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
387499d8f83cSChris Mason 	/*
387599d8f83cSChris Mason 	 * our goal is to get our slot at the start or end of a leaf.  If
387699d8f83cSChris Mason 	 * we've done so we're done
387799d8f83cSChris Mason 	 */
387899d8f83cSChris Mason 	if (path->slots[0] == 0 || path->slots[0] == nritems)
387999d8f83cSChris Mason 		return 0;
388099d8f83cSChris Mason 
388199d8f83cSChris Mason 	if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size)
388299d8f83cSChris Mason 		return 0;
388399d8f83cSChris Mason 
388499d8f83cSChris Mason 	/* try to push all the items before our slot into the next leaf */
388599d8f83cSChris Mason 	slot = path->slots[0];
388699d8f83cSChris Mason 	ret = push_leaf_left(trans, root, path, 1, data_size, 0, slot);
388799d8f83cSChris Mason 	if (ret < 0)
388899d8f83cSChris Mason 		return ret;
388999d8f83cSChris Mason 
389099d8f83cSChris Mason 	if (ret == 0)
389199d8f83cSChris Mason 		progress++;
389299d8f83cSChris Mason 
389399d8f83cSChris Mason 	if (progress)
389499d8f83cSChris Mason 		return 0;
389599d8f83cSChris Mason 	return 1;
389699d8f83cSChris Mason }
389799d8f83cSChris Mason 
389899d8f83cSChris Mason /*
389944871b1bSChris Mason  * split the path's leaf in two, making sure there is at least data_size
390044871b1bSChris Mason  * available for the resulting leaf level of the path.
390144871b1bSChris Mason  *
390244871b1bSChris Mason  * returns 0 if all went well and < 0 on failure.
390344871b1bSChris Mason  */
390444871b1bSChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans,
390544871b1bSChris Mason 			       struct btrfs_root *root,
390644871b1bSChris Mason 			       struct btrfs_key *ins_key,
390744871b1bSChris Mason 			       struct btrfs_path *path, int data_size,
390844871b1bSChris Mason 			       int extend)
390944871b1bSChris Mason {
39105d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
391144871b1bSChris Mason 	struct extent_buffer *l;
391244871b1bSChris Mason 	u32 nritems;
391344871b1bSChris Mason 	int mid;
391444871b1bSChris Mason 	int slot;
391544871b1bSChris Mason 	struct extent_buffer *right;
391644871b1bSChris Mason 	int ret = 0;
391744871b1bSChris Mason 	int wret;
39185d4f98a2SYan Zheng 	int split;
391944871b1bSChris Mason 	int num_doubles = 0;
392099d8f83cSChris Mason 	int tried_avoid_double = 0;
392144871b1bSChris Mason 
3922a5719521SYan, Zheng 	l = path->nodes[0];
3923a5719521SYan, Zheng 	slot = path->slots[0];
3924a5719521SYan, Zheng 	if (extend && data_size + btrfs_item_size_nr(l, slot) +
3925a5719521SYan, Zheng 	    sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(root))
3926a5719521SYan, Zheng 		return -EOVERFLOW;
3927a5719521SYan, Zheng 
392844871b1bSChris Mason 	/* first try to make some room by pushing left and right */
392999d8f83cSChris Mason 	if (data_size) {
393099d8f83cSChris Mason 		wret = push_leaf_right(trans, root, path, data_size,
393199d8f83cSChris Mason 				       data_size, 0, 0);
393244871b1bSChris Mason 		if (wret < 0)
393344871b1bSChris Mason 			return wret;
393444871b1bSChris Mason 		if (wret) {
393599d8f83cSChris Mason 			wret = push_leaf_left(trans, root, path, data_size,
393699d8f83cSChris Mason 					      data_size, 0, (u32)-1);
393744871b1bSChris Mason 			if (wret < 0)
393844871b1bSChris Mason 				return wret;
393944871b1bSChris Mason 		}
394044871b1bSChris Mason 		l = path->nodes[0];
394144871b1bSChris Mason 
394244871b1bSChris Mason 		/* did the pushes work? */
394344871b1bSChris Mason 		if (btrfs_leaf_free_space(root, l) >= data_size)
394444871b1bSChris Mason 			return 0;
394544871b1bSChris Mason 	}
394644871b1bSChris Mason 
394744871b1bSChris Mason 	if (!path->nodes[1]) {
394844871b1bSChris Mason 		ret = insert_new_root(trans, root, path, 1);
394944871b1bSChris Mason 		if (ret)
395044871b1bSChris Mason 			return ret;
395144871b1bSChris Mason 	}
395244871b1bSChris Mason again:
39535d4f98a2SYan Zheng 	split = 1;
395444871b1bSChris Mason 	l = path->nodes[0];
395544871b1bSChris Mason 	slot = path->slots[0];
395644871b1bSChris Mason 	nritems = btrfs_header_nritems(l);
395744871b1bSChris Mason 	mid = (nritems + 1) / 2;
395844871b1bSChris Mason 
39595d4f98a2SYan Zheng 	if (mid <= slot) {
39605d4f98a2SYan Zheng 		if (nritems == 1 ||
39615d4f98a2SYan Zheng 		    leaf_space_used(l, mid, nritems - mid) + data_size >
39625d4f98a2SYan Zheng 			BTRFS_LEAF_DATA_SIZE(root)) {
39635d4f98a2SYan Zheng 			if (slot >= nritems) {
39645d4f98a2SYan Zheng 				split = 0;
39655d4f98a2SYan Zheng 			} else {
39665d4f98a2SYan Zheng 				mid = slot;
39675d4f98a2SYan Zheng 				if (mid != nritems &&
39685d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
39695d4f98a2SYan Zheng 				    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
397099d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
397199d8f83cSChris Mason 						goto push_for_double;
39725d4f98a2SYan Zheng 					split = 2;
39735d4f98a2SYan Zheng 				}
39745d4f98a2SYan Zheng 			}
39755d4f98a2SYan Zheng 		}
39765d4f98a2SYan Zheng 	} else {
39775d4f98a2SYan Zheng 		if (leaf_space_used(l, 0, mid) + data_size >
39785d4f98a2SYan Zheng 			BTRFS_LEAF_DATA_SIZE(root)) {
39795d4f98a2SYan Zheng 			if (!extend && data_size && slot == 0) {
39805d4f98a2SYan Zheng 				split = 0;
39815d4f98a2SYan Zheng 			} else if ((extend || !data_size) && slot == 0) {
39825d4f98a2SYan Zheng 				mid = 1;
39835d4f98a2SYan Zheng 			} else {
39845d4f98a2SYan Zheng 				mid = slot;
39855d4f98a2SYan Zheng 				if (mid != nritems &&
39865d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
39875d4f98a2SYan Zheng 				    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
398899d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
398999d8f83cSChris Mason 						goto push_for_double;
39905d4f98a2SYan Zheng 					split = 2 ;
39915d4f98a2SYan Zheng 				}
39925d4f98a2SYan Zheng 			}
39935d4f98a2SYan Zheng 		}
39945d4f98a2SYan Zheng 	}
39955d4f98a2SYan Zheng 
39965d4f98a2SYan Zheng 	if (split == 0)
39975d4f98a2SYan Zheng 		btrfs_cpu_key_to_disk(&disk_key, ins_key);
39985d4f98a2SYan Zheng 	else
39995d4f98a2SYan Zheng 		btrfs_item_key(l, &disk_key, mid);
40005d4f98a2SYan Zheng 
40015d4f98a2SYan Zheng 	right = btrfs_alloc_free_block(trans, root, root->leafsize, 0,
400244871b1bSChris Mason 					root->root_key.objectid,
40035581a51aSJan Schmidt 					&disk_key, 0, l->start, 0);
4004f0486c68SYan, Zheng 	if (IS_ERR(right))
400544871b1bSChris Mason 		return PTR_ERR(right);
4006f0486c68SYan, Zheng 
4007f0486c68SYan, Zheng 	root_add_used(root, root->leafsize);
400844871b1bSChris Mason 
400944871b1bSChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
401044871b1bSChris Mason 	btrfs_set_header_bytenr(right, right->start);
401144871b1bSChris Mason 	btrfs_set_header_generation(right, trans->transid);
40125d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(right, BTRFS_MIXED_BACKREF_REV);
401344871b1bSChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
401444871b1bSChris Mason 	btrfs_set_header_level(right, 0);
401544871b1bSChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
401644871b1bSChris Mason 			    (unsigned long)btrfs_header_fsid(right),
401744871b1bSChris Mason 			    BTRFS_FSID_SIZE);
401844871b1bSChris Mason 
401944871b1bSChris Mason 	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
402044871b1bSChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
402144871b1bSChris Mason 			    BTRFS_UUID_SIZE);
402244871b1bSChris Mason 
40235d4f98a2SYan Zheng 	if (split == 0) {
402444871b1bSChris Mason 		if (mid <= slot) {
402544871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
4026143bede5SJeff Mahoney 			insert_ptr(trans, root, path, &disk_key, right->start,
4027c3e06965SJan Schmidt 				   path->slots[1] + 1, 1);
402844871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
402944871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
403044871b1bSChris Mason 			path->nodes[0] = right;
403144871b1bSChris Mason 			path->slots[0] = 0;
403244871b1bSChris Mason 			path->slots[1] += 1;
403344871b1bSChris Mason 		} else {
403444871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
4035143bede5SJeff Mahoney 			insert_ptr(trans, root, path, &disk_key, right->start,
4036c3e06965SJan Schmidt 					  path->slots[1], 1);
403744871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
403844871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
403944871b1bSChris Mason 			path->nodes[0] = right;
404044871b1bSChris Mason 			path->slots[0] = 0;
4041143bede5SJeff Mahoney 			if (path->slots[1] == 0)
4042143bede5SJeff Mahoney 				fixup_low_keys(trans, root, path,
4043143bede5SJeff Mahoney 					       &disk_key, 1);
40445d4f98a2SYan Zheng 		}
404544871b1bSChris Mason 		btrfs_mark_buffer_dirty(right);
404644871b1bSChris Mason 		return ret;
404744871b1bSChris Mason 	}
404844871b1bSChris Mason 
4049143bede5SJeff Mahoney 	copy_for_split(trans, root, path, l, right, slot, mid, nritems);
405044871b1bSChris Mason 
40515d4f98a2SYan Zheng 	if (split == 2) {
4052cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
4053cc0c5538SChris Mason 		num_doubles++;
4054cc0c5538SChris Mason 		goto again;
40553326d1b0SChris Mason 	}
405644871b1bSChris Mason 
4057143bede5SJeff Mahoney 	return 0;
405899d8f83cSChris Mason 
405999d8f83cSChris Mason push_for_double:
406099d8f83cSChris Mason 	push_for_double_split(trans, root, path, data_size);
406199d8f83cSChris Mason 	tried_avoid_double = 1;
406299d8f83cSChris Mason 	if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size)
406399d8f83cSChris Mason 		return 0;
406499d8f83cSChris Mason 	goto again;
4065be0e5c09SChris Mason }
4066be0e5c09SChris Mason 
4067ad48fd75SYan, Zheng static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans,
4068ad48fd75SYan, Zheng 					 struct btrfs_root *root,
4069ad48fd75SYan, Zheng 					 struct btrfs_path *path, int ins_len)
4070ad48fd75SYan, Zheng {
4071ad48fd75SYan, Zheng 	struct btrfs_key key;
4072ad48fd75SYan, Zheng 	struct extent_buffer *leaf;
4073ad48fd75SYan, Zheng 	struct btrfs_file_extent_item *fi;
4074ad48fd75SYan, Zheng 	u64 extent_len = 0;
4075ad48fd75SYan, Zheng 	u32 item_size;
4076ad48fd75SYan, Zheng 	int ret;
4077ad48fd75SYan, Zheng 
4078ad48fd75SYan, Zheng 	leaf = path->nodes[0];
4079ad48fd75SYan, Zheng 	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
4080ad48fd75SYan, Zheng 
4081ad48fd75SYan, Zheng 	BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY &&
4082ad48fd75SYan, Zheng 	       key.type != BTRFS_EXTENT_CSUM_KEY);
4083ad48fd75SYan, Zheng 
4084ad48fd75SYan, Zheng 	if (btrfs_leaf_free_space(root, leaf) >= ins_len)
4085ad48fd75SYan, Zheng 		return 0;
4086ad48fd75SYan, Zheng 
4087ad48fd75SYan, Zheng 	item_size = btrfs_item_size_nr(leaf, path->slots[0]);
4088ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
4089ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
4090ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
4091ad48fd75SYan, Zheng 		extent_len = btrfs_file_extent_num_bytes(leaf, fi);
4092ad48fd75SYan, Zheng 	}
4093b3b4aa74SDavid Sterba 	btrfs_release_path(path);
4094ad48fd75SYan, Zheng 
4095ad48fd75SYan, Zheng 	path->keep_locks = 1;
4096ad48fd75SYan, Zheng 	path->search_for_split = 1;
4097ad48fd75SYan, Zheng 	ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
4098ad48fd75SYan, Zheng 	path->search_for_split = 0;
4099ad48fd75SYan, Zheng 	if (ret < 0)
4100ad48fd75SYan, Zheng 		goto err;
4101ad48fd75SYan, Zheng 
4102ad48fd75SYan, Zheng 	ret = -EAGAIN;
4103ad48fd75SYan, Zheng 	leaf = path->nodes[0];
4104ad48fd75SYan, Zheng 	/* if our item isn't there or got smaller, return now */
4105ad48fd75SYan, Zheng 	if (ret > 0 || item_size != btrfs_item_size_nr(leaf, path->slots[0]))
4106ad48fd75SYan, Zheng 		goto err;
4107ad48fd75SYan, Zheng 
4108109f6aefSChris Mason 	/* the leaf has  changed, it now has room.  return now */
4109109f6aefSChris Mason 	if (btrfs_leaf_free_space(root, path->nodes[0]) >= ins_len)
4110109f6aefSChris Mason 		goto err;
4111109f6aefSChris Mason 
4112ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
4113ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
4114ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
4115ad48fd75SYan, Zheng 		if (extent_len != btrfs_file_extent_num_bytes(leaf, fi))
4116ad48fd75SYan, Zheng 			goto err;
4117ad48fd75SYan, Zheng 	}
4118ad48fd75SYan, Zheng 
4119ad48fd75SYan, Zheng 	btrfs_set_path_blocking(path);
4120ad48fd75SYan, Zheng 	ret = split_leaf(trans, root, &key, path, ins_len, 1);
4121f0486c68SYan, Zheng 	if (ret)
4122f0486c68SYan, Zheng 		goto err;
4123ad48fd75SYan, Zheng 
4124ad48fd75SYan, Zheng 	path->keep_locks = 0;
4125ad48fd75SYan, Zheng 	btrfs_unlock_up_safe(path, 1);
4126ad48fd75SYan, Zheng 	return 0;
4127ad48fd75SYan, Zheng err:
4128ad48fd75SYan, Zheng 	path->keep_locks = 0;
4129ad48fd75SYan, Zheng 	return ret;
4130ad48fd75SYan, Zheng }
4131ad48fd75SYan, Zheng 
4132ad48fd75SYan, Zheng static noinline int split_item(struct btrfs_trans_handle *trans,
4133459931ecSChris Mason 			       struct btrfs_root *root,
4134459931ecSChris Mason 			       struct btrfs_path *path,
4135459931ecSChris Mason 			       struct btrfs_key *new_key,
4136459931ecSChris Mason 			       unsigned long split_offset)
4137459931ecSChris Mason {
4138459931ecSChris Mason 	struct extent_buffer *leaf;
4139459931ecSChris Mason 	struct btrfs_item *item;
4140459931ecSChris Mason 	struct btrfs_item *new_item;
4141459931ecSChris Mason 	int slot;
4142ad48fd75SYan, Zheng 	char *buf;
4143459931ecSChris Mason 	u32 nritems;
4144ad48fd75SYan, Zheng 	u32 item_size;
4145459931ecSChris Mason 	u32 orig_offset;
4146459931ecSChris Mason 	struct btrfs_disk_key disk_key;
4147459931ecSChris Mason 
4148459931ecSChris Mason 	leaf = path->nodes[0];
4149b9473439SChris Mason 	BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item));
4150b9473439SChris Mason 
4151b4ce94deSChris Mason 	btrfs_set_path_blocking(path);
4152b4ce94deSChris Mason 
4153459931ecSChris Mason 	item = btrfs_item_nr(leaf, path->slots[0]);
4154459931ecSChris Mason 	orig_offset = btrfs_item_offset(leaf, item);
4155459931ecSChris Mason 	item_size = btrfs_item_size(leaf, item);
4156459931ecSChris Mason 
4157459931ecSChris Mason 	buf = kmalloc(item_size, GFP_NOFS);
4158ad48fd75SYan, Zheng 	if (!buf)
4159ad48fd75SYan, Zheng 		return -ENOMEM;
4160ad48fd75SYan, Zheng 
4161459931ecSChris Mason 	read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf,
4162459931ecSChris Mason 			    path->slots[0]), item_size);
4163ad48fd75SYan, Zheng 
4164459931ecSChris Mason 	slot = path->slots[0] + 1;
4165459931ecSChris Mason 	nritems = btrfs_header_nritems(leaf);
4166459931ecSChris Mason 	if (slot != nritems) {
4167459931ecSChris Mason 		/* shift the items */
4168459931ecSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
4169459931ecSChris Mason 				btrfs_item_nr_offset(slot),
4170459931ecSChris Mason 				(nritems - slot) * sizeof(struct btrfs_item));
4171459931ecSChris Mason 	}
4172459931ecSChris Mason 
4173459931ecSChris Mason 	btrfs_cpu_key_to_disk(&disk_key, new_key);
4174459931ecSChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
4175459931ecSChris Mason 
4176459931ecSChris Mason 	new_item = btrfs_item_nr(leaf, slot);
4177459931ecSChris Mason 
4178459931ecSChris Mason 	btrfs_set_item_offset(leaf, new_item, orig_offset);
4179459931ecSChris Mason 	btrfs_set_item_size(leaf, new_item, item_size - split_offset);
4180459931ecSChris Mason 
4181459931ecSChris Mason 	btrfs_set_item_offset(leaf, item,
4182459931ecSChris Mason 			      orig_offset + item_size - split_offset);
4183459931ecSChris Mason 	btrfs_set_item_size(leaf, item, split_offset);
4184459931ecSChris Mason 
4185459931ecSChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
4186459931ecSChris Mason 
4187459931ecSChris Mason 	/* write the data for the start of the original item */
4188459931ecSChris Mason 	write_extent_buffer(leaf, buf,
4189459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, path->slots[0]),
4190459931ecSChris Mason 			    split_offset);
4191459931ecSChris Mason 
4192459931ecSChris Mason 	/* write the data for the new item */
4193459931ecSChris Mason 	write_extent_buffer(leaf, buf + split_offset,
4194459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, slot),
4195459931ecSChris Mason 			    item_size - split_offset);
4196459931ecSChris Mason 	btrfs_mark_buffer_dirty(leaf);
4197459931ecSChris Mason 
4198ad48fd75SYan, Zheng 	BUG_ON(btrfs_leaf_free_space(root, leaf) < 0);
4199459931ecSChris Mason 	kfree(buf);
4200ad48fd75SYan, Zheng 	return 0;
4201ad48fd75SYan, Zheng }
4202ad48fd75SYan, Zheng 
4203ad48fd75SYan, Zheng /*
4204ad48fd75SYan, Zheng  * This function splits a single item into two items,
4205ad48fd75SYan, Zheng  * giving 'new_key' to the new item and splitting the
4206ad48fd75SYan, Zheng  * old one at split_offset (from the start of the item).
4207ad48fd75SYan, Zheng  *
4208ad48fd75SYan, Zheng  * The path may be released by this operation.  After
4209ad48fd75SYan, Zheng  * the split, the path is pointing to the old item.  The
4210ad48fd75SYan, Zheng  * new item is going to be in the same node as the old one.
4211ad48fd75SYan, Zheng  *
4212ad48fd75SYan, Zheng  * Note, the item being split must be smaller enough to live alone on
4213ad48fd75SYan, Zheng  * a tree block with room for one extra struct btrfs_item
4214ad48fd75SYan, Zheng  *
4215ad48fd75SYan, Zheng  * This allows us to split the item in place, keeping a lock on the
4216ad48fd75SYan, Zheng  * leaf the entire time.
4217ad48fd75SYan, Zheng  */
4218ad48fd75SYan, Zheng int btrfs_split_item(struct btrfs_trans_handle *trans,
4219ad48fd75SYan, Zheng 		     struct btrfs_root *root,
4220ad48fd75SYan, Zheng 		     struct btrfs_path *path,
4221ad48fd75SYan, Zheng 		     struct btrfs_key *new_key,
4222ad48fd75SYan, Zheng 		     unsigned long split_offset)
4223ad48fd75SYan, Zheng {
4224ad48fd75SYan, Zheng 	int ret;
4225ad48fd75SYan, Zheng 	ret = setup_leaf_for_split(trans, root, path,
4226ad48fd75SYan, Zheng 				   sizeof(struct btrfs_item));
4227ad48fd75SYan, Zheng 	if (ret)
4228459931ecSChris Mason 		return ret;
4229ad48fd75SYan, Zheng 
4230ad48fd75SYan, Zheng 	ret = split_item(trans, root, path, new_key, split_offset);
4231ad48fd75SYan, Zheng 	return ret;
4232ad48fd75SYan, Zheng }
4233ad48fd75SYan, Zheng 
4234ad48fd75SYan, Zheng /*
4235ad48fd75SYan, Zheng  * This function duplicate a item, giving 'new_key' to the new item.
4236ad48fd75SYan, Zheng  * It guarantees both items live in the same tree leaf and the new item
4237ad48fd75SYan, Zheng  * is contiguous with the original item.
4238ad48fd75SYan, Zheng  *
4239ad48fd75SYan, Zheng  * This allows us to split file extent in place, keeping a lock on the
4240ad48fd75SYan, Zheng  * leaf the entire time.
4241ad48fd75SYan, Zheng  */
4242ad48fd75SYan, Zheng int btrfs_duplicate_item(struct btrfs_trans_handle *trans,
4243ad48fd75SYan, Zheng 			 struct btrfs_root *root,
4244ad48fd75SYan, Zheng 			 struct btrfs_path *path,
4245ad48fd75SYan, Zheng 			 struct btrfs_key *new_key)
4246ad48fd75SYan, Zheng {
4247ad48fd75SYan, Zheng 	struct extent_buffer *leaf;
4248ad48fd75SYan, Zheng 	int ret;
4249ad48fd75SYan, Zheng 	u32 item_size;
4250ad48fd75SYan, Zheng 
4251ad48fd75SYan, Zheng 	leaf = path->nodes[0];
4252ad48fd75SYan, Zheng 	item_size = btrfs_item_size_nr(leaf, path->slots[0]);
4253ad48fd75SYan, Zheng 	ret = setup_leaf_for_split(trans, root, path,
4254ad48fd75SYan, Zheng 				   item_size + sizeof(struct btrfs_item));
4255ad48fd75SYan, Zheng 	if (ret)
4256ad48fd75SYan, Zheng 		return ret;
4257ad48fd75SYan, Zheng 
4258ad48fd75SYan, Zheng 	path->slots[0]++;
4259143bede5SJeff Mahoney 	setup_items_for_insert(trans, root, path, new_key, &item_size,
4260ad48fd75SYan, Zheng 			       item_size, item_size +
4261ad48fd75SYan, Zheng 			       sizeof(struct btrfs_item), 1);
4262ad48fd75SYan, Zheng 	leaf = path->nodes[0];
4263ad48fd75SYan, Zheng 	memcpy_extent_buffer(leaf,
4264ad48fd75SYan, Zheng 			     btrfs_item_ptr_offset(leaf, path->slots[0]),
4265ad48fd75SYan, Zheng 			     btrfs_item_ptr_offset(leaf, path->slots[0] - 1),
4266ad48fd75SYan, Zheng 			     item_size);
4267ad48fd75SYan, Zheng 	return 0;
4268459931ecSChris Mason }
4269459931ecSChris Mason 
4270459931ecSChris Mason /*
4271d352ac68SChris Mason  * make the item pointed to by the path smaller.  new_size indicates
4272d352ac68SChris Mason  * how small to make it, and from_end tells us if we just chop bytes
4273d352ac68SChris Mason  * off the end of the item or if we shift the item to chop bytes off
4274d352ac68SChris Mason  * the front.
4275d352ac68SChris Mason  */
4276143bede5SJeff Mahoney void btrfs_truncate_item(struct btrfs_trans_handle *trans,
4277b18c6685SChris Mason 			 struct btrfs_root *root,
4278b18c6685SChris Mason 			 struct btrfs_path *path,
4279179e29e4SChris Mason 			 u32 new_size, int from_end)
4280b18c6685SChris Mason {
4281b18c6685SChris Mason 	int slot;
42825f39d397SChris Mason 	struct extent_buffer *leaf;
42835f39d397SChris Mason 	struct btrfs_item *item;
4284b18c6685SChris Mason 	u32 nritems;
4285b18c6685SChris Mason 	unsigned int data_end;
4286b18c6685SChris Mason 	unsigned int old_data_start;
4287b18c6685SChris Mason 	unsigned int old_size;
4288b18c6685SChris Mason 	unsigned int size_diff;
4289b18c6685SChris Mason 	int i;
4290cfed81a0SChris Mason 	struct btrfs_map_token token;
4291cfed81a0SChris Mason 
4292cfed81a0SChris Mason 	btrfs_init_map_token(&token);
4293b18c6685SChris Mason 
42945f39d397SChris Mason 	leaf = path->nodes[0];
4295179e29e4SChris Mason 	slot = path->slots[0];
4296179e29e4SChris Mason 
4297179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
4298179e29e4SChris Mason 	if (old_size == new_size)
4299143bede5SJeff Mahoney 		return;
4300b18c6685SChris Mason 
43015f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
4302b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
4303b18c6685SChris Mason 
43045f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
4305179e29e4SChris Mason 
4306b18c6685SChris Mason 	size_diff = old_size - new_size;
4307b18c6685SChris Mason 
4308b18c6685SChris Mason 	BUG_ON(slot < 0);
4309b18c6685SChris Mason 	BUG_ON(slot >= nritems);
4310b18c6685SChris Mason 
4311b18c6685SChris Mason 	/*
4312b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
4313b18c6685SChris Mason 	 */
4314b18c6685SChris Mason 	/* first correct the data pointers */
4315b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
43165f39d397SChris Mason 		u32 ioff;
43175f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
4318db94535dSChris Mason 
4319cfed81a0SChris Mason 		ioff = btrfs_token_item_offset(leaf, item, &token);
4320cfed81a0SChris Mason 		btrfs_set_token_item_offset(leaf, item,
4321cfed81a0SChris Mason 					    ioff + size_diff, &token);
4322b18c6685SChris Mason 	}
4323db94535dSChris Mason 
4324b18c6685SChris Mason 	/* shift the data */
4325179e29e4SChris Mason 	if (from_end) {
43265f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
4327b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
4328b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
4329179e29e4SChris Mason 	} else {
4330179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
4331179e29e4SChris Mason 		u64 offset;
4332179e29e4SChris Mason 
4333179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
4334179e29e4SChris Mason 
4335179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
4336179e29e4SChris Mason 			unsigned long ptr;
4337179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
4338179e29e4SChris Mason 
4339179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
4340179e29e4SChris Mason 					    struct btrfs_file_extent_item);
4341179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
4342179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
4343179e29e4SChris Mason 
4344179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
4345179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
4346179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
4347179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
4348179e29e4SChris Mason 				      (unsigned long)fi,
4349179e29e4SChris Mason 				      offsetof(struct btrfs_file_extent_item,
4350179e29e4SChris Mason 						 disk_bytenr));
4351179e29e4SChris Mason 			}
4352179e29e4SChris Mason 		}
4353179e29e4SChris Mason 
4354179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
4355179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
4356179e29e4SChris Mason 			      data_end, old_data_start - data_end);
4357179e29e4SChris Mason 
4358179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
4359179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
4360179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
4361179e29e4SChris Mason 		if (slot == 0)
4362179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
4363179e29e4SChris Mason 	}
43645f39d397SChris Mason 
43655f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
43665f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
43675f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
4368b18c6685SChris Mason 
43695f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
43705f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
4371b18c6685SChris Mason 		BUG();
43725f39d397SChris Mason 	}
4373b18c6685SChris Mason }
4374b18c6685SChris Mason 
4375d352ac68SChris Mason /*
4376d352ac68SChris Mason  * make the item pointed to by the path bigger, data_size is the new size.
4377d352ac68SChris Mason  */
4378143bede5SJeff Mahoney void btrfs_extend_item(struct btrfs_trans_handle *trans,
43795f39d397SChris Mason 		       struct btrfs_root *root, struct btrfs_path *path,
43805f39d397SChris Mason 		       u32 data_size)
43816567e837SChris Mason {
43826567e837SChris Mason 	int slot;
43835f39d397SChris Mason 	struct extent_buffer *leaf;
43845f39d397SChris Mason 	struct btrfs_item *item;
43856567e837SChris Mason 	u32 nritems;
43866567e837SChris Mason 	unsigned int data_end;
43876567e837SChris Mason 	unsigned int old_data;
43886567e837SChris Mason 	unsigned int old_size;
43896567e837SChris Mason 	int i;
4390cfed81a0SChris Mason 	struct btrfs_map_token token;
4391cfed81a0SChris Mason 
4392cfed81a0SChris Mason 	btrfs_init_map_token(&token);
43936567e837SChris Mason 
43945f39d397SChris Mason 	leaf = path->nodes[0];
43956567e837SChris Mason 
43965f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
43976567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
43986567e837SChris Mason 
43995f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
44005f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
44016567e837SChris Mason 		BUG();
44025f39d397SChris Mason 	}
44036567e837SChris Mason 	slot = path->slots[0];
44045f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
44056567e837SChris Mason 
44066567e837SChris Mason 	BUG_ON(slot < 0);
44073326d1b0SChris Mason 	if (slot >= nritems) {
44083326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
4409d397712bSChris Mason 		printk(KERN_CRIT "slot %d too large, nritems %d\n",
4410d397712bSChris Mason 		       slot, nritems);
44113326d1b0SChris Mason 		BUG_ON(1);
44123326d1b0SChris Mason 	}
44136567e837SChris Mason 
44146567e837SChris Mason 	/*
44156567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
44166567e837SChris Mason 	 */
44176567e837SChris Mason 	/* first correct the data pointers */
44186567e837SChris Mason 	for (i = slot; i < nritems; i++) {
44195f39d397SChris Mason 		u32 ioff;
44205f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
4421db94535dSChris Mason 
4422cfed81a0SChris Mason 		ioff = btrfs_token_item_offset(leaf, item, &token);
4423cfed81a0SChris Mason 		btrfs_set_token_item_offset(leaf, item,
4424cfed81a0SChris Mason 					    ioff - data_size, &token);
44256567e837SChris Mason 	}
44265f39d397SChris Mason 
44276567e837SChris Mason 	/* shift the data */
44285f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
44296567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
44306567e837SChris Mason 		      data_end, old_data - data_end);
44315f39d397SChris Mason 
44326567e837SChris Mason 	data_end = old_data;
44335f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
44345f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
44355f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
44365f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
44376567e837SChris Mason 
44385f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
44395f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
44406567e837SChris Mason 		BUG();
44415f39d397SChris Mason 	}
44426567e837SChris Mason }
44436567e837SChris Mason 
444474123bd7SChris Mason /*
444544871b1bSChris Mason  * this is a helper for btrfs_insert_empty_items, the main goal here is
444644871b1bSChris Mason  * to save stack depth by doing the bulk of the work in a function
444744871b1bSChris Mason  * that doesn't call btrfs_search_slot
444874123bd7SChris Mason  */
4449143bede5SJeff Mahoney void setup_items_for_insert(struct btrfs_trans_handle *trans,
445044871b1bSChris Mason 			    struct btrfs_root *root, struct btrfs_path *path,
44519c58309dSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
445244871b1bSChris Mason 			    u32 total_data, u32 total_size, int nr)
4453be0e5c09SChris Mason {
44545f39d397SChris Mason 	struct btrfs_item *item;
44559c58309dSChris Mason 	int i;
44567518a238SChris Mason 	u32 nritems;
4457be0e5c09SChris Mason 	unsigned int data_end;
4458e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
445944871b1bSChris Mason 	struct extent_buffer *leaf;
446044871b1bSChris Mason 	int slot;
4461cfed81a0SChris Mason 	struct btrfs_map_token token;
4462cfed81a0SChris Mason 
4463cfed81a0SChris Mason 	btrfs_init_map_token(&token);
4464e2fa7227SChris Mason 
44655f39d397SChris Mason 	leaf = path->nodes[0];
446644871b1bSChris Mason 	slot = path->slots[0];
446774123bd7SChris Mason 
44685f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
4469123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
4470eb60ceacSChris Mason 
4471f25956ccSChris Mason 	if (btrfs_leaf_free_space(root, leaf) < total_size) {
44723326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
4473d397712bSChris Mason 		printk(KERN_CRIT "not enough freespace need %u have %d\n",
44749c58309dSChris Mason 		       total_size, btrfs_leaf_free_space(root, leaf));
4475be0e5c09SChris Mason 		BUG();
4476d4dbff95SChris Mason 	}
44775f39d397SChris Mason 
4478be0e5c09SChris Mason 	if (slot != nritems) {
44795f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
4480be0e5c09SChris Mason 
44815f39d397SChris Mason 		if (old_data < data_end) {
44825f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4483d397712bSChris Mason 			printk(KERN_CRIT "slot %d old_data %d data_end %d\n",
44845f39d397SChris Mason 			       slot, old_data, data_end);
44855f39d397SChris Mason 			BUG_ON(1);
44865f39d397SChris Mason 		}
4487be0e5c09SChris Mason 		/*
4488be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
4489be0e5c09SChris Mason 		 */
4490be0e5c09SChris Mason 		/* first correct the data pointers */
44910783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
44925f39d397SChris Mason 			u32 ioff;
4493db94535dSChris Mason 
44945f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
4495cfed81a0SChris Mason 			ioff = btrfs_token_item_offset(leaf, item, &token);
4496cfed81a0SChris Mason 			btrfs_set_token_item_offset(leaf, item,
4497cfed81a0SChris Mason 						    ioff - total_data, &token);
44980783fcfcSChris Mason 		}
4499be0e5c09SChris Mason 		/* shift the items */
45009c58309dSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
45015f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
45020783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
4503be0e5c09SChris Mason 
4504be0e5c09SChris Mason 		/* shift the data */
45055f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
45069c58309dSChris Mason 			      data_end - total_data, btrfs_leaf_data(leaf) +
4507be0e5c09SChris Mason 			      data_end, old_data - data_end);
4508be0e5c09SChris Mason 		data_end = old_data;
4509be0e5c09SChris Mason 	}
45105f39d397SChris Mason 
451162e2749eSChris Mason 	/* setup the item for the new data */
45129c58309dSChris Mason 	for (i = 0; i < nr; i++) {
45139c58309dSChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
45149c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
45159c58309dSChris Mason 		item = btrfs_item_nr(leaf, slot + i);
4516cfed81a0SChris Mason 		btrfs_set_token_item_offset(leaf, item,
4517cfed81a0SChris Mason 					    data_end - data_size[i], &token);
45189c58309dSChris Mason 		data_end -= data_size[i];
4519cfed81a0SChris Mason 		btrfs_set_token_item_size(leaf, item, data_size[i], &token);
45209c58309dSChris Mason 	}
452144871b1bSChris Mason 
45229c58309dSChris Mason 	btrfs_set_header_nritems(leaf, nritems + nr);
4523aa5d6bedSChris Mason 
45245a01a2e3SChris Mason 	if (slot == 0) {
45255a01a2e3SChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
4526143bede5SJeff Mahoney 		fixup_low_keys(trans, root, path, &disk_key, 1);
45275a01a2e3SChris Mason 	}
4528b9473439SChris Mason 	btrfs_unlock_up_safe(path, 1);
4529b9473439SChris Mason 	btrfs_mark_buffer_dirty(leaf);
4530aa5d6bedSChris Mason 
45315f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
45325f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
4533be0e5c09SChris Mason 		BUG();
45345f39d397SChris Mason 	}
453544871b1bSChris Mason }
453644871b1bSChris Mason 
453744871b1bSChris Mason /*
453844871b1bSChris Mason  * Given a key and some data, insert items into the tree.
453944871b1bSChris Mason  * This does all the path init required, making room in the tree if needed.
454044871b1bSChris Mason  */
454144871b1bSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
454244871b1bSChris Mason 			    struct btrfs_root *root,
454344871b1bSChris Mason 			    struct btrfs_path *path,
454444871b1bSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
454544871b1bSChris Mason 			    int nr)
454644871b1bSChris Mason {
454744871b1bSChris Mason 	int ret = 0;
454844871b1bSChris Mason 	int slot;
454944871b1bSChris Mason 	int i;
455044871b1bSChris Mason 	u32 total_size = 0;
455144871b1bSChris Mason 	u32 total_data = 0;
455244871b1bSChris Mason 
455344871b1bSChris Mason 	for (i = 0; i < nr; i++)
455444871b1bSChris Mason 		total_data += data_size[i];
455544871b1bSChris Mason 
455644871b1bSChris Mason 	total_size = total_data + (nr * sizeof(struct btrfs_item));
455744871b1bSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
455844871b1bSChris Mason 	if (ret == 0)
455944871b1bSChris Mason 		return -EEXIST;
456044871b1bSChris Mason 	if (ret < 0)
4561143bede5SJeff Mahoney 		return ret;
456244871b1bSChris Mason 
456344871b1bSChris Mason 	slot = path->slots[0];
456444871b1bSChris Mason 	BUG_ON(slot < 0);
456544871b1bSChris Mason 
4566143bede5SJeff Mahoney 	setup_items_for_insert(trans, root, path, cpu_key, data_size,
456744871b1bSChris Mason 			       total_data, total_size, nr);
4568143bede5SJeff Mahoney 	return 0;
456962e2749eSChris Mason }
457062e2749eSChris Mason 
457162e2749eSChris Mason /*
457262e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
457362e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
457462e2749eSChris Mason  */
4575e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
4576e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
4577e089f05cSChris Mason 		      data_size)
457862e2749eSChris Mason {
457962e2749eSChris Mason 	int ret = 0;
45802c90e5d6SChris Mason 	struct btrfs_path *path;
45815f39d397SChris Mason 	struct extent_buffer *leaf;
45825f39d397SChris Mason 	unsigned long ptr;
458362e2749eSChris Mason 
45842c90e5d6SChris Mason 	path = btrfs_alloc_path();
4585db5b493aSTsutomu Itoh 	if (!path)
4586db5b493aSTsutomu Itoh 		return -ENOMEM;
45872c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
458862e2749eSChris Mason 	if (!ret) {
45895f39d397SChris Mason 		leaf = path->nodes[0];
45905f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
45915f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
45925f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
459362e2749eSChris Mason 	}
45942c90e5d6SChris Mason 	btrfs_free_path(path);
4595aa5d6bedSChris Mason 	return ret;
4596be0e5c09SChris Mason }
4597be0e5c09SChris Mason 
459874123bd7SChris Mason /*
45995de08d7dSChris Mason  * delete the pointer from a given node.
460074123bd7SChris Mason  *
4601d352ac68SChris Mason  * the tree should have been previously balanced so the deletion does not
4602d352ac68SChris Mason  * empty a node.
460374123bd7SChris Mason  */
4604143bede5SJeff Mahoney static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
4605f3ea38daSJan Schmidt 		    struct btrfs_path *path, int level, int slot,
4606f3ea38daSJan Schmidt 		    int tree_mod_log)
4607be0e5c09SChris Mason {
46085f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
46097518a238SChris Mason 	u32 nritems;
4610f3ea38daSJan Schmidt 	int ret;
4611be0e5c09SChris Mason 
4612*6a7a665dSLiu Bo 	if (tree_mod_log && level) {
4613*6a7a665dSLiu Bo 		ret = tree_mod_log_insert_key(root->fs_info, parent, slot,
4614*6a7a665dSLiu Bo 					      MOD_LOG_KEY_REMOVE);
4615*6a7a665dSLiu Bo 		BUG_ON(ret < 0);
4616*6a7a665dSLiu Bo 	}
4617*6a7a665dSLiu Bo 
46185f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
4619be0e5c09SChris Mason 	if (slot != nritems - 1) {
4620f3ea38daSJan Schmidt 		if (tree_mod_log && level)
4621f3ea38daSJan Schmidt 			tree_mod_log_eb_move(root->fs_info, parent, slot,
4622f3ea38daSJan Schmidt 					     slot + 1, nritems - slot - 1);
46235f39d397SChris Mason 		memmove_extent_buffer(parent,
46245f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
46255f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
4626d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
4627d6025579SChris Mason 			      (nritems - slot - 1));
4628be0e5c09SChris Mason 	}
4629f3ea38daSJan Schmidt 
46307518a238SChris Mason 	nritems--;
46315f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
46327518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
46335f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
4634eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
46355f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
4636bb803951SChris Mason 	} else if (slot == 0) {
46375f39d397SChris Mason 		struct btrfs_disk_key disk_key;
46385f39d397SChris Mason 
46395f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
4640143bede5SJeff Mahoney 		fixup_low_keys(trans, root, path, &disk_key, level + 1);
4641be0e5c09SChris Mason 	}
4642d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
4643be0e5c09SChris Mason }
4644be0e5c09SChris Mason 
464574123bd7SChris Mason /*
4646323ac95bSChris Mason  * a helper function to delete the leaf pointed to by path->slots[1] and
46475d4f98a2SYan Zheng  * path->nodes[1].
4648323ac95bSChris Mason  *
4649323ac95bSChris Mason  * This deletes the pointer in path->nodes[1] and frees the leaf
4650323ac95bSChris Mason  * block extent.  zero is returned if it all worked out, < 0 otherwise.
4651323ac95bSChris Mason  *
4652323ac95bSChris Mason  * The path must have already been setup for deleting the leaf, including
4653323ac95bSChris Mason  * all the proper balancing.  path->nodes[1] must be locked.
4654323ac95bSChris Mason  */
4655143bede5SJeff Mahoney static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans,
4656323ac95bSChris Mason 				    struct btrfs_root *root,
46575d4f98a2SYan Zheng 				    struct btrfs_path *path,
46585d4f98a2SYan Zheng 				    struct extent_buffer *leaf)
4659323ac95bSChris Mason {
46605d4f98a2SYan Zheng 	WARN_ON(btrfs_header_generation(leaf) != trans->transid);
4661f3ea38daSJan Schmidt 	del_ptr(trans, root, path, 1, path->slots[1], 1);
4662323ac95bSChris Mason 
46634d081c41SChris Mason 	/*
46644d081c41SChris Mason 	 * btrfs_free_extent is expensive, we want to make sure we
46654d081c41SChris Mason 	 * aren't holding any locks when we call it
46664d081c41SChris Mason 	 */
46674d081c41SChris Mason 	btrfs_unlock_up_safe(path, 0);
46684d081c41SChris Mason 
4669f0486c68SYan, Zheng 	root_sub_used(root, leaf->len);
4670f0486c68SYan, Zheng 
46713083ee2eSJosef Bacik 	extent_buffer_get(leaf);
46725581a51aSJan Schmidt 	btrfs_free_tree_block(trans, root, leaf, 0, 1);
46733083ee2eSJosef Bacik 	free_extent_buffer_stale(leaf);
4674323ac95bSChris Mason }
4675323ac95bSChris Mason /*
467674123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
467774123bd7SChris Mason  * the leaf, remove it from the tree
467874123bd7SChris Mason  */
467985e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
468085e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
4681be0e5c09SChris Mason {
46825f39d397SChris Mason 	struct extent_buffer *leaf;
46835f39d397SChris Mason 	struct btrfs_item *item;
468485e21bacSChris Mason 	int last_off;
468585e21bacSChris Mason 	int dsize = 0;
4686aa5d6bedSChris Mason 	int ret = 0;
4687aa5d6bedSChris Mason 	int wret;
468885e21bacSChris Mason 	int i;
46897518a238SChris Mason 	u32 nritems;
4690cfed81a0SChris Mason 	struct btrfs_map_token token;
4691cfed81a0SChris Mason 
4692cfed81a0SChris Mason 	btrfs_init_map_token(&token);
4693be0e5c09SChris Mason 
46945f39d397SChris Mason 	leaf = path->nodes[0];
469585e21bacSChris Mason 	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
469685e21bacSChris Mason 
469785e21bacSChris Mason 	for (i = 0; i < nr; i++)
469885e21bacSChris Mason 		dsize += btrfs_item_size_nr(leaf, slot + i);
469985e21bacSChris Mason 
47005f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
4701be0e5c09SChris Mason 
470285e21bacSChris Mason 	if (slot + nr != nritems) {
4703123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
47045f39d397SChris Mason 
47055f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
4706d6025579SChris Mason 			      data_end + dsize,
4707123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
470885e21bacSChris Mason 			      last_off - data_end);
47095f39d397SChris Mason 
471085e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
47115f39d397SChris Mason 			u32 ioff;
4712db94535dSChris Mason 
47135f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
4714cfed81a0SChris Mason 			ioff = btrfs_token_item_offset(leaf, item, &token);
4715cfed81a0SChris Mason 			btrfs_set_token_item_offset(leaf, item,
4716cfed81a0SChris Mason 						    ioff + dsize, &token);
47170783fcfcSChris Mason 		}
4718db94535dSChris Mason 
47195f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
472085e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
47210783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
472285e21bacSChris Mason 			      (nritems - slot - nr));
4723be0e5c09SChris Mason 	}
472485e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
472585e21bacSChris Mason 	nritems -= nr;
47265f39d397SChris Mason 
472774123bd7SChris Mason 	/* delete the leaf if we've emptied it */
47287518a238SChris Mason 	if (nritems == 0) {
47295f39d397SChris Mason 		if (leaf == root->node) {
47305f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
47319a8dd150SChris Mason 		} else {
4732f0486c68SYan, Zheng 			btrfs_set_path_blocking(path);
4733f0486c68SYan, Zheng 			clean_tree_block(trans, root, leaf);
4734143bede5SJeff Mahoney 			btrfs_del_leaf(trans, root, path, leaf);
47359a8dd150SChris Mason 		}
4736be0e5c09SChris Mason 	} else {
47377518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
4738aa5d6bedSChris Mason 		if (slot == 0) {
47395f39d397SChris Mason 			struct btrfs_disk_key disk_key;
47405f39d397SChris Mason 
47415f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
4742143bede5SJeff Mahoney 			fixup_low_keys(trans, root, path, &disk_key, 1);
4743aa5d6bedSChris Mason 		}
4744aa5d6bedSChris Mason 
474574123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
4746d717aa1dSYan Zheng 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
4747be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
4748be0e5c09SChris Mason 			 * make sure the path still points to our leaf
4749be0e5c09SChris Mason 			 * for possible call to del_ptr below
4750be0e5c09SChris Mason 			 */
47514920c9acSChris Mason 			slot = path->slots[1];
47525f39d397SChris Mason 			extent_buffer_get(leaf);
47535f39d397SChris Mason 
4754b9473439SChris Mason 			btrfs_set_path_blocking(path);
475599d8f83cSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1,
475699d8f83cSChris Mason 					      1, (u32)-1);
475754aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
4758aa5d6bedSChris Mason 				ret = wret;
47595f39d397SChris Mason 
47605f39d397SChris Mason 			if (path->nodes[0] == leaf &&
47615f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
476299d8f83cSChris Mason 				wret = push_leaf_right(trans, root, path, 1,
476399d8f83cSChris Mason 						       1, 1, 0);
476454aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
4765aa5d6bedSChris Mason 					ret = wret;
4766aa5d6bedSChris Mason 			}
47675f39d397SChris Mason 
47685f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
4769323ac95bSChris Mason 				path->slots[1] = slot;
4770143bede5SJeff Mahoney 				btrfs_del_leaf(trans, root, path, leaf);
47715f39d397SChris Mason 				free_extent_buffer(leaf);
4772143bede5SJeff Mahoney 				ret = 0;
47735de08d7dSChris Mason 			} else {
4774925baeddSChris Mason 				/* if we're still in the path, make sure
4775925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
4776925baeddSChris Mason 				 * push_leaf functions must have already
4777925baeddSChris Mason 				 * dirtied this buffer
4778925baeddSChris Mason 				 */
4779925baeddSChris Mason 				if (path->nodes[0] == leaf)
47805f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
47815f39d397SChris Mason 				free_extent_buffer(leaf);
4782be0e5c09SChris Mason 			}
4783d5719762SChris Mason 		} else {
47845f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
4785be0e5c09SChris Mason 		}
47869a8dd150SChris Mason 	}
4787aa5d6bedSChris Mason 	return ret;
47889a8dd150SChris Mason }
47899a8dd150SChris Mason 
479097571fd0SChris Mason /*
4791925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
47927bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
47937bb86316SChris Mason  * returns < 0 on io errors.
4794d352ac68SChris Mason  *
4795d352ac68SChris Mason  * This may release the path, and so you may lose any locks held at the
4796d352ac68SChris Mason  * time you call it.
47977bb86316SChris Mason  */
47987bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
47997bb86316SChris Mason {
4800925baeddSChris Mason 	struct btrfs_key key;
4801925baeddSChris Mason 	struct btrfs_disk_key found_key;
4802925baeddSChris Mason 	int ret;
48037bb86316SChris Mason 
4804925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
4805925baeddSChris Mason 
4806925baeddSChris Mason 	if (key.offset > 0)
4807925baeddSChris Mason 		key.offset--;
4808925baeddSChris Mason 	else if (key.type > 0)
4809925baeddSChris Mason 		key.type--;
4810925baeddSChris Mason 	else if (key.objectid > 0)
4811925baeddSChris Mason 		key.objectid--;
4812925baeddSChris Mason 	else
48137bb86316SChris Mason 		return 1;
48147bb86316SChris Mason 
4815b3b4aa74SDavid Sterba 	btrfs_release_path(path);
4816925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
4817925baeddSChris Mason 	if (ret < 0)
4818925baeddSChris Mason 		return ret;
4819925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
4820925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
4821925baeddSChris Mason 	if (ret < 0)
48227bb86316SChris Mason 		return 0;
4823925baeddSChris Mason 	return 1;
48247bb86316SChris Mason }
48257bb86316SChris Mason 
48263f157a2fSChris Mason /*
48273f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
48283f157a2fSChris Mason  * for nodes or leaves that are either in cache or have a minimum
4829d352ac68SChris Mason  * transaction id.  This is used by the btree defrag code, and tree logging
48303f157a2fSChris Mason  *
48313f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
48323f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
48333f157a2fSChris Mason  * key and get a writable path.
48343f157a2fSChris Mason  *
48353f157a2fSChris Mason  * This does lock as it descends, and path->keep_locks should be set
48363f157a2fSChris Mason  * to 1 by the caller.
48373f157a2fSChris Mason  *
48383f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
48393f157a2fSChris Mason  * of the tree.
48403f157a2fSChris Mason  *
4841d352ac68SChris Mason  * min_trans indicates the oldest transaction that you are interested
4842d352ac68SChris Mason  * in walking through.  Any nodes or leaves older than min_trans are
4843d352ac68SChris Mason  * skipped over (without reading them).
4844d352ac68SChris Mason  *
48453f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
48463f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
48473f157a2fSChris Mason  */
48483f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
4849e02119d5SChris Mason 			 struct btrfs_key *max_key,
48503f157a2fSChris Mason 			 struct btrfs_path *path, int cache_only,
48513f157a2fSChris Mason 			 u64 min_trans)
48523f157a2fSChris Mason {
48533f157a2fSChris Mason 	struct extent_buffer *cur;
48543f157a2fSChris Mason 	struct btrfs_key found_key;
48553f157a2fSChris Mason 	int slot;
48569652480bSYan 	int sret;
48573f157a2fSChris Mason 	u32 nritems;
48583f157a2fSChris Mason 	int level;
48593f157a2fSChris Mason 	int ret = 1;
48603f157a2fSChris Mason 
4861934d375bSChris Mason 	WARN_ON(!path->keep_locks);
48623f157a2fSChris Mason again:
4863bd681513SChris Mason 	cur = btrfs_read_lock_root_node(root);
48643f157a2fSChris Mason 	level = btrfs_header_level(cur);
4865e02119d5SChris Mason 	WARN_ON(path->nodes[level]);
48663f157a2fSChris Mason 	path->nodes[level] = cur;
4867bd681513SChris Mason 	path->locks[level] = BTRFS_READ_LOCK;
48683f157a2fSChris Mason 
48693f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
48703f157a2fSChris Mason 		ret = 1;
48713f157a2fSChris Mason 		goto out;
48723f157a2fSChris Mason 	}
48733f157a2fSChris Mason 	while (1) {
48743f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
48753f157a2fSChris Mason 		level = btrfs_header_level(cur);
48769652480bSYan 		sret = bin_search(cur, min_key, level, &slot);
48773f157a2fSChris Mason 
4878323ac95bSChris Mason 		/* at the lowest level, we're done, setup the path and exit */
4879323ac95bSChris Mason 		if (level == path->lowest_level) {
4880e02119d5SChris Mason 			if (slot >= nritems)
4881e02119d5SChris Mason 				goto find_next_key;
48823f157a2fSChris Mason 			ret = 0;
48833f157a2fSChris Mason 			path->slots[level] = slot;
48843f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
48853f157a2fSChris Mason 			goto out;
48863f157a2fSChris Mason 		}
48879652480bSYan 		if (sret && slot > 0)
48889652480bSYan 			slot--;
48893f157a2fSChris Mason 		/*
48903f157a2fSChris Mason 		 * check this node pointer against the cache_only and
48913f157a2fSChris Mason 		 * min_trans parameters.  If it isn't in cache or is too
48923f157a2fSChris Mason 		 * old, skip to the next one.
48933f157a2fSChris Mason 		 */
48943f157a2fSChris Mason 		while (slot < nritems) {
48953f157a2fSChris Mason 			u64 blockptr;
48963f157a2fSChris Mason 			u64 gen;
48973f157a2fSChris Mason 			struct extent_buffer *tmp;
4898e02119d5SChris Mason 			struct btrfs_disk_key disk_key;
4899e02119d5SChris Mason 
49003f157a2fSChris Mason 			blockptr = btrfs_node_blockptr(cur, slot);
49013f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
49023f157a2fSChris Mason 			if (gen < min_trans) {
49033f157a2fSChris Mason 				slot++;
49043f157a2fSChris Mason 				continue;
49053f157a2fSChris Mason 			}
49063f157a2fSChris Mason 			if (!cache_only)
49073f157a2fSChris Mason 				break;
49083f157a2fSChris Mason 
4909e02119d5SChris Mason 			if (max_key) {
4910e02119d5SChris Mason 				btrfs_node_key(cur, &disk_key, slot);
4911e02119d5SChris Mason 				if (comp_keys(&disk_key, max_key) >= 0) {
4912e02119d5SChris Mason 					ret = 1;
4913e02119d5SChris Mason 					goto out;
4914e02119d5SChris Mason 				}
4915e02119d5SChris Mason 			}
4916e02119d5SChris Mason 
49173f157a2fSChris Mason 			tmp = btrfs_find_tree_block(root, blockptr,
49183f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
49193f157a2fSChris Mason 
4920b9fab919SChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen, 1) > 0) {
49213f157a2fSChris Mason 				free_extent_buffer(tmp);
49223f157a2fSChris Mason 				break;
49233f157a2fSChris Mason 			}
49243f157a2fSChris Mason 			if (tmp)
49253f157a2fSChris Mason 				free_extent_buffer(tmp);
49263f157a2fSChris Mason 			slot++;
49273f157a2fSChris Mason 		}
4928e02119d5SChris Mason find_next_key:
49293f157a2fSChris Mason 		/*
49303f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
49313f157a2fSChris Mason 		 * and find another one
49323f157a2fSChris Mason 		 */
49333f157a2fSChris Mason 		if (slot >= nritems) {
4934e02119d5SChris Mason 			path->slots[level] = slot;
4935b4ce94deSChris Mason 			btrfs_set_path_blocking(path);
4936e02119d5SChris Mason 			sret = btrfs_find_next_key(root, path, min_key, level,
49373f157a2fSChris Mason 						  cache_only, min_trans);
4938e02119d5SChris Mason 			if (sret == 0) {
4939b3b4aa74SDavid Sterba 				btrfs_release_path(path);
49403f157a2fSChris Mason 				goto again;
49413f157a2fSChris Mason 			} else {
49423f157a2fSChris Mason 				goto out;
49433f157a2fSChris Mason 			}
49443f157a2fSChris Mason 		}
49453f157a2fSChris Mason 		/* save our key for returning back */
49463f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
49473f157a2fSChris Mason 		path->slots[level] = slot;
49483f157a2fSChris Mason 		if (level == path->lowest_level) {
49493f157a2fSChris Mason 			ret = 0;
4950f7c79f30SChris Mason 			unlock_up(path, level, 1, 0, NULL);
49513f157a2fSChris Mason 			goto out;
49523f157a2fSChris Mason 		}
4953b4ce94deSChris Mason 		btrfs_set_path_blocking(path);
49543f157a2fSChris Mason 		cur = read_node_slot(root, cur, slot);
495579787eaaSJeff Mahoney 		BUG_ON(!cur); /* -ENOMEM */
49563f157a2fSChris Mason 
4957bd681513SChris Mason 		btrfs_tree_read_lock(cur);
4958b4ce94deSChris Mason 
4959bd681513SChris Mason 		path->locks[level - 1] = BTRFS_READ_LOCK;
49603f157a2fSChris Mason 		path->nodes[level - 1] = cur;
4961f7c79f30SChris Mason 		unlock_up(path, level, 1, 0, NULL);
4962bd681513SChris Mason 		btrfs_clear_path_blocking(path, NULL, 0);
49633f157a2fSChris Mason 	}
49643f157a2fSChris Mason out:
49653f157a2fSChris Mason 	if (ret == 0)
49663f157a2fSChris Mason 		memcpy(min_key, &found_key, sizeof(found_key));
4967b4ce94deSChris Mason 	btrfs_set_path_blocking(path);
49683f157a2fSChris Mason 	return ret;
49693f157a2fSChris Mason }
49703f157a2fSChris Mason 
49717069830aSAlexander Block static void tree_move_down(struct btrfs_root *root,
49727069830aSAlexander Block 			   struct btrfs_path *path,
49737069830aSAlexander Block 			   int *level, int root_level)
49747069830aSAlexander Block {
497574dd17fbSChris Mason 	BUG_ON(*level == 0);
49767069830aSAlexander Block 	path->nodes[*level - 1] = read_node_slot(root, path->nodes[*level],
49777069830aSAlexander Block 					path->slots[*level]);
49787069830aSAlexander Block 	path->slots[*level - 1] = 0;
49797069830aSAlexander Block 	(*level)--;
49807069830aSAlexander Block }
49817069830aSAlexander Block 
49827069830aSAlexander Block static int tree_move_next_or_upnext(struct btrfs_root *root,
49837069830aSAlexander Block 				    struct btrfs_path *path,
49847069830aSAlexander Block 				    int *level, int root_level)
49857069830aSAlexander Block {
49867069830aSAlexander Block 	int ret = 0;
49877069830aSAlexander Block 	int nritems;
49887069830aSAlexander Block 	nritems = btrfs_header_nritems(path->nodes[*level]);
49897069830aSAlexander Block 
49907069830aSAlexander Block 	path->slots[*level]++;
49917069830aSAlexander Block 
499274dd17fbSChris Mason 	while (path->slots[*level] >= nritems) {
49937069830aSAlexander Block 		if (*level == root_level)
49947069830aSAlexander Block 			return -1;
49957069830aSAlexander Block 
49967069830aSAlexander Block 		/* move upnext */
49977069830aSAlexander Block 		path->slots[*level] = 0;
49987069830aSAlexander Block 		free_extent_buffer(path->nodes[*level]);
49997069830aSAlexander Block 		path->nodes[*level] = NULL;
50007069830aSAlexander Block 		(*level)++;
50017069830aSAlexander Block 		path->slots[*level]++;
50027069830aSAlexander Block 
50037069830aSAlexander Block 		nritems = btrfs_header_nritems(path->nodes[*level]);
50047069830aSAlexander Block 		ret = 1;
50057069830aSAlexander Block 	}
50067069830aSAlexander Block 	return ret;
50077069830aSAlexander Block }
50087069830aSAlexander Block 
50097069830aSAlexander Block /*
50107069830aSAlexander Block  * Returns 1 if it had to move up and next. 0 is returned if it moved only next
50117069830aSAlexander Block  * or down.
50127069830aSAlexander Block  */
50137069830aSAlexander Block static int tree_advance(struct btrfs_root *root,
50147069830aSAlexander Block 			struct btrfs_path *path,
50157069830aSAlexander Block 			int *level, int root_level,
50167069830aSAlexander Block 			int allow_down,
50177069830aSAlexander Block 			struct btrfs_key *key)
50187069830aSAlexander Block {
50197069830aSAlexander Block 	int ret;
50207069830aSAlexander Block 
50217069830aSAlexander Block 	if (*level == 0 || !allow_down) {
50227069830aSAlexander Block 		ret = tree_move_next_or_upnext(root, path, level, root_level);
50237069830aSAlexander Block 	} else {
50247069830aSAlexander Block 		tree_move_down(root, path, level, root_level);
50257069830aSAlexander Block 		ret = 0;
50267069830aSAlexander Block 	}
50277069830aSAlexander Block 	if (ret >= 0) {
50287069830aSAlexander Block 		if (*level == 0)
50297069830aSAlexander Block 			btrfs_item_key_to_cpu(path->nodes[*level], key,
50307069830aSAlexander Block 					path->slots[*level]);
50317069830aSAlexander Block 		else
50327069830aSAlexander Block 			btrfs_node_key_to_cpu(path->nodes[*level], key,
50337069830aSAlexander Block 					path->slots[*level]);
50347069830aSAlexander Block 	}
50357069830aSAlexander Block 	return ret;
50367069830aSAlexander Block }
50377069830aSAlexander Block 
50387069830aSAlexander Block static int tree_compare_item(struct btrfs_root *left_root,
50397069830aSAlexander Block 			     struct btrfs_path *left_path,
50407069830aSAlexander Block 			     struct btrfs_path *right_path,
50417069830aSAlexander Block 			     char *tmp_buf)
50427069830aSAlexander Block {
50437069830aSAlexander Block 	int cmp;
50447069830aSAlexander Block 	int len1, len2;
50457069830aSAlexander Block 	unsigned long off1, off2;
50467069830aSAlexander Block 
50477069830aSAlexander Block 	len1 = btrfs_item_size_nr(left_path->nodes[0], left_path->slots[0]);
50487069830aSAlexander Block 	len2 = btrfs_item_size_nr(right_path->nodes[0], right_path->slots[0]);
50497069830aSAlexander Block 	if (len1 != len2)
50507069830aSAlexander Block 		return 1;
50517069830aSAlexander Block 
50527069830aSAlexander Block 	off1 = btrfs_item_ptr_offset(left_path->nodes[0], left_path->slots[0]);
50537069830aSAlexander Block 	off2 = btrfs_item_ptr_offset(right_path->nodes[0],
50547069830aSAlexander Block 				right_path->slots[0]);
50557069830aSAlexander Block 
50567069830aSAlexander Block 	read_extent_buffer(left_path->nodes[0], tmp_buf, off1, len1);
50577069830aSAlexander Block 
50587069830aSAlexander Block 	cmp = memcmp_extent_buffer(right_path->nodes[0], tmp_buf, off2, len1);
50597069830aSAlexander Block 	if (cmp)
50607069830aSAlexander Block 		return 1;
50617069830aSAlexander Block 	return 0;
50627069830aSAlexander Block }
50637069830aSAlexander Block 
50647069830aSAlexander Block #define ADVANCE 1
50657069830aSAlexander Block #define ADVANCE_ONLY_NEXT -1
50667069830aSAlexander Block 
50677069830aSAlexander Block /*
50687069830aSAlexander Block  * This function compares two trees and calls the provided callback for
50697069830aSAlexander Block  * every changed/new/deleted item it finds.
50707069830aSAlexander Block  * If shared tree blocks are encountered, whole subtrees are skipped, making
50717069830aSAlexander Block  * the compare pretty fast on snapshotted subvolumes.
50727069830aSAlexander Block  *
50737069830aSAlexander Block  * This currently works on commit roots only. As commit roots are read only,
50747069830aSAlexander Block  * we don't do any locking. The commit roots are protected with transactions.
50757069830aSAlexander Block  * Transactions are ended and rejoined when a commit is tried in between.
50767069830aSAlexander Block  *
50777069830aSAlexander Block  * This function checks for modifications done to the trees while comparing.
50787069830aSAlexander Block  * If it detects a change, it aborts immediately.
50797069830aSAlexander Block  */
50807069830aSAlexander Block int btrfs_compare_trees(struct btrfs_root *left_root,
50817069830aSAlexander Block 			struct btrfs_root *right_root,
50827069830aSAlexander Block 			btrfs_changed_cb_t changed_cb, void *ctx)
50837069830aSAlexander Block {
50847069830aSAlexander Block 	int ret;
50857069830aSAlexander Block 	int cmp;
50867069830aSAlexander Block 	struct btrfs_trans_handle *trans = NULL;
50877069830aSAlexander Block 	struct btrfs_path *left_path = NULL;
50887069830aSAlexander Block 	struct btrfs_path *right_path = NULL;
50897069830aSAlexander Block 	struct btrfs_key left_key;
50907069830aSAlexander Block 	struct btrfs_key right_key;
50917069830aSAlexander Block 	char *tmp_buf = NULL;
50927069830aSAlexander Block 	int left_root_level;
50937069830aSAlexander Block 	int right_root_level;
50947069830aSAlexander Block 	int left_level;
50957069830aSAlexander Block 	int right_level;
50967069830aSAlexander Block 	int left_end_reached;
50977069830aSAlexander Block 	int right_end_reached;
50987069830aSAlexander Block 	int advance_left;
50997069830aSAlexander Block 	int advance_right;
51007069830aSAlexander Block 	u64 left_blockptr;
51017069830aSAlexander Block 	u64 right_blockptr;
51027069830aSAlexander Block 	u64 left_start_ctransid;
51037069830aSAlexander Block 	u64 right_start_ctransid;
51047069830aSAlexander Block 	u64 ctransid;
51057069830aSAlexander Block 
51067069830aSAlexander Block 	left_path = btrfs_alloc_path();
51077069830aSAlexander Block 	if (!left_path) {
51087069830aSAlexander Block 		ret = -ENOMEM;
51097069830aSAlexander Block 		goto out;
51107069830aSAlexander Block 	}
51117069830aSAlexander Block 	right_path = btrfs_alloc_path();
51127069830aSAlexander Block 	if (!right_path) {
51137069830aSAlexander Block 		ret = -ENOMEM;
51147069830aSAlexander Block 		goto out;
51157069830aSAlexander Block 	}
51167069830aSAlexander Block 
51177069830aSAlexander Block 	tmp_buf = kmalloc(left_root->leafsize, GFP_NOFS);
51187069830aSAlexander Block 	if (!tmp_buf) {
51197069830aSAlexander Block 		ret = -ENOMEM;
51207069830aSAlexander Block 		goto out;
51217069830aSAlexander Block 	}
51227069830aSAlexander Block 
51237069830aSAlexander Block 	left_path->search_commit_root = 1;
51247069830aSAlexander Block 	left_path->skip_locking = 1;
51257069830aSAlexander Block 	right_path->search_commit_root = 1;
51267069830aSAlexander Block 	right_path->skip_locking = 1;
51277069830aSAlexander Block 
51287069830aSAlexander Block 	spin_lock(&left_root->root_times_lock);
51297069830aSAlexander Block 	left_start_ctransid = btrfs_root_ctransid(&left_root->root_item);
51307069830aSAlexander Block 	spin_unlock(&left_root->root_times_lock);
51317069830aSAlexander Block 
51327069830aSAlexander Block 	spin_lock(&right_root->root_times_lock);
51337069830aSAlexander Block 	right_start_ctransid = btrfs_root_ctransid(&right_root->root_item);
51347069830aSAlexander Block 	spin_unlock(&right_root->root_times_lock);
51357069830aSAlexander Block 
51367069830aSAlexander Block 	trans = btrfs_join_transaction(left_root);
51377069830aSAlexander Block 	if (IS_ERR(trans)) {
51387069830aSAlexander Block 		ret = PTR_ERR(trans);
51397069830aSAlexander Block 		trans = NULL;
51407069830aSAlexander Block 		goto out;
51417069830aSAlexander Block 	}
51427069830aSAlexander Block 
51437069830aSAlexander Block 	/*
51447069830aSAlexander Block 	 * Strategy: Go to the first items of both trees. Then do
51457069830aSAlexander Block 	 *
51467069830aSAlexander Block 	 * If both trees are at level 0
51477069830aSAlexander Block 	 *   Compare keys of current items
51487069830aSAlexander Block 	 *     If left < right treat left item as new, advance left tree
51497069830aSAlexander Block 	 *       and repeat
51507069830aSAlexander Block 	 *     If left > right treat right item as deleted, advance right tree
51517069830aSAlexander Block 	 *       and repeat
51527069830aSAlexander Block 	 *     If left == right do deep compare of items, treat as changed if
51537069830aSAlexander Block 	 *       needed, advance both trees and repeat
51547069830aSAlexander Block 	 * If both trees are at the same level but not at level 0
51557069830aSAlexander Block 	 *   Compare keys of current nodes/leafs
51567069830aSAlexander Block 	 *     If left < right advance left tree and repeat
51577069830aSAlexander Block 	 *     If left > right advance right tree and repeat
51587069830aSAlexander Block 	 *     If left == right compare blockptrs of the next nodes/leafs
51597069830aSAlexander Block 	 *       If they match advance both trees but stay at the same level
51607069830aSAlexander Block 	 *         and repeat
51617069830aSAlexander Block 	 *       If they don't match advance both trees while allowing to go
51627069830aSAlexander Block 	 *         deeper and repeat
51637069830aSAlexander Block 	 * If tree levels are different
51647069830aSAlexander Block 	 *   Advance the tree that needs it and repeat
51657069830aSAlexander Block 	 *
51667069830aSAlexander Block 	 * Advancing a tree means:
51677069830aSAlexander Block 	 *   If we are at level 0, try to go to the next slot. If that's not
51687069830aSAlexander Block 	 *   possible, go one level up and repeat. Stop when we found a level
51697069830aSAlexander Block 	 *   where we could go to the next slot. We may at this point be on a
51707069830aSAlexander Block 	 *   node or a leaf.
51717069830aSAlexander Block 	 *
51727069830aSAlexander Block 	 *   If we are not at level 0 and not on shared tree blocks, go one
51737069830aSAlexander Block 	 *   level deeper.
51747069830aSAlexander Block 	 *
51757069830aSAlexander Block 	 *   If we are not at level 0 and on shared tree blocks, go one slot to
51767069830aSAlexander Block 	 *   the right if possible or go up and right.
51777069830aSAlexander Block 	 */
51787069830aSAlexander Block 
51797069830aSAlexander Block 	left_level = btrfs_header_level(left_root->commit_root);
51807069830aSAlexander Block 	left_root_level = left_level;
51817069830aSAlexander Block 	left_path->nodes[left_level] = left_root->commit_root;
51827069830aSAlexander Block 	extent_buffer_get(left_path->nodes[left_level]);
51837069830aSAlexander Block 
51847069830aSAlexander Block 	right_level = btrfs_header_level(right_root->commit_root);
51857069830aSAlexander Block 	right_root_level = right_level;
51867069830aSAlexander Block 	right_path->nodes[right_level] = right_root->commit_root;
51877069830aSAlexander Block 	extent_buffer_get(right_path->nodes[right_level]);
51887069830aSAlexander Block 
51897069830aSAlexander Block 	if (left_level == 0)
51907069830aSAlexander Block 		btrfs_item_key_to_cpu(left_path->nodes[left_level],
51917069830aSAlexander Block 				&left_key, left_path->slots[left_level]);
51927069830aSAlexander Block 	else
51937069830aSAlexander Block 		btrfs_node_key_to_cpu(left_path->nodes[left_level],
51947069830aSAlexander Block 				&left_key, left_path->slots[left_level]);
51957069830aSAlexander Block 	if (right_level == 0)
51967069830aSAlexander Block 		btrfs_item_key_to_cpu(right_path->nodes[right_level],
51977069830aSAlexander Block 				&right_key, right_path->slots[right_level]);
51987069830aSAlexander Block 	else
51997069830aSAlexander Block 		btrfs_node_key_to_cpu(right_path->nodes[right_level],
52007069830aSAlexander Block 				&right_key, right_path->slots[right_level]);
52017069830aSAlexander Block 
52027069830aSAlexander Block 	left_end_reached = right_end_reached = 0;
52037069830aSAlexander Block 	advance_left = advance_right = 0;
52047069830aSAlexander Block 
52057069830aSAlexander Block 	while (1) {
52067069830aSAlexander Block 		/*
52077069830aSAlexander Block 		 * We need to make sure the transaction does not get committed
52087069830aSAlexander Block 		 * while we do anything on commit roots. This means, we need to
52097069830aSAlexander Block 		 * join and leave transactions for every item that we process.
52107069830aSAlexander Block 		 */
52117069830aSAlexander Block 		if (trans && btrfs_should_end_transaction(trans, left_root)) {
52127069830aSAlexander Block 			btrfs_release_path(left_path);
52137069830aSAlexander Block 			btrfs_release_path(right_path);
52147069830aSAlexander Block 
52157069830aSAlexander Block 			ret = btrfs_end_transaction(trans, left_root);
52167069830aSAlexander Block 			trans = NULL;
52177069830aSAlexander Block 			if (ret < 0)
52187069830aSAlexander Block 				goto out;
52197069830aSAlexander Block 		}
52207069830aSAlexander Block 		/* now rejoin the transaction */
52217069830aSAlexander Block 		if (!trans) {
52227069830aSAlexander Block 			trans = btrfs_join_transaction(left_root);
52237069830aSAlexander Block 			if (IS_ERR(trans)) {
52247069830aSAlexander Block 				ret = PTR_ERR(trans);
52257069830aSAlexander Block 				trans = NULL;
52267069830aSAlexander Block 				goto out;
52277069830aSAlexander Block 			}
52287069830aSAlexander Block 
52297069830aSAlexander Block 			spin_lock(&left_root->root_times_lock);
52307069830aSAlexander Block 			ctransid = btrfs_root_ctransid(&left_root->root_item);
52317069830aSAlexander Block 			spin_unlock(&left_root->root_times_lock);
52327069830aSAlexander Block 			if (ctransid != left_start_ctransid)
52337069830aSAlexander Block 				left_start_ctransid = 0;
52347069830aSAlexander Block 
52357069830aSAlexander Block 			spin_lock(&right_root->root_times_lock);
52367069830aSAlexander Block 			ctransid = btrfs_root_ctransid(&right_root->root_item);
52377069830aSAlexander Block 			spin_unlock(&right_root->root_times_lock);
52387069830aSAlexander Block 			if (ctransid != right_start_ctransid)
52397069830aSAlexander Block 				right_start_ctransid = 0;
52407069830aSAlexander Block 
52417069830aSAlexander Block 			if (!left_start_ctransid || !right_start_ctransid) {
52427069830aSAlexander Block 				WARN(1, KERN_WARNING
52437069830aSAlexander Block 					"btrfs: btrfs_compare_tree detected "
52447069830aSAlexander Block 					"a change in one of the trees while "
52457069830aSAlexander Block 					"iterating. This is probably a "
52467069830aSAlexander Block 					"bug.\n");
52477069830aSAlexander Block 				ret = -EIO;
52487069830aSAlexander Block 				goto out;
52497069830aSAlexander Block 			}
52507069830aSAlexander Block 
52517069830aSAlexander Block 			/*
52527069830aSAlexander Block 			 * the commit root may have changed, so start again
52537069830aSAlexander Block 			 * where we stopped
52547069830aSAlexander Block 			 */
52557069830aSAlexander Block 			left_path->lowest_level = left_level;
52567069830aSAlexander Block 			right_path->lowest_level = right_level;
52577069830aSAlexander Block 			ret = btrfs_search_slot(NULL, left_root,
52587069830aSAlexander Block 					&left_key, left_path, 0, 0);
52597069830aSAlexander Block 			if (ret < 0)
52607069830aSAlexander Block 				goto out;
52617069830aSAlexander Block 			ret = btrfs_search_slot(NULL, right_root,
52627069830aSAlexander Block 					&right_key, right_path, 0, 0);
52637069830aSAlexander Block 			if (ret < 0)
52647069830aSAlexander Block 				goto out;
52657069830aSAlexander Block 		}
52667069830aSAlexander Block 
52677069830aSAlexander Block 		if (advance_left && !left_end_reached) {
52687069830aSAlexander Block 			ret = tree_advance(left_root, left_path, &left_level,
52697069830aSAlexander Block 					left_root_level,
52707069830aSAlexander Block 					advance_left != ADVANCE_ONLY_NEXT,
52717069830aSAlexander Block 					&left_key);
52727069830aSAlexander Block 			if (ret < 0)
52737069830aSAlexander Block 				left_end_reached = ADVANCE;
52747069830aSAlexander Block 			advance_left = 0;
52757069830aSAlexander Block 		}
52767069830aSAlexander Block 		if (advance_right && !right_end_reached) {
52777069830aSAlexander Block 			ret = tree_advance(right_root, right_path, &right_level,
52787069830aSAlexander Block 					right_root_level,
52797069830aSAlexander Block 					advance_right != ADVANCE_ONLY_NEXT,
52807069830aSAlexander Block 					&right_key);
52817069830aSAlexander Block 			if (ret < 0)
52827069830aSAlexander Block 				right_end_reached = ADVANCE;
52837069830aSAlexander Block 			advance_right = 0;
52847069830aSAlexander Block 		}
52857069830aSAlexander Block 
52867069830aSAlexander Block 		if (left_end_reached && right_end_reached) {
52877069830aSAlexander Block 			ret = 0;
52887069830aSAlexander Block 			goto out;
52897069830aSAlexander Block 		} else if (left_end_reached) {
52907069830aSAlexander Block 			if (right_level == 0) {
52917069830aSAlexander Block 				ret = changed_cb(left_root, right_root,
52927069830aSAlexander Block 						left_path, right_path,
52937069830aSAlexander Block 						&right_key,
52947069830aSAlexander Block 						BTRFS_COMPARE_TREE_DELETED,
52957069830aSAlexander Block 						ctx);
52967069830aSAlexander Block 				if (ret < 0)
52977069830aSAlexander Block 					goto out;
52987069830aSAlexander Block 			}
52997069830aSAlexander Block 			advance_right = ADVANCE;
53007069830aSAlexander Block 			continue;
53017069830aSAlexander Block 		} else if (right_end_reached) {
53027069830aSAlexander Block 			if (left_level == 0) {
53037069830aSAlexander Block 				ret = changed_cb(left_root, right_root,
53047069830aSAlexander Block 						left_path, right_path,
53057069830aSAlexander Block 						&left_key,
53067069830aSAlexander Block 						BTRFS_COMPARE_TREE_NEW,
53077069830aSAlexander Block 						ctx);
53087069830aSAlexander Block 				if (ret < 0)
53097069830aSAlexander Block 					goto out;
53107069830aSAlexander Block 			}
53117069830aSAlexander Block 			advance_left = ADVANCE;
53127069830aSAlexander Block 			continue;
53137069830aSAlexander Block 		}
53147069830aSAlexander Block 
53157069830aSAlexander Block 		if (left_level == 0 && right_level == 0) {
53167069830aSAlexander Block 			cmp = btrfs_comp_cpu_keys(&left_key, &right_key);
53177069830aSAlexander Block 			if (cmp < 0) {
53187069830aSAlexander Block 				ret = changed_cb(left_root, right_root,
53197069830aSAlexander Block 						left_path, right_path,
53207069830aSAlexander Block 						&left_key,
53217069830aSAlexander Block 						BTRFS_COMPARE_TREE_NEW,
53227069830aSAlexander Block 						ctx);
53237069830aSAlexander Block 				if (ret < 0)
53247069830aSAlexander Block 					goto out;
53257069830aSAlexander Block 				advance_left = ADVANCE;
53267069830aSAlexander Block 			} else if (cmp > 0) {
53277069830aSAlexander Block 				ret = changed_cb(left_root, right_root,
53287069830aSAlexander Block 						left_path, right_path,
53297069830aSAlexander Block 						&right_key,
53307069830aSAlexander Block 						BTRFS_COMPARE_TREE_DELETED,
53317069830aSAlexander Block 						ctx);
53327069830aSAlexander Block 				if (ret < 0)
53337069830aSAlexander Block 					goto out;
53347069830aSAlexander Block 				advance_right = ADVANCE;
53357069830aSAlexander Block 			} else {
533674dd17fbSChris Mason 				WARN_ON(!extent_buffer_uptodate(left_path->nodes[0]));
53377069830aSAlexander Block 				ret = tree_compare_item(left_root, left_path,
53387069830aSAlexander Block 						right_path, tmp_buf);
53397069830aSAlexander Block 				if (ret) {
534074dd17fbSChris Mason 					WARN_ON(!extent_buffer_uptodate(left_path->nodes[0]));
53417069830aSAlexander Block 					ret = changed_cb(left_root, right_root,
53427069830aSAlexander Block 						left_path, right_path,
53437069830aSAlexander Block 						&left_key,
53447069830aSAlexander Block 						BTRFS_COMPARE_TREE_CHANGED,
53457069830aSAlexander Block 						ctx);
53467069830aSAlexander Block 					if (ret < 0)
53477069830aSAlexander Block 						goto out;
53487069830aSAlexander Block 				}
53497069830aSAlexander Block 				advance_left = ADVANCE;
53507069830aSAlexander Block 				advance_right = ADVANCE;
53517069830aSAlexander Block 			}
53527069830aSAlexander Block 		} else if (left_level == right_level) {
53537069830aSAlexander Block 			cmp = btrfs_comp_cpu_keys(&left_key, &right_key);
53547069830aSAlexander Block 			if (cmp < 0) {
53557069830aSAlexander Block 				advance_left = ADVANCE;
53567069830aSAlexander Block 			} else if (cmp > 0) {
53577069830aSAlexander Block 				advance_right = ADVANCE;
53587069830aSAlexander Block 			} else {
53597069830aSAlexander Block 				left_blockptr = btrfs_node_blockptr(
53607069830aSAlexander Block 						left_path->nodes[left_level],
53617069830aSAlexander Block 						left_path->slots[left_level]);
53627069830aSAlexander Block 				right_blockptr = btrfs_node_blockptr(
53637069830aSAlexander Block 						right_path->nodes[right_level],
53647069830aSAlexander Block 						right_path->slots[right_level]);
53657069830aSAlexander Block 				if (left_blockptr == right_blockptr) {
53667069830aSAlexander Block 					/*
53677069830aSAlexander Block 					 * As we're on a shared block, don't
53687069830aSAlexander Block 					 * allow to go deeper.
53697069830aSAlexander Block 					 */
53707069830aSAlexander Block 					advance_left = ADVANCE_ONLY_NEXT;
53717069830aSAlexander Block 					advance_right = ADVANCE_ONLY_NEXT;
53727069830aSAlexander Block 				} else {
53737069830aSAlexander Block 					advance_left = ADVANCE;
53747069830aSAlexander Block 					advance_right = ADVANCE;
53757069830aSAlexander Block 				}
53767069830aSAlexander Block 			}
53777069830aSAlexander Block 		} else if (left_level < right_level) {
53787069830aSAlexander Block 			advance_right = ADVANCE;
53797069830aSAlexander Block 		} else {
53807069830aSAlexander Block 			advance_left = ADVANCE;
53817069830aSAlexander Block 		}
53827069830aSAlexander Block 	}
53837069830aSAlexander Block 
53847069830aSAlexander Block out:
53857069830aSAlexander Block 	btrfs_free_path(left_path);
53867069830aSAlexander Block 	btrfs_free_path(right_path);
53877069830aSAlexander Block 	kfree(tmp_buf);
53887069830aSAlexander Block 
53897069830aSAlexander Block 	if (trans) {
53907069830aSAlexander Block 		if (!ret)
53917069830aSAlexander Block 			ret = btrfs_end_transaction(trans, left_root);
53927069830aSAlexander Block 		else
53937069830aSAlexander Block 			btrfs_end_transaction(trans, left_root);
53947069830aSAlexander Block 	}
53957069830aSAlexander Block 
53967069830aSAlexander Block 	return ret;
53977069830aSAlexander Block }
53987069830aSAlexander Block 
53993f157a2fSChris Mason /*
54003f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
54013f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
54023f157a2fSChris Mason  * tree based on the current path and the cache_only and min_trans
54033f157a2fSChris Mason  * parameters.
54043f157a2fSChris Mason  *
54053f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
54063f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
54073f157a2fSChris Mason  *
54083f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
54093f157a2fSChris Mason  * calling this function.
54103f157a2fSChris Mason  */
5411e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
541233c66f43SYan Zheng 			struct btrfs_key *key, int level,
54133f157a2fSChris Mason 			int cache_only, u64 min_trans)
5414e7a84565SChris Mason {
5415e7a84565SChris Mason 	int slot;
5416e7a84565SChris Mason 	struct extent_buffer *c;
5417e7a84565SChris Mason 
5418934d375bSChris Mason 	WARN_ON(!path->keep_locks);
5419e7a84565SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
5420e7a84565SChris Mason 		if (!path->nodes[level])
5421e7a84565SChris Mason 			return 1;
5422e7a84565SChris Mason 
5423e7a84565SChris Mason 		slot = path->slots[level] + 1;
5424e7a84565SChris Mason 		c = path->nodes[level];
54253f157a2fSChris Mason next:
5426e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
542733c66f43SYan Zheng 			int ret;
542833c66f43SYan Zheng 			int orig_lowest;
542933c66f43SYan Zheng 			struct btrfs_key cur_key;
543033c66f43SYan Zheng 			if (level + 1 >= BTRFS_MAX_LEVEL ||
543133c66f43SYan Zheng 			    !path->nodes[level + 1])
5432e7a84565SChris Mason 				return 1;
543333c66f43SYan Zheng 
543433c66f43SYan Zheng 			if (path->locks[level + 1]) {
543533c66f43SYan Zheng 				level++;
5436e7a84565SChris Mason 				continue;
5437e7a84565SChris Mason 			}
543833c66f43SYan Zheng 
543933c66f43SYan Zheng 			slot = btrfs_header_nritems(c) - 1;
544033c66f43SYan Zheng 			if (level == 0)
544133c66f43SYan Zheng 				btrfs_item_key_to_cpu(c, &cur_key, slot);
544233c66f43SYan Zheng 			else
544333c66f43SYan Zheng 				btrfs_node_key_to_cpu(c, &cur_key, slot);
544433c66f43SYan Zheng 
544533c66f43SYan Zheng 			orig_lowest = path->lowest_level;
5446b3b4aa74SDavid Sterba 			btrfs_release_path(path);
544733c66f43SYan Zheng 			path->lowest_level = level;
544833c66f43SYan Zheng 			ret = btrfs_search_slot(NULL, root, &cur_key, path,
544933c66f43SYan Zheng 						0, 0);
545033c66f43SYan Zheng 			path->lowest_level = orig_lowest;
545133c66f43SYan Zheng 			if (ret < 0)
545233c66f43SYan Zheng 				return ret;
545333c66f43SYan Zheng 
545433c66f43SYan Zheng 			c = path->nodes[level];
545533c66f43SYan Zheng 			slot = path->slots[level];
545633c66f43SYan Zheng 			if (ret == 0)
545733c66f43SYan Zheng 				slot++;
545833c66f43SYan Zheng 			goto next;
545933c66f43SYan Zheng 		}
546033c66f43SYan Zheng 
5461e7a84565SChris Mason 		if (level == 0)
5462e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
54633f157a2fSChris Mason 		else {
54643f157a2fSChris Mason 			u64 blockptr = btrfs_node_blockptr(c, slot);
54653f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
54663f157a2fSChris Mason 
54673f157a2fSChris Mason 			if (cache_only) {
54683f157a2fSChris Mason 				struct extent_buffer *cur;
54693f157a2fSChris Mason 				cur = btrfs_find_tree_block(root, blockptr,
54703f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
5471b9fab919SChris Mason 				if (!cur ||
5472b9fab919SChris Mason 				    btrfs_buffer_uptodate(cur, gen, 1) <= 0) {
54733f157a2fSChris Mason 					slot++;
54743f157a2fSChris Mason 					if (cur)
54753f157a2fSChris Mason 						free_extent_buffer(cur);
54763f157a2fSChris Mason 					goto next;
54773f157a2fSChris Mason 				}
54783f157a2fSChris Mason 				free_extent_buffer(cur);
54793f157a2fSChris Mason 			}
54803f157a2fSChris Mason 			if (gen < min_trans) {
54813f157a2fSChris Mason 				slot++;
54823f157a2fSChris Mason 				goto next;
54833f157a2fSChris Mason 			}
5484e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
54853f157a2fSChris Mason 		}
5486e7a84565SChris Mason 		return 0;
5487e7a84565SChris Mason 	}
5488e7a84565SChris Mason 	return 1;
5489e7a84565SChris Mason }
5490e7a84565SChris Mason 
54917bb86316SChris Mason /*
5492925baeddSChris Mason  * search the tree again to find a leaf with greater keys
54930f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
54940f70abe2SChris Mason  * returns < 0 on io errors.
549597571fd0SChris Mason  */
5496234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
5497d97e63b6SChris Mason {
54983d7806ecSJan Schmidt 	return btrfs_next_old_leaf(root, path, 0);
54993d7806ecSJan Schmidt }
55003d7806ecSJan Schmidt 
55013d7806ecSJan Schmidt int btrfs_next_old_leaf(struct btrfs_root *root, struct btrfs_path *path,
55023d7806ecSJan Schmidt 			u64 time_seq)
55033d7806ecSJan Schmidt {
5504d97e63b6SChris Mason 	int slot;
55058e73f275SChris Mason 	int level;
55065f39d397SChris Mason 	struct extent_buffer *c;
55078e73f275SChris Mason 	struct extent_buffer *next;
5508925baeddSChris Mason 	struct btrfs_key key;
5509925baeddSChris Mason 	u32 nritems;
5510925baeddSChris Mason 	int ret;
55118e73f275SChris Mason 	int old_spinning = path->leave_spinning;
5512bd681513SChris Mason 	int next_rw_lock = 0;
5513925baeddSChris Mason 
5514925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
5515d397712bSChris Mason 	if (nritems == 0)
5516925baeddSChris Mason 		return 1;
5517925baeddSChris Mason 
55188e73f275SChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
55198e73f275SChris Mason again:
55208e73f275SChris Mason 	level = 1;
55218e73f275SChris Mason 	next = NULL;
5522bd681513SChris Mason 	next_rw_lock = 0;
5523b3b4aa74SDavid Sterba 	btrfs_release_path(path);
55248e73f275SChris Mason 
5525a2135011SChris Mason 	path->keep_locks = 1;
55268e73f275SChris Mason 	path->leave_spinning = 1;
55278e73f275SChris Mason 
55283d7806ecSJan Schmidt 	if (time_seq)
55293d7806ecSJan Schmidt 		ret = btrfs_search_old_slot(root, &key, path, time_seq);
55303d7806ecSJan Schmidt 	else
5531925baeddSChris Mason 		ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
5532925baeddSChris Mason 	path->keep_locks = 0;
5533925baeddSChris Mason 
5534925baeddSChris Mason 	if (ret < 0)
5535925baeddSChris Mason 		return ret;
5536925baeddSChris Mason 
5537a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
5538168fd7d2SChris Mason 	/*
5539168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
5540168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
5541168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
5542168fd7d2SChris Mason 	 * advance the path if there are now more items available.
5543168fd7d2SChris Mason 	 */
5544a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
5545e457afecSYan Zheng 		if (ret == 0)
5546168fd7d2SChris Mason 			path->slots[0]++;
55478e73f275SChris Mason 		ret = 0;
5548925baeddSChris Mason 		goto done;
5549925baeddSChris Mason 	}
5550d97e63b6SChris Mason 
5551234b63a0SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
55528e73f275SChris Mason 		if (!path->nodes[level]) {
55538e73f275SChris Mason 			ret = 1;
55548e73f275SChris Mason 			goto done;
55558e73f275SChris Mason 		}
55565f39d397SChris Mason 
5557d97e63b6SChris Mason 		slot = path->slots[level] + 1;
5558d97e63b6SChris Mason 		c = path->nodes[level];
55595f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
5560d97e63b6SChris Mason 			level++;
55618e73f275SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
55628e73f275SChris Mason 				ret = 1;
55638e73f275SChris Mason 				goto done;
55648e73f275SChris Mason 			}
5565d97e63b6SChris Mason 			continue;
5566d97e63b6SChris Mason 		}
55675f39d397SChris Mason 
5568925baeddSChris Mason 		if (next) {
5569bd681513SChris Mason 			btrfs_tree_unlock_rw(next, next_rw_lock);
55705f39d397SChris Mason 			free_extent_buffer(next);
5571925baeddSChris Mason 		}
55725f39d397SChris Mason 
55738e73f275SChris Mason 		next = c;
5574bd681513SChris Mason 		next_rw_lock = path->locks[level];
55758e73f275SChris Mason 		ret = read_block_for_search(NULL, root, path, &next, level,
55765d9e75c4SJan Schmidt 					    slot, &key, 0);
55778e73f275SChris Mason 		if (ret == -EAGAIN)
55788e73f275SChris Mason 			goto again;
55795f39d397SChris Mason 
558076a05b35SChris Mason 		if (ret < 0) {
5581b3b4aa74SDavid Sterba 			btrfs_release_path(path);
558276a05b35SChris Mason 			goto done;
558376a05b35SChris Mason 		}
558476a05b35SChris Mason 
55855cd57b2cSChris Mason 		if (!path->skip_locking) {
5586bd681513SChris Mason 			ret = btrfs_try_tree_read_lock(next);
5587d42244a0SJan Schmidt 			if (!ret && time_seq) {
5588d42244a0SJan Schmidt 				/*
5589d42244a0SJan Schmidt 				 * If we don't get the lock, we may be racing
5590d42244a0SJan Schmidt 				 * with push_leaf_left, holding that lock while
5591d42244a0SJan Schmidt 				 * itself waiting for the leaf we've currently
5592d42244a0SJan Schmidt 				 * locked. To solve this situation, we give up
5593d42244a0SJan Schmidt 				 * on our lock and cycle.
5594d42244a0SJan Schmidt 				 */
5595cf538830SJan Schmidt 				free_extent_buffer(next);
5596d42244a0SJan Schmidt 				btrfs_release_path(path);
5597d42244a0SJan Schmidt 				cond_resched();
5598d42244a0SJan Schmidt 				goto again;
5599d42244a0SJan Schmidt 			}
56008e73f275SChris Mason 			if (!ret) {
56018e73f275SChris Mason 				btrfs_set_path_blocking(path);
5602bd681513SChris Mason 				btrfs_tree_read_lock(next);
5603bd681513SChris Mason 				btrfs_clear_path_blocking(path, next,
5604bd681513SChris Mason 							  BTRFS_READ_LOCK);
56058e73f275SChris Mason 			}
5606bd681513SChris Mason 			next_rw_lock = BTRFS_READ_LOCK;
5607bd681513SChris Mason 		}
5608d97e63b6SChris Mason 		break;
5609d97e63b6SChris Mason 	}
5610d97e63b6SChris Mason 	path->slots[level] = slot;
5611d97e63b6SChris Mason 	while (1) {
5612d97e63b6SChris Mason 		level--;
5613d97e63b6SChris Mason 		c = path->nodes[level];
5614925baeddSChris Mason 		if (path->locks[level])
5615bd681513SChris Mason 			btrfs_tree_unlock_rw(c, path->locks[level]);
56168e73f275SChris Mason 
56175f39d397SChris Mason 		free_extent_buffer(c);
5618d97e63b6SChris Mason 		path->nodes[level] = next;
5619d97e63b6SChris Mason 		path->slots[level] = 0;
5620a74a4b97SChris Mason 		if (!path->skip_locking)
5621bd681513SChris Mason 			path->locks[level] = next_rw_lock;
5622d97e63b6SChris Mason 		if (!level)
5623d97e63b6SChris Mason 			break;
5624b4ce94deSChris Mason 
56258e73f275SChris Mason 		ret = read_block_for_search(NULL, root, path, &next, level,
56265d9e75c4SJan Schmidt 					    0, &key, 0);
56278e73f275SChris Mason 		if (ret == -EAGAIN)
56288e73f275SChris Mason 			goto again;
56298e73f275SChris Mason 
563076a05b35SChris Mason 		if (ret < 0) {
5631b3b4aa74SDavid Sterba 			btrfs_release_path(path);
563276a05b35SChris Mason 			goto done;
563376a05b35SChris Mason 		}
563476a05b35SChris Mason 
56355cd57b2cSChris Mason 		if (!path->skip_locking) {
5636bd681513SChris Mason 			ret = btrfs_try_tree_read_lock(next);
56378e73f275SChris Mason 			if (!ret) {
56388e73f275SChris Mason 				btrfs_set_path_blocking(path);
5639bd681513SChris Mason 				btrfs_tree_read_lock(next);
5640bd681513SChris Mason 				btrfs_clear_path_blocking(path, next,
5641bd681513SChris Mason 							  BTRFS_READ_LOCK);
56428e73f275SChris Mason 			}
5643bd681513SChris Mason 			next_rw_lock = BTRFS_READ_LOCK;
5644bd681513SChris Mason 		}
5645d97e63b6SChris Mason 	}
56468e73f275SChris Mason 	ret = 0;
5647925baeddSChris Mason done:
5648f7c79f30SChris Mason 	unlock_up(path, 0, 1, 0, NULL);
56498e73f275SChris Mason 	path->leave_spinning = old_spinning;
56508e73f275SChris Mason 	if (!old_spinning)
56518e73f275SChris Mason 		btrfs_set_path_blocking(path);
56528e73f275SChris Mason 
56538e73f275SChris Mason 	return ret;
5654d97e63b6SChris Mason }
56550b86a832SChris Mason 
56563f157a2fSChris Mason /*
56573f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
56583f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
56593f157a2fSChris Mason  *
56603f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
56613f157a2fSChris Mason  */
56620b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
56630b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
56640b86a832SChris Mason 			int type)
56650b86a832SChris Mason {
56660b86a832SChris Mason 	struct btrfs_key found_key;
56670b86a832SChris Mason 	struct extent_buffer *leaf;
5668e02119d5SChris Mason 	u32 nritems;
56690b86a832SChris Mason 	int ret;
56700b86a832SChris Mason 
56710b86a832SChris Mason 	while (1) {
56720b86a832SChris Mason 		if (path->slots[0] == 0) {
5673b4ce94deSChris Mason 			btrfs_set_path_blocking(path);
56740b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
56750b86a832SChris Mason 			if (ret != 0)
56760b86a832SChris Mason 				return ret;
56770b86a832SChris Mason 		} else {
56780b86a832SChris Mason 			path->slots[0]--;
56790b86a832SChris Mason 		}
56800b86a832SChris Mason 		leaf = path->nodes[0];
5681e02119d5SChris Mason 		nritems = btrfs_header_nritems(leaf);
5682e02119d5SChris Mason 		if (nritems == 0)
5683e02119d5SChris Mason 			return 1;
5684e02119d5SChris Mason 		if (path->slots[0] == nritems)
5685e02119d5SChris Mason 			path->slots[0]--;
5686e02119d5SChris Mason 
56870b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
5688e02119d5SChris Mason 		if (found_key.objectid < min_objectid)
5689e02119d5SChris Mason 			break;
56900a4eefbbSYan Zheng 		if (found_key.type == type)
56910a4eefbbSYan Zheng 			return 0;
5692e02119d5SChris Mason 		if (found_key.objectid == min_objectid &&
5693e02119d5SChris Mason 		    found_key.type < type)
5694e02119d5SChris Mason 			break;
56950b86a832SChris Mason 	}
56960b86a832SChris Mason 	return 1;
56970b86a832SChris Mason }
5698