xref: /openbmc/linux/fs/btrfs/ctree.c (revision 4aec2b52327b3a3a20faa3f1af3102c0d97c7c92)
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"
249a8dd150SChris Mason 
25e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
26e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
27e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
28d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
29cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend);
305f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
315f39d397SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
325f39d397SChris Mason 			  struct extent_buffer *src);
335f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
345f39d397SChris Mason 			      struct btrfs_root *root,
355f39d397SChris Mason 			      struct extent_buffer *dst_buf,
365f39d397SChris Mason 			      struct extent_buffer *src_buf);
37e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
38e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot);
39d97e63b6SChris Mason 
40df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p)
41df24a2b9SChris Mason {
42df24a2b9SChris Mason 	memset(p, 0, sizeof(*p));
43df24a2b9SChris Mason }
44df24a2b9SChris Mason 
452c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
462c90e5d6SChris Mason {
47df24a2b9SChris Mason 	struct btrfs_path *path;
48df24a2b9SChris Mason 	path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
492cc58cf2SChris Mason 	if (path) {
50df24a2b9SChris Mason 		btrfs_init_path(path);
512cc58cf2SChris Mason 		path->reada = 1;
522cc58cf2SChris Mason 	}
53df24a2b9SChris Mason 	return path;
542c90e5d6SChris Mason }
552c90e5d6SChris Mason 
562c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
572c90e5d6SChris Mason {
58df24a2b9SChris Mason 	btrfs_release_path(NULL, p);
592c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
602c90e5d6SChris Mason }
612c90e5d6SChris Mason 
62234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
63eb60ceacSChris Mason {
64eb60ceacSChris Mason 	int i;
65234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
66eb60ceacSChris Mason 		if (!p->nodes[i])
67eb60ceacSChris Mason 			break;
685f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
69eb60ceacSChris Mason 	}
70aa5d6bedSChris Mason 	memset(p, 0, sizeof(*p));
71eb60ceacSChris Mason }
72eb60ceacSChris Mason 
73be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
74be20aa9dSChris Mason 		      struct btrfs_root *root,
75be20aa9dSChris Mason 		      struct extent_buffer *buf,
76be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
77be20aa9dSChris Mason {
78be20aa9dSChris Mason 	struct extent_buffer *cow;
79be20aa9dSChris Mason 	u32 nritems;
80be20aa9dSChris Mason 	int ret = 0;
81be20aa9dSChris Mason 	int level;
82be20aa9dSChris Mason 	struct btrfs_key first_key;
83*4aec2b52SChris Mason 	struct btrfs_root *new_root;
84be20aa9dSChris Mason 
85*4aec2b52SChris Mason 	new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
86*4aec2b52SChris Mason 	if (!new_root)
87*4aec2b52SChris Mason 		return -ENOMEM;
88*4aec2b52SChris Mason 
89*4aec2b52SChris Mason 	memcpy(new_root, root, sizeof(*new_root));
90*4aec2b52SChris Mason 	new_root->root_key.objectid = new_root_objectid;
91be20aa9dSChris Mason 
92be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
93be20aa9dSChris Mason 		root->fs_info->running_transaction->transid);
94be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
95be20aa9dSChris Mason 
96be20aa9dSChris Mason 	level = btrfs_header_level(buf);
97be20aa9dSChris Mason 	nritems = btrfs_header_nritems(buf);
98be20aa9dSChris Mason 	if (nritems) {
99be20aa9dSChris Mason 		if (level == 0)
100be20aa9dSChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
101be20aa9dSChris Mason 		else
102be20aa9dSChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
103be20aa9dSChris Mason 	} else {
104be20aa9dSChris Mason 		first_key.objectid = 0;
105be20aa9dSChris Mason 	}
106*4aec2b52SChris Mason 	cow = __btrfs_alloc_free_block(trans, new_root, buf->len,
107be20aa9dSChris Mason 				       new_root_objectid,
108be20aa9dSChris Mason 				       trans->transid, first_key.objectid,
109be20aa9dSChris Mason 				       level, buf->start, 0);
110*4aec2b52SChris Mason 	if (IS_ERR(cow)) {
111*4aec2b52SChris Mason 		kfree(new_root);
112be20aa9dSChris Mason 		return PTR_ERR(cow);
113*4aec2b52SChris Mason 	}
114be20aa9dSChris Mason 
115be20aa9dSChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
116be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
117be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
118be20aa9dSChris Mason 	btrfs_set_header_owner(cow, new_root_objectid);
119be20aa9dSChris Mason 
120be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
121*4aec2b52SChris Mason 	ret = btrfs_inc_ref(trans, new_root, buf);
122*4aec2b52SChris Mason 	kfree(new_root);
123*4aec2b52SChris Mason 
124be20aa9dSChris Mason 	if (ret)
125be20aa9dSChris Mason 		return ret;
126be20aa9dSChris Mason 
127be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
128be20aa9dSChris Mason 	*cow_ret = cow;
129be20aa9dSChris Mason 	return 0;
130be20aa9dSChris Mason }
131be20aa9dSChris Mason 
132be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans,
1335f39d397SChris Mason 			     struct btrfs_root *root,
1345f39d397SChris Mason 			     struct extent_buffer *buf,
1355f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
1365f39d397SChris Mason 			     struct extent_buffer **cow_ret,
1375f39d397SChris Mason 			     u64 search_start, u64 empty_size)
1386702ed49SChris Mason {
1397bb86316SChris Mason 	u64 root_gen;
1405f39d397SChris Mason 	struct extent_buffer *cow;
1417bb86316SChris Mason 	u32 nritems;
1426702ed49SChris Mason 	int ret = 0;
1436702ed49SChris Mason 	int different_trans = 0;
1447bb86316SChris Mason 	int level;
1457bb86316SChris Mason 	struct btrfs_key first_key;
1466702ed49SChris Mason 
1477bb86316SChris Mason 	if (root->ref_cows) {
1487bb86316SChris Mason 		root_gen = trans->transid;
1497bb86316SChris Mason 	} else {
1507bb86316SChris Mason 		root_gen = 0;
1517bb86316SChris Mason 	}
1527bb86316SChris Mason 
1537bb86316SChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
1547bb86316SChris Mason 		root->fs_info->running_transaction->transid);
1556702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
1565f39d397SChris Mason 
1577bb86316SChris Mason 	level = btrfs_header_level(buf);
1587bb86316SChris Mason 	nritems = btrfs_header_nritems(buf);
1597bb86316SChris Mason 	if (nritems) {
1607bb86316SChris Mason 		if (level == 0)
1617bb86316SChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
1627bb86316SChris Mason 		else
1637bb86316SChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
1647bb86316SChris Mason 	} else {
1657bb86316SChris Mason 		first_key.objectid = 0;
1667bb86316SChris Mason 	}
1677bb86316SChris Mason 	cow = __btrfs_alloc_free_block(trans, root, buf->len,
1687bb86316SChris Mason 				     root->root_key.objectid,
1697bb86316SChris Mason 				     root_gen, first_key.objectid, level,
170db94535dSChris Mason 				     search_start, empty_size);
1716702ed49SChris Mason 	if (IS_ERR(cow))
1726702ed49SChris Mason 		return PTR_ERR(cow);
1736702ed49SChris Mason 
1745f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
175db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
1765f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
1775f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
1786702ed49SChris Mason 
1795f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
1805f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
1816702ed49SChris Mason 		different_trans = 1;
1826702ed49SChris Mason 		ret = btrfs_inc_ref(trans, root, buf);
1836702ed49SChris Mason 		if (ret)
1846702ed49SChris Mason 			return ret;
1856702ed49SChris Mason 	} else {
1866702ed49SChris Mason 		clean_tree_block(trans, root, buf);
1876702ed49SChris Mason 	}
1886702ed49SChris Mason 
1896702ed49SChris Mason 	if (buf == root->node) {
1907bb86316SChris Mason 		root_gen = btrfs_header_generation(buf);
1916702ed49SChris Mason 		root->node = cow;
1925f39d397SChris Mason 		extent_buffer_get(cow);
1936702ed49SChris Mason 		if (buf != root->commit_root) {
194db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
1957bb86316SChris Mason 					  buf->len, root->root_key.objectid,
1967bb86316SChris Mason 					  root_gen, 0, 0, 1);
1976702ed49SChris Mason 		}
1985f39d397SChris Mason 		free_extent_buffer(buf);
1996702ed49SChris Mason 	} else {
2007bb86316SChris Mason 		root_gen = btrfs_header_generation(parent);
2015f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
202db94535dSChris Mason 					cow->start);
20374493f7aSChris Mason 		WARN_ON(trans->transid == 0);
20474493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
20574493f7aSChris Mason 					      trans->transid);
2066702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
2075f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
2087bb86316SChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len,
2097bb86316SChris Mason 				  btrfs_header_owner(parent), root_gen,
2107bb86316SChris Mason 				  0, 0, 1);
2116702ed49SChris Mason 	}
2125f39d397SChris Mason 	free_extent_buffer(buf);
2136702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
2146702ed49SChris Mason 	*cow_ret = cow;
2156702ed49SChris Mason 	return 0;
2166702ed49SChris Mason }
2176702ed49SChris Mason 
2185f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans,
2195f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
2205f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
2215f39d397SChris Mason 		    struct extent_buffer **cow_ret)
22202217ed2SChris Mason {
2236702ed49SChris Mason 	u64 search_start;
224f510cfecSChris Mason 	int ret;
225ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
226ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
227ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
228ccd467d6SChris Mason 		WARN_ON(1);
229ccd467d6SChris Mason 	}
230ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
231ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
232ccd467d6SChris Mason 		       root->fs_info->generation);
233ccd467d6SChris Mason 		WARN_ON(1);
234ccd467d6SChris Mason 	}
2355f39d397SChris Mason 	if (btrfs_header_generation(buf) == trans->transid) {
23602217ed2SChris Mason 		*cow_ret = buf;
23702217ed2SChris Mason 		return 0;
23802217ed2SChris Mason 	}
2396702ed49SChris Mason 
240db94535dSChris Mason 	search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
241f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
2426702ed49SChris Mason 				 parent_slot, cow_ret, search_start, 0);
243f510cfecSChris Mason 	return ret;
2442c90e5d6SChris Mason }
2456702ed49SChris Mason 
2466b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
2476702ed49SChris Mason {
2486b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
2496702ed49SChris Mason 		return 1;
2506b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
2516702ed49SChris Mason 		return 1;
25202217ed2SChris Mason 	return 0;
25302217ed2SChris Mason }
25402217ed2SChris Mason 
255081e9573SChris Mason /*
256081e9573SChris Mason  * compare two keys in a memcmp fashion
257081e9573SChris Mason  */
258081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
259081e9573SChris Mason {
260081e9573SChris Mason 	struct btrfs_key k1;
261081e9573SChris Mason 
262081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
263081e9573SChris Mason 
264081e9573SChris Mason 	if (k1.objectid > k2->objectid)
265081e9573SChris Mason 		return 1;
266081e9573SChris Mason 	if (k1.objectid < k2->objectid)
267081e9573SChris Mason 		return -1;
268081e9573SChris Mason 	if (k1.type > k2->type)
269081e9573SChris Mason 		return 1;
270081e9573SChris Mason 	if (k1.type < k2->type)
271081e9573SChris Mason 		return -1;
272081e9573SChris Mason 	if (k1.offset > k2->offset)
273081e9573SChris Mason 		return 1;
274081e9573SChris Mason 	if (k1.offset < k2->offset)
275081e9573SChris Mason 		return -1;
276081e9573SChris Mason 	return 0;
277081e9573SChris Mason }
278081e9573SChris Mason 
279081e9573SChris Mason 
2806702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
2815f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
282a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
283a6b6e75eSChris Mason 		       struct btrfs_key *progress)
2846702ed49SChris Mason {
2856b80053dSChris Mason 	struct extent_buffer *cur;
2866b80053dSChris Mason 	struct extent_buffer *tmp;
2876702ed49SChris Mason 	u64 blocknr;
288e9d0b13bSChris Mason 	u64 search_start = *last_ret;
289e9d0b13bSChris Mason 	u64 last_block = 0;
2906702ed49SChris Mason 	u64 other;
2916702ed49SChris Mason 	u32 parent_nritems;
2926702ed49SChris Mason 	int end_slot;
2936702ed49SChris Mason 	int i;
2946702ed49SChris Mason 	int err = 0;
295f2183bdeSChris Mason 	int parent_level;
2966b80053dSChris Mason 	int uptodate;
2976b80053dSChris Mason 	u32 blocksize;
298081e9573SChris Mason 	int progress_passed = 0;
299081e9573SChris Mason 	struct btrfs_disk_key disk_key;
3006702ed49SChris Mason 
3015708b959SChris Mason 	parent_level = btrfs_header_level(parent);
3025708b959SChris Mason 	if (cache_only && parent_level != 1)
3035708b959SChris Mason 		return 0;
3045708b959SChris Mason 
3056702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
3066702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3076702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
3086702ed49SChris Mason 		WARN_ON(1);
3096702ed49SChris Mason 	}
3106702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
3116702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3126702ed49SChris Mason 		       root->fs_info->generation);
3136702ed49SChris Mason 		WARN_ON(1);
3146702ed49SChris Mason 	}
31586479a04SChris Mason 
3166b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
3176b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
3186702ed49SChris Mason 	end_slot = parent_nritems;
3196702ed49SChris Mason 
3206702ed49SChris Mason 	if (parent_nritems == 1)
3216702ed49SChris Mason 		return 0;
3226702ed49SChris Mason 
3236702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
3246702ed49SChris Mason 		int close = 1;
325a6b6e75eSChris Mason 
3265708b959SChris Mason 		if (!parent->map_token) {
3275708b959SChris Mason 			map_extent_buffer(parent,
3285708b959SChris Mason 					btrfs_node_key_ptr_offset(i),
3295708b959SChris Mason 					sizeof(struct btrfs_key_ptr),
3305708b959SChris Mason 					&parent->map_token, &parent->kaddr,
3315708b959SChris Mason 					&parent->map_start, &parent->map_len,
3325708b959SChris Mason 					KM_USER1);
3335708b959SChris Mason 		}
334081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
335081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
336081e9573SChris Mason 			continue;
337081e9573SChris Mason 
338081e9573SChris Mason 		progress_passed = 1;
3396b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
340e9d0b13bSChris Mason 		if (last_block == 0)
341e9d0b13bSChris Mason 			last_block = blocknr;
3425708b959SChris Mason 
3436702ed49SChris Mason 		if (i > 0) {
3446b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
3456b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
3466702ed49SChris Mason 		}
3475708b959SChris Mason 		if (close && i < end_slot - 2) {
3486b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
3496b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
3506702ed49SChris Mason 		}
351e9d0b13bSChris Mason 		if (close) {
352e9d0b13bSChris Mason 			last_block = blocknr;
3536702ed49SChris Mason 			continue;
354e9d0b13bSChris Mason 		}
3555708b959SChris Mason 		if (parent->map_token) {
3565708b959SChris Mason 			unmap_extent_buffer(parent, parent->map_token,
3575708b959SChris Mason 					    KM_USER1);
3585708b959SChris Mason 			parent->map_token = NULL;
3595708b959SChris Mason 		}
3606702ed49SChris Mason 
3616b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
3626b80053dSChris Mason 		if (cur)
3636b80053dSChris Mason 			uptodate = btrfs_buffer_uptodate(cur);
3646b80053dSChris Mason 		else
3656b80053dSChris Mason 			uptodate = 0;
3665708b959SChris Mason 		if (!cur || !uptodate) {
3676702ed49SChris Mason 			if (cache_only) {
3686b80053dSChris Mason 				free_extent_buffer(cur);
3696702ed49SChris Mason 				continue;
3706702ed49SChris Mason 			}
3716b80053dSChris Mason 			if (!cur) {
3726b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
3736b80053dSChris Mason 							 blocksize);
3746b80053dSChris Mason 			} else if (!uptodate) {
3756b80053dSChris Mason 				btrfs_read_buffer(cur);
3766702ed49SChris Mason 			}
377f2183bdeSChris Mason 		}
378e9d0b13bSChris Mason 		if (search_start == 0)
3796b80053dSChris Mason 			search_start = last_block;
380e9d0b13bSChris Mason 
3816b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
3826b80053dSChris Mason 					&tmp, search_start,
3836b80053dSChris Mason 					min(16 * blocksize,
3846b80053dSChris Mason 					    (end_slot - i) * blocksize));
385252c38f0SYan 		if (err) {
3866b80053dSChris Mason 			free_extent_buffer(cur);
3876702ed49SChris Mason 			break;
388252c38f0SYan 		}
3896b80053dSChris Mason 		search_start = tmp->start;
3905708b959SChris Mason 		last_block = tmp->start;
391f2183bdeSChris Mason 		*last_ret = search_start;
392f2183bdeSChris Mason 		if (parent_level == 1)
3936b80053dSChris Mason 			btrfs_clear_buffer_defrag(tmp);
3946b80053dSChris Mason 		free_extent_buffer(tmp);
3956702ed49SChris Mason 	}
3965708b959SChris Mason 	if (parent->map_token) {
3975708b959SChris Mason 		unmap_extent_buffer(parent, parent->map_token,
3985708b959SChris Mason 				    KM_USER1);
3995708b959SChris Mason 		parent->map_token = NULL;
4005708b959SChris Mason 	}
4016702ed49SChris Mason 	return err;
4026702ed49SChris Mason }
4036702ed49SChris Mason 
40474123bd7SChris Mason /*
40574123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
40674123bd7SChris Mason  * this returns the address of the start of the last item,
40774123bd7SChris Mason  * which is the stop of the leaf data stack
40874123bd7SChris Mason  */
409123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
4105f39d397SChris Mason 					 struct extent_buffer *leaf)
411be0e5c09SChris Mason {
4125f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
413be0e5c09SChris Mason 	if (nr == 0)
414123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
4155f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
416be0e5c09SChris Mason }
417be0e5c09SChris Mason 
418123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
419123abc88SChris Mason 		      int level)
420aa5d6bedSChris Mason {
4215f39d397SChris Mason 	struct extent_buffer *parent = NULL;
4225f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
4235f39d397SChris Mason 	struct btrfs_disk_key parent_key;
4245f39d397SChris Mason 	struct btrfs_disk_key node_key;
425aa5d6bedSChris Mason 	int parent_slot;
4268d7be552SChris Mason 	int slot;
4278d7be552SChris Mason 	struct btrfs_key cpukey;
4285f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
429aa5d6bedSChris Mason 
430aa5d6bedSChris Mason 	if (path->nodes[level + 1])
4315f39d397SChris Mason 		parent = path->nodes[level + 1];
432a1f39630SAneesh 
4338d7be552SChris Mason 	slot = path->slots[level];
4347518a238SChris Mason 	BUG_ON(nritems == 0);
4357518a238SChris Mason 	if (parent) {
436a1f39630SAneesh 		parent_slot = path->slots[level + 1];
4375f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
4385f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
4395f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
440e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
4411d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
442db94535dSChris Mason 		       btrfs_header_bytenr(node));
443aa5d6bedSChris Mason 	}
444123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
4458d7be552SChris Mason 	if (slot != 0) {
4465f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
4475f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
4485f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
4498d7be552SChris Mason 	}
4508d7be552SChris Mason 	if (slot < nritems - 1) {
4515f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
4525f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
4535f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
454aa5d6bedSChris Mason 	}
455aa5d6bedSChris Mason 	return 0;
456aa5d6bedSChris Mason }
457aa5d6bedSChris Mason 
458123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
459123abc88SChris Mason 		      int level)
460aa5d6bedSChris Mason {
4615f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
4625f39d397SChris Mason 	struct extent_buffer *parent = NULL;
463aa5d6bedSChris Mason 	int parent_slot;
4648d7be552SChris Mason 	struct btrfs_key cpukey;
4655f39d397SChris Mason 	struct btrfs_disk_key parent_key;
4665f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
4675f39d397SChris Mason 	int slot = path->slots[0];
4688d7be552SChris Mason 
4695f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
470aa5d6bedSChris Mason 
471aa5d6bedSChris Mason 	if (path->nodes[level + 1])
4725f39d397SChris Mason 		parent = path->nodes[level + 1];
4737518a238SChris Mason 
4747518a238SChris Mason 	if (nritems == 0)
4757518a238SChris Mason 		return 0;
4767518a238SChris Mason 
4777518a238SChris Mason 	if (parent) {
478a1f39630SAneesh 		parent_slot = path->slots[level + 1];
4795f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
4805f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
4816702ed49SChris Mason 
4825f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
483e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
4841d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
485db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
486aa5d6bedSChris Mason 	}
4875f39d397SChris Mason #if 0
4885f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
4895f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
4905f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
4915f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
4925f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4935f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
4945f39d397SChris Mason 			BUG_ON(1);
4955f39d397SChris Mason 		}
4965f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
4975f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
4985f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4995f39d397SChris Mason 			printk("slot %d offset bad\n", i);
5005f39d397SChris Mason 			BUG_ON(1);
5015f39d397SChris Mason 		}
5025f39d397SChris Mason 		if (i == 0) {
5035f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
5045f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
5055f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
5065f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5075f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
5085f39d397SChris Mason 				BUG_ON(1);
5095f39d397SChris Mason 			}
5105f39d397SChris Mason 		}
5115f39d397SChris Mason 	}
5125f39d397SChris Mason 	if (nritems > 0) {
5135f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
5145f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5155f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
5165f39d397SChris Mason 				BUG_ON(1);
5175f39d397SChris Mason 		}
5185f39d397SChris Mason 	}
5195f39d397SChris Mason #endif
5205f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
5215f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
5225f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
5235f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
5245f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5255f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
5265f39d397SChris Mason 			BUG_ON(1);
5275f39d397SChris Mason 		}
5285f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
5295f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
5305f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5315f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
5325f39d397SChris Mason 			BUG_ON(1);
5335f39d397SChris Mason 		}
534aa5d6bedSChris Mason 	}
5358d7be552SChris Mason 	if (slot < nritems - 1) {
5365f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
5375f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
5385f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
5395f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
5405f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
5415f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5425f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
5435f39d397SChris Mason 			BUG_ON(1);
544aa5d6bedSChris Mason 		}
5455f39d397SChris Mason 	}
5465f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
5475f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
548aa5d6bedSChris Mason 	return 0;
549aa5d6bedSChris Mason }
550aa5d6bedSChris Mason 
551123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path,
552123abc88SChris Mason 			int level)
553aa5d6bedSChris Mason {
554810191ffSChris Mason 	return 0;
555db94535dSChris Mason #if 0
5565f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
5575f39d397SChris Mason 
558479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
559479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
560479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
5615f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
562db94535dSChris Mason 		return 1;
5635f39d397SChris Mason 	}
564db94535dSChris Mason #endif
565aa5d6bedSChris Mason 	if (level == 0)
566123abc88SChris Mason 		return check_leaf(root, path, level);
567123abc88SChris Mason 	return check_node(root, path, level);
568aa5d6bedSChris Mason }
569aa5d6bedSChris Mason 
57074123bd7SChris Mason /*
5715f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
5725f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
5735f39d397SChris Mason  *
57474123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
57574123bd7SChris Mason  * the place where you would insert key if it is not found in
57674123bd7SChris Mason  * the array.
57774123bd7SChris Mason  *
57874123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
57974123bd7SChris Mason  */
5805f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
5815f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
582be0e5c09SChris Mason 			      int max, int *slot)
583be0e5c09SChris Mason {
584be0e5c09SChris Mason 	int low = 0;
585be0e5c09SChris Mason 	int high = max;
586be0e5c09SChris Mason 	int mid;
587be0e5c09SChris Mason 	int ret;
588479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
5895f39d397SChris Mason 	struct btrfs_disk_key unaligned;
5905f39d397SChris Mason 	unsigned long offset;
5915f39d397SChris Mason 	char *map_token = NULL;
5925f39d397SChris Mason 	char *kaddr = NULL;
5935f39d397SChris Mason 	unsigned long map_start = 0;
5945f39d397SChris Mason 	unsigned long map_len = 0;
595479965d6SChris Mason 	int err;
596be0e5c09SChris Mason 
597be0e5c09SChris Mason 	while(low < high) {
598be0e5c09SChris Mason 		mid = (low + high) / 2;
5995f39d397SChris Mason 		offset = p + mid * item_size;
6005f39d397SChris Mason 
6015f39d397SChris Mason 		if (!map_token || offset < map_start ||
6025f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
6035f39d397SChris Mason 		    map_start + map_len) {
604479965d6SChris Mason 			if (map_token) {
6055f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
606479965d6SChris Mason 				map_token = NULL;
607479965d6SChris Mason 			}
608479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
609479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
610479965d6SChris Mason 						&map_token, &kaddr,
6115f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
6125f39d397SChris Mason 
613479965d6SChris Mason 			if (!err) {
614479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
615479965d6SChris Mason 							map_start);
616479965d6SChris Mason 			} else {
6175f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
6185f39d397SChris Mason 						   offset, sizeof(unaligned));
6195f39d397SChris Mason 				tmp = &unaligned;
620479965d6SChris Mason 			}
621479965d6SChris Mason 
6225f39d397SChris Mason 		} else {
6235f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
6245f39d397SChris Mason 							map_start);
6255f39d397SChris Mason 		}
626be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
627be0e5c09SChris Mason 
628be0e5c09SChris Mason 		if (ret < 0)
629be0e5c09SChris Mason 			low = mid + 1;
630be0e5c09SChris Mason 		else if (ret > 0)
631be0e5c09SChris Mason 			high = mid;
632be0e5c09SChris Mason 		else {
633be0e5c09SChris Mason 			*slot = mid;
634479965d6SChris Mason 			if (map_token)
6355f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
636be0e5c09SChris Mason 			return 0;
637be0e5c09SChris Mason 		}
638be0e5c09SChris Mason 	}
639be0e5c09SChris Mason 	*slot = low;
6405f39d397SChris Mason 	if (map_token)
6415f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
642be0e5c09SChris Mason 	return 1;
643be0e5c09SChris Mason }
644be0e5c09SChris Mason 
64597571fd0SChris Mason /*
64697571fd0SChris Mason  * simple bin_search frontend that does the right thing for
64797571fd0SChris Mason  * leaves vs nodes
64897571fd0SChris Mason  */
6495f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
6505f39d397SChris Mason 		      int level, int *slot)
651be0e5c09SChris Mason {
6525f39d397SChris Mason 	if (level == 0) {
6535f39d397SChris Mason 		return generic_bin_search(eb,
6545f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
6550783fcfcSChris Mason 					  sizeof(struct btrfs_item),
6565f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
6577518a238SChris Mason 					  slot);
658be0e5c09SChris Mason 	} else {
6595f39d397SChris Mason 		return generic_bin_search(eb,
6605f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
661123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
6625f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
6637518a238SChris Mason 					  slot);
664be0e5c09SChris Mason 	}
665be0e5c09SChris Mason 	return -1;
666be0e5c09SChris Mason }
667be0e5c09SChris Mason 
6685f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
6695f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
670bb803951SChris Mason {
671bb803951SChris Mason 	if (slot < 0)
672bb803951SChris Mason 		return NULL;
6735f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
674bb803951SChris Mason 		return NULL;
675db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
676db94535dSChris Mason 		       btrfs_level_size(root, btrfs_header_level(parent) - 1));
677bb803951SChris Mason }
678bb803951SChris Mason 
679e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
680e089f05cSChris Mason 			 *root, struct btrfs_path *path, int level)
681bb803951SChris Mason {
6825f39d397SChris Mason 	struct extent_buffer *right = NULL;
6835f39d397SChris Mason 	struct extent_buffer *mid;
6845f39d397SChris Mason 	struct extent_buffer *left = NULL;
6855f39d397SChris Mason 	struct extent_buffer *parent = NULL;
686bb803951SChris Mason 	int ret = 0;
687bb803951SChris Mason 	int wret;
688bb803951SChris Mason 	int pslot;
689bb803951SChris Mason 	int orig_slot = path->slots[level];
69054aa1f4dSChris Mason 	int err_on_enospc = 0;
69179f95c82SChris Mason 	u64 orig_ptr;
692bb803951SChris Mason 
693bb803951SChris Mason 	if (level == 0)
694bb803951SChris Mason 		return 0;
695bb803951SChris Mason 
6965f39d397SChris Mason 	mid = path->nodes[level];
6977bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
6987bb86316SChris Mason 
6991d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
70079f95c82SChris Mason 
701234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
7025f39d397SChris Mason 		parent = path->nodes[level + 1];
703bb803951SChris Mason 	pslot = path->slots[level + 1];
704bb803951SChris Mason 
70540689478SChris Mason 	/*
70640689478SChris Mason 	 * deal with the case where there is only one pointer in the root
70740689478SChris Mason 	 * by promoting the node below to a root
70840689478SChris Mason 	 */
7095f39d397SChris Mason 	if (!parent) {
7105f39d397SChris Mason 		struct extent_buffer *child;
711bb803951SChris Mason 
7125f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
713bb803951SChris Mason 			return 0;
714bb803951SChris Mason 
715bb803951SChris Mason 		/* promote the child to a root */
7165f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
717bb803951SChris Mason 		BUG_ON(!child);
718bb803951SChris Mason 		root->node = child;
719bb803951SChris Mason 		path->nodes[level] = NULL;
7205f39d397SChris Mason 		clean_tree_block(trans, root, mid);
7215f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
722bb803951SChris Mason 		/* once for the path */
7235f39d397SChris Mason 		free_extent_buffer(mid);
7247bb86316SChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
7257bb86316SChris Mason 					root->root_key.objectid,
7267bb86316SChris Mason 					btrfs_header_generation(mid), 0, 0, 1);
727bb803951SChris Mason 		/* once for the root ptr */
7285f39d397SChris Mason 		free_extent_buffer(mid);
729db94535dSChris Mason 		return ret;
730bb803951SChris Mason 	}
7315f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
732123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
733bb803951SChris Mason 		return 0;
734bb803951SChris Mason 
7355f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
73654aa1f4dSChris Mason 		err_on_enospc = 1;
73754aa1f4dSChris Mason 
7385f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
7395f39d397SChris Mason 	if (left) {
7405f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
7415f39d397SChris Mason 				       parent, pslot - 1, &left);
74254aa1f4dSChris Mason 		if (wret) {
74354aa1f4dSChris Mason 			ret = wret;
74454aa1f4dSChris Mason 			goto enospc;
74554aa1f4dSChris Mason 		}
7462cc58cf2SChris Mason 	}
7475f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
7485f39d397SChris Mason 	if (right) {
7495f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
7505f39d397SChris Mason 				       parent, pslot + 1, &right);
7512cc58cf2SChris Mason 		if (wret) {
7522cc58cf2SChris Mason 			ret = wret;
7532cc58cf2SChris Mason 			goto enospc;
7542cc58cf2SChris Mason 		}
7552cc58cf2SChris Mason 	}
7562cc58cf2SChris Mason 
7572cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
7585f39d397SChris Mason 	if (left) {
7595f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
7605f39d397SChris Mason 		wret = push_node_left(trans, root, left, mid);
76179f95c82SChris Mason 		if (wret < 0)
76279f95c82SChris Mason 			ret = wret;
7635f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
76454aa1f4dSChris Mason 			err_on_enospc = 1;
765bb803951SChris Mason 	}
76679f95c82SChris Mason 
76779f95c82SChris Mason 	/*
76879f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
76979f95c82SChris Mason 	 */
7705f39d397SChris Mason 	if (right) {
7715f39d397SChris Mason 		wret = push_node_left(trans, root, mid, right);
77254aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
77379f95c82SChris Mason 			ret = wret;
7745f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
775db94535dSChris Mason 			u64 bytenr = right->start;
7767bb86316SChris Mason 			u64 generation = btrfs_header_generation(parent);
777db94535dSChris Mason 			u32 blocksize = right->len;
778db94535dSChris Mason 
7795f39d397SChris Mason 			clean_tree_block(trans, root, right);
7805f39d397SChris Mason 			wait_on_tree_block_writeback(root, right);
7815f39d397SChris Mason 			free_extent_buffer(right);
782bb803951SChris Mason 			right = NULL;
783e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
784e089f05cSChris Mason 				       1);
785bb803951SChris Mason 			if (wret)
786bb803951SChris Mason 				ret = wret;
787db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
7887bb86316SChris Mason 						 blocksize,
7897bb86316SChris Mason 						 btrfs_header_owner(parent),
7907bb86316SChris Mason 						 generation, 0, 0, 1);
791bb803951SChris Mason 			if (wret)
792bb803951SChris Mason 				ret = wret;
793bb803951SChris Mason 		} else {
7945f39d397SChris Mason 			struct btrfs_disk_key right_key;
7955f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
7965f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
7975f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
798bb803951SChris Mason 		}
799bb803951SChris Mason 	}
8005f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
80179f95c82SChris Mason 		/*
80279f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
80379f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
80479f95c82SChris Mason 		 * could try to delete the only pointer in this node.
80579f95c82SChris Mason 		 * So, pull some keys from the left.
80679f95c82SChris Mason 		 * There has to be a left pointer at this point because
80779f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
80879f95c82SChris Mason 		 * right
80979f95c82SChris Mason 		 */
8105f39d397SChris Mason 		BUG_ON(!left);
8115f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
81254aa1f4dSChris Mason 		if (wret < 0) {
81379f95c82SChris Mason 			ret = wret;
81454aa1f4dSChris Mason 			goto enospc;
81554aa1f4dSChris Mason 		}
81679f95c82SChris Mason 		BUG_ON(wret == 1);
81779f95c82SChris Mason 	}
8185f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
81979f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
8207bb86316SChris Mason 		u64 root_gen = btrfs_header_generation(parent);
821db94535dSChris Mason 		u64 bytenr = mid->start;
822db94535dSChris Mason 		u32 blocksize = mid->len;
8235f39d397SChris Mason 		clean_tree_block(trans, root, mid);
8245f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
8255f39d397SChris Mason 		free_extent_buffer(mid);
826bb803951SChris Mason 		mid = NULL;
827e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
828bb803951SChris Mason 		if (wret)
829bb803951SChris Mason 			ret = wret;
8307bb86316SChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
8317bb86316SChris Mason 					 btrfs_header_owner(parent),
8327bb86316SChris Mason 					 root_gen, 0, 0, 1);
833bb803951SChris Mason 		if (wret)
834bb803951SChris Mason 			ret = wret;
83579f95c82SChris Mason 	} else {
83679f95c82SChris Mason 		/* update the parent key to reflect our changes */
8375f39d397SChris Mason 		struct btrfs_disk_key mid_key;
8385f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
8395f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
8405f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
84179f95c82SChris Mason 	}
842bb803951SChris Mason 
84379f95c82SChris Mason 	/* update the path */
8445f39d397SChris Mason 	if (left) {
8455f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
8465f39d397SChris Mason 			extent_buffer_get(left);
8475f39d397SChris Mason 			path->nodes[level] = left;
848bb803951SChris Mason 			path->slots[level + 1] -= 1;
849bb803951SChris Mason 			path->slots[level] = orig_slot;
8505f39d397SChris Mason 			if (mid)
8515f39d397SChris Mason 				free_extent_buffer(mid);
852bb803951SChris Mason 		} else {
8535f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
854bb803951SChris Mason 			path->slots[level] = orig_slot;
855bb803951SChris Mason 		}
856bb803951SChris Mason 	}
85779f95c82SChris Mason 	/* double check we haven't messed things up */
858123abc88SChris Mason 	check_block(root, path, level);
859e20d96d6SChris Mason 	if (orig_ptr !=
8605f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
86179f95c82SChris Mason 		BUG();
86254aa1f4dSChris Mason enospc:
8635f39d397SChris Mason 	if (right)
8645f39d397SChris Mason 		free_extent_buffer(right);
8655f39d397SChris Mason 	if (left)
8665f39d397SChris Mason 		free_extent_buffer(left);
867bb803951SChris Mason 	return ret;
868bb803951SChris Mason }
869bb803951SChris Mason 
870e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
871e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
872e66f709bSChris Mason 				struct btrfs_root *root,
873e66f709bSChris Mason 				struct btrfs_path *path, int level)
874e66f709bSChris Mason {
8755f39d397SChris Mason 	struct extent_buffer *right = NULL;
8765f39d397SChris Mason 	struct extent_buffer *mid;
8775f39d397SChris Mason 	struct extent_buffer *left = NULL;
8785f39d397SChris Mason 	struct extent_buffer *parent = NULL;
879e66f709bSChris Mason 	int ret = 0;
880e66f709bSChris Mason 	int wret;
881e66f709bSChris Mason 	int pslot;
882e66f709bSChris Mason 	int orig_slot = path->slots[level];
883e66f709bSChris Mason 	u64 orig_ptr;
884e66f709bSChris Mason 
885e66f709bSChris Mason 	if (level == 0)
886e66f709bSChris Mason 		return 1;
887e66f709bSChris Mason 
8885f39d397SChris Mason 	mid = path->nodes[level];
8897bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
890e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
891e66f709bSChris Mason 
892e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
8935f39d397SChris Mason 		parent = path->nodes[level + 1];
894e66f709bSChris Mason 	pslot = path->slots[level + 1];
895e66f709bSChris Mason 
8965f39d397SChris Mason 	if (!parent)
897e66f709bSChris Mason 		return 1;
898e66f709bSChris Mason 
8995f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
900e66f709bSChris Mason 
901e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
9025f39d397SChris Mason 	if (left) {
903e66f709bSChris Mason 		u32 left_nr;
9045f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
90533ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
90633ade1f8SChris Mason 			wret = 1;
90733ade1f8SChris Mason 		} else {
9085f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
9095f39d397SChris Mason 					      pslot - 1, &left);
91054aa1f4dSChris Mason 			if (ret)
91154aa1f4dSChris Mason 				wret = 1;
91254aa1f4dSChris Mason 			else {
91354aa1f4dSChris Mason 				wret = push_node_left(trans, root,
9145f39d397SChris Mason 						      left, mid);
91554aa1f4dSChris Mason 			}
91633ade1f8SChris Mason 		}
917e66f709bSChris Mason 		if (wret < 0)
918e66f709bSChris Mason 			ret = wret;
919e66f709bSChris Mason 		if (wret == 0) {
9205f39d397SChris Mason 			struct btrfs_disk_key disk_key;
921e66f709bSChris Mason 			orig_slot += left_nr;
9225f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
9235f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
9245f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
9255f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
9265f39d397SChris Mason 				path->nodes[level] = left;
927e66f709bSChris Mason 				path->slots[level + 1] -= 1;
928e66f709bSChris Mason 				path->slots[level] = orig_slot;
9295f39d397SChris Mason 				free_extent_buffer(mid);
930e66f709bSChris Mason 			} else {
931e66f709bSChris Mason 				orig_slot -=
9325f39d397SChris Mason 					btrfs_header_nritems(left);
933e66f709bSChris Mason 				path->slots[level] = orig_slot;
9345f39d397SChris Mason 				free_extent_buffer(left);
935e66f709bSChris Mason 			}
936e66f709bSChris Mason 			return 0;
937e66f709bSChris Mason 		}
9385f39d397SChris Mason 		free_extent_buffer(left);
939e66f709bSChris Mason 	}
9405f39d397SChris Mason 	right= read_node_slot(root, parent, pslot + 1);
941e66f709bSChris Mason 
942e66f709bSChris Mason 	/*
943e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
944e66f709bSChris Mason 	 */
9455f39d397SChris Mason 	if (right) {
94633ade1f8SChris Mason 		u32 right_nr;
9475f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
94833ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
94933ade1f8SChris Mason 			wret = 1;
95033ade1f8SChris Mason 		} else {
9515f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
9525f39d397SChris Mason 					      parent, pslot + 1,
9535f39d397SChris Mason 					      &right);
95454aa1f4dSChris Mason 			if (ret)
95554aa1f4dSChris Mason 				wret = 1;
95654aa1f4dSChris Mason 			else {
95733ade1f8SChris Mason 				wret = balance_node_right(trans, root,
9585f39d397SChris Mason 							  right, mid);
95933ade1f8SChris Mason 			}
96054aa1f4dSChris Mason 		}
961e66f709bSChris Mason 		if (wret < 0)
962e66f709bSChris Mason 			ret = wret;
963e66f709bSChris Mason 		if (wret == 0) {
9645f39d397SChris Mason 			struct btrfs_disk_key disk_key;
9655f39d397SChris Mason 
9665f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
9675f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
9685f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
9695f39d397SChris Mason 
9705f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
9715f39d397SChris Mason 				path->nodes[level] = right;
972e66f709bSChris Mason 				path->slots[level + 1] += 1;
973e66f709bSChris Mason 				path->slots[level] = orig_slot -
9745f39d397SChris Mason 					btrfs_header_nritems(mid);
9755f39d397SChris Mason 				free_extent_buffer(mid);
976e66f709bSChris Mason 			} else {
9775f39d397SChris Mason 				free_extent_buffer(right);
978e66f709bSChris Mason 			}
979e66f709bSChris Mason 			return 0;
980e66f709bSChris Mason 		}
9815f39d397SChris Mason 		free_extent_buffer(right);
982e66f709bSChris Mason 	}
983e66f709bSChris Mason 	return 1;
984e66f709bSChris Mason }
985e66f709bSChris Mason 
98674123bd7SChris Mason /*
9873c69faecSChris Mason  * readahead one full node of leaves
9883c69faecSChris Mason  */
9893c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
9906702ed49SChris Mason 			     int level, int slot)
9913c69faecSChris Mason {
9925f39d397SChris Mason 	struct extent_buffer *node;
9933c69faecSChris Mason 	u32 nritems;
9943c69faecSChris Mason 	u64 search;
9956b80053dSChris Mason 	u64 lowest_read;
9966b80053dSChris Mason 	u64 highest_read;
9976b80053dSChris Mason 	u64 nread = 0;
9983c69faecSChris Mason 	int direction = path->reada;
9995f39d397SChris Mason 	struct extent_buffer *eb;
10006b80053dSChris Mason 	u32 nr;
10016b80053dSChris Mason 	u32 blocksize;
10026b80053dSChris Mason 	u32 nscan = 0;
1003db94535dSChris Mason 
1004a6b6e75eSChris Mason 	if (level != 1)
10053c69faecSChris Mason 		return;
10063c69faecSChris Mason 
10076702ed49SChris Mason 	if (!path->nodes[level])
10086702ed49SChris Mason 		return;
10096702ed49SChris Mason 
10105f39d397SChris Mason 	node = path->nodes[level];
10113c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
10126b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
10136b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
10145f39d397SChris Mason 	if (eb) {
10155f39d397SChris Mason 		free_extent_buffer(eb);
10163c69faecSChris Mason 		return;
10173c69faecSChris Mason 	}
10183c69faecSChris Mason 
10196b80053dSChris Mason 	highest_read = search;
10206b80053dSChris Mason 	lowest_read = search;
10216b80053dSChris Mason 
10225f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
10236b80053dSChris Mason 	nr = slot;
10243c69faecSChris Mason 	while(1) {
10256b80053dSChris Mason 		if (direction < 0) {
10266b80053dSChris Mason 			if (nr == 0)
10273c69faecSChris Mason 				break;
10286b80053dSChris Mason 			nr--;
10296b80053dSChris Mason 		} else if (direction > 0) {
10306b80053dSChris Mason 			nr++;
10316b80053dSChris Mason 			if (nr >= nritems)
10326b80053dSChris Mason 				break;
10333c69faecSChris Mason 		}
10346b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
10356b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
10366b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
10376b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
10386b80053dSChris Mason 			readahead_tree_block(root, search, blocksize);
10396b80053dSChris Mason 			nread += blocksize;
10403c69faecSChris Mason 		}
10416b80053dSChris Mason 		nscan++;
10426b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
10436b80053dSChris Mason 			break;
10446b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
10456b80053dSChris Mason 			break;
10466b80053dSChris Mason 
10476b80053dSChris Mason 		if (search < lowest_read)
10486b80053dSChris Mason 			lowest_read = search;
10496b80053dSChris Mason 		if (search > highest_read)
10506b80053dSChris Mason 			highest_read = search;
10513c69faecSChris Mason 	}
10523c69faecSChris Mason }
10533c69faecSChris Mason /*
105474123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
105574123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
105674123bd7SChris Mason  * level of the path (level 0)
105774123bd7SChris Mason  *
105874123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
1059aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
1060aa5d6bedSChris Mason  * search a negative error number is returned.
106197571fd0SChris Mason  *
106297571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
106397571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
106497571fd0SChris Mason  * possible)
106574123bd7SChris Mason  */
1066e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
1067e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
1068e089f05cSChris Mason 		      ins_len, int cow)
1069be0e5c09SChris Mason {
10705f39d397SChris Mason 	struct extent_buffer *b;
1071db94535dSChris Mason 	u64 bytenr;
107274493f7aSChris Mason 	u64 ptr_gen;
1073be0e5c09SChris Mason 	int slot;
1074be0e5c09SChris Mason 	int ret;
1075be0e5c09SChris Mason 	int level;
10763c69faecSChris Mason 	int should_reada = p->reada;
10779f3a7427SChris Mason 	u8 lowest_level = 0;
10789f3a7427SChris Mason 
10796702ed49SChris Mason 	lowest_level = p->lowest_level;
10806702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
108122b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
108222b0ebdaSChris Mason 	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
1083bb803951SChris Mason again:
1084bb803951SChris Mason 	b = root->node;
10855f39d397SChris Mason 	extent_buffer_get(b);
1086eb60ceacSChris Mason 	while (b) {
10875f39d397SChris Mason 		level = btrfs_header_level(b);
108802217ed2SChris Mason 		if (cow) {
108902217ed2SChris Mason 			int wret;
1090e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
1091e20d96d6SChris Mason 					       p->nodes[level + 1],
1092e20d96d6SChris Mason 					       p->slots[level + 1],
1093252c38f0SYan 					       &b);
109454aa1f4dSChris Mason 			if (wret) {
10955f39d397SChris Mason 				free_extent_buffer(b);
109654aa1f4dSChris Mason 				return wret;
109754aa1f4dSChris Mason 			}
109802217ed2SChris Mason 		}
109902217ed2SChris Mason 		BUG_ON(!cow && ins_len);
11005f39d397SChris Mason 		if (level != btrfs_header_level(b))
11012c90e5d6SChris Mason 			WARN_ON(1);
11025f39d397SChris Mason 		level = btrfs_header_level(b);
1103eb60ceacSChris Mason 		p->nodes[level] = b;
1104123abc88SChris Mason 		ret = check_block(root, p, level);
1105aa5d6bedSChris Mason 		if (ret)
1106aa5d6bedSChris Mason 			return -1;
11075f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
11085f39d397SChris Mason 		if (level != 0) {
1109be0e5c09SChris Mason 			if (ret && slot > 0)
1110be0e5c09SChris Mason 				slot -= 1;
1111be0e5c09SChris Mason 			p->slots[level] = slot;
11125f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
1113d4dbff95SChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
1114e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
11155c680ed6SChris Mason 				BUG_ON(sret > 0);
11165c680ed6SChris Mason 				if (sret)
11175c680ed6SChris Mason 					return sret;
11185c680ed6SChris Mason 				b = p->nodes[level];
11195c680ed6SChris Mason 				slot = p->slots[level];
1120bb803951SChris Mason 			} else if (ins_len < 0) {
1121e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1122e089f05cSChris Mason 							 level);
1123bb803951SChris Mason 				if (sret)
1124bb803951SChris Mason 					return sret;
1125bb803951SChris Mason 				b = p->nodes[level];
1126f510cfecSChris Mason 				if (!b) {
1127f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1128bb803951SChris Mason 					goto again;
1129f510cfecSChris Mason 				}
1130bb803951SChris Mason 				slot = p->slots[level];
11315f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
11325c680ed6SChris Mason 			}
11339f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
11349f3a7427SChris Mason 			if (level == lowest_level)
11359f3a7427SChris Mason 				break;
1136db94535dSChris Mason 			bytenr = btrfs_node_blockptr(b, slot);
113774493f7aSChris Mason 			ptr_gen = btrfs_node_ptr_generation(b, slot);
11386702ed49SChris Mason 			if (should_reada)
11396702ed49SChris Mason 				reada_for_search(root, p, level, slot);
1140db94535dSChris Mason 			b = read_tree_block(root, bytenr,
1141db94535dSChris Mason 					    btrfs_level_size(root, level - 1));
114274493f7aSChris Mason 			if (ptr_gen != btrfs_header_generation(b)) {
114374493f7aSChris Mason 				printk("block %llu bad gen wanted %llu "
114474493f7aSChris Mason 				       "found %llu\n",
114574493f7aSChris Mason 			        (unsigned long long)b->start,
114674493f7aSChris Mason 				(unsigned long long)ptr_gen,
114774493f7aSChris Mason 			        (unsigned long long)btrfs_header_generation(b));
114874493f7aSChris Mason 			}
1149be0e5c09SChris Mason 		} else {
1150be0e5c09SChris Mason 			p->slots[level] = slot;
11515f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
11520783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1153d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1154cc0c5538SChris Mason 						      p, ins_len, ret == 0);
11555c680ed6SChris Mason 				BUG_ON(sret > 0);
11565c680ed6SChris Mason 				if (sret)
11575c680ed6SChris Mason 					return sret;
11585c680ed6SChris Mason 			}
1159be0e5c09SChris Mason 			return ret;
1160be0e5c09SChris Mason 		}
1161be0e5c09SChris Mason 	}
1162aa5d6bedSChris Mason 	return 1;
1163be0e5c09SChris Mason }
1164be0e5c09SChris Mason 
116574123bd7SChris Mason /*
116674123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
116774123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
116874123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
116974123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
117074123bd7SChris Mason  * higher levels
1171aa5d6bedSChris Mason  *
1172aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1173aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
117474123bd7SChris Mason  */
11755f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
11765f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
11775f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1178be0e5c09SChris Mason {
1179be0e5c09SChris Mason 	int i;
1180aa5d6bedSChris Mason 	int ret = 0;
11815f39d397SChris Mason 	struct extent_buffer *t;
11825f39d397SChris Mason 
1183234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1184be0e5c09SChris Mason 		int tslot = path->slots[i];
1185eb60ceacSChris Mason 		if (!path->nodes[i])
1186be0e5c09SChris Mason 			break;
11875f39d397SChris Mason 		t = path->nodes[i];
11885f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1189d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1190be0e5c09SChris Mason 		if (tslot != 0)
1191be0e5c09SChris Mason 			break;
1192be0e5c09SChris Mason 	}
1193aa5d6bedSChris Mason 	return ret;
1194be0e5c09SChris Mason }
1195be0e5c09SChris Mason 
119674123bd7SChris Mason /*
119774123bd7SChris Mason  * try to push data from one node into the next node left in the
119879f95c82SChris Mason  * tree.
1199aa5d6bedSChris Mason  *
1200aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1201aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
120274123bd7SChris Mason  */
1203e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
12045f39d397SChris Mason 			  *root, struct extent_buffer *dst,
12055f39d397SChris Mason 			  struct extent_buffer *src)
1206be0e5c09SChris Mason {
1207be0e5c09SChris Mason 	int push_items = 0;
1208bb803951SChris Mason 	int src_nritems;
1209bb803951SChris Mason 	int dst_nritems;
1210aa5d6bedSChris Mason 	int ret = 0;
1211be0e5c09SChris Mason 
12125f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
12135f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1214123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
12157bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
12167bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
121754aa1f4dSChris Mason 
1218eb60ceacSChris Mason 	if (push_items <= 0) {
1219be0e5c09SChris Mason 		return 1;
1220eb60ceacSChris Mason 	}
1221be0e5c09SChris Mason 
1222be0e5c09SChris Mason 	if (src_nritems < push_items)
1223be0e5c09SChris Mason 		push_items = src_nritems;
122479f95c82SChris Mason 
12255f39d397SChris Mason 	copy_extent_buffer(dst, src,
12265f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
12275f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1228123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
12295f39d397SChris Mason 
1230bb803951SChris Mason 	if (push_items < src_nritems) {
12315f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
12325f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1233e2fa7227SChris Mason 				      (src_nritems - push_items) *
1234123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1235bb803951SChris Mason 	}
12365f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
12375f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
12385f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
12395f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1240bb803951SChris Mason 	return ret;
1241be0e5c09SChris Mason }
1242be0e5c09SChris Mason 
124397571fd0SChris Mason /*
124479f95c82SChris Mason  * try to push data from one node into the next node right in the
124579f95c82SChris Mason  * tree.
124679f95c82SChris Mason  *
124779f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
124879f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
124979f95c82SChris Mason  *
125079f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
125179f95c82SChris Mason  */
12525f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
12535f39d397SChris Mason 			      struct btrfs_root *root,
12545f39d397SChris Mason 			      struct extent_buffer *dst,
12555f39d397SChris Mason 			      struct extent_buffer *src)
125679f95c82SChris Mason {
125779f95c82SChris Mason 	int push_items = 0;
125879f95c82SChris Mason 	int max_push;
125979f95c82SChris Mason 	int src_nritems;
126079f95c82SChris Mason 	int dst_nritems;
126179f95c82SChris Mason 	int ret = 0;
126279f95c82SChris Mason 
12637bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
12647bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
12657bb86316SChris Mason 
12665f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
12675f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1268123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
12695f39d397SChris Mason 	if (push_items <= 0)
127079f95c82SChris Mason 		return 1;
127179f95c82SChris Mason 
127279f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
127379f95c82SChris Mason 	/* don't try to empty the node */
1274252c38f0SYan 	if (max_push >= src_nritems)
127579f95c82SChris Mason 		return 1;
1276252c38f0SYan 
127779f95c82SChris Mason 	if (max_push < push_items)
127879f95c82SChris Mason 		push_items = max_push;
127979f95c82SChris Mason 
12805f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
12815f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
12825f39d397SChris Mason 				      (dst_nritems) *
12835f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1284d6025579SChris Mason 
12855f39d397SChris Mason 	copy_extent_buffer(dst, src,
12865f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
12875f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1288123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
128979f95c82SChris Mason 
12905f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
12915f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
129279f95c82SChris Mason 
12935f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
12945f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
129579f95c82SChris Mason 	return ret;
129679f95c82SChris Mason }
129779f95c82SChris Mason 
129879f95c82SChris Mason /*
129997571fd0SChris Mason  * helper function to insert a new root level in the tree.
130097571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
130197571fd0SChris Mason  * point to the existing root
1302aa5d6bedSChris Mason  *
1303aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
130497571fd0SChris Mason  */
13055f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans,
13065f39d397SChris Mason 			   struct btrfs_root *root,
13075f39d397SChris Mason 			   struct btrfs_path *path, int level)
130874123bd7SChris Mason {
13097bb86316SChris Mason 	u64 root_gen;
13107bb86316SChris Mason 	u64 lower_gen;
13115f39d397SChris Mason 	struct extent_buffer *lower;
13125f39d397SChris Mason 	struct extent_buffer *c;
13135f39d397SChris Mason 	struct btrfs_disk_key lower_key;
13145c680ed6SChris Mason 
13155c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
13165c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
13175c680ed6SChris Mason 
13187bb86316SChris Mason 	if (root->ref_cows)
13197bb86316SChris Mason 		root_gen = trans->transid;
13207bb86316SChris Mason 	else
13217bb86316SChris Mason 		root_gen = 0;
13227bb86316SChris Mason 
13237bb86316SChris Mason 	lower = path->nodes[level-1];
13247bb86316SChris Mason 	if (level == 1)
13257bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
13267bb86316SChris Mason 	else
13277bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
13287bb86316SChris Mason 
13297bb86316SChris Mason 	c = __btrfs_alloc_free_block(trans, root, root->nodesize,
13307bb86316SChris Mason 				   root->root_key.objectid,
13317bb86316SChris Mason 				   root_gen, lower_key.objectid, level,
1332db94535dSChris Mason 				   root->node->start, 0);
13335f39d397SChris Mason 	if (IS_ERR(c))
13345f39d397SChris Mason 		return PTR_ERR(c);
13355f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
13365f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
13375f39d397SChris Mason 	btrfs_set_header_level(c, level);
1338db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
13395f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
13405f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
1341d5719762SChris Mason 
13425f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
13435f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
13445f39d397SChris Mason 			    BTRFS_FSID_SIZE);
13455f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1346db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
13477bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
13487bb86316SChris Mason 	WARN_ON(lower_gen == 0);
13497bb86316SChris Mason 
13507bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
13515f39d397SChris Mason 
13525f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1353d5719762SChris Mason 
1354cfaa7295SChris Mason 	/* the super has an extra ref to root->node */
13555f39d397SChris Mason 	free_extent_buffer(root->node);
13565f39d397SChris Mason 	root->node = c;
13575f39d397SChris Mason 	extent_buffer_get(c);
13585f39d397SChris Mason 	path->nodes[level] = c;
135974123bd7SChris Mason 	path->slots[level] = 0;
13607bb86316SChris Mason 
13617bb86316SChris Mason 	if (root->ref_cows && lower_gen != trans->transid) {
13627bb86316SChris Mason 		struct btrfs_path *back_path = btrfs_alloc_path();
13637bb86316SChris Mason 		int ret;
13647bb86316SChris Mason 		ret = btrfs_insert_extent_backref(trans,
13657bb86316SChris Mason 						  root->fs_info->extent_root,
13667bb86316SChris Mason 						  path, lower->start,
13677bb86316SChris Mason 						  root->root_key.objectid,
13687bb86316SChris Mason 						  trans->transid, 0, 0);
13697bb86316SChris Mason 		BUG_ON(ret);
13707bb86316SChris Mason 		btrfs_free_path(back_path);
13717bb86316SChris Mason 	}
137274123bd7SChris Mason 	return 0;
137374123bd7SChris Mason }
13745c680ed6SChris Mason 
13755c680ed6SChris Mason /*
13765c680ed6SChris Mason  * worker function to insert a single pointer in a node.
13775c680ed6SChris Mason  * the node should have enough room for the pointer already
137897571fd0SChris Mason  *
13795c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
13805c680ed6SChris Mason  * blocknr is the block the key points to.
1381aa5d6bedSChris Mason  *
1382aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
13835c680ed6SChris Mason  */
1384e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1385e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1386db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
13875c680ed6SChris Mason {
13885f39d397SChris Mason 	struct extent_buffer *lower;
13895c680ed6SChris Mason 	int nritems;
13905c680ed6SChris Mason 
13915c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
13925f39d397SChris Mason 	lower = path->nodes[level];
13935f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
139474123bd7SChris Mason 	if (slot > nritems)
139574123bd7SChris Mason 		BUG();
1396123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
139774123bd7SChris Mason 		BUG();
139874123bd7SChris Mason 	if (slot != nritems) {
13995f39d397SChris Mason 		memmove_extent_buffer(lower,
14005f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
14015f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1402123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
140374123bd7SChris Mason 	}
14045f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1405db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
140674493f7aSChris Mason 	WARN_ON(trans->transid == 0);
140774493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
14085f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
14095f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
141074123bd7SChris Mason 	return 0;
141174123bd7SChris Mason }
141274123bd7SChris Mason 
141397571fd0SChris Mason /*
141497571fd0SChris Mason  * split the node at the specified level in path in two.
141597571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
141697571fd0SChris Mason  *
141797571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
141897571fd0SChris Mason  * left and right, if either one works, it returns right away.
1419aa5d6bedSChris Mason  *
1420aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
142197571fd0SChris Mason  */
1422e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1423e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1424be0e5c09SChris Mason {
14257bb86316SChris Mason 	u64 root_gen;
14265f39d397SChris Mason 	struct extent_buffer *c;
14275f39d397SChris Mason 	struct extent_buffer *split;
14285f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1429be0e5c09SChris Mason 	int mid;
14305c680ed6SChris Mason 	int ret;
1431aa5d6bedSChris Mason 	int wret;
14327518a238SChris Mason 	u32 c_nritems;
1433be0e5c09SChris Mason 
14345f39d397SChris Mason 	c = path->nodes[level];
14357bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
14365f39d397SChris Mason 	if (c == root->node) {
14375c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1438e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
14395c680ed6SChris Mason 		if (ret)
14405c680ed6SChris Mason 			return ret;
1441e66f709bSChris Mason 	} else {
1442e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
14435f39d397SChris Mason 		c = path->nodes[level];
14445f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1445e66f709bSChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1446e66f709bSChris Mason 			return 0;
144754aa1f4dSChris Mason 		if (ret < 0)
144854aa1f4dSChris Mason 			return ret;
14495c680ed6SChris Mason 	}
1450e66f709bSChris Mason 
14515f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
14527bb86316SChris Mason 	if (root->ref_cows)
14537bb86316SChris Mason 		root_gen = trans->transid;
14547bb86316SChris Mason 	else
14557bb86316SChris Mason 		root_gen = 0;
14567bb86316SChris Mason 
14577bb86316SChris Mason 	btrfs_node_key(c, &disk_key, 0);
14587bb86316SChris Mason 	split = __btrfs_alloc_free_block(trans, root, root->nodesize,
14597bb86316SChris Mason 					 root->root_key.objectid,
14607bb86316SChris Mason 					 root_gen,
14617bb86316SChris Mason 					 btrfs_disk_key_objectid(&disk_key),
14627bb86316SChris Mason 					 level, c->start, 0);
14635f39d397SChris Mason 	if (IS_ERR(split))
14645f39d397SChris Mason 		return PTR_ERR(split);
146554aa1f4dSChris Mason 
14665f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
14675f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1468db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
14695f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
14705f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
14715f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
14725f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
14735f39d397SChris Mason 			    BTRFS_FSID_SIZE);
14745f39d397SChris Mason 
14757518a238SChris Mason 	mid = (c_nritems + 1) / 2;
14765f39d397SChris Mason 
14775f39d397SChris Mason 	copy_extent_buffer(split, c,
14785f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
14795f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1480123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
14815f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
14825f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1483aa5d6bedSChris Mason 	ret = 0;
1484aa5d6bedSChris Mason 
14855f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
14865f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
14875f39d397SChris Mason 
14885f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1489db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
14905f39d397SChris Mason 			  path->slots[level + 1] + 1,
1491123abc88SChris Mason 			  level + 1);
1492aa5d6bedSChris Mason 	if (wret)
1493aa5d6bedSChris Mason 		ret = wret;
1494aa5d6bedSChris Mason 
14955de08d7dSChris Mason 	if (path->slots[level] >= mid) {
14965c680ed6SChris Mason 		path->slots[level] -= mid;
14975f39d397SChris Mason 		free_extent_buffer(c);
14985f39d397SChris Mason 		path->nodes[level] = split;
14995c680ed6SChris Mason 		path->slots[level + 1] += 1;
1500eb60ceacSChris Mason 	} else {
15015f39d397SChris Mason 		free_extent_buffer(split);
1502be0e5c09SChris Mason 	}
1503aa5d6bedSChris Mason 	return ret;
1504be0e5c09SChris Mason }
1505be0e5c09SChris Mason 
150674123bd7SChris Mason /*
150774123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
150874123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
150974123bd7SChris Mason  * space used both by the item structs and the item data
151074123bd7SChris Mason  */
15115f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1512be0e5c09SChris Mason {
1513be0e5c09SChris Mason 	int data_len;
15145f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1515d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1516be0e5c09SChris Mason 
1517be0e5c09SChris Mason 	if (!nr)
1518be0e5c09SChris Mason 		return 0;
15195f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
15205f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
15210783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1522d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1523be0e5c09SChris Mason 	return data_len;
1524be0e5c09SChris Mason }
1525be0e5c09SChris Mason 
152674123bd7SChris Mason /*
1527d4dbff95SChris Mason  * The space between the end of the leaf items and
1528d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1529d4dbff95SChris Mason  * the leaf has left for both items and data
1530d4dbff95SChris Mason  */
15315f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1532d4dbff95SChris Mason {
15335f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
15345f39d397SChris Mason 	int ret;
15355f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
15365f39d397SChris Mason 	if (ret < 0) {
15375f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1538ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
15395f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
15405f39d397SChris Mason 	}
15415f39d397SChris Mason 	return ret;
1542d4dbff95SChris Mason }
1543d4dbff95SChris Mason 
1544d4dbff95SChris Mason /*
154500ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
154600ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1547aa5d6bedSChris Mason  *
1548aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1549aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
155000ec4c51SChris Mason  */
1551e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
155234a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
155334a38218SChris Mason 			   int empty)
155400ec4c51SChris Mason {
15555f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
15565f39d397SChris Mason 	struct extent_buffer *right;
15575f39d397SChris Mason 	struct extent_buffer *upper;
15585f39d397SChris Mason 	struct btrfs_disk_key disk_key;
155900ec4c51SChris Mason 	int slot;
156034a38218SChris Mason 	u32 i;
156100ec4c51SChris Mason 	int free_space;
156200ec4c51SChris Mason 	int push_space = 0;
156300ec4c51SChris Mason 	int push_items = 0;
15640783fcfcSChris Mason 	struct btrfs_item *item;
15657518a238SChris Mason 	u32 left_nritems;
156634a38218SChris Mason 	u32 nr;
15677518a238SChris Mason 	u32 right_nritems;
15685f39d397SChris Mason 	u32 data_end;
1569db94535dSChris Mason 	u32 this_item_size;
157054aa1f4dSChris Mason 	int ret;
157100ec4c51SChris Mason 
157200ec4c51SChris Mason 	slot = path->slots[1];
157300ec4c51SChris Mason 	if (!path->nodes[1]) {
157400ec4c51SChris Mason 		return 1;
157500ec4c51SChris Mason 	}
157600ec4c51SChris Mason 	upper = path->nodes[1];
15775f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
157800ec4c51SChris Mason 		return 1;
15795f39d397SChris Mason 
1580db94535dSChris Mason 	right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1581db94535dSChris Mason 				root->leafsize);
1582123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
15830783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
15845f39d397SChris Mason 		free_extent_buffer(right);
158502217ed2SChris Mason 		return 1;
158602217ed2SChris Mason 	}
158702217ed2SChris Mason 
15885f39d397SChris Mason 	/* cow and double check */
15895f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
15905f39d397SChris Mason 			      slot + 1, &right);
15915f39d397SChris Mason 	if (ret) {
15925f39d397SChris Mason 		free_extent_buffer(right);
1593a429e513SChris Mason 		return 1;
1594a429e513SChris Mason 	}
15955f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
15965f39d397SChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
15975f39d397SChris Mason 		free_extent_buffer(right);
15985f39d397SChris Mason 		return 1;
15995f39d397SChris Mason 	}
16005f39d397SChris Mason 
16015f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
16025f39d397SChris Mason 	if (left_nritems == 0) {
16035f39d397SChris Mason 		free_extent_buffer(right);
16045f39d397SChris Mason 		return 1;
16055f39d397SChris Mason 	}
16065f39d397SChris Mason 
160734a38218SChris Mason 	if (empty)
160834a38218SChris Mason 		nr = 0;
160934a38218SChris Mason 	else
161034a38218SChris Mason 		nr = 1;
161134a38218SChris Mason 
161234a38218SChris Mason 	i = left_nritems - 1;
161334a38218SChris Mason 	while (i >= nr) {
16145f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1615db94535dSChris Mason 
161600ec4c51SChris Mason 		if (path->slots[0] == i)
161700ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1618db94535dSChris Mason 
1619db94535dSChris Mason 		if (!left->map_token) {
1620db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1621db94535dSChris Mason 					sizeof(struct btrfs_item),
1622db94535dSChris Mason 					&left->map_token, &left->kaddr,
1623db94535dSChris Mason 					&left->map_start, &left->map_len,
1624db94535dSChris Mason 					KM_USER1);
1625db94535dSChris Mason 		}
1626db94535dSChris Mason 
1627db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1628db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
162900ec4c51SChris Mason 			break;
163000ec4c51SChris Mason 		push_items++;
1631db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
163234a38218SChris Mason 		if (i == 0)
163334a38218SChris Mason 			break;
163434a38218SChris Mason 		i--;
1635db94535dSChris Mason 	}
1636db94535dSChris Mason 	if (left->map_token) {
1637db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1638db94535dSChris Mason 		left->map_token = NULL;
163900ec4c51SChris Mason 	}
16405f39d397SChris Mason 
164100ec4c51SChris Mason 	if (push_items == 0) {
16425f39d397SChris Mason 		free_extent_buffer(right);
164300ec4c51SChris Mason 		return 1;
164400ec4c51SChris Mason 	}
16455f39d397SChris Mason 
164634a38218SChris Mason 	if (!empty && push_items == left_nritems)
1647a429e513SChris Mason 		WARN_ON(1);
16485f39d397SChris Mason 
164900ec4c51SChris Mason 	/* push left to right */
16505f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
165134a38218SChris Mason 
16525f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1653123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
16545f39d397SChris Mason 
165500ec4c51SChris Mason 	/* make room in the right data area */
16565f39d397SChris Mason 	data_end = leaf_data_end(root, right);
16575f39d397SChris Mason 	memmove_extent_buffer(right,
16585f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
16595f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
16605f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
16615f39d397SChris Mason 
166200ec4c51SChris Mason 	/* copy from the left data area */
16635f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1664d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1665d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1666d6025579SChris Mason 		     push_space);
16675f39d397SChris Mason 
16685f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
16695f39d397SChris Mason 			      btrfs_item_nr_offset(0),
16700783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
16715f39d397SChris Mason 
167200ec4c51SChris Mason 	/* copy the items from left to right */
16735f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
16745f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
16750783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
167600ec4c51SChris Mason 
167700ec4c51SChris Mason 	/* update the item pointers */
16787518a238SChris Mason 	right_nritems += push_items;
16795f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1680123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
16817518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
16825f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1683db94535dSChris Mason 		if (!right->map_token) {
1684db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1685db94535dSChris Mason 					sizeof(struct btrfs_item),
1686db94535dSChris Mason 					&right->map_token, &right->kaddr,
1687db94535dSChris Mason 					&right->map_start, &right->map_len,
1688db94535dSChris Mason 					KM_USER1);
1689db94535dSChris Mason 		}
1690db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1691db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1692db94535dSChris Mason 	}
1693db94535dSChris Mason 
1694db94535dSChris Mason 	if (right->map_token) {
1695db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1696db94535dSChris Mason 		right->map_token = NULL;
169700ec4c51SChris Mason 	}
16987518a238SChris Mason 	left_nritems -= push_items;
16995f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
170000ec4c51SChris Mason 
170134a38218SChris Mason 	if (left_nritems)
17025f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
17035f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1704a429e513SChris Mason 
17055f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
17065f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1707d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
170802217ed2SChris Mason 
170900ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
17107518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
17117518a238SChris Mason 		path->slots[0] -= left_nritems;
17125f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
17135f39d397SChris Mason 		path->nodes[0] = right;
171400ec4c51SChris Mason 		path->slots[1] += 1;
171500ec4c51SChris Mason 	} else {
17165f39d397SChris Mason 		free_extent_buffer(right);
171700ec4c51SChris Mason 	}
171800ec4c51SChris Mason 	return 0;
171900ec4c51SChris Mason }
172000ec4c51SChris Mason /*
172174123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
172274123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
172374123bd7SChris Mason  */
1724e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
172534a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
172634a38218SChris Mason 			  int empty)
1727be0e5c09SChris Mason {
17285f39d397SChris Mason 	struct btrfs_disk_key disk_key;
17295f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
17305f39d397SChris Mason 	struct extent_buffer *left;
1731be0e5c09SChris Mason 	int slot;
1732be0e5c09SChris Mason 	int i;
1733be0e5c09SChris Mason 	int free_space;
1734be0e5c09SChris Mason 	int push_space = 0;
1735be0e5c09SChris Mason 	int push_items = 0;
17360783fcfcSChris Mason 	struct btrfs_item *item;
17377518a238SChris Mason 	u32 old_left_nritems;
17385f39d397SChris Mason 	u32 right_nritems;
173934a38218SChris Mason 	u32 nr;
1740aa5d6bedSChris Mason 	int ret = 0;
1741aa5d6bedSChris Mason 	int wret;
1742db94535dSChris Mason 	u32 this_item_size;
1743db94535dSChris Mason 	u32 old_left_item_size;
1744be0e5c09SChris Mason 
1745be0e5c09SChris Mason 	slot = path->slots[1];
17465f39d397SChris Mason 	if (slot == 0)
1747be0e5c09SChris Mason 		return 1;
17485f39d397SChris Mason 	if (!path->nodes[1])
1749be0e5c09SChris Mason 		return 1;
17505f39d397SChris Mason 
17513685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
17523685f791SChris Mason 	if (right_nritems == 0) {
17533685f791SChris Mason 		return 1;
17543685f791SChris Mason 	}
17553685f791SChris Mason 
17565f39d397SChris Mason 	left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
1757db94535dSChris Mason 			       slot - 1), root->leafsize);
1758123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
17590783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
17605f39d397SChris Mason 		free_extent_buffer(left);
1761be0e5c09SChris Mason 		return 1;
1762be0e5c09SChris Mason 	}
176302217ed2SChris Mason 
176402217ed2SChris Mason 	/* cow and double check */
17655f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
17665f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
176754aa1f4dSChris Mason 	if (ret) {
176854aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
17695f39d397SChris Mason 		free_extent_buffer(left);
177054aa1f4dSChris Mason 		return 1;
177154aa1f4dSChris Mason 	}
17723685f791SChris Mason 
1773123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
17740783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
17755f39d397SChris Mason 		free_extent_buffer(left);
177602217ed2SChris Mason 		return 1;
177702217ed2SChris Mason 	}
177802217ed2SChris Mason 
177934a38218SChris Mason 	if (empty)
178034a38218SChris Mason 		nr = right_nritems;
178134a38218SChris Mason 	else
178234a38218SChris Mason 		nr = right_nritems - 1;
178334a38218SChris Mason 
178434a38218SChris Mason 	for (i = 0; i < nr; i++) {
17855f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1786db94535dSChris Mason 		if (!right->map_token) {
1787db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1788db94535dSChris Mason 					sizeof(struct btrfs_item),
1789db94535dSChris Mason 					&right->map_token, &right->kaddr,
1790db94535dSChris Mason 					&right->map_start, &right->map_len,
1791db94535dSChris Mason 					KM_USER1);
1792db94535dSChris Mason 		}
1793db94535dSChris Mason 
1794be0e5c09SChris Mason 		if (path->slots[0] == i)
1795be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
1796db94535dSChris Mason 
1797db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
1798db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
1799be0e5c09SChris Mason 			break;
1800db94535dSChris Mason 
1801be0e5c09SChris Mason 		push_items++;
1802db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1803be0e5c09SChris Mason 	}
1804db94535dSChris Mason 
1805db94535dSChris Mason 	if (right->map_token) {
1806db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1807db94535dSChris Mason 		right->map_token = NULL;
1808db94535dSChris Mason 	}
1809db94535dSChris Mason 
1810be0e5c09SChris Mason 	if (push_items == 0) {
18115f39d397SChris Mason 		free_extent_buffer(left);
1812be0e5c09SChris Mason 		return 1;
1813be0e5c09SChris Mason 	}
181434a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
1815a429e513SChris Mason 		WARN_ON(1);
18165f39d397SChris Mason 
1817be0e5c09SChris Mason 	/* push data from right to left */
18185f39d397SChris Mason 	copy_extent_buffer(left, right,
18195f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
18205f39d397SChris Mason 			   btrfs_item_nr_offset(0),
18215f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
18225f39d397SChris Mason 
1823123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
18245f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
18255f39d397SChris Mason 
18265f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
1827d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
1828123abc88SChris Mason 		     btrfs_leaf_data(right) +
18295f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
1830be0e5c09SChris Mason 		     push_space);
18315f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
1832eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
1833eb60ceacSChris Mason 
1834db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
1835be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
18365f39d397SChris Mason 		u32 ioff;
1837db94535dSChris Mason 
18385f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1839db94535dSChris Mason 		if (!left->map_token) {
1840db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1841db94535dSChris Mason 					sizeof(struct btrfs_item),
1842db94535dSChris Mason 					&left->map_token, &left->kaddr,
1843db94535dSChris Mason 					&left->map_start, &left->map_len,
1844db94535dSChris Mason 					KM_USER1);
1845db94535dSChris Mason 		}
1846db94535dSChris Mason 
18475f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
18485f39d397SChris Mason 		btrfs_set_item_offset(left, item,
1849db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1850be0e5c09SChris Mason 	}
18515f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1852db94535dSChris Mason 	if (left->map_token) {
1853db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1854db94535dSChris Mason 		left->map_token = NULL;
1855db94535dSChris Mason 	}
1856be0e5c09SChris Mason 
1857be0e5c09SChris Mason 	/* fixup right node */
185834a38218SChris Mason 	if (push_items > right_nritems) {
185934a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
186034a38218SChris Mason 		WARN_ON(1);
186134a38218SChris Mason 	}
186234a38218SChris Mason 
186334a38218SChris Mason 	if (push_items < right_nritems) {
18645f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
1865123abc88SChris Mason 						  leaf_data_end(root, right);
18665f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
1867d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
1868d6025579SChris Mason 				      btrfs_leaf_data(right) +
1869123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
18705f39d397SChris Mason 
18715f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
18725f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
18735f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
18740783fcfcSChris Mason 			     sizeof(struct btrfs_item));
187534a38218SChris Mason 	}
1876eef1c494SYan 	right_nritems -= push_items;
1877eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
1878123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
18795f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
18805f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1881db94535dSChris Mason 
1882db94535dSChris Mason 		if (!right->map_token) {
1883db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1884db94535dSChris Mason 					sizeof(struct btrfs_item),
1885db94535dSChris Mason 					&right->map_token, &right->kaddr,
1886db94535dSChris Mason 					&right->map_start, &right->map_len,
1887db94535dSChris Mason 					KM_USER1);
1888db94535dSChris Mason 		}
1889db94535dSChris Mason 
1890db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
1891db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1892db94535dSChris Mason 	}
1893db94535dSChris Mason 	if (right->map_token) {
1894db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1895db94535dSChris Mason 		right->map_token = NULL;
1896be0e5c09SChris Mason 	}
1897eb60ceacSChris Mason 
18985f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
189934a38218SChris Mason 	if (right_nritems)
19005f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
1901098f59c2SChris Mason 
19025f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
19035f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1904aa5d6bedSChris Mason 	if (wret)
1905aa5d6bedSChris Mason 		ret = wret;
1906be0e5c09SChris Mason 
1907be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
1908be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
1909be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
19105f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19115f39d397SChris Mason 		path->nodes[0] = left;
1912be0e5c09SChris Mason 		path->slots[1] -= 1;
1913be0e5c09SChris Mason 	} else {
19145f39d397SChris Mason 		free_extent_buffer(left);
1915be0e5c09SChris Mason 		path->slots[0] -= push_items;
1916be0e5c09SChris Mason 	}
1917eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1918aa5d6bedSChris Mason 	return ret;
1919be0e5c09SChris Mason }
1920be0e5c09SChris Mason 
192174123bd7SChris Mason /*
192274123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
192374123bd7SChris Mason  * available for the resulting leaf level of the path.
1924aa5d6bedSChris Mason  *
1925aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
192674123bd7SChris Mason  */
1927e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1928d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
1929cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend)
1930be0e5c09SChris Mason {
19317bb86316SChris Mason 	u64 root_gen;
19325f39d397SChris Mason 	struct extent_buffer *l;
19337518a238SChris Mason 	u32 nritems;
1934eb60ceacSChris Mason 	int mid;
1935eb60ceacSChris Mason 	int slot;
19365f39d397SChris Mason 	struct extent_buffer *right;
19370783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
1938be0e5c09SChris Mason 	int data_copy_size;
1939be0e5c09SChris Mason 	int rt_data_off;
1940be0e5c09SChris Mason 	int i;
1941d4dbff95SChris Mason 	int ret = 0;
1942aa5d6bedSChris Mason 	int wret;
1943cc0c5538SChris Mason 	int double_split;
1944cc0c5538SChris Mason 	int num_doubles = 0;
1945d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
1946be0e5c09SChris Mason 
1947cc0c5538SChris Mason 	if (extend)
1948cc0c5538SChris Mason 		space_needed = data_size;
1949cc0c5538SChris Mason 
19507bb86316SChris Mason 	if (root->ref_cows)
19517bb86316SChris Mason 		root_gen = trans->transid;
19527bb86316SChris Mason 	else
19537bb86316SChris Mason 		root_gen = 0;
19547bb86316SChris Mason 
195540689478SChris Mason 	/* first try to make some room by pushing left and right */
19563685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
195734a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
19583326d1b0SChris Mason 		if (wret < 0) {
1959eaee50e8SChris Mason 			return wret;
19603326d1b0SChris Mason 		}
1961eaee50e8SChris Mason 		if (wret) {
196234a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
1963eaee50e8SChris Mason 			if (wret < 0)
1964eaee50e8SChris Mason 				return wret;
1965eaee50e8SChris Mason 		}
19665f39d397SChris Mason 		l = path->nodes[0];
1967aa5d6bedSChris Mason 
1968aa5d6bedSChris Mason 		/* did the pushes work? */
1969cc0c5538SChris Mason 		if (btrfs_leaf_free_space(root, l) >= space_needed)
1970be0e5c09SChris Mason 			return 0;
19713326d1b0SChris Mason 	}
1972aa5d6bedSChris Mason 
19735c680ed6SChris Mason 	if (!path->nodes[1]) {
1974e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
19755c680ed6SChris Mason 		if (ret)
19765c680ed6SChris Mason 			return ret;
19775c680ed6SChris Mason 	}
1978cc0c5538SChris Mason again:
1979cc0c5538SChris Mason 	double_split = 0;
1980cc0c5538SChris Mason 	l = path->nodes[0];
1981eb60ceacSChris Mason 	slot = path->slots[0];
19825f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
1983eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
198454aa1f4dSChris Mason 
19857bb86316SChris Mason 	btrfs_item_key(l, &disk_key, 0);
19867bb86316SChris Mason 
19877bb86316SChris Mason 	right = __btrfs_alloc_free_block(trans, root, root->leafsize,
19887bb86316SChris Mason 					 root->root_key.objectid,
19897bb86316SChris Mason 					 root_gen, disk_key.objectid, 0,
1990db94535dSChris Mason 					 l->start, 0);
19915f39d397SChris Mason 	if (IS_ERR(right))
19925f39d397SChris Mason 		return PTR_ERR(right);
199354aa1f4dSChris Mason 
19945f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1995db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
19965f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
19975f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
19985f39d397SChris Mason 	btrfs_set_header_level(right, 0);
19995f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
20005f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
20015f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2002d4dbff95SChris Mason 	if (mid <= slot) {
2003d4dbff95SChris Mason 		if (nritems == 1 ||
2004d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
2005d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2006d4dbff95SChris Mason 			if (slot >= nritems) {
2007d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
20085f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2009d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2010db94535dSChris Mason 						  &disk_key, right->start,
2011d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
2012d4dbff95SChris Mason 				if (wret)
2013d4dbff95SChris Mason 					ret = wret;
20145f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
20155f39d397SChris Mason 				path->nodes[0] = right;
2016d4dbff95SChris Mason 				path->slots[0] = 0;
2017d4dbff95SChris Mason 				path->slots[1] += 1;
2018d4dbff95SChris Mason 				return ret;
2019d4dbff95SChris Mason 			}
2020d4dbff95SChris Mason 			mid = slot;
20213326d1b0SChris Mason 			if (mid != nritems &&
20223326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
20233326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2024d4dbff95SChris Mason 				double_split = 1;
2025d4dbff95SChris Mason 			}
20263326d1b0SChris Mason 		}
2027d4dbff95SChris Mason 	} else {
2028d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
2029d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2030cc0c5538SChris Mason 			if (!extend && slot == 0) {
2031d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
20325f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2033d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2034d4dbff95SChris Mason 						  &disk_key,
2035db94535dSChris Mason 						  right->start,
2036098f59c2SChris Mason 						  path->slots[1], 1);
2037d4dbff95SChris Mason 				if (wret)
2038d4dbff95SChris Mason 					ret = wret;
20395f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
20405f39d397SChris Mason 				path->nodes[0] = right;
2041d4dbff95SChris Mason 				path->slots[0] = 0;
2042a429e513SChris Mason 				if (path->slots[1] == 0) {
2043a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
2044a429e513SChris Mason 					           path, &disk_key, 1);
2045a429e513SChris Mason 					if (wret)
2046a429e513SChris Mason 						ret = wret;
2047a429e513SChris Mason 				}
2048d4dbff95SChris Mason 				return ret;
2049cc0c5538SChris Mason 			} else if (extend && slot == 0) {
2050cc0c5538SChris Mason 				mid = 1;
2051cc0c5538SChris Mason 			} else {
2052d4dbff95SChris Mason 				mid = slot;
20535ee78ac7SChris Mason 				if (mid != nritems &&
20545ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
20555ee78ac7SChris Mason 				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2056d4dbff95SChris Mason 					double_split = 1;
2057d4dbff95SChris Mason 				}
2058d4dbff95SChris Mason 			}
20595ee78ac7SChris Mason 		}
2060cc0c5538SChris Mason 	}
20615f39d397SChris Mason 	nritems = nritems - mid;
20625f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
20635f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
20645f39d397SChris Mason 
20655f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
20665f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
20675f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
20685f39d397SChris Mason 
20695f39d397SChris Mason 	copy_extent_buffer(right, l,
2070d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2071123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
2072123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
207374123bd7SChris Mason 
20745f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
20755f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
20765f39d397SChris Mason 
20775f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
20785f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
2079db94535dSChris Mason 		u32 ioff;
2080db94535dSChris Mason 
2081db94535dSChris Mason 		if (!right->map_token) {
2082db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2083db94535dSChris Mason 					sizeof(struct btrfs_item),
2084db94535dSChris Mason 					&right->map_token, &right->kaddr,
2085db94535dSChris Mason 					&right->map_start, &right->map_len,
2086db94535dSChris Mason 					KM_USER1);
2087db94535dSChris Mason 		}
2088db94535dSChris Mason 
2089db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
20905f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
20910783fcfcSChris Mason 	}
209274123bd7SChris Mason 
2093db94535dSChris Mason 	if (right->map_token) {
2094db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2095db94535dSChris Mason 		right->map_token = NULL;
2096db94535dSChris Mason 	}
2097db94535dSChris Mason 
20985f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
2099aa5d6bedSChris Mason 	ret = 0;
21005f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
2101db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
2102db94535dSChris Mason 			  path->slots[1] + 1, 1);
2103aa5d6bedSChris Mason 	if (wret)
2104aa5d6bedSChris Mason 		ret = wret;
21055f39d397SChris Mason 
21065f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
21075f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
2108eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
21095f39d397SChris Mason 
2110be0e5c09SChris Mason 	if (mid <= slot) {
21115f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
21125f39d397SChris Mason 		path->nodes[0] = right;
2113be0e5c09SChris Mason 		path->slots[0] -= mid;
2114be0e5c09SChris Mason 		path->slots[1] += 1;
2115eb60ceacSChris Mason 	} else
21165f39d397SChris Mason 		free_extent_buffer(right);
21175f39d397SChris Mason 
2118eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2119d4dbff95SChris Mason 
2120cc0c5538SChris Mason 	if (double_split) {
2121cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
2122cc0c5538SChris Mason 		num_doubles++;
2123cc0c5538SChris Mason 		goto again;
21243326d1b0SChris Mason 	}
2125be0e5c09SChris Mason 	return ret;
2126be0e5c09SChris Mason }
2127be0e5c09SChris Mason 
2128b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2129b18c6685SChris Mason 			struct btrfs_root *root,
2130b18c6685SChris Mason 			struct btrfs_path *path,
2131179e29e4SChris Mason 			u32 new_size, int from_end)
2132b18c6685SChris Mason {
2133b18c6685SChris Mason 	int ret = 0;
2134b18c6685SChris Mason 	int slot;
2135b18c6685SChris Mason 	int slot_orig;
21365f39d397SChris Mason 	struct extent_buffer *leaf;
21375f39d397SChris Mason 	struct btrfs_item *item;
2138b18c6685SChris Mason 	u32 nritems;
2139b18c6685SChris Mason 	unsigned int data_end;
2140b18c6685SChris Mason 	unsigned int old_data_start;
2141b18c6685SChris Mason 	unsigned int old_size;
2142b18c6685SChris Mason 	unsigned int size_diff;
2143b18c6685SChris Mason 	int i;
2144b18c6685SChris Mason 
2145b18c6685SChris Mason 	slot_orig = path->slots[0];
21465f39d397SChris Mason 	leaf = path->nodes[0];
2147179e29e4SChris Mason 	slot = path->slots[0];
2148179e29e4SChris Mason 
2149179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
2150179e29e4SChris Mason 	if (old_size == new_size)
2151179e29e4SChris Mason 		return 0;
2152b18c6685SChris Mason 
21535f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2154b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
2155b18c6685SChris Mason 
21565f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
2157179e29e4SChris Mason 
2158b18c6685SChris Mason 	size_diff = old_size - new_size;
2159b18c6685SChris Mason 
2160b18c6685SChris Mason 	BUG_ON(slot < 0);
2161b18c6685SChris Mason 	BUG_ON(slot >= nritems);
2162b18c6685SChris Mason 
2163b18c6685SChris Mason 	/*
2164b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2165b18c6685SChris Mason 	 */
2166b18c6685SChris Mason 	/* first correct the data pointers */
2167b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
21685f39d397SChris Mason 		u32 ioff;
21695f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2170db94535dSChris Mason 
2171db94535dSChris Mason 		if (!leaf->map_token) {
2172db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2173db94535dSChris Mason 					sizeof(struct btrfs_item),
2174db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2175db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2176db94535dSChris Mason 					KM_USER1);
2177db94535dSChris Mason 		}
2178db94535dSChris Mason 
21795f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
21805f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2181b18c6685SChris Mason 	}
2182db94535dSChris Mason 
2183db94535dSChris Mason 	if (leaf->map_token) {
2184db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2185db94535dSChris Mason 		leaf->map_token = NULL;
2186db94535dSChris Mason 	}
2187db94535dSChris Mason 
2188b18c6685SChris Mason 	/* shift the data */
2189179e29e4SChris Mason 	if (from_end) {
21905f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2191b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2192b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
2193179e29e4SChris Mason 	} else {
2194179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
2195179e29e4SChris Mason 		u64 offset;
2196179e29e4SChris Mason 
2197179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
2198179e29e4SChris Mason 
2199179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2200179e29e4SChris Mason 			unsigned long ptr;
2201179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
2202179e29e4SChris Mason 
2203179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
2204179e29e4SChris Mason 					    struct btrfs_file_extent_item);
2205179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
2206179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
2207179e29e4SChris Mason 
2208179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
2209179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
2210179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
2211179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
2212179e29e4SChris Mason 				        (unsigned long)fi,
2213179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
2214179e29e4SChris Mason 						 disk_bytenr));
2215179e29e4SChris Mason 			}
2216179e29e4SChris Mason 		}
2217179e29e4SChris Mason 
2218179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2219179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2220179e29e4SChris Mason 			      data_end, old_data_start - data_end);
2221179e29e4SChris Mason 
2222179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
2223179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2224179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
2225179e29e4SChris Mason 		if (slot == 0)
2226179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
2227179e29e4SChris Mason 	}
22285f39d397SChris Mason 
22295f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
22305f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
22315f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2232b18c6685SChris Mason 
2233b18c6685SChris Mason 	ret = 0;
22345f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
22355f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2236b18c6685SChris Mason 		BUG();
22375f39d397SChris Mason 	}
2238b18c6685SChris Mason 	return ret;
2239b18c6685SChris Mason }
2240b18c6685SChris Mason 
22415f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
22425f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
22435f39d397SChris Mason 		      u32 data_size)
22446567e837SChris Mason {
22456567e837SChris Mason 	int ret = 0;
22466567e837SChris Mason 	int slot;
22476567e837SChris Mason 	int slot_orig;
22485f39d397SChris Mason 	struct extent_buffer *leaf;
22495f39d397SChris Mason 	struct btrfs_item *item;
22506567e837SChris Mason 	u32 nritems;
22516567e837SChris Mason 	unsigned int data_end;
22526567e837SChris Mason 	unsigned int old_data;
22536567e837SChris Mason 	unsigned int old_size;
22546567e837SChris Mason 	int i;
22556567e837SChris Mason 
22566567e837SChris Mason 	slot_orig = path->slots[0];
22575f39d397SChris Mason 	leaf = path->nodes[0];
22586567e837SChris Mason 
22595f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
22606567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
22616567e837SChris Mason 
22625f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
22635f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
22646567e837SChris Mason 		BUG();
22655f39d397SChris Mason 	}
22666567e837SChris Mason 	slot = path->slots[0];
22675f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
22686567e837SChris Mason 
22696567e837SChris Mason 	BUG_ON(slot < 0);
22703326d1b0SChris Mason 	if (slot >= nritems) {
22713326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
22723326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
22733326d1b0SChris Mason 		BUG_ON(1);
22743326d1b0SChris Mason 	}
22756567e837SChris Mason 
22766567e837SChris Mason 	/*
22776567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
22786567e837SChris Mason 	 */
22796567e837SChris Mason 	/* first correct the data pointers */
22806567e837SChris Mason 	for (i = slot; i < nritems; i++) {
22815f39d397SChris Mason 		u32 ioff;
22825f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2283db94535dSChris Mason 
2284db94535dSChris Mason 		if (!leaf->map_token) {
2285db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2286db94535dSChris Mason 					sizeof(struct btrfs_item),
2287db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2288db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2289db94535dSChris Mason 					KM_USER1);
2290db94535dSChris Mason 		}
22915f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
22925f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
22936567e837SChris Mason 	}
22945f39d397SChris Mason 
2295db94535dSChris Mason 	if (leaf->map_token) {
2296db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2297db94535dSChris Mason 		leaf->map_token = NULL;
2298db94535dSChris Mason 	}
2299db94535dSChris Mason 
23006567e837SChris Mason 	/* shift the data */
23015f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
23026567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
23036567e837SChris Mason 		      data_end, old_data - data_end);
23045f39d397SChris Mason 
23056567e837SChris Mason 	data_end = old_data;
23065f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
23075f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
23085f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
23095f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
23106567e837SChris Mason 
23116567e837SChris Mason 	ret = 0;
23125f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
23135f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
23146567e837SChris Mason 		BUG();
23155f39d397SChris Mason 	}
23166567e837SChris Mason 	return ret;
23176567e837SChris Mason }
23186567e837SChris Mason 
231974123bd7SChris Mason /*
232074123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
232174123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
232274123bd7SChris Mason  */
23235f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
23245f39d397SChris Mason 			    struct btrfs_root *root,
23255f39d397SChris Mason 			    struct btrfs_path *path,
23265f39d397SChris Mason 			    struct btrfs_key *cpu_key, u32 data_size)
2327be0e5c09SChris Mason {
23285f39d397SChris Mason 	struct extent_buffer *leaf;
23295f39d397SChris Mason 	struct btrfs_item *item;
2330aa5d6bedSChris Mason 	int ret = 0;
2331be0e5c09SChris Mason 	int slot;
2332eb60ceacSChris Mason 	int slot_orig;
23337518a238SChris Mason 	u32 nritems;
2334be0e5c09SChris Mason 	unsigned int data_end;
2335e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2336e2fa7227SChris Mason 
2337e2fa7227SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2338be0e5c09SChris Mason 
233974123bd7SChris Mason 	/* create a root if there isn't one */
23405c680ed6SChris Mason 	if (!root->node)
2341cfaa7295SChris Mason 		BUG();
23425f39d397SChris Mason 
2343e089f05cSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
2344eb60ceacSChris Mason 	if (ret == 0) {
2345f0930a37SChris Mason 		return -EEXIST;
2346aa5d6bedSChris Mason 	}
2347ed2ff2cbSChris Mason 	if (ret < 0)
2348ed2ff2cbSChris Mason 		goto out;
2349be0e5c09SChris Mason 
235062e2749eSChris Mason 	slot_orig = path->slots[0];
23515f39d397SChris Mason 	leaf = path->nodes[0];
235274123bd7SChris Mason 
23535f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2354123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2355eb60ceacSChris Mason 
2356123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
2357d4dbff95SChris Mason 	    sizeof(struct btrfs_item) + data_size) {
23583326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
23593326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
23603326d1b0SChris Mason 		       data_size, btrfs_leaf_free_space(root, leaf));
2361be0e5c09SChris Mason 		BUG();
2362d4dbff95SChris Mason 	}
23635f39d397SChris Mason 
236462e2749eSChris Mason 	slot = path->slots[0];
2365eb60ceacSChris Mason 	BUG_ON(slot < 0);
23665f39d397SChris Mason 
2367be0e5c09SChris Mason 	if (slot != nritems) {
2368be0e5c09SChris Mason 		int i;
23695f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2370be0e5c09SChris Mason 
23715f39d397SChris Mason 		if (old_data < data_end) {
23725f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
23735f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
23745f39d397SChris Mason 			       slot, old_data, data_end);
23755f39d397SChris Mason 			BUG_ON(1);
23765f39d397SChris Mason 		}
2377be0e5c09SChris Mason 		/*
2378be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2379be0e5c09SChris Mason 		 */
2380be0e5c09SChris Mason 		/* first correct the data pointers */
2381db94535dSChris Mason 		WARN_ON(leaf->map_token);
23820783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
23835f39d397SChris Mason 			u32 ioff;
2384db94535dSChris Mason 
23855f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2386db94535dSChris Mason 			if (!leaf->map_token) {
2387db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2388db94535dSChris Mason 					sizeof(struct btrfs_item),
2389db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2390db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2391db94535dSChris Mason 					KM_USER1);
2392db94535dSChris Mason 			}
2393db94535dSChris Mason 
23945f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
23955f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff - data_size);
23960783fcfcSChris Mason 		}
2397db94535dSChris Mason 		if (leaf->map_token) {
2398db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2399db94535dSChris Mason 			leaf->map_token = NULL;
2400db94535dSChris Mason 		}
2401be0e5c09SChris Mason 
2402be0e5c09SChris Mason 		/* shift the items */
24035f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
24045f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
24050783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2406be0e5c09SChris Mason 
2407be0e5c09SChris Mason 		/* shift the data */
24085f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2409d6025579SChris Mason 			      data_end - data_size, btrfs_leaf_data(leaf) +
2410be0e5c09SChris Mason 			      data_end, old_data - data_end);
2411be0e5c09SChris Mason 		data_end = old_data;
2412be0e5c09SChris Mason 	}
24135f39d397SChris Mason 
241462e2749eSChris Mason 	/* setup the item for the new data */
24155f39d397SChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
24165f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
24175f39d397SChris Mason 	btrfs_set_item_offset(leaf, item, data_end - data_size);
24185f39d397SChris Mason 	btrfs_set_item_size(leaf, item, data_size);
24195f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
24205f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2421aa5d6bedSChris Mason 
2422aa5d6bedSChris Mason 	ret = 0;
24238e19f2cdSChris Mason 	if (slot == 0)
2424e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2425aa5d6bedSChris Mason 
24265f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
24275f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2428be0e5c09SChris Mason 		BUG();
24295f39d397SChris Mason 	}
2430ed2ff2cbSChris Mason out:
243162e2749eSChris Mason 	return ret;
243262e2749eSChris Mason }
243362e2749eSChris Mason 
243462e2749eSChris Mason /*
243562e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
243662e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
243762e2749eSChris Mason  */
2438e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2439e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2440e089f05cSChris Mason 		      data_size)
244162e2749eSChris Mason {
244262e2749eSChris Mason 	int ret = 0;
24432c90e5d6SChris Mason 	struct btrfs_path *path;
24445f39d397SChris Mason 	struct extent_buffer *leaf;
24455f39d397SChris Mason 	unsigned long ptr;
244662e2749eSChris Mason 
24472c90e5d6SChris Mason 	path = btrfs_alloc_path();
24482c90e5d6SChris Mason 	BUG_ON(!path);
24492c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
245062e2749eSChris Mason 	if (!ret) {
24515f39d397SChris Mason 		leaf = path->nodes[0];
24525f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
24535f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
24545f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
245562e2749eSChris Mason 	}
24562c90e5d6SChris Mason 	btrfs_free_path(path);
2457aa5d6bedSChris Mason 	return ret;
2458be0e5c09SChris Mason }
2459be0e5c09SChris Mason 
246074123bd7SChris Mason /*
24615de08d7dSChris Mason  * delete the pointer from a given node.
246274123bd7SChris Mason  *
246374123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
246474123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
246574123bd7SChris Mason  * a leaf if all the nodes are emptied.
246674123bd7SChris Mason  */
2467e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2468e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2469be0e5c09SChris Mason {
24705f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
24717518a238SChris Mason 	u32 nritems;
2472aa5d6bedSChris Mason 	int ret = 0;
2473bb803951SChris Mason 	int wret;
2474be0e5c09SChris Mason 
24755f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2476be0e5c09SChris Mason 	if (slot != nritems -1) {
24775f39d397SChris Mason 		memmove_extent_buffer(parent,
24785f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
24795f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2480d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2481d6025579SChris Mason 			      (nritems - slot - 1));
2482be0e5c09SChris Mason 	}
24837518a238SChris Mason 	nritems--;
24845f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
24857518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
24865f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2487eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
24885f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2489bb803951SChris Mason 	} else if (slot == 0) {
24905f39d397SChris Mason 		struct btrfs_disk_key disk_key;
24915f39d397SChris Mason 
24925f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
24935f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
24940f70abe2SChris Mason 		if (wret)
24950f70abe2SChris Mason 			ret = wret;
2496be0e5c09SChris Mason 	}
2497d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2498aa5d6bedSChris Mason 	return ret;
2499be0e5c09SChris Mason }
2500be0e5c09SChris Mason 
250174123bd7SChris Mason /*
250274123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
250374123bd7SChris Mason  * the leaf, remove it from the tree
250474123bd7SChris Mason  */
2505e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2506e089f05cSChris Mason 		   struct btrfs_path *path)
2507be0e5c09SChris Mason {
2508be0e5c09SChris Mason 	int slot;
25095f39d397SChris Mason 	struct extent_buffer *leaf;
25105f39d397SChris Mason 	struct btrfs_item *item;
2511be0e5c09SChris Mason 	int doff;
2512be0e5c09SChris Mason 	int dsize;
2513aa5d6bedSChris Mason 	int ret = 0;
2514aa5d6bedSChris Mason 	int wret;
25157518a238SChris Mason 	u32 nritems;
2516be0e5c09SChris Mason 
25175f39d397SChris Mason 	leaf = path->nodes[0];
25184920c9acSChris Mason 	slot = path->slots[0];
25195f39d397SChris Mason 	doff = btrfs_item_offset_nr(leaf, slot);
25205f39d397SChris Mason 	dsize = btrfs_item_size_nr(leaf, slot);
25215f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2522be0e5c09SChris Mason 
25237518a238SChris Mason 	if (slot != nritems - 1) {
2524be0e5c09SChris Mason 		int i;
2525123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
25265f39d397SChris Mason 
25275f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2528d6025579SChris Mason 			      data_end + dsize,
2529123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
2530be0e5c09SChris Mason 			      doff - data_end);
25315f39d397SChris Mason 
25320783fcfcSChris Mason 		for (i = slot + 1; i < nritems; i++) {
25335f39d397SChris Mason 			u32 ioff;
2534db94535dSChris Mason 
25355f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2536db94535dSChris Mason 			if (!leaf->map_token) {
2537db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2538db94535dSChris Mason 					sizeof(struct btrfs_item),
2539db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2540db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2541db94535dSChris Mason 					KM_USER1);
2542db94535dSChris Mason 			}
25435f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
25445f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
25450783fcfcSChris Mason 		}
2546db94535dSChris Mason 
2547db94535dSChris Mason 		if (leaf->map_token) {
2548db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2549db94535dSChris Mason 			leaf->map_token = NULL;
2550db94535dSChris Mason 		}
2551db94535dSChris Mason 
25525f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
25535f39d397SChris Mason 			      btrfs_item_nr_offset(slot + 1),
25540783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
25557518a238SChris Mason 			      (nritems - slot - 1));
2556be0e5c09SChris Mason 	}
25575f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems - 1);
25587518a238SChris Mason 	nritems--;
25595f39d397SChris Mason 
256074123bd7SChris Mason 	/* delete the leaf if we've emptied it */
25617518a238SChris Mason 	if (nritems == 0) {
25625f39d397SChris Mason 		if (leaf == root->node) {
25635f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
25649a8dd150SChris Mason 		} else {
25657bb86316SChris Mason 			u64 root_gen = btrfs_header_generation(path->nodes[1]);
25665f39d397SChris Mason 			clean_tree_block(trans, root, leaf);
25675f39d397SChris Mason 			wait_on_tree_block_writeback(root, leaf);
2568e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2569aa5d6bedSChris Mason 			if (wret)
2570aa5d6bedSChris Mason 				ret = wret;
2571e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
25727bb86316SChris Mason 					 leaf->start, leaf->len,
25737bb86316SChris Mason 					 btrfs_header_owner(path->nodes[1]),
25747bb86316SChris Mason 					 root_gen, 0, 0, 1);
25750f70abe2SChris Mason 			if (wret)
25760f70abe2SChris Mason 				ret = wret;
25779a8dd150SChris Mason 		}
2578be0e5c09SChris Mason 	} else {
25797518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2580aa5d6bedSChris Mason 		if (slot == 0) {
25815f39d397SChris Mason 			struct btrfs_disk_key disk_key;
25825f39d397SChris Mason 
25835f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2584e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
25855f39d397SChris Mason 					      &disk_key, 1);
2586aa5d6bedSChris Mason 			if (wret)
2587aa5d6bedSChris Mason 				ret = wret;
2588aa5d6bedSChris Mason 		}
2589aa5d6bedSChris Mason 
259074123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
25917936ca38SChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
2592be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2593be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2594be0e5c09SChris Mason 			 * for possible call to del_ptr below
2595be0e5c09SChris Mason 			 */
25964920c9acSChris Mason 			slot = path->slots[1];
25975f39d397SChris Mason 			extent_buffer_get(leaf);
25985f39d397SChris Mason 
259934a38218SChris Mason 			wret = push_leaf_right(trans, root, path, 1, 1);
260054aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2601aa5d6bedSChris Mason 				ret = wret;
26025f39d397SChris Mason 
26035f39d397SChris Mason 			if (path->nodes[0] == leaf &&
26045f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
260534a38218SChris Mason 				wret = push_leaf_left(trans, root, path, 1, 1);
260654aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2607aa5d6bedSChris Mason 					ret = wret;
2608aa5d6bedSChris Mason 			}
26095f39d397SChris Mason 
26105f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
26117bb86316SChris Mason 				u64 root_gen;
2612db94535dSChris Mason 				u64 bytenr = leaf->start;
2613db94535dSChris Mason 				u32 blocksize = leaf->len;
26145f39d397SChris Mason 
26157bb86316SChris Mason 				root_gen = btrfs_header_generation(
26167bb86316SChris Mason 							   path->nodes[1]);
26177bb86316SChris Mason 
26185f39d397SChris Mason 				clean_tree_block(trans, root, leaf);
26195f39d397SChris Mason 				wait_on_tree_block_writeback(root, leaf);
26205f39d397SChris Mason 
2621e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2622aa5d6bedSChris Mason 				if (wret)
2623aa5d6bedSChris Mason 					ret = wret;
26245f39d397SChris Mason 
26255f39d397SChris Mason 				free_extent_buffer(leaf);
2626db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
26277bb86316SChris Mason 					     blocksize,
26287bb86316SChris Mason 					     btrfs_header_owner(path->nodes[1]),
26297bb86316SChris Mason 					     root_gen, 0, 0, 1);
26300f70abe2SChris Mason 				if (wret)
26310f70abe2SChris Mason 					ret = wret;
26325de08d7dSChris Mason 			} else {
26335f39d397SChris Mason 				btrfs_mark_buffer_dirty(leaf);
26345f39d397SChris Mason 				free_extent_buffer(leaf);
2635be0e5c09SChris Mason 			}
2636d5719762SChris Mason 		} else {
26375f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2638be0e5c09SChris Mason 		}
26399a8dd150SChris Mason 	}
2640aa5d6bedSChris Mason 	return ret;
26419a8dd150SChris Mason }
26429a8dd150SChris Mason 
264397571fd0SChris Mason /*
26447bb86316SChris Mason  * walk up the tree as far as required to find the previous leaf.
26457bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
26467bb86316SChris Mason  * returns < 0 on io errors.
26477bb86316SChris Mason  */
26487bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
26497bb86316SChris Mason {
26507bb86316SChris Mason 	int slot;
26517bb86316SChris Mason 	int level = 1;
26527bb86316SChris Mason 	u64 bytenr;
26537bb86316SChris Mason 	struct extent_buffer *c;
26547bb86316SChris Mason 	struct extent_buffer *next = NULL;
26557bb86316SChris Mason 
26567bb86316SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
26577bb86316SChris Mason 		if (!path->nodes[level])
26587bb86316SChris Mason 			return 1;
26597bb86316SChris Mason 
26607bb86316SChris Mason 		slot = path->slots[level];
26617bb86316SChris Mason 		c = path->nodes[level];
26627bb86316SChris Mason 		if (slot == 0) {
26637bb86316SChris Mason 			level++;
26647bb86316SChris Mason 			if (level == BTRFS_MAX_LEVEL)
26657bb86316SChris Mason 				return 1;
26667bb86316SChris Mason 			continue;
26677bb86316SChris Mason 		}
26687bb86316SChris Mason 		slot--;
26697bb86316SChris Mason 
26707bb86316SChris Mason 		bytenr = btrfs_node_blockptr(c, slot);
26717bb86316SChris Mason 		if (next)
26727bb86316SChris Mason 			free_extent_buffer(next);
26737bb86316SChris Mason 
26747bb86316SChris Mason 		if (path->reada < 0)
26757bb86316SChris Mason 			reada_for_search(root, path, level, slot);
26767bb86316SChris Mason 
26777bb86316SChris Mason 		next = read_tree_block(root, bytenr,
26787bb86316SChris Mason 				       btrfs_level_size(root, level - 1));
26797bb86316SChris Mason 		break;
26807bb86316SChris Mason 	}
26817bb86316SChris Mason 	path->slots[level] = slot;
26827bb86316SChris Mason 	while(1) {
26837bb86316SChris Mason 		level--;
26847bb86316SChris Mason 		c = path->nodes[level];
26857bb86316SChris Mason 		free_extent_buffer(c);
26867bb86316SChris Mason 		path->nodes[level] = next;
26877bb86316SChris Mason 		path->slots[level] = 0;
26887bb86316SChris Mason 		if (!level)
26897bb86316SChris Mason 			break;
26907bb86316SChris Mason 		if (path->reada)
26917bb86316SChris Mason 			reada_for_search(root, path, level, 0);
26927bb86316SChris Mason 		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
26937bb86316SChris Mason 				       btrfs_level_size(root, level - 1));
26947bb86316SChris Mason 	}
26957bb86316SChris Mason 	return 0;
26967bb86316SChris Mason }
26977bb86316SChris Mason 
26987bb86316SChris Mason /*
269997571fd0SChris Mason  * walk up the tree as far as required to find the next leaf.
27000f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
27010f70abe2SChris Mason  * returns < 0 on io errors.
270297571fd0SChris Mason  */
2703234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2704d97e63b6SChris Mason {
2705d97e63b6SChris Mason 	int slot;
2706d97e63b6SChris Mason 	int level = 1;
2707db94535dSChris Mason 	u64 bytenr;
27085f39d397SChris Mason 	struct extent_buffer *c;
27095f39d397SChris Mason 	struct extent_buffer *next = NULL;
2710d97e63b6SChris Mason 
2711234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
2712d97e63b6SChris Mason 		if (!path->nodes[level])
27130f70abe2SChris Mason 			return 1;
27145f39d397SChris Mason 
2715d97e63b6SChris Mason 		slot = path->slots[level] + 1;
2716d97e63b6SChris Mason 		c = path->nodes[level];
27175f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
2718d97e63b6SChris Mason 			level++;
27197bb86316SChris Mason 			if (level == BTRFS_MAX_LEVEL)
27207bb86316SChris Mason 				return 1;
2721d97e63b6SChris Mason 			continue;
2722d97e63b6SChris Mason 		}
27235f39d397SChris Mason 
2724db94535dSChris Mason 		bytenr = btrfs_node_blockptr(c, slot);
2725cfaa7295SChris Mason 		if (next)
27265f39d397SChris Mason 			free_extent_buffer(next);
27275f39d397SChris Mason 
27286702ed49SChris Mason 		if (path->reada)
27296702ed49SChris Mason 			reada_for_search(root, path, level, slot);
27305f39d397SChris Mason 
2731db94535dSChris Mason 		next = read_tree_block(root, bytenr,
2732db94535dSChris Mason 				       btrfs_level_size(root, level -1));
2733d97e63b6SChris Mason 		break;
2734d97e63b6SChris Mason 	}
2735d97e63b6SChris Mason 	path->slots[level] = slot;
2736d97e63b6SChris Mason 	while(1) {
2737d97e63b6SChris Mason 		level--;
2738d97e63b6SChris Mason 		c = path->nodes[level];
27395f39d397SChris Mason 		free_extent_buffer(c);
2740d97e63b6SChris Mason 		path->nodes[level] = next;
2741d97e63b6SChris Mason 		path->slots[level] = 0;
2742d97e63b6SChris Mason 		if (!level)
2743d97e63b6SChris Mason 			break;
27446702ed49SChris Mason 		if (path->reada)
274532020611SYan 			reada_for_search(root, path, level, 0);
2746db94535dSChris Mason 		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2747db94535dSChris Mason 				       btrfs_level_size(root, level - 1));
2748d97e63b6SChris Mason 	}
2749d97e63b6SChris Mason 	return 0;
2750d97e63b6SChris Mason }
2751