xref: /openbmc/linux/fs/btrfs/ctree.c (revision 3685f791659c9f21b763ee1702ac8ca58bc20f81)
16cbd5570SChris Mason /*
26cbd5570SChris Mason  * Copyright (C) 2007 Oracle.  All rights reserved.
36cbd5570SChris Mason  *
46cbd5570SChris Mason  * This program is free software; you can redistribute it and/or
56cbd5570SChris Mason  * modify it under the terms of the GNU General Public
66cbd5570SChris Mason  * License v2 as published by the Free Software Foundation.
76cbd5570SChris Mason  *
86cbd5570SChris Mason  * This program is distributed in the hope that it will be useful,
96cbd5570SChris Mason  * but WITHOUT ANY WARRANTY; without even the implied warranty of
106cbd5570SChris Mason  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
116cbd5570SChris Mason  * General Public License for more details.
126cbd5570SChris Mason  *
136cbd5570SChris Mason  * You should have received a copy of the GNU General Public
146cbd5570SChris Mason  * License along with this program; if not, write to the
156cbd5570SChris Mason  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
166cbd5570SChris Mason  * Boston, MA 021110-1307, USA.
176cbd5570SChris Mason  */
186cbd5570SChris Mason 
19a6b6e75eSChris Mason #include <linux/sched.h>
20eb60ceacSChris Mason #include "ctree.h"
21eb60ceacSChris Mason #include "disk-io.h"
227f5c1516SChris Mason #include "transaction.h"
235f39d397SChris Mason #include "print-tree.h"
249a8dd150SChris Mason 
25e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
26e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
27e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
28d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
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 
915f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
92db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
935f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
945f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
956702ed49SChris Mason 
965f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
975f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
986702ed49SChris Mason 		different_trans = 1;
996702ed49SChris Mason 		ret = btrfs_inc_ref(trans, root, buf);
1006702ed49SChris Mason 		if (ret)
1016702ed49SChris Mason 			return ret;
1026702ed49SChris Mason 	} else {
1036702ed49SChris Mason 		clean_tree_block(trans, root, buf);
1046702ed49SChris Mason 	}
1056702ed49SChris Mason 
1066702ed49SChris Mason 	if (buf == root->node) {
1076702ed49SChris Mason 		root->node = cow;
1085f39d397SChris Mason 		extent_buffer_get(cow);
1096702ed49SChris Mason 		if (buf != root->commit_root) {
110db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
111db94535dSChris Mason 					  buf->len, 1);
1126702ed49SChris Mason 		}
1135f39d397SChris Mason 		free_extent_buffer(buf);
1146702ed49SChris Mason 	} else {
1155f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
116db94535dSChris Mason 					cow->start);
1176702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
1185f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
119db94535dSChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len, 1);
1206702ed49SChris Mason 	}
1215f39d397SChris Mason 	free_extent_buffer(buf);
1226702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
1236702ed49SChris Mason 	*cow_ret = cow;
1246702ed49SChris Mason 	return 0;
1256702ed49SChris Mason }
1266702ed49SChris Mason 
1275f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans,
1285f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
1295f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
1305f39d397SChris Mason 		    struct extent_buffer **cow_ret)
13102217ed2SChris Mason {
1326702ed49SChris Mason 	u64 search_start;
133f510cfecSChris Mason 	int ret;
134ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
135ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
136ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
137ccd467d6SChris Mason 		WARN_ON(1);
138ccd467d6SChris Mason 	}
139ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
140ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
141ccd467d6SChris Mason 		       root->fs_info->generation);
142ccd467d6SChris Mason 		WARN_ON(1);
143ccd467d6SChris Mason 	}
1445f39d397SChris Mason 	if (btrfs_header_generation(buf) == trans->transid) {
14502217ed2SChris Mason 		*cow_ret = buf;
14602217ed2SChris Mason 		return 0;
14702217ed2SChris Mason 	}
1486702ed49SChris Mason 
149db94535dSChris Mason 	search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
150f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
1516702ed49SChris Mason 				 parent_slot, cow_ret, search_start, 0);
152f510cfecSChris Mason 	return ret;
1532c90e5d6SChris Mason }
1546702ed49SChris Mason 
1556b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
1566702ed49SChris Mason {
1576b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
1586702ed49SChris Mason 		return 1;
1596b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
1606702ed49SChris Mason 		return 1;
16102217ed2SChris Mason 	return 0;
16202217ed2SChris Mason }
16302217ed2SChris Mason 
1646b80053dSChris Mason static int should_defrag_leaf(struct extent_buffer *leaf)
1652cc58cf2SChris Mason {
1666b80053dSChris Mason 	struct btrfs_key key;
1672cc58cf2SChris Mason 	u32 nritems;
1682cc58cf2SChris Mason 
1696b80053dSChris Mason 	if (btrfs_buffer_defrag(leaf))
1702cc58cf2SChris Mason 		return 1;
1712cc58cf2SChris Mason 
1726b80053dSChris Mason 	nritems = btrfs_header_nritems(leaf);
1732cc58cf2SChris Mason 	if (nritems == 0)
1742cc58cf2SChris Mason 		return 0;
1752cc58cf2SChris Mason 
1766b80053dSChris Mason 	btrfs_item_key_to_cpu(leaf, &key, 0);
1776b80053dSChris Mason 	if (key.type == BTRFS_DIR_ITEM_KEY)
1782cc58cf2SChris Mason 		return 1;
1792cc58cf2SChris Mason 
1806b80053dSChris Mason 
1816b80053dSChris Mason 	btrfs_item_key_to_cpu(leaf, &key, nritems - 1);
1826b80053dSChris Mason 	if (key.type == BTRFS_DIR_ITEM_KEY)
1832cc58cf2SChris Mason 		return 1;
1842cc58cf2SChris Mason 	if (nritems > 4) {
1856b80053dSChris Mason 		btrfs_item_key_to_cpu(leaf, &key, nritems / 2);
1866b80053dSChris Mason 		if (key.type == BTRFS_DIR_ITEM_KEY)
1872cc58cf2SChris Mason 			return 1;
1882cc58cf2SChris Mason 	}
1892cc58cf2SChris Mason 	return 0;
1902cc58cf2SChris Mason }
1912cc58cf2SChris Mason 
1926702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
1935f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
194a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
195a6b6e75eSChris Mason 		       struct btrfs_key *progress)
1966702ed49SChris Mason {
1976b80053dSChris Mason 	struct extent_buffer *cur;
1986b80053dSChris Mason 	struct extent_buffer *tmp;
1996702ed49SChris Mason 	u64 blocknr;
200e9d0b13bSChris Mason 	u64 search_start = *last_ret;
201e9d0b13bSChris Mason 	u64 last_block = 0;
2026702ed49SChris Mason 	u64 other;
2036702ed49SChris Mason 	u32 parent_nritems;
2046702ed49SChris Mason 	int end_slot;
2056702ed49SChris Mason 	int i;
2066702ed49SChris Mason 	int err = 0;
207f2183bdeSChris Mason 	int parent_level;
2086b80053dSChris Mason 	int uptodate;
2096b80053dSChris Mason 	u32 blocksize;
2106702ed49SChris Mason 
2116702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
2126702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2136702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
2146702ed49SChris Mason 		WARN_ON(1);
2156702ed49SChris Mason 	}
2166702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
2176702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2186702ed49SChris Mason 		       root->fs_info->generation);
2196702ed49SChris Mason 		WARN_ON(1);
2206702ed49SChris Mason 	}
2210f1ebbd1SChris Mason 	parent_level = btrfs_header_level(parent);
22286479a04SChris Mason 
2236b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
2246b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
2256702ed49SChris Mason 	end_slot = parent_nritems;
2266702ed49SChris Mason 
2276702ed49SChris Mason 	if (parent_nritems == 1)
2286702ed49SChris Mason 		return 0;
2296702ed49SChris Mason 
230a6b6e75eSChris Mason 	if (root != root->fs_info->extent_root) {
231a6b6e75eSChris Mason 		struct btrfs_key first_key;
232a6b6e75eSChris Mason 		struct btrfs_key last_key;
233a6b6e75eSChris Mason 
234a6b6e75eSChris Mason 		btrfs_node_key_to_cpu(parent, &first_key, 0);
235a6b6e75eSChris Mason 		btrfs_node_key_to_cpu(parent, &last_key, parent_nritems - 1);
236a6b6e75eSChris Mason 		if (first_key.objectid != last_key.objectid)
237a6b6e75eSChris Mason 			return 0;
238a6b6e75eSChris Mason 	}
239a6b6e75eSChris Mason 
2406702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
2416702ed49SChris Mason 		int close = 1;
242a6b6e75eSChris Mason 
2436b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
244e9d0b13bSChris Mason 		if (last_block == 0)
245e9d0b13bSChris Mason 			last_block = blocknr;
2466702ed49SChris Mason 		if (i > 0) {
2476b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
2486b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2496702ed49SChris Mason 		}
2506702ed49SChris Mason 		if (close && i < end_slot - 1) {
2516b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
2526b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2536702ed49SChris Mason 		}
254e9d0b13bSChris Mason 		if (close) {
255e9d0b13bSChris Mason 			last_block = blocknr;
2566702ed49SChris Mason 			continue;
257e9d0b13bSChris Mason 		}
2586702ed49SChris Mason 
2596b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
2606b80053dSChris Mason 		if (cur)
2616b80053dSChris Mason 			uptodate = btrfs_buffer_uptodate(cur);
2626b80053dSChris Mason 		else
2636b80053dSChris Mason 			uptodate = 0;
2646b80053dSChris Mason 		if (!cur || !uptodate ||
2656b80053dSChris Mason 		    (parent_level != 1 && !btrfs_buffer_defrag(cur)) ||
2666b80053dSChris Mason 		    (parent_level == 1 && !should_defrag_leaf(cur))) {
2676702ed49SChris Mason 			if (cache_only) {
2686b80053dSChris Mason 				free_extent_buffer(cur);
2696702ed49SChris Mason 				continue;
2706702ed49SChris Mason 			}
2716b80053dSChris Mason 			if (!cur) {
2726b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
2736b80053dSChris Mason 							 blocksize);
2746b80053dSChris Mason 			} else if (!uptodate) {
2756b80053dSChris Mason 				btrfs_read_buffer(cur);
2766702ed49SChris Mason 			}
277f2183bdeSChris Mason 		}
278e9d0b13bSChris Mason 		if (search_start == 0)
2796b80053dSChris Mason 			search_start = last_block;
280e9d0b13bSChris Mason 
2816b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
2826b80053dSChris Mason 					&tmp, search_start,
2836b80053dSChris Mason 					min(16 * blocksize,
2846b80053dSChris Mason 					    (end_slot - i) * blocksize));
285252c38f0SYan 		if (err) {
2866b80053dSChris Mason 			free_extent_buffer(cur);
2876702ed49SChris Mason 			break;
288252c38f0SYan 		}
2896b80053dSChris Mason 		search_start = tmp->start;
290f2183bdeSChris Mason 		*last_ret = search_start;
291f2183bdeSChris Mason 		if (parent_level == 1)
2926b80053dSChris Mason 			btrfs_clear_buffer_defrag(tmp);
2936b80053dSChris Mason 		free_extent_buffer(tmp);
2946702ed49SChris Mason 	}
2956702ed49SChris Mason 	return err;
2966702ed49SChris Mason }
2976702ed49SChris Mason 
29874123bd7SChris Mason /*
29974123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
30074123bd7SChris Mason  * this returns the address of the start of the last item,
30174123bd7SChris Mason  * which is the stop of the leaf data stack
30274123bd7SChris Mason  */
303123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
3045f39d397SChris Mason 					 struct extent_buffer *leaf)
305be0e5c09SChris Mason {
3065f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
307be0e5c09SChris Mason 	if (nr == 0)
308123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
3095f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
310be0e5c09SChris Mason }
311be0e5c09SChris Mason 
31274123bd7SChris Mason /*
31374123bd7SChris Mason  * compare two keys in a memcmp fashion
31474123bd7SChris Mason  */
3159aca1d51SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
316be0e5c09SChris Mason {
317e2fa7227SChris Mason 	struct btrfs_key k1;
318e2fa7227SChris Mason 
319e2fa7227SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
320e2fa7227SChris Mason 
321e2fa7227SChris Mason 	if (k1.objectid > k2->objectid)
322be0e5c09SChris Mason 		return 1;
323e2fa7227SChris Mason 	if (k1.objectid < k2->objectid)
324be0e5c09SChris Mason 		return -1;
3255f39d397SChris Mason 	if (k1.type > k2->type)
326f254e52cSChris Mason 		return 1;
3275f39d397SChris Mason 	if (k1.type < k2->type)
328f254e52cSChris Mason 		return -1;
32970b2befdSChris Mason 	if (k1.offset > k2->offset)
33070b2befdSChris Mason 		return 1;
33170b2befdSChris Mason 	if (k1.offset < k2->offset)
33270b2befdSChris Mason 		return -1;
333be0e5c09SChris Mason 	return 0;
334be0e5c09SChris Mason }
33574123bd7SChris Mason 
336123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
337123abc88SChris Mason 		      int level)
338aa5d6bedSChris Mason {
3395f39d397SChris Mason 	struct extent_buffer *parent = NULL;
3405f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
3415f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3425f39d397SChris Mason 	struct btrfs_disk_key node_key;
343aa5d6bedSChris Mason 	int parent_slot;
3448d7be552SChris Mason 	int slot;
3458d7be552SChris Mason 	struct btrfs_key cpukey;
3465f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
347aa5d6bedSChris Mason 
348aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3495f39d397SChris Mason 		parent = path->nodes[level + 1];
350a1f39630SAneesh 
3518d7be552SChris Mason 	slot = path->slots[level];
3527518a238SChris Mason 	BUG_ON(nritems == 0);
3537518a238SChris Mason 	if (parent) {
354a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3555f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3565f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
3575f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
358e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
3591d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
360db94535dSChris Mason 		       btrfs_header_bytenr(node));
361aa5d6bedSChris Mason 	}
362123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
3638d7be552SChris Mason 	if (slot != 0) {
3645f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
3655f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3665f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
3678d7be552SChris Mason 	}
3688d7be552SChris Mason 	if (slot < nritems - 1) {
3695f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
3705f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3715f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
372aa5d6bedSChris Mason 	}
373aa5d6bedSChris Mason 	return 0;
374aa5d6bedSChris Mason }
375aa5d6bedSChris Mason 
376123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
377123abc88SChris Mason 		      int level)
378aa5d6bedSChris Mason {
3795f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
3805f39d397SChris Mason 	struct extent_buffer *parent = NULL;
381aa5d6bedSChris Mason 	int parent_slot;
3828d7be552SChris Mason 	struct btrfs_key cpukey;
3835f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3845f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
3855f39d397SChris Mason 	int slot = path->slots[0];
3868d7be552SChris Mason 
3875f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
388aa5d6bedSChris Mason 
389aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3905f39d397SChris Mason 		parent = path->nodes[level + 1];
3917518a238SChris Mason 
3927518a238SChris Mason 	if (nritems == 0)
3937518a238SChris Mason 		return 0;
3947518a238SChris Mason 
3957518a238SChris Mason 	if (parent) {
396a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3975f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3985f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
3996702ed49SChris Mason 
4005f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
401e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
4021d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
403db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
404aa5d6bedSChris Mason 	}
4055f39d397SChris Mason #if 0
4065f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
4075f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
4085f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
4095f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
4105f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4115f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
4125f39d397SChris Mason 			BUG_ON(1);
4135f39d397SChris Mason 		}
4145f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
4155f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
4165f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4175f39d397SChris Mason 			printk("slot %d offset bad\n", i);
4185f39d397SChris Mason 			BUG_ON(1);
4195f39d397SChris Mason 		}
4205f39d397SChris Mason 		if (i == 0) {
4215f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
4225f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
4235f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
4245f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4255f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
4265f39d397SChris Mason 				BUG_ON(1);
4275f39d397SChris Mason 			}
4285f39d397SChris Mason 		}
4295f39d397SChris Mason 	}
4305f39d397SChris Mason 	if (nritems > 0) {
4315f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
4325f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4335f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
4345f39d397SChris Mason 				BUG_ON(1);
4355f39d397SChris Mason 		}
4365f39d397SChris Mason 	}
4375f39d397SChris Mason #endif
4385f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
4395f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4405f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
4415f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
4425f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4435f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
4445f39d397SChris Mason 			BUG_ON(1);
4455f39d397SChris Mason 		}
4465f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
4475f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
4485f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4495f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4505f39d397SChris Mason 			BUG_ON(1);
4515f39d397SChris Mason 		}
452aa5d6bedSChris Mason 	}
4538d7be552SChris Mason 	if (slot < nritems - 1) {
4545f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4555f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
4565f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
4575f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
4585f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
4595f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4605f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4615f39d397SChris Mason 			BUG_ON(1);
462aa5d6bedSChris Mason 		}
4635f39d397SChris Mason 	}
4645f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
4655f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
466aa5d6bedSChris Mason 	return 0;
467aa5d6bedSChris Mason }
468aa5d6bedSChris Mason 
469123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path,
470123abc88SChris Mason 			int level)
471aa5d6bedSChris Mason {
472810191ffSChris Mason 	return 0;
473db94535dSChris Mason #if 0
4745f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
4755f39d397SChris Mason 
476479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
477479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
478479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
4795f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
480db94535dSChris Mason 		return 1;
4815f39d397SChris Mason 	}
482db94535dSChris Mason #endif
483aa5d6bedSChris Mason 	if (level == 0)
484123abc88SChris Mason 		return check_leaf(root, path, level);
485123abc88SChris Mason 	return check_node(root, path, level);
486aa5d6bedSChris Mason }
487aa5d6bedSChris Mason 
48874123bd7SChris Mason /*
4895f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
4905f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
4915f39d397SChris Mason  *
49274123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
49374123bd7SChris Mason  * the place where you would insert key if it is not found in
49474123bd7SChris Mason  * the array.
49574123bd7SChris Mason  *
49674123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
49774123bd7SChris Mason  */
4985f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
4995f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
500be0e5c09SChris Mason 			      int max, int *slot)
501be0e5c09SChris Mason {
502be0e5c09SChris Mason 	int low = 0;
503be0e5c09SChris Mason 	int high = max;
504be0e5c09SChris Mason 	int mid;
505be0e5c09SChris Mason 	int ret;
506479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
5075f39d397SChris Mason 	struct btrfs_disk_key unaligned;
5085f39d397SChris Mason 	unsigned long offset;
5095f39d397SChris Mason 	char *map_token = NULL;
5105f39d397SChris Mason 	char *kaddr = NULL;
5115f39d397SChris Mason 	unsigned long map_start = 0;
5125f39d397SChris Mason 	unsigned long map_len = 0;
513479965d6SChris Mason 	int err;
514be0e5c09SChris Mason 
515be0e5c09SChris Mason 	while(low < high) {
516be0e5c09SChris Mason 		mid = (low + high) / 2;
5175f39d397SChris Mason 		offset = p + mid * item_size;
5185f39d397SChris Mason 
5195f39d397SChris Mason 		if (!map_token || offset < map_start ||
5205f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
5215f39d397SChris Mason 		    map_start + map_len) {
522479965d6SChris Mason 			if (map_token) {
5235f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
524479965d6SChris Mason 				map_token = NULL;
525479965d6SChris Mason 			}
526479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
527479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
528479965d6SChris Mason 						&map_token, &kaddr,
5295f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
5305f39d397SChris Mason 
531479965d6SChris Mason 			if (!err) {
532479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
533479965d6SChris Mason 							map_start);
534479965d6SChris Mason 			} else {
5355f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
5365f39d397SChris Mason 						   offset, sizeof(unaligned));
5375f39d397SChris Mason 				tmp = &unaligned;
538479965d6SChris Mason 			}
539479965d6SChris Mason 
5405f39d397SChris Mason 		} else {
5415f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
5425f39d397SChris Mason 							map_start);
5435f39d397SChris Mason 		}
544be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
545be0e5c09SChris Mason 
546be0e5c09SChris Mason 		if (ret < 0)
547be0e5c09SChris Mason 			low = mid + 1;
548be0e5c09SChris Mason 		else if (ret > 0)
549be0e5c09SChris Mason 			high = mid;
550be0e5c09SChris Mason 		else {
551be0e5c09SChris Mason 			*slot = mid;
552479965d6SChris Mason 			if (map_token)
5535f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
554be0e5c09SChris Mason 			return 0;
555be0e5c09SChris Mason 		}
556be0e5c09SChris Mason 	}
557be0e5c09SChris Mason 	*slot = low;
5585f39d397SChris Mason 	if (map_token)
5595f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
560be0e5c09SChris Mason 	return 1;
561be0e5c09SChris Mason }
562be0e5c09SChris Mason 
56397571fd0SChris Mason /*
56497571fd0SChris Mason  * simple bin_search frontend that does the right thing for
56597571fd0SChris Mason  * leaves vs nodes
56697571fd0SChris Mason  */
5675f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
5685f39d397SChris Mason 		      int level, int *slot)
569be0e5c09SChris Mason {
5705f39d397SChris Mason 	if (level == 0) {
5715f39d397SChris Mason 		return generic_bin_search(eb,
5725f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
5730783fcfcSChris Mason 					  sizeof(struct btrfs_item),
5745f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5757518a238SChris Mason 					  slot);
576be0e5c09SChris Mason 	} else {
5775f39d397SChris Mason 		return generic_bin_search(eb,
5785f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
579123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
5805f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5817518a238SChris Mason 					  slot);
582be0e5c09SChris Mason 	}
583be0e5c09SChris Mason 	return -1;
584be0e5c09SChris Mason }
585be0e5c09SChris Mason 
5865f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
5875f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
588bb803951SChris Mason {
589bb803951SChris Mason 	if (slot < 0)
590bb803951SChris Mason 		return NULL;
5915f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
592bb803951SChris Mason 		return NULL;
593db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
594db94535dSChris Mason 		       btrfs_level_size(root, btrfs_header_level(parent) - 1));
595bb803951SChris Mason }
596bb803951SChris Mason 
597e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
598e089f05cSChris Mason 			 *root, struct btrfs_path *path, int level)
599bb803951SChris Mason {
6005f39d397SChris Mason 	struct extent_buffer *right = NULL;
6015f39d397SChris Mason 	struct extent_buffer *mid;
6025f39d397SChris Mason 	struct extent_buffer *left = NULL;
6035f39d397SChris Mason 	struct extent_buffer *parent = NULL;
604bb803951SChris Mason 	int ret = 0;
605bb803951SChris Mason 	int wret;
606bb803951SChris Mason 	int pslot;
607bb803951SChris Mason 	int orig_slot = path->slots[level];
60854aa1f4dSChris Mason 	int err_on_enospc = 0;
60979f95c82SChris Mason 	u64 orig_ptr;
610bb803951SChris Mason 
611bb803951SChris Mason 	if (level == 0)
612bb803951SChris Mason 		return 0;
613bb803951SChris Mason 
6145f39d397SChris Mason 	mid = path->nodes[level];
6151d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
61679f95c82SChris Mason 
617234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
6185f39d397SChris Mason 		parent = path->nodes[level + 1];
619bb803951SChris Mason 	pslot = path->slots[level + 1];
620bb803951SChris Mason 
62140689478SChris Mason 	/*
62240689478SChris Mason 	 * deal with the case where there is only one pointer in the root
62340689478SChris Mason 	 * by promoting the node below to a root
62440689478SChris Mason 	 */
6255f39d397SChris Mason 	if (!parent) {
6265f39d397SChris Mason 		struct extent_buffer *child;
627bb803951SChris Mason 
6285f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
629bb803951SChris Mason 			return 0;
630bb803951SChris Mason 
631bb803951SChris Mason 		/* promote the child to a root */
6325f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
633bb803951SChris Mason 		BUG_ON(!child);
634bb803951SChris Mason 		root->node = child;
635bb803951SChris Mason 		path->nodes[level] = NULL;
6365f39d397SChris Mason 		clean_tree_block(trans, root, mid);
6375f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
638bb803951SChris Mason 		/* once for the path */
6395f39d397SChris Mason 		free_extent_buffer(mid);
640db94535dSChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
641bb803951SChris Mason 		/* once for the root ptr */
6425f39d397SChris Mason 		free_extent_buffer(mid);
643db94535dSChris Mason 		return ret;
644bb803951SChris Mason 	}
6455f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
646123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
647bb803951SChris Mason 		return 0;
648bb803951SChris Mason 
6495f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
65054aa1f4dSChris Mason 		err_on_enospc = 1;
65154aa1f4dSChris Mason 
6525f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
6535f39d397SChris Mason 	if (left) {
6545f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
6555f39d397SChris Mason 				       parent, pslot - 1, &left);
65654aa1f4dSChris Mason 		if (wret) {
65754aa1f4dSChris Mason 			ret = wret;
65854aa1f4dSChris Mason 			goto enospc;
65954aa1f4dSChris Mason 		}
6602cc58cf2SChris Mason 	}
6615f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
6625f39d397SChris Mason 	if (right) {
6635f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
6645f39d397SChris Mason 				       parent, pslot + 1, &right);
6652cc58cf2SChris Mason 		if (wret) {
6662cc58cf2SChris Mason 			ret = wret;
6672cc58cf2SChris Mason 			goto enospc;
6682cc58cf2SChris Mason 		}
6692cc58cf2SChris Mason 	}
6702cc58cf2SChris Mason 
6712cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
6725f39d397SChris Mason 	if (left) {
6735f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
6745f39d397SChris Mason 		wret = push_node_left(trans, root, left, mid);
67579f95c82SChris Mason 		if (wret < 0)
67679f95c82SChris Mason 			ret = wret;
6775f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
67854aa1f4dSChris Mason 			err_on_enospc = 1;
679bb803951SChris Mason 	}
68079f95c82SChris Mason 
68179f95c82SChris Mason 	/*
68279f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
68379f95c82SChris Mason 	 */
6845f39d397SChris Mason 	if (right) {
6855f39d397SChris Mason 		wret = push_node_left(trans, root, mid, right);
68654aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
68779f95c82SChris Mason 			ret = wret;
6885f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
689db94535dSChris Mason 			u64 bytenr = right->start;
690db94535dSChris Mason 			u32 blocksize = right->len;
691db94535dSChris Mason 
6925f39d397SChris Mason 			clean_tree_block(trans, root, right);
6935f39d397SChris Mason 			wait_on_tree_block_writeback(root, right);
6945f39d397SChris Mason 			free_extent_buffer(right);
695bb803951SChris Mason 			right = NULL;
696e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
697e089f05cSChris Mason 				       1);
698bb803951SChris Mason 			if (wret)
699bb803951SChris Mason 				ret = wret;
700db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
701db94535dSChris Mason 						 blocksize, 1);
702bb803951SChris Mason 			if (wret)
703bb803951SChris Mason 				ret = wret;
704bb803951SChris Mason 		} else {
7055f39d397SChris Mason 			struct btrfs_disk_key right_key;
7065f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
7075f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
7085f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
709bb803951SChris Mason 		}
710bb803951SChris Mason 	}
7115f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
71279f95c82SChris Mason 		/*
71379f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
71479f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
71579f95c82SChris Mason 		 * could try to delete the only pointer in this node.
71679f95c82SChris Mason 		 * So, pull some keys from the left.
71779f95c82SChris Mason 		 * There has to be a left pointer at this point because
71879f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
71979f95c82SChris Mason 		 * right
72079f95c82SChris Mason 		 */
7215f39d397SChris Mason 		BUG_ON(!left);
7225f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
72354aa1f4dSChris Mason 		if (wret < 0) {
72479f95c82SChris Mason 			ret = wret;
72554aa1f4dSChris Mason 			goto enospc;
72654aa1f4dSChris Mason 		}
72779f95c82SChris Mason 		BUG_ON(wret == 1);
72879f95c82SChris Mason 	}
7295f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
73079f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
731db94535dSChris Mason 		u64 bytenr = mid->start;
732db94535dSChris Mason 		u32 blocksize = mid->len;
7335f39d397SChris Mason 		clean_tree_block(trans, root, mid);
7345f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
7355f39d397SChris Mason 		free_extent_buffer(mid);
736bb803951SChris Mason 		mid = NULL;
737e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
738bb803951SChris Mason 		if (wret)
739bb803951SChris Mason 			ret = wret;
740db94535dSChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
741bb803951SChris Mason 		if (wret)
742bb803951SChris Mason 			ret = wret;
74379f95c82SChris Mason 	} else {
74479f95c82SChris Mason 		/* update the parent key to reflect our changes */
7455f39d397SChris Mason 		struct btrfs_disk_key mid_key;
7465f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
7475f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
7485f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
74979f95c82SChris Mason 	}
750bb803951SChris Mason 
75179f95c82SChris Mason 	/* update the path */
7525f39d397SChris Mason 	if (left) {
7535f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
7545f39d397SChris Mason 			extent_buffer_get(left);
7555f39d397SChris Mason 			path->nodes[level] = left;
756bb803951SChris Mason 			path->slots[level + 1] -= 1;
757bb803951SChris Mason 			path->slots[level] = orig_slot;
7585f39d397SChris Mason 			if (mid)
7595f39d397SChris Mason 				free_extent_buffer(mid);
760bb803951SChris Mason 		} else {
7615f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
762bb803951SChris Mason 			path->slots[level] = orig_slot;
763bb803951SChris Mason 		}
764bb803951SChris Mason 	}
76579f95c82SChris Mason 	/* double check we haven't messed things up */
766123abc88SChris Mason 	check_block(root, path, level);
767e20d96d6SChris Mason 	if (orig_ptr !=
7685f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
76979f95c82SChris Mason 		BUG();
77054aa1f4dSChris Mason enospc:
7715f39d397SChris Mason 	if (right)
7725f39d397SChris Mason 		free_extent_buffer(right);
7735f39d397SChris Mason 	if (left)
7745f39d397SChris Mason 		free_extent_buffer(left);
775bb803951SChris Mason 	return ret;
776bb803951SChris Mason }
777bb803951SChris Mason 
778e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
779e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
780e66f709bSChris Mason 				struct btrfs_root *root,
781e66f709bSChris Mason 				struct btrfs_path *path, int level)
782e66f709bSChris Mason {
7835f39d397SChris Mason 	struct extent_buffer *right = NULL;
7845f39d397SChris Mason 	struct extent_buffer *mid;
7855f39d397SChris Mason 	struct extent_buffer *left = NULL;
7865f39d397SChris Mason 	struct extent_buffer *parent = NULL;
787e66f709bSChris Mason 	int ret = 0;
788e66f709bSChris Mason 	int wret;
789e66f709bSChris Mason 	int pslot;
790e66f709bSChris Mason 	int orig_slot = path->slots[level];
791e66f709bSChris Mason 	u64 orig_ptr;
792e66f709bSChris Mason 
793e66f709bSChris Mason 	if (level == 0)
794e66f709bSChris Mason 		return 1;
795e66f709bSChris Mason 
7965f39d397SChris Mason 	mid = path->nodes[level];
797e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
798e66f709bSChris Mason 
799e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
8005f39d397SChris Mason 		parent = path->nodes[level + 1];
801e66f709bSChris Mason 	pslot = path->slots[level + 1];
802e66f709bSChris Mason 
8035f39d397SChris Mason 	if (!parent)
804e66f709bSChris Mason 		return 1;
805e66f709bSChris Mason 
8065f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
807e66f709bSChris Mason 
808e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
8095f39d397SChris Mason 	if (left) {
810e66f709bSChris Mason 		u32 left_nr;
8115f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
81233ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
81333ade1f8SChris Mason 			wret = 1;
81433ade1f8SChris Mason 		} else {
8155f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
8165f39d397SChris Mason 					      pslot - 1, &left);
81754aa1f4dSChris Mason 			if (ret)
81854aa1f4dSChris Mason 				wret = 1;
81954aa1f4dSChris Mason 			else {
82054aa1f4dSChris Mason 				wret = push_node_left(trans, root,
8215f39d397SChris Mason 						      left, mid);
82254aa1f4dSChris Mason 			}
82333ade1f8SChris Mason 		}
824e66f709bSChris Mason 		if (wret < 0)
825e66f709bSChris Mason 			ret = wret;
826e66f709bSChris Mason 		if (wret == 0) {
8275f39d397SChris Mason 			struct btrfs_disk_key disk_key;
828e66f709bSChris Mason 			orig_slot += left_nr;
8295f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
8305f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
8315f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8325f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
8335f39d397SChris Mason 				path->nodes[level] = left;
834e66f709bSChris Mason 				path->slots[level + 1] -= 1;
835e66f709bSChris Mason 				path->slots[level] = orig_slot;
8365f39d397SChris Mason 				free_extent_buffer(mid);
837e66f709bSChris Mason 			} else {
838e66f709bSChris Mason 				orig_slot -=
8395f39d397SChris Mason 					btrfs_header_nritems(left);
840e66f709bSChris Mason 				path->slots[level] = orig_slot;
8415f39d397SChris Mason 				free_extent_buffer(left);
842e66f709bSChris Mason 			}
843e66f709bSChris Mason 			return 0;
844e66f709bSChris Mason 		}
8455f39d397SChris Mason 		free_extent_buffer(left);
846e66f709bSChris Mason 	}
8475f39d397SChris Mason 	right= read_node_slot(root, parent, pslot + 1);
848e66f709bSChris Mason 
849e66f709bSChris Mason 	/*
850e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
851e66f709bSChris Mason 	 */
8525f39d397SChris Mason 	if (right) {
85333ade1f8SChris Mason 		u32 right_nr;
8545f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
85533ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
85633ade1f8SChris Mason 			wret = 1;
85733ade1f8SChris Mason 		} else {
8585f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
8595f39d397SChris Mason 					      parent, pslot + 1,
8605f39d397SChris Mason 					      &right);
86154aa1f4dSChris Mason 			if (ret)
86254aa1f4dSChris Mason 				wret = 1;
86354aa1f4dSChris Mason 			else {
86433ade1f8SChris Mason 				wret = balance_node_right(trans, root,
8655f39d397SChris Mason 							  right, mid);
86633ade1f8SChris Mason 			}
86754aa1f4dSChris Mason 		}
868e66f709bSChris Mason 		if (wret < 0)
869e66f709bSChris Mason 			ret = wret;
870e66f709bSChris Mason 		if (wret == 0) {
8715f39d397SChris Mason 			struct btrfs_disk_key disk_key;
8725f39d397SChris Mason 
8735f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
8745f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
8755f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8765f39d397SChris Mason 
8775f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
8785f39d397SChris Mason 				path->nodes[level] = right;
879e66f709bSChris Mason 				path->slots[level + 1] += 1;
880e66f709bSChris Mason 				path->slots[level] = orig_slot -
8815f39d397SChris Mason 					btrfs_header_nritems(mid);
8825f39d397SChris Mason 				free_extent_buffer(mid);
883e66f709bSChris Mason 			} else {
8845f39d397SChris Mason 				free_extent_buffer(right);
885e66f709bSChris Mason 			}
886e66f709bSChris Mason 			return 0;
887e66f709bSChris Mason 		}
8885f39d397SChris Mason 		free_extent_buffer(right);
889e66f709bSChris Mason 	}
890e66f709bSChris Mason 	return 1;
891e66f709bSChris Mason }
892e66f709bSChris Mason 
89374123bd7SChris Mason /*
8943c69faecSChris Mason  * readahead one full node of leaves
8953c69faecSChris Mason  */
8963c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
8976702ed49SChris Mason 			     int level, int slot)
8983c69faecSChris Mason {
8995f39d397SChris Mason 	struct extent_buffer *node;
9003c69faecSChris Mason 	u32 nritems;
9013c69faecSChris Mason 	u64 search;
9026b80053dSChris Mason 	u64 lowest_read;
9036b80053dSChris Mason 	u64 highest_read;
9046b80053dSChris Mason 	u64 nread = 0;
9053c69faecSChris Mason 	int direction = path->reada;
9065f39d397SChris Mason 	struct extent_buffer *eb;
9076b80053dSChris Mason 	u32 nr;
9086b80053dSChris Mason 	u32 blocksize;
9096b80053dSChris Mason 	u32 nscan = 0;
910db94535dSChris Mason 
911a6b6e75eSChris Mason 	if (level != 1)
9123c69faecSChris Mason 		return;
9133c69faecSChris Mason 
9146702ed49SChris Mason 	if (!path->nodes[level])
9156702ed49SChris Mason 		return;
9166702ed49SChris Mason 
9175f39d397SChris Mason 	node = path->nodes[level];
9183c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
9196b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
9206b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
9215f39d397SChris Mason 	if (eb) {
9225f39d397SChris Mason 		free_extent_buffer(eb);
9233c69faecSChris Mason 		return;
9243c69faecSChris Mason 	}
9253c69faecSChris Mason 
9266b80053dSChris Mason 	highest_read = search;
9276b80053dSChris Mason 	lowest_read = search;
9286b80053dSChris Mason 
9295f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
9306b80053dSChris Mason 	nr = slot;
9313c69faecSChris Mason 	while(1) {
9326b80053dSChris Mason 		if (direction < 0) {
9336b80053dSChris Mason 			if (nr == 0)
9343c69faecSChris Mason 				break;
9356b80053dSChris Mason 			nr--;
9366b80053dSChris Mason 		} else if (direction > 0) {
9376b80053dSChris Mason 			nr++;
9386b80053dSChris Mason 			if (nr >= nritems)
9396b80053dSChris Mason 				break;
9403c69faecSChris Mason 		}
9416b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
9426b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
9436b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
9446b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
9456b80053dSChris Mason 			readahead_tree_block(root, search, blocksize);
9466b80053dSChris Mason 			nread += blocksize;
9473c69faecSChris Mason 		}
9486b80053dSChris Mason 		nscan++;
9496b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
9506b80053dSChris Mason 			break;
9516b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
9526b80053dSChris Mason 			break;
9536b80053dSChris Mason 
9546b80053dSChris Mason 		if (search < lowest_read)
9556b80053dSChris Mason 			lowest_read = search;
9566b80053dSChris Mason 		if (search > highest_read)
9576b80053dSChris Mason 			highest_read = search;
9583c69faecSChris Mason 	}
9593c69faecSChris Mason }
9603c69faecSChris Mason /*
96174123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
96274123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
96374123bd7SChris Mason  * level of the path (level 0)
96474123bd7SChris Mason  *
96574123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
966aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
967aa5d6bedSChris Mason  * search a negative error number is returned.
96897571fd0SChris Mason  *
96997571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
97097571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
97197571fd0SChris Mason  * possible)
97274123bd7SChris Mason  */
973e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
974e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
975e089f05cSChris Mason 		      ins_len, int cow)
976be0e5c09SChris Mason {
9775f39d397SChris Mason 	struct extent_buffer *b;
978db94535dSChris Mason 	u64 bytenr;
979be0e5c09SChris Mason 	int slot;
980be0e5c09SChris Mason 	int ret;
981be0e5c09SChris Mason 	int level;
9823c69faecSChris Mason 	int should_reada = p->reada;
9839f3a7427SChris Mason 	u8 lowest_level = 0;
9849f3a7427SChris Mason 
9856702ed49SChris Mason 	lowest_level = p->lowest_level;
9866702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
98722b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
98822b0ebdaSChris Mason 	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
989bb803951SChris Mason again:
990bb803951SChris Mason 	b = root->node;
9915f39d397SChris Mason 	extent_buffer_get(b);
992eb60ceacSChris Mason 	while (b) {
9935f39d397SChris Mason 		level = btrfs_header_level(b);
99402217ed2SChris Mason 		if (cow) {
99502217ed2SChris Mason 			int wret;
996e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
997e20d96d6SChris Mason 					       p->nodes[level + 1],
998e20d96d6SChris Mason 					       p->slots[level + 1],
999252c38f0SYan 					       &b);
100054aa1f4dSChris Mason 			if (wret) {
10015f39d397SChris Mason 				free_extent_buffer(b);
100254aa1f4dSChris Mason 				return wret;
100354aa1f4dSChris Mason 			}
100402217ed2SChris Mason 		}
100502217ed2SChris Mason 		BUG_ON(!cow && ins_len);
10065f39d397SChris Mason 		if (level != btrfs_header_level(b))
10072c90e5d6SChris Mason 			WARN_ON(1);
10085f39d397SChris Mason 		level = btrfs_header_level(b);
1009eb60ceacSChris Mason 		p->nodes[level] = b;
1010123abc88SChris Mason 		ret = check_block(root, p, level);
1011aa5d6bedSChris Mason 		if (ret)
1012aa5d6bedSChris Mason 			return -1;
10135f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
10145f39d397SChris Mason 		if (level != 0) {
1015be0e5c09SChris Mason 			if (ret && slot > 0)
1016be0e5c09SChris Mason 				slot -= 1;
1017be0e5c09SChris Mason 			p->slots[level] = slot;
10185f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
1019d4dbff95SChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
1020e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
10215c680ed6SChris Mason 				BUG_ON(sret > 0);
10225c680ed6SChris Mason 				if (sret)
10235c680ed6SChris Mason 					return sret;
10245c680ed6SChris Mason 				b = p->nodes[level];
10255c680ed6SChris Mason 				slot = p->slots[level];
1026bb803951SChris Mason 			} else if (ins_len < 0) {
1027e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1028e089f05cSChris Mason 							 level);
1029bb803951SChris Mason 				if (sret)
1030bb803951SChris Mason 					return sret;
1031bb803951SChris Mason 				b = p->nodes[level];
1032f510cfecSChris Mason 				if (!b) {
1033f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1034bb803951SChris Mason 					goto again;
1035f510cfecSChris Mason 				}
1036bb803951SChris Mason 				slot = p->slots[level];
10375f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
10385c680ed6SChris Mason 			}
10399f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
10409f3a7427SChris Mason 			if (level == lowest_level)
10419f3a7427SChris Mason 				break;
1042db94535dSChris Mason 			bytenr = btrfs_node_blockptr(b, slot);
10436702ed49SChris Mason 			if (should_reada)
10446702ed49SChris Mason 				reada_for_search(root, p, level, slot);
1045db94535dSChris Mason 			b = read_tree_block(root, bytenr,
1046db94535dSChris Mason 					    btrfs_level_size(root, level - 1));
1047be0e5c09SChris Mason 		} else {
1048be0e5c09SChris Mason 			p->slots[level] = slot;
10495f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
10500783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1051d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1052d4dbff95SChris Mason 						      p, ins_len);
10535c680ed6SChris Mason 				BUG_ON(sret > 0);
10545c680ed6SChris Mason 				if (sret)
10555c680ed6SChris Mason 					return sret;
10565c680ed6SChris Mason 			}
1057be0e5c09SChris Mason 			return ret;
1058be0e5c09SChris Mason 		}
1059be0e5c09SChris Mason 	}
1060aa5d6bedSChris Mason 	return 1;
1061be0e5c09SChris Mason }
1062be0e5c09SChris Mason 
106374123bd7SChris Mason /*
106474123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
106574123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
106674123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
106774123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
106874123bd7SChris Mason  * higher levels
1069aa5d6bedSChris Mason  *
1070aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1071aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
107274123bd7SChris Mason  */
10735f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
10745f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
10755f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1076be0e5c09SChris Mason {
1077be0e5c09SChris Mason 	int i;
1078aa5d6bedSChris Mason 	int ret = 0;
10795f39d397SChris Mason 	struct extent_buffer *t;
10805f39d397SChris Mason 
1081234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1082be0e5c09SChris Mason 		int tslot = path->slots[i];
1083eb60ceacSChris Mason 		if (!path->nodes[i])
1084be0e5c09SChris Mason 			break;
10855f39d397SChris Mason 		t = path->nodes[i];
10865f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1087d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1088be0e5c09SChris Mason 		if (tslot != 0)
1089be0e5c09SChris Mason 			break;
1090be0e5c09SChris Mason 	}
1091aa5d6bedSChris Mason 	return ret;
1092be0e5c09SChris Mason }
1093be0e5c09SChris Mason 
109474123bd7SChris Mason /*
109574123bd7SChris Mason  * try to push data from one node into the next node left in the
109679f95c82SChris Mason  * tree.
1097aa5d6bedSChris Mason  *
1098aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1099aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
110074123bd7SChris Mason  */
1101e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
11025f39d397SChris Mason 			  *root, struct extent_buffer *dst,
11035f39d397SChris Mason 			  struct extent_buffer *src)
1104be0e5c09SChris Mason {
1105be0e5c09SChris Mason 	int push_items = 0;
1106bb803951SChris Mason 	int src_nritems;
1107bb803951SChris Mason 	int dst_nritems;
1108aa5d6bedSChris Mason 	int ret = 0;
1109be0e5c09SChris Mason 
11105f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11115f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1112123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
111354aa1f4dSChris Mason 
1114eb60ceacSChris Mason 	if (push_items <= 0) {
1115be0e5c09SChris Mason 		return 1;
1116eb60ceacSChris Mason 	}
1117be0e5c09SChris Mason 
1118be0e5c09SChris Mason 	if (src_nritems < push_items)
1119be0e5c09SChris Mason 		push_items = src_nritems;
112079f95c82SChris Mason 
11215f39d397SChris Mason 	copy_extent_buffer(dst, src,
11225f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
11235f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1124123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
11255f39d397SChris Mason 
1126bb803951SChris Mason 	if (push_items < src_nritems) {
11275f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
11285f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1129e2fa7227SChris Mason 				      (src_nritems - push_items) *
1130123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1131bb803951SChris Mason 	}
11325f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11335f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
11345f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11355f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1136bb803951SChris Mason 	return ret;
1137be0e5c09SChris Mason }
1138be0e5c09SChris Mason 
113997571fd0SChris Mason /*
114079f95c82SChris Mason  * try to push data from one node into the next node right in the
114179f95c82SChris Mason  * tree.
114279f95c82SChris Mason  *
114379f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
114479f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
114579f95c82SChris Mason  *
114679f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
114779f95c82SChris Mason  */
11485f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
11495f39d397SChris Mason 			      struct btrfs_root *root,
11505f39d397SChris Mason 			      struct extent_buffer *dst,
11515f39d397SChris Mason 			      struct extent_buffer *src)
115279f95c82SChris Mason {
115379f95c82SChris Mason 	int push_items = 0;
115479f95c82SChris Mason 	int max_push;
115579f95c82SChris Mason 	int src_nritems;
115679f95c82SChris Mason 	int dst_nritems;
115779f95c82SChris Mason 	int ret = 0;
115879f95c82SChris Mason 
11595f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11605f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1161123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
11625f39d397SChris Mason 	if (push_items <= 0)
116379f95c82SChris Mason 		return 1;
116479f95c82SChris Mason 
116579f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
116679f95c82SChris Mason 	/* don't try to empty the node */
1167252c38f0SYan 	if (max_push >= src_nritems)
116879f95c82SChris Mason 		return 1;
1169252c38f0SYan 
117079f95c82SChris Mason 	if (max_push < push_items)
117179f95c82SChris Mason 		push_items = max_push;
117279f95c82SChris Mason 
11735f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
11745f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
11755f39d397SChris Mason 				      (dst_nritems) *
11765f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1177d6025579SChris Mason 
11785f39d397SChris Mason 	copy_extent_buffer(dst, src,
11795f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
11805f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1181123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
118279f95c82SChris Mason 
11835f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11845f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
118579f95c82SChris Mason 
11865f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11875f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
118879f95c82SChris Mason 	return ret;
118979f95c82SChris Mason }
119079f95c82SChris Mason 
119179f95c82SChris Mason /*
119297571fd0SChris Mason  * helper function to insert a new root level in the tree.
119397571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
119497571fd0SChris Mason  * point to the existing root
1195aa5d6bedSChris Mason  *
1196aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
119797571fd0SChris Mason  */
11985f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans,
11995f39d397SChris Mason 			   struct btrfs_root *root,
12005f39d397SChris Mason 			   struct btrfs_path *path, int level)
120174123bd7SChris Mason {
12025f39d397SChris Mason 	struct extent_buffer *lower;
12035f39d397SChris Mason 	struct extent_buffer *c;
12045f39d397SChris Mason 	struct btrfs_disk_key lower_key;
12055c680ed6SChris Mason 
12065c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
12075c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
12085c680ed6SChris Mason 
1209db94535dSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
1210db94535dSChris Mason 				   root->node->start, 0);
12115f39d397SChris Mason 	if (IS_ERR(c))
12125f39d397SChris Mason 		return PTR_ERR(c);
12135f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
12145f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
12155f39d397SChris Mason 	btrfs_set_header_level(c, level);
1216db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
12175f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
12185f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
12195f39d397SChris Mason 	lower = path->nodes[level-1];
1220d5719762SChris Mason 
12215f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
12225f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
12235f39d397SChris Mason 			    BTRFS_FSID_SIZE);
12245f39d397SChris Mason 	if (level == 1)
12255f39d397SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
12265f39d397SChris Mason 	else
12275f39d397SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
12285f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1229db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
12305f39d397SChris Mason 
12315f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1232d5719762SChris Mason 
1233cfaa7295SChris Mason 	/* the super has an extra ref to root->node */
12345f39d397SChris Mason 	free_extent_buffer(root->node);
12355f39d397SChris Mason 	root->node = c;
12365f39d397SChris Mason 	extent_buffer_get(c);
12375f39d397SChris Mason 	path->nodes[level] = c;
123874123bd7SChris Mason 	path->slots[level] = 0;
123974123bd7SChris Mason 	return 0;
124074123bd7SChris Mason }
12415c680ed6SChris Mason 
12425c680ed6SChris Mason /*
12435c680ed6SChris Mason  * worker function to insert a single pointer in a node.
12445c680ed6SChris Mason  * the node should have enough room for the pointer already
124597571fd0SChris Mason  *
12465c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
12475c680ed6SChris Mason  * blocknr is the block the key points to.
1248aa5d6bedSChris Mason  *
1249aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
12505c680ed6SChris Mason  */
1251e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1252e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1253db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
12545c680ed6SChris Mason {
12555f39d397SChris Mason 	struct extent_buffer *lower;
12565c680ed6SChris Mason 	int nritems;
12575c680ed6SChris Mason 
12585c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
12595f39d397SChris Mason 	lower = path->nodes[level];
12605f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
126174123bd7SChris Mason 	if (slot > nritems)
126274123bd7SChris Mason 		BUG();
1263123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
126474123bd7SChris Mason 		BUG();
126574123bd7SChris Mason 	if (slot != nritems) {
12665f39d397SChris Mason 		memmove_extent_buffer(lower,
12675f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
12685f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1269123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
127074123bd7SChris Mason 	}
12715f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1272db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
12735f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
12745f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
127574123bd7SChris Mason 	return 0;
127674123bd7SChris Mason }
127774123bd7SChris Mason 
127897571fd0SChris Mason /*
127997571fd0SChris Mason  * split the node at the specified level in path in two.
128097571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
128197571fd0SChris Mason  *
128297571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
128397571fd0SChris Mason  * left and right, if either one works, it returns right away.
1284aa5d6bedSChris Mason  *
1285aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
128697571fd0SChris Mason  */
1287e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1288e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1289be0e5c09SChris Mason {
12905f39d397SChris Mason 	struct extent_buffer *c;
12915f39d397SChris Mason 	struct extent_buffer *split;
12925f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1293be0e5c09SChris Mason 	int mid;
12945c680ed6SChris Mason 	int ret;
1295aa5d6bedSChris Mason 	int wret;
12967518a238SChris Mason 	u32 c_nritems;
1297be0e5c09SChris Mason 
12985f39d397SChris Mason 	c = path->nodes[level];
12995f39d397SChris Mason 	if (c == root->node) {
13005c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1301e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
13025c680ed6SChris Mason 		if (ret)
13035c680ed6SChris Mason 			return ret;
1304e66f709bSChris Mason 	} else {
1305e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
13065f39d397SChris Mason 		c = path->nodes[level];
13075f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1308e66f709bSChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1309e66f709bSChris Mason 			return 0;
131054aa1f4dSChris Mason 		if (ret < 0)
131154aa1f4dSChris Mason 			return ret;
13125c680ed6SChris Mason 	}
1313e66f709bSChris Mason 
13145f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
1315db94535dSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
1316db94535dSChris Mason 				       c->start, 0);
13175f39d397SChris Mason 	if (IS_ERR(split))
13185f39d397SChris Mason 		return PTR_ERR(split);
131954aa1f4dSChris Mason 
13205f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
13215f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1322db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
13235f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
13245f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
13255f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
13265f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
13275f39d397SChris Mason 			    BTRFS_FSID_SIZE);
13285f39d397SChris Mason 
13297518a238SChris Mason 	mid = (c_nritems + 1) / 2;
13305f39d397SChris Mason 
13315f39d397SChris Mason 	copy_extent_buffer(split, c,
13325f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
13335f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1334123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
13355f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
13365f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1337aa5d6bedSChris Mason 	ret = 0;
1338aa5d6bedSChris Mason 
13395f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
13405f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
13415f39d397SChris Mason 
13425f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1343db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
13445f39d397SChris Mason 			  path->slots[level + 1] + 1,
1345123abc88SChris Mason 			  level + 1);
1346aa5d6bedSChris Mason 	if (wret)
1347aa5d6bedSChris Mason 		ret = wret;
1348aa5d6bedSChris Mason 
13495de08d7dSChris Mason 	if (path->slots[level] >= mid) {
13505c680ed6SChris Mason 		path->slots[level] -= mid;
13515f39d397SChris Mason 		free_extent_buffer(c);
13525f39d397SChris Mason 		path->nodes[level] = split;
13535c680ed6SChris Mason 		path->slots[level + 1] += 1;
1354eb60ceacSChris Mason 	} else {
13555f39d397SChris Mason 		free_extent_buffer(split);
1356be0e5c09SChris Mason 	}
1357aa5d6bedSChris Mason 	return ret;
1358be0e5c09SChris Mason }
1359be0e5c09SChris Mason 
136074123bd7SChris Mason /*
136174123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
136274123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
136374123bd7SChris Mason  * space used both by the item structs and the item data
136474123bd7SChris Mason  */
13655f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1366be0e5c09SChris Mason {
1367be0e5c09SChris Mason 	int data_len;
13685f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1369d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1370be0e5c09SChris Mason 
1371be0e5c09SChris Mason 	if (!nr)
1372be0e5c09SChris Mason 		return 0;
13735f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
13745f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
13750783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1376d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1377be0e5c09SChris Mason 	return data_len;
1378be0e5c09SChris Mason }
1379be0e5c09SChris Mason 
138074123bd7SChris Mason /*
1381d4dbff95SChris Mason  * The space between the end of the leaf items and
1382d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1383d4dbff95SChris Mason  * the leaf has left for both items and data
1384d4dbff95SChris Mason  */
13855f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1386d4dbff95SChris Mason {
13875f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
13885f39d397SChris Mason 	int ret;
13895f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
13905f39d397SChris Mason 	if (ret < 0) {
13915f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1392ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
13935f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
13945f39d397SChris Mason 	}
13955f39d397SChris Mason 	return ret;
1396d4dbff95SChris Mason }
1397d4dbff95SChris Mason 
1398d4dbff95SChris Mason /*
139900ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
140000ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1401aa5d6bedSChris Mason  *
1402aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1403aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
140400ec4c51SChris Mason  */
1405e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1406e089f05cSChris Mason 			   *root, struct btrfs_path *path, int data_size)
140700ec4c51SChris Mason {
14085f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
14095f39d397SChris Mason 	struct extent_buffer *right;
14105f39d397SChris Mason 	struct extent_buffer *upper;
14115f39d397SChris Mason 	struct btrfs_disk_key disk_key;
141200ec4c51SChris Mason 	int slot;
141300ec4c51SChris Mason 	int i;
141400ec4c51SChris Mason 	int free_space;
141500ec4c51SChris Mason 	int push_space = 0;
141600ec4c51SChris Mason 	int push_items = 0;
14170783fcfcSChris Mason 	struct btrfs_item *item;
14187518a238SChris Mason 	u32 left_nritems;
14197518a238SChris Mason 	u32 right_nritems;
14205f39d397SChris Mason 	u32 data_end;
1421db94535dSChris Mason 	u32 this_item_size;
142254aa1f4dSChris Mason 	int ret;
142300ec4c51SChris Mason 
142400ec4c51SChris Mason 	slot = path->slots[1];
142500ec4c51SChris Mason 	if (!path->nodes[1]) {
142600ec4c51SChris Mason 		return 1;
142700ec4c51SChris Mason 	}
142800ec4c51SChris Mason 	upper = path->nodes[1];
14295f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
143000ec4c51SChris Mason 		return 1;
14315f39d397SChris Mason 
1432db94535dSChris Mason 	right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1433db94535dSChris Mason 				root->leafsize);
1434123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14350783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14365f39d397SChris Mason 		free_extent_buffer(right);
143702217ed2SChris Mason 		return 1;
143802217ed2SChris Mason 	}
143902217ed2SChris Mason 
14405f39d397SChris Mason 	/* cow and double check */
14415f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
14425f39d397SChris Mason 			      slot + 1, &right);
14435f39d397SChris Mason 	if (ret) {
14445f39d397SChris Mason 		free_extent_buffer(right);
1445a429e513SChris Mason 		return 1;
1446a429e513SChris Mason 	}
14475f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14485f39d397SChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14495f39d397SChris Mason 		free_extent_buffer(right);
14505f39d397SChris Mason 		return 1;
14515f39d397SChris Mason 	}
14525f39d397SChris Mason 
14535f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
14545f39d397SChris Mason 	if (left_nritems == 0) {
14555f39d397SChris Mason 		free_extent_buffer(right);
14565f39d397SChris Mason 		return 1;
14575f39d397SChris Mason 	}
14585f39d397SChris Mason 
1459a429e513SChris Mason 	for (i = left_nritems - 1; i >= 1; i--) {
14605f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1461db94535dSChris Mason 
146200ec4c51SChris Mason 		if (path->slots[0] == i)
146300ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1464db94535dSChris Mason 
1465db94535dSChris Mason 		if (!left->map_token) {
1466db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1467db94535dSChris Mason 					sizeof(struct btrfs_item),
1468db94535dSChris Mason 					&left->map_token, &left->kaddr,
1469db94535dSChris Mason 					&left->map_start, &left->map_len,
1470db94535dSChris Mason 					KM_USER1);
1471db94535dSChris Mason 		}
1472db94535dSChris Mason 
1473db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1474db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
147500ec4c51SChris Mason 			break;
147600ec4c51SChris Mason 		push_items++;
1477db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1478db94535dSChris Mason 	}
1479db94535dSChris Mason 	if (left->map_token) {
1480db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1481db94535dSChris Mason 		left->map_token = NULL;
148200ec4c51SChris Mason 	}
14835f39d397SChris Mason 
148400ec4c51SChris Mason 	if (push_items == 0) {
14855f39d397SChris Mason 		free_extent_buffer(right);
148600ec4c51SChris Mason 		return 1;
148700ec4c51SChris Mason 	}
14885f39d397SChris Mason 
1489a429e513SChris Mason 	if (push_items == left_nritems)
1490a429e513SChris Mason 		WARN_ON(1);
14915f39d397SChris Mason 
149200ec4c51SChris Mason 	/* push left to right */
14935f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
14945f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1495123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
14965f39d397SChris Mason 
149700ec4c51SChris Mason 	/* make room in the right data area */
14985f39d397SChris Mason 	data_end = leaf_data_end(root, right);
14995f39d397SChris Mason 	memmove_extent_buffer(right,
15005f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
15015f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
15025f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
15035f39d397SChris Mason 
150400ec4c51SChris Mason 	/* copy from the left data area */
15055f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1506d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1507d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1508d6025579SChris Mason 		     push_space);
15095f39d397SChris Mason 
15105f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
15115f39d397SChris Mason 			      btrfs_item_nr_offset(0),
15120783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
15135f39d397SChris Mason 
151400ec4c51SChris Mason 	/* copy the items from left to right */
15155f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
15165f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
15170783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
151800ec4c51SChris Mason 
151900ec4c51SChris Mason 	/* update the item pointers */
15207518a238SChris Mason 	right_nritems += push_items;
15215f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1522123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
1523db94535dSChris Mason 
15247518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
15255f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1526db94535dSChris Mason 		if (!right->map_token) {
1527db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1528db94535dSChris Mason 					sizeof(struct btrfs_item),
1529db94535dSChris Mason 					&right->map_token, &right->kaddr,
1530db94535dSChris Mason 					&right->map_start, &right->map_len,
1531db94535dSChris Mason 					KM_USER1);
1532db94535dSChris Mason 		}
1533db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1534db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1535db94535dSChris Mason 	}
1536db94535dSChris Mason 
1537db94535dSChris Mason 	if (right->map_token) {
1538db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1539db94535dSChris Mason 		right->map_token = NULL;
154000ec4c51SChris Mason 	}
15417518a238SChris Mason 	left_nritems -= push_items;
15425f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
154300ec4c51SChris Mason 
15445f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
15455f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1546a429e513SChris Mason 
15475f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
15485f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1549d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
155002217ed2SChris Mason 
155100ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
15527518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
15537518a238SChris Mason 		path->slots[0] -= left_nritems;
15545f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
15555f39d397SChris Mason 		path->nodes[0] = right;
155600ec4c51SChris Mason 		path->slots[1] += 1;
155700ec4c51SChris Mason 	} else {
15585f39d397SChris Mason 		free_extent_buffer(right);
155900ec4c51SChris Mason 	}
156000ec4c51SChris Mason 	return 0;
156100ec4c51SChris Mason }
156200ec4c51SChris Mason /*
156374123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
156474123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
156574123bd7SChris Mason  */
1566e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1567e089f05cSChris Mason 			  *root, struct btrfs_path *path, int data_size)
1568be0e5c09SChris Mason {
15695f39d397SChris Mason 	struct btrfs_disk_key disk_key;
15705f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
15715f39d397SChris Mason 	struct extent_buffer *left;
1572be0e5c09SChris Mason 	int slot;
1573be0e5c09SChris Mason 	int i;
1574be0e5c09SChris Mason 	int free_space;
1575be0e5c09SChris Mason 	int push_space = 0;
1576be0e5c09SChris Mason 	int push_items = 0;
15770783fcfcSChris Mason 	struct btrfs_item *item;
15787518a238SChris Mason 	u32 old_left_nritems;
15795f39d397SChris Mason 	u32 right_nritems;
1580aa5d6bedSChris Mason 	int ret = 0;
1581aa5d6bedSChris Mason 	int wret;
1582db94535dSChris Mason 	u32 this_item_size;
1583db94535dSChris Mason 	u32 old_left_item_size;
1584be0e5c09SChris Mason 
1585be0e5c09SChris Mason 	slot = path->slots[1];
15865f39d397SChris Mason 	if (slot == 0)
1587be0e5c09SChris Mason 		return 1;
15885f39d397SChris Mason 	if (!path->nodes[1])
1589be0e5c09SChris Mason 		return 1;
15905f39d397SChris Mason 
1591*3685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
1592*3685f791SChris Mason 	if (right_nritems == 0) {
1593*3685f791SChris Mason 		return 1;
1594*3685f791SChris Mason 	}
1595*3685f791SChris Mason 
15965f39d397SChris Mason 	left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
1597db94535dSChris Mason 			       slot - 1), root->leafsize);
1598123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
15990783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
16005f39d397SChris Mason 		free_extent_buffer(left);
1601be0e5c09SChris Mason 		return 1;
1602be0e5c09SChris Mason 	}
160302217ed2SChris Mason 
160402217ed2SChris Mason 	/* cow and double check */
16055f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
16065f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
160754aa1f4dSChris Mason 	if (ret) {
160854aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
16095f39d397SChris Mason 		free_extent_buffer(left);
161054aa1f4dSChris Mason 		return 1;
161154aa1f4dSChris Mason 	}
1612*3685f791SChris Mason 
1613123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
16140783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
16155f39d397SChris Mason 		free_extent_buffer(left);
161602217ed2SChris Mason 		return 1;
161702217ed2SChris Mason 	}
161802217ed2SChris Mason 
16195f39d397SChris Mason 	for (i = 0; i < right_nritems - 1; i++) {
16205f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1621db94535dSChris Mason 		if (!right->map_token) {
1622db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1623db94535dSChris Mason 					sizeof(struct btrfs_item),
1624db94535dSChris Mason 					&right->map_token, &right->kaddr,
1625db94535dSChris Mason 					&right->map_start, &right->map_len,
1626db94535dSChris Mason 					KM_USER1);
1627db94535dSChris Mason 		}
1628db94535dSChris Mason 
1629be0e5c09SChris Mason 		if (path->slots[0] == i)
1630be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
1631db94535dSChris Mason 
1632db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
1633db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
1634be0e5c09SChris Mason 			break;
1635db94535dSChris Mason 
1636be0e5c09SChris Mason 		push_items++;
1637db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1638be0e5c09SChris Mason 	}
1639db94535dSChris Mason 
1640db94535dSChris Mason 	if (right->map_token) {
1641db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1642db94535dSChris Mason 		right->map_token = NULL;
1643db94535dSChris Mason 	}
1644db94535dSChris Mason 
1645be0e5c09SChris Mason 	if (push_items == 0) {
16465f39d397SChris Mason 		free_extent_buffer(left);
1647be0e5c09SChris Mason 		return 1;
1648be0e5c09SChris Mason 	}
16495f39d397SChris Mason 	if (push_items == btrfs_header_nritems(right))
1650a429e513SChris Mason 		WARN_ON(1);
16515f39d397SChris Mason 
1652be0e5c09SChris Mason 	/* push data from right to left */
16535f39d397SChris Mason 	copy_extent_buffer(left, right,
16545f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
16555f39d397SChris Mason 			   btrfs_item_nr_offset(0),
16565f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
16575f39d397SChris Mason 
1658123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
16595f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
16605f39d397SChris Mason 
16615f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
1662d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
1663123abc88SChris Mason 		     btrfs_leaf_data(right) +
16645f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
1665be0e5c09SChris Mason 		     push_space);
16665f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
1667eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
1668eb60ceacSChris Mason 
1669db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
1670be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
16715f39d397SChris Mason 		u32 ioff;
1672db94535dSChris Mason 
16735f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1674db94535dSChris Mason 		if (!left->map_token) {
1675db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1676db94535dSChris Mason 					sizeof(struct btrfs_item),
1677db94535dSChris Mason 					&left->map_token, &left->kaddr,
1678db94535dSChris Mason 					&left->map_start, &left->map_len,
1679db94535dSChris Mason 					KM_USER1);
1680db94535dSChris Mason 		}
1681db94535dSChris Mason 
16825f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
16835f39d397SChris Mason 		btrfs_set_item_offset(left, item,
1684db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1685be0e5c09SChris Mason 	}
16865f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1687db94535dSChris Mason 	if (left->map_token) {
1688db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1689db94535dSChris Mason 		left->map_token = NULL;
1690db94535dSChris Mason 	}
1691be0e5c09SChris Mason 
1692be0e5c09SChris Mason 	/* fixup right node */
16935f39d397SChris Mason 	push_space = btrfs_item_offset_nr(right, push_items - 1) -
1694123abc88SChris Mason 					  leaf_data_end(root, right);
16955f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_leaf_data(right) +
1696d6025579SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - push_space,
1697d6025579SChris Mason 			      btrfs_leaf_data(right) +
1698123abc88SChris Mason 			      leaf_data_end(root, right), push_space);
16995f39d397SChris Mason 
17005f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(0),
17015f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
17025f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
17030783fcfcSChris Mason 			     sizeof(struct btrfs_item));
17045f39d397SChris Mason 
17055f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right) - push_items;
17065f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1707123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
1708eb60ceacSChris Mason 
17095f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
17105f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1711db94535dSChris Mason 
1712db94535dSChris Mason 		if (!right->map_token) {
1713db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1714db94535dSChris Mason 					sizeof(struct btrfs_item),
1715db94535dSChris Mason 					&right->map_token, &right->kaddr,
1716db94535dSChris Mason 					&right->map_start, &right->map_len,
1717db94535dSChris Mason 					KM_USER1);
1718db94535dSChris Mason 		}
1719db94535dSChris Mason 
1720db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
1721db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1722db94535dSChris Mason 	}
1723db94535dSChris Mason 	if (right->map_token) {
1724db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1725db94535dSChris Mason 		right->map_token = NULL;
1726be0e5c09SChris Mason 	}
1727eb60ceacSChris Mason 
17285f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
17295f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1730098f59c2SChris Mason 
17315f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
17325f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1733aa5d6bedSChris Mason 	if (wret)
1734aa5d6bedSChris Mason 		ret = wret;
1735be0e5c09SChris Mason 
1736be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
1737be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
1738be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
17395f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
17405f39d397SChris Mason 		path->nodes[0] = left;
1741be0e5c09SChris Mason 		path->slots[1] -= 1;
1742be0e5c09SChris Mason 	} else {
17435f39d397SChris Mason 		free_extent_buffer(left);
1744be0e5c09SChris Mason 		path->slots[0] -= push_items;
1745be0e5c09SChris Mason 	}
1746eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1747aa5d6bedSChris Mason 	return ret;
1748be0e5c09SChris Mason }
1749be0e5c09SChris Mason 
175074123bd7SChris Mason /*
175174123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
175274123bd7SChris Mason  * available for the resulting leaf level of the path.
1753aa5d6bedSChris Mason  *
1754aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
175574123bd7SChris Mason  */
1756e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1757d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
1758d4dbff95SChris Mason 		      struct btrfs_path *path, int data_size)
1759be0e5c09SChris Mason {
17605f39d397SChris Mason 	struct extent_buffer *l;
17617518a238SChris Mason 	u32 nritems;
1762eb60ceacSChris Mason 	int mid;
1763eb60ceacSChris Mason 	int slot;
17645f39d397SChris Mason 	struct extent_buffer *right;
17650783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
1766be0e5c09SChris Mason 	int data_copy_size;
1767be0e5c09SChris Mason 	int rt_data_off;
1768be0e5c09SChris Mason 	int i;
1769d4dbff95SChris Mason 	int ret = 0;
1770aa5d6bedSChris Mason 	int wret;
1771d4dbff95SChris Mason 	int double_split = 0;
1772d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
1773be0e5c09SChris Mason 
177440689478SChris Mason 	/* first try to make some room by pushing left and right */
1775*3685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
1776*3685f791SChris Mason 		wret = push_leaf_right(trans, root, path, data_size);
17773326d1b0SChris Mason 		if (wret < 0) {
1778eaee50e8SChris Mason 			return wret;
17793326d1b0SChris Mason 		}
1780eaee50e8SChris Mason 		if (wret) {
1781*3685f791SChris Mason 			wret = push_leaf_left(trans, root, path, data_size);
1782eaee50e8SChris Mason 			if (wret < 0)
1783eaee50e8SChris Mason 				return wret;
1784eaee50e8SChris Mason 		}
17855f39d397SChris Mason 		l = path->nodes[0];
1786aa5d6bedSChris Mason 
1787aa5d6bedSChris Mason 		/* did the pushes work? */
1788123abc88SChris Mason 		if (btrfs_leaf_free_space(root, l) >=
17893326d1b0SChris Mason 		    sizeof(struct btrfs_item) + data_size) {
1790be0e5c09SChris Mason 			return 0;
17913326d1b0SChris Mason 		}
1792*3685f791SChris Mason 	} else {
1793*3685f791SChris Mason 		l = path->nodes[0];
1794*3685f791SChris Mason 	}
1795aa5d6bedSChris Mason 
17965c680ed6SChris Mason 	if (!path->nodes[1]) {
1797e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
17985c680ed6SChris Mason 		if (ret)
17995c680ed6SChris Mason 			return ret;
18005c680ed6SChris Mason 	}
1801eb60ceacSChris Mason 	slot = path->slots[0];
18025f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
1803eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
180454aa1f4dSChris Mason 
1805db94535dSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
1806db94535dSChris Mason 				       l->start, 0);
18075f39d397SChris Mason 	if (IS_ERR(right))
18085f39d397SChris Mason 		return PTR_ERR(right);
180954aa1f4dSChris Mason 
18105f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1811db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
18125f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
18135f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
18145f39d397SChris Mason 	btrfs_set_header_level(right, 0);
18155f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
18165f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
18175f39d397SChris Mason 			    BTRFS_FSID_SIZE);
18185f39d397SChris Mason 
1819d4dbff95SChris Mason 	if (mid <= slot) {
1820d4dbff95SChris Mason 		if (nritems == 1 ||
1821d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
1822d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1823d4dbff95SChris Mason 			if (slot >= nritems) {
1824d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18255f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1826d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1827db94535dSChris Mason 						  &disk_key, right->start,
1828d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
1829d4dbff95SChris Mason 				if (wret)
1830d4dbff95SChris Mason 					ret = wret;
18315f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18325f39d397SChris Mason 				path->nodes[0] = right;
1833d4dbff95SChris Mason 				path->slots[0] = 0;
1834d4dbff95SChris Mason 				path->slots[1] += 1;
1835d4dbff95SChris Mason 				return ret;
1836d4dbff95SChris Mason 			}
1837d4dbff95SChris Mason 			mid = slot;
18383326d1b0SChris Mason 			if (mid != nritems &&
18393326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
18403326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
1841d4dbff95SChris Mason 				double_split = 1;
1842d4dbff95SChris Mason 			}
18433326d1b0SChris Mason 		}
1844d4dbff95SChris Mason 	} else {
1845d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
1846d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1847d4dbff95SChris Mason 			if (slot == 0) {
1848d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18495f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1850d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1851d4dbff95SChris Mason 						  &disk_key,
1852db94535dSChris Mason 						  right->start,
1853098f59c2SChris Mason 						  path->slots[1], 1);
1854d4dbff95SChris Mason 				if (wret)
1855d4dbff95SChris Mason 					ret = wret;
18565f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18575f39d397SChris Mason 				path->nodes[0] = right;
1858d4dbff95SChris Mason 				path->slots[0] = 0;
1859a429e513SChris Mason 				if (path->slots[1] == 0) {
1860a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
1861a429e513SChris Mason 					           path, &disk_key, 1);
1862a429e513SChris Mason 					if (wret)
1863a429e513SChris Mason 						ret = wret;
1864a429e513SChris Mason 				}
1865d4dbff95SChris Mason 				return ret;
1866d4dbff95SChris Mason 			}
1867d4dbff95SChris Mason 			mid = slot;
1868d4dbff95SChris Mason 			double_split = 1;
1869d4dbff95SChris Mason 		}
1870d4dbff95SChris Mason 	}
18715f39d397SChris Mason 	nritems = nritems - mid;
18725f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
18735f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
18745f39d397SChris Mason 
18755f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
18765f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
18775f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
18785f39d397SChris Mason 
18795f39d397SChris Mason 	copy_extent_buffer(right, l,
1880d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
1881123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
1882123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
188374123bd7SChris Mason 
18845f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
18855f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
18865f39d397SChris Mason 
18875f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
18885f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
1889db94535dSChris Mason 		u32 ioff;
1890db94535dSChris Mason 
1891db94535dSChris Mason 		if (!right->map_token) {
1892db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1893db94535dSChris Mason 					sizeof(struct btrfs_item),
1894db94535dSChris Mason 					&right->map_token, &right->kaddr,
1895db94535dSChris Mason 					&right->map_start, &right->map_len,
1896db94535dSChris Mason 					KM_USER1);
1897db94535dSChris Mason 		}
1898db94535dSChris Mason 
1899db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
19005f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
19010783fcfcSChris Mason 	}
190274123bd7SChris Mason 
1903db94535dSChris Mason 	if (right->map_token) {
1904db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1905db94535dSChris Mason 		right->map_token = NULL;
1906db94535dSChris Mason 	}
1907db94535dSChris Mason 
19085f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
1909aa5d6bedSChris Mason 	ret = 0;
19105f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
1911db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
1912db94535dSChris Mason 			  path->slots[1] + 1, 1);
1913aa5d6bedSChris Mason 	if (wret)
1914aa5d6bedSChris Mason 		ret = wret;
19155f39d397SChris Mason 
19165f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
19175f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
1918eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
19195f39d397SChris Mason 
1920be0e5c09SChris Mason 	if (mid <= slot) {
19215f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19225f39d397SChris Mason 		path->nodes[0] = right;
1923be0e5c09SChris Mason 		path->slots[0] -= mid;
1924be0e5c09SChris Mason 		path->slots[1] += 1;
1925eb60ceacSChris Mason 	} else
19265f39d397SChris Mason 		free_extent_buffer(right);
19275f39d397SChris Mason 
1928eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1929d4dbff95SChris Mason 
19303326d1b0SChris Mason 	if (!double_split) {
1931d4dbff95SChris Mason 		return ret;
19323326d1b0SChris Mason 	}
193354aa1f4dSChris Mason 
1934db94535dSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
1935db94535dSChris Mason 				       l->start, 0);
19365f39d397SChris Mason 	if (IS_ERR(right))
19375f39d397SChris Mason 		return PTR_ERR(right);
19385f39d397SChris Mason 
19395f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1940db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
19415f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
19425f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
19435f39d397SChris Mason 	btrfs_set_header_level(right, 0);
19445f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
19455f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
19465f39d397SChris Mason 			    BTRFS_FSID_SIZE);
19475f39d397SChris Mason 
1948d4dbff95SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, ins_key);
19495f39d397SChris Mason 	btrfs_set_header_nritems(right, 0);
1950d4dbff95SChris Mason 	wret = insert_ptr(trans, root, path,
1951db94535dSChris Mason 			  &disk_key, right->start,
1952d4dbff95SChris Mason 			  path->slots[1], 1);
1953d4dbff95SChris Mason 	if (wret)
1954d4dbff95SChris Mason 		ret = wret;
1955a429e513SChris Mason 	if (path->slots[1] == 0) {
1956a429e513SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1957a429e513SChris Mason 		if (wret)
1958a429e513SChris Mason 			ret = wret;
1959a429e513SChris Mason 	}
19605f39d397SChris Mason 	free_extent_buffer(path->nodes[0]);
19615f39d397SChris Mason 	path->nodes[0] = right;
1962d4dbff95SChris Mason 	path->slots[0] = 0;
1963be0e5c09SChris Mason 	return ret;
1964be0e5c09SChris Mason }
1965be0e5c09SChris Mason 
1966b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
1967b18c6685SChris Mason 			struct btrfs_root *root,
1968b18c6685SChris Mason 			struct btrfs_path *path,
1969b18c6685SChris Mason 			u32 new_size)
1970b18c6685SChris Mason {
1971b18c6685SChris Mason 	int ret = 0;
1972b18c6685SChris Mason 	int slot;
1973b18c6685SChris Mason 	int slot_orig;
19745f39d397SChris Mason 	struct extent_buffer *leaf;
19755f39d397SChris Mason 	struct btrfs_item *item;
1976b18c6685SChris Mason 	u32 nritems;
1977b18c6685SChris Mason 	unsigned int data_end;
1978b18c6685SChris Mason 	unsigned int old_data_start;
1979b18c6685SChris Mason 	unsigned int old_size;
1980b18c6685SChris Mason 	unsigned int size_diff;
1981b18c6685SChris Mason 	int i;
1982b18c6685SChris Mason 
1983b18c6685SChris Mason 	slot_orig = path->slots[0];
19845f39d397SChris Mason 	leaf = path->nodes[0];
1985b18c6685SChris Mason 
19865f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
1987b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
1988b18c6685SChris Mason 
1989b18c6685SChris Mason 	slot = path->slots[0];
19905f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
19915f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
1992b18c6685SChris Mason 	BUG_ON(old_size <= new_size);
1993b18c6685SChris Mason 	size_diff = old_size - new_size;
1994b18c6685SChris Mason 
1995b18c6685SChris Mason 	BUG_ON(slot < 0);
1996b18c6685SChris Mason 	BUG_ON(slot >= nritems);
1997b18c6685SChris Mason 
1998b18c6685SChris Mason 	/*
1999b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2000b18c6685SChris Mason 	 */
2001b18c6685SChris Mason 	/* first correct the data pointers */
2002b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
20035f39d397SChris Mason 		u32 ioff;
20045f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2005db94535dSChris Mason 
2006db94535dSChris Mason 		if (!leaf->map_token) {
2007db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2008db94535dSChris Mason 					sizeof(struct btrfs_item),
2009db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2010db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2011db94535dSChris Mason 					KM_USER1);
2012db94535dSChris Mason 		}
2013db94535dSChris Mason 
20145f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
20155f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2016b18c6685SChris Mason 	}
2017db94535dSChris Mason 
2018db94535dSChris Mason 	if (leaf->map_token) {
2019db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2020db94535dSChris Mason 		leaf->map_token = NULL;
2021db94535dSChris Mason 	}
2022db94535dSChris Mason 
2023b18c6685SChris Mason 	/* shift the data */
20245f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2025b18c6685SChris Mason 		      data_end + size_diff, btrfs_leaf_data(leaf) +
2026b18c6685SChris Mason 		      data_end, old_data_start + new_size - data_end);
20275f39d397SChris Mason 
20285f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
20295f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
20305f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2031b18c6685SChris Mason 
2032b18c6685SChris Mason 	ret = 0;
20335f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
20345f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2035b18c6685SChris Mason 		BUG();
20365f39d397SChris Mason 	}
2037b18c6685SChris Mason 	return ret;
2038b18c6685SChris Mason }
2039b18c6685SChris Mason 
20405f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
20415f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
20425f39d397SChris Mason 		      u32 data_size)
20436567e837SChris Mason {
20446567e837SChris Mason 	int ret = 0;
20456567e837SChris Mason 	int slot;
20466567e837SChris Mason 	int slot_orig;
20475f39d397SChris Mason 	struct extent_buffer *leaf;
20485f39d397SChris Mason 	struct btrfs_item *item;
20496567e837SChris Mason 	u32 nritems;
20506567e837SChris Mason 	unsigned int data_end;
20516567e837SChris Mason 	unsigned int old_data;
20526567e837SChris Mason 	unsigned int old_size;
20536567e837SChris Mason 	int i;
20546567e837SChris Mason 
20556567e837SChris Mason 	slot_orig = path->slots[0];
20565f39d397SChris Mason 	leaf = path->nodes[0];
20576567e837SChris Mason 
20585f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
20596567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
20606567e837SChris Mason 
20615f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
20625f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
20636567e837SChris Mason 		BUG();
20645f39d397SChris Mason 	}
20656567e837SChris Mason 	slot = path->slots[0];
20665f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
20676567e837SChris Mason 
20686567e837SChris Mason 	BUG_ON(slot < 0);
20693326d1b0SChris Mason 	if (slot >= nritems) {
20703326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
20713326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
20723326d1b0SChris Mason 		BUG_ON(1);
20733326d1b0SChris Mason 	}
20746567e837SChris Mason 
20756567e837SChris Mason 	/*
20766567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
20776567e837SChris Mason 	 */
20786567e837SChris Mason 	/* first correct the data pointers */
20796567e837SChris Mason 	for (i = slot; i < nritems; i++) {
20805f39d397SChris Mason 		u32 ioff;
20815f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2082db94535dSChris Mason 
2083db94535dSChris Mason 		if (!leaf->map_token) {
2084db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2085db94535dSChris Mason 					sizeof(struct btrfs_item),
2086db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2087db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2088db94535dSChris Mason 					KM_USER1);
2089db94535dSChris Mason 		}
20905f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
20915f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
20926567e837SChris Mason 	}
20935f39d397SChris Mason 
2094db94535dSChris Mason 	if (leaf->map_token) {
2095db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2096db94535dSChris Mason 		leaf->map_token = NULL;
2097db94535dSChris Mason 	}
2098db94535dSChris Mason 
20996567e837SChris Mason 	/* shift the data */
21005f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
21016567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
21026567e837SChris Mason 		      data_end, old_data - data_end);
21035f39d397SChris Mason 
21046567e837SChris Mason 	data_end = old_data;
21055f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
21065f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
21075f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
21085f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
21096567e837SChris Mason 
21106567e837SChris Mason 	ret = 0;
21115f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
21125f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
21136567e837SChris Mason 		BUG();
21145f39d397SChris Mason 	}
21156567e837SChris Mason 	return ret;
21166567e837SChris Mason }
21176567e837SChris Mason 
211874123bd7SChris Mason /*
211974123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
212074123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
212174123bd7SChris Mason  */
21225f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
21235f39d397SChris Mason 			    struct btrfs_root *root,
21245f39d397SChris Mason 			    struct btrfs_path *path,
21255f39d397SChris Mason 			    struct btrfs_key *cpu_key, u32 data_size)
2126be0e5c09SChris Mason {
21275f39d397SChris Mason 	struct extent_buffer *leaf;
21285f39d397SChris Mason 	struct btrfs_item *item;
2129aa5d6bedSChris Mason 	int ret = 0;
2130be0e5c09SChris Mason 	int slot;
2131eb60ceacSChris Mason 	int slot_orig;
21327518a238SChris Mason 	u32 nritems;
2133be0e5c09SChris Mason 	unsigned int data_end;
2134e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2135e2fa7227SChris Mason 
2136e2fa7227SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2137be0e5c09SChris Mason 
213874123bd7SChris Mason 	/* create a root if there isn't one */
21395c680ed6SChris Mason 	if (!root->node)
2140cfaa7295SChris Mason 		BUG();
21415f39d397SChris Mason 
2142e089f05cSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
2143eb60ceacSChris Mason 	if (ret == 0) {
2144f0930a37SChris Mason 		return -EEXIST;
2145aa5d6bedSChris Mason 	}
2146ed2ff2cbSChris Mason 	if (ret < 0)
2147ed2ff2cbSChris Mason 		goto out;
2148be0e5c09SChris Mason 
214962e2749eSChris Mason 	slot_orig = path->slots[0];
21505f39d397SChris Mason 	leaf = path->nodes[0];
215174123bd7SChris Mason 
21525f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2153123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2154eb60ceacSChris Mason 
2155123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
2156d4dbff95SChris Mason 	    sizeof(struct btrfs_item) + data_size) {
21573326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
21583326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
21593326d1b0SChris Mason 		       data_size, btrfs_leaf_free_space(root, leaf));
2160be0e5c09SChris Mason 		BUG();
2161d4dbff95SChris Mason 	}
21625f39d397SChris Mason 
216362e2749eSChris Mason 	slot = path->slots[0];
2164eb60ceacSChris Mason 	BUG_ON(slot < 0);
21655f39d397SChris Mason 
2166be0e5c09SChris Mason 	if (slot != nritems) {
2167be0e5c09SChris Mason 		int i;
21685f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2169be0e5c09SChris Mason 
21705f39d397SChris Mason 		if (old_data < data_end) {
21715f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
21725f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
21735f39d397SChris Mason 			       slot, old_data, data_end);
21745f39d397SChris Mason 			BUG_ON(1);
21755f39d397SChris Mason 		}
2176be0e5c09SChris Mason 		/*
2177be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2178be0e5c09SChris Mason 		 */
2179be0e5c09SChris Mason 		/* first correct the data pointers */
2180db94535dSChris Mason 		WARN_ON(leaf->map_token);
21810783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
21825f39d397SChris Mason 			u32 ioff;
2183db94535dSChris Mason 
21845f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2185db94535dSChris Mason 			if (!leaf->map_token) {
2186db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2187db94535dSChris Mason 					sizeof(struct btrfs_item),
2188db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2189db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2190db94535dSChris Mason 					KM_USER1);
2191db94535dSChris Mason 			}
2192db94535dSChris Mason 
21935f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
21945f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff - data_size);
21950783fcfcSChris Mason 		}
2196db94535dSChris Mason 		if (leaf->map_token) {
2197db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2198db94535dSChris Mason 			leaf->map_token = NULL;
2199db94535dSChris Mason 		}
2200be0e5c09SChris Mason 
2201be0e5c09SChris Mason 		/* shift the items */
22025f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
22035f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
22040783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2205be0e5c09SChris Mason 
2206be0e5c09SChris Mason 		/* shift the data */
22075f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2208d6025579SChris Mason 			      data_end - data_size, btrfs_leaf_data(leaf) +
2209be0e5c09SChris Mason 			      data_end, old_data - data_end);
2210be0e5c09SChris Mason 		data_end = old_data;
2211be0e5c09SChris Mason 	}
22125f39d397SChris Mason 
221362e2749eSChris Mason 	/* setup the item for the new data */
22145f39d397SChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
22155f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
22165f39d397SChris Mason 	btrfs_set_item_offset(leaf, item, data_end - data_size);
22175f39d397SChris Mason 	btrfs_set_item_size(leaf, item, data_size);
22185f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
22195f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2220aa5d6bedSChris Mason 
2221aa5d6bedSChris Mason 	ret = 0;
22228e19f2cdSChris Mason 	if (slot == 0)
2223e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2224aa5d6bedSChris Mason 
22255f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
22265f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2227be0e5c09SChris Mason 		BUG();
22285f39d397SChris Mason 	}
2229ed2ff2cbSChris Mason out:
223062e2749eSChris Mason 	return ret;
223162e2749eSChris Mason }
223262e2749eSChris Mason 
223362e2749eSChris Mason /*
223462e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
223562e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
223662e2749eSChris Mason  */
2237e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2238e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2239e089f05cSChris Mason 		      data_size)
224062e2749eSChris Mason {
224162e2749eSChris Mason 	int ret = 0;
22422c90e5d6SChris Mason 	struct btrfs_path *path;
22435f39d397SChris Mason 	struct extent_buffer *leaf;
22445f39d397SChris Mason 	unsigned long ptr;
224562e2749eSChris Mason 
22462c90e5d6SChris Mason 	path = btrfs_alloc_path();
22472c90e5d6SChris Mason 	BUG_ON(!path);
22482c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
224962e2749eSChris Mason 	if (!ret) {
22505f39d397SChris Mason 		leaf = path->nodes[0];
22515f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
22525f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
22535f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
225462e2749eSChris Mason 	}
22552c90e5d6SChris Mason 	btrfs_free_path(path);
2256aa5d6bedSChris Mason 	return ret;
2257be0e5c09SChris Mason }
2258be0e5c09SChris Mason 
225974123bd7SChris Mason /*
22605de08d7dSChris Mason  * delete the pointer from a given node.
226174123bd7SChris Mason  *
226274123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
226374123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
226474123bd7SChris Mason  * a leaf if all the nodes are emptied.
226574123bd7SChris Mason  */
2266e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2267e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2268be0e5c09SChris Mason {
22695f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
22707518a238SChris Mason 	u32 nritems;
2271aa5d6bedSChris Mason 	int ret = 0;
2272bb803951SChris Mason 	int wret;
2273be0e5c09SChris Mason 
22745f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2275be0e5c09SChris Mason 	if (slot != nritems -1) {
22765f39d397SChris Mason 		memmove_extent_buffer(parent,
22775f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
22785f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2279d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2280d6025579SChris Mason 			      (nritems - slot - 1));
2281be0e5c09SChris Mason 	}
22827518a238SChris Mason 	nritems--;
22835f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
22847518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
22855f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2286eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
22875f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2288bb803951SChris Mason 	} else if (slot == 0) {
22895f39d397SChris Mason 		struct btrfs_disk_key disk_key;
22905f39d397SChris Mason 
22915f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
22925f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
22930f70abe2SChris Mason 		if (wret)
22940f70abe2SChris Mason 			ret = wret;
2295be0e5c09SChris Mason 	}
2296d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2297aa5d6bedSChris Mason 	return ret;
2298be0e5c09SChris Mason }
2299be0e5c09SChris Mason 
230074123bd7SChris Mason /*
230174123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
230274123bd7SChris Mason  * the leaf, remove it from the tree
230374123bd7SChris Mason  */
2304e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2305e089f05cSChris Mason 		   struct btrfs_path *path)
2306be0e5c09SChris Mason {
2307be0e5c09SChris Mason 	int slot;
23085f39d397SChris Mason 	struct extent_buffer *leaf;
23095f39d397SChris Mason 	struct btrfs_item *item;
2310be0e5c09SChris Mason 	int doff;
2311be0e5c09SChris Mason 	int dsize;
2312aa5d6bedSChris Mason 	int ret = 0;
2313aa5d6bedSChris Mason 	int wret;
23147518a238SChris Mason 	u32 nritems;
2315be0e5c09SChris Mason 
23165f39d397SChris Mason 	leaf = path->nodes[0];
23174920c9acSChris Mason 	slot = path->slots[0];
23185f39d397SChris Mason 	doff = btrfs_item_offset_nr(leaf, slot);
23195f39d397SChris Mason 	dsize = btrfs_item_size_nr(leaf, slot);
23205f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2321be0e5c09SChris Mason 
23227518a238SChris Mason 	if (slot != nritems - 1) {
2323be0e5c09SChris Mason 		int i;
2324123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
23255f39d397SChris Mason 
23265f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2327d6025579SChris Mason 			      data_end + dsize,
2328123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
2329be0e5c09SChris Mason 			      doff - data_end);
23305f39d397SChris Mason 
23310783fcfcSChris Mason 		for (i = slot + 1; i < nritems; i++) {
23325f39d397SChris Mason 			u32 ioff;
2333db94535dSChris Mason 
23345f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2335db94535dSChris Mason 			if (!leaf->map_token) {
2336db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2337db94535dSChris Mason 					sizeof(struct btrfs_item),
2338db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2339db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2340db94535dSChris Mason 					KM_USER1);
2341db94535dSChris Mason 			}
23425f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
23435f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
23440783fcfcSChris Mason 		}
2345db94535dSChris Mason 
2346db94535dSChris Mason 		if (leaf->map_token) {
2347db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2348db94535dSChris Mason 			leaf->map_token = NULL;
2349db94535dSChris Mason 		}
2350db94535dSChris Mason 
23515f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
23525f39d397SChris Mason 			      btrfs_item_nr_offset(slot + 1),
23530783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
23547518a238SChris Mason 			      (nritems - slot - 1));
2355be0e5c09SChris Mason 	}
23565f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems - 1);
23577518a238SChris Mason 	nritems--;
23585f39d397SChris Mason 
235974123bd7SChris Mason 	/* delete the leaf if we've emptied it */
23607518a238SChris Mason 	if (nritems == 0) {
23615f39d397SChris Mason 		if (leaf == root->node) {
23625f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
23639a8dd150SChris Mason 		} else {
23645f39d397SChris Mason 			clean_tree_block(trans, root, leaf);
23655f39d397SChris Mason 			wait_on_tree_block_writeback(root, leaf);
2366e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2367aa5d6bedSChris Mason 			if (wret)
2368aa5d6bedSChris Mason 				ret = wret;
2369e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
2370db94535dSChris Mason 						 leaf->start, leaf->len, 1);
23710f70abe2SChris Mason 			if (wret)
23720f70abe2SChris Mason 				ret = wret;
23739a8dd150SChris Mason 		}
2374be0e5c09SChris Mason 	} else {
23757518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2376aa5d6bedSChris Mason 		if (slot == 0) {
23775f39d397SChris Mason 			struct btrfs_disk_key disk_key;
23785f39d397SChris Mason 
23795f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2380e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
23815f39d397SChris Mason 					      &disk_key, 1);
2382aa5d6bedSChris Mason 			if (wret)
2383aa5d6bedSChris Mason 				ret = wret;
2384aa5d6bedSChris Mason 		}
2385aa5d6bedSChris Mason 
238674123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
23877936ca38SChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
2388be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2389be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2390be0e5c09SChris Mason 			 * for possible call to del_ptr below
2391be0e5c09SChris Mason 			 */
23924920c9acSChris Mason 			slot = path->slots[1];
23935f39d397SChris Mason 			extent_buffer_get(leaf);
23945f39d397SChris Mason 
2395*3685f791SChris Mason 			wret = push_leaf_right(trans, root, path, 1);
239654aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2397aa5d6bedSChris Mason 				ret = wret;
23985f39d397SChris Mason 
23995f39d397SChris Mason 			if (path->nodes[0] == leaf &&
24005f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
2401*3685f791SChris Mason 				wret = push_leaf_left(trans, root, path, 1);
240254aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2403aa5d6bedSChris Mason 					ret = wret;
2404aa5d6bedSChris Mason 			}
24055f39d397SChris Mason 
24065f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
2407db94535dSChris Mason 				u64 bytenr = leaf->start;
2408db94535dSChris Mason 				u32 blocksize = leaf->len;
24095f39d397SChris Mason 
24105f39d397SChris Mason 				clean_tree_block(trans, root, leaf);
24115f39d397SChris Mason 				wait_on_tree_block_writeback(root, leaf);
24125f39d397SChris Mason 
2413e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2414aa5d6bedSChris Mason 				if (wret)
2415aa5d6bedSChris Mason 					ret = wret;
24165f39d397SChris Mason 
24175f39d397SChris Mason 				free_extent_buffer(leaf);
2418db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
2419db94535dSChris Mason 							 blocksize, 1);
24200f70abe2SChris Mason 				if (wret)
24210f70abe2SChris Mason 					ret = wret;
24225de08d7dSChris Mason 			} else {
24235f39d397SChris Mason 				btrfs_mark_buffer_dirty(leaf);
24245f39d397SChris Mason 				free_extent_buffer(leaf);
2425be0e5c09SChris Mason 			}
2426d5719762SChris Mason 		} else {
24275f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2428be0e5c09SChris Mason 		}
24299a8dd150SChris Mason 	}
2430aa5d6bedSChris Mason 	return ret;
24319a8dd150SChris Mason }
24329a8dd150SChris Mason 
243397571fd0SChris Mason /*
243497571fd0SChris Mason  * walk up the tree as far as required to find the next leaf.
24350f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
24360f70abe2SChris Mason  * returns < 0 on io errors.
243797571fd0SChris Mason  */
2438234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2439d97e63b6SChris Mason {
2440d97e63b6SChris Mason 	int slot;
2441d97e63b6SChris Mason 	int level = 1;
2442db94535dSChris Mason 	u64 bytenr;
24435f39d397SChris Mason 	struct extent_buffer *c;
24445f39d397SChris Mason 	struct extent_buffer *next = NULL;
2445d97e63b6SChris Mason 
2446234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
2447d97e63b6SChris Mason 		if (!path->nodes[level])
24480f70abe2SChris Mason 			return 1;
24495f39d397SChris Mason 
2450d97e63b6SChris Mason 		slot = path->slots[level] + 1;
2451d97e63b6SChris Mason 		c = path->nodes[level];
24525f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
2453d97e63b6SChris Mason 			level++;
2454d97e63b6SChris Mason 			continue;
2455d97e63b6SChris Mason 		}
24565f39d397SChris Mason 
2457db94535dSChris Mason 		bytenr = btrfs_node_blockptr(c, slot);
2458cfaa7295SChris Mason 		if (next)
24595f39d397SChris Mason 			free_extent_buffer(next);
24605f39d397SChris Mason 
24616702ed49SChris Mason 		if (path->reada)
24626702ed49SChris Mason 			reada_for_search(root, path, level, slot);
24635f39d397SChris Mason 
2464db94535dSChris Mason 		next = read_tree_block(root, bytenr,
2465db94535dSChris Mason 				       btrfs_level_size(root, level -1));
2466d97e63b6SChris Mason 		break;
2467d97e63b6SChris Mason 	}
2468d97e63b6SChris Mason 	path->slots[level] = slot;
2469d97e63b6SChris Mason 	while(1) {
2470d97e63b6SChris Mason 		level--;
2471d97e63b6SChris Mason 		c = path->nodes[level];
24725f39d397SChris Mason 		free_extent_buffer(c);
2473d97e63b6SChris Mason 		path->nodes[level] = next;
2474d97e63b6SChris Mason 		path->slots[level] = 0;
2475d97e63b6SChris Mason 		if (!level)
2476d97e63b6SChris Mason 			break;
24776702ed49SChris Mason 		if (path->reada)
247832020611SYan 			reada_for_search(root, path, level, 0);
2479db94535dSChris Mason 		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2480db94535dSChris Mason 				       btrfs_level_size(root, level - 1));
2481d97e63b6SChris Mason 	}
2482d97e63b6SChris Mason 	return 0;
2483d97e63b6SChris Mason }
2484