xref: /openbmc/linux/fs/btrfs/ctree.c (revision 6b80053d02be41886344b5007d04e345311ec0b5)
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 
195f39d397SChris Mason #include <linux/highmem.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,
29d4dbff95SChris Mason 		      struct btrfs_path *path, int data_size);
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 
735f39d397SChris Mason static int __btrfs_cow_block(struct btrfs_trans_handle *trans,
745f39d397SChris Mason 			     struct btrfs_root *root,
755f39d397SChris Mason 			     struct extent_buffer *buf,
765f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
775f39d397SChris Mason 			     struct extent_buffer **cow_ret,
785f39d397SChris Mason 			     u64 search_start, u64 empty_size)
796702ed49SChris Mason {
805f39d397SChris Mason 	struct extent_buffer *cow;
816702ed49SChris Mason 	int ret = 0;
826702ed49SChris Mason 	int different_trans = 0;
836702ed49SChris Mason 
846702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
855f39d397SChris Mason 
86db94535dSChris Mason 	cow = btrfs_alloc_free_block(trans, root, buf->len,
87db94535dSChris Mason 				     search_start, empty_size);
886702ed49SChris Mason 	if (IS_ERR(cow))
896702ed49SChris Mason 		return PTR_ERR(cow);
906702ed49SChris Mason 
91f510cfecSChris Mason 	cow->alloc_addr = (unsigned long)__builtin_return_address(0);
926702ed49SChris Mason 
935f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
94db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
955f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
965f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
976702ed49SChris Mason 
985f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
995f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
1006702ed49SChris Mason 		different_trans = 1;
1016702ed49SChris Mason 		ret = btrfs_inc_ref(trans, root, buf);
1026702ed49SChris Mason 		if (ret)
1036702ed49SChris Mason 			return ret;
1046702ed49SChris Mason 	} else {
1056702ed49SChris Mason 		clean_tree_block(trans, root, buf);
1066702ed49SChris Mason 	}
1076702ed49SChris Mason 
1086702ed49SChris Mason 	if (buf == root->node) {
1096702ed49SChris Mason 		root->node = cow;
1105f39d397SChris Mason 		extent_buffer_get(cow);
1116702ed49SChris Mason 		if (buf != root->commit_root) {
112db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
113db94535dSChris Mason 					  buf->len, 1);
1146702ed49SChris Mason 		}
1155f39d397SChris Mason 		free_extent_buffer(buf);
1166702ed49SChris Mason 	} else {
1175f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
118db94535dSChris Mason 					cow->start);
1196702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
1205f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
121db94535dSChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len, 1);
1226702ed49SChris Mason 	}
1235f39d397SChris Mason 	free_extent_buffer(buf);
1246702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
1256702ed49SChris Mason 	*cow_ret = cow;
1266702ed49SChris Mason 	return 0;
1276702ed49SChris Mason }
1286702ed49SChris Mason 
1295f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans,
1305f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
1315f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
1325f39d397SChris Mason 		    struct extent_buffer **cow_ret)
13302217ed2SChris Mason {
1346702ed49SChris Mason 	u64 search_start;
135f510cfecSChris Mason 	int ret;
136ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
137ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
138ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
139ccd467d6SChris Mason 		WARN_ON(1);
140ccd467d6SChris Mason 	}
141ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
142ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
143ccd467d6SChris Mason 		       root->fs_info->generation);
144ccd467d6SChris Mason 		WARN_ON(1);
145ccd467d6SChris Mason 	}
1465f39d397SChris Mason 	if (btrfs_header_generation(buf) == trans->transid) {
14702217ed2SChris Mason 		*cow_ret = buf;
14802217ed2SChris Mason 		return 0;
14902217ed2SChris Mason 	}
1506702ed49SChris Mason 
151db94535dSChris Mason 	search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
152f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
1536702ed49SChris Mason 				 parent_slot, cow_ret, search_start, 0);
154f510cfecSChris Mason 	(*cow_ret)->alloc_addr = (unsigned long)__builtin_return_address(0);
155f510cfecSChris Mason 	return ret;
1562c90e5d6SChris Mason }
1576702ed49SChris Mason 
158*6b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
1596702ed49SChris Mason {
160*6b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
1616702ed49SChris Mason 		return 1;
162*6b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
1636702ed49SChris Mason 		return 1;
16402217ed2SChris Mason 	return 0;
16502217ed2SChris Mason }
16602217ed2SChris Mason 
167*6b80053dSChris Mason static int should_defrag_leaf(struct extent_buffer *leaf)
1682cc58cf2SChris Mason {
169*6b80053dSChris Mason 	struct btrfs_key key;
1702cc58cf2SChris Mason 	u32 nritems;
1712cc58cf2SChris Mason 
172*6b80053dSChris Mason 	if (btrfs_buffer_defrag(leaf))
1732cc58cf2SChris Mason 		return 1;
1742cc58cf2SChris Mason 
175*6b80053dSChris Mason 	nritems = btrfs_header_nritems(leaf);
1762cc58cf2SChris Mason 	if (nritems == 0)
1772cc58cf2SChris Mason 		return 0;
1782cc58cf2SChris Mason 
179*6b80053dSChris Mason 	btrfs_item_key_to_cpu(leaf, &key, 0);
180*6b80053dSChris Mason 	if (key.type == BTRFS_DIR_ITEM_KEY)
1812cc58cf2SChris Mason 		return 1;
1822cc58cf2SChris Mason 
183*6b80053dSChris Mason 
184*6b80053dSChris Mason 	btrfs_item_key_to_cpu(leaf, &key, nritems - 1);
185*6b80053dSChris Mason 	if (key.type == BTRFS_DIR_ITEM_KEY)
1862cc58cf2SChris Mason 		return 1;
1872cc58cf2SChris Mason 	if (nritems > 4) {
188*6b80053dSChris Mason 		btrfs_item_key_to_cpu(leaf, &key, nritems / 2);
189*6b80053dSChris Mason 		if (key.type == BTRFS_DIR_ITEM_KEY)
1902cc58cf2SChris Mason 			return 1;
1912cc58cf2SChris Mason 	}
1922cc58cf2SChris Mason 	return 0;
1932cc58cf2SChris Mason }
1942cc58cf2SChris Mason 
1956702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
1965f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
197e9d0b13bSChris Mason 		       int cache_only, u64 *last_ret)
1986702ed49SChris Mason {
199*6b80053dSChris Mason 	struct extent_buffer *cur;
200*6b80053dSChris Mason 	struct extent_buffer *tmp;
2016702ed49SChris Mason 	u64 blocknr;
202e9d0b13bSChris Mason 	u64 search_start = *last_ret;
203e9d0b13bSChris Mason 	u64 last_block = 0;
2046702ed49SChris Mason 	u64 other;
2056702ed49SChris Mason 	u32 parent_nritems;
2066702ed49SChris Mason 	int start_slot;
2076702ed49SChris Mason 	int end_slot;
2086702ed49SChris Mason 	int i;
2096702ed49SChris Mason 	int err = 0;
210f2183bdeSChris Mason 	int parent_level;
211*6b80053dSChris Mason 	int uptodate;
212*6b80053dSChris Mason 	u32 blocksize;
2136702ed49SChris Mason 
2146702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
2156702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2166702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
2176702ed49SChris Mason 		WARN_ON(1);
2186702ed49SChris Mason 	}
2196702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
2206702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2216702ed49SChris Mason 		       root->fs_info->generation);
2226702ed49SChris Mason 		WARN_ON(1);
2236702ed49SChris Mason 	}
224*6b80053dSChris Mason 	if (btrfs_buffer_defrag_done(parent))
22586479a04SChris Mason 		return 0;
22686479a04SChris Mason 
227*6b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
228*6b80053dSChris Mason 	parent_level = btrfs_header_level(parent);
229*6b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
2306702ed49SChris Mason 
2316702ed49SChris Mason 	start_slot = 0;
2326702ed49SChris Mason 	end_slot = parent_nritems;
2336702ed49SChris Mason 
2346702ed49SChris Mason 	if (parent_nritems == 1)
2356702ed49SChris Mason 		return 0;
2366702ed49SChris Mason 
2376702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
2386702ed49SChris Mason 		int close = 1;
239*6b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
240e9d0b13bSChris Mason 		if (last_block == 0)
241e9d0b13bSChris Mason 			last_block = blocknr;
2426702ed49SChris Mason 		if (i > 0) {
243*6b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
244*6b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2456702ed49SChris Mason 		}
2466702ed49SChris Mason 		if (close && i < end_slot - 1) {
247*6b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
248*6b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2496702ed49SChris Mason 		}
250e9d0b13bSChris Mason 		if (close) {
251e9d0b13bSChris Mason 			last_block = blocknr;
2526702ed49SChris Mason 			continue;
253e9d0b13bSChris Mason 		}
2546702ed49SChris Mason 
255*6b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
256*6b80053dSChris Mason 		if (cur)
257*6b80053dSChris Mason 			uptodate = btrfs_buffer_uptodate(cur);
258*6b80053dSChris Mason 		else
259*6b80053dSChris Mason 			uptodate = 0;
260*6b80053dSChris Mason 		if (!cur || !uptodate ||
261*6b80053dSChris Mason 		    (parent_level != 1 && !btrfs_buffer_defrag(cur)) ||
262*6b80053dSChris Mason 		    (parent_level == 1 && !should_defrag_leaf(cur))) {
2636702ed49SChris Mason 			if (cache_only) {
264*6b80053dSChris Mason 				free_extent_buffer(cur);
2656702ed49SChris Mason 				continue;
2666702ed49SChris Mason 			}
267*6b80053dSChris Mason 			if (!cur) {
268*6b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
269*6b80053dSChris Mason 							 blocksize);
270*6b80053dSChris Mason 			} else if (!uptodate) {
271*6b80053dSChris Mason 				btrfs_read_buffer(cur);
2726702ed49SChris Mason 			}
273f2183bdeSChris Mason 		}
274e9d0b13bSChris Mason 		if (search_start == 0)
275*6b80053dSChris Mason 			search_start = last_block;
276e9d0b13bSChris Mason 
277*6b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
278*6b80053dSChris Mason 					&tmp, search_start,
279*6b80053dSChris Mason 					min(16 * blocksize,
280*6b80053dSChris Mason 					    (end_slot - i) * blocksize));
281252c38f0SYan 		if (err) {
282*6b80053dSChris Mason 			free_extent_buffer(cur);
2836702ed49SChris Mason 			break;
284252c38f0SYan 		}
285*6b80053dSChris Mason 		search_start = tmp->start;
286f2183bdeSChris Mason 		*last_ret = search_start;
287f2183bdeSChris Mason 		if (parent_level == 1)
288*6b80053dSChris Mason 			btrfs_clear_buffer_defrag(tmp);
289*6b80053dSChris Mason 		btrfs_set_buffer_defrag_done(tmp);
290*6b80053dSChris Mason 		free_extent_buffer(tmp);
2916702ed49SChris Mason 	}
2926702ed49SChris Mason 	return err;
2936702ed49SChris Mason }
2946702ed49SChris Mason 
29574123bd7SChris Mason /*
29674123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
29774123bd7SChris Mason  * this returns the address of the start of the last item,
29874123bd7SChris Mason  * which is the stop of the leaf data stack
29974123bd7SChris Mason  */
300123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
3015f39d397SChris Mason 					 struct extent_buffer *leaf)
302be0e5c09SChris Mason {
3035f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
304be0e5c09SChris Mason 	if (nr == 0)
305123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
3065f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
307be0e5c09SChris Mason }
308be0e5c09SChris Mason 
30974123bd7SChris Mason /*
31074123bd7SChris Mason  * compare two keys in a memcmp fashion
31174123bd7SChris Mason  */
3129aca1d51SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
313be0e5c09SChris Mason {
314e2fa7227SChris Mason 	struct btrfs_key k1;
315e2fa7227SChris Mason 
316e2fa7227SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
317e2fa7227SChris Mason 
318e2fa7227SChris Mason 	if (k1.objectid > k2->objectid)
319be0e5c09SChris Mason 		return 1;
320e2fa7227SChris Mason 	if (k1.objectid < k2->objectid)
321be0e5c09SChris Mason 		return -1;
3225f39d397SChris Mason 	if (k1.type > k2->type)
323f254e52cSChris Mason 		return 1;
3245f39d397SChris Mason 	if (k1.type < k2->type)
325f254e52cSChris Mason 		return -1;
32670b2befdSChris Mason 	if (k1.offset > k2->offset)
32770b2befdSChris Mason 		return 1;
32870b2befdSChris Mason 	if (k1.offset < k2->offset)
32970b2befdSChris Mason 		return -1;
330be0e5c09SChris Mason 	return 0;
331be0e5c09SChris Mason }
33274123bd7SChris Mason 
333123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
334123abc88SChris Mason 		      int level)
335aa5d6bedSChris Mason {
3365f39d397SChris Mason 	struct extent_buffer *parent = NULL;
3375f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
3385f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3395f39d397SChris Mason 	struct btrfs_disk_key node_key;
340aa5d6bedSChris Mason 	int parent_slot;
3418d7be552SChris Mason 	int slot;
3428d7be552SChris Mason 	struct btrfs_key cpukey;
3435f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
344aa5d6bedSChris Mason 
345aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3465f39d397SChris Mason 		parent = path->nodes[level + 1];
347a1f39630SAneesh 
3488d7be552SChris Mason 	slot = path->slots[level];
3497518a238SChris Mason 	BUG_ON(nritems == 0);
3507518a238SChris Mason 	if (parent) {
351a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3525f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3535f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
3545f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
355e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
3561d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
357db94535dSChris Mason 		       btrfs_header_bytenr(node));
358aa5d6bedSChris Mason 	}
359123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
3608d7be552SChris Mason 	if (slot != 0) {
3615f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
3625f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3635f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
3648d7be552SChris Mason 	}
3658d7be552SChris Mason 	if (slot < nritems - 1) {
3665f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
3675f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3685f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
369aa5d6bedSChris Mason 	}
370aa5d6bedSChris Mason 	return 0;
371aa5d6bedSChris Mason }
372aa5d6bedSChris Mason 
373123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
374123abc88SChris Mason 		      int level)
375aa5d6bedSChris Mason {
3765f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
3775f39d397SChris Mason 	struct extent_buffer *parent = NULL;
378aa5d6bedSChris Mason 	int parent_slot;
3798d7be552SChris Mason 	struct btrfs_key cpukey;
3805f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3815f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
3825f39d397SChris Mason 	int slot = path->slots[0];
3838d7be552SChris Mason 
3845f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
385aa5d6bedSChris Mason 
386aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3875f39d397SChris Mason 		parent = path->nodes[level + 1];
3887518a238SChris Mason 
3897518a238SChris Mason 	if (nritems == 0)
3907518a238SChris Mason 		return 0;
3917518a238SChris Mason 
3927518a238SChris Mason 	if (parent) {
393a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3945f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3955f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
3966702ed49SChris Mason 
3975f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
398e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
3991d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
400db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
401aa5d6bedSChris Mason 	}
4025f39d397SChris Mason #if 0
4035f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
4045f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
4055f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
4065f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
4075f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4085f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
4095f39d397SChris Mason 			BUG_ON(1);
4105f39d397SChris Mason 		}
4115f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
4125f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
4135f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4145f39d397SChris Mason 			printk("slot %d offset bad\n", i);
4155f39d397SChris Mason 			BUG_ON(1);
4165f39d397SChris Mason 		}
4175f39d397SChris Mason 		if (i == 0) {
4185f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
4195f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
4205f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
4215f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4225f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
4235f39d397SChris Mason 				BUG_ON(1);
4245f39d397SChris Mason 			}
4255f39d397SChris Mason 		}
4265f39d397SChris Mason 	}
4275f39d397SChris Mason 	if (nritems > 0) {
4285f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
4295f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4305f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
4315f39d397SChris Mason 				BUG_ON(1);
4325f39d397SChris Mason 		}
4335f39d397SChris Mason 	}
4345f39d397SChris Mason #endif
4355f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
4365f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4375f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
4385f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
4395f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4405f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
4415f39d397SChris Mason 			BUG_ON(1);
4425f39d397SChris Mason 		}
4435f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
4445f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
4455f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4465f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4475f39d397SChris Mason 			BUG_ON(1);
4485f39d397SChris Mason 		}
449aa5d6bedSChris Mason 	}
4508d7be552SChris Mason 	if (slot < nritems - 1) {
4515f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4525f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
4535f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
4545f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
4555f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
4565f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4575f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4585f39d397SChris Mason 			BUG_ON(1);
459aa5d6bedSChris Mason 		}
4605f39d397SChris Mason 	}
4615f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
4625f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
463aa5d6bedSChris Mason 	return 0;
464aa5d6bedSChris Mason }
465aa5d6bedSChris Mason 
466123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path,
467123abc88SChris Mason 			int level)
468aa5d6bedSChris Mason {
469db94535dSChris Mason #if 0
4705f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
4715f39d397SChris Mason 
472479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
473479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
474479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
4755f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
476db94535dSChris Mason 		return 1;
4775f39d397SChris Mason 	}
478db94535dSChris Mason #endif
479aa5d6bedSChris Mason 	if (level == 0)
480123abc88SChris Mason 		return check_leaf(root, path, level);
481123abc88SChris Mason 	return check_node(root, path, level);
482aa5d6bedSChris Mason }
483aa5d6bedSChris Mason 
48474123bd7SChris Mason /*
4855f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
4865f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
4875f39d397SChris Mason  *
48874123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
48974123bd7SChris Mason  * the place where you would insert key if it is not found in
49074123bd7SChris Mason  * the array.
49174123bd7SChris Mason  *
49274123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
49374123bd7SChris Mason  */
4945f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
4955f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
496be0e5c09SChris Mason 			      int max, int *slot)
497be0e5c09SChris Mason {
498be0e5c09SChris Mason 	int low = 0;
499be0e5c09SChris Mason 	int high = max;
500be0e5c09SChris Mason 	int mid;
501be0e5c09SChris Mason 	int ret;
502479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
5035f39d397SChris Mason 	struct btrfs_disk_key unaligned;
5045f39d397SChris Mason 	unsigned long offset;
5055f39d397SChris Mason 	char *map_token = NULL;
5065f39d397SChris Mason 	char *kaddr = NULL;
5075f39d397SChris Mason 	unsigned long map_start = 0;
5085f39d397SChris Mason 	unsigned long map_len = 0;
509479965d6SChris Mason 	int err;
510be0e5c09SChris Mason 
511be0e5c09SChris Mason 	while(low < high) {
512be0e5c09SChris Mason 		mid = (low + high) / 2;
5135f39d397SChris Mason 		offset = p + mid * item_size;
5145f39d397SChris Mason 
5155f39d397SChris Mason 		if (!map_token || offset < map_start ||
5165f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
5175f39d397SChris Mason 		    map_start + map_len) {
518479965d6SChris Mason 			if (map_token) {
5195f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
520479965d6SChris Mason 				map_token = NULL;
521479965d6SChris Mason 			}
522479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
523479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
524479965d6SChris Mason 						&map_token, &kaddr,
5255f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
5265f39d397SChris Mason 
527479965d6SChris Mason 			if (!err) {
528479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
529479965d6SChris Mason 							map_start);
530479965d6SChris Mason 			} else {
5315f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
5325f39d397SChris Mason 						   offset, sizeof(unaligned));
5335f39d397SChris Mason 				tmp = &unaligned;
534479965d6SChris Mason 			}
535479965d6SChris Mason 
5365f39d397SChris Mason 		} else {
5375f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
5385f39d397SChris Mason 							map_start);
5395f39d397SChris Mason 		}
540be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
541be0e5c09SChris Mason 
542be0e5c09SChris Mason 		if (ret < 0)
543be0e5c09SChris Mason 			low = mid + 1;
544be0e5c09SChris Mason 		else if (ret > 0)
545be0e5c09SChris Mason 			high = mid;
546be0e5c09SChris Mason 		else {
547be0e5c09SChris Mason 			*slot = mid;
548479965d6SChris Mason 			if (map_token)
5495f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
550be0e5c09SChris Mason 			return 0;
551be0e5c09SChris Mason 		}
552be0e5c09SChris Mason 	}
553be0e5c09SChris Mason 	*slot = low;
5545f39d397SChris Mason 	if (map_token)
5555f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
556be0e5c09SChris Mason 	return 1;
557be0e5c09SChris Mason }
558be0e5c09SChris Mason 
55997571fd0SChris Mason /*
56097571fd0SChris Mason  * simple bin_search frontend that does the right thing for
56197571fd0SChris Mason  * leaves vs nodes
56297571fd0SChris Mason  */
5635f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
5645f39d397SChris Mason 		      int level, int *slot)
565be0e5c09SChris Mason {
5665f39d397SChris Mason 	if (level == 0) {
5675f39d397SChris Mason 		return generic_bin_search(eb,
5685f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
5690783fcfcSChris Mason 					  sizeof(struct btrfs_item),
5705f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5717518a238SChris Mason 					  slot);
572be0e5c09SChris Mason 	} else {
5735f39d397SChris Mason 		return generic_bin_search(eb,
5745f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
575123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
5765f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5777518a238SChris Mason 					  slot);
578be0e5c09SChris Mason 	}
579be0e5c09SChris Mason 	return -1;
580be0e5c09SChris Mason }
581be0e5c09SChris Mason 
5825f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
5835f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
584bb803951SChris Mason {
585bb803951SChris Mason 	if (slot < 0)
586bb803951SChris Mason 		return NULL;
5875f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
588bb803951SChris Mason 		return NULL;
589db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
590db94535dSChris Mason 		       btrfs_level_size(root, btrfs_header_level(parent) - 1));
591bb803951SChris Mason }
592bb803951SChris Mason 
593e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
594e089f05cSChris Mason 			 *root, struct btrfs_path *path, int level)
595bb803951SChris Mason {
5965f39d397SChris Mason 	struct extent_buffer *right = NULL;
5975f39d397SChris Mason 	struct extent_buffer *mid;
5985f39d397SChris Mason 	struct extent_buffer *left = NULL;
5995f39d397SChris Mason 	struct extent_buffer *parent = NULL;
600bb803951SChris Mason 	int ret = 0;
601bb803951SChris Mason 	int wret;
602bb803951SChris Mason 	int pslot;
603bb803951SChris Mason 	int orig_slot = path->slots[level];
60454aa1f4dSChris Mason 	int err_on_enospc = 0;
60579f95c82SChris Mason 	u64 orig_ptr;
606bb803951SChris Mason 
607bb803951SChris Mason 	if (level == 0)
608bb803951SChris Mason 		return 0;
609bb803951SChris Mason 
6105f39d397SChris Mason 	mid = path->nodes[level];
6111d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
61279f95c82SChris Mason 
613234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
6145f39d397SChris Mason 		parent = path->nodes[level + 1];
615bb803951SChris Mason 	pslot = path->slots[level + 1];
616bb803951SChris Mason 
61740689478SChris Mason 	/*
61840689478SChris Mason 	 * deal with the case where there is only one pointer in the root
61940689478SChris Mason 	 * by promoting the node below to a root
62040689478SChris Mason 	 */
6215f39d397SChris Mason 	if (!parent) {
6225f39d397SChris Mason 		struct extent_buffer *child;
623bb803951SChris Mason 
6245f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
625bb803951SChris Mason 			return 0;
626bb803951SChris Mason 
627bb803951SChris Mason 		/* promote the child to a root */
6285f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
629bb803951SChris Mason 		BUG_ON(!child);
630bb803951SChris Mason 		root->node = child;
631bb803951SChris Mason 		path->nodes[level] = NULL;
6325f39d397SChris Mason 		clean_tree_block(trans, root, mid);
6335f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
634bb803951SChris Mason 		/* once for the path */
6355f39d397SChris Mason 		free_extent_buffer(mid);
636db94535dSChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
637bb803951SChris Mason 		/* once for the root ptr */
6385f39d397SChris Mason 		free_extent_buffer(mid);
639db94535dSChris Mason 		return ret;
640bb803951SChris Mason 	}
6415f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
642123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
643bb803951SChris Mason 		return 0;
644bb803951SChris Mason 
6455f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
64654aa1f4dSChris Mason 		err_on_enospc = 1;
64754aa1f4dSChris Mason 
6485f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
6495f39d397SChris Mason 	if (left) {
6505f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
6515f39d397SChris Mason 				       parent, pslot - 1, &left);
65254aa1f4dSChris Mason 		if (wret) {
65354aa1f4dSChris Mason 			ret = wret;
65454aa1f4dSChris Mason 			goto enospc;
65554aa1f4dSChris Mason 		}
6562cc58cf2SChris Mason 	}
6575f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
6585f39d397SChris Mason 	if (right) {
6595f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
6605f39d397SChris Mason 				       parent, pslot + 1, &right);
6612cc58cf2SChris Mason 		if (wret) {
6622cc58cf2SChris Mason 			ret = wret;
6632cc58cf2SChris Mason 			goto enospc;
6642cc58cf2SChris Mason 		}
6652cc58cf2SChris Mason 	}
6662cc58cf2SChris Mason 
6672cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
6685f39d397SChris Mason 	if (left) {
6695f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
6705f39d397SChris Mason 		wret = push_node_left(trans, root, left, mid);
67179f95c82SChris Mason 		if (wret < 0)
67279f95c82SChris Mason 			ret = wret;
6735f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
67454aa1f4dSChris Mason 			err_on_enospc = 1;
675bb803951SChris Mason 	}
67679f95c82SChris Mason 
67779f95c82SChris Mason 	/*
67879f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
67979f95c82SChris Mason 	 */
6805f39d397SChris Mason 	if (right) {
6815f39d397SChris Mason 		wret = push_node_left(trans, root, mid, right);
68254aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
68379f95c82SChris Mason 			ret = wret;
6845f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
685db94535dSChris Mason 			u64 bytenr = right->start;
686db94535dSChris Mason 			u32 blocksize = right->len;
687db94535dSChris Mason 
6885f39d397SChris Mason 			clean_tree_block(trans, root, right);
6895f39d397SChris Mason 			wait_on_tree_block_writeback(root, right);
6905f39d397SChris Mason 			free_extent_buffer(right);
691bb803951SChris Mason 			right = NULL;
692e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
693e089f05cSChris Mason 				       1);
694bb803951SChris Mason 			if (wret)
695bb803951SChris Mason 				ret = wret;
696db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
697db94535dSChris Mason 						 blocksize, 1);
698bb803951SChris Mason 			if (wret)
699bb803951SChris Mason 				ret = wret;
700bb803951SChris Mason 		} else {
7015f39d397SChris Mason 			struct btrfs_disk_key right_key;
7025f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
7035f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
7045f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
705bb803951SChris Mason 		}
706bb803951SChris Mason 	}
7075f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
70879f95c82SChris Mason 		/*
70979f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
71079f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
71179f95c82SChris Mason 		 * could try to delete the only pointer in this node.
71279f95c82SChris Mason 		 * So, pull some keys from the left.
71379f95c82SChris Mason 		 * There has to be a left pointer at this point because
71479f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
71579f95c82SChris Mason 		 * right
71679f95c82SChris Mason 		 */
7175f39d397SChris Mason 		BUG_ON(!left);
7185f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
71954aa1f4dSChris Mason 		if (wret < 0) {
72079f95c82SChris Mason 			ret = wret;
72154aa1f4dSChris Mason 			goto enospc;
72254aa1f4dSChris Mason 		}
72379f95c82SChris Mason 		BUG_ON(wret == 1);
72479f95c82SChris Mason 	}
7255f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
72679f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
727db94535dSChris Mason 		u64 bytenr = mid->start;
728db94535dSChris Mason 		u32 blocksize = mid->len;
7295f39d397SChris Mason 		clean_tree_block(trans, root, mid);
7305f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
7315f39d397SChris Mason 		free_extent_buffer(mid);
732bb803951SChris Mason 		mid = NULL;
733e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
734bb803951SChris Mason 		if (wret)
735bb803951SChris Mason 			ret = wret;
736db94535dSChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
737bb803951SChris Mason 		if (wret)
738bb803951SChris Mason 			ret = wret;
73979f95c82SChris Mason 	} else {
74079f95c82SChris Mason 		/* update the parent key to reflect our changes */
7415f39d397SChris Mason 		struct btrfs_disk_key mid_key;
7425f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
7435f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
7445f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
74579f95c82SChris Mason 	}
746bb803951SChris Mason 
74779f95c82SChris Mason 	/* update the path */
7485f39d397SChris Mason 	if (left) {
7495f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
7505f39d397SChris Mason 			extent_buffer_get(left);
7515f39d397SChris Mason 			path->nodes[level] = left;
752bb803951SChris Mason 			path->slots[level + 1] -= 1;
753bb803951SChris Mason 			path->slots[level] = orig_slot;
7545f39d397SChris Mason 			if (mid)
7555f39d397SChris Mason 				free_extent_buffer(mid);
756bb803951SChris Mason 		} else {
7575f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
758bb803951SChris Mason 			path->slots[level] = orig_slot;
759bb803951SChris Mason 		}
760bb803951SChris Mason 	}
76179f95c82SChris Mason 	/* double check we haven't messed things up */
762123abc88SChris Mason 	check_block(root, path, level);
763e20d96d6SChris Mason 	if (orig_ptr !=
7645f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
76579f95c82SChris Mason 		BUG();
76654aa1f4dSChris Mason enospc:
7675f39d397SChris Mason 	if (right)
7685f39d397SChris Mason 		free_extent_buffer(right);
7695f39d397SChris Mason 	if (left)
7705f39d397SChris Mason 		free_extent_buffer(left);
771bb803951SChris Mason 	return ret;
772bb803951SChris Mason }
773bb803951SChris Mason 
774e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
775e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
776e66f709bSChris Mason 				struct btrfs_root *root,
777e66f709bSChris Mason 				struct btrfs_path *path, int level)
778e66f709bSChris Mason {
7795f39d397SChris Mason 	struct extent_buffer *right = NULL;
7805f39d397SChris Mason 	struct extent_buffer *mid;
7815f39d397SChris Mason 	struct extent_buffer *left = NULL;
7825f39d397SChris Mason 	struct extent_buffer *parent = NULL;
783e66f709bSChris Mason 	int ret = 0;
784e66f709bSChris Mason 	int wret;
785e66f709bSChris Mason 	int pslot;
786e66f709bSChris Mason 	int orig_slot = path->slots[level];
787e66f709bSChris Mason 	u64 orig_ptr;
788e66f709bSChris Mason 
789e66f709bSChris Mason 	if (level == 0)
790e66f709bSChris Mason 		return 1;
791e66f709bSChris Mason 
7925f39d397SChris Mason 	mid = path->nodes[level];
793e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
794e66f709bSChris Mason 
795e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
7965f39d397SChris Mason 		parent = path->nodes[level + 1];
797e66f709bSChris Mason 	pslot = path->slots[level + 1];
798e66f709bSChris Mason 
7995f39d397SChris Mason 	if (!parent)
800e66f709bSChris Mason 		return 1;
801e66f709bSChris Mason 
8025f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
803e66f709bSChris Mason 
804e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
8055f39d397SChris Mason 	if (left) {
806e66f709bSChris Mason 		u32 left_nr;
8075f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
80833ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
80933ade1f8SChris Mason 			wret = 1;
81033ade1f8SChris Mason 		} else {
8115f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
8125f39d397SChris Mason 					      pslot - 1, &left);
81354aa1f4dSChris Mason 			if (ret)
81454aa1f4dSChris Mason 				wret = 1;
81554aa1f4dSChris Mason 			else {
81654aa1f4dSChris Mason 				wret = push_node_left(trans, root,
8175f39d397SChris Mason 						      left, mid);
81854aa1f4dSChris Mason 			}
81933ade1f8SChris Mason 		}
820e66f709bSChris Mason 		if (wret < 0)
821e66f709bSChris Mason 			ret = wret;
822e66f709bSChris Mason 		if (wret == 0) {
8235f39d397SChris Mason 			struct btrfs_disk_key disk_key;
824e66f709bSChris Mason 			orig_slot += left_nr;
8255f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
8265f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
8275f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8285f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
8295f39d397SChris Mason 				path->nodes[level] = left;
830e66f709bSChris Mason 				path->slots[level + 1] -= 1;
831e66f709bSChris Mason 				path->slots[level] = orig_slot;
8325f39d397SChris Mason 				free_extent_buffer(mid);
833e66f709bSChris Mason 			} else {
834e66f709bSChris Mason 				orig_slot -=
8355f39d397SChris Mason 					btrfs_header_nritems(left);
836e66f709bSChris Mason 				path->slots[level] = orig_slot;
8375f39d397SChris Mason 				free_extent_buffer(left);
838e66f709bSChris Mason 			}
839e66f709bSChris Mason 			return 0;
840e66f709bSChris Mason 		}
8415f39d397SChris Mason 		free_extent_buffer(left);
842e66f709bSChris Mason 	}
8435f39d397SChris Mason 	right= read_node_slot(root, parent, pslot + 1);
844e66f709bSChris Mason 
845e66f709bSChris Mason 	/*
846e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
847e66f709bSChris Mason 	 */
8485f39d397SChris Mason 	if (right) {
84933ade1f8SChris Mason 		u32 right_nr;
8505f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
85133ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
85233ade1f8SChris Mason 			wret = 1;
85333ade1f8SChris Mason 		} else {
8545f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
8555f39d397SChris Mason 					      parent, pslot + 1,
8565f39d397SChris Mason 					      &right);
85754aa1f4dSChris Mason 			if (ret)
85854aa1f4dSChris Mason 				wret = 1;
85954aa1f4dSChris Mason 			else {
86033ade1f8SChris Mason 				wret = balance_node_right(trans, root,
8615f39d397SChris Mason 							  right, mid);
86233ade1f8SChris Mason 			}
86354aa1f4dSChris Mason 		}
864e66f709bSChris Mason 		if (wret < 0)
865e66f709bSChris Mason 			ret = wret;
866e66f709bSChris Mason 		if (wret == 0) {
8675f39d397SChris Mason 			struct btrfs_disk_key disk_key;
8685f39d397SChris Mason 
8695f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
8705f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
8715f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8725f39d397SChris Mason 
8735f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
8745f39d397SChris Mason 				path->nodes[level] = right;
875e66f709bSChris Mason 				path->slots[level + 1] += 1;
876e66f709bSChris Mason 				path->slots[level] = orig_slot -
8775f39d397SChris Mason 					btrfs_header_nritems(mid);
8785f39d397SChris Mason 				free_extent_buffer(mid);
879e66f709bSChris Mason 			} else {
8805f39d397SChris Mason 				free_extent_buffer(right);
881e66f709bSChris Mason 			}
882e66f709bSChris Mason 			return 0;
883e66f709bSChris Mason 		}
8845f39d397SChris Mason 		free_extent_buffer(right);
885e66f709bSChris Mason 	}
886e66f709bSChris Mason 	return 1;
887e66f709bSChris Mason }
888e66f709bSChris Mason 
88974123bd7SChris Mason /*
8903c69faecSChris Mason  * readahead one full node of leaves
8913c69faecSChris Mason  */
8923c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
8936702ed49SChris Mason 			     int level, int slot)
8943c69faecSChris Mason {
8955f39d397SChris Mason 	struct extent_buffer *node;
8963c69faecSChris Mason 	u32 nritems;
8973c69faecSChris Mason 	u64 search;
898*6b80053dSChris Mason 	u64 lowest_read;
899*6b80053dSChris Mason 	u64 highest_read;
900*6b80053dSChris Mason 	u64 nread = 0;
9013c69faecSChris Mason 	int direction = path->reada;
9025f39d397SChris Mason 	struct extent_buffer *eb;
903*6b80053dSChris Mason 	u32 nr;
904*6b80053dSChris Mason 	u32 blocksize;
905*6b80053dSChris Mason 	u32 nscan = 0;
906db94535dSChris Mason 
9076702ed49SChris Mason 	if (level == 0)
9083c69faecSChris Mason 		return;
9093c69faecSChris Mason 
9106702ed49SChris Mason 	if (!path->nodes[level])
9116702ed49SChris Mason 		return;
9126702ed49SChris Mason 
9135f39d397SChris Mason 	node = path->nodes[level];
9143c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
915*6b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
916*6b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
9175f39d397SChris Mason 	if (eb) {
9185f39d397SChris Mason 		free_extent_buffer(eb);
9193c69faecSChris Mason 		return;
9203c69faecSChris Mason 	}
9213c69faecSChris Mason 
922*6b80053dSChris Mason 	highest_read = search;
923*6b80053dSChris Mason 	lowest_read = search;
924*6b80053dSChris Mason 
9255f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
926*6b80053dSChris Mason 	nr = slot;
9273c69faecSChris Mason 	while(1) {
928*6b80053dSChris Mason 		if (direction < 0) {
929*6b80053dSChris Mason 			if (nr == 0)
9303c69faecSChris Mason 				break;
931*6b80053dSChris Mason 			nr--;
932*6b80053dSChris Mason 		} else if (direction > 0) {
933*6b80053dSChris Mason 			nr++;
934*6b80053dSChris Mason 			if (nr >= nritems)
935*6b80053dSChris Mason 				break;
9363c69faecSChris Mason 		}
937*6b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
938*6b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
939*6b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
940*6b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
941*6b80053dSChris Mason 			readahead_tree_block(root, search, blocksize);
942*6b80053dSChris Mason 			nread += blocksize;
9433c69faecSChris Mason 		}
944*6b80053dSChris Mason 		nscan++;
945*6b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
946*6b80053dSChris Mason 			break;
947*6b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
948*6b80053dSChris Mason 			break;
949*6b80053dSChris Mason 
950*6b80053dSChris Mason 		if (search < lowest_read)
951*6b80053dSChris Mason 			lowest_read = search;
952*6b80053dSChris Mason 		if (search > highest_read)
953*6b80053dSChris Mason 			highest_read = search;
9543c69faecSChris Mason 	}
9553c69faecSChris Mason }
9563c69faecSChris Mason /*
95774123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
95874123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
95974123bd7SChris Mason  * level of the path (level 0)
96074123bd7SChris Mason  *
96174123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
962aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
963aa5d6bedSChris Mason  * search a negative error number is returned.
96497571fd0SChris Mason  *
96597571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
96697571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
96797571fd0SChris Mason  * possible)
96874123bd7SChris Mason  */
969e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
970e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
971e089f05cSChris Mason 		      ins_len, int cow)
972be0e5c09SChris Mason {
9735f39d397SChris Mason 	struct extent_buffer *b;
974db94535dSChris Mason 	u64 bytenr;
975be0e5c09SChris Mason 	int slot;
976be0e5c09SChris Mason 	int ret;
977be0e5c09SChris Mason 	int level;
9783c69faecSChris Mason 	int should_reada = p->reada;
9799f3a7427SChris Mason 	u8 lowest_level = 0;
9809f3a7427SChris Mason 
9816702ed49SChris Mason 	lowest_level = p->lowest_level;
9826702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
98322b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
98422b0ebdaSChris Mason 	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
985bb803951SChris Mason again:
986bb803951SChris Mason 	b = root->node;
9875f39d397SChris Mason 	extent_buffer_get(b);
988eb60ceacSChris Mason 	while (b) {
9895f39d397SChris Mason 		level = btrfs_header_level(b);
99002217ed2SChris Mason 		if (cow) {
99102217ed2SChris Mason 			int wret;
992e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
993e20d96d6SChris Mason 					       p->nodes[level + 1],
994e20d96d6SChris Mason 					       p->slots[level + 1],
995252c38f0SYan 					       &b);
99654aa1f4dSChris Mason 			if (wret) {
9975f39d397SChris Mason 				free_extent_buffer(b);
99854aa1f4dSChris Mason 				return wret;
99954aa1f4dSChris Mason 			}
100002217ed2SChris Mason 		}
100102217ed2SChris Mason 		BUG_ON(!cow && ins_len);
10025f39d397SChris Mason 		if (level != btrfs_header_level(b))
10032c90e5d6SChris Mason 			WARN_ON(1);
10045f39d397SChris Mason 		level = btrfs_header_level(b);
1005eb60ceacSChris Mason 		p->nodes[level] = b;
1006123abc88SChris Mason 		ret = check_block(root, p, level);
1007aa5d6bedSChris Mason 		if (ret)
1008aa5d6bedSChris Mason 			return -1;
10095f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
10105f39d397SChris Mason 		if (level != 0) {
1011be0e5c09SChris Mason 			if (ret && slot > 0)
1012be0e5c09SChris Mason 				slot -= 1;
1013be0e5c09SChris Mason 			p->slots[level] = slot;
10145f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
1015d4dbff95SChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
1016e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
10175c680ed6SChris Mason 				BUG_ON(sret > 0);
10185c680ed6SChris Mason 				if (sret)
10195c680ed6SChris Mason 					return sret;
10205c680ed6SChris Mason 				b = p->nodes[level];
10215c680ed6SChris Mason 				slot = p->slots[level];
1022bb803951SChris Mason 			} else if (ins_len < 0) {
1023e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1024e089f05cSChris Mason 							 level);
1025bb803951SChris Mason 				if (sret)
1026bb803951SChris Mason 					return sret;
1027bb803951SChris Mason 				b = p->nodes[level];
1028f510cfecSChris Mason 				if (!b) {
1029f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1030bb803951SChris Mason 					goto again;
1031f510cfecSChris Mason 				}
1032bb803951SChris Mason 				slot = p->slots[level];
10335f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
10345c680ed6SChris Mason 			}
10359f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
10369f3a7427SChris Mason 			if (level == lowest_level)
10379f3a7427SChris Mason 				break;
1038db94535dSChris Mason 			bytenr = btrfs_node_blockptr(b, slot);
10396702ed49SChris Mason 			if (should_reada)
10406702ed49SChris Mason 				reada_for_search(root, p, level, slot);
1041db94535dSChris Mason 			b = read_tree_block(root, bytenr,
1042db94535dSChris Mason 					    btrfs_level_size(root, level - 1));
1043be0e5c09SChris Mason 		} else {
1044be0e5c09SChris Mason 			p->slots[level] = slot;
10455f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
10460783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1047d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1048d4dbff95SChris Mason 						      p, ins_len);
10495c680ed6SChris Mason 				BUG_ON(sret > 0);
10505c680ed6SChris Mason 				if (sret)
10515c680ed6SChris Mason 					return sret;
10525c680ed6SChris Mason 			}
1053be0e5c09SChris Mason 			return ret;
1054be0e5c09SChris Mason 		}
1055be0e5c09SChris Mason 	}
1056aa5d6bedSChris Mason 	return 1;
1057be0e5c09SChris Mason }
1058be0e5c09SChris Mason 
105974123bd7SChris Mason /*
106074123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
106174123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
106274123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
106374123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
106474123bd7SChris Mason  * higher levels
1065aa5d6bedSChris Mason  *
1066aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1067aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
106874123bd7SChris Mason  */
10695f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
10705f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
10715f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1072be0e5c09SChris Mason {
1073be0e5c09SChris Mason 	int i;
1074aa5d6bedSChris Mason 	int ret = 0;
10755f39d397SChris Mason 	struct extent_buffer *t;
10765f39d397SChris Mason 
1077234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1078be0e5c09SChris Mason 		int tslot = path->slots[i];
1079eb60ceacSChris Mason 		if (!path->nodes[i])
1080be0e5c09SChris Mason 			break;
10815f39d397SChris Mason 		t = path->nodes[i];
10825f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1083d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1084be0e5c09SChris Mason 		if (tslot != 0)
1085be0e5c09SChris Mason 			break;
1086be0e5c09SChris Mason 	}
1087aa5d6bedSChris Mason 	return ret;
1088be0e5c09SChris Mason }
1089be0e5c09SChris Mason 
109074123bd7SChris Mason /*
109174123bd7SChris Mason  * try to push data from one node into the next node left in the
109279f95c82SChris Mason  * tree.
1093aa5d6bedSChris Mason  *
1094aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1095aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
109674123bd7SChris Mason  */
1097e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
10985f39d397SChris Mason 			  *root, struct extent_buffer *dst,
10995f39d397SChris Mason 			  struct extent_buffer *src)
1100be0e5c09SChris Mason {
1101be0e5c09SChris Mason 	int push_items = 0;
1102bb803951SChris Mason 	int src_nritems;
1103bb803951SChris Mason 	int dst_nritems;
1104aa5d6bedSChris Mason 	int ret = 0;
1105be0e5c09SChris Mason 
11065f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11075f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1108123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
110954aa1f4dSChris Mason 
1110eb60ceacSChris Mason 	if (push_items <= 0) {
1111be0e5c09SChris Mason 		return 1;
1112eb60ceacSChris Mason 	}
1113be0e5c09SChris Mason 
1114be0e5c09SChris Mason 	if (src_nritems < push_items)
1115be0e5c09SChris Mason 		push_items = src_nritems;
111679f95c82SChris Mason 
11175f39d397SChris Mason 	copy_extent_buffer(dst, src,
11185f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
11195f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1120123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
11215f39d397SChris Mason 
1122bb803951SChris Mason 	if (push_items < src_nritems) {
11235f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
11245f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1125e2fa7227SChris Mason 				      (src_nritems - push_items) *
1126123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1127bb803951SChris Mason 	}
11285f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11295f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
11305f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11315f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1132bb803951SChris Mason 	return ret;
1133be0e5c09SChris Mason }
1134be0e5c09SChris Mason 
113597571fd0SChris Mason /*
113679f95c82SChris Mason  * try to push data from one node into the next node right in the
113779f95c82SChris Mason  * tree.
113879f95c82SChris Mason  *
113979f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
114079f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
114179f95c82SChris Mason  *
114279f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
114379f95c82SChris Mason  */
11445f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
11455f39d397SChris Mason 			      struct btrfs_root *root,
11465f39d397SChris Mason 			      struct extent_buffer *dst,
11475f39d397SChris Mason 			      struct extent_buffer *src)
114879f95c82SChris Mason {
114979f95c82SChris Mason 	int push_items = 0;
115079f95c82SChris Mason 	int max_push;
115179f95c82SChris Mason 	int src_nritems;
115279f95c82SChris Mason 	int dst_nritems;
115379f95c82SChris Mason 	int ret = 0;
115479f95c82SChris Mason 
11555f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11565f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1157123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
11585f39d397SChris Mason 	if (push_items <= 0)
115979f95c82SChris Mason 		return 1;
116079f95c82SChris Mason 
116179f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
116279f95c82SChris Mason 	/* don't try to empty the node */
1163252c38f0SYan 	if (max_push >= src_nritems)
116479f95c82SChris Mason 		return 1;
1165252c38f0SYan 
116679f95c82SChris Mason 	if (max_push < push_items)
116779f95c82SChris Mason 		push_items = max_push;
116879f95c82SChris Mason 
11695f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
11705f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
11715f39d397SChris Mason 				      (dst_nritems) *
11725f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1173d6025579SChris Mason 
11745f39d397SChris Mason 	copy_extent_buffer(dst, src,
11755f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
11765f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1177123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
117879f95c82SChris Mason 
11795f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11805f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
118179f95c82SChris Mason 
11825f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11835f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
118479f95c82SChris Mason 	return ret;
118579f95c82SChris Mason }
118679f95c82SChris Mason 
118779f95c82SChris Mason /*
118897571fd0SChris Mason  * helper function to insert a new root level in the tree.
118997571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
119097571fd0SChris Mason  * point to the existing root
1191aa5d6bedSChris Mason  *
1192aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
119397571fd0SChris Mason  */
11945f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans,
11955f39d397SChris Mason 			   struct btrfs_root *root,
11965f39d397SChris Mason 			   struct btrfs_path *path, int level)
119774123bd7SChris Mason {
11985f39d397SChris Mason 	struct extent_buffer *lower;
11995f39d397SChris Mason 	struct extent_buffer *c;
12005f39d397SChris Mason 	struct btrfs_disk_key lower_key;
12015c680ed6SChris Mason 
12025c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
12035c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
12045c680ed6SChris Mason 
1205db94535dSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
1206db94535dSChris Mason 				   root->node->start, 0);
12075f39d397SChris Mason 	if (IS_ERR(c))
12085f39d397SChris Mason 		return PTR_ERR(c);
12095f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
12105f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
12115f39d397SChris Mason 	btrfs_set_header_level(c, level);
1212db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
12135f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
12145f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
12155f39d397SChris Mason 	lower = path->nodes[level-1];
1216d5719762SChris Mason 
12175f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
12185f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
12195f39d397SChris Mason 			    BTRFS_FSID_SIZE);
12205f39d397SChris Mason 	if (level == 1)
12215f39d397SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
12225f39d397SChris Mason 	else
12235f39d397SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
12245f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1225db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
12265f39d397SChris Mason 
12275f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1228d5719762SChris Mason 
1229cfaa7295SChris Mason 	/* the super has an extra ref to root->node */
12305f39d397SChris Mason 	free_extent_buffer(root->node);
12315f39d397SChris Mason 	root->node = c;
12325f39d397SChris Mason 	extent_buffer_get(c);
12335f39d397SChris Mason 	path->nodes[level] = c;
123474123bd7SChris Mason 	path->slots[level] = 0;
123574123bd7SChris Mason 	return 0;
123674123bd7SChris Mason }
12375c680ed6SChris Mason 
12385c680ed6SChris Mason /*
12395c680ed6SChris Mason  * worker function to insert a single pointer in a node.
12405c680ed6SChris Mason  * the node should have enough room for the pointer already
124197571fd0SChris Mason  *
12425c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
12435c680ed6SChris Mason  * blocknr is the block the key points to.
1244aa5d6bedSChris Mason  *
1245aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
12465c680ed6SChris Mason  */
1247e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1248e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1249db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
12505c680ed6SChris Mason {
12515f39d397SChris Mason 	struct extent_buffer *lower;
12525c680ed6SChris Mason 	int nritems;
12535c680ed6SChris Mason 
12545c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
12555f39d397SChris Mason 	lower = path->nodes[level];
12565f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
125774123bd7SChris Mason 	if (slot > nritems)
125874123bd7SChris Mason 		BUG();
1259123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
126074123bd7SChris Mason 		BUG();
126174123bd7SChris Mason 	if (slot != nritems) {
12625f39d397SChris Mason 		memmove_extent_buffer(lower,
12635f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
12645f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1265123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
126674123bd7SChris Mason 	}
12675f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1268db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
12695f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
12705f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
127174123bd7SChris Mason 	return 0;
127274123bd7SChris Mason }
127374123bd7SChris Mason 
127497571fd0SChris Mason /*
127597571fd0SChris Mason  * split the node at the specified level in path in two.
127697571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
127797571fd0SChris Mason  *
127897571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
127997571fd0SChris Mason  * left and right, if either one works, it returns right away.
1280aa5d6bedSChris Mason  *
1281aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
128297571fd0SChris Mason  */
1283e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1284e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1285be0e5c09SChris Mason {
12865f39d397SChris Mason 	struct extent_buffer *c;
12875f39d397SChris Mason 	struct extent_buffer *split;
12885f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1289be0e5c09SChris Mason 	int mid;
12905c680ed6SChris Mason 	int ret;
1291aa5d6bedSChris Mason 	int wret;
12927518a238SChris Mason 	u32 c_nritems;
1293be0e5c09SChris Mason 
12945f39d397SChris Mason 	c = path->nodes[level];
12955f39d397SChris Mason 	if (c == root->node) {
12965c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1297e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
12985c680ed6SChris Mason 		if (ret)
12995c680ed6SChris Mason 			return ret;
1300e66f709bSChris Mason 	} else {
1301e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
13025f39d397SChris Mason 		c = path->nodes[level];
13035f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1304e66f709bSChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1305e66f709bSChris Mason 			return 0;
130654aa1f4dSChris Mason 		if (ret < 0)
130754aa1f4dSChris Mason 			return ret;
13085c680ed6SChris Mason 	}
1309e66f709bSChris Mason 
13105f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
1311db94535dSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
1312db94535dSChris Mason 				       c->start, 0);
13135f39d397SChris Mason 	if (IS_ERR(split))
13145f39d397SChris Mason 		return PTR_ERR(split);
131554aa1f4dSChris Mason 
13165f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
13175f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1318db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
13195f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
13205f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
13215f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
13225f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
13235f39d397SChris Mason 			    BTRFS_FSID_SIZE);
13245f39d397SChris Mason 
13257518a238SChris Mason 	mid = (c_nritems + 1) / 2;
13265f39d397SChris Mason 
13275f39d397SChris Mason 	copy_extent_buffer(split, c,
13285f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
13295f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1330123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
13315f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
13325f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1333aa5d6bedSChris Mason 	ret = 0;
1334aa5d6bedSChris Mason 
13355f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
13365f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
13375f39d397SChris Mason 
13385f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1339db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
13405f39d397SChris Mason 			  path->slots[level + 1] + 1,
1341123abc88SChris Mason 			  level + 1);
1342aa5d6bedSChris Mason 	if (wret)
1343aa5d6bedSChris Mason 		ret = wret;
1344aa5d6bedSChris Mason 
13455de08d7dSChris Mason 	if (path->slots[level] >= mid) {
13465c680ed6SChris Mason 		path->slots[level] -= mid;
13475f39d397SChris Mason 		free_extent_buffer(c);
13485f39d397SChris Mason 		path->nodes[level] = split;
13495c680ed6SChris Mason 		path->slots[level + 1] += 1;
1350eb60ceacSChris Mason 	} else {
13515f39d397SChris Mason 		free_extent_buffer(split);
1352be0e5c09SChris Mason 	}
1353aa5d6bedSChris Mason 	return ret;
1354be0e5c09SChris Mason }
1355be0e5c09SChris Mason 
135674123bd7SChris Mason /*
135774123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
135874123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
135974123bd7SChris Mason  * space used both by the item structs and the item data
136074123bd7SChris Mason  */
13615f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1362be0e5c09SChris Mason {
1363be0e5c09SChris Mason 	int data_len;
13645f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1365d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1366be0e5c09SChris Mason 
1367be0e5c09SChris Mason 	if (!nr)
1368be0e5c09SChris Mason 		return 0;
13695f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
13705f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
13710783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1372d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1373be0e5c09SChris Mason 	return data_len;
1374be0e5c09SChris Mason }
1375be0e5c09SChris Mason 
137674123bd7SChris Mason /*
1377d4dbff95SChris Mason  * The space between the end of the leaf items and
1378d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1379d4dbff95SChris Mason  * the leaf has left for both items and data
1380d4dbff95SChris Mason  */
13815f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1382d4dbff95SChris Mason {
13835f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
13845f39d397SChris Mason 	int ret;
13855f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
13865f39d397SChris Mason 	if (ret < 0) {
13875f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
13885f39d397SChris Mason 		       ret, BTRFS_LEAF_DATA_SIZE(root),
13895f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
13905f39d397SChris Mason 	}
13915f39d397SChris Mason 	return ret;
1392d4dbff95SChris Mason }
1393d4dbff95SChris Mason 
1394d4dbff95SChris Mason /*
139500ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
139600ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1397aa5d6bedSChris Mason  *
1398aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1399aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
140000ec4c51SChris Mason  */
1401e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1402e089f05cSChris Mason 			   *root, struct btrfs_path *path, int data_size)
140300ec4c51SChris Mason {
14045f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
14055f39d397SChris Mason 	struct extent_buffer *right;
14065f39d397SChris Mason 	struct extent_buffer *upper;
14075f39d397SChris Mason 	struct btrfs_disk_key disk_key;
140800ec4c51SChris Mason 	int slot;
140900ec4c51SChris Mason 	int i;
141000ec4c51SChris Mason 	int free_space;
141100ec4c51SChris Mason 	int push_space = 0;
141200ec4c51SChris Mason 	int push_items = 0;
14130783fcfcSChris Mason 	struct btrfs_item *item;
14147518a238SChris Mason 	u32 left_nritems;
14157518a238SChris Mason 	u32 right_nritems;
14165f39d397SChris Mason 	u32 data_end;
1417db94535dSChris Mason 	u32 this_item_size;
141854aa1f4dSChris Mason 	int ret;
141900ec4c51SChris Mason 
142000ec4c51SChris Mason 	slot = path->slots[1];
142100ec4c51SChris Mason 	if (!path->nodes[1]) {
142200ec4c51SChris Mason 		return 1;
142300ec4c51SChris Mason 	}
142400ec4c51SChris Mason 	upper = path->nodes[1];
14255f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
142600ec4c51SChris Mason 		return 1;
14275f39d397SChris Mason 
1428db94535dSChris Mason 	right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1429db94535dSChris Mason 				root->leafsize);
1430123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14310783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14325f39d397SChris Mason 		free_extent_buffer(right);
143302217ed2SChris Mason 		return 1;
143402217ed2SChris Mason 	}
143502217ed2SChris Mason 
14365f39d397SChris Mason 	/* cow and double check */
14375f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
14385f39d397SChris Mason 			      slot + 1, &right);
14395f39d397SChris Mason 	if (ret) {
14405f39d397SChris Mason 		free_extent_buffer(right);
1441a429e513SChris Mason 		return 1;
1442a429e513SChris Mason 	}
14435f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14445f39d397SChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14455f39d397SChris Mason 		free_extent_buffer(right);
14465f39d397SChris Mason 		return 1;
14475f39d397SChris Mason 	}
14485f39d397SChris Mason 
14495f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
14505f39d397SChris Mason 	if (left_nritems == 0) {
14515f39d397SChris Mason 		free_extent_buffer(right);
14525f39d397SChris Mason 		return 1;
14535f39d397SChris Mason 	}
14545f39d397SChris Mason 
1455a429e513SChris Mason 	for (i = left_nritems - 1; i >= 1; i--) {
14565f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1457db94535dSChris Mason 
145800ec4c51SChris Mason 		if (path->slots[0] == i)
145900ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1460db94535dSChris Mason 
1461db94535dSChris Mason 		if (!left->map_token) {
1462db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1463db94535dSChris Mason 					sizeof(struct btrfs_item),
1464db94535dSChris Mason 					&left->map_token, &left->kaddr,
1465db94535dSChris Mason 					&left->map_start, &left->map_len,
1466db94535dSChris Mason 					KM_USER1);
1467db94535dSChris Mason 		}
1468db94535dSChris Mason 
1469db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1470db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
147100ec4c51SChris Mason 			break;
147200ec4c51SChris Mason 		push_items++;
1473db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1474db94535dSChris Mason 	}
1475db94535dSChris Mason 	if (left->map_token) {
1476db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1477db94535dSChris Mason 		left->map_token = NULL;
147800ec4c51SChris Mason 	}
14795f39d397SChris Mason 
148000ec4c51SChris Mason 	if (push_items == 0) {
14815f39d397SChris Mason 		free_extent_buffer(right);
148200ec4c51SChris Mason 		return 1;
148300ec4c51SChris Mason 	}
14845f39d397SChris Mason 
1485a429e513SChris Mason 	if (push_items == left_nritems)
1486a429e513SChris Mason 		WARN_ON(1);
14875f39d397SChris Mason 
148800ec4c51SChris Mason 	/* push left to right */
14895f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
14905f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1491123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
14925f39d397SChris Mason 
149300ec4c51SChris Mason 	/* make room in the right data area */
14945f39d397SChris Mason 	data_end = leaf_data_end(root, right);
14955f39d397SChris Mason 	memmove_extent_buffer(right,
14965f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
14975f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
14985f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
14995f39d397SChris Mason 
150000ec4c51SChris Mason 	/* copy from the left data area */
15015f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1502d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1503d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1504d6025579SChris Mason 		     push_space);
15055f39d397SChris Mason 
15065f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
15075f39d397SChris Mason 			      btrfs_item_nr_offset(0),
15080783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
15095f39d397SChris Mason 
151000ec4c51SChris Mason 	/* copy the items from left to right */
15115f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
15125f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
15130783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
151400ec4c51SChris Mason 
151500ec4c51SChris Mason 	/* update the item pointers */
15167518a238SChris Mason 	right_nritems += push_items;
15175f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1518123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
1519db94535dSChris Mason 
15207518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
15215f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1522db94535dSChris Mason 		if (!right->map_token) {
1523db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1524db94535dSChris Mason 					sizeof(struct btrfs_item),
1525db94535dSChris Mason 					&right->map_token, &right->kaddr,
1526db94535dSChris Mason 					&right->map_start, &right->map_len,
1527db94535dSChris Mason 					KM_USER1);
1528db94535dSChris Mason 		}
1529db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1530db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1531db94535dSChris Mason 	}
1532db94535dSChris Mason 
1533db94535dSChris Mason 	if (right->map_token) {
1534db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1535db94535dSChris Mason 		right->map_token = NULL;
153600ec4c51SChris Mason 	}
15377518a238SChris Mason 	left_nritems -= push_items;
15385f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
153900ec4c51SChris Mason 
15405f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
15415f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1542a429e513SChris Mason 
15435f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
15445f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1545d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
154602217ed2SChris Mason 
154700ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
15487518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
15497518a238SChris Mason 		path->slots[0] -= left_nritems;
15505f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
15515f39d397SChris Mason 		path->nodes[0] = right;
155200ec4c51SChris Mason 		path->slots[1] += 1;
155300ec4c51SChris Mason 	} else {
15545f39d397SChris Mason 		free_extent_buffer(right);
155500ec4c51SChris Mason 	}
155600ec4c51SChris Mason 	return 0;
155700ec4c51SChris Mason }
155800ec4c51SChris Mason /*
155974123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
156074123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
156174123bd7SChris Mason  */
1562e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1563e089f05cSChris Mason 			  *root, struct btrfs_path *path, int data_size)
1564be0e5c09SChris Mason {
15655f39d397SChris Mason 	struct btrfs_disk_key disk_key;
15665f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
15675f39d397SChris Mason 	struct extent_buffer *left;
1568be0e5c09SChris Mason 	int slot;
1569be0e5c09SChris Mason 	int i;
1570be0e5c09SChris Mason 	int free_space;
1571be0e5c09SChris Mason 	int push_space = 0;
1572be0e5c09SChris Mason 	int push_items = 0;
15730783fcfcSChris Mason 	struct btrfs_item *item;
15747518a238SChris Mason 	u32 old_left_nritems;
15755f39d397SChris Mason 	u32 right_nritems;
1576aa5d6bedSChris Mason 	int ret = 0;
1577aa5d6bedSChris Mason 	int wret;
1578db94535dSChris Mason 	u32 this_item_size;
1579db94535dSChris Mason 	u32 old_left_item_size;
1580be0e5c09SChris Mason 
1581be0e5c09SChris Mason 	slot = path->slots[1];
15825f39d397SChris Mason 	if (slot == 0)
1583be0e5c09SChris Mason 		return 1;
15845f39d397SChris Mason 	if (!path->nodes[1])
1585be0e5c09SChris Mason 		return 1;
15865f39d397SChris Mason 
15875f39d397SChris Mason 	left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
1588db94535dSChris Mason 			       slot - 1), root->leafsize);
1589123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
15900783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
15915f39d397SChris Mason 		free_extent_buffer(left);
1592be0e5c09SChris Mason 		return 1;
1593be0e5c09SChris Mason 	}
159402217ed2SChris Mason 
159502217ed2SChris Mason 	/* cow and double check */
15965f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
15975f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
159854aa1f4dSChris Mason 	if (ret) {
159954aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
16005f39d397SChris Mason 		free_extent_buffer(left);
160154aa1f4dSChris Mason 		return 1;
160254aa1f4dSChris Mason 	}
1603123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
16040783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
16055f39d397SChris Mason 		free_extent_buffer(left);
160602217ed2SChris Mason 		return 1;
160702217ed2SChris Mason 	}
160802217ed2SChris Mason 
16095f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
16105f39d397SChris Mason 	if (right_nritems == 0) {
16115f39d397SChris Mason 		free_extent_buffer(left);
1612a429e513SChris Mason 		return 1;
1613a429e513SChris Mason 	}
1614a429e513SChris Mason 
16155f39d397SChris Mason 	for (i = 0; i < right_nritems - 1; i++) {
16165f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1617db94535dSChris Mason 		if (!right->map_token) {
1618db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1619db94535dSChris Mason 					sizeof(struct btrfs_item),
1620db94535dSChris Mason 					&right->map_token, &right->kaddr,
1621db94535dSChris Mason 					&right->map_start, &right->map_len,
1622db94535dSChris Mason 					KM_USER1);
1623db94535dSChris Mason 		}
1624db94535dSChris Mason 
1625be0e5c09SChris Mason 		if (path->slots[0] == i)
1626be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
1627db94535dSChris Mason 
1628db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
1629db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
1630be0e5c09SChris Mason 			break;
1631db94535dSChris Mason 
1632be0e5c09SChris Mason 		push_items++;
1633db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1634be0e5c09SChris Mason 	}
1635db94535dSChris Mason 
1636db94535dSChris Mason 	if (right->map_token) {
1637db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1638db94535dSChris Mason 		right->map_token = NULL;
1639db94535dSChris Mason 	}
1640db94535dSChris Mason 
1641be0e5c09SChris Mason 	if (push_items == 0) {
16425f39d397SChris Mason 		free_extent_buffer(left);
1643be0e5c09SChris Mason 		return 1;
1644be0e5c09SChris Mason 	}
16455f39d397SChris Mason 	if (push_items == btrfs_header_nritems(right))
1646a429e513SChris Mason 		WARN_ON(1);
16475f39d397SChris Mason 
1648be0e5c09SChris Mason 	/* push data from right to left */
16495f39d397SChris Mason 	copy_extent_buffer(left, right,
16505f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
16515f39d397SChris Mason 			   btrfs_item_nr_offset(0),
16525f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
16535f39d397SChris Mason 
1654123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
16555f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
16565f39d397SChris Mason 
16575f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
1658d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
1659123abc88SChris Mason 		     btrfs_leaf_data(right) +
16605f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
1661be0e5c09SChris Mason 		     push_space);
16625f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
1663eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
1664eb60ceacSChris Mason 
1665db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
1666be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
16675f39d397SChris Mason 		u32 ioff;
1668db94535dSChris Mason 
16695f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1670db94535dSChris Mason 		if (!left->map_token) {
1671db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1672db94535dSChris Mason 					sizeof(struct btrfs_item),
1673db94535dSChris Mason 					&left->map_token, &left->kaddr,
1674db94535dSChris Mason 					&left->map_start, &left->map_len,
1675db94535dSChris Mason 					KM_USER1);
1676db94535dSChris Mason 		}
1677db94535dSChris Mason 
16785f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
16795f39d397SChris Mason 		btrfs_set_item_offset(left, item,
1680db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1681be0e5c09SChris Mason 	}
16825f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1683db94535dSChris Mason 	if (left->map_token) {
1684db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1685db94535dSChris Mason 		left->map_token = NULL;
1686db94535dSChris Mason 	}
1687be0e5c09SChris Mason 
1688be0e5c09SChris Mason 	/* fixup right node */
16895f39d397SChris Mason 	push_space = btrfs_item_offset_nr(right, push_items - 1) -
1690123abc88SChris Mason 					  leaf_data_end(root, right);
16915f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_leaf_data(right) +
1692d6025579SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - push_space,
1693d6025579SChris Mason 			      btrfs_leaf_data(right) +
1694123abc88SChris Mason 			      leaf_data_end(root, right), push_space);
16955f39d397SChris Mason 
16965f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(0),
16975f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
16985f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
16990783fcfcSChris Mason 			     sizeof(struct btrfs_item));
17005f39d397SChris Mason 
17015f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right) - push_items;
17025f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1703123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
1704eb60ceacSChris Mason 
17055f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
17065f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1707db94535dSChris Mason 
1708db94535dSChris Mason 		if (!right->map_token) {
1709db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1710db94535dSChris Mason 					sizeof(struct btrfs_item),
1711db94535dSChris Mason 					&right->map_token, &right->kaddr,
1712db94535dSChris Mason 					&right->map_start, &right->map_len,
1713db94535dSChris Mason 					KM_USER1);
1714db94535dSChris Mason 		}
1715db94535dSChris Mason 
1716db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
1717db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1718db94535dSChris Mason 	}
1719db94535dSChris Mason 	if (right->map_token) {
1720db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1721db94535dSChris Mason 		right->map_token = NULL;
1722be0e5c09SChris Mason 	}
1723eb60ceacSChris Mason 
17245f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
17255f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1726098f59c2SChris Mason 
17275f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
17285f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1729aa5d6bedSChris Mason 	if (wret)
1730aa5d6bedSChris Mason 		ret = wret;
1731be0e5c09SChris Mason 
1732be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
1733be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
1734be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
17355f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
17365f39d397SChris Mason 		path->nodes[0] = left;
1737be0e5c09SChris Mason 		path->slots[1] -= 1;
1738be0e5c09SChris Mason 	} else {
17395f39d397SChris Mason 		free_extent_buffer(left);
1740be0e5c09SChris Mason 		path->slots[0] -= push_items;
1741be0e5c09SChris Mason 	}
1742eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1743aa5d6bedSChris Mason 	return ret;
1744be0e5c09SChris Mason }
1745be0e5c09SChris Mason 
174674123bd7SChris Mason /*
174774123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
174874123bd7SChris Mason  * available for the resulting leaf level of the path.
1749aa5d6bedSChris Mason  *
1750aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
175174123bd7SChris Mason  */
1752e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1753d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
1754d4dbff95SChris Mason 		      struct btrfs_path *path, int data_size)
1755be0e5c09SChris Mason {
17565f39d397SChris Mason 	struct extent_buffer *l;
17577518a238SChris Mason 	u32 nritems;
1758eb60ceacSChris Mason 	int mid;
1759eb60ceacSChris Mason 	int slot;
17605f39d397SChris Mason 	struct extent_buffer *right;
17610783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
1762be0e5c09SChris Mason 	int data_copy_size;
1763be0e5c09SChris Mason 	int rt_data_off;
1764be0e5c09SChris Mason 	int i;
1765d4dbff95SChris Mason 	int ret = 0;
1766aa5d6bedSChris Mason 	int wret;
1767d4dbff95SChris Mason 	int double_split = 0;
1768d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
1769be0e5c09SChris Mason 
177040689478SChris Mason 	/* first try to make some room by pushing left and right */
1771e089f05cSChris Mason 	wret = push_leaf_left(trans, root, path, data_size);
1772eaee50e8SChris Mason 	if (wret < 0)
1773eaee50e8SChris Mason 		return wret;
1774eaee50e8SChris Mason 	if (wret) {
1775e089f05cSChris Mason 		wret = push_leaf_right(trans, root, path, data_size);
1776eaee50e8SChris Mason 		if (wret < 0)
1777eaee50e8SChris Mason 			return wret;
1778eaee50e8SChris Mason 	}
17795f39d397SChris Mason 	l = path->nodes[0];
1780aa5d6bedSChris Mason 
1781aa5d6bedSChris Mason 	/* did the pushes work? */
1782123abc88SChris Mason 	if (btrfs_leaf_free_space(root, l) >=
1783123abc88SChris Mason 	    sizeof(struct btrfs_item) + data_size)
1784be0e5c09SChris Mason 		return 0;
1785aa5d6bedSChris Mason 
17865c680ed6SChris Mason 	if (!path->nodes[1]) {
1787e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
17885c680ed6SChris Mason 		if (ret)
17895c680ed6SChris Mason 			return ret;
17905c680ed6SChris Mason 	}
1791eb60ceacSChris Mason 	slot = path->slots[0];
17925f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
1793eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
179454aa1f4dSChris Mason 
1795db94535dSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
1796db94535dSChris Mason 				       l->start, 0);
17975f39d397SChris Mason 	if (IS_ERR(right))
17985f39d397SChris Mason 		return PTR_ERR(right);
179954aa1f4dSChris Mason 
18005f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1801db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
18025f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
18035f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
18045f39d397SChris Mason 	btrfs_set_header_level(right, 0);
18055f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
18065f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
18075f39d397SChris Mason 			    BTRFS_FSID_SIZE);
18085f39d397SChris Mason 
1809d4dbff95SChris Mason 	if (mid <= slot) {
1810d4dbff95SChris Mason 		if (nritems == 1 ||
1811d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
1812d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1813d4dbff95SChris Mason 			if (slot >= nritems) {
1814d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18155f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1816d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1817db94535dSChris Mason 						  &disk_key, right->start,
1818d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
1819d4dbff95SChris Mason 				if (wret)
1820d4dbff95SChris Mason 					ret = wret;
18215f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18225f39d397SChris Mason 				path->nodes[0] = right;
1823d4dbff95SChris Mason 				path->slots[0] = 0;
1824d4dbff95SChris Mason 				path->slots[1] += 1;
1825d4dbff95SChris Mason 				return ret;
1826d4dbff95SChris Mason 			}
1827d4dbff95SChris Mason 			mid = slot;
1828d4dbff95SChris Mason 			double_split = 1;
1829d4dbff95SChris Mason 		}
1830d4dbff95SChris Mason 	} else {
1831d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
1832d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1833d4dbff95SChris Mason 			if (slot == 0) {
1834d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18355f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1836d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1837d4dbff95SChris Mason 						  &disk_key,
1838db94535dSChris Mason 						  right->start,
1839098f59c2SChris Mason 						  path->slots[1], 1);
1840d4dbff95SChris Mason 				if (wret)
1841d4dbff95SChris Mason 					ret = wret;
18425f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18435f39d397SChris Mason 				path->nodes[0] = right;
1844d4dbff95SChris Mason 				path->slots[0] = 0;
1845a429e513SChris Mason 				if (path->slots[1] == 0) {
1846a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
1847a429e513SChris Mason 					           path, &disk_key, 1);
1848a429e513SChris Mason 					if (wret)
1849a429e513SChris Mason 						ret = wret;
1850a429e513SChris Mason 				}
1851d4dbff95SChris Mason 				return ret;
1852d4dbff95SChris Mason 			}
1853d4dbff95SChris Mason 			mid = slot;
1854d4dbff95SChris Mason 			double_split = 1;
1855d4dbff95SChris Mason 		}
1856d4dbff95SChris Mason 	}
18575f39d397SChris Mason 	nritems = nritems - mid;
18585f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
18595f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
18605f39d397SChris Mason 
18615f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
18625f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
18635f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
18645f39d397SChris Mason 
18655f39d397SChris Mason 	copy_extent_buffer(right, l,
1866d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
1867123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
1868123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
186974123bd7SChris Mason 
18705f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
18715f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
18725f39d397SChris Mason 
18735f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
18745f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
1875db94535dSChris Mason 		u32 ioff;
1876db94535dSChris Mason 
1877db94535dSChris Mason 		if (!right->map_token) {
1878db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1879db94535dSChris Mason 					sizeof(struct btrfs_item),
1880db94535dSChris Mason 					&right->map_token, &right->kaddr,
1881db94535dSChris Mason 					&right->map_start, &right->map_len,
1882db94535dSChris Mason 					KM_USER1);
1883db94535dSChris Mason 		}
1884db94535dSChris Mason 
1885db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
18865f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
18870783fcfcSChris Mason 	}
188874123bd7SChris Mason 
1889db94535dSChris Mason 	if (right->map_token) {
1890db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1891db94535dSChris Mason 		right->map_token = NULL;
1892db94535dSChris Mason 	}
1893db94535dSChris Mason 
18945f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
1895aa5d6bedSChris Mason 	ret = 0;
18965f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
1897db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
1898db94535dSChris Mason 			  path->slots[1] + 1, 1);
1899aa5d6bedSChris Mason 	if (wret)
1900aa5d6bedSChris Mason 		ret = wret;
19015f39d397SChris Mason 
19025f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
19035f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
1904eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
19055f39d397SChris Mason 
1906be0e5c09SChris Mason 	if (mid <= slot) {
19075f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19085f39d397SChris Mason 		path->nodes[0] = right;
1909be0e5c09SChris Mason 		path->slots[0] -= mid;
1910be0e5c09SChris Mason 		path->slots[1] += 1;
1911eb60ceacSChris Mason 	} else
19125f39d397SChris Mason 		free_extent_buffer(right);
19135f39d397SChris Mason 
1914eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1915d4dbff95SChris Mason 
1916d4dbff95SChris Mason 	if (!double_split)
1917d4dbff95SChris Mason 		return ret;
191854aa1f4dSChris Mason 
1919db94535dSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
1920db94535dSChris Mason 				       l->start, 0);
19215f39d397SChris Mason 	if (IS_ERR(right))
19225f39d397SChris Mason 		return PTR_ERR(right);
19235f39d397SChris Mason 
19245f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1925db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
19265f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
19275f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
19285f39d397SChris Mason 	btrfs_set_header_level(right, 0);
19295f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
19305f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
19315f39d397SChris Mason 			    BTRFS_FSID_SIZE);
19325f39d397SChris Mason 
1933d4dbff95SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, ins_key);
19345f39d397SChris Mason 	btrfs_set_header_nritems(right, 0);
1935d4dbff95SChris Mason 	wret = insert_ptr(trans, root, path,
1936db94535dSChris Mason 			  &disk_key, right->start,
1937d4dbff95SChris Mason 			  path->slots[1], 1);
1938d4dbff95SChris Mason 	if (wret)
1939d4dbff95SChris Mason 		ret = wret;
1940a429e513SChris Mason 	if (path->slots[1] == 0) {
1941a429e513SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1942a429e513SChris Mason 		if (wret)
1943a429e513SChris Mason 			ret = wret;
1944a429e513SChris Mason 	}
19455f39d397SChris Mason 	free_extent_buffer(path->nodes[0]);
19465f39d397SChris Mason 	path->nodes[0] = right;
1947d4dbff95SChris Mason 	path->slots[0] = 0;
1948be0e5c09SChris Mason 	return ret;
1949be0e5c09SChris Mason }
1950be0e5c09SChris Mason 
1951b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
1952b18c6685SChris Mason 			struct btrfs_root *root,
1953b18c6685SChris Mason 			struct btrfs_path *path,
1954b18c6685SChris Mason 			u32 new_size)
1955b18c6685SChris Mason {
1956b18c6685SChris Mason 	int ret = 0;
1957b18c6685SChris Mason 	int slot;
1958b18c6685SChris Mason 	int slot_orig;
19595f39d397SChris Mason 	struct extent_buffer *leaf;
19605f39d397SChris Mason 	struct btrfs_item *item;
1961b18c6685SChris Mason 	u32 nritems;
1962b18c6685SChris Mason 	unsigned int data_end;
1963b18c6685SChris Mason 	unsigned int old_data_start;
1964b18c6685SChris Mason 	unsigned int old_size;
1965b18c6685SChris Mason 	unsigned int size_diff;
1966b18c6685SChris Mason 	int i;
1967b18c6685SChris Mason 
1968b18c6685SChris Mason 	slot_orig = path->slots[0];
19695f39d397SChris Mason 	leaf = path->nodes[0];
1970b18c6685SChris Mason 
19715f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
1972b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
1973b18c6685SChris Mason 
1974b18c6685SChris Mason 	slot = path->slots[0];
19755f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
19765f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
1977b18c6685SChris Mason 	BUG_ON(old_size <= new_size);
1978b18c6685SChris Mason 	size_diff = old_size - new_size;
1979b18c6685SChris Mason 
1980b18c6685SChris Mason 	BUG_ON(slot < 0);
1981b18c6685SChris Mason 	BUG_ON(slot >= nritems);
1982b18c6685SChris Mason 
1983b18c6685SChris Mason 	/*
1984b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
1985b18c6685SChris Mason 	 */
1986b18c6685SChris Mason 	/* first correct the data pointers */
1987b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
19885f39d397SChris Mason 		u32 ioff;
19895f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
1990db94535dSChris Mason 
1991db94535dSChris Mason 		if (!leaf->map_token) {
1992db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
1993db94535dSChris Mason 					sizeof(struct btrfs_item),
1994db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
1995db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
1996db94535dSChris Mason 					KM_USER1);
1997db94535dSChris Mason 		}
1998db94535dSChris Mason 
19995f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
20005f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2001b18c6685SChris Mason 	}
2002db94535dSChris Mason 
2003db94535dSChris Mason 	if (leaf->map_token) {
2004db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2005db94535dSChris Mason 		leaf->map_token = NULL;
2006db94535dSChris Mason 	}
2007db94535dSChris Mason 
2008b18c6685SChris Mason 	/* shift the data */
20095f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2010b18c6685SChris Mason 		      data_end + size_diff, btrfs_leaf_data(leaf) +
2011b18c6685SChris Mason 		      data_end, old_data_start + new_size - data_end);
20125f39d397SChris Mason 
20135f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
20145f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
20155f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2016b18c6685SChris Mason 
2017b18c6685SChris Mason 	ret = 0;
20185f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
20195f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2020b18c6685SChris Mason 		BUG();
20215f39d397SChris Mason 	}
2022b18c6685SChris Mason 	return ret;
2023b18c6685SChris Mason }
2024b18c6685SChris Mason 
20255f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
20265f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
20275f39d397SChris Mason 		      u32 data_size)
20286567e837SChris Mason {
20296567e837SChris Mason 	int ret = 0;
20306567e837SChris Mason 	int slot;
20316567e837SChris Mason 	int slot_orig;
20325f39d397SChris Mason 	struct extent_buffer *leaf;
20335f39d397SChris Mason 	struct btrfs_item *item;
20346567e837SChris Mason 	u32 nritems;
20356567e837SChris Mason 	unsigned int data_end;
20366567e837SChris Mason 	unsigned int old_data;
20376567e837SChris Mason 	unsigned int old_size;
20386567e837SChris Mason 	int i;
20396567e837SChris Mason 
20406567e837SChris Mason 	slot_orig = path->slots[0];
20415f39d397SChris Mason 	leaf = path->nodes[0];
20426567e837SChris Mason 
20435f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
20446567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
20456567e837SChris Mason 
20465f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
20475f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
20486567e837SChris Mason 		BUG();
20495f39d397SChris Mason 	}
20506567e837SChris Mason 	slot = path->slots[0];
20515f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
20526567e837SChris Mason 
20536567e837SChris Mason 	BUG_ON(slot < 0);
20546567e837SChris Mason 	BUG_ON(slot >= nritems);
20556567e837SChris Mason 
20566567e837SChris Mason 	/*
20576567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
20586567e837SChris Mason 	 */
20596567e837SChris Mason 	/* first correct the data pointers */
20606567e837SChris Mason 	for (i = slot; i < nritems; i++) {
20615f39d397SChris Mason 		u32 ioff;
20625f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2063db94535dSChris Mason 
2064db94535dSChris Mason 		if (!leaf->map_token) {
2065db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2066db94535dSChris Mason 					sizeof(struct btrfs_item),
2067db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2068db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2069db94535dSChris Mason 					KM_USER1);
2070db94535dSChris Mason 		}
20715f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
20725f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
20736567e837SChris Mason 	}
20745f39d397SChris Mason 
2075db94535dSChris Mason 	if (leaf->map_token) {
2076db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2077db94535dSChris Mason 		leaf->map_token = NULL;
2078db94535dSChris Mason 	}
2079db94535dSChris Mason 
20806567e837SChris Mason 	/* shift the data */
20815f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
20826567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
20836567e837SChris Mason 		      data_end, old_data - data_end);
20845f39d397SChris Mason 
20856567e837SChris Mason 	data_end = old_data;
20865f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
20875f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
20885f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
20895f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
20906567e837SChris Mason 
20916567e837SChris Mason 	ret = 0;
20925f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
20935f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
20946567e837SChris Mason 		BUG();
20955f39d397SChris Mason 	}
20966567e837SChris Mason 	return ret;
20976567e837SChris Mason }
20986567e837SChris Mason 
209974123bd7SChris Mason /*
210074123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
210174123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
210274123bd7SChris Mason  */
21035f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
21045f39d397SChris Mason 			    struct btrfs_root *root,
21055f39d397SChris Mason 			    struct btrfs_path *path,
21065f39d397SChris Mason 			    struct btrfs_key *cpu_key, u32 data_size)
2107be0e5c09SChris Mason {
21085f39d397SChris Mason 	struct extent_buffer *leaf;
21095f39d397SChris Mason 	struct btrfs_item *item;
2110aa5d6bedSChris Mason 	int ret = 0;
2111be0e5c09SChris Mason 	int slot;
2112eb60ceacSChris Mason 	int slot_orig;
21137518a238SChris Mason 	u32 nritems;
2114be0e5c09SChris Mason 	unsigned int data_end;
2115e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2116e2fa7227SChris Mason 
2117e2fa7227SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2118be0e5c09SChris Mason 
211974123bd7SChris Mason 	/* create a root if there isn't one */
21205c680ed6SChris Mason 	if (!root->node)
2121cfaa7295SChris Mason 		BUG();
21225f39d397SChris Mason 
2123e089f05cSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
2124eb60ceacSChris Mason 	if (ret == 0) {
2125f0930a37SChris Mason 		return -EEXIST;
2126aa5d6bedSChris Mason 	}
2127ed2ff2cbSChris Mason 	if (ret < 0)
2128ed2ff2cbSChris Mason 		goto out;
2129be0e5c09SChris Mason 
213062e2749eSChris Mason 	slot_orig = path->slots[0];
21315f39d397SChris Mason 	leaf = path->nodes[0];
213274123bd7SChris Mason 
21335f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2134123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2135eb60ceacSChris Mason 
2136123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
2137d4dbff95SChris Mason 	    sizeof(struct btrfs_item) + data_size) {
2138be0e5c09SChris Mason 		BUG();
2139d4dbff95SChris Mason 	}
21405f39d397SChris Mason 
214162e2749eSChris Mason 	slot = path->slots[0];
2142eb60ceacSChris Mason 	BUG_ON(slot < 0);
21435f39d397SChris Mason 
2144be0e5c09SChris Mason 	if (slot != nritems) {
2145be0e5c09SChris Mason 		int i;
21465f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2147be0e5c09SChris Mason 
21485f39d397SChris Mason 		if (old_data < data_end) {
21495f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
21505f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
21515f39d397SChris Mason 			       slot, old_data, data_end);
21525f39d397SChris Mason 			BUG_ON(1);
21535f39d397SChris Mason 		}
2154be0e5c09SChris Mason 		/*
2155be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2156be0e5c09SChris Mason 		 */
2157be0e5c09SChris Mason 		/* first correct the data pointers */
2158db94535dSChris Mason 		WARN_ON(leaf->map_token);
21590783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
21605f39d397SChris Mason 			u32 ioff;
2161db94535dSChris Mason 
21625f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2163db94535dSChris Mason 			if (!leaf->map_token) {
2164db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2165db94535dSChris Mason 					sizeof(struct btrfs_item),
2166db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2167db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2168db94535dSChris Mason 					KM_USER1);
2169db94535dSChris Mason 			}
2170db94535dSChris Mason 
21715f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
21725f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff - data_size);
21730783fcfcSChris Mason 		}
2174db94535dSChris Mason 		if (leaf->map_token) {
2175db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2176db94535dSChris Mason 			leaf->map_token = NULL;
2177db94535dSChris Mason 		}
2178be0e5c09SChris Mason 
2179be0e5c09SChris Mason 		/* shift the items */
21805f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
21815f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
21820783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2183be0e5c09SChris Mason 
2184be0e5c09SChris Mason 		/* shift the data */
21855f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2186d6025579SChris Mason 			      data_end - data_size, btrfs_leaf_data(leaf) +
2187be0e5c09SChris Mason 			      data_end, old_data - data_end);
2188be0e5c09SChris Mason 		data_end = old_data;
2189be0e5c09SChris Mason 	}
21905f39d397SChris Mason 
219162e2749eSChris Mason 	/* setup the item for the new data */
21925f39d397SChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
21935f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
21945f39d397SChris Mason 	btrfs_set_item_offset(leaf, item, data_end - data_size);
21955f39d397SChris Mason 	btrfs_set_item_size(leaf, item, data_size);
21965f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
21975f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2198aa5d6bedSChris Mason 
2199aa5d6bedSChris Mason 	ret = 0;
22008e19f2cdSChris Mason 	if (slot == 0)
2201e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2202aa5d6bedSChris Mason 
22035f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
22045f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2205be0e5c09SChris Mason 		BUG();
22065f39d397SChris Mason 	}
2207ed2ff2cbSChris Mason out:
220862e2749eSChris Mason 	return ret;
220962e2749eSChris Mason }
221062e2749eSChris Mason 
221162e2749eSChris Mason /*
221262e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
221362e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
221462e2749eSChris Mason  */
2215e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2216e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2217e089f05cSChris Mason 		      data_size)
221862e2749eSChris Mason {
221962e2749eSChris Mason 	int ret = 0;
22202c90e5d6SChris Mason 	struct btrfs_path *path;
22215f39d397SChris Mason 	struct extent_buffer *leaf;
22225f39d397SChris Mason 	unsigned long ptr;
222362e2749eSChris Mason 
22242c90e5d6SChris Mason 	path = btrfs_alloc_path();
22252c90e5d6SChris Mason 	BUG_ON(!path);
22262c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
222762e2749eSChris Mason 	if (!ret) {
22285f39d397SChris Mason 		leaf = path->nodes[0];
22295f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
22305f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
22315f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
223262e2749eSChris Mason 	}
22332c90e5d6SChris Mason 	btrfs_free_path(path);
2234aa5d6bedSChris Mason 	return ret;
2235be0e5c09SChris Mason }
2236be0e5c09SChris Mason 
223774123bd7SChris Mason /*
22385de08d7dSChris Mason  * delete the pointer from a given node.
223974123bd7SChris Mason  *
224074123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
224174123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
224274123bd7SChris Mason  * a leaf if all the nodes are emptied.
224374123bd7SChris Mason  */
2244e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2245e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2246be0e5c09SChris Mason {
22475f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
22487518a238SChris Mason 	u32 nritems;
2249aa5d6bedSChris Mason 	int ret = 0;
2250bb803951SChris Mason 	int wret;
2251be0e5c09SChris Mason 
22525f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2253be0e5c09SChris Mason 	if (slot != nritems -1) {
22545f39d397SChris Mason 		memmove_extent_buffer(parent,
22555f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
22565f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2257d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2258d6025579SChris Mason 			      (nritems - slot - 1));
2259be0e5c09SChris Mason 	}
22607518a238SChris Mason 	nritems--;
22615f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
22627518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
22635f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2264eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
22655f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2266bb803951SChris Mason 	} else if (slot == 0) {
22675f39d397SChris Mason 		struct btrfs_disk_key disk_key;
22685f39d397SChris Mason 
22695f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
22705f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
22710f70abe2SChris Mason 		if (wret)
22720f70abe2SChris Mason 			ret = wret;
2273be0e5c09SChris Mason 	}
2274d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2275aa5d6bedSChris Mason 	return ret;
2276be0e5c09SChris Mason }
2277be0e5c09SChris Mason 
227874123bd7SChris Mason /*
227974123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
228074123bd7SChris Mason  * the leaf, remove it from the tree
228174123bd7SChris Mason  */
2282e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2283e089f05cSChris Mason 		   struct btrfs_path *path)
2284be0e5c09SChris Mason {
2285be0e5c09SChris Mason 	int slot;
22865f39d397SChris Mason 	struct extent_buffer *leaf;
22875f39d397SChris Mason 	struct btrfs_item *item;
2288be0e5c09SChris Mason 	int doff;
2289be0e5c09SChris Mason 	int dsize;
2290aa5d6bedSChris Mason 	int ret = 0;
2291aa5d6bedSChris Mason 	int wret;
22927518a238SChris Mason 	u32 nritems;
2293be0e5c09SChris Mason 
22945f39d397SChris Mason 	leaf = path->nodes[0];
22954920c9acSChris Mason 	slot = path->slots[0];
22965f39d397SChris Mason 	doff = btrfs_item_offset_nr(leaf, slot);
22975f39d397SChris Mason 	dsize = btrfs_item_size_nr(leaf, slot);
22985f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2299be0e5c09SChris Mason 
23007518a238SChris Mason 	if (slot != nritems - 1) {
2301be0e5c09SChris Mason 		int i;
2302123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
23035f39d397SChris Mason 
23045f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2305d6025579SChris Mason 			      data_end + dsize,
2306123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
2307be0e5c09SChris Mason 			      doff - data_end);
23085f39d397SChris Mason 
23090783fcfcSChris Mason 		for (i = slot + 1; i < nritems; i++) {
23105f39d397SChris Mason 			u32 ioff;
2311db94535dSChris Mason 
23125f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2313db94535dSChris Mason 			if (!leaf->map_token) {
2314db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2315db94535dSChris Mason 					sizeof(struct btrfs_item),
2316db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2317db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2318db94535dSChris Mason 					KM_USER1);
2319db94535dSChris Mason 			}
23205f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
23215f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
23220783fcfcSChris Mason 		}
2323db94535dSChris Mason 
2324db94535dSChris Mason 		if (leaf->map_token) {
2325db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2326db94535dSChris Mason 			leaf->map_token = NULL;
2327db94535dSChris Mason 		}
2328db94535dSChris Mason 
23295f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
23305f39d397SChris Mason 			      btrfs_item_nr_offset(slot + 1),
23310783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
23327518a238SChris Mason 			      (nritems - slot - 1));
2333be0e5c09SChris Mason 	}
23345f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems - 1);
23357518a238SChris Mason 	nritems--;
23365f39d397SChris Mason 
233774123bd7SChris Mason 	/* delete the leaf if we've emptied it */
23387518a238SChris Mason 	if (nritems == 0) {
23395f39d397SChris Mason 		if (leaf == root->node) {
23405f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
23419a8dd150SChris Mason 		} else {
23425f39d397SChris Mason 			clean_tree_block(trans, root, leaf);
23435f39d397SChris Mason 			wait_on_tree_block_writeback(root, leaf);
2344e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2345aa5d6bedSChris Mason 			if (wret)
2346aa5d6bedSChris Mason 				ret = wret;
2347e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
2348db94535dSChris Mason 						 leaf->start, leaf->len, 1);
23490f70abe2SChris Mason 			if (wret)
23500f70abe2SChris Mason 				ret = wret;
23519a8dd150SChris Mason 		}
2352be0e5c09SChris Mason 	} else {
23537518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2354aa5d6bedSChris Mason 		if (slot == 0) {
23555f39d397SChris Mason 			struct btrfs_disk_key disk_key;
23565f39d397SChris Mason 
23575f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2358e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
23595f39d397SChris Mason 					      &disk_key, 1);
2360aa5d6bedSChris Mason 			if (wret)
2361aa5d6bedSChris Mason 				ret = wret;
2362aa5d6bedSChris Mason 		}
2363aa5d6bedSChris Mason 
236474123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
2365123abc88SChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
2366be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2367be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2368be0e5c09SChris Mason 			 * for possible call to del_ptr below
2369be0e5c09SChris Mason 			 */
23704920c9acSChris Mason 			slot = path->slots[1];
23715f39d397SChris Mason 			extent_buffer_get(leaf);
23725f39d397SChris Mason 
2373e089f05cSChris Mason 			wret = push_leaf_left(trans, root, path, 1);
237454aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2375aa5d6bedSChris Mason 				ret = wret;
23765f39d397SChris Mason 
23775f39d397SChris Mason 			if (path->nodes[0] == leaf &&
23785f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
2379e089f05cSChris Mason 				wret = push_leaf_right(trans, root, path, 1);
238054aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2381aa5d6bedSChris Mason 					ret = wret;
2382aa5d6bedSChris Mason 			}
23835f39d397SChris Mason 
23845f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
2385db94535dSChris Mason 				u64 bytenr = leaf->start;
2386db94535dSChris Mason 				u32 blocksize = leaf->len;
23875f39d397SChris Mason 
23885f39d397SChris Mason 				clean_tree_block(trans, root, leaf);
23895f39d397SChris Mason 				wait_on_tree_block_writeback(root, leaf);
23905f39d397SChris Mason 
2391e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2392aa5d6bedSChris Mason 				if (wret)
2393aa5d6bedSChris Mason 					ret = wret;
23945f39d397SChris Mason 
23955f39d397SChris Mason 				free_extent_buffer(leaf);
2396db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
2397db94535dSChris Mason 							 blocksize, 1);
23980f70abe2SChris Mason 				if (wret)
23990f70abe2SChris Mason 					ret = wret;
24005de08d7dSChris Mason 			} else {
24015f39d397SChris Mason 				btrfs_mark_buffer_dirty(leaf);
24025f39d397SChris Mason 				free_extent_buffer(leaf);
2403be0e5c09SChris Mason 			}
2404d5719762SChris Mason 		} else {
24055f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2406be0e5c09SChris Mason 		}
24079a8dd150SChris Mason 	}
2408aa5d6bedSChris Mason 	return ret;
24099a8dd150SChris Mason }
24109a8dd150SChris Mason 
241197571fd0SChris Mason /*
241297571fd0SChris Mason  * walk up the tree as far as required to find the next leaf.
24130f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
24140f70abe2SChris Mason  * returns < 0 on io errors.
241597571fd0SChris Mason  */
2416234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2417d97e63b6SChris Mason {
2418d97e63b6SChris Mason 	int slot;
2419d97e63b6SChris Mason 	int level = 1;
2420db94535dSChris Mason 	u64 bytenr;
24215f39d397SChris Mason 	struct extent_buffer *c;
24225f39d397SChris Mason 	struct extent_buffer *next = NULL;
2423d97e63b6SChris Mason 
2424234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
2425d97e63b6SChris Mason 		if (!path->nodes[level])
24260f70abe2SChris Mason 			return 1;
24275f39d397SChris Mason 
2428d97e63b6SChris Mason 		slot = path->slots[level] + 1;
2429d97e63b6SChris Mason 		c = path->nodes[level];
24305f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
2431d97e63b6SChris Mason 			level++;
2432d97e63b6SChris Mason 			continue;
2433d97e63b6SChris Mason 		}
24345f39d397SChris Mason 
2435db94535dSChris Mason 		bytenr = btrfs_node_blockptr(c, slot);
2436cfaa7295SChris Mason 		if (next)
24375f39d397SChris Mason 			free_extent_buffer(next);
24385f39d397SChris Mason 
24396702ed49SChris Mason 		if (path->reada)
24406702ed49SChris Mason 			reada_for_search(root, path, level, slot);
24415f39d397SChris Mason 
2442db94535dSChris Mason 		next = read_tree_block(root, bytenr,
2443db94535dSChris Mason 				       btrfs_level_size(root, level -1));
2444d97e63b6SChris Mason 		break;
2445d97e63b6SChris Mason 	}
2446d97e63b6SChris Mason 	path->slots[level] = slot;
2447d97e63b6SChris Mason 	while(1) {
2448d97e63b6SChris Mason 		level--;
2449d97e63b6SChris Mason 		c = path->nodes[level];
24505f39d397SChris Mason 		free_extent_buffer(c);
2451d97e63b6SChris Mason 		path->nodes[level] = next;
2452d97e63b6SChris Mason 		path->slots[level] = 0;
2453d97e63b6SChris Mason 		if (!level)
2454d97e63b6SChris Mason 			break;
24556702ed49SChris Mason 		if (path->reada)
245632020611SYan 			reada_for_search(root, path, level, 0);
2457db94535dSChris Mason 		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2458db94535dSChris Mason 				       btrfs_level_size(root, level - 1));
2459d97e63b6SChris Mason 	}
2460d97e63b6SChris Mason 	return 0;
2461d97e63b6SChris Mason }
2462