xref: /openbmc/linux/fs/btrfs/ctree.c (revision a213501153fd66e2359e091b1612841305ba6551)
16cbd5570SChris Mason /*
26cbd5570SChris Mason  * Copyright (C) 2007 Oracle.  All rights reserved.
36cbd5570SChris Mason  *
46cbd5570SChris Mason  * This program is free software; you can redistribute it and/or
56cbd5570SChris Mason  * modify it under the terms of the GNU General Public
66cbd5570SChris Mason  * License v2 as published by the Free Software Foundation.
76cbd5570SChris Mason  *
86cbd5570SChris Mason  * This program is distributed in the hope that it will be useful,
96cbd5570SChris Mason  * but WITHOUT ANY WARRANTY; without even the implied warranty of
106cbd5570SChris Mason  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
116cbd5570SChris Mason  * General Public License for more details.
126cbd5570SChris Mason  *
136cbd5570SChris Mason  * You should have received a copy of the GNU General Public
146cbd5570SChris Mason  * License along with this program; if not, write to the
156cbd5570SChris Mason  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
166cbd5570SChris Mason  * Boston, MA 021110-1307, USA.
176cbd5570SChris Mason  */
186cbd5570SChris Mason 
19a6b6e75eSChris Mason #include <linux/sched.h>
20eb60ceacSChris Mason #include "ctree.h"
21eb60ceacSChris Mason #include "disk-io.h"
227f5c1516SChris Mason #include "transaction.h"
235f39d397SChris Mason #include "print-tree.h"
24925baeddSChris Mason #include "locking.h"
259a8dd150SChris Mason 
26e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
27e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
28e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
29d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
30cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend);
315f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
325f39d397SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
33971a1f66SChris Mason 			  struct extent_buffer *src, int empty);
345f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
355f39d397SChris Mason 			      struct btrfs_root *root,
365f39d397SChris Mason 			      struct extent_buffer *dst_buf,
375f39d397SChris Mason 			      struct extent_buffer *src_buf);
38e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
39e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot);
40d97e63b6SChris Mason 
41df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p)
42df24a2b9SChris Mason {
43df24a2b9SChris Mason 	memset(p, 0, sizeof(*p));
44df24a2b9SChris Mason }
45df24a2b9SChris Mason 
462c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
472c90e5d6SChris Mason {
48df24a2b9SChris Mason 	struct btrfs_path *path;
49df24a2b9SChris Mason 	path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
502cc58cf2SChris Mason 	if (path) {
51df24a2b9SChris Mason 		btrfs_init_path(path);
522cc58cf2SChris Mason 		path->reada = 1;
532cc58cf2SChris Mason 	}
54df24a2b9SChris Mason 	return path;
552c90e5d6SChris Mason }
562c90e5d6SChris Mason 
572c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
582c90e5d6SChris Mason {
59df24a2b9SChris Mason 	btrfs_release_path(NULL, p);
602c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
612c90e5d6SChris Mason }
622c90e5d6SChris Mason 
63234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
64eb60ceacSChris Mason {
65eb60ceacSChris Mason 	int i;
66*a2135011SChris Mason 	int skip = p->skip_locking;
67*a2135011SChris Mason 	int keep = p->keep_locks;
68*a2135011SChris Mason 
69234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
70eb60ceacSChris Mason 		if (!p->nodes[i])
71925baeddSChris Mason 			continue;
72925baeddSChris Mason 		if (p->locks[i]) {
73925baeddSChris Mason 			btrfs_tree_unlock(p->nodes[i]);
74925baeddSChris Mason 			p->locks[i] = 0;
75925baeddSChris Mason 		}
765f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
77eb60ceacSChris Mason 	}
78aa5d6bedSChris Mason 	memset(p, 0, sizeof(*p));
79*a2135011SChris Mason 	p->skip_locking = skip;
80*a2135011SChris Mason 	p->keep_locks = keep;
81eb60ceacSChris Mason }
82eb60ceacSChris Mason 
83925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
84925baeddSChris Mason {
85925baeddSChris Mason 	struct extent_buffer *eb;
86925baeddSChris Mason 	spin_lock(&root->node_lock);
87925baeddSChris Mason 	eb = root->node;
88925baeddSChris Mason 	extent_buffer_get(eb);
89925baeddSChris Mason 	spin_unlock(&root->node_lock);
90925baeddSChris Mason 	return eb;
91925baeddSChris Mason }
92925baeddSChris Mason 
93925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
94925baeddSChris Mason {
95925baeddSChris Mason 	struct extent_buffer *eb;
96925baeddSChris Mason 
97925baeddSChris Mason 	while(1) {
98925baeddSChris Mason 		eb = btrfs_root_node(root);
99925baeddSChris Mason 		btrfs_tree_lock(eb);
100925baeddSChris Mason 
101925baeddSChris Mason 		spin_lock(&root->node_lock);
102925baeddSChris Mason 		if (eb == root->node) {
103925baeddSChris Mason 			spin_unlock(&root->node_lock);
104925baeddSChris Mason 			break;
105925baeddSChris Mason 		}
106925baeddSChris Mason 		spin_unlock(&root->node_lock);
107925baeddSChris Mason 
108925baeddSChris Mason 		btrfs_tree_unlock(eb);
109925baeddSChris Mason 		free_extent_buffer(eb);
110925baeddSChris Mason 	}
111925baeddSChris Mason 	return eb;
112925baeddSChris Mason }
113925baeddSChris Mason 
1140b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root)
1150b86a832SChris Mason {
1160b86a832SChris Mason 	if (root->track_dirty && list_empty(&root->dirty_list)) {
1170b86a832SChris Mason 		list_add(&root->dirty_list,
1180b86a832SChris Mason 			 &root->fs_info->dirty_cowonly_roots);
1190b86a832SChris Mason 	}
1200b86a832SChris Mason }
1210b86a832SChris Mason 
122be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
123be20aa9dSChris Mason 		      struct btrfs_root *root,
124be20aa9dSChris Mason 		      struct extent_buffer *buf,
125be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
126be20aa9dSChris Mason {
127be20aa9dSChris Mason 	struct extent_buffer *cow;
128be20aa9dSChris Mason 	u32 nritems;
129be20aa9dSChris Mason 	int ret = 0;
130be20aa9dSChris Mason 	int level;
131be20aa9dSChris Mason 	struct btrfs_key first_key;
1324aec2b52SChris Mason 	struct btrfs_root *new_root;
133be20aa9dSChris Mason 
1344aec2b52SChris Mason 	new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
1354aec2b52SChris Mason 	if (!new_root)
1364aec2b52SChris Mason 		return -ENOMEM;
1374aec2b52SChris Mason 
1384aec2b52SChris Mason 	memcpy(new_root, root, sizeof(*new_root));
1394aec2b52SChris Mason 	new_root->root_key.objectid = new_root_objectid;
140be20aa9dSChris Mason 
141be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
142be20aa9dSChris Mason 		root->fs_info->running_transaction->transid);
143be20aa9dSChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
144be20aa9dSChris Mason 
145be20aa9dSChris Mason 	level = btrfs_header_level(buf);
146be20aa9dSChris Mason 	nritems = btrfs_header_nritems(buf);
147be20aa9dSChris Mason 	if (nritems) {
148be20aa9dSChris Mason 		if (level == 0)
149be20aa9dSChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
150be20aa9dSChris Mason 		else
151be20aa9dSChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
152be20aa9dSChris Mason 	} else {
153be20aa9dSChris Mason 		first_key.objectid = 0;
154be20aa9dSChris Mason 	}
155925baeddSChris Mason 	cow = btrfs_alloc_free_block(trans, new_root, buf->len,
156be20aa9dSChris Mason 				       new_root_objectid,
157be20aa9dSChris Mason 				       trans->transid, first_key.objectid,
158be20aa9dSChris Mason 				       level, buf->start, 0);
1594aec2b52SChris Mason 	if (IS_ERR(cow)) {
1604aec2b52SChris Mason 		kfree(new_root);
161be20aa9dSChris Mason 		return PTR_ERR(cow);
1624aec2b52SChris Mason 	}
163be20aa9dSChris Mason 
164be20aa9dSChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
165be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
166be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
167be20aa9dSChris Mason 	btrfs_set_header_owner(cow, new_root_objectid);
16863b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
169be20aa9dSChris Mason 
170be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
1714aec2b52SChris Mason 	ret = btrfs_inc_ref(trans, new_root, buf);
1724aec2b52SChris Mason 	kfree(new_root);
1734aec2b52SChris Mason 
174be20aa9dSChris Mason 	if (ret)
175be20aa9dSChris Mason 		return ret;
176be20aa9dSChris Mason 
177be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
178be20aa9dSChris Mason 	*cow_ret = cow;
179be20aa9dSChris Mason 	return 0;
180be20aa9dSChris Mason }
181be20aa9dSChris Mason 
182be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans,
1835f39d397SChris Mason 			     struct btrfs_root *root,
1845f39d397SChris Mason 			     struct extent_buffer *buf,
1855f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
1865f39d397SChris Mason 			     struct extent_buffer **cow_ret,
1875f39d397SChris Mason 			     u64 search_start, u64 empty_size)
1886702ed49SChris Mason {
1897bb86316SChris Mason 	u64 root_gen;
1905f39d397SChris Mason 	struct extent_buffer *cow;
1917bb86316SChris Mason 	u32 nritems;
1926702ed49SChris Mason 	int ret = 0;
1936702ed49SChris Mason 	int different_trans = 0;
1947bb86316SChris Mason 	int level;
195925baeddSChris Mason 	int unlock_orig = 0;
1967bb86316SChris Mason 	struct btrfs_key first_key;
1976702ed49SChris Mason 
198925baeddSChris Mason 	if (*cow_ret == buf)
199925baeddSChris Mason 		unlock_orig = 1;
200925baeddSChris Mason 
201925baeddSChris Mason 	WARN_ON(!btrfs_tree_locked(buf));
202925baeddSChris Mason 
2037bb86316SChris Mason 	if (root->ref_cows) {
2047bb86316SChris Mason 		root_gen = trans->transid;
2057bb86316SChris Mason 	} else {
2067bb86316SChris Mason 		root_gen = 0;
2077bb86316SChris Mason 	}
2087bb86316SChris Mason 	WARN_ON(root->ref_cows && trans->transid !=
2097bb86316SChris Mason 		root->fs_info->running_transaction->transid);
2106702ed49SChris Mason 	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
2115f39d397SChris Mason 
2127bb86316SChris Mason 	level = btrfs_header_level(buf);
2137bb86316SChris Mason 	nritems = btrfs_header_nritems(buf);
2147bb86316SChris Mason 	if (nritems) {
2157bb86316SChris Mason 		if (level == 0)
2167bb86316SChris Mason 			btrfs_item_key_to_cpu(buf, &first_key, 0);
2177bb86316SChris Mason 		else
2187bb86316SChris Mason 			btrfs_node_key_to_cpu(buf, &first_key, 0);
2197bb86316SChris Mason 	} else {
2207bb86316SChris Mason 		first_key.objectid = 0;
2217bb86316SChris Mason 	}
222925baeddSChris Mason 	cow = btrfs_alloc_free_block(trans, root, buf->len,
2237bb86316SChris Mason 				     root->root_key.objectid,
2247bb86316SChris Mason 				     root_gen, first_key.objectid, level,
225db94535dSChris Mason 				     search_start, empty_size);
2266702ed49SChris Mason 	if (IS_ERR(cow))
2276702ed49SChris Mason 		return PTR_ERR(cow);
2286702ed49SChris Mason 
2295f39d397SChris Mason 	copy_extent_buffer(cow, buf, 0, 0, cow->len);
230db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
2315f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
2325f39d397SChris Mason 	btrfs_set_header_owner(cow, root->root_key.objectid);
23363b10fc4SChris Mason 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
2346702ed49SChris Mason 
2355f39d397SChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
2365f39d397SChris Mason 	if (btrfs_header_generation(buf) != trans->transid) {
2376702ed49SChris Mason 		different_trans = 1;
2386702ed49SChris Mason 		ret = btrfs_inc_ref(trans, root, buf);
2396702ed49SChris Mason 		if (ret)
2406702ed49SChris Mason 			return ret;
2416702ed49SChris Mason 	} else {
2426702ed49SChris Mason 		clean_tree_block(trans, root, buf);
2436702ed49SChris Mason 	}
2446702ed49SChris Mason 
2456702ed49SChris Mason 	if (buf == root->node) {
246925baeddSChris Mason 		WARN_ON(parent && parent != buf);
2477bb86316SChris Mason 		root_gen = btrfs_header_generation(buf);
248925baeddSChris Mason 
249925baeddSChris Mason 		spin_lock(&root->node_lock);
2506702ed49SChris Mason 		root->node = cow;
2515f39d397SChris Mason 		extent_buffer_get(cow);
252925baeddSChris Mason 		spin_unlock(&root->node_lock);
253925baeddSChris Mason 
2546702ed49SChris Mason 		if (buf != root->commit_root) {
255db94535dSChris Mason 			btrfs_free_extent(trans, root, buf->start,
2567bb86316SChris Mason 					  buf->len, root->root_key.objectid,
2577bb86316SChris Mason 					  root_gen, 0, 0, 1);
2586702ed49SChris Mason 		}
2595f39d397SChris Mason 		free_extent_buffer(buf);
2600b86a832SChris Mason 		add_root_to_dirty_list(root);
2616702ed49SChris Mason 	} else {
2627bb86316SChris Mason 		root_gen = btrfs_header_generation(parent);
2635f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
264db94535dSChris Mason 					cow->start);
26574493f7aSChris Mason 		WARN_ON(trans->transid == 0);
26674493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
26774493f7aSChris Mason 					      trans->transid);
2686702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
2695f39d397SChris Mason 		WARN_ON(btrfs_header_generation(parent) != trans->transid);
2707bb86316SChris Mason 		btrfs_free_extent(trans, root, buf->start, buf->len,
2717bb86316SChris Mason 				  btrfs_header_owner(parent), root_gen,
2727bb86316SChris Mason 				  0, 0, 1);
2736702ed49SChris Mason 	}
274925baeddSChris Mason 	if (unlock_orig)
275925baeddSChris Mason 		btrfs_tree_unlock(buf);
2765f39d397SChris Mason 	free_extent_buffer(buf);
2776702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
2786702ed49SChris Mason 	*cow_ret = cow;
2796702ed49SChris Mason 	return 0;
2806702ed49SChris Mason }
2816702ed49SChris Mason 
2825f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans,
2835f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
2845f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
2855f39d397SChris Mason 		    struct extent_buffer **cow_ret)
28602217ed2SChris Mason {
2876702ed49SChris Mason 	u64 search_start;
288dc17ff8fSChris Mason 	u64 header_trans;
289f510cfecSChris Mason 	int ret;
290dc17ff8fSChris Mason 
291ccd467d6SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
292ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
293ccd467d6SChris Mason 		       root->fs_info->running_transaction->transid);
294ccd467d6SChris Mason 		WARN_ON(1);
295ccd467d6SChris Mason 	}
296ccd467d6SChris Mason 	if (trans->transid != root->fs_info->generation) {
297ccd467d6SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
298ccd467d6SChris Mason 		       root->fs_info->generation);
299ccd467d6SChris Mason 		WARN_ON(1);
300ccd467d6SChris Mason 	}
301dc17ff8fSChris Mason 
302dc17ff8fSChris Mason 	header_trans = btrfs_header_generation(buf);
30363b10fc4SChris Mason 	spin_lock(&root->fs_info->hash_lock);
30463b10fc4SChris Mason 	if (header_trans == trans->transid &&
30563b10fc4SChris Mason 	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) {
30602217ed2SChris Mason 		*cow_ret = buf;
30763b10fc4SChris Mason 		spin_unlock(&root->fs_info->hash_lock);
30802217ed2SChris Mason 		return 0;
30902217ed2SChris Mason 	}
31063b10fc4SChris Mason 	spin_unlock(&root->fs_info->hash_lock);
3110b86a832SChris Mason 	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
312f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
3136702ed49SChris Mason 				 parent_slot, cow_ret, search_start, 0);
314f510cfecSChris Mason 	return ret;
3152c90e5d6SChris Mason }
3166702ed49SChris Mason 
3176b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
3186702ed49SChris Mason {
3196b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
3206702ed49SChris Mason 		return 1;
3216b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
3226702ed49SChris Mason 		return 1;
32302217ed2SChris Mason 	return 0;
32402217ed2SChris Mason }
32502217ed2SChris Mason 
326081e9573SChris Mason /*
327081e9573SChris Mason  * compare two keys in a memcmp fashion
328081e9573SChris Mason  */
329081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
330081e9573SChris Mason {
331081e9573SChris Mason 	struct btrfs_key k1;
332081e9573SChris Mason 
333081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
334081e9573SChris Mason 
335081e9573SChris Mason 	if (k1.objectid > k2->objectid)
336081e9573SChris Mason 		return 1;
337081e9573SChris Mason 	if (k1.objectid < k2->objectid)
338081e9573SChris Mason 		return -1;
339081e9573SChris Mason 	if (k1.type > k2->type)
340081e9573SChris Mason 		return 1;
341081e9573SChris Mason 	if (k1.type < k2->type)
342081e9573SChris Mason 		return -1;
343081e9573SChris Mason 	if (k1.offset > k2->offset)
344081e9573SChris Mason 		return 1;
345081e9573SChris Mason 	if (k1.offset < k2->offset)
346081e9573SChris Mason 		return -1;
347081e9573SChris Mason 	return 0;
348081e9573SChris Mason }
349081e9573SChris Mason 
350081e9573SChris Mason 
3516702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
3525f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
353a6b6e75eSChris Mason 		       int start_slot, int cache_only, u64 *last_ret,
354a6b6e75eSChris Mason 		       struct btrfs_key *progress)
3556702ed49SChris Mason {
3566b80053dSChris Mason 	struct extent_buffer *cur;
3576b80053dSChris Mason 	struct extent_buffer *tmp;
3586702ed49SChris Mason 	u64 blocknr;
359ca7a79adSChris Mason 	u64 gen;
360e9d0b13bSChris Mason 	u64 search_start = *last_ret;
361e9d0b13bSChris Mason 	u64 last_block = 0;
3626702ed49SChris Mason 	u64 other;
3636702ed49SChris Mason 	u32 parent_nritems;
3646702ed49SChris Mason 	int end_slot;
3656702ed49SChris Mason 	int i;
3666702ed49SChris Mason 	int err = 0;
367f2183bdeSChris Mason 	int parent_level;
3686b80053dSChris Mason 	int uptodate;
3696b80053dSChris Mason 	u32 blocksize;
370081e9573SChris Mason 	int progress_passed = 0;
371081e9573SChris Mason 	struct btrfs_disk_key disk_key;
3726702ed49SChris Mason 
373925baeddSChris Mason 	/* FIXME this code needs locking */
374925baeddSChris Mason 	return 0;
375925baeddSChris Mason 
3765708b959SChris Mason 	parent_level = btrfs_header_level(parent);
3775708b959SChris Mason 	if (cache_only && parent_level != 1)
3785708b959SChris Mason 		return 0;
3795708b959SChris Mason 
3806702ed49SChris Mason 	if (trans->transaction != root->fs_info->running_transaction) {
3816702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3826702ed49SChris Mason 		       root->fs_info->running_transaction->transid);
3836702ed49SChris Mason 		WARN_ON(1);
3846702ed49SChris Mason 	}
3856702ed49SChris Mason 	if (trans->transid != root->fs_info->generation) {
3866702ed49SChris Mason 		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
3876702ed49SChris Mason 		       root->fs_info->generation);
3886702ed49SChris Mason 		WARN_ON(1);
3896702ed49SChris Mason 	}
39086479a04SChris Mason 
3916b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
3926b80053dSChris Mason 	blocksize = btrfs_level_size(root, parent_level - 1);
3936702ed49SChris Mason 	end_slot = parent_nritems;
3946702ed49SChris Mason 
3956702ed49SChris Mason 	if (parent_nritems == 1)
3966702ed49SChris Mason 		return 0;
3976702ed49SChris Mason 
3986702ed49SChris Mason 	for (i = start_slot; i < end_slot; i++) {
3996702ed49SChris Mason 		int close = 1;
400a6b6e75eSChris Mason 
4015708b959SChris Mason 		if (!parent->map_token) {
4025708b959SChris Mason 			map_extent_buffer(parent,
4035708b959SChris Mason 					btrfs_node_key_ptr_offset(i),
4045708b959SChris Mason 					sizeof(struct btrfs_key_ptr),
4055708b959SChris Mason 					&parent->map_token, &parent->kaddr,
4065708b959SChris Mason 					&parent->map_start, &parent->map_len,
4075708b959SChris Mason 					KM_USER1);
4085708b959SChris Mason 		}
409081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
410081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
411081e9573SChris Mason 			continue;
412081e9573SChris Mason 
413081e9573SChris Mason 		progress_passed = 1;
4146b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
415ca7a79adSChris Mason 		gen = btrfs_node_ptr_generation(parent, i);
416e9d0b13bSChris Mason 		if (last_block == 0)
417e9d0b13bSChris Mason 			last_block = blocknr;
4185708b959SChris Mason 
4196702ed49SChris Mason 		if (i > 0) {
4206b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
4216b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
4226702ed49SChris Mason 		}
4230ef3e66bSChris Mason 		if (!close && i < end_slot - 2) {
4246b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
4256b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
4266702ed49SChris Mason 		}
427e9d0b13bSChris Mason 		if (close) {
428e9d0b13bSChris Mason 			last_block = blocknr;
4296702ed49SChris Mason 			continue;
430e9d0b13bSChris Mason 		}
4315708b959SChris Mason 		if (parent->map_token) {
4325708b959SChris Mason 			unmap_extent_buffer(parent, parent->map_token,
4335708b959SChris Mason 					    KM_USER1);
4345708b959SChris Mason 			parent->map_token = NULL;
4355708b959SChris Mason 		}
4366702ed49SChris Mason 
4376b80053dSChris Mason 		cur = btrfs_find_tree_block(root, blocknr, blocksize);
4386b80053dSChris Mason 		if (cur)
4391259ab75SChris Mason 			uptodate = btrfs_buffer_uptodate(cur, gen);
4406b80053dSChris Mason 		else
4416b80053dSChris Mason 			uptodate = 0;
4425708b959SChris Mason 		if (!cur || !uptodate) {
4436702ed49SChris Mason 			if (cache_only) {
4446b80053dSChris Mason 				free_extent_buffer(cur);
4456702ed49SChris Mason 				continue;
4466702ed49SChris Mason 			}
4476b80053dSChris Mason 			if (!cur) {
4486b80053dSChris Mason 				cur = read_tree_block(root, blocknr,
449ca7a79adSChris Mason 							 blocksize, gen);
4506b80053dSChris Mason 			} else if (!uptodate) {
451ca7a79adSChris Mason 				btrfs_read_buffer(cur, gen);
4526702ed49SChris Mason 			}
453f2183bdeSChris Mason 		}
454e9d0b13bSChris Mason 		if (search_start == 0)
4556b80053dSChris Mason 			search_start = last_block;
456e9d0b13bSChris Mason 
4576b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
4586b80053dSChris Mason 					&tmp, search_start,
4596b80053dSChris Mason 					min(16 * blocksize,
4606b80053dSChris Mason 					    (end_slot - i) * blocksize));
461252c38f0SYan 		if (err) {
4626b80053dSChris Mason 			free_extent_buffer(cur);
4636702ed49SChris Mason 			break;
464252c38f0SYan 		}
4656b80053dSChris Mason 		search_start = tmp->start;
4665708b959SChris Mason 		last_block = tmp->start;
467f2183bdeSChris Mason 		*last_ret = search_start;
468f2183bdeSChris Mason 		if (parent_level == 1)
4696b80053dSChris Mason 			btrfs_clear_buffer_defrag(tmp);
4706b80053dSChris Mason 		free_extent_buffer(tmp);
4716702ed49SChris Mason 	}
4725708b959SChris Mason 	if (parent->map_token) {
4735708b959SChris Mason 		unmap_extent_buffer(parent, parent->map_token,
4745708b959SChris Mason 				    KM_USER1);
4755708b959SChris Mason 		parent->map_token = NULL;
4765708b959SChris Mason 	}
4776702ed49SChris Mason 	return err;
4786702ed49SChris Mason }
4796702ed49SChris Mason 
48074123bd7SChris Mason /*
48174123bd7SChris Mason  * The leaf data grows from end-to-front in the node.
48274123bd7SChris Mason  * this returns the address of the start of the last item,
48374123bd7SChris Mason  * which is the stop of the leaf data stack
48474123bd7SChris Mason  */
485123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root,
4865f39d397SChris Mason 					 struct extent_buffer *leaf)
487be0e5c09SChris Mason {
4885f39d397SChris Mason 	u32 nr = btrfs_header_nritems(leaf);
489be0e5c09SChris Mason 	if (nr == 0)
490123abc88SChris Mason 		return BTRFS_LEAF_DATA_SIZE(root);
4915f39d397SChris Mason 	return btrfs_item_offset_nr(leaf, nr - 1);
492be0e5c09SChris Mason }
493be0e5c09SChris Mason 
494123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path,
495123abc88SChris Mason 		      int level)
496aa5d6bedSChris Mason {
4975f39d397SChris Mason 	struct extent_buffer *parent = NULL;
4985f39d397SChris Mason 	struct extent_buffer *node = path->nodes[level];
4995f39d397SChris Mason 	struct btrfs_disk_key parent_key;
5005f39d397SChris Mason 	struct btrfs_disk_key node_key;
501aa5d6bedSChris Mason 	int parent_slot;
5028d7be552SChris Mason 	int slot;
5038d7be552SChris Mason 	struct btrfs_key cpukey;
5045f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(node);
505aa5d6bedSChris Mason 
506aa5d6bedSChris Mason 	if (path->nodes[level + 1])
5075f39d397SChris Mason 		parent = path->nodes[level + 1];
508a1f39630SAneesh 
5098d7be552SChris Mason 	slot = path->slots[level];
5107518a238SChris Mason 	BUG_ON(nritems == 0);
5117518a238SChris Mason 	if (parent) {
512a1f39630SAneesh 		parent_slot = path->slots[level + 1];
5135f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
5145f39d397SChris Mason 		btrfs_node_key(node, &node_key, 0);
5155f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &node_key,
516e2fa7227SChris Mason 			      sizeof(struct btrfs_disk_key)));
5171d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
518db94535dSChris Mason 		       btrfs_header_bytenr(node));
519aa5d6bedSChris Mason 	}
520123abc88SChris Mason 	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
5218d7be552SChris Mason 	if (slot != 0) {
5225f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
5235f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
5245f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
5258d7be552SChris Mason 	}
5268d7be552SChris Mason 	if (slot < nritems - 1) {
5275f39d397SChris Mason 		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
5285f39d397SChris Mason 		btrfs_node_key(node, &node_key, slot);
5295f39d397SChris Mason 		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
530aa5d6bedSChris Mason 	}
531aa5d6bedSChris Mason 	return 0;
532aa5d6bedSChris Mason }
533aa5d6bedSChris Mason 
534123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
535123abc88SChris Mason 		      int level)
536aa5d6bedSChris Mason {
5375f39d397SChris Mason 	struct extent_buffer *leaf = path->nodes[level];
5385f39d397SChris Mason 	struct extent_buffer *parent = NULL;
539aa5d6bedSChris Mason 	int parent_slot;
5408d7be552SChris Mason 	struct btrfs_key cpukey;
5415f39d397SChris Mason 	struct btrfs_disk_key parent_key;
5425f39d397SChris Mason 	struct btrfs_disk_key leaf_key;
5435f39d397SChris Mason 	int slot = path->slots[0];
5448d7be552SChris Mason 
5455f39d397SChris Mason 	u32 nritems = btrfs_header_nritems(leaf);
546aa5d6bedSChris Mason 
547aa5d6bedSChris Mason 	if (path->nodes[level + 1])
5485f39d397SChris Mason 		parent = path->nodes[level + 1];
5497518a238SChris Mason 
5507518a238SChris Mason 	if (nritems == 0)
5517518a238SChris Mason 		return 0;
5527518a238SChris Mason 
5537518a238SChris Mason 	if (parent) {
554a1f39630SAneesh 		parent_slot = path->slots[level + 1];
5555f39d397SChris Mason 		btrfs_node_key(parent, &parent_key, parent_slot);
5565f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, 0);
5576702ed49SChris Mason 
5585f39d397SChris Mason 		BUG_ON(memcmp(&parent_key, &leaf_key,
559e2fa7227SChris Mason 		       sizeof(struct btrfs_disk_key)));
5601d4f8a0cSChris Mason 		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
561db94535dSChris Mason 		       btrfs_header_bytenr(leaf));
562aa5d6bedSChris Mason 	}
5635f39d397SChris Mason #if 0
5645f39d397SChris Mason 	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
5655f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
5665f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, i);
5675f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) >= 0) {
5685f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5695f39d397SChris Mason 			printk("slot %d offset bad key\n", i);
5705f39d397SChris Mason 			BUG_ON(1);
5715f39d397SChris Mason 		}
5725f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, i) !=
5735f39d397SChris Mason 			btrfs_item_end_nr(leaf, i + 1)) {
5745f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
5755f39d397SChris Mason 			printk("slot %d offset bad\n", i);
5765f39d397SChris Mason 			BUG_ON(1);
5775f39d397SChris Mason 		}
5785f39d397SChris Mason 		if (i == 0) {
5795f39d397SChris Mason 			if (btrfs_item_offset_nr(leaf, i) +
5805f39d397SChris Mason 			       btrfs_item_size_nr(leaf, i) !=
5815f39d397SChris Mason 			       BTRFS_LEAF_DATA_SIZE(root)) {
5825f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5835f39d397SChris Mason 				printk("slot %d first offset bad\n", i);
5845f39d397SChris Mason 				BUG_ON(1);
5855f39d397SChris Mason 			}
5865f39d397SChris Mason 		}
5875f39d397SChris Mason 	}
5885f39d397SChris Mason 	if (nritems > 0) {
5895f39d397SChris Mason 		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
5905f39d397SChris Mason 				btrfs_print_leaf(root, leaf);
5915f39d397SChris Mason 				printk("slot %d bad size \n", nritems - 1);
5925f39d397SChris Mason 				BUG_ON(1);
5935f39d397SChris Mason 		}
5945f39d397SChris Mason 	}
5955f39d397SChris Mason #endif
5965f39d397SChris Mason 	if (slot != 0 && slot < nritems - 1) {
5975f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
5985f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
5995f39d397SChris Mason 		if (comp_keys(&leaf_key, &cpukey) <= 0) {
6005f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6015f39d397SChris Mason 			printk("slot %d offset bad key\n", slot);
6025f39d397SChris Mason 			BUG_ON(1);
6035f39d397SChris Mason 		}
6045f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot - 1) !=
6055f39d397SChris Mason 		       btrfs_item_end_nr(leaf, slot)) {
6065f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6075f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
6085f39d397SChris Mason 			BUG_ON(1);
6095f39d397SChris Mason 		}
610aa5d6bedSChris Mason 	}
6118d7be552SChris Mason 	if (slot < nritems - 1) {
6125f39d397SChris Mason 		btrfs_item_key(leaf, &leaf_key, slot);
6135f39d397SChris Mason 		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
6145f39d397SChris Mason 		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
6155f39d397SChris Mason 		if (btrfs_item_offset_nr(leaf, slot) !=
6165f39d397SChris Mason 			btrfs_item_end_nr(leaf, slot + 1)) {
6175f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
6185f39d397SChris Mason 			printk("slot %d offset bad\n", slot);
6195f39d397SChris Mason 			BUG_ON(1);
620aa5d6bedSChris Mason 		}
6215f39d397SChris Mason 	}
6225f39d397SChris Mason 	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
6235f39d397SChris Mason 	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
624aa5d6bedSChris Mason 	return 0;
625aa5d6bedSChris Mason }
626aa5d6bedSChris Mason 
62798ed5174SChris Mason static int noinline check_block(struct btrfs_root *root,
62898ed5174SChris Mason 				struct btrfs_path *path, int level)
629aa5d6bedSChris Mason {
630f188591eSChris Mason 	u64 found_start;
63185d824c4SChris Mason 	return 0;
632f188591eSChris Mason 	if (btrfs_header_level(path->nodes[level]) != level)
633f188591eSChris Mason 	    printk("warning: bad level %Lu wanted %d found %d\n",
634f188591eSChris Mason 		   path->nodes[level]->start, level,
635f188591eSChris Mason 		   btrfs_header_level(path->nodes[level]));
636f188591eSChris Mason 	found_start = btrfs_header_bytenr(path->nodes[level]);
637f188591eSChris Mason 	if (found_start != path->nodes[level]->start) {
638f188591eSChris Mason 	    printk("warning: bad bytentr %Lu found %Lu\n",
639f188591eSChris Mason 		   path->nodes[level]->start, found_start);
640f188591eSChris Mason 	}
641db94535dSChris Mason #if 0
6425f39d397SChris Mason 	struct extent_buffer *buf = path->nodes[level];
6435f39d397SChris Mason 
644479965d6SChris Mason 	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
645479965d6SChris Mason 				 (unsigned long)btrfs_header_fsid(buf),
646479965d6SChris Mason 				 BTRFS_FSID_SIZE)) {
6475f39d397SChris Mason 		printk("warning bad block %Lu\n", buf->start);
648db94535dSChris Mason 		return 1;
6495f39d397SChris Mason 	}
650db94535dSChris Mason #endif
651aa5d6bedSChris Mason 	if (level == 0)
652123abc88SChris Mason 		return check_leaf(root, path, level);
653123abc88SChris Mason 	return check_node(root, path, level);
654aa5d6bedSChris Mason }
655aa5d6bedSChris Mason 
65674123bd7SChris Mason /*
6575f39d397SChris Mason  * search for key in the extent_buffer.  The items start at offset p,
6585f39d397SChris Mason  * and they are item_size apart.  There are 'max' items in p.
6595f39d397SChris Mason  *
66074123bd7SChris Mason  * the slot in the array is returned via slot, and it points to
66174123bd7SChris Mason  * the place where you would insert key if it is not found in
66274123bd7SChris Mason  * the array.
66374123bd7SChris Mason  *
66474123bd7SChris Mason  * slot may point to max if the key is bigger than all of the keys
66574123bd7SChris Mason  */
6665f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
6675f39d397SChris Mason 			      int item_size, struct btrfs_key *key,
668be0e5c09SChris Mason 			      int max, int *slot)
669be0e5c09SChris Mason {
670be0e5c09SChris Mason 	int low = 0;
671be0e5c09SChris Mason 	int high = max;
672be0e5c09SChris Mason 	int mid;
673be0e5c09SChris Mason 	int ret;
674479965d6SChris Mason 	struct btrfs_disk_key *tmp = NULL;
6755f39d397SChris Mason 	struct btrfs_disk_key unaligned;
6765f39d397SChris Mason 	unsigned long offset;
6775f39d397SChris Mason 	char *map_token = NULL;
6785f39d397SChris Mason 	char *kaddr = NULL;
6795f39d397SChris Mason 	unsigned long map_start = 0;
6805f39d397SChris Mason 	unsigned long map_len = 0;
681479965d6SChris Mason 	int err;
682be0e5c09SChris Mason 
683be0e5c09SChris Mason 	while(low < high) {
684be0e5c09SChris Mason 		mid = (low + high) / 2;
6855f39d397SChris Mason 		offset = p + mid * item_size;
6865f39d397SChris Mason 
6875f39d397SChris Mason 		if (!map_token || offset < map_start ||
6885f39d397SChris Mason 		    (offset + sizeof(struct btrfs_disk_key)) >
6895f39d397SChris Mason 		    map_start + map_len) {
690479965d6SChris Mason 			if (map_token) {
6915f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
692479965d6SChris Mason 				map_token = NULL;
693479965d6SChris Mason 			}
694479965d6SChris Mason 			err = map_extent_buffer(eb, offset,
695479965d6SChris Mason 						sizeof(struct btrfs_disk_key),
696479965d6SChris Mason 						&map_token, &kaddr,
6975f39d397SChris Mason 						&map_start, &map_len, KM_USER0);
6985f39d397SChris Mason 
699479965d6SChris Mason 			if (!err) {
700479965d6SChris Mason 				tmp = (struct btrfs_disk_key *)(kaddr + offset -
701479965d6SChris Mason 							map_start);
702479965d6SChris Mason 			} else {
7035f39d397SChris Mason 				read_extent_buffer(eb, &unaligned,
7045f39d397SChris Mason 						   offset, sizeof(unaligned));
7055f39d397SChris Mason 				tmp = &unaligned;
706479965d6SChris Mason 			}
707479965d6SChris Mason 
7085f39d397SChris Mason 		} else {
7095f39d397SChris Mason 			tmp = (struct btrfs_disk_key *)(kaddr + offset -
7105f39d397SChris Mason 							map_start);
7115f39d397SChris Mason 		}
712be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
713be0e5c09SChris Mason 
714be0e5c09SChris Mason 		if (ret < 0)
715be0e5c09SChris Mason 			low = mid + 1;
716be0e5c09SChris Mason 		else if (ret > 0)
717be0e5c09SChris Mason 			high = mid;
718be0e5c09SChris Mason 		else {
719be0e5c09SChris Mason 			*slot = mid;
720479965d6SChris Mason 			if (map_token)
7215f39d397SChris Mason 				unmap_extent_buffer(eb, map_token, KM_USER0);
722be0e5c09SChris Mason 			return 0;
723be0e5c09SChris Mason 		}
724be0e5c09SChris Mason 	}
725be0e5c09SChris Mason 	*slot = low;
7265f39d397SChris Mason 	if (map_token)
7275f39d397SChris Mason 		unmap_extent_buffer(eb, map_token, KM_USER0);
728be0e5c09SChris Mason 	return 1;
729be0e5c09SChris Mason }
730be0e5c09SChris Mason 
73197571fd0SChris Mason /*
73297571fd0SChris Mason  * simple bin_search frontend that does the right thing for
73397571fd0SChris Mason  * leaves vs nodes
73497571fd0SChris Mason  */
7355f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
7365f39d397SChris Mason 		      int level, int *slot)
737be0e5c09SChris Mason {
7385f39d397SChris Mason 	if (level == 0) {
7395f39d397SChris Mason 		return generic_bin_search(eb,
7405f39d397SChris Mason 					  offsetof(struct btrfs_leaf, items),
7410783fcfcSChris Mason 					  sizeof(struct btrfs_item),
7425f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
7437518a238SChris Mason 					  slot);
744be0e5c09SChris Mason 	} else {
7455f39d397SChris Mason 		return generic_bin_search(eb,
7465f39d397SChris Mason 					  offsetof(struct btrfs_node, ptrs),
747123abc88SChris Mason 					  sizeof(struct btrfs_key_ptr),
7485f39d397SChris Mason 					  key, btrfs_header_nritems(eb),
7497518a238SChris Mason 					  slot);
750be0e5c09SChris Mason 	}
751be0e5c09SChris Mason 	return -1;
752be0e5c09SChris Mason }
753be0e5c09SChris Mason 
7545f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root,
7555f39d397SChris Mason 				   struct extent_buffer *parent, int slot)
756bb803951SChris Mason {
757ca7a79adSChris Mason 	int level = btrfs_header_level(parent);
758bb803951SChris Mason 	if (slot < 0)
759bb803951SChris Mason 		return NULL;
7605f39d397SChris Mason 	if (slot >= btrfs_header_nritems(parent))
761bb803951SChris Mason 		return NULL;
762ca7a79adSChris Mason 
763ca7a79adSChris Mason 	BUG_ON(level == 0);
764ca7a79adSChris Mason 
765db94535dSChris Mason 	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
766ca7a79adSChris Mason 		       btrfs_level_size(root, level - 1),
767ca7a79adSChris Mason 		       btrfs_node_ptr_generation(parent, slot));
768bb803951SChris Mason }
769bb803951SChris Mason 
77098ed5174SChris Mason static int balance_level(struct btrfs_trans_handle *trans,
77198ed5174SChris Mason 			 struct btrfs_root *root,
77298ed5174SChris Mason 			 struct btrfs_path *path, int level)
773bb803951SChris 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;
778bb803951SChris Mason 	int ret = 0;
779bb803951SChris Mason 	int wret;
780bb803951SChris Mason 	int pslot;
781bb803951SChris Mason 	int orig_slot = path->slots[level];
78254aa1f4dSChris Mason 	int err_on_enospc = 0;
78379f95c82SChris Mason 	u64 orig_ptr;
784bb803951SChris Mason 
785bb803951SChris Mason 	if (level == 0)
786bb803951SChris Mason 		return 0;
787bb803951SChris Mason 
7885f39d397SChris Mason 	mid = path->nodes[level];
789925baeddSChris Mason 	WARN_ON(!path->locks[level]);
7907bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
7917bb86316SChris Mason 
7921d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
79379f95c82SChris Mason 
794234b63a0SChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
7955f39d397SChris Mason 		parent = path->nodes[level + 1];
796bb803951SChris Mason 	pslot = path->slots[level + 1];
797bb803951SChris Mason 
79840689478SChris Mason 	/*
79940689478SChris Mason 	 * deal with the case where there is only one pointer in the root
80040689478SChris Mason 	 * by promoting the node below to a root
80140689478SChris Mason 	 */
8025f39d397SChris Mason 	if (!parent) {
8035f39d397SChris Mason 		struct extent_buffer *child;
804bb803951SChris Mason 
8055f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
806bb803951SChris Mason 			return 0;
807bb803951SChris Mason 
808bb803951SChris Mason 		/* promote the child to a root */
8095f39d397SChris Mason 		child = read_node_slot(root, mid, 0);
810925baeddSChris Mason 		btrfs_tree_lock(child);
811bb803951SChris Mason 		BUG_ON(!child);
8122f375ab9SYan 		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
8132f375ab9SYan 		BUG_ON(ret);
8142f375ab9SYan 
815925baeddSChris Mason 		spin_lock(&root->node_lock);
816bb803951SChris Mason 		root->node = child;
817925baeddSChris Mason 		spin_unlock(&root->node_lock);
818925baeddSChris Mason 
8190b86a832SChris Mason 		add_root_to_dirty_list(root);
820925baeddSChris Mason 		btrfs_tree_unlock(child);
821925baeddSChris Mason 		path->locks[level] = 0;
822bb803951SChris Mason 		path->nodes[level] = NULL;
8235f39d397SChris Mason 		clean_tree_block(trans, root, mid);
824925baeddSChris Mason 		btrfs_tree_unlock(mid);
825bb803951SChris Mason 		/* once for the path */
8265f39d397SChris Mason 		free_extent_buffer(mid);
8277bb86316SChris Mason 		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
8287bb86316SChris Mason 					root->root_key.objectid,
8297bb86316SChris Mason 					btrfs_header_generation(mid), 0, 0, 1);
830bb803951SChris Mason 		/* once for the root ptr */
8315f39d397SChris Mason 		free_extent_buffer(mid);
832db94535dSChris Mason 		return ret;
833bb803951SChris Mason 	}
8345f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
835123abc88SChris Mason 	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
836bb803951SChris Mason 		return 0;
837bb803951SChris Mason 
8385f39d397SChris Mason 	if (btrfs_header_nritems(mid) < 2)
83954aa1f4dSChris Mason 		err_on_enospc = 1;
84054aa1f4dSChris Mason 
8415f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
8425f39d397SChris Mason 	if (left) {
843925baeddSChris Mason 		btrfs_tree_lock(left);
8445f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
8455f39d397SChris Mason 				       parent, pslot - 1, &left);
84654aa1f4dSChris Mason 		if (wret) {
84754aa1f4dSChris Mason 			ret = wret;
84854aa1f4dSChris Mason 			goto enospc;
84954aa1f4dSChris Mason 		}
8502cc58cf2SChris Mason 	}
8515f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
8525f39d397SChris Mason 	if (right) {
853925baeddSChris Mason 		btrfs_tree_lock(right);
8545f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
8555f39d397SChris Mason 				       parent, pslot + 1, &right);
8562cc58cf2SChris Mason 		if (wret) {
8572cc58cf2SChris Mason 			ret = wret;
8582cc58cf2SChris Mason 			goto enospc;
8592cc58cf2SChris Mason 		}
8602cc58cf2SChris Mason 	}
8612cc58cf2SChris Mason 
8622cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
8635f39d397SChris Mason 	if (left) {
8645f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
865bce4eae9SChris Mason 		wret = push_node_left(trans, root, left, mid, 1);
86679f95c82SChris Mason 		if (wret < 0)
86779f95c82SChris Mason 			ret = wret;
8685f39d397SChris Mason 		if (btrfs_header_nritems(mid) < 2)
86954aa1f4dSChris Mason 			err_on_enospc = 1;
870bb803951SChris Mason 	}
87179f95c82SChris Mason 
87279f95c82SChris Mason 	/*
87379f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
87479f95c82SChris Mason 	 */
8755f39d397SChris Mason 	if (right) {
876971a1f66SChris Mason 		wret = push_node_left(trans, root, mid, right, 1);
87754aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
87879f95c82SChris Mason 			ret = wret;
8795f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
880db94535dSChris Mason 			u64 bytenr = right->start;
8817bb86316SChris Mason 			u64 generation = btrfs_header_generation(parent);
882db94535dSChris Mason 			u32 blocksize = right->len;
883db94535dSChris Mason 
8845f39d397SChris Mason 			clean_tree_block(trans, root, right);
885925baeddSChris Mason 			btrfs_tree_unlock(right);
8865f39d397SChris Mason 			free_extent_buffer(right);
887bb803951SChris Mason 			right = NULL;
888e089f05cSChris Mason 			wret = del_ptr(trans, root, path, level + 1, pslot +
889e089f05cSChris Mason 				       1);
890bb803951SChris Mason 			if (wret)
891bb803951SChris Mason 				ret = wret;
892db94535dSChris Mason 			wret = btrfs_free_extent(trans, root, bytenr,
8937bb86316SChris Mason 						 blocksize,
8947bb86316SChris Mason 						 btrfs_header_owner(parent),
8957bb86316SChris Mason 						 generation, 0, 0, 1);
896bb803951SChris Mason 			if (wret)
897bb803951SChris Mason 				ret = wret;
898bb803951SChris Mason 		} else {
8995f39d397SChris Mason 			struct btrfs_disk_key right_key;
9005f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
9015f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
9025f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
903bb803951SChris Mason 		}
904bb803951SChris Mason 	}
9055f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
90679f95c82SChris Mason 		/*
90779f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
90879f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
90979f95c82SChris Mason 		 * could try to delete the only pointer in this node.
91079f95c82SChris Mason 		 * So, pull some keys from the left.
91179f95c82SChris Mason 		 * There has to be a left pointer at this point because
91279f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
91379f95c82SChris Mason 		 * right
91479f95c82SChris Mason 		 */
9155f39d397SChris Mason 		BUG_ON(!left);
9165f39d397SChris Mason 		wret = balance_node_right(trans, root, mid, left);
91754aa1f4dSChris Mason 		if (wret < 0) {
91879f95c82SChris Mason 			ret = wret;
91954aa1f4dSChris Mason 			goto enospc;
92054aa1f4dSChris Mason 		}
921bce4eae9SChris Mason 		if (wret == 1) {
922bce4eae9SChris Mason 			wret = push_node_left(trans, root, left, mid, 1);
923bce4eae9SChris Mason 			if (wret < 0)
924bce4eae9SChris Mason 				ret = wret;
925bce4eae9SChris Mason 		}
92679f95c82SChris Mason 		BUG_ON(wret == 1);
92779f95c82SChris Mason 	}
9285f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
92979f95c82SChris Mason 		/* we've managed to empty the middle node, drop it */
9307bb86316SChris Mason 		u64 root_gen = btrfs_header_generation(parent);
931db94535dSChris Mason 		u64 bytenr = mid->start;
932db94535dSChris Mason 		u32 blocksize = mid->len;
933925baeddSChris Mason 
9345f39d397SChris Mason 		clean_tree_block(trans, root, mid);
935925baeddSChris Mason 		btrfs_tree_unlock(mid);
9365f39d397SChris Mason 		free_extent_buffer(mid);
937bb803951SChris Mason 		mid = NULL;
938e089f05cSChris Mason 		wret = del_ptr(trans, root, path, level + 1, pslot);
939bb803951SChris Mason 		if (wret)
940bb803951SChris Mason 			ret = wret;
9417bb86316SChris Mason 		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
9427bb86316SChris Mason 					 btrfs_header_owner(parent),
9437bb86316SChris Mason 					 root_gen, 0, 0, 1);
944bb803951SChris Mason 		if (wret)
945bb803951SChris Mason 			ret = wret;
94679f95c82SChris Mason 	} else {
94779f95c82SChris Mason 		/* update the parent key to reflect our changes */
9485f39d397SChris Mason 		struct btrfs_disk_key mid_key;
9495f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
9505f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
9515f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
95279f95c82SChris Mason 	}
953bb803951SChris Mason 
95479f95c82SChris Mason 	/* update the path */
9555f39d397SChris Mason 	if (left) {
9565f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
9575f39d397SChris Mason 			extent_buffer_get(left);
958925baeddSChris Mason 			/* left was locked after cow */
9595f39d397SChris Mason 			path->nodes[level] = left;
960bb803951SChris Mason 			path->slots[level + 1] -= 1;
961bb803951SChris Mason 			path->slots[level] = orig_slot;
962925baeddSChris Mason 			if (mid) {
963925baeddSChris Mason 				btrfs_tree_unlock(mid);
9645f39d397SChris Mason 				free_extent_buffer(mid);
965925baeddSChris Mason 			}
966bb803951SChris Mason 		} else {
9675f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
968bb803951SChris Mason 			path->slots[level] = orig_slot;
969bb803951SChris Mason 		}
970bb803951SChris Mason 	}
97179f95c82SChris Mason 	/* double check we haven't messed things up */
972123abc88SChris Mason 	check_block(root, path, level);
973e20d96d6SChris Mason 	if (orig_ptr !=
9745f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
97579f95c82SChris Mason 		BUG();
97654aa1f4dSChris Mason enospc:
977925baeddSChris Mason 	if (right) {
978925baeddSChris Mason 		btrfs_tree_unlock(right);
9795f39d397SChris Mason 		free_extent_buffer(right);
980925baeddSChris Mason 	}
981925baeddSChris Mason 	if (left) {
982925baeddSChris Mason 		if (path->nodes[level] != left)
983925baeddSChris Mason 			btrfs_tree_unlock(left);
9845f39d397SChris Mason 		free_extent_buffer(left);
985925baeddSChris Mason 	}
986bb803951SChris Mason 	return ret;
987bb803951SChris Mason }
988bb803951SChris Mason 
989e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */
99098ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
991e66f709bSChris Mason 					  struct btrfs_root *root,
992e66f709bSChris Mason 					  struct btrfs_path *path, int level)
993e66f709bSChris Mason {
9945f39d397SChris Mason 	struct extent_buffer *right = NULL;
9955f39d397SChris Mason 	struct extent_buffer *mid;
9965f39d397SChris Mason 	struct extent_buffer *left = NULL;
9975f39d397SChris Mason 	struct extent_buffer *parent = NULL;
998e66f709bSChris Mason 	int ret = 0;
999e66f709bSChris Mason 	int wret;
1000e66f709bSChris Mason 	int pslot;
1001e66f709bSChris Mason 	int orig_slot = path->slots[level];
1002e66f709bSChris Mason 	u64 orig_ptr;
1003e66f709bSChris Mason 
1004e66f709bSChris Mason 	if (level == 0)
1005e66f709bSChris Mason 		return 1;
1006e66f709bSChris Mason 
10075f39d397SChris Mason 	mid = path->nodes[level];
10087bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1009e66f709bSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
1010e66f709bSChris Mason 
1011e66f709bSChris Mason 	if (level < BTRFS_MAX_LEVEL - 1)
10125f39d397SChris Mason 		parent = path->nodes[level + 1];
1013e66f709bSChris Mason 	pslot = path->slots[level + 1];
1014e66f709bSChris Mason 
10155f39d397SChris Mason 	if (!parent)
1016e66f709bSChris Mason 		return 1;
1017e66f709bSChris Mason 
10185f39d397SChris Mason 	left = read_node_slot(root, parent, pslot - 1);
1019e66f709bSChris Mason 
1020e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
10215f39d397SChris Mason 	if (left) {
1022e66f709bSChris Mason 		u32 left_nr;
1023925baeddSChris Mason 
1024925baeddSChris Mason 		btrfs_tree_lock(left);
10255f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
102633ade1f8SChris Mason 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
102733ade1f8SChris Mason 			wret = 1;
102833ade1f8SChris Mason 		} else {
10295f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
10305f39d397SChris Mason 					      pslot - 1, &left);
103154aa1f4dSChris Mason 			if (ret)
103254aa1f4dSChris Mason 				wret = 1;
103354aa1f4dSChris Mason 			else {
103454aa1f4dSChris Mason 				wret = push_node_left(trans, root,
1035971a1f66SChris Mason 						      left, mid, 0);
103654aa1f4dSChris Mason 			}
103733ade1f8SChris Mason 		}
1038e66f709bSChris Mason 		if (wret < 0)
1039e66f709bSChris Mason 			ret = wret;
1040e66f709bSChris Mason 		if (wret == 0) {
10415f39d397SChris Mason 			struct btrfs_disk_key disk_key;
1042e66f709bSChris Mason 			orig_slot += left_nr;
10435f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
10445f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
10455f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
10465f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
10475f39d397SChris Mason 				path->nodes[level] = left;
1048e66f709bSChris Mason 				path->slots[level + 1] -= 1;
1049e66f709bSChris Mason 				path->slots[level] = orig_slot;
1050925baeddSChris Mason 				btrfs_tree_unlock(mid);
10515f39d397SChris Mason 				free_extent_buffer(mid);
1052e66f709bSChris Mason 			} else {
1053e66f709bSChris Mason 				orig_slot -=
10545f39d397SChris Mason 					btrfs_header_nritems(left);
1055e66f709bSChris Mason 				path->slots[level] = orig_slot;
1056925baeddSChris Mason 				btrfs_tree_unlock(left);
10575f39d397SChris Mason 				free_extent_buffer(left);
1058e66f709bSChris Mason 			}
1059e66f709bSChris Mason 			return 0;
1060e66f709bSChris Mason 		}
1061925baeddSChris Mason 		btrfs_tree_unlock(left);
10625f39d397SChris Mason 		free_extent_buffer(left);
1063e66f709bSChris Mason 	}
10645f39d397SChris Mason 	right = read_node_slot(root, parent, pslot + 1);
1065e66f709bSChris Mason 
1066e66f709bSChris Mason 	/*
1067e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
1068e66f709bSChris Mason 	 */
10695f39d397SChris Mason 	if (right) {
107033ade1f8SChris Mason 		u32 right_nr;
1071925baeddSChris Mason 		btrfs_tree_lock(right);
10725f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
107333ade1f8SChris Mason 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
107433ade1f8SChris Mason 			wret = 1;
107533ade1f8SChris Mason 		} else {
10765f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
10775f39d397SChris Mason 					      parent, pslot + 1,
10785f39d397SChris Mason 					      &right);
107954aa1f4dSChris Mason 			if (ret)
108054aa1f4dSChris Mason 				wret = 1;
108154aa1f4dSChris Mason 			else {
108233ade1f8SChris Mason 				wret = balance_node_right(trans, root,
10835f39d397SChris Mason 							  right, mid);
108433ade1f8SChris Mason 			}
108554aa1f4dSChris Mason 		}
1086e66f709bSChris Mason 		if (wret < 0)
1087e66f709bSChris Mason 			ret = wret;
1088e66f709bSChris Mason 		if (wret == 0) {
10895f39d397SChris Mason 			struct btrfs_disk_key disk_key;
10905f39d397SChris Mason 
10915f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
10925f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
10935f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
10945f39d397SChris Mason 
10955f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
10965f39d397SChris Mason 				path->nodes[level] = right;
1097e66f709bSChris Mason 				path->slots[level + 1] += 1;
1098e66f709bSChris Mason 				path->slots[level] = orig_slot -
10995f39d397SChris Mason 					btrfs_header_nritems(mid);
1100925baeddSChris Mason 				btrfs_tree_unlock(mid);
11015f39d397SChris Mason 				free_extent_buffer(mid);
1102e66f709bSChris Mason 			} else {
1103925baeddSChris Mason 				btrfs_tree_unlock(right);
11045f39d397SChris Mason 				free_extent_buffer(right);
1105e66f709bSChris Mason 			}
1106e66f709bSChris Mason 			return 0;
1107e66f709bSChris Mason 		}
1108925baeddSChris Mason 		btrfs_tree_unlock(right);
11095f39d397SChris Mason 		free_extent_buffer(right);
1110e66f709bSChris Mason 	}
1111e66f709bSChris Mason 	return 1;
1112e66f709bSChris Mason }
1113e66f709bSChris Mason 
111474123bd7SChris Mason /*
11153c69faecSChris Mason  * readahead one full node of leaves
11163c69faecSChris Mason  */
11173c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
111801f46658SChris Mason 			     int level, int slot, u64 objectid)
11193c69faecSChris Mason {
11205f39d397SChris Mason 	struct extent_buffer *node;
112101f46658SChris Mason 	struct btrfs_disk_key disk_key;
11223c69faecSChris Mason 	u32 nritems;
11233c69faecSChris Mason 	u64 search;
11246b80053dSChris Mason 	u64 lowest_read;
11256b80053dSChris Mason 	u64 highest_read;
11266b80053dSChris Mason 	u64 nread = 0;
11273c69faecSChris Mason 	int direction = path->reada;
11285f39d397SChris Mason 	struct extent_buffer *eb;
11296b80053dSChris Mason 	u32 nr;
11306b80053dSChris Mason 	u32 blocksize;
11316b80053dSChris Mason 	u32 nscan = 0;
1132db94535dSChris Mason 
1133a6b6e75eSChris Mason 	if (level != 1)
11343c69faecSChris Mason 		return;
11353c69faecSChris Mason 
11366702ed49SChris Mason 	if (!path->nodes[level])
11376702ed49SChris Mason 		return;
11386702ed49SChris Mason 
11395f39d397SChris Mason 	node = path->nodes[level];
1140925baeddSChris Mason 	WARN_ON(!path->skip_locking && !btrfs_tree_locked(node));
1141925baeddSChris Mason 
11423c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
11436b80053dSChris Mason 	blocksize = btrfs_level_size(root, level - 1);
11446b80053dSChris Mason 	eb = btrfs_find_tree_block(root, search, blocksize);
11455f39d397SChris Mason 	if (eb) {
11465f39d397SChris Mason 		free_extent_buffer(eb);
11473c69faecSChris Mason 		return;
11483c69faecSChris Mason 	}
11493c69faecSChris Mason 
11506b80053dSChris Mason 	highest_read = search;
11516b80053dSChris Mason 	lowest_read = search;
11526b80053dSChris Mason 
11535f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
11546b80053dSChris Mason 	nr = slot;
11553c69faecSChris Mason 	while(1) {
11566b80053dSChris Mason 		if (direction < 0) {
11576b80053dSChris Mason 			if (nr == 0)
11583c69faecSChris Mason 				break;
11596b80053dSChris Mason 			nr--;
11606b80053dSChris Mason 		} else if (direction > 0) {
11616b80053dSChris Mason 			nr++;
11626b80053dSChris Mason 			if (nr >= nritems)
11636b80053dSChris Mason 				break;
11643c69faecSChris Mason 		}
116501f46658SChris Mason 		if (path->reada < 0 && objectid) {
116601f46658SChris Mason 			btrfs_node_key(node, &disk_key, nr);
116701f46658SChris Mason 			if (btrfs_disk_key_objectid(&disk_key) != objectid)
116801f46658SChris Mason 				break;
116901f46658SChris Mason 		}
11706b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
11716b80053dSChris Mason 		if ((search >= lowest_read && search <= highest_read) ||
11726b80053dSChris Mason 		    (search < lowest_read && lowest_read - search <= 32768) ||
11736b80053dSChris Mason 		    (search > highest_read && search - highest_read <= 32768)) {
1174ca7a79adSChris Mason 			readahead_tree_block(root, search, blocksize,
1175ca7a79adSChris Mason 				     btrfs_node_ptr_generation(node, nr));
11766b80053dSChris Mason 			nread += blocksize;
11773c69faecSChris Mason 		}
11786b80053dSChris Mason 		nscan++;
11796b80053dSChris Mason 		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
11806b80053dSChris Mason 			break;
11816b80053dSChris Mason 		if(nread > (1024 * 1024) || nscan > 128)
11826b80053dSChris Mason 			break;
11836b80053dSChris Mason 
11846b80053dSChris Mason 		if (search < lowest_read)
11856b80053dSChris Mason 			lowest_read = search;
11866b80053dSChris Mason 		if (search > highest_read)
11876b80053dSChris Mason 			highest_read = search;
11883c69faecSChris Mason 	}
11893c69faecSChris Mason }
1190925baeddSChris Mason 
1191925baeddSChris Mason static void unlock_up(struct btrfs_path *path, int level, int lowest_unlock)
1192925baeddSChris Mason {
1193925baeddSChris Mason 	int i;
1194925baeddSChris Mason 	int skip_level = level;
1195925baeddSChris Mason 	struct extent_buffer *t;
1196925baeddSChris Mason 
1197925baeddSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1198925baeddSChris Mason 		if (!path->nodes[i])
1199925baeddSChris Mason 			break;
1200925baeddSChris Mason 		if (!path->locks[i])
1201925baeddSChris Mason 			break;
1202925baeddSChris Mason 		if (path->slots[i] == 0) {
1203925baeddSChris Mason 			skip_level = i + 1;
1204925baeddSChris Mason 			continue;
1205925baeddSChris Mason 		}
1206925baeddSChris Mason 		if (path->keep_locks) {
1207925baeddSChris Mason 			u32 nritems;
1208925baeddSChris Mason 			t = path->nodes[i];
1209925baeddSChris Mason 			nritems = btrfs_header_nritems(t);
1210*a2135011SChris Mason 			if (nritems < 2 || path->slots[i] >= nritems - 2) {
1211*a2135011SChris Mason if (path->keep_locks) {
1212*a2135011SChris Mason //printk("path %p skip level now %d\n", path, skip_level);
1213*a2135011SChris Mason }
1214925baeddSChris Mason 				skip_level = i + 1;
1215925baeddSChris Mason 				continue;
1216925baeddSChris Mason 			}
1217925baeddSChris Mason 		}
1218925baeddSChris Mason 		t = path->nodes[i];
1219925baeddSChris Mason 		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
1220*a2135011SChris Mason if (path->keep_locks) {
1221*a2135011SChris Mason //printk("path %p unlocking level %d slot %d nritems %d skip_level %d\n", path, i, path->slots[i], btrfs_header_nritems(t), skip_level);
1222*a2135011SChris Mason }
1223925baeddSChris Mason 			btrfs_tree_unlock(t);
1224925baeddSChris Mason 			path->locks[i] = 0;
1225925baeddSChris Mason 		}
1226925baeddSChris Mason 	}
1227925baeddSChris Mason }
1228925baeddSChris Mason 
12293c69faecSChris Mason /*
123074123bd7SChris Mason  * look for key in the tree.  path is filled in with nodes along the way
123174123bd7SChris Mason  * if key is found, we return zero and you can find the item in the leaf
123274123bd7SChris Mason  * level of the path (level 0)
123374123bd7SChris Mason  *
123474123bd7SChris Mason  * If the key isn't found, the path points to the slot where it should
1235aa5d6bedSChris Mason  * be inserted, and 1 is returned.  If there are other errors during the
1236aa5d6bedSChris Mason  * search a negative error number is returned.
123797571fd0SChris Mason  *
123897571fd0SChris Mason  * if ins_len > 0, nodes and leaves will be split as we walk down the
123997571fd0SChris Mason  * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
124097571fd0SChris Mason  * possible)
124174123bd7SChris Mason  */
1242e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
1243e089f05cSChris Mason 		      *root, struct btrfs_key *key, struct btrfs_path *p, int
1244e089f05cSChris Mason 		      ins_len, int cow)
1245be0e5c09SChris Mason {
12465f39d397SChris Mason 	struct extent_buffer *b;
1247be0e5c09SChris Mason 	int slot;
1248be0e5c09SChris Mason 	int ret;
1249be0e5c09SChris Mason 	int level;
12503c69faecSChris Mason 	int should_reada = p->reada;
1251925baeddSChris Mason 	int lowest_unlock = 1;
12529f3a7427SChris Mason 	u8 lowest_level = 0;
12539f3a7427SChris Mason 
12546702ed49SChris Mason 	lowest_level = p->lowest_level;
12556702ed49SChris Mason 	WARN_ON(lowest_level && ins_len);
125622b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
1257925baeddSChris Mason 	WARN_ON(root == root->fs_info->extent_root &&
1258925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->alloc_mutex));
1259925baeddSChris Mason 	WARN_ON(root == root->fs_info->chunk_root &&
1260925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->chunk_mutex));
1261925baeddSChris Mason 	WARN_ON(root == root->fs_info->dev_root &&
1262925baeddSChris Mason 		!mutex_is_locked(&root->fs_info->chunk_mutex));
1263925baeddSChris Mason 	if (ins_len < 0)
1264925baeddSChris Mason 		lowest_unlock = 2;
1265bb803951SChris Mason again:
1266925baeddSChris Mason 	if (!p->skip_locking)
1267925baeddSChris Mason 		b = btrfs_lock_root_node(root);
1268925baeddSChris Mason 	else
1269925baeddSChris Mason 		b = btrfs_root_node(root);
1270925baeddSChris Mason 
1271eb60ceacSChris Mason 	while (b) {
12725f39d397SChris Mason 		level = btrfs_header_level(b);
127302217ed2SChris Mason 		if (cow) {
127402217ed2SChris Mason 			int wret;
1275e20d96d6SChris Mason 			wret = btrfs_cow_block(trans, root, b,
1276e20d96d6SChris Mason 					       p->nodes[level + 1],
1277e20d96d6SChris Mason 					       p->slots[level + 1],
1278252c38f0SYan 					       &b);
127954aa1f4dSChris Mason 			if (wret) {
12805f39d397SChris Mason 				free_extent_buffer(b);
128154aa1f4dSChris Mason 				return wret;
128254aa1f4dSChris Mason 			}
128302217ed2SChris Mason 		}
128402217ed2SChris Mason 		BUG_ON(!cow && ins_len);
12855f39d397SChris Mason 		if (level != btrfs_header_level(b))
12862c90e5d6SChris Mason 			WARN_ON(1);
12875f39d397SChris Mason 		level = btrfs_header_level(b);
1288eb60ceacSChris Mason 		p->nodes[level] = b;
1289925baeddSChris Mason 		if (!p->skip_locking)
1290925baeddSChris Mason 			p->locks[level] = 1;
1291123abc88SChris Mason 		ret = check_block(root, p, level);
1292aa5d6bedSChris Mason 		if (ret)
1293aa5d6bedSChris Mason 			return -1;
1294925baeddSChris Mason 
12955f39d397SChris Mason 		ret = bin_search(b, key, level, &slot);
12965f39d397SChris Mason 		if (level != 0) {
1297be0e5c09SChris Mason 			if (ret && slot > 0)
1298be0e5c09SChris Mason 				slot -= 1;
1299be0e5c09SChris Mason 			p->slots[level] = slot;
13005f39d397SChris Mason 			if (ins_len > 0 && btrfs_header_nritems(b) >=
13011514794eSChris Mason 			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1302e089f05cSChris Mason 				int sret = split_node(trans, root, p, level);
13035c680ed6SChris Mason 				BUG_ON(sret > 0);
13045c680ed6SChris Mason 				if (sret)
13055c680ed6SChris Mason 					return sret;
13065c680ed6SChris Mason 				b = p->nodes[level];
13075c680ed6SChris Mason 				slot = p->slots[level];
1308bb803951SChris Mason 			} else if (ins_len < 0) {
1309e089f05cSChris Mason 				int sret = balance_level(trans, root, p,
1310e089f05cSChris Mason 							 level);
1311bb803951SChris Mason 				if (sret)
1312bb803951SChris Mason 					return sret;
1313bb803951SChris Mason 				b = p->nodes[level];
1314f510cfecSChris Mason 				if (!b) {
1315f510cfecSChris Mason 					btrfs_release_path(NULL, p);
1316bb803951SChris Mason 					goto again;
1317f510cfecSChris Mason 				}
1318bb803951SChris Mason 				slot = p->slots[level];
13195f39d397SChris Mason 				BUG_ON(btrfs_header_nritems(b) == 1);
13205c680ed6SChris Mason 			}
13219f3a7427SChris Mason 			/* this is only true while dropping a snapshot */
1322925baeddSChris Mason 			if (level == lowest_level) {
1323925baeddSChris Mason 				unlock_up(p, level, lowest_unlock);
13249f3a7427SChris Mason 				break;
1325925baeddSChris Mason 			}
1326ca7a79adSChris Mason 
13276702ed49SChris Mason 			if (should_reada)
132801f46658SChris Mason 				reada_for_search(root, p, level, slot,
132901f46658SChris Mason 						 key->objectid);
1330ca7a79adSChris Mason 
1331ca7a79adSChris Mason 			b = read_node_slot(root, b, slot);
1332925baeddSChris Mason 			if (!p->skip_locking)
1333925baeddSChris Mason 				btrfs_tree_lock(b);
1334*a2135011SChris Mason 			unlock_up(p, level + 1, lowest_unlock);
1335be0e5c09SChris Mason 		} else {
1336be0e5c09SChris Mason 			p->slots[level] = slot;
13375f39d397SChris Mason 			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
13380783fcfcSChris Mason 			    sizeof(struct btrfs_item) + ins_len) {
1339d4dbff95SChris Mason 				int sret = split_leaf(trans, root, key,
1340cc0c5538SChris Mason 						      p, ins_len, ret == 0);
13415c680ed6SChris Mason 				BUG_ON(sret > 0);
13425c680ed6SChris Mason 				if (sret)
13435c680ed6SChris Mason 					return sret;
13445c680ed6SChris Mason 			}
1345925baeddSChris Mason 			unlock_up(p, level, lowest_unlock);
1346be0e5c09SChris Mason 			return ret;
1347be0e5c09SChris Mason 		}
1348be0e5c09SChris Mason 	}
1349aa5d6bedSChris Mason 	return 1;
1350be0e5c09SChris Mason }
1351be0e5c09SChris Mason 
135274123bd7SChris Mason /*
135374123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
135474123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
135574123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
135674123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
135774123bd7SChris Mason  * higher levels
1358aa5d6bedSChris Mason  *
1359aa5d6bedSChris Mason  * If this fails to write a tree block, it returns -1, but continues
1360aa5d6bedSChris Mason  * fixing up the blocks in ram so the tree is consistent.
136174123bd7SChris Mason  */
13625f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans,
13635f39d397SChris Mason 			  struct btrfs_root *root, struct btrfs_path *path,
13645f39d397SChris Mason 			  struct btrfs_disk_key *key, int level)
1365be0e5c09SChris Mason {
1366be0e5c09SChris Mason 	int i;
1367aa5d6bedSChris Mason 	int ret = 0;
13685f39d397SChris Mason 	struct extent_buffer *t;
13695f39d397SChris Mason 
1370234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1371be0e5c09SChris Mason 		int tslot = path->slots[i];
1372eb60ceacSChris Mason 		if (!path->nodes[i])
1373be0e5c09SChris Mason 			break;
13745f39d397SChris Mason 		t = path->nodes[i];
13755f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
1376925baeddSChris Mason 		if (!btrfs_tree_locked(path->nodes[i])) {
1377925baeddSChris Mason 			int ii;
1378925baeddSChris Mason printk("fixup without lock on level %d\n", btrfs_header_level(path->nodes[i]));
1379925baeddSChris Mason 			for (ii = 0; ii < BTRFS_MAX_LEVEL; ii++) {
1380925baeddSChris Mason printk("level %d slot %d\n", ii, path->slots[ii]);
1381925baeddSChris Mason 			}
1382925baeddSChris Mason 		}
1383d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
1384be0e5c09SChris Mason 		if (tslot != 0)
1385be0e5c09SChris Mason 			break;
1386be0e5c09SChris Mason 	}
1387aa5d6bedSChris Mason 	return ret;
1388be0e5c09SChris Mason }
1389be0e5c09SChris Mason 
139074123bd7SChris Mason /*
139174123bd7SChris Mason  * try to push data from one node into the next node left in the
139279f95c82SChris Mason  * tree.
1393aa5d6bedSChris Mason  *
1394aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1395aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
139674123bd7SChris Mason  */
139798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
139898ed5174SChris Mason 			  struct btrfs_root *root, struct extent_buffer *dst,
1399971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
1400be0e5c09SChris Mason {
1401be0e5c09SChris Mason 	int push_items = 0;
1402bb803951SChris Mason 	int src_nritems;
1403bb803951SChris Mason 	int dst_nritems;
1404aa5d6bedSChris Mason 	int ret = 0;
1405be0e5c09SChris Mason 
14065f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14075f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1408123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
14097bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14107bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
141154aa1f4dSChris Mason 
1412bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
1413971a1f66SChris Mason 		return 1;
1414971a1f66SChris Mason 
1415eb60ceacSChris Mason 	if (push_items <= 0) {
1416be0e5c09SChris Mason 		return 1;
1417eb60ceacSChris Mason 	}
1418be0e5c09SChris Mason 
1419bce4eae9SChris Mason 	if (empty) {
1420971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
1421bce4eae9SChris Mason 		if (push_items < src_nritems) {
1422bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
1423bce4eae9SChris Mason 			 * we aren't going to empty it
1424bce4eae9SChris Mason 			 */
1425bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
1426bce4eae9SChris Mason 				if (push_items <= 8)
1427bce4eae9SChris Mason 					return 1;
1428bce4eae9SChris Mason 				push_items -= 8;
1429bce4eae9SChris Mason 			}
1430bce4eae9SChris Mason 		}
1431bce4eae9SChris Mason 	} else
1432bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
143379f95c82SChris Mason 
14345f39d397SChris Mason 	copy_extent_buffer(dst, src,
14355f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
14365f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
1437123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
14385f39d397SChris Mason 
1439bb803951SChris Mason 	if (push_items < src_nritems) {
14405f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
14415f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
1442e2fa7227SChris Mason 				      (src_nritems - push_items) *
1443123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
1444bb803951SChris Mason 	}
14455f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
14465f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
14475f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
14485f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
1449bb803951SChris Mason 	return ret;
1450be0e5c09SChris Mason }
1451be0e5c09SChris Mason 
145297571fd0SChris Mason /*
145379f95c82SChris Mason  * try to push data from one node into the next node right in the
145479f95c82SChris Mason  * tree.
145579f95c82SChris Mason  *
145679f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
145779f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
145879f95c82SChris Mason  *
145979f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
146079f95c82SChris Mason  */
14615f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
14625f39d397SChris Mason 			      struct btrfs_root *root,
14635f39d397SChris Mason 			      struct extent_buffer *dst,
14645f39d397SChris Mason 			      struct extent_buffer *src)
146579f95c82SChris Mason {
146679f95c82SChris Mason 	int push_items = 0;
146779f95c82SChris Mason 	int max_push;
146879f95c82SChris Mason 	int src_nritems;
146979f95c82SChris Mason 	int dst_nritems;
147079f95c82SChris Mason 	int ret = 0;
147179f95c82SChris Mason 
14727bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
14737bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
14747bb86316SChris Mason 
14755f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
14765f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
1477123abc88SChris Mason 	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1478bce4eae9SChris Mason 	if (push_items <= 0) {
147979f95c82SChris Mason 		return 1;
1480bce4eae9SChris Mason 	}
1481bce4eae9SChris Mason 
1482bce4eae9SChris Mason 	if (src_nritems < 4) {
1483bce4eae9SChris Mason 		return 1;
1484bce4eae9SChris Mason 	}
148579f95c82SChris Mason 
148679f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
148779f95c82SChris Mason 	/* don't try to empty the node */
1488bce4eae9SChris Mason 	if (max_push >= src_nritems) {
148979f95c82SChris Mason 		return 1;
1490bce4eae9SChris Mason 	}
1491252c38f0SYan 
149279f95c82SChris Mason 	if (max_push < push_items)
149379f95c82SChris Mason 		push_items = max_push;
149479f95c82SChris Mason 
14955f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
14965f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
14975f39d397SChris Mason 				      (dst_nritems) *
14985f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
1499d6025579SChris Mason 
15005f39d397SChris Mason 	copy_extent_buffer(dst, src,
15015f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
15025f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
1503123abc88SChris Mason 		           push_items * sizeof(struct btrfs_key_ptr));
150479f95c82SChris Mason 
15055f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
15065f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
150779f95c82SChris Mason 
15085f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
15095f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
151079f95c82SChris Mason 	return ret;
151179f95c82SChris Mason }
151279f95c82SChris Mason 
151379f95c82SChris Mason /*
151497571fd0SChris Mason  * helper function to insert a new root level in the tree.
151597571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
151697571fd0SChris Mason  * point to the existing root
1517aa5d6bedSChris Mason  *
1518aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
151997571fd0SChris Mason  */
152098ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans,
15215f39d397SChris Mason 			   struct btrfs_root *root,
15225f39d397SChris Mason 			   struct btrfs_path *path, int level)
152374123bd7SChris Mason {
15247bb86316SChris Mason 	u64 root_gen;
15257bb86316SChris Mason 	u64 lower_gen;
15265f39d397SChris Mason 	struct extent_buffer *lower;
15275f39d397SChris Mason 	struct extent_buffer *c;
1528925baeddSChris Mason 	struct extent_buffer *old;
15295f39d397SChris Mason 	struct btrfs_disk_key lower_key;
15305c680ed6SChris Mason 
15315c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
15325c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
15335c680ed6SChris Mason 
15347bb86316SChris Mason 	if (root->ref_cows)
15357bb86316SChris Mason 		root_gen = trans->transid;
15367bb86316SChris Mason 	else
15377bb86316SChris Mason 		root_gen = 0;
15387bb86316SChris Mason 
15397bb86316SChris Mason 	lower = path->nodes[level-1];
15407bb86316SChris Mason 	if (level == 1)
15417bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
15427bb86316SChris Mason 	else
15437bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
15447bb86316SChris Mason 
1545925baeddSChris Mason 	c = btrfs_alloc_free_block(trans, root, root->nodesize,
15467bb86316SChris Mason 				   root->root_key.objectid,
15477bb86316SChris Mason 				   root_gen, lower_key.objectid, level,
1548db94535dSChris Mason 				   root->node->start, 0);
15495f39d397SChris Mason 	if (IS_ERR(c))
15505f39d397SChris Mason 		return PTR_ERR(c);
1551925baeddSChris Mason 
15525f39d397SChris Mason 	memset_extent_buffer(c, 0, 0, root->nodesize);
15535f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
15545f39d397SChris Mason 	btrfs_set_header_level(c, level);
1555db94535dSChris Mason 	btrfs_set_header_bytenr(c, c->start);
15565f39d397SChris Mason 	btrfs_set_header_generation(c, trans->transid);
15575f39d397SChris Mason 	btrfs_set_header_owner(c, root->root_key.objectid);
1558d5719762SChris Mason 
15595f39d397SChris Mason 	write_extent_buffer(c, root->fs_info->fsid,
15605f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(c),
15615f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1562e17cade2SChris Mason 
1563e17cade2SChris Mason 	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
1564e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
1565e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
1566e17cade2SChris Mason 
15675f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
1568db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
15697bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
15707bb86316SChris Mason 	WARN_ON(lower_gen == 0);
15717bb86316SChris Mason 
15727bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
15735f39d397SChris Mason 
15745f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
1575d5719762SChris Mason 
1576925baeddSChris Mason 	spin_lock(&root->node_lock);
1577925baeddSChris Mason 	old = root->node;
15785f39d397SChris Mason 	root->node = c;
1579925baeddSChris Mason 	spin_unlock(&root->node_lock);
1580925baeddSChris Mason 
1581925baeddSChris Mason 	/* the super has an extra ref to root->node */
1582925baeddSChris Mason 	free_extent_buffer(old);
1583925baeddSChris Mason 
15840b86a832SChris Mason 	add_root_to_dirty_list(root);
15855f39d397SChris Mason 	extent_buffer_get(c);
15865f39d397SChris Mason 	path->nodes[level] = c;
1587925baeddSChris Mason 	path->locks[level] = 1;
158874123bd7SChris Mason 	path->slots[level] = 0;
15897bb86316SChris Mason 
15907bb86316SChris Mason 	if (root->ref_cows && lower_gen != trans->transid) {
15917bb86316SChris Mason 		struct btrfs_path *back_path = btrfs_alloc_path();
15927bb86316SChris Mason 		int ret;
1593925baeddSChris Mason 		mutex_lock(&root->fs_info->alloc_mutex);
15947bb86316SChris Mason 		ret = btrfs_insert_extent_backref(trans,
15957bb86316SChris Mason 						  root->fs_info->extent_root,
15967bb86316SChris Mason 						  path, lower->start,
15977bb86316SChris Mason 						  root->root_key.objectid,
15987bb86316SChris Mason 						  trans->transid, 0, 0);
15997bb86316SChris Mason 		BUG_ON(ret);
1600925baeddSChris Mason 		mutex_unlock(&root->fs_info->alloc_mutex);
16017bb86316SChris Mason 		btrfs_free_path(back_path);
16027bb86316SChris Mason 	}
160374123bd7SChris Mason 	return 0;
160474123bd7SChris Mason }
16055c680ed6SChris Mason 
16065c680ed6SChris Mason /*
16075c680ed6SChris Mason  * worker function to insert a single pointer in a node.
16085c680ed6SChris Mason  * the node should have enough room for the pointer already
160997571fd0SChris Mason  *
16105c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
16115c680ed6SChris Mason  * blocknr is the block the key points to.
1612aa5d6bedSChris Mason  *
1613aa5d6bedSChris Mason  * returns zero on success and < 0 on any error
16145c680ed6SChris Mason  */
1615e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1616e089f05cSChris Mason 		      *root, struct btrfs_path *path, struct btrfs_disk_key
1617db94535dSChris Mason 		      *key, u64 bytenr, int slot, int level)
16185c680ed6SChris Mason {
16195f39d397SChris Mason 	struct extent_buffer *lower;
16205c680ed6SChris Mason 	int nritems;
16215c680ed6SChris Mason 
16225c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
16235f39d397SChris Mason 	lower = path->nodes[level];
16245f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
162574123bd7SChris Mason 	if (slot > nritems)
162674123bd7SChris Mason 		BUG();
1627123abc88SChris Mason 	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
162874123bd7SChris Mason 		BUG();
162974123bd7SChris Mason 	if (slot != nritems) {
16305f39d397SChris Mason 		memmove_extent_buffer(lower,
16315f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
16325f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
1633123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
163474123bd7SChris Mason 	}
16355f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
1636db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
163774493f7aSChris Mason 	WARN_ON(trans->transid == 0);
163874493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
16395f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
16405f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
164174123bd7SChris Mason 	return 0;
164274123bd7SChris Mason }
164374123bd7SChris Mason 
164497571fd0SChris Mason /*
164597571fd0SChris Mason  * split the node at the specified level in path in two.
164697571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
164797571fd0SChris Mason  *
164897571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
164997571fd0SChris Mason  * left and right, if either one works, it returns right away.
1650aa5d6bedSChris Mason  *
1651aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
165297571fd0SChris Mason  */
1653e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1654e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level)
1655be0e5c09SChris Mason {
16567bb86316SChris Mason 	u64 root_gen;
16575f39d397SChris Mason 	struct extent_buffer *c;
16585f39d397SChris Mason 	struct extent_buffer *split;
16595f39d397SChris Mason 	struct btrfs_disk_key disk_key;
1660be0e5c09SChris Mason 	int mid;
16615c680ed6SChris Mason 	int ret;
1662aa5d6bedSChris Mason 	int wret;
16637518a238SChris Mason 	u32 c_nritems;
1664be0e5c09SChris Mason 
16655f39d397SChris Mason 	c = path->nodes[level];
16667bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
16675f39d397SChris Mason 	if (c == root->node) {
16685c680ed6SChris Mason 		/* trying to split the root, lets make a new one */
1669e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, level + 1);
16705c680ed6SChris Mason 		if (ret)
16715c680ed6SChris Mason 			return ret;
1672e66f709bSChris Mason 	} else {
1673e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
16745f39d397SChris Mason 		c = path->nodes[level];
16755f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
1676c448acf0SChris Mason 		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
1677e66f709bSChris Mason 			return 0;
167854aa1f4dSChris Mason 		if (ret < 0)
167954aa1f4dSChris Mason 			return ret;
16805c680ed6SChris Mason 	}
1681e66f709bSChris Mason 
16825f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
16837bb86316SChris Mason 	if (root->ref_cows)
16847bb86316SChris Mason 		root_gen = trans->transid;
16857bb86316SChris Mason 	else
16867bb86316SChris Mason 		root_gen = 0;
16877bb86316SChris Mason 
16887bb86316SChris Mason 	btrfs_node_key(c, &disk_key, 0);
1689925baeddSChris Mason 	split = btrfs_alloc_free_block(trans, root, root->nodesize,
16907bb86316SChris Mason 					 root->root_key.objectid,
16917bb86316SChris Mason 					 root_gen,
16927bb86316SChris Mason 					 btrfs_disk_key_objectid(&disk_key),
16937bb86316SChris Mason 					 level, c->start, 0);
16945f39d397SChris Mason 	if (IS_ERR(split))
16955f39d397SChris Mason 		return PTR_ERR(split);
169654aa1f4dSChris Mason 
16975f39d397SChris Mason 	btrfs_set_header_flags(split, btrfs_header_flags(c));
16985f39d397SChris Mason 	btrfs_set_header_level(split, btrfs_header_level(c));
1699db94535dSChris Mason 	btrfs_set_header_bytenr(split, split->start);
17005f39d397SChris Mason 	btrfs_set_header_generation(split, trans->transid);
17015f39d397SChris Mason 	btrfs_set_header_owner(split, root->root_key.objectid);
170263b10fc4SChris Mason 	btrfs_set_header_flags(split, 0);
17035f39d397SChris Mason 	write_extent_buffer(split, root->fs_info->fsid,
17045f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(split),
17055f39d397SChris Mason 			    BTRFS_FSID_SIZE);
1706e17cade2SChris Mason 	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
1707e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
1708e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
17095f39d397SChris Mason 
17107518a238SChris Mason 	mid = (c_nritems + 1) / 2;
17115f39d397SChris Mason 
17125f39d397SChris Mason 	copy_extent_buffer(split, c,
17135f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
17145f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
1715123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
17165f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
17175f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
1718aa5d6bedSChris Mason 	ret = 0;
1719aa5d6bedSChris Mason 
17205f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
17215f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
17225f39d397SChris Mason 
17235f39d397SChris Mason 	btrfs_node_key(split, &disk_key, 0);
1724db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, split->start,
17255f39d397SChris Mason 			  path->slots[level + 1] + 1,
1726123abc88SChris Mason 			  level + 1);
1727aa5d6bedSChris Mason 	if (wret)
1728aa5d6bedSChris Mason 		ret = wret;
1729aa5d6bedSChris Mason 
17305de08d7dSChris Mason 	if (path->slots[level] >= mid) {
17315c680ed6SChris Mason 		path->slots[level] -= mid;
1732925baeddSChris Mason 		btrfs_tree_unlock(c);
17335f39d397SChris Mason 		free_extent_buffer(c);
17345f39d397SChris Mason 		path->nodes[level] = split;
17355c680ed6SChris Mason 		path->slots[level + 1] += 1;
1736eb60ceacSChris Mason 	} else {
1737925baeddSChris Mason 		btrfs_tree_unlock(split);
17385f39d397SChris Mason 		free_extent_buffer(split);
1739be0e5c09SChris Mason 	}
1740aa5d6bedSChris Mason 	return ret;
1741be0e5c09SChris Mason }
1742be0e5c09SChris Mason 
174374123bd7SChris Mason /*
174474123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
174574123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
174674123bd7SChris Mason  * space used both by the item structs and the item data
174774123bd7SChris Mason  */
17485f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1749be0e5c09SChris Mason {
1750be0e5c09SChris Mason 	int data_len;
17515f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
1752d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
1753be0e5c09SChris Mason 
1754be0e5c09SChris Mason 	if (!nr)
1755be0e5c09SChris Mason 		return 0;
17565f39d397SChris Mason 	data_len = btrfs_item_end_nr(l, start);
17575f39d397SChris Mason 	data_len = data_len - btrfs_item_offset_nr(l, end);
17580783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
1759d4dbff95SChris Mason 	WARN_ON(data_len < 0);
1760be0e5c09SChris Mason 	return data_len;
1761be0e5c09SChris Mason }
1762be0e5c09SChris Mason 
176374123bd7SChris Mason /*
1764d4dbff95SChris Mason  * The space between the end of the leaf items and
1765d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
1766d4dbff95SChris Mason  * the leaf has left for both items and data
1767d4dbff95SChris Mason  */
17685f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1769d4dbff95SChris Mason {
17705f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
17715f39d397SChris Mason 	int ret;
17725f39d397SChris Mason 	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
17735f39d397SChris Mason 	if (ret < 0) {
17745f39d397SChris Mason 		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1775ae2f5411SJens Axboe 		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
17765f39d397SChris Mason 		       leaf_space_used(leaf, 0, nritems), nritems);
17775f39d397SChris Mason 	}
17785f39d397SChris Mason 	return ret;
1779d4dbff95SChris Mason }
1780d4dbff95SChris Mason 
1781d4dbff95SChris Mason /*
178200ec4c51SChris Mason  * push some data in the path leaf to the right, trying to free up at
178300ec4c51SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
1784aa5d6bedSChris Mason  *
1785aa5d6bedSChris Mason  * returns 1 if the push failed because the other node didn't have enough
1786aa5d6bedSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
178700ec4c51SChris Mason  */
1788e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
178934a38218SChris Mason 			   *root, struct btrfs_path *path, int data_size,
179034a38218SChris Mason 			   int empty)
179100ec4c51SChris Mason {
17925f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
17935f39d397SChris Mason 	struct extent_buffer *right;
17945f39d397SChris Mason 	struct extent_buffer *upper;
17955f39d397SChris Mason 	struct btrfs_disk_key disk_key;
179600ec4c51SChris Mason 	int slot;
179734a38218SChris Mason 	u32 i;
179800ec4c51SChris Mason 	int free_space;
179900ec4c51SChris Mason 	int push_space = 0;
180000ec4c51SChris Mason 	int push_items = 0;
18010783fcfcSChris Mason 	struct btrfs_item *item;
18027518a238SChris Mason 	u32 left_nritems;
180334a38218SChris Mason 	u32 nr;
18047518a238SChris Mason 	u32 right_nritems;
18055f39d397SChris Mason 	u32 data_end;
1806db94535dSChris Mason 	u32 this_item_size;
180754aa1f4dSChris Mason 	int ret;
180800ec4c51SChris Mason 
180900ec4c51SChris Mason 	slot = path->slots[1];
181000ec4c51SChris Mason 	if (!path->nodes[1]) {
181100ec4c51SChris Mason 		return 1;
181200ec4c51SChris Mason 	}
181300ec4c51SChris Mason 	upper = path->nodes[1];
18145f39d397SChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
181500ec4c51SChris Mason 		return 1;
18165f39d397SChris Mason 
1817*a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
1818*a2135011SChris Mason 
1819ca7a79adSChris Mason 	right = read_node_slot(root, upper, slot + 1);
1820925baeddSChris Mason 	btrfs_tree_lock(right);
1821123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1822925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1823925baeddSChris Mason 		goto out_unlock;
182402217ed2SChris Mason 
18255f39d397SChris Mason 	/* cow and double check */
18265f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
18275f39d397SChris Mason 			      slot + 1, &right);
1828925baeddSChris Mason 	if (ret)
1829925baeddSChris Mason 		goto out_unlock;
1830925baeddSChris Mason 
18315f39d397SChris Mason 	free_space = btrfs_leaf_free_space(root, right);
1832925baeddSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item))
1833925baeddSChris Mason 		goto out_unlock;
18345f39d397SChris Mason 
18355f39d397SChris Mason 	left_nritems = btrfs_header_nritems(left);
1836925baeddSChris Mason 	if (left_nritems == 0)
1837925baeddSChris Mason 		goto out_unlock;
18385f39d397SChris Mason 
183934a38218SChris Mason 	if (empty)
184034a38218SChris Mason 		nr = 0;
184134a38218SChris Mason 	else
184234a38218SChris Mason 		nr = 1;
184334a38218SChris Mason 
184434a38218SChris Mason 	i = left_nritems - 1;
184534a38218SChris Mason 	while (i >= nr) {
18465f39d397SChris Mason 		item = btrfs_item_nr(left, i);
1847db94535dSChris Mason 
184800ec4c51SChris Mason 		if (path->slots[0] == i)
184900ec4c51SChris Mason 			push_space += data_size + sizeof(*item);
1850db94535dSChris Mason 
1851db94535dSChris Mason 		if (!left->map_token) {
1852db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
1853db94535dSChris Mason 					sizeof(struct btrfs_item),
1854db94535dSChris Mason 					&left->map_token, &left->kaddr,
1855db94535dSChris Mason 					&left->map_start, &left->map_len,
1856db94535dSChris Mason 					KM_USER1);
1857db94535dSChris Mason 		}
1858db94535dSChris Mason 
1859db94535dSChris Mason 		this_item_size = btrfs_item_size(left, item);
1860db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
186100ec4c51SChris Mason 			break;
186200ec4c51SChris Mason 		push_items++;
1863db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
186434a38218SChris Mason 		if (i == 0)
186534a38218SChris Mason 			break;
186634a38218SChris Mason 		i--;
1867db94535dSChris Mason 	}
1868db94535dSChris Mason 	if (left->map_token) {
1869db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
1870db94535dSChris Mason 		left->map_token = NULL;
187100ec4c51SChris Mason 	}
18725f39d397SChris Mason 
1873925baeddSChris Mason 	if (push_items == 0)
1874925baeddSChris Mason 		goto out_unlock;
18755f39d397SChris Mason 
187634a38218SChris Mason 	if (!empty && push_items == left_nritems)
1877a429e513SChris Mason 		WARN_ON(1);
18785f39d397SChris Mason 
187900ec4c51SChris Mason 	/* push left to right */
18805f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
188134a38218SChris Mason 
18825f39d397SChris Mason 	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
1883123abc88SChris Mason 	push_space -= leaf_data_end(root, left);
18845f39d397SChris Mason 
188500ec4c51SChris Mason 	/* make room in the right data area */
18865f39d397SChris Mason 	data_end = leaf_data_end(root, right);
18875f39d397SChris Mason 	memmove_extent_buffer(right,
18885f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end - push_space,
18895f39d397SChris Mason 			      btrfs_leaf_data(right) + data_end,
18905f39d397SChris Mason 			      BTRFS_LEAF_DATA_SIZE(root) - data_end);
18915f39d397SChris Mason 
189200ec4c51SChris Mason 	/* copy from the left data area */
18935f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
1894d6025579SChris Mason 		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
1895d6025579SChris Mason 		     btrfs_leaf_data(left) + leaf_data_end(root, left),
1896d6025579SChris Mason 		     push_space);
18975f39d397SChris Mason 
18985f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
18995f39d397SChris Mason 			      btrfs_item_nr_offset(0),
19000783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
19015f39d397SChris Mason 
190200ec4c51SChris Mason 	/* copy the items from left to right */
19035f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
19045f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
19050783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
190600ec4c51SChris Mason 
190700ec4c51SChris Mason 	/* update the item pointers */
19087518a238SChris Mason 	right_nritems += push_items;
19095f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
1910123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
19117518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
19125f39d397SChris Mason 		item = btrfs_item_nr(right, i);
1913db94535dSChris Mason 		if (!right->map_token) {
1914db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
1915db94535dSChris Mason 					sizeof(struct btrfs_item),
1916db94535dSChris Mason 					&right->map_token, &right->kaddr,
1917db94535dSChris Mason 					&right->map_start, &right->map_len,
1918db94535dSChris Mason 					KM_USER1);
1919db94535dSChris Mason 		}
1920db94535dSChris Mason 		push_space -= btrfs_item_size(right, item);
1921db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
1922db94535dSChris Mason 	}
1923db94535dSChris Mason 
1924db94535dSChris Mason 	if (right->map_token) {
1925db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
1926db94535dSChris Mason 		right->map_token = NULL;
192700ec4c51SChris Mason 	}
19287518a238SChris Mason 	left_nritems -= push_items;
19295f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
193000ec4c51SChris Mason 
193134a38218SChris Mason 	if (left_nritems)
19325f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
19335f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
1934a429e513SChris Mason 
19355f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
19365f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
1937d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
193802217ed2SChris Mason 
193900ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
19407518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
19417518a238SChris Mason 		path->slots[0] -= left_nritems;
1942925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
1943925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
1944925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
19455f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
19465f39d397SChris Mason 		path->nodes[0] = right;
194700ec4c51SChris Mason 		path->slots[1] += 1;
194800ec4c51SChris Mason 	} else {
1949925baeddSChris Mason 		btrfs_tree_unlock(right);
19505f39d397SChris Mason 		free_extent_buffer(right);
195100ec4c51SChris Mason 	}
195200ec4c51SChris Mason 	return 0;
1953925baeddSChris Mason 
1954925baeddSChris Mason out_unlock:
1955925baeddSChris Mason 	btrfs_tree_unlock(right);
1956925baeddSChris Mason 	free_extent_buffer(right);
1957925baeddSChris Mason 	return 1;
195800ec4c51SChris Mason }
1959925baeddSChris Mason 
196000ec4c51SChris Mason /*
196174123bd7SChris Mason  * push some data in the path leaf to the left, trying to free up at
196274123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
196374123bd7SChris Mason  */
1964e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
196534a38218SChris Mason 			  *root, struct btrfs_path *path, int data_size,
196634a38218SChris Mason 			  int empty)
1967be0e5c09SChris Mason {
19685f39d397SChris Mason 	struct btrfs_disk_key disk_key;
19695f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
19705f39d397SChris Mason 	struct extent_buffer *left;
1971be0e5c09SChris Mason 	int slot;
1972be0e5c09SChris Mason 	int i;
1973be0e5c09SChris Mason 	int free_space;
1974be0e5c09SChris Mason 	int push_space = 0;
1975be0e5c09SChris Mason 	int push_items = 0;
19760783fcfcSChris Mason 	struct btrfs_item *item;
19777518a238SChris Mason 	u32 old_left_nritems;
19785f39d397SChris Mason 	u32 right_nritems;
197934a38218SChris Mason 	u32 nr;
1980aa5d6bedSChris Mason 	int ret = 0;
1981aa5d6bedSChris Mason 	int wret;
1982db94535dSChris Mason 	u32 this_item_size;
1983db94535dSChris Mason 	u32 old_left_item_size;
1984be0e5c09SChris Mason 
1985be0e5c09SChris Mason 	slot = path->slots[1];
19865f39d397SChris Mason 	if (slot == 0)
1987be0e5c09SChris Mason 		return 1;
19885f39d397SChris Mason 	if (!path->nodes[1])
1989be0e5c09SChris Mason 		return 1;
19905f39d397SChris Mason 
19913685f791SChris Mason 	right_nritems = btrfs_header_nritems(right);
19923685f791SChris Mason 	if (right_nritems == 0) {
19933685f791SChris Mason 		return 1;
19943685f791SChris Mason 	}
19953685f791SChris Mason 
1996*a2135011SChris Mason 	WARN_ON(!btrfs_tree_locked(path->nodes[1]));
1997*a2135011SChris Mason 
1998ca7a79adSChris Mason 	left = read_node_slot(root, path->nodes[1], slot - 1);
1999925baeddSChris Mason 	btrfs_tree_lock(left);
2000123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20010783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2002925baeddSChris Mason 		ret = 1;
2003925baeddSChris Mason 		goto out;
2004be0e5c09SChris Mason 	}
200502217ed2SChris Mason 
200602217ed2SChris Mason 	/* cow and double check */
20075f39d397SChris Mason 	ret = btrfs_cow_block(trans, root, left,
20085f39d397SChris Mason 			      path->nodes[1], slot - 1, &left);
200954aa1f4dSChris Mason 	if (ret) {
201054aa1f4dSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
2011925baeddSChris Mason 		ret = 1;
2012925baeddSChris Mason 		goto out;
201354aa1f4dSChris Mason 	}
20143685f791SChris Mason 
2015123abc88SChris Mason 	free_space = btrfs_leaf_free_space(root, left);
20160783fcfcSChris Mason 	if (free_space < data_size + sizeof(struct btrfs_item)) {
2017925baeddSChris Mason 		ret = 1;
2018925baeddSChris Mason 		goto out;
201902217ed2SChris Mason 	}
202002217ed2SChris Mason 
202134a38218SChris Mason 	if (empty)
202234a38218SChris Mason 		nr = right_nritems;
202334a38218SChris Mason 	else
202434a38218SChris Mason 		nr = right_nritems - 1;
202534a38218SChris Mason 
202634a38218SChris Mason 	for (i = 0; i < nr; i++) {
20275f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2028db94535dSChris Mason 		if (!right->map_token) {
2029db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2030db94535dSChris Mason 					sizeof(struct btrfs_item),
2031db94535dSChris Mason 					&right->map_token, &right->kaddr,
2032db94535dSChris Mason 					&right->map_start, &right->map_len,
2033db94535dSChris Mason 					KM_USER1);
2034db94535dSChris Mason 		}
2035db94535dSChris Mason 
2036be0e5c09SChris Mason 		if (path->slots[0] == i)
2037be0e5c09SChris Mason 			push_space += data_size + sizeof(*item);
2038db94535dSChris Mason 
2039db94535dSChris Mason 		this_item_size = btrfs_item_size(right, item);
2040db94535dSChris Mason 		if (this_item_size + sizeof(*item) + push_space > free_space)
2041be0e5c09SChris Mason 			break;
2042db94535dSChris Mason 
2043be0e5c09SChris Mason 		push_items++;
2044db94535dSChris Mason 		push_space += this_item_size + sizeof(*item);
2045be0e5c09SChris Mason 	}
2046db94535dSChris Mason 
2047db94535dSChris Mason 	if (right->map_token) {
2048db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2049db94535dSChris Mason 		right->map_token = NULL;
2050db94535dSChris Mason 	}
2051db94535dSChris Mason 
2052be0e5c09SChris Mason 	if (push_items == 0) {
2053925baeddSChris Mason 		ret = 1;
2054925baeddSChris Mason 		goto out;
2055be0e5c09SChris Mason 	}
205634a38218SChris Mason 	if (!empty && push_items == btrfs_header_nritems(right))
2057a429e513SChris Mason 		WARN_ON(1);
20585f39d397SChris Mason 
2059be0e5c09SChris Mason 	/* push data from right to left */
20605f39d397SChris Mason 	copy_extent_buffer(left, right,
20615f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
20625f39d397SChris Mason 			   btrfs_item_nr_offset(0),
20635f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
20645f39d397SChris Mason 
2065123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root) -
20665f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items -1);
20675f39d397SChris Mason 
20685f39d397SChris Mason 	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
2069d6025579SChris Mason 		     leaf_data_end(root, left) - push_space,
2070123abc88SChris Mason 		     btrfs_leaf_data(right) +
20715f39d397SChris Mason 		     btrfs_item_offset_nr(right, push_items - 1),
2072be0e5c09SChris Mason 		     push_space);
20735f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
2074eb60ceacSChris Mason 	BUG_ON(old_left_nritems < 0);
2075eb60ceacSChris Mason 
2076db94535dSChris Mason 	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
2077be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
20785f39d397SChris Mason 		u32 ioff;
2079db94535dSChris Mason 
20805f39d397SChris Mason 		item = btrfs_item_nr(left, i);
2081db94535dSChris Mason 		if (!left->map_token) {
2082db94535dSChris Mason 			map_extent_buffer(left, (unsigned long)item,
2083db94535dSChris Mason 					sizeof(struct btrfs_item),
2084db94535dSChris Mason 					&left->map_token, &left->kaddr,
2085db94535dSChris Mason 					&left->map_start, &left->map_len,
2086db94535dSChris Mason 					KM_USER1);
2087db94535dSChris Mason 		}
2088db94535dSChris Mason 
20895f39d397SChris Mason 		ioff = btrfs_item_offset(left, item);
20905f39d397SChris Mason 		btrfs_set_item_offset(left, item,
2091db94535dSChris Mason 		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2092be0e5c09SChris Mason 	}
20935f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2094db94535dSChris Mason 	if (left->map_token) {
2095db94535dSChris Mason 		unmap_extent_buffer(left, left->map_token, KM_USER1);
2096db94535dSChris Mason 		left->map_token = NULL;
2097db94535dSChris Mason 	}
2098be0e5c09SChris Mason 
2099be0e5c09SChris Mason 	/* fixup right node */
210034a38218SChris Mason 	if (push_items > right_nritems) {
210134a38218SChris Mason 		printk("push items %d nr %u\n", push_items, right_nritems);
210234a38218SChris Mason 		WARN_ON(1);
210334a38218SChris Mason 	}
210434a38218SChris Mason 
210534a38218SChris Mason 	if (push_items < right_nritems) {
21065f39d397SChris Mason 		push_space = btrfs_item_offset_nr(right, push_items - 1) -
2107123abc88SChris Mason 						  leaf_data_end(root, right);
21085f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_leaf_data(right) +
2109d6025579SChris Mason 				      BTRFS_LEAF_DATA_SIZE(root) - push_space,
2110d6025579SChris Mason 				      btrfs_leaf_data(right) +
2111123abc88SChris Mason 				      leaf_data_end(root, right), push_space);
21125f39d397SChris Mason 
21135f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
21145f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
21155f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
21160783fcfcSChris Mason 			     sizeof(struct btrfs_item));
211734a38218SChris Mason 	}
2118eef1c494SYan 	right_nritems -= push_items;
2119eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
2120123abc88SChris Mason 	push_space = BTRFS_LEAF_DATA_SIZE(root);
21215f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
21225f39d397SChris Mason 		item = btrfs_item_nr(right, i);
2123db94535dSChris Mason 
2124db94535dSChris Mason 		if (!right->map_token) {
2125db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2126db94535dSChris Mason 					sizeof(struct btrfs_item),
2127db94535dSChris Mason 					&right->map_token, &right->kaddr,
2128db94535dSChris Mason 					&right->map_start, &right->map_len,
2129db94535dSChris Mason 					KM_USER1);
2130db94535dSChris Mason 		}
2131db94535dSChris Mason 
2132db94535dSChris Mason 		push_space = push_space - btrfs_item_size(right, item);
2133db94535dSChris Mason 		btrfs_set_item_offset(right, item, push_space);
2134db94535dSChris Mason 	}
2135db94535dSChris Mason 	if (right->map_token) {
2136db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2137db94535dSChris Mason 		right->map_token = NULL;
2138be0e5c09SChris Mason 	}
2139eb60ceacSChris Mason 
21405f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
214134a38218SChris Mason 	if (right_nritems)
21425f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
2143098f59c2SChris Mason 
21445f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
21455f39d397SChris Mason 	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
2146aa5d6bedSChris Mason 	if (wret)
2147aa5d6bedSChris Mason 		ret = wret;
2148be0e5c09SChris Mason 
2149be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
2150be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
2151be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
2152925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
2153925baeddSChris Mason 			clean_tree_block(trans, root, path->nodes[0]);
2154925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
21555f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
21565f39d397SChris Mason 		path->nodes[0] = left;
2157be0e5c09SChris Mason 		path->slots[1] -= 1;
2158be0e5c09SChris Mason 	} else {
2159925baeddSChris Mason 		btrfs_tree_unlock(left);
21605f39d397SChris Mason 		free_extent_buffer(left);
2161be0e5c09SChris Mason 		path->slots[0] -= push_items;
2162be0e5c09SChris Mason 	}
2163eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2164aa5d6bedSChris Mason 	return ret;
2165925baeddSChris Mason out:
2166925baeddSChris Mason 	btrfs_tree_unlock(left);
2167925baeddSChris Mason 	free_extent_buffer(left);
2168925baeddSChris Mason 	return ret;
2169be0e5c09SChris Mason }
2170be0e5c09SChris Mason 
217174123bd7SChris Mason /*
217274123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
217374123bd7SChris Mason  * available for the resulting leaf level of the path.
2174aa5d6bedSChris Mason  *
2175aa5d6bedSChris Mason  * returns 0 if all went well and < 0 on failure.
217674123bd7SChris Mason  */
2177e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
2178d4dbff95SChris Mason 		      *root, struct btrfs_key *ins_key,
2179cc0c5538SChris Mason 		      struct btrfs_path *path, int data_size, int extend)
2180be0e5c09SChris Mason {
21817bb86316SChris Mason 	u64 root_gen;
21825f39d397SChris Mason 	struct extent_buffer *l;
21837518a238SChris Mason 	u32 nritems;
2184eb60ceacSChris Mason 	int mid;
2185eb60ceacSChris Mason 	int slot;
21865f39d397SChris Mason 	struct extent_buffer *right;
21870783fcfcSChris Mason 	int space_needed = data_size + sizeof(struct btrfs_item);
2188be0e5c09SChris Mason 	int data_copy_size;
2189be0e5c09SChris Mason 	int rt_data_off;
2190be0e5c09SChris Mason 	int i;
2191d4dbff95SChris Mason 	int ret = 0;
2192aa5d6bedSChris Mason 	int wret;
2193cc0c5538SChris Mason 	int double_split;
2194cc0c5538SChris Mason 	int num_doubles = 0;
2195d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
2196be0e5c09SChris Mason 
2197cc0c5538SChris Mason 	if (extend)
2198cc0c5538SChris Mason 		space_needed = data_size;
2199cc0c5538SChris Mason 
22007bb86316SChris Mason 	if (root->ref_cows)
22017bb86316SChris Mason 		root_gen = trans->transid;
22027bb86316SChris Mason 	else
22037bb86316SChris Mason 		root_gen = 0;
22047bb86316SChris Mason 
220540689478SChris Mason 	/* first try to make some room by pushing left and right */
22063685f791SChris Mason 	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
220734a38218SChris Mason 		wret = push_leaf_right(trans, root, path, data_size, 0);
22083326d1b0SChris Mason 		if (wret < 0) {
2209eaee50e8SChris Mason 			return wret;
22103326d1b0SChris Mason 		}
2211eaee50e8SChris Mason 		if (wret) {
221234a38218SChris Mason 			wret = push_leaf_left(trans, root, path, data_size, 0);
2213eaee50e8SChris Mason 			if (wret < 0)
2214eaee50e8SChris Mason 				return wret;
2215eaee50e8SChris Mason 		}
22165f39d397SChris Mason 		l = path->nodes[0];
2217aa5d6bedSChris Mason 
2218aa5d6bedSChris Mason 		/* did the pushes work? */
2219cc0c5538SChris Mason 		if (btrfs_leaf_free_space(root, l) >= space_needed)
2220be0e5c09SChris Mason 			return 0;
22213326d1b0SChris Mason 	}
2222aa5d6bedSChris Mason 
22235c680ed6SChris Mason 	if (!path->nodes[1]) {
2224e089f05cSChris Mason 		ret = insert_new_root(trans, root, path, 1);
22255c680ed6SChris Mason 		if (ret)
22265c680ed6SChris Mason 			return ret;
22275c680ed6SChris Mason 	}
2228cc0c5538SChris Mason again:
2229cc0c5538SChris Mason 	double_split = 0;
2230cc0c5538SChris Mason 	l = path->nodes[0];
2231eb60ceacSChris Mason 	slot = path->slots[0];
22325f39d397SChris Mason 	nritems = btrfs_header_nritems(l);
2233eb60ceacSChris Mason 	mid = (nritems + 1)/ 2;
223454aa1f4dSChris Mason 
22357bb86316SChris Mason 	btrfs_item_key(l, &disk_key, 0);
22367bb86316SChris Mason 
2237925baeddSChris Mason 	right = btrfs_alloc_free_block(trans, root, root->leafsize,
22387bb86316SChris Mason 					 root->root_key.objectid,
22397bb86316SChris Mason 					 root_gen, disk_key.objectid, 0,
2240db94535dSChris Mason 					 l->start, 0);
2241cea9e445SChris Mason 	if (IS_ERR(right)) {
2242cea9e445SChris Mason 		BUG_ON(1);
22435f39d397SChris Mason 		return PTR_ERR(right);
2244cea9e445SChris Mason 	}
224554aa1f4dSChris Mason 
22465f39d397SChris Mason 	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2247db94535dSChris Mason 	btrfs_set_header_bytenr(right, right->start);
22485f39d397SChris Mason 	btrfs_set_header_generation(right, trans->transid);
22495f39d397SChris Mason 	btrfs_set_header_owner(right, root->root_key.objectid);
22505f39d397SChris Mason 	btrfs_set_header_level(right, 0);
22515f39d397SChris Mason 	write_extent_buffer(right, root->fs_info->fsid,
22525f39d397SChris Mason 			    (unsigned long)btrfs_header_fsid(right),
22535f39d397SChris Mason 			    BTRFS_FSID_SIZE);
2254e17cade2SChris Mason 
2255e17cade2SChris Mason 	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
2256e17cade2SChris Mason 			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
2257e17cade2SChris Mason 			    BTRFS_UUID_SIZE);
2258d4dbff95SChris Mason 	if (mid <= slot) {
2259d4dbff95SChris Mason 		if (nritems == 1 ||
2260d4dbff95SChris Mason 		    leaf_space_used(l, mid, nritems - mid) + space_needed >
2261d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2262d4dbff95SChris Mason 			if (slot >= nritems) {
2263d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
22645f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2265d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2266db94535dSChris Mason 						  &disk_key, right->start,
2267d4dbff95SChris Mason 						  path->slots[1] + 1, 1);
2268d4dbff95SChris Mason 				if (wret)
2269d4dbff95SChris Mason 					ret = wret;
2270925baeddSChris Mason 
2271925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
22725f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
22735f39d397SChris Mason 				path->nodes[0] = right;
2274d4dbff95SChris Mason 				path->slots[0] = 0;
2275d4dbff95SChris Mason 				path->slots[1] += 1;
22760ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2277d4dbff95SChris Mason 				return ret;
2278d4dbff95SChris Mason 			}
2279d4dbff95SChris Mason 			mid = slot;
22803326d1b0SChris Mason 			if (mid != nritems &&
22813326d1b0SChris Mason 			    leaf_space_used(l, mid, nritems - mid) +
22823326d1b0SChris Mason 			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2283d4dbff95SChris Mason 				double_split = 1;
2284d4dbff95SChris Mason 			}
22853326d1b0SChris Mason 		}
2286d4dbff95SChris Mason 	} else {
2287d4dbff95SChris Mason 		if (leaf_space_used(l, 0, mid + 1) + space_needed >
2288d4dbff95SChris Mason 			BTRFS_LEAF_DATA_SIZE(root)) {
2289cc0c5538SChris Mason 			if (!extend && slot == 0) {
2290d4dbff95SChris Mason 				btrfs_cpu_key_to_disk(&disk_key, ins_key);
22915f39d397SChris Mason 				btrfs_set_header_nritems(right, 0);
2292d4dbff95SChris Mason 				wret = insert_ptr(trans, root, path,
2293d4dbff95SChris Mason 						  &disk_key,
2294db94535dSChris Mason 						  right->start,
2295098f59c2SChris Mason 						  path->slots[1], 1);
2296d4dbff95SChris Mason 				if (wret)
2297d4dbff95SChris Mason 					ret = wret;
2298925baeddSChris Mason 				btrfs_tree_unlock(path->nodes[0]);
22995f39d397SChris Mason 				free_extent_buffer(path->nodes[0]);
23005f39d397SChris Mason 				path->nodes[0] = right;
2301d4dbff95SChris Mason 				path->slots[0] = 0;
2302a429e513SChris Mason 				if (path->slots[1] == 0) {
2303a429e513SChris Mason 					wret = fixup_low_keys(trans, root,
2304a429e513SChris Mason 					           path, &disk_key, 1);
2305a429e513SChris Mason 					if (wret)
2306a429e513SChris Mason 						ret = wret;
2307a429e513SChris Mason 				}
23080ef8b242SChris Mason 				btrfs_mark_buffer_dirty(right);
2309d4dbff95SChris Mason 				return ret;
2310cc0c5538SChris Mason 			} else if (extend && slot == 0) {
2311cc0c5538SChris Mason 				mid = 1;
2312cc0c5538SChris Mason 			} else {
2313d4dbff95SChris Mason 				mid = slot;
23145ee78ac7SChris Mason 				if (mid != nritems &&
23155ee78ac7SChris Mason 				    leaf_space_used(l, mid, nritems - mid) +
23165ee78ac7SChris Mason 				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
2317d4dbff95SChris Mason 					double_split = 1;
2318d4dbff95SChris Mason 				}
2319d4dbff95SChris Mason 			}
23205ee78ac7SChris Mason 		}
2321cc0c5538SChris Mason 	}
23225f39d397SChris Mason 	nritems = nritems - mid;
23235f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
23245f39d397SChris Mason 	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
23255f39d397SChris Mason 
23265f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
23275f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
23285f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
23295f39d397SChris Mason 
23305f39d397SChris Mason 	copy_extent_buffer(right, l,
2331d6025579SChris Mason 		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
2332123abc88SChris Mason 		     data_copy_size, btrfs_leaf_data(l) +
2333123abc88SChris Mason 		     leaf_data_end(root, l), data_copy_size);
233474123bd7SChris Mason 
23355f39d397SChris Mason 	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
23365f39d397SChris Mason 		      btrfs_item_end_nr(l, mid);
23375f39d397SChris Mason 
23385f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
23395f39d397SChris Mason 		struct btrfs_item *item = btrfs_item_nr(right, i);
2340db94535dSChris Mason 		u32 ioff;
2341db94535dSChris Mason 
2342db94535dSChris Mason 		if (!right->map_token) {
2343db94535dSChris Mason 			map_extent_buffer(right, (unsigned long)item,
2344db94535dSChris Mason 					sizeof(struct btrfs_item),
2345db94535dSChris Mason 					&right->map_token, &right->kaddr,
2346db94535dSChris Mason 					&right->map_start, &right->map_len,
2347db94535dSChris Mason 					KM_USER1);
2348db94535dSChris Mason 		}
2349db94535dSChris Mason 
2350db94535dSChris Mason 		ioff = btrfs_item_offset(right, item);
23515f39d397SChris Mason 		btrfs_set_item_offset(right, item, ioff + rt_data_off);
23520783fcfcSChris Mason 	}
235374123bd7SChris Mason 
2354db94535dSChris Mason 	if (right->map_token) {
2355db94535dSChris Mason 		unmap_extent_buffer(right, right->map_token, KM_USER1);
2356db94535dSChris Mason 		right->map_token = NULL;
2357db94535dSChris Mason 	}
2358db94535dSChris Mason 
23595f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
2360aa5d6bedSChris Mason 	ret = 0;
23615f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
2362db94535dSChris Mason 	wret = insert_ptr(trans, root, path, &disk_key, right->start,
2363db94535dSChris Mason 			  path->slots[1] + 1, 1);
2364aa5d6bedSChris Mason 	if (wret)
2365aa5d6bedSChris Mason 		ret = wret;
23665f39d397SChris Mason 
23675f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
23685f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
2369eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
23705f39d397SChris Mason 
2371be0e5c09SChris Mason 	if (mid <= slot) {
2372925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
23735f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
23745f39d397SChris Mason 		path->nodes[0] = right;
2375be0e5c09SChris Mason 		path->slots[0] -= mid;
2376be0e5c09SChris Mason 		path->slots[1] += 1;
2377925baeddSChris Mason 	} else {
2378925baeddSChris Mason 		btrfs_tree_unlock(right);
23795f39d397SChris Mason 		free_extent_buffer(right);
2380925baeddSChris Mason 	}
23815f39d397SChris Mason 
2382eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
2383d4dbff95SChris Mason 
2384cc0c5538SChris Mason 	if (double_split) {
2385cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
2386cc0c5538SChris Mason 		num_doubles++;
2387cc0c5538SChris Mason 		goto again;
23883326d1b0SChris Mason 	}
2389be0e5c09SChris Mason 	return ret;
2390be0e5c09SChris Mason }
2391be0e5c09SChris Mason 
2392b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans,
2393b18c6685SChris Mason 			struct btrfs_root *root,
2394b18c6685SChris Mason 			struct btrfs_path *path,
2395179e29e4SChris Mason 			u32 new_size, int from_end)
2396b18c6685SChris Mason {
2397b18c6685SChris Mason 	int ret = 0;
2398b18c6685SChris Mason 	int slot;
2399b18c6685SChris Mason 	int slot_orig;
24005f39d397SChris Mason 	struct extent_buffer *leaf;
24015f39d397SChris Mason 	struct btrfs_item *item;
2402b18c6685SChris Mason 	u32 nritems;
2403b18c6685SChris Mason 	unsigned int data_end;
2404b18c6685SChris Mason 	unsigned int old_data_start;
2405b18c6685SChris Mason 	unsigned int old_size;
2406b18c6685SChris Mason 	unsigned int size_diff;
2407b18c6685SChris Mason 	int i;
2408b18c6685SChris Mason 
2409b18c6685SChris Mason 	slot_orig = path->slots[0];
24105f39d397SChris Mason 	leaf = path->nodes[0];
2411179e29e4SChris Mason 	slot = path->slots[0];
2412179e29e4SChris Mason 
2413179e29e4SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
2414179e29e4SChris Mason 	if (old_size == new_size)
2415179e29e4SChris Mason 		return 0;
2416b18c6685SChris Mason 
24175f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2418b18c6685SChris Mason 	data_end = leaf_data_end(root, leaf);
2419b18c6685SChris Mason 
24205f39d397SChris Mason 	old_data_start = btrfs_item_offset_nr(leaf, slot);
2421179e29e4SChris Mason 
2422b18c6685SChris Mason 	size_diff = old_size - new_size;
2423b18c6685SChris Mason 
2424b18c6685SChris Mason 	BUG_ON(slot < 0);
2425b18c6685SChris Mason 	BUG_ON(slot >= nritems);
2426b18c6685SChris Mason 
2427b18c6685SChris Mason 	/*
2428b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2429b18c6685SChris Mason 	 */
2430b18c6685SChris Mason 	/* first correct the data pointers */
2431b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
24325f39d397SChris Mason 		u32 ioff;
24335f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2434db94535dSChris Mason 
2435db94535dSChris Mason 		if (!leaf->map_token) {
2436db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2437db94535dSChris Mason 					sizeof(struct btrfs_item),
2438db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2439db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2440db94535dSChris Mason 					KM_USER1);
2441db94535dSChris Mason 		}
2442db94535dSChris Mason 
24435f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
24445f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff + size_diff);
2445b18c6685SChris Mason 	}
2446db94535dSChris Mason 
2447db94535dSChris Mason 	if (leaf->map_token) {
2448db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2449db94535dSChris Mason 		leaf->map_token = NULL;
2450db94535dSChris Mason 	}
2451db94535dSChris Mason 
2452b18c6685SChris Mason 	/* shift the data */
2453179e29e4SChris Mason 	if (from_end) {
24545f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2455b18c6685SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2456b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
2457179e29e4SChris Mason 	} else {
2458179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
2459179e29e4SChris Mason 		u64 offset;
2460179e29e4SChris Mason 
2461179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
2462179e29e4SChris Mason 
2463179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
2464179e29e4SChris Mason 			unsigned long ptr;
2465179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
2466179e29e4SChris Mason 
2467179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
2468179e29e4SChris Mason 					    struct btrfs_file_extent_item);
2469179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
2470179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
2471179e29e4SChris Mason 
2472179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
2473179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
2474179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
2475179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
2476179e29e4SChris Mason 				        (unsigned long)fi,
2477179e29e4SChris Mason 				        offsetof(struct btrfs_file_extent_item,
2478179e29e4SChris Mason 						 disk_bytenr));
2479179e29e4SChris Mason 			}
2480179e29e4SChris Mason 		}
2481179e29e4SChris Mason 
2482179e29e4SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2483179e29e4SChris Mason 			      data_end + size_diff, btrfs_leaf_data(leaf) +
2484179e29e4SChris Mason 			      data_end, old_data_start - data_end);
2485179e29e4SChris Mason 
2486179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
2487179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
2488179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
2489179e29e4SChris Mason 		if (slot == 0)
2490179e29e4SChris Mason 			fixup_low_keys(trans, root, path, &disk_key, 1);
2491179e29e4SChris Mason 	}
24925f39d397SChris Mason 
24935f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
24945f39d397SChris Mason 	btrfs_set_item_size(leaf, item, new_size);
24955f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2496b18c6685SChris Mason 
2497b18c6685SChris Mason 	ret = 0;
24985f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
24995f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2500b18c6685SChris Mason 		BUG();
25015f39d397SChris Mason 	}
2502b18c6685SChris Mason 	return ret;
2503b18c6685SChris Mason }
2504b18c6685SChris Mason 
25055f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans,
25065f39d397SChris Mason 		      struct btrfs_root *root, struct btrfs_path *path,
25075f39d397SChris Mason 		      u32 data_size)
25086567e837SChris Mason {
25096567e837SChris Mason 	int ret = 0;
25106567e837SChris Mason 	int slot;
25116567e837SChris Mason 	int slot_orig;
25125f39d397SChris Mason 	struct extent_buffer *leaf;
25135f39d397SChris Mason 	struct btrfs_item *item;
25146567e837SChris Mason 	u32 nritems;
25156567e837SChris Mason 	unsigned int data_end;
25166567e837SChris Mason 	unsigned int old_data;
25176567e837SChris Mason 	unsigned int old_size;
25186567e837SChris Mason 	int i;
25196567e837SChris Mason 
25206567e837SChris Mason 	slot_orig = path->slots[0];
25215f39d397SChris Mason 	leaf = path->nodes[0];
25226567e837SChris Mason 
25235f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
25246567e837SChris Mason 	data_end = leaf_data_end(root, leaf);
25256567e837SChris Mason 
25265f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < data_size) {
25275f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25286567e837SChris Mason 		BUG();
25295f39d397SChris Mason 	}
25306567e837SChris Mason 	slot = path->slots[0];
25315f39d397SChris Mason 	old_data = btrfs_item_end_nr(leaf, slot);
25326567e837SChris Mason 
25336567e837SChris Mason 	BUG_ON(slot < 0);
25343326d1b0SChris Mason 	if (slot >= nritems) {
25353326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
25363326d1b0SChris Mason 		printk("slot %d too large, nritems %d\n", slot, nritems);
25373326d1b0SChris Mason 		BUG_ON(1);
25383326d1b0SChris Mason 	}
25396567e837SChris Mason 
25406567e837SChris Mason 	/*
25416567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
25426567e837SChris Mason 	 */
25436567e837SChris Mason 	/* first correct the data pointers */
25446567e837SChris Mason 	for (i = slot; i < nritems; i++) {
25455f39d397SChris Mason 		u32 ioff;
25465f39d397SChris Mason 		item = btrfs_item_nr(leaf, i);
2547db94535dSChris Mason 
2548db94535dSChris Mason 		if (!leaf->map_token) {
2549db94535dSChris Mason 			map_extent_buffer(leaf, (unsigned long)item,
2550db94535dSChris Mason 					sizeof(struct btrfs_item),
2551db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2552db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2553db94535dSChris Mason 					KM_USER1);
2554db94535dSChris Mason 		}
25555f39d397SChris Mason 		ioff = btrfs_item_offset(leaf, item);
25565f39d397SChris Mason 		btrfs_set_item_offset(leaf, item, ioff - data_size);
25576567e837SChris Mason 	}
25585f39d397SChris Mason 
2559db94535dSChris Mason 	if (leaf->map_token) {
2560db94535dSChris Mason 		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2561db94535dSChris Mason 		leaf->map_token = NULL;
2562db94535dSChris Mason 	}
2563db94535dSChris Mason 
25646567e837SChris Mason 	/* shift the data */
25655f39d397SChris Mason 	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
25666567e837SChris Mason 		      data_end - data_size, btrfs_leaf_data(leaf) +
25676567e837SChris Mason 		      data_end, old_data - data_end);
25685f39d397SChris Mason 
25696567e837SChris Mason 	data_end = old_data;
25705f39d397SChris Mason 	old_size = btrfs_item_size_nr(leaf, slot);
25715f39d397SChris Mason 	item = btrfs_item_nr(leaf, slot);
25725f39d397SChris Mason 	btrfs_set_item_size(leaf, item, old_size + data_size);
25735f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
25746567e837SChris Mason 
25756567e837SChris Mason 	ret = 0;
25765f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
25775f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
25786567e837SChris Mason 		BUG();
25795f39d397SChris Mason 	}
25806567e837SChris Mason 	return ret;
25816567e837SChris Mason }
25826567e837SChris Mason 
258374123bd7SChris Mason /*
258474123bd7SChris Mason  * Given a key and some data, insert an item into the tree.
258574123bd7SChris Mason  * This does all the path init required, making room in the tree if needed.
258674123bd7SChris Mason  */
25879c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
25885f39d397SChris Mason 			    struct btrfs_root *root,
25895f39d397SChris Mason 			    struct btrfs_path *path,
25909c58309dSChris Mason 			    struct btrfs_key *cpu_key, u32 *data_size,
25919c58309dSChris Mason 			    int nr)
2592be0e5c09SChris Mason {
25935f39d397SChris Mason 	struct extent_buffer *leaf;
25945f39d397SChris Mason 	struct btrfs_item *item;
2595aa5d6bedSChris Mason 	int ret = 0;
2596be0e5c09SChris Mason 	int slot;
2597eb60ceacSChris Mason 	int slot_orig;
25989c58309dSChris Mason 	int i;
25997518a238SChris Mason 	u32 nritems;
26009c58309dSChris Mason 	u32 total_size = 0;
26019c58309dSChris Mason 	u32 total_data = 0;
2602be0e5c09SChris Mason 	unsigned int data_end;
2603e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
2604e2fa7227SChris Mason 
26059c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26069c58309dSChris Mason 		total_data += data_size[i];
26079c58309dSChris Mason 	}
2608be0e5c09SChris Mason 
26099c58309dSChris Mason 	total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
26109c58309dSChris Mason 	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2611eb60ceacSChris Mason 	if (ret == 0) {
2612f0930a37SChris Mason 		return -EEXIST;
2613aa5d6bedSChris Mason 	}
2614ed2ff2cbSChris Mason 	if (ret < 0)
2615ed2ff2cbSChris Mason 		goto out;
2616be0e5c09SChris Mason 
261762e2749eSChris Mason 	slot_orig = path->slots[0];
26185f39d397SChris Mason 	leaf = path->nodes[0];
261974123bd7SChris Mason 
26205f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2621123abc88SChris Mason 	data_end = leaf_data_end(root, leaf);
2622eb60ceacSChris Mason 
2623123abc88SChris Mason 	if (btrfs_leaf_free_space(root, leaf) <
26249c58309dSChris Mason 	    sizeof(struct btrfs_item) + total_size) {
26253326d1b0SChris Mason 		btrfs_print_leaf(root, leaf);
26263326d1b0SChris Mason 		printk("not enough freespace need %u have %d\n",
26279c58309dSChris Mason 		       total_size, btrfs_leaf_free_space(root, leaf));
2628be0e5c09SChris Mason 		BUG();
2629d4dbff95SChris Mason 	}
26305f39d397SChris Mason 
263162e2749eSChris Mason 	slot = path->slots[0];
2632eb60ceacSChris Mason 	BUG_ON(slot < 0);
26335f39d397SChris Mason 
2634be0e5c09SChris Mason 	if (slot != nritems) {
2635be0e5c09SChris Mason 		int i;
26365f39d397SChris Mason 		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2637be0e5c09SChris Mason 
26385f39d397SChris Mason 		if (old_data < data_end) {
26395f39d397SChris Mason 			btrfs_print_leaf(root, leaf);
26405f39d397SChris Mason 			printk("slot %d old_data %d data_end %d\n",
26415f39d397SChris Mason 			       slot, old_data, data_end);
26425f39d397SChris Mason 			BUG_ON(1);
26435f39d397SChris Mason 		}
2644be0e5c09SChris Mason 		/*
2645be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2646be0e5c09SChris Mason 		 */
2647be0e5c09SChris Mason 		/* first correct the data pointers */
2648db94535dSChris Mason 		WARN_ON(leaf->map_token);
26490783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
26505f39d397SChris Mason 			u32 ioff;
2651db94535dSChris Mason 
26525f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2653db94535dSChris Mason 			if (!leaf->map_token) {
2654db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2655db94535dSChris Mason 					sizeof(struct btrfs_item),
2656db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2657db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2658db94535dSChris Mason 					KM_USER1);
2659db94535dSChris Mason 			}
2660db94535dSChris Mason 
26615f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
26629c58309dSChris Mason 			btrfs_set_item_offset(leaf, item, ioff - total_data);
26630783fcfcSChris Mason 		}
2664db94535dSChris Mason 		if (leaf->map_token) {
2665db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2666db94535dSChris Mason 			leaf->map_token = NULL;
2667db94535dSChris Mason 		}
2668be0e5c09SChris Mason 
2669be0e5c09SChris Mason 		/* shift the items */
26709c58309dSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
26715f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
26720783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
2673be0e5c09SChris Mason 
2674be0e5c09SChris Mason 		/* shift the data */
26755f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
26769c58309dSChris Mason 			      data_end - total_data, btrfs_leaf_data(leaf) +
2677be0e5c09SChris Mason 			      data_end, old_data - data_end);
2678be0e5c09SChris Mason 		data_end = old_data;
2679be0e5c09SChris Mason 	}
26805f39d397SChris Mason 
268162e2749eSChris Mason 	/* setup the item for the new data */
26829c58309dSChris Mason 	for (i = 0; i < nr; i++) {
26839c58309dSChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
26849c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
26859c58309dSChris Mason 		item = btrfs_item_nr(leaf, slot + i);
26869c58309dSChris Mason 		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
26879c58309dSChris Mason 		data_end -= data_size[i];
26889c58309dSChris Mason 		btrfs_set_item_size(leaf, item, data_size[i]);
26899c58309dSChris Mason 	}
26909c58309dSChris Mason 	btrfs_set_header_nritems(leaf, nritems + nr);
26915f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
2692aa5d6bedSChris Mason 
2693aa5d6bedSChris Mason 	ret = 0;
26945a01a2e3SChris Mason 	if (slot == 0) {
26955a01a2e3SChris Mason 		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2696e089f05cSChris Mason 		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
26975a01a2e3SChris Mason 	}
2698aa5d6bedSChris Mason 
26995f39d397SChris Mason 	if (btrfs_leaf_free_space(root, leaf) < 0) {
27005f39d397SChris Mason 		btrfs_print_leaf(root, leaf);
2701be0e5c09SChris Mason 		BUG();
27025f39d397SChris Mason 	}
2703ed2ff2cbSChris Mason out:
270462e2749eSChris Mason 	return ret;
270562e2749eSChris Mason }
270662e2749eSChris Mason 
270762e2749eSChris Mason /*
270862e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
270962e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
271062e2749eSChris Mason  */
2711e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2712e089f05cSChris Mason 		      *root, struct btrfs_key *cpu_key, void *data, u32
2713e089f05cSChris Mason 		      data_size)
271462e2749eSChris Mason {
271562e2749eSChris Mason 	int ret = 0;
27162c90e5d6SChris Mason 	struct btrfs_path *path;
27175f39d397SChris Mason 	struct extent_buffer *leaf;
27185f39d397SChris Mason 	unsigned long ptr;
271962e2749eSChris Mason 
27202c90e5d6SChris Mason 	path = btrfs_alloc_path();
27212c90e5d6SChris Mason 	BUG_ON(!path);
27222c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
272362e2749eSChris Mason 	if (!ret) {
27245f39d397SChris Mason 		leaf = path->nodes[0];
27255f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
27265f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
27275f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
272862e2749eSChris Mason 	}
27292c90e5d6SChris Mason 	btrfs_free_path(path);
2730aa5d6bedSChris Mason 	return ret;
2731be0e5c09SChris Mason }
2732be0e5c09SChris Mason 
273374123bd7SChris Mason /*
27345de08d7dSChris Mason  * delete the pointer from a given node.
273574123bd7SChris Mason  *
273674123bd7SChris Mason  * If the delete empties a node, the node is removed from the tree,
273774123bd7SChris Mason  * continuing all the way the root if required.  The root is converted into
273874123bd7SChris Mason  * a leaf if all the nodes are emptied.
273974123bd7SChris Mason  */
2740e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2741e089f05cSChris Mason 		   struct btrfs_path *path, int level, int slot)
2742be0e5c09SChris Mason {
27435f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
27447518a238SChris Mason 	u32 nritems;
2745aa5d6bedSChris Mason 	int ret = 0;
2746bb803951SChris Mason 	int wret;
2747be0e5c09SChris Mason 
27485f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
2749be0e5c09SChris Mason 	if (slot != nritems -1) {
27505f39d397SChris Mason 		memmove_extent_buffer(parent,
27515f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
27525f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
2753d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
2754d6025579SChris Mason 			      (nritems - slot - 1));
2755be0e5c09SChris Mason 	}
27567518a238SChris Mason 	nritems--;
27575f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
27587518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
27595f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
2760eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
27615f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
2762bb803951SChris Mason 	} else if (slot == 0) {
27635f39d397SChris Mason 		struct btrfs_disk_key disk_key;
27645f39d397SChris Mason 
27655f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
27665f39d397SChris Mason 		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
27670f70abe2SChris Mason 		if (wret)
27680f70abe2SChris Mason 			ret = wret;
2769be0e5c09SChris Mason 	}
2770d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
2771aa5d6bedSChris Mason 	return ret;
2772be0e5c09SChris Mason }
2773be0e5c09SChris Mason 
277474123bd7SChris Mason /*
277574123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
277674123bd7SChris Mason  * the leaf, remove it from the tree
277774123bd7SChris Mason  */
277885e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
277985e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
2780be0e5c09SChris Mason {
27815f39d397SChris Mason 	struct extent_buffer *leaf;
27825f39d397SChris Mason 	struct btrfs_item *item;
278385e21bacSChris Mason 	int last_off;
278485e21bacSChris Mason 	int dsize = 0;
2785aa5d6bedSChris Mason 	int ret = 0;
2786aa5d6bedSChris Mason 	int wret;
278785e21bacSChris Mason 	int i;
27887518a238SChris Mason 	u32 nritems;
2789be0e5c09SChris Mason 
27905f39d397SChris Mason 	leaf = path->nodes[0];
279185e21bacSChris Mason 	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);
279285e21bacSChris Mason 
279385e21bacSChris Mason 	for (i = 0; i < nr; i++)
279485e21bacSChris Mason 		dsize += btrfs_item_size_nr(leaf, slot + i);
279585e21bacSChris Mason 
27965f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
2797be0e5c09SChris Mason 
279885e21bacSChris Mason 	if (slot + nr != nritems) {
2799be0e5c09SChris Mason 		int i;
2800123abc88SChris Mason 		int data_end = leaf_data_end(root, leaf);
28015f39d397SChris Mason 
28025f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2803d6025579SChris Mason 			      data_end + dsize,
2804123abc88SChris Mason 			      btrfs_leaf_data(leaf) + data_end,
280585e21bacSChris Mason 			      last_off - data_end);
28065f39d397SChris Mason 
280785e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
28085f39d397SChris Mason 			u32 ioff;
2809db94535dSChris Mason 
28105f39d397SChris Mason 			item = btrfs_item_nr(leaf, i);
2811db94535dSChris Mason 			if (!leaf->map_token) {
2812db94535dSChris Mason 				map_extent_buffer(leaf, (unsigned long)item,
2813db94535dSChris Mason 					sizeof(struct btrfs_item),
2814db94535dSChris Mason 					&leaf->map_token, &leaf->kaddr,
2815db94535dSChris Mason 					&leaf->map_start, &leaf->map_len,
2816db94535dSChris Mason 					KM_USER1);
2817db94535dSChris Mason 			}
28185f39d397SChris Mason 			ioff = btrfs_item_offset(leaf, item);
28195f39d397SChris Mason 			btrfs_set_item_offset(leaf, item, ioff + dsize);
28200783fcfcSChris Mason 		}
2821db94535dSChris Mason 
2822db94535dSChris Mason 		if (leaf->map_token) {
2823db94535dSChris Mason 			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2824db94535dSChris Mason 			leaf->map_token = NULL;
2825db94535dSChris Mason 		}
2826db94535dSChris Mason 
28275f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
282885e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
28290783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
283085e21bacSChris Mason 			      (nritems - slot - nr));
2831be0e5c09SChris Mason 	}
283285e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
283385e21bacSChris Mason 	nritems -= nr;
28345f39d397SChris Mason 
283574123bd7SChris Mason 	/* delete the leaf if we've emptied it */
28367518a238SChris Mason 	if (nritems == 0) {
28375f39d397SChris Mason 		if (leaf == root->node) {
28385f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
28399a8dd150SChris Mason 		} else {
28407bb86316SChris Mason 			u64 root_gen = btrfs_header_generation(path->nodes[1]);
2841e089f05cSChris Mason 			wret = del_ptr(trans, root, path, 1, path->slots[1]);
2842aa5d6bedSChris Mason 			if (wret)
2843aa5d6bedSChris Mason 				ret = wret;
2844e089f05cSChris Mason 			wret = btrfs_free_extent(trans, root,
28457bb86316SChris Mason 					 leaf->start, leaf->len,
28467bb86316SChris Mason 					 btrfs_header_owner(path->nodes[1]),
28477bb86316SChris Mason 					 root_gen, 0, 0, 1);
28480f70abe2SChris Mason 			if (wret)
28490f70abe2SChris Mason 				ret = wret;
28509a8dd150SChris Mason 		}
2851be0e5c09SChris Mason 	} else {
28527518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
2853aa5d6bedSChris Mason 		if (slot == 0) {
28545f39d397SChris Mason 			struct btrfs_disk_key disk_key;
28555f39d397SChris Mason 
28565f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
2857e089f05cSChris Mason 			wret = fixup_low_keys(trans, root, path,
28585f39d397SChris Mason 					      &disk_key, 1);
2859aa5d6bedSChris Mason 			if (wret)
2860aa5d6bedSChris Mason 				ret = wret;
2861aa5d6bedSChris Mason 		}
2862aa5d6bedSChris Mason 
286374123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
286485e21bacSChris Mason 		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
2865be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
2866be0e5c09SChris Mason 			 * make sure the path still points to our leaf
2867be0e5c09SChris Mason 			 * for possible call to del_ptr below
2868be0e5c09SChris Mason 			 */
28694920c9acSChris Mason 			slot = path->slots[1];
28705f39d397SChris Mason 			extent_buffer_get(leaf);
28715f39d397SChris Mason 
287285e21bacSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1);
287354aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
2874aa5d6bedSChris Mason 				ret = wret;
28755f39d397SChris Mason 
28765f39d397SChris Mason 			if (path->nodes[0] == leaf &&
28775f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
287885e21bacSChris Mason 				wret = push_leaf_right(trans, root, path, 1, 1);
287954aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
2880aa5d6bedSChris Mason 					ret = wret;
2881aa5d6bedSChris Mason 			}
28825f39d397SChris Mason 
28835f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
28847bb86316SChris Mason 				u64 root_gen;
2885db94535dSChris Mason 				u64 bytenr = leaf->start;
2886db94535dSChris Mason 				u32 blocksize = leaf->len;
28875f39d397SChris Mason 
28887bb86316SChris Mason 				root_gen = btrfs_header_generation(
28897bb86316SChris Mason 							   path->nodes[1]);
28907bb86316SChris Mason 
2891e089f05cSChris Mason 				wret = del_ptr(trans, root, path, 1, slot);
2892aa5d6bedSChris Mason 				if (wret)
2893aa5d6bedSChris Mason 					ret = wret;
28945f39d397SChris Mason 
28955f39d397SChris Mason 				free_extent_buffer(leaf);
2896db94535dSChris Mason 				wret = btrfs_free_extent(trans, root, bytenr,
28977bb86316SChris Mason 					     blocksize,
28987bb86316SChris Mason 					     btrfs_header_owner(path->nodes[1]),
28997bb86316SChris Mason 					     root_gen, 0, 0, 1);
29000f70abe2SChris Mason 				if (wret)
29010f70abe2SChris Mason 					ret = wret;
29025de08d7dSChris Mason 			} else {
2903925baeddSChris Mason 				/* if we're still in the path, make sure
2904925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
2905925baeddSChris Mason 				 * push_leaf functions must have already
2906925baeddSChris Mason 				 * dirtied this buffer
2907925baeddSChris Mason 				 */
2908925baeddSChris Mason 				if (path->nodes[0] == leaf)
29095f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
29105f39d397SChris Mason 				free_extent_buffer(leaf);
2911be0e5c09SChris Mason 			}
2912d5719762SChris Mason 		} else {
29135f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
2914be0e5c09SChris Mason 		}
29159a8dd150SChris Mason 	}
2916aa5d6bedSChris Mason 	return ret;
29179a8dd150SChris Mason }
29189a8dd150SChris Mason 
291997571fd0SChris Mason /*
2920925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
29217bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
29227bb86316SChris Mason  * returns < 0 on io errors.
29237bb86316SChris Mason  */
29247bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
29257bb86316SChris Mason {
2926925baeddSChris Mason 	struct btrfs_key key;
2927925baeddSChris Mason 	struct btrfs_disk_key found_key;
2928925baeddSChris Mason 	int ret;
29297bb86316SChris Mason 
2930925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
2931925baeddSChris Mason 
2932925baeddSChris Mason 	if (key.offset > 0)
2933925baeddSChris Mason 		key.offset--;
2934925baeddSChris Mason 	else if (key.type > 0)
2935925baeddSChris Mason 		key.type--;
2936925baeddSChris Mason 	else if (key.objectid > 0)
2937925baeddSChris Mason 		key.objectid--;
2938925baeddSChris Mason 	else
29397bb86316SChris Mason 		return 1;
29407bb86316SChris Mason 
2941925baeddSChris Mason 	btrfs_release_path(root, path);
2942925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2943925baeddSChris Mason 	if (ret < 0)
2944925baeddSChris Mason 		return ret;
2945925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
2946925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
2947925baeddSChris Mason 	if (ret < 0)
29487bb86316SChris Mason 		return 0;
2949925baeddSChris Mason 	return 1;
29507bb86316SChris Mason }
29517bb86316SChris Mason 
29527bb86316SChris Mason /*
2953925baeddSChris Mason  * search the tree again to find a leaf with greater keys
29540f70abe2SChris Mason  * returns 0 if it found something or 1 if there are no greater leaves.
29550f70abe2SChris Mason  * returns < 0 on io errors.
295697571fd0SChris Mason  */
2957234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2958d97e63b6SChris Mason {
2959d97e63b6SChris Mason 	int slot;
2960d97e63b6SChris Mason 	int level = 1;
29615f39d397SChris Mason 	struct extent_buffer *c;
29625f39d397SChris Mason 	struct extent_buffer *next = NULL;
2963925baeddSChris Mason 	struct btrfs_key key;
2964925baeddSChris Mason 	u32 nritems;
2965925baeddSChris Mason 	int ret;
2966925baeddSChris Mason 
2967925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
2968925baeddSChris Mason 	if (nritems == 0) {
2969925baeddSChris Mason 		return 1;
2970925baeddSChris Mason 	}
2971925baeddSChris Mason 
2972925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
2973925baeddSChris Mason 
2974925baeddSChris Mason 	btrfs_release_path(root, path);
2975*a2135011SChris Mason 	path->keep_locks = 1;
2976925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2977925baeddSChris Mason 	path->keep_locks = 0;
2978925baeddSChris Mason 
2979925baeddSChris Mason 	if (ret < 0)
2980925baeddSChris Mason 		return ret;
2981925baeddSChris Mason 
2982*a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
2983*a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
2984925baeddSChris Mason 		goto done;
2985925baeddSChris Mason 	}
2986d97e63b6SChris Mason 
2987234b63a0SChris Mason 	while(level < BTRFS_MAX_LEVEL) {
2988d97e63b6SChris Mason 		if (!path->nodes[level])
29890f70abe2SChris Mason 			return 1;
29905f39d397SChris Mason 
2991d97e63b6SChris Mason 		slot = path->slots[level] + 1;
2992d97e63b6SChris Mason 		c = path->nodes[level];
29935f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
2994d97e63b6SChris Mason 			level++;
2995925baeddSChris Mason 			if (level == BTRFS_MAX_LEVEL) {
29967bb86316SChris Mason 				return 1;
2997925baeddSChris Mason 			}
2998d97e63b6SChris Mason 			continue;
2999d97e63b6SChris Mason 		}
30005f39d397SChris Mason 
3001925baeddSChris Mason 		if (next) {
3002925baeddSChris Mason 			btrfs_tree_unlock(next);
30035f39d397SChris Mason 			free_extent_buffer(next);
3004925baeddSChris Mason 		}
30055f39d397SChris Mason 
3006925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
300701f46658SChris Mason 			reada_for_search(root, path, level, slot, 0);
30085f39d397SChris Mason 
3009ca7a79adSChris Mason 		next = read_node_slot(root, c, slot);
3010*a2135011SChris Mason 		if (!path->skip_locking) {
3011*a2135011SChris Mason 			if (!btrfs_tree_locked(c)) {
3012*a2135011SChris Mason 				int i;
3013*a2135011SChris Mason 				WARN_ON(1);
3014*a2135011SChris Mason printk("path %p no lock on level %d\n", path, level);
3015*a2135011SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
3016*a2135011SChris Mason printk("path %p level %d slot %d nritems %d\n", path, i, path->slots[i], btrfs_header_nritems(path->nodes[i]));
3017*a2135011SChris Mason }
3018*a2135011SChris Mason 			}
3019925baeddSChris Mason 			btrfs_tree_lock(next);
3020*a2135011SChris Mason 		}
3021d97e63b6SChris Mason 		break;
3022d97e63b6SChris Mason 	}
3023d97e63b6SChris Mason 	path->slots[level] = slot;
3024d97e63b6SChris Mason 	while(1) {
3025d97e63b6SChris Mason 		level--;
3026d97e63b6SChris Mason 		c = path->nodes[level];
3027925baeddSChris Mason 		if (path->locks[level])
3028925baeddSChris Mason 			btrfs_tree_unlock(c);
30295f39d397SChris Mason 		free_extent_buffer(c);
3030d97e63b6SChris Mason 		path->nodes[level] = next;
3031d97e63b6SChris Mason 		path->slots[level] = 0;
3032925baeddSChris Mason 		path->locks[level] = 1;
3033d97e63b6SChris Mason 		if (!level)
3034d97e63b6SChris Mason 			break;
3035925baeddSChris Mason 		if (level == 1 && path->locks[1] && path->reada)
3036925baeddSChris Mason 			reada_for_search(root, path, level, slot, 0);
3037ca7a79adSChris Mason 		next = read_node_slot(root, next, 0);
3038*a2135011SChris Mason 		if (!path->skip_locking) {
3039*a2135011SChris Mason 			WARN_ON(!btrfs_tree_locked(path->nodes[level]));
3040925baeddSChris Mason 			btrfs_tree_lock(next);
3041d97e63b6SChris Mason 		}
3042*a2135011SChris Mason 	}
3043925baeddSChris Mason done:
3044925baeddSChris Mason 	unlock_up(path, 0, 1);
3045d97e63b6SChris Mason 	return 0;
3046d97e63b6SChris Mason }
30470b86a832SChris Mason 
30480b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
30490b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
30500b86a832SChris Mason 			int type)
30510b86a832SChris Mason {
30520b86a832SChris Mason 	struct btrfs_key found_key;
30530b86a832SChris Mason 	struct extent_buffer *leaf;
30540b86a832SChris Mason 	int ret;
30550b86a832SChris Mason 
30560b86a832SChris Mason 	while(1) {
30570b86a832SChris Mason 		if (path->slots[0] == 0) {
30580b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
30590b86a832SChris Mason 			if (ret != 0)
30600b86a832SChris Mason 				return ret;
30610b86a832SChris Mason 		} else {
30620b86a832SChris Mason 			path->slots[0]--;
30630b86a832SChris Mason 		}
30640b86a832SChris Mason 		leaf = path->nodes[0];
30650b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
30660b86a832SChris Mason 		if (found_key.type == type)
30670b86a832SChris Mason 			return 0;
30680b86a832SChris Mason 	}
30690b86a832SChris Mason 	return 1;
30700b86a832SChris Mason }
30710b86a832SChris Mason 
3072