xref: /openbmc/linux/fs/btrfs/ctree.c (revision 87b29b20)
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>
20eb60ceacSChris Mason #include "ctree.h"
21eb60ceacSChris Mason #include "disk-io.h"
227f5c1516SChris Mason #include "transaction.h"
235f39d397SChris Mason #include "print-tree.h"
24925baeddSChris Mason #include "locking.h"
259a8dd150SChris Mason 
26e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
27e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
28e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
29d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
30cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend);
315f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
325f39d397SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
33971a1f66SChris Mason 			  struct extent_buffer *src, int empty);
345f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
355f39d397SChris Mason 			      struct btrfs_root *root,
365f39d397SChris Mason 			      struct extent_buffer *dst_buf,
375f39d397SChris Mason 			      struct extent_buffer *src_buf);
38e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
39e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot);
40d97e63b6SChris Mason 
41df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p)
42df24a2b9SChris Mason {
43df24a2b9SChris Mason 	memset(p, 0, sizeof(*p));
44df24a2b9SChris Mason }
45df24a2b9SChris Mason 
462c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
472c90e5d6SChris Mason {
48df24a2b9SChris Mason 	struct btrfs_path *path;
49df24a2b9SChris Mason 	path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
502cc58cf2SChris Mason 	if (path) {
51df24a2b9SChris Mason 		btrfs_init_path(path);
522cc58cf2SChris Mason 		path->reada = 1;
532cc58cf2SChris Mason 	}
54df24a2b9SChris Mason 	return path;
552c90e5d6SChris Mason }
562c90e5d6SChris Mason 
57d352ac68SChris Mason /* this also releases the path */
582c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
592c90e5d6SChris Mason {
60df24a2b9SChris Mason 	btrfs_release_path(NULL, p);
612c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
622c90e5d6SChris Mason }
632c90e5d6SChris Mason 
64d352ac68SChris Mason /*
65d352ac68SChris Mason  * path release drops references on the extent buffers in the path
66d352ac68SChris Mason  * and it drops any locks held by this path
67d352ac68SChris Mason  *
68d352ac68SChris Mason  * It is safe to call this on paths that no locks or extent buffers held.
69d352ac68SChris Mason  */
70e02119d5SChris Mason void noinline btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
71eb60ceacSChris Mason {
72eb60ceacSChris Mason 	int i;
73a2135011SChris Mason 
74234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
753f157a2fSChris Mason 		p->slots[i] = 0;
76eb60ceacSChris Mason 		if (!p->nodes[i])
77925baeddSChris Mason 			continue;
78925baeddSChris Mason 		if (p->locks[i]) {
79925baeddSChris Mason 			btrfs_tree_unlock(p->nodes[i]);
80925baeddSChris Mason 			p->locks[i] = 0;
81925baeddSChris Mason 		}
825f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
833f157a2fSChris Mason 		p->nodes[i] = NULL;
84eb60ceacSChris Mason 	}
85eb60ceacSChris Mason }
86eb60ceacSChris Mason 
87d352ac68SChris Mason /*
88d352ac68SChris Mason  * safely gets a reference on the root node of a tree.  A lock
89d352ac68SChris Mason  * is not taken, so a concurrent writer may put a different node
90d352ac68SChris Mason  * at the root of the tree.  See btrfs_lock_root_node for the
91d352ac68SChris Mason  * looping required.
92d352ac68SChris Mason  *
93d352ac68SChris Mason  * The extent buffer returned by this has a reference taken, so
94d352ac68SChris Mason  * it won't disappear.  It may stop being the root of the tree
95d352ac68SChris Mason  * at any time because there are no locks held.
96d352ac68SChris Mason  */
97925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
98925baeddSChris Mason {
99925baeddSChris Mason 	struct extent_buffer *eb;
100925baeddSChris Mason 	spin_lock(&root->node_lock);
101925baeddSChris Mason 	eb = root->node;
102925baeddSChris Mason 	extent_buffer_get(eb);
103925baeddSChris Mason 	spin_unlock(&root->node_lock);
104925baeddSChris Mason 	return eb;
105925baeddSChris Mason }
106925baeddSChris Mason 
107d352ac68SChris Mason /* loop around taking references on and locking the root node of the
108d352ac68SChris Mason  * tree until you end up with a lock on the root.  A locked buffer
109d352ac68SChris Mason  * is returned, with a reference held.
110d352ac68SChris Mason  */
111925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
112925baeddSChris Mason {
113925baeddSChris Mason 	struct extent_buffer *eb;
114925baeddSChris Mason 
115925baeddSChris Mason 	while(1) {
116925baeddSChris Mason 		eb = btrfs_root_node(root);
117925baeddSChris Mason 		btrfs_tree_lock(eb);
118925baeddSChris Mason 
119925baeddSChris Mason 		spin_lock(&root->node_lock);
120925baeddSChris Mason 		if (eb == root->node) {
121925baeddSChris Mason 			spin_unlock(&root->node_lock);
122925baeddSChris Mason 			break;
123925baeddSChris Mason 		}
124925baeddSChris Mason 		spin_unlock(&root->node_lock);
125925baeddSChris Mason 
126925baeddSChris Mason 		btrfs_tree_unlock(eb);
127925baeddSChris Mason 		free_extent_buffer(eb);
128925baeddSChris Mason 	}
129925baeddSChris Mason 	return eb;
130925baeddSChris Mason }
131925baeddSChris Mason 
132d352ac68SChris Mason /* cowonly root (everything not a reference counted cow subvolume), just get
133d352ac68SChris Mason  * put onto a simple dirty list.  transaction.c walks this to make sure they
134d352ac68SChris Mason  * get properly updated on disk.
135d352ac68SChris Mason  */
1360b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root)
1370b86a832SChris Mason {
1380b86a832SChris Mason 	if (root->track_dirty && list_empty(&root->dirty_list)) {
1390b86a832SChris Mason 		list_add(&root->dirty_list,
1400b86a832SChris Mason 			 &root->fs_info->dirty_cowonly_roots);
1410b86a832SChris Mason 	}
1420b86a832SChris Mason }
1430b86a832SChris Mason 
144d352ac68SChris Mason /*
145d352ac68SChris Mason  * used by snapshot creation to make a copy of a root for a tree with
146d352ac68SChris Mason  * a given objectid.  The buffer with the new root node is returned in
147d352ac68SChris Mason  * cow_ret, and this func returns zero on success or a negative error code.
148d352ac68SChris Mason  */
149be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
150be20aa9dSChris Mason 		      struct btrfs_root *root,
151be20aa9dSChris Mason 		      struct extent_buffer *buf,
152be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
153be20aa9dSChris Mason {
154be20aa9dSChris Mason 	struct extent_buffer *cow;
155be20aa9dSChris Mason 	u32 nritems;
156be20aa9dSChris Mason 	int ret = 0;
157be20aa9dSChris Mason 	int level;
1584aec2b52SChris Mason 	struct btrfs_root *new_root;
159be20aa9dSChris Mason 
1604aec2b52SChris Mason 	new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
1614aec2b52SChris Mason 	if (!new_root)
1624aec2b52SChris Mason 		return -ENOMEM;
1634aec2b52SChris Mason 
1644aec2b52SChris Mason 	memcpy(new_root, root, sizeof(*new_root));
1654aec2b52SChris Mason 	new_root->root_key.objectid = new_root_objectid;
166be20aa9dSChris Mason 
167be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
168be20aa9dSChris Mason 		root->fs_info->running_transaction->transid);
169be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
170be20aa9dSChris Mason 
171be20aa9dSChris Mason 	level = btrfs_header_level(buf);
172be20aa9dSChris Mason 	nritems = btrfs_header_nritems(buf);
17331840ae1SZheng Yan 
17431840ae1SZheng Yan 	cow = btrfs_alloc_free_block(trans, new_root, buf->len, 0,
17531840ae1SZheng Yan 				     new_root_objectid, trans->transid,
176be20aa9dSChris Mason 				     level, buf->start, 0);
1774aec2b52SChris Mason 	if (IS_ERR(cow)) {
1784aec2b52SChris Mason 		kfree(new_root);
179be20aa9dSChris Mason 		return PTR_ERR(cow);
1804aec2b52SChris Mason 	}
181be20aa9dSChris Mason 
182be20aa9dSChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
183be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
184be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
185be20aa9dSChris Mason 	btrfs_set_header_owner(cow, new_root_objectid);
18663b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
187be20aa9dSChris Mason 
1882b82032cSYan Zheng 	write_extent_buffer(cow, root->fs_info->fsid,
1892b82032cSYan Zheng 			    (unsigned long)btrfs_header_fsid(cow),
1902b82032cSYan Zheng 			    BTRFS_FSID_SIZE);
1912b82032cSYan Zheng 
192be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
19331840ae1SZheng Yan 	ret = btrfs_inc_ref(trans, new_root, buf, cow, NULL);
1944aec2b52SChris Mason 	kfree(new_root);
1954aec2b52SChris Mason 
196be20aa9dSChris Mason 	if (ret)
197be20aa9dSChris Mason 		return ret;
198be20aa9dSChris Mason 
199be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
200be20aa9dSChris Mason 	*cow_ret = cow;
201be20aa9dSChris Mason 	return 0;
202be20aa9dSChris Mason }
203be20aa9dSChris Mason 
204d352ac68SChris Mason /*
205d352ac68SChris Mason  * does the dirty work in cow of a single block.  The parent block
206d352ac68SChris Mason  * (if supplied) is updated to point to the new cow copy.  The new
207d352ac68SChris Mason  * buffer is marked dirty and returned locked.  If you modify the block
208d352ac68SChris Mason  * it needs to be marked dirty again.
209d352ac68SChris Mason  *
210d352ac68SChris Mason  * search_start -- an allocation hint for the new block
211d352ac68SChris Mason  *
212d352ac68SChris Mason  * empty_size -- a hint that you plan on doing more cow.  This is the size in bytes
213d352ac68SChris Mason  * the allocator should try to find free next to the block it returns.  This is
214d352ac68SChris Mason  * just a hint and may be ignored by the allocator.
215d352ac68SChris Mason  *
216d352ac68SChris Mason  * prealloc_dest -- if you have already reserved a destination for the cow,
217d352ac68SChris Mason  * this uses that block instead of allocating a new one.  btrfs_alloc_reserved_extent
218d352ac68SChris Mason  * is used to finish the allocation.
219d352ac68SChris Mason  */
220b2950863SChristoph Hellwig static int noinline __btrfs_cow_block(struct btrfs_trans_handle *trans,
2215f39d397SChris Mason 			     struct btrfs_root *root,
2225f39d397SChris Mason 			     struct extent_buffer *buf,
2235f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
2245f39d397SChris Mason 			     struct extent_buffer **cow_ret,
22565b51a00SChris Mason 			     u64 search_start, u64 empty_size,
22665b51a00SChris Mason 			     u64 prealloc_dest)
2276702ed49SChris Mason {
22831840ae1SZheng Yan 	u64 parent_start;
2295f39d397SChris Mason 	struct extent_buffer *cow;
2307bb86316SChris Mason 	u32 nritems;
2316702ed49SChris Mason 	int ret = 0;
2327bb86316SChris Mason 	int level;
233925baeddSChris Mason 	int unlock_orig = 0;
2346702ed49SChris Mason 
235925baeddSChris Mason 	if (*cow_ret == buf)
236925baeddSChris Mason 		unlock_orig = 1;
237925baeddSChris Mason 
238925baeddSChris Mason 	WARN_ON(!btrfs_tree_locked(buf));
239925baeddSChris Mason 
24031840ae1SZheng Yan 	if (parent)
24131840ae1SZheng Yan 		parent_start = parent->start;
24231840ae1SZheng Yan 	else
24331840ae1SZheng Yan 		parent_start = 0;
24431840ae1SZheng Yan 
2457bb86316SChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
2467bb86316SChris Mason 		root->fs_info->running_transaction->transid);
2476702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
2485f39d397SChris Mason 
2497bb86316SChris Mason 	level = btrfs_header_level(buf);
2507bb86316SChris Mason 	nritems = btrfs_header_nritems(buf);
25131840ae1SZheng Yan 
25265b51a00SChris Mason 	if (prealloc_dest) {
25365b51a00SChris Mason 		struct btrfs_key ins;
25465b51a00SChris Mason 
25565b51a00SChris Mason 		ins.objectid = prealloc_dest;
25665b51a00SChris Mason 		ins.offset = buf->len;
25765b51a00SChris Mason 		ins.type = BTRFS_EXTENT_ITEM_KEY;
25865b51a00SChris Mason 
25931840ae1SZheng Yan 		ret = btrfs_alloc_reserved_extent(trans, root, parent_start,
26065b51a00SChris Mason 						  root->root_key.objectid,
2613bb1a1bcSYan Zheng 						  trans->transid, level, &ins);
26265b51a00SChris Mason 		BUG_ON(ret);
26365b51a00SChris Mason 		cow = btrfs_init_new_buffer(trans, root, prealloc_dest,
26465b51a00SChris Mason 					    buf->len);
26565b51a00SChris Mason 	} else {
266925baeddSChris Mason 		cow = btrfs_alloc_free_block(trans, root, buf->len,
26731840ae1SZheng Yan 					     parent_start,
2687bb86316SChris Mason 					     root->root_key.objectid,
26931840ae1SZheng Yan 					     trans->transid, level,
27031840ae1SZheng Yan 					     search_start, empty_size);
27165b51a00SChris Mason 	}
2726702ed49SChris Mason 	if (IS_ERR(cow))
2736702ed49SChris Mason 		return PTR_ERR(cow);
2746702ed49SChris Mason 
2755f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
276db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
2775f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
2785f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
27963b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
2806702ed49SChris Mason 
2812b82032cSYan Zheng 	write_extent_buffer(cow, root->fs_info->fsid,
2822b82032cSYan Zheng 			    (unsigned long)btrfs_header_fsid(cow),
2832b82032cSYan Zheng 			    BTRFS_FSID_SIZE);
2842b82032cSYan Zheng 
2855f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
2865f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
28731840ae1SZheng Yan 		u32 nr_extents;
28831840ae1SZheng Yan 		ret = btrfs_inc_ref(trans, root, buf, cow, &nr_extents);
2896702ed49SChris Mason 		if (ret)
2906702ed49SChris Mason 			return ret;
29131840ae1SZheng Yan 
29231840ae1SZheng Yan 		ret = btrfs_cache_ref(trans, root, buf, nr_extents);
29331840ae1SZheng Yan 		WARN_ON(ret);
2941a40e23bSZheng Yan 	} else if (btrfs_header_owner(buf) == BTRFS_TREE_RELOC_OBJECTID) {
2951a40e23bSZheng Yan 		/*
2961a40e23bSZheng Yan 		 * There are only two places that can drop reference to
2971a40e23bSZheng Yan 		 * tree blocks owned by living reloc trees, one is here,
298f82d02d9SYan Zheng 		 * the other place is btrfs_drop_subtree. In both places,
2991a40e23bSZheng Yan 		 * we check reference count while tree block is locked.
3001a40e23bSZheng Yan 		 * Furthermore, if reference count is one, it won't get
3011a40e23bSZheng Yan 		 * increased by someone else.
3021a40e23bSZheng Yan 		 */
3031a40e23bSZheng Yan 		u32 refs;
3041a40e23bSZheng Yan 		ret = btrfs_lookup_extent_ref(trans, root, buf->start,
3051a40e23bSZheng Yan 					      buf->len, &refs);
3061a40e23bSZheng Yan 		BUG_ON(ret);
3071a40e23bSZheng Yan 		if (refs == 1) {
3081a40e23bSZheng Yan 			ret = btrfs_update_ref(trans, root, buf, cow,
3091a40e23bSZheng Yan 					       0, nritems);
3101a40e23bSZheng Yan 			clean_tree_block(trans, root, buf);
3111a40e23bSZheng Yan 		} else {
3121a40e23bSZheng Yan 			ret = btrfs_inc_ref(trans, root, buf, cow, NULL);
3131a40e23bSZheng Yan 		}
3141a40e23bSZheng Yan 		BUG_ON(ret);
3156702ed49SChris Mason 	} else {
31631840ae1SZheng Yan 		ret = btrfs_update_ref(trans, root, buf, cow, 0, nritems);
31731840ae1SZheng Yan 		if (ret)
31831840ae1SZheng Yan 			return ret;
3196702ed49SChris Mason 		clean_tree_block(trans, root, buf);
3206702ed49SChris Mason 	}
3216702ed49SChris Mason 
3221a40e23bSZheng Yan 	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) {
3231a40e23bSZheng Yan 		ret = btrfs_reloc_tree_cache_ref(trans, root, cow, buf->start);
3241a40e23bSZheng Yan 		WARN_ON(ret);
3251a40e23bSZheng Yan 	}
3261a40e23bSZheng Yan 
3276702ed49SChris Mason 	if (buf == root->node) {
328925baeddSChris Mason 		WARN_ON(parent && parent != buf);
329925baeddSChris Mason 
330925baeddSChris Mason 		spin_lock(&root->node_lock);
3316702ed49SChris Mason 		root->node = cow;
3325f39d397SChris Mason 		extent_buffer_get(cow);
333925baeddSChris Mason 		spin_unlock(&root->node_lock);
334925baeddSChris Mason 
3356702ed49SChris Mason 		if (buf != root->commit_root) {
336db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
33731840ae1SZheng Yan 					  buf->len, buf->start,
33831840ae1SZheng Yan 					  root->root_key.objectid,
33931840ae1SZheng Yan 					  btrfs_header_generation(buf),
3403bb1a1bcSYan Zheng 					  level, 1);
3416702ed49SChris Mason 		}
3425f39d397SChris Mason 		free_extent_buffer(buf);
3430b86a832SChris Mason 		add_root_to_dirty_list(root);
3446702ed49SChris Mason 	} else {
3455f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
346db94535dSChris Mason 					cow->start);
34774493f7aSChris Mason 		WARN_ON(trans->transid == 0);
34874493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
34974493f7aSChris Mason 					      trans->transid);
3506702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
3515f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
3527bb86316SChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len,
35331840ae1SZheng Yan 				  parent_start, btrfs_header_owner(parent),
3543bb1a1bcSYan Zheng 				  btrfs_header_generation(parent), level, 1);
3556702ed49SChris Mason 	}
356925baeddSChris Mason 	if (unlock_orig)
357925baeddSChris Mason 		btrfs_tree_unlock(buf);
3585f39d397SChris Mason 	free_extent_buffer(buf);
3596702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
3606702ed49SChris Mason 	*cow_ret = cow;
3616702ed49SChris Mason 	return 0;
3626702ed49SChris Mason }
3636702ed49SChris Mason 
364d352ac68SChris Mason /*
365d352ac68SChris Mason  * cows a single block, see __btrfs_cow_block for the real work.
366d352ac68SChris Mason  * This version of it has extra checks so that a block isn't cow'd more than
367d352ac68SChris Mason  * once per transaction, as long as it hasn't been written yet
368d352ac68SChris Mason  */
369e02119d5SChris Mason int noinline btrfs_cow_block(struct btrfs_trans_handle *trans,
3705f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
3715f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
37265b51a00SChris Mason 		    struct extent_buffer **cow_ret, u64 prealloc_dest)
37302217ed2SChris Mason {
3746702ed49SChris Mason 	u64 search_start;
375f510cfecSChris Mason 	int ret;
376dc17ff8fSChris Mason 
377ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
378ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
379ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
380ccd467d6SChris Mason 		WARN_ON(1);
381ccd467d6SChris Mason 	}
382ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
383ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
384ccd467d6SChris Mason 		       root->fs_info->generation);
385ccd467d6SChris Mason 		WARN_ON(1);
386ccd467d6SChris Mason 	}
387dc17ff8fSChris Mason 
38863b10fc4SChris Mason 	spin_lock(&root->fs_info->hash_lock);
3895b21f2edSZheng Yan 	if (btrfs_header_generation(buf) == trans->transid &&
3905b21f2edSZheng Yan 	    btrfs_header_owner(buf) == root->root_key.objectid &&
39163b10fc4SChris Mason 	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) {
39202217ed2SChris Mason 		*cow_ret = buf;
39363b10fc4SChris Mason 		spin_unlock(&root->fs_info->hash_lock);
39465b51a00SChris Mason 		WARN_ON(prealloc_dest);
39502217ed2SChris Mason 		return 0;
39602217ed2SChris Mason 	}
39763b10fc4SChris Mason 	spin_unlock(&root->fs_info->hash_lock);
3980b86a832SChris Mason 	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
399f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
40065b51a00SChris Mason 				 parent_slot, cow_ret, search_start, 0,
40165b51a00SChris Mason 				 prealloc_dest);
402f510cfecSChris Mason 	return ret;
4032c90e5d6SChris Mason }
4046702ed49SChris Mason 
405d352ac68SChris Mason /*
406d352ac68SChris Mason  * helper function for defrag to decide if two blocks pointed to by a
407d352ac68SChris Mason  * node are actually close by
408d352ac68SChris Mason  */
4096b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
4106702ed49SChris Mason {
4116b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
4126702ed49SChris Mason 		return 1;
4136b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
4146702ed49SChris Mason 		return 1;
41502217ed2SChris Mason 	return 0;
41602217ed2SChris Mason }
41702217ed2SChris Mason 
418081e9573SChris Mason /*
419081e9573SChris Mason  * compare two keys in a memcmp fashion
420081e9573SChris Mason  */
421081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
422081e9573SChris Mason {
423081e9573SChris Mason 	struct btrfs_key k1;
424081e9573SChris Mason 
425081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
426081e9573SChris Mason 
427081e9573SChris Mason 	if (k1.objectid > k2->objectid)
428081e9573SChris Mason 		return 1;
429081e9573SChris Mason 	if (k1.objectid < k2->objectid)
430081e9573SChris Mason 		return -1;
431081e9573SChris Mason 	if (k1.type > k2->type)
432081e9573SChris Mason 		return 1;
433081e9573SChris Mason 	if (k1.type < k2->type)
434081e9573SChris Mason 		return -1;
435081e9573SChris Mason 	if (k1.offset > k2->offset)
436081e9573SChris Mason 		return 1;
437081e9573SChris Mason 	if (k1.offset < k2->offset)
438081e9573SChris Mason 		return -1;
439081e9573SChris Mason 	return 0;
440081e9573SChris Mason }
441081e9573SChris Mason 
442f3465ca4SJosef Bacik /*
443f3465ca4SJosef Bacik  * same as comp_keys only with two btrfs_key's
444f3465ca4SJosef Bacik  */
445f3465ca4SJosef Bacik static int comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2)
446f3465ca4SJosef Bacik {
447f3465ca4SJosef Bacik 	if (k1->objectid > k2->objectid)
448f3465ca4SJosef Bacik 		return 1;
449f3465ca4SJosef Bacik 	if (k1->objectid < k2->objectid)
450f3465ca4SJosef Bacik 		return -1;
451f3465ca4SJosef Bacik 	if (k1->type > k2->type)
452f3465ca4SJosef Bacik 		return 1;
453f3465ca4SJosef Bacik 	if (k1->type < k2->type)
454f3465ca4SJosef Bacik 		return -1;
455f3465ca4SJosef Bacik 	if (k1->offset > k2->offset)
456f3465ca4SJosef Bacik 		return 1;
457f3465ca4SJosef Bacik 	if (k1->offset < k2->offset)
458f3465ca4SJosef Bacik 		return -1;
459f3465ca4SJosef Bacik 	return 0;
460f3465ca4SJosef Bacik }
461081e9573SChris Mason 
462d352ac68SChris Mason /*
463d352ac68SChris Mason  * this is used by the defrag code to go through all the
464d352ac68SChris Mason  * leaves pointed to by a node and reallocate them so that
465d352ac68SChris Mason  * disk order is close to key order
466d352ac68SChris Mason  */
4676702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
4685f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
469a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
470a6b6e75eSChris Mason 		       struct btrfs_key *progress)
4716702ed49SChris Mason {
4726b80053dSChris Mason 	struct extent_buffer *cur;
4736702ed49SChris Mason 	u64 blocknr;
474ca7a79adSChris Mason 	u64 gen;
475e9d0b13bSChris Mason 	u64 search_start = *last_ret;
476e9d0b13bSChris Mason 	u64 last_block = 0;
4776702ed49SChris Mason 	u64 other;
4786702ed49SChris Mason 	u32 parent_nritems;
4796702ed49SChris Mason 	int end_slot;
4806702ed49SChris Mason 	int i;
4816702ed49SChris Mason 	int err = 0;
482f2183bdeSChris Mason 	int parent_level;
4836b80053dSChris Mason 	int uptodate;
4846b80053dSChris Mason 	u32 blocksize;
485081e9573SChris Mason 	int progress_passed = 0;
486081e9573SChris Mason 	struct btrfs_disk_key disk_key;
4876702ed49SChris Mason 
4885708b959SChris Mason 	parent_level = btrfs_header_level(parent);
4895708b959SChris Mason 	if (cache_only && parent_level != 1)
4905708b959SChris Mason 		return 0;
4915708b959SChris Mason 
4926702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
4936702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
4946702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
4956702ed49SChris Mason 		WARN_ON(1);
4966702ed49SChris Mason 	}
4976702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
4986702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
4996702ed49SChris Mason 		       root->fs_info->generation);
5006702ed49SChris Mason 		WARN_ON(1);
5016702ed49SChris Mason 	}
50286479a04SChris Mason 
5036b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
5046b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
5056702ed49SChris Mason 	end_slot = parent_nritems;
5066702ed49SChris Mason 
5076702ed49SChris Mason 	if (parent_nritems == 1)
5086702ed49SChris Mason 		return 0;
5096702ed49SChris Mason 
5106702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
5116702ed49SChris Mason 		int close = 1;
512a6b6e75eSChris Mason 
5135708b959SChris Mason 		if (!parent->map_token) {
5145708b959SChris Mason 			map_extent_buffer(parent,
5155708b959SChris Mason 					btrfs_node_key_ptr_offset(i),
5165708b959SChris Mason 					sizeof(struct btrfs_key_ptr),
5175708b959SChris Mason 					&parent->map_token, &parent->kaddr,
5185708b959SChris Mason 					&parent->map_start, &parent->map_len,
5195708b959SChris Mason 					KM_USER1);
5205708b959SChris Mason 		}
521081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
522081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
523081e9573SChris Mason 			continue;
524081e9573SChris Mason 
525081e9573SChris Mason 		progress_passed = 1;
5266b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
527ca7a79adSChris Mason 		gen = btrfs_node_ptr_generation(parent, i);
528e9d0b13bSChris Mason 		if (last_block == 0)
529e9d0b13bSChris Mason 			last_block = blocknr;
5305708b959SChris Mason 
5316702ed49SChris Mason 		if (i > 0) {
5326b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
5336b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
5346702ed49SChris Mason 		}
5350ef3e66bSChris Mason 		if (!close && i < end_slot - 2) {
5366b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
5376b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
5386702ed49SChris Mason 		}
539e9d0b13bSChris Mason 		if (close) {
540e9d0b13bSChris Mason 			last_block = blocknr;
5416702ed49SChris Mason 			continue;
542e9d0b13bSChris Mason 		}
5435708b959SChris Mason 		if (parent->map_token) {
5445708b959SChris Mason 			unmap_extent_buffer(parent, parent->map_token,
5455708b959SChris Mason 					    KM_USER1);
5465708b959SChris Mason 			parent->map_token = NULL;
5475708b959SChris Mason 		}
5486702ed49SChris Mason 
5496b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
5506b80053dSChris Mason 		if (cur)
5511259ab75SChris Mason 			uptodate = btrfs_buffer_uptodate(cur, gen);
5526b80053dSChris Mason 		else
5536b80053dSChris Mason 			uptodate = 0;
5545708b959SChris Mason 		if (!cur || !uptodate) {
5556702ed49SChris Mason 			if (cache_only) {
5566b80053dSChris Mason 				free_extent_buffer(cur);
5576702ed49SChris Mason 				continue;
5586702ed49SChris Mason 			}
5596b80053dSChris Mason 			if (!cur) {
5606b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
561ca7a79adSChris Mason 							 blocksize, gen);
5626b80053dSChris Mason 			} else if (!uptodate) {
563ca7a79adSChris Mason 				btrfs_read_buffer(cur, gen);
5646702ed49SChris Mason 			}
565f2183bdeSChris Mason 		}
566e9d0b13bSChris Mason 		if (search_start == 0)
5676b80053dSChris Mason 			search_start = last_block;
568e9d0b13bSChris Mason 
569e7a84565SChris Mason 		btrfs_tree_lock(cur);
5706b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
571e7a84565SChris Mason 					&cur, search_start,
5726b80053dSChris Mason 					min(16 * blocksize,
57365b51a00SChris Mason 					    (end_slot - i) * blocksize), 0);
574252c38f0SYan 		if (err) {
575e7a84565SChris Mason 			btrfs_tree_unlock(cur);
5766b80053dSChris Mason 			free_extent_buffer(cur);
5776702ed49SChris Mason 			break;
578252c38f0SYan 		}
579e7a84565SChris Mason 		search_start = cur->start;
580e7a84565SChris Mason 		last_block = cur->start;
581f2183bdeSChris Mason 		*last_ret = search_start;
582e7a84565SChris Mason 		btrfs_tree_unlock(cur);
583e7a84565SChris Mason 		free_extent_buffer(cur);
5846702ed49SChris Mason 	}
5855708b959SChris Mason 	if (parent->map_token) {
5865708b959SChris Mason 		unmap_extent_buffer(parent, parent->map_token,
5875708b959SChris Mason 				    KM_USER1);
5885708b959SChris Mason 		parent->map_token = NULL;
5895708b959SChris Mason 	}
5906702ed49SChris Mason 	return err;
5916702ed49SChris Mason }
5926702ed49SChris Mason 
59374123bd7SChris Mason /*
59474123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
59574123bd7SChris Mason  * this returns the address of the start of the last item,
59674123bd7SChris Mason  * which is the stop of the leaf data stack
59774123bd7SChris Mason  */
598123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
5995f39d397SChris Mason 					 struct extent_buffer *leaf)
600be0e5c09SChris Mason {
6015f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
602be0e5c09SChris Mason 	if (nr == 0)
603123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
6045f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
605be0e5c09SChris Mason }
606be0e5c09SChris Mason 
607d352ac68SChris Mason /*
608d352ac68SChris Mason  * extra debugging checks to make sure all the items in a key are
609d352ac68SChris Mason  * well formed and in the proper order
610d352ac68SChris Mason  */
611123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
612123abc88SChris Mason 		      int level)
613aa5d6bedSChris Mason {
6145f39d397SChris Mason 	struct extent_buffer *parent = NULL;
6155f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
6165f39d397SChris Mason 	struct btrfs_disk_key parent_key;
6175f39d397SChris Mason 	struct btrfs_disk_key node_key;
618aa5d6bedSChris Mason 	int parent_slot;
6198d7be552SChris Mason 	int slot;
6208d7be552SChris Mason 	struct btrfs_key cpukey;
6215f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
622aa5d6bedSChris Mason 
623aa5d6bedSChris Mason 	if (path->nodes[level + 1])
6245f39d397SChris Mason 		parent = path->nodes[level + 1];
625a1f39630SAneesh 
6268d7be552SChris Mason 	slot = path->slots[level];
6277518a238SChris Mason 	BUG_ON(nritems == 0);
6287518a238SChris Mason 	if (parent) {
629a1f39630SAneesh 		parent_slot = path->slots[level + 1];
6305f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
6315f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
6325f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
633e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
6341d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
635db94535dSChris Mason 		       btrfs_header_bytenr(node));
636aa5d6bedSChris Mason 	}
637123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
6388d7be552SChris Mason 	if (slot != 0) {
6395f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
6405f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
6415f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
6428d7be552SChris Mason 	}
6438d7be552SChris Mason 	if (slot < nritems - 1) {
6445f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
6455f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
6465f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
647aa5d6bedSChris Mason 	}
648aa5d6bedSChris Mason 	return 0;
649aa5d6bedSChris Mason }
650aa5d6bedSChris Mason 
651d352ac68SChris Mason /*
652d352ac68SChris Mason  * extra checking to make sure all the items in a leaf are
653d352ac68SChris Mason  * well formed and in the proper order
654d352ac68SChris Mason  */
655123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
656123abc88SChris Mason 		      int level)
657aa5d6bedSChris Mason {
6585f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
6595f39d397SChris Mason 	struct extent_buffer *parent = NULL;
660aa5d6bedSChris Mason 	int parent_slot;
6618d7be552SChris Mason 	struct btrfs_key cpukey;
6625f39d397SChris Mason 	struct btrfs_disk_key parent_key;
6635f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
6645f39d397SChris Mason 	int slot = path->slots[0];
6658d7be552SChris Mason 
6665f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
667aa5d6bedSChris Mason 
668aa5d6bedSChris Mason 	if (path->nodes[level + 1])
6695f39d397SChris Mason 		parent = path->nodes[level + 1];
6707518a238SChris Mason 
6717518a238SChris Mason 	if (nritems == 0)
6727518a238SChris Mason 		return 0;
6737518a238SChris Mason 
6747518a238SChris Mason 	if (parent) {
675a1f39630SAneesh 		parent_slot = path->slots[level + 1];
6765f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
6775f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
6786702ed49SChris Mason 
6795f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
680e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
6811d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
682db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
683aa5d6bedSChris Mason 	}
6845f39d397SChris Mason #if 0
6855f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
6865f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
6875f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
6885f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
6895f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6905f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
6915f39d397SChris Mason 			BUG_ON(1);
6925f39d397SChris Mason 		}
6935f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
6945f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
6955f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6965f39d397SChris Mason 			printk("slot %d offset bad\n", i);
6975f39d397SChris Mason 			BUG_ON(1);
6985f39d397SChris Mason 		}
6995f39d397SChris Mason 		if (i == 0) {
7005f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
7015f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
7025f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
7035f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
7045f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
7055f39d397SChris Mason 				BUG_ON(1);
7065f39d397SChris Mason 			}
7075f39d397SChris Mason 		}
7085f39d397SChris Mason 	}
7095f39d397SChris Mason 	if (nritems > 0) {
7105f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
7115f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
7125f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
7135f39d397SChris Mason 				BUG_ON(1);
7145f39d397SChris Mason 		}
7155f39d397SChris Mason 	}
7165f39d397SChris Mason #endif
7175f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
7185f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
7195f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
7205f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
7215f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
7225f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
7235f39d397SChris Mason 			BUG_ON(1);
7245f39d397SChris Mason 		}
7255f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
7265f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
7275f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
7285f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
7295f39d397SChris Mason 			BUG_ON(1);
7305f39d397SChris Mason 		}
731aa5d6bedSChris Mason 	}
7328d7be552SChris Mason 	if (slot < nritems - 1) {
7335f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
7345f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
7355f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
7365f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
7375f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
7385f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
7395f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
7405f39d397SChris Mason 			BUG_ON(1);
741aa5d6bedSChris Mason 		}
7425f39d397SChris Mason 	}
7435f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
7445f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
745aa5d6bedSChris Mason 	return 0;
746aa5d6bedSChris Mason }
747aa5d6bedSChris Mason 
74898ed5174SChris Mason static int noinline check_block(struct btrfs_root *root,
74998ed5174SChris Mason 				struct btrfs_path *path, int level)
750aa5d6bedSChris Mason {
751f188591eSChris Mason 	u64 found_start;
75285d824c4SChris Mason 	return 0;
753f188591eSChris Mason 	if (btrfs_header_level(path->nodes[level]) != level)
754f188591eSChris Mason 	    printk("warning: bad level %Lu wanted %d found %d\n",
755f188591eSChris Mason 		   path->nodes[level]->start, level,
756f188591eSChris Mason 		   btrfs_header_level(path->nodes[level]));
757f188591eSChris Mason 	found_start = btrfs_header_bytenr(path->nodes[level]);
758f188591eSChris Mason 	if (found_start != path->nodes[level]->start) {
759f188591eSChris Mason 	    printk("warning: bad bytentr %Lu found %Lu\n",
760f188591eSChris Mason 		   path->nodes[level]->start, found_start);
761f188591eSChris Mason 	}
762db94535dSChris Mason #if 0
7635f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
7645f39d397SChris Mason 
765479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
766479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
767479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
7685f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
769db94535dSChris Mason 		return 1;
7705f39d397SChris Mason 	}
771db94535dSChris Mason #endif
772aa5d6bedSChris Mason 	if (level == 0)
773123abc88SChris Mason 		return check_leaf(root, path, level);
774123abc88SChris Mason 	return check_node(root, path, level);
775aa5d6bedSChris Mason }
776aa5d6bedSChris Mason 
77774123bd7SChris Mason /*
7785f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
7795f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
7805f39d397SChris Mason  *
78174123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
78274123bd7SChris Mason  * the place where you would insert key if it is not found in
78374123bd7SChris Mason  * the array.
78474123bd7SChris Mason  *
78574123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
78674123bd7SChris Mason  */
787e02119d5SChris Mason static noinline int generic_bin_search(struct extent_buffer *eb,
788e02119d5SChris Mason 				       unsigned long p,
7895f39d397SChris Mason 				       int item_size, struct btrfs_key *key,
790be0e5c09SChris Mason 				       int max, int *slot)
791be0e5c09SChris Mason {
792be0e5c09SChris Mason 	int low = 0;
793be0e5c09SChris Mason 	int high = max;
794be0e5c09SChris Mason 	int mid;
795be0e5c09SChris Mason 	int ret;
796479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
7975f39d397SChris Mason 	struct btrfs_disk_key unaligned;
7985f39d397SChris Mason 	unsigned long offset;
7995f39d397SChris Mason 	char *map_token = NULL;
8005f39d397SChris Mason 	char *kaddr = NULL;
8015f39d397SChris Mason 	unsigned long map_start = 0;
8025f39d397SChris Mason 	unsigned long map_len = 0;
803479965d6SChris Mason 	int err;
804be0e5c09SChris Mason 
805be0e5c09SChris Mason 	while(low < high) {
806be0e5c09SChris Mason 		mid = (low + high) / 2;
8075f39d397SChris Mason 		offset = p + mid * item_size;
8085f39d397SChris Mason 
8095f39d397SChris Mason 		if (!map_token || offset < map_start ||
8105f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
8115f39d397SChris Mason 		    map_start + map_len) {
812479965d6SChris Mason 			if (map_token) {
8135f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
814479965d6SChris Mason 				map_token = NULL;
815479965d6SChris Mason 			}
816934d375bSChris Mason 
817934d375bSChris Mason 			err = map_private_extent_buffer(eb, offset,
818479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
819479965d6SChris Mason 						&map_token, &kaddr,
8205f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
8215f39d397SChris Mason 
822479965d6SChris Mason 			if (!err) {
823479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
824479965d6SChris Mason 							map_start);
825479965d6SChris Mason 			} else {
8265f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
8275f39d397SChris Mason 						   offset, sizeof(unaligned));
8285f39d397SChris Mason 				tmp = &unaligned;
829479965d6SChris Mason 			}
830479965d6SChris Mason 
8315f39d397SChris Mason 		} else {
8325f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
8335f39d397SChris Mason 							map_start);
8345f39d397SChris Mason 		}
835be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
836be0e5c09SChris Mason 
837be0e5c09SChris Mason 		if (ret < 0)
838be0e5c09SChris Mason 			low = mid + 1;
839be0e5c09SChris Mason 		else if (ret > 0)
840be0e5c09SChris Mason 			high = mid;
841be0e5c09SChris Mason 		else {
842be0e5c09SChris Mason 			*slot = mid;
843479965d6SChris Mason 			if (map_token)
8445f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
845be0e5c09SChris Mason 			return 0;
846be0e5c09SChris Mason 		}
847be0e5c09SChris Mason 	}
848be0e5c09SChris Mason 	*slot = low;
8495f39d397SChris Mason 	if (map_token)
8505f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
851be0e5c09SChris Mason 	return 1;
852be0e5c09SChris Mason }
853be0e5c09SChris Mason 
85497571fd0SChris Mason /*
85597571fd0SChris Mason  * simple bin_search frontend that does the right thing for
85697571fd0SChris Mason  * leaves vs nodes
85797571fd0SChris Mason  */
8585f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
8595f39d397SChris Mason 		      int level, int *slot)
860be0e5c09SChris Mason {
8615f39d397SChris Mason 	if (level == 0) {
8625f39d397SChris Mason 		return generic_bin_search(eb,
8635f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
8640783fcfcSChris Mason 					  sizeof(struct btrfs_item),
8655f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
8667518a238SChris Mason 					  slot);
867be0e5c09SChris Mason 	} else {
8685f39d397SChris Mason 		return generic_bin_search(eb,
8695f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
870123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
8715f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
8727518a238SChris Mason 					  slot);
873be0e5c09SChris Mason 	}
874be0e5c09SChris Mason 	return -1;
875be0e5c09SChris Mason }
876be0e5c09SChris Mason 
877d352ac68SChris Mason /* given a node and slot number, this reads the blocks it points to.  The
878d352ac68SChris Mason  * extent buffer is returned with a reference taken (but unlocked).
879d352ac68SChris Mason  * NULL is returned on error.
880d352ac68SChris Mason  */
881e02119d5SChris Mason static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root,
8825f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
883bb803951SChris Mason {
884ca7a79adSChris Mason 	int level = btrfs_header_level(parent);
885bb803951SChris Mason 	if (slot < 0)
886bb803951SChris Mason 		return NULL;
8875f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
888bb803951SChris Mason 		return NULL;
889ca7a79adSChris Mason 
890ca7a79adSChris Mason 	BUG_ON(level == 0);
891ca7a79adSChris Mason 
892db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
893ca7a79adSChris Mason 		       btrfs_level_size(root, level - 1),
894ca7a79adSChris Mason 		       btrfs_node_ptr_generation(parent, slot));
895bb803951SChris Mason }
896bb803951SChris Mason 
897d352ac68SChris Mason /*
898d352ac68SChris Mason  * node level balancing, used to make sure nodes are in proper order for
899d352ac68SChris Mason  * item deletion.  We balance from the top down, so we have to make sure
900d352ac68SChris Mason  * that a deletion won't leave an node completely empty later on.
901d352ac68SChris Mason  */
902e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans,
90398ed5174SChris Mason 			 struct btrfs_root *root,
90498ed5174SChris Mason 			 struct btrfs_path *path, int level)
905bb803951SChris Mason {
9065f39d397SChris Mason 	struct extent_buffer *right = NULL;
9075f39d397SChris Mason 	struct extent_buffer *mid;
9085f39d397SChris Mason 	struct extent_buffer *left = NULL;
9095f39d397SChris Mason 	struct extent_buffer *parent = NULL;
910bb803951SChris Mason 	int ret = 0;
911bb803951SChris Mason 	int wret;
912bb803951SChris Mason 	int pslot;
913bb803951SChris Mason 	int orig_slot = path->slots[level];
91454aa1f4dSChris Mason 	int err_on_enospc = 0;
91579f95c82SChris Mason 	u64 orig_ptr;
916bb803951SChris Mason 
917bb803951SChris Mason 	if (level == 0)
918bb803951SChris Mason 		return 0;
919bb803951SChris Mason 
9205f39d397SChris Mason 	mid = path->nodes[level];
921925baeddSChris Mason 	WARN_ON(!path->locks[level]);
9227bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
9237bb86316SChris Mason 
9241d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
92579f95c82SChris Mason 
926234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
9275f39d397SChris Mason 		parent = path->nodes[level + 1];
928bb803951SChris Mason 	pslot = path->slots[level + 1];
929bb803951SChris Mason 
93040689478SChris Mason 	/*
93140689478SChris Mason 	 * deal with the case where there is only one pointer in the root
93240689478SChris Mason 	 * by promoting the node below to a root
93340689478SChris Mason 	 */
9345f39d397SChris Mason 	if (!parent) {
9355f39d397SChris Mason 		struct extent_buffer *child;
936bb803951SChris Mason 
9375f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
938bb803951SChris Mason 			return 0;
939bb803951SChris Mason 
940bb803951SChris Mason 		/* promote the child to a root */
9415f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
942925baeddSChris Mason 		btrfs_tree_lock(child);
943bb803951SChris Mason 		BUG_ON(!child);
94465b51a00SChris Mason 		ret = btrfs_cow_block(trans, root, child, mid, 0, &child, 0);
9452f375ab9SYan 		BUG_ON(ret);
9462f375ab9SYan 
947925baeddSChris Mason 		spin_lock(&root->node_lock);
948bb803951SChris Mason 		root->node = child;
949925baeddSChris Mason 		spin_unlock(&root->node_lock);
950925baeddSChris Mason 
95131840ae1SZheng Yan 		ret = btrfs_update_extent_ref(trans, root, child->start,
95231840ae1SZheng Yan 					      mid->start, child->start,
95331840ae1SZheng Yan 					      root->root_key.objectid,
9543bb1a1bcSYan Zheng 					      trans->transid, level - 1);
95531840ae1SZheng Yan 		BUG_ON(ret);
95631840ae1SZheng Yan 
9570b86a832SChris Mason 		add_root_to_dirty_list(root);
958925baeddSChris Mason 		btrfs_tree_unlock(child);
959925baeddSChris Mason 		path->locks[level] = 0;
960bb803951SChris Mason 		path->nodes[level] = NULL;
9615f39d397SChris Mason 		clean_tree_block(trans, root, mid);
962925baeddSChris Mason 		btrfs_tree_unlock(mid);
963bb803951SChris Mason 		/* once for the path */
9645f39d397SChris Mason 		free_extent_buffer(mid);
9657bb86316SChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
96631840ae1SZheng Yan 					mid->start, root->root_key.objectid,
9673bb1a1bcSYan Zheng 					btrfs_header_generation(mid),
9683bb1a1bcSYan Zheng 					level, 1);
969bb803951SChris Mason 		/* once for the root ptr */
9705f39d397SChris Mason 		free_extent_buffer(mid);
971db94535dSChris Mason 		return ret;
972bb803951SChris Mason 	}
9735f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
974123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
975bb803951SChris Mason 		return 0;
976bb803951SChris Mason 
9775f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
97854aa1f4dSChris Mason 		err_on_enospc = 1;
97954aa1f4dSChris Mason 
9805f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
9815f39d397SChris Mason 	if (left) {
982925baeddSChris Mason 		btrfs_tree_lock(left);
9835f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
98465b51a00SChris Mason 				       parent, pslot - 1, &left, 0);
98554aa1f4dSChris Mason 		if (wret) {
98654aa1f4dSChris Mason 			ret = wret;
98754aa1f4dSChris Mason 			goto enospc;
98854aa1f4dSChris Mason 		}
9892cc58cf2SChris Mason 	}
9905f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
9915f39d397SChris Mason 	if (right) {
992925baeddSChris Mason 		btrfs_tree_lock(right);
9935f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
99465b51a00SChris Mason 				       parent, pslot + 1, &right, 0);
9952cc58cf2SChris Mason 		if (wret) {
9962cc58cf2SChris Mason 			ret = wret;
9972cc58cf2SChris Mason 			goto enospc;
9982cc58cf2SChris Mason 		}
9992cc58cf2SChris Mason 	}
10002cc58cf2SChris Mason 
10012cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
10025f39d397SChris Mason 	if (left) {
10035f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
1004bce4eae9SChris Mason 		wret = push_node_left(trans, root, left, mid, 1);
100579f95c82SChris Mason 		if (wret < 0)
100679f95c82SChris Mason 			ret = wret;
10075f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
100854aa1f4dSChris Mason 			err_on_enospc = 1;
1009bb803951SChris Mason 	}
101079f95c82SChris Mason 
101179f95c82SChris Mason 	/*
101279f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
101379f95c82SChris Mason 	 */
10145f39d397SChris Mason 	if (right) {
1015971a1f66SChris Mason 		wret = push_node_left(trans, root, mid, right, 1);
101654aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
101779f95c82SChris Mason 			ret = wret;
10185f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
1019db94535dSChris Mason 			u64 bytenr = right->start;
10207bb86316SChris Mason 			u64 generation = btrfs_header_generation(parent);
1021db94535dSChris Mason 			u32 blocksize = right->len;
1022db94535dSChris Mason 
10235f39d397SChris Mason 			clean_tree_block(trans, root, right);
1024925baeddSChris Mason 			btrfs_tree_unlock(right);
10255f39d397SChris Mason 			free_extent_buffer(right);
1026bb803951SChris Mason 			right = NULL;
1027e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
1028e089f05cSChris Mason 				       1);
1029bb803951SChris Mason 			if (wret)
1030bb803951SChris Mason 				ret = wret;
1031db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
103231840ae1SZheng Yan 						 blocksize, parent->start,
10337bb86316SChris Mason 						 btrfs_header_owner(parent),
10343bb1a1bcSYan Zheng 						 generation, level, 1);
1035bb803951SChris Mason 			if (wret)
1036bb803951SChris Mason 				ret = wret;
1037bb803951SChris Mason 		} else {
10385f39d397SChris Mason 			struct btrfs_disk_key right_key;
10395f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
10405f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
10415f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
1042bb803951SChris Mason 		}
1043bb803951SChris Mason 	}
10445f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
104579f95c82SChris Mason 		/*
104679f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
104779f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
104879f95c82SChris Mason 		 * could try to delete the only pointer in this node.
104979f95c82SChris Mason 		 * So, pull some keys from the left.
105079f95c82SChris Mason 		 * There has to be a left pointer at this point because
105179f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
105279f95c82SChris Mason 		 * right
105379f95c82SChris Mason 		 */
10545f39d397SChris Mason 		BUG_ON(!left);
10555f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
105654aa1f4dSChris Mason 		if (wret < 0) {
105779f95c82SChris Mason 			ret = wret;
105854aa1f4dSChris Mason 			goto enospc;
105954aa1f4dSChris Mason 		}
1060bce4eae9SChris Mason 		if (wret == 1) {
1061bce4eae9SChris Mason 			wret = push_node_left(trans, root, left, mid, 1);
1062bce4eae9SChris Mason 			if (wret < 0)
1063bce4eae9SChris Mason 				ret = wret;
1064bce4eae9SChris Mason 		}
106579f95c82SChris Mason 		BUG_ON(wret == 1);
106679f95c82SChris Mason 	}
10675f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
106879f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
10697bb86316SChris Mason 		u64 root_gen = btrfs_header_generation(parent);
1070db94535dSChris Mason 		u64 bytenr = mid->start;
1071db94535dSChris Mason 		u32 blocksize = mid->len;
1072925baeddSChris Mason 
10735f39d397SChris Mason 		clean_tree_block(trans, root, mid);
1074925baeddSChris Mason 		btrfs_tree_unlock(mid);
10755f39d397SChris Mason 		free_extent_buffer(mid);
1076bb803951SChris Mason 		mid = NULL;
1077e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
1078bb803951SChris Mason 		if (wret)
1079bb803951SChris Mason 			ret = wret;
10807bb86316SChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
108131840ae1SZheng Yan 					 parent->start,
10827bb86316SChris Mason 					 btrfs_header_owner(parent),
10833bb1a1bcSYan Zheng 					 root_gen, level, 1);
1084bb803951SChris Mason 		if (wret)
1085bb803951SChris Mason 			ret = wret;
108679f95c82SChris Mason 	} else {
108779f95c82SChris Mason 		/* update the parent key to reflect our changes */
10885f39d397SChris Mason 		struct btrfs_disk_key mid_key;
10895f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
10905f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
10915f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
109279f95c82SChris Mason 	}
1093bb803951SChris Mason 
109479f95c82SChris Mason 	/* update the path */
10955f39d397SChris Mason 	if (left) {
10965f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
10975f39d397SChris Mason 			extent_buffer_get(left);
1098925baeddSChris Mason 			/* left was locked after cow */
10995f39d397SChris Mason 			path->nodes[level] = left;
1100bb803951SChris Mason 			path->slots[level + 1] -= 1;
1101bb803951SChris Mason 			path->slots[level] = orig_slot;
1102925baeddSChris Mason 			if (mid) {
1103925baeddSChris Mason 				btrfs_tree_unlock(mid);
11045f39d397SChris Mason 				free_extent_buffer(mid);
1105925baeddSChris Mason 			}
1106bb803951SChris Mason 		} else {
11075f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
1108bb803951SChris Mason 			path->slots[level] = orig_slot;
1109bb803951SChris Mason 		}
1110bb803951SChris Mason 	}
111179f95c82SChris Mason 	/* double check we haven't messed things up */
1112123abc88SChris Mason 	check_block(root, path, level);
1113e20d96d6SChris Mason 	if (orig_ptr !=
11145f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
111579f95c82SChris Mason 		BUG();
111654aa1f4dSChris Mason enospc:
1117925baeddSChris Mason 	if (right) {
1118925baeddSChris Mason 		btrfs_tree_unlock(right);
11195f39d397SChris Mason 		free_extent_buffer(right);
1120925baeddSChris Mason 	}
1121925baeddSChris Mason 	if (left) {
1122925baeddSChris Mason 		if (path->nodes[level] != left)
1123925baeddSChris Mason 			btrfs_tree_unlock(left);
11245f39d397SChris Mason 		free_extent_buffer(left);
1125925baeddSChris Mason 	}
1126bb803951SChris Mason 	return ret;
1127bb803951SChris Mason }
1128bb803951SChris Mason 
1129d352ac68SChris Mason /* Node balancing for insertion.  Here we only split or push nodes around
1130d352ac68SChris Mason  * when they are completely full.  This is also done top down, so we
1131d352ac68SChris Mason  * have to be pessimistic.
1132d352ac68SChris Mason  */
113398ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
1134e66f709bSChris Mason 					  struct btrfs_root *root,
1135e66f709bSChris Mason 					  struct btrfs_path *path, int level)
1136e66f709bSChris Mason {
11375f39d397SChris Mason 	struct extent_buffer *right = NULL;
11385f39d397SChris Mason 	struct extent_buffer *mid;
11395f39d397SChris Mason 	struct extent_buffer *left = NULL;
11405f39d397SChris Mason 	struct extent_buffer *parent = NULL;
1141e66f709bSChris Mason 	int ret = 0;
1142e66f709bSChris Mason 	int wret;
1143e66f709bSChris Mason 	int pslot;
1144e66f709bSChris Mason 	int orig_slot = path->slots[level];
1145e66f709bSChris Mason 	u64 orig_ptr;
1146e66f709bSChris Mason 
1147e66f709bSChris Mason 	if (level == 0)
1148e66f709bSChris Mason 		return 1;
1149e66f709bSChris Mason 
11505f39d397SChris Mason 	mid = path->nodes[level];
11517bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1152e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
1153e66f709bSChris Mason 
1154e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
11555f39d397SChris Mason 		parent = path->nodes[level + 1];
1156e66f709bSChris Mason 	pslot = path->slots[level + 1];
1157e66f709bSChris Mason 
11585f39d397SChris Mason 	if (!parent)
1159e66f709bSChris Mason 		return 1;
1160e66f709bSChris Mason 
11615f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
1162e66f709bSChris Mason 
1163e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
11645f39d397SChris Mason 	if (left) {
1165e66f709bSChris Mason 		u32 left_nr;
1166925baeddSChris Mason 
1167925baeddSChris Mason 		btrfs_tree_lock(left);
11685f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
116933ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
117033ade1f8SChris Mason 			wret = 1;
117133ade1f8SChris Mason 		} else {
11725f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
117365b51a00SChris Mason 					      pslot - 1, &left, 0);
117454aa1f4dSChris Mason 			if (ret)
117554aa1f4dSChris Mason 				wret = 1;
117654aa1f4dSChris Mason 			else {
117754aa1f4dSChris Mason 				wret = push_node_left(trans, root,
1178971a1f66SChris Mason 						      left, mid, 0);
117954aa1f4dSChris Mason 			}
118033ade1f8SChris Mason 		}
1181e66f709bSChris Mason 		if (wret < 0)
1182e66f709bSChris Mason 			ret = wret;
1183e66f709bSChris Mason 		if (wret == 0) {
11845f39d397SChris Mason 			struct btrfs_disk_key disk_key;
1185e66f709bSChris Mason 			orig_slot += left_nr;
11865f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
11875f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
11885f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
11895f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
11905f39d397SChris Mason 				path->nodes[level] = left;
1191e66f709bSChris Mason 				path->slots[level + 1] -= 1;
1192e66f709bSChris Mason 				path->slots[level] = orig_slot;
1193925baeddSChris Mason 				btrfs_tree_unlock(mid);
11945f39d397SChris Mason 				free_extent_buffer(mid);
1195e66f709bSChris Mason 			} else {
1196e66f709bSChris Mason 				orig_slot -=
11975f39d397SChris Mason 					btrfs_header_nritems(left);
1198e66f709bSChris Mason 				path->slots[level] = orig_slot;
1199925baeddSChris Mason 				btrfs_tree_unlock(left);
12005f39d397SChris Mason 				free_extent_buffer(left);
1201e66f709bSChris Mason 			}
1202e66f709bSChris Mason 			return 0;
1203e66f709bSChris Mason 		}
1204925baeddSChris Mason 		btrfs_tree_unlock(left);
12055f39d397SChris Mason 		free_extent_buffer(left);
1206e66f709bSChris Mason 	}
12075f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
1208e66f709bSChris Mason 
1209e66f709bSChris Mason 	/*
1210e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
1211e66f709bSChris Mason 	 */
12125f39d397SChris Mason 	if (right) {
121333ade1f8SChris Mason 		u32 right_nr;
1214925baeddSChris Mason 		btrfs_tree_lock(right);
12155f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
121633ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
121733ade1f8SChris Mason 			wret = 1;
121833ade1f8SChris Mason 		} else {
12195f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
12205f39d397SChris Mason 					      parent, pslot + 1,
122165b51a00SChris Mason 					      &right, 0);
122254aa1f4dSChris Mason 			if (ret)
122354aa1f4dSChris Mason 				wret = 1;
122454aa1f4dSChris Mason 			else {
122533ade1f8SChris Mason 				wret = balance_node_right(trans, root,
12265f39d397SChris Mason 							  right, mid);
122733ade1f8SChris Mason 			}
122854aa1f4dSChris Mason 		}
1229e66f709bSChris Mason 		if (wret < 0)
1230e66f709bSChris Mason 			ret = wret;
1231e66f709bSChris Mason 		if (wret == 0) {
12325f39d397SChris Mason 			struct btrfs_disk_key disk_key;
12335f39d397SChris Mason 
12345f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
12355f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
12365f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
12375f39d397SChris Mason 
12385f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
12395f39d397SChris Mason 				path->nodes[level] = right;
1240e66f709bSChris Mason 				path->slots[level + 1] += 1;
1241e66f709bSChris Mason 				path->slots[level] = orig_slot -
12425f39d397SChris Mason 					btrfs_header_nritems(mid);
1243925baeddSChris Mason 				btrfs_tree_unlock(mid);
12445f39d397SChris Mason 				free_extent_buffer(mid);
1245e66f709bSChris Mason 			} else {
1246925baeddSChris Mason 				btrfs_tree_unlock(right);
12475f39d397SChris Mason 				free_extent_buffer(right);
1248e66f709bSChris Mason 			}
1249e66f709bSChris Mason 			return 0;
1250e66f709bSChris Mason 		}
1251925baeddSChris Mason 		btrfs_tree_unlock(right);
12525f39d397SChris Mason 		free_extent_buffer(right);
1253e66f709bSChris Mason 	}
1254e66f709bSChris Mason 	return 1;
1255e66f709bSChris Mason }
1256e66f709bSChris Mason 
125774123bd7SChris Mason /*
1258d352ac68SChris Mason  * readahead one full node of leaves, finding things that are close
1259d352ac68SChris Mason  * to the block in 'slot', and triggering ra on them.
12603c69faecSChris Mason  */
1261e02119d5SChris Mason static noinline void reada_for_search(struct btrfs_root *root,
1262e02119d5SChris Mason 				      struct btrfs_path *path,
126301f46658SChris Mason 				      int level, int slot, u64 objectid)
12643c69faecSChris Mason {
12655f39d397SChris Mason 	struct extent_buffer *node;
126601f46658SChris Mason 	struct btrfs_disk_key disk_key;
12673c69faecSChris Mason 	u32 nritems;
12683c69faecSChris Mason 	u64 search;
12696b80053dSChris Mason 	u64 lowest_read;
12706b80053dSChris Mason 	u64 highest_read;
12716b80053dSChris Mason 	u64 nread = 0;
12723c69faecSChris Mason 	int direction = path->reada;
12735f39d397SChris Mason 	struct extent_buffer *eb;
12746b80053dSChris Mason 	u32 nr;
12756b80053dSChris Mason 	u32 blocksize;
12766b80053dSChris Mason 	u32 nscan = 0;
1277db94535dSChris Mason 
1278a6b6e75eSChris Mason 	if (level != 1)
12793c69faecSChris Mason 		return;
12803c69faecSChris Mason 
12816702ed49SChris Mason 	if (!path->nodes[level])
12826702ed49SChris Mason 		return;
12836702ed49SChris Mason 
12845f39d397SChris Mason 	node = path->nodes[level];
1285925baeddSChris Mason 
12863c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
12876b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
12886b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
12895f39d397SChris Mason 	if (eb) {
12905f39d397SChris Mason 		free_extent_buffer(eb);
12913c69faecSChris Mason 		return;
12923c69faecSChris Mason 	}
12933c69faecSChris Mason 
12946b80053dSChris Mason 	highest_read = search;
12956b80053dSChris Mason 	lowest_read = search;
12966b80053dSChris Mason 
12975f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
12986b80053dSChris Mason 	nr = slot;
12993c69faecSChris Mason 	while(1) {
13006b80053dSChris Mason 		if (direction < 0) {
13016b80053dSChris Mason 			if (nr == 0)
13023c69faecSChris Mason 				break;
13036b80053dSChris Mason 			nr--;
13046b80053dSChris Mason 		} else if (direction > 0) {
13056b80053dSChris Mason 			nr++;
13066b80053dSChris Mason 			if (nr >= nritems)
13076b80053dSChris Mason 				break;
13083c69faecSChris Mason 		}
130901f46658SChris Mason 		if (path->reada < 0 && objectid) {
131001f46658SChris Mason 			btrfs_node_key(node, &disk_key, nr);
131101f46658SChris Mason 			if (btrfs_disk_key_objectid(&disk_key) != objectid)
131201f46658SChris Mason 				break;
131301f46658SChris Mason 		}
13146b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
13156b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
13166f3577bdSChris Mason 		    (search < lowest_read && lowest_read - search <= 16384) ||
13176f3577bdSChris Mason 		    (search > highest_read && search - highest_read <= 16384)) {
1318ca7a79adSChris Mason 			readahead_tree_block(root, search, blocksize,
1319ca7a79adSChris Mason 				     btrfs_node_ptr_generation(node, nr));
13206b80053dSChris Mason 			nread += blocksize;
13213c69faecSChris Mason 		}
13226b80053dSChris Mason 		nscan++;
13236f3577bdSChris Mason 		if (path->reada < 2 && (nread > (64 * 1024) || nscan > 32))
13246b80053dSChris Mason 			break;
13256f3577bdSChris Mason 		if(nread > (256 * 1024) || nscan > 128)
13266b80053dSChris Mason 			break;
13276b80053dSChris Mason 
13286b80053dSChris Mason 		if (search < lowest_read)
13296b80053dSChris Mason 			lowest_read = search;
13306b80053dSChris Mason 		if (search > highest_read)
13316b80053dSChris Mason 			highest_read = search;
13323c69faecSChris Mason 	}
13333c69faecSChris Mason }
1334925baeddSChris Mason 
1335d352ac68SChris Mason /*
1336d352ac68SChris Mason  * when we walk down the tree, it is usually safe to unlock the higher layers in
1337d352ac68SChris Mason  * the tree.  The exceptions are when our path goes through slot 0, because operations
1338d352ac68SChris Mason  * on the tree might require changing key pointers higher up in the tree.
1339d352ac68SChris Mason  *
1340d352ac68SChris Mason  * callers might also have set path->keep_locks, which tells this code to
1341d352ac68SChris Mason  * keep the lock if the path points to the last slot in the block.  This is
1342d352ac68SChris Mason  * part of walking through the tree, and selecting the next slot in the higher
1343d352ac68SChris Mason  * block.
1344d352ac68SChris Mason  *
1345d352ac68SChris Mason  * lowest_unlock sets the lowest level in the tree we're allowed to unlock.
1346d352ac68SChris Mason  * so if lowest_unlock is 1, level 0 won't be unlocked
1347d352ac68SChris Mason  */
1348e02119d5SChris Mason static noinline void unlock_up(struct btrfs_path *path, int level,
1349e02119d5SChris Mason 			       int lowest_unlock)
1350925baeddSChris Mason {
1351925baeddSChris Mason 	int i;
1352925baeddSChris Mason 	int skip_level = level;
1353051e1b9fSChris Mason 	int no_skips = 0;
1354925baeddSChris Mason 	struct extent_buffer *t;
1355925baeddSChris Mason 
1356925baeddSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1357925baeddSChris Mason 		if (!path->nodes[i])
1358925baeddSChris Mason 			break;
1359925baeddSChris Mason 		if (!path->locks[i])
1360925baeddSChris Mason 			break;
1361051e1b9fSChris Mason 		if (!no_skips && path->slots[i] == 0) {
1362925baeddSChris Mason 			skip_level = i + 1;
1363925baeddSChris Mason 			continue;
1364925baeddSChris Mason 		}
1365051e1b9fSChris Mason 		if (!no_skips && path->keep_locks) {
1366925baeddSChris Mason 			u32 nritems;
1367925baeddSChris Mason 			t = path->nodes[i];
1368925baeddSChris Mason 			nritems = btrfs_header_nritems(t);
1369051e1b9fSChris Mason 			if (nritems < 1 || path->slots[i] >= nritems - 1) {
1370925baeddSChris Mason 				skip_level = i + 1;
1371925baeddSChris Mason 				continue;
1372925baeddSChris Mason 			}
1373925baeddSChris Mason 		}
1374051e1b9fSChris Mason 		if (skip_level < i && i >= lowest_unlock)
1375051e1b9fSChris Mason 			no_skips = 1;
1376051e1b9fSChris Mason 
1377925baeddSChris Mason 		t = path->nodes[i];
1378925baeddSChris Mason 		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
1379925baeddSChris Mason 			btrfs_tree_unlock(t);
1380925baeddSChris Mason 			path->locks[i] = 0;
1381925baeddSChris Mason 		}
1382925baeddSChris Mason 	}
1383925baeddSChris Mason }
1384925baeddSChris Mason 
13853c69faecSChris Mason /*
138674123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
138774123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
138874123bd7SChris Mason  * level of the path (level 0)
138974123bd7SChris Mason  *
139074123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
1391aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
1392aa5d6bedSChris Mason  * search a negative error number is returned.
139397571fd0SChris Mason  *
139497571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
139597571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
139697571fd0SChris Mason  * possible)
139774123bd7SChris Mason  */
1398e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
1399e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
1400e089f05cSChris Mason 		      ins_len, int cow)
1401be0e5c09SChris Mason {
14025f39d397SChris Mason 	struct extent_buffer *b;
1403051e1b9fSChris Mason 	struct extent_buffer *tmp;
1404be0e5c09SChris Mason 	int slot;
1405be0e5c09SChris Mason 	int ret;
1406be0e5c09SChris Mason 	int level;
14073c69faecSChris Mason 	int should_reada = p->reada;
1408925baeddSChris Mason 	int lowest_unlock = 1;
1409594a24ebSChris Mason 	int blocksize;
14109f3a7427SChris Mason 	u8 lowest_level = 0;
1411594a24ebSChris Mason 	u64 blocknr;
1412594a24ebSChris Mason 	u64 gen;
141365b51a00SChris Mason 	struct btrfs_key prealloc_block;
14149f3a7427SChris Mason 
14156702ed49SChris Mason 	lowest_level = p->lowest_level;
1416323ac95bSChris Mason 	WARN_ON(lowest_level && ins_len > 0);
141722b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
141825179201SJosef Bacik 
1419925baeddSChris Mason 	if (ins_len < 0)
1420925baeddSChris Mason 		lowest_unlock = 2;
142165b51a00SChris Mason 
142265b51a00SChris Mason 	prealloc_block.objectid = 0;
142365b51a00SChris Mason 
1424bb803951SChris Mason again:
14255cd57b2cSChris Mason 	if (p->skip_locking)
14265cd57b2cSChris Mason 		b = btrfs_root_node(root);
14275cd57b2cSChris Mason 	else
1428925baeddSChris Mason 		b = btrfs_lock_root_node(root);
1429925baeddSChris Mason 
1430eb60ceacSChris Mason 	while (b) {
14315f39d397SChris Mason 		level = btrfs_header_level(b);
143265b51a00SChris Mason 
143365b51a00SChris Mason 		/*
143465b51a00SChris Mason 		 * setup the path here so we can release it under lock
143565b51a00SChris Mason 		 * contention with the cow code
143665b51a00SChris Mason 		 */
143765b51a00SChris Mason 		p->nodes[level] = b;
143865b51a00SChris Mason 		if (!p->skip_locking)
143965b51a00SChris Mason 			p->locks[level] = 1;
144065b51a00SChris Mason 
144102217ed2SChris Mason 		if (cow) {
144202217ed2SChris Mason 			int wret;
144365b51a00SChris Mason 
144465b51a00SChris Mason 			/* is a cow on this block not required */
144565b51a00SChris Mason 			spin_lock(&root->fs_info->hash_lock);
144665b51a00SChris Mason 			if (btrfs_header_generation(b) == trans->transid &&
14475b21f2edSZheng Yan 			    btrfs_header_owner(b) == root->root_key.objectid &&
144865b51a00SChris Mason 			    !btrfs_header_flag(b, BTRFS_HEADER_FLAG_WRITTEN)) {
144965b51a00SChris Mason 				spin_unlock(&root->fs_info->hash_lock);
145065b51a00SChris Mason 				goto cow_done;
145165b51a00SChris Mason 			}
145265b51a00SChris Mason 			spin_unlock(&root->fs_info->hash_lock);
145365b51a00SChris Mason 
145465b51a00SChris Mason 			/* ok, we have to cow, is our old prealloc the right
145565b51a00SChris Mason 			 * size?
145665b51a00SChris Mason 			 */
145765b51a00SChris Mason 			if (prealloc_block.objectid &&
145865b51a00SChris Mason 			    prealloc_block.offset != b->len) {
145965b51a00SChris Mason 				btrfs_free_reserved_extent(root,
146065b51a00SChris Mason 					   prealloc_block.objectid,
146165b51a00SChris Mason 					   prealloc_block.offset);
146265b51a00SChris Mason 				prealloc_block.objectid = 0;
146365b51a00SChris Mason 			}
146465b51a00SChris Mason 
146565b51a00SChris Mason 			/*
146665b51a00SChris Mason 			 * for higher level blocks, try not to allocate blocks
146765b51a00SChris Mason 			 * with the block and the parent locks held.
146865b51a00SChris Mason 			 */
146965b51a00SChris Mason 			if (level > 1 && !prealloc_block.objectid &&
147065b51a00SChris Mason 			    btrfs_path_lock_waiting(p, level)) {
147165b51a00SChris Mason 				u32 size = b->len;
147265b51a00SChris Mason 				u64 hint = b->start;
147365b51a00SChris Mason 
147465b51a00SChris Mason 				btrfs_release_path(root, p);
147565b51a00SChris Mason 				ret = btrfs_reserve_extent(trans, root,
147665b51a00SChris Mason 							   size, size, 0,
147765b51a00SChris Mason 							   hint, (u64)-1,
147865b51a00SChris Mason 							   &prealloc_block, 0);
147965b51a00SChris Mason 				BUG_ON(ret);
148065b51a00SChris Mason 				goto again;
148165b51a00SChris Mason 			}
148265b51a00SChris Mason 
1483e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
1484e20d96d6SChris Mason 					       p->nodes[level + 1],
1485e20d96d6SChris Mason 					       p->slots[level + 1],
148665b51a00SChris Mason 					       &b, prealloc_block.objectid);
148765b51a00SChris Mason 			prealloc_block.objectid = 0;
148854aa1f4dSChris Mason 			if (wret) {
14895f39d397SChris Mason 				free_extent_buffer(b);
149065b51a00SChris Mason 				ret = wret;
149165b51a00SChris Mason 				goto done;
149254aa1f4dSChris Mason 			}
149302217ed2SChris Mason 		}
149465b51a00SChris Mason cow_done:
149502217ed2SChris Mason 		BUG_ON(!cow && ins_len);
14965f39d397SChris Mason 		if (level != btrfs_header_level(b))
14972c90e5d6SChris Mason 			WARN_ON(1);
14985f39d397SChris Mason 		level = btrfs_header_level(b);
149965b51a00SChris Mason 
1500eb60ceacSChris Mason 		p->nodes[level] = b;
15015cd57b2cSChris Mason 		if (!p->skip_locking)
1502925baeddSChris Mason 			p->locks[level] = 1;
150365b51a00SChris Mason 
1504123abc88SChris Mason 		ret = check_block(root, p, level);
150565b51a00SChris Mason 		if (ret) {
150665b51a00SChris Mason 			ret = -1;
150765b51a00SChris Mason 			goto done;
150865b51a00SChris Mason 		}
1509925baeddSChris Mason 
15105f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
15115f39d397SChris Mason 		if (level != 0) {
1512be0e5c09SChris Mason 			if (ret && slot > 0)
1513be0e5c09SChris Mason 				slot -= 1;
1514be0e5c09SChris Mason 			p->slots[level] = slot;
1515459931ecSChris Mason 			if ((p->search_for_split || ins_len > 0) &&
1516459931ecSChris Mason 			    btrfs_header_nritems(b) >=
15171514794eSChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1518e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
15195c680ed6SChris Mason 				BUG_ON(sret > 0);
152065b51a00SChris Mason 				if (sret) {
152165b51a00SChris Mason 					ret = sret;
152265b51a00SChris Mason 					goto done;
152365b51a00SChris Mason 				}
15245c680ed6SChris Mason 				b = p->nodes[level];
15255c680ed6SChris Mason 				slot = p->slots[level];
1526bb803951SChris Mason 			} else if (ins_len < 0) {
1527e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1528e089f05cSChris Mason 							 level);
152965b51a00SChris Mason 				if (sret) {
153065b51a00SChris Mason 					ret = sret;
153165b51a00SChris Mason 					goto done;
153265b51a00SChris Mason 				}
1533bb803951SChris Mason 				b = p->nodes[level];
1534f510cfecSChris Mason 				if (!b) {
1535f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1536bb803951SChris Mason 					goto again;
1537f510cfecSChris Mason 				}
1538bb803951SChris Mason 				slot = p->slots[level];
15395f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
15405c680ed6SChris Mason 			}
1541f9efa9c7SChris Mason 			unlock_up(p, level, lowest_unlock);
1542f9efa9c7SChris Mason 
15439f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
1544925baeddSChris Mason 			if (level == lowest_level) {
15455b21f2edSZheng Yan 				ret = 0;
15465b21f2edSZheng Yan 				goto done;
1547925baeddSChris Mason 			}
1548ca7a79adSChris Mason 
1549594a24ebSChris Mason 			blocknr = btrfs_node_blockptr(b, slot);
1550594a24ebSChris Mason 			gen = btrfs_node_ptr_generation(b, slot);
1551594a24ebSChris Mason 			blocksize = btrfs_level_size(root, level - 1);
1552594a24ebSChris Mason 
1553594a24ebSChris Mason 			tmp = btrfs_find_tree_block(root, blocknr, blocksize);
1554594a24ebSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
1555051e1b9fSChris Mason 				b = tmp;
1556051e1b9fSChris Mason 			} else {
1557051e1b9fSChris Mason 				/*
1558051e1b9fSChris Mason 				 * reduce lock contention at high levels
1559051e1b9fSChris Mason 				 * of the btree by dropping locks before
1560051e1b9fSChris Mason 				 * we read.
1561051e1b9fSChris Mason 				 */
1562051e1b9fSChris Mason 				if (level > 1) {
1563051e1b9fSChris Mason 					btrfs_release_path(NULL, p);
1564051e1b9fSChris Mason 					if (tmp)
1565051e1b9fSChris Mason 						free_extent_buffer(tmp);
1566f9efa9c7SChris Mason 					if (should_reada)
1567f9efa9c7SChris Mason 						reada_for_search(root, p,
1568f9efa9c7SChris Mason 								 level, slot,
1569f9efa9c7SChris Mason 								 key->objectid);
1570f9efa9c7SChris Mason 
1571594a24ebSChris Mason 					tmp = read_tree_block(root, blocknr,
1572594a24ebSChris Mason 							 blocksize, gen);
1573594a24ebSChris Mason 					if (tmp)
1574594a24ebSChris Mason 						free_extent_buffer(tmp);
1575051e1b9fSChris Mason 					goto again;
1576051e1b9fSChris Mason 				} else {
1577a74a4b97SChris Mason 					if (tmp)
1578a74a4b97SChris Mason 						free_extent_buffer(tmp);
1579f9efa9c7SChris Mason 					if (should_reada)
1580f9efa9c7SChris Mason 						reada_for_search(root, p,
1581f9efa9c7SChris Mason 								 level, slot,
1582f9efa9c7SChris Mason 								 key->objectid);
1583ca7a79adSChris Mason 					b = read_node_slot(root, b, slot);
1584051e1b9fSChris Mason 				}
1585051e1b9fSChris Mason 			}
15865cd57b2cSChris Mason 			if (!p->skip_locking)
1587925baeddSChris Mason 				btrfs_tree_lock(b);
1588be0e5c09SChris Mason 		} else {
1589be0e5c09SChris Mason 			p->slots[level] = slot;
159087b29b20SYan Zheng 			if (ins_len > 0 &&
159187b29b20SYan Zheng 			    btrfs_leaf_free_space(root, b) < ins_len) {
1592d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1593cc0c5538SChris Mason 						      p, ins_len, ret == 0);
15945c680ed6SChris Mason 				BUG_ON(sret > 0);
159565b51a00SChris Mason 				if (sret) {
159665b51a00SChris Mason 					ret = sret;
159765b51a00SChris Mason 					goto done;
159865b51a00SChris Mason 				}
15995c680ed6SChris Mason 			}
1600459931ecSChris Mason 			if (!p->search_for_split)
1601925baeddSChris Mason 				unlock_up(p, level, lowest_unlock);
160265b51a00SChris Mason 			goto done;
160365b51a00SChris Mason 		}
160465b51a00SChris Mason 	}
160565b51a00SChris Mason 	ret = 1;
160665b51a00SChris Mason done:
160765b51a00SChris Mason 	if (prealloc_block.objectid) {
160865b51a00SChris Mason 		btrfs_free_reserved_extent(root,
160965b51a00SChris Mason 			   prealloc_block.objectid,
161065b51a00SChris Mason 			   prealloc_block.offset);
161165b51a00SChris Mason 	}
161265b51a00SChris Mason 
1613be0e5c09SChris Mason 	return ret;
1614be0e5c09SChris Mason }
1615be0e5c09SChris Mason 
16161a40e23bSZheng Yan int btrfs_merge_path(struct btrfs_trans_handle *trans,
16171a40e23bSZheng Yan 		     struct btrfs_root *root,
16181a40e23bSZheng Yan 		     struct btrfs_key *node_keys,
16191a40e23bSZheng Yan 		     u64 *nodes, int lowest_level)
16201a40e23bSZheng Yan {
16211a40e23bSZheng Yan 	struct extent_buffer *eb;
16221a40e23bSZheng Yan 	struct extent_buffer *parent;
16231a40e23bSZheng Yan 	struct btrfs_key key;
16241a40e23bSZheng Yan 	u64 bytenr;
16251a40e23bSZheng Yan 	u64 generation;
16261a40e23bSZheng Yan 	u32 blocksize;
16271a40e23bSZheng Yan 	int level;
16281a40e23bSZheng Yan 	int slot;
16291a40e23bSZheng Yan 	int key_match;
16301a40e23bSZheng Yan 	int ret;
16311a40e23bSZheng Yan 
16321a40e23bSZheng Yan 	eb = btrfs_lock_root_node(root);
16331a40e23bSZheng Yan 	ret = btrfs_cow_block(trans, root, eb, NULL, 0, &eb, 0);
16341a40e23bSZheng Yan 	BUG_ON(ret);
16351a40e23bSZheng Yan 
16361a40e23bSZheng Yan 	parent = eb;
16371a40e23bSZheng Yan 	while (1) {
16381a40e23bSZheng Yan 		level = btrfs_header_level(parent);
16391a40e23bSZheng Yan 		if (level == 0 || level <= lowest_level)
16401a40e23bSZheng Yan 			break;
16411a40e23bSZheng Yan 
16421a40e23bSZheng Yan 		ret = bin_search(parent, &node_keys[lowest_level], level,
16431a40e23bSZheng Yan 				 &slot);
16441a40e23bSZheng Yan 		if (ret && slot > 0)
16451a40e23bSZheng Yan 			slot--;
16461a40e23bSZheng Yan 
16471a40e23bSZheng Yan 		bytenr = btrfs_node_blockptr(parent, slot);
16481a40e23bSZheng Yan 		if (nodes[level - 1] == bytenr)
16491a40e23bSZheng Yan 			break;
16501a40e23bSZheng Yan 
16511a40e23bSZheng Yan 		blocksize = btrfs_level_size(root, level - 1);
16521a40e23bSZheng Yan 		generation = btrfs_node_ptr_generation(parent, slot);
16531a40e23bSZheng Yan 		btrfs_node_key_to_cpu(eb, &key, slot);
16541a40e23bSZheng Yan 		key_match = !memcmp(&key, &node_keys[level - 1], sizeof(key));
16551a40e23bSZheng Yan 
1656f82d02d9SYan Zheng 		if (generation == trans->transid) {
16571a40e23bSZheng Yan 			eb = read_tree_block(root, bytenr, blocksize,
16581a40e23bSZheng Yan 					     generation);
16591a40e23bSZheng Yan 			btrfs_tree_lock(eb);
1660f82d02d9SYan Zheng 		}
1661f82d02d9SYan Zheng 
1662f82d02d9SYan Zheng 		/*
1663f82d02d9SYan Zheng 		 * if node keys match and node pointer hasn't been modified
1664f82d02d9SYan Zheng 		 * in the running transaction, we can merge the path. for
1665f82d02d9SYan Zheng 		 * blocks owened by reloc trees, the node pointer check is
1666f82d02d9SYan Zheng 		 * skipped, this is because these blocks are fully controlled
1667f82d02d9SYan Zheng 		 * by the space balance code, no one else can modify them.
1668f82d02d9SYan Zheng 		 */
1669f82d02d9SYan Zheng 		if (!nodes[level - 1] || !key_match ||
1670f82d02d9SYan Zheng 		    (generation == trans->transid &&
1671f82d02d9SYan Zheng 		     btrfs_header_owner(eb) != BTRFS_TREE_RELOC_OBJECTID)) {
1672f82d02d9SYan Zheng 			if (level == 1 || level == lowest_level + 1) {
1673f82d02d9SYan Zheng 				if (generation == trans->transid) {
1674f82d02d9SYan Zheng 					btrfs_tree_unlock(eb);
1675f82d02d9SYan Zheng 					free_extent_buffer(eb);
1676f82d02d9SYan Zheng 				}
1677f82d02d9SYan Zheng 				break;
1678f82d02d9SYan Zheng 			}
1679f82d02d9SYan Zheng 
1680f82d02d9SYan Zheng 			if (generation != trans->transid) {
1681f82d02d9SYan Zheng 				eb = read_tree_block(root, bytenr, blocksize,
1682f82d02d9SYan Zheng 						generation);
1683f82d02d9SYan Zheng 				btrfs_tree_lock(eb);
1684f82d02d9SYan Zheng 			}
16851a40e23bSZheng Yan 
16861a40e23bSZheng Yan 			ret = btrfs_cow_block(trans, root, eb, parent, slot,
16871a40e23bSZheng Yan 					      &eb, 0);
16881a40e23bSZheng Yan 			BUG_ON(ret);
16891a40e23bSZheng Yan 
1690f82d02d9SYan Zheng 			if (root->root_key.objectid ==
1691f82d02d9SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID) {
1692f82d02d9SYan Zheng 				if (!nodes[level - 1]) {
1693f82d02d9SYan Zheng 					nodes[level - 1] = eb->start;
1694f82d02d9SYan Zheng 					memcpy(&node_keys[level - 1], &key,
1695f82d02d9SYan Zheng 					       sizeof(node_keys[0]));
1696f82d02d9SYan Zheng 				} else {
1697f82d02d9SYan Zheng 					WARN_ON(1);
1698f82d02d9SYan Zheng 				}
1699f82d02d9SYan Zheng 			}
1700f82d02d9SYan Zheng 
17011a40e23bSZheng Yan 			btrfs_tree_unlock(parent);
17021a40e23bSZheng Yan 			free_extent_buffer(parent);
17031a40e23bSZheng Yan 			parent = eb;
17041a40e23bSZheng Yan 			continue;
17051a40e23bSZheng Yan 		}
17061a40e23bSZheng Yan 
17071a40e23bSZheng Yan 		btrfs_set_node_blockptr(parent, slot, nodes[level - 1]);
17081a40e23bSZheng Yan 		btrfs_set_node_ptr_generation(parent, slot, trans->transid);
17091a40e23bSZheng Yan 		btrfs_mark_buffer_dirty(parent);
17101a40e23bSZheng Yan 
17111a40e23bSZheng Yan 		ret = btrfs_inc_extent_ref(trans, root,
17121a40e23bSZheng Yan 					nodes[level - 1],
17131a40e23bSZheng Yan 					blocksize, parent->start,
17141a40e23bSZheng Yan 					btrfs_header_owner(parent),
17151a40e23bSZheng Yan 					btrfs_header_generation(parent),
17163bb1a1bcSYan Zheng 					level - 1);
17171a40e23bSZheng Yan 		BUG_ON(ret);
1718f82d02d9SYan Zheng 
1719f82d02d9SYan Zheng 		/*
1720f82d02d9SYan Zheng 		 * If the block was created in the running transaction,
1721f82d02d9SYan Zheng 		 * it's possible this is the last reference to it, so we
1722f82d02d9SYan Zheng 		 * should drop the subtree.
1723f82d02d9SYan Zheng 		 */
1724f82d02d9SYan Zheng 		if (generation == trans->transid) {
1725f82d02d9SYan Zheng 			ret = btrfs_drop_subtree(trans, root, eb, parent);
1726f82d02d9SYan Zheng 			BUG_ON(ret);
1727f82d02d9SYan Zheng 			btrfs_tree_unlock(eb);
1728f82d02d9SYan Zheng 			free_extent_buffer(eb);
1729f82d02d9SYan Zheng 		} else {
17301a40e23bSZheng Yan 			ret = btrfs_free_extent(trans, root, bytenr,
17311a40e23bSZheng Yan 					blocksize, parent->start,
17321a40e23bSZheng Yan 					btrfs_header_owner(parent),
17331a40e23bSZheng Yan 					btrfs_header_generation(parent),
17343bb1a1bcSYan Zheng 					level - 1, 1);
17351a40e23bSZheng Yan 			BUG_ON(ret);
17361a40e23bSZheng Yan 		}
17371a40e23bSZheng Yan 		break;
17381a40e23bSZheng Yan 	}
17391a40e23bSZheng Yan 	btrfs_tree_unlock(parent);
17401a40e23bSZheng Yan 	free_extent_buffer(parent);
17411a40e23bSZheng Yan 	return 0;
17421a40e23bSZheng Yan }
17431a40e23bSZheng Yan 
174474123bd7SChris Mason /*
174574123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
174674123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
174774123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
174874123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
174974123bd7SChris Mason  * higher levels
1750aa5d6bedSChris Mason  *
1751aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1752aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
175374123bd7SChris Mason  */
17545f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
17555f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
17565f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1757be0e5c09SChris Mason {
1758be0e5c09SChris Mason 	int i;
1759aa5d6bedSChris Mason 	int ret = 0;
17605f39d397SChris Mason 	struct extent_buffer *t;
17615f39d397SChris Mason 
1762234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1763be0e5c09SChris Mason 		int tslot = path->slots[i];
1764eb60ceacSChris Mason 		if (!path->nodes[i])
1765be0e5c09SChris Mason 			break;
17665f39d397SChris Mason 		t = path->nodes[i];
17675f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1768d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1769be0e5c09SChris Mason 		if (tslot != 0)
1770be0e5c09SChris Mason 			break;
1771be0e5c09SChris Mason 	}
1772aa5d6bedSChris Mason 	return ret;
1773be0e5c09SChris Mason }
1774be0e5c09SChris Mason 
177574123bd7SChris Mason /*
177631840ae1SZheng Yan  * update item key.
177731840ae1SZheng Yan  *
177831840ae1SZheng Yan  * This function isn't completely safe. It's the caller's responsibility
177931840ae1SZheng Yan  * that the new key won't break the order
178031840ae1SZheng Yan  */
178131840ae1SZheng Yan int btrfs_set_item_key_safe(struct btrfs_trans_handle *trans,
178231840ae1SZheng Yan 			    struct btrfs_root *root, struct btrfs_path *path,
178331840ae1SZheng Yan 			    struct btrfs_key *new_key)
178431840ae1SZheng Yan {
178531840ae1SZheng Yan 	struct btrfs_disk_key disk_key;
178631840ae1SZheng Yan 	struct extent_buffer *eb;
178731840ae1SZheng Yan 	int slot;
178831840ae1SZheng Yan 
178931840ae1SZheng Yan 	eb = path->nodes[0];
179031840ae1SZheng Yan 	slot = path->slots[0];
179131840ae1SZheng Yan 	if (slot > 0) {
179231840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot - 1);
179331840ae1SZheng Yan 		if (comp_keys(&disk_key, new_key) >= 0)
179431840ae1SZheng Yan 			return -1;
179531840ae1SZheng Yan 	}
179631840ae1SZheng Yan 	if (slot < btrfs_header_nritems(eb) - 1) {
179731840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot + 1);
179831840ae1SZheng Yan 		if (comp_keys(&disk_key, new_key) <= 0)
179931840ae1SZheng Yan 			return -1;
180031840ae1SZheng Yan 	}
180131840ae1SZheng Yan 
180231840ae1SZheng Yan 	btrfs_cpu_key_to_disk(&disk_key, new_key);
180331840ae1SZheng Yan 	btrfs_set_item_key(eb, &disk_key, slot);
180431840ae1SZheng Yan 	btrfs_mark_buffer_dirty(eb);
180531840ae1SZheng Yan 	if (slot == 0)
180631840ae1SZheng Yan 		fixup_low_keys(trans, root, path, &disk_key, 1);
180731840ae1SZheng Yan 	return 0;
180831840ae1SZheng Yan }
180931840ae1SZheng Yan 
181031840ae1SZheng Yan /*
181174123bd7SChris Mason  * try to push data from one node into the next node left in the
181279f95c82SChris Mason  * tree.
1813aa5d6bedSChris Mason  *
1814aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1815aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
181674123bd7SChris Mason  */
181798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
181898ed5174SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
1819971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
1820be0e5c09SChris Mason {
1821be0e5c09SChris Mason 	int push_items = 0;
1822bb803951SChris Mason 	int src_nritems;
1823bb803951SChris Mason 	int dst_nritems;
1824aa5d6bedSChris Mason 	int ret = 0;
1825be0e5c09SChris Mason 
18265f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
18275f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1828123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
18297bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
18307bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
183154aa1f4dSChris Mason 
1832bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
1833971a1f66SChris Mason 		return 1;
1834971a1f66SChris Mason 
1835eb60ceacSChris Mason 	if (push_items <= 0) {
1836be0e5c09SChris Mason 		return 1;
1837eb60ceacSChris Mason 	}
1838be0e5c09SChris Mason 
1839bce4eae9SChris Mason 	if (empty) {
1840971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
1841bce4eae9SChris Mason 		if (push_items < src_nritems) {
1842bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
1843bce4eae9SChris Mason 			 * we aren't going to empty it
1844bce4eae9SChris Mason 			 */
1845bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
1846bce4eae9SChris Mason 				if (push_items <= 8)
1847bce4eae9SChris Mason 					return 1;
1848bce4eae9SChris Mason 				push_items -= 8;
1849bce4eae9SChris Mason 			}
1850bce4eae9SChris Mason 		}
1851bce4eae9SChris Mason 	} else
1852bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
185379f95c82SChris Mason 
18545f39d397SChris Mason 	copy_extent_buffer(dst, src,
18555f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
18565f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1857123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
18585f39d397SChris Mason 
1859bb803951SChris Mason 	if (push_items < src_nritems) {
18605f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
18615f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1862e2fa7227SChris Mason 				      (src_nritems - push_items) *
1863123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1864bb803951SChris Mason 	}
18655f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
18665f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
18675f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
18685f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
186931840ae1SZheng Yan 
187031840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, src, dst, dst_nritems, push_items);
187131840ae1SZheng Yan 	BUG_ON(ret);
187231840ae1SZheng Yan 
1873bb803951SChris Mason 	return ret;
1874be0e5c09SChris Mason }
1875be0e5c09SChris Mason 
187697571fd0SChris Mason /*
187779f95c82SChris Mason  * try to push data from one node into the next node right in the
187879f95c82SChris Mason  * tree.
187979f95c82SChris Mason  *
188079f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
188179f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
188279f95c82SChris Mason  *
188379f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
188479f95c82SChris Mason  */
18855f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
18865f39d397SChris Mason 			      struct btrfs_root *root,
18875f39d397SChris Mason 			      struct extent_buffer *dst,
18885f39d397SChris Mason 			      struct extent_buffer *src)
188979f95c82SChris Mason {
189079f95c82SChris Mason 	int push_items = 0;
189179f95c82SChris Mason 	int max_push;
189279f95c82SChris Mason 	int src_nritems;
189379f95c82SChris Mason 	int dst_nritems;
189479f95c82SChris Mason 	int ret = 0;
189579f95c82SChris Mason 
18967bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
18977bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
18987bb86316SChris Mason 
18995f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
19005f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1901123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1902bce4eae9SChris Mason 	if (push_items <= 0) {
190379f95c82SChris Mason 		return 1;
1904bce4eae9SChris Mason 	}
1905bce4eae9SChris Mason 
1906bce4eae9SChris Mason 	if (src_nritems < 4) {
1907bce4eae9SChris Mason 		return 1;
1908bce4eae9SChris Mason 	}
190979f95c82SChris Mason 
191079f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
191179f95c82SChris Mason 	/* don't try to empty the node */
1912bce4eae9SChris Mason 	if (max_push >= src_nritems) {
191379f95c82SChris Mason 		return 1;
1914bce4eae9SChris Mason 	}
1915252c38f0SYan 
191679f95c82SChris Mason 	if (max_push < push_items)
191779f95c82SChris Mason 		push_items = max_push;
191879f95c82SChris Mason 
19195f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
19205f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
19215f39d397SChris Mason 				      (dst_nritems) *
19225f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1923d6025579SChris Mason 
19245f39d397SChris Mason 	copy_extent_buffer(dst, src,
19255f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
19265f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1927123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
192879f95c82SChris Mason 
19295f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
19305f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
193179f95c82SChris Mason 
19325f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
19335f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
193431840ae1SZheng Yan 
193531840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, src, dst, 0, push_items);
193631840ae1SZheng Yan 	BUG_ON(ret);
193731840ae1SZheng Yan 
193879f95c82SChris Mason 	return ret;
193979f95c82SChris Mason }
194079f95c82SChris Mason 
194179f95c82SChris Mason /*
194297571fd0SChris Mason  * helper function to insert a new root level in the tree.
194397571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
194497571fd0SChris Mason  * point to the existing root
1945aa5d6bedSChris Mason  *
1946aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
194797571fd0SChris Mason  */
194898ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans,
19495f39d397SChris Mason 			   struct btrfs_root *root,
19505f39d397SChris Mason 			   struct btrfs_path *path, int level)
195174123bd7SChris Mason {
19527bb86316SChris Mason 	u64 lower_gen;
19535f39d397SChris Mason 	struct extent_buffer *lower;
19545f39d397SChris Mason 	struct extent_buffer *c;
1955925baeddSChris Mason 	struct extent_buffer *old;
19565f39d397SChris Mason 	struct btrfs_disk_key lower_key;
195731840ae1SZheng Yan 	int ret;
19585c680ed6SChris Mason 
19595c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
19605c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
19615c680ed6SChris Mason 
19627bb86316SChris Mason 	lower = path->nodes[level-1];
19637bb86316SChris Mason 	if (level == 1)
19647bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
19657bb86316SChris Mason 	else
19667bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
19677bb86316SChris Mason 
196831840ae1SZheng Yan 	c = btrfs_alloc_free_block(trans, root, root->nodesize, 0,
196931840ae1SZheng Yan 				   root->root_key.objectid, trans->transid,
1970ad3d81baSChristoph Hellwig 				   level, root->node->start, 0);
19715f39d397SChris Mason 	if (IS_ERR(c))
19725f39d397SChris Mason 		return PTR_ERR(c);
1973925baeddSChris Mason 
19745f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
19755f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
19765f39d397SChris Mason 	btrfs_set_header_level(c, level);
1977db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
19785f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
19795f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
1980d5719762SChris Mason 
19815f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
19825f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
19835f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1984e17cade2SChris Mason 
1985e17cade2SChris Mason 	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
1986e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
1987e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
1988e17cade2SChris Mason 
19895f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1990db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
19917bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
199231840ae1SZheng Yan 	WARN_ON(lower_gen != trans->transid);
19937bb86316SChris Mason 
19947bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
19955f39d397SChris Mason 
19965f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1997d5719762SChris Mason 
1998925baeddSChris Mason 	spin_lock(&root->node_lock);
1999925baeddSChris Mason 	old = root->node;
20005f39d397SChris Mason 	root->node = c;
2001925baeddSChris Mason 	spin_unlock(&root->node_lock);
2002925baeddSChris Mason 
200331840ae1SZheng Yan 	ret = btrfs_update_extent_ref(trans, root, lower->start,
200431840ae1SZheng Yan 				      lower->start, c->start,
200531840ae1SZheng Yan 				      root->root_key.objectid,
20063bb1a1bcSYan Zheng 				      trans->transid, level - 1);
200731840ae1SZheng Yan 	BUG_ON(ret);
200831840ae1SZheng Yan 
2009925baeddSChris Mason 	/* the super has an extra ref to root->node */
2010925baeddSChris Mason 	free_extent_buffer(old);
2011925baeddSChris Mason 
20120b86a832SChris Mason 	add_root_to_dirty_list(root);
20135f39d397SChris Mason 	extent_buffer_get(c);
20145f39d397SChris Mason 	path->nodes[level] = c;
2015925baeddSChris Mason 	path->locks[level] = 1;
201674123bd7SChris Mason 	path->slots[level] = 0;
201774123bd7SChris Mason 	return 0;
201874123bd7SChris Mason }
20195c680ed6SChris Mason 
20205c680ed6SChris Mason /*
20215c680ed6SChris Mason  * worker function to insert a single pointer in a node.
20225c680ed6SChris Mason  * the node should have enough room for the pointer already
202397571fd0SChris Mason  *
20245c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
20255c680ed6SChris Mason  * blocknr is the block the key points to.
2026aa5d6bedSChris Mason  *
2027aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
20285c680ed6SChris Mason  */
2029e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
2030e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
2031db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
20325c680ed6SChris Mason {
20335f39d397SChris Mason 	struct extent_buffer *lower;
20345c680ed6SChris Mason 	int nritems;
20355c680ed6SChris Mason 
20365c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
20375f39d397SChris Mason 	lower = path->nodes[level];
20385f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
203974123bd7SChris Mason 	if (slot > nritems)
204074123bd7SChris Mason 		BUG();
2041123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
204274123bd7SChris Mason 		BUG();
204374123bd7SChris Mason 	if (slot != nritems) {
20445f39d397SChris Mason 		memmove_extent_buffer(lower,
20455f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
20465f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
2047123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
204874123bd7SChris Mason 	}
20495f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
2050db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
205174493f7aSChris Mason 	WARN_ON(trans->transid == 0);
205274493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
20535f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
20545f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
205574123bd7SChris Mason 	return 0;
205674123bd7SChris Mason }
205774123bd7SChris Mason 
205897571fd0SChris Mason /*
205997571fd0SChris Mason  * split the node at the specified level in path in two.
206097571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
206197571fd0SChris Mason  *
206297571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
206397571fd0SChris Mason  * left and right, if either one works, it returns right away.
2064aa5d6bedSChris Mason  *
2065aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
206697571fd0SChris Mason  */
2067e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans,
2068e02119d5SChris Mason 			       struct btrfs_root *root,
2069e02119d5SChris Mason 			       struct btrfs_path *path, int level)
2070be0e5c09SChris Mason {
20715f39d397SChris Mason 	struct extent_buffer *c;
20725f39d397SChris Mason 	struct extent_buffer *split;
20735f39d397SChris Mason 	struct btrfs_disk_key disk_key;
2074be0e5c09SChris Mason 	int mid;
20755c680ed6SChris Mason 	int ret;
2076aa5d6bedSChris Mason 	int wret;
20777518a238SChris Mason 	u32 c_nritems;
2078be0e5c09SChris Mason 
20795f39d397SChris Mason 	c = path->nodes[level];
20807bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
20815f39d397SChris Mason 	if (c == root->node) {
20825c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
2083e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
20845c680ed6SChris Mason 		if (ret)
20855c680ed6SChris Mason 			return ret;
2086e66f709bSChris Mason 	} else {
2087e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
20885f39d397SChris Mason 		c = path->nodes[level];
20895f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
2090c448acf0SChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
2091e66f709bSChris Mason 			return 0;
209254aa1f4dSChris Mason 		if (ret < 0)
209354aa1f4dSChris Mason 			return ret;
20945c680ed6SChris Mason 	}
2095e66f709bSChris Mason 
20965f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
20977bb86316SChris Mason 
2098925baeddSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
209931840ae1SZheng Yan 					path->nodes[level + 1]->start,
21007bb86316SChris Mason 					root->root_key.objectid,
210131840ae1SZheng Yan 					trans->transid, level, c->start, 0);
21025f39d397SChris Mason 	if (IS_ERR(split))
21035f39d397SChris Mason 		return PTR_ERR(split);
210454aa1f4dSChris Mason 
21055f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
21065f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
2107db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
21085f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
21095f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
211063b10fc4SChris Mason 	btrfs_set_header_flags(split, 0);
21115f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
21125f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
21135f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2114e17cade2SChris Mason 	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
2115e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
2116e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
21175f39d397SChris Mason 
21187518a238SChris Mason 	mid = (c_nritems + 1) / 2;
21195f39d397SChris Mason 
21205f39d397SChris Mason 	copy_extent_buffer(split, c,
21215f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
21225f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
2123123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
21245f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
21255f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
2126aa5d6bedSChris Mason 	ret = 0;
2127aa5d6bedSChris Mason 
21285f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
21295f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
21305f39d397SChris Mason 
21315f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
2132db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
21335f39d397SChris Mason 			  path->slots[level + 1] + 1,
2134123abc88SChris Mason 			  level + 1);
2135aa5d6bedSChris Mason 	if (wret)
2136aa5d6bedSChris Mason 		ret = wret;
2137aa5d6bedSChris Mason 
213831840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, c, split, 0, c_nritems - mid);
213931840ae1SZheng Yan 	BUG_ON(ret);
214031840ae1SZheng Yan 
21415de08d7dSChris Mason 	if (path->slots[level] >= mid) {
21425c680ed6SChris Mason 		path->slots[level] -= mid;
2143925baeddSChris Mason 		btrfs_tree_unlock(c);
21445f39d397SChris Mason 		free_extent_buffer(c);
21455f39d397SChris Mason 		path->nodes[level] = split;
21465c680ed6SChris Mason 		path->slots[level + 1] += 1;
2147eb60ceacSChris Mason 	} else {
2148925baeddSChris Mason 		btrfs_tree_unlock(split);
21495f39d397SChris Mason 		free_extent_buffer(split);
2150be0e5c09SChris Mason 	}
2151aa5d6bedSChris Mason 	return ret;
2152be0e5c09SChris Mason }
2153be0e5c09SChris Mason 
215474123bd7SChris Mason /*
215574123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
215674123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
215774123bd7SChris Mason  * space used both by the item structs and the item data
215874123bd7SChris Mason  */
21595f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
2160be0e5c09SChris Mason {
2161be0e5c09SChris Mason 	int data_len;
21625f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
2163d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
2164be0e5c09SChris Mason 
2165be0e5c09SChris Mason 	if (!nr)
2166be0e5c09SChris Mason 		return 0;
21675f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
21685f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
21690783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
2170d4dbff95SChris Mason 	WARN_ON(data_len < 0);
2171be0e5c09SChris Mason 	return data_len;
2172be0e5c09SChris Mason }
2173be0e5c09SChris Mason 
217474123bd7SChris Mason /*
2175d4dbff95SChris Mason  * The space between the end of the leaf items and
2176d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
2177d4dbff95SChris Mason  * the leaf has left for both items and data
2178d4dbff95SChris Mason  */
2179e02119d5SChris Mason int noinline btrfs_leaf_free_space(struct btrfs_root *root,
2180e02119d5SChris Mason 				   struct extent_buffer *leaf)
2181d4dbff95SChris Mason {
21825f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
21835f39d397SChris Mason 	int ret;
21845f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
21855f39d397SChris Mason 	if (ret < 0) {
21865f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
2187ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
21885f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
21895f39d397SChris Mason 	}
21905f39d397SChris Mason 	return ret;
2191d4dbff95SChris Mason }
2192d4dbff95SChris Mason 
2193d4dbff95SChris Mason /*
219400ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
219500ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
2196aa5d6bedSChris Mason  *
2197aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
2198aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
219900ec4c51SChris Mason  */
2200e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
220134a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
220234a38218SChris Mason 			   int empty)
220300ec4c51SChris Mason {
22045f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
22055f39d397SChris Mason 	struct extent_buffer *right;
22065f39d397SChris Mason 	struct extent_buffer *upper;
22075f39d397SChris Mason 	struct btrfs_disk_key disk_key;
220800ec4c51SChris Mason 	int slot;
220934a38218SChris Mason 	u32 i;
221000ec4c51SChris Mason 	int free_space;
221100ec4c51SChris Mason 	int push_space = 0;
221200ec4c51SChris Mason 	int push_items = 0;
22130783fcfcSChris Mason 	struct btrfs_item *item;
22147518a238SChris Mason 	u32 left_nritems;
221534a38218SChris Mason 	u32 nr;
22167518a238SChris Mason 	u32 right_nritems;
22175f39d397SChris Mason 	u32 data_end;
2218db94535dSChris Mason 	u32 this_item_size;
221954aa1f4dSChris Mason 	int ret;
222000ec4c51SChris Mason 
222100ec4c51SChris Mason 	slot = path->slots[1];
222200ec4c51SChris Mason 	if (!path->nodes[1]) {
222300ec4c51SChris Mason 		return 1;
222400ec4c51SChris Mason 	}
222500ec4c51SChris Mason 	upper = path->nodes[1];
22265f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
222700ec4c51SChris Mason 		return 1;
22285f39d397SChris Mason 
2229a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
2230a2135011SChris Mason 
2231ca7a79adSChris Mason 	right = read_node_slot(root, upper, slot + 1);
2232925baeddSChris Mason 	btrfs_tree_lock(right);
2233123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
223487b29b20SYan Zheng 	if (free_space < data_size)
2235925baeddSChris Mason 		goto out_unlock;
223602217ed2SChris Mason 
22375f39d397SChris Mason 	/* cow and double check */
22385f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
223965b51a00SChris Mason 			      slot + 1, &right, 0);
2240925baeddSChris Mason 	if (ret)
2241925baeddSChris Mason 		goto out_unlock;
2242925baeddSChris Mason 
22435f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
224487b29b20SYan Zheng 	if (free_space < data_size)
2245925baeddSChris Mason 		goto out_unlock;
22465f39d397SChris Mason 
22475f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
2248925baeddSChris Mason 	if (left_nritems == 0)
2249925baeddSChris Mason 		goto out_unlock;
22505f39d397SChris Mason 
225134a38218SChris Mason 	if (empty)
225234a38218SChris Mason 		nr = 0;
225334a38218SChris Mason 	else
225434a38218SChris Mason 		nr = 1;
225534a38218SChris Mason 
225631840ae1SZheng Yan 	if (path->slots[0] >= left_nritems)
225787b29b20SYan Zheng 		push_space += data_size;
225831840ae1SZheng Yan 
225934a38218SChris Mason 	i = left_nritems - 1;
226034a38218SChris Mason 	while (i >= nr) {
22615f39d397SChris Mason 		item = btrfs_item_nr(left, i);
2262db94535dSChris Mason 
226331840ae1SZheng Yan 		if (!empty && push_items > 0) {
226431840ae1SZheng Yan 			if (path->slots[0] > i)
226531840ae1SZheng Yan 				break;
226631840ae1SZheng Yan 			if (path->slots[0] == i) {
226731840ae1SZheng Yan 				int space = btrfs_leaf_free_space(root, left);
226831840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
226931840ae1SZheng Yan 					break;
227031840ae1SZheng Yan 			}
227131840ae1SZheng Yan 		}
227231840ae1SZheng Yan 
227300ec4c51SChris Mason 		if (path->slots[0] == i)
227487b29b20SYan Zheng 			push_space += data_size;
2275db94535dSChris Mason 
2276db94535dSChris Mason 		if (!left->map_token) {
2277db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
2278db94535dSChris Mason 					sizeof(struct btrfs_item),
2279db94535dSChris Mason 					&left->map_token, &left->kaddr,
2280db94535dSChris Mason 					&left->map_start, &left->map_len,
2281db94535dSChris Mason 					KM_USER1);
2282db94535dSChris Mason 		}
2283db94535dSChris Mason 
2284db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
2285db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
228600ec4c51SChris Mason 			break;
228731840ae1SZheng Yan 
228800ec4c51SChris Mason 		push_items++;
2289db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
229034a38218SChris Mason 		if (i == 0)
229134a38218SChris Mason 			break;
229234a38218SChris Mason 		i--;
2293db94535dSChris Mason 	}
2294db94535dSChris Mason 	if (left->map_token) {
2295db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
2296db94535dSChris Mason 		left->map_token = NULL;
229700ec4c51SChris Mason 	}
22985f39d397SChris Mason 
2299925baeddSChris Mason 	if (push_items == 0)
2300925baeddSChris Mason 		goto out_unlock;
23015f39d397SChris Mason 
230234a38218SChris Mason 	if (!empty && push_items == left_nritems)
2303a429e513SChris Mason 		WARN_ON(1);
23045f39d397SChris Mason 
230500ec4c51SChris Mason 	/* push left to right */
23065f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
230734a38218SChris Mason 
23085f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
2309123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
23105f39d397SChris Mason 
231100ec4c51SChris Mason 	/* make room in the right data area */
23125f39d397SChris Mason 	data_end = leaf_data_end(root, right);
23135f39d397SChris Mason 	memmove_extent_buffer(right,
23145f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
23155f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
23165f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
23175f39d397SChris Mason 
231800ec4c51SChris Mason 	/* copy from the left data area */
23195f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
2320d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
2321d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
2322d6025579SChris Mason 		     push_space);
23235f39d397SChris Mason 
23245f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
23255f39d397SChris Mason 			      btrfs_item_nr_offset(0),
23260783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
23275f39d397SChris Mason 
232800ec4c51SChris Mason 	/* copy the items from left to right */
23295f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
23305f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
23310783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
233200ec4c51SChris Mason 
233300ec4c51SChris Mason 	/* update the item pointers */
23347518a238SChris Mason 	right_nritems += push_items;
23355f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
2336123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
23377518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
23385f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2339db94535dSChris Mason 		if (!right->map_token) {
2340db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2341db94535dSChris Mason 					sizeof(struct btrfs_item),
2342db94535dSChris Mason 					&right->map_token, &right->kaddr,
2343db94535dSChris Mason 					&right->map_start, &right->map_len,
2344db94535dSChris Mason 					KM_USER1);
2345db94535dSChris Mason 		}
2346db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
2347db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
2348db94535dSChris Mason 	}
2349db94535dSChris Mason 
2350db94535dSChris Mason 	if (right->map_token) {
2351db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2352db94535dSChris Mason 		right->map_token = NULL;
235300ec4c51SChris Mason 	}
23547518a238SChris Mason 	left_nritems -= push_items;
23555f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
235600ec4c51SChris Mason 
235734a38218SChris Mason 	if (left_nritems)
23585f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
23595f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
2360a429e513SChris Mason 
236131840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, left, right, 0, push_items);
236231840ae1SZheng Yan 	BUG_ON(ret);
236331840ae1SZheng Yan 
23645f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
23655f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
2366d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
236702217ed2SChris Mason 
236800ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
23697518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
23707518a238SChris Mason 		path->slots[0] -= left_nritems;
2371925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
2372925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
2373925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
23745f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
23755f39d397SChris Mason 		path->nodes[0] = right;
237600ec4c51SChris Mason 		path->slots[1] += 1;
237700ec4c51SChris Mason 	} else {
2378925baeddSChris Mason 		btrfs_tree_unlock(right);
23795f39d397SChris Mason 		free_extent_buffer(right);
238000ec4c51SChris Mason 	}
238100ec4c51SChris Mason 	return 0;
2382925baeddSChris Mason 
2383925baeddSChris Mason out_unlock:
2384925baeddSChris Mason 	btrfs_tree_unlock(right);
2385925baeddSChris Mason 	free_extent_buffer(right);
2386925baeddSChris Mason 	return 1;
238700ec4c51SChris Mason }
2388925baeddSChris Mason 
238900ec4c51SChris Mason /*
239074123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
239174123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
239274123bd7SChris Mason  */
2393e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
239434a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
239534a38218SChris Mason 			  int empty)
2396be0e5c09SChris Mason {
23975f39d397SChris Mason 	struct btrfs_disk_key disk_key;
23985f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
23995f39d397SChris Mason 	struct extent_buffer *left;
2400be0e5c09SChris Mason 	int slot;
2401be0e5c09SChris Mason 	int i;
2402be0e5c09SChris Mason 	int free_space;
2403be0e5c09SChris Mason 	int push_space = 0;
2404be0e5c09SChris Mason 	int push_items = 0;
24050783fcfcSChris Mason 	struct btrfs_item *item;
24067518a238SChris Mason 	u32 old_left_nritems;
24075f39d397SChris Mason 	u32 right_nritems;
240834a38218SChris Mason 	u32 nr;
2409aa5d6bedSChris Mason 	int ret = 0;
2410aa5d6bedSChris Mason 	int wret;
2411db94535dSChris Mason 	u32 this_item_size;
2412db94535dSChris Mason 	u32 old_left_item_size;
2413be0e5c09SChris Mason 
2414be0e5c09SChris Mason 	slot = path->slots[1];
24155f39d397SChris Mason 	if (slot == 0)
2416be0e5c09SChris Mason 		return 1;
24175f39d397SChris Mason 	if (!path->nodes[1])
2418be0e5c09SChris Mason 		return 1;
24195f39d397SChris Mason 
24203685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
24213685f791SChris Mason 	if (right_nritems == 0) {
24223685f791SChris Mason 		return 1;
24233685f791SChris Mason 	}
24243685f791SChris Mason 
2425a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
2426a2135011SChris Mason 
2427ca7a79adSChris Mason 	left = read_node_slot(root, path->nodes[1], slot - 1);
2428925baeddSChris Mason 	btrfs_tree_lock(left);
2429123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
243087b29b20SYan Zheng 	if (free_space < data_size) {
2431925baeddSChris Mason 		ret = 1;
2432925baeddSChris Mason 		goto out;
2433be0e5c09SChris Mason 	}
243402217ed2SChris Mason 
243502217ed2SChris Mason 	/* cow and double check */
24365f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
243765b51a00SChris Mason 			      path->nodes[1], slot - 1, &left, 0);
243854aa1f4dSChris Mason 	if (ret) {
243954aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
2440925baeddSChris Mason 		ret = 1;
2441925baeddSChris Mason 		goto out;
244254aa1f4dSChris Mason 	}
24433685f791SChris Mason 
2444123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
244587b29b20SYan Zheng 	if (free_space < data_size) {
2446925baeddSChris Mason 		ret = 1;
2447925baeddSChris Mason 		goto out;
244802217ed2SChris Mason 	}
244902217ed2SChris Mason 
245034a38218SChris Mason 	if (empty)
245134a38218SChris Mason 		nr = right_nritems;
245234a38218SChris Mason 	else
245334a38218SChris Mason 		nr = right_nritems - 1;
245434a38218SChris Mason 
245534a38218SChris Mason 	for (i = 0; i < nr; i++) {
24565f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2457db94535dSChris Mason 		if (!right->map_token) {
2458db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2459db94535dSChris Mason 					sizeof(struct btrfs_item),
2460db94535dSChris Mason 					&right->map_token, &right->kaddr,
2461db94535dSChris Mason 					&right->map_start, &right->map_len,
2462db94535dSChris Mason 					KM_USER1);
2463db94535dSChris Mason 		}
2464db94535dSChris Mason 
246531840ae1SZheng Yan 		if (!empty && push_items > 0) {
246631840ae1SZheng Yan 			if (path->slots[0] < i)
246731840ae1SZheng Yan 				break;
246831840ae1SZheng Yan 			if (path->slots[0] == i) {
246931840ae1SZheng Yan 				int space = btrfs_leaf_free_space(root, right);
247031840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
247131840ae1SZheng Yan 					break;
247231840ae1SZheng Yan 			}
247331840ae1SZheng Yan 		}
247431840ae1SZheng Yan 
2475be0e5c09SChris Mason 		if (path->slots[0] == i)
247687b29b20SYan Zheng 			push_space += data_size;
2477db94535dSChris Mason 
2478db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
2479db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
2480be0e5c09SChris Mason 			break;
2481db94535dSChris Mason 
2482be0e5c09SChris Mason 		push_items++;
2483db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
2484be0e5c09SChris Mason 	}
2485db94535dSChris Mason 
2486db94535dSChris Mason 	if (right->map_token) {
2487db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2488db94535dSChris Mason 		right->map_token = NULL;
2489db94535dSChris Mason 	}
2490db94535dSChris Mason 
2491be0e5c09SChris Mason 	if (push_items == 0) {
2492925baeddSChris Mason 		ret = 1;
2493925baeddSChris Mason 		goto out;
2494be0e5c09SChris Mason 	}
249534a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
2496a429e513SChris Mason 		WARN_ON(1);
24975f39d397SChris Mason 
2498be0e5c09SChris Mason 	/* push data from right to left */
24995f39d397SChris Mason 	copy_extent_buffer(left, right,
25005f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
25015f39d397SChris Mason 			   btrfs_item_nr_offset(0),
25025f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
25035f39d397SChris Mason 
2504123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
25055f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
25065f39d397SChris Mason 
25075f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
2508d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
2509123abc88SChris Mason 		     btrfs_leaf_data(right) +
25105f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
2511be0e5c09SChris Mason 		     push_space);
25125f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
251387b29b20SYan Zheng 	BUG_ON(old_left_nritems <= 0);
2514eb60ceacSChris Mason 
2515db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
2516be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
25175f39d397SChris Mason 		u32 ioff;
2518db94535dSChris Mason 
25195f39d397SChris Mason 		item = btrfs_item_nr(left, i);
2520db94535dSChris Mason 		if (!left->map_token) {
2521db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
2522db94535dSChris Mason 					sizeof(struct btrfs_item),
2523db94535dSChris Mason 					&left->map_token, &left->kaddr,
2524db94535dSChris Mason 					&left->map_start, &left->map_len,
2525db94535dSChris Mason 					KM_USER1);
2526db94535dSChris Mason 		}
2527db94535dSChris Mason 
25285f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
25295f39d397SChris Mason 		btrfs_set_item_offset(left, item,
2530db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2531be0e5c09SChris Mason 	}
25325f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2533db94535dSChris Mason 	if (left->map_token) {
2534db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
2535db94535dSChris Mason 		left->map_token = NULL;
2536db94535dSChris Mason 	}
2537be0e5c09SChris Mason 
2538be0e5c09SChris Mason 	/* fixup right node */
253934a38218SChris Mason 	if (push_items > right_nritems) {
254034a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
254134a38218SChris Mason 		WARN_ON(1);
254234a38218SChris Mason 	}
254334a38218SChris Mason 
254434a38218SChris Mason 	if (push_items < right_nritems) {
25455f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
2546123abc88SChris Mason 						  leaf_data_end(root, right);
25475f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
2548d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
2549d6025579SChris Mason 				      btrfs_leaf_data(right) +
2550123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
25515f39d397SChris Mason 
25525f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
25535f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
25545f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
25550783fcfcSChris Mason 			     sizeof(struct btrfs_item));
255634a38218SChris Mason 	}
2557eef1c494SYan 	right_nritems -= push_items;
2558eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
2559123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
25605f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
25615f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2562db94535dSChris Mason 
2563db94535dSChris Mason 		if (!right->map_token) {
2564db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2565db94535dSChris Mason 					sizeof(struct btrfs_item),
2566db94535dSChris Mason 					&right->map_token, &right->kaddr,
2567db94535dSChris Mason 					&right->map_start, &right->map_len,
2568db94535dSChris Mason 					KM_USER1);
2569db94535dSChris Mason 		}
2570db94535dSChris Mason 
2571db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
2572db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
2573db94535dSChris Mason 	}
2574db94535dSChris Mason 	if (right->map_token) {
2575db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2576db94535dSChris Mason 		right->map_token = NULL;
2577be0e5c09SChris Mason 	}
2578eb60ceacSChris Mason 
25795f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
258034a38218SChris Mason 	if (right_nritems)
25815f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
2582098f59c2SChris Mason 
258331840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, right, left,
258431840ae1SZheng Yan 			       old_left_nritems, push_items);
258531840ae1SZheng Yan 	BUG_ON(ret);
258631840ae1SZheng Yan 
25875f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
25885f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
2589aa5d6bedSChris Mason 	if (wret)
2590aa5d6bedSChris Mason 		ret = wret;
2591be0e5c09SChris Mason 
2592be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
2593be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
2594be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
2595925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
2596925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
2597925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
25985f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
25995f39d397SChris Mason 		path->nodes[0] = left;
2600be0e5c09SChris Mason 		path->slots[1] -= 1;
2601be0e5c09SChris Mason 	} else {
2602925baeddSChris Mason 		btrfs_tree_unlock(left);
26035f39d397SChris Mason 		free_extent_buffer(left);
2604be0e5c09SChris Mason 		path->slots[0] -= push_items;
2605be0e5c09SChris Mason 	}
2606eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2607aa5d6bedSChris Mason 	return ret;
2608925baeddSChris Mason out:
2609925baeddSChris Mason 	btrfs_tree_unlock(left);
2610925baeddSChris Mason 	free_extent_buffer(left);
2611925baeddSChris Mason 	return ret;
2612be0e5c09SChris Mason }
2613be0e5c09SChris Mason 
261474123bd7SChris Mason /*
261574123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
261674123bd7SChris Mason  * available for the resulting leaf level of the path.
2617aa5d6bedSChris Mason  *
2618aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
261974123bd7SChris Mason  */
2620e02119d5SChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans,
2621e02119d5SChris Mason 			       struct btrfs_root *root,
2622e02119d5SChris Mason 			       struct btrfs_key *ins_key,
2623e02119d5SChris Mason 			       struct btrfs_path *path, int data_size,
2624e02119d5SChris Mason 			       int extend)
2625be0e5c09SChris Mason {
26265f39d397SChris Mason 	struct extent_buffer *l;
26277518a238SChris Mason 	u32 nritems;
2628eb60ceacSChris Mason 	int mid;
2629eb60ceacSChris Mason 	int slot;
26305f39d397SChris Mason 	struct extent_buffer *right;
2631be0e5c09SChris Mason 	int data_copy_size;
2632be0e5c09SChris Mason 	int rt_data_off;
2633be0e5c09SChris Mason 	int i;
2634d4dbff95SChris Mason 	int ret = 0;
2635aa5d6bedSChris Mason 	int wret;
2636cc0c5538SChris Mason 	int double_split;
2637cc0c5538SChris Mason 	int num_doubles = 0;
2638d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
2639be0e5c09SChris Mason 
264040689478SChris Mason 	/* first try to make some room by pushing left and right */
2641459931ecSChris Mason 	if (data_size && ins_key->type != BTRFS_DIR_ITEM_KEY) {
264234a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
26433326d1b0SChris Mason 		if (wret < 0) {
2644eaee50e8SChris Mason 			return wret;
26453326d1b0SChris Mason 		}
2646eaee50e8SChris Mason 		if (wret) {
264734a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
2648eaee50e8SChris Mason 			if (wret < 0)
2649eaee50e8SChris Mason 				return wret;
2650eaee50e8SChris Mason 		}
26515f39d397SChris Mason 		l = path->nodes[0];
2652aa5d6bedSChris Mason 
2653aa5d6bedSChris Mason 		/* did the pushes work? */
265487b29b20SYan Zheng 		if (btrfs_leaf_free_space(root, l) >= data_size)
2655be0e5c09SChris Mason 			return 0;
26563326d1b0SChris Mason 	}
2657aa5d6bedSChris Mason 
26585c680ed6SChris Mason 	if (!path->nodes[1]) {
2659e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
26605c680ed6SChris Mason 		if (ret)
26615c680ed6SChris Mason 			return ret;
26625c680ed6SChris Mason 	}
2663cc0c5538SChris Mason again:
2664cc0c5538SChris Mason 	double_split = 0;
2665cc0c5538SChris Mason 	l = path->nodes[0];
2666eb60ceacSChris Mason 	slot = path->slots[0];
26675f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
2668eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
266954aa1f4dSChris Mason 
2670925baeddSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
267131840ae1SZheng Yan 					path->nodes[1]->start,
26727bb86316SChris Mason 					root->root_key.objectid,
267331840ae1SZheng Yan 					trans->transid, 0, l->start, 0);
2674cea9e445SChris Mason 	if (IS_ERR(right)) {
2675cea9e445SChris Mason 		BUG_ON(1);
26765f39d397SChris Mason 		return PTR_ERR(right);
2677cea9e445SChris Mason 	}
267854aa1f4dSChris Mason 
26795f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2680db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
26815f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
26825f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
26835f39d397SChris Mason 	btrfs_set_header_level(right, 0);
26845f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
26855f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
26865f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2687e17cade2SChris Mason 
2688e17cade2SChris Mason 	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
2689e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
2690e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
2691d4dbff95SChris Mason 	if (mid <= slot) {
2692d4dbff95SChris Mason 		if (nritems == 1 ||
269387b29b20SYan Zheng 		    leaf_space_used(l, mid, nritems - mid) + data_size >
2694d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2695d4dbff95SChris Mason 			if (slot >= nritems) {
2696d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
26975f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2698d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2699db94535dSChris Mason 						  &disk_key, right->start,
2700d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
2701d4dbff95SChris Mason 				if (wret)
2702d4dbff95SChris Mason 					ret = wret;
2703925baeddSChris Mason 
2704925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
27055f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
27065f39d397SChris Mason 				path->nodes[0] = right;
2707d4dbff95SChris Mason 				path->slots[0] = 0;
2708d4dbff95SChris Mason 				path->slots[1] += 1;
27090ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2710d4dbff95SChris Mason 				return ret;
2711d4dbff95SChris Mason 			}
2712d4dbff95SChris Mason 			mid = slot;
27133326d1b0SChris Mason 			if (mid != nritems &&
27143326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
271587b29b20SYan Zheng 			    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
2716d4dbff95SChris Mason 				double_split = 1;
2717d4dbff95SChris Mason 			}
27183326d1b0SChris Mason 		}
2719d4dbff95SChris Mason 	} else {
272087b29b20SYan Zheng 		if (leaf_space_used(l, 0, mid) + data_size >
2721d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2722459931ecSChris Mason 			if (!extend && data_size && slot == 0) {
2723d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
27245f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2725d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2726d4dbff95SChris Mason 						  &disk_key,
2727db94535dSChris Mason 						  right->start,
2728098f59c2SChris Mason 						  path->slots[1], 1);
2729d4dbff95SChris Mason 				if (wret)
2730d4dbff95SChris Mason 					ret = wret;
2731925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
27325f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
27335f39d397SChris Mason 				path->nodes[0] = right;
2734d4dbff95SChris Mason 				path->slots[0] = 0;
2735a429e513SChris Mason 				if (path->slots[1] == 0) {
2736a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
2737a429e513SChris Mason 					           path, &disk_key, 1);
2738a429e513SChris Mason 					if (wret)
2739a429e513SChris Mason 						ret = wret;
2740a429e513SChris Mason 				}
27410ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2742d4dbff95SChris Mason 				return ret;
2743459931ecSChris Mason 			} else if ((extend || !data_size) && slot == 0) {
2744cc0c5538SChris Mason 				mid = 1;
2745cc0c5538SChris Mason 			} else {
2746d4dbff95SChris Mason 				mid = slot;
27475ee78ac7SChris Mason 				if (mid != nritems &&
27485ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
274987b29b20SYan Zheng 				    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
2750d4dbff95SChris Mason 					double_split = 1;
2751d4dbff95SChris Mason 				}
2752d4dbff95SChris Mason 			}
27535ee78ac7SChris Mason 		}
2754cc0c5538SChris Mason 	}
27555f39d397SChris Mason 	nritems = nritems - mid;
27565f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
27575f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
27585f39d397SChris Mason 
27595f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
27605f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
27615f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
27625f39d397SChris Mason 
27635f39d397SChris Mason 	copy_extent_buffer(right, l,
2764d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2765123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
2766123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
276774123bd7SChris Mason 
27685f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
27695f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
27705f39d397SChris Mason 
27715f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
27725f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
2773db94535dSChris Mason 		u32 ioff;
2774db94535dSChris Mason 
2775db94535dSChris Mason 		if (!right->map_token) {
2776db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2777db94535dSChris Mason 					sizeof(struct btrfs_item),
2778db94535dSChris Mason 					&right->map_token, &right->kaddr,
2779db94535dSChris Mason 					&right->map_start, &right->map_len,
2780db94535dSChris Mason 					KM_USER1);
2781db94535dSChris Mason 		}
2782db94535dSChris Mason 
2783db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
27845f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
27850783fcfcSChris Mason 	}
278674123bd7SChris Mason 
2787db94535dSChris Mason 	if (right->map_token) {
2788db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2789db94535dSChris Mason 		right->map_token = NULL;
2790db94535dSChris Mason 	}
2791db94535dSChris Mason 
27925f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
2793aa5d6bedSChris Mason 	ret = 0;
27945f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
2795db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
2796db94535dSChris Mason 			  path->slots[1] + 1, 1);
2797aa5d6bedSChris Mason 	if (wret)
2798aa5d6bedSChris Mason 		ret = wret;
27995f39d397SChris Mason 
28005f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
28015f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
2802eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
28035f39d397SChris Mason 
280431840ae1SZheng Yan 	ret = btrfs_update_ref(trans, root, l, right, 0, nritems);
280531840ae1SZheng Yan 	BUG_ON(ret);
280631840ae1SZheng Yan 
2807be0e5c09SChris Mason 	if (mid <= slot) {
2808925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
28095f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
28105f39d397SChris Mason 		path->nodes[0] = right;
2811be0e5c09SChris Mason 		path->slots[0] -= mid;
2812be0e5c09SChris Mason 		path->slots[1] += 1;
2813925baeddSChris Mason 	} else {
2814925baeddSChris Mason 		btrfs_tree_unlock(right);
28155f39d397SChris Mason 		free_extent_buffer(right);
2816925baeddSChris Mason 	}
28175f39d397SChris Mason 
2818eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2819d4dbff95SChris Mason 
2820cc0c5538SChris Mason 	if (double_split) {
2821cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
2822cc0c5538SChris Mason 		num_doubles++;
2823cc0c5538SChris Mason 		goto again;
28243326d1b0SChris Mason 	}
2825be0e5c09SChris Mason 	return ret;
2826be0e5c09SChris Mason }
2827be0e5c09SChris Mason 
2828d352ac68SChris Mason /*
2829459931ecSChris Mason  * This function splits a single item into two items,
2830459931ecSChris Mason  * giving 'new_key' to the new item and splitting the
2831459931ecSChris Mason  * old one at split_offset (from the start of the item).
2832459931ecSChris Mason  *
2833459931ecSChris Mason  * The path may be released by this operation.  After
2834459931ecSChris Mason  * the split, the path is pointing to the old item.  The
2835459931ecSChris Mason  * new item is going to be in the same node as the old one.
2836459931ecSChris Mason  *
2837459931ecSChris Mason  * Note, the item being split must be smaller enough to live alone on
2838459931ecSChris Mason  * a tree block with room for one extra struct btrfs_item
2839459931ecSChris Mason  *
2840459931ecSChris Mason  * This allows us to split the item in place, keeping a lock on the
2841459931ecSChris Mason  * leaf the entire time.
2842459931ecSChris Mason  */
2843459931ecSChris Mason int btrfs_split_item(struct btrfs_trans_handle *trans,
2844459931ecSChris Mason 		     struct btrfs_root *root,
2845459931ecSChris Mason 		     struct btrfs_path *path,
2846459931ecSChris Mason 		     struct btrfs_key *new_key,
2847459931ecSChris Mason 		     unsigned long split_offset)
2848459931ecSChris Mason {
2849459931ecSChris Mason 	u32 item_size;
2850459931ecSChris Mason 	struct extent_buffer *leaf;
2851459931ecSChris Mason 	struct btrfs_key orig_key;
2852459931ecSChris Mason 	struct btrfs_item *item;
2853459931ecSChris Mason 	struct btrfs_item *new_item;
2854459931ecSChris Mason 	int ret = 0;
2855459931ecSChris Mason 	int slot;
2856459931ecSChris Mason 	u32 nritems;
2857459931ecSChris Mason 	u32 orig_offset;
2858459931ecSChris Mason 	struct btrfs_disk_key disk_key;
2859459931ecSChris Mason 	char *buf;
2860459931ecSChris Mason 
2861459931ecSChris Mason 	leaf = path->nodes[0];
2862459931ecSChris Mason 	btrfs_item_key_to_cpu(leaf, &orig_key, path->slots[0]);
2863459931ecSChris Mason 	if (btrfs_leaf_free_space(root, leaf) >= sizeof(struct btrfs_item))
2864459931ecSChris Mason 		goto split;
2865459931ecSChris Mason 
2866459931ecSChris Mason 	item_size = btrfs_item_size_nr(leaf, path->slots[0]);
2867459931ecSChris Mason 	btrfs_release_path(root, path);
2868459931ecSChris Mason 
2869459931ecSChris Mason 	path->search_for_split = 1;
2870459931ecSChris Mason 	path->keep_locks = 1;
2871459931ecSChris Mason 
2872459931ecSChris Mason 	ret = btrfs_search_slot(trans, root, &orig_key, path, 0, 1);
2873459931ecSChris Mason 	path->search_for_split = 0;
2874459931ecSChris Mason 
2875459931ecSChris Mason 	/* if our item isn't there or got smaller, return now */
2876459931ecSChris Mason 	if (ret != 0 || item_size != btrfs_item_size_nr(path->nodes[0],
2877459931ecSChris Mason 							path->slots[0])) {
2878459931ecSChris Mason 		path->keep_locks = 0;
2879459931ecSChris Mason 		return -EAGAIN;
2880459931ecSChris Mason 	}
2881459931ecSChris Mason 
288287b29b20SYan Zheng 	ret = split_leaf(trans, root, &orig_key, path,
288387b29b20SYan Zheng 			 sizeof(struct btrfs_item), 1);
2884459931ecSChris Mason 	path->keep_locks = 0;
2885459931ecSChris Mason 	BUG_ON(ret);
2886459931ecSChris Mason 
2887459931ecSChris Mason 	leaf = path->nodes[0];
288842dc7babSChris Mason 	BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item));
2889459931ecSChris Mason 
2890459931ecSChris Mason split:
2891459931ecSChris Mason 	item = btrfs_item_nr(leaf, path->slots[0]);
2892459931ecSChris Mason 	orig_offset = btrfs_item_offset(leaf, item);
2893459931ecSChris Mason 	item_size = btrfs_item_size(leaf, item);
2894459931ecSChris Mason 
2895459931ecSChris Mason 
2896459931ecSChris Mason 	buf = kmalloc(item_size, GFP_NOFS);
2897459931ecSChris Mason 	read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf,
2898459931ecSChris Mason 			    path->slots[0]), item_size);
2899459931ecSChris Mason 	slot = path->slots[0] + 1;
2900459931ecSChris Mason 	leaf = path->nodes[0];
2901459931ecSChris Mason 
2902459931ecSChris Mason 	nritems = btrfs_header_nritems(leaf);
2903459931ecSChris Mason 
2904459931ecSChris Mason 	if (slot != nritems) {
2905459931ecSChris Mason 		/* shift the items */
2906459931ecSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
2907459931ecSChris Mason 			      btrfs_item_nr_offset(slot),
2908459931ecSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2909459931ecSChris Mason 
2910459931ecSChris Mason 	}
2911459931ecSChris Mason 
2912459931ecSChris Mason 	btrfs_cpu_key_to_disk(&disk_key, new_key);
2913459931ecSChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
2914459931ecSChris Mason 
2915459931ecSChris Mason 	new_item = btrfs_item_nr(leaf, slot);
2916459931ecSChris Mason 
2917459931ecSChris Mason 	btrfs_set_item_offset(leaf, new_item, orig_offset);
2918459931ecSChris Mason 	btrfs_set_item_size(leaf, new_item, item_size - split_offset);
2919459931ecSChris Mason 
2920459931ecSChris Mason 	btrfs_set_item_offset(leaf, item,
2921459931ecSChris Mason 			      orig_offset + item_size - split_offset);
2922459931ecSChris Mason 	btrfs_set_item_size(leaf, item, split_offset);
2923459931ecSChris Mason 
2924459931ecSChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
2925459931ecSChris Mason 
2926459931ecSChris Mason 	/* write the data for the start of the original item */
2927459931ecSChris Mason 	write_extent_buffer(leaf, buf,
2928459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, path->slots[0]),
2929459931ecSChris Mason 			    split_offset);
2930459931ecSChris Mason 
2931459931ecSChris Mason 	/* write the data for the new item */
2932459931ecSChris Mason 	write_extent_buffer(leaf, buf + split_offset,
2933459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, slot),
2934459931ecSChris Mason 			    item_size - split_offset);
2935459931ecSChris Mason 	btrfs_mark_buffer_dirty(leaf);
2936459931ecSChris Mason 
2937459931ecSChris Mason 	ret = 0;
2938459931ecSChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
2939459931ecSChris Mason 		btrfs_print_leaf(root, leaf);
2940459931ecSChris Mason 		BUG();
2941459931ecSChris Mason 	}
2942459931ecSChris Mason 	kfree(buf);
2943459931ecSChris Mason 	return ret;
2944459931ecSChris Mason }
2945459931ecSChris Mason 
2946459931ecSChris Mason /*
2947d352ac68SChris Mason  * make the item pointed to by the path smaller.  new_size indicates
2948d352ac68SChris Mason  * how small to make it, and from_end tells us if we just chop bytes
2949d352ac68SChris Mason  * off the end of the item or if we shift the item to chop bytes off
2950d352ac68SChris Mason  * the front.
2951d352ac68SChris Mason  */
2952b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2953b18c6685SChris Mason 			struct btrfs_root *root,
2954b18c6685SChris Mason 			struct btrfs_path *path,
2955179e29e4SChris Mason 			u32 new_size, int from_end)
2956b18c6685SChris Mason {
2957b18c6685SChris Mason 	int ret = 0;
2958b18c6685SChris Mason 	int slot;
2959b18c6685SChris Mason 	int slot_orig;
29605f39d397SChris Mason 	struct extent_buffer *leaf;
29615f39d397SChris Mason 	struct btrfs_item *item;
2962b18c6685SChris Mason 	u32 nritems;
2963b18c6685SChris Mason 	unsigned int data_end;
2964b18c6685SChris Mason 	unsigned int old_data_start;
2965b18c6685SChris Mason 	unsigned int old_size;
2966b18c6685SChris Mason 	unsigned int size_diff;
2967b18c6685SChris Mason 	int i;
2968b18c6685SChris Mason 
2969b18c6685SChris Mason 	slot_orig = path->slots[0];
29705f39d397SChris Mason 	leaf = path->nodes[0];
2971179e29e4SChris Mason 	slot = path->slots[0];
2972179e29e4SChris Mason 
2973179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
2974179e29e4SChris Mason 	if (old_size == new_size)
2975179e29e4SChris Mason 		return 0;
2976b18c6685SChris Mason 
29775f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2978b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
2979b18c6685SChris Mason 
29805f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
2981179e29e4SChris Mason 
2982b18c6685SChris Mason 	size_diff = old_size - new_size;
2983b18c6685SChris Mason 
2984b18c6685SChris Mason 	BUG_ON(slot < 0);
2985b18c6685SChris Mason 	BUG_ON(slot >= nritems);
2986b18c6685SChris Mason 
2987b18c6685SChris Mason 	/*
2988b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2989b18c6685SChris Mason 	 */
2990b18c6685SChris Mason 	/* first correct the data pointers */
2991b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
29925f39d397SChris Mason 		u32 ioff;
29935f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2994db94535dSChris Mason 
2995db94535dSChris Mason 		if (!leaf->map_token) {
2996db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2997db94535dSChris Mason 					sizeof(struct btrfs_item),
2998db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2999db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
3000db94535dSChris Mason 					KM_USER1);
3001db94535dSChris Mason 		}
3002db94535dSChris Mason 
30035f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
30045f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
3005b18c6685SChris Mason 	}
3006db94535dSChris Mason 
3007db94535dSChris Mason 	if (leaf->map_token) {
3008db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
3009db94535dSChris Mason 		leaf->map_token = NULL;
3010db94535dSChris Mason 	}
3011db94535dSChris Mason 
3012b18c6685SChris Mason 	/* shift the data */
3013179e29e4SChris Mason 	if (from_end) {
30145f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
3015b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
3016b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
3017179e29e4SChris Mason 	} else {
3018179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
3019179e29e4SChris Mason 		u64 offset;
3020179e29e4SChris Mason 
3021179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
3022179e29e4SChris Mason 
3023179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
3024179e29e4SChris Mason 			unsigned long ptr;
3025179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
3026179e29e4SChris Mason 
3027179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
3028179e29e4SChris Mason 					    struct btrfs_file_extent_item);
3029179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
3030179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
3031179e29e4SChris Mason 
3032179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
3033179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
3034179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
3035179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
3036179e29e4SChris Mason 				        (unsigned long)fi,
3037179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
3038179e29e4SChris Mason 						 disk_bytenr));
3039179e29e4SChris Mason 			}
3040179e29e4SChris Mason 		}
3041179e29e4SChris Mason 
3042179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
3043179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
3044179e29e4SChris Mason 			      data_end, old_data_start - data_end);
3045179e29e4SChris Mason 
3046179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
3047179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
3048179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
3049179e29e4SChris Mason 		if (slot == 0)
3050179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
3051179e29e4SChris Mason 	}
30525f39d397SChris Mason 
30535f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
30545f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
30555f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3056b18c6685SChris Mason 
3057b18c6685SChris Mason 	ret = 0;
30585f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
30595f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
3060b18c6685SChris Mason 		BUG();
30615f39d397SChris Mason 	}
3062b18c6685SChris Mason 	return ret;
3063b18c6685SChris Mason }
3064b18c6685SChris Mason 
3065d352ac68SChris Mason /*
3066d352ac68SChris Mason  * make the item pointed to by the path bigger, data_size is the new size.
3067d352ac68SChris Mason  */
30685f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
30695f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
30705f39d397SChris Mason 		      u32 data_size)
30716567e837SChris Mason {
30726567e837SChris Mason 	int ret = 0;
30736567e837SChris Mason 	int slot;
30746567e837SChris Mason 	int slot_orig;
30755f39d397SChris Mason 	struct extent_buffer *leaf;
30765f39d397SChris Mason 	struct btrfs_item *item;
30776567e837SChris Mason 	u32 nritems;
30786567e837SChris Mason 	unsigned int data_end;
30796567e837SChris Mason 	unsigned int old_data;
30806567e837SChris Mason 	unsigned int old_size;
30816567e837SChris Mason 	int i;
30826567e837SChris Mason 
30836567e837SChris Mason 	slot_orig = path->slots[0];
30845f39d397SChris Mason 	leaf = path->nodes[0];
30856567e837SChris Mason 
30865f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
30876567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
30886567e837SChris Mason 
30895f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
30905f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
30916567e837SChris Mason 		BUG();
30925f39d397SChris Mason 	}
30936567e837SChris Mason 	slot = path->slots[0];
30945f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
30956567e837SChris Mason 
30966567e837SChris Mason 	BUG_ON(slot < 0);
30973326d1b0SChris Mason 	if (slot >= nritems) {
30983326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
30993326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
31003326d1b0SChris Mason 		BUG_ON(1);
31013326d1b0SChris Mason 	}
31026567e837SChris Mason 
31036567e837SChris Mason 	/*
31046567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
31056567e837SChris Mason 	 */
31066567e837SChris Mason 	/* first correct the data pointers */
31076567e837SChris Mason 	for (i = slot; i < nritems; i++) {
31085f39d397SChris Mason 		u32 ioff;
31095f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
3110db94535dSChris Mason 
3111db94535dSChris Mason 		if (!leaf->map_token) {
3112db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
3113db94535dSChris Mason 					sizeof(struct btrfs_item),
3114db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
3115db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
3116db94535dSChris Mason 					KM_USER1);
3117db94535dSChris Mason 		}
31185f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
31195f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
31206567e837SChris Mason 	}
31215f39d397SChris Mason 
3122db94535dSChris Mason 	if (leaf->map_token) {
3123db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
3124db94535dSChris Mason 		leaf->map_token = NULL;
3125db94535dSChris Mason 	}
3126db94535dSChris Mason 
31276567e837SChris Mason 	/* shift the data */
31285f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
31296567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
31306567e837SChris Mason 		      data_end, old_data - data_end);
31315f39d397SChris Mason 
31326567e837SChris Mason 	data_end = old_data;
31335f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
31345f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
31355f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
31365f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
31376567e837SChris Mason 
31386567e837SChris Mason 	ret = 0;
31395f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
31405f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
31416567e837SChris Mason 		BUG();
31425f39d397SChris Mason 	}
31436567e837SChris Mason 	return ret;
31446567e837SChris Mason }
31456567e837SChris Mason 
314674123bd7SChris Mason /*
3147d352ac68SChris Mason  * Given a key and some data, insert items into the tree.
314874123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
3149f3465ca4SJosef Bacik  * Returns the number of keys that were inserted.
3150f3465ca4SJosef Bacik  */
3151f3465ca4SJosef Bacik int btrfs_insert_some_items(struct btrfs_trans_handle *trans,
3152f3465ca4SJosef Bacik 			    struct btrfs_root *root,
3153f3465ca4SJosef Bacik 			    struct btrfs_path *path,
3154f3465ca4SJosef Bacik 			    struct btrfs_key *cpu_key, u32 *data_size,
3155f3465ca4SJosef Bacik 			    int nr)
3156f3465ca4SJosef Bacik {
3157f3465ca4SJosef Bacik 	struct extent_buffer *leaf;
3158f3465ca4SJosef Bacik 	struct btrfs_item *item;
3159f3465ca4SJosef Bacik 	int ret = 0;
3160f3465ca4SJosef Bacik 	int slot;
3161f3465ca4SJosef Bacik 	int i;
3162f3465ca4SJosef Bacik 	u32 nritems;
3163f3465ca4SJosef Bacik 	u32 total_data = 0;
3164f3465ca4SJosef Bacik 	u32 total_size = 0;
3165f3465ca4SJosef Bacik 	unsigned int data_end;
3166f3465ca4SJosef Bacik 	struct btrfs_disk_key disk_key;
3167f3465ca4SJosef Bacik 	struct btrfs_key found_key;
3168f3465ca4SJosef Bacik 
316987b29b20SYan Zheng 	for (i = 0; i < nr; i++) {
317087b29b20SYan Zheng 		if (total_size + data_size[i] + sizeof(struct btrfs_item) >
317187b29b20SYan Zheng 		    BTRFS_LEAF_DATA_SIZE(root)) {
317287b29b20SYan Zheng 			break;
317387b29b20SYan Zheng 			nr = i;
317487b29b20SYan Zheng 		}
3175f3465ca4SJosef Bacik 		total_data += data_size[i];
317687b29b20SYan Zheng 		total_size += data_size[i] + sizeof(struct btrfs_item);
317787b29b20SYan Zheng 	}
317887b29b20SYan Zheng 	BUG_ON(nr == 0);
3179f3465ca4SJosef Bacik 
3180f3465ca4SJosef Bacik 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
3181f3465ca4SJosef Bacik 	if (ret == 0)
3182f3465ca4SJosef Bacik 		return -EEXIST;
3183f3465ca4SJosef Bacik 	if (ret < 0)
3184f3465ca4SJosef Bacik 		goto out;
3185f3465ca4SJosef Bacik 
3186f3465ca4SJosef Bacik 	leaf = path->nodes[0];
3187f3465ca4SJosef Bacik 
3188f3465ca4SJosef Bacik 	nritems = btrfs_header_nritems(leaf);
3189f3465ca4SJosef Bacik 	data_end = leaf_data_end(root, leaf);
3190f3465ca4SJosef Bacik 
3191f3465ca4SJosef Bacik 	if (btrfs_leaf_free_space(root, leaf) < total_size) {
3192f3465ca4SJosef Bacik 		for (i = nr; i >= 0; i--) {
3193f3465ca4SJosef Bacik 			total_data -= data_size[i];
3194f3465ca4SJosef Bacik 			total_size -= data_size[i] + sizeof(struct btrfs_item);
3195f3465ca4SJosef Bacik 			if (total_size < btrfs_leaf_free_space(root, leaf))
3196f3465ca4SJosef Bacik 				break;
3197f3465ca4SJosef Bacik 		}
3198f3465ca4SJosef Bacik 		nr = i;
3199f3465ca4SJosef Bacik 	}
3200f3465ca4SJosef Bacik 
3201f3465ca4SJosef Bacik 	slot = path->slots[0];
3202f3465ca4SJosef Bacik 	BUG_ON(slot < 0);
3203f3465ca4SJosef Bacik 
3204f3465ca4SJosef Bacik 	if (slot != nritems) {
3205f3465ca4SJosef Bacik 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
3206f3465ca4SJosef Bacik 
3207f3465ca4SJosef Bacik 		item = btrfs_item_nr(leaf, slot);
3208f3465ca4SJosef Bacik 		btrfs_item_key_to_cpu(leaf, &found_key, slot);
3209f3465ca4SJosef Bacik 
3210f3465ca4SJosef Bacik 		/* figure out how many keys we can insert in here */
3211f3465ca4SJosef Bacik 		total_data = data_size[0];
3212f3465ca4SJosef Bacik 		for (i = 1; i < nr; i++) {
3213f3465ca4SJosef Bacik 			if (comp_cpu_keys(&found_key, cpu_key + i) <= 0)
3214f3465ca4SJosef Bacik 				break;
3215f3465ca4SJosef Bacik 			total_data += data_size[i];
3216f3465ca4SJosef Bacik 		}
3217f3465ca4SJosef Bacik 		nr = i;
3218f3465ca4SJosef Bacik 
3219f3465ca4SJosef Bacik 		if (old_data < data_end) {
3220f3465ca4SJosef Bacik 			btrfs_print_leaf(root, leaf);
3221f3465ca4SJosef Bacik 			printk("slot %d old_data %d data_end %d\n",
3222f3465ca4SJosef Bacik 			       slot, old_data, data_end);
3223f3465ca4SJosef Bacik 			BUG_ON(1);
3224f3465ca4SJosef Bacik 		}
3225f3465ca4SJosef Bacik 		/*
3226f3465ca4SJosef Bacik 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3227f3465ca4SJosef Bacik 		 */
3228f3465ca4SJosef Bacik 		/* first correct the data pointers */
3229f3465ca4SJosef Bacik 		WARN_ON(leaf->map_token);
3230f3465ca4SJosef Bacik 		for (i = slot; i < nritems; i++) {
3231f3465ca4SJosef Bacik 			u32 ioff;
3232f3465ca4SJosef Bacik 
3233f3465ca4SJosef Bacik 			item = btrfs_item_nr(leaf, i);
3234f3465ca4SJosef Bacik 			if (!leaf->map_token) {
3235f3465ca4SJosef Bacik 				map_extent_buffer(leaf, (unsigned long)item,
3236f3465ca4SJosef Bacik 					sizeof(struct btrfs_item),
3237f3465ca4SJosef Bacik 					&leaf->map_token, &leaf->kaddr,
3238f3465ca4SJosef Bacik 					&leaf->map_start, &leaf->map_len,
3239f3465ca4SJosef Bacik 					KM_USER1);
3240f3465ca4SJosef Bacik 			}
3241f3465ca4SJosef Bacik 
3242f3465ca4SJosef Bacik 			ioff = btrfs_item_offset(leaf, item);
3243f3465ca4SJosef Bacik 			btrfs_set_item_offset(leaf, item, ioff - total_data);
3244f3465ca4SJosef Bacik 		}
3245f3465ca4SJosef Bacik 		if (leaf->map_token) {
3246f3465ca4SJosef Bacik 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
3247f3465ca4SJosef Bacik 			leaf->map_token = NULL;
3248f3465ca4SJosef Bacik 		}
3249f3465ca4SJosef Bacik 
3250f3465ca4SJosef Bacik 		/* shift the items */
3251f3465ca4SJosef Bacik 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
3252f3465ca4SJosef Bacik 			      btrfs_item_nr_offset(slot),
3253f3465ca4SJosef Bacik 			      (nritems - slot) * sizeof(struct btrfs_item));
3254f3465ca4SJosef Bacik 
3255f3465ca4SJosef Bacik 		/* shift the data */
3256f3465ca4SJosef Bacik 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
3257f3465ca4SJosef Bacik 			      data_end - total_data, btrfs_leaf_data(leaf) +
3258f3465ca4SJosef Bacik 			      data_end, old_data - data_end);
3259f3465ca4SJosef Bacik 		data_end = old_data;
3260f3465ca4SJosef Bacik 	} else {
3261f3465ca4SJosef Bacik 		/*
3262f3465ca4SJosef Bacik 		 * this sucks but it has to be done, if we are inserting at
3263f3465ca4SJosef Bacik 		 * the end of the leaf only insert 1 of the items, since we
3264f3465ca4SJosef Bacik 		 * have no way of knowing whats on the next leaf and we'd have
3265f3465ca4SJosef Bacik 		 * to drop our current locks to figure it out
3266f3465ca4SJosef Bacik 		 */
3267f3465ca4SJosef Bacik 		nr = 1;
3268f3465ca4SJosef Bacik 	}
3269f3465ca4SJosef Bacik 
3270f3465ca4SJosef Bacik 	/* setup the item for the new data */
3271f3465ca4SJosef Bacik 	for (i = 0; i < nr; i++) {
3272f3465ca4SJosef Bacik 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
3273f3465ca4SJosef Bacik 		btrfs_set_item_key(leaf, &disk_key, slot + i);
3274f3465ca4SJosef Bacik 		item = btrfs_item_nr(leaf, slot + i);
3275f3465ca4SJosef Bacik 		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
3276f3465ca4SJosef Bacik 		data_end -= data_size[i];
3277f3465ca4SJosef Bacik 		btrfs_set_item_size(leaf, item, data_size[i]);
3278f3465ca4SJosef Bacik 	}
3279f3465ca4SJosef Bacik 	btrfs_set_header_nritems(leaf, nritems + nr);
3280f3465ca4SJosef Bacik 	btrfs_mark_buffer_dirty(leaf);
3281f3465ca4SJosef Bacik 
3282f3465ca4SJosef Bacik 	ret = 0;
3283f3465ca4SJosef Bacik 	if (slot == 0) {
3284f3465ca4SJosef Bacik 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
3285f3465ca4SJosef Bacik 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
3286f3465ca4SJosef Bacik 	}
3287f3465ca4SJosef Bacik 
3288f3465ca4SJosef Bacik 	if (btrfs_leaf_free_space(root, leaf) < 0) {
3289f3465ca4SJosef Bacik 		btrfs_print_leaf(root, leaf);
3290f3465ca4SJosef Bacik 		BUG();
3291f3465ca4SJosef Bacik 	}
3292f3465ca4SJosef Bacik out:
3293f3465ca4SJosef Bacik 	if (!ret)
3294f3465ca4SJosef Bacik 		ret = nr;
3295f3465ca4SJosef Bacik 	return ret;
3296f3465ca4SJosef Bacik }
3297f3465ca4SJosef Bacik 
3298f3465ca4SJosef Bacik /*
3299f3465ca4SJosef Bacik  * Given a key and some data, insert items into the tree.
3300f3465ca4SJosef Bacik  * This does all the path init required, making room in the tree if needed.
330174123bd7SChris Mason  */
33029c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
33035f39d397SChris Mason 			    struct btrfs_root *root,
33045f39d397SChris Mason 			    struct btrfs_path *path,
33059c58309dSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
33069c58309dSChris Mason 			    int nr)
3307be0e5c09SChris Mason {
33085f39d397SChris Mason 	struct extent_buffer *leaf;
33095f39d397SChris Mason 	struct btrfs_item *item;
3310aa5d6bedSChris Mason 	int ret = 0;
3311be0e5c09SChris Mason 	int slot;
3312eb60ceacSChris Mason 	int slot_orig;
33139c58309dSChris Mason 	int i;
33147518a238SChris Mason 	u32 nritems;
33159c58309dSChris Mason 	u32 total_size = 0;
33169c58309dSChris Mason 	u32 total_data = 0;
3317be0e5c09SChris Mason 	unsigned int data_end;
3318e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
3319e2fa7227SChris Mason 
33209c58309dSChris Mason 	for (i = 0; i < nr; i++) {
33219c58309dSChris Mason 		total_data += data_size[i];
33229c58309dSChris Mason 	}
3323be0e5c09SChris Mason 
33247b128766SJosef Bacik 	total_size = total_data + (nr * sizeof(struct btrfs_item));
33259c58309dSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
33260f9dd46cSJosef Bacik 	if (ret == 0)
3327f0930a37SChris Mason 		return -EEXIST;
3328ed2ff2cbSChris Mason 	if (ret < 0)
3329ed2ff2cbSChris Mason 		goto out;
3330be0e5c09SChris Mason 
333162e2749eSChris Mason 	slot_orig = path->slots[0];
33325f39d397SChris Mason 	leaf = path->nodes[0];
333374123bd7SChris Mason 
33345f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
3335123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
3336eb60ceacSChris Mason 
3337f25956ccSChris Mason 	if (btrfs_leaf_free_space(root, leaf) < total_size) {
33383326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
33393326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
33409c58309dSChris Mason 		       total_size, btrfs_leaf_free_space(root, leaf));
3341be0e5c09SChris Mason 		BUG();
3342d4dbff95SChris Mason 	}
33435f39d397SChris Mason 
334462e2749eSChris Mason 	slot = path->slots[0];
3345eb60ceacSChris Mason 	BUG_ON(slot < 0);
33465f39d397SChris Mason 
3347be0e5c09SChris Mason 	if (slot != nritems) {
33485f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
3349be0e5c09SChris Mason 
33505f39d397SChris Mason 		if (old_data < data_end) {
33515f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
33525f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
33535f39d397SChris Mason 			       slot, old_data, data_end);
33545f39d397SChris Mason 			BUG_ON(1);
33555f39d397SChris Mason 		}
3356be0e5c09SChris Mason 		/*
3357be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3358be0e5c09SChris Mason 		 */
3359be0e5c09SChris Mason 		/* first correct the data pointers */
3360db94535dSChris Mason 		WARN_ON(leaf->map_token);
33610783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
33625f39d397SChris Mason 			u32 ioff;
3363db94535dSChris Mason 
33645f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
3365db94535dSChris Mason 			if (!leaf->map_token) {
3366db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
3367db94535dSChris Mason 					sizeof(struct btrfs_item),
3368db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
3369db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
3370db94535dSChris Mason 					KM_USER1);
3371db94535dSChris Mason 			}
3372db94535dSChris Mason 
33735f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
33749c58309dSChris Mason 			btrfs_set_item_offset(leaf, item, ioff - total_data);
33750783fcfcSChris Mason 		}
3376db94535dSChris Mason 		if (leaf->map_token) {
3377db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
3378db94535dSChris Mason 			leaf->map_token = NULL;
3379db94535dSChris Mason 		}
3380be0e5c09SChris Mason 
3381be0e5c09SChris Mason 		/* shift the items */
33829c58309dSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
33835f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
33840783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
3385be0e5c09SChris Mason 
3386be0e5c09SChris Mason 		/* shift the data */
33875f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
33889c58309dSChris Mason 			      data_end - total_data, btrfs_leaf_data(leaf) +
3389be0e5c09SChris Mason 			      data_end, old_data - data_end);
3390be0e5c09SChris Mason 		data_end = old_data;
3391be0e5c09SChris Mason 	}
33925f39d397SChris Mason 
339362e2749eSChris Mason 	/* setup the item for the new data */
33949c58309dSChris Mason 	for (i = 0; i < nr; i++) {
33959c58309dSChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
33969c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
33979c58309dSChris Mason 		item = btrfs_item_nr(leaf, slot + i);
33989c58309dSChris Mason 		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
33999c58309dSChris Mason 		data_end -= data_size[i];
34009c58309dSChris Mason 		btrfs_set_item_size(leaf, item, data_size[i]);
34019c58309dSChris Mason 	}
34029c58309dSChris Mason 	btrfs_set_header_nritems(leaf, nritems + nr);
34035f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3404aa5d6bedSChris Mason 
3405aa5d6bedSChris Mason 	ret = 0;
34065a01a2e3SChris Mason 	if (slot == 0) {
34075a01a2e3SChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
3408e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
34095a01a2e3SChris Mason 	}
3410aa5d6bedSChris Mason 
34115f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
34125f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
3413be0e5c09SChris Mason 		BUG();
34145f39d397SChris Mason 	}
3415ed2ff2cbSChris Mason out:
341662e2749eSChris Mason 	return ret;
341762e2749eSChris Mason }
341862e2749eSChris Mason 
341962e2749eSChris Mason /*
342062e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
342162e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
342262e2749eSChris Mason  */
3423e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
3424e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
3425e089f05cSChris Mason 		      data_size)
342662e2749eSChris Mason {
342762e2749eSChris Mason 	int ret = 0;
34282c90e5d6SChris Mason 	struct btrfs_path *path;
34295f39d397SChris Mason 	struct extent_buffer *leaf;
34305f39d397SChris Mason 	unsigned long ptr;
343162e2749eSChris Mason 
34322c90e5d6SChris Mason 	path = btrfs_alloc_path();
34332c90e5d6SChris Mason 	BUG_ON(!path);
34342c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
343562e2749eSChris Mason 	if (!ret) {
34365f39d397SChris Mason 		leaf = path->nodes[0];
34375f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
34385f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
34395f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
344062e2749eSChris Mason 	}
34412c90e5d6SChris Mason 	btrfs_free_path(path);
3442aa5d6bedSChris Mason 	return ret;
3443be0e5c09SChris Mason }
3444be0e5c09SChris Mason 
344574123bd7SChris Mason /*
34465de08d7dSChris Mason  * delete the pointer from a given node.
344774123bd7SChris Mason  *
3448d352ac68SChris Mason  * the tree should have been previously balanced so the deletion does not
3449d352ac68SChris Mason  * empty a node.
345074123bd7SChris Mason  */
3451e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
3452e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
3453be0e5c09SChris Mason {
34545f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
34557518a238SChris Mason 	u32 nritems;
3456aa5d6bedSChris Mason 	int ret = 0;
3457bb803951SChris Mason 	int wret;
3458be0e5c09SChris Mason 
34595f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
3460be0e5c09SChris Mason 	if (slot != nritems -1) {
34615f39d397SChris Mason 		memmove_extent_buffer(parent,
34625f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
34635f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
3464d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
3465d6025579SChris Mason 			      (nritems - slot - 1));
3466be0e5c09SChris Mason 	}
34677518a238SChris Mason 	nritems--;
34685f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
34697518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
34705f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
3471eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
34725f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
3473bb803951SChris Mason 	} else if (slot == 0) {
34745f39d397SChris Mason 		struct btrfs_disk_key disk_key;
34755f39d397SChris Mason 
34765f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
34775f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
34780f70abe2SChris Mason 		if (wret)
34790f70abe2SChris Mason 			ret = wret;
3480be0e5c09SChris Mason 	}
3481d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
3482aa5d6bedSChris Mason 	return ret;
3483be0e5c09SChris Mason }
3484be0e5c09SChris Mason 
348574123bd7SChris Mason /*
3486323ac95bSChris Mason  * a helper function to delete the leaf pointed to by path->slots[1] and
3487323ac95bSChris Mason  * path->nodes[1].  bytenr is the node block pointer, but since the callers
3488323ac95bSChris Mason  * already know it, it is faster to have them pass it down than to
3489323ac95bSChris Mason  * read it out of the node again.
3490323ac95bSChris Mason  *
3491323ac95bSChris Mason  * This deletes the pointer in path->nodes[1] and frees the leaf
3492323ac95bSChris Mason  * block extent.  zero is returned if it all worked out, < 0 otherwise.
3493323ac95bSChris Mason  *
3494323ac95bSChris Mason  * The path must have already been setup for deleting the leaf, including
3495323ac95bSChris Mason  * all the proper balancing.  path->nodes[1] must be locked.
3496323ac95bSChris Mason  */
3497323ac95bSChris Mason noinline int btrfs_del_leaf(struct btrfs_trans_handle *trans,
3498323ac95bSChris Mason 			    struct btrfs_root *root,
3499323ac95bSChris Mason 			    struct btrfs_path *path, u64 bytenr)
3500323ac95bSChris Mason {
3501323ac95bSChris Mason 	int ret;
3502323ac95bSChris Mason 	u64 root_gen = btrfs_header_generation(path->nodes[1]);
3503323ac95bSChris Mason 
3504323ac95bSChris Mason 	ret = del_ptr(trans, root, path, 1, path->slots[1]);
3505323ac95bSChris Mason 	if (ret)
3506323ac95bSChris Mason 		return ret;
3507323ac95bSChris Mason 
3508323ac95bSChris Mason 	ret = btrfs_free_extent(trans, root, bytenr,
3509323ac95bSChris Mason 				btrfs_level_size(root, 0),
3510323ac95bSChris Mason 				path->nodes[1]->start,
3511323ac95bSChris Mason 				btrfs_header_owner(path->nodes[1]),
35123bb1a1bcSYan Zheng 				root_gen, 0, 1);
3513323ac95bSChris Mason 	return ret;
3514323ac95bSChris Mason }
3515323ac95bSChris Mason /*
351674123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
351774123bd7SChris Mason  * the leaf, remove it from the tree
351874123bd7SChris Mason  */
351985e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
352085e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
3521be0e5c09SChris Mason {
35225f39d397SChris Mason 	struct extent_buffer *leaf;
35235f39d397SChris Mason 	struct btrfs_item *item;
352485e21bacSChris Mason 	int last_off;
352585e21bacSChris Mason 	int dsize = 0;
3526aa5d6bedSChris Mason 	int ret = 0;
3527aa5d6bedSChris Mason 	int wret;
352885e21bacSChris Mason 	int i;
35297518a238SChris Mason 	u32 nritems;
3530be0e5c09SChris Mason 
35315f39d397SChris Mason 	leaf = path->nodes[0];
353285e21bacSChris Mason 	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
353385e21bacSChris Mason 
353485e21bacSChris Mason 	for (i = 0; i < nr; i++)
353585e21bacSChris Mason 		dsize += btrfs_item_size_nr(leaf, slot + i);
353685e21bacSChris Mason 
35375f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
3538be0e5c09SChris Mason 
353985e21bacSChris Mason 	if (slot + nr != nritems) {
3540123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
35415f39d397SChris Mason 
35425f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
3543d6025579SChris Mason 			      data_end + dsize,
3544123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
354585e21bacSChris Mason 			      last_off - data_end);
35465f39d397SChris Mason 
354785e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
35485f39d397SChris Mason 			u32 ioff;
3549db94535dSChris Mason 
35505f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
3551db94535dSChris Mason 			if (!leaf->map_token) {
3552db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
3553db94535dSChris Mason 					sizeof(struct btrfs_item),
3554db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
3555db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
3556db94535dSChris Mason 					KM_USER1);
3557db94535dSChris Mason 			}
35585f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
35595f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
35600783fcfcSChris Mason 		}
3561db94535dSChris Mason 
3562db94535dSChris Mason 		if (leaf->map_token) {
3563db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
3564db94535dSChris Mason 			leaf->map_token = NULL;
3565db94535dSChris Mason 		}
3566db94535dSChris Mason 
35675f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
356885e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
35690783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
357085e21bacSChris Mason 			      (nritems - slot - nr));
3571be0e5c09SChris Mason 	}
357285e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
357385e21bacSChris Mason 	nritems -= nr;
35745f39d397SChris Mason 
357574123bd7SChris Mason 	/* delete the leaf if we've emptied it */
35767518a238SChris Mason 	if (nritems == 0) {
35775f39d397SChris Mason 		if (leaf == root->node) {
35785f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
35799a8dd150SChris Mason 		} else {
3580323ac95bSChris Mason 			ret = btrfs_del_leaf(trans, root, path, leaf->start);
3581323ac95bSChris Mason 			BUG_ON(ret);
35829a8dd150SChris Mason 		}
3583be0e5c09SChris Mason 	} else {
35847518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
3585aa5d6bedSChris Mason 		if (slot == 0) {
35865f39d397SChris Mason 			struct btrfs_disk_key disk_key;
35875f39d397SChris Mason 
35885f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
3589e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
35905f39d397SChris Mason 					      &disk_key, 1);
3591aa5d6bedSChris Mason 			if (wret)
3592aa5d6bedSChris Mason 				ret = wret;
3593aa5d6bedSChris Mason 		}
3594aa5d6bedSChris Mason 
359574123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
359685e21bacSChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
3597be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
3598be0e5c09SChris Mason 			 * make sure the path still points to our leaf
3599be0e5c09SChris Mason 			 * for possible call to del_ptr below
3600be0e5c09SChris Mason 			 */
36014920c9acSChris Mason 			slot = path->slots[1];
36025f39d397SChris Mason 			extent_buffer_get(leaf);
36035f39d397SChris Mason 
360485e21bacSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1);
360554aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
3606aa5d6bedSChris Mason 				ret = wret;
36075f39d397SChris Mason 
36085f39d397SChris Mason 			if (path->nodes[0] == leaf &&
36095f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
361085e21bacSChris Mason 				wret = push_leaf_right(trans, root, path, 1, 1);
361154aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
3612aa5d6bedSChris Mason 					ret = wret;
3613aa5d6bedSChris Mason 			}
36145f39d397SChris Mason 
36155f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
3616323ac95bSChris Mason 				path->slots[1] = slot;
3617323ac95bSChris Mason 				ret = btrfs_del_leaf(trans, root, path, leaf->start);
3618323ac95bSChris Mason 				BUG_ON(ret);
36195f39d397SChris Mason 				free_extent_buffer(leaf);
36205de08d7dSChris Mason 			} else {
3621925baeddSChris Mason 				/* if we're still in the path, make sure
3622925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
3623925baeddSChris Mason 				 * push_leaf functions must have already
3624925baeddSChris Mason 				 * dirtied this buffer
3625925baeddSChris Mason 				 */
3626925baeddSChris Mason 				if (path->nodes[0] == leaf)
36275f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
36285f39d397SChris Mason 				free_extent_buffer(leaf);
3629be0e5c09SChris Mason 			}
3630d5719762SChris Mason 		} else {
36315f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
3632be0e5c09SChris Mason 		}
36339a8dd150SChris Mason 	}
3634aa5d6bedSChris Mason 	return ret;
36359a8dd150SChris Mason }
36369a8dd150SChris Mason 
363797571fd0SChris Mason /*
3638925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
36397bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
36407bb86316SChris Mason  * returns < 0 on io errors.
3641d352ac68SChris Mason  *
3642d352ac68SChris Mason  * This may release the path, and so you may lose any locks held at the
3643d352ac68SChris Mason  * time you call it.
36447bb86316SChris Mason  */
36457bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
36467bb86316SChris Mason {
3647925baeddSChris Mason 	struct btrfs_key key;
3648925baeddSChris Mason 	struct btrfs_disk_key found_key;
3649925baeddSChris Mason 	int ret;
36507bb86316SChris Mason 
3651925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
3652925baeddSChris Mason 
3653925baeddSChris Mason 	if (key.offset > 0)
3654925baeddSChris Mason 		key.offset--;
3655925baeddSChris Mason 	else if (key.type > 0)
3656925baeddSChris Mason 		key.type--;
3657925baeddSChris Mason 	else if (key.objectid > 0)
3658925baeddSChris Mason 		key.objectid--;
3659925baeddSChris Mason 	else
36607bb86316SChris Mason 		return 1;
36617bb86316SChris Mason 
3662925baeddSChris Mason 	btrfs_release_path(root, path);
3663925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
3664925baeddSChris Mason 	if (ret < 0)
3665925baeddSChris Mason 		return ret;
3666925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
3667925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
3668925baeddSChris Mason 	if (ret < 0)
36697bb86316SChris Mason 		return 0;
3670925baeddSChris Mason 	return 1;
36717bb86316SChris Mason }
36727bb86316SChris Mason 
36733f157a2fSChris Mason /*
36743f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
36753f157a2fSChris Mason  * for nodes or leaves that are either in cache or have a minimum
3676d352ac68SChris Mason  * transaction id.  This is used by the btree defrag code, and tree logging
36773f157a2fSChris Mason  *
36783f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
36793f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
36803f157a2fSChris Mason  * key and get a writable path.
36813f157a2fSChris Mason  *
36823f157a2fSChris Mason  * This does lock as it descends, and path->keep_locks should be set
36833f157a2fSChris Mason  * to 1 by the caller.
36843f157a2fSChris Mason  *
36853f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
36863f157a2fSChris Mason  * of the tree.
36873f157a2fSChris Mason  *
3688d352ac68SChris Mason  * min_trans indicates the oldest transaction that you are interested
3689d352ac68SChris Mason  * in walking through.  Any nodes or leaves older than min_trans are
3690d352ac68SChris Mason  * skipped over (without reading them).
3691d352ac68SChris Mason  *
36923f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
36933f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
36943f157a2fSChris Mason  */
36953f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
3696e02119d5SChris Mason 			 struct btrfs_key *max_key,
36973f157a2fSChris Mason 			 struct btrfs_path *path, int cache_only,
36983f157a2fSChris Mason 			 u64 min_trans)
36993f157a2fSChris Mason {
37003f157a2fSChris Mason 	struct extent_buffer *cur;
37013f157a2fSChris Mason 	struct btrfs_key found_key;
37023f157a2fSChris Mason 	int slot;
37039652480bSYan 	int sret;
37043f157a2fSChris Mason 	u32 nritems;
37053f157a2fSChris Mason 	int level;
37063f157a2fSChris Mason 	int ret = 1;
37073f157a2fSChris Mason 
3708934d375bSChris Mason 	WARN_ON(!path->keep_locks);
37093f157a2fSChris Mason again:
37103f157a2fSChris Mason 	cur = btrfs_lock_root_node(root);
37113f157a2fSChris Mason 	level = btrfs_header_level(cur);
3712e02119d5SChris Mason 	WARN_ON(path->nodes[level]);
37133f157a2fSChris Mason 	path->nodes[level] = cur;
37143f157a2fSChris Mason 	path->locks[level] = 1;
37153f157a2fSChris Mason 
37163f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
37173f157a2fSChris Mason 		ret = 1;
37183f157a2fSChris Mason 		goto out;
37193f157a2fSChris Mason 	}
37203f157a2fSChris Mason 	while(1) {
37213f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
37223f157a2fSChris Mason 		level = btrfs_header_level(cur);
37239652480bSYan 		sret = bin_search(cur, min_key, level, &slot);
37243f157a2fSChris Mason 
3725323ac95bSChris Mason 		/* at the lowest level, we're done, setup the path and exit */
3726323ac95bSChris Mason 		if (level == path->lowest_level) {
3727e02119d5SChris Mason 			if (slot >= nritems)
3728e02119d5SChris Mason 				goto find_next_key;
37293f157a2fSChris Mason 			ret = 0;
37303f157a2fSChris Mason 			path->slots[level] = slot;
37313f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
37323f157a2fSChris Mason 			goto out;
37333f157a2fSChris Mason 		}
37349652480bSYan 		if (sret && slot > 0)
37359652480bSYan 			slot--;
37363f157a2fSChris Mason 		/*
37373f157a2fSChris Mason 		 * check this node pointer against the cache_only and
37383f157a2fSChris Mason 		 * min_trans parameters.  If it isn't in cache or is too
37393f157a2fSChris Mason 		 * old, skip to the next one.
37403f157a2fSChris Mason 		 */
37413f157a2fSChris Mason 		while(slot < nritems) {
37423f157a2fSChris Mason 			u64 blockptr;
37433f157a2fSChris Mason 			u64 gen;
37443f157a2fSChris Mason 			struct extent_buffer *tmp;
3745e02119d5SChris Mason 			struct btrfs_disk_key disk_key;
3746e02119d5SChris Mason 
37473f157a2fSChris Mason 			blockptr = btrfs_node_blockptr(cur, slot);
37483f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
37493f157a2fSChris Mason 			if (gen < min_trans) {
37503f157a2fSChris Mason 				slot++;
37513f157a2fSChris Mason 				continue;
37523f157a2fSChris Mason 			}
37533f157a2fSChris Mason 			if (!cache_only)
37543f157a2fSChris Mason 				break;
37553f157a2fSChris Mason 
3756e02119d5SChris Mason 			if (max_key) {
3757e02119d5SChris Mason 				btrfs_node_key(cur, &disk_key, slot);
3758e02119d5SChris Mason 				if (comp_keys(&disk_key, max_key) >= 0) {
3759e02119d5SChris Mason 					ret = 1;
3760e02119d5SChris Mason 					goto out;
3761e02119d5SChris Mason 				}
3762e02119d5SChris Mason 			}
3763e02119d5SChris Mason 
37643f157a2fSChris Mason 			tmp = btrfs_find_tree_block(root, blockptr,
37653f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
37663f157a2fSChris Mason 
37673f157a2fSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
37683f157a2fSChris Mason 				free_extent_buffer(tmp);
37693f157a2fSChris Mason 				break;
37703f157a2fSChris Mason 			}
37713f157a2fSChris Mason 			if (tmp)
37723f157a2fSChris Mason 				free_extent_buffer(tmp);
37733f157a2fSChris Mason 			slot++;
37743f157a2fSChris Mason 		}
3775e02119d5SChris Mason find_next_key:
37763f157a2fSChris Mason 		/*
37773f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
37783f157a2fSChris Mason 		 * and find another one
37793f157a2fSChris Mason 		 */
37803f157a2fSChris Mason 		if (slot >= nritems) {
3781e02119d5SChris Mason 			path->slots[level] = slot;
3782e02119d5SChris Mason 			sret = btrfs_find_next_key(root, path, min_key, level,
37833f157a2fSChris Mason 						  cache_only, min_trans);
3784e02119d5SChris Mason 			if (sret == 0) {
37853f157a2fSChris Mason 				btrfs_release_path(root, path);
37863f157a2fSChris Mason 				goto again;
37873f157a2fSChris Mason 			} else {
37883f157a2fSChris Mason 				goto out;
37893f157a2fSChris Mason 			}
37903f157a2fSChris Mason 		}
37913f157a2fSChris Mason 		/* save our key for returning back */
37923f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
37933f157a2fSChris Mason 		path->slots[level] = slot;
37943f157a2fSChris Mason 		if (level == path->lowest_level) {
37953f157a2fSChris Mason 			ret = 0;
37963f157a2fSChris Mason 			unlock_up(path, level, 1);
37973f157a2fSChris Mason 			goto out;
37983f157a2fSChris Mason 		}
37993f157a2fSChris Mason 		cur = read_node_slot(root, cur, slot);
38003f157a2fSChris Mason 
38013f157a2fSChris Mason 		btrfs_tree_lock(cur);
38023f157a2fSChris Mason 		path->locks[level - 1] = 1;
38033f157a2fSChris Mason 		path->nodes[level - 1] = cur;
38043f157a2fSChris Mason 		unlock_up(path, level, 1);
38053f157a2fSChris Mason 	}
38063f157a2fSChris Mason out:
38073f157a2fSChris Mason 	if (ret == 0)
38083f157a2fSChris Mason 		memcpy(min_key, &found_key, sizeof(found_key));
38093f157a2fSChris Mason 	return ret;
38103f157a2fSChris Mason }
38113f157a2fSChris Mason 
38123f157a2fSChris Mason /*
38133f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
38143f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
38153f157a2fSChris Mason  * tree based on the current path and the cache_only and min_trans
38163f157a2fSChris Mason  * parameters.
38173f157a2fSChris Mason  *
38183f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
38193f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
38203f157a2fSChris Mason  *
38213f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
38223f157a2fSChris Mason  * calling this function.
38233f157a2fSChris Mason  */
3824e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
38253f157a2fSChris Mason 			struct btrfs_key *key, int lowest_level,
38263f157a2fSChris Mason 			int cache_only, u64 min_trans)
3827e7a84565SChris Mason {
3828e7a84565SChris Mason 	int level = lowest_level;
3829e7a84565SChris Mason 	int slot;
3830e7a84565SChris Mason 	struct extent_buffer *c;
3831e7a84565SChris Mason 
3832934d375bSChris Mason 	WARN_ON(!path->keep_locks);
3833e7a84565SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3834e7a84565SChris Mason 		if (!path->nodes[level])
3835e7a84565SChris Mason 			return 1;
3836e7a84565SChris Mason 
3837e7a84565SChris Mason 		slot = path->slots[level] + 1;
3838e7a84565SChris Mason 		c = path->nodes[level];
38393f157a2fSChris Mason next:
3840e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3841e7a84565SChris Mason 			level++;
3842e7a84565SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
3843e7a84565SChris Mason 				return 1;
3844e7a84565SChris Mason 			}
3845e7a84565SChris Mason 			continue;
3846e7a84565SChris Mason 		}
3847e7a84565SChris Mason 		if (level == 0)
3848e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
38493f157a2fSChris Mason 		else {
38503f157a2fSChris Mason 			u64 blockptr = btrfs_node_blockptr(c, slot);
38513f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
38523f157a2fSChris Mason 
38533f157a2fSChris Mason 			if (cache_only) {
38543f157a2fSChris Mason 				struct extent_buffer *cur;
38553f157a2fSChris Mason 				cur = btrfs_find_tree_block(root, blockptr,
38563f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
38573f157a2fSChris Mason 				if (!cur || !btrfs_buffer_uptodate(cur, gen)) {
38583f157a2fSChris Mason 					slot++;
38593f157a2fSChris Mason 					if (cur)
38603f157a2fSChris Mason 						free_extent_buffer(cur);
38613f157a2fSChris Mason 					goto next;
38623f157a2fSChris Mason 				}
38633f157a2fSChris Mason 				free_extent_buffer(cur);
38643f157a2fSChris Mason 			}
38653f157a2fSChris Mason 			if (gen < min_trans) {
38663f157a2fSChris Mason 				slot++;
38673f157a2fSChris Mason 				goto next;
38683f157a2fSChris Mason 			}
3869e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
38703f157a2fSChris Mason 		}
3871e7a84565SChris Mason 		return 0;
3872e7a84565SChris Mason 	}
3873e7a84565SChris Mason 	return 1;
3874e7a84565SChris Mason }
3875e7a84565SChris Mason 
38767bb86316SChris Mason /*
3877925baeddSChris Mason  * search the tree again to find a leaf with greater keys
38780f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
38790f70abe2SChris Mason  * returns < 0 on io errors.
388097571fd0SChris Mason  */
3881234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
3882d97e63b6SChris Mason {
3883d97e63b6SChris Mason 	int slot;
3884d97e63b6SChris Mason 	int level = 1;
38855f39d397SChris Mason 	struct extent_buffer *c;
38865f39d397SChris Mason 	struct extent_buffer *next = NULL;
3887925baeddSChris Mason 	struct btrfs_key key;
3888925baeddSChris Mason 	u32 nritems;
3889925baeddSChris Mason 	int ret;
3890925baeddSChris Mason 
3891925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3892925baeddSChris Mason 	if (nritems == 0) {
3893925baeddSChris Mason 		return 1;
3894925baeddSChris Mason 	}
3895925baeddSChris Mason 
3896925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
3897925baeddSChris Mason 
3898925baeddSChris Mason 	btrfs_release_path(root, path);
3899a2135011SChris Mason 	path->keep_locks = 1;
3900925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
3901925baeddSChris Mason 	path->keep_locks = 0;
3902925baeddSChris Mason 
3903925baeddSChris Mason 	if (ret < 0)
3904925baeddSChris Mason 		return ret;
3905925baeddSChris Mason 
3906a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3907168fd7d2SChris Mason 	/*
3908168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
3909168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
3910168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
3911168fd7d2SChris Mason 	 * advance the path if there are now more items available.
3912168fd7d2SChris Mason 	 */
3913a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
3914168fd7d2SChris Mason 		path->slots[0]++;
3915925baeddSChris Mason 		goto done;
3916925baeddSChris Mason 	}
3917d97e63b6SChris Mason 
3918234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3919d97e63b6SChris Mason 		if (!path->nodes[level])
39200f70abe2SChris Mason 			return 1;
39215f39d397SChris Mason 
3922d97e63b6SChris Mason 		slot = path->slots[level] + 1;
3923d97e63b6SChris Mason 		c = path->nodes[level];
39245f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3925d97e63b6SChris Mason 			level++;
3926925baeddSChris Mason 			if (level == BTRFS_MAX_LEVEL) {
39277bb86316SChris Mason 				return 1;
3928925baeddSChris Mason 			}
3929d97e63b6SChris Mason 			continue;
3930d97e63b6SChris Mason 		}
39315f39d397SChris Mason 
3932925baeddSChris Mason 		if (next) {
3933925baeddSChris Mason 			btrfs_tree_unlock(next);
39345f39d397SChris Mason 			free_extent_buffer(next);
3935925baeddSChris Mason 		}
39365f39d397SChris Mason 
39370bd40a71SChris Mason 		if (level == 1 && (path->locks[1] || path->skip_locking) &&
39380bd40a71SChris Mason 		    path->reada)
393901f46658SChris Mason 			reada_for_search(root, path, level, slot, 0);
39405f39d397SChris Mason 
3941ca7a79adSChris Mason 		next = read_node_slot(root, c, slot);
39425cd57b2cSChris Mason 		if (!path->skip_locking) {
3943051e1b9fSChris Mason 			WARN_ON(!btrfs_tree_locked(c));
3944925baeddSChris Mason 			btrfs_tree_lock(next);
39455cd57b2cSChris Mason 		}
3946d97e63b6SChris Mason 		break;
3947d97e63b6SChris Mason 	}
3948d97e63b6SChris Mason 	path->slots[level] = slot;
3949d97e63b6SChris Mason 	while(1) {
3950d97e63b6SChris Mason 		level--;
3951d97e63b6SChris Mason 		c = path->nodes[level];
3952925baeddSChris Mason 		if (path->locks[level])
3953925baeddSChris Mason 			btrfs_tree_unlock(c);
39545f39d397SChris Mason 		free_extent_buffer(c);
3955d97e63b6SChris Mason 		path->nodes[level] = next;
3956d97e63b6SChris Mason 		path->slots[level] = 0;
3957a74a4b97SChris Mason 		if (!path->skip_locking)
3958925baeddSChris Mason 			path->locks[level] = 1;
3959d97e63b6SChris Mason 		if (!level)
3960d97e63b6SChris Mason 			break;
3961925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
3962925baeddSChris Mason 			reada_for_search(root, path, level, slot, 0);
3963ca7a79adSChris Mason 		next = read_node_slot(root, next, 0);
39645cd57b2cSChris Mason 		if (!path->skip_locking) {
3965a2135011SChris Mason 			WARN_ON(!btrfs_tree_locked(path->nodes[level]));
3966925baeddSChris Mason 			btrfs_tree_lock(next);
3967d97e63b6SChris Mason 		}
39685cd57b2cSChris Mason 	}
3969925baeddSChris Mason done:
3970925baeddSChris Mason 	unlock_up(path, 0, 1);
3971d97e63b6SChris Mason 	return 0;
3972d97e63b6SChris Mason }
39730b86a832SChris Mason 
39743f157a2fSChris Mason /*
39753f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
39763f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
39773f157a2fSChris Mason  *
39783f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
39793f157a2fSChris Mason  */
39800b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
39810b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
39820b86a832SChris Mason 			int type)
39830b86a832SChris Mason {
39840b86a832SChris Mason 	struct btrfs_key found_key;
39850b86a832SChris Mason 	struct extent_buffer *leaf;
3986e02119d5SChris Mason 	u32 nritems;
39870b86a832SChris Mason 	int ret;
39880b86a832SChris Mason 
39890b86a832SChris Mason 	while(1) {
39900b86a832SChris Mason 		if (path->slots[0] == 0) {
39910b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
39920b86a832SChris Mason 			if (ret != 0)
39930b86a832SChris Mason 				return ret;
39940b86a832SChris Mason 		} else {
39950b86a832SChris Mason 			path->slots[0]--;
39960b86a832SChris Mason 		}
39970b86a832SChris Mason 		leaf = path->nodes[0];
3998e02119d5SChris Mason 		nritems = btrfs_header_nritems(leaf);
3999e02119d5SChris Mason 		if (nritems == 0)
4000e02119d5SChris Mason 			return 1;
4001e02119d5SChris Mason 		if (path->slots[0] == nritems)
4002e02119d5SChris Mason 			path->slots[0]--;
4003e02119d5SChris Mason 
40040b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
40050b86a832SChris Mason 		if (found_key.type == type)
40060b86a832SChris Mason 			return 0;
4007e02119d5SChris Mason 		if (found_key.objectid < min_objectid)
4008e02119d5SChris Mason 			break;
4009e02119d5SChris Mason 		if (found_key.objectid == min_objectid &&
4010e02119d5SChris Mason 		    found_key.type < type)
4011e02119d5SChris Mason 			break;
40120b86a832SChris Mason 	}
40130b86a832SChris Mason 	return 1;
40140b86a832SChris Mason }
4015