xref: /openbmc/linux/fs/btrfs/ctree.c (revision 34a3821873aeabff2607c8093bce82cd1fbcfd60)
16cbd5570SChris Mason /*
26cbd5570SChris Mason  * Copyright (C) 2007 Oracle.  All rights reserved.
36cbd5570SChris Mason  *
46cbd5570SChris Mason  * This program is free software; you can redistribute it and/or
56cbd5570SChris Mason  * modify it under the terms of the GNU General Public
66cbd5570SChris Mason  * License v2 as published by the Free Software Foundation.
76cbd5570SChris Mason  *
86cbd5570SChris Mason  * This program is distributed in the hope that it will be useful,
96cbd5570SChris Mason  * but WITHOUT ANY WARRANTY; without even the implied warranty of
106cbd5570SChris Mason  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
116cbd5570SChris Mason  * General Public License for more details.
126cbd5570SChris Mason  *
136cbd5570SChris Mason  * You should have received a copy of the GNU General Public
146cbd5570SChris Mason  * License along with this program; if not, write to the
156cbd5570SChris Mason  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
166cbd5570SChris Mason  * Boston, MA 021110-1307, USA.
176cbd5570SChris Mason  */
186cbd5570SChris Mason 
19a6b6e75eSChris Mason #include <linux/sched.h>
20eb60ceacSChris Mason #include "ctree.h"
21eb60ceacSChris Mason #include "disk-io.h"
227f5c1516SChris Mason #include "transaction.h"
235f39d397SChris Mason #include "print-tree.h"
249a8dd150SChris Mason 
25e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
26e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
27e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
28d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
29cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend);
305f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
315f39d397SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
325f39d397SChris Mason 			  struct extent_buffer *src);
335f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
345f39d397SChris Mason 			      struct btrfs_root *root,
355f39d397SChris Mason 			      struct extent_buffer *dst_buf,
365f39d397SChris Mason 			      struct extent_buffer *src_buf);
37e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
38e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot);
39d97e63b6SChris Mason 
40df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p)
41df24a2b9SChris Mason {
42df24a2b9SChris Mason 	memset(p, 0, sizeof(*p));
43df24a2b9SChris Mason }
44df24a2b9SChris Mason 
452c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
462c90e5d6SChris Mason {
47df24a2b9SChris Mason 	struct btrfs_path *path;
48df24a2b9SChris Mason 	path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
492cc58cf2SChris Mason 	if (path) {
50df24a2b9SChris Mason 		btrfs_init_path(path);
512cc58cf2SChris Mason 		path->reada = 1;
522cc58cf2SChris Mason 	}
53df24a2b9SChris Mason 	return path;
542c90e5d6SChris Mason }
552c90e5d6SChris Mason 
562c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
572c90e5d6SChris Mason {
58df24a2b9SChris Mason 	btrfs_release_path(NULL, p);
592c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
602c90e5d6SChris Mason }
612c90e5d6SChris Mason 
62234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
63eb60ceacSChris Mason {
64eb60ceacSChris Mason 	int i;
65234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
66eb60ceacSChris Mason 		if (!p->nodes[i])
67eb60ceacSChris Mason 			break;
685f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
69eb60ceacSChris Mason 	}
70aa5d6bedSChris Mason 	memset(p, 0, sizeof(*p));
71eb60ceacSChris Mason }
72eb60ceacSChris Mason 
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 
164081e9573SChris Mason /*
165081e9573SChris Mason  * compare two keys in a memcmp fashion
166081e9573SChris Mason  */
167081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
168081e9573SChris Mason {
169081e9573SChris Mason 	struct btrfs_key k1;
170081e9573SChris Mason 
171081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
172081e9573SChris Mason 
173081e9573SChris Mason 	if (k1.objectid > k2->objectid)
174081e9573SChris Mason 		return 1;
175081e9573SChris Mason 	if (k1.objectid < k2->objectid)
176081e9573SChris Mason 		return -1;
177081e9573SChris Mason 	if (k1.type > k2->type)
178081e9573SChris Mason 		return 1;
179081e9573SChris Mason 	if (k1.type < k2->type)
180081e9573SChris Mason 		return -1;
181081e9573SChris Mason 	if (k1.offset > k2->offset)
182081e9573SChris Mason 		return 1;
183081e9573SChris Mason 	if (k1.offset < k2->offset)
184081e9573SChris Mason 		return -1;
185081e9573SChris Mason 	return 0;
186081e9573SChris Mason }
187081e9573SChris Mason 
188081e9573SChris Mason 
1896702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
1905f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
191a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
192a6b6e75eSChris Mason 		       struct btrfs_key *progress)
1936702ed49SChris Mason {
1946b80053dSChris Mason 	struct extent_buffer *cur;
1956b80053dSChris Mason 	struct extent_buffer *tmp;
1966702ed49SChris Mason 	u64 blocknr;
197e9d0b13bSChris Mason 	u64 search_start = *last_ret;
198e9d0b13bSChris Mason 	u64 last_block = 0;
1996702ed49SChris Mason 	u64 other;
2006702ed49SChris Mason 	u32 parent_nritems;
2016702ed49SChris Mason 	int end_slot;
2026702ed49SChris Mason 	int i;
2036702ed49SChris Mason 	int err = 0;
204f2183bdeSChris Mason 	int parent_level;
2056b80053dSChris Mason 	int uptodate;
2066b80053dSChris Mason 	u32 blocksize;
207081e9573SChris Mason 	int progress_passed = 0;
208081e9573SChris Mason 	struct btrfs_disk_key disk_key;
2096702ed49SChris Mason 
2105708b959SChris Mason 	parent_level = btrfs_header_level(parent);
2115708b959SChris Mason 	if (cache_only && parent_level != 1)
2125708b959SChris Mason 		return 0;
2135708b959SChris Mason 
2146702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
2156702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2166702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
2176702ed49SChris Mason 		WARN_ON(1);
2186702ed49SChris Mason 	}
2196702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
2206702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
2216702ed49SChris Mason 		       root->fs_info->generation);
2226702ed49SChris Mason 		WARN_ON(1);
2236702ed49SChris Mason 	}
22486479a04SChris Mason 
2256b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
2266b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
2276702ed49SChris Mason 	end_slot = parent_nritems;
2286702ed49SChris Mason 
2296702ed49SChris Mason 	if (parent_nritems == 1)
2306702ed49SChris Mason 		return 0;
2316702ed49SChris Mason 
2326702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
2336702ed49SChris Mason 		int close = 1;
234a6b6e75eSChris Mason 
2355708b959SChris Mason 		if (!parent->map_token) {
2365708b959SChris Mason 			map_extent_buffer(parent,
2375708b959SChris Mason 					btrfs_node_key_ptr_offset(i),
2385708b959SChris Mason 					sizeof(struct btrfs_key_ptr),
2395708b959SChris Mason 					&parent->map_token, &parent->kaddr,
2405708b959SChris Mason 					&parent->map_start, &parent->map_len,
2415708b959SChris Mason 					KM_USER1);
2425708b959SChris Mason 		}
243081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
244081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
245081e9573SChris Mason 			continue;
246081e9573SChris Mason 
247081e9573SChris Mason 		progress_passed = 1;
2486b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
249e9d0b13bSChris Mason 		if (last_block == 0)
250e9d0b13bSChris Mason 			last_block = blocknr;
2515708b959SChris Mason 
2526702ed49SChris Mason 		if (i > 0) {
2536b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
2546b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2556702ed49SChris Mason 		}
2565708b959SChris Mason 		if (close && i < end_slot - 2) {
2576b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
2586b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
2596702ed49SChris Mason 		}
260e9d0b13bSChris Mason 		if (close) {
261e9d0b13bSChris Mason 			last_block = blocknr;
2626702ed49SChris Mason 			continue;
263e9d0b13bSChris Mason 		}
2645708b959SChris Mason 		if (parent->map_token) {
2655708b959SChris Mason 			unmap_extent_buffer(parent, parent->map_token,
2665708b959SChris Mason 					    KM_USER1);
2675708b959SChris Mason 			parent->map_token = NULL;
2685708b959SChris Mason 		}
2696702ed49SChris Mason 
2706b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
2716b80053dSChris Mason 		if (cur)
2726b80053dSChris Mason 			uptodate = btrfs_buffer_uptodate(cur);
2736b80053dSChris Mason 		else
2746b80053dSChris Mason 			uptodate = 0;
2755708b959SChris Mason 		if (!cur || !uptodate) {
2766702ed49SChris Mason 			if (cache_only) {
2776b80053dSChris Mason 				free_extent_buffer(cur);
2786702ed49SChris Mason 				continue;
2796702ed49SChris Mason 			}
2806b80053dSChris Mason 			if (!cur) {
2816b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
2826b80053dSChris Mason 							 blocksize);
2836b80053dSChris Mason 			} else if (!uptodate) {
2846b80053dSChris Mason 				btrfs_read_buffer(cur);
2856702ed49SChris Mason 			}
286f2183bdeSChris Mason 		}
287e9d0b13bSChris Mason 		if (search_start == 0)
2886b80053dSChris Mason 			search_start = last_block;
289e9d0b13bSChris Mason 
2906b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
2916b80053dSChris Mason 					&tmp, search_start,
2926b80053dSChris Mason 					min(16 * blocksize,
2936b80053dSChris Mason 					    (end_slot - i) * blocksize));
294252c38f0SYan 		if (err) {
2956b80053dSChris Mason 			free_extent_buffer(cur);
2966702ed49SChris Mason 			break;
297252c38f0SYan 		}
2986b80053dSChris Mason 		search_start = tmp->start;
2995708b959SChris Mason 		last_block = tmp->start;
300f2183bdeSChris Mason 		*last_ret = search_start;
301f2183bdeSChris Mason 		if (parent_level == 1)
3026b80053dSChris Mason 			btrfs_clear_buffer_defrag(tmp);
3036b80053dSChris Mason 		free_extent_buffer(tmp);
3046702ed49SChris Mason 	}
3055708b959SChris Mason 	if (parent->map_token) {
3065708b959SChris Mason 		unmap_extent_buffer(parent, parent->map_token,
3075708b959SChris Mason 				    KM_USER1);
3085708b959SChris Mason 		parent->map_token = NULL;
3095708b959SChris Mason 	}
3106702ed49SChris Mason 	return err;
3116702ed49SChris Mason }
3126702ed49SChris Mason 
31374123bd7SChris Mason /*
31474123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
31574123bd7SChris Mason  * this returns the address of the start of the last item,
31674123bd7SChris Mason  * which is the stop of the leaf data stack
31774123bd7SChris Mason  */
318123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
3195f39d397SChris Mason 					 struct extent_buffer *leaf)
320be0e5c09SChris Mason {
3215f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
322be0e5c09SChris Mason 	if (nr == 0)
323123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
3245f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
325be0e5c09SChris Mason }
326be0e5c09SChris Mason 
327123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
328123abc88SChris Mason 		      int level)
329aa5d6bedSChris Mason {
3305f39d397SChris Mason 	struct extent_buffer *parent = NULL;
3315f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
3325f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3335f39d397SChris Mason 	struct btrfs_disk_key node_key;
334aa5d6bedSChris Mason 	int parent_slot;
3358d7be552SChris Mason 	int slot;
3368d7be552SChris Mason 	struct btrfs_key cpukey;
3375f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
338aa5d6bedSChris Mason 
339aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3405f39d397SChris Mason 		parent = path->nodes[level + 1];
341a1f39630SAneesh 
3428d7be552SChris Mason 	slot = path->slots[level];
3437518a238SChris Mason 	BUG_ON(nritems == 0);
3447518a238SChris Mason 	if (parent) {
345a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3465f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3475f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
3485f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
349e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
3501d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
351db94535dSChris Mason 		       btrfs_header_bytenr(node));
352aa5d6bedSChris Mason 	}
353123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
3548d7be552SChris Mason 	if (slot != 0) {
3555f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
3565f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3575f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
3588d7be552SChris Mason 	}
3598d7be552SChris Mason 	if (slot < nritems - 1) {
3605f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
3615f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
3625f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
363aa5d6bedSChris Mason 	}
364aa5d6bedSChris Mason 	return 0;
365aa5d6bedSChris Mason }
366aa5d6bedSChris Mason 
367123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
368123abc88SChris Mason 		      int level)
369aa5d6bedSChris Mason {
3705f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
3715f39d397SChris Mason 	struct extent_buffer *parent = NULL;
372aa5d6bedSChris Mason 	int parent_slot;
3738d7be552SChris Mason 	struct btrfs_key cpukey;
3745f39d397SChris Mason 	struct btrfs_disk_key parent_key;
3755f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
3765f39d397SChris Mason 	int slot = path->slots[0];
3778d7be552SChris Mason 
3785f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
379aa5d6bedSChris Mason 
380aa5d6bedSChris Mason 	if (path->nodes[level + 1])
3815f39d397SChris Mason 		parent = path->nodes[level + 1];
3827518a238SChris Mason 
3837518a238SChris Mason 	if (nritems == 0)
3847518a238SChris Mason 		return 0;
3857518a238SChris Mason 
3867518a238SChris Mason 	if (parent) {
387a1f39630SAneesh 		parent_slot = path->slots[level + 1];
3885f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
3895f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
3906702ed49SChris Mason 
3915f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
392e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
3931d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
394db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
395aa5d6bedSChris Mason 	}
3965f39d397SChris Mason #if 0
3975f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
3985f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
3995f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
4005f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
4015f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4025f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
4035f39d397SChris Mason 			BUG_ON(1);
4045f39d397SChris Mason 		}
4055f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
4065f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
4075f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4085f39d397SChris Mason 			printk("slot %d offset bad\n", i);
4095f39d397SChris Mason 			BUG_ON(1);
4105f39d397SChris Mason 		}
4115f39d397SChris Mason 		if (i == 0) {
4125f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
4135f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
4145f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
4155f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4165f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
4175f39d397SChris Mason 				BUG_ON(1);
4185f39d397SChris Mason 			}
4195f39d397SChris Mason 		}
4205f39d397SChris Mason 	}
4215f39d397SChris Mason 	if (nritems > 0) {
4225f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
4235f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
4245f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
4255f39d397SChris Mason 				BUG_ON(1);
4265f39d397SChris Mason 		}
4275f39d397SChris Mason 	}
4285f39d397SChris Mason #endif
4295f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
4305f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4315f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
4325f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
4335f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4345f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
4355f39d397SChris Mason 			BUG_ON(1);
4365f39d397SChris Mason 		}
4375f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
4385f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
4395f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4405f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4415f39d397SChris Mason 			BUG_ON(1);
4425f39d397SChris Mason 		}
443aa5d6bedSChris Mason 	}
4448d7be552SChris Mason 	if (slot < nritems - 1) {
4455f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
4465f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
4475f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
4485f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
4495f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
4505f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
4515f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
4525f39d397SChris Mason 			BUG_ON(1);
453aa5d6bedSChris Mason 		}
4545f39d397SChris Mason 	}
4555f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
4565f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
457aa5d6bedSChris Mason 	return 0;
458aa5d6bedSChris Mason }
459aa5d6bedSChris Mason 
460123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path,
461123abc88SChris Mason 			int level)
462aa5d6bedSChris Mason {
463810191ffSChris Mason 	return 0;
464db94535dSChris Mason #if 0
4655f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
4665f39d397SChris Mason 
467479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
468479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
469479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
4705f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
471db94535dSChris Mason 		return 1;
4725f39d397SChris Mason 	}
473db94535dSChris Mason #endif
474aa5d6bedSChris Mason 	if (level == 0)
475123abc88SChris Mason 		return check_leaf(root, path, level);
476123abc88SChris Mason 	return check_node(root, path, level);
477aa5d6bedSChris Mason }
478aa5d6bedSChris Mason 
47974123bd7SChris Mason /*
4805f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
4815f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
4825f39d397SChris Mason  *
48374123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
48474123bd7SChris Mason  * the place where you would insert key if it is not found in
48574123bd7SChris Mason  * the array.
48674123bd7SChris Mason  *
48774123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
48874123bd7SChris Mason  */
4895f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
4905f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
491be0e5c09SChris Mason 			      int max, int *slot)
492be0e5c09SChris Mason {
493be0e5c09SChris Mason 	int low = 0;
494be0e5c09SChris Mason 	int high = max;
495be0e5c09SChris Mason 	int mid;
496be0e5c09SChris Mason 	int ret;
497479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
4985f39d397SChris Mason 	struct btrfs_disk_key unaligned;
4995f39d397SChris Mason 	unsigned long offset;
5005f39d397SChris Mason 	char *map_token = NULL;
5015f39d397SChris Mason 	char *kaddr = NULL;
5025f39d397SChris Mason 	unsigned long map_start = 0;
5035f39d397SChris Mason 	unsigned long map_len = 0;
504479965d6SChris Mason 	int err;
505be0e5c09SChris Mason 
506be0e5c09SChris Mason 	while(low < high) {
507be0e5c09SChris Mason 		mid = (low + high) / 2;
5085f39d397SChris Mason 		offset = p + mid * item_size;
5095f39d397SChris Mason 
5105f39d397SChris Mason 		if (!map_token || offset < map_start ||
5115f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
5125f39d397SChris Mason 		    map_start + map_len) {
513479965d6SChris Mason 			if (map_token) {
5145f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
515479965d6SChris Mason 				map_token = NULL;
516479965d6SChris Mason 			}
517479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
518479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
519479965d6SChris Mason 						&map_token, &kaddr,
5205f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
5215f39d397SChris Mason 
522479965d6SChris Mason 			if (!err) {
523479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
524479965d6SChris Mason 							map_start);
525479965d6SChris Mason 			} else {
5265f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
5275f39d397SChris Mason 						   offset, sizeof(unaligned));
5285f39d397SChris Mason 				tmp = &unaligned;
529479965d6SChris Mason 			}
530479965d6SChris Mason 
5315f39d397SChris Mason 		} else {
5325f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
5335f39d397SChris Mason 							map_start);
5345f39d397SChris Mason 		}
535be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
536be0e5c09SChris Mason 
537be0e5c09SChris Mason 		if (ret < 0)
538be0e5c09SChris Mason 			low = mid + 1;
539be0e5c09SChris Mason 		else if (ret > 0)
540be0e5c09SChris Mason 			high = mid;
541be0e5c09SChris Mason 		else {
542be0e5c09SChris Mason 			*slot = mid;
543479965d6SChris Mason 			if (map_token)
5445f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
545be0e5c09SChris Mason 			return 0;
546be0e5c09SChris Mason 		}
547be0e5c09SChris Mason 	}
548be0e5c09SChris Mason 	*slot = low;
5495f39d397SChris Mason 	if (map_token)
5505f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
551be0e5c09SChris Mason 	return 1;
552be0e5c09SChris Mason }
553be0e5c09SChris Mason 
55497571fd0SChris Mason /*
55597571fd0SChris Mason  * simple bin_search frontend that does the right thing for
55697571fd0SChris Mason  * leaves vs nodes
55797571fd0SChris Mason  */
5585f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
5595f39d397SChris Mason 		      int level, int *slot)
560be0e5c09SChris Mason {
5615f39d397SChris Mason 	if (level == 0) {
5625f39d397SChris Mason 		return generic_bin_search(eb,
5635f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
5640783fcfcSChris Mason 					  sizeof(struct btrfs_item),
5655f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5667518a238SChris Mason 					  slot);
567be0e5c09SChris Mason 	} else {
5685f39d397SChris Mason 		return generic_bin_search(eb,
5695f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
570123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
5715f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
5727518a238SChris Mason 					  slot);
573be0e5c09SChris Mason 	}
574be0e5c09SChris Mason 	return -1;
575be0e5c09SChris Mason }
576be0e5c09SChris Mason 
5775f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
5785f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
579bb803951SChris Mason {
580bb803951SChris Mason 	if (slot < 0)
581bb803951SChris Mason 		return NULL;
5825f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
583bb803951SChris Mason 		return NULL;
584db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
585db94535dSChris Mason 		       btrfs_level_size(root, btrfs_header_level(parent) - 1));
586bb803951SChris Mason }
587bb803951SChris Mason 
588e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
589e089f05cSChris Mason 			 *root, struct btrfs_path *path, int level)
590bb803951SChris Mason {
5915f39d397SChris Mason 	struct extent_buffer *right = NULL;
5925f39d397SChris Mason 	struct extent_buffer *mid;
5935f39d397SChris Mason 	struct extent_buffer *left = NULL;
5945f39d397SChris Mason 	struct extent_buffer *parent = NULL;
595bb803951SChris Mason 	int ret = 0;
596bb803951SChris Mason 	int wret;
597bb803951SChris Mason 	int pslot;
598bb803951SChris Mason 	int orig_slot = path->slots[level];
59954aa1f4dSChris Mason 	int err_on_enospc = 0;
60079f95c82SChris Mason 	u64 orig_ptr;
601bb803951SChris Mason 
602bb803951SChris Mason 	if (level == 0)
603bb803951SChris Mason 		return 0;
604bb803951SChris Mason 
6055f39d397SChris Mason 	mid = path->nodes[level];
6061d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
60779f95c82SChris Mason 
608234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
6095f39d397SChris Mason 		parent = path->nodes[level + 1];
610bb803951SChris Mason 	pslot = path->slots[level + 1];
611bb803951SChris Mason 
61240689478SChris Mason 	/*
61340689478SChris Mason 	 * deal with the case where there is only one pointer in the root
61440689478SChris Mason 	 * by promoting the node below to a root
61540689478SChris Mason 	 */
6165f39d397SChris Mason 	if (!parent) {
6175f39d397SChris Mason 		struct extent_buffer *child;
618bb803951SChris Mason 
6195f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
620bb803951SChris Mason 			return 0;
621bb803951SChris Mason 
622bb803951SChris Mason 		/* promote the child to a root */
6235f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
624bb803951SChris Mason 		BUG_ON(!child);
625bb803951SChris Mason 		root->node = child;
626bb803951SChris Mason 		path->nodes[level] = NULL;
6275f39d397SChris Mason 		clean_tree_block(trans, root, mid);
6285f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
629bb803951SChris Mason 		/* once for the path */
6305f39d397SChris Mason 		free_extent_buffer(mid);
631db94535dSChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
632bb803951SChris Mason 		/* once for the root ptr */
6335f39d397SChris Mason 		free_extent_buffer(mid);
634db94535dSChris Mason 		return ret;
635bb803951SChris Mason 	}
6365f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
637123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
638bb803951SChris Mason 		return 0;
639bb803951SChris Mason 
6405f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
64154aa1f4dSChris Mason 		err_on_enospc = 1;
64254aa1f4dSChris Mason 
6435f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
6445f39d397SChris Mason 	if (left) {
6455f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
6465f39d397SChris Mason 				       parent, pslot - 1, &left);
64754aa1f4dSChris Mason 		if (wret) {
64854aa1f4dSChris Mason 			ret = wret;
64954aa1f4dSChris Mason 			goto enospc;
65054aa1f4dSChris Mason 		}
6512cc58cf2SChris Mason 	}
6525f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
6535f39d397SChris Mason 	if (right) {
6545f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
6555f39d397SChris Mason 				       parent, pslot + 1, &right);
6562cc58cf2SChris Mason 		if (wret) {
6572cc58cf2SChris Mason 			ret = wret;
6582cc58cf2SChris Mason 			goto enospc;
6592cc58cf2SChris Mason 		}
6602cc58cf2SChris Mason 	}
6612cc58cf2SChris Mason 
6622cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
6635f39d397SChris Mason 	if (left) {
6645f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
6655f39d397SChris Mason 		wret = push_node_left(trans, root, left, mid);
66679f95c82SChris Mason 		if (wret < 0)
66779f95c82SChris Mason 			ret = wret;
6685f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
66954aa1f4dSChris Mason 			err_on_enospc = 1;
670bb803951SChris Mason 	}
67179f95c82SChris Mason 
67279f95c82SChris Mason 	/*
67379f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
67479f95c82SChris Mason 	 */
6755f39d397SChris Mason 	if (right) {
6765f39d397SChris Mason 		wret = push_node_left(trans, root, mid, right);
67754aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
67879f95c82SChris Mason 			ret = wret;
6795f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
680db94535dSChris Mason 			u64 bytenr = right->start;
681db94535dSChris Mason 			u32 blocksize = right->len;
682db94535dSChris Mason 
6835f39d397SChris Mason 			clean_tree_block(trans, root, right);
6845f39d397SChris Mason 			wait_on_tree_block_writeback(root, right);
6855f39d397SChris Mason 			free_extent_buffer(right);
686bb803951SChris Mason 			right = NULL;
687e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
688e089f05cSChris Mason 				       1);
689bb803951SChris Mason 			if (wret)
690bb803951SChris Mason 				ret = wret;
691db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
692db94535dSChris Mason 						 blocksize, 1);
693bb803951SChris Mason 			if (wret)
694bb803951SChris Mason 				ret = wret;
695bb803951SChris Mason 		} else {
6965f39d397SChris Mason 			struct btrfs_disk_key right_key;
6975f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
6985f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
6995f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
700bb803951SChris Mason 		}
701bb803951SChris Mason 	}
7025f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
70379f95c82SChris Mason 		/*
70479f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
70579f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
70679f95c82SChris Mason 		 * could try to delete the only pointer in this node.
70779f95c82SChris Mason 		 * So, pull some keys from the left.
70879f95c82SChris Mason 		 * There has to be a left pointer at this point because
70979f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
71079f95c82SChris Mason 		 * right
71179f95c82SChris Mason 		 */
7125f39d397SChris Mason 		BUG_ON(!left);
7135f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
71454aa1f4dSChris Mason 		if (wret < 0) {
71579f95c82SChris Mason 			ret = wret;
71654aa1f4dSChris Mason 			goto enospc;
71754aa1f4dSChris Mason 		}
71879f95c82SChris Mason 		BUG_ON(wret == 1);
71979f95c82SChris Mason 	}
7205f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
72179f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
722db94535dSChris Mason 		u64 bytenr = mid->start;
723db94535dSChris Mason 		u32 blocksize = mid->len;
7245f39d397SChris Mason 		clean_tree_block(trans, root, mid);
7255f39d397SChris Mason 		wait_on_tree_block_writeback(root, mid);
7265f39d397SChris Mason 		free_extent_buffer(mid);
727bb803951SChris Mason 		mid = NULL;
728e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
729bb803951SChris Mason 		if (wret)
730bb803951SChris Mason 			ret = wret;
731db94535dSChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
732bb803951SChris Mason 		if (wret)
733bb803951SChris Mason 			ret = wret;
73479f95c82SChris Mason 	} else {
73579f95c82SChris Mason 		/* update the parent key to reflect our changes */
7365f39d397SChris Mason 		struct btrfs_disk_key mid_key;
7375f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
7385f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
7395f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
74079f95c82SChris Mason 	}
741bb803951SChris Mason 
74279f95c82SChris Mason 	/* update the path */
7435f39d397SChris Mason 	if (left) {
7445f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
7455f39d397SChris Mason 			extent_buffer_get(left);
7465f39d397SChris Mason 			path->nodes[level] = left;
747bb803951SChris Mason 			path->slots[level + 1] -= 1;
748bb803951SChris Mason 			path->slots[level] = orig_slot;
7495f39d397SChris Mason 			if (mid)
7505f39d397SChris Mason 				free_extent_buffer(mid);
751bb803951SChris Mason 		} else {
7525f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
753bb803951SChris Mason 			path->slots[level] = orig_slot;
754bb803951SChris Mason 		}
755bb803951SChris Mason 	}
75679f95c82SChris Mason 	/* double check we haven't messed things up */
757123abc88SChris Mason 	check_block(root, path, level);
758e20d96d6SChris Mason 	if (orig_ptr !=
7595f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
76079f95c82SChris Mason 		BUG();
76154aa1f4dSChris Mason enospc:
7625f39d397SChris Mason 	if (right)
7635f39d397SChris Mason 		free_extent_buffer(right);
7645f39d397SChris Mason 	if (left)
7655f39d397SChris Mason 		free_extent_buffer(left);
766bb803951SChris Mason 	return ret;
767bb803951SChris Mason }
768bb803951SChris Mason 
769e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
770e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
771e66f709bSChris Mason 				struct btrfs_root *root,
772e66f709bSChris Mason 				struct btrfs_path *path, int level)
773e66f709bSChris Mason {
7745f39d397SChris Mason 	struct extent_buffer *right = NULL;
7755f39d397SChris Mason 	struct extent_buffer *mid;
7765f39d397SChris Mason 	struct extent_buffer *left = NULL;
7775f39d397SChris Mason 	struct extent_buffer *parent = NULL;
778e66f709bSChris Mason 	int ret = 0;
779e66f709bSChris Mason 	int wret;
780e66f709bSChris Mason 	int pslot;
781e66f709bSChris Mason 	int orig_slot = path->slots[level];
782e66f709bSChris Mason 	u64 orig_ptr;
783e66f709bSChris Mason 
784e66f709bSChris Mason 	if (level == 0)
785e66f709bSChris Mason 		return 1;
786e66f709bSChris Mason 
7875f39d397SChris Mason 	mid = path->nodes[level];
788e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
789e66f709bSChris Mason 
790e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
7915f39d397SChris Mason 		parent = path->nodes[level + 1];
792e66f709bSChris Mason 	pslot = path->slots[level + 1];
793e66f709bSChris Mason 
7945f39d397SChris Mason 	if (!parent)
795e66f709bSChris Mason 		return 1;
796e66f709bSChris Mason 
7975f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
798e66f709bSChris Mason 
799e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
8005f39d397SChris Mason 	if (left) {
801e66f709bSChris Mason 		u32 left_nr;
8025f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
80333ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
80433ade1f8SChris Mason 			wret = 1;
80533ade1f8SChris Mason 		} else {
8065f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
8075f39d397SChris Mason 					      pslot - 1, &left);
80854aa1f4dSChris Mason 			if (ret)
80954aa1f4dSChris Mason 				wret = 1;
81054aa1f4dSChris Mason 			else {
81154aa1f4dSChris Mason 				wret = push_node_left(trans, root,
8125f39d397SChris Mason 						      left, mid);
81354aa1f4dSChris Mason 			}
81433ade1f8SChris Mason 		}
815e66f709bSChris Mason 		if (wret < 0)
816e66f709bSChris Mason 			ret = wret;
817e66f709bSChris Mason 		if (wret == 0) {
8185f39d397SChris Mason 			struct btrfs_disk_key disk_key;
819e66f709bSChris Mason 			orig_slot += left_nr;
8205f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
8215f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
8225f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8235f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
8245f39d397SChris Mason 				path->nodes[level] = left;
825e66f709bSChris Mason 				path->slots[level + 1] -= 1;
826e66f709bSChris Mason 				path->slots[level] = orig_slot;
8275f39d397SChris Mason 				free_extent_buffer(mid);
828e66f709bSChris Mason 			} else {
829e66f709bSChris Mason 				orig_slot -=
8305f39d397SChris Mason 					btrfs_header_nritems(left);
831e66f709bSChris Mason 				path->slots[level] = orig_slot;
8325f39d397SChris Mason 				free_extent_buffer(left);
833e66f709bSChris Mason 			}
834e66f709bSChris Mason 			return 0;
835e66f709bSChris Mason 		}
8365f39d397SChris Mason 		free_extent_buffer(left);
837e66f709bSChris Mason 	}
8385f39d397SChris Mason 	right= read_node_slot(root, parent, pslot + 1);
839e66f709bSChris Mason 
840e66f709bSChris Mason 	/*
841e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
842e66f709bSChris Mason 	 */
8435f39d397SChris Mason 	if (right) {
84433ade1f8SChris Mason 		u32 right_nr;
8455f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
84633ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
84733ade1f8SChris Mason 			wret = 1;
84833ade1f8SChris Mason 		} else {
8495f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
8505f39d397SChris Mason 					      parent, pslot + 1,
8515f39d397SChris Mason 					      &right);
85254aa1f4dSChris Mason 			if (ret)
85354aa1f4dSChris Mason 				wret = 1;
85454aa1f4dSChris Mason 			else {
85533ade1f8SChris Mason 				wret = balance_node_right(trans, root,
8565f39d397SChris Mason 							  right, mid);
85733ade1f8SChris Mason 			}
85854aa1f4dSChris Mason 		}
859e66f709bSChris Mason 		if (wret < 0)
860e66f709bSChris Mason 			ret = wret;
861e66f709bSChris Mason 		if (wret == 0) {
8625f39d397SChris Mason 			struct btrfs_disk_key disk_key;
8635f39d397SChris Mason 
8645f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
8655f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
8665f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
8675f39d397SChris Mason 
8685f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
8695f39d397SChris Mason 				path->nodes[level] = right;
870e66f709bSChris Mason 				path->slots[level + 1] += 1;
871e66f709bSChris Mason 				path->slots[level] = orig_slot -
8725f39d397SChris Mason 					btrfs_header_nritems(mid);
8735f39d397SChris Mason 				free_extent_buffer(mid);
874e66f709bSChris Mason 			} else {
8755f39d397SChris Mason 				free_extent_buffer(right);
876e66f709bSChris Mason 			}
877e66f709bSChris Mason 			return 0;
878e66f709bSChris Mason 		}
8795f39d397SChris Mason 		free_extent_buffer(right);
880e66f709bSChris Mason 	}
881e66f709bSChris Mason 	return 1;
882e66f709bSChris Mason }
883e66f709bSChris Mason 
88474123bd7SChris Mason /*
8853c69faecSChris Mason  * readahead one full node of leaves
8863c69faecSChris Mason  */
8873c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
8886702ed49SChris Mason 			     int level, int slot)
8893c69faecSChris Mason {
8905f39d397SChris Mason 	struct extent_buffer *node;
8913c69faecSChris Mason 	u32 nritems;
8923c69faecSChris Mason 	u64 search;
8936b80053dSChris Mason 	u64 lowest_read;
8946b80053dSChris Mason 	u64 highest_read;
8956b80053dSChris Mason 	u64 nread = 0;
8963c69faecSChris Mason 	int direction = path->reada;
8975f39d397SChris Mason 	struct extent_buffer *eb;
8986b80053dSChris Mason 	u32 nr;
8996b80053dSChris Mason 	u32 blocksize;
9006b80053dSChris Mason 	u32 nscan = 0;
901db94535dSChris Mason 
902a6b6e75eSChris Mason 	if (level != 1)
9033c69faecSChris Mason 		return;
9043c69faecSChris Mason 
9056702ed49SChris Mason 	if (!path->nodes[level])
9066702ed49SChris Mason 		return;
9076702ed49SChris Mason 
9085f39d397SChris Mason 	node = path->nodes[level];
9093c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
9106b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
9116b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
9125f39d397SChris Mason 	if (eb) {
9135f39d397SChris Mason 		free_extent_buffer(eb);
9143c69faecSChris Mason 		return;
9153c69faecSChris Mason 	}
9163c69faecSChris Mason 
9176b80053dSChris Mason 	highest_read = search;
9186b80053dSChris Mason 	lowest_read = search;
9196b80053dSChris Mason 
9205f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
9216b80053dSChris Mason 	nr = slot;
9223c69faecSChris Mason 	while(1) {
9236b80053dSChris Mason 		if (direction < 0) {
9246b80053dSChris Mason 			if (nr == 0)
9253c69faecSChris Mason 				break;
9266b80053dSChris Mason 			nr--;
9276b80053dSChris Mason 		} else if (direction > 0) {
9286b80053dSChris Mason 			nr++;
9296b80053dSChris Mason 			if (nr >= nritems)
9306b80053dSChris Mason 				break;
9313c69faecSChris Mason 		}
9326b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
9336b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
9346b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
9356b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
9366b80053dSChris Mason 			readahead_tree_block(root, search, blocksize);
9376b80053dSChris Mason 			nread += blocksize;
9383c69faecSChris Mason 		}
9396b80053dSChris Mason 		nscan++;
9406b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
9416b80053dSChris Mason 			break;
9426b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
9436b80053dSChris Mason 			break;
9446b80053dSChris Mason 
9456b80053dSChris Mason 		if (search < lowest_read)
9466b80053dSChris Mason 			lowest_read = search;
9476b80053dSChris Mason 		if (search > highest_read)
9486b80053dSChris Mason 			highest_read = search;
9493c69faecSChris Mason 	}
9503c69faecSChris Mason }
9513c69faecSChris Mason /*
95274123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
95374123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
95474123bd7SChris Mason  * level of the path (level 0)
95574123bd7SChris Mason  *
95674123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
957aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
958aa5d6bedSChris Mason  * search a negative error number is returned.
95997571fd0SChris Mason  *
96097571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
96197571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
96297571fd0SChris Mason  * possible)
96374123bd7SChris Mason  */
964e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
965e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
966e089f05cSChris Mason 		      ins_len, int cow)
967be0e5c09SChris Mason {
9685f39d397SChris Mason 	struct extent_buffer *b;
969db94535dSChris Mason 	u64 bytenr;
970be0e5c09SChris Mason 	int slot;
971be0e5c09SChris Mason 	int ret;
972be0e5c09SChris Mason 	int level;
9733c69faecSChris Mason 	int should_reada = p->reada;
9749f3a7427SChris Mason 	u8 lowest_level = 0;
9759f3a7427SChris Mason 
9766702ed49SChris Mason 	lowest_level = p->lowest_level;
9776702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
97822b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
97922b0ebdaSChris Mason 	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
980bb803951SChris Mason again:
981bb803951SChris Mason 	b = root->node;
9825f39d397SChris Mason 	extent_buffer_get(b);
983eb60ceacSChris Mason 	while (b) {
9845f39d397SChris Mason 		level = btrfs_header_level(b);
98502217ed2SChris Mason 		if (cow) {
98602217ed2SChris Mason 			int wret;
987e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
988e20d96d6SChris Mason 					       p->nodes[level + 1],
989e20d96d6SChris Mason 					       p->slots[level + 1],
990252c38f0SYan 					       &b);
99154aa1f4dSChris Mason 			if (wret) {
9925f39d397SChris Mason 				free_extent_buffer(b);
99354aa1f4dSChris Mason 				return wret;
99454aa1f4dSChris Mason 			}
99502217ed2SChris Mason 		}
99602217ed2SChris Mason 		BUG_ON(!cow && ins_len);
9975f39d397SChris Mason 		if (level != btrfs_header_level(b))
9982c90e5d6SChris Mason 			WARN_ON(1);
9995f39d397SChris Mason 		level = btrfs_header_level(b);
1000eb60ceacSChris Mason 		p->nodes[level] = b;
1001123abc88SChris Mason 		ret = check_block(root, p, level);
1002aa5d6bedSChris Mason 		if (ret)
1003aa5d6bedSChris Mason 			return -1;
10045f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
10055f39d397SChris Mason 		if (level != 0) {
1006be0e5c09SChris Mason 			if (ret && slot > 0)
1007be0e5c09SChris Mason 				slot -= 1;
1008be0e5c09SChris Mason 			p->slots[level] = slot;
10095f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
1010d4dbff95SChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
1011e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
10125c680ed6SChris Mason 				BUG_ON(sret > 0);
10135c680ed6SChris Mason 				if (sret)
10145c680ed6SChris Mason 					return sret;
10155c680ed6SChris Mason 				b = p->nodes[level];
10165c680ed6SChris Mason 				slot = p->slots[level];
1017bb803951SChris Mason 			} else if (ins_len < 0) {
1018e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1019e089f05cSChris Mason 							 level);
1020bb803951SChris Mason 				if (sret)
1021bb803951SChris Mason 					return sret;
1022bb803951SChris Mason 				b = p->nodes[level];
1023f510cfecSChris Mason 				if (!b) {
1024f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1025bb803951SChris Mason 					goto again;
1026f510cfecSChris Mason 				}
1027bb803951SChris Mason 				slot = p->slots[level];
10285f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
10295c680ed6SChris Mason 			}
10309f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
10319f3a7427SChris Mason 			if (level == lowest_level)
10329f3a7427SChris Mason 				break;
1033db94535dSChris Mason 			bytenr = btrfs_node_blockptr(b, slot);
10346702ed49SChris Mason 			if (should_reada)
10356702ed49SChris Mason 				reada_for_search(root, p, level, slot);
1036db94535dSChris Mason 			b = read_tree_block(root, bytenr,
1037db94535dSChris Mason 					    btrfs_level_size(root, level - 1));
1038be0e5c09SChris Mason 		} else {
1039be0e5c09SChris Mason 			p->slots[level] = slot;
10405f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
10410783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1042d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1043cc0c5538SChris Mason 						      p, ins_len, ret == 0);
10445c680ed6SChris Mason 				BUG_ON(sret > 0);
10455c680ed6SChris Mason 				if (sret)
10465c680ed6SChris Mason 					return sret;
10475c680ed6SChris Mason 			}
1048be0e5c09SChris Mason 			return ret;
1049be0e5c09SChris Mason 		}
1050be0e5c09SChris Mason 	}
1051aa5d6bedSChris Mason 	return 1;
1052be0e5c09SChris Mason }
1053be0e5c09SChris Mason 
105474123bd7SChris Mason /*
105574123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
105674123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
105774123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
105874123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
105974123bd7SChris Mason  * higher levels
1060aa5d6bedSChris Mason  *
1061aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1062aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
106374123bd7SChris Mason  */
10645f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
10655f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
10665f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1067be0e5c09SChris Mason {
1068be0e5c09SChris Mason 	int i;
1069aa5d6bedSChris Mason 	int ret = 0;
10705f39d397SChris Mason 	struct extent_buffer *t;
10715f39d397SChris Mason 
1072234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1073be0e5c09SChris Mason 		int tslot = path->slots[i];
1074eb60ceacSChris Mason 		if (!path->nodes[i])
1075be0e5c09SChris Mason 			break;
10765f39d397SChris Mason 		t = path->nodes[i];
10775f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1078d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1079be0e5c09SChris Mason 		if (tslot != 0)
1080be0e5c09SChris Mason 			break;
1081be0e5c09SChris Mason 	}
1082aa5d6bedSChris Mason 	return ret;
1083be0e5c09SChris Mason }
1084be0e5c09SChris Mason 
108574123bd7SChris Mason /*
108674123bd7SChris Mason  * try to push data from one node into the next node left in the
108779f95c82SChris Mason  * tree.
1088aa5d6bedSChris Mason  *
1089aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1090aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
109174123bd7SChris Mason  */
1092e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
10935f39d397SChris Mason 			  *root, struct extent_buffer *dst,
10945f39d397SChris Mason 			  struct extent_buffer *src)
1095be0e5c09SChris Mason {
1096be0e5c09SChris Mason 	int push_items = 0;
1097bb803951SChris Mason 	int src_nritems;
1098bb803951SChris Mason 	int dst_nritems;
1099aa5d6bedSChris Mason 	int ret = 0;
1100be0e5c09SChris Mason 
11015f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11025f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1103123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
110454aa1f4dSChris Mason 
1105eb60ceacSChris Mason 	if (push_items <= 0) {
1106be0e5c09SChris Mason 		return 1;
1107eb60ceacSChris Mason 	}
1108be0e5c09SChris Mason 
1109be0e5c09SChris Mason 	if (src_nritems < push_items)
1110be0e5c09SChris Mason 		push_items = src_nritems;
111179f95c82SChris Mason 
11125f39d397SChris Mason 	copy_extent_buffer(dst, src,
11135f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
11145f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1115123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
11165f39d397SChris Mason 
1117bb803951SChris Mason 	if (push_items < src_nritems) {
11185f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
11195f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1120e2fa7227SChris Mason 				      (src_nritems - push_items) *
1121123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1122bb803951SChris Mason 	}
11235f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11245f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
11255f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11265f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1127bb803951SChris Mason 	return ret;
1128be0e5c09SChris Mason }
1129be0e5c09SChris Mason 
113097571fd0SChris Mason /*
113179f95c82SChris Mason  * try to push data from one node into the next node right in the
113279f95c82SChris Mason  * tree.
113379f95c82SChris Mason  *
113479f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
113579f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
113679f95c82SChris Mason  *
113779f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
113879f95c82SChris Mason  */
11395f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
11405f39d397SChris Mason 			      struct btrfs_root *root,
11415f39d397SChris Mason 			      struct extent_buffer *dst,
11425f39d397SChris Mason 			      struct extent_buffer *src)
114379f95c82SChris Mason {
114479f95c82SChris Mason 	int push_items = 0;
114579f95c82SChris Mason 	int max_push;
114679f95c82SChris Mason 	int src_nritems;
114779f95c82SChris Mason 	int dst_nritems;
114879f95c82SChris Mason 	int ret = 0;
114979f95c82SChris Mason 
11505f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
11515f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1152123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
11535f39d397SChris Mason 	if (push_items <= 0)
115479f95c82SChris Mason 		return 1;
115579f95c82SChris Mason 
115679f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
115779f95c82SChris Mason 	/* don't try to empty the node */
1158252c38f0SYan 	if (max_push >= src_nritems)
115979f95c82SChris Mason 		return 1;
1160252c38f0SYan 
116179f95c82SChris Mason 	if (max_push < push_items)
116279f95c82SChris Mason 		push_items = max_push;
116379f95c82SChris Mason 
11645f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
11655f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
11665f39d397SChris Mason 				      (dst_nritems) *
11675f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1168d6025579SChris Mason 
11695f39d397SChris Mason 	copy_extent_buffer(dst, src,
11705f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
11715f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1172123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
117379f95c82SChris Mason 
11745f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
11755f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
117679f95c82SChris Mason 
11775f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
11785f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
117979f95c82SChris Mason 	return ret;
118079f95c82SChris Mason }
118179f95c82SChris Mason 
118279f95c82SChris Mason /*
118397571fd0SChris Mason  * helper function to insert a new root level in the tree.
118497571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
118597571fd0SChris Mason  * point to the existing root
1186aa5d6bedSChris Mason  *
1187aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
118897571fd0SChris Mason  */
11895f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans,
11905f39d397SChris Mason 			   struct btrfs_root *root,
11915f39d397SChris Mason 			   struct btrfs_path *path, int level)
119274123bd7SChris Mason {
11935f39d397SChris Mason 	struct extent_buffer *lower;
11945f39d397SChris Mason 	struct extent_buffer *c;
11955f39d397SChris Mason 	struct btrfs_disk_key lower_key;
11965c680ed6SChris Mason 
11975c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
11985c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
11995c680ed6SChris Mason 
1200db94535dSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
1201db94535dSChris Mason 				   root->node->start, 0);
12025f39d397SChris Mason 	if (IS_ERR(c))
12035f39d397SChris Mason 		return PTR_ERR(c);
12045f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
12055f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
12065f39d397SChris Mason 	btrfs_set_header_level(c, level);
1207db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
12085f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
12095f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
12105f39d397SChris Mason 	lower = path->nodes[level-1];
1211d5719762SChris Mason 
12125f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
12135f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
12145f39d397SChris Mason 			    BTRFS_FSID_SIZE);
12155f39d397SChris Mason 	if (level == 1)
12165f39d397SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
12175f39d397SChris Mason 	else
12185f39d397SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
12195f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1220db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
12215f39d397SChris Mason 
12225f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1223d5719762SChris Mason 
1224cfaa7295SChris Mason 	/* the super has an extra ref to root->node */
12255f39d397SChris Mason 	free_extent_buffer(root->node);
12265f39d397SChris Mason 	root->node = c;
12275f39d397SChris Mason 	extent_buffer_get(c);
12285f39d397SChris Mason 	path->nodes[level] = c;
122974123bd7SChris Mason 	path->slots[level] = 0;
123074123bd7SChris Mason 	return 0;
123174123bd7SChris Mason }
12325c680ed6SChris Mason 
12335c680ed6SChris Mason /*
12345c680ed6SChris Mason  * worker function to insert a single pointer in a node.
12355c680ed6SChris Mason  * the node should have enough room for the pointer already
123697571fd0SChris Mason  *
12375c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
12385c680ed6SChris Mason  * blocknr is the block the key points to.
1239aa5d6bedSChris Mason  *
1240aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
12415c680ed6SChris Mason  */
1242e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1243e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1244db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
12455c680ed6SChris Mason {
12465f39d397SChris Mason 	struct extent_buffer *lower;
12475c680ed6SChris Mason 	int nritems;
12485c680ed6SChris Mason 
12495c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
12505f39d397SChris Mason 	lower = path->nodes[level];
12515f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
125274123bd7SChris Mason 	if (slot > nritems)
125374123bd7SChris Mason 		BUG();
1254123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
125574123bd7SChris Mason 		BUG();
125674123bd7SChris Mason 	if (slot != nritems) {
12575f39d397SChris Mason 		memmove_extent_buffer(lower,
12585f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
12595f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1260123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
126174123bd7SChris Mason 	}
12625f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1263db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
12645f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
12655f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
126674123bd7SChris Mason 	return 0;
126774123bd7SChris Mason }
126874123bd7SChris Mason 
126997571fd0SChris Mason /*
127097571fd0SChris Mason  * split the node at the specified level in path in two.
127197571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
127297571fd0SChris Mason  *
127397571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
127497571fd0SChris Mason  * left and right, if either one works, it returns right away.
1275aa5d6bedSChris Mason  *
1276aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
127797571fd0SChris Mason  */
1278e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1279e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1280be0e5c09SChris Mason {
12815f39d397SChris Mason 	struct extent_buffer *c;
12825f39d397SChris Mason 	struct extent_buffer *split;
12835f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1284be0e5c09SChris Mason 	int mid;
12855c680ed6SChris Mason 	int ret;
1286aa5d6bedSChris Mason 	int wret;
12877518a238SChris Mason 	u32 c_nritems;
1288be0e5c09SChris Mason 
12895f39d397SChris Mason 	c = path->nodes[level];
12905f39d397SChris Mason 	if (c == root->node) {
12915c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1292e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
12935c680ed6SChris Mason 		if (ret)
12945c680ed6SChris Mason 			return ret;
1295e66f709bSChris Mason 	} else {
1296e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
12975f39d397SChris Mason 		c = path->nodes[level];
12985f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1299e66f709bSChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1300e66f709bSChris Mason 			return 0;
130154aa1f4dSChris Mason 		if (ret < 0)
130254aa1f4dSChris Mason 			return ret;
13035c680ed6SChris Mason 	}
1304e66f709bSChris Mason 
13055f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
1306db94535dSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
1307db94535dSChris Mason 				       c->start, 0);
13085f39d397SChris Mason 	if (IS_ERR(split))
13095f39d397SChris Mason 		return PTR_ERR(split);
131054aa1f4dSChris Mason 
13115f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
13125f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1313db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
13145f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
13155f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
13165f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
13175f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
13185f39d397SChris Mason 			    BTRFS_FSID_SIZE);
13195f39d397SChris Mason 
13207518a238SChris Mason 	mid = (c_nritems + 1) / 2;
13215f39d397SChris Mason 
13225f39d397SChris Mason 	copy_extent_buffer(split, c,
13235f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
13245f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1325123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
13265f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
13275f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1328aa5d6bedSChris Mason 	ret = 0;
1329aa5d6bedSChris Mason 
13305f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
13315f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
13325f39d397SChris Mason 
13335f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1334db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
13355f39d397SChris Mason 			  path->slots[level + 1] + 1,
1336123abc88SChris Mason 			  level + 1);
1337aa5d6bedSChris Mason 	if (wret)
1338aa5d6bedSChris Mason 		ret = wret;
1339aa5d6bedSChris Mason 
13405de08d7dSChris Mason 	if (path->slots[level] >= mid) {
13415c680ed6SChris Mason 		path->slots[level] -= mid;
13425f39d397SChris Mason 		free_extent_buffer(c);
13435f39d397SChris Mason 		path->nodes[level] = split;
13445c680ed6SChris Mason 		path->slots[level + 1] += 1;
1345eb60ceacSChris Mason 	} else {
13465f39d397SChris Mason 		free_extent_buffer(split);
1347be0e5c09SChris Mason 	}
1348aa5d6bedSChris Mason 	return ret;
1349be0e5c09SChris Mason }
1350be0e5c09SChris Mason 
135174123bd7SChris Mason /*
135274123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
135374123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
135474123bd7SChris Mason  * space used both by the item structs and the item data
135574123bd7SChris Mason  */
13565f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1357be0e5c09SChris Mason {
1358be0e5c09SChris Mason 	int data_len;
13595f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1360d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1361be0e5c09SChris Mason 
1362be0e5c09SChris Mason 	if (!nr)
1363be0e5c09SChris Mason 		return 0;
13645f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
13655f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
13660783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1367d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1368be0e5c09SChris Mason 	return data_len;
1369be0e5c09SChris Mason }
1370be0e5c09SChris Mason 
137174123bd7SChris Mason /*
1372d4dbff95SChris Mason  * The space between the end of the leaf items and
1373d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1374d4dbff95SChris Mason  * the leaf has left for both items and data
1375d4dbff95SChris Mason  */
13765f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1377d4dbff95SChris Mason {
13785f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
13795f39d397SChris Mason 	int ret;
13805f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
13815f39d397SChris Mason 	if (ret < 0) {
13825f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1383ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
13845f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
13855f39d397SChris Mason 	}
13865f39d397SChris Mason 	return ret;
1387d4dbff95SChris Mason }
1388d4dbff95SChris Mason 
1389d4dbff95SChris Mason /*
139000ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
139100ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1392aa5d6bedSChris Mason  *
1393aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1394aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
139500ec4c51SChris Mason  */
1396e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1397*34a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
1398*34a38218SChris Mason 			   int empty)
139900ec4c51SChris Mason {
14005f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
14015f39d397SChris Mason 	struct extent_buffer *right;
14025f39d397SChris Mason 	struct extent_buffer *upper;
14035f39d397SChris Mason 	struct btrfs_disk_key disk_key;
140400ec4c51SChris Mason 	int slot;
1405*34a38218SChris Mason 	u32 i;
140600ec4c51SChris Mason 	int free_space;
140700ec4c51SChris Mason 	int push_space = 0;
140800ec4c51SChris Mason 	int push_items = 0;
14090783fcfcSChris Mason 	struct btrfs_item *item;
14107518a238SChris Mason 	u32 left_nritems;
1411*34a38218SChris Mason 	u32 nr;
14127518a238SChris Mason 	u32 right_nritems;
14135f39d397SChris Mason 	u32 data_end;
1414db94535dSChris Mason 	u32 this_item_size;
141554aa1f4dSChris Mason 	int ret;
141600ec4c51SChris Mason 
141700ec4c51SChris Mason 	slot = path->slots[1];
141800ec4c51SChris Mason 	if (!path->nodes[1]) {
141900ec4c51SChris Mason 		return 1;
142000ec4c51SChris Mason 	}
142100ec4c51SChris Mason 	upper = path->nodes[1];
14225f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
142300ec4c51SChris Mason 		return 1;
14245f39d397SChris Mason 
1425db94535dSChris Mason 	right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1426db94535dSChris Mason 				root->leafsize);
1427123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14280783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14295f39d397SChris Mason 		free_extent_buffer(right);
143002217ed2SChris Mason 		return 1;
143102217ed2SChris Mason 	}
143202217ed2SChris Mason 
14335f39d397SChris Mason 	/* cow and double check */
14345f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
14355f39d397SChris Mason 			      slot + 1, &right);
14365f39d397SChris Mason 	if (ret) {
14375f39d397SChris Mason 		free_extent_buffer(right);
1438a429e513SChris Mason 		return 1;
1439a429e513SChris Mason 	}
14405f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
14415f39d397SChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
14425f39d397SChris Mason 		free_extent_buffer(right);
14435f39d397SChris Mason 		return 1;
14445f39d397SChris Mason 	}
14455f39d397SChris Mason 
14465f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
14475f39d397SChris Mason 	if (left_nritems == 0) {
14485f39d397SChris Mason 		free_extent_buffer(right);
14495f39d397SChris Mason 		return 1;
14505f39d397SChris Mason 	}
14515f39d397SChris Mason 
1452*34a38218SChris Mason 	if (empty)
1453*34a38218SChris Mason 		nr = 0;
1454*34a38218SChris Mason 	else
1455*34a38218SChris Mason 		nr = 1;
1456*34a38218SChris Mason 
1457*34a38218SChris Mason 	i = left_nritems - 1;
1458*34a38218SChris Mason 	while (i >= nr) {
14595f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1460db94535dSChris Mason 
146100ec4c51SChris Mason 		if (path->slots[0] == i)
146200ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1463db94535dSChris Mason 
1464db94535dSChris Mason 		if (!left->map_token) {
1465db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1466db94535dSChris Mason 					sizeof(struct btrfs_item),
1467db94535dSChris Mason 					&left->map_token, &left->kaddr,
1468db94535dSChris Mason 					&left->map_start, &left->map_len,
1469db94535dSChris Mason 					KM_USER1);
1470db94535dSChris Mason 		}
1471db94535dSChris Mason 
1472db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1473db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
147400ec4c51SChris Mason 			break;
147500ec4c51SChris Mason 		push_items++;
1476db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1477*34a38218SChris Mason 		if (i == 0)
1478*34a38218SChris Mason 			break;
1479*34a38218SChris Mason 		i--;
1480db94535dSChris Mason 	}
1481db94535dSChris Mason 	if (left->map_token) {
1482db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1483db94535dSChris Mason 		left->map_token = NULL;
148400ec4c51SChris Mason 	}
14855f39d397SChris Mason 
148600ec4c51SChris Mason 	if (push_items == 0) {
14875f39d397SChris Mason 		free_extent_buffer(right);
148800ec4c51SChris Mason 		return 1;
148900ec4c51SChris Mason 	}
14905f39d397SChris Mason 
1491*34a38218SChris Mason 	if (!empty && push_items == left_nritems)
1492a429e513SChris Mason 		WARN_ON(1);
14935f39d397SChris Mason 
149400ec4c51SChris Mason 	/* push left to right */
14955f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
1496*34a38218SChris Mason 
14975f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1498123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
14995f39d397SChris Mason 
150000ec4c51SChris Mason 	/* make room in the right data area */
15015f39d397SChris Mason 	data_end = leaf_data_end(root, right);
15025f39d397SChris Mason 	memmove_extent_buffer(right,
15035f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
15045f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
15055f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
15065f39d397SChris Mason 
150700ec4c51SChris Mason 	/* copy from the left data area */
15085f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1509d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1510d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1511d6025579SChris Mason 		     push_space);
15125f39d397SChris Mason 
15135f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
15145f39d397SChris Mason 			      btrfs_item_nr_offset(0),
15150783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
15165f39d397SChris Mason 
151700ec4c51SChris Mason 	/* copy the items from left to right */
15185f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
15195f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
15200783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
152100ec4c51SChris Mason 
152200ec4c51SChris Mason 	/* update the item pointers */
15237518a238SChris Mason 	right_nritems += push_items;
15245f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1525123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
15267518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
15275f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1528db94535dSChris Mason 		if (!right->map_token) {
1529db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1530db94535dSChris Mason 					sizeof(struct btrfs_item),
1531db94535dSChris Mason 					&right->map_token, &right->kaddr,
1532db94535dSChris Mason 					&right->map_start, &right->map_len,
1533db94535dSChris Mason 					KM_USER1);
1534db94535dSChris Mason 		}
1535db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1536db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1537db94535dSChris Mason 	}
1538db94535dSChris Mason 
1539db94535dSChris Mason 	if (right->map_token) {
1540db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1541db94535dSChris Mason 		right->map_token = NULL;
154200ec4c51SChris Mason 	}
15437518a238SChris Mason 	left_nritems -= push_items;
15445f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
154500ec4c51SChris Mason 
1546*34a38218SChris Mason 	if (left_nritems)
15475f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
15485f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1549a429e513SChris Mason 
15505f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
15515f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1552d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
155302217ed2SChris Mason 
155400ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
15557518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
15567518a238SChris Mason 		path->slots[0] -= left_nritems;
15575f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
15585f39d397SChris Mason 		path->nodes[0] = right;
155900ec4c51SChris Mason 		path->slots[1] += 1;
156000ec4c51SChris Mason 	} else {
15615f39d397SChris Mason 		free_extent_buffer(right);
156200ec4c51SChris Mason 	}
156300ec4c51SChris Mason 	return 0;
156400ec4c51SChris Mason }
156500ec4c51SChris Mason /*
156674123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
156774123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
156874123bd7SChris Mason  */
1569e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1570*34a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
1571*34a38218SChris Mason 			  int empty)
1572be0e5c09SChris Mason {
15735f39d397SChris Mason 	struct btrfs_disk_key disk_key;
15745f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
15755f39d397SChris Mason 	struct extent_buffer *left;
1576be0e5c09SChris Mason 	int slot;
1577be0e5c09SChris Mason 	int i;
1578be0e5c09SChris Mason 	int free_space;
1579be0e5c09SChris Mason 	int push_space = 0;
1580be0e5c09SChris Mason 	int push_items = 0;
15810783fcfcSChris Mason 	struct btrfs_item *item;
15827518a238SChris Mason 	u32 old_left_nritems;
15835f39d397SChris Mason 	u32 right_nritems;
1584*34a38218SChris Mason 	u32 nr;
1585aa5d6bedSChris Mason 	int ret = 0;
1586aa5d6bedSChris Mason 	int wret;
1587db94535dSChris Mason 	u32 this_item_size;
1588db94535dSChris Mason 	u32 old_left_item_size;
1589be0e5c09SChris Mason 
1590be0e5c09SChris Mason 	slot = path->slots[1];
15915f39d397SChris Mason 	if (slot == 0)
1592be0e5c09SChris Mason 		return 1;
15935f39d397SChris Mason 	if (!path->nodes[1])
1594be0e5c09SChris Mason 		return 1;
15955f39d397SChris Mason 
15963685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
15973685f791SChris Mason 	if (right_nritems == 0) {
15983685f791SChris Mason 		return 1;
15993685f791SChris Mason 	}
16003685f791SChris Mason 
16015f39d397SChris Mason 	left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
1602db94535dSChris Mason 			       slot - 1), root->leafsize);
1603123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
16040783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
16055f39d397SChris Mason 		free_extent_buffer(left);
1606be0e5c09SChris Mason 		return 1;
1607be0e5c09SChris Mason 	}
160802217ed2SChris Mason 
160902217ed2SChris Mason 	/* cow and double check */
16105f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
16115f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
161254aa1f4dSChris Mason 	if (ret) {
161354aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
16145f39d397SChris Mason 		free_extent_buffer(left);
161554aa1f4dSChris Mason 		return 1;
161654aa1f4dSChris Mason 	}
16173685f791SChris Mason 
1618123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
16190783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
16205f39d397SChris Mason 		free_extent_buffer(left);
162102217ed2SChris Mason 		return 1;
162202217ed2SChris Mason 	}
162302217ed2SChris Mason 
1624*34a38218SChris Mason 	if (empty)
1625*34a38218SChris Mason 		nr = right_nritems;
1626*34a38218SChris Mason 	else
1627*34a38218SChris Mason 		nr = right_nritems - 1;
1628*34a38218SChris Mason 
1629*34a38218SChris Mason 	for (i = 0; i < nr; i++) {
16305f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1631db94535dSChris Mason 		if (!right->map_token) {
1632db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1633db94535dSChris Mason 					sizeof(struct btrfs_item),
1634db94535dSChris Mason 					&right->map_token, &right->kaddr,
1635db94535dSChris Mason 					&right->map_start, &right->map_len,
1636db94535dSChris Mason 					KM_USER1);
1637db94535dSChris Mason 		}
1638db94535dSChris Mason 
1639be0e5c09SChris Mason 		if (path->slots[0] == i)
1640be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
1641db94535dSChris Mason 
1642db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
1643db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
1644be0e5c09SChris Mason 			break;
1645db94535dSChris Mason 
1646be0e5c09SChris Mason 		push_items++;
1647db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
1648be0e5c09SChris Mason 	}
1649db94535dSChris Mason 
1650db94535dSChris Mason 	if (right->map_token) {
1651db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1652db94535dSChris Mason 		right->map_token = NULL;
1653db94535dSChris Mason 	}
1654db94535dSChris Mason 
1655be0e5c09SChris Mason 	if (push_items == 0) {
16565f39d397SChris Mason 		free_extent_buffer(left);
1657be0e5c09SChris Mason 		return 1;
1658be0e5c09SChris Mason 	}
1659*34a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
1660a429e513SChris Mason 		WARN_ON(1);
16615f39d397SChris Mason 
1662be0e5c09SChris Mason 	/* push data from right to left */
16635f39d397SChris Mason 	copy_extent_buffer(left, right,
16645f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
16655f39d397SChris Mason 			   btrfs_item_nr_offset(0),
16665f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
16675f39d397SChris Mason 
1668123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
16695f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
16705f39d397SChris Mason 
16715f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
1672d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
1673123abc88SChris Mason 		     btrfs_leaf_data(right) +
16745f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
1675be0e5c09SChris Mason 		     push_space);
16765f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
1677eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
1678eb60ceacSChris Mason 
1679db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
1680be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
16815f39d397SChris Mason 		u32 ioff;
1682db94535dSChris Mason 
16835f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1684db94535dSChris Mason 		if (!left->map_token) {
1685db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1686db94535dSChris Mason 					sizeof(struct btrfs_item),
1687db94535dSChris Mason 					&left->map_token, &left->kaddr,
1688db94535dSChris Mason 					&left->map_start, &left->map_len,
1689db94535dSChris Mason 					KM_USER1);
1690db94535dSChris Mason 		}
1691db94535dSChris Mason 
16925f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
16935f39d397SChris Mason 		btrfs_set_item_offset(left, item,
1694db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1695be0e5c09SChris Mason 	}
16965f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1697db94535dSChris Mason 	if (left->map_token) {
1698db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1699db94535dSChris Mason 		left->map_token = NULL;
1700db94535dSChris Mason 	}
1701be0e5c09SChris Mason 
1702be0e5c09SChris Mason 	/* fixup right node */
1703*34a38218SChris Mason 	if (push_items > right_nritems) {
1704*34a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
1705*34a38218SChris Mason 		WARN_ON(1);
1706*34a38218SChris Mason 	}
1707*34a38218SChris Mason 
1708*34a38218SChris Mason 	if (push_items < right_nritems) {
17095f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
1710123abc88SChris Mason 						  leaf_data_end(root, right);
17115f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
1712d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
1713d6025579SChris Mason 				      btrfs_leaf_data(right) +
1714123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
17155f39d397SChris Mason 
17165f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
17175f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
17185f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
17190783fcfcSChris Mason 			     sizeof(struct btrfs_item));
17205f39d397SChris Mason 
1721*34a38218SChris Mason 	}
1722*34a38218SChris Mason 	btrfs_set_header_nritems(right, right_nritems - push_items);
1723123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
1724eb60ceacSChris Mason 
17255f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
17265f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1727db94535dSChris Mason 
1728db94535dSChris Mason 		if (!right->map_token) {
1729db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1730db94535dSChris Mason 					sizeof(struct btrfs_item),
1731db94535dSChris Mason 					&right->map_token, &right->kaddr,
1732db94535dSChris Mason 					&right->map_start, &right->map_len,
1733db94535dSChris Mason 					KM_USER1);
1734db94535dSChris Mason 		}
1735db94535dSChris Mason 
1736db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
1737db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1738db94535dSChris Mason 	}
1739db94535dSChris Mason 	if (right->map_token) {
1740db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1741db94535dSChris Mason 		right->map_token = NULL;
1742be0e5c09SChris Mason 	}
1743eb60ceacSChris Mason 
17445f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
1745*34a38218SChris Mason 	if (right_nritems)
17465f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
1747098f59c2SChris Mason 
17485f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
17495f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1750aa5d6bedSChris Mason 	if (wret)
1751aa5d6bedSChris Mason 		ret = wret;
1752be0e5c09SChris Mason 
1753be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
1754be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
1755be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
17565f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
17575f39d397SChris Mason 		path->nodes[0] = left;
1758be0e5c09SChris Mason 		path->slots[1] -= 1;
1759be0e5c09SChris Mason 	} else {
17605f39d397SChris Mason 		free_extent_buffer(left);
1761be0e5c09SChris Mason 		path->slots[0] -= push_items;
1762be0e5c09SChris Mason 	}
1763eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1764aa5d6bedSChris Mason 	return ret;
1765be0e5c09SChris Mason }
1766be0e5c09SChris Mason 
176774123bd7SChris Mason /*
176874123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
176974123bd7SChris Mason  * available for the resulting leaf level of the path.
1770aa5d6bedSChris Mason  *
1771aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
177274123bd7SChris Mason  */
1773e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1774d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
1775cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend)
1776be0e5c09SChris Mason {
17775f39d397SChris Mason 	struct extent_buffer *l;
17787518a238SChris Mason 	u32 nritems;
1779eb60ceacSChris Mason 	int mid;
1780eb60ceacSChris Mason 	int slot;
17815f39d397SChris Mason 	struct extent_buffer *right;
17820783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
1783be0e5c09SChris Mason 	int data_copy_size;
1784be0e5c09SChris Mason 	int rt_data_off;
1785be0e5c09SChris Mason 	int i;
1786d4dbff95SChris Mason 	int ret = 0;
1787aa5d6bedSChris Mason 	int wret;
1788cc0c5538SChris Mason 	int double_split;
1789cc0c5538SChris Mason 	int num_doubles = 0;
1790d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
1791be0e5c09SChris Mason 
1792cc0c5538SChris Mason 	if (extend)
1793cc0c5538SChris Mason 		space_needed = data_size;
1794cc0c5538SChris Mason 
179540689478SChris Mason 	/* first try to make some room by pushing left and right */
17963685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
1797*34a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
17983326d1b0SChris Mason 		if (wret < 0) {
1799eaee50e8SChris Mason 			return wret;
18003326d1b0SChris Mason 		}
1801eaee50e8SChris Mason 		if (wret) {
1802*34a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
1803eaee50e8SChris Mason 			if (wret < 0)
1804eaee50e8SChris Mason 				return wret;
1805eaee50e8SChris Mason 		}
18065f39d397SChris Mason 		l = path->nodes[0];
1807aa5d6bedSChris Mason 
1808aa5d6bedSChris Mason 		/* did the pushes work? */
1809cc0c5538SChris Mason 		if (btrfs_leaf_free_space(root, l) >= space_needed)
1810be0e5c09SChris Mason 			return 0;
18113326d1b0SChris Mason 	}
1812aa5d6bedSChris Mason 
18135c680ed6SChris Mason 	if (!path->nodes[1]) {
1814e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
18155c680ed6SChris Mason 		if (ret)
18165c680ed6SChris Mason 			return ret;
18175c680ed6SChris Mason 	}
1818cc0c5538SChris Mason again:
1819cc0c5538SChris Mason 	double_split = 0;
1820cc0c5538SChris Mason 	l = path->nodes[0];
1821eb60ceacSChris Mason 	slot = path->slots[0];
18225f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
1823eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
182454aa1f4dSChris Mason 
1825db94535dSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
1826db94535dSChris Mason 				       l->start, 0);
18275f39d397SChris Mason 	if (IS_ERR(right))
18285f39d397SChris Mason 		return PTR_ERR(right);
182954aa1f4dSChris Mason 
18305f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1831db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
18325f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
18335f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
18345f39d397SChris Mason 	btrfs_set_header_level(right, 0);
18355f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
18365f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
18375f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1838d4dbff95SChris Mason 	if (mid <= slot) {
1839d4dbff95SChris Mason 		if (nritems == 1 ||
1840d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
1841d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1842d4dbff95SChris Mason 			if (slot >= nritems) {
1843d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18445f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1845d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1846db94535dSChris Mason 						  &disk_key, right->start,
1847d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
1848d4dbff95SChris Mason 				if (wret)
1849d4dbff95SChris Mason 					ret = wret;
18505f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18515f39d397SChris Mason 				path->nodes[0] = right;
1852d4dbff95SChris Mason 				path->slots[0] = 0;
1853d4dbff95SChris Mason 				path->slots[1] += 1;
1854d4dbff95SChris Mason 				return ret;
1855d4dbff95SChris Mason 			}
1856d4dbff95SChris Mason 			mid = slot;
18573326d1b0SChris Mason 			if (mid != nritems &&
18583326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
18593326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
1860d4dbff95SChris Mason 				double_split = 1;
1861d4dbff95SChris Mason 			}
18623326d1b0SChris Mason 		}
1863d4dbff95SChris Mason 	} else {
1864d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
1865d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
1866cc0c5538SChris Mason 			if (!extend && slot == 0) {
1867d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
18685f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
1869d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
1870d4dbff95SChris Mason 						  &disk_key,
1871db94535dSChris Mason 						  right->start,
1872098f59c2SChris Mason 						  path->slots[1], 1);
1873d4dbff95SChris Mason 				if (wret)
1874d4dbff95SChris Mason 					ret = wret;
18755f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
18765f39d397SChris Mason 				path->nodes[0] = right;
1877d4dbff95SChris Mason 				path->slots[0] = 0;
1878a429e513SChris Mason 				if (path->slots[1] == 0) {
1879a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
1880a429e513SChris Mason 					           path, &disk_key, 1);
1881a429e513SChris Mason 					if (wret)
1882a429e513SChris Mason 						ret = wret;
1883a429e513SChris Mason 				}
1884d4dbff95SChris Mason 				return ret;
1885cc0c5538SChris Mason 			} else if (extend && slot == 0) {
1886cc0c5538SChris Mason 				mid = 1;
1887cc0c5538SChris Mason 			} else {
1888d4dbff95SChris Mason 				mid = slot;
18895ee78ac7SChris Mason 				if (mid != nritems &&
18905ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
18915ee78ac7SChris Mason 				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
1892d4dbff95SChris Mason 					double_split = 1;
1893d4dbff95SChris Mason 				}
1894d4dbff95SChris Mason 			}
18955ee78ac7SChris Mason 		}
1896cc0c5538SChris Mason 	}
18975f39d397SChris Mason 	nritems = nritems - mid;
18985f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
18995f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
19005f39d397SChris Mason 
19015f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
19025f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
19035f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
19045f39d397SChris Mason 
19055f39d397SChris Mason 	copy_extent_buffer(right, l,
1906d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
1907123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
1908123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
190974123bd7SChris Mason 
19105f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
19115f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
19125f39d397SChris Mason 
19135f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
19145f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
1915db94535dSChris Mason 		u32 ioff;
1916db94535dSChris Mason 
1917db94535dSChris Mason 		if (!right->map_token) {
1918db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1919db94535dSChris Mason 					sizeof(struct btrfs_item),
1920db94535dSChris Mason 					&right->map_token, &right->kaddr,
1921db94535dSChris Mason 					&right->map_start, &right->map_len,
1922db94535dSChris Mason 					KM_USER1);
1923db94535dSChris Mason 		}
1924db94535dSChris Mason 
1925db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
19265f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
19270783fcfcSChris Mason 	}
192874123bd7SChris Mason 
1929db94535dSChris Mason 	if (right->map_token) {
1930db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1931db94535dSChris Mason 		right->map_token = NULL;
1932db94535dSChris Mason 	}
1933db94535dSChris Mason 
19345f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
1935aa5d6bedSChris Mason 	ret = 0;
19365f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
1937db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
1938db94535dSChris Mason 			  path->slots[1] + 1, 1);
1939aa5d6bedSChris Mason 	if (wret)
1940aa5d6bedSChris Mason 		ret = wret;
19415f39d397SChris Mason 
19425f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
19435f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
1944eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
19455f39d397SChris Mason 
1946be0e5c09SChris Mason 	if (mid <= slot) {
19475f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19485f39d397SChris Mason 		path->nodes[0] = right;
1949be0e5c09SChris Mason 		path->slots[0] -= mid;
1950be0e5c09SChris Mason 		path->slots[1] += 1;
1951eb60ceacSChris Mason 	} else
19525f39d397SChris Mason 		free_extent_buffer(right);
19535f39d397SChris Mason 
1954eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
1955d4dbff95SChris Mason 
1956cc0c5538SChris Mason 	if (double_split) {
1957cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
1958cc0c5538SChris Mason 		num_doubles++;
1959cc0c5538SChris Mason 		goto again;
19603326d1b0SChris Mason 	}
1961be0e5c09SChris Mason 	return ret;
1962be0e5c09SChris Mason }
1963be0e5c09SChris Mason 
1964b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
1965b18c6685SChris Mason 			struct btrfs_root *root,
1966b18c6685SChris Mason 			struct btrfs_path *path,
1967179e29e4SChris Mason 			u32 new_size, int from_end)
1968b18c6685SChris Mason {
1969b18c6685SChris Mason 	int ret = 0;
1970b18c6685SChris Mason 	int slot;
1971b18c6685SChris Mason 	int slot_orig;
19725f39d397SChris Mason 	struct extent_buffer *leaf;
19735f39d397SChris Mason 	struct btrfs_item *item;
1974b18c6685SChris Mason 	u32 nritems;
1975b18c6685SChris Mason 	unsigned int data_end;
1976b18c6685SChris Mason 	unsigned int old_data_start;
1977b18c6685SChris Mason 	unsigned int old_size;
1978b18c6685SChris Mason 	unsigned int size_diff;
1979b18c6685SChris Mason 	int i;
1980b18c6685SChris Mason 
1981b18c6685SChris Mason 	slot_orig = path->slots[0];
19825f39d397SChris Mason 	leaf = path->nodes[0];
1983179e29e4SChris Mason 	slot = path->slots[0];
1984179e29e4SChris Mason 
1985179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
1986179e29e4SChris Mason 	if (old_size == new_size)
1987179e29e4SChris Mason 		return 0;
1988b18c6685SChris Mason 
19895f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
1990b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
1991b18c6685SChris Mason 
19925f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
1993179e29e4SChris Mason 
1994b18c6685SChris Mason 	size_diff = old_size - new_size;
1995b18c6685SChris Mason 
1996b18c6685SChris Mason 	BUG_ON(slot < 0);
1997b18c6685SChris Mason 	BUG_ON(slot >= nritems);
1998b18c6685SChris Mason 
1999b18c6685SChris Mason 	/*
2000b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2001b18c6685SChris Mason 	 */
2002b18c6685SChris Mason 	/* first correct the data pointers */
2003b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
20045f39d397SChris Mason 		u32 ioff;
20055f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2006db94535dSChris Mason 
2007db94535dSChris Mason 		if (!leaf->map_token) {
2008db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2009db94535dSChris Mason 					sizeof(struct btrfs_item),
2010db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2011db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2012db94535dSChris Mason 					KM_USER1);
2013db94535dSChris Mason 		}
2014db94535dSChris Mason 
20155f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
20165f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2017b18c6685SChris Mason 	}
2018db94535dSChris Mason 
2019db94535dSChris Mason 	if (leaf->map_token) {
2020db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2021db94535dSChris Mason 		leaf->map_token = NULL;
2022db94535dSChris Mason 	}
2023db94535dSChris Mason 
2024b18c6685SChris Mason 	/* shift the data */
2025179e29e4SChris Mason 	if (from_end) {
20265f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2027b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2028b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
2029179e29e4SChris Mason 	} else {
2030179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
2031179e29e4SChris Mason 		u64 offset;
2032179e29e4SChris Mason 
2033179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
2034179e29e4SChris Mason 
2035179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2036179e29e4SChris Mason 			unsigned long ptr;
2037179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
2038179e29e4SChris Mason 
2039179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
2040179e29e4SChris Mason 					    struct btrfs_file_extent_item);
2041179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
2042179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
2043179e29e4SChris Mason 
2044179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
2045179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
2046179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
2047179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
2048179e29e4SChris Mason 				        (unsigned long)fi,
2049179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
2050179e29e4SChris Mason 						 disk_bytenr));
2051179e29e4SChris Mason 			}
2052179e29e4SChris Mason 		}
2053179e29e4SChris Mason 
2054179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2055179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2056179e29e4SChris Mason 			      data_end, old_data_start - data_end);
2057179e29e4SChris Mason 
2058179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
2059179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2060179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
2061179e29e4SChris Mason 		if (slot == 0)
2062179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
2063179e29e4SChris Mason 	}
20645f39d397SChris Mason 
20655f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
20665f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
20675f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2068b18c6685SChris Mason 
2069b18c6685SChris Mason 	ret = 0;
20705f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
20715f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2072b18c6685SChris Mason 		BUG();
20735f39d397SChris Mason 	}
2074b18c6685SChris Mason 	return ret;
2075b18c6685SChris Mason }
2076b18c6685SChris Mason 
20775f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
20785f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
20795f39d397SChris Mason 		      u32 data_size)
20806567e837SChris Mason {
20816567e837SChris Mason 	int ret = 0;
20826567e837SChris Mason 	int slot;
20836567e837SChris Mason 	int slot_orig;
20845f39d397SChris Mason 	struct extent_buffer *leaf;
20855f39d397SChris Mason 	struct btrfs_item *item;
20866567e837SChris Mason 	u32 nritems;
20876567e837SChris Mason 	unsigned int data_end;
20886567e837SChris Mason 	unsigned int old_data;
20896567e837SChris Mason 	unsigned int old_size;
20906567e837SChris Mason 	int i;
20916567e837SChris Mason 
20926567e837SChris Mason 	slot_orig = path->slots[0];
20935f39d397SChris Mason 	leaf = path->nodes[0];
20946567e837SChris Mason 
20955f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
20966567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
20976567e837SChris Mason 
20985f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
20995f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
21006567e837SChris Mason 		BUG();
21015f39d397SChris Mason 	}
21026567e837SChris Mason 	slot = path->slots[0];
21035f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
21046567e837SChris Mason 
21056567e837SChris Mason 	BUG_ON(slot < 0);
21063326d1b0SChris Mason 	if (slot >= nritems) {
21073326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
21083326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
21093326d1b0SChris Mason 		BUG_ON(1);
21103326d1b0SChris Mason 	}
21116567e837SChris Mason 
21126567e837SChris Mason 	/*
21136567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
21146567e837SChris Mason 	 */
21156567e837SChris Mason 	/* first correct the data pointers */
21166567e837SChris Mason 	for (i = slot; i < nritems; i++) {
21175f39d397SChris Mason 		u32 ioff;
21185f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2119db94535dSChris Mason 
2120db94535dSChris Mason 		if (!leaf->map_token) {
2121db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2122db94535dSChris Mason 					sizeof(struct btrfs_item),
2123db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2124db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2125db94535dSChris Mason 					KM_USER1);
2126db94535dSChris Mason 		}
21275f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
21285f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
21296567e837SChris Mason 	}
21305f39d397SChris Mason 
2131db94535dSChris Mason 	if (leaf->map_token) {
2132db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2133db94535dSChris Mason 		leaf->map_token = NULL;
2134db94535dSChris Mason 	}
2135db94535dSChris Mason 
21366567e837SChris Mason 	/* shift the data */
21375f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
21386567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
21396567e837SChris Mason 		      data_end, old_data - data_end);
21405f39d397SChris Mason 
21416567e837SChris Mason 	data_end = old_data;
21425f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
21435f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
21445f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
21455f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
21466567e837SChris Mason 
21476567e837SChris Mason 	ret = 0;
21485f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
21495f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
21506567e837SChris Mason 		BUG();
21515f39d397SChris Mason 	}
21526567e837SChris Mason 	return ret;
21536567e837SChris Mason }
21546567e837SChris Mason 
215574123bd7SChris Mason /*
215674123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
215774123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
215874123bd7SChris Mason  */
21595f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
21605f39d397SChris Mason 			    struct btrfs_root *root,
21615f39d397SChris Mason 			    struct btrfs_path *path,
21625f39d397SChris Mason 			    struct btrfs_key *cpu_key, u32 data_size)
2163be0e5c09SChris Mason {
21645f39d397SChris Mason 	struct extent_buffer *leaf;
21655f39d397SChris Mason 	struct btrfs_item *item;
2166aa5d6bedSChris Mason 	int ret = 0;
2167be0e5c09SChris Mason 	int slot;
2168eb60ceacSChris Mason 	int slot_orig;
21697518a238SChris Mason 	u32 nritems;
2170be0e5c09SChris Mason 	unsigned int data_end;
2171e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2172e2fa7227SChris Mason 
2173e2fa7227SChris Mason 	btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2174be0e5c09SChris Mason 
217574123bd7SChris Mason 	/* create a root if there isn't one */
21765c680ed6SChris Mason 	if (!root->node)
2177cfaa7295SChris Mason 		BUG();
21785f39d397SChris Mason 
2179e089f05cSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
2180eb60ceacSChris Mason 	if (ret == 0) {
2181f0930a37SChris Mason 		return -EEXIST;
2182aa5d6bedSChris Mason 	}
2183ed2ff2cbSChris Mason 	if (ret < 0)
2184ed2ff2cbSChris Mason 		goto out;
2185be0e5c09SChris Mason 
218662e2749eSChris Mason 	slot_orig = path->slots[0];
21875f39d397SChris Mason 	leaf = path->nodes[0];
218874123bd7SChris Mason 
21895f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2190123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2191eb60ceacSChris Mason 
2192123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
2193d4dbff95SChris Mason 	    sizeof(struct btrfs_item) + data_size) {
21943326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
21953326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
21963326d1b0SChris Mason 		       data_size, btrfs_leaf_free_space(root, leaf));
2197be0e5c09SChris Mason 		BUG();
2198d4dbff95SChris Mason 	}
21995f39d397SChris Mason 
220062e2749eSChris Mason 	slot = path->slots[0];
2201eb60ceacSChris Mason 	BUG_ON(slot < 0);
22025f39d397SChris Mason 
2203be0e5c09SChris Mason 	if (slot != nritems) {
2204be0e5c09SChris Mason 		int i;
22055f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2206be0e5c09SChris Mason 
22075f39d397SChris Mason 		if (old_data < data_end) {
22085f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
22095f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
22105f39d397SChris Mason 			       slot, old_data, data_end);
22115f39d397SChris Mason 			BUG_ON(1);
22125f39d397SChris Mason 		}
2213be0e5c09SChris Mason 		/*
2214be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2215be0e5c09SChris Mason 		 */
2216be0e5c09SChris Mason 		/* first correct the data pointers */
2217db94535dSChris Mason 		WARN_ON(leaf->map_token);
22180783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
22195f39d397SChris Mason 			u32 ioff;
2220db94535dSChris Mason 
22215f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2222db94535dSChris Mason 			if (!leaf->map_token) {
2223db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2224db94535dSChris Mason 					sizeof(struct btrfs_item),
2225db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2226db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2227db94535dSChris Mason 					KM_USER1);
2228db94535dSChris Mason 			}
2229db94535dSChris Mason 
22305f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
22315f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff - data_size);
22320783fcfcSChris Mason 		}
2233db94535dSChris Mason 		if (leaf->map_token) {
2234db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2235db94535dSChris Mason 			leaf->map_token = NULL;
2236db94535dSChris Mason 		}
2237be0e5c09SChris Mason 
2238be0e5c09SChris Mason 		/* shift the items */
22395f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
22405f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
22410783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2242be0e5c09SChris Mason 
2243be0e5c09SChris Mason 		/* shift the data */
22445f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2245d6025579SChris Mason 			      data_end - data_size, btrfs_leaf_data(leaf) +
2246be0e5c09SChris Mason 			      data_end, old_data - data_end);
2247be0e5c09SChris Mason 		data_end = old_data;
2248be0e5c09SChris Mason 	}
22495f39d397SChris Mason 
225062e2749eSChris Mason 	/* setup the item for the new data */
22515f39d397SChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
22525f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
22535f39d397SChris Mason 	btrfs_set_item_offset(leaf, item, data_end - data_size);
22545f39d397SChris Mason 	btrfs_set_item_size(leaf, item, data_size);
22555f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
22565f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2257aa5d6bedSChris Mason 
2258aa5d6bedSChris Mason 	ret = 0;
22598e19f2cdSChris Mason 	if (slot == 0)
2260e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2261aa5d6bedSChris Mason 
22625f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
22635f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2264be0e5c09SChris Mason 		BUG();
22655f39d397SChris Mason 	}
2266ed2ff2cbSChris Mason out:
226762e2749eSChris Mason 	return ret;
226862e2749eSChris Mason }
226962e2749eSChris Mason 
227062e2749eSChris Mason /*
227162e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
227262e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
227362e2749eSChris Mason  */
2274e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2275e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2276e089f05cSChris Mason 		      data_size)
227762e2749eSChris Mason {
227862e2749eSChris Mason 	int ret = 0;
22792c90e5d6SChris Mason 	struct btrfs_path *path;
22805f39d397SChris Mason 	struct extent_buffer *leaf;
22815f39d397SChris Mason 	unsigned long ptr;
228262e2749eSChris Mason 
22832c90e5d6SChris Mason 	path = btrfs_alloc_path();
22842c90e5d6SChris Mason 	BUG_ON(!path);
22852c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
228662e2749eSChris Mason 	if (!ret) {
22875f39d397SChris Mason 		leaf = path->nodes[0];
22885f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
22895f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
22905f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
229162e2749eSChris Mason 	}
22922c90e5d6SChris Mason 	btrfs_free_path(path);
2293aa5d6bedSChris Mason 	return ret;
2294be0e5c09SChris Mason }
2295be0e5c09SChris Mason 
229674123bd7SChris Mason /*
22975de08d7dSChris Mason  * delete the pointer from a given node.
229874123bd7SChris Mason  *
229974123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
230074123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
230174123bd7SChris Mason  * a leaf if all the nodes are emptied.
230274123bd7SChris Mason  */
2303e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2304e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2305be0e5c09SChris Mason {
23065f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
23077518a238SChris Mason 	u32 nritems;
2308aa5d6bedSChris Mason 	int ret = 0;
2309bb803951SChris Mason 	int wret;
2310be0e5c09SChris Mason 
23115f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2312be0e5c09SChris Mason 	if (slot != nritems -1) {
23135f39d397SChris Mason 		memmove_extent_buffer(parent,
23145f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
23155f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2316d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2317d6025579SChris Mason 			      (nritems - slot - 1));
2318be0e5c09SChris Mason 	}
23197518a238SChris Mason 	nritems--;
23205f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
23217518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
23225f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2323eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
23245f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2325bb803951SChris Mason 	} else if (slot == 0) {
23265f39d397SChris Mason 		struct btrfs_disk_key disk_key;
23275f39d397SChris Mason 
23285f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
23295f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
23300f70abe2SChris Mason 		if (wret)
23310f70abe2SChris Mason 			ret = wret;
2332be0e5c09SChris Mason 	}
2333d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2334aa5d6bedSChris Mason 	return ret;
2335be0e5c09SChris Mason }
2336be0e5c09SChris Mason 
233774123bd7SChris Mason /*
233874123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
233974123bd7SChris Mason  * the leaf, remove it from the tree
234074123bd7SChris Mason  */
2341e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2342e089f05cSChris Mason 		   struct btrfs_path *path)
2343be0e5c09SChris Mason {
2344be0e5c09SChris Mason 	int slot;
23455f39d397SChris Mason 	struct extent_buffer *leaf;
23465f39d397SChris Mason 	struct btrfs_item *item;
2347be0e5c09SChris Mason 	int doff;
2348be0e5c09SChris Mason 	int dsize;
2349aa5d6bedSChris Mason 	int ret = 0;
2350aa5d6bedSChris Mason 	int wret;
23517518a238SChris Mason 	u32 nritems;
2352be0e5c09SChris Mason 
23535f39d397SChris Mason 	leaf = path->nodes[0];
23544920c9acSChris Mason 	slot = path->slots[0];
23555f39d397SChris Mason 	doff = btrfs_item_offset_nr(leaf, slot);
23565f39d397SChris Mason 	dsize = btrfs_item_size_nr(leaf, slot);
23575f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2358be0e5c09SChris Mason 
23597518a238SChris Mason 	if (slot != nritems - 1) {
2360be0e5c09SChris Mason 		int i;
2361123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
23625f39d397SChris Mason 
23635f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2364d6025579SChris Mason 			      data_end + dsize,
2365123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
2366be0e5c09SChris Mason 			      doff - data_end);
23675f39d397SChris Mason 
23680783fcfcSChris Mason 		for (i = slot + 1; i < nritems; i++) {
23695f39d397SChris Mason 			u32 ioff;
2370db94535dSChris Mason 
23715f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2372db94535dSChris Mason 			if (!leaf->map_token) {
2373db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2374db94535dSChris Mason 					sizeof(struct btrfs_item),
2375db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2376db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2377db94535dSChris Mason 					KM_USER1);
2378db94535dSChris Mason 			}
23795f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
23805f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
23810783fcfcSChris Mason 		}
2382db94535dSChris Mason 
2383db94535dSChris Mason 		if (leaf->map_token) {
2384db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2385db94535dSChris Mason 			leaf->map_token = NULL;
2386db94535dSChris Mason 		}
2387db94535dSChris Mason 
23885f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
23895f39d397SChris Mason 			      btrfs_item_nr_offset(slot + 1),
23900783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
23917518a238SChris Mason 			      (nritems - slot - 1));
2392be0e5c09SChris Mason 	}
23935f39d397SChris Mason 	btrfs_set_header_nritems(leaf, nritems - 1);
23947518a238SChris Mason 	nritems--;
23955f39d397SChris Mason 
239674123bd7SChris Mason 	/* delete the leaf if we've emptied it */
23977518a238SChris Mason 	if (nritems == 0) {
23985f39d397SChris Mason 		if (leaf == root->node) {
23995f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
24009a8dd150SChris Mason 		} else {
24015f39d397SChris Mason 			clean_tree_block(trans, root, leaf);
24025f39d397SChris Mason 			wait_on_tree_block_writeback(root, leaf);
2403e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2404aa5d6bedSChris Mason 			if (wret)
2405aa5d6bedSChris Mason 				ret = wret;
2406e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
2407db94535dSChris Mason 						 leaf->start, leaf->len, 1);
24080f70abe2SChris Mason 			if (wret)
24090f70abe2SChris Mason 				ret = wret;
24109a8dd150SChris Mason 		}
2411be0e5c09SChris Mason 	} else {
24127518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2413aa5d6bedSChris Mason 		if (slot == 0) {
24145f39d397SChris Mason 			struct btrfs_disk_key disk_key;
24155f39d397SChris Mason 
24165f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2417e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
24185f39d397SChris Mason 					      &disk_key, 1);
2419aa5d6bedSChris Mason 			if (wret)
2420aa5d6bedSChris Mason 				ret = wret;
2421aa5d6bedSChris Mason 		}
2422aa5d6bedSChris Mason 
242374123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
24247936ca38SChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
2425be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2426be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2427be0e5c09SChris Mason 			 * for possible call to del_ptr below
2428be0e5c09SChris Mason 			 */
24294920c9acSChris Mason 			slot = path->slots[1];
24305f39d397SChris Mason 			extent_buffer_get(leaf);
24315f39d397SChris Mason 
2432*34a38218SChris Mason 			wret = push_leaf_right(trans, root, path, 1, 1);
243354aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2434aa5d6bedSChris Mason 				ret = wret;
24355f39d397SChris Mason 
24365f39d397SChris Mason 			if (path->nodes[0] == leaf &&
24375f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
2438*34a38218SChris Mason 				wret = push_leaf_left(trans, root, path, 1, 1);
243954aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2440aa5d6bedSChris Mason 					ret = wret;
2441aa5d6bedSChris Mason 			}
24425f39d397SChris Mason 
24435f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
2444db94535dSChris Mason 				u64 bytenr = leaf->start;
2445db94535dSChris Mason 				u32 blocksize = leaf->len;
24465f39d397SChris Mason 
24475f39d397SChris Mason 				clean_tree_block(trans, root, leaf);
24485f39d397SChris Mason 				wait_on_tree_block_writeback(root, leaf);
24495f39d397SChris Mason 
2450e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2451aa5d6bedSChris Mason 				if (wret)
2452aa5d6bedSChris Mason 					ret = wret;
24535f39d397SChris Mason 
24545f39d397SChris Mason 				free_extent_buffer(leaf);
2455db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
2456db94535dSChris Mason 							 blocksize, 1);
24570f70abe2SChris Mason 				if (wret)
24580f70abe2SChris Mason 					ret = wret;
24595de08d7dSChris Mason 			} else {
24605f39d397SChris Mason 				btrfs_mark_buffer_dirty(leaf);
24615f39d397SChris Mason 				free_extent_buffer(leaf);
2462be0e5c09SChris Mason 			}
2463d5719762SChris Mason 		} else {
24645f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2465be0e5c09SChris Mason 		}
24669a8dd150SChris Mason 	}
2467aa5d6bedSChris Mason 	return ret;
24689a8dd150SChris Mason }
24699a8dd150SChris Mason 
247097571fd0SChris Mason /*
247197571fd0SChris Mason  * walk up the tree as far as required to find the next leaf.
24720f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
24730f70abe2SChris Mason  * returns < 0 on io errors.
247497571fd0SChris Mason  */
2475234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2476d97e63b6SChris Mason {
2477d97e63b6SChris Mason 	int slot;
2478d97e63b6SChris Mason 	int level = 1;
2479db94535dSChris Mason 	u64 bytenr;
24805f39d397SChris Mason 	struct extent_buffer *c;
24815f39d397SChris Mason 	struct extent_buffer *next = NULL;
2482d97e63b6SChris Mason 
2483234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
2484d97e63b6SChris Mason 		if (!path->nodes[level])
24850f70abe2SChris Mason 			return 1;
24865f39d397SChris Mason 
2487d97e63b6SChris Mason 		slot = path->slots[level] + 1;
2488d97e63b6SChris Mason 		c = path->nodes[level];
24895f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
2490d97e63b6SChris Mason 			level++;
2491d97e63b6SChris Mason 			continue;
2492d97e63b6SChris Mason 		}
24935f39d397SChris Mason 
2494db94535dSChris Mason 		bytenr = btrfs_node_blockptr(c, slot);
2495cfaa7295SChris Mason 		if (next)
24965f39d397SChris Mason 			free_extent_buffer(next);
24975f39d397SChris Mason 
24986702ed49SChris Mason 		if (path->reada)
24996702ed49SChris Mason 			reada_for_search(root, path, level, slot);
25005f39d397SChris Mason 
2501db94535dSChris Mason 		next = read_tree_block(root, bytenr,
2502db94535dSChris Mason 				       btrfs_level_size(root, level -1));
2503d97e63b6SChris Mason 		break;
2504d97e63b6SChris Mason 	}
2505d97e63b6SChris Mason 	path->slots[level] = slot;
2506d97e63b6SChris Mason 	while(1) {
2507d97e63b6SChris Mason 		level--;
2508d97e63b6SChris Mason 		c = path->nodes[level];
25095f39d397SChris Mason 		free_extent_buffer(c);
2510d97e63b6SChris Mason 		path->nodes[level] = next;
2511d97e63b6SChris Mason 		path->slots[level] = 0;
2512d97e63b6SChris Mason 		if (!level)
2513d97e63b6SChris Mason 			break;
25146702ed49SChris Mason 		if (path->reada)
251532020611SYan 			reada_for_search(root, path, level, 0);
2516db94535dSChris Mason 		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2517db94535dSChris Mason 				       btrfs_level_size(root, level - 1));
2518d97e63b6SChris Mason 	}
2519d97e63b6SChris Mason 	return 0;
2520d97e63b6SChris Mason }
2521