xref: /openbmc/linux/fs/btrfs/ctree.c (revision 9652480b)
16cbd5570SChris Mason /*
26cbd5570SChris Mason  * Copyright (C) 2007 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 
572c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
582c90e5d6SChris Mason {
59df24a2b9SChris Mason 	btrfs_release_path(NULL, p);
602c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
612c90e5d6SChris Mason }
622c90e5d6SChris Mason 
63234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
64eb60ceacSChris Mason {
65eb60ceacSChris Mason 	int i;
66a2135011SChris Mason 
67234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
683f157a2fSChris Mason 		p->slots[i] = 0;
69eb60ceacSChris Mason 		if (!p->nodes[i])
70925baeddSChris Mason 			continue;
71925baeddSChris Mason 		if (p->locks[i]) {
72925baeddSChris Mason 			btrfs_tree_unlock(p->nodes[i]);
73925baeddSChris Mason 			p->locks[i] = 0;
74925baeddSChris Mason 		}
755f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
763f157a2fSChris Mason 		p->nodes[i] = NULL;
77eb60ceacSChris Mason 	}
78eb60ceacSChris Mason }
79eb60ceacSChris Mason 
80925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
81925baeddSChris Mason {
82925baeddSChris Mason 	struct extent_buffer *eb;
83925baeddSChris Mason 	spin_lock(&root->node_lock);
84925baeddSChris Mason 	eb = root->node;
85925baeddSChris Mason 	extent_buffer_get(eb);
86925baeddSChris Mason 	spin_unlock(&root->node_lock);
87925baeddSChris Mason 	return eb;
88925baeddSChris Mason }
89925baeddSChris Mason 
90925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
91925baeddSChris Mason {
92925baeddSChris Mason 	struct extent_buffer *eb;
93925baeddSChris Mason 
94925baeddSChris Mason 	while(1) {
95925baeddSChris Mason 		eb = btrfs_root_node(root);
96925baeddSChris Mason 		btrfs_tree_lock(eb);
97925baeddSChris Mason 
98925baeddSChris Mason 		spin_lock(&root->node_lock);
99925baeddSChris Mason 		if (eb == root->node) {
100925baeddSChris Mason 			spin_unlock(&root->node_lock);
101925baeddSChris Mason 			break;
102925baeddSChris Mason 		}
103925baeddSChris Mason 		spin_unlock(&root->node_lock);
104925baeddSChris Mason 
105925baeddSChris Mason 		btrfs_tree_unlock(eb);
106925baeddSChris Mason 		free_extent_buffer(eb);
107925baeddSChris Mason 	}
108925baeddSChris Mason 	return eb;
109925baeddSChris Mason }
110925baeddSChris Mason 
1110b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root)
1120b86a832SChris Mason {
1130b86a832SChris Mason 	if (root->track_dirty && list_empty(&root->dirty_list)) {
1140b86a832SChris Mason 		list_add(&root->dirty_list,
1150b86a832SChris Mason 			 &root->fs_info->dirty_cowonly_roots);
1160b86a832SChris Mason 	}
1170b86a832SChris Mason }
1180b86a832SChris Mason 
119be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
120be20aa9dSChris Mason 		      struct btrfs_root *root,
121be20aa9dSChris Mason 		      struct extent_buffer *buf,
122be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
123be20aa9dSChris Mason {
124be20aa9dSChris Mason 	struct extent_buffer *cow;
125be20aa9dSChris Mason 	u32 nritems;
126be20aa9dSChris Mason 	int ret = 0;
127be20aa9dSChris Mason 	int level;
128be20aa9dSChris Mason 	struct btrfs_key first_key;
1294aec2b52SChris Mason 	struct btrfs_root *new_root;
130be20aa9dSChris Mason 
1314aec2b52SChris Mason 	new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
1324aec2b52SChris Mason 	if (!new_root)
1334aec2b52SChris Mason 		return -ENOMEM;
1344aec2b52SChris Mason 
1354aec2b52SChris Mason 	memcpy(new_root, root, sizeof(*new_root));
1364aec2b52SChris Mason 	new_root->root_key.objectid = new_root_objectid;
137be20aa9dSChris Mason 
138be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
139be20aa9dSChris Mason 		root->fs_info->running_transaction->transid);
140be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
141be20aa9dSChris Mason 
142be20aa9dSChris Mason 	level = btrfs_header_level(buf);
143be20aa9dSChris Mason 	nritems = btrfs_header_nritems(buf);
144be20aa9dSChris Mason 	if (nritems) {
145be20aa9dSChris Mason 		if (level == 0)
146be20aa9dSChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
147be20aa9dSChris Mason 		else
148be20aa9dSChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
149be20aa9dSChris Mason 	} else {
150be20aa9dSChris Mason 		first_key.objectid = 0;
151be20aa9dSChris Mason 	}
152925baeddSChris Mason 	cow = btrfs_alloc_free_block(trans, new_root, buf->len,
153be20aa9dSChris Mason 				       new_root_objectid,
154be20aa9dSChris Mason 				       trans->transid, first_key.objectid,
155be20aa9dSChris Mason 				       level, buf->start, 0);
1564aec2b52SChris Mason 	if (IS_ERR(cow)) {
1574aec2b52SChris Mason 		kfree(new_root);
158be20aa9dSChris Mason 		return PTR_ERR(cow);
1594aec2b52SChris Mason 	}
160be20aa9dSChris Mason 
161be20aa9dSChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
162be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
163be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
164be20aa9dSChris Mason 	btrfs_set_header_owner(cow, new_root_objectid);
16563b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
166be20aa9dSChris Mason 
167be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
1684aec2b52SChris Mason 	ret = btrfs_inc_ref(trans, new_root, buf);
1694aec2b52SChris Mason 	kfree(new_root);
1704aec2b52SChris Mason 
171be20aa9dSChris Mason 	if (ret)
172be20aa9dSChris Mason 		return ret;
173be20aa9dSChris Mason 
174be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
175be20aa9dSChris Mason 	*cow_ret = cow;
176be20aa9dSChris Mason 	return 0;
177be20aa9dSChris Mason }
178be20aa9dSChris Mason 
179be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans,
1805f39d397SChris Mason 			     struct btrfs_root *root,
1815f39d397SChris Mason 			     struct extent_buffer *buf,
1825f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
1835f39d397SChris Mason 			     struct extent_buffer **cow_ret,
1845f39d397SChris Mason 			     u64 search_start, u64 empty_size)
1856702ed49SChris Mason {
1867bb86316SChris Mason 	u64 root_gen;
1875f39d397SChris Mason 	struct extent_buffer *cow;
1887bb86316SChris Mason 	u32 nritems;
1896702ed49SChris Mason 	int ret = 0;
1906702ed49SChris Mason 	int different_trans = 0;
1917bb86316SChris Mason 	int level;
192925baeddSChris Mason 	int unlock_orig = 0;
1937bb86316SChris Mason 	struct btrfs_key first_key;
1946702ed49SChris Mason 
195925baeddSChris Mason 	if (*cow_ret == buf)
196925baeddSChris Mason 		unlock_orig = 1;
197925baeddSChris Mason 
198925baeddSChris Mason 	WARN_ON(!btrfs_tree_locked(buf));
199925baeddSChris Mason 
2007bb86316SChris Mason 	if (root->ref_cows) {
2017bb86316SChris Mason 		root_gen = trans->transid;
2027bb86316SChris Mason 	} else {
2037bb86316SChris Mason 		root_gen = 0;
2047bb86316SChris Mason 	}
2057bb86316SChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
2067bb86316SChris Mason 		root->fs_info->running_transaction->transid);
2076702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
2085f39d397SChris Mason 
2097bb86316SChris Mason 	level = btrfs_header_level(buf);
2107bb86316SChris Mason 	nritems = btrfs_header_nritems(buf);
2117bb86316SChris Mason 	if (nritems) {
2127bb86316SChris Mason 		if (level == 0)
2137bb86316SChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
2147bb86316SChris Mason 		else
2157bb86316SChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
2167bb86316SChris Mason 	} else {
2177bb86316SChris Mason 		first_key.objectid = 0;
2187bb86316SChris Mason 	}
219925baeddSChris Mason 	cow = btrfs_alloc_free_block(trans, root, buf->len,
2207bb86316SChris Mason 				     root->root_key.objectid,
2217bb86316SChris Mason 				     root_gen, first_key.objectid, level,
222db94535dSChris Mason 				     search_start, empty_size);
2236702ed49SChris Mason 	if (IS_ERR(cow))
2246702ed49SChris Mason 		return PTR_ERR(cow);
2256702ed49SChris Mason 
2265f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
227db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
2285f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
2295f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
23063b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
2316702ed49SChris Mason 
2325f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
2335f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
2346702ed49SChris Mason 		different_trans = 1;
2356702ed49SChris Mason 		ret = btrfs_inc_ref(trans, root, buf);
2366702ed49SChris Mason 		if (ret)
2376702ed49SChris Mason 			return ret;
2386702ed49SChris Mason 	} else {
2396702ed49SChris Mason 		clean_tree_block(trans, root, buf);
2406702ed49SChris Mason 	}
2416702ed49SChris Mason 
2426702ed49SChris Mason 	if (buf == root->node) {
243925baeddSChris Mason 		WARN_ON(parent && parent != buf);
2447bb86316SChris Mason 		root_gen = btrfs_header_generation(buf);
245925baeddSChris Mason 
246925baeddSChris Mason 		spin_lock(&root->node_lock);
2476702ed49SChris Mason 		root->node = cow;
2485f39d397SChris Mason 		extent_buffer_get(cow);
249925baeddSChris Mason 		spin_unlock(&root->node_lock);
250925baeddSChris Mason 
2516702ed49SChris Mason 		if (buf != root->commit_root) {
252db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
2537bb86316SChris Mason 					  buf->len, root->root_key.objectid,
2547bb86316SChris Mason 					  root_gen, 0, 0, 1);
2556702ed49SChris Mason 		}
2565f39d397SChris Mason 		free_extent_buffer(buf);
2570b86a832SChris Mason 		add_root_to_dirty_list(root);
2586702ed49SChris Mason 	} else {
2597bb86316SChris Mason 		root_gen = btrfs_header_generation(parent);
2605f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
261db94535dSChris Mason 					cow->start);
26274493f7aSChris Mason 		WARN_ON(trans->transid == 0);
26374493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
26474493f7aSChris Mason 					      trans->transid);
2656702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
2665f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
2677bb86316SChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len,
2687bb86316SChris Mason 				  btrfs_header_owner(parent), root_gen,
2697bb86316SChris Mason 				  0, 0, 1);
2706702ed49SChris Mason 	}
271925baeddSChris Mason 	if (unlock_orig)
272925baeddSChris Mason 		btrfs_tree_unlock(buf);
2735f39d397SChris Mason 	free_extent_buffer(buf);
2746702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
2756702ed49SChris Mason 	*cow_ret = cow;
2766702ed49SChris Mason 	return 0;
2776702ed49SChris Mason }
2786702ed49SChris Mason 
2795f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans,
2805f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
2815f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
2825f39d397SChris Mason 		    struct extent_buffer **cow_ret)
28302217ed2SChris Mason {
2846702ed49SChris Mason 	u64 search_start;
285dc17ff8fSChris Mason 	u64 header_trans;
286f510cfecSChris Mason 	int ret;
287dc17ff8fSChris Mason 
288ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
289ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
290ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
291ccd467d6SChris Mason 		WARN_ON(1);
292ccd467d6SChris Mason 	}
293ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
294ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
295ccd467d6SChris Mason 		       root->fs_info->generation);
296ccd467d6SChris Mason 		WARN_ON(1);
297ccd467d6SChris Mason 	}
298dc17ff8fSChris Mason 
299dc17ff8fSChris Mason 	header_trans = btrfs_header_generation(buf);
30063b10fc4SChris Mason 	spin_lock(&root->fs_info->hash_lock);
30163b10fc4SChris Mason 	if (header_trans == trans->transid &&
30263b10fc4SChris Mason 	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) {
30302217ed2SChris Mason 		*cow_ret = buf;
30463b10fc4SChris Mason 		spin_unlock(&root->fs_info->hash_lock);
30502217ed2SChris Mason 		return 0;
30602217ed2SChris Mason 	}
30763b10fc4SChris Mason 	spin_unlock(&root->fs_info->hash_lock);
3080b86a832SChris Mason 	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
309f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
3106702ed49SChris Mason 				 parent_slot, cow_ret, search_start, 0);
311f510cfecSChris Mason 	return ret;
3122c90e5d6SChris Mason }
3136702ed49SChris Mason 
3146b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
3156702ed49SChris Mason {
3166b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
3176702ed49SChris Mason 		return 1;
3186b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
3196702ed49SChris Mason 		return 1;
32002217ed2SChris Mason 	return 0;
32102217ed2SChris Mason }
32202217ed2SChris Mason 
323081e9573SChris Mason /*
324081e9573SChris Mason  * compare two keys in a memcmp fashion
325081e9573SChris Mason  */
326081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
327081e9573SChris Mason {
328081e9573SChris Mason 	struct btrfs_key k1;
329081e9573SChris Mason 
330081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
331081e9573SChris Mason 
332081e9573SChris Mason 	if (k1.objectid > k2->objectid)
333081e9573SChris Mason 		return 1;
334081e9573SChris Mason 	if (k1.objectid < k2->objectid)
335081e9573SChris Mason 		return -1;
336081e9573SChris Mason 	if (k1.type > k2->type)
337081e9573SChris Mason 		return 1;
338081e9573SChris Mason 	if (k1.type < k2->type)
339081e9573SChris Mason 		return -1;
340081e9573SChris Mason 	if (k1.offset > k2->offset)
341081e9573SChris Mason 		return 1;
342081e9573SChris Mason 	if (k1.offset < k2->offset)
343081e9573SChris Mason 		return -1;
344081e9573SChris Mason 	return 0;
345081e9573SChris Mason }
346081e9573SChris Mason 
347081e9573SChris Mason 
3486702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
3495f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
350a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
351a6b6e75eSChris Mason 		       struct btrfs_key *progress)
3526702ed49SChris Mason {
3536b80053dSChris Mason 	struct extent_buffer *cur;
3546702ed49SChris Mason 	u64 blocknr;
355ca7a79adSChris Mason 	u64 gen;
356e9d0b13bSChris Mason 	u64 search_start = *last_ret;
357e9d0b13bSChris Mason 	u64 last_block = 0;
3586702ed49SChris Mason 	u64 other;
3596702ed49SChris Mason 	u32 parent_nritems;
3606702ed49SChris Mason 	int end_slot;
3616702ed49SChris Mason 	int i;
3626702ed49SChris Mason 	int err = 0;
363f2183bdeSChris Mason 	int parent_level;
3646b80053dSChris Mason 	int uptodate;
3656b80053dSChris Mason 	u32 blocksize;
366081e9573SChris Mason 	int progress_passed = 0;
367081e9573SChris Mason 	struct btrfs_disk_key disk_key;
3686702ed49SChris Mason 
3695708b959SChris Mason 	parent_level = btrfs_header_level(parent);
3705708b959SChris Mason 	if (cache_only && parent_level != 1)
3715708b959SChris Mason 		return 0;
3725708b959SChris Mason 
3736702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
3746702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3756702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
3766702ed49SChris Mason 		WARN_ON(1);
3776702ed49SChris Mason 	}
3786702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
3796702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3806702ed49SChris Mason 		       root->fs_info->generation);
3816702ed49SChris Mason 		WARN_ON(1);
3826702ed49SChris Mason 	}
38386479a04SChris Mason 
3846b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
3856b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
3866702ed49SChris Mason 	end_slot = parent_nritems;
3876702ed49SChris Mason 
3886702ed49SChris Mason 	if (parent_nritems == 1)
3896702ed49SChris Mason 		return 0;
3906702ed49SChris Mason 
3916702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
3926702ed49SChris Mason 		int close = 1;
393a6b6e75eSChris Mason 
3945708b959SChris Mason 		if (!parent->map_token) {
3955708b959SChris Mason 			map_extent_buffer(parent,
3965708b959SChris Mason 					btrfs_node_key_ptr_offset(i),
3975708b959SChris Mason 					sizeof(struct btrfs_key_ptr),
3985708b959SChris Mason 					&parent->map_token, &parent->kaddr,
3995708b959SChris Mason 					&parent->map_start, &parent->map_len,
4005708b959SChris Mason 					KM_USER1);
4015708b959SChris Mason 		}
402081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
403081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
404081e9573SChris Mason 			continue;
405081e9573SChris Mason 
406081e9573SChris Mason 		progress_passed = 1;
4076b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
408ca7a79adSChris Mason 		gen = btrfs_node_ptr_generation(parent, i);
409e9d0b13bSChris Mason 		if (last_block == 0)
410e9d0b13bSChris Mason 			last_block = blocknr;
4115708b959SChris Mason 
4126702ed49SChris Mason 		if (i > 0) {
4136b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
4146b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
4156702ed49SChris Mason 		}
4160ef3e66bSChris Mason 		if (!close && i < end_slot - 2) {
4176b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
4186b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
4196702ed49SChris Mason 		}
420e9d0b13bSChris Mason 		if (close) {
421e9d0b13bSChris Mason 			last_block = blocknr;
4226702ed49SChris Mason 			continue;
423e9d0b13bSChris Mason 		}
4245708b959SChris Mason 		if (parent->map_token) {
4255708b959SChris Mason 			unmap_extent_buffer(parent, parent->map_token,
4265708b959SChris Mason 					    KM_USER1);
4275708b959SChris Mason 			parent->map_token = NULL;
4285708b959SChris Mason 		}
4296702ed49SChris Mason 
4306b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
4316b80053dSChris Mason 		if (cur)
4321259ab75SChris Mason 			uptodate = btrfs_buffer_uptodate(cur, gen);
4336b80053dSChris Mason 		else
4346b80053dSChris Mason 			uptodate = 0;
4355708b959SChris Mason 		if (!cur || !uptodate) {
4366702ed49SChris Mason 			if (cache_only) {
4376b80053dSChris Mason 				free_extent_buffer(cur);
4386702ed49SChris Mason 				continue;
4396702ed49SChris Mason 			}
4406b80053dSChris Mason 			if (!cur) {
4416b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
442ca7a79adSChris Mason 							 blocksize, gen);
4436b80053dSChris Mason 			} else if (!uptodate) {
444ca7a79adSChris Mason 				btrfs_read_buffer(cur, gen);
4456702ed49SChris Mason 			}
446f2183bdeSChris Mason 		}
447e9d0b13bSChris Mason 		if (search_start == 0)
4486b80053dSChris Mason 			search_start = last_block;
449e9d0b13bSChris Mason 
450e7a84565SChris Mason 		btrfs_tree_lock(cur);
4516b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
452e7a84565SChris Mason 					&cur, search_start,
4536b80053dSChris Mason 					min(16 * blocksize,
4546b80053dSChris Mason 					    (end_slot - i) * blocksize));
455252c38f0SYan 		if (err) {
456e7a84565SChris Mason 			btrfs_tree_unlock(cur);
4576b80053dSChris Mason 			free_extent_buffer(cur);
4586702ed49SChris Mason 			break;
459252c38f0SYan 		}
460e7a84565SChris Mason 		search_start = cur->start;
461e7a84565SChris Mason 		last_block = cur->start;
462f2183bdeSChris Mason 		*last_ret = search_start;
463e7a84565SChris Mason 		btrfs_tree_unlock(cur);
464e7a84565SChris Mason 		free_extent_buffer(cur);
4656702ed49SChris Mason 	}
4665708b959SChris Mason 	if (parent->map_token) {
4675708b959SChris Mason 		unmap_extent_buffer(parent, parent->map_token,
4685708b959SChris Mason 				    KM_USER1);
4695708b959SChris Mason 		parent->map_token = NULL;
4705708b959SChris Mason 	}
4716702ed49SChris Mason 	return err;
4726702ed49SChris Mason }
4736702ed49SChris Mason 
47474123bd7SChris Mason /*
47574123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
47674123bd7SChris Mason  * this returns the address of the start of the last item,
47774123bd7SChris Mason  * which is the stop of the leaf data stack
47874123bd7SChris Mason  */
479123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
4805f39d397SChris Mason 					 struct extent_buffer *leaf)
481be0e5c09SChris Mason {
4825f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
483be0e5c09SChris Mason 	if (nr == 0)
484123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
4855f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
486be0e5c09SChris Mason }
487be0e5c09SChris Mason 
488123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
489123abc88SChris Mason 		      int level)
490aa5d6bedSChris Mason {
4915f39d397SChris Mason 	struct extent_buffer *parent = NULL;
4925f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
4935f39d397SChris Mason 	struct btrfs_disk_key parent_key;
4945f39d397SChris Mason 	struct btrfs_disk_key node_key;
495aa5d6bedSChris Mason 	int parent_slot;
4968d7be552SChris Mason 	int slot;
4978d7be552SChris Mason 	struct btrfs_key cpukey;
4985f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
499aa5d6bedSChris Mason 
500aa5d6bedSChris Mason 	if (path->nodes[level + 1])
5015f39d397SChris Mason 		parent = path->nodes[level + 1];
502a1f39630SAneesh 
5038d7be552SChris Mason 	slot = path->slots[level];
5047518a238SChris Mason 	BUG_ON(nritems == 0);
5057518a238SChris Mason 	if (parent) {
506a1f39630SAneesh 		parent_slot = path->slots[level + 1];
5075f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
5085f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
5095f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
510e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
5111d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
512db94535dSChris Mason 		       btrfs_header_bytenr(node));
513aa5d6bedSChris Mason 	}
514123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
5158d7be552SChris Mason 	if (slot != 0) {
5165f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
5175f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
5185f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
5198d7be552SChris Mason 	}
5208d7be552SChris Mason 	if (slot < nritems - 1) {
5215f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
5225f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
5235f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
524aa5d6bedSChris Mason 	}
525aa5d6bedSChris Mason 	return 0;
526aa5d6bedSChris Mason }
527aa5d6bedSChris Mason 
528123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
529123abc88SChris Mason 		      int level)
530aa5d6bedSChris Mason {
5315f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
5325f39d397SChris Mason 	struct extent_buffer *parent = NULL;
533aa5d6bedSChris Mason 	int parent_slot;
5348d7be552SChris Mason 	struct btrfs_key cpukey;
5355f39d397SChris Mason 	struct btrfs_disk_key parent_key;
5365f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
5375f39d397SChris Mason 	int slot = path->slots[0];
5388d7be552SChris Mason 
5395f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
540aa5d6bedSChris Mason 
541aa5d6bedSChris Mason 	if (path->nodes[level + 1])
5425f39d397SChris Mason 		parent = path->nodes[level + 1];
5437518a238SChris Mason 
5447518a238SChris Mason 	if (nritems == 0)
5457518a238SChris Mason 		return 0;
5467518a238SChris Mason 
5477518a238SChris Mason 	if (parent) {
548a1f39630SAneesh 		parent_slot = path->slots[level + 1];
5495f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
5505f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
5516702ed49SChris Mason 
5525f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
553e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
5541d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
555db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
556aa5d6bedSChris Mason 	}
5575f39d397SChris Mason #if 0
5585f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
5595f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
5605f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
5615f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
5625f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5635f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
5645f39d397SChris Mason 			BUG_ON(1);
5655f39d397SChris Mason 		}
5665f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
5675f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
5685f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5695f39d397SChris Mason 			printk("slot %d offset bad\n", i);
5705f39d397SChris Mason 			BUG_ON(1);
5715f39d397SChris Mason 		}
5725f39d397SChris Mason 		if (i == 0) {
5735f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
5745f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
5755f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
5765f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5775f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
5785f39d397SChris Mason 				BUG_ON(1);
5795f39d397SChris Mason 			}
5805f39d397SChris Mason 		}
5815f39d397SChris Mason 	}
5825f39d397SChris Mason 	if (nritems > 0) {
5835f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
5845f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5855f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
5865f39d397SChris Mason 				BUG_ON(1);
5875f39d397SChris Mason 		}
5885f39d397SChris Mason 	}
5895f39d397SChris Mason #endif
5905f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
5915f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
5925f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
5935f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
5945f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5955f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
5965f39d397SChris Mason 			BUG_ON(1);
5975f39d397SChris Mason 		}
5985f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
5995f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
6005f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6015f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
6025f39d397SChris Mason 			BUG_ON(1);
6035f39d397SChris Mason 		}
604aa5d6bedSChris Mason 	}
6058d7be552SChris Mason 	if (slot < nritems - 1) {
6065f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
6075f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
6085f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
6095f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
6105f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
6115f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6125f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
6135f39d397SChris Mason 			BUG_ON(1);
614aa5d6bedSChris Mason 		}
6155f39d397SChris Mason 	}
6165f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
6175f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
618aa5d6bedSChris Mason 	return 0;
619aa5d6bedSChris Mason }
620aa5d6bedSChris Mason 
62198ed5174SChris Mason static int noinline check_block(struct btrfs_root *root,
62298ed5174SChris Mason 				struct btrfs_path *path, int level)
623aa5d6bedSChris Mason {
624f188591eSChris Mason 	u64 found_start;
62585d824c4SChris Mason 	return 0;
626f188591eSChris Mason 	if (btrfs_header_level(path->nodes[level]) != level)
627f188591eSChris Mason 	    printk("warning: bad level %Lu wanted %d found %d\n",
628f188591eSChris Mason 		   path->nodes[level]->start, level,
629f188591eSChris Mason 		   btrfs_header_level(path->nodes[level]));
630f188591eSChris Mason 	found_start = btrfs_header_bytenr(path->nodes[level]);
631f188591eSChris Mason 	if (found_start != path->nodes[level]->start) {
632f188591eSChris Mason 	    printk("warning: bad bytentr %Lu found %Lu\n",
633f188591eSChris Mason 		   path->nodes[level]->start, found_start);
634f188591eSChris Mason 	}
635db94535dSChris Mason #if 0
6365f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
6375f39d397SChris Mason 
638479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
639479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
640479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
6415f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
642db94535dSChris Mason 		return 1;
6435f39d397SChris Mason 	}
644db94535dSChris Mason #endif
645aa5d6bedSChris Mason 	if (level == 0)
646123abc88SChris Mason 		return check_leaf(root, path, level);
647123abc88SChris Mason 	return check_node(root, path, level);
648aa5d6bedSChris Mason }
649aa5d6bedSChris Mason 
65074123bd7SChris Mason /*
6515f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
6525f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
6535f39d397SChris Mason  *
65474123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
65574123bd7SChris Mason  * the place where you would insert key if it is not found in
65674123bd7SChris Mason  * the array.
65774123bd7SChris Mason  *
65874123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
65974123bd7SChris Mason  */
6605f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
6615f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
662be0e5c09SChris Mason 			      int max, int *slot)
663be0e5c09SChris Mason {
664be0e5c09SChris Mason 	int low = 0;
665be0e5c09SChris Mason 	int high = max;
666be0e5c09SChris Mason 	int mid;
667be0e5c09SChris Mason 	int ret;
668479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
6695f39d397SChris Mason 	struct btrfs_disk_key unaligned;
6705f39d397SChris Mason 	unsigned long offset;
6715f39d397SChris Mason 	char *map_token = NULL;
6725f39d397SChris Mason 	char *kaddr = NULL;
6735f39d397SChris Mason 	unsigned long map_start = 0;
6745f39d397SChris Mason 	unsigned long map_len = 0;
675479965d6SChris Mason 	int err;
676be0e5c09SChris Mason 
677be0e5c09SChris Mason 	while(low < high) {
678be0e5c09SChris Mason 		mid = (low + high) / 2;
6795f39d397SChris Mason 		offset = p + mid * item_size;
6805f39d397SChris Mason 
6815f39d397SChris Mason 		if (!map_token || offset < map_start ||
6825f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
6835f39d397SChris Mason 		    map_start + map_len) {
684479965d6SChris Mason 			if (map_token) {
6855f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
686479965d6SChris Mason 				map_token = NULL;
687479965d6SChris Mason 			}
688479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
689479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
690479965d6SChris Mason 						&map_token, &kaddr,
6915f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
6925f39d397SChris Mason 
693479965d6SChris Mason 			if (!err) {
694479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
695479965d6SChris Mason 							map_start);
696479965d6SChris Mason 			} else {
6975f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
6985f39d397SChris Mason 						   offset, sizeof(unaligned));
6995f39d397SChris Mason 				tmp = &unaligned;
700479965d6SChris Mason 			}
701479965d6SChris Mason 
7025f39d397SChris Mason 		} else {
7035f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
7045f39d397SChris Mason 							map_start);
7055f39d397SChris Mason 		}
706be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
707be0e5c09SChris Mason 
708be0e5c09SChris Mason 		if (ret < 0)
709be0e5c09SChris Mason 			low = mid + 1;
710be0e5c09SChris Mason 		else if (ret > 0)
711be0e5c09SChris Mason 			high = mid;
712be0e5c09SChris Mason 		else {
713be0e5c09SChris Mason 			*slot = mid;
714479965d6SChris Mason 			if (map_token)
7155f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
716be0e5c09SChris Mason 			return 0;
717be0e5c09SChris Mason 		}
718be0e5c09SChris Mason 	}
719be0e5c09SChris Mason 	*slot = low;
7205f39d397SChris Mason 	if (map_token)
7215f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
722be0e5c09SChris Mason 	return 1;
723be0e5c09SChris Mason }
724be0e5c09SChris Mason 
72597571fd0SChris Mason /*
72697571fd0SChris Mason  * simple bin_search frontend that does the right thing for
72797571fd0SChris Mason  * leaves vs nodes
72897571fd0SChris Mason  */
7295f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
7305f39d397SChris Mason 		      int level, int *slot)
731be0e5c09SChris Mason {
7325f39d397SChris Mason 	if (level == 0) {
7335f39d397SChris Mason 		return generic_bin_search(eb,
7345f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
7350783fcfcSChris Mason 					  sizeof(struct btrfs_item),
7365f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
7377518a238SChris Mason 					  slot);
738be0e5c09SChris Mason 	} else {
7395f39d397SChris Mason 		return generic_bin_search(eb,
7405f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
741123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
7425f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
7437518a238SChris Mason 					  slot);
744be0e5c09SChris Mason 	}
745be0e5c09SChris Mason 	return -1;
746be0e5c09SChris Mason }
747be0e5c09SChris Mason 
7485f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
7495f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
750bb803951SChris Mason {
751ca7a79adSChris Mason 	int level = btrfs_header_level(parent);
752bb803951SChris Mason 	if (slot < 0)
753bb803951SChris Mason 		return NULL;
7545f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
755bb803951SChris Mason 		return NULL;
756ca7a79adSChris Mason 
757ca7a79adSChris Mason 	BUG_ON(level == 0);
758ca7a79adSChris Mason 
759db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
760ca7a79adSChris Mason 		       btrfs_level_size(root, level - 1),
761ca7a79adSChris Mason 		       btrfs_node_ptr_generation(parent, slot));
762bb803951SChris Mason }
763bb803951SChris Mason 
76498ed5174SChris Mason static int balance_level(struct btrfs_trans_handle *trans,
76598ed5174SChris Mason 			 struct btrfs_root *root,
76698ed5174SChris Mason 			 struct btrfs_path *path, int level)
767bb803951SChris Mason {
7685f39d397SChris Mason 	struct extent_buffer *right = NULL;
7695f39d397SChris Mason 	struct extent_buffer *mid;
7705f39d397SChris Mason 	struct extent_buffer *left = NULL;
7715f39d397SChris Mason 	struct extent_buffer *parent = NULL;
772bb803951SChris Mason 	int ret = 0;
773bb803951SChris Mason 	int wret;
774bb803951SChris Mason 	int pslot;
775bb803951SChris Mason 	int orig_slot = path->slots[level];
77654aa1f4dSChris Mason 	int err_on_enospc = 0;
77779f95c82SChris Mason 	u64 orig_ptr;
778bb803951SChris Mason 
779bb803951SChris Mason 	if (level == 0)
780bb803951SChris Mason 		return 0;
781bb803951SChris Mason 
7825f39d397SChris Mason 	mid = path->nodes[level];
783925baeddSChris Mason 	WARN_ON(!path->locks[level]);
7847bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
7857bb86316SChris Mason 
7861d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
78779f95c82SChris Mason 
788234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
7895f39d397SChris Mason 		parent = path->nodes[level + 1];
790bb803951SChris Mason 	pslot = path->slots[level + 1];
791bb803951SChris Mason 
79240689478SChris Mason 	/*
79340689478SChris Mason 	 * deal with the case where there is only one pointer in the root
79440689478SChris Mason 	 * by promoting the node below to a root
79540689478SChris Mason 	 */
7965f39d397SChris Mason 	if (!parent) {
7975f39d397SChris Mason 		struct extent_buffer *child;
798bb803951SChris Mason 
7995f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
800bb803951SChris Mason 			return 0;
801bb803951SChris Mason 
802bb803951SChris Mason 		/* promote the child to a root */
8035f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
804925baeddSChris Mason 		btrfs_tree_lock(child);
805bb803951SChris Mason 		BUG_ON(!child);
8062f375ab9SYan 		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
8072f375ab9SYan 		BUG_ON(ret);
8082f375ab9SYan 
809925baeddSChris Mason 		spin_lock(&root->node_lock);
810bb803951SChris Mason 		root->node = child;
811925baeddSChris Mason 		spin_unlock(&root->node_lock);
812925baeddSChris Mason 
8130b86a832SChris Mason 		add_root_to_dirty_list(root);
814925baeddSChris Mason 		btrfs_tree_unlock(child);
815925baeddSChris Mason 		path->locks[level] = 0;
816bb803951SChris Mason 		path->nodes[level] = NULL;
8175f39d397SChris Mason 		clean_tree_block(trans, root, mid);
818925baeddSChris Mason 		btrfs_tree_unlock(mid);
819bb803951SChris Mason 		/* once for the path */
8205f39d397SChris Mason 		free_extent_buffer(mid);
8217bb86316SChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
8227bb86316SChris Mason 					root->root_key.objectid,
8237bb86316SChris Mason 					btrfs_header_generation(mid), 0, 0, 1);
824bb803951SChris Mason 		/* once for the root ptr */
8255f39d397SChris Mason 		free_extent_buffer(mid);
826db94535dSChris Mason 		return ret;
827bb803951SChris Mason 	}
8285f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
829123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
830bb803951SChris Mason 		return 0;
831bb803951SChris Mason 
8325f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
83354aa1f4dSChris Mason 		err_on_enospc = 1;
83454aa1f4dSChris Mason 
8355f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
8365f39d397SChris Mason 	if (left) {
837925baeddSChris Mason 		btrfs_tree_lock(left);
8385f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
8395f39d397SChris Mason 				       parent, pslot - 1, &left);
84054aa1f4dSChris Mason 		if (wret) {
84154aa1f4dSChris Mason 			ret = wret;
84254aa1f4dSChris Mason 			goto enospc;
84354aa1f4dSChris Mason 		}
8442cc58cf2SChris Mason 	}
8455f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
8465f39d397SChris Mason 	if (right) {
847925baeddSChris Mason 		btrfs_tree_lock(right);
8485f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
8495f39d397SChris Mason 				       parent, pslot + 1, &right);
8502cc58cf2SChris Mason 		if (wret) {
8512cc58cf2SChris Mason 			ret = wret;
8522cc58cf2SChris Mason 			goto enospc;
8532cc58cf2SChris Mason 		}
8542cc58cf2SChris Mason 	}
8552cc58cf2SChris Mason 
8562cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
8575f39d397SChris Mason 	if (left) {
8585f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
859bce4eae9SChris Mason 		wret = push_node_left(trans, root, left, mid, 1);
86079f95c82SChris Mason 		if (wret < 0)
86179f95c82SChris Mason 			ret = wret;
8625f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
86354aa1f4dSChris Mason 			err_on_enospc = 1;
864bb803951SChris Mason 	}
86579f95c82SChris Mason 
86679f95c82SChris Mason 	/*
86779f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
86879f95c82SChris Mason 	 */
8695f39d397SChris Mason 	if (right) {
870971a1f66SChris Mason 		wret = push_node_left(trans, root, mid, right, 1);
87154aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
87279f95c82SChris Mason 			ret = wret;
8735f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
874db94535dSChris Mason 			u64 bytenr = right->start;
8757bb86316SChris Mason 			u64 generation = btrfs_header_generation(parent);
876db94535dSChris Mason 			u32 blocksize = right->len;
877db94535dSChris Mason 
8785f39d397SChris Mason 			clean_tree_block(trans, root, right);
879925baeddSChris Mason 			btrfs_tree_unlock(right);
8805f39d397SChris Mason 			free_extent_buffer(right);
881bb803951SChris Mason 			right = NULL;
882e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
883e089f05cSChris Mason 				       1);
884bb803951SChris Mason 			if (wret)
885bb803951SChris Mason 				ret = wret;
886db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
8877bb86316SChris Mason 						 blocksize,
8887bb86316SChris Mason 						 btrfs_header_owner(parent),
8897bb86316SChris Mason 						 generation, 0, 0, 1);
890bb803951SChris Mason 			if (wret)
891bb803951SChris Mason 				ret = wret;
892bb803951SChris Mason 		} else {
8935f39d397SChris Mason 			struct btrfs_disk_key right_key;
8945f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
8955f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
8965f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
897bb803951SChris Mason 		}
898bb803951SChris Mason 	}
8995f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
90079f95c82SChris Mason 		/*
90179f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
90279f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
90379f95c82SChris Mason 		 * could try to delete the only pointer in this node.
90479f95c82SChris Mason 		 * So, pull some keys from the left.
90579f95c82SChris Mason 		 * There has to be a left pointer at this point because
90679f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
90779f95c82SChris Mason 		 * right
90879f95c82SChris Mason 		 */
9095f39d397SChris Mason 		BUG_ON(!left);
9105f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
91154aa1f4dSChris Mason 		if (wret < 0) {
91279f95c82SChris Mason 			ret = wret;
91354aa1f4dSChris Mason 			goto enospc;
91454aa1f4dSChris Mason 		}
915bce4eae9SChris Mason 		if (wret == 1) {
916bce4eae9SChris Mason 			wret = push_node_left(trans, root, left, mid, 1);
917bce4eae9SChris Mason 			if (wret < 0)
918bce4eae9SChris Mason 				ret = wret;
919bce4eae9SChris Mason 		}
92079f95c82SChris Mason 		BUG_ON(wret == 1);
92179f95c82SChris Mason 	}
9225f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
92379f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
9247bb86316SChris Mason 		u64 root_gen = btrfs_header_generation(parent);
925db94535dSChris Mason 		u64 bytenr = mid->start;
926db94535dSChris Mason 		u32 blocksize = mid->len;
927925baeddSChris Mason 
9285f39d397SChris Mason 		clean_tree_block(trans, root, mid);
929925baeddSChris Mason 		btrfs_tree_unlock(mid);
9305f39d397SChris Mason 		free_extent_buffer(mid);
931bb803951SChris Mason 		mid = NULL;
932e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
933bb803951SChris Mason 		if (wret)
934bb803951SChris Mason 			ret = wret;
9357bb86316SChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
9367bb86316SChris Mason 					 btrfs_header_owner(parent),
9377bb86316SChris Mason 					 root_gen, 0, 0, 1);
938bb803951SChris Mason 		if (wret)
939bb803951SChris Mason 			ret = wret;
94079f95c82SChris Mason 	} else {
94179f95c82SChris Mason 		/* update the parent key to reflect our changes */
9425f39d397SChris Mason 		struct btrfs_disk_key mid_key;
9435f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
9445f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
9455f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
94679f95c82SChris Mason 	}
947bb803951SChris Mason 
94879f95c82SChris Mason 	/* update the path */
9495f39d397SChris Mason 	if (left) {
9505f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
9515f39d397SChris Mason 			extent_buffer_get(left);
952925baeddSChris Mason 			/* left was locked after cow */
9535f39d397SChris Mason 			path->nodes[level] = left;
954bb803951SChris Mason 			path->slots[level + 1] -= 1;
955bb803951SChris Mason 			path->slots[level] = orig_slot;
956925baeddSChris Mason 			if (mid) {
957925baeddSChris Mason 				btrfs_tree_unlock(mid);
9585f39d397SChris Mason 				free_extent_buffer(mid);
959925baeddSChris Mason 			}
960bb803951SChris Mason 		} else {
9615f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
962bb803951SChris Mason 			path->slots[level] = orig_slot;
963bb803951SChris Mason 		}
964bb803951SChris Mason 	}
96579f95c82SChris Mason 	/* double check we haven't messed things up */
966123abc88SChris Mason 	check_block(root, path, level);
967e20d96d6SChris Mason 	if (orig_ptr !=
9685f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
96979f95c82SChris Mason 		BUG();
97054aa1f4dSChris Mason enospc:
971925baeddSChris Mason 	if (right) {
972925baeddSChris Mason 		btrfs_tree_unlock(right);
9735f39d397SChris Mason 		free_extent_buffer(right);
974925baeddSChris Mason 	}
975925baeddSChris Mason 	if (left) {
976925baeddSChris Mason 		if (path->nodes[level] != left)
977925baeddSChris Mason 			btrfs_tree_unlock(left);
9785f39d397SChris Mason 		free_extent_buffer(left);
979925baeddSChris Mason 	}
980bb803951SChris Mason 	return ret;
981bb803951SChris Mason }
982bb803951SChris Mason 
983e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
98498ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
985e66f709bSChris Mason 					  struct btrfs_root *root,
986e66f709bSChris Mason 					  struct btrfs_path *path, int level)
987e66f709bSChris Mason {
9885f39d397SChris Mason 	struct extent_buffer *right = NULL;
9895f39d397SChris Mason 	struct extent_buffer *mid;
9905f39d397SChris Mason 	struct extent_buffer *left = NULL;
9915f39d397SChris Mason 	struct extent_buffer *parent = NULL;
992e66f709bSChris Mason 	int ret = 0;
993e66f709bSChris Mason 	int wret;
994e66f709bSChris Mason 	int pslot;
995e66f709bSChris Mason 	int orig_slot = path->slots[level];
996e66f709bSChris Mason 	u64 orig_ptr;
997e66f709bSChris Mason 
998e66f709bSChris Mason 	if (level == 0)
999e66f709bSChris Mason 		return 1;
1000e66f709bSChris Mason 
10015f39d397SChris Mason 	mid = path->nodes[level];
10027bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1003e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
1004e66f709bSChris Mason 
1005e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
10065f39d397SChris Mason 		parent = path->nodes[level + 1];
1007e66f709bSChris Mason 	pslot = path->slots[level + 1];
1008e66f709bSChris Mason 
10095f39d397SChris Mason 	if (!parent)
1010e66f709bSChris Mason 		return 1;
1011e66f709bSChris Mason 
10125f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
1013e66f709bSChris Mason 
1014e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
10155f39d397SChris Mason 	if (left) {
1016e66f709bSChris Mason 		u32 left_nr;
1017925baeddSChris Mason 
1018925baeddSChris Mason 		btrfs_tree_lock(left);
10195f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
102033ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
102133ade1f8SChris Mason 			wret = 1;
102233ade1f8SChris Mason 		} else {
10235f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
10245f39d397SChris Mason 					      pslot - 1, &left);
102554aa1f4dSChris Mason 			if (ret)
102654aa1f4dSChris Mason 				wret = 1;
102754aa1f4dSChris Mason 			else {
102854aa1f4dSChris Mason 				wret = push_node_left(trans, root,
1029971a1f66SChris Mason 						      left, mid, 0);
103054aa1f4dSChris Mason 			}
103133ade1f8SChris Mason 		}
1032e66f709bSChris Mason 		if (wret < 0)
1033e66f709bSChris Mason 			ret = wret;
1034e66f709bSChris Mason 		if (wret == 0) {
10355f39d397SChris Mason 			struct btrfs_disk_key disk_key;
1036e66f709bSChris Mason 			orig_slot += left_nr;
10375f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
10385f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
10395f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
10405f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
10415f39d397SChris Mason 				path->nodes[level] = left;
1042e66f709bSChris Mason 				path->slots[level + 1] -= 1;
1043e66f709bSChris Mason 				path->slots[level] = orig_slot;
1044925baeddSChris Mason 				btrfs_tree_unlock(mid);
10455f39d397SChris Mason 				free_extent_buffer(mid);
1046e66f709bSChris Mason 			} else {
1047e66f709bSChris Mason 				orig_slot -=
10485f39d397SChris Mason 					btrfs_header_nritems(left);
1049e66f709bSChris Mason 				path->slots[level] = orig_slot;
1050925baeddSChris Mason 				btrfs_tree_unlock(left);
10515f39d397SChris Mason 				free_extent_buffer(left);
1052e66f709bSChris Mason 			}
1053e66f709bSChris Mason 			return 0;
1054e66f709bSChris Mason 		}
1055925baeddSChris Mason 		btrfs_tree_unlock(left);
10565f39d397SChris Mason 		free_extent_buffer(left);
1057e66f709bSChris Mason 	}
10585f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
1059e66f709bSChris Mason 
1060e66f709bSChris Mason 	/*
1061e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
1062e66f709bSChris Mason 	 */
10635f39d397SChris Mason 	if (right) {
106433ade1f8SChris Mason 		u32 right_nr;
1065925baeddSChris Mason 		btrfs_tree_lock(right);
10665f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
106733ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
106833ade1f8SChris Mason 			wret = 1;
106933ade1f8SChris Mason 		} else {
10705f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
10715f39d397SChris Mason 					      parent, pslot + 1,
10725f39d397SChris Mason 					      &right);
107354aa1f4dSChris Mason 			if (ret)
107454aa1f4dSChris Mason 				wret = 1;
107554aa1f4dSChris Mason 			else {
107633ade1f8SChris Mason 				wret = balance_node_right(trans, root,
10775f39d397SChris Mason 							  right, mid);
107833ade1f8SChris Mason 			}
107954aa1f4dSChris Mason 		}
1080e66f709bSChris Mason 		if (wret < 0)
1081e66f709bSChris Mason 			ret = wret;
1082e66f709bSChris Mason 		if (wret == 0) {
10835f39d397SChris Mason 			struct btrfs_disk_key disk_key;
10845f39d397SChris Mason 
10855f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
10865f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
10875f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
10885f39d397SChris Mason 
10895f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
10905f39d397SChris Mason 				path->nodes[level] = right;
1091e66f709bSChris Mason 				path->slots[level + 1] += 1;
1092e66f709bSChris Mason 				path->slots[level] = orig_slot -
10935f39d397SChris Mason 					btrfs_header_nritems(mid);
1094925baeddSChris Mason 				btrfs_tree_unlock(mid);
10955f39d397SChris Mason 				free_extent_buffer(mid);
1096e66f709bSChris Mason 			} else {
1097925baeddSChris Mason 				btrfs_tree_unlock(right);
10985f39d397SChris Mason 				free_extent_buffer(right);
1099e66f709bSChris Mason 			}
1100e66f709bSChris Mason 			return 0;
1101e66f709bSChris Mason 		}
1102925baeddSChris Mason 		btrfs_tree_unlock(right);
11035f39d397SChris Mason 		free_extent_buffer(right);
1104e66f709bSChris Mason 	}
1105e66f709bSChris Mason 	return 1;
1106e66f709bSChris Mason }
1107e66f709bSChris Mason 
110874123bd7SChris Mason /*
11093c69faecSChris Mason  * readahead one full node of leaves
11103c69faecSChris Mason  */
11113c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
111201f46658SChris Mason 			     int level, int slot, u64 objectid)
11133c69faecSChris Mason {
11145f39d397SChris Mason 	struct extent_buffer *node;
111501f46658SChris Mason 	struct btrfs_disk_key disk_key;
11163c69faecSChris Mason 	u32 nritems;
11173c69faecSChris Mason 	u64 search;
11186b80053dSChris Mason 	u64 lowest_read;
11196b80053dSChris Mason 	u64 highest_read;
11206b80053dSChris Mason 	u64 nread = 0;
11213c69faecSChris Mason 	int direction = path->reada;
11225f39d397SChris Mason 	struct extent_buffer *eb;
11236b80053dSChris Mason 	u32 nr;
11246b80053dSChris Mason 	u32 blocksize;
11256b80053dSChris Mason 	u32 nscan = 0;
1126db94535dSChris Mason 
1127a6b6e75eSChris Mason 	if (level != 1)
11283c69faecSChris Mason 		return;
11293c69faecSChris Mason 
11306702ed49SChris Mason 	if (!path->nodes[level])
11316702ed49SChris Mason 		return;
11326702ed49SChris Mason 
11335f39d397SChris Mason 	node = path->nodes[level];
1134925baeddSChris Mason 
11353c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
11366b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
11376b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
11385f39d397SChris Mason 	if (eb) {
11395f39d397SChris Mason 		free_extent_buffer(eb);
11403c69faecSChris Mason 		return;
11413c69faecSChris Mason 	}
11423c69faecSChris Mason 
11436b80053dSChris Mason 	highest_read = search;
11446b80053dSChris Mason 	lowest_read = search;
11456b80053dSChris Mason 
11465f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
11476b80053dSChris Mason 	nr = slot;
11483c69faecSChris Mason 	while(1) {
11496b80053dSChris Mason 		if (direction < 0) {
11506b80053dSChris Mason 			if (nr == 0)
11513c69faecSChris Mason 				break;
11526b80053dSChris Mason 			nr--;
11536b80053dSChris Mason 		} else if (direction > 0) {
11546b80053dSChris Mason 			nr++;
11556b80053dSChris Mason 			if (nr >= nritems)
11566b80053dSChris Mason 				break;
11573c69faecSChris Mason 		}
115801f46658SChris Mason 		if (path->reada < 0 && objectid) {
115901f46658SChris Mason 			btrfs_node_key(node, &disk_key, nr);
116001f46658SChris Mason 			if (btrfs_disk_key_objectid(&disk_key) != objectid)
116101f46658SChris Mason 				break;
116201f46658SChris Mason 		}
11636b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
11646b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
11656b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
11666b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
1167ca7a79adSChris Mason 			readahead_tree_block(root, search, blocksize,
1168ca7a79adSChris Mason 				     btrfs_node_ptr_generation(node, nr));
11696b80053dSChris Mason 			nread += blocksize;
11703c69faecSChris Mason 		}
11716b80053dSChris Mason 		nscan++;
11726b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
11736b80053dSChris Mason 			break;
11746b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
11756b80053dSChris Mason 			break;
11766b80053dSChris Mason 
11776b80053dSChris Mason 		if (search < lowest_read)
11786b80053dSChris Mason 			lowest_read = search;
11796b80053dSChris Mason 		if (search > highest_read)
11806b80053dSChris Mason 			highest_read = search;
11813c69faecSChris Mason 	}
11823c69faecSChris Mason }
1183925baeddSChris Mason 
1184925baeddSChris Mason static void unlock_up(struct btrfs_path *path, int level, int lowest_unlock)
1185925baeddSChris Mason {
1186925baeddSChris Mason 	int i;
1187925baeddSChris Mason 	int skip_level = level;
1188051e1b9fSChris Mason 	int no_skips = 0;
1189925baeddSChris Mason 	struct extent_buffer *t;
1190925baeddSChris Mason 
1191925baeddSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1192925baeddSChris Mason 		if (!path->nodes[i])
1193925baeddSChris Mason 			break;
1194925baeddSChris Mason 		if (!path->locks[i])
1195925baeddSChris Mason 			break;
1196051e1b9fSChris Mason 		if (!no_skips && path->slots[i] == 0) {
1197925baeddSChris Mason 			skip_level = i + 1;
1198925baeddSChris Mason 			continue;
1199925baeddSChris Mason 		}
1200051e1b9fSChris Mason 		if (!no_skips && path->keep_locks) {
1201925baeddSChris Mason 			u32 nritems;
1202925baeddSChris Mason 			t = path->nodes[i];
1203925baeddSChris Mason 			nritems = btrfs_header_nritems(t);
1204051e1b9fSChris Mason 			if (nritems < 1 || path->slots[i] >= nritems - 1) {
1205925baeddSChris Mason 				skip_level = i + 1;
1206925baeddSChris Mason 				continue;
1207925baeddSChris Mason 			}
1208925baeddSChris Mason 		}
1209051e1b9fSChris Mason 		if (skip_level < i && i >= lowest_unlock)
1210051e1b9fSChris Mason 			no_skips = 1;
1211051e1b9fSChris Mason 
1212925baeddSChris Mason 		t = path->nodes[i];
1213925baeddSChris Mason 		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
1214925baeddSChris Mason 			btrfs_tree_unlock(t);
1215925baeddSChris Mason 			path->locks[i] = 0;
1216925baeddSChris Mason 		}
1217925baeddSChris Mason 	}
1218925baeddSChris Mason }
1219925baeddSChris Mason 
12203c69faecSChris Mason /*
122174123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
122274123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
122374123bd7SChris Mason  * level of the path (level 0)
122474123bd7SChris Mason  *
122574123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
1226aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
1227aa5d6bedSChris Mason  * search a negative error number is returned.
122897571fd0SChris Mason  *
122997571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
123097571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
123197571fd0SChris Mason  * possible)
123274123bd7SChris Mason  */
1233e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
1234e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
1235e089f05cSChris Mason 		      ins_len, int cow)
1236be0e5c09SChris Mason {
12375f39d397SChris Mason 	struct extent_buffer *b;
1238051e1b9fSChris Mason 	struct extent_buffer *tmp;
1239be0e5c09SChris Mason 	int slot;
1240be0e5c09SChris Mason 	int ret;
1241be0e5c09SChris Mason 	int level;
12423c69faecSChris Mason 	int should_reada = p->reada;
1243925baeddSChris Mason 	int lowest_unlock = 1;
1244594a24ebSChris Mason 	int blocksize;
12459f3a7427SChris Mason 	u8 lowest_level = 0;
1246594a24ebSChris Mason 	u64 blocknr;
1247594a24ebSChris Mason 	u64 gen;
12489f3a7427SChris Mason 
12496702ed49SChris Mason 	lowest_level = p->lowest_level;
12506702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
125122b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
1252333db94cSChris Mason 	WARN_ON(cow && root == root->fs_info->extent_root &&
1253925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->alloc_mutex));
1254925baeddSChris Mason 	if (ins_len < 0)
1255925baeddSChris Mason 		lowest_unlock = 2;
1256bb803951SChris Mason again:
12575cd57b2cSChris Mason 	if (p->skip_locking)
12585cd57b2cSChris Mason 		b = btrfs_root_node(root);
12595cd57b2cSChris Mason 	else
1260925baeddSChris Mason 		b = btrfs_lock_root_node(root);
1261925baeddSChris Mason 
1262eb60ceacSChris Mason 	while (b) {
12635f39d397SChris Mason 		level = btrfs_header_level(b);
126402217ed2SChris Mason 		if (cow) {
126502217ed2SChris Mason 			int wret;
1266e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
1267e20d96d6SChris Mason 					       p->nodes[level + 1],
1268e20d96d6SChris Mason 					       p->slots[level + 1],
1269252c38f0SYan 					       &b);
127054aa1f4dSChris Mason 			if (wret) {
12715f39d397SChris Mason 				free_extent_buffer(b);
127254aa1f4dSChris Mason 				return wret;
127354aa1f4dSChris Mason 			}
127402217ed2SChris Mason 		}
127502217ed2SChris Mason 		BUG_ON(!cow && ins_len);
12765f39d397SChris Mason 		if (level != btrfs_header_level(b))
12772c90e5d6SChris Mason 			WARN_ON(1);
12785f39d397SChris Mason 		level = btrfs_header_level(b);
1279eb60ceacSChris Mason 		p->nodes[level] = b;
12805cd57b2cSChris Mason 		if (!p->skip_locking)
1281925baeddSChris Mason 			p->locks[level] = 1;
1282123abc88SChris Mason 		ret = check_block(root, p, level);
1283aa5d6bedSChris Mason 		if (ret)
1284aa5d6bedSChris Mason 			return -1;
1285925baeddSChris Mason 
12865f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
12875f39d397SChris Mason 		if (level != 0) {
1288be0e5c09SChris Mason 			if (ret && slot > 0)
1289be0e5c09SChris Mason 				slot -= 1;
1290be0e5c09SChris Mason 			p->slots[level] = slot;
12915f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
12921514794eSChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1293e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
12945c680ed6SChris Mason 				BUG_ON(sret > 0);
12955c680ed6SChris Mason 				if (sret)
12965c680ed6SChris Mason 					return sret;
12975c680ed6SChris Mason 				b = p->nodes[level];
12985c680ed6SChris Mason 				slot = p->slots[level];
1299bb803951SChris Mason 			} else if (ins_len < 0) {
1300e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1301e089f05cSChris Mason 							 level);
1302bb803951SChris Mason 				if (sret)
1303bb803951SChris Mason 					return sret;
1304bb803951SChris Mason 				b = p->nodes[level];
1305f510cfecSChris Mason 				if (!b) {
1306f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1307bb803951SChris Mason 					goto again;
1308f510cfecSChris Mason 				}
1309bb803951SChris Mason 				slot = p->slots[level];
13105f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
13115c680ed6SChris Mason 			}
1312f9efa9c7SChris Mason 			unlock_up(p, level, lowest_unlock);
1313f9efa9c7SChris Mason 
13149f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
1315925baeddSChris Mason 			if (level == lowest_level) {
13169f3a7427SChris Mason 				break;
1317925baeddSChris Mason 			}
1318ca7a79adSChris Mason 
1319594a24ebSChris Mason 			blocknr = btrfs_node_blockptr(b, slot);
1320594a24ebSChris Mason 			gen = btrfs_node_ptr_generation(b, slot);
1321594a24ebSChris Mason 			blocksize = btrfs_level_size(root, level - 1);
1322594a24ebSChris Mason 
1323594a24ebSChris Mason 			tmp = btrfs_find_tree_block(root, blocknr, blocksize);
1324594a24ebSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
1325051e1b9fSChris Mason 				b = tmp;
1326051e1b9fSChris Mason 			} else {
1327051e1b9fSChris Mason 				/*
1328051e1b9fSChris Mason 				 * reduce lock contention at high levels
1329051e1b9fSChris Mason 				 * of the btree by dropping locks before
1330051e1b9fSChris Mason 				 * we read.
1331051e1b9fSChris Mason 				 */
1332051e1b9fSChris Mason 				if (level > 1) {
1333051e1b9fSChris Mason 					btrfs_release_path(NULL, p);
1334051e1b9fSChris Mason 					if (tmp)
1335051e1b9fSChris Mason 						free_extent_buffer(tmp);
1336f9efa9c7SChris Mason 					if (should_reada)
1337f9efa9c7SChris Mason 						reada_for_search(root, p,
1338f9efa9c7SChris Mason 								 level, slot,
1339f9efa9c7SChris Mason 								 key->objectid);
1340f9efa9c7SChris Mason 
1341594a24ebSChris Mason 					tmp = read_tree_block(root, blocknr,
1342594a24ebSChris Mason 							 blocksize, gen);
1343594a24ebSChris Mason 					if (tmp)
1344594a24ebSChris Mason 						free_extent_buffer(tmp);
1345051e1b9fSChris Mason 					goto again;
1346051e1b9fSChris Mason 				} else {
1347a74a4b97SChris Mason 					if (tmp)
1348a74a4b97SChris Mason 						free_extent_buffer(tmp);
1349f9efa9c7SChris Mason 					if (should_reada)
1350f9efa9c7SChris Mason 						reada_for_search(root, p,
1351f9efa9c7SChris Mason 								 level, slot,
1352f9efa9c7SChris Mason 								 key->objectid);
1353ca7a79adSChris Mason 					b = read_node_slot(root, b, slot);
1354051e1b9fSChris Mason 				}
1355051e1b9fSChris Mason 			}
13565cd57b2cSChris Mason 			if (!p->skip_locking)
1357925baeddSChris Mason 				btrfs_tree_lock(b);
1358be0e5c09SChris Mason 		} else {
1359be0e5c09SChris Mason 			p->slots[level] = slot;
13605f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
13610783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1362d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1363cc0c5538SChris Mason 						      p, ins_len, ret == 0);
13645c680ed6SChris Mason 				BUG_ON(sret > 0);
13655c680ed6SChris Mason 				if (sret)
13665c680ed6SChris Mason 					return sret;
13675c680ed6SChris Mason 			}
1368925baeddSChris Mason 			unlock_up(p, level, lowest_unlock);
1369be0e5c09SChris Mason 			return ret;
1370be0e5c09SChris Mason 		}
1371be0e5c09SChris Mason 	}
1372aa5d6bedSChris Mason 	return 1;
1373be0e5c09SChris Mason }
1374be0e5c09SChris Mason 
137574123bd7SChris Mason /*
137674123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
137774123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
137874123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
137974123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
138074123bd7SChris Mason  * higher levels
1381aa5d6bedSChris Mason  *
1382aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1383aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
138474123bd7SChris Mason  */
13855f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
13865f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
13875f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1388be0e5c09SChris Mason {
1389be0e5c09SChris Mason 	int i;
1390aa5d6bedSChris Mason 	int ret = 0;
13915f39d397SChris Mason 	struct extent_buffer *t;
13925f39d397SChris Mason 
1393234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1394be0e5c09SChris Mason 		int tslot = path->slots[i];
1395eb60ceacSChris Mason 		if (!path->nodes[i])
1396be0e5c09SChris Mason 			break;
13975f39d397SChris Mason 		t = path->nodes[i];
13985f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1399d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1400be0e5c09SChris Mason 		if (tslot != 0)
1401be0e5c09SChris Mason 			break;
1402be0e5c09SChris Mason 	}
1403aa5d6bedSChris Mason 	return ret;
1404be0e5c09SChris Mason }
1405be0e5c09SChris Mason 
140674123bd7SChris Mason /*
140774123bd7SChris Mason  * try to push data from one node into the next node left in the
140879f95c82SChris Mason  * tree.
1409aa5d6bedSChris Mason  *
1410aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1411aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
141274123bd7SChris Mason  */
141398ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
141498ed5174SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
1415971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
1416be0e5c09SChris Mason {
1417be0e5c09SChris Mason 	int push_items = 0;
1418bb803951SChris Mason 	int src_nritems;
1419bb803951SChris Mason 	int dst_nritems;
1420aa5d6bedSChris Mason 	int ret = 0;
1421be0e5c09SChris Mason 
14225f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14235f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1424123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
14257bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14267bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
142754aa1f4dSChris Mason 
1428bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
1429971a1f66SChris Mason 		return 1;
1430971a1f66SChris Mason 
1431eb60ceacSChris Mason 	if (push_items <= 0) {
1432be0e5c09SChris Mason 		return 1;
1433eb60ceacSChris Mason 	}
1434be0e5c09SChris Mason 
1435bce4eae9SChris Mason 	if (empty) {
1436971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
1437bce4eae9SChris Mason 		if (push_items < src_nritems) {
1438bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
1439bce4eae9SChris Mason 			 * we aren't going to empty it
1440bce4eae9SChris Mason 			 */
1441bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
1442bce4eae9SChris Mason 				if (push_items <= 8)
1443bce4eae9SChris Mason 					return 1;
1444bce4eae9SChris Mason 				push_items -= 8;
1445bce4eae9SChris Mason 			}
1446bce4eae9SChris Mason 		}
1447bce4eae9SChris Mason 	} else
1448bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
144979f95c82SChris Mason 
14505f39d397SChris Mason 	copy_extent_buffer(dst, src,
14515f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
14525f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1453123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
14545f39d397SChris Mason 
1455bb803951SChris Mason 	if (push_items < src_nritems) {
14565f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
14575f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1458e2fa7227SChris Mason 				      (src_nritems - push_items) *
1459123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1460bb803951SChris Mason 	}
14615f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
14625f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
14635f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
14645f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1465bb803951SChris Mason 	return ret;
1466be0e5c09SChris Mason }
1467be0e5c09SChris Mason 
146897571fd0SChris Mason /*
146979f95c82SChris Mason  * try to push data from one node into the next node right in the
147079f95c82SChris Mason  * tree.
147179f95c82SChris Mason  *
147279f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
147379f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
147479f95c82SChris Mason  *
147579f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
147679f95c82SChris Mason  */
14775f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
14785f39d397SChris Mason 			      struct btrfs_root *root,
14795f39d397SChris Mason 			      struct extent_buffer *dst,
14805f39d397SChris Mason 			      struct extent_buffer *src)
148179f95c82SChris Mason {
148279f95c82SChris Mason 	int push_items = 0;
148379f95c82SChris Mason 	int max_push;
148479f95c82SChris Mason 	int src_nritems;
148579f95c82SChris Mason 	int dst_nritems;
148679f95c82SChris Mason 	int ret = 0;
148779f95c82SChris Mason 
14887bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14897bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
14907bb86316SChris Mason 
14915f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14925f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1493123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1494bce4eae9SChris Mason 	if (push_items <= 0) {
149579f95c82SChris Mason 		return 1;
1496bce4eae9SChris Mason 	}
1497bce4eae9SChris Mason 
1498bce4eae9SChris Mason 	if (src_nritems < 4) {
1499bce4eae9SChris Mason 		return 1;
1500bce4eae9SChris Mason 	}
150179f95c82SChris Mason 
150279f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
150379f95c82SChris Mason 	/* don't try to empty the node */
1504bce4eae9SChris Mason 	if (max_push >= src_nritems) {
150579f95c82SChris Mason 		return 1;
1506bce4eae9SChris Mason 	}
1507252c38f0SYan 
150879f95c82SChris Mason 	if (max_push < push_items)
150979f95c82SChris Mason 		push_items = max_push;
151079f95c82SChris Mason 
15115f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
15125f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
15135f39d397SChris Mason 				      (dst_nritems) *
15145f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1515d6025579SChris Mason 
15165f39d397SChris Mason 	copy_extent_buffer(dst, src,
15175f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
15185f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1519123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
152079f95c82SChris Mason 
15215f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
15225f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
152379f95c82SChris Mason 
15245f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
15255f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
152679f95c82SChris Mason 	return ret;
152779f95c82SChris Mason }
152879f95c82SChris Mason 
152979f95c82SChris Mason /*
153097571fd0SChris Mason  * helper function to insert a new root level in the tree.
153197571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
153297571fd0SChris Mason  * point to the existing root
1533aa5d6bedSChris Mason  *
1534aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
153597571fd0SChris Mason  */
153698ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans,
15375f39d397SChris Mason 			   struct btrfs_root *root,
15385f39d397SChris Mason 			   struct btrfs_path *path, int level)
153974123bd7SChris Mason {
15407bb86316SChris Mason 	u64 root_gen;
15417bb86316SChris Mason 	u64 lower_gen;
15425f39d397SChris Mason 	struct extent_buffer *lower;
15435f39d397SChris Mason 	struct extent_buffer *c;
1544925baeddSChris Mason 	struct extent_buffer *old;
15455f39d397SChris Mason 	struct btrfs_disk_key lower_key;
15465c680ed6SChris Mason 
15475c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
15485c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
15495c680ed6SChris Mason 
15507bb86316SChris Mason 	if (root->ref_cows)
15517bb86316SChris Mason 		root_gen = trans->transid;
15527bb86316SChris Mason 	else
15537bb86316SChris Mason 		root_gen = 0;
15547bb86316SChris Mason 
15557bb86316SChris Mason 	lower = path->nodes[level-1];
15567bb86316SChris Mason 	if (level == 1)
15577bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
15587bb86316SChris Mason 	else
15597bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
15607bb86316SChris Mason 
1561925baeddSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
15627bb86316SChris Mason 				   root->root_key.objectid,
15637bb86316SChris Mason 				   root_gen, lower_key.objectid, level,
1564db94535dSChris Mason 				   root->node->start, 0);
15655f39d397SChris Mason 	if (IS_ERR(c))
15665f39d397SChris Mason 		return PTR_ERR(c);
1567925baeddSChris Mason 
15685f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
15695f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
15705f39d397SChris Mason 	btrfs_set_header_level(c, level);
1571db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
15725f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
15735f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
1574d5719762SChris Mason 
15755f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
15765f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
15775f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1578e17cade2SChris Mason 
1579e17cade2SChris Mason 	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
1580e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
1581e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
1582e17cade2SChris Mason 
15835f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1584db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
15857bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
15867bb86316SChris Mason 	WARN_ON(lower_gen == 0);
15877bb86316SChris Mason 
15887bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
15895f39d397SChris Mason 
15905f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1591d5719762SChris Mason 
1592925baeddSChris Mason 	spin_lock(&root->node_lock);
1593925baeddSChris Mason 	old = root->node;
15945f39d397SChris Mason 	root->node = c;
1595925baeddSChris Mason 	spin_unlock(&root->node_lock);
1596925baeddSChris Mason 
1597925baeddSChris Mason 	/* the super has an extra ref to root->node */
1598925baeddSChris Mason 	free_extent_buffer(old);
1599925baeddSChris Mason 
16000b86a832SChris Mason 	add_root_to_dirty_list(root);
16015f39d397SChris Mason 	extent_buffer_get(c);
16025f39d397SChris Mason 	path->nodes[level] = c;
1603925baeddSChris Mason 	path->locks[level] = 1;
160474123bd7SChris Mason 	path->slots[level] = 0;
16057bb86316SChris Mason 
16067bb86316SChris Mason 	if (root->ref_cows && lower_gen != trans->transid) {
16077bb86316SChris Mason 		struct btrfs_path *back_path = btrfs_alloc_path();
16087bb86316SChris Mason 		int ret;
1609925baeddSChris Mason 		mutex_lock(&root->fs_info->alloc_mutex);
16107bb86316SChris Mason 		ret = btrfs_insert_extent_backref(trans,
16117bb86316SChris Mason 						  root->fs_info->extent_root,
16127bb86316SChris Mason 						  path, lower->start,
16137bb86316SChris Mason 						  root->root_key.objectid,
16147bb86316SChris Mason 						  trans->transid, 0, 0);
16157bb86316SChris Mason 		BUG_ON(ret);
1616925baeddSChris Mason 		mutex_unlock(&root->fs_info->alloc_mutex);
16177bb86316SChris Mason 		btrfs_free_path(back_path);
16187bb86316SChris Mason 	}
161974123bd7SChris Mason 	return 0;
162074123bd7SChris Mason }
16215c680ed6SChris Mason 
16225c680ed6SChris Mason /*
16235c680ed6SChris Mason  * worker function to insert a single pointer in a node.
16245c680ed6SChris Mason  * the node should have enough room for the pointer already
162597571fd0SChris Mason  *
16265c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
16275c680ed6SChris Mason  * blocknr is the block the key points to.
1628aa5d6bedSChris Mason  *
1629aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
16305c680ed6SChris Mason  */
1631e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1632e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1633db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
16345c680ed6SChris Mason {
16355f39d397SChris Mason 	struct extent_buffer *lower;
16365c680ed6SChris Mason 	int nritems;
16375c680ed6SChris Mason 
16385c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
16395f39d397SChris Mason 	lower = path->nodes[level];
16405f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
164174123bd7SChris Mason 	if (slot > nritems)
164274123bd7SChris Mason 		BUG();
1643123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
164474123bd7SChris Mason 		BUG();
164574123bd7SChris Mason 	if (slot != nritems) {
16465f39d397SChris Mason 		memmove_extent_buffer(lower,
16475f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
16485f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1649123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
165074123bd7SChris Mason 	}
16515f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1652db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
165374493f7aSChris Mason 	WARN_ON(trans->transid == 0);
165474493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
16555f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
16565f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
165774123bd7SChris Mason 	return 0;
165874123bd7SChris Mason }
165974123bd7SChris Mason 
166097571fd0SChris Mason /*
166197571fd0SChris Mason  * split the node at the specified level in path in two.
166297571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
166397571fd0SChris Mason  *
166497571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
166597571fd0SChris Mason  * left and right, if either one works, it returns right away.
1666aa5d6bedSChris Mason  *
1667aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
166897571fd0SChris Mason  */
1669e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1670e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1671be0e5c09SChris Mason {
16727bb86316SChris Mason 	u64 root_gen;
16735f39d397SChris Mason 	struct extent_buffer *c;
16745f39d397SChris Mason 	struct extent_buffer *split;
16755f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1676be0e5c09SChris Mason 	int mid;
16775c680ed6SChris Mason 	int ret;
1678aa5d6bedSChris Mason 	int wret;
16797518a238SChris Mason 	u32 c_nritems;
1680be0e5c09SChris Mason 
16815f39d397SChris Mason 	c = path->nodes[level];
16827bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
16835f39d397SChris Mason 	if (c == root->node) {
16845c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1685e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
16865c680ed6SChris Mason 		if (ret)
16875c680ed6SChris Mason 			return ret;
1688e66f709bSChris Mason 	} else {
1689e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
16905f39d397SChris Mason 		c = path->nodes[level];
16915f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1692c448acf0SChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
1693e66f709bSChris Mason 			return 0;
169454aa1f4dSChris Mason 		if (ret < 0)
169554aa1f4dSChris Mason 			return ret;
16965c680ed6SChris Mason 	}
1697e66f709bSChris Mason 
16985f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
16997bb86316SChris Mason 	if (root->ref_cows)
17007bb86316SChris Mason 		root_gen = trans->transid;
17017bb86316SChris Mason 	else
17027bb86316SChris Mason 		root_gen = 0;
17037bb86316SChris Mason 
17047bb86316SChris Mason 	btrfs_node_key(c, &disk_key, 0);
1705925baeddSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
17067bb86316SChris Mason 					 root->root_key.objectid,
17077bb86316SChris Mason 					 root_gen,
17087bb86316SChris Mason 					 btrfs_disk_key_objectid(&disk_key),
17097bb86316SChris Mason 					 level, c->start, 0);
17105f39d397SChris Mason 	if (IS_ERR(split))
17115f39d397SChris Mason 		return PTR_ERR(split);
171254aa1f4dSChris Mason 
17135f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
17145f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1715db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
17165f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
17175f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
171863b10fc4SChris Mason 	btrfs_set_header_flags(split, 0);
17195f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
17205f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
17215f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1722e17cade2SChris Mason 	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
1723e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
1724e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
17255f39d397SChris Mason 
17267518a238SChris Mason 	mid = (c_nritems + 1) / 2;
17275f39d397SChris Mason 
17285f39d397SChris Mason 	copy_extent_buffer(split, c,
17295f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
17305f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1731123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
17325f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
17335f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1734aa5d6bedSChris Mason 	ret = 0;
1735aa5d6bedSChris Mason 
17365f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
17375f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
17385f39d397SChris Mason 
17395f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1740db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
17415f39d397SChris Mason 			  path->slots[level + 1] + 1,
1742123abc88SChris Mason 			  level + 1);
1743aa5d6bedSChris Mason 	if (wret)
1744aa5d6bedSChris Mason 		ret = wret;
1745aa5d6bedSChris Mason 
17465de08d7dSChris Mason 	if (path->slots[level] >= mid) {
17475c680ed6SChris Mason 		path->slots[level] -= mid;
1748925baeddSChris Mason 		btrfs_tree_unlock(c);
17495f39d397SChris Mason 		free_extent_buffer(c);
17505f39d397SChris Mason 		path->nodes[level] = split;
17515c680ed6SChris Mason 		path->slots[level + 1] += 1;
1752eb60ceacSChris Mason 	} else {
1753925baeddSChris Mason 		btrfs_tree_unlock(split);
17545f39d397SChris Mason 		free_extent_buffer(split);
1755be0e5c09SChris Mason 	}
1756aa5d6bedSChris Mason 	return ret;
1757be0e5c09SChris Mason }
1758be0e5c09SChris Mason 
175974123bd7SChris Mason /*
176074123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
176174123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
176274123bd7SChris Mason  * space used both by the item structs and the item data
176374123bd7SChris Mason  */
17645f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1765be0e5c09SChris Mason {
1766be0e5c09SChris Mason 	int data_len;
17675f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1768d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1769be0e5c09SChris Mason 
1770be0e5c09SChris Mason 	if (!nr)
1771be0e5c09SChris Mason 		return 0;
17725f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
17735f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
17740783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1775d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1776be0e5c09SChris Mason 	return data_len;
1777be0e5c09SChris Mason }
1778be0e5c09SChris Mason 
177974123bd7SChris Mason /*
1780d4dbff95SChris Mason  * The space between the end of the leaf items and
1781d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1782d4dbff95SChris Mason  * the leaf has left for both items and data
1783d4dbff95SChris Mason  */
17845f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1785d4dbff95SChris Mason {
17865f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
17875f39d397SChris Mason 	int ret;
17885f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
17895f39d397SChris Mason 	if (ret < 0) {
17905f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1791ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
17925f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
17935f39d397SChris Mason 	}
17945f39d397SChris Mason 	return ret;
1795d4dbff95SChris Mason }
1796d4dbff95SChris Mason 
1797d4dbff95SChris Mason /*
179800ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
179900ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1800aa5d6bedSChris Mason  *
1801aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1802aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
180300ec4c51SChris Mason  */
1804e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
180534a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
180634a38218SChris Mason 			   int empty)
180700ec4c51SChris Mason {
18085f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
18095f39d397SChris Mason 	struct extent_buffer *right;
18105f39d397SChris Mason 	struct extent_buffer *upper;
18115f39d397SChris Mason 	struct btrfs_disk_key disk_key;
181200ec4c51SChris Mason 	int slot;
181334a38218SChris Mason 	u32 i;
181400ec4c51SChris Mason 	int free_space;
181500ec4c51SChris Mason 	int push_space = 0;
181600ec4c51SChris Mason 	int push_items = 0;
18170783fcfcSChris Mason 	struct btrfs_item *item;
18187518a238SChris Mason 	u32 left_nritems;
181934a38218SChris Mason 	u32 nr;
18207518a238SChris Mason 	u32 right_nritems;
18215f39d397SChris Mason 	u32 data_end;
1822db94535dSChris Mason 	u32 this_item_size;
182354aa1f4dSChris Mason 	int ret;
182400ec4c51SChris Mason 
182500ec4c51SChris Mason 	slot = path->slots[1];
182600ec4c51SChris Mason 	if (!path->nodes[1]) {
182700ec4c51SChris Mason 		return 1;
182800ec4c51SChris Mason 	}
182900ec4c51SChris Mason 	upper = path->nodes[1];
18305f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
183100ec4c51SChris Mason 		return 1;
18325f39d397SChris Mason 
1833a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
1834a2135011SChris Mason 
1835ca7a79adSChris Mason 	right = read_node_slot(root, upper, slot + 1);
1836925baeddSChris Mason 	btrfs_tree_lock(right);
1837123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1838925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1839925baeddSChris Mason 		goto out_unlock;
184002217ed2SChris Mason 
18415f39d397SChris Mason 	/* cow and double check */
18425f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
18435f39d397SChris Mason 			      slot + 1, &right);
1844925baeddSChris Mason 	if (ret)
1845925baeddSChris Mason 		goto out_unlock;
1846925baeddSChris Mason 
18475f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1848925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1849925baeddSChris Mason 		goto out_unlock;
18505f39d397SChris Mason 
18515f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
1852925baeddSChris Mason 	if (left_nritems == 0)
1853925baeddSChris Mason 		goto out_unlock;
18545f39d397SChris Mason 
185534a38218SChris Mason 	if (empty)
185634a38218SChris Mason 		nr = 0;
185734a38218SChris Mason 	else
185834a38218SChris Mason 		nr = 1;
185934a38218SChris Mason 
186034a38218SChris Mason 	i = left_nritems - 1;
186134a38218SChris Mason 	while (i >= nr) {
18625f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1863db94535dSChris Mason 
186400ec4c51SChris Mason 		if (path->slots[0] == i)
186500ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1866db94535dSChris Mason 
1867db94535dSChris Mason 		if (!left->map_token) {
1868db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1869db94535dSChris Mason 					sizeof(struct btrfs_item),
1870db94535dSChris Mason 					&left->map_token, &left->kaddr,
1871db94535dSChris Mason 					&left->map_start, &left->map_len,
1872db94535dSChris Mason 					KM_USER1);
1873db94535dSChris Mason 		}
1874db94535dSChris Mason 
1875db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1876db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
187700ec4c51SChris Mason 			break;
187800ec4c51SChris Mason 		push_items++;
1879db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
188034a38218SChris Mason 		if (i == 0)
188134a38218SChris Mason 			break;
188234a38218SChris Mason 		i--;
1883db94535dSChris Mason 	}
1884db94535dSChris Mason 	if (left->map_token) {
1885db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1886db94535dSChris Mason 		left->map_token = NULL;
188700ec4c51SChris Mason 	}
18885f39d397SChris Mason 
1889925baeddSChris Mason 	if (push_items == 0)
1890925baeddSChris Mason 		goto out_unlock;
18915f39d397SChris Mason 
189234a38218SChris Mason 	if (!empty && push_items == left_nritems)
1893a429e513SChris Mason 		WARN_ON(1);
18945f39d397SChris Mason 
189500ec4c51SChris Mason 	/* push left to right */
18965f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
189734a38218SChris Mason 
18985f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1899123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
19005f39d397SChris Mason 
190100ec4c51SChris Mason 	/* make room in the right data area */
19025f39d397SChris Mason 	data_end = leaf_data_end(root, right);
19035f39d397SChris Mason 	memmove_extent_buffer(right,
19045f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
19055f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
19065f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
19075f39d397SChris Mason 
190800ec4c51SChris Mason 	/* copy from the left data area */
19095f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1910d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1911d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1912d6025579SChris Mason 		     push_space);
19135f39d397SChris Mason 
19145f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
19155f39d397SChris Mason 			      btrfs_item_nr_offset(0),
19160783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
19175f39d397SChris Mason 
191800ec4c51SChris Mason 	/* copy the items from left to right */
19195f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
19205f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
19210783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
192200ec4c51SChris Mason 
192300ec4c51SChris Mason 	/* update the item pointers */
19247518a238SChris Mason 	right_nritems += push_items;
19255f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1926123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
19277518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
19285f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1929db94535dSChris Mason 		if (!right->map_token) {
1930db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1931db94535dSChris Mason 					sizeof(struct btrfs_item),
1932db94535dSChris Mason 					&right->map_token, &right->kaddr,
1933db94535dSChris Mason 					&right->map_start, &right->map_len,
1934db94535dSChris Mason 					KM_USER1);
1935db94535dSChris Mason 		}
1936db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1937db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1938db94535dSChris Mason 	}
1939db94535dSChris Mason 
1940db94535dSChris Mason 	if (right->map_token) {
1941db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1942db94535dSChris Mason 		right->map_token = NULL;
194300ec4c51SChris Mason 	}
19447518a238SChris Mason 	left_nritems -= push_items;
19455f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
194600ec4c51SChris Mason 
194734a38218SChris Mason 	if (left_nritems)
19485f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
19495f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1950a429e513SChris Mason 
19515f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
19525f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1953d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
195402217ed2SChris Mason 
195500ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
19567518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
19577518a238SChris Mason 		path->slots[0] -= left_nritems;
1958925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
1959925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
1960925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
19615f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19625f39d397SChris Mason 		path->nodes[0] = right;
196300ec4c51SChris Mason 		path->slots[1] += 1;
196400ec4c51SChris Mason 	} else {
1965925baeddSChris Mason 		btrfs_tree_unlock(right);
19665f39d397SChris Mason 		free_extent_buffer(right);
196700ec4c51SChris Mason 	}
196800ec4c51SChris Mason 	return 0;
1969925baeddSChris Mason 
1970925baeddSChris Mason out_unlock:
1971925baeddSChris Mason 	btrfs_tree_unlock(right);
1972925baeddSChris Mason 	free_extent_buffer(right);
1973925baeddSChris Mason 	return 1;
197400ec4c51SChris Mason }
1975925baeddSChris Mason 
197600ec4c51SChris Mason /*
197774123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
197874123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
197974123bd7SChris Mason  */
1980e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
198134a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
198234a38218SChris Mason 			  int empty)
1983be0e5c09SChris Mason {
19845f39d397SChris Mason 	struct btrfs_disk_key disk_key;
19855f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
19865f39d397SChris Mason 	struct extent_buffer *left;
1987be0e5c09SChris Mason 	int slot;
1988be0e5c09SChris Mason 	int i;
1989be0e5c09SChris Mason 	int free_space;
1990be0e5c09SChris Mason 	int push_space = 0;
1991be0e5c09SChris Mason 	int push_items = 0;
19920783fcfcSChris Mason 	struct btrfs_item *item;
19937518a238SChris Mason 	u32 old_left_nritems;
19945f39d397SChris Mason 	u32 right_nritems;
199534a38218SChris Mason 	u32 nr;
1996aa5d6bedSChris Mason 	int ret = 0;
1997aa5d6bedSChris Mason 	int wret;
1998db94535dSChris Mason 	u32 this_item_size;
1999db94535dSChris Mason 	u32 old_left_item_size;
2000be0e5c09SChris Mason 
2001be0e5c09SChris Mason 	slot = path->slots[1];
20025f39d397SChris Mason 	if (slot == 0)
2003be0e5c09SChris Mason 		return 1;
20045f39d397SChris Mason 	if (!path->nodes[1])
2005be0e5c09SChris Mason 		return 1;
20065f39d397SChris Mason 
20073685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
20083685f791SChris Mason 	if (right_nritems == 0) {
20093685f791SChris Mason 		return 1;
20103685f791SChris Mason 	}
20113685f791SChris Mason 
2012a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
2013a2135011SChris Mason 
2014ca7a79adSChris Mason 	left = read_node_slot(root, path->nodes[1], slot - 1);
2015925baeddSChris Mason 	btrfs_tree_lock(left);
2016123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20170783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2018925baeddSChris Mason 		ret = 1;
2019925baeddSChris Mason 		goto out;
2020be0e5c09SChris Mason 	}
202102217ed2SChris Mason 
202202217ed2SChris Mason 	/* cow and double check */
20235f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
20245f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
202554aa1f4dSChris Mason 	if (ret) {
202654aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
2027925baeddSChris Mason 		ret = 1;
2028925baeddSChris Mason 		goto out;
202954aa1f4dSChris Mason 	}
20303685f791SChris Mason 
2031123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20320783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2033925baeddSChris Mason 		ret = 1;
2034925baeddSChris Mason 		goto out;
203502217ed2SChris Mason 	}
203602217ed2SChris Mason 
203734a38218SChris Mason 	if (empty)
203834a38218SChris Mason 		nr = right_nritems;
203934a38218SChris Mason 	else
204034a38218SChris Mason 		nr = right_nritems - 1;
204134a38218SChris Mason 
204234a38218SChris Mason 	for (i = 0; i < nr; i++) {
20435f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2044db94535dSChris Mason 		if (!right->map_token) {
2045db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2046db94535dSChris Mason 					sizeof(struct btrfs_item),
2047db94535dSChris Mason 					&right->map_token, &right->kaddr,
2048db94535dSChris Mason 					&right->map_start, &right->map_len,
2049db94535dSChris Mason 					KM_USER1);
2050db94535dSChris Mason 		}
2051db94535dSChris Mason 
2052be0e5c09SChris Mason 		if (path->slots[0] == i)
2053be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
2054db94535dSChris Mason 
2055db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
2056db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
2057be0e5c09SChris Mason 			break;
2058db94535dSChris Mason 
2059be0e5c09SChris Mason 		push_items++;
2060db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
2061be0e5c09SChris Mason 	}
2062db94535dSChris Mason 
2063db94535dSChris Mason 	if (right->map_token) {
2064db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2065db94535dSChris Mason 		right->map_token = NULL;
2066db94535dSChris Mason 	}
2067db94535dSChris Mason 
2068be0e5c09SChris Mason 	if (push_items == 0) {
2069925baeddSChris Mason 		ret = 1;
2070925baeddSChris Mason 		goto out;
2071be0e5c09SChris Mason 	}
207234a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
2073a429e513SChris Mason 		WARN_ON(1);
20745f39d397SChris Mason 
2075be0e5c09SChris Mason 	/* push data from right to left */
20765f39d397SChris Mason 	copy_extent_buffer(left, right,
20775f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
20785f39d397SChris Mason 			   btrfs_item_nr_offset(0),
20795f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
20805f39d397SChris Mason 
2081123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
20825f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
20835f39d397SChris Mason 
20845f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
2085d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
2086123abc88SChris Mason 		     btrfs_leaf_data(right) +
20875f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
2088be0e5c09SChris Mason 		     push_space);
20895f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
2090eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
2091eb60ceacSChris Mason 
2092db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
2093be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
20945f39d397SChris Mason 		u32 ioff;
2095db94535dSChris Mason 
20965f39d397SChris Mason 		item = btrfs_item_nr(left, i);
2097db94535dSChris Mason 		if (!left->map_token) {
2098db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
2099db94535dSChris Mason 					sizeof(struct btrfs_item),
2100db94535dSChris Mason 					&left->map_token, &left->kaddr,
2101db94535dSChris Mason 					&left->map_start, &left->map_len,
2102db94535dSChris Mason 					KM_USER1);
2103db94535dSChris Mason 		}
2104db94535dSChris Mason 
21055f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
21065f39d397SChris Mason 		btrfs_set_item_offset(left, item,
2107db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2108be0e5c09SChris Mason 	}
21095f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2110db94535dSChris Mason 	if (left->map_token) {
2111db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
2112db94535dSChris Mason 		left->map_token = NULL;
2113db94535dSChris Mason 	}
2114be0e5c09SChris Mason 
2115be0e5c09SChris Mason 	/* fixup right node */
211634a38218SChris Mason 	if (push_items > right_nritems) {
211734a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
211834a38218SChris Mason 		WARN_ON(1);
211934a38218SChris Mason 	}
212034a38218SChris Mason 
212134a38218SChris Mason 	if (push_items < right_nritems) {
21225f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
2123123abc88SChris Mason 						  leaf_data_end(root, right);
21245f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
2125d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
2126d6025579SChris Mason 				      btrfs_leaf_data(right) +
2127123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
21285f39d397SChris Mason 
21295f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
21305f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
21315f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
21320783fcfcSChris Mason 			     sizeof(struct btrfs_item));
213334a38218SChris Mason 	}
2134eef1c494SYan 	right_nritems -= push_items;
2135eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
2136123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
21375f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
21385f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2139db94535dSChris Mason 
2140db94535dSChris Mason 		if (!right->map_token) {
2141db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2142db94535dSChris Mason 					sizeof(struct btrfs_item),
2143db94535dSChris Mason 					&right->map_token, &right->kaddr,
2144db94535dSChris Mason 					&right->map_start, &right->map_len,
2145db94535dSChris Mason 					KM_USER1);
2146db94535dSChris Mason 		}
2147db94535dSChris Mason 
2148db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
2149db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
2150db94535dSChris Mason 	}
2151db94535dSChris Mason 	if (right->map_token) {
2152db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2153db94535dSChris Mason 		right->map_token = NULL;
2154be0e5c09SChris Mason 	}
2155eb60ceacSChris Mason 
21565f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
215734a38218SChris Mason 	if (right_nritems)
21585f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
2159098f59c2SChris Mason 
21605f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
21615f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
2162aa5d6bedSChris Mason 	if (wret)
2163aa5d6bedSChris Mason 		ret = wret;
2164be0e5c09SChris Mason 
2165be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
2166be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
2167be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
2168925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
2169925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
2170925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
21715f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
21725f39d397SChris Mason 		path->nodes[0] = left;
2173be0e5c09SChris Mason 		path->slots[1] -= 1;
2174be0e5c09SChris Mason 	} else {
2175925baeddSChris Mason 		btrfs_tree_unlock(left);
21765f39d397SChris Mason 		free_extent_buffer(left);
2177be0e5c09SChris Mason 		path->slots[0] -= push_items;
2178be0e5c09SChris Mason 	}
2179eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2180aa5d6bedSChris Mason 	return ret;
2181925baeddSChris Mason out:
2182925baeddSChris Mason 	btrfs_tree_unlock(left);
2183925baeddSChris Mason 	free_extent_buffer(left);
2184925baeddSChris Mason 	return ret;
2185be0e5c09SChris Mason }
2186be0e5c09SChris Mason 
218774123bd7SChris Mason /*
218874123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
218974123bd7SChris Mason  * available for the resulting leaf level of the path.
2190aa5d6bedSChris Mason  *
2191aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
219274123bd7SChris Mason  */
2193e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
2194d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
2195cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend)
2196be0e5c09SChris Mason {
21977bb86316SChris Mason 	u64 root_gen;
21985f39d397SChris Mason 	struct extent_buffer *l;
21997518a238SChris Mason 	u32 nritems;
2200eb60ceacSChris Mason 	int mid;
2201eb60ceacSChris Mason 	int slot;
22025f39d397SChris Mason 	struct extent_buffer *right;
22030783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
2204be0e5c09SChris Mason 	int data_copy_size;
2205be0e5c09SChris Mason 	int rt_data_off;
2206be0e5c09SChris Mason 	int i;
2207d4dbff95SChris Mason 	int ret = 0;
2208aa5d6bedSChris Mason 	int wret;
2209cc0c5538SChris Mason 	int double_split;
2210cc0c5538SChris Mason 	int num_doubles = 0;
2211d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
2212be0e5c09SChris Mason 
2213cc0c5538SChris Mason 	if (extend)
2214cc0c5538SChris Mason 		space_needed = data_size;
2215cc0c5538SChris Mason 
22167bb86316SChris Mason 	if (root->ref_cows)
22177bb86316SChris Mason 		root_gen = trans->transid;
22187bb86316SChris Mason 	else
22197bb86316SChris Mason 		root_gen = 0;
22207bb86316SChris Mason 
222140689478SChris Mason 	/* first try to make some room by pushing left and right */
22223685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
222334a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
22243326d1b0SChris Mason 		if (wret < 0) {
2225eaee50e8SChris Mason 			return wret;
22263326d1b0SChris Mason 		}
2227eaee50e8SChris Mason 		if (wret) {
222834a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
2229eaee50e8SChris Mason 			if (wret < 0)
2230eaee50e8SChris Mason 				return wret;
2231eaee50e8SChris Mason 		}
22325f39d397SChris Mason 		l = path->nodes[0];
2233aa5d6bedSChris Mason 
2234aa5d6bedSChris Mason 		/* did the pushes work? */
2235cc0c5538SChris Mason 		if (btrfs_leaf_free_space(root, l) >= space_needed)
2236be0e5c09SChris Mason 			return 0;
22373326d1b0SChris Mason 	}
2238aa5d6bedSChris Mason 
22395c680ed6SChris Mason 	if (!path->nodes[1]) {
2240e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
22415c680ed6SChris Mason 		if (ret)
22425c680ed6SChris Mason 			return ret;
22435c680ed6SChris Mason 	}
2244cc0c5538SChris Mason again:
2245cc0c5538SChris Mason 	double_split = 0;
2246cc0c5538SChris Mason 	l = path->nodes[0];
2247eb60ceacSChris Mason 	slot = path->slots[0];
22485f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
2249eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
225054aa1f4dSChris Mason 
22517bb86316SChris Mason 	btrfs_item_key(l, &disk_key, 0);
22527bb86316SChris Mason 
2253925baeddSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
22547bb86316SChris Mason 					 root->root_key.objectid,
22557bb86316SChris Mason 					 root_gen, disk_key.objectid, 0,
2256db94535dSChris Mason 					 l->start, 0);
2257cea9e445SChris Mason 	if (IS_ERR(right)) {
2258cea9e445SChris Mason 		BUG_ON(1);
22595f39d397SChris Mason 		return PTR_ERR(right);
2260cea9e445SChris Mason 	}
226154aa1f4dSChris Mason 
22625f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2263db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
22645f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
22655f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
22665f39d397SChris Mason 	btrfs_set_header_level(right, 0);
22675f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
22685f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
22695f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2270e17cade2SChris Mason 
2271e17cade2SChris Mason 	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
2272e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
2273e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
2274d4dbff95SChris Mason 	if (mid <= slot) {
2275d4dbff95SChris Mason 		if (nritems == 1 ||
2276d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
2277d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2278d4dbff95SChris Mason 			if (slot >= nritems) {
2279d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
22805f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2281d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2282db94535dSChris Mason 						  &disk_key, right->start,
2283d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
2284d4dbff95SChris Mason 				if (wret)
2285d4dbff95SChris Mason 					ret = wret;
2286925baeddSChris Mason 
2287925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
22885f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
22895f39d397SChris Mason 				path->nodes[0] = right;
2290d4dbff95SChris Mason 				path->slots[0] = 0;
2291d4dbff95SChris Mason 				path->slots[1] += 1;
22920ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2293d4dbff95SChris Mason 				return ret;
2294d4dbff95SChris Mason 			}
2295d4dbff95SChris Mason 			mid = slot;
22963326d1b0SChris Mason 			if (mid != nritems &&
22973326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
22983326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2299d4dbff95SChris Mason 				double_split = 1;
2300d4dbff95SChris Mason 			}
23013326d1b0SChris Mason 		}
2302d4dbff95SChris Mason 	} else {
2303d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
2304d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2305cc0c5538SChris Mason 			if (!extend && slot == 0) {
2306d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
23075f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2308d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2309d4dbff95SChris Mason 						  &disk_key,
2310db94535dSChris Mason 						  right->start,
2311098f59c2SChris Mason 						  path->slots[1], 1);
2312d4dbff95SChris Mason 				if (wret)
2313d4dbff95SChris Mason 					ret = wret;
2314925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
23155f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
23165f39d397SChris Mason 				path->nodes[0] = right;
2317d4dbff95SChris Mason 				path->slots[0] = 0;
2318a429e513SChris Mason 				if (path->slots[1] == 0) {
2319a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
2320a429e513SChris Mason 					           path, &disk_key, 1);
2321a429e513SChris Mason 					if (wret)
2322a429e513SChris Mason 						ret = wret;
2323a429e513SChris Mason 				}
23240ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2325d4dbff95SChris Mason 				return ret;
2326cc0c5538SChris Mason 			} else if (extend && slot == 0) {
2327cc0c5538SChris Mason 				mid = 1;
2328cc0c5538SChris Mason 			} else {
2329d4dbff95SChris Mason 				mid = slot;
23305ee78ac7SChris Mason 				if (mid != nritems &&
23315ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
23325ee78ac7SChris Mason 				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2333d4dbff95SChris Mason 					double_split = 1;
2334d4dbff95SChris Mason 				}
2335d4dbff95SChris Mason 			}
23365ee78ac7SChris Mason 		}
2337cc0c5538SChris Mason 	}
23385f39d397SChris Mason 	nritems = nritems - mid;
23395f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
23405f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
23415f39d397SChris Mason 
23425f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
23435f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
23445f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
23455f39d397SChris Mason 
23465f39d397SChris Mason 	copy_extent_buffer(right, l,
2347d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2348123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
2349123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
235074123bd7SChris Mason 
23515f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
23525f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
23535f39d397SChris Mason 
23545f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
23555f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
2356db94535dSChris Mason 		u32 ioff;
2357db94535dSChris Mason 
2358db94535dSChris Mason 		if (!right->map_token) {
2359db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2360db94535dSChris Mason 					sizeof(struct btrfs_item),
2361db94535dSChris Mason 					&right->map_token, &right->kaddr,
2362db94535dSChris Mason 					&right->map_start, &right->map_len,
2363db94535dSChris Mason 					KM_USER1);
2364db94535dSChris Mason 		}
2365db94535dSChris Mason 
2366db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
23675f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
23680783fcfcSChris Mason 	}
236974123bd7SChris Mason 
2370db94535dSChris Mason 	if (right->map_token) {
2371db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2372db94535dSChris Mason 		right->map_token = NULL;
2373db94535dSChris Mason 	}
2374db94535dSChris Mason 
23755f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
2376aa5d6bedSChris Mason 	ret = 0;
23775f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
2378db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
2379db94535dSChris Mason 			  path->slots[1] + 1, 1);
2380aa5d6bedSChris Mason 	if (wret)
2381aa5d6bedSChris Mason 		ret = wret;
23825f39d397SChris Mason 
23835f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
23845f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
2385eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
23865f39d397SChris Mason 
2387be0e5c09SChris Mason 	if (mid <= slot) {
2388925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
23895f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
23905f39d397SChris Mason 		path->nodes[0] = right;
2391be0e5c09SChris Mason 		path->slots[0] -= mid;
2392be0e5c09SChris Mason 		path->slots[1] += 1;
2393925baeddSChris Mason 	} else {
2394925baeddSChris Mason 		btrfs_tree_unlock(right);
23955f39d397SChris Mason 		free_extent_buffer(right);
2396925baeddSChris Mason 	}
23975f39d397SChris Mason 
2398eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2399d4dbff95SChris Mason 
2400cc0c5538SChris Mason 	if (double_split) {
2401cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
2402cc0c5538SChris Mason 		num_doubles++;
2403cc0c5538SChris Mason 		goto again;
24043326d1b0SChris Mason 	}
2405be0e5c09SChris Mason 	return ret;
2406be0e5c09SChris Mason }
2407be0e5c09SChris Mason 
2408b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2409b18c6685SChris Mason 			struct btrfs_root *root,
2410b18c6685SChris Mason 			struct btrfs_path *path,
2411179e29e4SChris Mason 			u32 new_size, int from_end)
2412b18c6685SChris Mason {
2413b18c6685SChris Mason 	int ret = 0;
2414b18c6685SChris Mason 	int slot;
2415b18c6685SChris Mason 	int slot_orig;
24165f39d397SChris Mason 	struct extent_buffer *leaf;
24175f39d397SChris Mason 	struct btrfs_item *item;
2418b18c6685SChris Mason 	u32 nritems;
2419b18c6685SChris Mason 	unsigned int data_end;
2420b18c6685SChris Mason 	unsigned int old_data_start;
2421b18c6685SChris Mason 	unsigned int old_size;
2422b18c6685SChris Mason 	unsigned int size_diff;
2423b18c6685SChris Mason 	int i;
2424b18c6685SChris Mason 
2425b18c6685SChris Mason 	slot_orig = path->slots[0];
24265f39d397SChris Mason 	leaf = path->nodes[0];
2427179e29e4SChris Mason 	slot = path->slots[0];
2428179e29e4SChris Mason 
2429179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
2430179e29e4SChris Mason 	if (old_size == new_size)
2431179e29e4SChris Mason 		return 0;
2432b18c6685SChris Mason 
24335f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2434b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
2435b18c6685SChris Mason 
24365f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
2437179e29e4SChris Mason 
2438b18c6685SChris Mason 	size_diff = old_size - new_size;
2439b18c6685SChris Mason 
2440b18c6685SChris Mason 	BUG_ON(slot < 0);
2441b18c6685SChris Mason 	BUG_ON(slot >= nritems);
2442b18c6685SChris Mason 
2443b18c6685SChris Mason 	/*
2444b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2445b18c6685SChris Mason 	 */
2446b18c6685SChris Mason 	/* first correct the data pointers */
2447b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
24485f39d397SChris Mason 		u32 ioff;
24495f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2450db94535dSChris Mason 
2451db94535dSChris Mason 		if (!leaf->map_token) {
2452db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2453db94535dSChris Mason 					sizeof(struct btrfs_item),
2454db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2455db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2456db94535dSChris Mason 					KM_USER1);
2457db94535dSChris Mason 		}
2458db94535dSChris Mason 
24595f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
24605f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2461b18c6685SChris Mason 	}
2462db94535dSChris Mason 
2463db94535dSChris Mason 	if (leaf->map_token) {
2464db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2465db94535dSChris Mason 		leaf->map_token = NULL;
2466db94535dSChris Mason 	}
2467db94535dSChris Mason 
2468b18c6685SChris Mason 	/* shift the data */
2469179e29e4SChris Mason 	if (from_end) {
24705f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2471b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2472b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
2473179e29e4SChris Mason 	} else {
2474179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
2475179e29e4SChris Mason 		u64 offset;
2476179e29e4SChris Mason 
2477179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
2478179e29e4SChris Mason 
2479179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2480179e29e4SChris Mason 			unsigned long ptr;
2481179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
2482179e29e4SChris Mason 
2483179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
2484179e29e4SChris Mason 					    struct btrfs_file_extent_item);
2485179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
2486179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
2487179e29e4SChris Mason 
2488179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
2489179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
2490179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
2491179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
2492179e29e4SChris Mason 				        (unsigned long)fi,
2493179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
2494179e29e4SChris Mason 						 disk_bytenr));
2495179e29e4SChris Mason 			}
2496179e29e4SChris Mason 		}
2497179e29e4SChris Mason 
2498179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2499179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2500179e29e4SChris Mason 			      data_end, old_data_start - data_end);
2501179e29e4SChris Mason 
2502179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
2503179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2504179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
2505179e29e4SChris Mason 		if (slot == 0)
2506179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
2507179e29e4SChris Mason 	}
25085f39d397SChris Mason 
25095f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
25105f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
25115f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2512b18c6685SChris Mason 
2513b18c6685SChris Mason 	ret = 0;
25145f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
25155f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2516b18c6685SChris Mason 		BUG();
25175f39d397SChris Mason 	}
2518b18c6685SChris Mason 	return ret;
2519b18c6685SChris Mason }
2520b18c6685SChris Mason 
25215f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
25225f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
25235f39d397SChris Mason 		      u32 data_size)
25246567e837SChris Mason {
25256567e837SChris Mason 	int ret = 0;
25266567e837SChris Mason 	int slot;
25276567e837SChris Mason 	int slot_orig;
25285f39d397SChris Mason 	struct extent_buffer *leaf;
25295f39d397SChris Mason 	struct btrfs_item *item;
25306567e837SChris Mason 	u32 nritems;
25316567e837SChris Mason 	unsigned int data_end;
25326567e837SChris Mason 	unsigned int old_data;
25336567e837SChris Mason 	unsigned int old_size;
25346567e837SChris Mason 	int i;
25356567e837SChris Mason 
25366567e837SChris Mason 	slot_orig = path->slots[0];
25375f39d397SChris Mason 	leaf = path->nodes[0];
25386567e837SChris Mason 
25395f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
25406567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
25416567e837SChris Mason 
25425f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
25435f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25446567e837SChris Mason 		BUG();
25455f39d397SChris Mason 	}
25466567e837SChris Mason 	slot = path->slots[0];
25475f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
25486567e837SChris Mason 
25496567e837SChris Mason 	BUG_ON(slot < 0);
25503326d1b0SChris Mason 	if (slot >= nritems) {
25513326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
25523326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
25533326d1b0SChris Mason 		BUG_ON(1);
25543326d1b0SChris Mason 	}
25556567e837SChris Mason 
25566567e837SChris Mason 	/*
25576567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
25586567e837SChris Mason 	 */
25596567e837SChris Mason 	/* first correct the data pointers */
25606567e837SChris Mason 	for (i = slot; i < nritems; i++) {
25615f39d397SChris Mason 		u32 ioff;
25625f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2563db94535dSChris Mason 
2564db94535dSChris Mason 		if (!leaf->map_token) {
2565db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2566db94535dSChris Mason 					sizeof(struct btrfs_item),
2567db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2568db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2569db94535dSChris Mason 					KM_USER1);
2570db94535dSChris Mason 		}
25715f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
25725f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
25736567e837SChris Mason 	}
25745f39d397SChris Mason 
2575db94535dSChris Mason 	if (leaf->map_token) {
2576db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2577db94535dSChris Mason 		leaf->map_token = NULL;
2578db94535dSChris Mason 	}
2579db94535dSChris Mason 
25806567e837SChris Mason 	/* shift the data */
25815f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
25826567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
25836567e837SChris Mason 		      data_end, old_data - data_end);
25845f39d397SChris Mason 
25856567e837SChris Mason 	data_end = old_data;
25865f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
25875f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
25885f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
25895f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
25906567e837SChris Mason 
25916567e837SChris Mason 	ret = 0;
25925f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
25935f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25946567e837SChris Mason 		BUG();
25955f39d397SChris Mason 	}
25966567e837SChris Mason 	return ret;
25976567e837SChris Mason }
25986567e837SChris Mason 
259974123bd7SChris Mason /*
260074123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
260174123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
260274123bd7SChris Mason  */
26039c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
26045f39d397SChris Mason 			    struct btrfs_root *root,
26055f39d397SChris Mason 			    struct btrfs_path *path,
26069c58309dSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
26079c58309dSChris Mason 			    int nr)
2608be0e5c09SChris Mason {
26095f39d397SChris Mason 	struct extent_buffer *leaf;
26105f39d397SChris Mason 	struct btrfs_item *item;
2611aa5d6bedSChris Mason 	int ret = 0;
2612be0e5c09SChris Mason 	int slot;
2613eb60ceacSChris Mason 	int slot_orig;
26149c58309dSChris Mason 	int i;
26157518a238SChris Mason 	u32 nritems;
26169c58309dSChris Mason 	u32 total_size = 0;
26179c58309dSChris Mason 	u32 total_data = 0;
2618be0e5c09SChris Mason 	unsigned int data_end;
2619e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2620e2fa7227SChris Mason 
26219c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26229c58309dSChris Mason 		total_data += data_size[i];
26239c58309dSChris Mason 	}
2624be0e5c09SChris Mason 
26257b128766SJosef Bacik 	total_size = total_data + (nr * sizeof(struct btrfs_item));
26269c58309dSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2627eb60ceacSChris Mason 	if (ret == 0) {
2628f0930a37SChris Mason 		return -EEXIST;
2629aa5d6bedSChris Mason 	}
2630ed2ff2cbSChris Mason 	if (ret < 0)
2631ed2ff2cbSChris Mason 		goto out;
2632be0e5c09SChris Mason 
263362e2749eSChris Mason 	slot_orig = path->slots[0];
26345f39d397SChris Mason 	leaf = path->nodes[0];
263574123bd7SChris Mason 
26365f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2637123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2638eb60ceacSChris Mason 
2639123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
26409c58309dSChris Mason 	    sizeof(struct btrfs_item) + total_size) {
26413326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
26423326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
26439c58309dSChris Mason 		       total_size, btrfs_leaf_free_space(root, leaf));
2644be0e5c09SChris Mason 		BUG();
2645d4dbff95SChris Mason 	}
26465f39d397SChris Mason 
264762e2749eSChris Mason 	slot = path->slots[0];
2648eb60ceacSChris Mason 	BUG_ON(slot < 0);
26495f39d397SChris Mason 
2650be0e5c09SChris Mason 	if (slot != nritems) {
2651be0e5c09SChris Mason 		int i;
26525f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2653be0e5c09SChris Mason 
26545f39d397SChris Mason 		if (old_data < data_end) {
26555f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
26565f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
26575f39d397SChris Mason 			       slot, old_data, data_end);
26585f39d397SChris Mason 			BUG_ON(1);
26595f39d397SChris Mason 		}
2660be0e5c09SChris Mason 		/*
2661be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2662be0e5c09SChris Mason 		 */
2663be0e5c09SChris Mason 		/* first correct the data pointers */
2664db94535dSChris Mason 		WARN_ON(leaf->map_token);
26650783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
26665f39d397SChris Mason 			u32 ioff;
2667db94535dSChris Mason 
26685f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2669db94535dSChris Mason 			if (!leaf->map_token) {
2670db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2671db94535dSChris Mason 					sizeof(struct btrfs_item),
2672db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2673db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2674db94535dSChris Mason 					KM_USER1);
2675db94535dSChris Mason 			}
2676db94535dSChris Mason 
26775f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
26789c58309dSChris Mason 			btrfs_set_item_offset(leaf, item, ioff - total_data);
26790783fcfcSChris Mason 		}
2680db94535dSChris Mason 		if (leaf->map_token) {
2681db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2682db94535dSChris Mason 			leaf->map_token = NULL;
2683db94535dSChris Mason 		}
2684be0e5c09SChris Mason 
2685be0e5c09SChris Mason 		/* shift the items */
26869c58309dSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
26875f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
26880783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2689be0e5c09SChris Mason 
2690be0e5c09SChris Mason 		/* shift the data */
26915f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
26929c58309dSChris Mason 			      data_end - total_data, btrfs_leaf_data(leaf) +
2693be0e5c09SChris Mason 			      data_end, old_data - data_end);
2694be0e5c09SChris Mason 		data_end = old_data;
2695be0e5c09SChris Mason 	}
26965f39d397SChris Mason 
269762e2749eSChris Mason 	/* setup the item for the new data */
26989c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26999c58309dSChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
27009c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
27019c58309dSChris Mason 		item = btrfs_item_nr(leaf, slot + i);
27029c58309dSChris Mason 		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
27039c58309dSChris Mason 		data_end -= data_size[i];
27049c58309dSChris Mason 		btrfs_set_item_size(leaf, item, data_size[i]);
27059c58309dSChris Mason 	}
27069c58309dSChris Mason 	btrfs_set_header_nritems(leaf, nritems + nr);
27075f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2708aa5d6bedSChris Mason 
2709aa5d6bedSChris Mason 	ret = 0;
27105a01a2e3SChris Mason 	if (slot == 0) {
27115a01a2e3SChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2712e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
27135a01a2e3SChris Mason 	}
2714aa5d6bedSChris Mason 
27155f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
27165f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2717be0e5c09SChris Mason 		BUG();
27185f39d397SChris Mason 	}
2719ed2ff2cbSChris Mason out:
272062e2749eSChris Mason 	return ret;
272162e2749eSChris Mason }
272262e2749eSChris Mason 
272362e2749eSChris Mason /*
272462e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
272562e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
272662e2749eSChris Mason  */
2727e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2728e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2729e089f05cSChris Mason 		      data_size)
273062e2749eSChris Mason {
273162e2749eSChris Mason 	int ret = 0;
27322c90e5d6SChris Mason 	struct btrfs_path *path;
27335f39d397SChris Mason 	struct extent_buffer *leaf;
27345f39d397SChris Mason 	unsigned long ptr;
273562e2749eSChris Mason 
27362c90e5d6SChris Mason 	path = btrfs_alloc_path();
27372c90e5d6SChris Mason 	BUG_ON(!path);
27382c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
273962e2749eSChris Mason 	if (!ret) {
27405f39d397SChris Mason 		leaf = path->nodes[0];
27415f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
27425f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
27435f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
274462e2749eSChris Mason 	}
27452c90e5d6SChris Mason 	btrfs_free_path(path);
2746aa5d6bedSChris Mason 	return ret;
2747be0e5c09SChris Mason }
2748be0e5c09SChris Mason 
274974123bd7SChris Mason /*
27505de08d7dSChris Mason  * delete the pointer from a given node.
275174123bd7SChris Mason  *
275274123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
275374123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
275474123bd7SChris Mason  * a leaf if all the nodes are emptied.
275574123bd7SChris Mason  */
2756e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2757e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2758be0e5c09SChris Mason {
27595f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
27607518a238SChris Mason 	u32 nritems;
2761aa5d6bedSChris Mason 	int ret = 0;
2762bb803951SChris Mason 	int wret;
2763be0e5c09SChris Mason 
27645f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2765be0e5c09SChris Mason 	if (slot != nritems -1) {
27665f39d397SChris Mason 		memmove_extent_buffer(parent,
27675f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
27685f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2769d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2770d6025579SChris Mason 			      (nritems - slot - 1));
2771be0e5c09SChris Mason 	}
27727518a238SChris Mason 	nritems--;
27735f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
27747518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
27755f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2776eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
27775f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2778bb803951SChris Mason 	} else if (slot == 0) {
27795f39d397SChris Mason 		struct btrfs_disk_key disk_key;
27805f39d397SChris Mason 
27815f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
27825f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
27830f70abe2SChris Mason 		if (wret)
27840f70abe2SChris Mason 			ret = wret;
2785be0e5c09SChris Mason 	}
2786d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2787aa5d6bedSChris Mason 	return ret;
2788be0e5c09SChris Mason }
2789be0e5c09SChris Mason 
279074123bd7SChris Mason /*
279174123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
279274123bd7SChris Mason  * the leaf, remove it from the tree
279374123bd7SChris Mason  */
279485e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
279585e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
2796be0e5c09SChris Mason {
27975f39d397SChris Mason 	struct extent_buffer *leaf;
27985f39d397SChris Mason 	struct btrfs_item *item;
279985e21bacSChris Mason 	int last_off;
280085e21bacSChris Mason 	int dsize = 0;
2801aa5d6bedSChris Mason 	int ret = 0;
2802aa5d6bedSChris Mason 	int wret;
280385e21bacSChris Mason 	int i;
28047518a238SChris Mason 	u32 nritems;
2805be0e5c09SChris Mason 
28065f39d397SChris Mason 	leaf = path->nodes[0];
280785e21bacSChris Mason 	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
280885e21bacSChris Mason 
280985e21bacSChris Mason 	for (i = 0; i < nr; i++)
281085e21bacSChris Mason 		dsize += btrfs_item_size_nr(leaf, slot + i);
281185e21bacSChris Mason 
28125f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2813be0e5c09SChris Mason 
281485e21bacSChris Mason 	if (slot + nr != nritems) {
2815be0e5c09SChris Mason 		int i;
2816123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
28175f39d397SChris Mason 
28185f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2819d6025579SChris Mason 			      data_end + dsize,
2820123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
282185e21bacSChris Mason 			      last_off - data_end);
28225f39d397SChris Mason 
282385e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
28245f39d397SChris Mason 			u32 ioff;
2825db94535dSChris Mason 
28265f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2827db94535dSChris Mason 			if (!leaf->map_token) {
2828db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2829db94535dSChris Mason 					sizeof(struct btrfs_item),
2830db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2831db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2832db94535dSChris Mason 					KM_USER1);
2833db94535dSChris Mason 			}
28345f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
28355f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
28360783fcfcSChris Mason 		}
2837db94535dSChris Mason 
2838db94535dSChris Mason 		if (leaf->map_token) {
2839db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2840db94535dSChris Mason 			leaf->map_token = NULL;
2841db94535dSChris Mason 		}
2842db94535dSChris Mason 
28435f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
284485e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
28450783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
284685e21bacSChris Mason 			      (nritems - slot - nr));
2847be0e5c09SChris Mason 	}
284885e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
284985e21bacSChris Mason 	nritems -= nr;
28505f39d397SChris Mason 
285174123bd7SChris Mason 	/* delete the leaf if we've emptied it */
28527518a238SChris Mason 	if (nritems == 0) {
28535f39d397SChris Mason 		if (leaf == root->node) {
28545f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
28559a8dd150SChris Mason 		} else {
28567bb86316SChris Mason 			u64 root_gen = btrfs_header_generation(path->nodes[1]);
2857e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2858aa5d6bedSChris Mason 			if (wret)
2859aa5d6bedSChris Mason 				ret = wret;
2860e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
28617bb86316SChris Mason 					 leaf->start, leaf->len,
28627bb86316SChris Mason 					 btrfs_header_owner(path->nodes[1]),
28637bb86316SChris Mason 					 root_gen, 0, 0, 1);
28640f70abe2SChris Mason 			if (wret)
28650f70abe2SChris Mason 				ret = wret;
28669a8dd150SChris Mason 		}
2867be0e5c09SChris Mason 	} else {
28687518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2869aa5d6bedSChris Mason 		if (slot == 0) {
28705f39d397SChris Mason 			struct btrfs_disk_key disk_key;
28715f39d397SChris Mason 
28725f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2873e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
28745f39d397SChris Mason 					      &disk_key, 1);
2875aa5d6bedSChris Mason 			if (wret)
2876aa5d6bedSChris Mason 				ret = wret;
2877aa5d6bedSChris Mason 		}
2878aa5d6bedSChris Mason 
287974123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
288085e21bacSChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
2881be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2882be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2883be0e5c09SChris Mason 			 * for possible call to del_ptr below
2884be0e5c09SChris Mason 			 */
28854920c9acSChris Mason 			slot = path->slots[1];
28865f39d397SChris Mason 			extent_buffer_get(leaf);
28875f39d397SChris Mason 
288885e21bacSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1);
288954aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2890aa5d6bedSChris Mason 				ret = wret;
28915f39d397SChris Mason 
28925f39d397SChris Mason 			if (path->nodes[0] == leaf &&
28935f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
289485e21bacSChris Mason 				wret = push_leaf_right(trans, root, path, 1, 1);
289554aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2896aa5d6bedSChris Mason 					ret = wret;
2897aa5d6bedSChris Mason 			}
28985f39d397SChris Mason 
28995f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
29007bb86316SChris Mason 				u64 root_gen;
2901db94535dSChris Mason 				u64 bytenr = leaf->start;
2902db94535dSChris Mason 				u32 blocksize = leaf->len;
29035f39d397SChris Mason 
29047bb86316SChris Mason 				root_gen = btrfs_header_generation(
29057bb86316SChris Mason 							   path->nodes[1]);
29067bb86316SChris Mason 
2907e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2908aa5d6bedSChris Mason 				if (wret)
2909aa5d6bedSChris Mason 					ret = wret;
29105f39d397SChris Mason 
29115f39d397SChris Mason 				free_extent_buffer(leaf);
2912db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
29137bb86316SChris Mason 					     blocksize,
29147bb86316SChris Mason 					     btrfs_header_owner(path->nodes[1]),
29157bb86316SChris Mason 					     root_gen, 0, 0, 1);
29160f70abe2SChris Mason 				if (wret)
29170f70abe2SChris Mason 					ret = wret;
29185de08d7dSChris Mason 			} else {
2919925baeddSChris Mason 				/* if we're still in the path, make sure
2920925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
2921925baeddSChris Mason 				 * push_leaf functions must have already
2922925baeddSChris Mason 				 * dirtied this buffer
2923925baeddSChris Mason 				 */
2924925baeddSChris Mason 				if (path->nodes[0] == leaf)
29255f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
29265f39d397SChris Mason 				free_extent_buffer(leaf);
2927be0e5c09SChris Mason 			}
2928d5719762SChris Mason 		} else {
29295f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2930be0e5c09SChris Mason 		}
29319a8dd150SChris Mason 	}
2932aa5d6bedSChris Mason 	return ret;
29339a8dd150SChris Mason }
29349a8dd150SChris Mason 
293597571fd0SChris Mason /*
2936925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
29377bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
29387bb86316SChris Mason  * returns < 0 on io errors.
29397bb86316SChris Mason  */
29407bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
29417bb86316SChris Mason {
2942925baeddSChris Mason 	struct btrfs_key key;
2943925baeddSChris Mason 	struct btrfs_disk_key found_key;
2944925baeddSChris Mason 	int ret;
29457bb86316SChris Mason 
2946925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
2947925baeddSChris Mason 
2948925baeddSChris Mason 	if (key.offset > 0)
2949925baeddSChris Mason 		key.offset--;
2950925baeddSChris Mason 	else if (key.type > 0)
2951925baeddSChris Mason 		key.type--;
2952925baeddSChris Mason 	else if (key.objectid > 0)
2953925baeddSChris Mason 		key.objectid--;
2954925baeddSChris Mason 	else
29557bb86316SChris Mason 		return 1;
29567bb86316SChris Mason 
2957925baeddSChris Mason 	btrfs_release_path(root, path);
2958925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2959925baeddSChris Mason 	if (ret < 0)
2960925baeddSChris Mason 		return ret;
2961925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
2962925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
2963925baeddSChris Mason 	if (ret < 0)
29647bb86316SChris Mason 		return 0;
2965925baeddSChris Mason 	return 1;
29667bb86316SChris Mason }
29677bb86316SChris Mason 
29683f157a2fSChris Mason /*
29693f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
29703f157a2fSChris Mason  * for nodes or leaves that are either in cache or have a minimum
29713f157a2fSChris Mason  * transaction id.  This is used by the btree defrag code, but could
29723f157a2fSChris Mason  * also be used to search for blocks that have changed since a given
29733f157a2fSChris Mason  * transaction id.
29743f157a2fSChris Mason  *
29753f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
29763f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
29773f157a2fSChris Mason  * key and get a writable path.
29783f157a2fSChris Mason  *
29793f157a2fSChris Mason  * This does lock as it descends, and path->keep_locks should be set
29803f157a2fSChris Mason  * to 1 by the caller.
29813f157a2fSChris Mason  *
29823f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
29833f157a2fSChris Mason  * of the tree.
29843f157a2fSChris Mason  *
29853f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
29863f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
29873f157a2fSChris Mason  */
29883f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
29893f157a2fSChris Mason 			 struct btrfs_path *path, int cache_only,
29903f157a2fSChris Mason 			 u64 min_trans)
29913f157a2fSChris Mason {
29923f157a2fSChris Mason 	struct extent_buffer *cur;
29933f157a2fSChris Mason 	struct btrfs_key found_key;
29943f157a2fSChris Mason 	int slot;
29959652480bSYan 	int sret;
29963f157a2fSChris Mason 	u32 nritems;
29973f157a2fSChris Mason 	int level;
29983f157a2fSChris Mason 	int ret = 1;
29993f157a2fSChris Mason 
30003f157a2fSChris Mason again:
30013f157a2fSChris Mason 	cur = btrfs_lock_root_node(root);
30023f157a2fSChris Mason 	level = btrfs_header_level(cur);
30033f157a2fSChris Mason 	path->nodes[level] = cur;
30043f157a2fSChris Mason 	path->locks[level] = 1;
30053f157a2fSChris Mason 
30063f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
30073f157a2fSChris Mason 		ret = 1;
30083f157a2fSChris Mason 		goto out;
30093f157a2fSChris Mason 	}
30103f157a2fSChris Mason 	while(1) {
30113f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
30123f157a2fSChris Mason 		level = btrfs_header_level(cur);
30139652480bSYan 		sret = bin_search(cur, min_key, level, &slot);
30143f157a2fSChris Mason 
30153f157a2fSChris Mason 		/* at level = 0, we're done, setup the path and exit */
30163f157a2fSChris Mason 		if (level == 0) {
30173f157a2fSChris Mason 			ret = 0;
30183f157a2fSChris Mason 			path->slots[level] = slot;
30193f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
30203f157a2fSChris Mason 			goto out;
30213f157a2fSChris Mason 		}
30229652480bSYan 		if (sret && slot > 0)
30239652480bSYan 			slot--;
30243f157a2fSChris Mason 		/*
30253f157a2fSChris Mason 		 * check this node pointer against the cache_only and
30263f157a2fSChris Mason 		 * min_trans parameters.  If it isn't in cache or is too
30273f157a2fSChris Mason 		 * old, skip to the next one.
30283f157a2fSChris Mason 		 */
30293f157a2fSChris Mason 		while(slot < nritems) {
30303f157a2fSChris Mason 			u64 blockptr;
30313f157a2fSChris Mason 			u64 gen;
30323f157a2fSChris Mason 			struct extent_buffer *tmp;
30333f157a2fSChris Mason 			blockptr = btrfs_node_blockptr(cur, slot);
30343f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
30353f157a2fSChris Mason 			if (gen < min_trans) {
30363f157a2fSChris Mason 				slot++;
30373f157a2fSChris Mason 				continue;
30383f157a2fSChris Mason 			}
30393f157a2fSChris Mason 			if (!cache_only)
30403f157a2fSChris Mason 				break;
30413f157a2fSChris Mason 
30423f157a2fSChris Mason 			tmp = btrfs_find_tree_block(root, blockptr,
30433f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
30443f157a2fSChris Mason 
30453f157a2fSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
30463f157a2fSChris Mason 				free_extent_buffer(tmp);
30473f157a2fSChris Mason 				break;
30483f157a2fSChris Mason 			}
30493f157a2fSChris Mason 			if (tmp)
30503f157a2fSChris Mason 				free_extent_buffer(tmp);
30513f157a2fSChris Mason 			slot++;
30523f157a2fSChris Mason 		}
30533f157a2fSChris Mason 		/*
30543f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
30553f157a2fSChris Mason 		 * and find another one
30563f157a2fSChris Mason 		 */
30573f157a2fSChris Mason 		if (slot >= nritems) {
30583f157a2fSChris Mason 			ret = btrfs_find_next_key(root, path, min_key, level,
30593f157a2fSChris Mason 						  cache_only, min_trans);
30603f157a2fSChris Mason 			if (ret == 0) {
30613f157a2fSChris Mason 				btrfs_release_path(root, path);
30623f157a2fSChris Mason 				goto again;
30633f157a2fSChris Mason 			} else {
30643f157a2fSChris Mason 				goto out;
30653f157a2fSChris Mason 			}
30663f157a2fSChris Mason 		}
30673f157a2fSChris Mason 		/* save our key for returning back */
30683f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
30693f157a2fSChris Mason 		path->slots[level] = slot;
30703f157a2fSChris Mason 		if (level == path->lowest_level) {
30713f157a2fSChris Mason 			ret = 0;
30723f157a2fSChris Mason 			unlock_up(path, level, 1);
30733f157a2fSChris Mason 			goto out;
30743f157a2fSChris Mason 		}
30753f157a2fSChris Mason 		cur = read_node_slot(root, cur, slot);
30763f157a2fSChris Mason 
30773f157a2fSChris Mason 		btrfs_tree_lock(cur);
30783f157a2fSChris Mason 		path->locks[level - 1] = 1;
30793f157a2fSChris Mason 		path->nodes[level - 1] = cur;
30803f157a2fSChris Mason 		unlock_up(path, level, 1);
30813f157a2fSChris Mason 	}
30823f157a2fSChris Mason out:
30833f157a2fSChris Mason 	if (ret == 0)
30843f157a2fSChris Mason 		memcpy(min_key, &found_key, sizeof(found_key));
30853f157a2fSChris Mason 	return ret;
30863f157a2fSChris Mason }
30873f157a2fSChris Mason 
30883f157a2fSChris Mason /*
30893f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
30903f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
30913f157a2fSChris Mason  * tree based on the current path and the cache_only and min_trans
30923f157a2fSChris Mason  * parameters.
30933f157a2fSChris Mason  *
30943f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
30953f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
30963f157a2fSChris Mason  *
30973f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
30983f157a2fSChris Mason  * calling this function.
30993f157a2fSChris Mason  */
3100e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
31013f157a2fSChris Mason 			struct btrfs_key *key, int lowest_level,
31023f157a2fSChris Mason 			int cache_only, u64 min_trans)
3103e7a84565SChris Mason {
3104e7a84565SChris Mason 	int level = lowest_level;
3105e7a84565SChris Mason 	int slot;
3106e7a84565SChris Mason 	struct extent_buffer *c;
3107e7a84565SChris Mason 
3108e7a84565SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3109e7a84565SChris Mason 		if (!path->nodes[level])
3110e7a84565SChris Mason 			return 1;
3111e7a84565SChris Mason 
3112e7a84565SChris Mason 		slot = path->slots[level] + 1;
3113e7a84565SChris Mason 		c = path->nodes[level];
31143f157a2fSChris Mason next:
3115e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3116e7a84565SChris Mason 			level++;
3117e7a84565SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
3118e7a84565SChris Mason 				return 1;
3119e7a84565SChris Mason 			}
3120e7a84565SChris Mason 			continue;
3121e7a84565SChris Mason 		}
3122e7a84565SChris Mason 		if (level == 0)
3123e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
31243f157a2fSChris Mason 		else {
31253f157a2fSChris Mason 			u64 blockptr = btrfs_node_blockptr(c, slot);
31263f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
31273f157a2fSChris Mason 
31283f157a2fSChris Mason 			if (cache_only) {
31293f157a2fSChris Mason 				struct extent_buffer *cur;
31303f157a2fSChris Mason 				cur = btrfs_find_tree_block(root, blockptr,
31313f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
31323f157a2fSChris Mason 				if (!cur || !btrfs_buffer_uptodate(cur, gen)) {
31333f157a2fSChris Mason 					slot++;
31343f157a2fSChris Mason 					if (cur)
31353f157a2fSChris Mason 						free_extent_buffer(cur);
31363f157a2fSChris Mason 					goto next;
31373f157a2fSChris Mason 				}
31383f157a2fSChris Mason 				free_extent_buffer(cur);
31393f157a2fSChris Mason 			}
31403f157a2fSChris Mason 			if (gen < min_trans) {
31413f157a2fSChris Mason 				slot++;
31423f157a2fSChris Mason 				goto next;
31433f157a2fSChris Mason 			}
3144e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
31453f157a2fSChris Mason 		}
3146e7a84565SChris Mason 		return 0;
3147e7a84565SChris Mason 	}
3148e7a84565SChris Mason 	return 1;
3149e7a84565SChris Mason }
3150e7a84565SChris Mason 
31517bb86316SChris Mason /*
3152925baeddSChris Mason  * search the tree again to find a leaf with greater keys
31530f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
31540f70abe2SChris Mason  * returns < 0 on io errors.
315597571fd0SChris Mason  */
3156234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
3157d97e63b6SChris Mason {
3158d97e63b6SChris Mason 	int slot;
3159d97e63b6SChris Mason 	int level = 1;
31605f39d397SChris Mason 	struct extent_buffer *c;
31615f39d397SChris Mason 	struct extent_buffer *next = NULL;
3162925baeddSChris Mason 	struct btrfs_key key;
3163925baeddSChris Mason 	u32 nritems;
3164925baeddSChris Mason 	int ret;
3165925baeddSChris Mason 
3166925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3167925baeddSChris Mason 	if (nritems == 0) {
3168925baeddSChris Mason 		return 1;
3169925baeddSChris Mason 	}
3170925baeddSChris Mason 
3171925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
3172925baeddSChris Mason 
3173925baeddSChris Mason 	btrfs_release_path(root, path);
3174a2135011SChris Mason 	path->keep_locks = 1;
3175925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
3176925baeddSChris Mason 	path->keep_locks = 0;
3177925baeddSChris Mason 
3178925baeddSChris Mason 	if (ret < 0)
3179925baeddSChris Mason 		return ret;
3180925baeddSChris Mason 
3181a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3182168fd7d2SChris Mason 	/*
3183168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
3184168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
3185168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
3186168fd7d2SChris Mason 	 * advance the path if there are now more items available.
3187168fd7d2SChris Mason 	 */
3188a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
3189168fd7d2SChris Mason 		path->slots[0]++;
3190925baeddSChris Mason 		goto done;
3191925baeddSChris Mason 	}
3192d97e63b6SChris Mason 
3193234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3194d97e63b6SChris Mason 		if (!path->nodes[level])
31950f70abe2SChris Mason 			return 1;
31965f39d397SChris Mason 
3197d97e63b6SChris Mason 		slot = path->slots[level] + 1;
3198d97e63b6SChris Mason 		c = path->nodes[level];
31995f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3200d97e63b6SChris Mason 			level++;
3201925baeddSChris Mason 			if (level == BTRFS_MAX_LEVEL) {
32027bb86316SChris Mason 				return 1;
3203925baeddSChris Mason 			}
3204d97e63b6SChris Mason 			continue;
3205d97e63b6SChris Mason 		}
32065f39d397SChris Mason 
3207925baeddSChris Mason 		if (next) {
3208925baeddSChris Mason 			btrfs_tree_unlock(next);
32095f39d397SChris Mason 			free_extent_buffer(next);
3210925baeddSChris Mason 		}
32115f39d397SChris Mason 
32120bd40a71SChris Mason 		if (level == 1 && (path->locks[1] || path->skip_locking) &&
32130bd40a71SChris Mason 		    path->reada)
321401f46658SChris Mason 			reada_for_search(root, path, level, slot, 0);
32155f39d397SChris Mason 
3216ca7a79adSChris Mason 		next = read_node_slot(root, c, slot);
32175cd57b2cSChris Mason 		if (!path->skip_locking) {
3218051e1b9fSChris Mason 			WARN_ON(!btrfs_tree_locked(c));
3219925baeddSChris Mason 			btrfs_tree_lock(next);
32205cd57b2cSChris Mason 		}
3221d97e63b6SChris Mason 		break;
3222d97e63b6SChris Mason 	}
3223d97e63b6SChris Mason 	path->slots[level] = slot;
3224d97e63b6SChris Mason 	while(1) {
3225d97e63b6SChris Mason 		level--;
3226d97e63b6SChris Mason 		c = path->nodes[level];
3227925baeddSChris Mason 		if (path->locks[level])
3228925baeddSChris Mason 			btrfs_tree_unlock(c);
32295f39d397SChris Mason 		free_extent_buffer(c);
3230d97e63b6SChris Mason 		path->nodes[level] = next;
3231d97e63b6SChris Mason 		path->slots[level] = 0;
3232a74a4b97SChris Mason 		if (!path->skip_locking)
3233925baeddSChris Mason 			path->locks[level] = 1;
3234d97e63b6SChris Mason 		if (!level)
3235d97e63b6SChris Mason 			break;
3236925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
3237925baeddSChris Mason 			reada_for_search(root, path, level, slot, 0);
3238ca7a79adSChris Mason 		next = read_node_slot(root, next, 0);
32395cd57b2cSChris Mason 		if (!path->skip_locking) {
3240a2135011SChris Mason 			WARN_ON(!btrfs_tree_locked(path->nodes[level]));
3241925baeddSChris Mason 			btrfs_tree_lock(next);
3242d97e63b6SChris Mason 		}
32435cd57b2cSChris Mason 	}
3244925baeddSChris Mason done:
3245925baeddSChris Mason 	unlock_up(path, 0, 1);
3246d97e63b6SChris Mason 	return 0;
3247d97e63b6SChris Mason }
32480b86a832SChris Mason 
32493f157a2fSChris Mason /*
32503f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
32513f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
32523f157a2fSChris Mason  *
32533f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
32543f157a2fSChris Mason  */
32550b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
32560b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
32570b86a832SChris Mason 			int type)
32580b86a832SChris Mason {
32590b86a832SChris Mason 	struct btrfs_key found_key;
32600b86a832SChris Mason 	struct extent_buffer *leaf;
32610b86a832SChris Mason 	int ret;
32620b86a832SChris Mason 
32630b86a832SChris Mason 	while(1) {
32640b86a832SChris Mason 		if (path->slots[0] == 0) {
32650b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
32660b86a832SChris Mason 			if (ret != 0)
32670b86a832SChris Mason 				return ret;
32680b86a832SChris Mason 		} else {
32690b86a832SChris Mason 			path->slots[0]--;
32700b86a832SChris Mason 		}
32710b86a832SChris Mason 		leaf = path->nodes[0];
32720b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
32730b86a832SChris Mason 		if (found_key.type == type)
32740b86a832SChris Mason 			return 0;
32750b86a832SChris Mason 	}
32760b86a832SChris Mason 	return 1;
32770b86a832SChris Mason }
32780b86a832SChris Mason 
3279