xref: /openbmc/linux/fs/btrfs/ctree.c (revision 3f157a2fd2ad731e1ed9964fecdc5f459f04a4a4)
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++) {
68*3f157a2fSChris 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]);
76*3f157a2fSChris 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 	WARN_ON(root == root->fs_info->chunk_root &&
1255925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->chunk_mutex));
1256925baeddSChris Mason 	WARN_ON(root == root->fs_info->dev_root &&
1257925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->chunk_mutex));
1258925baeddSChris Mason 	if (ins_len < 0)
1259925baeddSChris Mason 		lowest_unlock = 2;
1260bb803951SChris Mason again:
12615cd57b2cSChris Mason 	if (p->skip_locking)
12625cd57b2cSChris Mason 		b = btrfs_root_node(root);
12635cd57b2cSChris Mason 	else
1264925baeddSChris Mason 		b = btrfs_lock_root_node(root);
1265925baeddSChris Mason 
1266eb60ceacSChris Mason 	while (b) {
12675f39d397SChris Mason 		level = btrfs_header_level(b);
126802217ed2SChris Mason 		if (cow) {
126902217ed2SChris Mason 			int wret;
1270e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
1271e20d96d6SChris Mason 					       p->nodes[level + 1],
1272e20d96d6SChris Mason 					       p->slots[level + 1],
1273252c38f0SYan 					       &b);
127454aa1f4dSChris Mason 			if (wret) {
12755f39d397SChris Mason 				free_extent_buffer(b);
127654aa1f4dSChris Mason 				return wret;
127754aa1f4dSChris Mason 			}
127802217ed2SChris Mason 		}
127902217ed2SChris Mason 		BUG_ON(!cow && ins_len);
12805f39d397SChris Mason 		if (level != btrfs_header_level(b))
12812c90e5d6SChris Mason 			WARN_ON(1);
12825f39d397SChris Mason 		level = btrfs_header_level(b);
1283eb60ceacSChris Mason 		p->nodes[level] = b;
12845cd57b2cSChris Mason 		if (!p->skip_locking)
1285925baeddSChris Mason 			p->locks[level] = 1;
1286123abc88SChris Mason 		ret = check_block(root, p, level);
1287aa5d6bedSChris Mason 		if (ret)
1288aa5d6bedSChris Mason 			return -1;
1289925baeddSChris Mason 
12905f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
12915f39d397SChris Mason 		if (level != 0) {
1292be0e5c09SChris Mason 			if (ret && slot > 0)
1293be0e5c09SChris Mason 				slot -= 1;
1294be0e5c09SChris Mason 			p->slots[level] = slot;
12955f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
12961514794eSChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1297e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
12985c680ed6SChris Mason 				BUG_ON(sret > 0);
12995c680ed6SChris Mason 				if (sret)
13005c680ed6SChris Mason 					return sret;
13015c680ed6SChris Mason 				b = p->nodes[level];
13025c680ed6SChris Mason 				slot = p->slots[level];
1303bb803951SChris Mason 			} else if (ins_len < 0) {
1304e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1305e089f05cSChris Mason 							 level);
1306bb803951SChris Mason 				if (sret)
1307bb803951SChris Mason 					return sret;
1308bb803951SChris Mason 				b = p->nodes[level];
1309f510cfecSChris Mason 				if (!b) {
1310f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1311bb803951SChris Mason 					goto again;
1312f510cfecSChris Mason 				}
1313bb803951SChris Mason 				slot = p->slots[level];
13145f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
13155c680ed6SChris Mason 			}
13169f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
1317925baeddSChris Mason 			if (level == lowest_level) {
1318925baeddSChris Mason 				unlock_up(p, level, lowest_unlock);
13199f3a7427SChris Mason 				break;
1320925baeddSChris Mason 			}
1321ca7a79adSChris Mason 
13226702ed49SChris Mason 			if (should_reada)
132301f46658SChris Mason 				reada_for_search(root, p, level, slot,
132401f46658SChris Mason 						 key->objectid);
1325ca7a79adSChris Mason 
1326594a24ebSChris Mason 			blocknr = btrfs_node_blockptr(b, slot);
1327594a24ebSChris Mason 			gen = btrfs_node_ptr_generation(b, slot);
1328594a24ebSChris Mason 			blocksize = btrfs_level_size(root, level - 1);
1329594a24ebSChris Mason 
1330594a24ebSChris Mason 			tmp = btrfs_find_tree_block(root, blocknr, blocksize);
1331594a24ebSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
1332051e1b9fSChris Mason 				b = tmp;
1333051e1b9fSChris Mason 			} else {
1334051e1b9fSChris Mason 				/*
1335051e1b9fSChris Mason 				 * reduce lock contention at high levels
1336051e1b9fSChris Mason 				 * of the btree by dropping locks before
1337051e1b9fSChris Mason 				 * we read.
1338051e1b9fSChris Mason 				 */
1339051e1b9fSChris Mason 				if (level > 1) {
1340051e1b9fSChris Mason 					btrfs_release_path(NULL, p);
1341051e1b9fSChris Mason 					if (tmp)
1342051e1b9fSChris Mason 						free_extent_buffer(tmp);
1343594a24ebSChris Mason 					tmp = read_tree_block(root, blocknr,
1344594a24ebSChris Mason 							 blocksize, gen);
1345594a24ebSChris Mason 					if (tmp)
1346594a24ebSChris Mason 						free_extent_buffer(tmp);
1347051e1b9fSChris Mason 					goto again;
1348051e1b9fSChris Mason 				} else {
1349a74a4b97SChris Mason 					if (tmp)
1350a74a4b97SChris Mason 						free_extent_buffer(tmp);
1351ca7a79adSChris Mason 					b = read_node_slot(root, b, slot);
1352051e1b9fSChris Mason 				}
1353051e1b9fSChris Mason 			}
13545cd57b2cSChris Mason 			if (!p->skip_locking)
1355925baeddSChris Mason 				btrfs_tree_lock(b);
1356051e1b9fSChris Mason 			unlock_up(p, level, lowest_unlock);
1357be0e5c09SChris Mason 		} else {
1358be0e5c09SChris Mason 			p->slots[level] = slot;
13595f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
13600783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1361d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1362cc0c5538SChris Mason 						      p, ins_len, ret == 0);
13635c680ed6SChris Mason 				BUG_ON(sret > 0);
13645c680ed6SChris Mason 				if (sret)
13655c680ed6SChris Mason 					return sret;
13665c680ed6SChris Mason 			}
1367925baeddSChris Mason 			unlock_up(p, level, lowest_unlock);
1368be0e5c09SChris Mason 			return ret;
1369be0e5c09SChris Mason 		}
1370be0e5c09SChris Mason 	}
1371aa5d6bedSChris Mason 	return 1;
1372be0e5c09SChris Mason }
1373be0e5c09SChris Mason 
137474123bd7SChris Mason /*
137574123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
137674123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
137774123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
137874123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
137974123bd7SChris Mason  * higher levels
1380aa5d6bedSChris Mason  *
1381aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1382aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
138374123bd7SChris Mason  */
13845f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
13855f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
13865f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1387be0e5c09SChris Mason {
1388be0e5c09SChris Mason 	int i;
1389aa5d6bedSChris Mason 	int ret = 0;
13905f39d397SChris Mason 	struct extent_buffer *t;
13915f39d397SChris Mason 
1392234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1393be0e5c09SChris Mason 		int tslot = path->slots[i];
1394eb60ceacSChris Mason 		if (!path->nodes[i])
1395be0e5c09SChris Mason 			break;
13965f39d397SChris Mason 		t = path->nodes[i];
13975f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1398d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1399be0e5c09SChris Mason 		if (tslot != 0)
1400be0e5c09SChris Mason 			break;
1401be0e5c09SChris Mason 	}
1402aa5d6bedSChris Mason 	return ret;
1403be0e5c09SChris Mason }
1404be0e5c09SChris Mason 
140574123bd7SChris Mason /*
140674123bd7SChris Mason  * try to push data from one node into the next node left in the
140779f95c82SChris Mason  * tree.
1408aa5d6bedSChris Mason  *
1409aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1410aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
141174123bd7SChris Mason  */
141298ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
141398ed5174SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
1414971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
1415be0e5c09SChris Mason {
1416be0e5c09SChris Mason 	int push_items = 0;
1417bb803951SChris Mason 	int src_nritems;
1418bb803951SChris Mason 	int dst_nritems;
1419aa5d6bedSChris Mason 	int ret = 0;
1420be0e5c09SChris Mason 
14215f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14225f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1423123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
14247bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14257bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
142654aa1f4dSChris Mason 
1427bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
1428971a1f66SChris Mason 		return 1;
1429971a1f66SChris Mason 
1430eb60ceacSChris Mason 	if (push_items <= 0) {
1431be0e5c09SChris Mason 		return 1;
1432eb60ceacSChris Mason 	}
1433be0e5c09SChris Mason 
1434bce4eae9SChris Mason 	if (empty) {
1435971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
1436bce4eae9SChris Mason 		if (push_items < src_nritems) {
1437bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
1438bce4eae9SChris Mason 			 * we aren't going to empty it
1439bce4eae9SChris Mason 			 */
1440bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
1441bce4eae9SChris Mason 				if (push_items <= 8)
1442bce4eae9SChris Mason 					return 1;
1443bce4eae9SChris Mason 				push_items -= 8;
1444bce4eae9SChris Mason 			}
1445bce4eae9SChris Mason 		}
1446bce4eae9SChris Mason 	} else
1447bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
144879f95c82SChris Mason 
14495f39d397SChris Mason 	copy_extent_buffer(dst, src,
14505f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
14515f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1452123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
14535f39d397SChris Mason 
1454bb803951SChris Mason 	if (push_items < src_nritems) {
14555f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
14565f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1457e2fa7227SChris Mason 				      (src_nritems - push_items) *
1458123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1459bb803951SChris Mason 	}
14605f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
14615f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
14625f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
14635f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1464bb803951SChris Mason 	return ret;
1465be0e5c09SChris Mason }
1466be0e5c09SChris Mason 
146797571fd0SChris Mason /*
146879f95c82SChris Mason  * try to push data from one node into the next node right in the
146979f95c82SChris Mason  * tree.
147079f95c82SChris Mason  *
147179f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
147279f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
147379f95c82SChris Mason  *
147479f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
147579f95c82SChris Mason  */
14765f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
14775f39d397SChris Mason 			      struct btrfs_root *root,
14785f39d397SChris Mason 			      struct extent_buffer *dst,
14795f39d397SChris Mason 			      struct extent_buffer *src)
148079f95c82SChris Mason {
148179f95c82SChris Mason 	int push_items = 0;
148279f95c82SChris Mason 	int max_push;
148379f95c82SChris Mason 	int src_nritems;
148479f95c82SChris Mason 	int dst_nritems;
148579f95c82SChris Mason 	int ret = 0;
148679f95c82SChris Mason 
14877bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14887bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
14897bb86316SChris Mason 
14905f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14915f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1492123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1493bce4eae9SChris Mason 	if (push_items <= 0) {
149479f95c82SChris Mason 		return 1;
1495bce4eae9SChris Mason 	}
1496bce4eae9SChris Mason 
1497bce4eae9SChris Mason 	if (src_nritems < 4) {
1498bce4eae9SChris Mason 		return 1;
1499bce4eae9SChris Mason 	}
150079f95c82SChris Mason 
150179f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
150279f95c82SChris Mason 	/* don't try to empty the node */
1503bce4eae9SChris Mason 	if (max_push >= src_nritems) {
150479f95c82SChris Mason 		return 1;
1505bce4eae9SChris Mason 	}
1506252c38f0SYan 
150779f95c82SChris Mason 	if (max_push < push_items)
150879f95c82SChris Mason 		push_items = max_push;
150979f95c82SChris Mason 
15105f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
15115f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
15125f39d397SChris Mason 				      (dst_nritems) *
15135f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1514d6025579SChris Mason 
15155f39d397SChris Mason 	copy_extent_buffer(dst, src,
15165f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
15175f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1518123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
151979f95c82SChris Mason 
15205f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
15215f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
152279f95c82SChris Mason 
15235f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
15245f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
152579f95c82SChris Mason 	return ret;
152679f95c82SChris Mason }
152779f95c82SChris Mason 
152879f95c82SChris Mason /*
152997571fd0SChris Mason  * helper function to insert a new root level in the tree.
153097571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
153197571fd0SChris Mason  * point to the existing root
1532aa5d6bedSChris Mason  *
1533aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
153497571fd0SChris Mason  */
153598ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans,
15365f39d397SChris Mason 			   struct btrfs_root *root,
15375f39d397SChris Mason 			   struct btrfs_path *path, int level)
153874123bd7SChris Mason {
15397bb86316SChris Mason 	u64 root_gen;
15407bb86316SChris Mason 	u64 lower_gen;
15415f39d397SChris Mason 	struct extent_buffer *lower;
15425f39d397SChris Mason 	struct extent_buffer *c;
1543925baeddSChris Mason 	struct extent_buffer *old;
15445f39d397SChris Mason 	struct btrfs_disk_key lower_key;
15455c680ed6SChris Mason 
15465c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
15475c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
15485c680ed6SChris Mason 
15497bb86316SChris Mason 	if (root->ref_cows)
15507bb86316SChris Mason 		root_gen = trans->transid;
15517bb86316SChris Mason 	else
15527bb86316SChris Mason 		root_gen = 0;
15537bb86316SChris Mason 
15547bb86316SChris Mason 	lower = path->nodes[level-1];
15557bb86316SChris Mason 	if (level == 1)
15567bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
15577bb86316SChris Mason 	else
15587bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
15597bb86316SChris Mason 
1560925baeddSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
15617bb86316SChris Mason 				   root->root_key.objectid,
15627bb86316SChris Mason 				   root_gen, lower_key.objectid, level,
1563db94535dSChris Mason 				   root->node->start, 0);
15645f39d397SChris Mason 	if (IS_ERR(c))
15655f39d397SChris Mason 		return PTR_ERR(c);
1566925baeddSChris Mason 
15675f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
15685f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
15695f39d397SChris Mason 	btrfs_set_header_level(c, level);
1570db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
15715f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
15725f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
1573d5719762SChris Mason 
15745f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
15755f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
15765f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1577e17cade2SChris Mason 
1578e17cade2SChris Mason 	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
1579e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
1580e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
1581e17cade2SChris Mason 
15825f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1583db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
15847bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
15857bb86316SChris Mason 	WARN_ON(lower_gen == 0);
15867bb86316SChris Mason 
15877bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
15885f39d397SChris Mason 
15895f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1590d5719762SChris Mason 
1591925baeddSChris Mason 	spin_lock(&root->node_lock);
1592925baeddSChris Mason 	old = root->node;
15935f39d397SChris Mason 	root->node = c;
1594925baeddSChris Mason 	spin_unlock(&root->node_lock);
1595925baeddSChris Mason 
1596925baeddSChris Mason 	/* the super has an extra ref to root->node */
1597925baeddSChris Mason 	free_extent_buffer(old);
1598925baeddSChris Mason 
15990b86a832SChris Mason 	add_root_to_dirty_list(root);
16005f39d397SChris Mason 	extent_buffer_get(c);
16015f39d397SChris Mason 	path->nodes[level] = c;
1602925baeddSChris Mason 	path->locks[level] = 1;
160374123bd7SChris Mason 	path->slots[level] = 0;
16047bb86316SChris Mason 
16057bb86316SChris Mason 	if (root->ref_cows && lower_gen != trans->transid) {
16067bb86316SChris Mason 		struct btrfs_path *back_path = btrfs_alloc_path();
16077bb86316SChris Mason 		int ret;
1608925baeddSChris Mason 		mutex_lock(&root->fs_info->alloc_mutex);
16097bb86316SChris Mason 		ret = btrfs_insert_extent_backref(trans,
16107bb86316SChris Mason 						  root->fs_info->extent_root,
16117bb86316SChris Mason 						  path, lower->start,
16127bb86316SChris Mason 						  root->root_key.objectid,
16137bb86316SChris Mason 						  trans->transid, 0, 0);
16147bb86316SChris Mason 		BUG_ON(ret);
1615925baeddSChris Mason 		mutex_unlock(&root->fs_info->alloc_mutex);
16167bb86316SChris Mason 		btrfs_free_path(back_path);
16177bb86316SChris Mason 	}
161874123bd7SChris Mason 	return 0;
161974123bd7SChris Mason }
16205c680ed6SChris Mason 
16215c680ed6SChris Mason /*
16225c680ed6SChris Mason  * worker function to insert a single pointer in a node.
16235c680ed6SChris Mason  * the node should have enough room for the pointer already
162497571fd0SChris Mason  *
16255c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
16265c680ed6SChris Mason  * blocknr is the block the key points to.
1627aa5d6bedSChris Mason  *
1628aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
16295c680ed6SChris Mason  */
1630e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1631e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1632db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
16335c680ed6SChris Mason {
16345f39d397SChris Mason 	struct extent_buffer *lower;
16355c680ed6SChris Mason 	int nritems;
16365c680ed6SChris Mason 
16375c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
16385f39d397SChris Mason 	lower = path->nodes[level];
16395f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
164074123bd7SChris Mason 	if (slot > nritems)
164174123bd7SChris Mason 		BUG();
1642123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
164374123bd7SChris Mason 		BUG();
164474123bd7SChris Mason 	if (slot != nritems) {
16455f39d397SChris Mason 		memmove_extent_buffer(lower,
16465f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
16475f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1648123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
164974123bd7SChris Mason 	}
16505f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1651db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
165274493f7aSChris Mason 	WARN_ON(trans->transid == 0);
165374493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
16545f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
16555f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
165674123bd7SChris Mason 	return 0;
165774123bd7SChris Mason }
165874123bd7SChris Mason 
165997571fd0SChris Mason /*
166097571fd0SChris Mason  * split the node at the specified level in path in two.
166197571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
166297571fd0SChris Mason  *
166397571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
166497571fd0SChris Mason  * left and right, if either one works, it returns right away.
1665aa5d6bedSChris Mason  *
1666aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
166797571fd0SChris Mason  */
1668e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1669e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1670be0e5c09SChris Mason {
16717bb86316SChris Mason 	u64 root_gen;
16725f39d397SChris Mason 	struct extent_buffer *c;
16735f39d397SChris Mason 	struct extent_buffer *split;
16745f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1675be0e5c09SChris Mason 	int mid;
16765c680ed6SChris Mason 	int ret;
1677aa5d6bedSChris Mason 	int wret;
16787518a238SChris Mason 	u32 c_nritems;
1679be0e5c09SChris Mason 
16805f39d397SChris Mason 	c = path->nodes[level];
16817bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
16825f39d397SChris Mason 	if (c == root->node) {
16835c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1684e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
16855c680ed6SChris Mason 		if (ret)
16865c680ed6SChris Mason 			return ret;
1687e66f709bSChris Mason 	} else {
1688e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
16895f39d397SChris Mason 		c = path->nodes[level];
16905f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1691c448acf0SChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
1692e66f709bSChris Mason 			return 0;
169354aa1f4dSChris Mason 		if (ret < 0)
169454aa1f4dSChris Mason 			return ret;
16955c680ed6SChris Mason 	}
1696e66f709bSChris Mason 
16975f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
16987bb86316SChris Mason 	if (root->ref_cows)
16997bb86316SChris Mason 		root_gen = trans->transid;
17007bb86316SChris Mason 	else
17017bb86316SChris Mason 		root_gen = 0;
17027bb86316SChris Mason 
17037bb86316SChris Mason 	btrfs_node_key(c, &disk_key, 0);
1704925baeddSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
17057bb86316SChris Mason 					 root->root_key.objectid,
17067bb86316SChris Mason 					 root_gen,
17077bb86316SChris Mason 					 btrfs_disk_key_objectid(&disk_key),
17087bb86316SChris Mason 					 level, c->start, 0);
17095f39d397SChris Mason 	if (IS_ERR(split))
17105f39d397SChris Mason 		return PTR_ERR(split);
171154aa1f4dSChris Mason 
17125f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
17135f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1714db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
17155f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
17165f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
171763b10fc4SChris Mason 	btrfs_set_header_flags(split, 0);
17185f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
17195f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
17205f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1721e17cade2SChris Mason 	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
1722e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
1723e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
17245f39d397SChris Mason 
17257518a238SChris Mason 	mid = (c_nritems + 1) / 2;
17265f39d397SChris Mason 
17275f39d397SChris Mason 	copy_extent_buffer(split, c,
17285f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
17295f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1730123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
17315f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
17325f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1733aa5d6bedSChris Mason 	ret = 0;
1734aa5d6bedSChris Mason 
17355f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
17365f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
17375f39d397SChris Mason 
17385f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1739db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
17405f39d397SChris Mason 			  path->slots[level + 1] + 1,
1741123abc88SChris Mason 			  level + 1);
1742aa5d6bedSChris Mason 	if (wret)
1743aa5d6bedSChris Mason 		ret = wret;
1744aa5d6bedSChris Mason 
17455de08d7dSChris Mason 	if (path->slots[level] >= mid) {
17465c680ed6SChris Mason 		path->slots[level] -= mid;
1747925baeddSChris Mason 		btrfs_tree_unlock(c);
17485f39d397SChris Mason 		free_extent_buffer(c);
17495f39d397SChris Mason 		path->nodes[level] = split;
17505c680ed6SChris Mason 		path->slots[level + 1] += 1;
1751eb60ceacSChris Mason 	} else {
1752925baeddSChris Mason 		btrfs_tree_unlock(split);
17535f39d397SChris Mason 		free_extent_buffer(split);
1754be0e5c09SChris Mason 	}
1755aa5d6bedSChris Mason 	return ret;
1756be0e5c09SChris Mason }
1757be0e5c09SChris Mason 
175874123bd7SChris Mason /*
175974123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
176074123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
176174123bd7SChris Mason  * space used both by the item structs and the item data
176274123bd7SChris Mason  */
17635f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1764be0e5c09SChris Mason {
1765be0e5c09SChris Mason 	int data_len;
17665f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1767d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1768be0e5c09SChris Mason 
1769be0e5c09SChris Mason 	if (!nr)
1770be0e5c09SChris Mason 		return 0;
17715f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
17725f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
17730783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1774d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1775be0e5c09SChris Mason 	return data_len;
1776be0e5c09SChris Mason }
1777be0e5c09SChris Mason 
177874123bd7SChris Mason /*
1779d4dbff95SChris Mason  * The space between the end of the leaf items and
1780d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1781d4dbff95SChris Mason  * the leaf has left for both items and data
1782d4dbff95SChris Mason  */
17835f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1784d4dbff95SChris Mason {
17855f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
17865f39d397SChris Mason 	int ret;
17875f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
17885f39d397SChris Mason 	if (ret < 0) {
17895f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1790ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
17915f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
17925f39d397SChris Mason 	}
17935f39d397SChris Mason 	return ret;
1794d4dbff95SChris Mason }
1795d4dbff95SChris Mason 
1796d4dbff95SChris Mason /*
179700ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
179800ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1799aa5d6bedSChris Mason  *
1800aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1801aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
180200ec4c51SChris Mason  */
1803e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
180434a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
180534a38218SChris Mason 			   int empty)
180600ec4c51SChris Mason {
18075f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
18085f39d397SChris Mason 	struct extent_buffer *right;
18095f39d397SChris Mason 	struct extent_buffer *upper;
18105f39d397SChris Mason 	struct btrfs_disk_key disk_key;
181100ec4c51SChris Mason 	int slot;
181234a38218SChris Mason 	u32 i;
181300ec4c51SChris Mason 	int free_space;
181400ec4c51SChris Mason 	int push_space = 0;
181500ec4c51SChris Mason 	int push_items = 0;
18160783fcfcSChris Mason 	struct btrfs_item *item;
18177518a238SChris Mason 	u32 left_nritems;
181834a38218SChris Mason 	u32 nr;
18197518a238SChris Mason 	u32 right_nritems;
18205f39d397SChris Mason 	u32 data_end;
1821db94535dSChris Mason 	u32 this_item_size;
182254aa1f4dSChris Mason 	int ret;
182300ec4c51SChris Mason 
182400ec4c51SChris Mason 	slot = path->slots[1];
182500ec4c51SChris Mason 	if (!path->nodes[1]) {
182600ec4c51SChris Mason 		return 1;
182700ec4c51SChris Mason 	}
182800ec4c51SChris Mason 	upper = path->nodes[1];
18295f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
183000ec4c51SChris Mason 		return 1;
18315f39d397SChris Mason 
1832a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
1833a2135011SChris Mason 
1834ca7a79adSChris Mason 	right = read_node_slot(root, upper, slot + 1);
1835925baeddSChris Mason 	btrfs_tree_lock(right);
1836123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1837925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1838925baeddSChris Mason 		goto out_unlock;
183902217ed2SChris Mason 
18405f39d397SChris Mason 	/* cow and double check */
18415f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
18425f39d397SChris Mason 			      slot + 1, &right);
1843925baeddSChris Mason 	if (ret)
1844925baeddSChris Mason 		goto out_unlock;
1845925baeddSChris Mason 
18465f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1847925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1848925baeddSChris Mason 		goto out_unlock;
18495f39d397SChris Mason 
18505f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
1851925baeddSChris Mason 	if (left_nritems == 0)
1852925baeddSChris Mason 		goto out_unlock;
18535f39d397SChris Mason 
185434a38218SChris Mason 	if (empty)
185534a38218SChris Mason 		nr = 0;
185634a38218SChris Mason 	else
185734a38218SChris Mason 		nr = 1;
185834a38218SChris Mason 
185934a38218SChris Mason 	i = left_nritems - 1;
186034a38218SChris Mason 	while (i >= nr) {
18615f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1862db94535dSChris Mason 
186300ec4c51SChris Mason 		if (path->slots[0] == i)
186400ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1865db94535dSChris Mason 
1866db94535dSChris Mason 		if (!left->map_token) {
1867db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1868db94535dSChris Mason 					sizeof(struct btrfs_item),
1869db94535dSChris Mason 					&left->map_token, &left->kaddr,
1870db94535dSChris Mason 					&left->map_start, &left->map_len,
1871db94535dSChris Mason 					KM_USER1);
1872db94535dSChris Mason 		}
1873db94535dSChris Mason 
1874db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1875db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
187600ec4c51SChris Mason 			break;
187700ec4c51SChris Mason 		push_items++;
1878db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
187934a38218SChris Mason 		if (i == 0)
188034a38218SChris Mason 			break;
188134a38218SChris Mason 		i--;
1882db94535dSChris Mason 	}
1883db94535dSChris Mason 	if (left->map_token) {
1884db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1885db94535dSChris Mason 		left->map_token = NULL;
188600ec4c51SChris Mason 	}
18875f39d397SChris Mason 
1888925baeddSChris Mason 	if (push_items == 0)
1889925baeddSChris Mason 		goto out_unlock;
18905f39d397SChris Mason 
189134a38218SChris Mason 	if (!empty && push_items == left_nritems)
1892a429e513SChris Mason 		WARN_ON(1);
18935f39d397SChris Mason 
189400ec4c51SChris Mason 	/* push left to right */
18955f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
189634a38218SChris Mason 
18975f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1898123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
18995f39d397SChris Mason 
190000ec4c51SChris Mason 	/* make room in the right data area */
19015f39d397SChris Mason 	data_end = leaf_data_end(root, right);
19025f39d397SChris Mason 	memmove_extent_buffer(right,
19035f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
19045f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
19055f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
19065f39d397SChris Mason 
190700ec4c51SChris Mason 	/* copy from the left data area */
19085f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1909d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1910d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1911d6025579SChris Mason 		     push_space);
19125f39d397SChris Mason 
19135f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
19145f39d397SChris Mason 			      btrfs_item_nr_offset(0),
19150783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
19165f39d397SChris Mason 
191700ec4c51SChris Mason 	/* copy the items from left to right */
19185f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
19195f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
19200783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
192100ec4c51SChris Mason 
192200ec4c51SChris Mason 	/* update the item pointers */
19237518a238SChris Mason 	right_nritems += push_items;
19245f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1925123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
19267518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
19275f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1928db94535dSChris Mason 		if (!right->map_token) {
1929db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1930db94535dSChris Mason 					sizeof(struct btrfs_item),
1931db94535dSChris Mason 					&right->map_token, &right->kaddr,
1932db94535dSChris Mason 					&right->map_start, &right->map_len,
1933db94535dSChris Mason 					KM_USER1);
1934db94535dSChris Mason 		}
1935db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1936db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1937db94535dSChris Mason 	}
1938db94535dSChris Mason 
1939db94535dSChris Mason 	if (right->map_token) {
1940db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1941db94535dSChris Mason 		right->map_token = NULL;
194200ec4c51SChris Mason 	}
19437518a238SChris Mason 	left_nritems -= push_items;
19445f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
194500ec4c51SChris Mason 
194634a38218SChris Mason 	if (left_nritems)
19475f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
19485f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1949a429e513SChris Mason 
19505f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
19515f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1952d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
195302217ed2SChris Mason 
195400ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
19557518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
19567518a238SChris Mason 		path->slots[0] -= left_nritems;
1957925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
1958925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
1959925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
19605f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19615f39d397SChris Mason 		path->nodes[0] = right;
196200ec4c51SChris Mason 		path->slots[1] += 1;
196300ec4c51SChris Mason 	} else {
1964925baeddSChris Mason 		btrfs_tree_unlock(right);
19655f39d397SChris Mason 		free_extent_buffer(right);
196600ec4c51SChris Mason 	}
196700ec4c51SChris Mason 	return 0;
1968925baeddSChris Mason 
1969925baeddSChris Mason out_unlock:
1970925baeddSChris Mason 	btrfs_tree_unlock(right);
1971925baeddSChris Mason 	free_extent_buffer(right);
1972925baeddSChris Mason 	return 1;
197300ec4c51SChris Mason }
1974925baeddSChris Mason 
197500ec4c51SChris Mason /*
197674123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
197774123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
197874123bd7SChris Mason  */
1979e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
198034a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
198134a38218SChris Mason 			  int empty)
1982be0e5c09SChris Mason {
19835f39d397SChris Mason 	struct btrfs_disk_key disk_key;
19845f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
19855f39d397SChris Mason 	struct extent_buffer *left;
1986be0e5c09SChris Mason 	int slot;
1987be0e5c09SChris Mason 	int i;
1988be0e5c09SChris Mason 	int free_space;
1989be0e5c09SChris Mason 	int push_space = 0;
1990be0e5c09SChris Mason 	int push_items = 0;
19910783fcfcSChris Mason 	struct btrfs_item *item;
19927518a238SChris Mason 	u32 old_left_nritems;
19935f39d397SChris Mason 	u32 right_nritems;
199434a38218SChris Mason 	u32 nr;
1995aa5d6bedSChris Mason 	int ret = 0;
1996aa5d6bedSChris Mason 	int wret;
1997db94535dSChris Mason 	u32 this_item_size;
1998db94535dSChris Mason 	u32 old_left_item_size;
1999be0e5c09SChris Mason 
2000be0e5c09SChris Mason 	slot = path->slots[1];
20015f39d397SChris Mason 	if (slot == 0)
2002be0e5c09SChris Mason 		return 1;
20035f39d397SChris Mason 	if (!path->nodes[1])
2004be0e5c09SChris Mason 		return 1;
20055f39d397SChris Mason 
20063685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
20073685f791SChris Mason 	if (right_nritems == 0) {
20083685f791SChris Mason 		return 1;
20093685f791SChris Mason 	}
20103685f791SChris Mason 
2011a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
2012a2135011SChris Mason 
2013ca7a79adSChris Mason 	left = read_node_slot(root, path->nodes[1], slot - 1);
2014925baeddSChris Mason 	btrfs_tree_lock(left);
2015123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20160783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2017925baeddSChris Mason 		ret = 1;
2018925baeddSChris Mason 		goto out;
2019be0e5c09SChris Mason 	}
202002217ed2SChris Mason 
202102217ed2SChris Mason 	/* cow and double check */
20225f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
20235f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
202454aa1f4dSChris Mason 	if (ret) {
202554aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
2026925baeddSChris Mason 		ret = 1;
2027925baeddSChris Mason 		goto out;
202854aa1f4dSChris Mason 	}
20293685f791SChris Mason 
2030123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20310783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2032925baeddSChris Mason 		ret = 1;
2033925baeddSChris Mason 		goto out;
203402217ed2SChris Mason 	}
203502217ed2SChris Mason 
203634a38218SChris Mason 	if (empty)
203734a38218SChris Mason 		nr = right_nritems;
203834a38218SChris Mason 	else
203934a38218SChris Mason 		nr = right_nritems - 1;
204034a38218SChris Mason 
204134a38218SChris Mason 	for (i = 0; i < nr; i++) {
20425f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2043db94535dSChris Mason 		if (!right->map_token) {
2044db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2045db94535dSChris Mason 					sizeof(struct btrfs_item),
2046db94535dSChris Mason 					&right->map_token, &right->kaddr,
2047db94535dSChris Mason 					&right->map_start, &right->map_len,
2048db94535dSChris Mason 					KM_USER1);
2049db94535dSChris Mason 		}
2050db94535dSChris Mason 
2051be0e5c09SChris Mason 		if (path->slots[0] == i)
2052be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
2053db94535dSChris Mason 
2054db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
2055db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
2056be0e5c09SChris Mason 			break;
2057db94535dSChris Mason 
2058be0e5c09SChris Mason 		push_items++;
2059db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
2060be0e5c09SChris Mason 	}
2061db94535dSChris Mason 
2062db94535dSChris Mason 	if (right->map_token) {
2063db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2064db94535dSChris Mason 		right->map_token = NULL;
2065db94535dSChris Mason 	}
2066db94535dSChris Mason 
2067be0e5c09SChris Mason 	if (push_items == 0) {
2068925baeddSChris Mason 		ret = 1;
2069925baeddSChris Mason 		goto out;
2070be0e5c09SChris Mason 	}
207134a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
2072a429e513SChris Mason 		WARN_ON(1);
20735f39d397SChris Mason 
2074be0e5c09SChris Mason 	/* push data from right to left */
20755f39d397SChris Mason 	copy_extent_buffer(left, right,
20765f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
20775f39d397SChris Mason 			   btrfs_item_nr_offset(0),
20785f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
20795f39d397SChris Mason 
2080123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
20815f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
20825f39d397SChris Mason 
20835f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
2084d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
2085123abc88SChris Mason 		     btrfs_leaf_data(right) +
20865f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
2087be0e5c09SChris Mason 		     push_space);
20885f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
2089eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
2090eb60ceacSChris Mason 
2091db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
2092be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
20935f39d397SChris Mason 		u32 ioff;
2094db94535dSChris Mason 
20955f39d397SChris Mason 		item = btrfs_item_nr(left, i);
2096db94535dSChris Mason 		if (!left->map_token) {
2097db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
2098db94535dSChris Mason 					sizeof(struct btrfs_item),
2099db94535dSChris Mason 					&left->map_token, &left->kaddr,
2100db94535dSChris Mason 					&left->map_start, &left->map_len,
2101db94535dSChris Mason 					KM_USER1);
2102db94535dSChris Mason 		}
2103db94535dSChris Mason 
21045f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
21055f39d397SChris Mason 		btrfs_set_item_offset(left, item,
2106db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2107be0e5c09SChris Mason 	}
21085f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2109db94535dSChris Mason 	if (left->map_token) {
2110db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
2111db94535dSChris Mason 		left->map_token = NULL;
2112db94535dSChris Mason 	}
2113be0e5c09SChris Mason 
2114be0e5c09SChris Mason 	/* fixup right node */
211534a38218SChris Mason 	if (push_items > right_nritems) {
211634a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
211734a38218SChris Mason 		WARN_ON(1);
211834a38218SChris Mason 	}
211934a38218SChris Mason 
212034a38218SChris Mason 	if (push_items < right_nritems) {
21215f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
2122123abc88SChris Mason 						  leaf_data_end(root, right);
21235f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
2124d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
2125d6025579SChris Mason 				      btrfs_leaf_data(right) +
2126123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
21275f39d397SChris Mason 
21285f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
21295f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
21305f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
21310783fcfcSChris Mason 			     sizeof(struct btrfs_item));
213234a38218SChris Mason 	}
2133eef1c494SYan 	right_nritems -= push_items;
2134eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
2135123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
21365f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
21375f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2138db94535dSChris Mason 
2139db94535dSChris Mason 		if (!right->map_token) {
2140db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2141db94535dSChris Mason 					sizeof(struct btrfs_item),
2142db94535dSChris Mason 					&right->map_token, &right->kaddr,
2143db94535dSChris Mason 					&right->map_start, &right->map_len,
2144db94535dSChris Mason 					KM_USER1);
2145db94535dSChris Mason 		}
2146db94535dSChris Mason 
2147db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
2148db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
2149db94535dSChris Mason 	}
2150db94535dSChris Mason 	if (right->map_token) {
2151db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2152db94535dSChris Mason 		right->map_token = NULL;
2153be0e5c09SChris Mason 	}
2154eb60ceacSChris Mason 
21555f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
215634a38218SChris Mason 	if (right_nritems)
21575f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
2158098f59c2SChris Mason 
21595f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
21605f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
2161aa5d6bedSChris Mason 	if (wret)
2162aa5d6bedSChris Mason 		ret = wret;
2163be0e5c09SChris Mason 
2164be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
2165be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
2166be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
2167925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
2168925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
2169925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
21705f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
21715f39d397SChris Mason 		path->nodes[0] = left;
2172be0e5c09SChris Mason 		path->slots[1] -= 1;
2173be0e5c09SChris Mason 	} else {
2174925baeddSChris Mason 		btrfs_tree_unlock(left);
21755f39d397SChris Mason 		free_extent_buffer(left);
2176be0e5c09SChris Mason 		path->slots[0] -= push_items;
2177be0e5c09SChris Mason 	}
2178eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2179aa5d6bedSChris Mason 	return ret;
2180925baeddSChris Mason out:
2181925baeddSChris Mason 	btrfs_tree_unlock(left);
2182925baeddSChris Mason 	free_extent_buffer(left);
2183925baeddSChris Mason 	return ret;
2184be0e5c09SChris Mason }
2185be0e5c09SChris Mason 
218674123bd7SChris Mason /*
218774123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
218874123bd7SChris Mason  * available for the resulting leaf level of the path.
2189aa5d6bedSChris Mason  *
2190aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
219174123bd7SChris Mason  */
2192e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
2193d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
2194cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend)
2195be0e5c09SChris Mason {
21967bb86316SChris Mason 	u64 root_gen;
21975f39d397SChris Mason 	struct extent_buffer *l;
21987518a238SChris Mason 	u32 nritems;
2199eb60ceacSChris Mason 	int mid;
2200eb60ceacSChris Mason 	int slot;
22015f39d397SChris Mason 	struct extent_buffer *right;
22020783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
2203be0e5c09SChris Mason 	int data_copy_size;
2204be0e5c09SChris Mason 	int rt_data_off;
2205be0e5c09SChris Mason 	int i;
2206d4dbff95SChris Mason 	int ret = 0;
2207aa5d6bedSChris Mason 	int wret;
2208cc0c5538SChris Mason 	int double_split;
2209cc0c5538SChris Mason 	int num_doubles = 0;
2210d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
2211be0e5c09SChris Mason 
2212cc0c5538SChris Mason 	if (extend)
2213cc0c5538SChris Mason 		space_needed = data_size;
2214cc0c5538SChris Mason 
22157bb86316SChris Mason 	if (root->ref_cows)
22167bb86316SChris Mason 		root_gen = trans->transid;
22177bb86316SChris Mason 	else
22187bb86316SChris Mason 		root_gen = 0;
22197bb86316SChris Mason 
222040689478SChris Mason 	/* first try to make some room by pushing left and right */
22213685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
222234a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
22233326d1b0SChris Mason 		if (wret < 0) {
2224eaee50e8SChris Mason 			return wret;
22253326d1b0SChris Mason 		}
2226eaee50e8SChris Mason 		if (wret) {
222734a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
2228eaee50e8SChris Mason 			if (wret < 0)
2229eaee50e8SChris Mason 				return wret;
2230eaee50e8SChris Mason 		}
22315f39d397SChris Mason 		l = path->nodes[0];
2232aa5d6bedSChris Mason 
2233aa5d6bedSChris Mason 		/* did the pushes work? */
2234cc0c5538SChris Mason 		if (btrfs_leaf_free_space(root, l) >= space_needed)
2235be0e5c09SChris Mason 			return 0;
22363326d1b0SChris Mason 	}
2237aa5d6bedSChris Mason 
22385c680ed6SChris Mason 	if (!path->nodes[1]) {
2239e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
22405c680ed6SChris Mason 		if (ret)
22415c680ed6SChris Mason 			return ret;
22425c680ed6SChris Mason 	}
2243cc0c5538SChris Mason again:
2244cc0c5538SChris Mason 	double_split = 0;
2245cc0c5538SChris Mason 	l = path->nodes[0];
2246eb60ceacSChris Mason 	slot = path->slots[0];
22475f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
2248eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
224954aa1f4dSChris Mason 
22507bb86316SChris Mason 	btrfs_item_key(l, &disk_key, 0);
22517bb86316SChris Mason 
2252925baeddSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
22537bb86316SChris Mason 					 root->root_key.objectid,
22547bb86316SChris Mason 					 root_gen, disk_key.objectid, 0,
2255db94535dSChris Mason 					 l->start, 0);
2256cea9e445SChris Mason 	if (IS_ERR(right)) {
2257cea9e445SChris Mason 		BUG_ON(1);
22585f39d397SChris Mason 		return PTR_ERR(right);
2259cea9e445SChris Mason 	}
226054aa1f4dSChris Mason 
22615f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2262db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
22635f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
22645f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
22655f39d397SChris Mason 	btrfs_set_header_level(right, 0);
22665f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
22675f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
22685f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2269e17cade2SChris Mason 
2270e17cade2SChris Mason 	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
2271e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
2272e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
2273d4dbff95SChris Mason 	if (mid <= slot) {
2274d4dbff95SChris Mason 		if (nritems == 1 ||
2275d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
2276d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2277d4dbff95SChris Mason 			if (slot >= nritems) {
2278d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
22795f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2280d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2281db94535dSChris Mason 						  &disk_key, right->start,
2282d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
2283d4dbff95SChris Mason 				if (wret)
2284d4dbff95SChris Mason 					ret = wret;
2285925baeddSChris Mason 
2286925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
22875f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
22885f39d397SChris Mason 				path->nodes[0] = right;
2289d4dbff95SChris Mason 				path->slots[0] = 0;
2290d4dbff95SChris Mason 				path->slots[1] += 1;
22910ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2292d4dbff95SChris Mason 				return ret;
2293d4dbff95SChris Mason 			}
2294d4dbff95SChris Mason 			mid = slot;
22953326d1b0SChris Mason 			if (mid != nritems &&
22963326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
22973326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2298d4dbff95SChris Mason 				double_split = 1;
2299d4dbff95SChris Mason 			}
23003326d1b0SChris Mason 		}
2301d4dbff95SChris Mason 	} else {
2302d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
2303d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2304cc0c5538SChris Mason 			if (!extend && slot == 0) {
2305d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
23065f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2307d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2308d4dbff95SChris Mason 						  &disk_key,
2309db94535dSChris Mason 						  right->start,
2310098f59c2SChris Mason 						  path->slots[1], 1);
2311d4dbff95SChris Mason 				if (wret)
2312d4dbff95SChris Mason 					ret = wret;
2313925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
23145f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
23155f39d397SChris Mason 				path->nodes[0] = right;
2316d4dbff95SChris Mason 				path->slots[0] = 0;
2317a429e513SChris Mason 				if (path->slots[1] == 0) {
2318a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
2319a429e513SChris Mason 					           path, &disk_key, 1);
2320a429e513SChris Mason 					if (wret)
2321a429e513SChris Mason 						ret = wret;
2322a429e513SChris Mason 				}
23230ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2324d4dbff95SChris Mason 				return ret;
2325cc0c5538SChris Mason 			} else if (extend && slot == 0) {
2326cc0c5538SChris Mason 				mid = 1;
2327cc0c5538SChris Mason 			} else {
2328d4dbff95SChris Mason 				mid = slot;
23295ee78ac7SChris Mason 				if (mid != nritems &&
23305ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
23315ee78ac7SChris Mason 				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2332d4dbff95SChris Mason 					double_split = 1;
2333d4dbff95SChris Mason 				}
2334d4dbff95SChris Mason 			}
23355ee78ac7SChris Mason 		}
2336cc0c5538SChris Mason 	}
23375f39d397SChris Mason 	nritems = nritems - mid;
23385f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
23395f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
23405f39d397SChris Mason 
23415f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
23425f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
23435f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
23445f39d397SChris Mason 
23455f39d397SChris Mason 	copy_extent_buffer(right, l,
2346d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2347123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
2348123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
234974123bd7SChris Mason 
23505f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
23515f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
23525f39d397SChris Mason 
23535f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
23545f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
2355db94535dSChris Mason 		u32 ioff;
2356db94535dSChris Mason 
2357db94535dSChris Mason 		if (!right->map_token) {
2358db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2359db94535dSChris Mason 					sizeof(struct btrfs_item),
2360db94535dSChris Mason 					&right->map_token, &right->kaddr,
2361db94535dSChris Mason 					&right->map_start, &right->map_len,
2362db94535dSChris Mason 					KM_USER1);
2363db94535dSChris Mason 		}
2364db94535dSChris Mason 
2365db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
23665f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
23670783fcfcSChris Mason 	}
236874123bd7SChris Mason 
2369db94535dSChris Mason 	if (right->map_token) {
2370db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2371db94535dSChris Mason 		right->map_token = NULL;
2372db94535dSChris Mason 	}
2373db94535dSChris Mason 
23745f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
2375aa5d6bedSChris Mason 	ret = 0;
23765f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
2377db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
2378db94535dSChris Mason 			  path->slots[1] + 1, 1);
2379aa5d6bedSChris Mason 	if (wret)
2380aa5d6bedSChris Mason 		ret = wret;
23815f39d397SChris Mason 
23825f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
23835f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
2384eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
23855f39d397SChris Mason 
2386be0e5c09SChris Mason 	if (mid <= slot) {
2387925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
23885f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
23895f39d397SChris Mason 		path->nodes[0] = right;
2390be0e5c09SChris Mason 		path->slots[0] -= mid;
2391be0e5c09SChris Mason 		path->slots[1] += 1;
2392925baeddSChris Mason 	} else {
2393925baeddSChris Mason 		btrfs_tree_unlock(right);
23945f39d397SChris Mason 		free_extent_buffer(right);
2395925baeddSChris Mason 	}
23965f39d397SChris Mason 
2397eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2398d4dbff95SChris Mason 
2399cc0c5538SChris Mason 	if (double_split) {
2400cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
2401cc0c5538SChris Mason 		num_doubles++;
2402cc0c5538SChris Mason 		goto again;
24033326d1b0SChris Mason 	}
2404be0e5c09SChris Mason 	return ret;
2405be0e5c09SChris Mason }
2406be0e5c09SChris Mason 
2407b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2408b18c6685SChris Mason 			struct btrfs_root *root,
2409b18c6685SChris Mason 			struct btrfs_path *path,
2410179e29e4SChris Mason 			u32 new_size, int from_end)
2411b18c6685SChris Mason {
2412b18c6685SChris Mason 	int ret = 0;
2413b18c6685SChris Mason 	int slot;
2414b18c6685SChris Mason 	int slot_orig;
24155f39d397SChris Mason 	struct extent_buffer *leaf;
24165f39d397SChris Mason 	struct btrfs_item *item;
2417b18c6685SChris Mason 	u32 nritems;
2418b18c6685SChris Mason 	unsigned int data_end;
2419b18c6685SChris Mason 	unsigned int old_data_start;
2420b18c6685SChris Mason 	unsigned int old_size;
2421b18c6685SChris Mason 	unsigned int size_diff;
2422b18c6685SChris Mason 	int i;
2423b18c6685SChris Mason 
2424b18c6685SChris Mason 	slot_orig = path->slots[0];
24255f39d397SChris Mason 	leaf = path->nodes[0];
2426179e29e4SChris Mason 	slot = path->slots[0];
2427179e29e4SChris Mason 
2428179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
2429179e29e4SChris Mason 	if (old_size == new_size)
2430179e29e4SChris Mason 		return 0;
2431b18c6685SChris Mason 
24325f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2433b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
2434b18c6685SChris Mason 
24355f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
2436179e29e4SChris Mason 
2437b18c6685SChris Mason 	size_diff = old_size - new_size;
2438b18c6685SChris Mason 
2439b18c6685SChris Mason 	BUG_ON(slot < 0);
2440b18c6685SChris Mason 	BUG_ON(slot >= nritems);
2441b18c6685SChris Mason 
2442b18c6685SChris Mason 	/*
2443b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2444b18c6685SChris Mason 	 */
2445b18c6685SChris Mason 	/* first correct the data pointers */
2446b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
24475f39d397SChris Mason 		u32 ioff;
24485f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2449db94535dSChris Mason 
2450db94535dSChris Mason 		if (!leaf->map_token) {
2451db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2452db94535dSChris Mason 					sizeof(struct btrfs_item),
2453db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2454db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2455db94535dSChris Mason 					KM_USER1);
2456db94535dSChris Mason 		}
2457db94535dSChris Mason 
24585f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
24595f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2460b18c6685SChris Mason 	}
2461db94535dSChris Mason 
2462db94535dSChris Mason 	if (leaf->map_token) {
2463db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2464db94535dSChris Mason 		leaf->map_token = NULL;
2465db94535dSChris Mason 	}
2466db94535dSChris Mason 
2467b18c6685SChris Mason 	/* shift the data */
2468179e29e4SChris Mason 	if (from_end) {
24695f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2470b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2471b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
2472179e29e4SChris Mason 	} else {
2473179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
2474179e29e4SChris Mason 		u64 offset;
2475179e29e4SChris Mason 
2476179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
2477179e29e4SChris Mason 
2478179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2479179e29e4SChris Mason 			unsigned long ptr;
2480179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
2481179e29e4SChris Mason 
2482179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
2483179e29e4SChris Mason 					    struct btrfs_file_extent_item);
2484179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
2485179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
2486179e29e4SChris Mason 
2487179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
2488179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
2489179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
2490179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
2491179e29e4SChris Mason 				        (unsigned long)fi,
2492179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
2493179e29e4SChris Mason 						 disk_bytenr));
2494179e29e4SChris Mason 			}
2495179e29e4SChris Mason 		}
2496179e29e4SChris Mason 
2497179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2498179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2499179e29e4SChris Mason 			      data_end, old_data_start - data_end);
2500179e29e4SChris Mason 
2501179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
2502179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2503179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
2504179e29e4SChris Mason 		if (slot == 0)
2505179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
2506179e29e4SChris Mason 	}
25075f39d397SChris Mason 
25085f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
25095f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
25105f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2511b18c6685SChris Mason 
2512b18c6685SChris Mason 	ret = 0;
25135f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
25145f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2515b18c6685SChris Mason 		BUG();
25165f39d397SChris Mason 	}
2517b18c6685SChris Mason 	return ret;
2518b18c6685SChris Mason }
2519b18c6685SChris Mason 
25205f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
25215f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
25225f39d397SChris Mason 		      u32 data_size)
25236567e837SChris Mason {
25246567e837SChris Mason 	int ret = 0;
25256567e837SChris Mason 	int slot;
25266567e837SChris Mason 	int slot_orig;
25275f39d397SChris Mason 	struct extent_buffer *leaf;
25285f39d397SChris Mason 	struct btrfs_item *item;
25296567e837SChris Mason 	u32 nritems;
25306567e837SChris Mason 	unsigned int data_end;
25316567e837SChris Mason 	unsigned int old_data;
25326567e837SChris Mason 	unsigned int old_size;
25336567e837SChris Mason 	int i;
25346567e837SChris Mason 
25356567e837SChris Mason 	slot_orig = path->slots[0];
25365f39d397SChris Mason 	leaf = path->nodes[0];
25376567e837SChris Mason 
25385f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
25396567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
25406567e837SChris Mason 
25415f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
25425f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25436567e837SChris Mason 		BUG();
25445f39d397SChris Mason 	}
25456567e837SChris Mason 	slot = path->slots[0];
25465f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
25476567e837SChris Mason 
25486567e837SChris Mason 	BUG_ON(slot < 0);
25493326d1b0SChris Mason 	if (slot >= nritems) {
25503326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
25513326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
25523326d1b0SChris Mason 		BUG_ON(1);
25533326d1b0SChris Mason 	}
25546567e837SChris Mason 
25556567e837SChris Mason 	/*
25566567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
25576567e837SChris Mason 	 */
25586567e837SChris Mason 	/* first correct the data pointers */
25596567e837SChris Mason 	for (i = slot; i < nritems; i++) {
25605f39d397SChris Mason 		u32 ioff;
25615f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2562db94535dSChris Mason 
2563db94535dSChris Mason 		if (!leaf->map_token) {
2564db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2565db94535dSChris Mason 					sizeof(struct btrfs_item),
2566db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2567db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2568db94535dSChris Mason 					KM_USER1);
2569db94535dSChris Mason 		}
25705f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
25715f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
25726567e837SChris Mason 	}
25735f39d397SChris Mason 
2574db94535dSChris Mason 	if (leaf->map_token) {
2575db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2576db94535dSChris Mason 		leaf->map_token = NULL;
2577db94535dSChris Mason 	}
2578db94535dSChris Mason 
25796567e837SChris Mason 	/* shift the data */
25805f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
25816567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
25826567e837SChris Mason 		      data_end, old_data - data_end);
25835f39d397SChris Mason 
25846567e837SChris Mason 	data_end = old_data;
25855f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
25865f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
25875f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
25885f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
25896567e837SChris Mason 
25906567e837SChris Mason 	ret = 0;
25915f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
25925f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25936567e837SChris Mason 		BUG();
25945f39d397SChris Mason 	}
25956567e837SChris Mason 	return ret;
25966567e837SChris Mason }
25976567e837SChris Mason 
259874123bd7SChris Mason /*
259974123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
260074123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
260174123bd7SChris Mason  */
26029c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
26035f39d397SChris Mason 			    struct btrfs_root *root,
26045f39d397SChris Mason 			    struct btrfs_path *path,
26059c58309dSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
26069c58309dSChris Mason 			    int nr)
2607be0e5c09SChris Mason {
26085f39d397SChris Mason 	struct extent_buffer *leaf;
26095f39d397SChris Mason 	struct btrfs_item *item;
2610aa5d6bedSChris Mason 	int ret = 0;
2611be0e5c09SChris Mason 	int slot;
2612eb60ceacSChris Mason 	int slot_orig;
26139c58309dSChris Mason 	int i;
26147518a238SChris Mason 	u32 nritems;
26159c58309dSChris Mason 	u32 total_size = 0;
26169c58309dSChris Mason 	u32 total_data = 0;
2617be0e5c09SChris Mason 	unsigned int data_end;
2618e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2619e2fa7227SChris Mason 
26209c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26219c58309dSChris Mason 		total_data += data_size[i];
26229c58309dSChris Mason 	}
2623be0e5c09SChris Mason 
26249c58309dSChris Mason 	total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
26259c58309dSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2626eb60ceacSChris Mason 	if (ret == 0) {
2627f0930a37SChris Mason 		return -EEXIST;
2628aa5d6bedSChris Mason 	}
2629ed2ff2cbSChris Mason 	if (ret < 0)
2630ed2ff2cbSChris Mason 		goto out;
2631be0e5c09SChris Mason 
263262e2749eSChris Mason 	slot_orig = path->slots[0];
26335f39d397SChris Mason 	leaf = path->nodes[0];
263474123bd7SChris Mason 
26355f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2636123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2637eb60ceacSChris Mason 
2638123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
26399c58309dSChris Mason 	    sizeof(struct btrfs_item) + total_size) {
26403326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
26413326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
26429c58309dSChris Mason 		       total_size, btrfs_leaf_free_space(root, leaf));
2643be0e5c09SChris Mason 		BUG();
2644d4dbff95SChris Mason 	}
26455f39d397SChris Mason 
264662e2749eSChris Mason 	slot = path->slots[0];
2647eb60ceacSChris Mason 	BUG_ON(slot < 0);
26485f39d397SChris Mason 
2649be0e5c09SChris Mason 	if (slot != nritems) {
2650be0e5c09SChris Mason 		int i;
26515f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2652be0e5c09SChris Mason 
26535f39d397SChris Mason 		if (old_data < data_end) {
26545f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
26555f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
26565f39d397SChris Mason 			       slot, old_data, data_end);
26575f39d397SChris Mason 			BUG_ON(1);
26585f39d397SChris Mason 		}
2659be0e5c09SChris Mason 		/*
2660be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2661be0e5c09SChris Mason 		 */
2662be0e5c09SChris Mason 		/* first correct the data pointers */
2663db94535dSChris Mason 		WARN_ON(leaf->map_token);
26640783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
26655f39d397SChris Mason 			u32 ioff;
2666db94535dSChris Mason 
26675f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2668db94535dSChris Mason 			if (!leaf->map_token) {
2669db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2670db94535dSChris Mason 					sizeof(struct btrfs_item),
2671db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2672db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2673db94535dSChris Mason 					KM_USER1);
2674db94535dSChris Mason 			}
2675db94535dSChris Mason 
26765f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
26779c58309dSChris Mason 			btrfs_set_item_offset(leaf, item, ioff - total_data);
26780783fcfcSChris Mason 		}
2679db94535dSChris Mason 		if (leaf->map_token) {
2680db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2681db94535dSChris Mason 			leaf->map_token = NULL;
2682db94535dSChris Mason 		}
2683be0e5c09SChris Mason 
2684be0e5c09SChris Mason 		/* shift the items */
26859c58309dSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
26865f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
26870783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2688be0e5c09SChris Mason 
2689be0e5c09SChris Mason 		/* shift the data */
26905f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
26919c58309dSChris Mason 			      data_end - total_data, btrfs_leaf_data(leaf) +
2692be0e5c09SChris Mason 			      data_end, old_data - data_end);
2693be0e5c09SChris Mason 		data_end = old_data;
2694be0e5c09SChris Mason 	}
26955f39d397SChris Mason 
269662e2749eSChris Mason 	/* setup the item for the new data */
26979c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26989c58309dSChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
26999c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
27009c58309dSChris Mason 		item = btrfs_item_nr(leaf, slot + i);
27019c58309dSChris Mason 		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
27029c58309dSChris Mason 		data_end -= data_size[i];
27039c58309dSChris Mason 		btrfs_set_item_size(leaf, item, data_size[i]);
27049c58309dSChris Mason 	}
27059c58309dSChris Mason 	btrfs_set_header_nritems(leaf, nritems + nr);
27065f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2707aa5d6bedSChris Mason 
2708aa5d6bedSChris Mason 	ret = 0;
27095a01a2e3SChris Mason 	if (slot == 0) {
27105a01a2e3SChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2711e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
27125a01a2e3SChris Mason 	}
2713aa5d6bedSChris Mason 
27145f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
27155f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2716be0e5c09SChris Mason 		BUG();
27175f39d397SChris Mason 	}
2718ed2ff2cbSChris Mason out:
271962e2749eSChris Mason 	return ret;
272062e2749eSChris Mason }
272162e2749eSChris Mason 
272262e2749eSChris Mason /*
272362e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
272462e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
272562e2749eSChris Mason  */
2726e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2727e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2728e089f05cSChris Mason 		      data_size)
272962e2749eSChris Mason {
273062e2749eSChris Mason 	int ret = 0;
27312c90e5d6SChris Mason 	struct btrfs_path *path;
27325f39d397SChris Mason 	struct extent_buffer *leaf;
27335f39d397SChris Mason 	unsigned long ptr;
273462e2749eSChris Mason 
27352c90e5d6SChris Mason 	path = btrfs_alloc_path();
27362c90e5d6SChris Mason 	BUG_ON(!path);
27372c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
273862e2749eSChris Mason 	if (!ret) {
27395f39d397SChris Mason 		leaf = path->nodes[0];
27405f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
27415f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
27425f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
274362e2749eSChris Mason 	}
27442c90e5d6SChris Mason 	btrfs_free_path(path);
2745aa5d6bedSChris Mason 	return ret;
2746be0e5c09SChris Mason }
2747be0e5c09SChris Mason 
274874123bd7SChris Mason /*
27495de08d7dSChris Mason  * delete the pointer from a given node.
275074123bd7SChris Mason  *
275174123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
275274123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
275374123bd7SChris Mason  * a leaf if all the nodes are emptied.
275474123bd7SChris Mason  */
2755e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2756e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2757be0e5c09SChris Mason {
27585f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
27597518a238SChris Mason 	u32 nritems;
2760aa5d6bedSChris Mason 	int ret = 0;
2761bb803951SChris Mason 	int wret;
2762be0e5c09SChris Mason 
27635f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2764be0e5c09SChris Mason 	if (slot != nritems -1) {
27655f39d397SChris Mason 		memmove_extent_buffer(parent,
27665f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
27675f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2768d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2769d6025579SChris Mason 			      (nritems - slot - 1));
2770be0e5c09SChris Mason 	}
27717518a238SChris Mason 	nritems--;
27725f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
27737518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
27745f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2775eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
27765f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2777bb803951SChris Mason 	} else if (slot == 0) {
27785f39d397SChris Mason 		struct btrfs_disk_key disk_key;
27795f39d397SChris Mason 
27805f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
27815f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
27820f70abe2SChris Mason 		if (wret)
27830f70abe2SChris Mason 			ret = wret;
2784be0e5c09SChris Mason 	}
2785d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2786aa5d6bedSChris Mason 	return ret;
2787be0e5c09SChris Mason }
2788be0e5c09SChris Mason 
278974123bd7SChris Mason /*
279074123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
279174123bd7SChris Mason  * the leaf, remove it from the tree
279274123bd7SChris Mason  */
279385e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
279485e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
2795be0e5c09SChris Mason {
27965f39d397SChris Mason 	struct extent_buffer *leaf;
27975f39d397SChris Mason 	struct btrfs_item *item;
279885e21bacSChris Mason 	int last_off;
279985e21bacSChris Mason 	int dsize = 0;
2800aa5d6bedSChris Mason 	int ret = 0;
2801aa5d6bedSChris Mason 	int wret;
280285e21bacSChris Mason 	int i;
28037518a238SChris Mason 	u32 nritems;
2804be0e5c09SChris Mason 
28055f39d397SChris Mason 	leaf = path->nodes[0];
280685e21bacSChris Mason 	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
280785e21bacSChris Mason 
280885e21bacSChris Mason 	for (i = 0; i < nr; i++)
280985e21bacSChris Mason 		dsize += btrfs_item_size_nr(leaf, slot + i);
281085e21bacSChris Mason 
28115f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2812be0e5c09SChris Mason 
281385e21bacSChris Mason 	if (slot + nr != nritems) {
2814be0e5c09SChris Mason 		int i;
2815123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
28165f39d397SChris Mason 
28175f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2818d6025579SChris Mason 			      data_end + dsize,
2819123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
282085e21bacSChris Mason 			      last_off - data_end);
28215f39d397SChris Mason 
282285e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
28235f39d397SChris Mason 			u32 ioff;
2824db94535dSChris Mason 
28255f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2826db94535dSChris Mason 			if (!leaf->map_token) {
2827db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2828db94535dSChris Mason 					sizeof(struct btrfs_item),
2829db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2830db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2831db94535dSChris Mason 					KM_USER1);
2832db94535dSChris Mason 			}
28335f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
28345f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
28350783fcfcSChris Mason 		}
2836db94535dSChris Mason 
2837db94535dSChris Mason 		if (leaf->map_token) {
2838db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2839db94535dSChris Mason 			leaf->map_token = NULL;
2840db94535dSChris Mason 		}
2841db94535dSChris Mason 
28425f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
284385e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
28440783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
284585e21bacSChris Mason 			      (nritems - slot - nr));
2846be0e5c09SChris Mason 	}
284785e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
284885e21bacSChris Mason 	nritems -= nr;
28495f39d397SChris Mason 
285074123bd7SChris Mason 	/* delete the leaf if we've emptied it */
28517518a238SChris Mason 	if (nritems == 0) {
28525f39d397SChris Mason 		if (leaf == root->node) {
28535f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
28549a8dd150SChris Mason 		} else {
28557bb86316SChris Mason 			u64 root_gen = btrfs_header_generation(path->nodes[1]);
2856e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2857aa5d6bedSChris Mason 			if (wret)
2858aa5d6bedSChris Mason 				ret = wret;
2859e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
28607bb86316SChris Mason 					 leaf->start, leaf->len,
28617bb86316SChris Mason 					 btrfs_header_owner(path->nodes[1]),
28627bb86316SChris Mason 					 root_gen, 0, 0, 1);
28630f70abe2SChris Mason 			if (wret)
28640f70abe2SChris Mason 				ret = wret;
28659a8dd150SChris Mason 		}
2866be0e5c09SChris Mason 	} else {
28677518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2868aa5d6bedSChris Mason 		if (slot == 0) {
28695f39d397SChris Mason 			struct btrfs_disk_key disk_key;
28705f39d397SChris Mason 
28715f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2872e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
28735f39d397SChris Mason 					      &disk_key, 1);
2874aa5d6bedSChris Mason 			if (wret)
2875aa5d6bedSChris Mason 				ret = wret;
2876aa5d6bedSChris Mason 		}
2877aa5d6bedSChris Mason 
287874123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
287985e21bacSChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
2880be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2881be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2882be0e5c09SChris Mason 			 * for possible call to del_ptr below
2883be0e5c09SChris Mason 			 */
28844920c9acSChris Mason 			slot = path->slots[1];
28855f39d397SChris Mason 			extent_buffer_get(leaf);
28865f39d397SChris Mason 
288785e21bacSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1);
288854aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2889aa5d6bedSChris Mason 				ret = wret;
28905f39d397SChris Mason 
28915f39d397SChris Mason 			if (path->nodes[0] == leaf &&
28925f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
289385e21bacSChris Mason 				wret = push_leaf_right(trans, root, path, 1, 1);
289454aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2895aa5d6bedSChris Mason 					ret = wret;
2896aa5d6bedSChris Mason 			}
28975f39d397SChris Mason 
28985f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
28997bb86316SChris Mason 				u64 root_gen;
2900db94535dSChris Mason 				u64 bytenr = leaf->start;
2901db94535dSChris Mason 				u32 blocksize = leaf->len;
29025f39d397SChris Mason 
29037bb86316SChris Mason 				root_gen = btrfs_header_generation(
29047bb86316SChris Mason 							   path->nodes[1]);
29057bb86316SChris Mason 
2906e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2907aa5d6bedSChris Mason 				if (wret)
2908aa5d6bedSChris Mason 					ret = wret;
29095f39d397SChris Mason 
29105f39d397SChris Mason 				free_extent_buffer(leaf);
2911db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
29127bb86316SChris Mason 					     blocksize,
29137bb86316SChris Mason 					     btrfs_header_owner(path->nodes[1]),
29147bb86316SChris Mason 					     root_gen, 0, 0, 1);
29150f70abe2SChris Mason 				if (wret)
29160f70abe2SChris Mason 					ret = wret;
29175de08d7dSChris Mason 			} else {
2918925baeddSChris Mason 				/* if we're still in the path, make sure
2919925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
2920925baeddSChris Mason 				 * push_leaf functions must have already
2921925baeddSChris Mason 				 * dirtied this buffer
2922925baeddSChris Mason 				 */
2923925baeddSChris Mason 				if (path->nodes[0] == leaf)
29245f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
29255f39d397SChris Mason 				free_extent_buffer(leaf);
2926be0e5c09SChris Mason 			}
2927d5719762SChris Mason 		} else {
29285f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2929be0e5c09SChris Mason 		}
29309a8dd150SChris Mason 	}
2931aa5d6bedSChris Mason 	return ret;
29329a8dd150SChris Mason }
29339a8dd150SChris Mason 
293497571fd0SChris Mason /*
2935925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
29367bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
29377bb86316SChris Mason  * returns < 0 on io errors.
29387bb86316SChris Mason  */
29397bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
29407bb86316SChris Mason {
2941925baeddSChris Mason 	struct btrfs_key key;
2942925baeddSChris Mason 	struct btrfs_disk_key found_key;
2943925baeddSChris Mason 	int ret;
29447bb86316SChris Mason 
2945925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
2946925baeddSChris Mason 
2947925baeddSChris Mason 	if (key.offset > 0)
2948925baeddSChris Mason 		key.offset--;
2949925baeddSChris Mason 	else if (key.type > 0)
2950925baeddSChris Mason 		key.type--;
2951925baeddSChris Mason 	else if (key.objectid > 0)
2952925baeddSChris Mason 		key.objectid--;
2953925baeddSChris Mason 	else
29547bb86316SChris Mason 		return 1;
29557bb86316SChris Mason 
2956925baeddSChris Mason 	btrfs_release_path(root, path);
2957925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2958925baeddSChris Mason 	if (ret < 0)
2959925baeddSChris Mason 		return ret;
2960925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
2961925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
2962925baeddSChris Mason 	if (ret < 0)
29637bb86316SChris Mason 		return 0;
2964925baeddSChris Mason 	return 1;
29657bb86316SChris Mason }
29667bb86316SChris Mason 
2967*3f157a2fSChris Mason /*
2968*3f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
2969*3f157a2fSChris Mason  * for nodes or leaves that are either in cache or have a minimum
2970*3f157a2fSChris Mason  * transaction id.  This is used by the btree defrag code, but could
2971*3f157a2fSChris Mason  * also be used to search for blocks that have changed since a given
2972*3f157a2fSChris Mason  * transaction id.
2973*3f157a2fSChris Mason  *
2974*3f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
2975*3f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
2976*3f157a2fSChris Mason  * key and get a writable path.
2977*3f157a2fSChris Mason  *
2978*3f157a2fSChris Mason  * This does lock as it descends, and path->keep_locks should be set
2979*3f157a2fSChris Mason  * to 1 by the caller.
2980*3f157a2fSChris Mason  *
2981*3f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
2982*3f157a2fSChris Mason  * of the tree.
2983*3f157a2fSChris Mason  *
2984*3f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
2985*3f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
2986*3f157a2fSChris Mason  */
2987*3f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
2988*3f157a2fSChris Mason 			 struct btrfs_path *path, int cache_only,
2989*3f157a2fSChris Mason 			 u64 min_trans)
2990*3f157a2fSChris Mason {
2991*3f157a2fSChris Mason 	struct extent_buffer *cur;
2992*3f157a2fSChris Mason 	struct btrfs_key found_key;
2993*3f157a2fSChris Mason 	int slot;
2994*3f157a2fSChris Mason 	u32 nritems;
2995*3f157a2fSChris Mason 	int level;
2996*3f157a2fSChris Mason 	int ret = 1;
2997*3f157a2fSChris Mason 
2998*3f157a2fSChris Mason again:
2999*3f157a2fSChris Mason 	cur = btrfs_lock_root_node(root);
3000*3f157a2fSChris Mason 	level = btrfs_header_level(cur);
3001*3f157a2fSChris Mason 	path->nodes[level] = cur;
3002*3f157a2fSChris Mason 	path->locks[level] = 1;
3003*3f157a2fSChris Mason 
3004*3f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
3005*3f157a2fSChris Mason 		ret = 1;
3006*3f157a2fSChris Mason 		goto out;
3007*3f157a2fSChris Mason 	}
3008*3f157a2fSChris Mason 	while(1) {
3009*3f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
3010*3f157a2fSChris Mason 		level = btrfs_header_level(cur);
3011*3f157a2fSChris Mason 		bin_search(cur, min_key, level, &slot);
3012*3f157a2fSChris Mason 
3013*3f157a2fSChris Mason 		/* at level = 0, we're done, setup the path and exit */
3014*3f157a2fSChris Mason 		if (level == 0) {
3015*3f157a2fSChris Mason 			ret = 0;
3016*3f157a2fSChris Mason 			path->slots[level] = slot;
3017*3f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
3018*3f157a2fSChris Mason 			goto out;
3019*3f157a2fSChris Mason 		}
3020*3f157a2fSChris Mason 		/*
3021*3f157a2fSChris Mason 		 * check this node pointer against the cache_only and
3022*3f157a2fSChris Mason 		 * min_trans parameters.  If it isn't in cache or is too
3023*3f157a2fSChris Mason 		 * old, skip to the next one.
3024*3f157a2fSChris Mason 		 */
3025*3f157a2fSChris Mason 		while(slot < nritems) {
3026*3f157a2fSChris Mason 			u64 blockptr;
3027*3f157a2fSChris Mason 			u64 gen;
3028*3f157a2fSChris Mason 			struct extent_buffer *tmp;
3029*3f157a2fSChris Mason 			blockptr = btrfs_node_blockptr(cur, slot);
3030*3f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
3031*3f157a2fSChris Mason 			if (gen < min_trans) {
3032*3f157a2fSChris Mason 				slot++;
3033*3f157a2fSChris Mason 				continue;
3034*3f157a2fSChris Mason 			}
3035*3f157a2fSChris Mason 			if (!cache_only)
3036*3f157a2fSChris Mason 				break;
3037*3f157a2fSChris Mason 
3038*3f157a2fSChris Mason 			tmp = btrfs_find_tree_block(root, blockptr,
3039*3f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
3040*3f157a2fSChris Mason 
3041*3f157a2fSChris Mason 			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
3042*3f157a2fSChris Mason 				free_extent_buffer(tmp);
3043*3f157a2fSChris Mason 				break;
3044*3f157a2fSChris Mason 			}
3045*3f157a2fSChris Mason 			if (tmp)
3046*3f157a2fSChris Mason 				free_extent_buffer(tmp);
3047*3f157a2fSChris Mason 			slot++;
3048*3f157a2fSChris Mason 		}
3049*3f157a2fSChris Mason 		/*
3050*3f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
3051*3f157a2fSChris Mason 		 * and find another one
3052*3f157a2fSChris Mason 		 */
3053*3f157a2fSChris Mason 		if (slot >= nritems) {
3054*3f157a2fSChris Mason 			ret = btrfs_find_next_key(root, path, min_key, level,
3055*3f157a2fSChris Mason 						  cache_only, min_trans);
3056*3f157a2fSChris Mason 			if (ret == 0) {
3057*3f157a2fSChris Mason 				btrfs_release_path(root, path);
3058*3f157a2fSChris Mason 				goto again;
3059*3f157a2fSChris Mason 			} else {
3060*3f157a2fSChris Mason 				goto out;
3061*3f157a2fSChris Mason 			}
3062*3f157a2fSChris Mason 		}
3063*3f157a2fSChris Mason 		/* save our key for returning back */
3064*3f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
3065*3f157a2fSChris Mason 		path->slots[level] = slot;
3066*3f157a2fSChris Mason 		if (level == path->lowest_level) {
3067*3f157a2fSChris Mason 			ret = 0;
3068*3f157a2fSChris Mason 			unlock_up(path, level, 1);
3069*3f157a2fSChris Mason 			goto out;
3070*3f157a2fSChris Mason 		}
3071*3f157a2fSChris Mason 		cur = read_node_slot(root, cur, slot);
3072*3f157a2fSChris Mason 
3073*3f157a2fSChris Mason 		btrfs_tree_lock(cur);
3074*3f157a2fSChris Mason 		path->locks[level - 1] = 1;
3075*3f157a2fSChris Mason 		path->nodes[level - 1] = cur;
3076*3f157a2fSChris Mason 		unlock_up(path, level, 1);
3077*3f157a2fSChris Mason 	}
3078*3f157a2fSChris Mason out:
3079*3f157a2fSChris Mason 	if (ret == 0)
3080*3f157a2fSChris Mason 		memcpy(min_key, &found_key, sizeof(found_key));
3081*3f157a2fSChris Mason 	return ret;
3082*3f157a2fSChris Mason }
3083*3f157a2fSChris Mason 
3084*3f157a2fSChris Mason /*
3085*3f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
3086*3f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
3087*3f157a2fSChris Mason  * tree based on the current path and the cache_only and min_trans
3088*3f157a2fSChris Mason  * parameters.
3089*3f157a2fSChris Mason  *
3090*3f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
3091*3f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
3092*3f157a2fSChris Mason  *
3093*3f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
3094*3f157a2fSChris Mason  * calling this function.
3095*3f157a2fSChris Mason  */
3096e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
3097*3f157a2fSChris Mason 			struct btrfs_key *key, int lowest_level,
3098*3f157a2fSChris Mason 			int cache_only, u64 min_trans)
3099e7a84565SChris Mason {
3100e7a84565SChris Mason 	int level = lowest_level;
3101e7a84565SChris Mason 	int slot;
3102e7a84565SChris Mason 	struct extent_buffer *c;
3103e7a84565SChris Mason 
3104e7a84565SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3105e7a84565SChris Mason 		if (!path->nodes[level])
3106e7a84565SChris Mason 			return 1;
3107e7a84565SChris Mason 
3108e7a84565SChris Mason 		slot = path->slots[level] + 1;
3109e7a84565SChris Mason 		c = path->nodes[level];
3110*3f157a2fSChris Mason next:
3111e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3112e7a84565SChris Mason 			level++;
3113e7a84565SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
3114e7a84565SChris Mason 				return 1;
3115e7a84565SChris Mason 			}
3116e7a84565SChris Mason 			continue;
3117e7a84565SChris Mason 		}
3118e7a84565SChris Mason 		if (level == 0)
3119e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
3120*3f157a2fSChris Mason 		else {
3121*3f157a2fSChris Mason 			u64 blockptr = btrfs_node_blockptr(c, slot);
3122*3f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
3123*3f157a2fSChris Mason 
3124*3f157a2fSChris Mason 			if (cache_only) {
3125*3f157a2fSChris Mason 				struct extent_buffer *cur;
3126*3f157a2fSChris Mason 				cur = btrfs_find_tree_block(root, blockptr,
3127*3f157a2fSChris Mason 					    btrfs_level_size(root, level - 1));
3128*3f157a2fSChris Mason 				if (!cur || !btrfs_buffer_uptodate(cur, gen)) {
3129*3f157a2fSChris Mason 					slot++;
3130*3f157a2fSChris Mason 					if (cur)
3131*3f157a2fSChris Mason 						free_extent_buffer(cur);
3132*3f157a2fSChris Mason 					goto next;
3133*3f157a2fSChris Mason 				}
3134*3f157a2fSChris Mason 				free_extent_buffer(cur);
3135*3f157a2fSChris Mason 			}
3136*3f157a2fSChris Mason 			if (gen < min_trans) {
3137*3f157a2fSChris Mason 				slot++;
3138*3f157a2fSChris Mason 				goto next;
3139*3f157a2fSChris Mason 			}
3140e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
3141*3f157a2fSChris Mason 		}
3142e7a84565SChris Mason 		return 0;
3143e7a84565SChris Mason 	}
3144e7a84565SChris Mason 	return 1;
3145e7a84565SChris Mason }
3146e7a84565SChris Mason 
31477bb86316SChris Mason /*
3148925baeddSChris Mason  * search the tree again to find a leaf with greater keys
31490f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
31500f70abe2SChris Mason  * returns < 0 on io errors.
315197571fd0SChris Mason  */
3152234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
3153d97e63b6SChris Mason {
3154d97e63b6SChris Mason 	int slot;
3155d97e63b6SChris Mason 	int level = 1;
31565f39d397SChris Mason 	struct extent_buffer *c;
31575f39d397SChris Mason 	struct extent_buffer *next = NULL;
3158925baeddSChris Mason 	struct btrfs_key key;
3159925baeddSChris Mason 	u32 nritems;
3160925baeddSChris Mason 	int ret;
3161925baeddSChris Mason 
3162925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3163925baeddSChris Mason 	if (nritems == 0) {
3164925baeddSChris Mason 		return 1;
3165925baeddSChris Mason 	}
3166925baeddSChris Mason 
3167925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
3168925baeddSChris Mason 
3169925baeddSChris Mason 	btrfs_release_path(root, path);
3170a2135011SChris Mason 	path->keep_locks = 1;
3171925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
3172925baeddSChris Mason 	path->keep_locks = 0;
3173925baeddSChris Mason 
3174925baeddSChris Mason 	if (ret < 0)
3175925baeddSChris Mason 		return ret;
3176925baeddSChris Mason 
3177a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
3178168fd7d2SChris Mason 	/*
3179168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
3180168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
3181168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
3182168fd7d2SChris Mason 	 * advance the path if there are now more items available.
3183168fd7d2SChris Mason 	 */
3184a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
3185168fd7d2SChris Mason 		path->slots[0]++;
3186925baeddSChris Mason 		goto done;
3187925baeddSChris Mason 	}
3188d97e63b6SChris Mason 
3189234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
3190d97e63b6SChris Mason 		if (!path->nodes[level])
31910f70abe2SChris Mason 			return 1;
31925f39d397SChris Mason 
3193d97e63b6SChris Mason 		slot = path->slots[level] + 1;
3194d97e63b6SChris Mason 		c = path->nodes[level];
31955f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
3196d97e63b6SChris Mason 			level++;
3197925baeddSChris Mason 			if (level == BTRFS_MAX_LEVEL) {
31987bb86316SChris Mason 				return 1;
3199925baeddSChris Mason 			}
3200d97e63b6SChris Mason 			continue;
3201d97e63b6SChris Mason 		}
32025f39d397SChris Mason 
3203925baeddSChris Mason 		if (next) {
3204925baeddSChris Mason 			btrfs_tree_unlock(next);
32055f39d397SChris Mason 			free_extent_buffer(next);
3206925baeddSChris Mason 		}
32075f39d397SChris Mason 
3208925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
320901f46658SChris Mason 			reada_for_search(root, path, level, slot, 0);
32105f39d397SChris Mason 
3211ca7a79adSChris Mason 		next = read_node_slot(root, c, slot);
32125cd57b2cSChris Mason 		if (!path->skip_locking) {
3213051e1b9fSChris Mason 			WARN_ON(!btrfs_tree_locked(c));
3214925baeddSChris Mason 			btrfs_tree_lock(next);
32155cd57b2cSChris Mason 		}
3216d97e63b6SChris Mason 		break;
3217d97e63b6SChris Mason 	}
3218d97e63b6SChris Mason 	path->slots[level] = slot;
3219d97e63b6SChris Mason 	while(1) {
3220d97e63b6SChris Mason 		level--;
3221d97e63b6SChris Mason 		c = path->nodes[level];
3222925baeddSChris Mason 		if (path->locks[level])
3223925baeddSChris Mason 			btrfs_tree_unlock(c);
32245f39d397SChris Mason 		free_extent_buffer(c);
3225d97e63b6SChris Mason 		path->nodes[level] = next;
3226d97e63b6SChris Mason 		path->slots[level] = 0;
3227a74a4b97SChris Mason 		if (!path->skip_locking)
3228925baeddSChris Mason 			path->locks[level] = 1;
3229d97e63b6SChris Mason 		if (!level)
3230d97e63b6SChris Mason 			break;
3231925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
3232925baeddSChris Mason 			reada_for_search(root, path, level, slot, 0);
3233ca7a79adSChris Mason 		next = read_node_slot(root, next, 0);
32345cd57b2cSChris Mason 		if (!path->skip_locking) {
3235a2135011SChris Mason 			WARN_ON(!btrfs_tree_locked(path->nodes[level]));
3236925baeddSChris Mason 			btrfs_tree_lock(next);
3237d97e63b6SChris Mason 		}
32385cd57b2cSChris Mason 	}
3239925baeddSChris Mason done:
3240925baeddSChris Mason 	unlock_up(path, 0, 1);
3241d97e63b6SChris Mason 	return 0;
3242d97e63b6SChris Mason }
32430b86a832SChris Mason 
3244*3f157a2fSChris Mason /*
3245*3f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
3246*3f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
3247*3f157a2fSChris Mason  *
3248*3f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
3249*3f157a2fSChris Mason  */
32500b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
32510b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
32520b86a832SChris Mason 			int type)
32530b86a832SChris Mason {
32540b86a832SChris Mason 	struct btrfs_key found_key;
32550b86a832SChris Mason 	struct extent_buffer *leaf;
32560b86a832SChris Mason 	int ret;
32570b86a832SChris Mason 
32580b86a832SChris Mason 	while(1) {
32590b86a832SChris Mason 		if (path->slots[0] == 0) {
32600b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
32610b86a832SChris Mason 			if (ret != 0)
32620b86a832SChris Mason 				return ret;
32630b86a832SChris Mason 		} else {
32640b86a832SChris Mason 			path->slots[0]--;
32650b86a832SChris Mason 		}
32660b86a832SChris Mason 		leaf = path->nodes[0];
32670b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
32680b86a832SChris Mason 		if (found_key.type == type)
32690b86a832SChris Mason 			return 0;
32700b86a832SChris Mason 	}
32710b86a832SChris Mason 	return 1;
32720b86a832SChris Mason }
32730b86a832SChris Mason 
3274